ecb8816bc9
I think it could be a little cleaner, but it would require refactoring, we can do that post Trial-3. Also we might have to figure out a way to get file saving progress feedback at some point.
211 lines
8.3 KiB
Python
211 lines
8.3 KiB
Python
# Copyright (C) 2006, 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 logging
|
|
import os
|
|
import shutil
|
|
import urlparse
|
|
import tempfile
|
|
|
|
import dbus
|
|
import dbus.service
|
|
|
|
from sugar import env
|
|
from sugar import util
|
|
from sugar.objects import mime
|
|
|
|
from clipboardobject import ClipboardObject, Format
|
|
|
|
NAME_KEY = 'NAME'
|
|
PERCENT_KEY = 'PERCENT'
|
|
ICON_KEY = 'ICON'
|
|
PREVIEW_KEY = 'PREVIEW'
|
|
ACTIVITIES_KEY = 'ACTIVITIES'
|
|
FORMATS_KEY = 'FORMATS'
|
|
|
|
TYPE_KEY = 'TYPE'
|
|
DATA_KEY = 'DATA'
|
|
ON_DISK_KEY = 'ON_DISK'
|
|
|
|
class ClipboardService(dbus.service.Object):
|
|
|
|
_CLIPBOARD_DBUS_INTERFACE = "org.laptop.Clipboard"
|
|
_CLIPBOARD_OBJECT_PATH = "/org/laptop/Clipboard"
|
|
_CLIPBOARD_OBJECTS_PATH = _CLIPBOARD_OBJECT_PATH + "/Objects/"
|
|
|
|
def __init__(self):
|
|
self._objects = {}
|
|
self._next_id = 0
|
|
|
|
bus = dbus.SessionBus()
|
|
bus_name = dbus.service.BusName(self._CLIPBOARD_DBUS_INTERFACE, bus=bus)
|
|
dbus.service.Object.__init__(self, bus_name, self._CLIPBOARD_OBJECT_PATH)
|
|
|
|
def _get_next_object_id(self):
|
|
self._next_id += 1
|
|
return self._next_id
|
|
|
|
# dbus methods
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="s", out_signature="o")
|
|
def add_object(self, name):
|
|
logging.debug('ClipboardService.add_object')
|
|
op = self._CLIPBOARD_OBJECTS_PATH + "%d" % self._get_next_object_id()
|
|
self._objects[op] = ClipboardObject(op, name)
|
|
self.object_added(dbus.ObjectPath(op), name)
|
|
logging.debug('Added object ' + op + ' with name ' + name)
|
|
return dbus.ObjectPath(op)
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="ssayb", out_signature="", byte_arrays=True)
|
|
def add_object_format(self, object_path, format_type, data, on_disk):
|
|
logging.debug('ClipboardService.add_object_format')
|
|
cb_object = self._objects[str(object_path)]
|
|
|
|
if format_type == 'XdndDirectSave0':
|
|
cb_object.add_format(Format(format_type, data, on_disk))
|
|
elif on_disk and cb_object.get_percent() == 100:
|
|
new_uri = self._copy_file(data)
|
|
cb_object.add_format(Format(format_type, new_uri, on_disk))
|
|
logging.debug('Added format of type ' + format_type + ' with path at ' + new_uri)
|
|
else:
|
|
cb_object.add_format(Format(format_type, data, on_disk))
|
|
logging.debug('Added in-memory format of type ' + format_type + '.')
|
|
|
|
self.object_state_changed(object_path, {NAME_KEY: cb_object.get_name(),
|
|
PERCENT_KEY: cb_object.get_percent(),
|
|
ICON_KEY: cb_object.get_icon(),
|
|
PREVIEW_KEY: cb_object.get_preview(),
|
|
ACTIVITIES_KEY: cb_object.get_activities()})
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="o", out_signature="")
|
|
def delete_object(self, object_path):
|
|
cb_object = self._objects.pop(str(object_path))
|
|
cb_object.destroy()
|
|
self.object_deleted(object_path)
|
|
logging.debug('Deleted object with object_id ' + object_path)
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="oi", out_signature="")
|
|
def set_object_percent(self, object_path, percent):
|
|
cb_object = self._objects[str(object_path)]
|
|
if percent < 0 or percent > 100:
|
|
raise ValueError("invalid percentage")
|
|
if cb_object.get_percent() > percent:
|
|
raise ValueError("invalid percentage; less than current percent")
|
|
if cb_object.get_percent() == percent:
|
|
# ignore setting same percentage
|
|
return
|
|
|
|
cb_object.set_percent(percent)
|
|
|
|
if percent == 100:
|
|
formats = cb_object.get_formats()
|
|
for format_name, format in formats.iteritems():
|
|
if format.is_on_disk() and format_name != 'XdndDirectSave0':
|
|
new_uri = self._copy_file(format.get_data())
|
|
format.set_data(new_uri)
|
|
|
|
# Add a text/plain format to objects that are text but lack it
|
|
if 'text/plain' not in formats.keys():
|
|
if 'UTF8_STRING' in formats.keys():
|
|
self.add_object_format(object_path,
|
|
'text/plain',
|
|
data=formats['UTF8_STRING'].get_data(),
|
|
on_disk=False)
|
|
elif 'text/unicode' in formats.keys():
|
|
self.add_object_format(object_path,
|
|
'text/plain',
|
|
data=formats['UTF8_STRING'].get_data(),
|
|
on_disk=False)
|
|
|
|
self.object_state_changed(object_path, {NAME_KEY: cb_object.get_name(),
|
|
PERCENT_KEY: percent,
|
|
ICON_KEY: cb_object.get_icon(),
|
|
PREVIEW_KEY: cb_object.get_preview(),
|
|
ACTIVITIES_KEY: cb_object.get_activities()})
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="o", out_signature="a{sv}")
|
|
def get_object(self, object_path):
|
|
logging.debug('ClipboardService.get_object')
|
|
|
|
if not self._objects.has_key(str(object_path)):
|
|
return dbus.Dictionary({}, signature='sv')
|
|
|
|
cb_object = self._objects[str(object_path)]
|
|
formats = cb_object.get_formats()
|
|
format_types = dbus.Array([], signature='s')
|
|
|
|
for type, format in formats.iteritems():
|
|
format_types.append(type)
|
|
|
|
result_dict = {NAME_KEY: cb_object.get_name(),
|
|
PERCENT_KEY: cb_object.get_percent(),
|
|
ICON_KEY: cb_object.get_icon(),
|
|
PREVIEW_KEY: cb_object.get_preview(),
|
|
ACTIVITIES_KEY: cb_object.get_activities(),
|
|
FORMATS_KEY: format_types}
|
|
return dbus.Dictionary(result_dict)
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="os", out_signature="a{sv}")
|
|
def get_object_data(self, object_path, format_type):
|
|
logging.debug('ClipboardService.get_object_data')
|
|
cb_object = self._objects[str(object_path)]
|
|
format = cb_object.get_formats()[format_type]
|
|
result_dict = {TYPE_KEY: format.get_type(),
|
|
DATA_KEY: dbus.ByteArray(format.get_data()),
|
|
ON_DISK_KEY: format.is_on_disk()}
|
|
return dbus.Dictionary(result_dict)
|
|
|
|
# dbus signals
|
|
@dbus.service.signal(_CLIPBOARD_DBUS_INTERFACE, signature="os")
|
|
def object_added(self, object_path, name):
|
|
pass
|
|
|
|
@dbus.service.signal(_CLIPBOARD_DBUS_INTERFACE, signature="o")
|
|
def object_deleted(self, object_path):
|
|
pass
|
|
|
|
@dbus.service.signal(_CLIPBOARD_DBUS_INTERFACE, signature="oa{sv}")
|
|
def object_state_changed(self, object_path, values):
|
|
pass
|
|
|
|
def _copy_file(self, original_uri):
|
|
uri = urlparse.urlparse(original_uri)
|
|
path, file_name = os.path.split(uri.path)
|
|
|
|
root, ext = os.path.splitext(file_name)
|
|
if not ext or ext == '.':
|
|
mime_type = mime.get_for_file(uri.path)
|
|
ext = '.' + mime.get_primary_extension(mime_type)
|
|
|
|
f, new_file_path = tempfile.mkstemp(ext, root)
|
|
del f
|
|
shutil.copyfile(uri.path, new_file_path)
|
|
|
|
return 'file://' + new_file_path
|
|
|
|
_instance = None
|
|
|
|
def get_instance():
|
|
global _instance
|
|
if not _instance:
|
|
_instance = ClipboardService()
|
|
return _instance
|