git.maemo.org Git - theonering/blob - src/channel/text.py

   1 import time
   2 import datetime
   3 import logging
   4
   5 import telepathy
   6
   7 import util.go_utils as gobject_utils
   8 import util.coroutines as coroutines
   9 import gtk_toolbox
  10
  11
  12 _moduleLogger = logging.getLogger("channel.text")
  13
  14
  15 class TextChannel(telepathy.server.ChannelTypeText):
  16         """
  17         Look into implementing ChannelInterfaceMessages for rich text formatting
  18         """
  19
  20         def __init__(self, connection, h):
  21                 telepathy.server.ChannelTypeText.__init__(self, connection, h)
  22                 self._nextRecievedId = 0
  23                 self._lastMessageTimestamp = datetime.datetime(1, 1, 1)
  24
  25                 self._otherHandle = h
  26
  27                 self._callback = coroutines.func_sink(
  28                         coroutines.expand_positional(
  29                                 self._on_conversations_updated
  30                         )
  31                 )
  32                 self._conn.session.conversations.updateSignalHandler.register_sink(
  33                         self._callback
  34                 )
  35
  36                 # The only reason there should be anything in the conversation is if
  37                 # its new, so report it all
  38                 try:
  39                         mergedConversations = self._conn.session.conversations.get_conversation(self._contactKey)
  40                 except KeyError:
  41                         pass
  42                 else:
  43                         self._report_conversation(mergedConversations)
  44
  45         @gtk_toolbox.log_exception(_moduleLogger)
  46         def Send(self, messageType, text):
  47                 if messageType != telepathy.CHANNEL_TEXT_MESSAGE_TYPE_NORMAL:
  48                         raise telepathy.errors.NotImplemented("Unhandled message type: %r" % messageType)
  49
  50                 self._conn.session.backend.send_sms(self._otherHandle.phoneNumber, text)
  51
  52                 self.Sent(int(time.time()), messageType, text)
  53
  54         @gtk_toolbox.log_exception(_moduleLogger)
  55         def Close(self):
  56                 try:
  57                         # Clear since the user has seen it all and it should start a new conversation
  58                         self._conn.session.conversations.clear_conversation(self._contactKey)
  59
  60                         self._conn.session.conversations.updateSignalHandler.unregister_sink(
  61                                 self._callback
  62                         )
  63                 finally:
  64                         telepathy.server.ChannelTypeText.Close(self)
  65                         self.remove_from_connection()
  66
  67         @property
  68         def _contactKey(self):
  69                 contactKey = self._otherHandle.contactID, self._otherHandle.phoneNumber
  70                 return contactKey
  71
  72         @gobject_utils.async
  73         @gtk_toolbox.log_exception(_moduleLogger)
  74         def _on_conversations_updated(self, conv, conversationIds):
  75                 if self._contactKey not in conversationIds:
  76                         return
  77                 _moduleLogger.info("Incoming messages from %r for existing conversation" % (self._contactKey, ))
  78                 mergedConversations = self._conn.session.conversations.get_conversation(self._contactKey)
  79                 self._report_conversation(mergedConversations)
  80
  81         def _report_conversation(self, mergedConversations):
  82                 newConversations = mergedConversations.conversations
  83                 newConversations = list(newConversations)
  84                 lenMerged = len(newConversations)
  85                 newConversations = self._filter_out_reported(newConversations)
  86                 newConversations = list(newConversations)
  87                 lenUnreported = len(newConversations)
  88                 newConversations = self._filter_out_read(newConversations)
  89                 newConversations = list(newConversations)
  90                 lenUnread = len(newConversations)
  91                 if not newConversations:
  92                         _moduleLogger.info(
  93                                 "New messages for %r have already been read externally" % (self._contactKey, )
  94                         )
  95                         return
  96                 _moduleLogger.debug("%s, %s, %s" % (lenMerged, lenUnreported, lenUnread))
  97                 self._lastMessageTimestamp = newConversations[-1].time
  98
  99                 for newConversation in newConversations:
 100                         for newMessage in newConversations.messages:
 101                                 if newMessage.name != "Me:":
 102                                         formattedMessage = self._format_message(newMessage)
 103                                         self._report_new_message(formattedMessage)
 104
 105         def _filter_out_reported(self, conversations):
 106                 return (
 107                         conversation
 108                         for conversation in conversations
 109                         if self._lastMessageTimestamp < conversation.time
 110                 )
 111
 112         def _filter_out_read(self, conversations):
 113                 return (
 114                         conversation
 115                         for conversation in conversations
 116                         if not conversation.isRead and not conversation.isArchived
 117                 )
 118
 119         def _format_message(self, message):
 120                 return " ".join(part.text.strip() for part in message.body)
 121
 122         def _report_new_message(self, message):
 123                 currentReceivedId = self._nextRecievedId
 124
 125                 timestamp = int(time.time())
 126                 type = telepathy.CHANNEL_TEXT_MESSAGE_TYPE_NORMAL
 127
 128                 _moduleLogger.info("Received message from User %r" % self._otherHandle)
 129                 self.Received(currentReceivedId, timestamp, self._otherHandle, type, 0, message)
 130
 131                 self._nextRecievedId += 1