X-Git-Url: https://git.camperquake.de/gitweb.cgi?a=blobdiff_plain;f=videosite%2FFileGetter.pm;h=5e54935c15076aecbb5368fc3d71bf3a964500e4;hb=refs%2Fheads%2Fmaster;hp=8c3420d30d69c3fc3a092977f086f35b1e612131;hpb=2044d5ff60e639a2eafbeed3a70224564e94c84c;p=videosite.git

diff --git a/videosite/FileGetter.pm b/videosite/FileGetter.pm
index 8c3420d..5e54935 100644
--- a/videosite/FileGetter.pm
+++ b/videosite/FileGetter.pm
@@ -4,26 +4,26 @@
 # A getter which will download the media to a local file storage
 #
 
-package FileGetter;
+package videosite::FileGetter;
 
-use GetterBase;
-@ISA = qw(GetterBase);
+use videosite::GetterBase;
+@ISA = qw(videosite::GetterBase);
 
 use strict;
-use LWP::Simple qw(!get);
 use File::Basename;
 
 sub new {
     my $class = shift;
-    my $self = $class->SUPER::new();
-
-    $self->{'NAME'} = 'filegetter';
-    $self->{'_PARAMS'} = {'MINFREE' => ['500000', 'The amount of space that needs to be available on the filesystem before the video is downloaded (in kilobytes)'], 'FILEPATTERN', => ['/tmp/%s - %s - %s.flv', "The file name to save the file under. This is a string which is passed to a sprintf call later on. The parameters passed to that sprintf call, in order, are:\n- The site the video is from\n- The ID of the video\n- The title of the video\n- The URL of the video file itself\n- The URL of the site the video was taken from\nAll parameters are encoded (space and / replaced by _)"]};
-
-    bless($self, $class);
-    $self->_prepare_parameters();
-
-    return $self;
+    my $self = $class->SUPER::new(
+        NAME => 'filegetter',
+        _PARAMS => {
+            MINFREE => ['500000', 'The amount of space that needs to be available on the filesystem before the video is downloaded (in kilobytes)'],
+            FILEPATTERN => ['/tmp/%s - %s - %s.flv', "The file name to save the file under. This is a string which is passed to a sprintf call later on. The parameters passed to that sprintf call, in order, are:\n- The site the video is from\n- The ID of the video\n- The title of the video\n- The URL of the video file itself\n- The URL of the site the video was taken from\nAll parameters are encoded (space and / replaced by _)"]
+        },
+        @_,
+    );
+
+    return bless($self, $class);
 }
 
 sub get {
@@ -45,17 +45,42 @@ sub get {
         return 0;
     }
 
-    $self->debug('Going to download %s to %s', $video->{'DLURL'}, $dlfile);
+    if (exists($video->{'CONNECTOR'})) {
+        $self->selectconn($video->{'CONNECTOR'});
+    }
 
-    if (200 != LWP::Simple::mirror($video->{'DLURL'}, $dlfile)) {
-        $self->error('Could not download %s to %s', $video->{'DLURL'}, $dlfile);
-        return 0;
+
+    $self->_download( $video->{'DLURL'}, $dlfile, $video )
+	or return 0;
+
+    if (exists($video->{'DLURL_AUDIO'})) {
+        $self->_download( $video->{'DLURL_AUDIO'}, $dlfile . '.audio', $video )
+	    or return 0;
     }
 
     return 1;
 }
 
 
+sub _download {
+    my $self = shift;
+    my $dlurl = shift;
+    my $dlfile = shift;
+    my $video = shift;
+    my $res;
+
+    $self->debug('Going to download %s to %s', $dlurl, $dlfile);
+
+    $res = $self->ua()->mirror($dlurl, $dlfile);
+    
+    if (!$res->is_success()) {
+        $self->error('Could not download %s to %s (%s)', $dlurl, $dlfile, $res->code());
+        return 0;
+    }
+
+    return 1;
+}
+
 sub _encode {
     my $self = shift;
     my $s = shift;