Youtube: specify video formats more clearly

[videosite.git] / videosite / AsyncFileGetter.pm

diff --git a/videosite/AsyncFileGetter.pm b/videosite/AsyncFileGetter.pm
index 04e8226..7a67b43 100644 (file)
--- a/videosite/AsyncFileGetter.pm
+++ b/videosite/AsyncFileGetter.pm
@@ -1,5 +1,5 @@
 # (c) 2007 by Ralf Ertzinger <ralf@camperquake.de>
-#     2008 by Christian Garbs <mitch@cgarbs.de>
+#     2008-2009 by Christian Garbs <mitch@cgarbs.de>
 #
 # licensed under GNU GPL v2
 #
@@ -7,14 +7,16 @@
 # in the background
 #
 
-package AsyncFileGetter;
+package videosite::AsyncFileGetter;
 
-use FileGetter;
-@ISA = qw(FileGetter);
+use videosite::FileGetter;
+@ISA = qw(videosite::FileGetter);
 
 use strict;
 use LWP::Simple qw(!get);
 use File::Basename;
+use File::Temp qw(tempfile);
+use MIME::Base64;
 
 sub new {
     my $class = shift;
@@ -33,6 +35,7 @@ sub get {
     my $video = shift;
     my $dlfile;
     my $dirname;
+    my $cookie = "";
 
     $dlfile = sprintf($self->_getval('FILEPATTERN'),
         $self->_encode($video->{'SOURCE'}),
@@ -49,7 +52,11 @@ sub get {
 
     $self->debug('Going to download %s to %s', $video->{'DLURL'}, $dlfile);
 
-    my $cmdline = "GET \"$video->{'DLURL'}\" > \"$dlfile\" &";
+    my (undef, $tmpfile) = tempfile('videosite.tmp.XXXXXXXXXXXX', DIR => $dirname);
+
+    $cookie = "-H \"Cookie: $video->{'COOKIE'}\"" if (defined $video->{'COOKIE'});
+    my $cmdline = "( GET $cookie \"$video->{'DLURL'}\" > \"$tmpfile\" && mv \"$tmpfile\" \"$dlfile\" && chmod =rw \"$dlfile\" || rm -f \"$tmpfile\" ) &";
+    $self->debug(encode_base64($cmdline));
     system($cmdline);
 
     return 1;