Merge presence window and chat in one window

1 files changed, 168 insertions, 0 deletions

diff --git a/shell/PresenceView.py b/shell/PresenceView.py
new file mode 100644
index 0000000..364ad15
--- /dev/null
+++ b/shell/PresenceView.py
@@ -0,0 +1,168 @@
+import pygtk
+pygtk.require('2.0')
+import gtk
+import gobject
+import dbus
+
+from sugar.presence.PresenceService import PresenceService
+from sugar.presence.Service import Service
+from sugar.chat.BuddyChat import BuddyChat
+
+from gettext import gettext as _
+
+class PresenceView(gtk.VBox):
+	_MODEL_COL_NICK = 0
+	_MODEL_COL_ICON = 1
+	_MODEL_COL_BUDDY = 2
+	_MODEL_COL_VISIBLE = 3
+		
+	def __init__(self, shell):
+		gtk.VBox.__init__(self, False, 6)
+		
+		self._activity = None
+		self._shell = shell
+
+		self._pservice = PresenceService.get_instance()
+		self._pservice.connect("buddy-appeared", self._on_buddy_appeared_cb)
+		self._pservice.connect("buddy-disappeared", self._on_buddy_disappeared_cb)
+		self._pservice.start()
+		
+		self._pservice.track_service_type(BuddyChat.SERVICE_TYPE)
+		
+		self._setup_ui()
+
+	def _is_buddy_visible(self, buddy):
+		if self._activity:
+			activity_type = self._activity.get_default_type()
+			service = buddy.get_service_of_type(activity_type, self._activity)
+			return service is not None
+		else:
+			return True
+
+	def _update_buddies_visibility(self):
+		for row in self._buddy_store:
+			row[self._MODEL_COL_VISIBLE] = self._is_buddy_visible(row[self._MODEL_COL_BUDDY])
+
+	def set_activity(self, activity):
+		self._activity = activity
+		self._update_buddies_visibility()
+		if activity:
+			if self._activity.get_shared():
+				self._share_button.set_sensitive(False)
+			else:
+				self._share_button.set_sensitive(True)
+		else:
+			self._share_button.set_sensitive(False)
+
+	def _setup_ui(self):
+		self.set_size_request(120, -1)
+
+		label = gtk.Label(_("Who's around:"))
+		label.set_alignment(0.0, 0.5)
+		self.pack_start(label, False)
+		label.show()
+
+		self._buddy_store = gtk.ListStore(gobject.TYPE_STRING,
+					 	 		  		  gtk.gdk.Pixbuf,
+					  			  		  gobject.TYPE_PYOBJECT,
+					  			  		  bool)
+		buddy_list_model = self._buddy_store.filter_new()
+		buddy_list_model.set_visible_column(self._MODEL_COL_VISIBLE)
+
+		sw = gtk.ScrolledWindow()
+		sw.set_shadow_type(gtk.SHADOW_IN)
+		sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
+
+		self._buddy_list_view = gtk.TreeView(buddy_list_model)
+		self._buddy_list_view.set_headers_visible(False)
+		self._buddy_list_view.connect("cursor-changed", self._on_buddyList_buddy_selected)
+		self._buddy_list_view.connect("row-activated", self._on_buddyList_buddy_double_clicked)
+
+		sw.add(self._buddy_list_view)
+		self._buddy_list_view.show()
+
+		renderer = gtk.CellRendererPixbuf()
+		column = gtk.TreeViewColumn("", renderer, pixbuf=self._MODEL_COL_ICON)
+		column.set_resizable(False)
+		column.set_expand(False);
+		self._buddy_list_view.append_column(column)
+
+		renderer = gtk.CellRendererText()
+		column = gtk.TreeViewColumn("", renderer, text=self._MODEL_COL_NICK)
+		column.set_resizable(True)
+		column.set_sizing("GTK_TREE_VIEW_COLUMN_GROW_ONLY");
+		column.set_expand(True);
+		self._buddy_list_view.append_column(column)
+
+		self.pack_start(sw)
+		sw.show()
+
+		button_box = gtk.HButtonBox()
+
+		self._share_button = gtk.Button(_('Share'))
+		self._share_button.connect('clicked', self._share_button_clicked_cb)
+		button_box.pack_start(self._share_button)
+		self._share_button.show()
+
+		self.pack_start(button_box, False)
+		button_box.show()
+	
+	def _share_button_clicked_cb(self, button):
+		self._shell.get_current_activity().publish()
+	
+	def _on_buddyList_buddy_selected(self, view, *args):
+		(model, aniter) = view.get_selection().get_selected()
+		name = model.get(aniter, self._MODEL_COL_NICK)
+
+	def _on_buddyList_buddy_double_clicked(self, view, *args):
+		""" Select the chat for this buddy or group """
+		(model, aniter) = view.get_selection().get_selected()
+		chat = None
+		buddy = view.get_model().get_value(aniter, self._MODEL_COL_BUDDY)
+		if buddy:
+			chat_service = buddy.get_service_of_type(BuddyChat.SERVICE_TYPE)
+			if chat_service:
+				bus = dbus.SessionBus()
+				proxy_obj = bus.get_object('com.redhat.Sugar.Chat', '/com/redhat/Sugar/Chat')
+				chat_shell = dbus.Interface(proxy_obj, 'com.redhat.Sugar.ChatShell')
+				serialized_service = Service.serialize(chat_service)
+				chat_shell.open_chat(serialized_service)
+			else:
+				print 'Could not find buddy chat'
+
+	def __buddy_icon_changed_cb(self, buddy):
+		it = self._get_iter_for_buddy(buddy)
+		self._buddy_store.set(it, self._MODEL_COL_ICON, buddy.get_icon_pixbuf())
+
+	def _on_buddy_appeared_cb(self, pservice, buddy):
+		if buddy.is_owner():
+			# Do not show ourself in the buddy list
+			return
+
+		aniter = self._buddy_store.append(None)
+		self._buddy_store.set(aniter,
+						  self._MODEL_COL_NICK, buddy.get_nick_name(),
+						  self._MODEL_COL_BUDDY, buddy,
+						  self._MODEL_COL_VISIBLE, self._is_buddy_visible(buddy))
+		buddy.connect('icon-changed', self.__buddy_icon_changed_cb)
+		buddy.connect('service-added', self.__buddy_service_added_cb)
+		buddy.connect('service-removed', self.__buddy_service_removed_cb)
+
+	def __buddy_service_added_cb(self, buddy, service):
+		self._update_buddies_visibility()
+
+	def __buddy_service_removed_cb(self, buddy, service):
+		self._update_buddies_visibility()
+		
+	def _on_buddy_disappeared_cb(self, pservice, buddy):
+		aniter = self._get_iter_for_buddy(buddy)
+		if aniter:
+			self._buddy_store.remove(aniter)
+
+	def _get_iter_for_buddy(self, buddy):
+		aniter = self._buddy_store.get_iter_first()
+		while aniter:
+			list_buddy = self._buddy_store.get_value(aniter, self._MODEL_COL_BUDDY)
+			if buddy == list_buddy:
+				return aniter
+			aniter = self._buddy_store.iter_next(aniter)