source: perl/modules/Jabber/lib/BarnOwl/Module/Jabber.pm @ 86840c5

barnowl_perlaimdebianrelease-1.4release-1.5release-1.6release-1.7release-1.8release-1.9
Last change on this file since 86840c5 was 86840c5, checked in by Nelson Elhage <nelhage@mit.edu>, 14 years ago
Validate JIDs passed to jmuc join. closes #25
  • Property mode set to 100644
File size: 38.9 KB
Line 
1use strict;
2use warnings;
3
4package BarnOwl::Module::Jabber;
5
6=head1 NAME
7
8BarnOwl::Module::Jabber
9
10=head1 DESCRIPTION
11
12This module implements Jabber support for barnowl.
13
14=cut
15
16use BarnOwl;
17use BarnOwl::Hooks;
18use BarnOwl::Message::Jabber;
19use BarnOwl::Module::Jabber::Connection;
20use BarnOwl::Module::Jabber::ConnectionManager;
21
22use Authen::SASL qw(Perl);
23use Net::Jabber;
24use Net::Jabber::MUC;
25use Net::DNS;
26use Getopt::Long;
27
28our $VERSION = 0.1;
29
30BEGIN {
31    if(eval {require IO::Socket::SSL;}) {
32        if($IO::Socket::SSL::VERSION eq "0.97") {
33            BarnOwl::error("You are using IO::Socket:SSL 0.97, which \n" .
34                           "contains bugs causing it not to work with barnowl's\n" .
35                           "jabber support. We recommend updating to the latest\n" .
36                           "IO::Socket::SSL from CPAN. \n");
37            die("Not loading Jabber.par\n");
38        }
39    }
40}
41
42no warnings 'redefine';
43
44################################################################################
45# owl perl jabber support
46#
47# XXX Todo:
48# Rosters for MUCs
49# More user feedback
50#  * joining MUC
51#  * parting MUC
52#  * presence (Roster and MUC)
53# Implementing formatting and logging callbacks for C
54# Appropriate callbacks for presence subscription messages.
55#
56################################################################################
57
58our $conn;
59$conn ||= BarnOwl::Module::Jabber::ConnectionManager->new;
60our %vars;
61
62sub onStart {
63    if ( *BarnOwl::queue_message{CODE} ) {
64        register_owl_commands();
65        register_keybindings();
66        register_filters();
67        $BarnOwl::Hooks::mainLoop->add(\&onMainLoop);
68        $BarnOwl::Hooks::getBuddyList->add(\&onGetBuddyList);
69        $vars{show} = '';
70        BarnOwl::new_variable_bool("jabber:show_offline_buddies",
71                                   { default => 1,
72                                     summary => 'Show offline or pending buddies.'});
73        BarnOwl::new_variable_int("jabber:auto_away_timeout",
74                                  { default => 5,
75                                    summary => 'After minutes idle, auto away.',
76                                  });
77        BarnOwl::new_variable_int("jabber:auto_xa_timeout",
78                                  { default => 15,
79                                    summary => 'After minutes idle, auto extended away.'
80                                });
81    } else {
82        # Our owl doesn't support queue_message. Unfortunately, this
83        # means it probably *also* doesn't support BarnOwl::error. So just
84        # give up silently.
85    }
86}
87
88$BarnOwl::Hooks::startup->add(\&onStart);
89
90sub onMainLoop {
91    return if ( !$conn->connected() );
92
93    $vars{status_changed} = 0;
94    my $auto_away = BarnOwl::getvar('jabber:auto_away_timeout');
95    my $auto_xa = BarnOwl::getvar('jabber:auto_xa_timeout');
96    my $idletime = BarnOwl::getidletime();
97    if ($auto_xa != 0 && $idletime >= (60 * $auto_xa) && ($vars{show} eq 'away' || $vars{show} eq '' )) {
98        $vars{show} = 'xa';
99        $vars{status} = 'Auto extended-away after '.$auto_xa.' minute'.($auto_xa == 1 ? '' : 's').' idle.';
100        $vars{status_changed} = 1;
101    } elsif ($auto_away != 0 && $idletime >= (60 * $auto_away) && $vars{show} eq '') {
102        $vars{show} = 'away';
103        $vars{status} = 'Auto away after '.$auto_away.' minute'.($auto_away == 1 ? '' : 's').' idle.';
104        $vars{status_changed} = 1;
105    } elsif ($idletime == 0 && $vars{show} ne '') {
106        $vars{show} = '';
107        $vars{status} = '';
108        $vars{status_changed} = 1;
109    }
110
111    foreach my $jid ( $conn->getJIDs() ) {
112        my $client = $conn->getConnectionFromJID($jid);
113
114        unless($client) {
115            $conn->removeConnection($jid);
116            BarnOwl::error("Connection for $jid undefined -- error in reload?");
117        }
118
119        my $status = $client->Process(0);
120        if ( !defined($status) ) {
121            BarnOwl::error("Jabber account $jid disconnected!");
122            do_logout($jid);
123        }
124        if ($::shutdown) {
125            do_logout($jid);
126            return;
127        }
128        if ($vars{status_changed}) {
129            my $p = new Net::Jabber::Presence;
130            $p->SetShow($vars{show}) if $vars{show};
131            $p->SetStatus($vars{status}) if $vars{status};
132            $client->Send($p);
133        }
134    }
135}
136
137sub blist_listBuddy {
138    my $roster = shift;
139    my $buddy  = shift;
140    my $blistStr .= "    ";
141    my %jq  = $roster->query($buddy);
142    my $res = $roster->resource($buddy);
143
144    my $name = $jq{name} || $buddy->GetUserID();
145
146    $blistStr .= sprintf '%-15s %s', $name, $buddy->GetJID();
147
148    if ($res) {
149        my %rq = $roster->resourceQuery( $buddy, $res );
150        $blistStr .= " [" . ( $rq{show} ? $rq{show} : 'online' ) . "]";
151        $blistStr .= " " . $rq{status} if $rq{status};
152        $blistStr = BarnOwl::Style::boldify($blistStr);
153    }
154    else {
155        return '' if (BarnOwl::getvar('jabber:show_offline_buddies') eq 'off');
156        if ($jq{ask}) {
157            $blistStr .= " [pending]";
158        }
159        elsif ($jq{subscription} eq 'none' || $jq{subscription} eq 'from') {
160            $blistStr .= " [not subscribed]";
161        }
162        else {
163            $blistStr .= " [offline]";
164        }
165    }
166    return $blistStr . "\n";
167}
168
169sub getSingleBuddyList {
170    my $jid = shift;
171    $jid = resolveConnectedJID($jid);
172    return "" unless $jid;
173    my $blist = "";
174    my $roster = $conn->getRosterFromJID($jid);
175    if ($roster) {
176        $blist .= "\n" . BarnOwl::Style::boldify("Jabber Roster for $jid\n");
177
178        foreach my $group ( $roster->groups() ) {
179            $blist .= "  Group: $group\n";
180            my @buddies = $roster->jids( 'group', $group );
181            foreach my $buddy ( @buddies ) {
182                $blist .= blist_listBuddy( $roster, $buddy );
183            }
184        }
185
186        my @unsorted = $roster->jids('nogroup');
187        if (@unsorted) {
188            $blist .= "  [unsorted]\n";
189            foreach my $buddy (@unsorted) {
190                $blist .= blist_listBuddy( $roster, $buddy );
191            }
192        }
193    }
194    return $blist;
195}
196
197sub onGetBuddyList {
198    my $blist = "";
199    foreach my $jid ($conn->getJIDs()) {
200        $blist .= getSingleBuddyList($jid);
201    }
202    return $blist;
203}
204
205################################################################################
206### Owl Commands
207sub register_owl_commands() {
208    BarnOwl::new_command(
209        jabberlogin => \&cmd_login,
210        {
211            summary => "Log into jabber",
212            usage   => "jabberlogin JID [PASSWORD]"
213        }
214    );
215    BarnOwl::new_command(
216        jabberlogout => \&cmd_logout,
217        { summary => "Log out of jabber" }
218    );
219    BarnOwl::new_command(
220        jwrite => \&cmd_jwrite,
221        {
222            summary => "Send a Jabber Message",
223            usage   => "jwrite JID [-t thread] [-s subject]"
224        }
225    );
226    BarnOwl::new_command(
227        jlist => \&cmd_jlist,
228        {
229            summary => "Show your Jabber roster.",
230            usage   => "jlist"
231        }
232    );
233    BarnOwl::new_command(
234        jmuc => \&cmd_jmuc,
235        {
236            summary     => "Jabber MUC related commands.",
237            description => "jmuc sends jabber commands related to muc.\n\n"
238              . "The following commands are available\n\n"
239              . "join <muc>  Join a muc.\n\n"
240              . "part <muc>  Part a muc.\n"
241              . "            The muc is taken from the current message if not supplied.\n\n"
242              . "invite <jid> <muc>\n"
243              . "            Invite <jid> to <muc>.\n"
244              . "            The muc is taken from the current message if not supplied.\n\n"
245              . "configure <muc>\n"
246              . "            Configures a MUC.\n"
247              . "            Necessary to initalize a new MUC.\n"
248              . "            At present, only the default configuration is supported.\n"
249              . "            The muc is taken from the current message if not supplied.\n\n"
250              . "presence <muc>\n"
251              . "            Shows the roster for <muc>.\n"
252              . "            The muc is taken from the current message if not supplied.\n\n"
253              . "presence -a\n"
254              . "            Shows rosters for all MUCs you're participating in.\n\n",
255            usage => "jmuc COMMAND ARGS"
256        }
257    );
258    BarnOwl::new_command(
259        jroster => \&cmd_jroster,
260        {
261            summary     => "Jabber Roster related commands.",
262            description => "jroster sends jabber commands related to rosters.\n\n"
263              . "The following commands are available\n\n"
264              . "sub <jid>     Subscribe to <jid>'s presence. (implicit add)\n\n"
265              . "add <jid>     Adds <jid> to your roster.\n\n"
266              . "unsub <jid>   Unsubscribe from <jid>'s presence.\n\n"
267              . "remove <jid>  Removes <jid> to your roster. (implicit unsub)\n\n"
268              . "auth <jid>    Authorizes <jid> to subscribe to your presence.\n\n"
269              . "deauth <jid>  De-authorizes <jid>'s subscription to your presence.\n\n"
270              . "The following arguments are supported for all commands\n\n"
271              . "-a <jid>      Specify which account to make the roster changes on.\n"
272              . "              Required if you're signed into more than one account.\n\n"
273              . "The following arguments only work with the add and sub commands.\n\n"
274              . "-g <group>    Add <jid> to group <group>.\n"
275              . "              May be specified more than once, will not remove <jid> from any groups.\n\n"
276              . "-p            Purge. Removes <jid> from all groups.\n"
277              . "              May be combined with -g completely alter <jid>'s groups.\n\n"
278              . "-n <name>     Sets <name> as <jid>'s short name.\n\n"
279              . "Note: Unless -n is used, you can specify multiple <jid> arguments.\n",
280            usage       => "jroster COMMAND ARGS"
281        }
282    );
283}
284
285sub register_keybindings {
286    BarnOwl::bindkey("recv j command start-command jwrite ");
287}
288
289sub register_filters {
290    BarnOwl::filter('jabber type ^jabber$');
291}
292
293sub cmd_login {
294    my $cmd = shift;
295    my $jid = new Net::Jabber::JID;
296    $jid->SetJID(shift);
297    my $password = '';
298    $password = shift if @_;
299
300    my $uid           = $jid->GetUserID();
301    my $componentname = $jid->GetServer();
302    my $resource      = $jid->GetResource() || 'owl';
303    $jid->SetResource($resource);
304    my $jidStr = $jid->GetJID('full');
305
306    if ( !$uid || !$componentname ) {
307        BarnOwl::error("usage: $cmd JID");
308        return;
309    }
310
311    if ( $conn->jidExists($jidStr) ) {
312        BarnOwl::error("Already logged in as $jidStr.");
313        return;
314    }
315
316    my ( $server, $port ) = getServerFromJID($jid);
317
318    $vars{jlogin_jid} = $jidStr;
319    $vars{jlogin_connhash} = {
320        hostname      => $server,
321        tls           => 1,
322        port          => $port,
323        componentname => $componentname
324    };
325    $vars{jlogin_authhash} =
326      { username => $uid,
327        resource => $resource,
328    };
329
330    return do_login($password);
331}
332
333sub do_login {
334    $vars{jlogin_password} = shift;
335    $vars{jlogin_authhash}->{password} = sub { return $vars{jlogin_password} || '' };
336    my $jidStr = $vars{jlogin_jid};
337    if ( !$jidStr && $vars{jlogin_havepass}) {
338        BarnOwl::error("Got password but have no jid!");
339    }
340    else
341    {
342        my $client = $conn->addConnection($jidStr);
343
344        #XXX Todo: Add more callbacks.
345        # * MUC presence handlers
346        # We use the anonymous subrefs in order to have the correct behavior
347        # when we reload
348        $client->SetMessageCallBacks(
349            chat      => sub { BarnOwl::Module::Jabber::process_incoming_chat_message(@_) },
350            error     => sub { BarnOwl::Module::Jabber::process_incoming_error_message(@_) },
351            groupchat => sub { BarnOwl::Module::Jabber::process_incoming_groupchat_message(@_) },
352            headline  => sub { BarnOwl::Module::Jabber::process_incoming_headline_message(@_) },
353            normal    => sub { BarnOwl::Module::Jabber::process_incoming_normal_message(@_) }
354        );
355        $client->SetPresenceCallBacks(
356            available    => sub { BarnOwl::Module::Jabber::process_presence_available(@_) },
357            unavailable  => sub { BarnOwl::Module::Jabber::process_presence_available(@_) },
358            subscribe    => sub { BarnOwl::Module::Jabber::process_presence_subscribe(@_) },
359            subscribed   => sub { BarnOwl::Module::Jabber::process_presence_subscribed(@_) },
360            unsubscribe  => sub { BarnOwl::Module::Jabber::process_presence_unsubscribe(@_) },
361            unsubscribed => sub { BarnOwl::Module::Jabber::process_presence_unsubscribed(@_) },
362            error        => sub { BarnOwl::Module::Jabber::process_presence_error(@_) });
363
364        my $status = $client->Connect( %{ $vars{jlogin_connhash} } );
365        if ( !$status ) {
366            $conn->removeConnection($jidStr);
367            BarnOwl::error("We failed to connect");
368        } else {
369            my @result = $client->AuthSend( %{ $vars{jlogin_authhash} } );
370
371            if ( !@result || $result[0] ne 'ok' ) {
372                if ( !$vars{jlogin_havepass} && ( !@result || $result[0] eq '401' ) ) {
373                    $vars{jlogin_havepass} = 1;
374                    $conn->removeConnection($jidStr);
375                    BarnOwl::start_password( "Password for $jidStr: ", \&do_login );
376                    return "";
377                }
378                $conn->removeConnection($jidStr);
379                BarnOwl::error( "Error in connect: " . join( " ", @result ) );
380            } else {
381                $conn->getRosterFromJID($jidStr)->fetch();
382                $client->PresenceSend( priority => 1 );
383                my $fullJid = $client->{SESSION}->{FULLJID} || $jidStr;
384                $conn->renameConnection($jidStr, $fullJid);
385                queue_admin_msg("Connected to jabber as $fullJid");
386            }
387        }
388    }
389    delete $vars{jlogin_jid};
390    $vars{jlogin_password} =~ tr/\0-\377/x/ if $vars{jlogin_password};
391    delete $vars{jlogin_password};
392    delete $vars{jlogin_havepass};
393    delete $vars{jlogin_connhash};
394    delete $vars{jlogin_authhash};
395
396    return "";
397}
398
399sub do_logout {
400    my $jid = shift;
401    my $disconnected = $conn->removeConnection($jid);
402    queue_admin_msg("Jabber disconnected ($jid).") if $disconnected;
403}
404
405sub cmd_logout {
406    # Logged into multiple accounts
407    if ( $conn->connected() > 1 ) {
408        # Logged into multiple accounts, no accout specified.
409        if ( !$_[1] ) {
410            my $errStr =
411              "You are logged into multiple accounts. Please specify an account to log out of.\n";
412            foreach my $jid ( $conn->getJIDs() ) {
413                $errStr .= "\t$jid\n";
414            }
415            queue_admin_msg($errStr);
416        }
417        # Logged into multiple accounts, account specified.
418        else {
419            if ( $_[1] eq '-a' )    #All accounts.
420            {
421                foreach my $jid ( $conn->getJIDs() ) {
422                    do_logout($jid);
423                }
424            }
425            else                    #One account.
426            {
427                my $jid = resolveConnectedJID( $_[1] );
428                do_logout($jid) if ( $jid ne '' );
429            }
430        }
431    }
432    else                            # Only one account logged in.
433    {
434        do_logout( ( $conn->getJIDs() )[0] );
435    }
436    return "";
437}
438
439sub cmd_jlist {
440    if ( !( scalar $conn->getJIDs() ) ) {
441        BarnOwl::error("You are not logged in to Jabber.");
442        return;
443    }
444    BarnOwl::popless_ztext( onGetBuddyList() );
445}
446
447sub cmd_jwrite {
448    if ( !$conn->connected() ) {
449        BarnOwl::error("You are not logged in to Jabber.");
450        return;
451    }
452
453    my $jwrite_to      = "";
454    my $jwrite_from    = "";
455    my $jwrite_sid     = "";
456    my $jwrite_thread  = "";
457    my $jwrite_subject = "";
458    my ($to, $from);
459    my $jwrite_type    = "chat";
460
461    my @args = @_;
462    shift;
463    local @ARGV = @_;
464    my $gc;
465    GetOptions(
466        'thread=s'  => \$jwrite_thread,
467        'subject=s' => \$jwrite_subject,
468        'account=s' => \$from,
469        'id=s'     =>  \$jwrite_sid,
470    );
471    $jwrite_type = 'groupchat' if $gc;
472
473    if ( scalar @ARGV != 1 ) {
474        BarnOwl::error(
475            "Usage: jwrite JID [-t thread] [-s 'subject'] [-a account]");
476        return;
477    }
478    else {
479        $to = shift @ARGV;
480    }
481
482    my @candidates = guess_jwrite($from, $to);
483
484    unless(scalar @candidates) {
485        die("Unable to resolve JID $to");
486    }
487
488    @candidates = grep {defined $_->[0]} @candidates;
489
490    unless(scalar @candidates) {
491        if(!$from) {
492            die("You must specify an account with -a");
493        } else {
494            die("Unable to resolve account $from");
495        }
496    }
497
498
499    ($jwrite_from, $jwrite_to, $jwrite_type) = @{$candidates[0]};
500
501    $vars{jwrite} = {
502        to      => $jwrite_to,
503        from    => $jwrite_from,
504        sid     => $jwrite_sid,
505        subject => $jwrite_subject,
506        thread  => $jwrite_thread,
507        type    => $jwrite_type
508    };
509
510    if(scalar @candidates > 1) {
511        BarnOwl::message(
512            "Warning: Guessing account and/or destination JID"
513           );
514    } else  {
515        BarnOwl::message(
516            "Type your message below.  End with a dot on a line by itself.  ^C will quit."
517           );
518    }
519
520    my $cmd = "jwrite $jwrite_to -a $jwrite_from";
521    $cmd .= " -t $jwrite_thread" if $jwrite_thread;
522    $cmd .= " -s $jwrite_subject" if $jwrite_subject;
523    BarnOwl::start_edit_win( $cmd, \&process_owl_jwrite );
524}
525
526sub cmd_jmuc {
527    die "You are not logged in to Jabber" unless $conn->connected();
528    my $ocmd = shift;
529    my $cmd  = shift;
530    if ( !$cmd ) {
531
532        #XXX TODO: Write general usage for jmuc command.
533        return;
534    }
535
536    my %jmuc_commands = (
537        join      => \&jmuc_join,
538        part      => \&jmuc_part,
539        invite    => \&jmuc_invite,
540        configure => \&jmuc_configure,
541        presence  => \&jmuc_presence
542    );
543    my $func = $jmuc_commands{$cmd};
544    if ( !$func ) {
545        BarnOwl::error("jmuc: Unknown command: $cmd");
546        return;
547    }
548
549    {
550        local @ARGV = @_;
551        my $jid;
552        my $muc;
553        my $m = BarnOwl::getcurmsg();
554        if ( $m && $m->is_jabber && $m->{jtype} eq 'groupchat' ) {
555            $muc = $m->{room};
556            $jid = $m->{to};
557        }
558
559        my $getopt = Getopt::Long::Parser->new;
560        $getopt->configure('pass_through');
561        $getopt->getoptions( 'account=s' => \$jid );
562        $jid ||= defaultJID();
563        if ($jid) {
564            $jid = resolveConnectedJID($jid);
565            return unless $jid;
566        }
567        else {
568            BarnOwl::error('You must specify an account with -a {jid}');
569        }
570        return $func->( $jid, $muc, @ARGV );
571    }
572}
573
574sub jmuc_join {
575    my ( $jid, $muc, @args ) = @_;
576    local @ARGV = @args;
577    my $password;
578    GetOptions( 'password=s' => \$password );
579
580    $muc = shift @ARGV
581      or die("Usage: jmuc join MUC [-p password] [-a account]");
582
583    die("Error: Must specify a fully-qualified MUC name (e.g. barnowl\@conference.mit.edu)\n")
584        unless $muc =~ /@/;
585    $muc = Net::Jabber::JID->new($muc);
586    $jid = Net::Jabber::JID->new($jid);
587    $muc->SetResource($jid->GetJID('full')) unless length $muc->GetResource();
588
589    $conn->getConnectionFromJID($jid)->MUCJoin(JID      => $muc,
590                                               Password => $password,
591                                               History  => {
592                                                   MaxChars => 0
593                                                  });
594    return;
595}
596
597sub jmuc_part {
598    my ( $jid, $muc, @args ) = @_;
599
600    $muc = shift @args if scalar @args;
601    die("Usage: jmuc part MUC [-a account]") unless $muc;
602
603    $conn->getConnectionFromJID($jid)->MUCLeave(JID => $muc);
604    queue_admin_msg("$jid has left $muc.");
605}
606
607sub jmuc_invite {
608    my ( $jid, $muc, @args ) = @_;
609
610    my $invite_jid = shift @args;
611    $muc = shift @args if scalar @args;
612
613    die('Usage: jmuc invite JID [muc] [-a account]')
614      unless $muc && $invite_jid;
615
616    my $message = Net::Jabber::Message->new();
617    $message->SetTo($muc);
618    my $x = $message->NewChild('http://jabber.org/protocol/muc#user');
619    $x->AddInvite();
620    $x->GetInvite()->SetTo($invite_jid);
621    $conn->getConnectionFromJID($jid)->Send($message);
622    queue_admin_msg("$jid has invited $invite_jid to $muc.");
623}
624
625sub jmuc_configure {
626    my ( $jid, $muc, @args ) = @_;
627    $muc = shift @args if scalar @args;
628    die("Usage: jmuc configure [muc]") unless $muc;
629    my $iq = Net::Jabber::IQ->new();
630    $iq->SetTo($muc);
631    $iq->SetType('set');
632    my $query = $iq->NewQuery("http://jabber.org/protocol/muc#owner");
633    my $x     = $query->NewChild("jabber:x:data");
634    $x->SetType('submit');
635
636    $conn->getConnectionFromJID($jid)->Send($iq);
637    queue_admin_msg("Accepted default instant configuration for $muc");
638}
639
640sub jmuc_presence_single {
641    my $m = shift;
642    my @jids = $m->Presence();
643    return "JIDs present in " . $m->BaseJID . "\n\t"
644      . join("\n\t", map {$_->GetResource}@jids) . "\n";
645}
646
647sub jmuc_presence {
648    my ( $jid, $muc, @args ) = @_;
649
650    $muc = shift @args if scalar @args;
651    die("Usage: jmuc presence MUC") unless $muc;
652
653    if ($muc eq '-a') {
654        my $str = "";
655        foreach my $jid ($conn->getJIDs()) {
656            $str .= BarnOwl::Style::boldify("Conferences for $jid:\n");
657            my $connection = $conn->getConnectionFromJID($jid);
658            foreach my $muc ($connection->MUCs) {
659                $str .= jmuc_presence_single($muc)."\n";
660            }
661        }
662        BarnOwl::popless_ztext($str);
663    }
664    else {
665        my $m = $conn->getConnectionFromJID($jid)->FindMUC(jid => $muc);
666        die("No such muc: $muc") unless $m;
667        BarnOwl::popless_ztext(jmuc_presence_single($m));
668    }
669}
670
671
672#XXX TODO: Consider merging this with jmuc and selecting off the first two args.
673sub cmd_jroster {
674    die "You are not logged in to Jabber" unless $conn->connected();
675    my $ocmd = shift;
676    my $cmd  = shift;
677    if ( !$cmd ) {
678
679        #XXX TODO: Write general usage for jroster command.
680        return;
681    }
682
683    my %jroster_commands = (
684        sub      => \&jroster_sub,
685        unsub    => \&jroster_unsub,
686        add      => \&jroster_add,
687        remove   => \&jroster_remove,
688        auth     => \&jroster_auth,
689        deauth   => \&jroster_deauth
690    );
691    my $func = $jroster_commands{$cmd};
692    if ( !$func ) {
693        BarnOwl::error("jroster: Unknown command: $cmd");
694        return;
695    }
696
697    {
698        local @ARGV = @_;
699        my $jid;
700        my $name;
701        my @groups;
702        my $purgeGroups;
703        my $getopt = Getopt::Long::Parser->new;
704        $getopt->configure('pass_through');
705        $getopt->getoptions(
706            'account=s' => \$jid,
707            'group=s' => \@groups,
708            'purgegroups' => \$purgeGroups,
709            'name=s' => \$name
710        );
711        $jid ||= defaultJID();
712        if ($jid) {
713            $jid = resolveConnectedJID($jid);
714            return unless $jid;
715        }
716        else {
717            BarnOwl::error('You must specify an account with -a {jid}');
718        }
719        return $func->( $jid, $name, \@groups, $purgeGroups,  @ARGV );
720    }
721}
722
723sub jroster_sub {
724    my $jid = shift;
725    my $name = shift;
726    my @groups = @{ shift() };
727    my $purgeGroups = shift;
728    my $baseJID = baseJID($jid);
729
730    my $roster = $conn->getRosterFromJID($jid);
731
732    # Adding lots of users with the same name is a bad idea.
733    $name = "" unless (1 == scalar(@ARGV));
734
735    my $p = new Net::Jabber::Presence;
736    $p->SetType('subscribe');
737
738    foreach my $to (@ARGV) {
739        jroster_add($jid, $name, \@groups, $purgeGroups, ($to)) unless ($roster->exists($to));
740
741        $p->SetTo($to);
742        $conn->getConnectionFromJID($jid)->Send($p);
743        queue_admin_msg("You ($baseJID) have requested a subscription to ($to)'s presence.");
744    }
745}
746
747sub jroster_unsub {
748    my $jid = shift;
749    my $name = shift;
750    my @groups = @{ shift() };
751    my $purgeGroups = shift;
752    my $baseJID = baseJID($jid);
753
754    my $p = new Net::Jabber::Presence;
755    $p->SetType('unsubscribe');
756    foreach my $to (@ARGV) {
757        $p->SetTo($to);
758        $conn->getConnectionFromJID($jid)->Send($p);
759        queue_admin_msg("You ($baseJID) have unsubscribed from ($to)'s presence.");
760    }
761}
762
763sub jroster_add {
764    my $jid = shift;
765    my $name = shift;
766    my @groups = @{ shift() };
767    my $purgeGroups = shift;
768    my $baseJID = baseJID($jid);
769
770    my $roster = $conn->getRosterFromJID($jid);
771
772    # Adding lots of users with the same name is a bad idea.
773    $name = "" unless (1 == scalar(@ARGV));
774
775    foreach my $to (@ARGV) {
776        my %jq  = $roster->query($to);
777        my $iq = new Net::Jabber::IQ;
778        $iq->SetType('set');
779        my $item = new XML::Stream::Node('item');
780        $iq->NewChild('jabber:iq:roster')->AddChild($item);
781
782        my %allGroups = ();
783
784        foreach my $g (@groups) {
785            $allGroups{$g} = $g;
786        }
787
788        unless ($purgeGroups) {
789            foreach my $g (@{$jq{groups}}) {
790                $allGroups{$g} = $g;
791            }
792        }
793
794        foreach my $g (keys %allGroups) {
795            $item->add_child('group')->add_cdata($g);
796        }
797
798        $item->put_attrib(jid => $to);
799        $item->put_attrib(name => $name) if $name;
800        $conn->getConnectionFromJID($jid)->Send($iq);
801        my $msg = "$baseJID: "
802          . ($name ? "$name ($to)" : "($to)")
803          . " is on your roster in the following groups: { "
804          . join(" , ", keys %allGroups)
805          . " }";
806        queue_admin_msg($msg);
807    }
808}
809
810sub jroster_remove {
811    my $jid = shift;
812    my $name = shift;
813    my @groups = @{ shift() };
814    my $purgeGroups = shift;
815    my $baseJID = baseJID($jid);
816
817    my $iq = new Net::Jabber::IQ;
818    $iq->SetType('set');
819    my $item = new XML::Stream::Node('item');
820    $iq->NewChild('jabber:iq:roster')->AddChild($item);
821    $item->put_attrib(subscription=> 'remove');
822    foreach my $to (@ARGV) {
823        $item->put_attrib(jid => $to);
824        $conn->getConnectionFromJID($jid)->Send($iq);
825        queue_admin_msg("You ($baseJID) have removed ($to) from your roster.");
826    }
827}
828
829sub jroster_auth {
830    my $jid = shift;
831    my $name = shift;
832    my @groups = @{ shift() };
833    my $purgeGroups = shift;
834    my $baseJID = baseJID($jid);
835
836    my $p = new Net::Jabber::Presence;
837    $p->SetType('subscribed');
838    foreach my $to (@ARGV) {
839        $p->SetTo($to);
840        $conn->getConnectionFromJID($jid)->Send($p);
841        queue_admin_msg("($to) has been subscribed to your ($baseJID) presence.");
842    }
843}
844
845sub jroster_deauth {
846    my $jid = shift;
847    my $name = shift;
848    my @groups = @{ shift() };
849    my $purgeGroups = shift;
850    my $baseJID = baseJID($jid);
851
852    my $p = new Net::Jabber::Presence;
853    $p->SetType('unsubscribed');
854    foreach my $to (@ARGV) {
855        $p->SetTo($to);
856        $conn->getConnectionFromJID($jid)->Send($p);
857        queue_admin_msg("($to) has been unsubscribed from your ($baseJID) presence.");
858    }
859}
860
861################################################################################
862### Owl Callbacks
863sub process_owl_jwrite {
864    my $body = shift;
865
866    my $j = new Net::Jabber::Message;
867    $body =~ s/\n\z//;
868    $j->SetMessage(
869        to   => $vars{jwrite}{to},
870        from => $vars{jwrite}{from},
871        type => $vars{jwrite}{type},
872        body => $body
873    );
874
875    $j->SetThread( $vars{jwrite}{thread} )   if ( $vars{jwrite}{thread} );
876    $j->SetSubject( $vars{jwrite}{subject} ) if ( $vars{jwrite}{subject} );
877
878    my $m = j2o( $j, { direction => 'out' } );
879    if ( $vars{jwrite}{type} ne 'groupchat') {
880        BarnOwl::queue_message($m);
881    }
882
883    $j->RemoveFrom(); # Kludge to get around gtalk's random bits after the resource.
884    if ($vars{jwrite}{sid} && $conn->sidExists( $vars{jwrite}{sid} )) {
885        $conn->getConnectionFromSid($vars{jwrite}{sid})->Send($j);
886    }
887    else {
888        $conn->getConnectionFromJID($vars{jwrite}{from})->Send($j);
889    }
890
891    delete $vars{jwrite};
892    BarnOwl::message("");   # Kludge to make the ``type your message...'' message go away
893}
894
895### XMPP Callbacks
896
897sub process_incoming_chat_message {
898    my ( $sid, $j ) = @_;
899    BarnOwl::queue_message( j2o( $j, { direction => 'in',
900                                   sid => $sid } ) );
901}
902
903sub process_incoming_error_message {
904    my ( $sid, $j ) = @_;
905    my %jhash = j2hash( $j, { direction => 'in',
906                              sid => $sid } );
907    $jhash{type} = 'admin';
908   
909    BarnOwl::queue_message( BarnOwl::Message->new(%jhash) );
910}
911
912sub process_incoming_groupchat_message {
913    my ( $sid, $j ) = @_;
914
915    # HACK IN PROGRESS (ignoring delayed messages)
916    return if ( $j->DefinedX('jabber:x:delay') && $j->GetX('jabber:x:delay') );
917    BarnOwl::queue_message( j2o( $j, { direction => 'in',
918                                   sid => $sid } ) );
919}
920
921sub process_incoming_headline_message {
922    my ( $sid, $j ) = @_;
923    BarnOwl::queue_message( j2o( $j, { direction => 'in',
924                                   sid => $sid } ) );
925}
926
927sub process_incoming_normal_message {
928    my ( $sid, $j ) = @_;
929    my %jhash = j2hash( $j, { direction => 'in',
930                              sid => $sid } );
931
932    # XXX TODO: handle things such as MUC invites here.
933
934    #    if ($j->HasX('http://jabber.org/protocol/muc#user'))
935    #    {
936    #   my $x = $j->GetX('http://jabber.org/protocol/muc#user');
937    #   if ($x->HasChild('invite'))
938    #   {
939    #       $props
940    #   }
941    #    }
942    #
943    BarnOwl::queue_message( BarnOwl::Message->new(%jhash) );
944}
945
946sub process_muc_presence {
947    my ( $sid, $p ) = @_;
948    return unless ( $p->HasX('http://jabber.org/protocol/muc#user') );
949}
950
951
952sub process_presence_available {
953    my ( $sid, $p ) = @_;
954    my $from = $p->GetFrom();
955    my $to = $p->GetTo();
956    my $type = $p->GetType();
957    my %props = (
958        to => $to,
959        from => $from,
960        recipient => $to,
961        sender => $from,
962        type => 'jabber',
963        jtype => $p->GetType(),
964        status => $p->GetStatus(),
965        show => $p->GetShow(),
966        xml => $p->GetXML(),
967        direction => 'in');
968
969    if ($type eq '' || $type eq 'available') {
970        $props{body} = "$from is now online. ";
971        $props{loginout} = 'login';
972    }
973    else {
974        $props{body} = "$from is now offline. ";
975        $props{loginout} = 'logout';
976    }
977    $props{replysendercmd} = $props{replycmd} = "jwrite $from -i $sid";
978    if(BarnOwl::getvar('debug') eq 'on') {
979        BarnOwl::queue_message(BarnOwl::Message->new(%props));
980    }
981}
982
983sub process_presence_subscribe {
984    my ( $sid, $p ) = @_;
985    my $from = $p->GetFrom();
986    my $to = $p->GetTo();
987    my %props = (
988        to => $to,
989        from => $from,
990        xml => $p->GetXML(),
991        type => 'admin',
992        adminheader => 'Jabber presence: subscribe',
993        direction => 'in');
994
995    $props{body} = "Allow user ($from) to subscribe to your ($to) presence?\n" .
996                   "(Answer with the `yes' or `no' commands)";
997    $props{yescommand} = "jroster auth $from -a $to";
998    $props{nocommand} = "jroster deauth $from -a $to";
999    $props{question} = "true";
1000    BarnOwl::queue_message(BarnOwl::Message->new(%props));
1001}
1002
1003sub process_presence_unsubscribe {
1004    my ( $sid, $p ) = @_;
1005    my $from = $p->GetFrom();
1006    my $to = $p->GetTo();
1007    my %props = (
1008        to => $to,
1009        from => $from,
1010        xml => $p->GetXML(),
1011        type => 'admin',
1012        adminheader => 'Jabber presence: unsubscribe',
1013        direction => 'in');
1014
1015    $props{body} = "The user ($from) has been unsubscribed from your ($to) presence.\n";
1016    BarnOwl::queue_message(BarnOwl::Message->new(%props));
1017
1018    # Find a connection to reply with.
1019    foreach my $jid ($conn->getJIDs()) {
1020        my $cJID = new Net::Jabber::JID;
1021        $cJID->SetJID($jid);
1022        if ($to eq $cJID->GetJID('base') ||
1023            $to eq $cJID->GetJID('full')) {
1024            my $reply = $p->Reply(type=>"unsubscribed");
1025            $conn->getConnectionFromJID($jid)->Send($reply);
1026            return;
1027        }
1028    }
1029}
1030
1031sub process_presence_subscribed {
1032    my ( $sid, $p ) = @_;
1033    queue_admin_msg("ignoring:".$p->GetXML());
1034    # RFC 3921 says we should respond to this with a "subscribe"
1035    # but this causes a flood of sub/sub'd presence packets with
1036    # some servers, so we won't. We may want to detect this condition
1037    # later, and have per-server settings.
1038    return;
1039}
1040
1041sub process_presence_unsubscribed {
1042    my ( $sid, $p ) = @_;
1043    queue_admin_msg("ignoring:".$p->GetXML());
1044    # RFC 3921 says we should respond to this with a "subscribe"
1045    # but this causes a flood of unsub/unsub'd presence packets with
1046    # some servers, so we won't. We may want to detect this condition
1047    # later, and have per-server settings.
1048    return;
1049}
1050
1051sub process_presence_error {
1052    my ( $sid, $p ) = @_;
1053    my $code = $p->GetErrorCode();
1054    my $error = $p->GetError();
1055    BarnOwl::error("Jabber: $code $error");
1056}
1057
1058
1059### Helper functions
1060
1061sub j2hash {
1062    my $j   = shift;
1063    my %initProps = %{ shift() };
1064
1065    my $dir = 'none';
1066    my %props = ( type => 'jabber' );
1067
1068    foreach my $k (keys %initProps) {
1069        $dir = $initProps{$k} if ($k eq 'direction');
1070        $props{$k} = $initProps{$k};
1071    }
1072
1073    my $jtype = $props{jtype} = $j->GetType();
1074    my $from = $j->GetFrom('jid');
1075    my $to   = $j->GetTo('jid');
1076
1077    $props{from} = $from->GetJID('full');
1078    $props{to}   = $to->GetJID('full');
1079
1080    $props{recipient}  = $to->GetJID('base');
1081    $props{sender}     = $from->GetJID('base');
1082    $props{subject}    = $j->GetSubject() if ( $j->DefinedSubject() );
1083    $props{thread}     = $j->GetThread() if ( $j->DefinedThread() );
1084    $props{body}       = $j->GetBody() if ( $j->DefinedBody() );
1085    $props{error}      = $j->GetError() if ( $j->DefinedError() );
1086    $props{error_code} = $j->GetErrorCode() if ( $j->DefinedErrorCode() );
1087    $props{xml}        = $j->GetXML();
1088
1089    if ( $jtype eq 'chat' ) {
1090        $props{replycmd} =
1091          "jwrite " . ( ( $dir eq 'in' ) ? $props{from} : $props{to} );
1092        $props{replycmd} .=
1093          " -a " . ( ( $dir eq 'out' ) ? $props{from} : $props{to} );
1094        $props{private} = 1;
1095
1096        my $connection;
1097        if ($dir eq 'in') {
1098            $connection = $conn->getConnectionFromSid($props{sid});
1099        }
1100        else {
1101            $connection = $conn->getConnectionFromJID($props{from});
1102        }
1103
1104        # Check to see if we're doing personals with someone in a muc.
1105        # If we are, show the full jid because the base jid is the room.
1106        if ($connection) {
1107            $props{sender} = $props{from}
1108              if ($connection->FindMUC(jid => $from));
1109            $props{recipient} = $props{to}
1110              if ($connection->FindMUC(jid => $to));
1111        }
1112    }
1113    elsif ( $jtype eq 'groupchat' ) {
1114        my $nick = $props{nick} = $from->GetResource();
1115        my $room = $props{room} = $from->GetJID('base');
1116        $props{replycmd} = "jwrite $room";
1117        $props{replycmd} .=
1118          " -a " . ( ( $dir eq 'out' ) ? $props{from} : $props{to} );
1119        if ($props{subject}) {
1120          $props{replycmd} .= " -s " . $props{subject}
1121        }
1122
1123        if ($dir eq 'out') {
1124            $props{replysendercmd} = "jwrite ".$props{to}." -a ".$props{from};
1125        }
1126        else {
1127            $props{replysendercmd} = "jwrite ".$props{from}." -a ".$props{to};
1128        }
1129
1130        $props{sender} = $nick || $room;
1131        $props{recipient} = $room;
1132
1133        if ( $props{subject} && !$props{body} ) {
1134            $props{body} =
1135              '[' . $nick . " has set the topic to: " . $props{subject} . "]";
1136        }
1137    }
1138    elsif ( $jtype eq 'normal' ) {
1139        $props{replycmd}  = "";
1140        $props{private} = 1;
1141    }
1142    elsif ( $jtype eq 'headline' ) {
1143        $props{replycmd} = "";
1144    }
1145    elsif ( $jtype eq 'error' ) {
1146        $props{replycmd} = "";
1147        $props{body}     = "Error "
1148          . $props{error_code}
1149          . " sending to "
1150          . $props{from} . "\n"
1151          . $props{error};
1152    }
1153
1154    $props{replysendercmd} = $props{replycmd} unless $props{replysendercmd};
1155    return %props;
1156}
1157
1158sub j2o {
1159    return BarnOwl::Message->new( j2hash(@_) );
1160}
1161
1162sub queue_admin_msg {
1163    my $err = shift;
1164    BarnOwl::admin_message("Jabber", $err);
1165}
1166
1167sub getServerFromJID {
1168    my $jid = shift;
1169    my $res = new Net::DNS::Resolver;
1170    my $packet =
1171      $res->search( '_xmpp-client._tcp.' . $jid->GetServer(), 'srv' );
1172
1173    if ($packet)    # Got srv record.
1174    {
1175        my @answer = $packet->answer;
1176        return $answer[0]{target}, $answer[0]{port};
1177    }
1178
1179    return $jid->GetServer(), 5222;
1180}
1181
1182sub defaultJID {
1183    return ( $conn->getJIDs() )[0] if ( $conn->connected() == 1 );
1184    return;
1185}
1186
1187sub baseJID {
1188    my $givenJIDStr = shift;
1189    my $givenJID    = new Net::Jabber::JID;
1190    $givenJID->SetJID($givenJIDStr);
1191    return $givenJID->GetJID('base');
1192}
1193
1194sub resolveConnectedJID {
1195    my $givenJIDStr = shift;
1196    my $givenJID    = new Net::Jabber::JID;
1197    $givenJID->SetJID($givenJIDStr);
1198
1199    # Account fully specified.
1200    if ( $givenJID->GetResource() ) {
1201        # Specified account exists
1202        return $givenJIDStr if ($conn->jidExists($givenJIDStr) );
1203        die("Invalid account: $givenJIDStr");
1204    }
1205
1206    # Disambiguate.
1207    else {
1208        my $matchingJID = "";
1209        my $errStr =
1210          "Ambiguous account reference. Please specify a resource.\n";
1211        my $ambiguous = 0;
1212
1213        foreach my $jid ( $conn->getJIDs() ) {
1214            my $cJID = new Net::Jabber::JID;
1215            $cJID->SetJID($jid);
1216            if ( $givenJIDStr eq $cJID->GetJID('base') ) {
1217                $ambiguous = 1 if ( $matchingJID ne "" );
1218                $matchingJID = $jid;
1219                $errStr .= "\t$jid\n";
1220            }
1221        }
1222
1223        # Need further disambiguation.
1224        if ($ambiguous) {
1225            die($errStr);
1226        }
1227
1228        # Not one of ours.
1229        elsif ( $matchingJID eq "" ) {
1230            die("Invalid account: $givenJIDStr");
1231        }
1232
1233        # It's this one.
1234        else {
1235            return $matchingJID;
1236        }
1237    }
1238    return "";
1239}
1240
1241sub resolveDestJID {
1242    my ($to, $from) = @_;
1243    my $jid = Net::Jabber::JID->new($to);
1244
1245    my $roster = $conn->getRosterFromJID($from);
1246    my @jids = $roster->jids('all');
1247    for my $j (@jids) {
1248        if(($roster->query($j, 'name') || $j->GetUserID()) eq $to) {
1249            return $j->GetJID('full');
1250        } elsif($j->GetJID('base') eq baseJID($to)) {
1251            return $jid->GetJID('full');
1252        }
1253    }
1254
1255    # If we found nothing being clever, check to see if our input was
1256    # sane enough to look like a jid with a UserID.
1257    return $jid->GetJID('full') if $jid->GetUserID();
1258    return undef;
1259}
1260
1261sub resolveType {
1262    my $to = shift;
1263    my $from = shift;
1264    return unless $from;
1265    my @mucs = $conn->getConnectionFromJID($from)->MUCs;
1266    if(grep {$_->BaseJID eq $to } @mucs) {
1267        return 'groupchat';
1268    } else {
1269        return 'chat';
1270    }
1271}
1272
1273sub guess_jwrite {
1274    # Heuristically guess what jids a jwrite was meant to be going to/from
1275    my ($from, $to) = (@_);
1276    my ($from_jid, $to_jid);
1277    my @matches;
1278    if($from) {
1279        $from_jid = resolveConnectedJID($from);
1280        die("Unable to resolve account $from") unless $from_jid;
1281        $to_jid = resolveDestJID($to, $from_jid);
1282        push @matches, [$from_jid, $to_jid] if $to_jid;
1283    } else {
1284        for my $f ($conn->getJIDs) {
1285            $to_jid = resolveDestJID($to, $f);
1286            if(defined($to_jid)) {
1287                push @matches, [$f, $to_jid];
1288            }
1289        }
1290        if($to =~ /@/) {
1291            push @matches, [$_, $to]
1292               for ($conn->getJIDs);
1293        }
1294    }
1295
1296    for my $m (@matches) {
1297        my $type = resolveType($m->[1], $m->[0]);
1298        push @$m, $type;
1299    }
1300
1301    return @matches;
1302}
1303
13041;
Note: See TracBrowser for help on using the repository browser.