SolarFM/plugins/searcher/mixins/grep_search_mixin.py

87 lines
2.8 KiB
Python
Raw Normal View History

2022-10-04 03:14:18 +00:00
# Python imports
import ctypes
import threading
import subprocess
import signal
import json
import shlex
from datetime import datetime
libgcc_s = ctypes.CDLL('libgcc_s.so.1')
2022-10-04 03:14:18 +00:00
# Lib imports
import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk
from gi.repository import GLib
2022-10-04 03:14:18 +00:00
# Application imports
from ..widgets.grep_preview_widget import GrepPreviewWidget
# NOTE: Threads WILL NOT die with parent's destruction.
def threaded(fn):
def wrapper(*args, **kwargs):
threading.Thread(target=fn, args=args, kwargs=kwargs, daemon=False).start()
2022-10-07 01:48:44 +00:00
2022-10-04 03:14:18 +00:00
return wrapper
# NOTE: Threads WILL die with parent's destruction.
def daemon_threaded(fn):
def wrapper(*args, **kwargs):
threading.Thread(target=fn, args=args, kwargs=kwargs, daemon=True).start()
2022-10-07 01:48:44 +00:00
2022-10-04 03:14:18 +00:00
return wrapper
class GrepSearchMixin:
def _run_grep_query(self, widget=None, eve=None):
self._handle_grep_query(query=widget)
2022-10-07 01:48:44 +00:00
# TODO: Merge this logic with nearly the exact same thing in file_search_mixin
2022-10-04 03:14:18 +00:00
@daemon_threaded
def _handle_grep_query(self, widget=None, eve=None, query=None):
# NOTE: Freeze IPC consumption
self.pause_fifo_update = True
2022-10-05 03:58:27 +00:00
self.grep_query = ""
dt = datetime.now()
self.grep_time_stamp = datetime.timestamp(dt) # NOTE: Get timestamp
2022-10-04 03:14:18 +00:00
# NOTE: Kill the former process
if self._grep_proc:
2022-10-07 01:48:44 +00:00
if self._grep_proc.poll() == None:
self._grep_proc.terminate()
while self._grep_proc.poll() == None:
...
2022-10-04 03:14:18 +00:00
2022-10-07 01:48:44 +00:00
self._grep_proc = None
2022-10-04 03:14:18 +00:00
2022-10-05 03:58:27 +00:00
# NOTE: Clear children from ui and make sure ui thread redraws
2022-10-07 01:48:44 +00:00
GLib.idle_add(self.reset_grep_box)
2022-10-04 03:14:18 +00:00
# NOTE: If query create new process and do all new loop.
if query:
self.pause_fifo_update = False
2022-10-04 03:14:18 +00:00
GLib.idle_add(self._exec_grep_query, query)
def _exec_grep_query(self, widget=None, eve=None):
query = widget.get_text()
if not query.strip() in ("", None):
2022-10-05 03:58:27 +00:00
self.grep_query = query
2022-10-04 03:14:18 +00:00
target_dir = shlex.quote( self._fm_state.tab.get_current_directory() )
command = ["python", f"{self.path}/utils/search.py", "-t", "grep_search", "-d", f"{target_dir}", "-q", f"{query}"]
self._grep_proc = subprocess.Popen(command, cwd=self.path, stdin=None, stdout=None, stderr=None)
def _load_grep_ui(self, data):
Gtk.main_iteration()
2022-10-05 03:58:27 +00:00
if not data in ("", None):
2022-10-04 03:14:18 +00:00
jdata = json.loads( data )
jkeys = jdata.keys()
for key in jkeys:
sub_keys = jdata[key].keys()
grep_result = jdata[key]
2022-10-05 03:58:27 +00:00
widget = GrepPreviewWidget(key, sub_keys, grep_result, self.grep_query)
self._grep_list.add(widget)