1 files changed, 139 insertions, 0 deletions
diff --git a/shell/view/frame/friendstray.py b/shell/view/frame/friendstray.py
new file mode 100644
index 0000000..b5c35d6
--- /dev/null
+++ b/shell/view/frame/friendstray.py
@@ -0,0 +1,139 @@
+# Copyright (C) 2006-2007 Red Hat, Inc.
+#
+# 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 hippo
+
+from sugar.presence import presenceservice
+from sugar.graphics.tray import VTray, TrayIcon
+
+from view.BuddyMenu import BuddyMenu
+from view.frame.frameinvoker import FrameWidgetInvoker
+from model.BuddyModel import BuddyModel
+
+class FriendIcon(TrayIcon):
+    def __init__(self, shell, buddy):
+        TrayIcon.__init__(self, icon_name='computer-xo',
+                          xo_color=buddy.get_color())
+
+        palette = BuddyMenu(shell, buddy)
+        self.set_palette(palette)
+        palette.set_group_id('frame')
+        palette.props.invoker = FrameWidgetInvoker(self)
+
+class FriendsTray(VTray):
+    def __init__(self, shell):
+        VTray.__init__(self)
+
+        self._shell = shell
+        self._activity_ps = None
+        self._joined_hid = -1
+        self._left_hid = -1
+        self._buddies = {}
+
+        self._pservice = presenceservice.get_instance()
+        self._pservice.connect('activity-appeared',
+                               self.__activity_appeared_cb)
+
+        # Add initial activities the PS knows about
+        self._pservice.get_activities_async(reply_handler=self._get_activities_cb)
+
+        home_model = shell.get_model().get_home()
+        home_model.connect('active-activity-changed',
+                           self._active_activity_changed_cb)
+
+    def _get_activities_cb(self, list):
+        for activity in list:
+            self.__activity_appeared_cb(self._pservice, activity)
+
+    def add_buddy(self, buddy):
+        if self._buddies.has_key(buddy.props.key):
+            return
+
+        model = BuddyModel(buddy=buddy)
+ 
+        icon = FriendIcon(self._shell, model)
+        self.add_item(icon)
+        icon.show()
+
+        self._buddies[buddy.props.key] = icon
+
+    def remove_buddy(self, buddy):
+        if not self._buddies.has_key(buddy.props.key):
+            return
+
+        self.remove_item(self._buddies[buddy.props.key])
+
+    def clear(self):
+        for item in self.get_children():
+            self.remove_item(item)
+        self._buddies = {}
+
+    def __activity_appeared_cb(self, pservice, activity_ps):
+        activity = self._shell.get_current_activity()
+        if activity and activity_ps.props.id == activity.get_id():
+            self._set_activity_ps(activity_ps)
+
+    def _set_activity_ps(self, activity_ps):
+        if self._activity_ps == activity_ps:
+            return
+
+        if self._joined_hid > 0:
+            self._activity_ps.disconnect(self._joined_hid)
+            self._joined_hid = -1
+        if self._left_hid > 0:
+            self._activity_ps.disconnect(self._left_hid)
+            self._left_hid = -1
+
+        self._activity_ps = activity_ps
+
+        self.clear()
+
+        if activity_ps != None:
+            for buddy in activity_ps.get_joined_buddies():
+                self.add_buddy(buddy)
+
+            self._joined_hid = activity_ps.connect(
+                            'buddy-joined', self.__buddy_joined_cb)
+            self._left_hid = activity_ps.connect(
+                            'buddy-left', self.__buddy_left_cb)
+
+    def _active_activity_changed_cb(self, home_model, home_activity):
+        if not home_activity:
+            self._set_activity_ps(None)
+            return
+
+        activity_id = home_activity.get_activity_id()
+        if not activity_id:
+            self._set_activity_ps(None)
+            return
+        
+        # HACK to suppress warning in logs when activity isn't found
+        # (if it's locally launched and not shared yet)
+        activity = None
+        for act in self._pservice.get_activities():
+            if activity_id == act.props.id:
+                activity = act
+                break
+        if activity:
+            self._set_activity_ps(activity)
+        else:
+            self._set_activity_ps(None)
+
+    def __buddy_joined_cb(self, activity, buddy):
+        self.add_buddy(buddy)
+
+    def __buddy_left_cb(self, activity, buddy):
+        self.remove_buddy(buddy)