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('', '') title = title.replace('', '') # 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 = '' + cgi.escape(title) + '' 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