From f6a44319d709eb824a6e47fe3037a3d228938880 Mon Sep 17 00:00:00 2001
From: Andrés Ambrois <andresambrois@gmail.com>
Date: Mon, 20 Apr 2009 08:06:30 +0000
Subject: Initial take at backend for overlay chat.

---
(limited to 'src/sugar')

diff --git a/src/sugar/presence/chat.py b/src/sugar/presence/chat.py
new file mode 100644
index 0000000..8695022
--- /dev/null
+++ b/src/sugar/presence/chat.py
@@ -0,0 +1,125 @@
+# -*- coding: UTF-8 -*-
+# Copyright (C) 2009 Andrés Ambrois
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+import gobject
+import telepathy
+from sugar.presence import presenceservice
+
+
+class BaseChat(gobject.GObject):
+    ''' Base class for Chat models '''
+
+    channels = {}
+
+    def __init__(self, activity_ps, keys=[]):
+        ''' activity_ps -- sugar.presence.activity instance where we get the
+                telepathy channels from.
+
+            keys -- If we need to act on received messages, give us a list of the
+                public keys of these buddies.
+        '''
+        gobject.GObject.__init__(self)
+        self.keys = dict.fromkeys(keys)
+        self._ps = presenceservice.get_instance()
+        self.channel = None
+
+        try:
+            self._find_channel(activity_ps)
+        except KeyError:
+            activity_ps.connect('new-channel',
+                lambda obj, path: self._find_channel(activity_ps))
+
+    def _find_channel(self, activity_ps):
+        id = activity_ps.props.id
+        try:
+            self.channel = BaseChat.channels[id]
+        except KeyError:
+            # Get the activity text chan from the presence service
+            bus_name, conn_path, channel_paths = activity_ps.get_channels()
+            for channel_path in channel_paths:
+                channel = telepathy.client.Channel(bus_name, channel_path)
+                htype, handle = channel.GetHandle()
+                if htype == telepathy.HANDLE_TYPE_ROOM:
+                    if channel.GetChannelType() == telepathy.CHANNEL_TYPE_TEXT:
+                        BaseChat.channels[id] = channel
+                        self.channel = channel
+
+                        self._received_sh = self.channel[
+                            telepathy.CHANNEL_TYPE_TEXT].connect_to_signal(
+                            'Received', self.message_received_cb)
+
+                        for id, timestamp, sender, type, flags, text in \
+                                self.channel[telepathy.CHANNEL_TYPE_TEXT
+                                ].ListPendingMessages(False):
+                            self.message_received_cb(id, timestamp, sender,
+                            type, flags, text)
+            if self.channel == None:
+                raise KeyError
+
+    def message_received_cb(self, id, timestamp, sender, type, flags, text):
+        if self.keys:
+            self._find_handles()
+
+    def _find_handles(self):
+        group = self.channel[telepathy.CHANNEL_INTERFACE_GROUP]
+        conn_name, conn_path = self._ps.get_preferred_connection()
+        if group.GetGroupFlags() & \
+        telepathy.CHANNEL_GROUP_FLAG_CHANNEL_SPECIFIC_HANDLES:
+            owners = self.channel.GetAll(telepathy.CHANNEL_INTERFACE_GROUP,
+                dbus_interface='org.freedesktop.DBus.Properties').get(
+                'HandleOwners')
+            for handle, owner in owners.items():
+                buddy = self._ps.get_buddy_by_telepathy_handle(
+                    conn_name, conn_path, owner)
+                if buddy is None:
+                    continue
+                if buddy.props.key in self.keys.keys(): # keys, keys, keys!!
+                    self.keys[buddy.props.key] = handle
+        else:
+            buddy = self._ps.get_buddy_by_telepathy_handle(
+                conn_name, conn_path, sender)
+            if buddy.props.key in self.keys.keys():
+                self.keys[buddy.props.key] = sender
+
+    def disconnect_signals(self, object=None):
+        self._received_sh.remove()
+
+class ChatCatcher(BaseChat):
+    __gsignals__ = {
+        'chat-received':            (gobject.SIGNAL_RUN_FIRST,
+                                    gobject.TYPE_NONE,
+                                    ([gobject.TYPE_STRING])),
+    }
+
+    def __init__(self, activity_ps, key):
+        BaseChat.__init__(self, activity_ps, [key])
+
+    def message_received_cb(self, id, timestamp, sender, type, flags, text):
+        BaseChat.message_received_cb(self, id, timestamp, sender, type, flags, text)
+        if sender in self.keys.values():
+            self.emit('chat-received', text)
+            self.channel[
+                telepathy.CHANNEL_TYPE_TEXT].AcknowledgePendingMessages([id])
+
+class ChatPitcher(BaseChat):
+    def __init__(self, activity_ps):
+        BaseChat.__init__(self, activity_ps)
+
+    def send(self, text):
+        if self.channel is not None:
+            self.channel[telepathy.CHANNEL_TYPE_TEXT].Send(
+                telepathy.CHANNEL_TEXT_MESSAGE_TYPE_NORMAL, text)
--
cgit v0.9.1