diff options
Diffstat (limited to 'shell/StartPage.py')
-rw-r--r-- | shell/StartPage.py | 295 |
1 files changed, 295 insertions, 0 deletions
diff --git a/shell/StartPage.py b/shell/StartPage.py new file mode 100644 index 0000000..4ceeec3 --- /dev/null +++ b/shell/StartPage.py @@ -0,0 +1,295 @@ +import pygtk +pygtk.require('2.0') +import gtk +import pango +import dbus +import cgi +import xml.sax.saxutils +import gobject + +from google import google +from sugar.presence.PresenceService import PresenceService + +_BROWSER_ACTIVITY_TYPE = "_web_olpc._udp" + +_COLUMN_TITLE = 0 +_COLUMN_ADDRESS = 1 +_COLUMN_SUBTITLE = 2 +_COLUMN_SERVICE = 3 + +class SearchHelper(object): + def __init__(self, activity_uid): + self.search_uid = activity_uid + self.found = False + +class SearchModel(gtk.ListStore): + def __init__(self, activities_model, search_text): + gtk.ListStore.__init__(self, gobject.TYPE_STRING, gobject.TYPE_STRING, + gobject.TYPE_STRING, gobject.TYPE_PYOBJECT) + + for row in activities_model: + title = row[_COLUMN_TITLE] + address = row[_COLUMN_ADDRESS] + if title.find(search_text) >= 0 or address.find(search_text) >= 0: + self.append([ title, address, row[_COLUMN_SUBTITLE], row[_COLUMN_SERVICE] ]) + + google.LICENSE_KEY = '1As9KaJQFHIJ1L0W5EZPl6vBOFvh/Vaf' + data = google.doGoogleSearch(search_text) + + for result in data.results: + title = result.title + + # FIXME what tags should we actually strip? + title = title.replace('<b>', '') + title = title.replace('</b>', '') + + # FIXME I'm sure there is a better way to + # unescape these. + title = title.replace('"', '"') + title = title.replace('&', '&') + + self.append([ title, result.URL, None, None ]) + +class ActivitiesModel(gtk.ListStore): + def __init__(self): + gtk.ListStore.__init__(self, gobject.TYPE_STRING, gobject.TYPE_STRING, + gobject.TYPE_STRING, gobject.TYPE_PYOBJECT) + + def _filter_dupe_activities(self, model, path, it, user_data): + """Search the list of list rows for an existing service that + has the activity ID we're looking for.""" + helper = user_data + (service, ) = model.get(it, _COLUMN_SERVICE) + if not service: + return False + if service.get_activity_uid() == helper.search_uid: + helper.found = True + return True + return False + + def add_activity(self, buddy, service): + # Web Activity check + activity_uid = service.get_activity_uid() + if activity_uid is None: + return + # Don't show dupes + helper = SearchHelper(activity_uid) + self.foreach(self._filter_dupe_activities, helper) + if helper.found == True: + return + + # Only accept browser activities for now + if service.get_type() == _BROWSER_ACTIVITY_TYPE: + escaped_title = service.get_one_property('Title') + escaped_uri = service.get_one_property('URI') + if escaped_title and escaped_uri: + title = xml.sax.saxutils.unescape(escaped_title) + address = xml.sax.saxutils.unescape(escaped_uri) + subtitle = 'Shared by %s' % buddy.get_nick_name() + self.append([ title, address, subtitle, service ]) + +class ActivitiesView(gtk.TreeView): + def __init__(self, model): + gtk.TreeView.__init__(self, model) + + self._owner = None + + self.set_headers_visible(False) + + theme = gtk.icon_theme_get_default() + size = 48 + self._web_pixbuf = theme.load_icon('emblem-web', size, 0) + self._share_pixbuf = theme.load_icon('emblem-people', size, 0) + + column = gtk.TreeViewColumn('') + self.append_column(column) + + cell = gtk.CellRendererPixbuf() + column.pack_start(cell, False) + column.set_cell_data_func(cell, self._icon_cell_data_func) + + cell = gtk.CellRendererText() + column.pack_start(cell) + column.set_cell_data_func(cell, self._cell_data_func) + + self.connect('row-activated', self._row_activated_cb) + + def _icon_cell_data_func(self, column, cell, model, it): + if model.get_value(it, _COLUMN_SERVICE) == None: + cell.set_property('pixbuf', self._web_pixbuf) + else: + cell.set_property('pixbuf', self._share_pixbuf) + + def _cell_data_func(self, column, cell, model, it): + title = model.get_value(it, _COLUMN_TITLE) + subtitle = model.get_value(it, _COLUMN_SUBTITLE) + if subtitle is None: + subtitle = model.get_value(it, _COLUMN_ADDRESS) + + markup = '<big><b>' + cgi.escape(title) + '</b></big>' + markup += '\n' + cgi.escape(subtitle) + + cell.set_property('markup', markup) + cell.set_property('ellipsize', pango.ELLIPSIZE_END) + + def set_owner(self, owner): + self._owner = owner + + def _row_activated_cb(self, treeview, path, column): + bus = dbus.SessionBus() + proxy_obj = bus.get_object('com.redhat.Sugar.Browser', '/com/redhat/Sugar/Browser') + browser_shell = dbus.Interface(proxy_obj, 'com.redhat.Sugar.BrowserShell') + + model = self.get_model() + address = model.get_value(model.get_iter(path), _COLUMN_ADDRESS) + service = model.get_value(model.get_iter(path), _COLUMN_SERVICE) + + print 'Activated row %s' % address + + if service is None: + browser_shell.open_browser(address) + else: + if not self._owner: + raise RuntimeError("We don't have an owner yet!") + serialized_service = service.serialize(self._owner) + browser_shell.open_browser(address, serialized_service) + +class StartPage(gtk.HBox): + def __init__(self, ac_signal_object): + gtk.HBox.__init__(self) + + self._ac_signal_object = ac_signal_object + self._ac_signal_object.connect("local-activity-started", + self._on_local_activity_started_cb) + self._ac_signal_object.connect("local-activity-ended", + self._on_local_activity_ended_cb) + + self._pservice = PresenceService.get_instance() + self._pservice.connect("activity-announced", self._on_activity_announced_cb) + self._pservice.connect("new-service-adv", self._on_new_service_adv_cb) + self._pservice.connect("buddy-appeared", self._on_buddy_appeared_cb) + self._pservice.connect("buddy-disappeared", self._on_buddy_disappeared_cb) + self._pservice.start() + self._pservice.track_service_type(_BROWSER_ACTIVITY_TYPE) + if self._pservice.get_owner(): + self._on_buddy_appeared_cb(self._pservice, self._pservice.get_owner()) + + vbox = gtk.VBox() + + search_box = gtk.HBox(False, 6) + search_box.set_border_width(24) + + self._search_entry = gtk.Entry() + self._search_entry.connect('activate', self._search_entry_activate_cb) + search_box.pack_start(self._search_entry) + self._search_entry.show() + + search_button = gtk.Button("Search") + search_button.connect('clicked', self._search_button_clicked_cb) + search_box.pack_start(search_button, False) + search_button.show() + + vbox.pack_start(search_box, False, True) + search_box.show() + + exp_space = gtk.Label('') + vbox.pack_start(exp_space) + exp_space.show() + + self.pack_start(vbox) + vbox.show() + + vbox = gtk.VBox() + + self._search_close_box = gtk.HBox() + + self._search_close_label = gtk.Label() + self._search_close_label.set_alignment(0.0, 0.5) + self._search_close_box.pack_start(self._search_close_label) + self._search_close_label.show() + + close_image = gtk.Image() + close_image.set_from_stock (gtk.STOCK_CLOSE, gtk.ICON_SIZE_MENU) + close_image.show() + + search_close_button = gtk.Button() + rcstyle = gtk.RcStyle(); + rcstyle.xthickness = rcstyle.ythickness = 0; + search_close_button.modify_style (rcstyle); + search_close_button.add(close_image) + search_close_button.set_relief(gtk.RELIEF_NONE) + search_close_button.set_focus_on_click(False) + search_close_button.connect("clicked", self.__search_close_button_clicked_cb) + + self._search_close_box.pack_start(search_close_button, False) + search_close_button.show() + + vbox.pack_start(self._search_close_box, False) + + sw = gtk.ScrolledWindow() + sw.set_size_request(320, -1) + sw.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC) + + self._activities_model = ActivitiesModel() + + owner = self._pservice.get_owner() + self._activities = ActivitiesView(self._activities_model) + sw.add(self._activities) + self._activities.show() + + vbox.pack_start(sw) + sw.show() + + self.pack_start(vbox) + vbox.show() + + def __search_close_button_clicked_cb(self, button): + self._search(None) + + def _on_local_activity_started_cb(self, helper, activity_container, activity_id): + print "new local activity %s" % activity_id + + def _on_local_activity_ended_cb(self, helper, activity_container, activity_id): + print "local activity %s disappeared" % activity_id + + def _on_new_service_adv_cb(self, pservice, activity_id, short_stype): + if activity_id: + self._pservice.track_service_type(short_stype) + + def _on_buddy_appeared_cb(self, pservice, buddy): + if buddy.is_owner(): + self._activities.set_owner(buddy) + + def _on_buddy_disappeared_cb(self, pservice, buddy): + if buddy.is_owner(): + self._activities.set_owner(None) + + def _on_activity_announced_cb(self, pservice, service, buddy): + print "Found new activity with type %s" % service.get_full_type() + self._activities_model.add_activity(buddy, service) + if self._activities.get_model() != self._activities_model: + self._search(self._last_search) + + def _search_entry_activate_cb(self, entry): + self._search() + self._search_entry.set_text('') + + def _search_button_clicked_cb(self, button): + self._search() + self._search_entry.set_text('') + + def _search(self, text = None): + if text == None: + text = self._search_entry.get_text() + + if text == None or len(text) == 0: + self._activities.set_model(self._activities_model) + self._search_close_box.hide() + else: + search_model = SearchModel(self._activities_model, text) + self._activities.set_model(search_model) + + self._search_close_label.set_text('Search for %s' % (text)) + self._search_close_box.show() + + self._last_search = text |