sugar-toolkit-gtk3/shell/view/home/MeshBox.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 random

import hippo
import gobject

from sugar.graphics.spreadbox import SpreadBox
from sugar.graphics.snowflakebox import SnowflakeBox
from sugar.graphics.canvasicon import CanvasIcon
from hardware import hardwaremanager
from view.BuddyIcon import BuddyIcon

class AccessPointView(CanvasIcon):
    def __init__(self, model):
        CanvasIcon.__init__(self, tooltip=model.get_name())
        self._model = model

        self.connect('activated', self._activate_cb)
        self._update_icon()

    def _activate_cb(self, icon):
        network_manager = hardwaremanager.get_network_manager()

        device = self._model.get_nm_device()
        network = self._model.get_nm_network()

        network_manager.set_active_device(device, network)

    def _update_icon(self):
        strength = self._model.props.strength
        if strength < 21:
            self.props.icon_name = 'theme:stock-net-wireless-00'
        elif strength < 41:
            self.props.icon_name = 'theme:stock-net-wireless-21-40'
        elif strength < 61:
            self.props.icon_name = 'theme:stock-net-wireless-41-60'
        elif strength < 81:
            self.props.icon_name = 'theme:stock-net-wireless-61-80'
        else:
            self.props.icon_name = 'theme:stock-net-wireless-81-100'

class ActivityView(SnowflakeBox):
    def __init__(self, shell, menu_shell, model):
        SnowflakeBox.__init__(self)

        self._shell = shell
        self._model = model
        self._icons = {}

        icon = CanvasIcon(icon_name=model.get_icon_name(),
                          xo_color=model.get_color(), box_width=80)
        icon.connect('activated', self._clicked_cb)
        self.append(icon, hippo.PACK_FIXED)
        self.set_root(icon)

    def has_buddy_icon(self, name):
        return self._icons.has_key(name)

    def add_buddy_icon(self, name, icon):
        self._icons[name] = icon
        self.append(icon, hippo.PACK_FIXED)

    def remove_buddy_icon(self, name):
        icon = self._icons[name]
        self.remove(icon)
        del self._icons[name]

    def _clicked_cb(self, item):
        bundle_id = self._model.get_service().get_type()
        self._shell.join_activity(bundle_id, self._model.get_id())

class MeshBox(SpreadBox):
    def __init__(self, shell, menu_shell):
        SpreadBox.__init__(self, background_color=0xe2e2e2ff)

        self._shell = shell
        self._menu_shell = menu_shell
        self._model = shell.get_model().get_mesh()
        self._buddies = {}
        self._activities = {}
        self._access_points = {}
        self._buddy_to_activity = {}

        for buddy_model in self._model.get_buddies():
            self._add_alone_buddy(buddy_model)

        self._model.connect('buddy-added', self._buddy_added_cb)
        self._model.connect('buddy-removed', self._buddy_removed_cb)
        self._model.connect('buddy-moved', self._buddy_moved_cb)

        for activity_model in self._model.get_activities():
            self._add_activity(activity_model)

        self._model.connect('activity-added', self._activity_added_cb)
        self._model.connect('activity-removed', self._activity_removed_cb)

        for ap_model in self._model.get_access_points():
            self._add_access_point(ap_model)

        self._model.connect('access-point-added',
                            self._access_point_added_cb)
        self._model.connect('access-point-removed',
                            self._access_point_removed_cb)

    def _buddy_added_cb(self, model, buddy_model):
        self._add_alone_buddy(buddy_model)

    def _buddy_removed_cb(self, model, buddy_model):
        self._remove_buddy(buddy_model) 

    def _buddy_moved_cb(self, model, buddy_model, activity_model):
        self._move_buddy(buddy_model, activity_model)

    def _activity_added_cb(self, model, activity_model):
        self._add_activity(activity_model)

    def _activity_removed_cb(self, model, activity_model):
        self._remove_activity(activity_model) 

    def _access_point_added_cb(self, model, ap_model):
        self._add_access_point(ap_model)

    def _access_point_removed_cb(self, model, ap_model):
        self._add_access_point(ap_model) 

    def _add_alone_buddy(self, buddy_model):
        icon = BuddyIcon(self._shell, self._menu_shell, buddy_model)
        self.add_item(icon)

        self._buddies[buddy_model.get_name()] = icon

    def _remove_alone_buddy(self, buddy_model):
        icon = self._buddies[buddy_model.get_name()]
        self.remove_item(icon)
        del self._buddies[buddy_model.get_name()]

    def _remove_buddy(self, buddy_model):
        name = buddy_model.get_name()
        if self._buddies.has_key(name):
            self._remove_alone_buddy(buddy_model)
        else:
            for activity in self._activities.values():
                if activity.has_buddy_icon(name):
                    activity.remove_buddy_icon(name)

    def _move_buddy(self, buddy_model, activity_model):
        name = buddy_model.get_name()

        self._remove_buddy(buddy_model)

        if activity_model == None:
            self._add_alone_buddy(buddy_model)
        else:
            activity = self._activities[activity_model.get_id()]

            icon = BuddyIcon(self._shell, self._menu_shell, buddy_model)
            activity.add_buddy_icon(buddy_model.get_name(), icon)

    def _add_activity(self, activity_model):
        icon = ActivityView(self._shell, self._menu_shell, activity_model)
        self.add_item(icon)

        self._activities[activity_model.get_id()] = icon

    def _remove_activity(self, activity_model):
        icon = self._activities[activity_model.get_id()]
        self.remove_item(icon)
        del self._activities[activity_model.get_id()]

    def _add_access_point(self, ap_model):
        icon = AccessPointView(ap_model)
        self.add_item(icon)

        self._access_points[ap_model.get_id()] = icon

    def _remove_access_point(self, ap_model):
        icon = self._access_points[ap_model.get_id()]
        self.remove_item(icon)
        del self._access_points[ap_model.get_id()]