#!/usr/bin/perl -w

#
# This perl script is intended to perform movie data lookups based on
# the themoviedb.org website
#
# For more information on MythVideo's external movie lookup mechanism, see
# the README file in this directory.
#
# Author: William Stewart
# 
# v0.1
# - Initial script

use File::Basename;
use lib dirname($0);

use MythTV::MythVideoCommon;

eval "use DateTime::Format::Strptime"; my $has_date_format = $@ ? 0 : 1;

use vars qw($opt_h $opt_r $opt_d $opt_i $opt_v $opt_D $opt_M $opt_P);
use Getopt::Std;
use Data::Dumper;  
use LWP::Simple; 
use XML::Simple; 

$title = "themoviedb Query";
$version = "v0.1.0";
$author = "William Stewart";
push(@MythTV::MythVideoCommon::URL_get_extras, ($title, $version));

my @countries = qw(USA UK Canada Japan);

my $base_url       = "http://api.themoviedb.org/2.0/";
my $posterimdb_url = "http://api.themoviedb.org/cover.php?imdb=";

# themoviedb.org api key given by Travis Bell for Mythtv
my $api_key        = "c27cb71cff5bd76e1a7a009380562c62";

binmode(STDOUT, ":utf8");

# display usage
sub usage {
   print "usage: $0 -hdrviMPD [parameters]\n";
   print "       -h           help\n";
   print "       -d           debug\n";
   print "       -r           dump raw query result data only\n";
   print "       -v           display version\n";
   print "       -i           display info\n";
   print "\n";
   print "       -M [options] <query>    get movie list\n";
   print "               some known options are:\n";
   print "                  type=[fuzy]         looser search\n";
   print "                  from_year=[int]     limit matches to year\n";
   print "                  to_year=[int]       limit matches to year\n";
   print "                  sort=[smart]        ??\n";
   print "                  tv=[no|both|only]   limits between tv and movies\n";
   print "               Note: multiple options must be separated by ';'\n";
   print "       -P <movieid>  get movie poster\n";
   print "       -D <movieid>  get movie data\n";
   exit(-1);
}

# display 1-line of info that describes the version of the program
sub version {
   print "$title ($version) by $author\n"
}

# display 1-line of info that can describe the type of query used
sub info {
   print "Performs queries using the themoviedb.org website.\n";
}

# display detailed help
sub help {
   version();
   info();
   usage();
}

# get Movie Data
sub getMovieData {
   my ($movieid)=@_; # grab movieid parameter
   if (defined $opt_d) { printf("# looking for movie id: '%s'\n", $movieid);}

   # get the search results page via Movie.imdbLookup
   my $request = $base_url . "Movie.imdbLookup?imdb_id=tt" . $movieid . "&api_key=" . $api_key;
   if (defined $opt_d) { printf("# request: '%s'\n", $request); }
   my ($rc, $response) = myth_url_get($request);
   if (defined $opt_r) { printf("%s", $response); }

   my $xs = new XML::Simple(suppressempty => '');   
   my $xml = $xs->XMLin($response);

   if ($xml->{"opensearch:totalResults"} > 0) {
        # 
        # now get the movie data via Movie.getInfo, Movie.imdbLookup does not provide us all the data
        #
        my $tmdbid = $xml->{moviematches}->{movie}->{id};
        $request = $base_url . "Movie.getInfo?id=" . $tmdbid . "&api_key=" . $api_key;
        if (defined $opt_d) { printf("# request: '%s'\n", $request); }
        my ($rc, $response) = myth_url_get($request);
        if (defined $opt_r) { printf("%s", $response); }

        $xml = $xs->XMLin($response);
  
        my $title       = $xml->{moviematches}->{movie}->{title}; 
        my $releasedate = $xml->{moviematches}->{movie}->{release};
        my $year        = substr($releasedate, 0, 4);
        my $plot        = $xml->{moviematches}->{movie}->{short_overview};
        my $userrating  = $xml->{moviematches}->{movie}->{rating};
        my $runtime     = $xml->{moviematches}->{movie}->{runtime};
        my $budget      = $xml->{moviematches}->{movie}->{budget};
        my $revenue     = $xml->{moviematches}->{movie}->{revenue};
        my $trailer     = $xml->{moviematches}->{movie}->{trailer}->{content};
        my $homepage    = $xml->{moviematches}->{movie}->{homepage};

        # Genre
        my @lgenres;
        my $j = 0;  
        foreach my $key (keys (%{$xml->{moviematches}->{movie}->{categories}->{category}}))
        {
           $lgenres[$j++] = $key;
        }
        my $genres = join(',', @lgenres);

        # People
        my @lcast;
        my @ldirector;
        my $dc = 0;
        my $cc = 0;  
        foreach my $key (keys (%{$xml->{moviematches}->{movie}->{people}->{person}}))
        {        
            if ($xml->{moviematches}->{movie}->{people}->{person}->{$key}->{job} eq "actor")
            {
              $lcast[$cc++] = $key;             
            }
            
            if ($xml->{moviematches}->{movie}->{people}->{person}->{$key}->{job} eq "director")
            {
              $ldirector[$dc++] = $key;             
            }
        }
        my $cast     = join(',', @lcast);
        my $director = join(',', @ldirector);

        # output fields (these field names must match what MythVideo is looking for)
        print "Title:$title\n";
        print "Year:$year\n";
        print "ReleaseDate:$releasedate\n";
        print "Director:$director\n";
        print "Plot:$plot\n";
        print "UserRating:$userrating\n";
#   print "MovieRating:$movierating\n";
        print "Runtime:$runtime\n";
#        print "Writers: $writer\n";
        print "Cast:$cast\n";
        print "Genres: $genres\n";
#        print "Countries: $lcountries\n";
        print "Budget: $budget\n";
        print "Revenue: $revenue\n";
        if ($trailer) {print "trailer: $trailer\n";}
        if ($homepage) {print "Homepage: $homepage\n";}
    }
}

# dump Movie Poster
sub getMoviePoster {
   my ($movieid)=@_; # grab movieid parameter
   if (defined $opt_d) { printf("# looking for poster for movie id: '%s'\n", $movieid);}

   # get the search results  page
   my $request = $base_url . "Movie.imdbLookup?imdb_id=tt" . $movieid . "&api_key=" . $api_key;
   if (defined $opt_d) { printf("# request: '%s'\n", $request); }
   my ($rc, $response) = myth_url_get($request);
   if (defined $opt_r) { printf("%s", $response); }
   
   my $xml = XMLin($response);
   
   my $poster = $xml->{moviematches}->{movie}->{poster}->[0]->{content};
   if ($poster) {print "$poster\n";}
    
   @poster_urls = getPosterURLsFromIMDB($movieid, $opt_d, $opt_r);
   for $url (@poster_urls) { print("$url\n"); }
}

# dump Movie list:  1 entry per line, each line as 'movieid:Movie Title'
sub getMovieList {
   my ($filename, $options)=@_; # grab parameters

   #
   # Convert filename into a query string
   # (use same rules that Metadata::guesTitle does)
   my $query = cleanTitleQuery($filename);
   if (!$options) { $options = "" ;}
   if (defined $opt_d) {
      printf("# query: '%s'\n", $query);
   }

   # get the search results  page
   my $request = $base_url . "Movie.search?title=" . $query . "&api_key=" . $api_key;
   if (defined $opt_d) { printf("# request: '%s'\n", $request); }
   my ($rc, $response) = myth_url_get($request);
   if (defined $opt_r) {
      print $response;
      exit(0);
   }

   my $xs = new XML::Simple(suppressempty => '');
   my $xml = $xs->XMLin($response);
   my @movies;
   my $j = 0;
   
   foreach my $key (keys (%{$xml->{moviematches}->{movie}}))
   {
      my $movienum  = $xml->{moviematches}->{movie}->{$key}->{imdb};
      my $moviename = $xml->{moviematches}->{movie}->{$key}->{title};
      my $release   = $xml->{moviematches}->{movie}->{$key}->{release};
      my $movieyear = substr($release, 0, 4);
      
      if ($movienum) {
         if ($release) {
            $movies[$j++]= substr($movienum,2) . ":" . $moviename . " (". $movieyear . ")";
         }
         else {
            $movies[$j++]= substr($movienum,2) . ":" . $moviename;
         }
      }
   }
   # display array of values

   for $movie (@movies) { print "$movie\n"; }   
}

#
# Main Program
#

# parse command line arguments
getopts('ohrdivDMP');

# print out info
if (defined $opt_v) { version(); exit 1; }
if (defined $opt_i) { info(); exit 1; }

# print out usage if needed
if (defined $opt_h || $#ARGV<0) { help(); }

if (defined $opt_D) {
   # take movieid from cmdline arg
   $movieid = shift || die "Usage : $0 -D <movieid>\n";
   getMovieData($movieid);
}

elsif (defined $opt_P) {
   # take movieid from cmdline arg
   $movieid = shift || die "Usage : $0 -P <movieid>\n";
   getMoviePoster($movieid);
}

elsif (defined $opt_M) {
   # take query from cmdline arg
   $options = shift || die "Usage : $0 -M [options] <query>\n";
   $query = shift;
   if (!$query) {
      $query = $options;
      $options = "";
   }
   getMovieList($query, $options);
}
# vim: set expandtab ts=3 sw=3 :