#!/usr/bin/perl
# tv2html.pl - Converts sorted XMLTV files to HTML
# Copyright (C) 2002-2003  Toby A Inkster
#    
# This program 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.
#
# This program 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 this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

# input must be a sorted XMLTV file.
# output is an HTML TV guide.

use XMLTV;

$debuglevel = 0;

&read_options;
&read_goods;

# Find input file name
$filename = '-' if (@ARGV == 0);
$filename = $ARGV[0] if (@ARGV == 1);
usage() if (@ARGV > 1);

# This bit is shamelessly stolen from tv_sort -- obtain list of programs
($encoding, $credits, $channels, $progs) = @{XMLTV::parsefile($filename)};
@progs = @$progs;
%channelDetails = %$channels;

# Put channels in correct order
@channelOrder = split(/ /, $options{'ChannelOrder'});
foreach $c (keys %$channels) { $aC{$c} = 0; }
$i = 1; foreach $c (@channelOrder) { $aC{$c} = $i; $i++; }
foreach $c (keys %aC) {
  if ($aC{$c} > 0) {
    @allChannels[$aC{$c}] = $c;
  } else {
    @buttEnd = (@buttEnd, $c);
  }
}
if ( @buttEnd and ($options{ChannelHideUnordered} != 1) ) {
  @allChannels = (@allChannels, @buttEnd);
}
$firstElement = shift @allChannels;
@allChannels = ($firstElement, @allChannels) unless ($firstElement eq '');

# Set date and time to some sort of pre-historic values:
# 1 June 1980 was when I was born. Hint, hint... nudge, nudge...
$myDate  = '19800601';
$myHour  = '25';
$myClass = 'listCell1';

# boolean (1 or 0) indicating whether the top of the table has been
# printed.
$myTop   = 0;

# loop through all programs
foreach my $thisProg (@progs)
{
  # just in case
  %pd = ();

  # gather program details
  $pd{time}         = $thisProg->{start};
  $pd{channel}      = $thisProg->{channel};
  $pd{title}        = $thisProg->{title}->[0]->[0];
  $pd{subtitle}     = $thisProg->{'sub-title'}->[0]->[0];
  $pd{description}  = $thisProg->{desc}->[0]->[0];
  $pd{category}     = $thisProg->{category}->[0]->[0];
                      # I don't really care about things like
                      # subtitles, stereo, star rating, ep number,
                      # etc. If you do, feel free to add them.

  # prepare less immediate info
  $pd{date}         = substr($pd{time}, 0, 8);
  $pd{hour}         = substr($pd{time}, 8, 2);
  $pd{time}         = substr($pd{time}, 8, 4);   # for pretty
                                                 # printing
  if($myTop == 0 && $myDate ne '19800601')
  {
   $myTop = 1;
   &dump_html_head();
  }

  # if we're on to a new hour, then print out the TV guide for
  # the previous hour and then reset our accumulator.
  if ( $pd{hour} != $myHour)
  {
    &dump_out_html();
    $myHour = $pd{hour};  $myDate = $pd{date};  %guideHash = ();
  }

  # NASTY DATA FORMAT!!!
  #   Examples -- 8pm, Friday:
  #     $guideHash{'bbc1.bbc.co.uk'}[0]{title} = 'EastEnders'
  #     $guideHash{'bbc1.bbc.co.uk'}[0]{time}  = '20:00'
  #     $guideHash{'bbc1.bbc.co.uk'}[1]{title} = 'Alistair McGowan's Big Impression'
  #     $guideHash{'bbc1.bbc.co.uk'}[1]{time}  = '20:30'
  #  This is a hash within an array within a hash
  push @{ $guideHash{$pd{channel}} }, { %pd };
}

# Print out the last hour
&dump_out_html();

&dump_html_foot();


###########################################################################
###########################################################################
###########################################################################
# Everything below here really needs a clean-up. Will do this after 0.2.0 #
###########################################################################
###########################################################################
###########################################################################


# Function for debugging. Plain text is a little easier to read on the
# console than HTML.
sub dump_out_text() {
  print "------------------------------------\n";
  print "Date: $myDate   Hour: $myHour\n";
  foreach $channel (@allChannels) {
    print " $channel:\n";
    @progList = @{ $guideHash{$channel} };
    foreach $prog (@progList) {
      print "  $prog->{time} - $prog->{title} ($prog->{category})\n";
    }
  }
}


# HTML Output
sub dump_out_html() {

  if ($options{StyleByHTML} == 1) {
    $htmlBgPale 	= 'bgcolor="#c0c0ff"';
    $htmlBgMedium 	= 'bgcolor="#0000ff"';
    $htmlBgDark 	= 'bgcolor="#000080"';
    $htmlFontWhite	= '<font color="white">';
    $htmlFontBlue	= '<font color="#0000ff">';
    $htmlFontEnd	= '</font>';
    $htmlBold		= '<b>';
    $htmlBoldEnd	= '</b>';
    $htmlItalic		= '<i>';
    $htmlItalicEnd	= '</i>';
  }


  # dummy date
  if ($myDate ne '19800601') {

    # print out a banner if it's midnight.
    if ($myHour eq '00') {
      $d = substr($myDate, 6, 2) . "/" . substr($myDate, 4, 2) . "/" . substr($myDate, 0, 4);
      $c = $#allChannels + 2;
      print "<tr><th colspan=\"$c\" class=\"day\" $htmlBgMedium><a name=\"$myDate\">$htmlFontWhite$d$htmlFontEnd</a></th></tr>\n";
      print "<tr>\n";
      print " <th class=\"remind\" $htmlBgDark>$htmlFontWhite Time $htmlFontEnd</th>\n";
      foreach $channel (@allChannels) {
        my ($cName, $cLang) = @{XMLTV::best_name($langs, $channelDetails{$channel}->{'display-name'})};
        my ($cIcon) = $channelDetails{$channel}->{'icon'}->[0]{src};
        if ($options{'DisplayChannels2'} eq 'Name') {
          print " <th class=\"remind\" $htmlBgDark>$htmlFontWhite$cName$htmlFontEnd</th>\n";
        } elsif ($options{'DisplayChannels2'} eq 'Caption') {
          print " <th class=\"remind\" $htmlBgDark>$htmlFontWhite$cName$htmlFontEnd<br><img src=\"$cIcon\" height=\"48\" width=\"48\" alt=\"$cName\"></th>\n";
        } elsif ($options{'DisplayChannels2'} eq 'Icon') {
          print " <th class=\"remind\" $htmlBgDark><img src=\"$cIcon\" height=\"48\" width=\"48\" alt=\"$cName\"></th>\n";
        } else {
          print " <th class=\"remind\" $htmlBgDark>$htmlFontWhite$channel$htmlFontEnd</th>\n";
        }
      }
      print "</tr>\n";
    }

    # print out this hour's TV!
    print "<tr>\n";
    print " <td class=\"time\" valign=\"top\" $htmlBgMedium>$htmlFontWhite $myHour:00 $htmlFontEnd</td>\n";

    foreach $channel (@allChannels) {
      print " <td class=\"$myClass\" valign=\"top\" $myHtml>\n";
      @progList = @{ $guideHash{$channel} };
      foreach $prog (@progList) {
	$good = '';
        foreach $i (@goods) {
          if ($prog->{title} =~ /$i/i) {
            $good = ' class="good"';
            #break;
          }
        }
        print "  <p$good>\n";
	$formattedTime = substr($prog->{time},0,2) . ":" . substr($prog->{time},2,2);
        print "   <span class=\"time\">$htmlBold$htmlFontBlue "     . $formattedTime       .  " $htmlFontEnd$htmlBoldEnd</span>\n";
        print "   <span class=\"title\">$htmlBold "    . $prog->{title}       . " $htmlBoldEnd</span><br>\n";
        print "   <span class=\"subtitle\"> " . $prog->{subtitle}    . " </span><br>\n" if (length($prog->{subtitle}) > 2);
        if ($options{'DisplayDescriptions'} == 1) {
	  print "   <span class=\"category\"> " . $prog->{category}    . " </span>";
          print    "<span class=\"desc\">$htmlItalic "     . $prog->{description} . " $htmlItalicEnd</span>\n";
        } elsif ($options{'DisplayDescriptions'} == 2) {
	  $prog->{description} =~ s/\"/&quot;/g;
	  print "   <span class=\"category\" title=\"" . $prog->{description} . "\">" . $prog->{category} . " </span>";
	}
	print "  </p>\n";
      }
      print "  </td>\n";
    }
    print "</tr>\n";
  }

  # alternate colours by row
  if ($myClass eq 'listCell1') { $myClass = 'listCell2'; $myHtml = $htmlBgPale; } 
  else                         { $myClass = 'listCell1'; $myHtml = ''; }
}

sub dump_html_head() {

  $d = substr($myDate, 6, 2) . "/" . substr($myDate, 4, 2) . "/" . substr($myDate, 0, 4);
  $c = $#allChannels + 2;

  if ($options{StyleByHTML} == 1) {
    $htmlBgMedium 	= 'bgcolor="#0000ff"';
    $htmlBgDark 	= 'bgcolor="#000080"';
    $htmlFontWhite	= '<font color="white">';
    $htmlFontEnd	= '</font>';
    $htmlBorder		= 'border="1"';
  }

  print <<EOF;
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">
<html>
<head>
 <title>TV Guide</title>
EOF
  if ($options{StyleByCSS} == 1) {
    print <<EOF;
    
 <style type="text/css"><!--
body { color: black; background-color: white; }
h1 { font-size: 34px; }
table { background-color: #007; margin:0; border:0; }
th { background-color: #007; color: white; font-size: 14px; font-weight: bold; padding: 4px; margin:0; border:0;}
th.remind { font-size: 14px; font-weight: normal; }
th.day { color: white; background-color: #00F; font-size: 18px; }
td { padding: 4px; margin:0; border:0;}
td.listCell1 { background-color: white; }
td.listCell2 { background-color: #c0c0ff; }
td.time { color: white; background-color: #00F; font-size: 14px; font-weight: bold; }
span.time { font-weight: bold; color: #00F; }
span.title { font-weight: bold; }
span.subtitle { font-size: 11px; }
span.desc { font-style: italic; font-size: 11px; }
span.category { font-weight: bold; font-size: 11px; }
p.foot { font-size: 11px; }
p.good ( background-color: #FFC; }
 --></style>
    
EOF
  }
  print <<EOF;
 <meta http-equiv="Content-type" content="text/html; charset=ISO-8859-1">
 <meta http-equiv="Content-language" content="en">
</head>
<body>
<h1>TV Guide</h1>
<table $htmlBorder>
<thead>
<tr><th colspan="$c" class="day" $htmlBgMedium>$htmlFontWhite$d$htmlFontEnd</th></tr>
<tr>
 <th $htmlBgDark>$htmlFontWhite Time $htmlFontEnd</th>
EOF
  foreach $channel (@allChannels) {
    my ($cName, $cLang) = @{XMLTV::best_name($langs, $channelDetails{$channel}->{'display-name'})};
    my ($cIcon) = $channelDetails{$channel}->{'icon'}->[0]{src};
    if ($options{'DisplayChannels'} eq 'Name') {
      print " <th $htmlBgDark>$htmlFontWhite$cName$htmlFontEnd</th>\n";
    } elsif ($options{'DisplayChannels'} eq 'Caption') {
      print " <th $htmlBgDark>$htmlFontWhite$cName$htmlFontEnd<br><img src=\"$cIcon\" height=\"48\" width=\"48\" alt=\"$cName\"></th>\n";
    } elsif ($options{'DisplayChannels'} eq 'Icon') {
      print " <th $htmlBgDark><img src=\"$cIcon\" height=\"48\" width=\"48\" alt=\"$cName\"></th>\n";
    } else {
      print " <th $htmlBgDark>$htmlFontWhite$channel$htmlFontEnd</th>\n";
    }
  }
  print <<EOF;
</tr>
</thead>
<tbody>
EOF
}

sub dump_html_foot() {
  print <<EOF;
</tbody>
</table>
<p class="foot">Data from <a href="http://www.ananova.com/">Ananova</a>.
<a href="http://membled.com/work/apps/xmltv/">XMLTV framework</a>
by <a href="mailto:ed%40membled.com">Ed Avis</a>. <a 
href="http://www.goddamn.co.uk/tobyink/?id=204">HTML output</a> by
<a href="http://www.goddamn.co.uk/tobyink/?id=132">Toby A Inkster</a>.</p>
<p class="valid"><a href="http://validator.w3.org/check/referer">Valid 
HTML 4.01.</a></p>
</body>
</html>

EOF

}

sub read_options() {
  debug('Beginning reading settings...');
  debugUp();
  $H = $ENV{'HOME'};
  foreach $f ('/etc/tv2html.conf', "$H/.tv2html.conf", "$H/etc/tv2html.conf", '/etc/tv2html.important.conf') {
    debug("Tryng to open $f");
    if (open(FH, $f) ) {
      debug("Reading settings from $f");
      debugUp();
      while(<FH>) {
        chomp;
        if ($_ !~ m/^\s*\#/) {
          ($optionName, @args) = split(/ /, $_);
          $options{$optionName} = join(' ',@args);
          debug("Option $optionName read.");
        }
      }
      debugDown();
    } else {
      debug('failed!');
    }
  }
  debugDown();
  debug('done!');
}

sub read_goods() {
  debug('Beginning reading favourites...');
  debugUp();
  $H = $ENV{'HOME'};
  foreach $f ('/etc/tv2html.good', "$H/.tv2html.good", "$H/etc/tv2html.good") {
    if (open(FH, $f) ) {
      debug("Reading favourites from $f");
      debugUp();
      while(<FH>) {  
        chomp;
	debug("$_ is good.");
        push @goods, $_;
      }
      debugDown();
      close(FH);
    }
  }
  debugDown();
  debug('done!');
}




sub debug {
  $d = shift @_;
  for ($i = 0; $i<$debuglevel; $i++) {
    print STDERR "\t";
  }
  print STDERR "$d\n";
}

sub debugUp {
  $debuglevel++;
}

sub debugDown {
  $debuglevel--;
}