class WebActivity(activity.Activity): def __init__(self, handle): activity.Activity.__init__(self, handle) _logger.debug('Starting the web activity') # figure out if we're an SSB self.is_ssb = ssb.get_is_ssb(self) self._browser = Browser() _set_accept_languages() _seed_xs_cookie() _set_dbus_globals(self.get_bundle_id()) # don't pick up the sugar theme - use the native mozilla one instead cls = components.classes['@mozilla.org/preferences-service;1'] pref_service = cls.getService(components.interfaces.nsIPrefService) branch = pref_service.getBranch("mozilla.widget.") branch.setBoolPref("disable-native-theme", True) toolbox = activity.ActivityToolbox(self) self._edit_toolbar = EditToolbar(self._browser) toolbox.add_toolbar(_('Edit'), self._edit_toolbar) self._edit_toolbar.show() self._web_toolbar = WebToolbar(self) toolbox.add_toolbar(_('Browse'), self._web_toolbar) self._web_toolbar.show() self._tray = HTray() self.set_tray(self._tray, gtk.POS_BOTTOM) self._tray.show() self._view_toolbar = ViewToolbar(self) toolbox.add_toolbar(_('View'), self._view_toolbar) self._view_toolbar.show() # the bookmarklet bar doesn't show up if empty self._bm_toolbar = None self.set_toolbox(toolbox) toolbox.show() self.set_canvas(self._browser) self._browser.show() self._browser.history.connect('session-link-changed', self._session_history_changed_cb) self._web_toolbar.connect('add-link', self._link_add_button_cb) self._browser.connect("notify::title", self._title_changed_cb) self._bm_store = bookmarklets.get_store() self._bm_store.connect('add_bookmarklet', self._add_bookmarklet_cb) self._bm_store.connect('overwrite_bookmarklet', self._overwrite_bookmarklet_cb) for name in self._bm_store.list(): self._add_bookmarklet(name) self.model = Model() self.model.connect('add_link', self._add_link_model_cb) self.current = _('blank') self.webtitle = _('blank') self.connect('key-press-event', self._key_press_cb) self.toolbox.set_current_toolbar(_TOOLBAR_BROWSE) if self.is_ssb: # set permanent homepage for SSBs f = open(os.path.join(activity.get_bundle_path(), 'data/homepage')) self.homepage = f.read() f.close() # enable userscript saving self._browser.userscript.connect('userscript-found', self._userscript_found_cb) # enable userscript injecting self._browser.userscript.connect('userscript-inject', self._userscript_inject_cb) if handle.uri: self._browser.load_uri(handle.uri) elif not self._jobject.file_path: # TODO: we need this hack until we extend the activity API for # opening URIs and default docs. self._load_homepage() self.messenger = None self.connect('shared', self._shared_cb) # Get the Presence Service self.pservice = presenceservice.get_instance() try: name, path = self.pservice.get_preferred_connection() self.tp_conn_name = name self.tp_conn_path = path self.conn = telepathy.client.Connection(name, path) except TypeError: _logger.debug('Offline') self.initiating = None if self._shared_activity is not None: _logger.debug('shared: %s' %self._shared_activity.props.joined) if self._shared_activity is not None: # We are joining the activity _logger.debug('Joined activity') self.connect('joined', self._joined_cb) if self.get_shared(): # We've already joined self._joined_cb() else: _logger.debug('Created activity') def _shared_cb(self, activity_): _logger.debug('My activity was shared') self.initiating = True self._setup() _logger.debug('This is my activity: making a tube...') self.tubes_chan[telepathy.CHANNEL_TYPE_TUBES].OfferDBusTube(SERVICE, {}) def _setup(self): if self._shared_activity is None: _logger.debug('Failed to share or join activity') return bus_name, conn_path, channel_paths = \ self._shared_activity.get_channels() # Work out what our room is called and whether we have Tubes already room = None tubes_chan = None text_chan = None for channel_path in channel_paths: channel = telepathy.client.Channel(bus_name, channel_path) htype, handle = channel.GetHandle() if htype == telepathy.HANDLE_TYPE_ROOM: _logger.debug('Found our room: it has handle#%d "%s"' %(handle, self.conn.InspectHandles(htype, [handle])[0])) room = handle ctype = channel.GetChannelType() if ctype == telepathy.CHANNEL_TYPE_TUBES: _logger.debug('Found our Tubes channel at %s'%channel_path) tubes_chan = channel elif ctype == telepathy.CHANNEL_TYPE_TEXT: _logger.debug('Found our Text channel at %s'%channel_path) text_chan = channel if room is None: _logger.debug("Presence service didn't create a room") return if text_chan is None: _logger.debug("Presence service didn't create a text channel") return # Make sure we have a Tubes channel - PS doesn't yet provide one if tubes_chan is None: _logger.debug("Didn't find our Tubes channel, requesting one...") tubes_chan = self.conn.request_channel(telepathy.CHANNEL_TYPE_TUBES, telepathy.HANDLE_TYPE_ROOM, room, True) self.tubes_chan = tubes_chan self.text_chan = text_chan tubes_chan[telepathy.CHANNEL_TYPE_TUBES].connect_to_signal( \ 'NewTube', self._new_tube_cb) def _list_tubes_reply_cb(self, tubes): for tube_info in tubes: self._new_tube_cb(*tube_info) def _list_tubes_error_cb(self, e): _logger.debug('ListTubes() failed: %s'%e) def _joined_cb(self, activity_): if not self._shared_activity: return _logger.debug('Joined an existing shared activity') self.initiating = False self._setup() _logger.debug('This is not my activity: waiting for a tube...') self.tubes_chan[telepathy.CHANNEL_TYPE_TUBES].ListTubes( reply_handler=self._list_tubes_reply_cb, error_handler=self._list_tubes_error_cb) def _new_tube_cb(self, identifier, initiator, type, service, params, state): _logger.debug('New tube: ID=%d initator=%d type=%d service=%s ' 'params=%r state=%d' %(identifier, initiator, type, service, params, state)) if (type == telepathy.TUBE_TYPE_DBUS and service == SERVICE): if state == telepathy.TUBE_STATE_LOCAL_PENDING: self.tubes_chan[telepathy.CHANNEL_TYPE_TUBES].AcceptDBusTube( identifier) self.tube_conn = TubeConnection(self.conn, self.tubes_chan[telepathy.CHANNEL_TYPE_TUBES], identifier, group_iface = self.text_chan[ telepathy.CHANNEL_INTERFACE_GROUP]) _logger.debug('Tube created') self.messenger = Messenger(self.tube_conn, self.initiating, self.model) def _load_homepage(self): if self.is_ssb: self._browser.load_uri(self.homepage) elif os.path.isfile(_LIBRARY_PATH): self._browser.load_uri('file://' + _LIBRARY_PATH) else: default_page = os.path.join(activity.get_bundle_path(), "data/index.html") self._browser.load_uri(default_page) def _session_history_changed_cb(self, session_history, link): _logger.debug('NewPage: %s.' %link) self.current = link def _title_changed_cb(self, embed, pspec): if embed.props.title is not '': _logger.debug('Title changed=%s' % embed.props.title) self.webtitle = embed.props.title def _get_data_from_file_path(self, file_path): fd = open(file_path, 'r') try: data = fd.read() finally: fd.close() return data def read_file(self, file_path): if self.metadata['mime_type'] == 'text/plain': data = self._get_data_from_file_path(file_path) self.model.deserialize(data) for link in self.model.data['shared_links']: _logger.debug('read: url=%s title=%s d=%s' % (link['url'], link['title'], link['color'])) self._add_link_totray(link['url'], base64.b64decode(link['thumb']), link['color'], link['title'], link['owner'], -1, link['hash']) self._browser.set_session(self.model.data['history']) elif self.metadata['mime_type'] == 'text/uri-list': data = self._get_data_from_file_path(file_path) uris = mime.split_uri_list(data) if len(uris) == 1: self._browser.load_uri(uris[0]) else: _logger.error('Open uri-list: Does not support' 'list of multiple uris by now.') elif self.metadata['mime_type'] == 'application/zip': z = zipfile.ZipFile(file_path, 'r') html = None for i in z.namelist(): if i.endswith('.html') or i.endswith('.htm'): html = i if i == 'index.html': break if file_name != None: self._browser.load_uri('jar:file://%!%s' % (file_path, html)) else: _logger.error('Open jar file: No html file to be opened') else: self._browser.load_uri(file_path) def write_file(self, file_path): if not self.metadata['mime_type']: self.metadata['mime_type'] = 'text/plain' if self.metadata['mime_type'] == 'text/plain': if not self._jobject.metadata['title_set_by_user'] == '1': if self._browser.props.title: self.metadata['title'] = self._browser.props.title self.model.data['history'] = self._browser.get_session() f = open(file_path, 'w') try: f.write(self.model.serialize()) finally: f.close() def save_document(self): logging.debug('Saving document to %s' % bundle_path) cls = components.classes[ \ '@mozilla.org/embedding/browser/nsWebBrowserPersist;1'] persist = cls.createInstance(interfaces.nsIWebBrowserPersist) persist.persistFlags = interfaces.nsIWebBrowserPersist \ .PERSIST_FLAGS_REPLACE_EXISTING_FILES local = components.classes["@mozilla.org/file/local;1"] local_file = local.createInstance(interfaces.nsILocalFile) local_data = local.createInstance(interfaces.nsILocalFile) temp_dir = tempfile.mkdtemp() local_file.initWithPath(os.path.join(temp_dir, 'index.html')) local_data.initWithPath(os.path.join(temp_dir, 'data')) persist.saveDocument(self._browser.dom_window.document, local_file, local_data, None, 0, 0) bundle_path = os.path.join(temp_dir, 'bundle.jar') bundle = zipfile.ZipFile(bundle_path, 'w') bundle.write(local_file.path) for i in os.listdir(local_data.path): bundle.write(os.path.join(local_data.path, i), zipfile.ZIP_DEFLATED) bundle.close() jobject = datastore.create() jobject.metadata['title'] = self.title jobject.metadata['mime_type'] = 'application/zip' jobject.metadata['icon-color'] = profile.get_color().to_string() jobject.metadata['activity'] = 'org.laptop.WebActivity' jobject.file_path = bundle_path datastore.write(jobject) activity.show_object_in_journal(jobject.object_id) # cleanup shutil.rmtree(temp_dir) def _link_add_button_cb(self, button): _logger.debug('button: Add link: %s.' % self.current) self._add_link() def _key_press_cb(self, widget, event): if event.state & gtk.gdk.CONTROL_MASK: if gtk.gdk.keyval_name(event.keyval) == "d": _logger.debug('keyboard: Add link: %s.' % self.current) self._add_link() return True elif gtk.gdk.keyval_name(event.keyval) == "f": _logger.debug('keyboard: Find') self.toolbox.set_current_toolbar(_TOOLBAR_EDIT) self._edit_toolbar.search_entry.grab_focus() return True elif gtk.gdk.keyval_name(event.keyval) == "l": _logger.debug('keyboard: Focus url entry') self.toolbox.set_current_toolbar(_TOOLBAR_BROWSE) self._web_toolbar.entry.grab_focus() return True elif gtk.gdk.keyval_name(event.keyval) == "minus": _logger.debug('keyboard: Zoom out') self._browser.zoom_out() return True elif gtk.gdk.keyval_name(event.keyval) == "plus" \ or gtk.gdk.keyval_name(event.keyval) == "equal" : _logger.debug('keyboard: Zoom in') self._browser.zoom_in() return True return False def _add_bookmarklet(self, name): '''add bookmarklet button and, if needed, the toolbar''' if self._bm_toolbar is None: self._bm_toolbar = BookmarkletToolbar(self) self.toolbox.add_toolbar(_('Bookmarklets'), self._bm_toolbar) self._bm_toolbar.show() if name not in self._bm_toolbar.bookmarklets: self._bm_toolbar.add_bookmarklet(name) return self._bm_toolbar.bookmarklets[name] def _add_bookmarklet_cb(self, store, name): '''receive name of new bookmarklet from the store''' bm = self._add_bookmarklet(name) bm.flash() self.toolbox.set_current_toolbar(_TOOLBAR_BOOKMARKLETS) def _overwrite_bookmarklet_cb(self, store, name, url): '''Ask for confirmation''' alert = ConfirmationAlert() alert.props.title = _('Add bookmarklet') alert.props.msg = _('"%s" already exists. Overwrite?') % name alert.connect('response', self._overwrite_bookmarklet_response_cb) # send the arguments through the alert object alert._bm = (name, url) self.add_alert(alert) def _overwrite_bookmarklet_response_cb(self, alert, response_id): self.remove_alert(alert) name, url = alert._bm # unpack the argument if response_id is gtk.RESPONSE_OK: self._bm_store.remove(name) self._bm_store.add(name, url) def _userscript_found_cb(self, listener, location): '''Ask user whether to install the userscript''' alert = ConfirmationAlert() alert.props.title = _('Add userscript') if usercode.script_exists(location): alert.props.msg = _('Userscript already exists. Overwrite?') else: alert.props.msg = _('Do you want to add this userscript?') alert.connect('response', self._userscript_found_response_cb) # send the argument through the alert object alert._location = location self.add_alert(alert) def _userscript_found_response_cb(self, alert, response_id): self.remove_alert(alert) if response_id is gtk.RESPONSE_OK: usercode.add_script(alert._location) def _userscript_inject_cb(self, listener, script_path): logging.debug('Injecting %s' % script_path) usercode.Injector(script_path).attach_to(self._browser.dom_window) def _add_link(self): ''' take screenshot and add link info to the model ''' for link in self.model.data['shared_links']: if link['hash'] == sha.new(self.current).hexdigest(): _logger.debug('_add_link: link exist already a=%s b=%s' %( link['hash'], sha.new(self.current).hexdigest())) return buf = self._get_screenshot() timestamp = time.time() self.model.add_link(self.current, self.webtitle, buf, profile.get_nick_name(), profile.get_color().to_string(), timestamp) if self.messenger is not None: self.messenger._add_link(self.current, self.webtitle, profile.get_color().to_string(), profile.get_nick_name(), base64.b64encode(buf), timestamp) def _add_link_model_cb(self, model, index): ''' receive index of new link from the model ''' link = self.model.data['shared_links'][index] self._add_link_totray(link['url'], base64.b64decode(link['thumb']), link['color'], link['title'], link['owner'], index, link['hash']) def _add_link_totray(self, url, buf, color, title, owner, index, hash): ''' add a link to the tray ''' item = LinkButton(url, buf, color, title, owner, index, hash) item.connect('clicked', self._link_clicked_cb, url) item.connect('remove_link', self._link_removed_cb) self._tray.add_item(item, index) # use index to add to the tray item.show() if self._tray.props.visible is False: self._tray.show() self._view_toolbar.traybutton.props.sensitive = True def _link_removed_cb(self, button, hash): ''' remove a link from tray and delete it in the model ''' self.model.remove_link(hash) self._tray.remove_item(button) if len(self._tray.get_children()) == 0: self._view_toolbar.traybutton.props.sensitive = False def _link_clicked_cb(self, button, url): ''' an item of the link tray has been clicked ''' self._browser.load_uri(url) def _pixbuf_save_cb(self, buf, data): data[0] += buf return True def get_buffer(self, pixbuf): data = [""] pixbuf.save_to_callback(self._pixbuf_save_cb, "png", {}, data) return str(data[0]) def _get_screenshot(self): window = self._browser.window width, height = window.get_size() screenshot = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, has_alpha=False, bits_per_sample=8, width=width, height=height) screenshot.get_from_drawable(window, window.get_colormap(), 0, 0, 0, 0, width, height) screenshot = screenshot.scale_simple(style.zoom(100), style.zoom(80), gtk.gdk.INTERP_BILINEAR) buf = self.get_buffer(screenshot) return buf def can_close(self): if downloadmanager.can_quit(): return True else: alert = Alert() alert.props.title = _('Download in progress') alert.props.msg = _('Stopping now will cancel your download') cancel_icon = Icon(icon_name='dialog-cancel') alert.add_button(gtk.RESPONSE_CANCEL, _('Cancel'), cancel_icon) stop_icon = Icon(icon_name='dialog-ok') alert.add_button(gtk.RESPONSE_OK, _('Stop'), stop_icon) stop_icon.show() self.add_alert(alert) alert.connect('response', self.__inprogress_response_cb) alert.show() self.present() def __inprogress_response_cb(self, alert, response_id): self.remove_alert(alert) if response_id is gtk.RESPONSE_CANCEL: logging.debug('Keep on') elif response_id == gtk.RESPONSE_OK: logging.debug('Stop downloads and quit') downloadmanager.remove_all_downloads() self.close(force=True) #def handle_view_source(self): # logging.debug('##### local view source') # logging.debug('@@@@@ %s' % usercode.STYLE_PATH) # view_source = viewsource.ViewSource(self.get_xid(), # self.get_bundle_path(), # usercode.STYLE_PATH, # self.get_title()) # view_source.show() def get_document_path(self, async_cb, async_err_cb): self._browser.get_source(async_cb, async_err_cb)