X-Git-Url: https://git.camperquake.de/gitweb.cgi?a=blobdiff_plain;f=videosite.pl;h=d87b5984fff28b5e01e19f961dc19a13d263c9c5;hb=c5b3d8327f58f02ce2cdcfba5742a4ca4fa2994d;hp=8f56fa1605c0ccf23063343e6aeb096ca2d9da08;hpb=b8b430497f9015b88686d189838d30267f1c6ac2;p=videosite.git

diff --git a/videosite.pl b/videosite.pl
index 8f56fa1..d87b598 100644
--- a/videosite.pl
+++ b/videosite.pl
@@ -7,12 +7,24 @@
 # which in turn is
 # based on trigger.pl by Wouter Coekaerts <wouter@coekaerts.be>
 
+
+BEGIN {
+    # Get rid of a (possibly old) version of BettIrssi
+    # This is a hack to prevent having to reload irssi just
+    # because BettIrssi.pm changed
+
+    delete($INC{'BettIrssi.pm'});
+}
+
 use strict;
 use Irssi 20020324 qw (command_bind command_runsub signal_add_first signal_add_last);
 use vars qw($VERSION %IRSSI);
 use XML::Simple;
 use Data::Dumper;
 use File::Spec;
+use File::Temp qw(tempfile);
+use BettIrssi 101 qw(_bcb _bcs);
+use LWP::UserAgent;
 
 my @grabbers;
 my @getters;
@@ -21,6 +33,7 @@ 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' => '',
@@ -32,23 +45,34 @@ my $PARAMS = {
 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
+# This does not use BettIrssi (yet)
 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();
@@ -91,7 +115,7 @@ my $videosite_commands = {
         foreach (@grabbers, @getters) {
             $_->setdebug(1);
         }
-        write_irssi(undef, 'Enabled debugging');
+        write_irssi('Enabled debugging');
     },
 
     'nodebug' => sub {
@@ -99,91 +123,155 @@ my $videosite_commands = {
         foreach (@grabbers, @getters) {
             $_->setdebug(0);
         }
-        write_irssi(undef, 'Disabled debugging');
+        write_irssi('Disabled debugging');
     },
 };
 
 sub write_irssi {
-    my $witem = shift;
     my @text = @_;
+    my $output = $outputstack[0];
+
+    my $format = "%%mvideosite: %%n" . shift(@text);
 
-    $text[0] = 'videosite: ' . $text[0];
+    # escape % in parameters from irssi
+    s/%/%%/g foreach @text;
 
-    if (defined $witem) {
-        $witem->print(sprintf(shift(@text), @text), MSGLEVEL_CLIENTCRAP);
+    if (defined $output) {
+        $output->(sprintf($format, @text), MSGLEVEL_CLIENTCRAP);
     } else {
-        Irssi::print(sprintf(shift(@text), @text));
+        Irssi::print(sprintf($format, @text));
     }
 
 }
 
 sub write_debug {
     if ($debug) {
-        write_irssi(shift, @_);
+        write_irssi(@_);
     }
 }
 
+sub expand_url_shortener {
+    my $s = shift;
+    my $os = '';
+    my @urlshortener = (
+        'is\.gd/[[:alnum:]]+',
+        'otf\.me/[[:alnum:]]+',
+        'hel\.me/[[:alnum:]]+',
+        '7ax\.de/[[:alnum:]]+',
+        'ow\.ly/[[:alnum:]]+',
+        'j\.mp/[[:alnum:]]+',
+        'bit\.ly/[[:alnum:]]+',
+        'tinyurl\.com/[[:alnum:]]+',
+        'pop\.is/[[:alnum:]]+',
+        'post\.ly/[[:alnum:]]+',
+        '1\.ly/[[:alnum:]]+',
+        '2\.ly/[[:alnum:]]+',
+        't\.co/[[:alnum:]]+',
+        'shar\.es/[[:alnum:]]+',
+        'goo\.gl/[[:alnum:]]+',
+        );
+    my $ua = LWP::UserAgent->new(agent => 'Mozilla', max_redirect => 0);
+    my $i = 10;
+
+    OUTER: while (($os ne $s) and ($i > 0)) {
+        study($s);
+        $os = $s;
+        $i--;
+
+        foreach my $pattern (@urlshortener) {
+            my $p = "https?:\/\/" . $pattern;
+
+            write_debug("Matching %s against %s", $p, $s);
+            if ($s =~ m|($p)|) {
+                my $matched = $1;
+                my $res;
+
+                write_debug("Found %s", $matched);
+                $res = $ua->head($matched);
+                if ($res->is_redirect()) {
+                    my $new = $res->headers()->header("Location");
+
+                    write_debug("Replacing %s with %s", $matched, $new);
+                    $s =~ s/$matched/$new/;
+                    next OUTER;
+                } else {
+                    write_debug("Error resolving %s", $matched);
+                }
+            }
+        }
+    }
+
+    if ($i == 0) {
+        write_debug("Loop terminated by counter");
+    }
+
+    write_debug("Final string: %s", $s);
+
+    return $s;
+}
+
+
 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(?:\s|$),) {
         return;
     }
 
+    push_output($event->ewpf);
+    $message = expand_url_shortener($message);
+
     study($message);
 
     # Offer the message to all Grabbers in turn
-    foreach $g (@grabbers) {
+    GRABBER: foreach $g (@grabbers) {
         ($m, $p) = $g->get($message);
         while (defined($m)) {
-            write_debug($witem, 'Metadata: %s', Dumper($m));
+            write_debug('Metadata: %s', Dumper($m));
             if ('download' eq ($conf->{'videosite'}->{'mode'})) {
-                write_irssi($witem, '%%R>>> %%NSaving %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
+                write_irssi('%%R>>> %%NSaving %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
                 unless($getter->get($m)) {
-                    write_irssi($witem, '%%R>>> FAILED');
+                    write_irssi('%%R>>> FAILED');
                 }
             } elsif ('display' eq ($conf->{'videosite'}->{'mode'})) {
-                write_irssi($witem, '%%M>>> %%NSaw %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
+                write_irssi('%%M>>> %%NSaw %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
             } else {
-                write_irssi($witem, '%%R>>> Invalid operation mode');
+                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);
+            last GRABBER if ($message =~ /^\s*$/);
 
             ($m, $p) = $g->get($message);
         }
     }
+
+    pop_output();
 }
 
 sub cmd_save {
 
+
     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);
+        my ($tempfile, $tempfn) = tempfile("videosite.xml.XXXXXX", dir => Irssi::get_irssi_dir());
+        print $tempfile XML::Simple::XMLout($conf, KeepRoot => 1, KeyAttr => {'config' => 'module', 'option' => 'key'});
+        close($tempfile);
+        rename($tempfn, $conffile);
     };
     if ($@) {
-        write_irssi(undef, 'Could not save config to %s: %s', ($conffile, $@));
+        write_irssi('Could not save config to %s: %s', ($conffile, $@));
     } else {
-        write_irssi(undef, 'configuration saved to %s', $conffile);
+        write_irssi('configuration saved to %s', $conffile);
     }
 }
 
@@ -199,7 +287,7 @@ sub cmd_set {
             return;
         }
     }
-    write_irssi(undef, 'No such module');
+    write_irssi('No such module');
 }
 
 
@@ -213,7 +301,7 @@ sub cmd_enable {
             return;
         }
     }
-    write_irssi(undef, 'No such module');
+    write_irssi('No such module');
 }
 
 
@@ -227,7 +315,7 @@ sub cmd_disable {
             return;
         }
     }
-    write_irssi(undef, 'No such module');
+    write_irssi('No such module');
 }
 
 
@@ -239,21 +327,21 @@ sub cmd_show {
     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 (* denotes enabled modules):');
+        write_irssi('Loaded grabbers (* denotes enabled modules):');
         foreach $p (@grabbers) {
             $e = $p->_getval('enabled');
-            write_irssi(undef, ' %s%s', $p->{'NAME'}, $e?'*':'');
+            write_irssi(' %s%s', $p->{'NAME'}, $e?'*':'');
         };
 
-        write_irssi(undef, 'Loaded getters:');
+        write_irssi('Loaded getters:');
         foreach $p (@getters) {
-            write_irssi(undef, ' %s', $p->{'NAME'});
+            write_irssi(' %s', $p->{'NAME'});
         };
     }
 }
@@ -265,17 +353,17 @@ 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, <<'EOT');
+        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
+ 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)
@@ -297,12 +385,13 @@ sub cmd_getter {
             if ($p->{'NAME'} eq $target) {
                 $getter = $p;
                 $conf->{'videosite'}->{'getter'} = $target;
+                write_irssi("Getter changed to %s", $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'});
     }
 }
 
@@ -313,11 +402,12 @@ sub cmd_mode {
         $mode = lc($mode);
         if (('download' eq $mode) or ('display' eq $mode)) {
             $conf->{'videosite'}->{'mode'} = $mode;
+            write_irssi('Now using %s mode', $mode);
         } else {
-            write_irssi(undef, 'Invalid mode: %s', $mode);
+            write_irssi('Invalid mode: %s', $mode);
         }
     } else {
-        write_irssi(undef, 'Current mode: %s', $conf->{'videosite'}->{'mode'});
+        write_irssi('Current mode: %s', $conf->{'videosite'}->{'mode'});
     }
 }
 
@@ -345,32 +435,32 @@ 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 videosite::$p; };
         if ($@) {
-            write_irssi(undef, "Failed to load plugin: $@");
+            write_irssi("Failed to load plugin: $@");
             next;
         }
 
         $g = eval qq{ videosite::$p->new(); };
         if ($@) {
-            write_irssi(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(sub {Irssi::print(shift)});
+            $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;
 }
@@ -381,7 +471,7 @@ sub _load_modules($) {
 
     foreach (keys(%INC)) {
         if ($INC{$_} =~ m|^$path|) {
-            write_debug(undef, "Removing %s from \$INC", $_);
+            write_debug("Removing %s from \$INC", $_);
             delete($INC{$_});
         }
     }
@@ -397,7 +487,7 @@ sub init_videosite {
 
     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');
+        write_debug('No config found, using defaults');
         $conf = { 'videosite' => { }};
     }
     foreach (keys(%{$PARAMS})) {
@@ -409,7 +499,7 @@ sub init_videosite {
     _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;
     }
 
@@ -419,7 +509,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
@@ -434,10 +524,10 @@ sub init_videosite {
         Irssi::signal_add('setup saved', 'cmd_save');
 
 
-        Irssi::command_bind('videosite' => \&cmdhandler);
+        Irssi::command_bind(_bcb('videosite' => \&cmdhandler));
     }
 
-    write_irssi(undef, 'videosite initialized');
+    write_irssi('initialized successfully');
 }
 
 sub sig_complete {
@@ -485,12 +575,16 @@ sub sig_complete {
 }
 
 sub cmdhandler {
-    my ($data, $server, $item) = @_;
-    my ($cmd, @params) = split(/\s+/, $data);
+    my $event = shift;
+    my ($cmd, @params) = split(/\s+/, $event->message());
+
+    push_output($event->ewpf);
 
     if (exists($videosite_commands->{$cmd})) {
         $videosite_commands->{$cmd}->(@params);
     }
+
+    pop_output();
 }
 
 unshift(@INC, $scriptdir);