twirssi.pl

   1 use strict;
   2 use Irssi;
   3 use Irssi::Irc;
   4 use HTTP::Date;
   5 use HTML::Entities;
   6 use File::Temp;
   7 use LWP::Simple;
   8 use Data::Dumper;
   9 $Data::Dumper::Indent = 1;
  10
  11 BEGIN {
  12     $ENV{JSON_ANY_ORDER} = "JSON Syck DWIW";
  13     require JSON::Any;
  14     import JSON::Any;
  15     require Net::Twitter;
  16     import Net::Twitter;
  17 }
  18
  19 use vars qw($VERSION %IRSSI);
  20
  21 $VERSION = "1.7.3";
  22 my ($REV) = '$Rev: 352 $' =~ /(\d+)/;
  23 %IRSSI = (
  24     authors     => 'Dan Boger',
  25     contact     => 'zigdon@gmail.com',
  26     name        => 'twirssi',
  27     description => 'Send twitter updates using /tweet.  '
  28       . 'Can optionally set your bitlbee /away message to same',
  29     license => 'GNU GPL v2',
  30     url     => 'http://tinyurl.com/twirssi',
  31     changed => '$Date: 2009-01-08 13:46:50 -0800 (Thu, 08 Jan 2009) $',
  32 );
  33
  34 my $window;
  35 my $twit;
  36 my %twits;
  37 my $user;
  38 my $poll;
  39 my %nicks;
  40 my %friends;
  41 my $last_poll = time - 300;
  42 my %tweet_cache;
  43 my %id_map;
  44
  45 sub cmd_direct {
  46     my ( $data, $server, $win ) = @_;
  47
  48     unless ($twit) {
  49         &notice("Not logged in!  Use /twitter_login username pass!");
  50         return;
  51     }
  52
  53     my ( $target, $text ) = split ' ', $data, 2;
  54     unless ( $target and $text ) {
  55         &notice("Usage: /dm <nick> <message>");
  56         return;
  57     }
  58
  59     &cmd_direct_as( "$user $data", $server, $win );
  60 }
  61
  62 sub cmd_direct_as {
  63     my ( $data, $server, $win ) = @_;
  64
  65     unless ($twit) {
  66         &notice("Not logged in!  Use /twitter_login username pass!");
  67         return;
  68     }
  69
  70     my ( $username, $target, $text ) = split ' ', $data, 3;
  71     unless ( $username and $target and $text ) {
  72         &notice("Usage: /dm_as <username> <nick> <message>");
  73         return;
  74     }
  75
  76     unless ( exists $twits{$username} ) {
  77         &notice("Unknown username $username");
  78         return;
  79     }
  80
  81     eval {
  82         unless ( $twits{$username}
  83             ->new_direct_message( { user => $target, text => $text } ) )
  84         {
  85             &notice("DM to $target failed");
  86             return;
  87         }
  88     };
  89
  90     if ($@) {
  91         &notice("DM caused an error.  Aborted");
  92         return;
  93     }
  94
  95     &notice("DM sent to $target");
  96     $nicks{$target} = time;
  97 }
  98
  99 sub cmd_tweet {
 100     my ( $data, $server, $win ) = @_;
 101
 102     unless ($twit) {
 103         &notice("Not logged in!  Use /twitter_login username pass!");
 104         return;
 105     }
 106
 107     $data =~ s/^\s+|\s+$//;
 108     unless ($data) {
 109         &notice("Usage: /tweet <update>");
 110         return;
 111     }
 112
 113     &cmd_tweet_as( "$user $data", $server, $win );
 114 }
 115
 116 sub cmd_tweet_as {
 117     my ( $data, $server, $win ) = @_;
 118
 119     unless ($twit) {
 120         &notice("Not logged in!  Use /twitter_login username pass!");
 121         return;
 122     }
 123
 124     $data =~ s/^\s+|\s+$//;
 125     my ( $username, $data ) = split ' ', $data, 2;
 126
 127     unless ( $username and $data ) {
 128         &notice("Usage: /tweet_as <username> <update>");
 129         return;
 130     }
 131
 132     unless ( exists $twits{$username} ) {
 133         &notice("Unknown username $username");
 134         return;
 135     }
 136
 137     if ( Irssi::settings_get_str("short_url_provider") ) {
 138         foreach my $url ( $data =~ /(https?:\/\/\S+[\w\/])/g ) {
 139             eval {
 140                 my $short = makeashorterlink($url);
 141                 $data =~ s/\Q$url/$short/g;
 142             };
 143         }
 144     }
 145
 146     if ( length $data > 140 ) {
 147         &notice(
 148             "Tweet too long (" . length($data) . " characters) - aborted" );
 149         return;
 150     }
 151
 152     eval {
 153         unless ( $twits{$username}->update($data) )
 154         {
 155             &notice("Update failed");
 156             return;
 157         }
 158     };
 159
 160     if ($@) {
 161         &notice("Update caused an error.  Aborted.");
 162         return;
 163     }
 164
 165     foreach ( $data =~ /@([-\w]+)/ ) {
 166         $nicks{$1} = time;
 167     }
 168
 169     my $away = 0;
 170     if (    Irssi::settings_get_bool("tweet_to_away")
 171         and $data !~ /\@\w/
 172         and $data !~ /^[dD] / )
 173     {
 174         my $server =
 175           Irssi::server_find_tag( Irssi::settings_get_str("bitlbee_server") );
 176         if ($server) {
 177             $server->send_raw("away :$data");
 178             $away = 1;
 179         } else {
 180             &notice( "Can't find bitlbee server.",
 181                 "Update bitlbee_server or disable tweet_to_away" );
 182         }
 183     }
 184
 185     &notice( "Update sent" . ( $away ? " (and away msg set)" : "" ) );
 186 }
 187
 188 sub cmd_reply {
 189     my ( $data, $server, $win ) = @_;
 190
 191     unless ($twit) {
 192         &notice("Not logged in!  Use /twitter_login username pass!");
 193         return;
 194     }
 195
 196     $data =~ s/^\s+|\s+$//;
 197     unless ($data) {
 198         &notice("Usage: /reply <nick[:num]> <update>");
 199         return;
 200     }
 201
 202     $data =~ s/^\s+|\s+$//;
 203     my ( $id, $data ) = split ' ', $data, 2;
 204     unless ( $id and $data ) {
 205         &notice("Usage: /reply_as <nick[:num]> <update>");
 206         return;
 207     }
 208
 209     &cmd_reply_as( "$user $id $data", $server, $win );
 210 }
 211
 212 sub cmd_reply_as {
 213     my ( $data, $server, $win ) = @_;
 214
 215     unless ( Irssi::settings_get_bool("twirssi_track_replies") ) {
 216         &notice("twirssi_track_replies is required in order to reply to "
 217               . "specific tweets.  Either enable it, or just use /tweet "
 218               . "\@username <text>." );
 219         return;
 220     }
 221
 222     unless ($twit) {
 223         &notice("Not logged in!  Use /twitter_login username pass!");
 224         return;
 225     }
 226
 227     $data =~ s/^\s+|\s+$//;
 228     my ( $username, $id, $data ) = split ' ', $data, 3;
 229
 230     unless ( $username and $data ) {
 231         &notice("Usage: /reply_as <username> <nick[:num]> <update>");
 232         return;
 233     }
 234
 235     unless ( exists $twits{$username} ) {
 236         &notice("Unknown username $username");
 237         return;
 238     }
 239
 240     my $nick;
 241     $id =~ s/[^\w\d\-:]+//g;
 242     ( $nick, $id ) = split /:/, $id;
 243     unless ( exists $id_map{ lc $nick } ) {
 244         &notice("Can't find a tweet from $nick to reply to!");
 245         return;
 246     }
 247
 248     $id = $id_map{__indexes}{$nick} unless $id;
 249     unless ( $id_map{ lc $nick }[$id] ) {
 250         &notice("Can't find a tweet numbered $id from $nick to reply to!");
 251         return;
 252     }
 253
 254     # remove any @nick at the beginning of the reply, as we'll add it anyway
 255     $data =~ s/^\s*\@?$nick\s*//;
 256     $data = "\@$nick " . $data;
 257
 258     if ( Irssi::settings_get_str("short_url_provider") ) {
 259         foreach my $url ( $data =~ /(https?:\/\/\S+[\w\/])/g ) {
 260             eval {
 261                 my $short = makeashorterlink($url);
 262                 $data =~ s/\Q$url/$short/g;
 263             };
 264         }
 265     }
 266
 267     if ( length $data > 140 ) {
 268         &notice(
 269             "Tweet too long (" . length($data) . " characters) - aborted" );
 270         return;
 271     }
 272
 273     eval {
 274         unless (
 275             $twits{$username}->update(
 276                 {
 277                     status                => $data,
 278                     in_reply_to_status_id => $id_map{ lc $nick }[$id]
 279                 }
 280             )
 281           )
 282         {
 283             &notice("Update failed");
 284             return;
 285         }
 286     };
 287
 288     if ($@) {
 289         &notice("Update caused an error.  Aborted");
 290         return;
 291     }
 292
 293     foreach ( $data =~ /@([-\w]+)/ ) {
 294         $nicks{$1} = time;
 295     }
 296
 297     my $away = 0;
 298     if (    Irssi::settings_get_bool("tweet_to_away")
 299         and $data !~ /\@\w/
 300         and $data !~ /^[dD] / )
 301     {
 302         my $server =
 303           Irssi::server_find_tag( Irssi::settings_get_str("bitlbee_server") );
 304         if ($server) {
 305             $server->send_raw("away :$data");
 306             $away = 1;
 307         } else {
 308             &notice( "Can't find bitlbee server.",
 309                 "Update bitlbee_server or disalbe tweet_to_away" );
 310         }
 311     }
 312
 313     &notice( "Update sent" . ( $away ? " (and away msg set)" : "" ) );
 314 }
 315
 316 sub gen_cmd {
 317     my ( $usage_str, $api_name, $post_ref ) = @_;
 318
 319     return sub {
 320         my ( $data, $server, $win ) = @_;
 321
 322         unless ($twit) {
 323             &notice("Not logged in!  Use /twitter_login username pass!");
 324             return;
 325         }
 326
 327         $data =~ s/^\s+|\s+$//;
 328         unless ($data) {
 329             &notice("Usage: $usage_str");
 330             return;
 331         }
 332
 333         eval {
 334             unless ( $twit->$api_name($data) )
 335             {
 336                 &notice("$api_name failed");
 337                 return;
 338             }
 339         };
 340
 341         if ($@) {
 342             &notice("$api_name caused an error.  Aborted.");
 343             return;
 344         }
 345
 346         &$post_ref($data) if $post_ref;
 347       }
 348 }
 349
 350 sub cmd_switch {
 351     my ( $data, $server, $win ) = @_;
 352
 353     $data =~ s/^\s+|\s+$//g;
 354     if ( exists $twits{$data} ) {
 355         &notice("Switching to $data");
 356         $twit = $twits{$data};
 357         $user = $data;
 358     } else {
 359         &notice("Unknown user $data");
 360     }
 361 }
 362
 363 sub cmd_logout {
 364     my ( $data, $server, $win ) = @_;
 365
 366     $data =~ s/^\s+|\s+$//g;
 367     if ( $data and exists $twits{$data} ) {
 368         &notice("Logging out $data...");
 369         $twits{$data}->end_session();
 370         delete $twits{$data};
 371     } elsif ($data) {
 372         &notice("Unknown username '$data'");
 373     } else {
 374         &notice("Logging out $user...");
 375         $twit->end_session();
 376         undef $twit;
 377         delete $twits{$user};
 378         if ( keys %twits ) {
 379             &cmd_switch( ( keys %twits )[0], $server, $win );
 380         } else {
 381             Irssi::timeout_remove($poll) if $poll;
 382             undef $poll;
 383         }
 384     }
 385 }
 386
 387 sub cmd_login {
 388     my ( $data, $server, $win ) = @_;
 389     my $pass;
 390     if ($data) {
 391         ( $user, $pass ) = split ' ', $data, 2;
 392     } elsif ( my $autouser = Irssi::settings_get_str("twitter_usernames")
 393         and my $autopass = Irssi::settings_get_str("twitter_passwords") )
 394     {
 395         my @user = split /\s*,\s*/, $autouser;
 396         my @pass = split /\s*,\s*/, $autopass;
 397         if ( @user != @pass ) {
 398             &notice("Number of usernames doesn't match "
 399                   . "the number of passwords - auto-login failed" );
 400         } else {
 401             my ( $u, $p );
 402             while ( @user and @pass ) {
 403                 $u = shift @user;
 404                 $p = shift @pass;
 405                 &cmd_login("$u $p");
 406             }
 407             return;
 408         }
 409     } else {
 410         &notice("/twitter_login requires either a username and password "
 411               . "or twitter_usernames and twitter_passwords to be set." );
 412         return;
 413     }
 414
 415     %friends = %nicks = ();
 416
 417     $twit = Net::Twitter->new(
 418         username => $user,
 419         password => $pass,
 420         source   => "twirssi"
 421     );
 422
 423     unless ( $twit->verify_credentials() ) {
 424         &notice("Login as $user failed");
 425         $twit = undef;
 426         if ( keys %twits ) {
 427             &cmd_switch( ( keys %twits )[0], $server, $win );
 428         }
 429         return;
 430     }
 431
 432     if ($twit) {
 433         my $rate_limit = $twit->rate_limit_status();
 434         if ( $rate_limit and $rate_limit->{remaining_hits} < 1 ) {
 435             &notice("Rate limit exceeded, try again later");
 436             $twit = undef;
 437             return;
 438         }
 439
 440         $twits{$user} = $twit;
 441         Irssi::timeout_remove($poll) if $poll;
 442         $poll = Irssi::timeout_add( 300 * 1000, \&get_updates, "" );
 443         &notice("Logged in as $user, loading friends list...");
 444         &load_friends();
 445         &notice( "loaded friends: ", scalar keys %friends );
 446         if ( Irssi::settings_get_bool("twirssi_first_run") ) {
 447             Irssi::settings_set_bool( "twirssi_first_run", 0 );
 448             unless ( exists $friends{twirssi} ) {
 449                 &notice("Welcome to twirssi!"
 450                       . "  Perhaps you should add \@twirssi to your friends list,"
 451                       . " so you can be notified when a new version is release?"
 452                       . "  Just type /twitter_friend twirssi." );
 453             }
 454         }
 455         %nicks = %friends;
 456         $nicks{$user} = 0;
 457         &get_updates;
 458     } else {
 459         &notice("Login failed");
 460     }
 461 }
 462
 463 sub cmd_upgrade {
 464     my ( $data, $server, $win ) = @_;
 465
 466     my $loc = Irssi::settings_get_str("twirssi_location");
 467     unless ( -w $loc ) {
 468         &notice(
 469 "$loc isn't writable, can't upgrade.  Perhaps you need to /set twirssi_location?"
 470         );
 471         return;
 472     }
 473
 474     if ( not -x "/usr/bin/md5sum" and not $data ) {
 475         &notice(
 476 "/usr/bin/md5sum can't be found - try '/twirssi_upgrade nomd5' to skip MD5 verification"
 477         );
 478         return;
 479     }
 480
 481     my $md5;
 482     unless ($data) {
 483         eval { use Digest::MD5; };
 484
 485         if ($@) {
 486             &notice(
 487 "Failed to load Digest::MD5.  Try '/twirssi_upgrade nomd5' to skip MD5 verification"
 488             );
 489             return;
 490         }
 491
 492         $md5 = get("http://twirssi.com/md5sum");
 493         chomp $md5;
 494         $md5 =~ s/ .*//;
 495         unless ($md5) {
 496             &notice("Failed to download md5sum from peeron!  Aborting.");
 497             return;
 498         }
 499
 500         unless ( open( CUR, $loc ) ) {
 501             &notice(
 502 "Failed to read $loc.  Check that /set twirssi_location is set to the correct location."
 503             );
 504             return;
 505         }
 506
 507         my $cur_md5 = Digest::MD5::md5_hex(<CUR>);
 508         close CUR;
 509
 510         if ( $cur_md5 eq $md5 ) {
 511             &notice("Current twirssi seems to be up to date.");
 512             return;
 513         }
 514     }
 515
 516     my $URL = "http://twirssi.com/twirssi.pl";
 517     &notice("Downloading twirssi from $URL");
 518     LWP::Simple::getstore( $URL, "$loc.upgrade" );
 519
 520     unless ($data) {
 521         unless ( open( NEW, "$loc.upgrade" ) ) {
 522             &notice(
 523 "Failed to read $loc.upgrade.  Check that /set twirssi_location is set to the correct location."
 524             );
 525             return;
 526         }
 527
 528         my $new_md5 = Digest::MD5::md5_hex(<NEW>);
 529         close NEW;
 530
 531         if ( $new_md5 ne $md5 ) {
 532             &notice("MD5 verification failed. expected $md5, got $new_md5");
 533             return;
 534         }
 535     }
 536
 537     rename $loc, "$loc.backup"
 538       or &notice("Failed to back up $loc: $!.  Aborting")
 539       and return;
 540     rename "$loc.upgrade", $loc
 541       or &notice("Failed to rename $loc.upgrade: $!.  Aborting")
 542       and return;
 543
 544     my ( $dir, $file ) = ( $loc =~ m{(.*)/([^/]+)$} );
 545     if ( -e "$dir/autorun/$file" ) {
 546         &notice("Updating $dir/autorun/$file");
 547         unlink "$dir/autorun/$file"
 548           or &notice("Failed to remove old $file from autorun: $!");
 549         symlink "../$file", "$dir/autorun/$file"
 550           or &notice("Failed to create symlink in autorun directory: $!");
 551     }
 552
 553     &notice("Download complete.  Reload twirssi with /script load $file");
 554 }
 555
 556 sub load_friends {
 557     my $fh   = shift;
 558     my $page = 1;
 559     my %new_friends;
 560     eval {
 561         while (1)
 562         {
 563             print $fh "type:debug Loading friends page $page...\n"
 564               if ( $fh and &debug );
 565             my $friends = $twit->friends( { page => $page } );
 566             last unless $friends;
 567             $new_friends{ $_->{screen_name} } = time foreach @$friends;
 568             $page++;
 569             last if @$friends == 0 or $page == 10;
 570         }
 571     };
 572
 573     if ($@) {
 574         print $fh "type:debug Error during friends list update.  Aborted.\n";
 575         return;
 576     }
 577
 578     my ( $added, $removed ) = ( 0, 0 );
 579     print $fh "type:debug Scanning for new friends...\n" if ( $fh and &debug );
 580     foreach ( keys %new_friends ) {
 581         next if exists $friends{$_};
 582         $friends{$_} = time;
 583         $added++;
 584     }
 585
 586     print $fh "type:debug Scanning for removed friends...\n"
 587       if ( $fh and &debug );
 588     foreach ( keys %friends ) {
 589         next if exists $new_friends{$_};
 590         delete $friends{$_};
 591         $removed++;
 592     }
 593
 594     return ( $added, $removed );
 595 }
 596
 597 sub get_updates {
 598     print scalar localtime, " - get_updates starting" if &debug;
 599
 600     $window =
 601       Irssi::window_find_name( Irssi::settings_get_str('twitter_window') );
 602     unless ($window) {
 603         Irssi::active_win()
 604           ->print( "Can't find a window named '"
 605               . Irssi::settings_get_str('twitter_window')
 606               . "'.  Create it or change the value of twitter_window" );
 607     }
 608     unless ($twit) {
 609         &notice("Not logged in!  Use /twitter_login username pass!");
 610         return;
 611     }
 612
 613     my ( $fh, $filename ) = File::Temp::tempfile();
 614     my $pid = fork();
 615
 616     if ($pid) {    # parent
 617         Irssi::timeout_add_once( 5000, 'monitor_child', [ $filename, 0 ] );
 618         Irssi::pidwait_add($pid);
 619     } elsif ( defined $pid ) {    # child
 620         close STDIN;
 621         close STDOUT;
 622         close STDERR;
 623
 624         my $new_poll = time;
 625
 626         my $error = 0;
 627         $error += &do_updates( $fh, $user, $twit );
 628         foreach ( keys %twits ) {
 629             next if $_ eq $user;
 630             $error += &do_updates( $fh, $_, $twits{$_} );
 631         }
 632
 633         my ( $added, $removed ) = &load_friends($fh);
 634         if ( $added + $removed ) {
 635             print $fh "type:debug %R***%n Friends list updated: ",
 636               join( ", ",
 637                 sprintf( "%d added",   $added ),
 638                 sprintf( "%d removed", $removed ) ),
 639               "\n";
 640         }
 641         print $fh "__friends__\n";
 642         foreach ( sort keys %friends ) {
 643             print $fh "$_ $friends{$_}\n";
 644         }
 645
 646         if ($error) {
 647             print $fh "type:debug Update encountered errors.  Aborted\n";
 648             print $fh $last_poll;
 649         } else {
 650             print $fh $new_poll;
 651         }
 652         close $fh;
 653         exit;
 654     }
 655     print scalar localtime, " - get_updates ends" if &debug;
 656 }
 657
 658 sub do_updates {
 659     my ( $fh, $username, $obj ) = @_;
 660
 661     print scalar localtime, " - Polling for updates for $username" if &debug;
 662     my $tweets;
 663     eval {
 664         $tweets = $obj->friends_timeline(
 665             { since => HTTP::Date::time2str($last_poll) } );
 666     };
 667
 668     if ($@) {
 669         print $fh "type:debug Error during friends_timeline call.  Aborted.\n";
 670         return 1;
 671     }
 672
 673     unless ( ref $tweets ) {
 674         if ( $obj->can("get_error") ) {
 675             print $fh "type:debug API Error during friends_timeline call: ",
 676               JSON::Any->jsonToObj( $obj->get_error() ), "  Aborted.\n";
 677         } else {
 678             print $fh
 679               "type:debug API Error during friends_timeline call. Aborted.\n";
 680         }
 681         return 1;
 682     }
 683
 684     foreach my $t ( reverse @$tweets ) {
 685         my $text = decode_entities( $t->{text} );
 686         $text =~ s/%/%%/g;
 687         $text =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g;
 688         my $reply = "tweet";
 689         if (    Irssi::settings_get_bool("show_reply_context")
 690             and $t->{in_reply_to_screen_name} ne $username
 691             and $t->{in_reply_to_screen_name}
 692             and not exists $friends{ $t->{in_reply_to_screen_name} } )
 693         {
 694             $nicks{ $t->{in_reply_to_screen_name} } = time;
 695             my $context;
 696             eval {
 697               $context = $obj->show_status( $t->{in_reply_to_status_id} );
 698             };
 699
 700             if ($context) {
 701                 my $ctext = decode_entities( $context->{text} );
 702                 $ctext =~ s/%/%%/g;
 703                 $ctext =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g;
 704                 printf $fh "id:%d account:%s nick:%s type:tweet %s\n",
 705                   $context->{id}, $username,
 706                   $context->{user}{screen_name}, $ctext;
 707                 $reply = "reply";
 708             } elsif ($@) {
 709                 print $fh "type:debug request to get context failed: $@";
 710             } else {
 711                 print $fh "type:debug Failed to get context from $t->{in_reply_to_screen_name}"
 712                   if &debug;
 713             }
 714         }
 715         next
 716           if $t->{user}{screen_name} eq $username
 717               and not Irssi::settings_get_bool("show_own_tweets");
 718         printf $fh "id:%d account:%s nick:%s type:%s %s\n",
 719           $t->{id}, $username, $t->{user}{screen_name}, $reply, $text;
 720     }
 721
 722     print scalar localtime, " - Polling for replies" if &debug;
 723     eval {
 724         $tweets = $obj->replies( { since => HTTP::Date::time2str($last_poll) } )
 725           || [];
 726     };
 727
 728     if ($@) {
 729         print $fh "type:debug Error during replies call.  Aborted.\n";
 730         return 1;
 731     }
 732
 733     foreach my $t ( reverse @$tweets ) {
 734         next
 735           if exists $friends{ $t->{user}{screen_name} };
 736
 737         my $text = decode_entities( $t->{text} );
 738         $text =~ s/%/%%/g;
 739         $text =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g;
 740         printf $fh "id:%d account:%s nick:%s type:tweet %s\n",
 741           $t->{id}, $username, $t->{user}{screen_name}, $text;
 742     }
 743
 744     print scalar localtime, " - Polling for DMs" if &debug;
 745     eval {
 746         $tweets =
 747           $obj->direct_messages( { since => HTTP::Date::time2str($last_poll) } )
 748           || [];
 749     };
 750
 751     if ($@) {
 752         print $fh "type:debug Error during direct_messages call.  Aborted.\n";
 753         return 1;
 754     }
 755
 756     foreach my $t ( reverse @$tweets ) {
 757         my $text = decode_entities( $t->{text} );
 758         $text =~ s/%/%%/g;
 759         $text =~ s/(^|\W)\@([-\w]+)/$1%B\@$2%n/g;
 760         printf $fh "id:%d account:%s nick:%s type:dm %s\n",
 761           $t->{id}, $username, $t->{sender_screen_name}, $text;
 762     }
 763     print scalar localtime, " - Done" if &debug;
 764
 765     return 0;
 766 }
 767
 768 sub monitor_child {
 769     my ($data)   = @_;
 770     my $filename = $data->[0];
 771     my $attempt  = $data->[1];
 772
 773     print scalar localtime, " - checking child log at $filename ($attempt)"
 774       if &debug;
 775     my $new_last_poll;
 776     if ( open FILE, $filename ) {
 777         my @lines;
 778         while (<FILE>) {
 779             chomp;
 780             last if /^__friends__/;
 781             my %meta;
 782             foreach my $key (qw/id account nick type/) {
 783                 if (s/^$key:(\S+)\s*//) {
 784                     $meta{$key} = $1;
 785                 }
 786             }
 787
 788             next if exists $meta{id} and exists $tweet_cache{ $meta{id} };
 789             $tweet_cache{ $meta{id} } = time;
 790             my $account = "";
 791             if ( $meta{account} ne $user ) {
 792                 $account = "$meta{account}: ";
 793             }
 794
 795             my $marker = "";
 796             if (    $meta{type} ne 'dm'
 797                 and Irssi::settings_get_bool("twirssi_track_replies")
 798                 and $meta{nick}
 799                 and $meta{id} )
 800             {
 801                 $marker = ( $id_map{__indexes}{ $meta{nick} } + 1 ) % 100;
 802                 $id_map{ lc $meta{nick} }[$marker] = $meta{id};
 803                 $id_map{__indexes}{ $meta{nick} }  = $marker;
 804                 $marker                            = ":$marker";
 805             }
 806
 807             if ( $meta{type} eq 'tweet' ) {
 808                 push @lines, "[$account%B\@$meta{nick}%n$marker] $_\n",;
 809             } elsif ( $meta{type} eq 'reply' ) {
 810                 push @lines, "[$account\\--> %B\@$meta{nick}%n$marker] $_\n",;
 811             } elsif ( $meta{type} eq 'dm' ) {
 812                 push @lines, "[$account%B\@$meta{nick}%n (%WDM%n)] $_\n",;
 813             } elsif ( $meta{type} eq 'error' ) {
 814                 push @lines, "ERROR: $_\n";
 815             } elsif ( $meta{type} eq 'debug' ) {
 816                 print "$_" if &debug,;
 817             } else {
 818                 print "Unknown line type $meta{type}: $_" if &debug,;
 819             }
 820         }
 821
 822         %friends = ();
 823         while (<FILE>) {
 824             if (/^\d+$/) {
 825                 $new_last_poll = $_;
 826                 last;
 827             }
 828             my ( $f, $t ) = split ' ', $_;
 829             $nicks{$f} = $friends{$f} = $t;
 830         }
 831
 832         if ($new_last_poll) {
 833             print "new last_poll = $new_last_poll" if &debug;
 834             foreach my $line (@lines) {
 835                 chomp $line;
 836                 $window->print( $line, MSGLEVEL_PUBLIC );
 837                 foreach ( $line =~ /\@([-\w]+)/ ) {
 838                     $nicks{$1} = time;
 839                 }
 840             }
 841
 842             close FILE;
 843             unlink $filename
 844               or warn "Failed to remove $filename: $!"
 845               unless &debug;
 846
 847             # keep enough cached tweets, to make sure we don't show duplicates.
 848             foreach ( keys %tweet_cache ) {
 849                 next if $tweet_cache{$_} >= $last_poll;
 850                 delete $tweet_cache{$_};
 851             }
 852             $last_poll = $new_last_poll;
 853
 854             # save id_map hash
 855             if ( keys %id_map
 856                 and my $file =
 857                 Irssi::settings_get_str("twirssi_replies_store") )
 858             {
 859                 if ( open JSON, ">$file" ) {
 860                     print JSON JSON::Any->objToJson( \%id_map );
 861                     close JSON;
 862                 } else {
 863                     &notice("Failed to write replies to $file: $!");
 864                 }
 865             }
 866             return;
 867         }
 868     }
 869
 870     close FILE;
 871
 872     if ( $attempt < 12 ) {
 873         Irssi::timeout_add_once( 5000, 'monitor_child',
 874             [ $filename, $attempt + 1 ] );
 875     } else {
 876         &notice("Giving up on polling $filename");
 877         unlink $filename unless &debug;
 878     }
 879 }
 880
 881 sub debug {
 882     return Irssi::settings_get_bool("twirssi_debug");
 883 }
 884
 885 sub notice {
 886     $window->print( "%R***%n @_", MSGLEVEL_PUBLIC );
 887 }
 888
 889 sub sig_complete {
 890     my ( $complist, $window, $word, $linestart, $want_space ) = @_;
 891
 892     if (
 893         $linestart =~ /^\/twitter_reply(?:_as)?\s*$/
 894         or ( Irssi::settings_get_bool("twirssi_use_reply_aliases")
 895             and $linestart =~ /^\/reply(?:_as)?\s*$/ )
 896       )
 897     {    # /twitter_reply gets a nick:num
 898         @$complist = grep /^\Q$word/i, sort keys %{ $id_map{__indexes} };
 899     }
 900
 901     # /tweet, /tweet_as, /dm, /dm_as - complete @nicks (and nicks as the first
 902     # arg to dm)
 903     if ( $linestart =~ /^\/(?:tweet|dm)/ ) {
 904         my $prefix = $word =~ s/^@//;
 905         $prefix = 0 if $linestart eq '/dm' or $linestart eq '/dm_as';
 906         push @$complist, grep /^\Q$word/i,
 907           sort { $nicks{$b} <=> $nicks{$a} } keys %nicks;
 908         @$complist = map { "\@$_" } @$complist if $prefix;
 909     }
 910 }
 911
 912 Irssi::settings_add_str( "twirssi", "twitter_window",     "twitter" );
 913 Irssi::settings_add_str( "twirssi", "bitlbee_server",     "bitlbee" );
 914 Irssi::settings_add_str( "twirssi", "short_url_provider", "TinyURL" );
 915 Irssi::settings_add_str( "twirssi", "twirssi_location",
 916     ".irssi/scripts/twirssi.pl" );
 917 Irssi::settings_add_str( "twirssi", "twitter_usernames", undef );
 918 Irssi::settings_add_str( "twirssi", "twitter_passwords", undef );
 919 Irssi::settings_add_str( "twirssi", "twirssi_replies_store",
 920     ".irssi/scripts/twirssi.json" );
 921 Irssi::settings_add_bool( "twirssi", "tweet_to_away",             0 );
 922 Irssi::settings_add_bool( "twirssi", "show_reply_context",        0 );
 923 Irssi::settings_add_bool( "twirssi", "show_own_tweets",           1 );
 924 Irssi::settings_add_bool( "twirssi", "twirssi_debug",             0 );
 925 Irssi::settings_add_bool( "twirssi", "twirssi_first_run",         1 );
 926 Irssi::settings_add_bool( "twirssi", "twirssi_track_replies",     1 );
 927 Irssi::settings_add_bool( "twirssi", "twirssi_use_reply_aliases", 0 );
 928 $window = Irssi::window_find_name( Irssi::settings_get_str('twitter_window') );
 929
 930 if ($window) {
 931     Irssi::command_bind( "dm",               "cmd_direct" );
 932     Irssi::command_bind( "dm_as",            "cmd_direct_as" );
 933     Irssi::command_bind( "tweet",            "cmd_tweet" );
 934     Irssi::command_bind( "tweet_as",         "cmd_tweet_as" );
 935     Irssi::command_bind( "twitter_reply",    "cmd_reply" );
 936     Irssi::command_bind( "twitter_reply_as", "cmd_reply_as" );
 937     Irssi::command_bind( "twitter_login",    "cmd_login" );
 938     Irssi::command_bind( "twitter_logout",   "cmd_logout" );
 939     Irssi::command_bind( "twitter_switch",   "cmd_switch" );
 940     Irssi::command_bind( "twirssi_upgrade",  "cmd_upgrade" );
 941     if ( Irssi::settings_get_bool("twirssi_use_reply_aliases") ) {
 942         Irssi::command_bind( "reply",    "cmd_reply" );
 943         Irssi::command_bind( "reply_as", "cmd_reply_as" );
 944     }
 945     Irssi::command_bind(
 946         "twirssi_dump",
 947         sub {
 948             print "twits: ", join ", ",
 949               map { "u: $_->{username}" } values %twits;
 950             print "friends: ", join ", ", sort keys %friends;
 951             print "nicks: ",   join ", ", sort keys %nicks;
 952             print "id_map: ", Dumper \%{ $id_map{__indexes} };
 953             print "last poll: $last_poll";
 954         }
 955     );
 956     Irssi::command_bind(
 957         "twirssi_version",
 958         sub {
 959             &notice("Twirssi v$VERSION (r$REV); "
 960                   . "Net::Twitter v$Net::Twitter::VERSION. "
 961                   . "JSON in use: "
 962                   . JSON::Any::handler()
 963                   . ".  See details at http://twirssi.com/" );
 964         }
 965     );
 966     Irssi::command_bind(
 967         "twitter_friend",
 968         &gen_cmd(
 969             "/twitter_friend <username>",
 970             "create_friend",
 971             sub { &notice("Following $_[0]"); $nicks{ $_[0] } = time; }
 972         )
 973     );
 974     Irssi::command_bind(
 975         "twitter_unfriend",
 976         &gen_cmd(
 977             "/twitter_unfriend <username>",
 978             "destroy_friend",
 979             sub { &notice("Stopped following $_[0]"); delete $nicks{ $_[0] }; }
 980         )
 981     );
 982     Irssi::command_bind( "twitter_updates", "get_updates" );
 983     Irssi::signal_add_last( 'complete word' => \&sig_complete );
 984
 985     &notice("  %Y<%C(%B^%C)%N                   TWIRSSI v%R$VERSION%N (r$REV)");
 986     &notice("   %C(_(\\%N           http://twirssi.com/ for full docs");
 987     &notice(
 988         "    %Y||%C `%N Log in with /twitter_login, send updates with /tweet");
 989
 990     my $file = Irssi::settings_get_str("twirssi_replies_store");
 991     if ( $file and -r $file ) {
 992         if ( open( JSON, $file ) ) {
 993             local $/;
 994             my $json = <JSON>;
 995             close JSON;
 996             eval {
 997                 my $ref = JSON::Any->jsonToObj($json);
 998                 %id_map = %$ref;
 999                 my $num = keys %{ $id_map{__indexes} };
1000                 &notice( sprintf "Loaded old replies from %d contact%s.",
1001                     $num, ( $num == 1 ? "" : "s" ) );
1002             };
1003         } else {
1004             &notice("Failed to load old replies from $file: $!");
1005         }
1006     }
1007
1008     if ( my $provider = Irssi::settings_get_str("short_url_provider") ) {
1009         eval "use WWW::Shorten::$provider;";
1010
1011         if ($@) {
1012             &notice(
1013 "Failed to load WWW::Shorten::$provider - either clear short_url_provider or install the CPAN module"
1014             );
1015         }
1016     }
1017
1018     if (    my $autouser = Irssi::settings_get_str("twitter_usernames")
1019         and my $autopass = Irssi::settings_get_str("twitter_passwords") )
1020     {
1021         &cmd_login();
1022     }
1023
1024 } else {
1025     Irssi::active_win()
1026       ->print( "Create a window named "
1027           . Irssi::settings_get_str('twitter_window')
1028           . " or change the value of twitter_window.  Then, reload twirssi." );
1029 }
1030