#TvRageIRCListener.java#

/*
 * This file is part of DrFTPD, Distributed FTP Daemon.
 *
 * DrFTPD is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * DrFTPD is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with DrFTPD; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package net.lhmods.plugins.irc.tvrage;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Properties;

import net.sf.drftpd.util.ReplacerUtils;

import org.apache.log4j.Logger;
import org.drftpd.GlobalContext;
import org.drftpd.plugins.SiteBot;
import org.drftpd.sitebot.IRCCommand;
import org.tanesha.replacer.ReplacerEnvironment;

import f00f.net.irc.martyr.commands.MessageCommand;

/**
 * @author Teflon
 */
public class TvRageIRCListener extends IRCCommand {
    private static final Logger logger = Logger.getLogger(TvRageIRCListener.class);
    
    public TvRageIRCListener(GlobalContext gctx) {
	super(gctx); 
    }

    public ArrayList<String> doTvRageInfo(String args, MessageCommand msgc) {
        ArrayList<String> out = new ArrayList<String>();
        ReplacerEnvironment env = new ReplacerEnvironment(SiteBot.GLOBAL_ENV);
	env.add("ircnick", msgc.getSource().getNick());	
        env.add("searchstr", args);

        if (args.equals("")) {
             out.add(ReplacerUtils.jprintf("tvrage.syntax",env,TvRageParser.class));
             return out;
        }
        
        TvRageParser tvrage = new TvRageParser("INFO",args);
        if (!tvrage.foundShow()) {
            out.add(ReplacerUtils.jprintf("tvrage.notfound", env, TvRageParser.class));
        } else {
            out.add(ReplacerUtils.jprintf("tvrage.info", tvrage.getInfoEnv(), TvRageParser.class));
        }
        return out;
    }

    public ArrayList<String> doTvRageToday(String args, MessageCommand msgc) {
        ArrayList<String> out = new ArrayList<String>();
        ReplacerEnvironment env = new ReplacerEnvironment(SiteBot.GLOBAL_ENV);
	env.add("ircnick", msgc.getSource().getNick());	
        env.add("countrycode", args);
        TvRageParser tvrage = new TvRageParser("TODAY",args);
        out.add(ReplacerUtils.jprintf("tvrage.today", tvrage.getTodayEnv(), TvRageParser.class));
        return out;
    }

    public ArrayList<String> doTvRageTomorrow(String args, MessageCommand msgc) {
        ArrayList<String> out = new ArrayList<String>();
        ReplacerEnvironment env = new ReplacerEnvironment(SiteBot.GLOBAL_ENV);
	env.add("ircnick", msgc.getSource().getNick());	
        env.add("countrycode", args);
        TvRageParser tvrage = new TvRageParser("TOMORROW",args);
        out.add(ReplacerUtils.jprintf("tvrage.tomorrow", tvrage.getTomorrowEnv(), TvRageParser.class));
        return out;
    }
}

# TvRageParser.java #

/*
 * This file is part of DrFTPD, Distributed FTP Daemon.
 *
 * DrFTPD is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * DrFTPD is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with DrFTPD; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package net.lhmods.plugins.irc.tvrage;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.log4j.Logger;
import org.drftpd.plugins.SiteBot;
import org.tanesha.replacer.ReplacerEnvironment;

/**
 * @author Teflon
 */
public class TvRageParser {
    private static final Logger logger = Logger.getLogger(TvRageParser.class); 
    
    private static final String _searchUrl = "http://www.tvrage.com/quickinfo.php?show=";
    private static final String _scheduleUrl = "http://www.tvrage.com/quickschedule.php?country=";
    
    // For INFO
    private boolean _foundShow;
    private String _title;
    private String _url;
    private String _premiered;
    private String _lastep_episode;
    private String _lastep_title;
    private String _lastep_date;
    private String _nextep_episode;
    private String _nextep_title;
    private String _nextep_date;
    private String _country;
    private String _status;
    private String _classification;
    private String _genres;
    private String _network;
    private String _airtime;    
    
    //For Today
    private String _todaydate;
    private String _todaydata;
    
    //For Tomorrow
    private String _tomorrowdate;
    private String _tomorrowdata;
    
    public TvRageParser(String selection, String args) {
	if (selection.equals("INFO"))
	    _foundShow = getInfo(args);
	else if (selection.equals("TODAY"))
	    _foundShow = getToday(args);
	else
	    _foundShow = getTomorrow(args);
    }

    // For INFO
    public String getTitle()          { return foundShow() ? _title          : "N/A"; }
    public String getURL()            { return foundShow() ? _url            : "N/A"; }
    public String getPremiered()      { return foundShow() ? _premiered      : "N/A"; }
    public String getLastepEpisode()  { return foundShow() ? _lastep_episode : "N/A"; }
    public String getLastepTitle()    { return foundShow() ? _lastep_title   : "N/A"; }
    public String getLastepDate()     { return foundShow() ? _lastep_date    : "N/A"; }
    public String getNextepEpisode()  { return foundShow() ? _nextep_episode : "N/A"; }
    public String getNextepTitle()    { return foundShow() ? _nextep_title   : "N/A"; }
    public String getNextepDate()     { return foundShow() ? _nextep_date    : "N/A"; }
    public String getCountry()        { return foundShow() ? _country        : "N/A"; }
    public String getStatus()         { return foundShow() ? _status         : "N/A"; }
    public String getClassification() { return foundShow() ? _classification : "N/A"; }
    public String getGenres()         { return foundShow() ? _genres         : "N/A"; }
    public String getNetwork()        { return foundShow() ? _network        : "N/A"; }
    public String getAirtime()        { return foundShow() ? _airtime        : "N/A"; }
    public boolean foundShow()        { return _foundShow; }

    // For TODAY
    public String getTodayDate()      { return foundShow() ? _todaydate      : "N/A"; }
    public String getTodayData()      { return foundShow() ? _todaydata      : "N/A"; }

    // For TOMORROW
    public String getTomorrowDate()   { return foundShow() ? _tomorrowdate   : "N/A"; }
    public String getTomorrowData()   { return foundShow() ? _tomorrowdata   : "N/A"; }
    
    private boolean getInfo(String searchString) {
        try {
            String urlString = _searchUrl + searchString;
            URL url = new URL(urlString);
            URLConnection urlConn = url.openConnection();
            
            if (!(urlConn.getContent() instanceof InputStream))
                return false;

            boolean redirect = false;
            String data = "";
            BufferedReader in = null;
            String line;
            try {
				in = new BufferedReader(new InputStreamReader(urlConn
						.getInputStream()));
				while ((line = in.readLine()) != null) {
					data += line + "::";
				}
			} finally {
				if (in != null) {
					in.close();
				}
			}

            if (data.indexOf("No Show Results Were Found For") > -1)
                return false;
            
            String[] dataArray = data.split("::");

            _title = getValFromArray(dataArray, "Show Name");
            _url = getValFromArray(dataArray, "Show URL");
            _premiered = getValFromArray(dataArray, "Premiered");
            _lastep_episode = getValFromArray(dataArray, "Latest Episode", 0);
            _lastep_title = getValFromArray(dataArray, "Latest Episode", 1);
            _lastep_date = getValFromArray(dataArray, "Latest Episode", 2);
            _nextep_episode = getValFromArray(dataArray, "Next Episode", 0);
            _nextep_title = getValFromArray(dataArray, "Next Episode", 1);
            _nextep_date = getValFromArray(dataArray, "Next Episode", 2);
            _country = getValFromArray(dataArray, "Country");
            _status = getValFromArray(dataArray, "Status");
            _classification = getValFromArray(dataArray, "Classification");
            _genres = getValFromArray(dataArray, "Genres");
            _network = getValFromArray(dataArray, "Network");
            _airtime = getValFromArray(dataArray, "Airtime"); 
  
        } catch (MalformedURLException e) {
            e.printStackTrace();
            return false;
        } catch (IOException e) {
            e.printStackTrace();
            return false;
        }
 
        return true;
    }

    private boolean getToday(String countrycode) {
        try {
	    String urlString = _scheduleUrl;
	    if (countrycode.toLowerCase().equals("us") || 
		countrycode.toLowerCase().equals("gb") || 
		countrycode.toLowerCase().equals("ca") || 
		countrycode.toLowerCase().equals("au"))
            	urlString += countrycode;
	    else
		urlString += "US";
            URL url = new URL(urlString);
            URLConnection urlConn = url.openConnection();
            
            if (!(urlConn.getContent() instanceof InputStream))
                return false;

            boolean redirect = false;
            String data = "";
            BufferedReader in = null;
            String line;
            try {
				in = new BufferedReader(new InputStreamReader(urlConn
						.getInputStream()));
				while ((line = in.readLine()) != null) {
					data += line + "::";
				}
			} finally {
				if (in != null) {
					in.close();
				}
			}
	
	    data = data.substring(0,data.indexOf("::::::"));
            String[] dataArray = data.split("::");
            _todaydate = dataArray[0].replaceAll("\\[/?DAY\\]","");
            _todaydata = getSchedule(dataArray);
	    System.out.println("DATE: " + _todaydata);
  
        } catch (MalformedURLException e) {
            e.printStackTrace();
	    return false;
        } catch (IOException e) {
            e.printStackTrace();
	    return false;
        }
	return true;
    }

    private boolean getTomorrow(String countrycode) {
        try {
	    String urlString = _scheduleUrl;
	    if (countrycode.toLowerCase().equals("us") || 
		countrycode.toLowerCase().equals("gb") || 
		countrycode.toLowerCase().equals("ca") || 
		countrycode.toLowerCase().equals("au"))
            	urlString += countrycode;
	    else
		urlString += "US";
            URL url = new URL(urlString);
            URLConnection urlConn = url.openConnection();
            
            if (!(urlConn.getContent() instanceof InputStream))
                return false;

            boolean redirect = false;
            String data = "";
            BufferedReader in = null;
            String line;
            try {
				in = new BufferedReader(new InputStreamReader(urlConn
						.getInputStream()));
				while ((line = in.readLine()) != null) {
					data += line + "::";
				}
			} finally {
				if (in != null) {
					in.close();
				}
			}

	    int startParse = data.indexOf("::::::") + 6;
	    int endParse = startParse - data.indexOf("::::::",startParse);
	    if (endParse > 0 && startParse + endParse < data.length()) {
	    	data = data.substring(startParse,endParse);
            	String[] dataArray = data.split("::");
            	_todaydate = dataArray[0].replaceAll("\\[/?DAY\\]","");
            	_todaydata = getSchedule(dataArray);
	    } else 
            	return false;

        } catch (MalformedURLException e) {
            e.printStackTrace();
	    return false;
        } catch (IOException e) {
            e.printStackTrace();
	    return false;
        }
	return true;
    }
    
    public ReplacerEnvironment getInfoEnv() {
        ReplacerEnvironment env = new ReplacerEnvironment(SiteBot.GLOBAL_ENV);
        env.add("title", getTitle());
        env.add("url", getURL());
        env.add("premiered", getPremiered());
        env.add("lastepnr", getLastepEpisode());
        env.add("lasteptitle", getLastepTitle());
        env.add("lastepdate", getLastepDate());
        env.add("nextepnr", getNextepEpisode());
        env.add("nexteptitle", getNextepTitle());
        env.add("nextepdate", getNextepDate());
        env.add("country", getCountry());
        env.add("status", getStatus());
        env.add("classification", getClassification());
        env.add("genres", getGenres());
        env.add("network", getNetwork());
        env.add("airtime", getAirtime());
        return env;
    }

    public ReplacerEnvironment getTodayEnv() {
        ReplacerEnvironment env = new ReplacerEnvironment(SiteBot.GLOBAL_ENV);
        env.add("date", getTodayDate());
        env.add("data", getTodayData());
        return env;
    }

    public ReplacerEnvironment getTomorrowEnv() {
        ReplacerEnvironment env = new ReplacerEnvironment(SiteBot.GLOBAL_ENV);
        env.add("date", getTomorrowDate());
        env.add("data", getTomorrowData());
        return env;
    }
    
    private String getValFromArray(String[] dataArray, String field) {
	for (int i = 0; i < dataArray.length; i++) {
	    String[] dataArray2 = dataArray[i].split("@");
	    if (dataArray2.length > 0)
	    	if (dataArray2[0].equals(field))
		    return dataArray2[1];
        }        
        return "N/A";
    }

    private String getValFromArray(String[] dataArray, String field, int value) {
	for (int i = 0; i < dataArray.length; i++) {
	    String[] dataArray2 = dataArray[i].split("@");
	    if (dataArray2.length > 0) {
	    	if (dataArray2[0].equals(field)) {
		    String[] tmpDataArray = dataArray2[1].split("\\^");
		    if (value-1 < tmpDataArray.length)
		    	return tmpDataArray[value];
	        }
	    }
        }      
        return "N/A";
    }

    private String getSchedule(String[] dataArray) {
	String schedulestring = "";
	for (int i = 1; i < dataArray.length; i++) {
	    if (dataArray[i].indexOf("[TIME]") > -1)
		schedulestring += "[" + dataArray[i].replaceAll("\\[/?TIME\\]","") + "]";
	    else
		schedulestring += " - " + dataArray[i].replaceAll("\\[/?SHOW\\]","");
        }        
        return schedulestring;
    }
    
}

# TvRageParser.properties #

#!tv tags: title, url, premiered, lastepnr, lasteptitle, lastepdate, nextepnr, nexteptitle,
#      nextepdate, country, status, classification, genres, network, airtime

#!tv trigger
tvrage.info=${color}15${bold}TvRage${bold}${coloroff} -> ${bold}${title}${bold} - ${underline}${url}${underline} \n Premiered: ${bold}${premiered}${bold} - Country: ${bold}${country}${bold} - Genre: ${bold}${genres}${bold} - Airing: ${bold}${airtime}${bold} on ${bold}${network}${bold} \n Last Episode: ${bold}${lasteptitle}${bold} - ${bold}${lastepnr}${bold} - ${bold}${lastepdate}${bold} \n Next Episode: ${bold}${nexteptitle}${bold} - ${bold}${nextepnr}${bold} - ${bold}${nextepdate}${bold}

#(!tv)TvRage info not found
tvrage.notfound=${color}15${bold}TvRage${bold}${coloroff} -> No info found for ${bold}${searchstr}${bold}

#(!tv)args missing
tvrage.syntax=${color}4${bold}TvRage${bold}${coloroff} -> SYNTAX: !tv <show>

#!today trigger
tvrage.today=${color}15${bold}TvRage${bold}${coloroff} -> ${bold}${date}${bold} \n ${data}

#!tomorrow trigger
tvrage.today=${color}15${bold}TvRage${bold}${coloroff} -> ${bold}${date}${bold} \n ${data}