use strict; use Irssi; use Irssi::Irc; use Net::Twitter; use HTTP::Date; use HTML::Entities; use File::Temp; use vars qw($VERSION %IRSSI); use constant { DEBUG => 0 }; $VERSION = "1.1"; my ($REV) = '$Rev: 309 $' =~ /(\d+)/; %IRSSI = ( authors => 'Dan Boger', contact => 'zigdon@gmail.com', name => 'twirssi', description => 'Send twitter updates using /tweet. ' . 'Can optionally set your bitlbee /away message to same', license => 'GNU GPL v2', url => 'http://tinyurl.com/twirssi', changed => 'Mon Dec 1 15:36:01 PST 2008', ); my $window; my $twit; my %twits; my $user; my $poll; my %nicks; my %friends; my $last_poll = time - 300; sub cmd_direct { my ( $data, $server, $win ) = @_; unless ($twit) { ¬ice("Not logged in! Use /twitter_login username pass!"); return; } my ( $target, $text ) = split ' ', $data, 2; unless ( $target and $text ) { ¬ice("Usage: /dm "); return; } &cmd_direct_as( "$user $data", $server, $win ); } sub cmd_direct_as { my ( $data, $server, $win ) = @_; unless ($twit) { ¬ice("Not logged in! Use /twitter_login username pass!"); return; } my ( $username, $target, $text ) = split ' ', $data, 3; unless ( $username and $target and $text ) { ¬ice("Usage: /dm_as "); return; } unless ( exists $twits{$username} ) { ¬ice("Unknown username $username"); return; } unless ( $twits{$username} ->new_direct_message( { user => $target, text => $text } ) ) { ¬ice("DM to $target failed"); return; } ¬ice("DM sent to $target"); $nicks{$target} = time; } sub cmd_tweet { my ( $data, $server, $win ) = @_; unless ($twit) { ¬ice("Not logged in! Use /twitter_login username pass!"); return; } $data =~ s/^\s+|\s+$//; unless ($data) { ¬ice("Usage: /tweet "); return; } &cmd_tweet_as( "$user $data", $server, $win ); } sub cmd_tweet_as { my ( $data, $server, $win ) = @_; unless ($twit) { ¬ice("Not logged in! Use /twitter_login username pass!"); return; } $data =~ s/^\s+|\s+$//; my ( $username, $data ) = split ' ', $data, 2; unless ( $username and $data ) { ¬ice("Usage: /tweet_as "); return; } unless ( exists $twits{$username} ) { ¬ice("Unknown username $username"); return; } foreach my $url ( $data =~ /(https?:\/\/\S+[\w\/])/g ) { eval { my $short = makeashorterlink($url); $data =~ s/\Q$url/$short/g; }; } unless ( $twits{$username}->update($data) ) { ¬ice("Update failed"); return; } foreach ( $data =~ /@([-\w]+)/ ) { $nicks{$1} = time; } my $away = 0; if ( Irssi::settings_get_bool("tweet_to_away") and $data !~ /\@\w/ and $data !~ /^[dD] / ) { my $server = Irssi::server_find_tag( Irssi::settings_get_str("bitlbee_server") ); if ($server) { $server->send_raw("away :$data"); $away = 1; } else { ¬ice( "Can't find bitlbee server.", "Update bitlbee_server or disalbe tweet_to_away" ); } } ¬ice( "Update sent" . ( $away ? " (and away msg set)" : "" ) ); } sub gen_cmd { my ( $usage_str, $api_name, $post_ref ) = @_; return sub { my ( $data, $server, $win ) = @_; unless ($twit) { ¬ice("Not logged in! Use /twitter_login username pass!"); return; } $data =~ s/^\s+|\s+$//; unless ($data) { ¬ice("Usage: $usage_str"); return; } unless ( $twit->$api_name($data) ) { ¬ice("$api_name failed"); return; } &$post_ref($data) if $post_ref; } } sub cmd_switch { my ( $data, $server, $win ) = @_; $data =~ s/^\s+|\s+$//g; if ( exists $twits{$data} ) { ¬ice("Switching to $data"); $twit = $twits{$data}; $user = $data; } else { ¬ice("Unknown user $data"); } } sub cmd_logout { my ( $data, $server, $win ) = @_; $data =~ s/^\s+|\s+$//g; if ( $data and exists $twits{$data} ) { ¬ice("Logging out $data..."); $twits{$data}->end_session(); delete $twits{$data}; } elsif ($data) { ¬ice("Unknown username '$data'"); } else { ¬ice("Logging out $user..."); $twit->end_session(); undef $twit; delete $twits{$user}; if ( keys %twits ) { &cmd_switch( ( keys %twits )[0], $server, $win ); } else { Irssi::timeout_remove($poll) if $poll; undef $poll; } } } sub cmd_login { my ( $data, $server, $win ) = @_; my $pass; ( $user, $pass ) = split ' ', $data, 2; %friends = %nicks = (); $twit = Net::Twitter->new( username => $user, password => $pass, source => "twirssi" ); unless ( $twit->verify_credentials() ) { ¬ice("Login failed"); $twit = undef; return; } if ($twit) { $twits{$user} = $twit; Irssi::timeout_remove($poll) if $poll; $poll = Irssi::timeout_add( 300 * 1000, \&get_updates, "" ); ¬ice("Logged in as $user, loading friends list..."); &load_friends; ¬ice( "loaded friends: ", scalar keys %nicks ); %nicks = %friends; $nicks{$user} = 0; &get_updates; } else { ¬ice("Login failed"); } } sub load_friends { my $page = 1; my %new_friends; while (1) { my $friends = $twit->friends( { page => $page } ); last unless $friends; $new_friends{ $_->{screen_name} } = time foreach @$friends; $page++; last if @$friends == 0 or $page == 10; $friends = $twit->friends( page => $page ); } foreach ( keys %new_friends ) { next if exists $friends{$_}; $friends{$_} = time; } foreach ( keys %friends ) { delete $friends{$_} unless exists $new_friends{$_}; } } sub get_updates { $window = Irssi::window_find_name( Irssi::settings_get_str('twitter_window') ); unless ($window) { Irssi::active_win() ->print( "Can't find a window named '" . Irssi::settings_get_str('twitter_window') . "'. Create it or change the value of twitter_window" ); } unless ($twit) { ¬ice("Not logged in! Use /twitter_login username pass!"); return; } my ( $fh, $filename ) = File::Temp::tempfile(); my $pid = fork(); if ($pid) { # parent Irssi::timeout_add_once( 5000, 'monitor_child', [$filename] ); } elsif ( defined $pid ) { # child close STDIN; close STDOUT; close STDERR; my $new_poll = time; &do_updates( $fh, $user, $twit ); foreach ( keys %twits ) { next if $_ eq $user; &do_updates( $fh, $_, $twits{$_} ); } print $fh "__friends__\n"; &load_friends; foreach ( sort keys %friends ) { print $fh "$_ $friends{$_}\n"; } print $fh $new_poll; close $fh; exit; } } sub do_updates { my ( $fh, $username, $obj ) = @_; print scalar localtime, " - Polling for updates for $username" if DEBUG; my $tweets = $obj->friends_timeline( { since => HTTP::Date::time2str($last_poll) } ) || []; foreach my $t ( reverse @$tweets ) { my $text = decode_entities( $t->{text} ); $text =~ s/%/%%/g; $text =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g; my $prefix = ""; if ( Irssi::settings_get_bool("show_reply_context") and $t->{in_reply_to_screen_name} ne $username and $t->{in_reply_to_screen_name} and not exists $friends{ $t->{in_reply_to_screen_name} } ) { $nicks{ $t->{in_reply_to_screen_name} } = time; my $context = $obj->show_status( $t->{in_reply_to_status_id} ); if ($context) { my $ctext = decode_entities( $context->{text} ); $ctext =~ s/%/%%/g; $ctext =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g; printf $fh "[%s%%B\@%s%%n] %s\n", ( $username ne $user ? "$username: " : "" ), $context->{user}{screen_name}, $ctext; $prefix = "\--> "; } } next if $t->{user}{screen_name} eq $username and not Irssi::settings_get_bool("show_own_tweets"); printf $fh "%s[%s%%B\@%s%%n] %s\n", $prefix, ( $username ne $user ? "$username: " : "" ), $t->{user}{screen_name}, $text; } print scalar localtime, " - Polling for replies" if DEBUG; $tweets = $obj->replies( { since => HTTP::Date::time2str($last_poll) } ) || []; foreach my $t ( reverse @$tweets ) { next if exists $friends{ $t->{user}{screen_name} }; my $text = decode_entities( $t->{text} ); $text =~ s/%/%%/g; $text =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g; printf $fh "[%s%%B\@%s%%n] %s\n", ( $username ne $user ? "$username: " : "" ), $t->{user}{screen_name}, $text; } print scalar localtime, " - Polling for DMs" if DEBUG; $tweets = $obj->direct_messages( { since => HTTP::Date::time2str($last_poll) } ) || []; foreach my $t ( reverse @$tweets ) { my $text = decode_entities( $t->{text} ); $text =~ s/%/%%/g; $text =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g; printf $fh "[%s%%B\@%s%%n (%%WDM%%n)] %s\n", ( $username ne $user ? "$username: " : "" ), $t->{sender_screen_name}, $text; } print scalar localtime, " - Done" if DEBUG; } sub monitor_child { my $data = shift; my $filename = $data->[0]; print scalar localtime, " - checking child log at $filename" if DEBUG; if ( open FILE, $filename ) { my @lines; while () { chomp; last if /^__friends__/; push @lines, $_ unless /^__friends__/; } %friends = (); while () { if (/^\d+$/) { $last_poll = $_; last; } my ( $f, $t ) = split ' ', $_; $nicks{$f} = $friends{$f} = $t; } print "new last_poll = $last_poll" if DEBUG; foreach my $line (@lines) { chomp $line; $window->print( $line, MSGLEVEL_PUBLIC ); foreach ( $line =~ /\@([-\w]+)/ ) { $nicks{$1} = time; } } close FILE; unlink $filename or warn "Failed to remove $filename: $!"; return; } Irssi::timeout_add_once( 5000, 'monitor_child', [$filename] ); } sub notice { $window->print( "%R***%n @_", MSGLEVEL_PUBLIC ); } sub sig_complete { my ( $complist, $window, $word, $linestart, $want_space ) = @_; return unless $linestart =~ /^\/(?:tweet|dm)/; return if $linestart eq '/tweet' and $word !~ s/^@//; push @$complist, grep /^\Q$word/i, sort { $nicks{$b} <=> $nicks{$a} } keys %nicks; @$complist = map { "\@$_" } @$complist if $linestart eq '/tweet'; } Irssi::settings_add_str( "twirssi", "twitter_window", "twitter" ); Irssi::settings_add_str( "twirssi", "bitlbee_server", "bitlbee" ); Irssi::settings_add_str( "twirssi", "short_url_provider", "TinyURL" ); Irssi::settings_add_bool( "twirssi", "tweet_to_away", 0 ); Irssi::settings_add_bool( "twirssi", "show_reply_context", 0 ); Irssi::settings_add_bool( "twirssi", "show_own_tweets", 1 ); $window = Irssi::window_find_name( Irssi::settings_get_str('twitter_window') ); if ($window) { Irssi::command_bind( "dm", "cmd_direct" ); Irssi::command_bind( "tweet", "cmd_tweet" ); Irssi::command_bind( "dm_as", "cmd_direct_as" ); Irssi::command_bind( "tweet_as", "cmd_tweet_as" ); Irssi::command_bind( "twitter_login", "cmd_login" ); Irssi::command_bind( "twitter_logout", "cmd_logout" ); Irssi::command_bind( "twitter_switch", "cmd_switch" ); Irssi::command_bind( "twirssi_version", sub { ¬ice( "Twirssi v$VERSION (r$REV). See details at http://tinyurl.com/twirssi" ); } ); Irssi::command_bind( "twitter_friend", &gen_cmd( "/twitter_friend ", "create_friend", sub { ¬ice("Following $_[0]"); $nicks{ $_[0] } = time; } ) ); Irssi::command_bind( "twitter_unfriend", &gen_cmd( "/twitter_unfriend ", "destroy_friend", sub { ¬ice("Stopped following $_[0]"); delete $nicks{ $_[0] }; } ) ); Irssi::command_bind( "twitter_updates", "get_updates" ); Irssi::signal_add_last( 'complete word' => \&sig_complete ); ¬ice(" %Y<%C(%B^%C)%N TWIRSSI v%R$VERSION%N (r$REV)"); ¬ice(" %C(_(\\%N http://tinyurl.com/twirssi for full docs"); ¬ice( " %Y||%C `%N Log in with /twitter_login, send updates with /tweet"); if ( my $provider = Irssi::settings_get_str("short_url_provider") ) { eval "use WWW::Shorten::$provider;"; if ($@) { ¬ice( "Failed to load WWW::Shorten::$provider - either clear short_url_provider or install the CPAN module" ); } } } else { Irssi::active_win() ->print( "Create a window named " . Irssi::settings_get_str('twitter_window') . " or change the value of twitter_window. Then, reload twirssi." ); }