Adding aliasing support to give pretty names to contacts

diff --git a/src/aliasing.py b/src/aliasing.py
new file mode 100644 (file)
index 0000000..02a483e
--- /dev/null
+++ b/src/aliasing.py
@@ -0,0 +1,113 @@
+import logging
+
+import telepathy
+
+import gtk_toolbox
+import handle
+
+
+_moduleLogger = logging.getLogger('aliasing')
+
+
+def make_pretty(phonenumber):
+       """
+       Function to take a phone number and return the pretty version
+       pretty numbers:
+               if phonenumber begins with 0:
+                       ...-(...)-...-....
+               if phonenumber begins with 1: ( for gizmo callback numbers )
+                       1 (...)-...-....
+               if phonenumber is 13 digits:
+                       (...)-...-....
+               if phonenumber is 10 digits:
+                       ...-....
+       >>> make_pretty("12")
+       '12'
+       >>> make_pretty("1234567")
+       '123-4567'
+       >>> make_pretty("2345678901")
+       '(234)-567-8901'
+       >>> make_pretty("12345678901")
+       '1 (234)-567-8901'
+       >>> make_pretty("01234567890")
+       '+012-(345)-678-90'
+       """
+       if phonenumber is None or phonenumber is "":
+               return ""
+
+       phonenumber = handle.strip_number(phonenumber)
+
+       if len(phonenumber) < 3:
+               return phonenumber
+
+       if phonenumber[0] == "0":
+               prettynumber = ""
+               prettynumber += "+%s" % phonenumber[0:3]
+               if 3 < len(phonenumber):
+                       prettynumber += "-(%s)" % phonenumber[3:6]
+                       if 6 < len(phonenumber):
+                               prettynumber += "-%s" % phonenumber[6:9]
+                               if 9 < len(phonenumber):
+                                       prettynumber += "-%s" % phonenumber[9:]
+               return prettynumber
+       elif len(phonenumber) <= 7:
+               prettynumber = "%s-%s" % (phonenumber[0:3], phonenumber[3:])
+       elif len(phonenumber) > 8 and phonenumber[0] == "1":
+               prettynumber = "1 (%s)-%s-%s" % (phonenumber[1:4], phonenumber[4:7], phonenumber[7:])
+       elif len(phonenumber) > 7:
+               prettynumber = "(%s)-%s-%s" % (phonenumber[0:3], phonenumber[3:6], phonenumber[6:])
+       return prettynumber
+
+
+class AliasingMixin(telepathy.server.ConnectionInterfaceAliasing):
+
+       def __init__(self):
+               telepathy.server.ConnectionInterfaceAliasing.__init__(self)
+
+       @property
+       def session(self):
+               """
+               @abstract
+               """
+               raise NotImplementedError("Abstract property called")
+
+       @property
+       def handle(self):
+               """
+               @abstract
+               """
+               raise NotImplementedError("Abstract property called")
+
+       @gtk_toolbox.log_exception(_moduleLogger)
+       def GetAliasFlags(self):
+               return 0
+
+       @gtk_toolbox.log_exception(_moduleLogger)
+       def RequestAliases(self, contactHandleIds):
+               _moduleLogger.debug("Called RequestAliases")
+               return [self._get_alias(handleId) for handleId in contactHandleIds]
+
+       @gtk_toolbox.log_exception(_moduleLogger)
+       def GetAliases(self, contactHandleIds):
+               _moduleLogger.debug("Called GetAliases")
+
+               idToAlias = dict(
+                       (handleId, self._get_alias(handleId))
+                       for handleId in contactHandleIds
+               )
+               return idToAlias
+
+       @gtk_toolbox.log_exception(_moduleLogger)
+       def SetAliases(self, aliases):
+               # @todo Look into setting callback number through this mechanism
+               raise telepathy.PermissionDenied("No user customizable aliases")
+
+       def _get_alias(self, handleId):
+               h = self.handle(telepathy.HANDLE_TYPE_CONTACT, handleId)
+               if isinstance(h, handle.ConnectionHandle):
+                       callbackNumber = self.session.backend.get_callback_number()
+                       userAlias = make_pretty(callbackNumber)
+                       return userAlias
+               else:
+                       contactAlias = self.session.addressbook.get_contact_name(h.contactID)
+                       return contactAlias

diff --git a/src/connection.py b/src/connection.py
index 2c2b8e2..eb2abc6 100644 (file)
--- a/src/connection.py
+++ b/src/connection.py
@@ -7,13 +7,17 @@ import constants
 import gtk_toolbox
 import gvoice
 import handle
+import aliasing
 import channel_manager
 
 
 _moduleLogger = logging.getLogger("connection")
 
 
-class TheOneRingConnection(telepathy.server.Connection):
+class TheOneRingConnection(
+       telepathy.server.Connection,
+       aliasing.AliasingMixin
+):
 
        # Overriding a base class variable
        _mandatory_parameters = {
@@ -32,12 +36,14 @@ class TheOneRingConnection(telepathy.server.Connection):
                        self.check_parameters(parameters)
                        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)
 
                        self._manager = weakref.proxy(manager)
                        self._credentials = (