generated from itdominator/Python-With-Gtk-Template
220 lines
7.6 KiB
Python
220 lines
7.6 KiB
Python
# Python imports
|
|
import os
|
|
import re
|
|
import threading
|
|
|
|
# Lib imports
|
|
import gi
|
|
gi.require_version('Gtk', '3.0')
|
|
gi.require_version('Gdk', '3.0')
|
|
from gi.repository import Gtk
|
|
from gi.repository import Gdk
|
|
from gi.repository import GLib
|
|
|
|
# Application imports
|
|
from plugins.plugin_base import PluginBase
|
|
from .styling_mixin import StylingMixin
|
|
from .replace_mixin import ReplaceMixin
|
|
|
|
|
|
|
|
class Plugin(StylingMixin, ReplaceMixin, PluginBase):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
self.name = "Search/Replace" # 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_replace.glade"
|
|
|
|
self._search_replace_dialog = None
|
|
self._find_entry = None
|
|
self._replace_entry = None
|
|
self._active_src_view = None
|
|
self._buffer = None
|
|
self._tag_table = None
|
|
|
|
self.use_regex = False
|
|
self.use_case_sensitive = False
|
|
self.search_only_in_selection = False
|
|
self.use_whole_word_search = False
|
|
|
|
self.timer = None
|
|
self.search_time = 0.35
|
|
self.find_text = ""
|
|
self.search_tag = "search_tag"
|
|
self.highlight_color = "#FBF719"
|
|
self.text_color = "#000000"
|
|
self.alpha_num_under = re.compile(r"[a-zA-Z0-9_]")
|
|
|
|
|
|
def run(self):
|
|
self._builder = Gtk.Builder()
|
|
self._builder.add_from_file(self._GLADE_FILE)
|
|
self._connect_builder_signals(self, self._builder)
|
|
|
|
separator_botton = self._ui_objects[0]
|
|
self._search_replace_dialog = self._builder.get_object("search_replace_dialog")
|
|
self._find_status_lbl = self._builder.get_object("find_status_lbl")
|
|
self._find_options_lbl = self._builder.get_object("find_options_lbl")
|
|
|
|
self._find_entry = self._builder.get_object("find_entry")
|
|
self._replace_entry = self._builder.get_object("replace_entry")
|
|
|
|
self._search_replace_dialog.set_relative_to(separator_botton)
|
|
self._search_replace_dialog.set_hexpand(True)
|
|
|
|
def generate_reference_ui_element(self):
|
|
...
|
|
|
|
def subscribe_to_events(self):
|
|
self._event_system.subscribe("tggl_search_replace", self._tggl_search_replace)
|
|
self._event_system.subscribe("set_active_src_view", self._set_active_src_view)
|
|
|
|
def _set_active_src_view(self, source_view):
|
|
self._active_src_view = source_view
|
|
self._buffer = self._active_src_view.get_buffer()
|
|
self._tag_table = self._buffer.get_tag_table()
|
|
self.search_for_string(self._find_entry)
|
|
|
|
def _show_search_replace(self, widget = None, eve = None):
|
|
self._search_replace_dialog.popup()
|
|
|
|
def _tggl_search_replace(self, widget = None, eve = None):
|
|
is_visible = self._search_replace_dialog.is_visible()
|
|
buffer = self._active_src_view.get_buffer()
|
|
data = None
|
|
|
|
if buffer.get_has_selection():
|
|
start, end = buffer.get_selection_bounds()
|
|
data = buffer.get_text(start, end, include_hidden_chars = False)
|
|
|
|
if data:
|
|
self._find_entry.set_text(data)
|
|
|
|
if not is_visible:
|
|
self._search_replace_dialog.popup();
|
|
self._find_entry.grab_focus()
|
|
elif not data and is_visible:
|
|
self._search_replace_dialog.popdown()
|
|
self._find_entry.set_text("")
|
|
|
|
|
|
def get_search_tag(self, buffer):
|
|
tag_table = buffer.get_tag_table()
|
|
search_tag = tag_table.lookup(self.search_tag)
|
|
if not search_tag:
|
|
search_tag = buffer.create_tag(self.search_tag, background = self.highlight_color, foreground = self.text_color)
|
|
|
|
buffer.remove_tag_by_name(self.search_tag, buffer.get_start_iter(), buffer.get_end_iter())
|
|
return search_tag
|
|
|
|
|
|
def cancel_timer(self):
|
|
if self.timer:
|
|
self.timer.cancel()
|
|
GLib.idle_remove_by_data(None)
|
|
|
|
def delay_search_glib(self):
|
|
GLib.idle_add(self._do_highlight)
|
|
|
|
def delay_search(self):
|
|
wait_time = self.search_time / len(self.find_text)
|
|
wait_time = max(wait_time, 0.05)
|
|
|
|
self.timer = threading.Timer(wait_time, self.delay_search_glib)
|
|
self.timer.daemon = True
|
|
self.timer.start()
|
|
|
|
|
|
def on_enter_search(self, widget, eve):
|
|
text = widget.get_text()
|
|
if not text: return
|
|
|
|
keyname = Gdk.keyval_name(eve.keyval)
|
|
if keyname == "Return":
|
|
self.find_next(widget)
|
|
|
|
def search_for_string(self, widget):
|
|
self.cancel_timer()
|
|
|
|
self.find_text = widget.get_text()
|
|
if len(self.find_text) > 0 and len(self.find_text) < 5:
|
|
self.delay_search()
|
|
else:
|
|
self._do_highlight(self.find_text)
|
|
|
|
|
|
def _do_highlight(self, query = None):
|
|
query = self.find_text if not query else query
|
|
buffer = self._active_src_view.get_buffer()
|
|
# Also clears tag from buffer so if no query we're clean in ui
|
|
search_tag = self.get_search_tag(buffer)
|
|
|
|
self.update_style(1)
|
|
if not query:
|
|
self._find_status_lbl.set_label(f"Find in current buffer")
|
|
self.update_style(0)
|
|
return
|
|
|
|
start_itr = buffer.get_start_iter()
|
|
end_itr = buffer.get_end_iter()
|
|
|
|
results, total_count = self.search(start_itr, query)
|
|
self._update_status_lbl(total_count, query)
|
|
for start, end in results:
|
|
buffer.apply_tag(search_tag, start, end)
|
|
|
|
def search(self, start_itr = None, query = None, limit = None):
|
|
if not start_itr or not query: return None, None
|
|
|
|
flags = Gtk.TextSearchFlags.VISIBLE_ONLY | Gtk.TextSearchFlags.TEXT_ONLY
|
|
if not self.use_case_sensitive:
|
|
flags = flags | Gtk.TextSearchFlags.CASE_INSENSITIVE
|
|
|
|
if self.search_only_in_selection and self._buffer.get_has_selection():
|
|
start_itr, limit = self._buffer.get_selection_bounds()
|
|
|
|
_results = []
|
|
while True:
|
|
result = start_itr.forward_search(query, flags, limit)
|
|
if not result: break
|
|
|
|
_results.append(result)
|
|
start_itr = result[1]
|
|
|
|
results = self.apply_filters(_results, query)
|
|
return results, len(results)
|
|
|
|
def apply_filters(self, _results, query):
|
|
results = []
|
|
for start, end in _results:
|
|
text = self._buffer.get_slice(start, end, include_hidden_chars = False)
|
|
if self.use_whole_word_search:
|
|
if not self.is_whole_word(start, end):
|
|
continue
|
|
|
|
results.append([start, end])
|
|
|
|
return results
|
|
|
|
def find_next(self, widget, eve = None, use_data = None):
|
|
mark = self._buffer.get_insert()
|
|
iter = self._buffer.get_iter_at_mark(mark)
|
|
iter.forward_line()
|
|
|
|
search_tag = self._tag_table.lookup(self.search_tag)
|
|
next_tag_found = iter.forward_to_tag_toggle(search_tag)
|
|
if not next_tag_found:
|
|
self._buffer.place_cursor( self._buffer.get_start_iter() )
|
|
mark = self._buffer.get_insert()
|
|
iter = self._buffer.get_iter_at_mark(mark)
|
|
iter.forward_to_tag_toggle(search_tag)
|
|
|
|
self._buffer.place_cursor(iter)
|
|
self._active_src_view.scroll_to_mark( self._buffer.get_insert(), 0.0, True, 0.0, 0.0 )
|
|
|
|
|
|
def find_all(self, widget):
|
|
...
|