Redtube: Fix grabber

[videosite.git] / videosite / GrabberBase.pm

diff --git a/videosite/GrabberBase.pm b/videosite/GrabberBase.pm
index 5dbb36b..e39be83 100644 (file)
--- a/videosite/GrabberBase.pm
+++ b/videosite/GrabberBase.pm
@@ -1,12 +1,17 @@
-package GrabberBase;
-use Base;
-@ISA = qw(Base);
+# (c) 2007 by Ralf Ertzinger <ralf@camperquake.de>
+# licensed under GNU GPL v2
+
+package videosite::GrabberBase;
+use videosite::Base;
+@ISA = qw(videosite::Base);
 
 use strict;
 
 sub new {
     my $class = shift;
-    my $self = {
+    my $self = $class->SUPER::new();
+
+    $self = {%{$self},
         NAME => 'FlashGrab',
         TYPE => 'grabber',
         PATTERNS => [],
@@ -18,18 +23,44 @@ sub get($$) {
     my $self = shift;
     my $url = shift;
     my $pattern;
+    my $res;
+
+    return undef unless $self->_getval('enabled');
 
     foreach $pattern (@{$self->{'PATTERNS'}}) {
         $self->debug("Matching %s against %s", $pattern, $url);
         if ($url =~ m|$pattern|) {
             $self->debug("Match");
-            return $self->_parse($url, $pattern);
+            foreach ($self->connectors()) {
+                $self->debug("Using connector %s", $_->{'name'});
+                $self->selectconn($_);
+                $res = $self->_parse($url, $pattern);
+                if (defined($res)) {
+                    $res->{'CONNECTOR'} = $_;
+                    last;
+                }
+            }
+            return wantarray?($res, $pattern):$res;
         }
     }
 
     return undef;
 }
 
+sub enable {
+    my $self = shift;
+
+    $self->debug('Enabling %s grabber', $self->{'NAME'});
+    $self->setval('enabled', '1');
+}
+
+sub disable {
+    my $self = shift;
+
+    $self->debug('Disabling %s grabber', $self->{'NAME'});
+    $self->setval('enabled', '0');
+}
+
 sub _parse {
     my $self = shift;
     my $url = shift;