Let Kiu change activities randomly

2 files changed, 80 insertions, 15 deletions

diff --git a/sugar/simulator.py b/sugar/simulator.py
index 96fa498..b46b9c3 100644
--- a/sugar/simulator.py
+++ b/sugar/simulator.py
@@ -2,6 +2,8 @@ import os
 
 import gtk
 import gobject
+import base64
+import dbus
 
 from sugar.session.TestSession import TestSession
 from sugar.presence import PresenceService
@@ -17,12 +19,18 @@ class _SimulatedActivity:
 	def get_id(self):
 		return self._id
 
-class _ShellOwner(object):
-	def __init__(self, nick, color):
+class _SimulatedShellOwner(object):
+	def __init__(self, nick, color, icon_file=None):
 		self._pservice = PresenceService.get_instance()
 		self._color = color
 		self._nick = nick
 
+		self._icon = None
+		if icon_file:
+			fd = open(icon_file, "r")
+			self._icon = fd.read()
+			fd.close()
+
 	def announce(self):
 		props = { 'color':  self._color.to_string() }
 		self._service = self._pservice.register_service(self._nick,
@@ -32,11 +40,16 @@ class _ShellOwner(object):
 		self._stream.register_reader_handler(self._handle_invite, "invite")
 
 	def _handle_buddy_icon_request(self):
+		if self._icon:
+			return base64.b64encode(self._icon)
 		return ''
 
 	def _handle_invite(self, issuer, bundle_id, activity_id):
 		return ''
 
+	def set_current_activity(self, activity_id):
+		self._service.set_published_value('curact', dbus.String(activity_id))
+
 class _Timeline:
 	def __init__(self, time_factor):
 		self._time_factor = time_factor
@@ -50,22 +63,27 @@ class _Timeline:
 		return False
 
 class ShareActivityAction:
-	def __init__(self, title, activity_type):
+	def __init__(self, title, activity_type, callback=None):
 		self._title = title
 		self._type = activity_type
+		self._callback = callback
 
 	def execute(self):
 		activity = _SimulatedActivity()
 		properties = { 'title' : self._title }
 		
 		pservice = PresenceService.get_instance()
-		pservice.share_activity(activity, self._type, properties)		
+		act_service = pservice.share_activity(activity, self._type, properties)		
+		if self._callback is not None:
+			self._callback(activity, act_service)
 
 class Bot:
 	def __init__(self, nick, color):
 		self._nick = nick
 		self._color = color
 		self._timeline = _Timeline(0.01)
+		self._owner = None
+		self._icon_file = None
 
 		os.environ['SUGAR_NICK_NAME'] = nick
 		os.environ['SUGAR_COLOR'] = color.to_string()
@@ -76,10 +94,10 @@ class Bot:
 
 		PresenceService.start()
 
-		owner = _ShellOwner(self._nick, self._color)
-		owner.announce()
+		self._owner = _SimulatedShellOwner(self._nick, self._color, self._icon_file)
+		self._owner.announce()
 
-		pservice = PresenceService.get_instance()
+		self._pservice = PresenceService.get_instance()
 
 		gtk.main()
 
diff --git a/tests/simulator/kiu.py b/tests/simulator/kiu.py
index af4166d..ac07453 100755
--- a/tests/simulator/kiu.py
+++ b/tests/simulator/kiu.py
@@ -3,15 +3,62 @@
 from sugar.simulator import Bot
 from sugar.simulator import ShareActivityAction
 from sugar.canvas.IconColor import IconColor
+import os, random, gobject
 
-bot = Bot('kiu', IconColor('#5E4505,#0F8A0F'))
+class KiuBot(Bot):
+	def __init__(self):
+		Bot.__init__(self, 'kiu', IconColor('#5E4505,#0F8A0F'))
+		self._olpc_channel_service = None
+		self._sugar_channel_service = None
+		self._activity_switch_timeout = None
 
-action = ShareActivityAction('OLPC channel',
-							 '_GroupChatActivity_Sugar_redhat_com._udp')
-bot.add_action(action, 10)
+		action = ShareActivityAction('OLPC channel',
+							 '_GroupChatActivity_Sugar_redhat_com._udp',
+							 self.__share_olpc_channel_cb)
+		self.add_action(action, 10)
 
-action = ShareActivityAction('Sugar channel',
-							 '_GroupChatActivity_Sugar_redhat_com._udp')
-bot.add_action(action, 20)
+		action = ShareActivityAction('Sugar channel',
+							 '_GroupChatActivity_Sugar_redhat_com._udp',
+							 self.__share_sugar_channel_cb)
+		self.add_action(action, 20)
 
-bot.start()
+		self._icon_file = os.path.abspath("kiu.jpg")
+
+	def __activity_switch_cb(self):
+		self._activity_switch_timeout = None
+		which = random.randint(1, 2)
+		if which == 1:
+			actid = self._olpc_channel_activity.get_id()
+		elif which == 2:
+			actid = self._sugar_channel_activity.get_id()
+		else:
+			raise RuntimeError("WTF? unexpected value")
+		print "KIU: now setting current activity to %s" % actid
+		self._owner.set_current_activity(actid)
+		self._schedule_activity_switch_timeout()
+		return False
+
+	def _schedule_activity_switch_timeout(self):
+		if self._activity_switch_timeout:
+			return
+		interval = random.randint(1000, 20000)
+		self._activity_switch_timeout = gobject.timeout_add(interval,
+				self.__activity_switch_cb)
+
+	def __share_olpc_channel_cb(self, sim_activity, service):
+		self._olpc_channel_service = service
+		self._olpc_channel_activity = sim_activity
+		self._schedule_activity_switch_timeout()
+
+	def __share_sugar_channel_cb(self, sim_activity, service):
+		self._sugar_channel_service = service
+		self._sugar_channel_activity = sim_activity
+		self._schedule_activity_switch_timeout()
+
+def main():
+	bot = KiuBot()
+	bot.start()
+
+
+if __name__ == "__main__":
+	main()