1 files changed, 54 insertions, 0 deletions
diff --git a/shell/model/Invites.py b/shell/model/Invites.py
new file mode 100644
index 0000000..24f9913
--- /dev/null
+++ b/shell/model/Invites.py
@@ -0,0 +1,54 @@
+import gobject
+
+import conf
+from sugar.presence import PresenceService
+from sugar.canvas.IconColor import IconColor
+
+class Invite:
+	def __init__(self, issuer, bundle_id, activity_id):
+		self._issuer = issuer
+		self._activity_id = activity_id
+		self._bundle_id = bundle_id
+
+	def get_icon(self):
+		reg = conf.get_activity_registry()
+		return reg.get_activity(self._bundle_id).get_icon()
+
+	def get_color(self):
+		pservice = PresenceService.get_instance()
+		buddy = pservice.get_buddy_by_name(self._issuer)
+		if buddy != None:
+			return IconColor(buddy.get_color())
+		else:
+			return IconColor('white')
+
+	def get_activity_id(self):
+		return self._activity_id
+
+	def get_bundle_id(self):
+		return self._bundle_id
+
+class Invites(gobject.GObject):
+	__gsignals__ = {
+		'invite-added':   (gobject.SIGNAL_RUN_FIRST,
+						   gobject.TYPE_NONE, ([object])),
+		'invite-removed': (gobject.SIGNAL_RUN_FIRST,
+						   gobject.TYPE_NONE, ([object])),
+	}
+
+	def __init__(self):
+		gobject.GObject.__init__(self)
+
+		self._list = []
+
+	def add_invite(self, issuer, bundle_id, activity_id):
+		invite = Invite(issuer, bundle_id, activity_id)
+		self._list.append(invite)
+		self.emit('invite-added', invite)
+
+	def remove_invite(self, invite):
+		self._list.remove(invite)
+		self.emit('invite-removed', invite)
+
+	def __iter__(self):
+		return self._list.__iter__()