From 966a203ff8409972aefe3dd7d243d36adaf21fce Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?John=20Ankarstr=C3=B6m?= <john@ankarstrom.se>
Date: Thu, 7 Apr 2022 00:22:37 +0200
Subject: Add "View TV Originals" option.

---
 pl/cfg.pl          | 7 +++++++
 pl/episode_data.pl | 5 ++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

(limited to 'pl')

diff --git a/pl/cfg.pl b/pl/cfg.pl
index 90b1e94..c2eae86 100644
--- a/pl/cfg.pl
+++ b/pl/cfg.pl
@@ -38,3 +38,10 @@ get_view_watched(V) :-
 	    integer(V), !
 	;   V = 1
 	).
+
+set_view_tv_original(V) :- set_key('ViewTVOriginal', V).
+get_view_tv_original(V) :-
+	(   get_key('ViewTVOriginal', V),
+	    integer(V), !
+	;   V = 1
+	).
diff --git a/pl/episode_data.pl b/pl/episode_data.pl
index e944243..557cddf 100644
--- a/pl/episode_data.pl
+++ b/pl/episode_data.pl
@@ -2,7 +2,8 @@
 			 retract_episode/1,
 			 episode_count/1,
 			 rate_episode/2,
-			 episode_rating/2]).
+			 episode_rating/2,
+			 tv_original/1]).
 
 :- use_module(library(clpfd)).
 :- use_module(library(dcg/basics)).
@@ -54,6 +55,8 @@ rate_episode(Ep, R) :-
 	;   assert_episode_rating(Ep, R)
 	).
 
+tv_original(Ep) :- episode_datum(Ep, 'Source', 'TV Original').
+
 % Remote data retrieval.
 
 episode_number(Ep) --> integer(Ep).
-- 
cgit v1.2.3