videosite.pl

   1 # autodownload flash videos
   2 #
   3 # (c) 2007-2008 by Ralf Ertzinger <ralf@camperquake.de>
   4 # licensed under GNU GPL v2
   5 #
   6 # Based on youtube.pl by Christian Garbs <mitch@cgarbs.de>
   7 # which in turn is
   8 # based on trigger.pl by Wouter Coekaerts <wouter@coekaerts.be>
   9
  10 use strict;
  11 use Irssi 20020324 qw (command_bind command_runsub signal_add_first signal_add_last);
  12 use vars qw($VERSION %IRSSI);
  13 use XML::Simple;
  14 use Data::Dumper;
  15 use File::Spec;
  16
  17 my @grabbers;
  18 my @getters;
  19 my $getter;
  20 my $conf;
  21 my $conffile = File::Spec->catfile(Irssi::get_irssi_dir(), 'videosite.xml');
  22
  23 # activate debug here
  24 my $debug = 0;
  25
  26 # "message public", SERVER_REC, char *msg, char *nick, char *address, char *target
  27 signal_add_last("message public" => sub {check_for_link(\@_,1,4,2,0);});
  28 # "message own_public", SERVER_REC, char *msg, char *target
  29 signal_add_last("message own_public" => sub {check_for_link(\@_,1,2,-1,0);});
  30
  31 # "message private", SERVER_REC, char *msg, char *nick, char *address
  32 signal_add_last("message private" => sub {check_for_link(\@_,1,-1,2,0);});
  33 # "message own_private", SERVER_REC, char *msg, char *target, char *orig_target
  34 signal_add_last("message own_private" => sub {check_for_link(\@_,1,2,-1,0);});
  35
  36 # "message irc action", SERVER_REC, char *msg, char *nick, char *address, char *target
  37 signal_add_last("message irc action" => sub {check_for_link(\@_,1,4,2,0);});
  38 # "message irc own_action", SERVER_REC, char *msg, char *target
  39 signal_add_last("message irc own_action" => sub {check_for_link(\@_,1,2,-1,0);});
  40
  41 sub write_irssi {
  42     my $witem = shift;
  43     my @text = @_;
  44
  45     $text[0] = 'videosite: ' . $text[0];
  46
  47     if (defined $witem) {
  48         $witem->print(sprintf(shift(@text), @text), MSGLEVEL_CLIENTCRAP);
  49     } else {
  50         Irssi::print(sprintf(shift(@text), @text));
  51     }
  52
  53 }
  54
  55 sub write_debug {
  56     if ($debug) {
  57         write_irssi(shift, @_);
  58     }
  59 }
  60
  61 sub check_for_link {
  62     my ($signal,$parammessage,$paramchannel,$paramnick,$paramserver) = @_;
  63     my $server = $signal->[$paramserver];
  64     my $target = $signal->[$paramchannel];
  65     my $message = ($parammessage == -1) ? '' : $signal->[$parammessage];
  66     my $g;
  67     my $m;
  68
  69
  70     my $witem;
  71     if (defined $server) {
  72         $witem = $server->window_item_find($target);
  73     } else {
  74         $witem = Irssi::window_item_find($target);
  75     }
  76
  77     # Look if we should ignore this line
  78     if ($message =~ m,(?:\s|^)/nosave(?:\s|$),) {
  79         return;
  80     }
  81
  82     # Offer the message to all Grabbers in turn
  83     foreach $g (@grabbers) {
  84         if (defined($m = $g->get($message))) {
  85             write_debug($witem, 'Metadata: %s', Dumper($m));
  86             write_irssi($witem, '%%R>>> %%NSaving %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
  87             unless($getter->get($m)) {
  88                 write_irssi($witem, '%%R>>> FAILED');
  89             }
  90         }
  91     }
  92 }
  93
  94 sub cmd_save {
  95
  96     eval {
  97         open(CONF, '>'.$conffile) or die 'Could not open config file';
  98         print CONF XML::Simple::XMLout($conf, KeepRoot => 1, KeyAttr => {'config' => 'module', 'option' => 'key'});
  99         close(CONF);
 100     };
 101     if ($@) {
 102         write_irssi(undef, 'Could not save config to %s: %s', ($conffile, $@));
 103     } else {
 104         write_irssi(undef, 'configuration saved to %s', $conffile);
 105     }
 106 }
 107
 108 sub cmd_set {
 109     my $target = shift;
 110     my $key = shift;
 111     my $val = shift;
 112     my $p;
 113
 114     foreach $p (@getters, @grabbers) {
 115         if ($p->{'NAME'} eq $target) {
 116             $p->setval($key, $val);
 117             return;
 118         }
 119     }
 120     write_irssi(undef, 'No such module');
 121 }
 122
 123 sub cmd_show {
 124     my $target = shift;
 125     my $p;
 126
 127     if (defined($target)) {
 128         foreach $p (@getters, @grabbers) {
 129             if ($p->{'NAME'} eq $target) {
 130                 write_irssi(undef, $p->getconfstr());
 131                 return;
 132             }
 133         }
 134         write_irssi(undef, 'No such module');
 135     } else {
 136         write_irssi(undef, 'Loaded grabbers:');
 137         foreach $p (@grabbers) {
 138             write_irssi(undef, ' ' . $p->{'NAME'});
 139         };
 140
 141         write_irssi(undef, 'Loaded getters:');
 142         foreach $p (@getters) {
 143             write_irssi(undef, ' ' . $p->{'NAME'});
 144         };
 145     }
 146 }
 147
 148 sub cmd_help {
 149     my $target = shift;
 150     my $p;
 151
 152     if (defined($target)) {
 153         foreach $p (@getters, @grabbers) {
 154             if ($p->{'NAME'} eq $target) {
 155                 write_irssi(undef, $p->gethelpstr());
 156                 return;
 157             }
 158         }
 159         write_irssi(undef, 'No such module');
 160     } else {
 161         write_irssi(undef, <<'EOT');
 162 Supported commands:
 163  save: Save the current configuration
 164  help [modulename]: Display this help, or module specific help
 165  show [modulename]: Show loaded modules, or the current parameters of a module
 166  set modulename parameter value: set a module parameter to a new value
 167  getter [modulename]: display or set the getter to use
 168  debug: enable debugging messages
 169  nodebug: disable debugging messages
 170 EOT
 171     }
 172 }
 173
 174 sub cmd_getter {
 175     my $target = shift;
 176     my $p;
 177
 178     if (defined($target)) {
 179         foreach $p (@getters, @grabbers) {
 180             if (($p->{'NAME'} eq $target) && ($p->{'TYPE'} eq 'getter')) {
 181                 $getter = $p;
 182                 $conf->{'videosite'}->{'getter'} = $target;
 183                 return;
 184             }
 185         }
 186         write_irssi(undef, 'No such getter');
 187     } else {
 188         write_irssi(undef, 'Current getter: %s', $conf->{'videosite'}->{'getter'});
 189     }
 190 }
 191
 192
 193 # save on unload
 194 sub sig_command_script_unload {
 195     my $script = shift;
 196     if ($script =~ /(.*\/)?videosite(\.pl)?$/) {
 197         cmd_save();
 198     }
 199 }
 200
 201 sub ploader {
 202
 203     my $dir = shift;
 204     my $pattern = shift;
 205     my $type = shift;
 206     my @list;
 207     my $p;
 208     my $g;
 209     my @g = ();
 210
 211     opendir(D, $dir) || return ();
 212     @list = grep {/$pattern/ && -f File::Spec->catfile($dir, $_) } readdir(D);
 213     closedir(D);
 214
 215     foreach $p (@list) {
 216         write_debug(undef, "Trying to load $p:");
 217         $p =~ s/\.pm$//;
 218         eval qq{ require $p; };
 219         if ($@) {
 220             write_debug(undef, "Failed to load plugin: $@");
 221             next;
 222         }
 223
 224         $g = eval $p.q{->new();};
 225         if ($@) {
 226             write_debug(undef, "Failed to instanciate: $@");
 227             next;
 228         }
 229
 230         write_debug(undef, "found $g->{'TYPE'} $g->{'NAME'}");
 231         if ($type eq $g->{'TYPE'}) {
 232             push(@g, $g);
 233         } else {
 234             write_irssi(undef, '%s has wrong type (got %s, expected %s)', $p, $g->{'TYPE'}, $type);
 235         }
 236     }
 237
 238     write_debug(undef, "Loaded %d plugins", $#g+1);
 239
 240     return @g;
 241 }
 242
 243 sub init_videosite {
 244
 245     my $p;
 246
 247     unless(-r $conffile && defined($conf = XML::Simple::XMLin($conffile, ForceArray => ['config', 'option'], KeepRoot => 1, KeyAttr => {'config' => 'module', 'option' => 'key'}))) {
 248         # No config, start with an empty one
 249         write_debug(undef, 'No config found, using defaults');
 250         $conf = { 'videosite' => { 'getter' => '' }};
 251     }
 252
 253     unshift(@INC, File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts', 'videosite'));
 254     @grabbers = ploader(File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts', 'videosite'), '.*Grabber\.pm$', 'grabber');
 255     @getters = ploader(File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts', 'videosite'), '.*Getter\.pm$', 'getter');
 256
 257     unless (defined(@grabbers) && defined(@getters)) {
 258         write_irssi(undef, 'No grabbers or no getters found, can not proceed.');
 259         return;
 260     }
 261
 262     $getter = $getters[0];
 263     foreach $p (@getters) {
 264         if ($conf->{'videosite'}->{'getter'} eq $p->{'NAME'}) {
 265             $getter = $p;
 266         }
 267     }
 268     write_debug(undef, 'Selected %s as getter', $getter->{'NAME'});
 269     $conf->{'videosite'}->{'getter'} = $getter->{'NAME'};
 270
 271     # Loop through all plugins and load the config
 272     foreach $p (@grabbers, @getters) {
 273         $conf->{'videosite'}->{'config'}->{$p->{'NAME'}} = $p->mergeconfig($conf->{'videosite'}->{'config'}->{$p->{'NAME'}});
 274     }
 275
 276     Irssi::signal_add_first('command script load', 'sig_command_script_unload');
 277     Irssi::signal_add_first('command script unload', 'sig_command_script_unload');
 278     Irssi::signal_add('setup saved', 'cmd_save');
 279
 280
 281     Irssi::command_bind('videosite' => \&cmdhandler);
 282 }
 283
 284 sub cmdhandler {
 285     my ($data, $server, $item) = @_;
 286     my @params = split(/\s+/, $data);
 287
 288     if ($params[0] eq 'save') {
 289         cmd_save();
 290     } elsif ($params[0] eq 'set') {
 291         shift(@params);
 292         cmd_set(@params);
 293     } elsif ($params[0] eq 'show') {
 294         shift(@params);
 295         cmd_show(@params);
 296     } elsif ($params[0] eq 'help') {
 297         shift(@params);
 298         cmd_help(@params);
 299     } elsif ($params[0] eq 'getter') {
 300         shift(@params);
 301         cmd_getter(@params);
 302     } elsif ($params[0] eq 'debug') {
 303         $debug = 1;
 304         foreach (@grabbers, @getters) {
 305             $_->setdebug(1);
 306         }
 307         write_irssi(undef, 'Enabled debugging');
 308     } elsif ($params[0] eq 'nodebug') {
 309         $debug = 0;
 310         foreach (@grabbers, @getters) {
 311             $_->setdebug(0);
 312         }
 313         write_irssi(undef, 'Disabled debugging');
 314     }
 315 }
 316
 317 init_videosite();