Write a mesh model and start using it in the mesh view

3 files changed, 114 insertions, 0 deletions

diff --git a/shell/model/Makefile.am b/shell/model/Makefile.am
index 322e732..50fe145 100644
--- a/shell/model/Makefile.am
+++ b/shell/model/Makefile.am
@@ -5,4 +5,5 @@ sugar_PYTHON =			\
 	Friends.py		\
 	Invites.py		\
 	Owner.py		\
+	MeshModel.py		\
 	ShellModel.py
diff --git a/shell/model/MeshModel.py b/shell/model/MeshModel.py
new file mode 100644
index 0000000..2b47baf
--- /dev/null
+++ b/shell/model/MeshModel.py
@@ -0,0 +1,108 @@
+import gobject
+
+import conf
+from sugar.canvas.IconItem import IconColor
+from sugar.presence import PresenceService
+from model.BuddyModel import BuddyModel
+
+class ActivityModel:
+	def __init__(self, activity, service):
+		self._service = service
+		self._activity = activity
+
+	def get_id(self):
+		return self._activity.get_id()
+		
+	def get_icon_name(self):
+		registry = conf.get_activity_registry()
+		info = registry.get_activity_from_type(self._service.get_type())
+
+		return info.get_icon()
+	
+	def get_color(self):
+		return IconColor(self._activity.get_color())
+
+	def get_service(self):
+		return self._service
+
+class MeshModel(gobject.GObject):
+	__gsignals__ = {
+		'activity-added':   (gobject.SIGNAL_RUN_FIRST,
+							 gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
+		'activity-removed': (gobject.SIGNAL_RUN_FIRST,
+							 gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
+		'buddy-added':      (gobject.SIGNAL_RUN_FIRST,
+							 gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
+		'buddy-moved':      (gobject.SIGNAL_RUN_FIRST,
+							 gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT,
+												  gobject.TYPE_PYOBJECT])),
+		'buddy-removed':    (gobject.SIGNAL_RUN_FIRST,
+							 gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT]))
+	}
+
+	def __init__(self):
+		gobject.GObject.__init__(self)
+
+		self._activities = {}
+		self._buddies = {}
+
+		self._pservice = PresenceService.get_instance()
+		self._pservice.connect("service-appeared",
+							   self._service_appeared_cb)
+		self._pservice.connect('activity-disappeared',
+							   self._activity_disappeared_cb)
+		self._pservice.connect("buddy-appeared",
+							   self._buddy_appeared_cb)
+		self._pservice.connect("buddy-disappeared",
+							   self._buddy_disappeared_cb)
+
+		for service in self._pservice.get_services():
+			self._check_service(service)
+
+	def get_activities(self):
+		return self._activities
+
+	def get_buddies(self):
+		return self._buddies
+
+	def _buddy_activity_changed_cb(self, buddy, cur_activity):
+		buddy_model = self._buddies[buddy.get_name()]
+		activity_model = self._activities[cur_activity.get_id()]
+		self.emit('buddy-moved', buddy_model, activity_model)
+
+	def _buddy_appeared_cb(self, pservice, buddy):
+		model = BuddyModel(buddy=buddy)
+		model.connect('current-activity-changed',
+					  self._buddy_activity_changed_cb)
+		self._buddies[model.get_name()] = model
+		self.emit('buddy-added', model)
+
+	def _buddy_disappeared_cb(self, pservice, buddy):
+		self.emit('buddy-removed', buddy)
+		del self._buddies[buddy.get_name()]
+
+	def _service_appeared_cb(self, pservice, service):
+		self._check_service(service)
+
+	def _check_service(self, service):
+		registry = conf.get_activity_registry()
+		if registry.get_activity_from_type(service.get_type()) != None:
+			activity_id = service.get_activity_id()
+			if not self.has_activity(activity_id):
+				activity = self._pservice.get_activity(activity_id)
+				if activity != None:
+					self.add_activity(activity, service)
+
+	def has_activity(self, activity_id):
+		return self._activities.has_key(activity_id)
+
+	def add_activity(self, activity, service):
+		model = ActivityModel(activity, service)
+		self._activities[model.get_id()] = model
+		self.emit('activity-added', model)
+
+	def _activity_disappeared_cb(self, pservice, activity):
+		if self._activities.has_key(activity.get_id()):
+			activity_model = self._activities[activity.get_id()]
+			self.emit('activity-removed', activity_model)
+			del self._activities[activity.get_id()]
diff --git a/shell/model/ShellModel.py b/shell/model/ShellModel.py
index 5872b02..9b6d598 100644
--- a/shell/model/ShellModel.py
+++ b/shell/model/ShellModel.py
@@ -1,5 +1,6 @@
 from sugar.presence import PresenceService
 from model.Friends import Friends
+from model.MeshModel import MeshModel
 from model.Owner import ShellOwner
 
 class ShellModel:
@@ -12,6 +13,10 @@ class ShellModel:
 		self._owner = ShellOwner()
 		self._owner.announce()
 		self._friends = Friends()
+		self._mesh = MeshModel()
+
+	def get_mesh(self):
+		return self._mesh
 
 	def get_friends(self):
 		return self._friends