# (c) 2007 by Ralf Ertzinger <ralf@camperquake.de>
-# 2008 by Christian Garbs <mitch@cgarbs.de>
+# 2008-2009,2011-2012 by Christian Garbs <mitch@cgarbs.de>
#
# licensed under GNU GPL v2
#
use strict;
use LWP::Simple qw(!get);
use File::Basename;
+use File::Temp qw(tempfile);
use MIME::Base64;
sub new {
my $video = shift;
my $dlfile;
my $dirname;
+ my $cookie = "";
$dlfile = sprintf($self->_getval('FILEPATTERN'),
$self->_encode($video->{'SOURCE'}),
$self->_encode($video->{'DLURL'}),
$self->_encode($video)->{'URL'});
+ $dlfile =~ s/([\\"])/\\$1/g;
+
$dirname = dirname($dlfile);
if ($self->_diskfree($dirname) < $self->_getval('MINFREE')) {
$self->error("Not enough free space to download");
$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);
+
+ my %saved_env;
+
+ if (exists($video->{'CONNECTOR'})) {
+ my $schemas = $video->{'CONNECTOR'}->{'schemas'};
+ foreach my $schemakey(keys(%{$schemas})) {
+ $self->debug("Setting %s_proxy to %s", $schemakey, $schemas->{$schemakey});
+ my $envkey = $schemakey.'_proxy';
+ $saved_env{ $envkey } = $ENV{ $envkey };
+ $ENV{ $envkey } = $schemas->{$schemakey};
+ }
+ }
+
+ $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);
+ # restore environment
+ foreach my $envkey (keys %saved_env) {
+ $self->debug("Restoring environment: %s=%s", $envkey, $saved_env{ $envkey} );
+ $ENV{ $envkey } = $saved_env{ $envkey };
+ }
+
return 1;
}