nautilus: first working version

Improve the nautilus extension to get to a first working version:
- Single and multiple selections are working.
- The menu item only is there if mat2 has a chance to work on the
  selected files.
- Errors are reported using notifications.

Signed-off-by: Antoine Tenart <antoine.tenart@ack.tf>

1 files changed, 123 insertions, 10 deletions

diff --git a/nautilus/nautilus_mat2.py b/nautilus/nautilus_mat2.py
index 1216470..409df72 100644
--- a/nautilus/nautilus_mat2.py
+++ b/nautilus/nautilus_mat2.py
@@ -2,28 +2,141 @@
 
 import gi
 gi.require_version('Nautilus', '3.0')
-from gi.repository import Nautilus, GObject
+gi.require_version('Gtk', '3.0')
+from gi.repository import Nautilus, GObject, Gtk
+from urllib.parse import unquote
+
+import os
+
+from libmat2 import parser_factory
+
+class Mat2Wrapper():
+    def __init__(self, filepath):
+        self.filepath = filepath
+
+class StatusWindow(Gtk.Window):
+    def __init__(self, items):
+        self.window = Gtk.Window()
+        self.window.set_border_width(10)
+
+        self.items = items
+        self.confirm_window()
+
+    def confirm_window(self):
+        # Header bar
+        hb = Gtk.HeaderBar()
+        self.window.set_titlebar(hb)
+        hb.props.title = "Remove metadata"
+
+        cancel = Gtk.Button("Cancel")
+        cancel.connect("clicked", self.cancel_btn)
+        hb.pack_start(cancel)
+
+        self.remove = Gtk.Button("Remove")
+        self.remove.get_style_context().add_class(Gtk.STYLE_CLASS_SUGGESTED_ACTION);
+        self.remove.connect("clicked", self.remove_btn)
+        hb.pack_end(self.remove)
+
+        self.main_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
+        self.window.add(self.main_box)
+
+        # Disclaimer
+        nitems = len(self.items)
+        if nitems > 1:
+            disclaimer = "Remove metadata from the %d following items:" % nitems
+        else:
+            disclaimer = "Remove metadata from the following item:"
+        self.main_box.pack_start(Gtk.Label(disclaimer, xalign=0), True, True, 0)
+
+        # List of files to clean
+        listbox = Gtk.ListBox()
+        self.main_box.pack_start(listbox, True, True, 0)
+        listbox.set_selection_mode(Gtk.SelectionMode.NONE)
+        for i in self.items:
+            listbox.add(Gtk.Label(os.path.basename(i), xalign=0))
+        listbox.show_all()
+
+        # Options
+        hbox = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=10)
+        self.main_box.pack_start(hbox, True, True, 0)
+        label = Gtk.Label(xalign=0)
+        label.set_markup("Lightweight mode (only remove <i>some</i> metadata)")
+        hbox.pack_start(label, False, True, 0)
+        hbox.pack_start(Gtk.CheckButton(), False, True, 0)
+
+        self.window.show_all()
+
+    def error_window(self, items):
+        self.window.remove(self.main_box)
+
+        box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
+        self.window.add(box)
+
+        # Disclaimer
+        box.pack_start(Gtk.Label("Could not remove metadata from the following items:",
+                                 xalign=0), True, True, 0)
+
+        # List of failed files
+        listbox = Gtk.ListBox()
+        box.pack_start(listbox, True, True, 0)
+        listbox.set_selection_mode(Gtk.SelectionMode.NONE)
+        for i in items:
+            listbox.add(Gtk.Label(os.path.basename(i), xalign=0))
+        listbox.show_all()
+
+        self.window.show_all()
+        self.remove.hide()
+
+    def cancel_btn(self, button):
+        self.window.close()
+
+    def remove_btn(self, button):
+        failed = []
+        for i in self.items:
+            p, mtype = parser_factory.get_parser(i)
+            if p is not None and p.remove_all():
+                continue
+            failed.append(i)
+
+        # Everything went the right way, exit
+        if not len(failed):
+            self.window.close()
+
+        self.error_window(failed)
 
 class ColumnExtension(GObject.GObject, Nautilus.MenuProvider):
-    def menu_activate_cb(self, menu, file):
-        print "menu_activate_cb", file
-        # TODO: clean metadata here
+    def __validate(self, file):
+        if file.get_uri_scheme() != "file" or file.is_directory():
+            return False
+        if not file.can_write():
+            return False
+        return True
+
+    def __validate_set(self, files):
+        for f in files:
+            if self.__validate(f):
+                return True
+        return False
+
+    def menu_activate_cb(self, menu, files):
+        items = list(map(lambda x: unquote(x.get_uri()[7:]), files))
+        StatusWindow(items)
 
     def get_background_items(self, window, file):
         """ https://bugzilla.gnome.org/show_bug.cgi?id=784278 """
         return None
 
     def get_file_items(self, window, files):
-        if len(files) != 1:  # we're not supporting multiple files for now
+        # Do not show the menu item if not a single file has a chance to be
+        # processed by mat2.
+        if not self.__validate_set(files):
             return
 
-        file = files[0]
-
         item = Nautilus.MenuItem(
             name="MAT2::Remove_metadata",
-            label="Remove metadata from %s" % file.get_name(),
-            tip="Remove metadata from %s" % file.get_name()
+            label="Remove metadata",
+            tip="Remove metadata"
         )
-        item.connect('activate', self.menu_activate_cb, file)
+        item.connect('activate', self.menu_activate_cb, files)
 
         return [item]