# Python imports import os import time # Lib imports import gi gi.require_version('Gtk', '3.0') from gi.repository import Gtk from gi.repository import GLib # Application imports from plugins.plugin_base import PluginBase from .mixins.file_search_mixin import FileSearchMixin from .mixins.grep_search_mixin import GrepSearchMixin from .utils.ipc_server import IPCServer class Plugin(IPCServer, FileSearchMixin, GrepSearchMixin, PluginBase): def __init__(self): super().__init__() self.name = "Search" # NOTE: Need to remove after establishing private bidirectional 1-1 message bus # where self.name should not be needed for message comms self.path = os.path.dirname(os.path.realpath(__file__)) self._GLADE_FILE = f"{self.path}/search_dialog.glade" self.update_list_ui_buffer = () self._search_dialog = None self._spinner = None self._active_path = None self.file_list_parent = None self.grep_list_parent = None self._file_list = None self._grep_list = None self._grep_proc = None self._list_proc = None self.pause_fifo_update = False self.grep_query = "" self.grep_time_stamp = None self._queue_grep = False self._grep_watcher_running = False self.search_query = "" self.fsearch_time_stamp = None self._queue_search = False self._search_watcher_running = False def run(self): self._builder = Gtk.Builder() self._builder.add_from_file(self._GLADE_FILE) self._connect_builder_signals(self, self._builder) self._search_dialog = self._builder.get_object("search_dialog") self.fsearch = self._builder.get_object("fsearch") self._spinner = self._builder.get_object("spinner") self.grep_list_parent = self._builder.get_object("grep_list_parent") self.file_list_parent = self._builder.get_object("file_list_parent") self._event_system.subscribe("update-file-ui", self._load_file_ui) self._event_system.subscribe("update-grep-ui", self._load_grep_ui) self._event_system.subscribe("show_search_page", self._show_page) self.create_ipc_listener() def generate_reference_ui_element(self): item = Gtk.ImageMenuItem(self.name) item.set_image( Gtk.Image(stock=Gtk.STOCK_FIND) ) item.connect("activate", self._show_page) item.set_always_show_image(True) return item def stop_spinner(self, ret_code): print(f"Return Code: {ret_code}") self._spinner.stop() def _show_page(self, widget=None, eve=None): self._event_system.emit("get_current_state") state = self._fm_state self._event_message = None self._active_path = state.tab.get_current_directory() response = self._search_dialog.run() self._search_dialog.hide() # TODO: Merge the below methods into some unified logic def reset_grep_box(self) -> None: try: child = self.grep_list_parent.get_children()[0] self._grep_list = None self.grep_list_parent.remove(child) except Exception: ... self._grep_list = Gtk.Box() self._grep_list.set_orientation(Gtk.Orientation.VERTICAL) self.grep_list_parent.add(self._grep_list) self.grep_list_parent.show_all() time.sleep(0.05) Gtk.main_iteration() def reset_file_list_box(self) -> None: try: child = self.file_list_parent.get_children()[0] self._file_list = None self.file_list_parent.remove(child) except Exception: ... self._file_list = Gtk.Box() self._file_list.set_orientation(Gtk.Orientation.VERTICAL) self.file_list_parent.add(self._file_list) self.file_list_parent.show_all() time.sleep(0.05) Gtk.main_iteration()