source: perl/modules/jabber.pl @ cb769bb

barnowl_perlaimdebianrelease-1.4release-1.5release-1.6release-1.7release-1.8release-1.9
Last change on this file since cb769bb was cb769bb, checked in by Nelson Elhage <nelhage@mit.edu>, 15 years ago
r15874@phanatique: nelhage | 2006-12-24 18:25:33 -0500 Don't quit if we can't contact the hostmaster. r15884@phanatique: nelhage | 2006-12-24 18:56:03 -0500 Respect the displayoutgoing variable r15885@phanatique: nelhage | 2006-12-24 20:10:44 -0500 You can now write filters based off arbitrary message attributes r15887@phanatique: nelhage | 2006-12-24 22:59:39 -0500 r15886@phanatique (orig r476): nelhage | 2006-12-24 22:59:10 -0500 r24493@heretique: nelhage | 2006-12-24 22:59:02 -0500 Moving zephyr initialization later, so that zephyr works again r15891@phanatique: nelhage | 2006-12-25 14:40:08 -0500 * perl messages hashes use `private', not `isprivate' * get rid of a perl warning if login fails r15900@phanatique: nelhage | 2006-12-25 21:04:15 -0500 Merging in filter regression tests from my local branch. r15926@phanatique: nelhage | 2006-12-26 00:57:07 -0500 r15901@phanatique: nelhage | 2006-12-25 21:08:47 -0500 Base framework for the filter rewrite system. Only understands regexes and true/false so far. r15927@phanatique: nelhage | 2006-12-26 00:57:08 -0500 r15902@phanatique: nelhage | 2006-12-25 23:03:33 -0500 support for negation and parentheses r15928@phanatique: nelhage | 2006-12-26 00:57:08 -0500 r15924@phanatique: nelhage | 2006-12-26 00:16:30 -0500 Now passing all tests except for recursion detection r15929@phanatique: nelhage | 2006-12-26 00:57:08 -0500 r15925@phanatique: nelhage | 2006-12-26 00:52:09 -0500 Checking for filter recursion loops
  • Property mode set to 100644
File size: 31.7 KB
Line 
1# -*- mode: cperl; cperl-indent-level: 4; indent-tabs-mode: nil -*-
2package owl_jabber;
3use warnings;
4use strict;
5
6use Authen::SASL qw(Perl);
7use Net::Jabber;
8use Net::DNS;
9use Getopt::Long;
10
11no warnings 'redefine';
12
13################################################################################
14# owl perl jabber support
15#
16# XXX Todo:
17# Rosters for MUCs
18# More user feedback
19#  * joining MUC
20#  * parting MUC
21#  * presence (Roster and MUC)
22# Implementing formatting and logging callbacks for C
23# Appropriate callbacks for presence subscription messages.
24#  * Current behavior is auto-accept (default for Net::Jabber)
25#
26################################################################################
27
28
29################################################################################
30################################################################################
31package owl_jabber::ConnectionManager;
32sub new {
33    my $class = shift;
34    return bless { }, $class;
35}
36
37sub addConnection {
38    my $self = shift;
39    my $jidStr = shift;
40
41    my %args = ();
42    if(owl::getvar('debug') eq 'on') {
43        $args{debuglevel} = 1;
44        $args{debugfile} = 'jabber.log';
45    }
46    my $client = Net::Jabber::Client->new(%args);
47
48    $self->{Client}->{$jidStr} = $client;
49    $self->{Roster}->{$jidStr} = $client->Roster();
50    return $client;
51}
52
53sub removeConnection {
54    my $self = shift;
55    my $jidStr = shift;
56    return 0 unless exists $self->{Client}->{$jidStr};
57   
58    $self->{Client}->{$jidStr}->Disconnect();
59    delete $self->{Roster}->{$jidStr};
60    delete $self->{Client}->{$jidStr};
61   
62    return 1;
63}
64
65sub connected {
66    my $self = shift;
67    return scalar keys %{ $self->{Client} };
68}
69
70sub getJids {
71    my $self = shift;
72    return keys %{ $self->{Client} };
73}
74
75sub jidExists {
76    my $self = shift;
77    my $jidStr = shift;
78    return exists $self->{Client}->{$jidStr};
79}
80
81sub sidExists {
82    my $self = shift;
83    my $sid = shift || "";
84    foreach my $j ( keys %{ $self->{Client} } ) {
85        return 1 if ($self->{Client}->{$j}->{SESSION}->{id} eq $sid);
86    }
87    return 0;
88}
89
90sub getConnectionFromSid {
91    my $self = shift;
92    my $sid = shift;
93    foreach my $c (values %{ $self->{Client} }) {
94        return $c if $c->{SESSION}->{id} eq $sid;
95    }
96    return undef;
97}
98
99sub getConnectionFromJidStr {
100    my $self = shift;
101    my $jidStr = shift;
102    return $self->{Client}->{$jidStr};
103}
104
105sub getRosterFromSid {
106    my $self = shift;
107    my $sid = shift;
108    foreach my $j ( $self->getJids ) {
109        return $self->{Roster}->{$j}
110          if $self->{Client}->{$j}->{SESSION}->{id} eq $sid;
111    }
112    return undef;
113}
114
115sub getRosterFromJidStr {
116    my $self = shift;
117    my $jidStr = shift;
118    return $self->{Roster}->{$jidStr};
119    return undef;
120}
121################################################################################
122
123package owl_jabber;
124
125our $conn = new owl_jabber::ConnectionManager unless $conn;;
126our %vars;
127
128sub onStart {
129    if ( eval { \&owl::queue_message } ) {
130        register_owl_commands();
131        push @::onMainLoop,     sub { owl_jabber::onMainLoop(@_) };
132        push @::onGetBuddyList, sub { owl_jabber::onGetBuddyList(@_) };
133    }
134    else {
135        # Our owl doesn't support queue_message. Unfortunately, this
136        # means it probably *also* doesn't support owl::error. So just
137        # give up silently.
138    }
139}
140
141push @::onStartSubs, sub { owl_jabber::onStart(@_) };
142
143sub onMainLoop {
144    return if ( !$conn->connected() );
145
146    foreach my $jid ( $conn->getJids() ) {
147        my $client = $conn->getConnectionFromJidStr($jid);
148
149        my $status = $client->Process(0);
150        if ( !defined($status) ) {
151            owl::error("Jabber account $jid disconnected!");
152            do_logout($jid);
153        }
154        if ($::shutdown) {
155            do_logout($jid);
156            return;
157        }
158    }
159}
160
161sub blist_listBuddy {
162    my $roster = shift;
163    my $buddy  = shift;
164    my $blistStr .= "    ";
165    my %jq  = $roster->query($buddy);
166    my $res = $roster->resource($buddy);
167
168    $blistStr .= $jq{name} ? $jq{name} . "\t(" .$buddy->GetJID() . ')' : $buddy->GetJID();
169
170    if ($res) {
171        my %rq = $roster->resourceQuery( $buddy, $res );
172        $blistStr .= " [" . ( $rq{show} ? $rq{show} : 'online' ) . "]";
173        $blistStr .= " " . $rq{status} if $rq{status};
174        $blistStr = boldify($blistStr);
175    }
176    else {
177        if ($jq{ask}) {
178            $blistStr .= " [pending]";
179        }
180        elsif ($jq{subscription} eq 'none' || $jq{subscription} eq 'from') {
181            $blistStr .= " [not subscribed]";
182        }
183        else {
184            $blistStr .= " [offline]";
185        }
186    }
187    return $blistStr . "\n";
188}
189
190sub getSingleBuddyList {
191    my $jid = shift;
192    $jid = resolveJID($jid);
193    return "" unless $jid;
194    my $blist = "";
195    my $roster = $conn->getRosterFromJidStr($jid);
196    if ($roster) {
197        $blist .= "\n" . boldify("Jabber Roster for $jid\n");
198
199        foreach my $group ( $roster->groups() ) {
200            $blist .= "  Group: $group\n";
201            foreach my $buddy ( $roster->jids( 'group', $group ) ) {
202                $blist .= blist_listBuddy( $roster, $buddy );
203            }
204        }
205
206        my @unsorted = $roster->jids('nogroup');
207        if (@unsorted) {
208            $blist .= "  [unsorted]\n";
209            foreach my $buddy (@unsorted) {
210                $blist .= blist_listBuddy( $roster, $buddy );
211            }
212        }
213    }
214    return $blist;
215}
216
217sub onGetBuddyList {
218    my $blist = "";
219    foreach my $jid ($conn->getJids()) {
220        $blist .= getSingleBuddyList($jid);
221    }
222    return $blist;
223}
224
225################################################################################
226### Owl Commands
227sub register_owl_commands() {
228    owl::new_command(
229        jabberlogin => \&cmd_login,
230        { summary => "Log into jabber", }
231    );
232    owl::new_command(
233        jabberlogout => \&cmd_logout,
234        { summary => "Log out of jabber" }
235    );
236    owl::new_command(
237        jwrite => \&cmd_jwrite,
238        {
239            summary => "Send a Jabber Message",
240            usage   => "jwrite JID [-g] [-t thread] [-s subject]"
241        }
242    );
243    owl::new_command(
244        jlist => \&cmd_jlist,
245        {
246            summary => "Show your Jabber roster.",
247            usage   => "jlist"
248        }
249    );
250    owl::new_command(
251        jmuc => \&cmd_jmuc,
252        {
253            summary     => "Jabber MUC related commands.",
254            description => "jmuc sends jabber commands related to muc.\n\n"
255              . "The following commands are available\n\n"
256              . "join {muc}  Join a muc.\n\n"
257              . "part [muc]  Part a muc.\n"
258              . "            The muc is taken from the current message if not supplied.\n\n"
259              . "invite {jid} [muc]\n"
260              . "            Invite {jid} to [muc].\n"
261              . "            The muc is taken from the current message if not supplied.\n\n"
262              . "configure [muc]\n"
263              . "            Configure [muc].\n"
264              . "            Necessary to initalize a new MUC",
265            usage => "jmuc {command} {args}"
266        }
267    );
268    owl::new_command(
269        jroster => \&cmd_jroster,
270        {
271            summary     => "Jabber Roster related commands.",
272            description => "jroster sends jabber commands related to rosters.\n\n",
273            usage       => "jroster {command} {args}"
274        }
275    );
276}
277
278sub cmd_login {
279    my $cmd = shift;
280    my $jid = new Net::XMPP::JID;
281    $jid->SetJID(shift);
282
283    my $uid           = $jid->GetUserID();
284    my $componentname = $jid->GetServer();
285    my $resource      = $jid->GetResource() || 'owl';
286    $jid->SetResource($resource);
287    my $jidStr = $jid->GetJID('full');
288
289    if ( !$uid || !$componentname ) {
290        owl::error("usage: $cmd {jid}");
291        return;
292    }
293
294    if ( $conn->jidExists($jidStr) ) {
295        owl::error("Already logged in as $jidStr.");
296        return;
297    }
298
299    my ( $server, $port ) = getServerFromJID($jid);
300
301    $vars{jlogin_jid} = $jidStr;
302    $vars{jlogin_havepass} = 0;
303    $vars{jlogin_connhash} = {
304        hostname      => $server,
305        tls           => 1,
306        port          => $port,
307        componentname => $componentname
308    };
309    $vars{jlogin_authhash} =
310      { username => $uid,
311        resource => $resource,
312    };
313
314    return do_login('');
315}
316
317sub do_login {
318    $vars{jlogin_password} = shift;
319    $vars{jlogin_authhash}->{password} = sub { return $vars{jlogin_password} || '' };
320    my $jidStr = $vars{jlogin_jid};
321    if ( !$jidStr && $vars{jlogin_havepass}) {
322        owl::error("Got password but have no jid!");
323    }
324    else
325    {
326        my $client = $conn->addConnection($jidStr);
327
328        #XXX Todo: Add more callbacks.
329        # * MUC presence handlers
330        $client->SetMessageCallBacks(
331            chat      => sub { owl_jabber::process_incoming_chat_message(@_) },
332            error     => sub { owl_jabber::process_incoming_error_message(@_) },
333            groupchat => sub { owl_jabber::process_incoming_groupchat_message(@_) },
334            headline  => sub { owl_jabber::process_incoming_headline_message(@_) },
335            normal    => sub { owl_jabber::process_incoming_normal_message(@_) }
336        );
337        $client->SetPresenceCallBacks(
338#            available    => sub { owl_jabber::process_presence_available(@_) },
339#            unavailable  => sub { owl_jabber::process_presence_available(@_) },
340            subscribe    => sub { owl_jabber::process_presence_subscribe(@_) },
341            subscribed   => sub { owl_jabber::process_presence_subscribed(@_) },
342            unsubscribe  => sub { owl_jabber::process_presence_unsubscribe(@_) },
343            unsubscribed => sub { owl_jabber::process_presence_unsubscribed(@_) });
344
345        my $status = $client->Connect( %{ $vars{jlogin_connhash} } );
346        if ( !$status ) {
347            $conn->removeConnection($jidStr);
348            owl::error("We failed to connect");
349        }
350        else {
351            my @result = $client->AuthSend( %{ $vars{jlogin_authhash} } );
352
353            if ( $result[0] ne 'ok' ) {
354            if ( !$vars{jlogin_havepass} && $result[0] eq '401' ) {
355                $vars{jlogin_havepass} = 1;
356                $conn->removeConnection($jidStr);
357                owl::start_password( "Password for $jidStr: ", \&do_login );
358                return "";
359            }
360            $conn->removeConnection($jidStr);
361            owl::error(
362                "Error in connect: " . join( " ", @result ) );
363        }
364            else {
365                $conn->getRosterFromJidStr($jidStr)->fetch();
366                $client->PresenceSend( priority => 1 );
367                queue_admin_msg("Connected to jabber as $jidStr");
368            }
369        }
370    }
371    delete $vars{jlogin_jid};
372    $vars{jlogin_password} =~ tr/\0-\377/x/;
373    delete $vars{jlogin_password};
374    delete $vars{jlogin_havepass};
375    delete $vars{jlogin_connhash};
376    delete $vars{jlogin_authhash};
377    return "";
378}
379
380sub do_logout {
381    my $jid = shift;
382    my $disconnected = $conn->removeConnection($jid);
383    queue_admin_msg("Jabber disconnected ($jid).") if $disconnected;
384}
385
386sub cmd_logout {
387    # Logged into multiple accounts
388    if ( $conn->connected() > 1 ) {
389        # Logged into multiple accounts, no accout specified.
390        if ( !$_[1] ) {
391            my $errStr =
392              "You are logged into multiple accounts. Please specify an account to log out of.\n";
393            foreach my $jid ( $conn->getJids() ) {
394                $errStr .= "\t$jid\n";
395            }
396            queue_admin_msg($errStr);
397        }
398        # Logged into multiple accounts, account specified.
399        else {
400            if ( $_[1] eq '-a' )    #All accounts.
401            {
402                foreach my $jid ( $conn->getJids() ) {
403                    do_logout($jid);
404                }
405            }
406            else                    #One account.
407            {
408                my $jid = resolveJID( $_[1] );
409                do_logout($jid) if ( $jid ne '' );
410            }
411        }
412    }
413    else                            # Only one account logged in.
414    {
415        do_logout( ( $conn->getJids() )[0] );
416    }
417    return "";
418}
419
420sub cmd_jlist {
421    if ( !( scalar $conn->getJids() ) ) {
422        owl::error("You are not logged in to Jabber.");
423        return;
424    }
425    owl::popless_ztext( onGetBuddyList() );
426}
427
428sub cmd_jwrite {
429    if ( !$conn->connected() ) {
430        owl::error("You are not logged in to Jabber.");
431        return;
432    }
433
434    my $jwrite_to      = "";
435    my $jwrite_from    = "";
436    my $jwrite_sid     = "";
437    my $jwrite_thread  = "";
438    my $jwrite_subject = "";
439    my $jwrite_type    = "chat";
440
441    my @args = @_;
442    shift;
443    local @ARGV = @_;
444    my $gc;
445    GetOptions(
446        'thread=s'  => \$jwrite_thread,
447        'subject=s' => \$jwrite_subject,
448        'account=s' => \$jwrite_from,
449        'id=s'     =>  \$jwrite_sid,
450        'groupchat' => \$gc
451    );
452    $jwrite_type = 'groupchat' if $gc;
453
454    if ( scalar @ARGV != 1 ) {
455        owl::error(
456            "Usage: jwrite JID [-g] [-t thread] [-s 'subject'] [-a account]");
457        return;
458    }
459    else {
460        $jwrite_to = shift @ARGV;
461    }
462
463    if ( !$jwrite_from ) {
464        if ( $conn->connected() == 1 ) {
465            $jwrite_from = ( $conn->getJids() )[0];
466        }
467        else {
468            owl::error("Please specify an account with -a {jid}");
469            return;
470        }
471    }
472    else {
473        $jwrite_from = resolveJID($jwrite_from);
474        return unless $jwrite_from;
475    }
476
477    $vars{jwrite} = {
478        to      => $jwrite_to,
479        from    => $jwrite_from,
480        sid     => $jwrite_sid,
481        subject => $jwrite_subject,
482        thread  => $jwrite_thread,
483        type    => $jwrite_type
484    };
485
486    owl::message(
487"Type your message below.  End with a dot on a line by itself.  ^C will quit."
488    );
489    owl::start_edit_win( join( ' ', @args ), \&process_owl_jwrite );
490}
491
492sub cmd_jmuc {
493    die "You are not logged in to Jabber" unless $conn->connected();
494    my $ocmd = shift;
495    my $cmd  = shift;
496    if ( !$cmd ) {
497
498        #XXX TODO: Write general usage for jmuc command.
499        return;
500    }
501
502    my %jmuc_commands = (
503        join      => \&jmuc_join,
504        part      => \&jmuc_part,
505        invite    => \&jmuc_invite,
506        configure => \&jmuc_configure
507    );
508    my $func = $jmuc_commands{$cmd};
509    if ( !$func ) {
510        owl::error("jmuc: Unknown command: $cmd");
511        return;
512    }
513
514    {
515        local @ARGV = @_;
516        my $jid;
517        my $muc;
518        my $m = owl::getcurmsg();
519        if ( $m->is_jabber && $m->{jtype} eq 'groupchat' ) {
520            $muc = $m->{room};
521            $jid = $m->{to};
522        }
523
524        my $getopt = Getopt::Long::Parser->new;
525        $getopt->configure('pass_through');
526        $getopt->getoptions( 'account=s' => \$jid );
527        $jid ||= defaultJID();
528        if ($jid) {
529            $jid = resolveJID($jid);
530            return unless $jid;
531        }
532        else {
533            owl::error('You must specify an account with -a {jid}');
534        }
535        return $func->( $jid, $muc, @ARGV );
536    }
537}
538
539sub jmuc_join {
540    my ( $jid, $muc, @args ) = @_;
541    local @ARGV = @args;
542    my $password;
543    GetOptions( 'password=s' => \$password );
544
545    $muc = shift @ARGV
546      or die("Usage: jmuc join {muc} [-p password] [-a account]");
547
548    my $presence = new Net::Jabber::Presence;
549    $presence->SetPresence( to => $muc );
550    my $x = $presence->NewChild('http://jabber.org/protocol/muc');
551    $x->AddHistory()->SetMaxChars(0);
552    if ($password) {
553        $x->SetPassword($password);
554    }
555
556    $conn->getConnectionFromJidStr($jid)->Send($presence);
557}
558
559sub jmuc_part {
560    my ( $jid, $muc, @args ) = @_;
561
562    $muc = shift @args if scalar @args;
563    die("Usage: jmuc part {muc} [-a account]") unless $muc;
564
565    $conn->getConnectionFromJidStr($jid)->PresenceSend( to => $muc, type => 'unavailable' );
566    queue_admin_msg("$jid has left $muc.");
567}
568
569sub jmuc_invite {
570    my ( $jid, $muc, @args ) = @_;
571
572    my $invite_jid = shift @args;
573    $muc = shift @args if scalar @args;
574
575    die('Usage: jmuc invite {jid} [muc] [-a account]')
576      unless $muc && $invite_jid;
577
578    my $message = Net::Jabber::Message->new();
579    $message->SetTo($muc);
580    my $x = $message->NewChild('http://jabber.org/protocol/muc#user');
581    $x->AddInvite();
582    $x->GetInvite()->SetTo($invite_jid);
583    $conn->getConnectionFromJidStr($jid)->Send($message);
584    queue_admin_msg("$jid has invited $invite_jid to $muc.");
585}
586
587sub jmuc_configure {
588    my ( $jid, $muc, @args ) = @_;
589    $muc = shift @args if scalar @args;
590    die("Usage: jmuc configure [muc]") unless $muc;
591    my $iq = Net::Jabber::IQ->new();
592    $iq->SetTo($muc);
593    $iq->SetType('set');
594    my $query = $iq->NewQuery("http://jabber.org/protocol/muc#owner");
595    my $x     = $query->NewChild("jabber:x:data");
596    $x->SetType('submit');
597
598    $conn->getConnectionFromJidStr($jid)->Send($iq);
599    queue_admin_msg("Accepted default instant configuration for $muc");
600}
601
602
603#XXX TODO: Consider merging this with jmuc and selecting off the first two args.
604sub cmd_jroster {
605    die "You are not logged in to Jabber" unless $conn->connected();
606    my $ocmd = shift;
607    my $cmd  = shift;
608    if ( !$cmd ) {
609
610        #XXX TODO: Write general usage for jroster command.
611        return;
612    }
613
614    my %jroster_commands = (
615        sub      => \&jroster_sub,
616        unsub    => \&jroster_unsub,
617        add      => \&jroster_add,
618        remove   => \&jroster_remove,
619        auth     => \&jroster_auth,
620        deauth   => \&jroster_deauth
621    );
622
623    my $func = $jroster_commands{$cmd};
624    if ( !$func ) {
625        owl::error("jroster: Unknown command: $cmd");
626        return;
627    }
628
629    {
630        local @ARGV = @_;
631        my $jid;
632        my $name;
633        my @groups;
634        my $purgeGroups;
635        my $getopt = Getopt::Long::Parser->new;
636        $getopt->configure('pass_through');
637        $getopt->getoptions(
638            'account=s' => \$jid,
639            'group=s' => \@groups,
640            'purgegroups' => \$purgeGroups,
641            'name=s' => \$name
642        );
643        $jid ||= defaultJID();
644        if ($jid) {
645            $jid = resolveJID($jid);
646            return unless $jid;
647        }
648        else {
649            owl::error('You must specify an account with -a {jid}');
650        }
651        return $func->( $jid, $name, \@groups, $purgeGroups,  @ARGV );
652    }
653}
654
655sub jroster_sub {
656    my $jid = shift;
657    my $name = shift;
658    my @groups = @{ shift() };
659    my $purgeGroups = shift;
660    my $baseJid = baseJID($jid);
661
662    my $roster = $conn->getRosterFromJidStr($jid);
663
664    # Adding lots of users with the same name is a bad idea.
665    $name = "" unless (1 == scalar(@ARGV));
666
667    my $p = new Net::XMPP::Presence;
668    $p->SetType('subscribe');
669
670    foreach my $to (@ARGV) {
671        jroster_add($jid, $name, \@groups, $purgeGroups, ($to)) unless ($roster->exists($to));
672
673        $p->SetTo($to);
674        $conn->getConnectionFromJidStr($jid)->Send($p);
675        queue_admin_msg("You ($baseJid) have requested a subscription to ($to)'s presence.");
676    }
677}
678
679sub jroster_unsub {
680    my $jid = shift;
681    my $name = shift;
682    my @groups = @{ shift() };
683    my $purgeGroups = shift;
684    my $baseJid = baseJID($jid);
685
686    my $p = new Net::XMPP::Presence;
687    $p->SetType('unsubscribe');
688    foreach my $to (@ARGV) {
689        $p->SetTo($to);
690        $conn->getConnectionFromJidStr($jid)->Send($p);
691        queue_admin_msg("You ($baseJid) have unsubscribed from ($to)'s presence.");
692    }
693}
694
695sub jroster_add {
696    my $jid = shift;
697    my $name = shift;
698    my @groups = @{ shift() };
699    my $purgeGroups = shift;
700    my $baseJid = baseJID($jid);
701
702    my $roster = $conn->getRosterFromJidStr($jid);
703
704    # Adding lots of users with the same name is a bad idea.
705    $name = "" unless (1 == scalar(@ARGV));
706
707    foreach my $to (@ARGV) {
708        my %jq  = $roster->query($to);
709        my $iq = new Net::XMPP::IQ;
710        $iq->SetType('set');
711        my $item = new XML::Stream::Node('item');
712        $iq->NewChild('jabber:iq:roster')->AddChild($item);
713
714        my %allGroups = ();
715
716        foreach my $g (@groups) {
717            $allGroups{$g} = $g;
718        }
719
720        unless ($purgeGroups) {
721            foreach my $g (@{$jq{groups}}) {
722                $allGroups{$g} = $g;
723            }
724        }
725
726        foreach my $g (keys %allGroups) {
727            $item->add_child('group')->add_cdata($g);
728        }
729
730        $item->put_attrib(jid => $to);
731        $item->put_attrib(name => $name) if $name;
732        $conn->getConnectionFromJidStr($jid)->Send($iq);
733        my $msg = "$baseJid: "
734          . ($name ? "$name ($to)" : "($to)")
735          . " is on your roster in the following groups: { "
736          . join(" , ", keys %allGroups)
737          . " }";
738        queue_admin_msg($msg);
739    }
740}
741
742sub jroster_remove {
743    my $jid = shift;
744    my $name = shift;
745    my @groups = @{ shift() };
746    my $purgeGroups = shift;
747    my $baseJid = baseJID($jid);
748
749    my $iq = new Net::XMPP::IQ;
750    $iq->SetType('set');
751    my $item = new XML::Stream::Node('item');
752    $iq->NewChild('jabber:iq:roster')->AddChild($item);
753    $item->put_attrib(subscription=> 'remove');
754    foreach my $to (@ARGV) {
755        $item->put_attrib(jid => $to);
756        $conn->getConnectionFromJidStr($jid)->Send($iq);
757        queue_admin_msg("You ($baseJid) have removed ($to) from your roster.");
758    }
759}
760
761sub jroster_auth {
762    my $jid = shift;
763    my $name = shift;
764    my @groups = @{ shift() };
765    my $purgeGroups = shift;
766    my $baseJid = baseJID($jid);
767
768    my $p = new Net::XMPP::Presence;
769    $p->SetType('subscribed');
770    foreach my $to (@ARGV) {
771        $p->SetTo($to);
772        $conn->getConnectionFromJidStr($jid)->Send($p);
773        queue_admin_msg("($to) has been subscribed to your ($baseJid) presence.");
774    }
775}
776
777sub jroster_deauth {
778    my $jid = shift;
779    my $name = shift;
780    my @groups = @{ shift() };
781    my $purgeGroups = shift;
782    my $baseJid = baseJID($jid);
783
784    my $p = new Net::XMPP::Presence;
785    $p->SetType('unsubscribed');
786    foreach my $to (@ARGV) {
787        $p->SetTo($to);
788        $conn->getConnectionFromJidStr($jid)->Send($p);
789        queue_admin_msg("($to) has been unsubscribed from your ($baseJid) presence.");
790    }
791}
792
793################################################################################
794### Owl Callbacks
795sub process_owl_jwrite {
796    my $body = shift;
797
798    my $j = new Net::XMPP::Message;
799    $body =~ s/\n\z//;
800    $j->SetMessage(
801        to   => $vars{jwrite}{to},
802        from => $vars{jwrite}{from},
803        type => $vars{jwrite}{type},
804        body => $body
805    );
806
807    $j->SetThread( $vars{jwrite}{thread} )   if ( $vars{jwrite}{thread} );
808    $j->SetSubject( $vars{jwrite}{subject} ) if ( $vars{jwrite}{subject} );
809
810    my $m = j2o( $j, { direction => 'out' } );
811    if ( $vars{jwrite}{type} ne 'groupchat' && owl::getvar('displayoutgoing') eq 'on') {
812        owl::queue_message($m);
813    }
814
815    if ($vars{jwrite}{sid} && $conn->sidExists( $vars{jwrite}{sid} )) {
816        $conn->getConnectionFromSid($vars{jwrite}{sid})->Send($j);
817    }
818    else {
819        $conn->getConnectionFromJidStr($vars{jwrite}{from})->Send($j);
820    }
821
822    delete $vars{jwrite};
823    owl::message("");   # Kludge to make the ``type your message...'' message go away
824}
825
826### XMPP Callbacks
827
828sub process_incoming_chat_message {
829    my ( $sid, $j ) = @_;
830    owl::queue_message( j2o( $j, { direction => 'in',
831                                   sid => $sid } ) );
832}
833
834sub process_incoming_error_message {
835    my ( $sid, $j ) = @_;
836    my %jhash = j2hash( $j, { direction => 'in',
837                              sid => $sid } );
838    $jhash{type} = 'admin';
839    owl::queue_message( owl::Message->new(%jhash) );
840}
841
842sub process_incoming_groupchat_message {
843    my ( $sid, $j ) = @_;
844
845    # HACK IN PROGRESS (ignoring delayed messages)
846    return if ( $j->DefinedX('jabber:x:delay') && $j->GetX('jabber:x:delay') );
847    owl::queue_message( j2o( $j, { direction => 'in',
848                                   sid => $sid } ) );
849}
850
851sub process_incoming_headline_message {
852    my ( $sid, $j ) = @_;
853    owl::queue_message( j2o( $j, { direction => 'in',
854                                   sid => $sid } ) );
855}
856
857sub process_incoming_normal_message {
858    my ( $sid, $j ) = @_;
859    my %jhash = j2hash( $j, { direction => 'in',
860                              sid => $sid } );
861
862    # XXX TODO: handle things such as MUC invites here.
863
864    #    if ($j->HasX('http://jabber.org/protocol/muc#user'))
865    #    {
866    #   my $x = $j->GetX('http://jabber.org/protocol/muc#user');
867    #   if ($x->HasChild('invite'))
868    #   {
869    #       $props
870    #   }
871    #    }
872    #
873    owl::queue_message( owl::Message->new(%jhash) );
874}
875
876sub process_muc_presence {
877    my ( $sid, $p ) = @_;
878    return unless ( $p->HasX('http://jabber.org/protocol/muc#user') );
879}
880
881
882sub process_presence_available {
883    my ( $sid, $p ) = @_;
884    my $from = $p->GetFrom();
885    my $to = $p->GetTo();
886    my $type = $p->GetType();
887    my %props = (
888        to => $to,
889        from => $from,
890        recipient => $to,
891        sender => $from,
892        type => 'jabber',
893        jtype => $p->GetType(),
894        status => $p->GetStatus(),
895        show => $p->GetShow(),
896        xml => $p->GetXML(),
897        direction => 'in');
898
899    if ($type eq '' || $type eq 'available') {
900        $props{body} = "$from is now online. ";
901        $props{loginout} = 'login';
902    }
903    else {
904        $props{body} = "$from is now offline. ";
905        $props{loginout} = 'logout';
906    }
907    $props{replysendercmd} = $props{replycmd} = "jwrite $from -i $sid";
908    owl::queue_message(owl::Message->new(%props));
909}
910
911sub process_presence_subscribe {
912    my ( $sid, $p ) = @_;
913    my $from = $p->GetFrom();
914    my $to = $p->GetTo();
915    my %props = (
916        to => $to,
917        from => $from,
918        xml => $p->GetXML(),
919        type => 'admin',
920        adminheader => 'Jabber presence: subscribe',
921        direction => 'in');
922
923    $props{body} = "The user ($from) wants to subscribe to your ($to) presence.\nReply (r) will authorize, reply-sender (R) will deny.";
924    $props{replycmd} = "jroster auth $from -a $to";
925    $props{replysendercmd} = "jroster deauth $from -a $to";
926    owl::queue_message(owl::Message->new(%props));
927}
928
929sub process_presence_unsubscribe {
930    my ( $sid, $p ) = @_;
931    my $from = $p->GetFrom();
932    my $to = $p->GetTo();
933    my %props = (
934        to => $to,
935        from => $from,
936        xml => $p->GetXML(),
937        type => 'admin',
938        adminheader => 'Jabber presence: unsubscribe',
939        direction => 'in');
940
941    $props{body} = "The user ($from) has been unsubscribed from your ($to) presence.\n";
942    owl::queue_message(owl::Message->new(%props));
943
944    # Find a connection to reply with.
945    foreach my $jid ($conn->getJids()) {
946        my $cJid = new Net::XMPP::JID;
947        $cJid->SetJID($jid);
948        if ($to eq $cJid->GetJID('base') ||
949            $to eq $cJid->GetJID('full')) {
950            my $reply = $p->Reply(type=>"unsubscribed");
951            $conn->getConnectionFromJidStr($jid)->Send($reply);
952            return;
953        }
954    }
955}
956
957sub process_presence_subscribed {
958    my ( $sid, $p ) = @_;
959    queue_admin_msg("ignoring:".$p->GetXML());
960    # RFC 3921 says we should respond to this with a "subscribe"
961    # but this causes a flood of sub/sub'd presence packets with
962    # some servers, so we won't. We may want to detect this condition
963    # later, and have per-server settings.
964    return;
965}
966
967sub process_presence_unsubscribed {
968    my ( $sid, $p ) = @_;
969    queue_admin_msg("ignoring:".$p->GetXML());
970    # RFC 3921 says we should respond to this with a "subscribe"
971    # but this causes a flood of unsub/unsub'd presence packets with
972    # some servers, so we won't. We may want to detect this condition
973    # later, and have per-server settings.
974    return;
975}
976
977
978### Helper functions
979
980sub j2hash {
981    my $j   = shift;
982    my %initProps = %{ shift() };
983
984    my $dir = 'none';
985    my %props = ( type => 'jabber' );
986
987    foreach my $k (keys %initProps) {
988        $dir = $initProps{$k} if ($k eq 'direction');
989        $props{$k} = $initProps{$k};
990    }
991
992    my $jtype = $props{jtype} = $j->GetType();
993    my $from = $j->GetFrom('jid');
994    my $to   = $j->GetTo('jid');
995
996    $props{from} = $from->GetJID('full');
997    $props{to}   = $to->GetJID('full');
998
999    $props{recipient}  = $to->GetJID('base');
1000    $props{sender}     = $from->GetJID('base');
1001    $props{subject}    = $j->GetSubject() if ( $j->DefinedSubject() );
1002    $props{thread}     = $j->GetThread() if ( $j->DefinedThread() );
1003    $props{body}       = $j->GetBody() if ( $j->DefinedBody() );
1004    $props{error}      = $j->GetError() if ( $j->DefinedError() );
1005    $props{error_code} = $j->GetErrorCode() if ( $j->DefinedErrorCode() );
1006    $props{xml}        = $j->GetXML();
1007
1008    if ( $jtype eq 'chat' ) {
1009        $props{replycmd} =
1010          "jwrite " . ( ( $dir eq 'in' ) ? $props{from} : $props{to} );
1011        $props{replycmd} .=
1012          " -a " . ( ( $dir eq 'out' ) ? $props{from} : $props{to} );
1013        $props{private} = 1;
1014    }
1015    elsif ( $jtype eq 'groupchat' ) {
1016        my $nick = $props{nick} = $from->GetResource();
1017        my $room = $props{room} = $from->GetJID('base');
1018        $props{replycmd} = "jwrite -g $room";
1019        $props{replycmd} .=
1020          " -a " . ( ( $dir eq 'out' ) ? $props{from} : $props{to} );
1021
1022        $props{sender} = $nick || $room;
1023        $props{recipient} = $room;
1024
1025        if ( $props{subject} && !$props{body} ) {
1026            $props{body} =
1027              '[' . $nick . " has set the topic to: " . $props{subject} . "]";
1028        }
1029    }
1030    elsif ( $jtype eq 'normal' ) {
1031        $props{replycmd}  = undef;
1032        $props{private} = 1;
1033    }
1034    elsif ( $jtype eq 'headline' ) {
1035        $props{replycmd} = undef;
1036    }
1037    elsif ( $jtype eq 'error' ) {
1038        $props{replycmd} = undef;
1039        $props{body}     = "Error "
1040          . $props{error_code}
1041          . " sending to "
1042          . $props{from} . "\n"
1043          . $props{error};
1044    }
1045
1046    $props{replysendercmd} = $props{replycmd};
1047    return %props;
1048}
1049
1050sub j2o {
1051    return owl::Message->new( j2hash(@_) );
1052}
1053
1054sub queue_admin_msg {
1055    my $err = shift;
1056    my $m   = owl::Message->new(
1057        type      => 'admin',
1058        direction => 'none',
1059        body      => $err
1060    );
1061    owl::queue_message($m);
1062}
1063
1064sub boldify($) {
1065    my $str = shift;
1066
1067    return '@b(' . $str . ')' if ( $str !~ /\)/ );
1068    return '@b<' . $str . '>' if ( $str !~ /\>/ );
1069    return '@b{' . $str . '}' if ( $str !~ /\}/ );
1070    return '@b[' . $str . ']' if ( $str !~ /\]/ );
1071
1072    my $txt = "\@b($str";
1073    $txt =~ s/\)/\)\@b\[\)\]\@b\(/g;
1074    return $txt . ')';
1075}
1076
1077sub getServerFromJID {
1078    my $jid = shift;
1079    my $res = new Net::DNS::Resolver;
1080    my $packet =
1081      $res->search( '_xmpp-client._tcp.' . $jid->GetServer(), 'srv' );
1082
1083    if ($packet)    # Got srv record.
1084    {
1085        my @answer = $packet->answer;
1086        return $answer[0]{target}, $answer[0]{port};
1087    }
1088
1089    return $jid->GetServer(), 5222;
1090}
1091
1092sub defaultJID {
1093    return ( $conn->getJids() )[0] if ( $conn->connected() == 1 );
1094    return;
1095}
1096
1097sub baseJID {
1098    my $givenJidStr = shift;
1099    my $givenJid    = new Net::XMPP::JID;
1100    $givenJid->SetJID($givenJidStr);
1101    return $givenJid->GetJID('base');
1102}
1103
1104sub resolveJID {
1105    my $givenJidStr = shift;
1106    my $givenJid    = new Net::XMPP::JID;
1107    $givenJid->SetJID($givenJidStr);
1108
1109    # Account fully specified.
1110    if ( $givenJid->GetResource() ) {
1111        # Specified account exists
1112        return $givenJidStr if ($conn->jidExists($givenJidStr) );
1113        owl::error("Invalid account: $givenJidStr");
1114    }
1115
1116    # Disambiguate.
1117    else {
1118        my $matchingJid = "";
1119        my $errStr =
1120          "Ambiguous account reference. Please specify a resource.\n";
1121        my $ambiguous = 0;
1122
1123        foreach my $jid ( $conn->getJids() ) {
1124            my $cJid = new Net::XMPP::JID;
1125            $cJid->SetJID($jid);
1126            if ( $givenJidStr eq $cJid->GetJID('base') ) {
1127                $ambiguous = 1 if ( $matchingJid ne "" );
1128                $matchingJid = $jid;
1129                $errStr .= "\t$jid\n";
1130            }
1131        }
1132
1133        # Need further disambiguation.
1134        if ($ambiguous) {
1135            queue_admin_msg($errStr);
1136        }
1137
1138        # Not one of ours.
1139        elsif ( $matchingJid eq "" ) {
1140            owl::error("Invalid account: $givenJidStr");
1141        }
1142
1143        # It's this one.
1144        else {
1145            return $matchingJid;
1146        }
1147    }
1148    return "";
1149}
1150
11511;
Note: See TracBrowser for help on using the repository browser.