1 files changed, 82 insertions, 0 deletions
diff --git a/src/jarabe/session.py b/src/jarabe/session.py
new file mode 100644
index 0000000..c1f8d0f
--- /dev/null
+++ b/src/jarabe/session.py
@@ -0,0 +1,82 @@
+# Copyright (C) 2008, Red Hat, Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+import gtk
+import dbus
+import os
+import signal
+
+from sugar import session
+from sugar import env
+
+from jarabe.hardware import hardwaremanager
+
+_session_manager = None
+
+class SessionManager(session.SessionManager):
+    MODE_LOGOUT = 0
+    MODE_SHUTDOWN = 1
+    MODE_REBOOT = 2
+
+    def __init__(self):
+        session.SessionManager.__init__(self)
+        self._logout_mode = None
+
+    def logout(self):
+        self._logout_mode = self.MODE_LOGOUT
+        self.initiate_shutdown()
+
+    def shutdown(self):
+        self._logout_mode = self.MODE_SHUTDOWN
+        self.initiate_shutdown()
+
+    def reboot(self):
+        self._logout_mode = self.MODE_REBOOT
+        self.initiate_shutdown()
+
+    def shutdown_completed(self):
+        session.SessionManager.shutdown_completed(self)
+
+        hw_manager = hardwaremanager.get_manager()
+        hw_manager.shutdown()
+
+        bus = dbus.SystemBus()
+        proxy = bus.get_object('org.freedesktop.Hal', 
+                               '/org/freedesktop/Hal/devices/computer')
+        pm = dbus.Interface(proxy, \
+                            'org.freedesktop.Hal.Device.SystemPowerManagement') 
+
+        if env.is_emulator():
+            self._close_emulator()
+        else:
+            if self._logout_mode == self.MODE_SHUTDOWN:
+                pm.Shutdown()
+            elif self._logout_mode == self.MODE_REBOOT:
+                pm.Reboot()
+
+            gtk.main_quit()
+
+    def _close_emulator(self):
+        if os.environ.has_key('SUGAR_EMULATOR_PID'):
+            pid = int(os.environ['SUGAR_EMULATOR_PID'])
+            os.kill(pid, signal.SIGTERM)
+
+def get_session_manager():
+    global _session_manager
+
+    if _session_manager == None:
+        _session_manager = SessionManager()
+    return _session_manager