Get back presence list to work

1 files changed, 84 insertions, 0 deletions

diff --git a/shell/view/frame/FriendsBox.py b/shell/view/frame/FriendsBox.py
new file mode 100644
index 0000000..8d48a0e
--- /dev/null
+++ b/shell/view/frame/FriendsBox.py
@@ -0,0 +1,84 @@
+import hippo
+
+from sugar.graphics.canvasicon import CanvasIcon
+from sugar.graphics.iconcolor import IconColor
+from sugar.graphics import style
+from sugar.presence import PresenceService
+from view.BuddyIcon import BuddyIcon
+from model.BuddyModel import BuddyModel
+from view.frame.MenuStrategy import MenuStrategy
+
+class FriendsBox(hippo.CanvasBox):
+	def __init__(self, shell, menu_shell):
+		hippo.CanvasBox.__init__(self)
+		self._shell = shell
+		self._menu_shell = menu_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)
+
+		shell.connect('activity-changed', self.__activity_changed_cb)
+
+	def add(self, buddy):
+		model = BuddyModel(buddy=buddy)
+		icon = BuddyIcon(self._shell, self._menu_shell, model)
+		style.apply_stylesheet(icon, 'frame.BuddyIcon')
+		icon.set_menu_strategy(MenuStrategy())
+		self.append(icon)
+
+		self._buddies[buddy.get_name()] = icon
+
+	def remove(self, buddy):
+		self.remove(self._buddies[buddy.get_name()])
+
+	def clear(self):
+		for item in self.get_children():
+			self.remove(item)
+		self._buddies = {}
+
+	def __activity_appeared_cb(self, pservice, activity_ps):
+		activity = self._shell.get_current_activity()
+		if activity and activity_ps.get_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)
+
+			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 __activity_changed_cb(self, group, activity):
+		if activity:
+			ps = self._pservice.get_activity(activity.get_id())
+			self._set_activity_ps(ps)
+		else:
+			self._set_activity_ps(None)
+
+	def __buddy_joined_cb(self, activity, buddy):
+		self.add(buddy)
+
+	def __buddy_left_cb(self, activity, buddy):
+		self.remove(buddy)