added preliminary support for identi.ca
[twirssi-net-twitter-lite.git] / twirssi.pl
index 757cb708f9b5c1b718f36fbacc0d66de0f944409..7f2f496d28996af95cb1b7106ce64bdc2098d38f 100644 (file)
@@ -6,22 +6,23 @@ use HTML::Entities;
 use File::Temp;
 use LWP::Simple;
 use Data::Dumper;
+use Net::Identica;
 use Net::Twitter;
 $Data::Dumper::Indent = 1;
 
 use vars qw($VERSION %IRSSI);
 
-$VERSION = "2.0.2";
-my ($REV) = '$Rev: 452 $' =~ /(\d+)/;
+$VERSION = "2.0.6";
+my ($REV) = '$Rev: 483 $' =~ /(\d+)/;
 %IRSSI = (
     authors     => 'Dan Boger',
     contact     => 'zigdon@gmail.com',
     name        => 'twirssi',
-    description => 'Send twitter updates using /tweet.  '
+    description => 'Send twitter/identica updates using /tweet or /dent.  '
       . 'Can optionally set your bitlbee /away message to same',
     license => 'GNU GPL v2',
     url     => 'http://twirssi.com',
-    changed => '$Date: 2009-02-03 10:38:51 -0800 (Tue, 03 Feb 2009) $',
+    changed => '$Date: 2009-02-20 14:37:28 -0800 (Fri, 20 Feb 2009) $',
 );
 
 my $window;
@@ -81,21 +82,26 @@ sub cmd_direct_as {
     return unless &valid_username($username);
 
     eval {
-        unless ( $twits{$username}
+        if ( $twits{$username}
             ->new_direct_message( { user => $target, text => $text } ) )
         {
+            &notice("DM sent to $target");
+            $nicks{$target} = time;
+        } else {
+            my $error;
+            eval {
+                $error = JSON::Any->jsonToObj( $twits{$username}->get_error() );
+                $error = $error->{error};
+            };
+            die $error if $error;
             &notice("DM to $target failed");
-            return;
         }
     };
 
     if ($@) {
-        &notice("DM caused an error.  Aborted");
+        &notice("DM caused an error: $@");
         return;
     }
-
-    &notice("DM sent to $target");
-    $nicks{$target} = time;
 }
 
 sub cmd_tweet {
@@ -118,6 +124,7 @@ sub cmd_tweet_as {
     return unless &logged_in($twit);
 
     $data =~ s/^\s+|\s+$//;
+    $data =~ s/\s\s+/ /g;
     my ( $username, $data ) = split ' ', $data, 2;
 
     unless ( $username and $data ) {
@@ -140,16 +147,18 @@ sub cmd_tweet_as {
 
     return if &too_long($data);
 
+    my $success = 1;
     eval {
         unless ( $twits{$username}->update($data) )
         {
             &notice("Update failed");
-            return;
+            $success = 0;
         }
     };
+    return unless $success;
 
     if ($@) {
-        &notice("Update caused an error.  Aborted.");
+        &notice("Update caused an error: $@.  Aborted.");
         return;
     }
 
@@ -226,17 +235,20 @@ sub cmd_reply_as {
         $data = "\@$nick " . $data;
     }
 
-    if ( Irssi::settings_get_str("short_url_provider") ) {
-        foreach my $url ( $data =~ /(https?:\/\/\S+[\w\/])/g ) {
-            eval {
-                my $short = makeashorterlink($url);
-                $data =~ s/\Q$url/$short/g;
-            };
+    if ( &too_long( $data, 1 ) ) {
+        if ( Irssi::settings_get_str("short_url_provider") ) {
+            foreach my $url ( $data =~ /(https?:\/\/\S+[\w\/])/g ) {
+                eval {
+                    my $short = makeashorterlink($url);
+                    $data =~ s/\Q$url/$short/g;
+                };
+            }
         }
     }
 
     return if &too_long($data);
 
+    my $success = 1;
     eval {
         unless (
             $twits{$username}->update(
@@ -248,12 +260,13 @@ sub cmd_reply_as {
           )
         {
             &notice("Update failed");
-            return;
+            $success = 0;
         }
     };
+    return unless $success;
 
     if ($@) {
-        &notice("Update caused an error.  Aborted");
+        &notice("Update caused an error: $@.  Aborted");
         return;
     }
 
@@ -280,13 +293,15 @@ sub gen_cmd {
             return;
         }
 
+        my $success = 1;
         eval {
             unless ( $twit->$api_name($data) )
             {
                 &notice("$api_name failed");
-                return;
+                $success = 0;
             }
         };
+        return unless $success;
 
         if ($@) {
             &notice("$api_name caused an error.  Aborted.");
@@ -310,7 +325,26 @@ sub cmd_switch {
     }
 }
 
-sub cmd_logout {
+sub cmd_logout_twitter {
+    my ( $data, $server, $win ) = @_;
+
+    $data =~ s/^\s+|\s+$//g;
+    $data = $user unless $data;
+    return unless &valid_username($data);
+
+    &notice("Logging out $data...");
+    $twits{$data}->end_session();
+    delete $twits{$data};
+    undef $twit;
+    if ( keys %twits ) {
+        &cmd_switch( ( keys %twits )[0], $server, $win );
+    } else {
+        Irssi::timeout_remove($poll) if $poll;
+        undef $poll;
+    }
+}
+
+sub cmd_logout_identica {
     my ( $data, $server, $win ) = @_;
 
     $data =~ s/^\s+|\s+$//g;
@@ -329,13 +363,13 @@ sub cmd_logout {
     }
 }
 
-sub cmd_login {
+sub cmd_login_twitter {
     my ( $data, $server, $win ) = @_;
     my $pass;
     if ($data) {
         ( $user, $pass ) = split ' ', $data, 2;
-    } elsif ( my $autouser = Irssi::settings_get_str("twitter_usernames")
-        and my $autopass = Irssi::settings_get_str("twitter_passwords") )
+    } elsif ( my $autouser = Irssi::settings_get_str("twitter_usernames_twitter")
+        and my $autopass = Irssi::settings_get_str("twitter_passwords_twitter") )
     {
         my @user = split /\s*,\s*/, $autouser;
         my @pass = split /\s*,\s*/, $autopass;
@@ -347,13 +381,13 @@ sub cmd_login {
             while ( @user and @pass ) {
                 $u = shift @user;
                 $p = shift @pass;
-                &cmd_login("$u $p");
+                &cmd_login_twitter("$u $p");
             }
             return;
         }
     } else {
         &notice("/twitter_login requires either a username and password "
-              . "or twitter_usernames and twitter_passwords to be set." );
+              . "or twitter_usernames_twitter and twitter_passwords_twitter to be set." );
         return;
     }
 
@@ -407,7 +441,85 @@ sub cmd_login {
     }
 }
 
-sub cmd_add_search {
+sub cmd_login_identica {
+    my ( $data, $server, $win ) = @_;
+    my $pass;
+    if ($data) {
+        ( $user, $pass ) = split ' ', $data, 2;
+    } elsif ( my $autouser = Irssi::settings_get_str("twitter_usernames_identica")
+        and my $autopass = Irssi::settings_get_str("twitter_passwords_identica") )
+    {
+        my @user = split /\s*,\s*/, $autouser;
+        my @pass = split /\s*,\s*/, $autopass;
+        if ( @user != @pass ) {
+            &notice("Number of usernames doesn't match "
+                  . "the number of passwords - auto-login failed" );
+        } else {
+            my ( $u, $p );
+            while ( @user and @pass ) {
+                $u = shift @user;
+                $p = shift @pass;
+                &cmd_login_identica("$u $p");
+            }
+            return;
+        }
+    } else {
+        &notice("/identica_login requires either a username and password "
+              . "or twitter_usernames_identica and twitter_passwords_identica to be set." );
+        return;
+    }
+
+    %friends = %nicks = ();
+
+    $twit = Net::Identica->new(
+        username => $user,
+        password => $pass,
+        source   => "twirssi"
+    );
+
+    unless ( $twit->verify_credentials() ) {
+        &notice("Login as $user failed");
+        $twit = undef;
+        if ( keys %twits ) {
+            &cmd_switch( ( keys %twits )[0], $server, $win );
+        }
+        return;
+    }
+
+    if ($twit) {
+        my $rate_limit = $twit->rate_limit_status();
+        if ( $rate_limit and $rate_limit->{remaining_hits} < 1 ) {
+            &notice(
+                "Rate limit exceeded, try again after $rate_limit->{reset_time}"
+            );
+            $twit = undef;
+            return;
+        }
+
+        $twits{$user} = $twit;
+        Irssi::timeout_remove($poll) if $poll;
+        $poll = Irssi::timeout_add( &get_poll_time * 1000, \&get_updates, "" );
+        &notice("Logged in as $user, loading friends list...");
+        &load_friends();
+        &notice( "loaded friends: ", scalar keys %friends );
+        if ( Irssi::settings_get_bool("twirssi_first_run") ) {
+            Irssi::settings_set_bool( "twirssi_first_run", 0 );
+            unless ( exists $friends{twirssi} ) {
+                &notice("Welcome to twirssi!"
+                      . "  Perhaps you should add \@twirssi to your friends list,"
+                      . " so you can be notified when a new version is release?"
+                      . "  Just type /twitter_friend twirssi." );
+            }
+        }
+        %nicks = %friends;
+        $nicks{$user} = 0;
+        &get_updates;
+    } else {
+        &notice("Login failed");
+    }
+}
+
+sub cmd_add_search_twitter {
     my ( $data, $server, $win ) = @_;
 
     unless ( $twit and $twit->can('search') ) {
@@ -433,7 +545,33 @@ sub cmd_add_search {
     &notice("Added subscription for '$data'");
 }
 
-sub cmd_del_search {
+sub cmd_add_search_identica {
+    my ( $data, $server, $win ) = @_;
+
+    unless ( $twit and $twit->can('search') ) {
+        &notice("ERROR: Your version of Net::Identica ($Net::Identica::VERSION) "
+              . "doesn't support searches." );
+        return;
+    }
+
+    $data =~ s/^\s+|\s+$//;
+    $data = lc $data;
+
+    unless ($data) {
+        &notice("Usage: /twitter_subscribe <topic>");
+        return;
+    }
+
+    if ( exists $id_map{__searches}{$user}{$data} ) {
+        &notice("Already had a subscription for '$data'");
+        return;
+    }
+
+    $id_map{__searches}{$user}{$data} = 1;
+    &notice("Added subscription for '$data'");
+}
+
+sub cmd_del_search_twitter {
     my ( $data, $server, $win ) = @_;
 
     unless ( $twit and $twit->can('search') ) {
@@ -458,6 +596,31 @@ sub cmd_del_search {
     &notice("Removed subscription for '$data'");
 }
 
+sub cmd_del_search_identica {
+    my ( $data, $server, $win ) = @_;
+
+    unless ( $twit and $twit->can('search') ) {
+        &notice("ERROR: Your version of Net::Identica ($Net::Identica::VERSION) "
+              . "doesn't support searches." );
+        return;
+    }
+    $data =~ s/^\s+|\s+$//;
+    $data = lc $data;
+
+    unless ($data) {
+        &notice("Usage: /identica_unsubscribe <topic>");
+        return;
+    }
+
+    unless ( exists $id_map{__searches}{$user}{$data} ) {
+        &notice("No subscription found for '$data'");
+        return;
+    }
+
+    delete $id_map{__searches}{$user}{$data};
+    &notice("Removed subscription for '$data'");
+}
+
 sub cmd_list_search {
     my ( $data, $server, $win ) = @_;
 
@@ -489,13 +652,6 @@ sub cmd_upgrade {
         return;
     }
 
-    if ( not -x "/usr/bin/md5sum" and not $data ) {
-        &notice(
-"/usr/bin/md5sum can't be found - try '/twirssi_upgrade nomd5' to skip MD5 verification"
-        );
-        return;
-    }
-
     my $md5;
     unless ($data) {
         eval { use Digest::MD5; };
@@ -688,7 +844,8 @@ sub do_updates {
     };
 
     if ($@) {
-        print $fh "type:debug Error during friends_timeline call.  Aborted.\n";
+        print $fh
+          "type:debug Error during friends_timeline call: $@.  Aborted.\n";
         return 1;
     }
 
@@ -850,8 +1007,11 @@ sub monitor_child {
                 }
             }
 
-            next if exists $meta{id} and exists $tweet_cache{ $meta{id} };
-            $tweet_cache{ $meta{id} } = time;
+            if ( not $meta{type} or $meta{type} ne 'searchid' ) {
+                next if exists $meta{id} and exists $tweet_cache{ $meta{id} };
+                $tweet_cache{ $meta{id} } = time;
+            }
+
             my $account = "";
             if ( $meta{account} ne $user ) {
                 $account = "$meta{account}: ";
@@ -871,7 +1031,9 @@ sub monitor_child {
 
             my $hilight_color =
               $irssi_to_mirc_colors{ Irssi::settings_get_str("hilight_color") };
-            if ( $_ =~ /\@$meta{account}\W/i ) {
+            if ( ( $_ =~ /\@$meta{account}\W/i )
+                && Irssi::settings_get_bool("twirssi_hilights") )
+            {
                 $meta{nick} = "\cC$hilight_color$meta{nick}\cO";
                 $hilight = MSGLEVEL_HILIGHT;
             }
@@ -889,6 +1051,12 @@ sub monitor_child {
                     $meta{type}, $account, $meta{topic},
                     $meta{nick}, $marker,  $_
                   ];
+                if ( $meta{id} >
+                    $id_map{__searches}{ $meta{account} }{ $meta{topic} } )
+                {
+                    $id_map{__searches}{ $meta{account} }{ $meta{topic} } =
+                      $meta{id};
+                }
             } elsif ( $meta{type} eq 'dm' ) {
                 push @lines,
                   [
@@ -905,7 +1073,6 @@ sub monitor_child {
                 } elsif (&debug) {
                     print "Search '$meta{topic}' returned invalid id $meta{id}";
                 }
-                print "Search '$meta{topic}' id set to $meta{id}" if &debug;
             } elsif ( $meta{type} eq 'error' ) {
                 push @lines, [ MSGLEVEL_MSGS, $_ ];
             } elsif ( $meta{type} eq 'debug' ) {
@@ -965,12 +1132,23 @@ sub monitor_child {
 
     close FILE;
 
-    if ( $attempt < 12 ) {
+    if ( $attempt < 24 ) {
         Irssi::timeout_add_once( 5000, 'monitor_child',
             [ $filename, $attempt + 1 ] );
     } else {
-        &notice("Giving up on polling $filename");
+        print "Giving up on polling $filename" if &debug;
         unlink $filename unless &debug;
+
+        return unless Irssi::settings_get_bool("twirssi_notify_timeouts");
+
+        my $since;
+        my @time = localtime($last_poll);
+        if ( time - $last_poll < 24 * 60 * 60 ) {
+            $since = sprintf( "%d:%02d", @time[ 2, 1 ] );
+        } else {
+            $since = scalar localtime($last_poll);
+        }
+        &notice("Haven't been able to get updated tweets since $since");
     }
 }
 
@@ -1085,8 +1263,16 @@ sub get_poll_time {
 sub hilight {
     my $text = shift;
 
-    $text =~ s/(^|\W)\@([-\w]+)/$1\cC12\@$2\cO/g;
-    $text =~ s/(^|\W)\#([-\w]+)/$1\cC5\#$2\cO/g;
+    if ( Irssi::settings_get_str("twirssi_nick_color") ) {
+        my $c = Irssi::settings_get_str("twirssi_nick_color");
+        $c = $irssi_to_mirc_colors{$c};
+        $text =~ s/(^|\W)\@([-\w]+)/$1\cC$c\@$2\cO/g if $c;
+    }
+    if ( Irssi::settings_get_str("twirssi_topic_color") ) {
+        my $c = Irssi::settings_get_str("twirssi_topic_color");
+        $c = $irssi_to_mirc_colors{$c};
+        $text =~ s/(^|\W)\#([-\w]+)/$1\cC$c\#$2\cO/g if $c;
+    }
     $text =~ s/[\n\r]/ /g;
 
     return $text;
@@ -1114,6 +1300,8 @@ Irssi::settings_add_str( "twirssi", "twitter_usernames", undef );
 Irssi::settings_add_str( "twirssi", "twitter_passwords", undef );
 Irssi::settings_add_str( "twirssi", "twirssi_replies_store",
     ".irssi/scripts/twirssi.json" );
+Irssi::settings_add_str( "twirssi", "twirssi_nick_color",  "%B" );
+Irssi::settings_add_str( "twirssi", "twirssi_topic_color", "%r" );
 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 );
@@ -1122,6 +1310,8 @@ Irssi::settings_add_bool( "twirssi", "twirssi_first_run",         1 );
 Irssi::settings_add_bool( "twirssi", "twirssi_track_replies",     1 );
 Irssi::settings_add_bool( "twirssi", "twirssi_replies_autonick",  1 );
 Irssi::settings_add_bool( "twirssi", "twirssi_use_reply_aliases", 0 );
+Irssi::settings_add_bool( "twirssi", "twirssi_notify_timeouts",   1 );
+Irssi::settings_add_bool( "twirssi", "twirssi_hilights",          1 );
 Irssi::settings_add_bool( "twirssi", "tweet_window_input",        0 );
 
 $last_poll = time - &get_poll_time;
@@ -1137,14 +1327,22 @@ if ($window) {
     Irssi::command_bind( "dm",                         "cmd_direct" );
     Irssi::command_bind( "dm_as",                      "cmd_direct_as" );
     Irssi::command_bind( "tweet",                      "cmd_tweet" );
+    Irssi::command_bind( "dent",                       "cmd_tweet" );
     Irssi::command_bind( "tweet_as",                   "cmd_tweet_as" );
+    Irssi::command_bind( "dent_as",                   "cmd_tweet_as" );
     Irssi::command_bind( "twitter_reply",              "cmd_reply" );
+    Irssi::command_bind( "identica_reply",             "cmd_reply" );
     Irssi::command_bind( "twitter_reply_as",           "cmd_reply_as" );
-    Irssi::command_bind( "twitter_login",              "cmd_login" );
-    Irssi::command_bind( "twitter_logout",             "cmd_logout" );
+    Irssi::command_bind( "identica_reply_as",          "cmd_reply_as" );
+    Irssi::command_bind( "twitter_login",              "cmd_login_twitter" );
+    Irssi::command_bind( "twitter_logout",             "cmd_logout_twitter" );
+    Irssi::command_bind( "identica_login",             "cmd_login_identica" );
+    Irssi::command_bind( "identica_logout",            "cmd_logout_identica" );
     Irssi::command_bind( "twitter_switch",             "cmd_switch" );
-    Irssi::command_bind( "twitter_subscribe",          "cmd_add_search" );
-    Irssi::command_bind( "twitter_unsubscribe",        "cmd_del_search" );
+    Irssi::command_bind( "twitter_subscribe",          "cmd_add_search_twitter" );
+    Irssi::command_bind( "identica_subscribe",         "cmd_add_search_identica" );
+    Irssi::command_bind( "twitter_unsubscribe",        "cmd_del_search_twitter" );
+    Irssi::command_bind( "identica_unsubscribe",       "cmd_del_search_identica" );
     Irssi::command_bind( "twitter_list_subscriptions", "cmd_list_search" );
     Irssi::command_bind( "twirssi_upgrade",            "cmd_upgrade" );
     if ( Irssi::settings_get_bool("twirssi_use_reply_aliases") ) {
@@ -1166,7 +1364,8 @@ if ($window) {
         "twirssi_version",
         sub {
             &notice("Twirssi v$VERSION (r$REV); "
-                  . "Net::Twitter v$Net::Twitter::VERSION. "
+                  . "Net::Identica v$Net::Identica::VERSION. "
+                 . "Net::Twitter v$Net::Twitter::VERSION."
                   . "JSON in use: "
                   . JSON::Any::handler()
                   . ".  See details at http://twirssi.com/" );
@@ -1194,7 +1393,7 @@ if ($window) {
     &notice("  %Y<%C(%B^%C)%N                   TWIRSSI v%R$VERSION%N (r$REV)");
     &notice("   %C(_(\\%N           http://twirssi.com/ for full docs");
     &notice(
-        "    %Y||%C `%N Log in with /twitter_login, send updates with /tweet");
+        "    %Y||%C `%N Log in with /twitter_login or /identica_login, send updates with /tweet or /dent");
 
     my $file = Irssi::settings_get_str("twirssi_replies_store");
     if ( $file and -r $file ) {
@@ -1224,10 +1423,16 @@ if ($window) {
         }
     }
 
-    if (    my $autouser = Irssi::settings_get_str("twitter_usernames")
-        and my $autopass = Irssi::settings_get_str("twitter_passwords") )
+    if (    my $autouser = Irssi::settings_get_str("twitter_usernames_twitter")
+        and my $autopass = Irssi::settings_get_str("twitter_passwords_twitter") )
     {
-        &cmd_login();
+        &cmd_login_twitter();
+    }
+
+    if (    my $autouser = Irssi::settings_get_str("twitter_usernames_identica")
+        and my $autopass = Irssi::settings_get_str("twitter_passwords_identica") )
+    {
+        &cmd_login_identica();
     }
 
 } else {
@@ -1235,6 +1440,4 @@ if ($window) {
       ->print( "Create a window named "
           . Irssi::settings_get_str('twitter_window')
           . " or change the value of twitter_window.  Then, reload twirssi." );
-}
-
-# vim: set sts=4 expandtab:
+}
\ No newline at end of file