Adding old-school presence so I can get presence support working in Empathy, wahooo
[theonering] / src / connection.py
index 32b47c2..f359cd5 100644 (file)
@@ -4,52 +4,72 @@ import logging
 import telepathy
 
 import constants
+import util.go_utils as gobject_utils
+import util.coroutines as coroutines
+import gtk_toolbox
 import gvoice
 import handle
-import channel_manager
+import aliasing
 import simple_presence
+import presence
+import capabilities
+import channel_manager
 
 
 _moduleLogger = logging.getLogger("connection")
 
 
-class TheOneRingConnection(telepathy.server.Connection, simple_presence.SimplePresenceMixin):
+class TheOneRingConnection(
+       telepathy.server.Connection,
+       aliasing.AliasingMixin,
+       simple_presence.SimplePresenceMixin,
+       presence.PresenceMixin,
+       capabilities.CapabilitiesMixin,
+):
 
-       MANDATORY_PARAMETERS = {
-               'account' : 's',
+       # Overriding a base class variable
+       # Should the forwarding number be handled by the alias or by an option?
+       _mandatory_parameters = {
+               'username' : 's',
                'password' : 's',
                'forward' : 's',
        }
-       OPTIONAL_PARAMETERS = {
+       # Overriding a base class variable
+       _optional_parameters = {
        }
-       PARAMETER_DEFAULTS = {
+       _parameter_defaults = {
        }
 
        def __init__(self, manager, parameters):
+               self.check_parameters(parameters)
                try:
-                       self.check_parameters(parameters)
-                       account = unicode(parameters['account'])
+                       account = unicode(parameters['username'])
 
+                       # Connection init must come first
                        telepathy.server.Connection.__init__(
                                self,
                                constants._telepathy_protocol_name_,
                                account,
                                constants._telepathy_implementation_name_
                        )
+                       aliasing.AliasingMixin.__init__(self)
+                       simple_presence.SimplePresenceMixin.__init__(self)
+                       presence.PresenceMixin.__init__(self)
+                       capabilities.CapabilitiesMixin.__init__(self)
 
                        self._manager = weakref.proxy(manager)
                        self._credentials = (
-                               parameters['account'].encode('utf-8'),
+                               parameters['username'].encode('utf-8'),
                                parameters['password'].encode('utf-8'),
                        )
                        self._callbackNumber = parameters['forward'].encode('utf-8')
                        self._channelManager = channel_manager.ChannelManager(self)
 
-                       cookieFilePath = "%s/cookies.txt" % constants._data_path_
-                       self._session = gvoice.session.Session(cookieFilePath)
+                       self._session = gvoice.session.Session(None)
 
                        self.set_self_handle(handle.create_handle(self, 'connection'))
 
+                       self._callback = None
                        _moduleLogger.info("Connection to the account %s created" % account)
                except Exception, e:
                        _moduleLogger.exception("Failed to create Connection")
@@ -65,45 +85,71 @@ class TheOneRingConnection(telepathy.server.Connection, simple_presence.SimplePr
 
        @property
        def username(self):
-               self._credentials[0]
+               return self._credentials[0]
 
        def handle(self, handleType, handleId):
                self.check_handle(handleType, handleId)
                return self._handles[handleType, handleId]
 
+       @gtk_toolbox.log_exception(_moduleLogger)
        def Connect(self):
                """
                For org.freedesktop.telepathy.Connection
                """
+               _moduleLogger.info("Connecting...")
                self.StatusChanged(
                        telepathy.CONNECTION_STATUS_CONNECTING,
                        telepathy.CONNECTION_STATUS_REASON_REQUESTED
                )
                try:
+                       cookieFilePath = None
+                       self._session = gvoice.session.Session(cookieFilePath)
+
+                       self._callback = coroutines.func_sink(
+                               coroutines.expand_positional(
+                                       self._on_conversations_updated
+                               )
+                       )
+                       self.session.conversations.updateSignalHandler.register_sink(
+                               self._callback
+                       )
                        self.session.login(*self._credentials)
                        self.session.backend.set_callback_number(self._callbackNumber)
-               except gvoice.backend.NetworkError:
+               except gvoice.backend.NetworkError, e:
+                       _moduleLogger.exception("Connection Failed")
                        self.StatusChanged(
                                telepathy.CONNECTION_STATUS_DISCONNECTED,
                                telepathy.CONNECTION_STATUS_REASON_NETWORK_ERROR
                        )
-               except Exception:
+               except Exception, e:
+                       _moduleLogger.exception("Connection Failed")
                        self.StatusChanged(
                                telepathy.CONNECTION_STATUS_DISCONNECTED,
                                telepathy.CONNECTION_STATUS_REASON_AUTHENTICATION_FAILED
                        )
                else:
+                       _moduleLogger.info("Connected")
                        self.StatusChanged(
                                telepathy.CONNECTION_STATUS_CONNECTED,
                                telepathy.CONNECTION_STATUS_REASON_REQUESTED
                        )
 
+       @gtk_toolbox.log_exception(_moduleLogger)
        def Disconnect(self):
                """
                For org.freedesktop.telepathy.Connection
+               @bug Not properly logging out.  Cookie files need to be per connection and removed
                """
+               _moduleLogger.info("Disconnecting")
                try:
+                       self.session.conversations.updateSignalHandler.unregister_sink(
+                               self._callback
+                       )
+                       self._callback = None
+                       self._channelManager.close()
                        self.session.logout()
+                       self.session.close()
+                       self._session = None
                        _moduleLogger.info("Disconnected")
                except Exception:
                        _moduleLogger.exception("Disconnecting Failed")
@@ -111,7 +157,9 @@ class TheOneRingConnection(telepathy.server.Connection, simple_presence.SimplePr
                        telepathy.CONNECTION_STATUS_DISCONNECTED,
                        telepathy.CONNECTION_STATUS_REASON_REQUESTED
                )
+               self.manager.disconnected(self)
 
+       @gtk_toolbox.log_exception(_moduleLogger)
        def RequestChannel(self, type, handleType, handleId, suppressHandler):
                """
                For org.freedesktop.telepathy.Connection
@@ -122,29 +170,32 @@ class TheOneRingConnection(telepathy.server.Connection, simple_presence.SimplePr
                @returns DBus object path for the channel created or retrieved
                """
                self.check_connected()
+               self.check_handle(handleType, handleId)
 
                channel = None
                channelManager = self._channelManager
                handle = self.handle(handleType, handleId)
 
                if type == telepathy.CHANNEL_TYPE_CONTACT_LIST:
+                       _moduleLogger.info("RequestChannel ContactList")
                        channel = channelManager.channel_for_list(handle, suppressHandler)
                elif type == telepathy.CHANNEL_TYPE_TEXT:
-                       if handleType != telepathy.HANDLE_TYPE_CONTACT:
-                               raise telepathy.NotImplemented("Only Contacts are allowed")
-                       channel = channelManager.channel_for_text(handle, None, suppressHandler)
+                       _moduleLogger.info("RequestChannel Text")
+                       channel = channelManager.channel_for_text(handle, suppressHandler)
                elif type == telepathy.CHANNEL_TYPE_STREAMED_MEDIA:
-                       if handleType != telepathy.HANDLE_TYPE_CONTACT:
-                               raise telepathy.NotImplemented("Only Contacts are allowed")
-                       channel = channelManager.channel_for_text(handle, None, suppressHandler)
+                       _moduleLogger.info("RequestChannel Media")
+                       channel = channelManager.channel_for_call(handle, suppressHandler)
                else:
-                       raise telepathy.NotImplemented("unknown channel type %s" % type)
+                       raise telepathy.errors.NotImplemented("unknown channel type %s" % type)
 
+               _moduleLogger.info("RequestChannel Object Path: %s" % channel._object_path)
                return channel._object_path
 
+       @gtk_toolbox.log_exception(_moduleLogger)
        def RequestHandles(self, handleType, names, sender):
                """
                For org.freedesktop.telepathy.Connection
+               Overiding telepathy.server.Connecton to allow custom handles
                """
                self.check_connected()
                self.check_handle_type(handleType)
@@ -153,29 +204,33 @@ class TheOneRingConnection(telepathy.server.Connection, simple_presence.SimplePr
                for name in names:
                        name = name.encode('utf-8')
                        if handleType == telepathy.HANDLE_TYPE_CONTACT:
+                               _moduleLogger.info("RequestHandles Contact: %s" % name)
                                h = self._create_contact_handle(name)
                        elif handleType == telepathy.HANDLE_TYPE_LIST:
                                # Support only server side (immutable) lists
+                               _moduleLogger.info("RequestHandles List: %s" % name)
                                h = handle.create_handle(self, 'list', name)
                        else:
-                               raise telepathy.NotAvailable('Handle type unsupported %d' % handleType)
+                               raise telepathy.errors.NotAvailable('Handle type unsupported %d' % handleType)
                        handles.append(h.id)
-                       self.add_client_handle(handle, sender)
+                       self.add_client_handle(h, sender)
                return handles
 
-       def _create_contact_handle(self, name):
-               requestedContactId = name
-
-               contacts = self.session.addressbook.get_contacts()
-               contactsFound = [
-                       contactId for contactId in contacts
-                       if contactId == requestedContactId
-               ]
-
-               if 0 < len(contactsFound):
-                       contactId = contactsFound[0]
-                       if len(contactsFound) != 1:
-                               _moduleLogger.error("Contact ID was not unique: %s for %s" % (contactId, ))
-               else:
-                       contactId = requestedContactId
-               h = handle.create_handle(self, 'contact', contactId)
+       def _create_contact_handle(self, requestedHandleName):
+               requestedContactId, requestedContactNumber = handle.ContactHandle.from_handle_name(
+                       requestedHandleName
+               )
+               h = handle.create_handle(self, 'contact', requestedContactId, requestedContactNumber)
+               return h
+
+       @gobject_utils.async
+       @gtk_toolbox.log_exception(_moduleLogger)
+       def _on_conversations_updated(self, conv, conversationIds):
+               # @todo get conversations update running
+               # @todo test conversatiuons
+               _moduleLogger.info("Incoming messages from: %r" % (conversationIds, ))
+               channelManager = self._channelManager
+               for contactId, phoneNumber in conversationIds:
+                       h = handle.create_handle(self, 'contact', contactId, phoneNumber)
+                       # Just let the TextChannel decide whether it should be reported to the user or not
+                       channel = channelManager.channel_for_text(h)