X-Git-Url: https://git.camperquake.de/gitweb.cgi?a=blobdiff_plain;f=videosite.pl;h=ec41009d2a5310862358088f96961911fc7f17e8;hb=12db405291947c6b02c2555051c3dc52f3995cbc;hp=35c0809371becc31aab39e3925685afa568e21fd;hpb=0e960b95a2d7eadb3c748ff82035f33670406172;p=videosite.git

diff --git a/videosite.pl b/videosite.pl
index 35c0809..ec41009 100644
--- a/videosite.pl
+++ b/videosite.pl
@@ -1,11 +1,12 @@
 # autodownload flash videos
 #
-# (c) 2007 by Ralf Ertzinger <ralf@camperquake.de>
+# (c) 2007-2008 by Ralf Ertzinger <ralf@camperquake.de>
 # licensed under GNU GPL v2
 #
 # Based on youtube.pl by Christian Garbs <mitch@cgarbs.de>
 # which in turn is
 # based on trigger.pl by Wouter Coekaerts <wouter@coekaerts.be>
+# This is helena
 
 use strict;
 use Irssi 20020324 qw (command_bind command_runsub signal_add_first signal_add_last);
@@ -13,92 +14,186 @@ use vars qw($VERSION %IRSSI);
 use XML::Simple;
 use Data::Dumper;
 use File::Spec;
+delete($INC{'BettIrssi.pm'});
+use BettIrssi qw(_bcb _bcs);
 
 my @grabbers;
 my @getters;
 my $getter;
 my $conf;
 my $conffile = File::Spec->catfile(Irssi::get_irssi_dir(), 'videosite.xml');
+my $scriptdir = File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts');
+my $plugindir = File::Spec->catfile($scriptdir, 'videosite');
+my @outputstack = (undef);
+
+my $PARAMS = {
+    'getter' => '',
+    'mode' => 'download'
+};
+
 
 # activate debug here
 my $debug = 0;
 
 # "message public", SERVER_REC, char *msg, char *nick, char *address, char *target
-signal_add_last("message public" => sub {check_for_link(\@_,1,4,2,0);});
+signal_add_last(_bcs("message public" => sub {check_for_link(@_)}));
 # "message own_public", SERVER_REC, char *msg, char *target
-signal_add_last("message own_public" => sub {check_for_link(\@_,1,2,-1,0);});
+signal_add_last(_bcs("message own_public" => sub {check_for_link(@_)}));
 
 # "message private", SERVER_REC, char *msg, char *nick, char *address
-signal_add_last("message private" => sub {check_for_link(\@_,1,-1,2,0);});
+signal_add_last(_bcs("message private" => sub {check_for_link(@_)}));
 # "message own_private", SERVER_REC, char *msg, char *target, char *orig_target
-signal_add_last("message own_private" => sub {check_for_link(\@_,1,2,-1,0);});
+signal_add_last(_bcs("message own_private" => sub {check_for_link(@_)}));
 
 # "message irc action", SERVER_REC, char *msg, char *nick, char *address, char *target
-signal_add_last("message irc action" => sub {check_for_link(\@_,1,4,2,0);});
+signal_add_last(_bcs("message irc action" => sub {check_for_link(@_)}));
 # "message irc own_action", SERVER_REC, char *msg, char *target
-signal_add_last("message irc own_action" => sub {check_for_link(\@_,1,2,-1,0);});
+signal_add_last(_bcs("message irc own_action" => sub {check_for_link(@_)}));
+
+# For tab completion
+signal_add_first('complete word', \&sig_complete);
+
+sub push_output {
+    unshift(@outputstack, shift);
+}
+
+sub pop_output {
+    shift(@outputstack);
+
+    @outputstack = (undef) unless (@outputstack);
+}
+
+my $videosite_commands = {
+    'save' => sub {
+        cmd_save();
+    },
+
+    'set' => sub {
+        cmd_set(@_);
+    },
+    
+    'show' => sub {
+        cmd_show(@_);
+    },
+
+    'help' => sub {
+        cmd_help(@_);
+    },
+
+    'getter' => sub {
+        cmd_getter(@_);
+    },
+
+    'enable' => sub {
+        cmd_enable(@_);
+    },
+
+    'disable' => sub {
+        cmd_disable(@_);
+    },
+
+    'reload' => sub {
+        init_videosite(0);
+    },
+
+    'mode' => sub {
+        cmd_mode(@_);
+    },
+
+    'debug' => sub {
+        $debug = 1;
+        foreach (@grabbers, @getters) {
+            $_->setdebug(1);
+        }
+        write_irssi('Enabled debugging');
+    },
+
+    'nodebug' => sub {
+        $debug = 0;
+        foreach (@grabbers, @getters) {
+            $_->setdebug(0);
+        }
+        write_irssi('Disabled debugging');
+    },
+};
 
 sub write_irssi {
-    my $witem = shift;
     my @text = @_;
+    my $output = $outputstack[0];
 
-    if (defined $witem) {
-        $witem->print(sprintf(shift(@text), @text), MSGLEVEL_CLIENTCRAP);
-    } else {
-        Irssi::print(sprintf(shift(@text), @text));
-    }
+    my $format = "%%mvideosite: %%n" . shift(@text);
 
-}
+    # escape % in parameters from irssi
+    s/%/%%/g foreach @text;
 
-sub write_verbose {
-    if (Irssi::settings_get_bool('youtube_verbose')) {
-        write_irssi(shift, @_);
+    if (defined $output) {
+        $output->(sprintf($format, @text), MSGLEVEL_CLIENTCRAP);
+    } else {
+        Irssi::print(sprintf($format, @text));
     }
+
 }
 
 sub write_debug {
     if ($debug) {
-        write_irssi(shift, @_);
+        write_irssi(@_);
     }
 }
 
 sub check_for_link {
-    my ($signal,$parammessage,$paramchannel,$paramnick,$paramserver) = @_;
-    my $server = $signal->[$paramserver];
-    my $target = $signal->[$paramchannel];
-    my $message = ($parammessage == -1) ? '' : $signal->[$parammessage];
+    my $event = shift;
+    my $message = $event->message();
+    my $witem = $event->channel();
     my $g;
     my $m;
+    my $p;
 
 
-    my $witem;
-    if (defined $server) {
-        $witem = $server->window_item_find($target);
-    } else {
-        $witem = Irssi::window_item_find($target);
-    }
-
     # Look if we should ignore this line
-    if ($message =~ m,(?:\s|^)/nosave,) {
+    if ($message =~ m,(?:\s|^)/nosave(?:\s|$),) {
         return;
     }
 
+    study($message);
+
     # Offer the message to all Grabbers in turn
     foreach $g (@grabbers) {
-        if (defined($m = $g->get($message))) {
-            write_irssi($witem, '%%R>>> %%NSaving %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
-            unless($getter->get($m)) {
-                write_irssi($witem, '%%R>>> FAILED');
+        ($m, $p) = $g->get($message);
+        while (defined($m)) {
+            write_debug('Metadata: %s', Dumper($m));
+            if ('download' eq ($conf->{'videosite'}->{'mode'})) {
+                write_irssi('%%R>>> %%NSaving %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
+                unless($getter->get($m)) {
+                    write_irssi('%%R>>> FAILED');
+                }
+            } elsif ('display' eq ($conf->{'videosite'}->{'mode'})) {
+                write_irssi('%%M>>> %%NSaw %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
+            } else {
+                write_irssi('%%R>>> Invalid operation mode');
             }
+
+            # Remove the matched part from the message and try again (there may be
+            # more!)
+            $message =~ s/$p//;
+            study($message);
+
+            ($m, $p) = $g->get($message);
         }
     }
 }
 
 sub cmd_save {
 
-    open(CONF, '>'.$conffile);
-    print CONF XML::Simple::XMLout($conf, KeepRoot => 1, KeyAttr => ['module', 'key']);
-    close(CONF);
+    eval {
+        open(CONF, '>'.$conffile) or die 'Could not open config file';
+        print CONF XML::Simple::XMLout($conf, KeepRoot => 1, KeyAttr => {'config' => 'module', 'option' => 'key'});
+        close(CONF);
+    };
+    if ($@) {
+        write_irssi('Could not save config to %s: %s', ($conffile, $@));
+    } else {
+        write_irssi('configuration saved to %s', $conffile);
+    }
 }
 
 sub cmd_set {
@@ -113,30 +208,61 @@ sub cmd_set {
             return;
         }
     }
-    write_irssi(undef, 'No such module');
+    write_irssi('No such module');
+}
+
+
+sub cmd_enable {
+    my $target = shift;
+    my $p;
+
+    foreach $p (@grabbers) {
+        if ($p->{'NAME'} eq $target) {
+            $p->enable();
+            return;
+        }
+    }
+    write_irssi('No such module');
+}
+
+
+sub cmd_disable {
+    my $target = shift;
+    my $p;
+
+    foreach $p (@grabbers) {
+        if ($p->{'NAME'} eq $target) {
+            $p->disable();
+            return;
+        }
+    }
+    write_irssi('No such module');
 }
 
+
 sub cmd_show {
     my $target = shift;
     my $p;
+    my $e;
 
     if (defined($target)) {
         foreach $p (@getters, @grabbers) {
             if ($p->{'NAME'} eq $target) {
-                write_irssi(undef, $p->getconfstr());
+                write_irssi($p->getconfstr());
                 return;
             }
         }
-        write_irssi(undef, 'No such module');
+        write_irssi('No such module');
     } else {
-        write_irssi(undef, 'Loaded grabbers:');
+        write_irssi('Loaded grabbers (* denotes enabled modules):');
         foreach $p (@grabbers) {
-            write_irssi(undef, ' ' . $p->{'NAME'});
+            $e = $p->_getval('enabled');
+            write_irssi(' %s%s', $p->{'NAME'}, $e?'*':'');
         };
 
-        write_irssi(undef, 'Loaded getters:');
+        write_irssi('Loaded getters:');
         foreach $p (@getters) {
-            write_irssi(undef, ' ' . $p->{'NAME'});
+            write_irssi(' %s', $p->{'NAME'});
         };
     }
 }
@@ -148,13 +274,26 @@ sub cmd_help {
     if (defined($target)) {
         foreach $p (@getters, @grabbers) {
             if ($p->{'NAME'} eq $target) {
-                write_irssi(undef, $p->gethelpstr());
+                write_irssi($p->gethelpstr());
                 return;
             }
         }
-        write_irssi(undef, 'No such module');
+        write_irssi('No such module');
     } else {
-        write_irssi(undef, "Supported commands:\n save: Save the current configuration\n help [modulename]: Display this help, or module specific help\n show [modulename]: Show loaded modules, or the current parameters of a module\n set modulename parameter value: set a module parameter to a new value\n getter [modulename]: display or set the getter to use");
+        write_irssi(<<'EOT');
+Supported commands:
+ save: save the current configuration
+ help [modulename]: display this help, or module specific help
+ show [modulename]: show loaded modules, or the current parameters of a module
+ set modulename parameter value: set a module parameter to a new value
+ getter [modulename]: display or set the getter to use
+ enable [modulename]: enable the usage of this module (grabbers only)
+ disable [modulename]: disable the usage of this module (grabbers only)
+ reload: reload all modules (this is somewhat experimental)
+ mode [modename]: display or set the operation mode (download/display)
+ debug: enable debugging messages
+ nodebug: disable debugging messages
+EOT
     }
 }
 
@@ -163,16 +302,31 @@ sub cmd_getter {
     my $p;
 
     if (defined($target)) {
-        foreach $p (@getters, @grabbers) {
-            if (($p->{'NAME'} eq $target) && ($p->{'TYPE'} eq 'getter')) {
+        foreach $p (@getters) {
+            if ($p->{'NAME'} eq $target) {
                 $getter = $p;
                 $conf->{'videosite'}->{'getter'} = $target;
                 return;
             }
         }
-        write_irssi(undef, 'No such getter');
+        write_irssi('No such getter');
     } else {
-        write_irssi(undef, 'Current getter: %s', $conf->{'videosite'}->{'getter'});
+        write_irssi('Current getter: %s', $conf->{'videosite'}->{'getter'});
+    }
+}
+
+sub cmd_mode {
+    my $mode = shift;
+
+    if (defined($mode)) {
+        $mode = lc($mode);
+        if (('download' eq $mode) or ('display' eq $mode)) {
+            $conf->{'videosite'}->{'mode'} = $mode;
+        } else {
+            write_irssi('Invalid mode: %s', $mode);
+        }
+    } else {
+        write_irssi('Current mode: %s', $conf->{'videosite'}->{'mode'});
     }
 }
 
@@ -200,49 +354,71 @@ sub ploader {
     closedir(D);
 
     foreach $p (@list) {
-        write_debug(undef, "Trying to load $p:");
+        write_debug("Trying to load $p:");
         $p =~ s/\.pm$//;
-        eval qq{ require $p; };
+        eval qq{ require videosite::$p; };
         if ($@) {
-            write_debug(undef, "Failed to load plugin: $@");
+            write_irssi("Failed to load plugin: $@");
             next;
         }
 
-        $g = eval $p.q{->new();};
+        $g = eval qq{ videosite::$p->new(); };
         if ($@) {
-            write_debug(undef, "Failed to instanciate: $@");
+            write_irssi("Failed to instanciate: $@");
+            delete($INC{$p});
             next;
         }
 
-        write_debug(undef, "found $g->{'TYPE'} $g->{'NAME'}");
+        write_debug("found $g->{'TYPE'} $g->{'NAME'}");
         if ($type eq $g->{'TYPE'}) {
             push(@g, $g);
+            $g->setio(\&write_irssi);
         } else {
-            write_irssi(undef, '%s has wrong type (got %s, expected %s)', $p, $g->{'TYPE'}, $type);
+            write_irssi('%s has wrong type (got %s, expected %s)', $p, $g->{'TYPE'}, $type);
+            delete($INC{$p});
         }
     }
 
-    write_debug(undef, "Loaded %d plugins", $#g+1);
+    write_debug("Loaded %d plugins", $#g+1);
     
     return @g;
 }
 
+sub _load_modules($) {
+
+    my $path = shift;
+
+    foreach (keys(%INC)) {
+        if ($INC{$_} =~ m|^$path|) {
+            write_debug("Removing %s from \$INC", $_);
+            delete($INC{$_});
+        }
+    }
+    @grabbers = ploader($path, '.*Grabber\.pm$', 'grabber');
+    @getters = ploader($path, '.*Getter\.pm$', 'getter');
+}
+
+
 sub init_videosite {
 
+    my $bindings = shift;
     my $p;
 
-    unless(-r $conffile && defined($conf = XML::Simple::XMLin($conffile, ForceArray => ['config', 'option'], KeepRoot => 1, KeyAttr => ['module', 'key']))) {
+    unless(-r $conffile && defined($conf = XML::Simple::XMLin($conffile, ForceArray => ['config', 'option'], KeepRoot => 1, KeyAttr => {'config' => 'module', 'option' => 'key'}))) {
         # No config, start with an empty one
-        write_debug(undef, 'No config found, using defaults');
-        $conf = { 'videosite' => { 'getter' => '' }};
+        write_debug('No config found, using defaults');
+        $conf = { 'videosite' => { }};
+    }
+    foreach (keys(%{$PARAMS})) {
+        unless (exists($conf->{'videosite'}->{$_})) {
+            $conf->{'videosite'}->{$_} = $PARAMS->{$_};
+        }
     }
 
-    unshift(@INC, File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts', 'videosite'));
-    @grabbers = ploader(File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts', 'videosite'), '.*Grabber\.pm$', 'grabber');
-    @getters = ploader(File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts', 'videosite'), '.*Getter\.pm$', 'getter');
+    _load_modules($plugindir);
 
     unless (defined(@grabbers) && defined(@getters)) {
-        write_irssi(undef, 'No grabbers or no getters found, can not proceed.');
+        write_irssi('No grabbers or no getters found, can not proceed.');
         return;
     }
 
@@ -252,7 +428,7 @@ sub init_videosite {
             $getter = $p;
         }
     }
-    write_debug(undef, 'Selected %s as getter', $getter->{'NAME'});
+    write_debug('Selected %s as getter', $getter->{'NAME'});
     $conf->{'videosite'}->{'getter'} = $getter->{'NAME'};
 
     # Loop through all plugins and load the config
@@ -260,43 +436,71 @@ sub init_videosite {
         $conf->{'videosite'}->{'config'}->{$p->{'NAME'}} = $p->mergeconfig($conf->{'videosite'}->{'config'}->{$p->{'NAME'}});
     }
 
-    Irssi::signal_add_first('command script load', 'sig_command_script_unload');
-    Irssi::signal_add_first('command script unload', 'sig_command_script_unload');
+    if ($bindings) {
+
+        Irssi::signal_add_first('command script load', 'sig_command_script_unload');
+        Irssi::signal_add_first('command script unload', 'sig_command_script_unload');
+        Irssi::signal_add('setup saved', 'cmd_save');
 
-    Irssi::command_bind('videosite' => \&cmdhandler);
+
+        Irssi::command_bind(_bcb('videosite' => \&cmdhandler));
+    }
+
+    write_irssi('initialized successfully');
 }
 
-sub cmdhandler {
-    my ($data, $server, $item) = @_;
-    my @params = split(/\s+/, $data);
+sub sig_complete {
+    my ($complist, $window, $word, $linestart, $want_space) = @_;
+    my @matches;
 
-    if ($params[0] eq 'save') {
-        cmd_save();
-    } elsif ($params[0] eq 'set') {
-        shift(@params);
-        cmd_set(@params);
-    } elsif ($params[0] eq 'show') {
-        shift(@params);
-        cmd_show(@params);
-    } elsif ($params[0] eq 'help') {
-        shift(@params);
-        cmd_help(@params);
-    } elsif ($params[0] eq 'getter') {
-        shift(@params);
-        cmd_getter(@params);
-    } elsif ($params[0] eq 'debug') {
-        $debug = 1;
+    if ($linestart !~ m|^/videosite\b|) {
+        return;
+    }
+
+    if ('/videosite' eq $linestart) {
+        # No command enterd so far. Produce a list of possible follow-ups
+        @matches = grep {/^$word/} keys (%{$videosite_commands});
+    } elsif ('/videosite set' eq $linestart) {
+        # 'set' command entered. Produce a list of modules
         foreach (@grabbers, @getters) {
-            $_->setdebug(1);
+            push(@matches, $_->{'NAME'}) if $_->{'NAME'} =~ m|^$word|;
+        };
+    } elsif ($linestart =~ m|^/videosite set (\w+)$|) {
+        my $module = $1;
+
+        foreach my $p (@getters, @grabbers) {
+            if ($p->{'NAME'} eq $module) {
+                @matches = $p->getparamlist($word);
+                last;
+            }
         }
-        write_irssi(undef, 'Enabled debugging');
-    } elsif ($params[0] eq 'nodebug') {
-        $debug = 0;
-        foreach (@grabbers, @getters) {
-            $_->setdebug(0);
+    } elsif ($linestart =~ m|/videosite set (\w+) (\w+)$|) {
+        my $module = $1;
+        my $param = $2;
+
+        foreach my $p (@getters, @grabbers) {
+            if ($p->{'NAME'} eq $module) {
+                @matches = $p->getparamvalues($param, $word);
+                last;
+            }
         }
-        write_irssi(undef, 'Disabled debugging');
+    }
+
+
+    push(@{$complist}, sort @matches);
+    ${$want_space} = 0;
+
+    Irssi::signal_stop();
+}
+
+sub cmdhandler {
+    my $event = shift;
+    my ($cmd, @params) = split(/\s+/, $event->message());
+
+    if (exists($videosite_commands->{$cmd})) {
+        $videosite_commands->{$cmd}->(@params);
     }
 }
 
-init_videosite();
+unshift(@INC, $scriptdir);
+init_videosite(1);