import time
-import weakref
+import datetime
+import logging
import telepathy
-import handle
+import util.go_utils as gobject_utils
+import util.coroutines as coroutines
+import gtk_toolbox
-class ButterflyTextChannel(
- telepathy.server.ChannelTypeText,
- telepathy.server.ChannelInterfaceGroup,
- telepathy.server.ChannelInterfaceChatState
- ):
+_moduleLogger = logging.getLogger("channel.text")
- def __init__(self, connection, conversation):
- self._recv_id = 0
- self._conversation = conversation
- self._conn_ref = weakref.ref(connection)
- telepathy.server.ChannelTypeText.__init__(self, connection, None)
- telepathy.server.ChannelInterfaceGroup.__init__(self)
- telepathy.server.ChannelInterfaceChatState.__init__(self)
+class TextChannel(telepathy.server.ChannelTypeText):
+ """
+ Look into implementing ChannelInterfaceMessages for rich text formatting
+ """
- self.GroupFlagsChanged(telepathy.CHANNEL_GROUP_FLAG_CAN_ADD, 0)
- self.__add_initial_participants()
+ def __init__(self, connection, manager, props, contactHandle):
+ self._manager = manager
+ self._props = props
- def SetChatState(self, state):
- if state == telepathy.CHANNEL_CHAT_STATE_COMPOSING:
- self._conversation.send_typing_notification()
- h = handle.create_handle(self._conn_ref(), 'self')
- self.ChatStateChanged(h, state)
+ try:
+ # HACK Older python-telepathy way
+ telepathy.server.ChannelTypeText.__init__(self, connection, contactHandle)
+ except TypeError:
+ # HACK Newer python-telepathy way
+ telepathy.server.ChannelTypeText.__init__(self, connection, manager, props)
+ self._nextRecievedId = 0
+ self._lastMessageTimestamp = datetime.datetime(1, 1, 1)
- def Send(self, messageType, text):
- if messageType == telepathy.CHANNEL_TEXT_MESSAGE_TYPE_NORMAL:
- self._conversation.send_text_message(text)
- elif messageType == telepathy.CHANNEL_TEXT_MESSAGE_TYPE_ACTION and text == u"nudge":
- self._conversation.send_nudge()
+ self._otherHandle = contactHandle
+
+ self._callback = coroutines.func_sink(
+ coroutines.expand_positional(
+ self._on_conversations_updated
+ )
+ )
+ self._conn.session.conversations.updateSignalHandler.register_sink(
+ self._callback
+ )
+
+ # The only reason there should be anything in the conversation is if
+ # its new, so report it all
+ try:
+ mergedConversations = self._conn.session.conversations.get_conversation(self._contactKey)
+ except KeyError:
+ pass
else:
- raise telepathy.NotImplemented("Unhandled message type")
+ self._report_conversation(mergedConversations)
+
+ @gtk_toolbox.log_exception(_moduleLogger)
+ def Send(self, messageType, text):
+ if messageType != telepathy.CHANNEL_TEXT_MESSAGE_TYPE_NORMAL:
+ raise telepathy.errors.NotImplemented("Unhandled message type: %r" % messageType)
+
+ self._conn.session.backend.send_sms(self._otherHandle.phoneNumber, text)
+ self._conn.session.stateMachine.reset_timers()
+
self.Sent(int(time.time()), messageType, text)
+ @gtk_toolbox.log_exception(_moduleLogger)
def Close(self):
- self._conversation.leave()
+ self.close()
+
+ def close(self):
+ self._conn.session.conversations.updateSignalHandler.unregister_sink(
+ self._callback
+ )
+ self._callback = None
+
telepathy.server.ChannelTypeText.Close(self)
+ if self._manager.channel_exists(self._props):
+ # HACK Older python-telepathy requires doing this manually
+ self._manager.remove_channel(self)
self.remove_from_connection()
+
+ @property
+ def _contactKey(self):
+ contactKey = self._otherHandle.contactID, self._otherHandle.phoneNumber
+ return contactKey
+
+ @gobject_utils.async
+ @gtk_toolbox.log_exception(_moduleLogger)
+ def _on_conversations_updated(self, conv, conversationIds):
+ if self._contactKey not in conversationIds:
+ return
+ _moduleLogger.info("Incoming messages from %r for existing conversation" % (self._contactKey, ))
+ mergedConversations = self._conn.session.conversations.get_conversation(self._contactKey)
+ self._report_conversation(mergedConversations)
+
+ def _report_conversation(self, mergedConversations):
+ newConversations = mergedConversations.conversations
+ newConversations = self._filter_out_reported(newConversations)
+ newConversations = self._filter_out_read(newConversations)
+ newConversations = list(newConversations)
+ if not newConversations:
+ _moduleLogger.info(
+ "New messages for %r have already been read externally" % (self._contactKey, )
+ )
+ return
+ self._lastMessageTimestamp = newConversations[-1].time
+
+ messages = [
+ newMessage
+ for newConversation in newConversations
+ for newMessage in newConversation.messages
+ if newMessage.whoFrom != "Me:"
+ ]
+ if not newConversations:
+ _moduleLogger.info(
+ "All incoming messages were really outbound messages for %r" % (self._contactKey, )
+ )
+ return
+
+ for newMessage in messages:
+ formattedMessage = self._format_message(newMessage)
+ self._report_new_message(formattedMessage)
+
+ def _filter_out_reported(self, conversations):
+ return (
+ conversation
+ for conversation in conversations
+ if self._lastMessageTimestamp < conversation.time
+ )
+
+ def _filter_out_read(self, conversations):
+ return (
+ conversation
+ for conversation in conversations
+ if not conversation.isRead and not conversation.isArchived
+ )
+
+ def _format_message(self, message):
+ return " ".join(part.text.strip() for part in message.body)
+
+ def _report_new_message(self, message):
+ currentReceivedId = self._nextRecievedId
+
+ timestamp = int(time.time())
+ type = telepathy.CHANNEL_TEXT_MESSAGE_TYPE_NORMAL
+
+ _moduleLogger.info("Received message from User %r" % self._otherHandle)
+ self.Received(currentReceivedId, timestamp, self._otherHandle, type, 0, message)
+
+ self._nextRecievedId += 1