From 7e0912f8c8e0198877891e3ac5095d2dc851d718 Mon Sep 17 00:00:00 2001
From: Jan-Pascal van Best <janpascal@vanbest.org>
Date: Wed, 2 May 2012 16:05:09 +0200
Subject: [PATCH] Implement kijkwijzer for RTL source. Fixes #1

---
 .../org/vanbest/xmltv/AbstractEPGSource.java  | 26 +++++++++++++++++++
 src/main/java/org/vanbest/xmltv/RTL.java      | 19 +++++++++++++-
 src/main/java/org/vanbest/xmltv/TvGids.java   | 24 -----------------
 3 files changed, 44 insertions(+), 25 deletions(-)

diff --git a/src/main/java/org/vanbest/xmltv/AbstractEPGSource.java b/src/main/java/org/vanbest/xmltv/AbstractEPGSource.java
index 31eb930..3d59f8a 100644
--- a/src/main/java/org/vanbest/xmltv/AbstractEPGSource.java
+++ b/src/main/java/org/vanbest/xmltv/AbstractEPGSource.java
@@ -80,4 +80,30 @@ public abstract class AbstractEPGSource implements EPGSource {
 	public void clearCache() {
 		cache.clear(sourceId);
 	}
+
+	List<String> parseKijkwijzer(String s) {
+		List<String> result = new ArrayList<String>();
+		for (int i=0; i<s.length(); i++) {
+			char c = s.charAt(i);
+			switch(c) {
+			case 'a':result.add("Angst"); break;
+			case 'd':result.add("Discriminatie"); break;
+			case 's':result.add("Seks"); break;
+			case 'h':result.add("Drugs/Alcohol"); break;
+			case 'g':result.add("Geweld"); break;
+			case 't':result.add("Grof taalgebruik"); break;
+			case '1':result.add("Voor alle leeftijden"); break;
+			case '2':result.add("Afgeraden voor kinderen jonger dan 6 jaar"); break;
+			case '9':result.add("Afgeraden voor kinderen jonger dan 9 jaar"); break;
+			case '3':result.add("Afgeraden voor kinderen jonger dan 12 jaar"); break;
+			case '4':result.add("Afgeraden voor kinderen jonger dan 16 jaar"); break;
+			case '5': break; // Lijkt op een foutje van RTL, bedoeld wordt wrsch "12 jaar en ouder". Wordt op RTL tvgids niet weegegeven.
+			default: if (!config.quiet) {
+				logger.warn("Unknown kijkwijzer character: " + c + " in string " + s);
+				}
+			}
+		}
+		return result;
+	}
+
 }
\ No newline at end of file
diff --git a/src/main/java/org/vanbest/xmltv/RTL.java b/src/main/java/org/vanbest/xmltv/RTL.java
index b8e7a95..da2384f 100644
--- a/src/main/java/org/vanbest/xmltv/RTL.java
+++ b/src/main/java/org/vanbest/xmltv/RTL.java
@@ -50,6 +50,8 @@ import net.sf.json.JSONArray;
 import net.sf.json.JSONObject;
 
 import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang.StringUtils;
+import org.apache.log4j.Level;
 import org.apache.log4j.Logger;
 import org.vanbest.xmltv.EPGSource.Stats;
 import org.w3c.dom.DOMException;
@@ -281,7 +283,20 @@ public class RTL extends AbstractEPGSource implements EPGSource  {
 			prog.endTime = parseTime(e.getTextContent(), dateStatus, DateStatus.END_TIME);
 		} else if (tag.equals("omroep")) {
 		} else if (tag.equals("kijkwijzer")) {
-			//System.out.println("Kijkwijzer: \"" + e.getTextContent() + "\"");
+			logger.trace(prog.toString());
+			logger.trace("Kijkwijzer: \"" + e.getTextContent() + "\"");
+			String kijkwijzer = e.getTextContent();
+			List<String> list = parseKijkwijzer(kijkwijzer);
+			if (config.joinKijkwijzerRatings) {
+				// mythtv doesn't understand multiple <rating> tags
+				prog.addRating("kijkwijzer", StringUtils.join(list, ","));
+			} else {
+				for(String rating: list) {
+					prog.addRating("kijkwijzer", rating);
+				}
+			}
+			logger.trace("Kijkwijzer: \"" + StringUtils.join(list, ",") + "\"");
+			// TODO add kijkwijzer icons?
 		} else if (tag.equals("presentatie")) {
 			// A; A en B; A, B, C en D
 			String[] presentatoren = e.getTextContent().split(", | en ");
@@ -421,6 +436,8 @@ public class RTL extends AbstractEPGSource implements EPGSource  {
 	 */
 	public static void main(String[] args) throws FileNotFoundException {
 		debug = true;
+        Logger.getRootLogger().setLevel(Level.TRACE);
+
 		Config config = Config.getDefaultConfig();
 		config.niceMilliseconds = 50;
 		RTL rtl = new RTL(2, config);
diff --git a/src/main/java/org/vanbest/xmltv/TvGids.java b/src/main/java/org/vanbest/xmltv/TvGids.java
index 04f127c..fa4f73c 100644
--- a/src/main/java/org/vanbest/xmltv/TvGids.java
+++ b/src/main/java/org/vanbest/xmltv/TvGids.java
@@ -107,30 +107,6 @@ public class TvGids extends AbstractEPGSource implements EPGSource {
 		return new URL(s.toString());
 	}
 
-	List<String> parseKijkwijzer(String s) {
-		List<String> result = new ArrayList<String>();
-		for (int i=0; i<s.length(); i++) {
-			char c = s.charAt(i);
-			switch(c) {
-			case 'a':result.add("Angst"); break;
-			case 'd':result.add("Discriminatie"); break;
-			case 's':result.add("Seks"); break;
-			case 'h':result.add("Drugs/Alcohol"); break;
-			case 'g':result.add("Geweld"); break;
-			case 't':result.add("Grof taalgebruik"); break;
-			case '1':result.add("Voor alle leeftijden"); break;
-			case '2':result.add("Afgeraden voor kinderen jonger dan 6 jaar"); break;
-			case '9':result.add("Afgeraden voor kinderen jonger dan 9 jaar"); break;
-			case '3':result.add("Afgeraden voor kinderen jonger dan 12 jaar"); break;
-			case '4':result.add("Afgeraden voor kinderen jonger dan 16 jaar"); break;
-			default: if (!config.quiet) {
-				logger.warn("Unknown kijkwijzer character: " + c + " in string " + s);
-				}
-			}
-		}
-		return result;
-	}
-
 	/* (non-Javadoc)
 	 * @see org.vanbest.xmltv.EPGSource#getChannels()
 	 */
-- 
2.39.5