X-Git-Url: https://git.camperquake.de/gitweb.cgi?a=blobdiff_plain;f=videosite.pl;h=3ae1e83fdfe0fd03a82e0475f37226af3a10a5ce;hb=bd9a8a86176b546d3336587a8f28f9c25299f5f6;hp=9be1fee50689715693aff49f53fc53d03dec2e0f;hpb=c52f373793933cf72d90ca23f942c367bc8f32ed;p=videosite.git

diff --git a/videosite.pl b/videosite.pl
index 9be1fee..3ae1e83 100644
--- a/videosite.pl
+++ b/videosite.pl
@@ -1,6 +1,6 @@
 # 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>
@@ -19,6 +19,13 @@ my @getters;
 my $getter;
 my $conf;
 my $conffile = File::Spec->catfile(Irssi::get_irssi_dir(), 'videosite.xml');
+my $plugindir = File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts', 'videosite');
+
+my $PARAMS = {
+    'getter' => '',
+    'mode' => 'download'
+};
+
 
 # activate debug here
 my $debug = 0;
@@ -42,6 +49,8 @@ sub write_irssi {
     my $witem = shift;
     my @text = @_;
 
+    $text[0] = 'videosite: ' . $text[0];
+
     if (defined $witem) {
         $witem->print(sprintf(shift(@text), @text), MSGLEVEL_CLIENTCRAP);
     } else {
@@ -50,12 +59,6 @@ sub write_irssi {
 
 }
 
-sub write_verbose {
-    if (Irssi::settings_get_bool('youtube_verbose')) {
-        write_irssi(shift, @_);
-    }
-}
-
 sub write_debug {
     if ($debug) {
         write_irssi(shift, @_);
@@ -69,6 +72,7 @@ sub check_for_link {
     my $message = ($parammessage == -1) ? '' : $signal->[$parammessage];
     my $g;
     my $m;
+    my $p;
 
 
     my $witem;
@@ -85,20 +89,41 @@ sub check_for_link {
 
     # 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($witem, 'Metadata: %s', Dumper($m));
+            if ('download' eq ($conf->{'videosite'}->{'mode'})) {
+                write_irssi($witem, '%%R>>> %%NSaving %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
+                unless($getter->get($m)) {
+                    write_irssi($witem, '%%R>>> FAILED');
+                }
+            } elsif ('display' eq ($conf->{'videosite'}->{'mode'})) {
+                write_irssi($witem, '%%M>>> %%NSaw %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
+            } else {
+                write_irssi($witem, '%%R>>> Invalid operation mode');
             }
+
+            # Remove the matched part from the message and try again (there may be
+            # more!)
+            $message =~ s/$p//;
+
+            ($m, $p) = $g->get($message);
         }
     }
 }
 
 sub cmd_save {
 
-    open(CONF, '>'.$conffile);
-    print CONF XML::Simple::XMLout($conf, KeepRoot => 1, KeyAttr => {'config' => 'module', 'option' => '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(undef, 'Could not save config to %s: %s', ($conffile, $@));
+    } else {
+        write_irssi(undef, 'configuration saved to %s', $conffile);
+    }
 }
 
 sub cmd_set {
@@ -116,9 +141,39 @@ sub cmd_set {
     write_irssi(undef, 'No such module');
 }
 
+
+sub cmd_enable {
+    my $target = shift;
+    my $p;
+
+    foreach $p (@grabbers) {
+        if ($p->{'NAME'} eq $target) {
+            $p->enable();
+            return;
+        }
+    }
+    write_irssi(undef, 'No such module');
+}
+
+
+sub cmd_disable {
+    my $target = shift;
+    my $p;
+
+    foreach $p (@grabbers) {
+        if ($p->{'NAME'} eq $target) {
+            $p->disable();
+            return;
+        }
+    }
+    write_irssi(undef, 'No such module');
+}
+
+
 sub cmd_show {
     my $target = shift;
     my $p;
+    my $e;
 
     if (defined($target)) {
         foreach $p (@getters, @grabbers) {
@@ -129,14 +184,15 @@ sub cmd_show {
         }
         write_irssi(undef, 'No such module');
     } else {
-        write_irssi(undef, 'Loaded grabbers:');
+        write_irssi(undef, 'Loaded grabbers (* denotes enabled modules):');
         foreach $p (@grabbers) {
-            write_irssi(undef, ' ' . $p->{'NAME'});
+            $e = $p->_getval('enabled');
+            write_irssi(undef, ' %s%s', $p->{'NAME'}, $e?'*':'');
         };
 
         write_irssi(undef, 'Loaded getters:');
         foreach $p (@getters) {
-            write_irssi(undef, ' ' . $p->{'NAME'});
+            write_irssi(undef, ' %s', $p->{'NAME'});
         };
     }
 }
@@ -154,16 +210,20 @@ sub cmd_help {
         }
         write_irssi(undef, '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\n".
-		    " debug: enable debugging messages\n".
-		    " nodebug: disable debugging messages"
-		    );
+        write_irssi(undef, <<'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
     }
 }
 
@@ -185,6 +245,21 @@ sub cmd_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(undef, 'Invalid mode: %s', $mode);
+        }
+    } else {
+        write_irssi(undef, 'Current mode: %s', $conf->{'videosite'}->{'mode'});
+    }
+}
+
 
 # save on unload
 sub sig_command_script_unload {
@@ -220,14 +295,17 @@ sub ploader {
         $g = eval $p.q{->new();};
         if ($@) {
             write_debug(undef, "Failed to instanciate: $@");
+            delete($INC{$p});
             next;
         }
 
         write_debug(undef, "found $g->{'TYPE'} $g->{'NAME'}");
         if ($type eq $g->{'TYPE'}) {
             push(@g, $g);
+            $g->setio(sub {Irssi::print(shift)});
         } else {
             write_irssi(undef, '%s has wrong type (got %s, expected %s)', $p, $g->{'TYPE'}, $type);
+            delete($INC{$p});
         }
     }
 
@@ -236,19 +314,38 @@ sub ploader {
     return @g;
 }
 
+sub _load_modules($) {
+
+    my $path = shift;
+
+    foreach (keys(%INC)) {
+        if ($INC{$_} =~ m|^$path|) {
+            write_debug(undef, "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 => {'config' => 'module', 'option' => 'key'}))) {
         # No config, start with an empty one
         write_debug(undef, 'No config found, using defaults');
-        $conf = { 'videosite' => { 'getter' => '' }};
+        $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.');
@@ -269,10 +366,17 @@ 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::command_bind('videosite' => \&cmdhandler);
+        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);
+    }
+
+    write_irssi(undef, 'videosite initialized');
 }
 
 sub cmdhandler {
@@ -293,6 +397,17 @@ sub cmdhandler {
     } elsif ($params[0] eq 'getter') {
         shift(@params);
         cmd_getter(@params);
+    } elsif ($params[0] eq 'enable') {
+        shift(@params);
+        cmd_enable(@params);
+    } elsif ($params[0] eq 'disable') {
+        shift(@params);
+        cmd_disable(@params);
+    } elsif ($params[0] eq 'reload') {
+        init_videosite(0);
+    } elsif ($params[0] eq 'mode') {
+        shift(@params);
+        cmd_mode(@params);
     } elsif ($params[0] eq 'debug') {
         $debug = 1;
         foreach (@grabbers, @getters) {
@@ -308,4 +423,5 @@ sub cmdhandler {
     }
 }
 
-init_videosite();
+unshift(@INC, $plugindir);
+init_videosite(1);