Allow proper subclassing of complete Getters/Grabbers

[videosite.git] / videosite / DailyMotionGrabber.pm

diff --git a/videosite/DailyMotionGrabber.pm b/videosite/DailyMotionGrabber.pm
index 981fef7..3e1e77f 100644 (file)
--- a/videosite/DailyMotionGrabber.pm
+++ b/videosite/DailyMotionGrabber.pm
@@ -8,7 +8,6 @@ package videosite::DailyMotionGrabber;
 use videosite::GrabberBase;
 @ISA = qw(videosite::GrabberBase);
 
-use LWP::Simple qw(!get);
 use HTML::Parser;
 use videosite::JSArrayParser;
 use Data::Dumper;
@@ -17,16 +16,13 @@ use strict;
 
 sub new {
     my $class = shift;
-    my $self = $class->SUPER::new();
+    my $self = $class->SUPER::new(
+        NAME => 'dailymotion',
+        PATTERNS => ['(http://(?:[-a-zA-Z0-9_.]+\.)*dailymotion.com/(?:[^/]+/)*video/([-a-zA-Z0-9_]+))'],
+        @_,
+    );
 
-    $self->{'NAME'} = 'dailymotion';
-    $self->{'PATTERNS'} = ['(http://(?:[-a-zA-Z0-9_.]+\.)*dailymotion.com/(?:[^/]+/)*video/([-a-zA-Z0-9_]+))'];
-
-    bless($self, $class);
-
-    $self->_prepare_parameters();
-
-    return $self;
+    return bless($self, $class);
 }
 
 sub _parse {
@@ -50,7 +46,7 @@ sub _parse {
     $metadata->{'TITLE'} = undef;
     $metadata->{'DLURL'} = undef;
 
-    unless(defined($content = LWP::Simple::get(sprintf('http://www.dailymotion.com/video/%s', $2)))) {
+    unless(defined($content = $self->simple_get(sprintf('http://www.dailymotion.com/video/%s', $2)))) {
         $self->error('Could not download %s', $url);
         return undef;
     }
@@ -77,7 +73,7 @@ sub _parse {
         if ($e->[0] =~ m|\.addVariable\("sequence",\s*"([^\"]+)"|) {
             my $sequence = $1;
             my $jsp = videosite::JSArrayParser->new();
-            my $main;
+            my $l;
             my $s;
 
             $sequence =~ s/%(..)/chr(hex($1))/ge;
@@ -93,31 +89,22 @@ sub _parse {
             } else {
                 $self->debug("Parsed sequence: %s", Dumper($sequence));
 
-                foreach (@{$sequence}) {
-                    if (exists($_->{'name'}) && ($_->{'name'} eq 'main')) {
-                        # Found main section
-                        $main = $_->{'layerList'};
-                    }
-                }
-                unless(defined($main)) {
-                    $self->error("Could not find layerList[main]");
+                $l = $self->_fetch_layer($sequence, "root/layerList", "background/sequenceList", "main/layerList", "video/param");
+                unless(defined($l)) {
+                    $self->error("Could not find video layer");
                     return undef;
                 }
 
-                foreach (@{$main}) {
-                    if (exists($_->{'name'}) && ($_->{'name'} eq 'video')) {
-                        # Found video section
-                        if (exists($_->{'param'}->{'hdURL'})) {
-                            $metadata->{'DLURL'} = $_->{'param'}->{'hdURL'};
-                        } elsif (exists($_->{'param'}->{'hqURL'})) {
-                            $metadata->{'DLURL'} = $_->{'param'}->{'hqURL'};
-                        } elsif (exists($_->{'param'}->{'hqURL'})) {
-                            $metadata->{'DLURL'} = $_->{'param'}->{'sdURL'};
-                        } else {
-                            $self->error("Video section found, but no URLs");
-                            return undef;
-                        }
-                    }
+                # Found video section
+                if (exists($l->{'videoPluginParameters'}->{'hdURL'})) {
+                    $metadata->{'DLURL'} = $l->{'videoPluginParameters'}->{'hdURL'};
+                } elsif (exists($l->{'videoPluginParameters'}->{'hqURL'})) {
+                    $metadata->{'DLURL'} = $l->{'videoPluginParameters'}->{'hqURL'};
+                } elsif (exists($l->{'videoPluginParameters'}->{'hqURL'})) {
+                    $metadata->{'DLURL'} = $l->{'videoPluginParameters'}->{'sdURL'};
+                } else {
+                    $self->error("Video section found, but no URLs");
+                    return undef;
                 }
             }
         }
@@ -131,4 +118,37 @@ sub _parse {
     return $metadata;
 }
 
+sub _fetch_layer {
+    my $self = shift;
+    my $sequence = shift;
+    my $point = shift;
+    my $next;
+    my @points = @_;
+    my $l;
+
+    $self->debug("Looking for %s in %s", $point, Dumper($sequence));
+
+    unless(defined($point)) {
+        $self->debug("Reached last point");
+        return $sequence;
+    }
+    ($point, $next) = split(/\//, $point, 2);
+
+    foreach (@{$sequence}) {
+        if (exists($_->{'name'}) and ($_->{'name'} eq $point)) {
+            if (exists($_->{$next})) {
+                $self->debug("Using %s in %s", $next, $point);
+                return $self->_fetch_layer($_->{$next}, @points);
+            } else {
+                $self->debug("%s found, but no %s", $point, $next);
+                return undef;
+            }
+
+        }
+    }
+
+    $self->debug("Could not find entry named %s", $point);
+    return undef;
+}
+
 1;