Merge pull request #117 from karjonas/kanal5play

Kanal5play.se: Adding quality selection and subtitles

1 files changed, 71 insertions, 12 deletions

diff --git a/lib/FlashVideo/Site/Kanal5play.pm b/lib/FlashVideo/Site/Kanal5play.pm
index 125a627..16081f2 100644
--- a/lib/FlashVideo/Site/Kanal5play.pm
+++ b/lib/FlashVideo/Site/Kanal5play.pm
@@ -6,9 +6,15 @@ use warnings;
 use FlashVideo::Utils;
 use FlashVideo::JSON;
 
-our $VERSION = '0.02';
+our $VERSION = '0.03';
 sub Version() { $VERSION;}
 
+my $bitrate_index = {
+  high   => 0,
+  medium => 1,
+  low    => 2
+};
+
 sub find_video {
   my ($self, $browser, $embed_url, $prefs) = @_;
   if (!($browser->uri->as_string =~ m/video\/([0-9]*)/)) {
@@ -24,22 +30,18 @@ sub find_video {
 
   my $jsonstr  = $browser->content;
   my $json     = from_json($jsonstr);
-
   my $name     = $json->{program}->{name};
   my $episode  = $json->{episodeNumber};
   my $season   = $json->{seasonNumber};
+  my $subtitle = $json->{hasSubtitle};
   my $filename = sprintf "%s - S%02dE%02d", $name, $season, $episode;
   my $rtmp     = "rtmp://fl1.c00608.cdn.qbrick.com:1935/00608";
   my $playpath = $json->{streams}[0]->{source};
-  my $max_rate = 0;
+  my %paths=();
 
-  # Always take the highest bitrate stream
+  # Put the streams into the hash
   foreach my $stream (@{$json->{streams}}) {
-    my $rate = int($stream->{bitrate});
-    if ($rate > $max_rate) {
-      $playpath = $stream->{source};
-      $max_rate = $rate;
-    }
+    $paths{int($stream->{bitrate})} = $stream->{source};
   }
 
   # Check if the maximum quality stream is available.
@@ -56,9 +58,66 @@ sub find_video {
     swfVfy   => "http://www.kanal5play.se/flash/K5StandardPlayer.swf"
   };
 
-  # If the stream was not found we revert the playpath
-  if (!$downloader->try_download($args)) {
-    $args->{playpath} = $playpath;
+  # If the stream was found we push it to the list of playpaths
+  if ($downloader->try_download($args)) {
+    $paths{1600000} = $playpath_max;
+  }
+
+  # Sort the paths and select the suitable one based upon quality preference
+  my $quality = $bitrate_index->{$prefs->{quality}};
+  my $min = $quality < scalar(keys(%paths)) ? $quality : scalar(keys(%paths));
+  my $key = (sort {int($b) <=> int($a)} keys %paths)[$min];
+  $args->{playpath} = $paths{$key};
+
+  # Check for subtitles
+  if ($prefs->{subtitles} and $subtitle) {
+    my $subtitle_url = "http://www.kanal5play.se/api/subtitles/$video_id";
+    $browser->get($subtitle_url);
+    if (!$browser->success) {
+      die "Couldn't download $subtitle_url: " . $browser->response->status_line;
+    }
+    $jsonstr = $browser->content;
+    $json = from_json($jsonstr);
+
+    # The format is a list of hashmap with the following keys:
+    # startMillis : int
+    # endMillis : int
+    # text : string
+    # posX : int
+    # posY : int
+    # colorR : int
+    # colorG : int
+    # colorB : int
+    #
+    # We convert this to an srt
+
+    my $srt_filename = title_to_filename($filename, "srt");
+    open my $srt_fh, '>', $srt_filename
+      or die "Can't open subtitles file $srt_filename: $!";
+
+    my $i = 1;
+
+    foreach my $line (@{$json}) {
+      my $text  = $line->{text};
+      my $hour  = int($line->{startMillis}) / 3600000;
+      my $min   = (int($line->{startMillis}) / 60000) % 60;
+      my $sec   = (int($line->{startMillis}) / 1000) % 60;
+      my $milli = int($line->{startMillis}) % 1000;
+
+      my $start = sprintf "%02d:%02d:%02d,%03d", $hour, $min, $sec, $milli;
+
+      $hour  = int($line->{endMillis}) / 3600000;
+      $min   = (int($line->{endMillis}) / 60000) % 60;
+      $sec   = (int($line->{endMillis}) / 1000) % 60;
+      $milli = int($line->{endMillis}) % 1000;
+
+      my $end = sprintf "%02d:%02d:%02d,%03d", $hour, $min, $sec, $milli;
+
+      print $srt_fh "$i\n" . "$start --> $end\n" . "$text\n\n";
+
+      $i++;
+    }
+    close $srt_fh;
   }
 
   return $args;