X-Git-Url: http://git.maemo.org/git/?p=theonering;a=blobdiff_plain;f=src%2Fchannel%2Ftext.py;h=5212cd74d7e260ee1410d3f5ad4b54c32d0bb98d;hp=e6cc6aff62d3645c0568de4f77768d9e3bb1bfd2;hb=17ed78514215ad8e1358e329fb2e77ad749fe777;hpb=ab4da214c520d763bbbddeb828c8e7c838374fe3 diff --git a/src/channel/text.py b/src/channel/text.py index e6cc6af..5212cd7 100644 --- a/src/channel/text.py +++ b/src/channel/text.py @@ -1,45 +1,184 @@ import time -import weakref +import datetime +import logging import telepathy -import handle +import tp +import util.coroutines as coroutines +import util.misc as misc_utils +import util.go_utils as gobject_utils +import gvoice -class ButterflyTextChannel( - telepathy.server.ChannelTypeText, - telepathy.server.ChannelInterfaceGroup, - telepathy.server.ChannelInterfaceChatState - ): +_moduleLogger = logging.getLogger(__name__) - 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(tp.ChannelTypeText): - self.GroupFlagsChanged(telepathy.CHANNEL_GROUP_FLAG_CAN_ADD, 0) - self.__add_initial_participants() + OLDEST_MESSAGE_WINDOW = datetime.timedelta(days=1) - 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) + def __init__(self, connection, manager, props, contactHandle): + self.__manager = manager + self.__props = props - 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() + tp.ChannelTypeText.__init__(self, connection, manager, props) + self.__nextRecievedId = 0 + + self.__otherHandle = contactHandle + + self.__callback = coroutines.func_sink( + coroutines.expand_positional( + self._on_conversations_updated + ) + ) + self._conn.session.voicemails.updateSignalHandler.register_sink( + self.__callback + ) + self._conn.session.texts.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.voicemails.get_conversation(self._contactKey) + except KeyError: + _moduleLogger.debug("No voicemails in the conversation yet for %r" % (self._contactKey, )) else: - raise telepathy.NotImplemented("Unhandled message type") + self._report_conversation(mergedConversations) + try: + mergedConversations = self._conn.session.texts.get_conversation(self._contactKey) + except KeyError: + _moduleLogger.debug("No texts conversation yet for %r" % (self._contactKey, )) + else: + self._report_conversation(mergedConversations) + + @misc_utils.log_exception(_moduleLogger) + def Send(self, messageType, text): + le = gobject_utils.AsyncLinearExecution(self._conn.session.pool, self._send) + le.start(messageType, text) + + @misc_utils.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) + + _moduleLogger.info("Sending message to %r" % (self.__otherHandle, )) + try: + result = yield ( + self._conn.session.backend.send_sms, + ([self.__otherHandle.phoneNumber], text), + {}, + ) + except Exception: + _moduleLogger.exception("Oh no, what happened?") + return + + self._conn.session.textsStateMachine.reset_timers() + self.Sent(int(time.time()), messageType, text) + @misc_utils.log_exception(_moduleLogger) + def _on_send_sms_failed(self, error): + _moduleLogger.error(error) + + @misc_utils.log_exception(_moduleLogger) def Close(self): - self._conversation.leave() - telepathy.server.ChannelTypeText.Close(self) + self.close() + + def close(self): + _moduleLogger.debug("Closing text") + self._conn.session.voicemails.updateSignalHandler.unregister_sink( + self.__callback + ) + self._conn.session.texts.updateSignalHandler.unregister_sink( + self.__callback + ) + self.__callback = None + + tp.ChannelTypeText.Close(self) self.remove_from_connection() + + @property + def _contactKey(self): + contactKey = self.__otherHandle.phoneNumber + return contactKey + + @misc_utils.log_exception(_moduleLogger) + def _on_conversations_updated(self, conv, conversationIds): + if self._contactKey not in conversationIds: + return + _moduleLogger.debug("Incoming messages from %r for existing conversation" % (self._contactKey, )) + mergedConversations = conv.get_conversation(self._contactKey) + self._report_conversation(mergedConversations) + + def _report_conversation(self, mergedConversations): + newConversations = mergedConversations.conversations + if not newConversations: + _moduleLogger.info( + "No messages ended up existing for %r" % (self._contactKey, ) + ) + return + + # Can't filter out messages in a texting conversation that came in + # before the last one sent because that creates a race condition of two + # people sending at about the same time, which happens quite a bit + postUpdateLen = len(newConversations) + newConversations = gvoice.conversations.filter_out_self(newConversations) + newConversations = list(newConversations) + postSelfLen = len(newConversations) + if postSelfLen < postUpdateLen: + _moduleLogger.info("Dropped %s messages due to being from self" % (postUpdateLen - postSelfLen)) + if not newConversations: + _moduleLogger.debug( + "New messages for %r are from yourself" % (self._contactKey, ) + ) + return + + newConversations = gvoice.conversations.filter_out_read(newConversations) + newConversations = list(newConversations) + postReadLen = len(newConversations) + if postReadLen < postSelfLen: + _moduleLogger.info("Dropped %s messages due to already being read" % (postSelfLen- postReadLen)) + if not newConversations: + _moduleLogger.debug( + "New messages for %r have already been read externally" % (self._contactKey, ) + ) + return + + messages = [ + (newMessage, newConversation) + for newConversation in newConversations + for newMessage in newConversation.messages + if not gvoice.conversations.is_message_from_self(newMessage) + ] + if not messages: + _moduleLogger.debug( + "How did this happen for %r?" % (self._contactKey, ) + ) + return + + now = datetime.datetime.now() + for newMessage, conv in messages: + if self.OLDEST_MESSAGE_WINDOW < (now - conv.time): + _moduleLogger.warning("Why are we reporting a message that is so old?") + _moduleLogger.warning("\t%r %r (%r) with %r messages" % (conv.number, conv.id, conv.time, len(conv.messages))) + formattedMessage = self._format_message(newMessage) + self._report_new_message(formattedMessage, conv.time) + + for conv in mergedConversations.conversations: + conv.isRead = True + + def _format_message(self, message): + return " ".join(part.text.strip() for part in message.body) + + def _report_new_message(self, message, convTime): + currentReceivedId = self.__nextRecievedId + timestamp = time.mktime(convTime.timetuple()) + 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