- __gtype_name__ = "QuickOpenPopup"

- def __init__(self, window, paths, handler):

- Gtk.Dialog.__init__(self,

- title=_('Quick Open'),

- parent=window,

- flags=Gtk.DialogFlags.DESTROY_WITH_PARENT | Gtk.DialogFlags.MODAL,

- buttons=(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL))

- self._open_button = self.add_button(Gtk.STOCK_OPEN, Gtk.ResponseType.ACCEPT)

- self._handler = handler

- self._build_ui()

- self._size = (0, 0)

- self._dirs = []

- self._cache = {}

- self._theme = None

- self._cursor = None

- self._shift_start = None

- self._busy_cursor = Gdk.Cursor(Gdk.CursorType.WATCH)

- accel_group = Gtk.AccelGroup()

- accel_group.connect(Gdk.KEY_l, Gdk.ModifierType.CONTROL_MASK, 0, self.on_focus_entry)

- self.add_accel_group(accel_group)

- unique = []

- for path in paths:

- if not path.get_uri() in unique:

- self._dirs.append(path)

- unique.append(path.get_uri())

- def get_final_size(self):

- return self._size

- def _build_ui(self):

- vbox = self.get_content_area()

- vbox.set_spacing(3)

- self._entry = Gtk.Entry()

- self._entry.connect('changed', self.on_changed)

- self._entry.connect('key-press-event', self.on_key_press_event)

- sw = Gtk.ScrolledWindow()

- sw.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)

- sw.set_shadow_type(Gtk.ShadowType.OUT)

- tv = Gtk.TreeView()

- tv.set_headers_visible(False)

- self._store = Gtk.ListStore(Gio.Icon, str, GObject.Object, Gio.FileType)

- tv.set_model(self._store)

- self._treeview = tv

- tv.connect('row-activated', self.on_row_activated)

- renderer = Gtk.CellRendererPixbuf()

- column = Gtk.TreeViewColumn()

- column.pack_start(renderer, False)

- column.add_attribute(renderer, "gicon", 0)

- renderer = Gtk.CellRendererText()

- column.pack_start(renderer, True)

- column.add_attribute(renderer, "markup", 1)

- column.set_cell_data_func(renderer, self.on_cell_data_cb, None)

- tv.append_column(column)

- sw.add(tv)

-

- selection = tv.get_selection()

- selection.connect('changed', self.on_selection_changed)

- selection.set_mode(Gtk.SelectionMode.MULTIPLE)

- vbox.pack_start(self._entry, False, False, 0)

- vbox.pack_start(sw, True, True, 0)

- lbl = Gtk.Label()

- lbl.set_alignment(0, 0.5)

- lbl.set_ellipsize(Pango.EllipsizeMode.MIDDLE)

- self._info_label = lbl

- vbox.pack_start(lbl, False, False, 0)

- # Initial selection

- self.on_selection_changed(tv.get_selection())

- vbox.show_all()

- def on_cell_data_cb(self, column, cell, model, piter, user_data):

- path = model.get_path(piter)

-

- if self._cursor and path == self._cursor.get_path():

- style = self._treeview.get_style()

- bg = style.bg[Gtk.StateType.PRELIGHT]

-

- cell.set_property('cell-background-gdk', bg)

- cell.set_property('style', Pango.Style.ITALIC)

- else:

- cell.set_property('cell-background-set', False)

- cell.set_property('style-set', False)

- def _icon_from_stock(self, stock):

- theme = Gtk.icon_theme_get_default()

- size = Gtk.icon_size_lookup(Gtk.IconSize.MENU)

- pixbuf = theme.load_icon(stock, size[0], Gtk.IconLookupFlags.USE_BUILTIN)

- return pixbuf

- def _list_dir(self, gfile):

- entries = []

- try:

- ret = gfile.enumerate_children("standard::*", Gio.FileQueryInfoFlags.NONE, None)

- except GLib.GError:

- pass

- if isinstance(ret, Gio.FileEnumerator):

- while True:

- entry = ret.next_file(None)

- if not entry:

- break

- entries.append((gfile.get_child(entry.get_name()), entry))

- else:

- entries = ret

- children = []

- for entry in entries:

- children.append((entry[0], entry[1].get_name(), entry[1].get_file_type(), entry[1].get_icon()))

- return children

- def _compare_entries(self, a, b, lpart):

- if lpart in a:

- if lpart in b:

- return cmp(a.index(lpart), b.index(lpart))

- else:

- return -1

- elif lpart in b:

- return 1

- else:

- return 0

- def _match_glob(self, s, glob):

- if glob:

- glob += '*'

- return fnmatch.fnmatch(s, glob)

- def do_search_dir(self, parts, d):

- if not parts or not d:

- return []

- if not d in self._cache:

- entries = self._list_dir(d)

- entries.sort(lambda x, y: cmp(x[1].lower(), y[1].lower()))

- self._cache[d] = entries

- else:

- entries = self._cache[d]

- found = []

- newdirs = []

- lpart = parts[0].lower()

- for entry in entries:

- if not entry:

- continue

- lentry = entry[1].lower()

- if not lpart or lpart in lentry or self._match_glob(lentry, lpart):

- if entry[2] == Gio.FileType.DIRECTORY:

- if len(parts) > 1:

- newdirs.append(entry[0])

- else:

- found.append(entry)

- elif entry[2] == Gio.FileType.REGULAR and \

- (not lpart or len(parts) == 1):

- found.append(entry)

- found.sort(lambda a, b: self._compare_entries(a[1].lower(), b[1].lower(), lpart))

- if lpart == '..':

- newdirs.append(d.get_parent())

- for dd in newdirs:

- found.extend(self.do_search_dir(parts[1:], dd))

- return found

- def _replace_insensitive(self, s, find, rep):

- out = ''

- l = s.lower()

- find = find.lower()

- last = 0

- if len(find) == 0:

- return xml.sax.saxutils.escape(s)

- while True:

- m = l.find(find, last)

- if m == -1:

- break

- else:

- out += xml.sax.saxutils.escape(s[last:m]) + rep % (xml.sax.saxutils.escape(s[m:m + len(find)]),)

- last = m + len(find)

- return out + xml.sax.saxutils.escape(s[last:])

- def make_markup(self, parts, path):

- out = []

- for i in range(0, len(parts)):

- out.append(self._replace_insensitive(path[i], parts[i], "<b>%s</b>"))

- return os.sep.join(out)

- def _get_icon(self, f):

- query = f.query_info(Gio.FILE_ATTRIBUTE_STANDARD_ICON, Gio.FileQueryInfoFlags.NONE, None)

- if not query:

- return None

- else:

- return query.get_icon()

- def _make_parts(self, parent, child, pp):

- parts = []

- # We went from parent, to child, using pp

- idx = len(pp) - 1

- while idx >= 0:

- if pp[idx] == '..':

- parts.insert(0, '..')

- else:

- parts.insert(0, child.get_basename())

- child = child.get_parent()

- idx -= 1

- return parts

- def normalize_relative(self, parts):

- if not parts:

- return []

- out = self.normalize_relative(parts[:-1])

- if parts[-1] == '..':

- if not out or (out[-1] == '..') or len(out) == 1:

- out.append('..')

- else:

- del out[-1]

- else:

- out.append(parts[-1])

- return out

- def _append_to_store(self, item):

- if not item in self._stored_items:

- self._store.append(item)

- self._stored_items[item] = True

- def _clear_store(self):

- self._store.clear()

- self._stored_items = {}

- def _show_virtuals(self):

- for d in self._dirs:

- if isinstance(d, VirtualDirectory):

- for entry in d.enumerate_children("standard::*", 0, None):

- self._append_to_store((entry[1].get_icon(), xml.sax.saxutils.escape(entry[1].get_name()), entry[0], entry[1].get_file_type()))

- def _set_busy(self, busy):

- if busy:

- self.get_window().set_cursor(self._busy_cursor)

- else:

- self.get_window().set_cursor(None)

- Gdk.flush()

- def _remove_cursor(self):

- if self._cursor:

- path = self._cursor.get_path()

- self._cursor = None

- self._store.row_changed(path, self._store.get_iter(path))

- def do_search(self):

- self._set_busy(True)

- self._remove_cursor()

- text = self._entry.get_text().strip()

- self._clear_store()

- if text == '':

- self._show_virtuals()

- else:

- parts = self.normalize_relative(text.split(os.sep))

- files = []

- for d in self._dirs:

- for entry in self.do_search_dir(parts, d):

- pathparts = self._make_parts(d, entry[0], parts)

- self._append_to_store((entry[3], self.make_markup(parts, pathparts), entry[0], entry[2]))

- piter = self._store.get_iter_first()

- if piter:

- self._treeview.get_selection().select_path(self._store.get_path(piter))

- self.get_window().set_cursor(None)

- self._set_busy(False)

- def do_show(self):

- Gtk.Window.do_show(self)

- self._entry.grab_focus()

- self._entry.set_text("")

-

- self.do_search()

-

- def on_changed(self, editable):

- self.do_search()

- self.on_selection_changed(self._treeview.get_selection())

-

- def _shift_extend(self, towhere):

- selection = self._treeview.get_selection()

-

- if not self._shift_start:

- model, rows = selection.get_selected_rows()

- start = rows[0]

-

- self._shift_start = Gtk.TreeRowReference(self._store, start)

- else:

- start = self._shift_start.get_path()

-

- selection.unselect_all()

- selection.select_range(start, towhere)

-

- def _select_index(self, idx, hasctrl, hasshift):

- path = (idx,)

-

- if not (hasctrl or hasshift):

- self._treeview.get_selection().unselect_all()

-

- if hasshift:

- self._shift_extend(path)

- else:

- self._shift_start = None

-

- if not hasctrl:

- self._treeview.get_selection().select_path(path)

-

- self._treeview.scroll_to_cell(path, None, True, 0.5, 0)

- self._remove_cursor()

-

- if hasctrl or hasshift:

- self._cursor = Gtk.TreeRowReference(self._store, path)

-

- piter = self._store.get_iter(path)

- self._store.row_changed(path, piter)

-

- def _move_selection(self, howmany, hasctrl, hasshift):

- num = self._store.iter_n_children(None)

-

- if num == 0:

- return True

-

- # Test for cursor

- path = None

-

- if self._cursor:

- path = self._cursor.get_path()

- else:

- model, rows = self._treeview.get_selection().get_selected_rows()

-

- if len(rows) == 1:

- path = rows[0]

-

- if not path:

- if howmany > 0:

- self._select_index(0, hasctrl, hasshift)

- else:

- self._select_index(num - 1, hasctrl, hasshift)

- else:

- idx = path[0]

-

- if idx + howmany < 0:

- self._select_index(0, hasctrl, hasshift)

- elif idx + howmany >= num:

- self._select_index(num - 1, hasctrl, hasshift)

- else:

- self._select_index(idx + howmany, hasctrl, hasshift)

- return True

- def _direct_file(self):

- uri = self._entry.get_text()

- gfile = None

-

- if Pluma.utils_is_valid_uri(uri):

- gfile = Gio.file_new_for_uri(uri)

- elif os.path.isabs(uri):

- f = Gio.file_new_for_uri(uri)

-

- if f.query_exists():

- gfile = f

-

- return gfile

-

- def _activate(self):

- model, rows = self._treeview.get_selection().get_selected_rows()

- ret = True

-

- for row in rows:

- s = model.get_iter(row)

- info = model.get(s, 2, 3)

-

- if info[1] != Gio.FileType.DIRECTORY:

- ret = ret and self._handler(info[0])

- else:

- text = self._entry.get_text()

-

- for i in range(len(text) - 1, -1, -1):

- if text[i] == os.sep:

- break

-

- self._entry.set_text(os.path.join(text[:i], os.path.basename(info[0].get_uri())) + os.sep)

- self._entry.set_position(-1)

- self._entry.grab_focus()

- return True

-

- if rows and ret:

- self.destroy()

-

- if not rows:

- gfile = self._direct_file()

-

- if gfile and self._handler(gfile):

- self.destroy()

- else:

- ret = False

- else:

- ret = False

-

- return ret

-

- def toggle_cursor(self):

- if not self._cursor:

- return

-

- path = self._cursor.get_path()

- selection = self._treeview.get_selection()

-

- if selection.path_is_selected(path):

- selection.unselect_path(path)

- else:

- selection.select_path(path)

-

- def on_key_press_event(self, widget, event):

- move_mapping = {

- Gdk.KEY_Down: 1,

- Gdk.KEY_Up: -1,

- Gdk.KEY_Page_Down: 5,

- Gdk.KEY_Page_Up: -5

- }

-

- if event.keyval == Gdk.KEY_Escape:

- self.destroy()

- return True

- elif event.keyval in move_mapping:

- return self._move_selection(move_mapping[event.keyval], event.state & Gdk.ModifierType.CONTROL_MASK, event.state & Gdk.ModifierType.SHIFT_MASK)

- elif event.keyval in [Gdk.KEY_Return, Gdk.KEY_KP_Enter, Gdk.KEY_Tab, Gdk.KEY_ISO_Left_Tab]:

- return self._activate()

- elif event.keyval == Gdk.KEY_space and event.state & Gdk.ModifierType.CONTROL_MASK:

- self.toggle_cursor()

-

- return False

-

- def on_row_activated(self, view, path, column):

- self._activate()

-

- def do_response(self, response):

- if response != Gtk.ResponseType.ACCEPT or not self._activate():

- self.destroy()

-

- def do_configure_event(self, event):

- if self.get_realized():

- alloc = self.get_allocation()

- self._size = (alloc.width, alloc.height)

-

- return Gtk.Dialog.do_configure_event(self, event)

-

- def on_selection_changed(self, selection):

- model, rows = selection.get_selected_rows()

-

- gfile = None

- fname = None

-

- if not rows:

- gfile = self._direct_file()

- elif len(rows) == 1:

- gfile = model.get(model.get_iter(rows[0]), 2)[0]

- else:

- fname = ''

-

- if gfile:

- if gfile.is_native():

- fname = xml.sax.saxutils.escape(gfile.get_path())

- else:

- fname = xml.sax.saxutils.escape(gfile.get_uri())

-

- self._open_button.set_sensitive(fname != None)

- self._info_label.set_markup(fname or '')

-

- def on_focus_entry(self, group, accel, keyval, modifier):

-# ex:ts=8:et: