# Copyright (C) 2008, One Laptop Per Child # Copyright (C) 2009, Tomeu Vizoso, Simon Schampijer # # 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 os import tempfile import urlparse from gettext import gettext as _ from gi.repository import Gtk from gi.repository import GObject from sugar3.graphics.palette import Palette, Invoker from sugar3.graphics.menuitem import MenuItem from sugar3.graphics.icon import Icon from sugar3 import profile from sugar3.activity import activity #import downloadmanager class ContentInvoker(Invoker): def __init__(self, browser): Invoker.__init__(self) self._position_hint = self.AT_CURSOR self._browser = browser self._mouseout_listener = None self._popdown_handler_id = None def get_default_position(self): return self.AT_CURSOR def get_rect(self): return () def get_toplevel(self): return None def handleEvent(self, event): if event.button != 2: return target = event.target if target.tagName.lower() == 'a': if target.firstChild: title = target.firstChild.nodeValue else: title = None self.palette = LinkPalette(self._browser, title, target.href, target.ownerDocument) self.notify_right_click() elif target.tagName.lower() == 'img': if target.title: title = target.title elif target.title: title = target.alt elif target.name: title = target.name else: title = os.path.basename(urlparse.urlparse(target.src).path) self.palette = ImagePalette(title, target.src, target.ownerDocument) self.notify_right_click() else: return if self._popdown_handler_id is not None: self._popdown_handler_id = self.palette.connect( \ 'popdown', self.__palette_popdown_cb) self._mouseout_listener = MouseOutListener(target) wrapper = xpcom.server.WrapObject(self._mouseout_listener, interfaces.nsIDOMEventListener) target.addEventListener('mouseout', wrapper, False) self._mouseout_listener.connect('mouse-out', self.__moved_out_cb) def __moved_out_cb(self, listener): self.palette.popdown() def __palette_popdown_cb(self, palette): if self._mouseout_listener is not None: wrapper = xpcom.server.WrapObject(self._mouseout_listener, interfaces.nsIDOMEventListener) self._mouseout_listener.target.removeEventListener('mouseout', wrapper, False) del self._mouseout_listener class LinkPalette(Palette): def __init__(self, browser, title, url, owner_document): Palette.__init__(self) self._browser = browser self._title = title self._url = url self._owner_document = owner_document if title is not None: self.props.primary_text = title self.props.secondary_text = url else: self.props.primary_text = url menu_item = MenuItem(_('Follow link'), 'browse-follow-link') menu_item.connect('activate', self.__follow_activate_cb) self.menu.append(menu_item) menu_item.show() menu_item = MenuItem(_('Follow link in new tab'), 'browse-follow-link-new-tab') menu_item.connect('activate', self.__follow_activate_cb, True) self.menu.append(menu_item) menu_item.show() menu_item = MenuItem(_('Keep link')) icon = Icon(icon_name='document-save', xo_color=profile.get_color(), icon_size=Gtk.IconSize.MENU) menu_item.set_image(icon) menu_item.connect('activate', self.__download_activate_cb) self.menu.append(menu_item) menu_item.show() menu_item = MenuItem(_('Copy link')) icon = Icon(icon_name='edit-copy', xo_color=profile.get_color(), icon_size=Gtk.IconSize.MENU) menu_item.set_image(icon) menu_item.connect('activate', self.__copy_activate_cb) self.menu.append(menu_item) menu_item.show() def __follow_activate_cb(self, menu_item, new_tab=False): if new_tab: new_browser = self._browser.open_new_tab(self._url) else: self._browser.load_uri(self._url) self._browser.grab_focus() def __copy_activate_cb(self, menu_item): clipboard = Gtk.Clipboard() targets = Gtk.target_list_add_uri_targets() targets = Gtk.target_list_add_text_targets(targets) targets.append(('text/x-moz-url', 0, 0)) clipboard.set_with_data(targets, self.__clipboard_get_func_cb, self.__clipboard_clear_func_cb) def __clipboard_get_func_cb(self, clipboard, selection_data, info, data): uri_targets = \ [target[0] for target in Gtk.target_list_add_uri_targets()] text_targets = \ [target[0] for target in Gtk.target_list_add_text_targets()] if selection_data.target in uri_targets: selection_data.set_uris([self._url]) elif selection_data.target in text_targets: selection_data.set_text(self._url) elif selection_data.target == 'text/x-moz-url': selection_data.set('text/x-moz-url', 8, self._url) def __clipboard_clear_func_cb(self, clipboard, data): pass def __download_activate_cb(self, menu_item): downloadmanager.save_link(self._url, self._title, self._owner_document) class ImagePalette(Palette): def __init__(self, title, url, owner_document): Palette.__init__(self) self._title = title self._url = url self._owner_document = owner_document self.props.primary_text = title self.props.secondary_text = url menu_item = MenuItem(_('Keep image')) icon = Icon(icon_name='document-save', xo_color=profile.get_color(), icon_size=Gtk.IconSize.MENU) menu_item.set_image(icon) menu_item.connect('activate', self.__download_activate_cb) self.menu.append(menu_item) menu_item.show() menu_item = MenuItem(_('Copy image')) icon = Icon(icon_name='edit-copy', xo_color=profile.get_color(), icon_size=Gtk.IconSize.MENU) menu_item.set_image(icon) menu_item.connect('activate', self.__copy_activate_cb) self.menu.append(menu_item) menu_item.show() def __copy_activate_cb(self, menu_item): file_name = os.path.basename(urlparse.urlparse(self._url).path) if '.' in file_name: base_name, extension = file_name.split('.') extension = '.' + extension else: base_name = file_name extension = '' temp_path = os.path.join(activity.get_activity_root(), 'instance') fd, temp_file = tempfile.mkstemp(dir=temp_path, prefix=base_name, suffix=extension) os.close(fd) os.chmod(temp_file, 0664) cls = components.classes['@mozilla.org/network/io-service;1'] io_service = cls.getService(interfaces.nsIIOService) uri = io_service.newURI(self._url, None, None) cls = components.classes['@mozilla.org/file/local;1'] target_file = cls.createInstance(interfaces.nsILocalFile) target_file.initWithPath(temp_file) cls = components.classes[ \ '@mozilla.org/embedding/browser/nsWebBrowserPersist;1'] persist = cls.createInstance(interfaces.nsIWebBrowserPersist) persist.persistFlags = 1 # PERSIST_FLAGS_FROM_CACHE listener = xpcom.server.WrapObject(_ImageProgressListener(temp_file), interfaces.nsIWebProgressListener) persist.progressListener = listener persist.saveURI(uri, None, None, None, None, target_file) def __download_activate_cb(self, menu_item): downloadmanager.save_link(self._url, self._title, self._owner_document) class _ImageProgressListener(object): def __init__(self, temp_file): self._temp_file = temp_file def onLocationChange(self, webProgress, request, location): pass def onProgressChange(self, webProgress, request, curSelfProgress, maxSelfProgress, curTotalProgress, maxTotalProgress): pass def onSecurityChange(self, webProgress, request, state): pass def onStatusChange(self, webProgress, request, status, message): pass def onStateChange(self, webProgress, request, stateFlags, status): if (stateFlags & interfaces.nsIWebProgressListener.STATE_IS_REQUEST and stateFlags & interfaces.nsIWebProgressListener.STATE_STOP): clipboard = Gtk.Clipboard() clipboard.set_with_data([('text/uri-list', 0, 0)], _clipboard_get_func_cb, _clipboard_clear_func_cb, self._temp_file) def _clipboard_get_func_cb(clipboard, selection_data, info, temp_file): selection_data.set_uris(['file://' + temp_file]) def _clipboard_clear_func_cb(clipboard, temp_file): if os.path.exists(temp_file): os.remove(temp_file)