source: perl/modules/Jabber/lib/BarnOwl/Module/Jabber/ConnectionManager.pm @ f3678c3

release-1.5release-1.6release-1.7release-1.8release-1.9
Last change on this file since f3678c3 was f3678c3, checked in by Alex Vandiver <alexmv@mit.edu>, 12 years ago
Add a jabber:reconnect (default to on) which controls auto-reconnect
  • Property mode set to 100644
File size: 4.7 KB
Line 
1use warnings;
2use strict;
3
4=head1 NAME
5
6BarnOwl::Module::Jabber::ConnectionManager
7
8=head1 DESCRIPTION
9
10A class to keep track of all the active connection in the barnowl
11jabber module
12
13=cut
14
15package BarnOwl::Module::Jabber::ConnectionManager;
16
17sub new {
18    my $class = shift;
19    return bless { }, $class;
20}
21
22sub addConnection {
23    my $self = shift;
24    my $jidStr = shift;
25
26    my $client = BarnOwl::Module::Jabber::Connection->new;
27
28    $self->{$jidStr}->{Client} = $client;
29    $self->{$jidStr}->{Roster} = $client->Roster();
30    $self->{$jidStr}->{Status} = "available";
31    return $client;
32}
33
34sub removeConnection {
35    my $self = shift;
36    my $jidStr = shift;
37    return 0 unless exists $self->{$jidStr};
38
39    BarnOwl::remove_dispatch($self->{$jidStr}->{Client}->{fileno}) if $self->{$jidStr}->{Client}->{fileno};
40    $self->{$jidStr}->{Client}->Disconnect()
41      if $self->{$jidStr}->{Client};
42    delete $self->{$jidStr};
43
44    return 1;
45}
46
47sub scheduleReconnect {
48    my $self = shift;
49    my $jidStr = shift;
50    return 0 unless exists $self->{$jidStr};
51    BarnOwl::admin_message(Jabber => "Disconnected from jabber account $jidStr");
52
53    unless (BarnOwl::getvar('jabber:reconnect') eq 'on') {
54        return $self->removeConnection($jidStr);
55    }
56
57    BarnOwl::remove_dispatch($self->{$jidStr}->{Client}->{fileno}) if $self->{$jidStr}->{Client}->{fileno};
58    $self->{$jidStr}->{Client}->Disconnect()
59      if $self->{$jidStr}->{Client};
60
61    $self->{$jidStr}->{Status} = "reconnecting";
62    $self->{$jidStr}->{ReconnectBackoff} = 5;
63    $self->{$jidStr}->{ReconnectAt} = time + $self->{$jidStr}->{ReconnectBackoff};
64    return 1;
65}
66
67sub setAuth {
68    my $self = shift;
69    my $jidStr = shift;
70    $self->{$jidStr}->{Auth} = shift;
71}
72
73sub tryReconnect {
74    my $self = shift;
75    my $jidStr = shift;
76    my $force = shift;
77
78    return 0 unless exists $self->{$jidStr};
79    return 0 unless $self->{$jidStr}{Status} eq "reconnecting";
80    return 0 unless $force or (time > $self->{$jidStr}{ReconnectAt});
81
82    $self->{$jidStr}->{ReconnectBackoff} *= 2;
83    $self->{$jidStr}->{ReconnectBackoff} = 60*5
84        if $self->{$jidStr}->{ReconnectBackoff} > 60*5;
85    $self->{$jidStr}->{ReconnectAt} = time + $self->{$jidStr}->{ReconnectBackoff};
86
87    my $status = $self->{$jidStr}->{Client}->Connect;
88    return 0 unless $status;
89
90    my @result = $self->{$jidStr}->{Client}->AuthSend( %{ $self->{$jidStr}->{Auth} } );
91    if ( !@result || $result[0] ne 'ok' ) {
92        $self->removeConnection($jidStr);
93        BarnOwl::error( "Error in jabber reconnect: " . join( " ", @result ) );
94        return 0;
95    }
96
97    BarnOwl::admin_message(Jabber => "Reconnected to jabber as $jidStr");
98    $self->{$jidStr}{Status} = "available";
99
100    return 1;
101}
102
103sub renameConnection {
104    my $self = shift;
105    my $oldJidStr = shift;
106    my $newJidStr = shift;
107    return 0 unless exists $self->{$oldJidStr};
108    return 0 if $oldJidStr eq $newJidStr;
109
110    $self->{$newJidStr} = $self->{$oldJidStr};
111    delete $self->{$oldJidStr};
112    return 1;
113}
114
115sub connected {
116    my $self = shift;
117    return scalar keys %{ $self };
118}
119
120sub getJIDs {
121    my $self = shift;
122    return keys %{ $self };
123}
124
125sub jidExists {
126    my $self = shift;
127    my $jidStr = shift;
128    return exists $self->{$jidStr};
129}
130
131sub baseJIDExists {
132    my $self = shift;
133    my $jidStr = shift;
134    my $jid = new Net::Jabber::JID;
135    $jid->SetJID($jidStr);
136    my $baseJID = $jid->GetJID('base');
137
138    foreach my $cjidStr ( keys %{ $self } ) {
139        my $cJID = new Net::Jabber::JID;
140        $cJID->SetJID($cjidStr);
141        return $cjidStr if ($cJID->GetJID('base') eq $baseJID);
142    }
143    return 0;
144}
145
146sub jidActive {
147    my $self = shift;
148    my $jidStr = shift;
149    return(exists $self->{$jidStr} and $self->{$jidStr}{Status} eq "available");
150}
151
152sub sidExists {
153    my $self = shift;
154    my $sid = shift || "";
155    foreach my $c ( values %{ $self } ) {
156        return 1 if ($c->{Client}->{SESSION}->{id} eq $sid);
157    }
158    return 0;
159}
160
161sub getConnectionFromSid {
162    my $self = shift;
163    my $sid = shift;
164    foreach my $c (values %{ $self }) {
165        return $c->{Client} if $c->{Client}->{SESSION}->{id} eq $sid;
166    }
167    return undef;
168}
169
170sub getConnectionFromJID {
171    my $self = shift;
172    my $jid = shift;
173    $jid = $jid->GetJID('full') if UNIVERSAL::isa($jid, 'Net::XMPP::JID');
174    return $self->{$jid}->{Client} if $self->jidActive($jid);
175}
176
177sub getRosterFromSid {
178    my $self = shift;
179    my $sid = shift;
180    foreach my $c (values %{ $self }) {
181        return $c->{Roster}
182          if $c->{Client}->{SESSION}->{id} eq $sid;
183    }
184    return undef;
185}
186
187sub getRosterFromJID {
188    my $self = shift;
189    my $jid = shift;
190    $jid = $jid->GetJID('full') if UNIVERSAL::isa($jid, 'Net::XMPP::JID');
191    return $self->{$jid}->{Roster} if $self->jidExists($jid);
192}
193
194
1951;
Note: See TracBrowser for help on using the repository browser.