#!/usr/bin/perl -w
# grab_fi_icons_xmltvids
# By Ilkka Tengvall 14.10.2007
# Shamelesly copied from grabukicons.pl made by Justin Hornsby 19 October 2006
# 
# A script for downloading Finnish TV channel icons and set the xmltvid fields.
# Also generates videosource xml file.
#
# Usage info will be output if the script is run with no arguments (or
# insufficient arguments)
#

# PERL MODULES WE WILL BE USING
use DBI;
use DBD::mysql

#
$host     = '192.168.1.10';
$database = 'mythconverg';
$user     = 'mythtv';
$pass     = $user;

$db = undef;
$connect = undef;
my @name;
my @data;
my $item;

######################################
#                                    #
#        SUBROUTINES ARE HERE        #
#                                    #
######################################


# Try to find the database
#
sub findDatabase()
{
    my $found = 0;
    my @mysql = ('/usr/share/mythtv/mysql.txt',
        '/usr/share/mythtv/mysql.txt',
        '/etc/mythtv/mysql.txt',
        '/usr/etc/mythtv/mysql.txt',
        "$ENV{HOME}/.mythtv/mysql.txt",
        'mysql.txt'
    );
    foreach my $file (@mysql) {
        next unless (-e $file);
        $found = 1;
        open(CONF, $file) or die "Unable to open $file:  $!\n\n";
        while (my $line = <CONF>) {
            # Cleanup
            next if ($line =~ /^\s*#/);
            $line =~ s/^str //;
            chomp($line);

            # Split off the var=val pairs
            my ($var, $val) = split(/\=/, $line, 2);
            next unless ($var && $var =~ /\w/);
            if ($var eq 'DBHostName') {
                $host = $val;
            } elsif ($var eq 'DBUserName') {
                $user = $val;
            } elsif ($var eq 'DBName') {
                $database = $val;
            } elsif ($var eq 'DBPassword') {
                $pass = $val;
            }
        }
        close CONF;
    }

if ( ! $found )
    {   die "Unable to locate mysql.txt:  $!\n\n"  }
}


#  DB connect subroutine
# =======================
# openDatabase() - Connect or die
#

sub openDatabase()
{
    $db = DBI->connect("dbi:mysql:database=$database" .
                       ":host=$host", $user, $pass)
    or die "Cannot connect to database: $!\n\n";

    return $db;
}


$usage = "\nPlease pass the location of the channel icons as an argument (e.g. /usr/share/mythtv/icons) \n ";

# find the database

findDatabase();

# get this script's ARGS
#

# if user hasn't passed enough arguments, die and print the usage info

unless ($ARGV[0])
{
    die "$usage";
}

$location = $ARGV[0];

$connect = openDatabase(); 

$query = "select name from channel where name != NULL or name NOT like ".'""'.";";
$query_handle = $connect->prepare($query);
$query_handle->execute()  || die "Unable to query channel table";


open(FILE, "lookup.txt") or die("Unable to open file");
                                  @data = <FILE>;
                                  close(FILE);
 
foreach $chan (@data)
{
    @name_url = split('\|', $chan);
    $xmltvid_map{$name_url[0]} = $name_url[1];
    $icon_map{$name_url[0]} = $name_url[2];
}

while (@channame = $query_handle->fetchrow())
{
    foreach $id (@channame)
    {
        if ($xmltvid_map{$id})
        {
            $xmltvid=$xmltvid_map{$id};
            $xmltvid=~ s/\s+$//;
            $xmltvid=~ s/^\s+//;
            if ($xmltvid ne "none")
            { 
                $db_xmltvid{$id} = $xmltvid;
            }
        } 
        if ($icon_map{$id})
        {
            $iconloc=$icon_map{$id};
            $iconloc =~ s/\s+$//;
            $iconloc =~ s/^\s+//;
            if ($iconloc ne "none")
            { 
                @name = split('/', $iconloc);
                $num=@name;
                $filename=$name[$num-1];
                $command = "wget -q $iconloc -O $location/$filename";
                print "grabbing $iconloc ..";
                system ($command);
                print ".. done \n";
                $dbicon{$id} = $location."/".$filename;
            }
        } 
    }
}
$query_handle->finish;

print "Icons grabbed and downloaded to $location \n.",
      "  Updating the database....\n";
foreach $item (keys %dbicon) {
    print "setting channel $item to have icon from $dbicon{$item} \n";
    $query = "UPDATE channel SET icon = ".'"'.$dbicon{$item}.'"'." WHERE name = ".'"'.$item.'"'.";";
    $query_handle = $connect->prepare($query);
    $query_handle->execute()  || die "Unable to query channel table";
}

foreach $item (keys %db_xmltvid) {
    print "setting channel $item to have xmltvid $db_xmltvid{$item} \n";
    $query = "UPDATE channel SET xmltvid = ".'"'.$db_xmltvid{$item}.'"'." WHERE name = ".'"'.$item.'"'.";";
    $query_handle = $connect->prepare($query);
    $query_handle->execute()  || die "Unable to query channel table";
}

print "To get xmltvdata with mythfilldatabase you need to copy paste \n",
      "the following text into videosource file:\n",
      "(trying to find name from database...)\n"; 
$query = "SELECT name FROM videosource;";
$query_handle = $connect->prepare($query);
$query_handle->execute()  || die "Unable to query channel table";
while ($videosource_file = $query_handle->fetchrow())
{
    print "Possible videosource filename: ~/.mythtv/$videosource_file.xmltv\n";
} 
$query_handle->finish;

print "--------------- cut beginning from here ----------------------------\n";
foreach $item (keys %db_xmltvid) {
    $ch_id = $db_xmltvid{$item}; 
    $ch_id =~ s/\.+.*$//;
    print "channel $ch_id $item\n";
}
print "--------------- end cut --------------------------------------------\n";
print "Now remember to save text above and run mythfilldatabase --update!\n";

print "done! \n";