1 # autodownload flash videos
3 # (c) 2007-2008 by Ralf Ertzinger <ralf@camperquake.de>
4 # licensed under GNU GPL v2
6 # Based on youtube.pl by Christian Garbs <mitch@cgarbs.de>
8 # based on trigger.pl by Wouter Coekaerts <wouter@coekaerts.be>
12 # Get rid of a (possibly old) version of BettIrssi
13 # This is a hack to prevent having to reload irssi just
14 # because BettIrssi.pm changed
16 delete($INC{'BettIrssi.pm'});
20 use Irssi 20020324 qw (command_bind command_runsub signal_add_first signal_add_last);
21 use vars qw($VERSION %IRSSI);
25 use File::Temp qw(tempfile);
26 use BettIrssi 101 qw(_bcb _bcs);
28 use JSON -support_by_pp;
34 my $xmlconffile = File::Spec->catfile(Irssi::get_irssi_dir(), 'videosite.xml');
35 my $conffile = File::Spec->catfile(Irssi::get_irssi_dir(), 'videosite.json');
36 my $scriptdir = File::Spec->catfile(Irssi::get_irssi_dir(), 'scripts');
37 my $plugindir = File::Spec->catfile($scriptdir, 'videosite');
38 my @outputstack = (undef);
43 'connectorlist' => ['direct'],
51 # "message public", SERVER_REC, char *msg, char *nick, char *address, char *target
52 signal_add_last(_bcs("message public" => sub {check_for_link(@_)}));
53 # "message own_public", SERVER_REC, char *msg, char *target
54 signal_add_last(_bcs("message own_public" => sub {check_for_link(@_)}));
56 # "message private", SERVER_REC, char *msg, char *nick, char *address
57 signal_add_last(_bcs("message private" => sub {check_for_link(@_)}));
58 # "message own_private", SERVER_REC, char *msg, char *target, char *orig_target
59 signal_add_last(_bcs("message own_private" => sub {check_for_link(@_)}));
61 # "message irc action", SERVER_REC, char *msg, char *nick, char *address, char *target
62 signal_add_last(_bcs("message irc action" => sub {check_for_link(@_)}));
63 # "message irc own_action", SERVER_REC, char *msg, char *target
64 signal_add_last(_bcs("message irc own_action" => sub {check_for_link(@_)}));
67 # This does not use BettIrssi (yet)
68 signal_add_first('complete word', \&sig_complete);
71 unshift(@outputstack, shift);
77 @outputstack = (undef) unless (@outputstack);
80 my $videosite_commands = {
123 foreach (@grabbers, @getters) {
126 write_irssi('Enabled debugging');
131 foreach (@grabbers, @getters) {
134 write_irssi('Disabled debugging');
140 my $output = $outputstack[0];
142 my $format = "%%mvideosite: %%n" . shift(@text);
144 # escape % in parameters from irssi
145 s/%/%%/g foreach @text;
147 if (defined $output) {
148 $output->(sprintf($format, @text), MSGLEVEL_CLIENTCRAP);
150 Irssi::print(sprintf($format, @text));
161 sub expand_url_shortener {
165 'is\.gd/[[:alnum:]]+',
166 'otf\.me/[[:alnum:]]+',
167 'hel\.me/[[:alnum:]]+',
168 '7ax\.de/[[:alnum:]]+',
169 'ow\.ly/[[:alnum:]]+',
170 'j\.mp/[[:alnum:]]+',
171 'bit\.ly/[[:alnum:]]+',
172 'tinyurl\.com/[[:alnum:]]+',
173 'pop\.is/[[:alnum:]]+',
174 'post\.ly/[[:alnum:]]+',
175 '1\.ly/[[:alnum:]]+',
176 '2\.ly/[[:alnum:]]+',
177 't\.co/[[:alnum:]]+',
178 'shar\.es/[[:alnum:]]+',
179 'goo\.gl/[[:alnum:]]+',
181 my $ua = LWP::UserAgent->new(agent => 'Mozilla', max_redirect => 0, timeout => 5);
184 OUTER: while (($os ne $s) and ($i > 0)) {
189 foreach my $pattern (@urlshortener) {
190 my $p = "https?:\/\/" . $pattern;
192 write_debug("Matching %s against %s", $p, $s);
197 write_debug("Found %s", $matched);
198 $res = $ua->head($matched);
199 if ($res->is_redirect()) {
200 my $new = $res->headers()->header("Location");
202 write_debug("Replacing %s with %s", $matched, $new);
203 $s =~ s/$matched/$new/;
206 write_debug("Error resolving %s", $matched);
213 write_debug("Loop terminated by counter");
216 write_debug("Final string: %s", $s);
224 foreach (@{$conf->{'videosite'}->{'connectorlist'}}) {
225 push(@c, $conf->{'videosite'}->{'connectors'}->{$_});
234 my $message = $event->message();
235 my $witem = $event->channel();
241 # Look if we should ignore this line
242 if ($message =~ m,(?:\s|^)/nosave(?:\s|$),) {
246 push_output($event->ewpf);
247 $message = expand_url_shortener($message);
251 # Offer the message to all Grabbers in turn
252 GRABBER: foreach $g (@grabbers) {
253 ($m, $p) = $g->get($message);
254 while (defined($m)) {
255 write_debug('Metadata: %s', Dumper($m));
256 if ('download' eq ($conf->{'videosite'}->{'mode'})) {
257 write_irssi('%%R>>> %%NSaving %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
258 unless($getter->get($m)) {
259 write_irssi('%%R>>> FAILED');
261 } elsif ('display' eq ($conf->{'videosite'}->{'mode'})) {
262 write_irssi('%%M>>> %%NSaw %%Y%s%%N %%G%s', $m->{'SOURCE'}, $m->{'TITLE'});
264 write_irssi('%%R>>> Invalid operation mode');
267 # Remove the matched part from the message and try again (there may be
271 last GRABBER if ($message =~ /^\s*$/);
273 ($m, $p) = $g->get($message);
284 my ($tempfile, $tempfn) = tempfile("videosite.json.XXXXXX", dir => Irssi::get_irssi_dir());
285 print $tempfile JSON->new->pretty->utf8->encode($conf);
287 rename($tempfn, $conffile);
290 write_irssi('Could not save config to %s: %s', ($conffile, $@));
292 write_irssi('configuration saved to %s', $conffile);
302 foreach $p (@getters, @grabbers) {
303 if ($p->{'NAME'} eq $target) {
304 $p->setval($key, $val);
308 write_irssi('No such module');
316 foreach $p (@grabbers) {
317 if ($p->{'NAME'} eq $target) {
322 write_irssi('No such module');
330 foreach $p (@grabbers) {
331 if ($p->{'NAME'} eq $target) {
336 write_irssi('No such module');
345 if (defined($target)) {
346 foreach $p (@getters, @grabbers) {
347 if ($p->{'NAME'} eq $target) {
348 write_irssi($p->getconfstr());
352 write_irssi('No such module');
354 write_irssi('Loaded grabbers (* denotes enabled modules):');
355 foreach $p (@grabbers) {
356 $e = $p->_getval('enabled');
357 write_irssi(' %s%s', $p->{'NAME'}, $e?'*':'');
360 write_irssi('Loaded getters:');
361 foreach $p (@getters) {
362 write_irssi(' %s', $p->{'NAME'});
371 if (defined($target)) {
372 foreach $p (@getters, @grabbers) {
373 if ($p->{'NAME'} eq $target) {
374 write_irssi($p->gethelpstr());
378 write_irssi('No such module');
380 write_irssi(<<'EOT');
382 save: save the current configuration
383 help [modulename]: display this help, or module specific help
384 show [modulename]: show loaded modules, or the current parameters of a module
385 set modulename parameter value: set a module parameter to a new value
386 getter [modulename]: display or set the getter to use
387 enable [modulename]: enable the usage of this module (grabbers only)
388 disable [modulename]: disable the usage of this module (grabbers only)
389 reload: reload all modules (this is somewhat experimental)
390 mode [modename]: display or set the operation mode (download/display)
391 connector [subcommand]: manage connectors (proxies)
392 debug: enable debugging messages
393 nodebug: disable debugging messages
402 if (defined($target)) {
403 foreach $p (@getters) {
404 if ($p->{'NAME'} eq $target) {
406 $conf->{'videosite'}->{'getter'} = $target;
407 write_irssi("Getter changed to %s", $target);
411 write_irssi('No such getter');
413 write_irssi('Current getter: %s', $conf->{'videosite'}->{'getter'});
420 if (defined($mode)) {
422 if (('download' eq $mode) or ('display' eq $mode)) {
423 $conf->{'videosite'}->{'mode'} = $mode;
424 write_irssi('Now using %s mode', $mode);
426 write_irssi('Invalid mode: %s', $mode);
429 write_irssi('Current mode: %s', $conf->{'videosite'}->{'mode'});
435 my $connconf = $conf->{'videosite'}->{'connectors'};
437 unless(defined($subcmd)) {
441 $subcmd = lc($subcmd);
443 if ($subcmd eq 'list') {
444 write_irssi("Defined connectors");
445 foreach (keys(%{$connconf})) {
447 my $schemas = $connconf->{$_}->{'schemas'};
448 if (scalar(keys(%{$schemas})) == 0) {
449 write_irssi(" No schemas defined");
451 foreach (keys(%{$schemas})) {
452 write_irssi(' %s: %s', $_, $schemas->{$_});
458 write_irssi("Selected connectors: %s", join(", ", @{$conf->{'videosite'}->{'connectorlist'}}));
459 } elsif ($subcmd eq 'add') {
462 unless(defined($name)) {
463 write_irssi("No name given");
469 if (exists($connconf->{$_})) {
470 write_irssi("Connector already exists");
474 $connconf->{$name} = {'name' => $name, 'schemas' => {}};
475 } elsif ($subcmd eq 'del') {
478 unless(defined($name)) {
479 write_irssi("No name given");
485 unless (exists($connconf->{$name})) {
486 write_irssi("Connector does not exist");
490 if (exists($connconf->{$name}->{'_immutable'})) {
491 write_irssi("Connector cannot be removed");
495 delete($connconf->{$name});
497 # Remove from list of active connectors
498 $conf->{'videosite'}->{'connectorlist'} =
499 [ grep { $_ ne $name } @{$conf->{'videosite'}->{'connectorlist'}} ];
501 if (scalar(@{$conf->{'videosite'}->{'connectorlist'}}) == 0) {
502 write_irssi("List of selected connectors is empty, resetting to direct");
503 $conf->{'videosite'}->{'connectorlist'} = [ 'direct' ];
505 } elsif ($subcmd eq 'addschema') {
506 my ($conn, $schema, $proxy) = @_;
508 unless(defined($conn)) {
509 write_irssi("No connector name given");
515 if (exists($connconf->{$conn}->{'_immutable'})) {
516 write_irssi("Connector cannot be modified");
520 unless(defined($schema)) {
521 write_irssi("No schema given");
525 $schema = lc($schema);
527 unless(defined($proxy)) {
528 write_irssi("No proxy given");
532 unless(exists($connconf->{$conn})) {
533 write_irssi("Connector does not exist");
537 $connconf->{$conn}->{'schemas'}->{$schema} = $proxy;
538 } elsif ($subcmd eq 'delschema') {
539 my ($conn, $schema) = @_;
541 unless(defined($conn)) {
542 write_irssi("No connector name given");
548 if (exists($connconf->{$conn}->{'_immutable'})) {
549 write_irssi("Connector cannot be modified");
553 unless(defined($schema)) {
554 write_irssi("No schema given");
558 $schema = lc($schema);
560 unless(exists($connconf->{$conn})) {
561 write_irssi("Connector does not exist");
565 delete($connconf->{$conn}->{'schemas'}->{$schema});
566 } elsif ($subcmd eq 'select') {
567 my @connlist = map { lc } @_;
569 if (scalar(@connlist) == 0) {
570 write_irssi("No connectors given");
574 foreach (@connlist) {
575 unless(exists($connconf->{$_})) {
576 write_irssi("Connector %s does not exist", $_);
581 $conf->{'videosite'}->{'connectorlist'} = [ @connlist ];
583 write_irssi("connector [list|add|del|addschema|delschema|help] <options>");
584 write_irssi(" help: Show this help");
585 write_irssi(" list: List the defined connectors");
586 write_irssi(" add <name>: Add a connector with name <name>");
587 write_irssi(" del <name>: Delete the connector with name <name>");
588 write_irssi(" addschema <name> <schema> <proxy>: Add proxy to connector for the given schema");
589 write_irssi(" delschema <name> <schema>: Remove the schema from the connector");
590 write_irssi(" select <name> [<name>...]: Select the connectors to use");
598 sub sig_command_script_unload {
600 if ($script =~ /(.*\/)?videosite(\.pl)?$/) {
615 opendir(D, $dir) || return ();
616 @list = grep {/$pattern/ && -f File::Spec->catfile($dir, $_) } readdir(D);
620 write_debug("Trying to load $p:");
622 eval qq{ require videosite::$p; };
624 write_irssi("Failed to load plugin: $@");
628 $g = eval qq{ videosite::$p->new(); };
630 write_irssi("Failed to instanciate: $@");
635 write_debug("found $g->{'TYPE'} $g->{'NAME'}");
636 if ($type eq $g->{'TYPE'}) {
638 $g->setio(\&write_irssi);
639 $g->setconn(\&connectorlist);
641 write_irssi('%s has wrong type (got %s, expected %s)', $p, $g->{'TYPE'}, $type);
646 write_debug("Loaded %d plugins", $#g+1);
651 sub _load_modules($) {
655 foreach (keys(%INC)) {
656 if ($INC{$_} =~ m|^$path|) {
657 write_debug("Removing %s from \$INC", $_);
661 @grabbers = ploader($path, '.*Grabber\.pm$', 'grabber');
662 @getters = ploader($path, '.*Getter\.pm$', 'getter');
668 my $bindings = shift;
672 write_debug("Attempting JSON config load from %s", $conffile);
675 open(CONF, '<', $conffile);
676 $conf = JSON->new->utf8->decode(<CONF>);
679 } elsif (-r $xmlconffile) {
680 write_debug("Attempting XML config load from %s", $xmlconffile);
681 $conf = XML::Simple::XMLin($xmlconffile, ForceArray => ['config', 'option', 'connectorlist'], KeepRoot => 1, KeyAttr => {'connector' => '+name', 'config' => 'module', 'option' => 'key'});
684 unless(defined($conf)) {
685 # No config, start with an empty one
686 write_debug('No config found, using defaults');
687 $conf = { 'videosite' => { }};
690 foreach (keys(%{$PARAMS})) {
691 unless (exists($conf->{'videosite'}->{$_})) {
692 $conf->{'videosite'}->{$_} = $PARAMS->{$_};
696 # Make sure there is a connector called 'direct', which defines no
698 unless (exists($conf->{'videosite'}->{'connectors'}->{'direct'})) {
699 $conf->{'videosite'}->{'connectors'}->{'direct'} = {
706 _load_modules($plugindir);
708 unless (defined(@grabbers) && defined(@getters)) {
709 write_irssi('No grabbers or no getters found, can not proceed.');
713 $getter = $getters[0];
714 foreach $p (@getters) {
715 if ($conf->{'videosite'}->{'getter'} eq $p->{'NAME'}) {
719 write_debug('Selected %s as getter', $getter->{'NAME'});
720 $conf->{'videosite'}->{'getter'} = $getter->{'NAME'};
722 # Loop through all plugins and load the config
723 foreach $p (@grabbers, @getters) {
724 $conf->{'videosite'}->{'config'}->{$p->{'NAME'}} = $p->mergeconfig($conf->{'videosite'}->{'config'}->{$p->{'NAME'}});
729 Irssi::signal_add_first('command script load', 'sig_command_script_unload');
730 Irssi::signal_add_first('command script unload', 'sig_command_script_unload');
731 Irssi::signal_add('setup saved', 'cmd_save');
734 Irssi::command_bind(_bcb('videosite' => \&cmdhandler));
737 write_irssi('initialized successfully');
741 my ($complist, $window, $word, $linestart, $want_space) = @_;
744 if ($linestart !~ m|^/videosite\b|) {
748 if ('/videosite' eq $linestart) {
749 # No command enterd so far. Produce a list of possible follow-ups
750 @matches = grep {/^$word/} keys (%{$videosite_commands});
751 } elsif ('/videosite set' eq $linestart) {
752 # 'set' command entered. Produce a list of modules
753 foreach (@grabbers, @getters) {
754 push(@matches, $_->{'NAME'}) if $_->{'NAME'} =~ m|^$word|;
756 } elsif ($linestart =~ m|^/videosite set (\w+)$|) {
759 foreach my $p (@getters, @grabbers) {
760 if ($p->{'NAME'} eq $module) {
761 @matches = $p->getparamlist($word);
765 } elsif ($linestart =~ m|/videosite set (\w+) (\w+)$|) {
769 foreach my $p (@getters, @grabbers) {
770 if ($p->{'NAME'} eq $module) {
771 @matches = $p->getparamvalues($param, $word);
778 push(@{$complist}, sort @matches);
781 Irssi::signal_stop();
786 my ($cmd, @params) = split(/\s+/, $event->message());
788 push_output($event->ewpf);
790 if (exists($videosite_commands->{$cmd})) {
791 $videosite_commands->{$cmd}->(@params);
797 unshift(@INC, $scriptdir);