Reducing not-helpful traces
[theonering] / src / connection.py
1 import os
2 import weakref
3 import logging
4
5 import telepathy
6
7 import constants
8 import tp
9 import util.go_utils as gobject_utils
10 import util.misc as misc_utils
11
12 import gvoice
13 import handle
14
15 import aliasing
16 import avatars
17 import capabilities
18 import contacts
19 import presence
20 import requests
21 import simple_presence
22
23 import autogv
24 import channel_manager
25
26
27 _moduleLogger = logging.getLogger(__name__)
28
29
30 class TheOneRingOptions(object):
31
32         useGVContacts = True
33
34         assert gvoice.session.Session._DEFAULTS["contacts"][1] == "hours"
35         contactsPollPeriodInHours = gvoice.session.Session._DEFAULTS["contacts"][0]
36
37         assert gvoice.session.Session._DEFAULTS["voicemail"][1] == "minutes"
38         voicemailPollPeriodInMinutes = gvoice.session.Session._DEFAULTS["voicemail"][0]
39
40         assert gvoice.session.Session._DEFAULTS["texts"][1] == "minutes"
41         textsPollPeriodInMinutes = gvoice.session.Session._DEFAULTS["texts"][0]
42
43         def __init__(self, parameters = None):
44                 if parameters is None:
45                         return
46                 self.useGVContacts = parameters["use-gv-contacts"]
47                 self.contactsPollPeriodInHours = parameters['contacts-poll-period-in-hours']
48                 self.voicemailPollPeriodInMinutes = parameters['voicemail-poll-period-in-minutes']
49                 self.textsPollPeriodInMinutes = parameters['texts-poll-period-in-minutes']
50
51
52 class TheOneRingConnection(
53         tp.Connection,
54         aliasing.AliasingMixin,
55         avatars.AvatarsMixin,
56         capabilities.CapabilitiesMixin,
57         contacts.ContactsMixin,
58         presence.PresenceMixin,
59         requests.RequestsMixin,
60         simple_presence.SimplePresenceMixin,
61 ):
62
63         # overiding base class variable
64         _mandatory_parameters = {
65                 'account': 's',
66                 'password': 's',
67         }
68         # overiding base class variable
69         _optional_parameters = {
70                 'forward': 's',
71                 'use-gv-contacts': 'b',
72                 'contacts-poll-period-in-hours': 'i',
73                 'voicemail-poll-period-in-minutes': 'i',
74                 'texts-poll-period-in-minutes': 'i',
75         }
76         _parameter_defaults = {
77                 'forward': '',
78                 'use-gv-contacts': TheOneRingOptions.useGVContacts,
79                 'contacts-poll-period-in-hours': TheOneRingOptions.contactsPollPeriodInHours,
80                 'voicemail-poll-period-in-minutes': TheOneRingOptions.voicemailPollPeriodInMinutes,
81                 'texts-poll-period-in-minutes': TheOneRingOptions.textsPollPeriodInMinutes,
82         }
83         _secret_parameters = set((
84                 "password",
85         ))
86
87         @misc_utils.log_exception(_moduleLogger)
88         def __init__(self, manager, parameters):
89                 self.check_parameters(parameters)
90                 account = unicode(parameters['account'])
91                 encodedAccount = parameters['account'].encode('utf-8')
92                 encodedPassword = parameters['password'].encode('utf-8')
93                 encodedCallback = misc_utils.normalize_number(parameters['forward'].encode('utf-8'))
94                 if encodedCallback and not misc_utils.is_valid_number(encodedCallback):
95                         raise telepathy.errors.InvalidArgument("Invalid forwarding number")
96
97                 # Connection init must come first
98                 self.__options = TheOneRingOptions(parameters)
99                 self.__session = gvoice.session.Session(
100                         cookiePath = None,
101                         defaults = {
102                                 "contacts": (self.__options.contactsPollPeriodInHours, "hours"),
103                                 "voicemail": (self.__options.voicemailPollPeriodInMinutes, "minutes"),
104                                 "texts": (self.__options.textsPollPeriodInMinutes, "minutes"),
105                         },
106                 )
107                 tp.Connection.__init__(
108                         self,
109                         constants._telepathy_protocol_name_,
110                         account,
111                         constants._telepathy_implementation_name_
112                 )
113                 aliasing.AliasingMixin.__init__(self)
114                 avatars.AvatarsMixin.__init__(self)
115                 capabilities.CapabilitiesMixin.__init__(self)
116                 contacts.ContactsMixin.__init__(self)
117                 presence.PresenceMixin.__init__(self)
118                 requests.RequestsMixin.__init__(self)
119                 simple_presence.SimplePresenceMixin.__init__(self)
120
121                 self.__manager = weakref.proxy(manager)
122                 self.__credentials = (
123                         encodedAccount,
124                         encodedPassword,
125                 )
126                 self.__callbackNumberParameter = encodedCallback
127                 self.__channelManager = channel_manager.ChannelManager(self)
128
129                 self.__cachePath = os.sep.join((constants._data_path_, "cache", self.username))
130                 try:
131                         os.makedirs(self.__cachePath)
132                 except OSError, e:
133                         if e.errno != 17:
134                                 raise
135
136                 self.set_self_handle(handle.create_handle(self, 'connection'))
137                 self._plumbing = [
138                         autogv.NewGVConversations(weakref.ref(self)),
139                         autogv.RefreshVoicemail(weakref.ref(self)),
140                         autogv.AutoDisconnect(weakref.ref(self)),
141                 ]
142                 self._delayedConnect = gobject_utils.Async(self._delayed_connect)
143
144                 _moduleLogger.info("Connection to the account %s created" % account)
145
146         @property
147         def manager(self):
148                 return self.__manager
149
150         @property
151         def session(self):
152                 return self.__session
153
154         @property
155         def options(self):
156                 return self.__options
157
158         @property
159         def username(self):
160                 return self.__credentials[0]
161
162         @property
163         def callbackNumberParameter(self):
164                 return self.__callbackNumberParameter
165
166         def get_handle_by_name(self, handleType, handleName):
167                 requestedHandleName = handleName.encode('utf-8')
168                 if handleType == telepathy.HANDLE_TYPE_CONTACT:
169                         h = handle.create_handle(self, 'contact', requestedHandleName)
170                 elif handleType == telepathy.HANDLE_TYPE_LIST:
171                         # Support only server side (immutable) lists
172                         h = handle.create_handle(self, 'list', requestedHandleName)
173                 else:
174                         raise telepathy.errors.NotAvailable('Handle type unsupported %d' % handleType)
175                 return h
176
177         @property
178         def _channel_manager(self):
179                 return self.__channelManager
180
181         @misc_utils.log_exception(_moduleLogger)
182         def Connect(self):
183                 """
184                 For org.freedesktop.telepathy.Connection
185                 """
186                 if self._status != telepathy.CONNECTION_STATUS_DISCONNECTED:
187                         _moduleLogger.info("Attempting connect when not disconnected")
188                         return
189                 _moduleLogger.info("Kicking off connect")
190                 self._delayedConnect.start()
191
192         @misc_utils.log_exception(_moduleLogger)
193         def _delayed_connect(self):
194                 _moduleLogger.info("Connecting...")
195                 self.StatusChanged(
196                         telepathy.CONNECTION_STATUS_CONNECTING,
197                         telepathy.CONNECTION_STATUS_REASON_REQUESTED
198                 )
199                 try:
200                         self.__session.load(self.__cachePath)
201
202                         for plumber in self._plumbing:
203                                 plumber.start()
204                         self.session.login(*self.__credentials)
205                         if not self.__callbackNumberParameter:
206                                 callback = gvoice.backend.get_sane_callback(
207                                         self.session.backend
208                                 )
209                                 self.__callbackNumberParameter = misc_utils.normalize_number(callback)
210                         self.session.backend.set_callback_number(self.__callbackNumberParameter)
211
212                         subscribeHandle = self.get_handle_by_name(telepathy.HANDLE_TYPE_LIST, "subscribe")
213                         subscribeProps = self.generate_props(telepathy.CHANNEL_TYPE_CONTACT_LIST, subscribeHandle, False)
214                         self.__channelManager.channel_for_props(subscribeProps, signal=True)
215                         publishHandle = self.get_handle_by_name(telepathy.HANDLE_TYPE_LIST, "publish")
216                         publishProps = self.generate_props(telepathy.CHANNEL_TYPE_CONTACT_LIST, publishHandle, False)
217                         self.__channelManager.channel_for_props(publishProps, signal=True)
218                 except gvoice.backend.NetworkError:
219                         _moduleLogger.exception("Connection Failed")
220                         self.disconnect(telepathy.CONNECTION_STATUS_REASON_NETWORK_ERROR)
221                         return
222                 except Exception:
223                         _moduleLogger.exception("Connection Failed")
224                         self.disconnect(telepathy.CONNECTION_STATUS_REASON_AUTHENTICATION_FAILED)
225                         return
226
227                 _moduleLogger.info("Connected")
228                 self.StatusChanged(
229                         telepathy.CONNECTION_STATUS_CONNECTED,
230                         telepathy.CONNECTION_STATUS_REASON_REQUESTED
231                 )
232
233         @misc_utils.log_exception(_moduleLogger)
234         def Disconnect(self):
235                 """
236                 For org.freedesktop.telepathy.Connection
237                 """
238                 _moduleLogger.info("Kicking off disconnect")
239                 self.disconnect(telepathy.CONNECTION_STATUS_REASON_REQUESTED)
240
241         @misc_utils.log_exception(_moduleLogger)
242         def RequestChannel(self, type, handleType, handleId, suppressHandler):
243                 """
244                 For org.freedesktop.telepathy.Connection
245
246                 @param type DBus interface name for base channel type
247                 @param handleId represents a contact, list, etc according to handleType
248
249                 @returns DBus object path for the channel created or retrieved
250                 """
251                 self.check_connected()
252                 self.check_handle(handleType, handleId)
253
254                 h = self.get_handle_by_id(handleType, handleId) if handleId != 0 else None
255                 props = self.generate_props(type, h, suppressHandler)
256                 self._validate_handle(props)
257
258                 chan = self.__channelManager.channel_for_props(props, signal=True)
259                 path = chan._object_path
260                 _moduleLogger.info("RequestChannel Object Path (%s): %s" % (type.rsplit(".", 1)[-1], path))
261                 return path
262
263         def generate_props(self, channelType, handleObj, suppressHandler, initiatorHandle=None):
264                 targetHandle = 0 if handleObj is None else handleObj.get_id()
265                 targetHandleType = telepathy.HANDLE_TYPE_NONE if handleObj is None else handleObj.get_type()
266                 props = {
267                         telepathy.CHANNEL_INTERFACE + '.ChannelType': channelType,
268                         telepathy.CHANNEL_INTERFACE + '.TargetHandle': targetHandle,
269                         telepathy.CHANNEL_INTERFACE + '.TargetHandleType': targetHandleType,
270                         telepathy.CHANNEL_INTERFACE + '.Requested': suppressHandler
271                 }
272
273                 if initiatorHandle is not None:
274                         props[telepathy.CHANNEL_INTERFACE + '.InitiatorHandle'] = initiatorHandle.id
275
276                 return props
277
278         def disconnect(self, reason):
279                 _moduleLogger.info("Disconnecting")
280
281                 self._delayedConnect.cancel()
282
283                 # Not having the disconnect first can cause weird behavior with clients
284                 # including not being able to reconnect or even crashing
285                 self.StatusChanged(
286                         telepathy.CONNECTION_STATUS_DISCONNECTED,
287                         reason,
288                 )
289
290                 for plumber in self._plumbing:
291                         plumber.stop()
292
293                 self.__channelManager.close()
294                 self.manager.disconnected(self)
295
296                 self.session.save(self.__cachePath)
297                 self.session.logout()
298                 self.session.close()
299
300                 _moduleLogger.info("Disconnected")