+/*
+ * Copyright 2005 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.wamblee.crawler.kiss;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.InputStream;
+import java.io.PrintStream;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import org.apache.commons.httpclient.HttpClient;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.dom4j.Element;
+import org.wamblee.conditions.Condition;
+import org.wamblee.conditions.OrCondition;
+import org.wamblee.crawler.Action;
+import org.wamblee.crawler.Configuration;
+import org.wamblee.crawler.Crawler;
+import org.wamblee.crawler.Page;
+import org.wamblee.crawler.impl.ConfigurationParser;
+import org.wamblee.crawler.impl.CrawlerImpl;
+
+/**
+ *
+ */
+public class KissCrawler {
+
+ private static final Log LOG = LogFactory.getLog(KissCrawler.class);
+
+ private static final String LOG_FILE = "kiss.log";
+
+ private static final String START_URL = "http://epg.kml.kiss-technology.com/login_core.php";
+
+ private static final String CRAWLER_CONFIG = "config.xml";
+
+ private static final String PROGRAM_CONFIG = "programs.xml";
+
+ private static final String TIME_REGEX = "([0-9]{2}):([0-9]{2})[^0-9]*([0-9]{2}):([0-9]{2}).*";
+
+ private Pattern _pattern;
+
+ public KissCrawler(String aStartUrl, String aCrawlerConfig, String aProgramConfig) throws Exception {
+
+ _pattern = Pattern.compile(TIME_REGEX);
+
+ FileOutputStream fos = new FileOutputStream(new File(LOG_FILE));
+ PrintStream os = new PrintStream(fos);
+
+ try {
+ ConfigurationParser parser = new ConfigurationParser(os);
+ InputStream crawlerConfigFile = new FileInputStream(new File(aCrawlerConfig));
+ Configuration config = parser.parse(crawlerConfigFile);
+
+ InputStream programConfigFile = new FileInputStream(new File(aProgramConfig));
+ Condition<Program> programCondition = new ProgramConfigurationParser().parse(programConfigFile);
+
+
+ HttpClient client = new HttpClient();
+ // client.getHostConfiguration().setProxy("localhost", 3128);
+
+ Crawler crawler = new CrawlerImpl(client, config);
+
+ Page page = crawler.getPage(aStartUrl);
+ showPage(page);
+ page = page.getAction("channels-favorites").execute();
+ TVGuide guide = createGuide(page);
+ PrintVisitor printer = new PrintVisitor(System.out);
+ guide.accept(printer);
+
+ MatchVisitor matcher = new MatchVisitor(programCondition);
+ guide.accept(matcher);
+ List<Program> programs = matcher.getMatches();
+ for (Program program: programs) {
+ System.out.println("Found: " + program + " record: " + program.record() );
+ }
+
+ } finally {
+ os.flush();
+ os.close();
+ System.out.println("Output written on '" + LOG_FILE + "'");
+ }
+ }
+
+ public static void main(String[] args) throws Exception {
+ new KissCrawler(START_URL, CRAWLER_CONFIG, PROGRAM_CONFIG);
+ }
+
+ private void showPage(Page aPage) {
+ Action[] links = aPage.getActions();
+ for (Action link : links) {
+ System.out.println("Link found '" + link.getName() + "'");
+ }
+ Element element = aPage.getContent();
+ System.out.println("Retrieved content: " + element.asXML());
+ }
+
+ private TVGuide createGuide(Page page) {
+ LOG.info("Obtaining full TV guide");
+ Action[] actions = page.getActions();
+ List<Channel> channels = new ArrayList<Channel>();
+ for (Action action : actions) {
+ Channel channel = createChannel(action.getName(), action.execute()
+ .getAction("right-now").execute());
+ channels.add(channel);
+ }
+ return new TVGuide(channels);
+ }
+
+ private Channel createChannel(String aChannel, Page aPage) {
+ LOG.info("Obtaining program for " + aChannel);
+ Action[] programActions = aPage.getActions();
+ List<Program> programs = new ArrayList<Program>();
+ for (Action action : programActions) {
+ String time = action.getContent().element("time").getText().trim();
+ Matcher matcher = _pattern.matcher(time);
+ if (matcher.matches()) {
+ Time begin = new Time(Integer.parseInt(matcher.group(1)),
+ Integer.parseInt(matcher.group(2)));
+ Time end = new Time(Integer.parseInt(matcher.group(3)),
+ Integer.parseInt(matcher.group(4)));
+ TimeInterval interval = new TimeInterval(begin, end);
+ //Page programInfo = action.execute();
+ //String description = programInfo.getContent().element("description").getText().trim();
+ //String keywords = programInfo.getContent().element("keywords").getText().trim();
+ String description = "";
+ String keywords = "";
+ Program program = new Program(aChannel, action.getName(), description, keywords, interval, action);
+
+ LOG.debug("Got program " + program);
+ programs.add(program);
+ }
+ }
+ return new Channel(aChannel, programs);
+ }
+}