sugar-toolkit-gtk3/shell/view/Shell.py

# Copyright (C) 2006, 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 logging

import gtk
import gobject
import wnck
import dbus

import view.stylesheet
from sugar.graphics import style
from view.home.HomeWindow import HomeWindow
from sugar.presence import PresenceService
from view.ActivityHost import ActivityHost
from sugar.activity import ActivityFactory
from sugar.activity import Activity
from view.frame.Frame import Frame
from view.dconmanager import DCONManager
from _sugar import KeyGrabber
import sugar

class Shell(gobject.GObject):
    __gsignals__ = {
        'activity-opened':  (gobject.SIGNAL_RUN_FIRST,
                             gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
        'activity-changed': (gobject.SIGNAL_RUN_FIRST,
                             gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
        'activity-closed':  (gobject.SIGNAL_RUN_FIRST,
                             gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT]))
    }

    def __init__(self, model):
        gobject.GObject.__init__(self)

        self._model = model
        self._hosts = {}
        self._screen = wnck.screen_get_default()
        self._current_host = None

        style.load_stylesheet(view.stylesheet)

        self._dcon_manager = DCONManager()

        self._key_grabber = KeyGrabber()
        self._key_grabber.connect('key-pressed',
                                  self.__global_key_pressed_cb)
        self._key_grabber.connect('key-released',
                                  self.__global_key_released_cb)
        self._key_grabber.grab('F1')
        self._key_grabber.grab('F2')
        self._key_grabber.grab('F3')
        self._key_grabber.grab('F4')
        self._key_grabber.grab('F5')
        self._key_grabber.grab('F6')
        self._key_grabber.grab('F7')
        self._key_grabber.grab('F8')
        self._key_grabber.grab('0xDC') # Camera key
        self._key_grabber.grab('0xE0') # Overlay key
        self._key_grabber.grab('0x93') # Frame key

        # For non-OLPC machines
        self._key_grabber.grab('<shft><alt>F9')
        self._key_grabber.grab('<shft><alt>F10')

        self._home_window = HomeWindow(self)
        self._home_window.show()
        self.set_zoom_level(sugar.ZOOM_HOME)

        self._screen.connect('window-opened', self.__window_opened_cb)
        self._screen.connect('window-closed', self.__window_closed_cb)
        self._screen.connect('active-window-changed',
                             self.__active_window_changed_cb)

        self._frame = Frame(self)
        self._frame.show_and_hide(3)

    def _open_terminal_cb(self):
        self.start_activity('org.sugar.Terminal')
        return False

    def __global_key_pressed_cb(self, grabber, key):
        if key == 'F1':
            self.set_zoom_level(sugar.ZOOM_MESH)
        elif key == 'F2':
            self.set_zoom_level(sugar.ZOOM_FRIENDS)
        elif key == 'F3':
            self.set_zoom_level(sugar.ZOOM_HOME)
        elif key == 'F4':
            self.set_zoom_level(sugar.ZOOM_ACTIVITY)
        elif key == 'F5':
            self._dcon_manager.decrease_brightness()
        elif key == 'F6':
            self._dcon_manager.increase_brightness()
        elif key == 'F7':
            self._dcon_manager.set_mode(DCONManager.COLOR_MODE)
        elif key == 'F8':
            self._dcon_manager.set_mode(DCONManager.BLACK_AND_WHITE_MODE)
        elif key == '<shft><alt>F9':
            self._frame.notify_key_press()
        elif key == '<shft><alt>F10':
            self.toggle_chat_visibility()
        elif key == '0xDC': # Camera key
            pass
        elif key == '0xE0': # Overlay key
            self.toggle_chat_visibility()
        elif key == '0x93': # Frame key
            self._frame.notify_key_press()

    def __global_key_released_cb(self, grabber, key):
        if key == '<shft><alt>F9':
            self._frame.notify_key_release()
        elif key == '0x93':
            self._frame.notify_key_release()

    def __window_opened_cb(self, screen, window):
        logging.debug('Shell.__window_opened_cb')
        if window.get_window_type() == wnck.WINDOW_NORMAL:
            try:
                activity_host = ActivityHost(self.get_model(), window)
            except dbus.DBusException:
                logging.debug('Shell.__window_opened_cb: opened unknown window ' +
                              window.get_name() + ' with xid ' + 
                              str(window.get_xid()))
                return
                        
            self._hosts[activity_host.get_xid()] = activity_host
            self.emit('activity-opened', activity_host)

    def __active_window_changed_cb(self, screen):
        logging.debug('Shell.__active_window_changed_cb')
        window = screen.get_active_window()
        if not window or window.get_window_type() != wnck.WINDOW_NORMAL:
            return
        if not self._hosts.has_key(window.get_xid()):
            return

        activity_host = self._hosts[window.get_xid()]
        current = self._model.get_current_activity()
        if activity_host.get_id() == current:
            return

        self._set_current_activity(activity_host)

    def __window_closed_cb(self, screen, window):
        logging.debug('Shell.__window_closed_cb')
        if window.get_window_type() != wnck.WINDOW_NORMAL:
            return

        if not self._hosts.has_key(window.get_xid()):
            return

        host = self._hosts[window.get_xid()]
        host.destroy()

        self.emit('activity-closed', host)
        del self._hosts[window.get_xid()]

        if len(self._hosts) == 0:
            self._set_current_activity(None)

    def _set_current_activity(self, host):
        if host:
            self._model.set_current_activity(host.get_id())
        else:
            self._model.set_current_activity(None)

        if self._current_host:
            self._current_host.set_active(False)

        self._current_host = host

        if self._current_host:
            self._current_host.set_active(True)

        self.emit('activity-changed', host)

    def get_model(self):
        return self._model

    def join_activity(self, bundle_id, activity_id):
        pservice = PresenceService.get_instance()

        activity = self._get_activity(activity_id)
        if activity:
            activity.present()
        else:
            activity_ps = pservice.get_activity(activity_id)

            if activity_ps:
                # Get the service name for this activity, if
                # we have a bundle on the system capable of handling
                # this activity type
                breg = self._model.get_bundle_registry()
                bundle = breg.find_by_default_type(bundle_id)
                if bundle:
                    serv_name = bundle.get_service_name()
                    try:
                        activity = ActivityFactory.create(serv_name)
                    except DBusException, e:
                        logging.error("Couldn't launch activity %s:\n%s" % (serv_name, e))
                    else:
                        logging.debug("Joining activity type %s id %s" % (serv_name, activity_id))
                        activity.join(activity_ps.object_path())
                else:
                    logging.error("Couldn't find activity for type %s" % bundle_id)
            else:
                logging.error('Cannot start activity.')

    def start_activity(self, activity_type):
        logging.debug('Shell.start_activity')
        activity = ActivityFactory.create(activity_type)
        activity.start()
        return activity

    def set_zoom_level(self, level):
        if level == sugar.ZOOM_ACTIVITY:
            self._screen.toggle_showing_desktop(False)
        else:
            self._screen.toggle_showing_desktop(True)
            self._home_window.set_zoom_level(level)

    def get_current_activity(self):
        activity_id = self._model.get_current_activity()
        if activity_id:
            return self._get_activity(activity_id)
        else:
            return None

    def _get_activity(self, activity_id):
        for host in self._hosts.values():
            if host.get_id() == activity_id:
                return host
        return None

    def toggle_chat_visibility(self):
        act = self.get_current_activity()
        if not act:
            return
        is_visible = self._frame.is_visible()
        if act.is_chat_visible():
            frame_was_visible = act.chat_hide()
            if not frame_was_visible:
                self._frame.do_slide_out()
        else:
            if not is_visible:
                self._frame.do_slide_in()
            act.chat_show(is_visible)