1 files changed, 71 insertions, 0 deletions
diff --git a/button.py b/button.py
new file mode 100644
index 0000000..66cf80b
--- /dev/null
+++ b/button.py
@@ -0,0 +1,71 @@
+import gobject
+import gtk
+from gettext import gettext as _
+
+from sugar.graphics.palette import Palette
+from sugar.graphics.tray import TrayButton
+import constants
+import utils
+
+class RecdButton(TrayButton):
+    __gsignals__ = {
+        'remove-requested': (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, ()),
+        'copy-clipboard-requested': (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, ()),
+    }
+
+    def __init__(self, recd):
+        super(RecdButton, self).__init__()
+        self._recd = recd
+
+        self.set_icon_widget(self.get_image())
+        self._copy_menu_item_handler = None
+
+        palette = Palette(recd.title)
+        self.set_palette(palette)
+
+        self._rem_menu_item = gtk.MenuItem(_('Remove'))
+        self._rem_menu_item_handler = self._rem_menu_item.connect('activate', self._remove_clicked)
+        palette.menu.append(self._rem_menu_item)
+        self._rem_menu_item.show()
+
+        self._add_copy_menu_item()
+
+    def _add_copy_menu_item( self ):
+        if self._recd.buddy and not self._recd.downloadedFromBuddy:
+            return
+
+        self._copy_menu_item = gtk.MenuItem(_('Copy to clipboard'))
+        self._copy_menu_item_handler = self._copy_menu_item.connect('activate', self._copy_clipboard_clicked)
+        self.get_palette().menu.append(self._copy_menu_item)
+        self._copy_menu_item.show()
+ 
+    def get_recd(self):
+        return self._recd
+
+    def get_image(self):
+        img = gtk.Image()
+        ipb = self._recd.getThumbPixbuf()
+        if self._recd.type == constants.TYPE_PHOTO:
+            path = 'object-photo.svg'
+        elif self._recd.type == constants.TYPE_VIDEO:
+            path = 'object-video.svg'
+        elif self._recd.type == constants.TYPE_AUDIO:
+            path = 'object-audio.svg'
+
+        pixbuf = utils.load_colored_svg(path, self._recd.colorStroke, self._recd.colorFill)
+        if ipb:
+            ipb.composite(pixbuf, 8, 8, ipb.get_width(), ipb.get_height(), 8, 8, 1, 1, gtk.gdk.INTERP_BILINEAR, 255)
+        img.set_from_pixbuf(pixbuf)
+        img.show()
+        return img
+
+    def cleanup(self):
+        self._rem_menu_item.disconnect(self._rem_menu_item_handler)
+        if self._copy_menu_item_handler != None:
+            self._copy_menu_item.disconnect(self._copy_menu_item_handler)
+
+    def _remove_clicked(self, widget):
+        self.emit('remove-requested')
+
+    def _copy_clipboard_clicked(self, widget):
+        self.emit('copy-clipboard-requested')