diff options
Diffstat (limited to 'src/model/MeshModel.py')
| -rw-r--r-- | src/model/MeshModel.py | 235 |
1 files changed, 235 insertions, 0 deletions
diff --git a/src/model/MeshModel.py b/src/model/MeshModel.py new file mode 100644 index 0000000..da5b3c2 --- /dev/null +++ b/src/model/MeshModel.py @@ -0,0 +1,235 @@ +# Copyright (C) 2006-2007 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 gobject + +from sugar.graphics.xocolor import XoColor +from sugar.presence import presenceservice +from sugar import activity + +from model.BuddyModel import BuddyModel +from model.accesspointmodel import AccessPointModel +from hardware import hardwaremanager +from hardware import nmclient + +class ActivityModel: + def __init__(self, activity, bundle): + self.activity = activity + self.bundle = bundle + + def get_id(self): + return self.activity.props.id + + def get_icon_name(self): + return self.bundle.icon + + def get_color(self): + return XoColor(self.activity.props.color) + + def get_bundle_id(self): + return self.bundle.bundle_id + +class MeshModel(gobject.GObject): + __gsignals__ = { + 'activity-added': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), + 'activity-removed': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), + 'buddy-added': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), + 'buddy-moved': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, + ([gobject.TYPE_PYOBJECT, + gobject.TYPE_PYOBJECT])), + 'buddy-removed': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), + 'access-point-added': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), + 'access-point-removed': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), + 'mesh-added': (gobject.SIGNAL_RUN_FIRST, + gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), + 'mesh-removed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])) + } + + def __init__(self): + gobject.GObject.__init__(self) + + self._activities = {} + self._buddies = {} + self._access_points = {} + self._mesh = None + + self._pservice = presenceservice.get_instance() + self._pservice.connect("activity-appeared", + self._activity_appeared_cb) + self._pservice.connect('activity-disappeared', + self._activity_disappeared_cb) + self._pservice.connect("buddy-appeared", + self._buddy_appeared_cb) + self._pservice.connect("buddy-disappeared", + self._buddy_disappeared_cb) + + # Add any buddies the PS knows about already + self._pservice.get_buddies_async(reply_handler=self._get_buddies_cb) + + self._pservice.get_activities_async(reply_handler=self._get_activities_cb) + + network_manager = hardwaremanager.get_network_manager() + if network_manager: + for nm_device in network_manager.get_devices(): + self._add_network_device(nm_device) + network_manager.connect('device-added', + self._nm_device_added_cb) + network_manager.connect('device-removed', + self._nm_device_removed_cb) + + def _get_buddies_cb(self, list): + for buddy in list: + self._buddy_appeared_cb(self._pservice, buddy) + + def _get_activities_cb(self, list): + for activity in list: + self._check_activity(activity) + + def _nm_device_added_cb(self, manager, nm_device): + self._add_network_device(nm_device) + + def _nm_device_removed_cb(self, manager, nm_device): + self._remove_network_device(nm_device) + + def _nm_network_appeared_cb(self, nm_device, nm_network): + self._add_access_point(nm_device, nm_network) + + def _nm_network_disappeared_cb(self, nm_device, nm_network): + if self._access_points.has_key(nm_network.get_op()): + ap = self._access_points[nm_network.get_op()] + self._remove_access_point(ap) + + def _add_network_device(self, nm_device): + dtype = nm_device.get_type() + if dtype == nmclient.DEVICE_TYPE_802_11_WIRELESS: + for nm_network in nm_device.get_networks(): + self._add_access_point(nm_device, nm_network) + + nm_device.connect('network-appeared', + self._nm_network_appeared_cb) + nm_device.connect('network-disappeared', + self._nm_network_disappeared_cb) + elif dtype == nmclient.DEVICE_TYPE_802_11_MESH_OLPC: + self._mesh = nm_device + self.emit('mesh-added', self._mesh) + + def _remove_network_device(self, nm_device): + if nm_device == self._mesh: + self._mesh = None + self.emit('mesh-removed') + elif nm_device.get_type() == nmclient.DEVICE_TYPE_802_11_WIRELESS: + aplist = self._access_points.values() + for ap in aplist: + if ap.get_nm_device() == nm_device: + self._remove_access_point(ap) + + def _add_access_point(self, nm_device, nm_network): + model = AccessPointModel(nm_device, nm_network) + self._access_points[model.get_id()] = model + self.emit('access-point-added', model) + + def _remove_access_point(self, ap): + if not self._access_points.has_key(ap.get_id()): + return + self.emit('access-point-removed', ap) + del self._access_points[ap.get_id()] + + def get_mesh(self): + return self._mesh + + def get_access_points(self): + return self._access_points.values() + + def get_activities(self): + return self._activities.values() + + def get_buddies(self): + return self._buddies.values() + + def _buddy_activity_changed_cb(self, model, cur_activity): + if not self._buddies.has_key(model.get_key()): + return + if cur_activity and self._activities.has_key(cur_activity.props.id): + activity_model = self._activities[cur_activity.props.id] + self.emit('buddy-moved', model, activity_model) + else: + self.emit('buddy-moved', model, None) + + def _buddy_appeared_cb(self, pservice, buddy): + if self._buddies.has_key(buddy.props.key): + return + + model = BuddyModel(buddy=buddy) + model.connect('current-activity-changed', + self._buddy_activity_changed_cb) + self._buddies[buddy.props.key] = model + self.emit('buddy-added', model) + + cur_activity = buddy.props.current_activity + if cur_activity: + self._buddy_activity_changed_cb(model, cur_activity) + + def _buddy_disappeared_cb(self, pservice, buddy): + if not self._buddies.has_key(buddy.props.key): + return + self.emit('buddy-removed', self._buddies[buddy.props.key]) + del self._buddies[buddy.props.key] + + def _activity_appeared_cb(self, pservice, activity): + self._check_activity(activity) + + def _check_activity(self, presence_activity): + registry = activity.get_registry() + bundle = registry.get_activity(presence_activity.props.type) + if not bundle: + return + if self.has_activity(presence_activity.props.id): + return + self.add_activity(bundle, presence_activity) + + def has_activity(self, activity_id): + return self._activities.has_key(activity_id) + + def get_activity(self, activity_id): + if self.has_activity(activity_id): + return self._activities[activity_id] + else: + return None + + def add_activity(self, bundle, activity): + model = ActivityModel(activity, bundle) + self._activities[model.get_id()] = model + self.emit('activity-added', model) + + for buddy in self._pservice.get_buddies(): + cur_activity = buddy.props.current_activity + key = buddy.props.key + if cur_activity == activity and self._buddies.has_key(key): + buddy_model = self._buddies[key] + self.emit('buddy-moved', buddy_model, model) + + def _activity_disappeared_cb(self, pservice, activity): + if self._activities.has_key(activity.props.id): + activity_model = self._activities[activity.props.id] + self.emit('activity-removed', activity_model) + del self._activities[activity.props.id] |