Accept raw X windows in the shell

3 files changed, 43 insertions, 35 deletions

diff --git a/shell/model/Makefile.am b/shell/model/Makefile.am
index d8a843e..9e24dc4 100644
--- a/shell/model/Makefile.am
+++ b/shell/model/Makefile.am
@@ -11,4 +11,5 @@ sugar_PYTHON =			\
 	MeshModel.py		\
 	ShellModel.py		\
 	homeactivity.py		\
-	homemodel.py
+	homemodel.py		\
+	homerawwindow.py
diff --git a/shell/model/homeactivity.py b/shell/model/homeactivity.py
index f823f59..8772888 100644
--- a/shell/model/homeactivity.py
+++ b/shell/model/homeactivity.py
@@ -24,9 +24,6 @@ from sugar.graphics.xocolor import XoColor
 from sugar.presence import PresenceService
 from sugar import profile
 
-_ACTIVITY_SERVICE_NAME = "org.laptop.Activity"
-_ACTIVITY_SERVICE_PATH = "/org/laptop/Activity"
-
 class HomeActivity(gobject.GObject):
     __gsignals__ = {
         'launch-timeout':          (gobject.SIGNAL_RUN_FIRST,
@@ -39,29 +36,32 @@ class HomeActivity(gobject.GObject):
         self._window = None
         self._xid = None
         self._service = None
-        self._id = activity_id
-        self._type = bundle.get_service_name()
-        self._icon_name = bundle.get_icon()
+        self._activity_id = activity_id
+        self._bundle = bundle
 
         self._launch_time = time.time()
         self._launched = False
-        self._launch_timeout_id = gobject.timeout_add(20000, self._launch_timeout_cb)
+        self._launch_timeout_id = gobject.timeout_add(
+                                    20000, self._launch_timeout_cb)
 
-        logging.debug("Activity %s (%s) launching..." % (self._id, self._type))
+        logging.debug("Activity %s (%s) launching..." %
+                      (self._activity_id, self.get_type))
 
     def __del__(self):
         gobject.source_remove(self._launch_timeout_id)
         self._launch_timeout_id = 0
 
     def _launch_timeout_cb(self, user_data=None):
-        logging.debug("Activity %s (%s) launch timed out" % (self._id, self._type))
+        logging.debug("Activity %s (%s) launch timed out" %
+                      (self._activity_id, self.get_type))
         self._launch_timeout_id = 0
         self.emit('launch-timeout')
         return False
 
     def set_window(self, window):
         """An activity is 'launched' once we get its window."""
-        logging.debug("Activity %s (%s) finished launching" % (self._id, self._type))
+        logging.debug("Activity %s (%s) finished launching" %
+                      (self._activity_id, self.get_type))
         self._launched = True
         gobject.source_remove(self._launch_timeout_id)
         self._launch_timeout_id = 0
@@ -74,17 +74,8 @@ class HomeActivity(gobject.GObject):
         self._window = window
         self._xid = window.get_xid()
 
-        bus = dbus.SessionBus()
-        self._service = bus.get_object(_ACTIVITY_SERVICE_NAME + '%d' % self._xid,
-                                       _ACTIVITY_SERVICE_PATH + "/%s" % self._xid)
-
-        # verify id and type details
-        act_id = self._service.get_id()
-        if act_id != self._id:
-            raise RuntimeError("Activity's real ID (%s) didn't match expected (%s)." % (act_id, self._id))
-        act_type = self._service.get_service_name()
-        if act_type != self._type:
-            raise RuntimeError("Activity's real type (%s) didn't match expected (%s)." % (act_type, self._type))
+    def set_service(self, service):
+        self._service = service
 
     def get_service(self):
         return self._service
@@ -95,17 +86,18 @@ class HomeActivity(gobject.GObject):
         return self._window.get_name()
 
     def get_icon_name(self):
-        return self._icon_name
+        return self._bundle.get_icon()
     
     def get_icon_color(self):
-        activity = PresenceService.get_instance().get_activity(self._id)
+        pservice = PresenceService.get_instance()
+        activity = pservice.get_activity(self._activity_id)
         if activity != None:
             return XoColor(activity.get_color())
         else:
             return profile.get_color()
         
-    def get_id(self):
-        return self._id
+    def get_activity_id(self):
+        return self._activity_id
 
     def get_xid(self):
         if not self._launched:
@@ -118,7 +110,7 @@ class HomeActivity(gobject.GObject):
         return self._window
 
     def get_type(self):
-        return self._type
+        return self._bundle.get_service_name()
 
     def get_shared(self):
         if not self._launched:
diff --git a/shell/model/homemodel.py b/shell/model/homemodel.py
index 598562b..9aa47e2 100644
--- a/shell/model/homemodel.py
+++ b/shell/model/homemodel.py
@@ -21,10 +21,11 @@ import wnck
 import dbus
 
 from model.homeactivity import HomeActivity
+from model.homerawwindow import HomeRawWindow
 from sugar.activity import bundleregistry
 
-_ACTIVITY_SERVICE_NAME = "org.laptop.Activity"
-_ACTIVITY_SERVICE_PATH = "/org/laptop/Activity"
+_SERVICE_NAME = "org.laptop.Activity"
+_SERVICE_PATH = "/org/laptop/Activity"
 
 class HomeModel(gobject.GObject):
 
@@ -116,15 +117,28 @@ class HomeModel(gobject.GObject):
             logging.error('Model for window %d does not exist.' % xid)
 
         self.emit('active-activity-changed', self._current_activity)
-        
+
+    def _add_window(self, window):
+        home_window = HomeRawWindow(window)
+        self._activities[home_window.get_activity_id()] = home_window
+        self.emit('activity-added', home_window)
+
     def _add_activity(self, window):
         bus = dbus.SessionBus()
         xid = window.get_xid()
-        act_service = bus.get_object(_ACTIVITY_SERVICE_NAME + '%d' % xid,
-                                     _ACTIVITY_SERVICE_PATH + "/%s" % xid)
-        act_id = act_service.get_id()
+        try:
+            service = bus.get_object(_SERVICE_NAME + '%d' % xid,
+                                     _SERVICE_PATH + "/%s" % xid)
+        except dbus.DBusException:
+            service = None
+
+        if not service:
+            self._add_window(window)
+            return
 
         activity = None
+        act_id = service.get_id()
+        act_type = service.get_service_name()
         if self._activities.has_key(act_id):
             activity = self._activities[act_id]
         else:
@@ -138,6 +152,7 @@ class HomeModel(gobject.GObject):
             activity = HomeActivity(bundle, act_id)
             self._activities[act_id] = activity
 
+        activity.set_service(service)
         activity.set_window(window)
         self.emit('activity-added', activity)
 
@@ -146,7 +161,7 @@ class HomeModel(gobject.GObject):
             self._current_activity = None
 
         self.emit('activity-removed', activity)
-        act_id = activity.get_id()
+        act_id = activity.get_activity_id()
         del self._activities[act_id]
         
     def _remove_activity(self, xid):
@@ -157,7 +172,7 @@ class HomeModel(gobject.GObject):
             logging.error('Model for window %d does not exist.' % xid)
 
     def _activity_launch_timeout_cb(self, activity):
-        act_id = activity.get_id()
+        act_id = activity.get_activity_id()
         if not act_id in self._activities.keys():
             return
         self._internal_remove_activity(activity)