videosite-dl.pl

   1 #!/usr/bin/perl -w
   2
   3 use strict;
   4 use Getopt::Long;
   5 use File::Spec;
   6 use File::Basename;
   7 use Cwd qw(realpath);
   8
   9 sub ploader {
  10
  11     my $dir = shift;
  12     my $pattern = shift;
  13     my $type = shift;
  14     my @list;
  15     my $p;
  16     my $g;
  17     my @g = ();
  18
  19     unshift(@INC, $dir);
  20
  21     opendir(D, $dir) || return ();
  22     @list = grep {/$pattern/ && -f File::Spec->catfile($dir, $_) } readdir(D);
  23     closedir(D);
  24
  25     foreach $p (@list) {
  26         $p =~ s/\.pm$//;
  27         eval qq{ require videosite::$p; };
  28         if ($@) {
  29             print("Failed to load plugin: $@");
  30             next;
  31         }
  32
  33         $g = eval qq{ videosite::$p->new();};
  34         if ($@) {
  35             print("Failed to instanciate: $@");
  36             delete($INC{$p});
  37             next;
  38         }
  39
  40         if ($type eq $g->{'TYPE'}) {
  41             push(@g, $g);
  42         } else {
  43             printf('%s has wrong type (got %s, expected %s)', $p, $g->{'TYPE'}, $type);
  44             delete($INC{$p});
  45         }
  46     }
  47
  48     return @g;
  49 }
  50
  51 my $hq = 0;
  52 my $ext = '.flv';
  53 my $y;
  54 my $f;
  55 my $m;
  56 my @g;
  57 my $bp;
  58 my $info = 0;
  59 my $debug = 0;
  60
  61 GetOptions("i" => \$info, "d" => \$debug);
  62
  63 # This is some dark magic to find out our real base directory,
  64 # where we hope to find our plugins.
  65 $bp = File::Spec->catdir(dirname(realpath($0)), 'videosite');
  66 unshift(@INC, dirname(realpath($0)));
  67
  68 @g = ploader($bp, '.*Grabber\.pm$', 'grabber');
  69 ($f) = ploader($bp, '^FileGetter\.pm$', 'getter');
  70
  71 unless(@g and defined($f)) {
  72     print("No plugins could be loaded\n");
  73     exit 1;
  74 }
  75
  76 foreach (@g, $f) {
  77     $_->setio(sub { printf(@_); print("\n"); } );
  78
  79     if ($debug) {
  80         $_->setdebug(1);
  81     }
  82 }
  83
  84 $f->setval('FILEPATTERN', './%3$s' . $ext);
  85
  86 foreach (@ARGV) {
  87     foreach $y (@g) {
  88         ($m, undef) = $y->get($_);
  89         if (defined($m)) {
  90             if ($info) {
  91                 foreach (keys(%{$m})) {
  92                     printf("%s: %s\n", $_, defined($m->{$_})?$m->{$_}:'(undef)');
  93                 }
  94             } else {
  95                 print("Downloading $m->{'TITLE'}\n");
  96                 $f->get($m);
  97             }
  98         }
  99     }
 100 }