187 lines
7.1 KiB
Python
187 lines
7.1 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 gobject
|
|
import os
|
|
import shutil
|
|
import dbus
|
|
import dbus.service
|
|
from sugar import env
|
|
from sugar import util
|
|
from clipboardobject import ClipboardObject, Format
|
|
import typeregistry
|
|
|
|
NAME_KEY = 'NAME'
|
|
PERCENT_KEY = 'PERCENT'
|
|
ICON_KEY = 'ICON'
|
|
PREVIEW_KEY = 'PREVIEW'
|
|
ACTIVITY_KEY = 'ACTIVITY'
|
|
FORMATS_KEY = 'FORMATS'
|
|
|
|
class ClipboardDBusServiceHelper(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, parent):
|
|
self._parent = parent
|
|
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
|
|
|
|
def _handle_file_completed(self, cb_object):
|
|
"""If the object is an on-disk file, and it's at 100%, and we care about
|
|
it's file type, copy that file to $HOME and upate the clipboard object's
|
|
data to point to the new location"""
|
|
formats = cb_object.get_formats()
|
|
if not len(formats) or len(formats) > 1:
|
|
return
|
|
|
|
format = formats.values()[0]
|
|
if not format.get_on_disk():
|
|
return
|
|
|
|
if not len(cb_object.get_activity()):
|
|
# no activity to handle this, don't autosave it
|
|
return
|
|
|
|
# copy to homedir
|
|
src = format.get_data()
|
|
if not os.path.exists(src):
|
|
logging.debug("File %s doesn't appear to exist" % src)
|
|
return
|
|
dst = os.path.join(os.path.expanduser("~"), os.path.basename(src))
|
|
try:
|
|
shutil.move(src, dst)
|
|
format._set_data(dst)
|
|
except IOError, e:
|
|
logging.debug("Couldn't move file %s to %s: %s" % (src, dst, e))
|
|
|
|
# dbus methods
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="s", out_signature="o")
|
|
def add_object(self, name):
|
|
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):
|
|
cb_object = self._objects[str(object_path)]
|
|
cb_object.add_format(Format(format_type, data, on_disk))
|
|
|
|
if on_disk:
|
|
logging.debug('Added format of type ' + format_type + ' with path at ' + data)
|
|
else:
|
|
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(),
|
|
ACTIVITY_KEY: cb_object.get_activity()})
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="o", out_signature="")
|
|
def delete_object(self, object_path):
|
|
del self._objects[str(object_path)]
|
|
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:
|
|
self._handle_file_completed(cb_object)
|
|
|
|
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(),
|
|
ACTIVITY_KEY: cb_object.get_activity()})
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="o", out_signature="a{sv}")
|
|
def get_object(self, object_path):
|
|
cb_object = self._objects[str(object_path)]
|
|
formats = cb_object.get_formats()
|
|
format_types = dbus.Array([], '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(),
|
|
ACTIVITY_KEY: cb_object.get_activity(),
|
|
FORMATS_KEY: format_types}
|
|
return dbus.Dictionary(result_dict)
|
|
|
|
@dbus.service.method(_CLIPBOARD_DBUS_INTERFACE,
|
|
in_signature="os", out_signature="ay")
|
|
def get_object_data(self, object_path, format_type):
|
|
cb_object = self._objects[str(object_path)]
|
|
formats = cb_object.get_formats()
|
|
return dbus.ByteArray(formats[format_type].get_data())
|
|
|
|
# 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
|
|
|
|
class ClipboardService(object):
|
|
def __init__(self):
|
|
self._dbus_helper = ClipboardDBusServiceHelper(self)
|
|
|
|
def run(self):
|
|
loop = gobject.MainLoop()
|
|
try:
|
|
loop.run()
|
|
except KeyboardInterrupt:
|
|
print 'Ctrl+C pressed, exiting...'
|