From 1df4adac5caf0b999b22d026a2952b758cde426f Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Tue, 9 Mar 2021 20:27:16 +0530
Subject: [PATCH] Edit book: Add a command line flag to allow selecting the
 specified text when opening a book.

---
 src/calibre/gui2/tweak_book/boss.py        | 25 ++++++++++++++++++++--
 src/calibre/gui2/tweak_book/main.py        |  3 ++-
 src/calibre/gui2/tweak_book/text_search.py | 16 ++++++++------
 3 files changed, 35 insertions(+), 9 deletions(-)

diff --git a/src/calibre/gui2/tweak_book/boss.py b/src/calibre/gui2/tweak_book/boss.py
index 75ab6a1d3e..cf46022f86 100644
--- a/src/calibre/gui2/tweak_book/boss.py
+++ b/src/calibre/gui2/tweak_book/boss.py
@@ -286,7 +286,7 @@ class Boss(QObject):
                 return get_container(dest, tdir=tdir)
             self.gui.blocking_job('import_book', _('Importing book, please wait...'), self.book_opened, func, src, dest, tdir=self.mkdtemp())
 
-    def open_book(self, path=None, edit_file=None, clear_notify_data=True, open_folder=False):
+    def open_book(self, path=None, edit_file=None, clear_notify_data=True, open_folder=False, search_text=None):
         '''
         Open the e-book at ``path`` for editing. Will show an error if the e-book is not in a supported format or the current book has unsaved changes.
 
@@ -336,13 +336,15 @@ class Boss(QObject):
         # temp file cleaners from nuking ebooks. See https://bugs.launchpad.net/bugs/1740460
         self.tdir = tdir_in_cache('ee')
         self._edit_file_on_open = edit_file
+        self._search_text_on_open = search_text
         self._clear_notify_data = clear_notify_data
         self.gui.blocking_job('open_book', _('Opening book, please wait...'), self.book_opened, get_container, path, tdir=self.mkdtemp())
 
     def book_opened(self, job):
         ef = getattr(self, '_edit_file_on_open', None)
         cn = getattr(self, '_clear_notify_data', True)
-        self._edit_file_on_open = None
+        st = getattr(self, '_search_text_on_open', None)
+        self._edit_file_on_open = self._search_text_on_open = None
 
         if job.traceback is not None:
             if 'DRMError:' in job.traceback:
@@ -396,6 +398,8 @@ class Boss(QObject):
                     self.restore_book_edit_state()
             self.gui.toc_view.update_if_visible()
             self.add_savepoint(_('Start of editing session'))
+            if st:
+                self.find_initial_text(st)
 
     def update_editors_from_container(self, container=None, names=None):
         c = container or current_container()
@@ -1051,6 +1055,23 @@ class Boss(QObject):
         if getattr(ed, 'has_line_numbers', False):
             ed.editor.setFocus(Qt.FocusReason.OtherFocusReason)
 
+    def find_initial_text(self, text):
+        from calibre.gui2.tweak_book.search import get_search_regex
+        from calibre.gui2.tweak_book.text_search import file_matches_pattern
+        search = {'find': text, 'mode': 'normal', 'case_sensitive': True, 'direction': 'down'}
+        pat = get_search_regex(search)
+        searchable_names = set(self.gui.file_list.searchable_names['text'])
+        for name, ed in iteritems(editors):
+            searchable_names.discard(name)
+            if ed.find_text(pat, complete=True):
+                self.show_editor(name)
+                return
+        for name in searchable_names:
+            if file_matches_pattern(name, pat):
+                self.edit_file(name)
+                if editors[name].find_text(pat, complete=True):
+                    return
+
     def find_word(self, word, locations):
         # Go to a word from the spell check dialog
         ed = self.gui.central.current_editor
diff --git a/src/calibre/gui2/tweak_book/main.py b/src/calibre/gui2/tweak_book/main.py
index be2e092ecc..d89d4d339e 100644
--- a/src/calibre/gui2/tweak_book/main.py
+++ b/src/calibre/gui2/tweak_book/main.py
@@ -34,6 +34,7 @@ files inside the book which will be opened for editing automatically.
         )
     )
     setup_gui_option_parser(parser)
+    parser.add_option('--select-text', default=None, help=_('The text to select on open'))
     return parser
 
 
@@ -79,7 +80,7 @@ def _run(args, notify=None):
     main.show()
     app.shutdown_signal_received.connect(main.boss.quit)
     if len(args) > 1:
-        main.boss.open_book(args[1], edit_file=args[2:], clear_notify_data=False)
+        main.boss.open_book(args[1], edit_file=args[2:], clear_notify_data=False, search_text=opts.select_text)
     else:
         for path in reversed(app.file_event_hook.events):
             main.boss.open_book(path)
diff --git a/src/calibre/gui2/tweak_book/text_search.py b/src/calibre/gui2/tweak_book/text_search.py
index f3bf749edb..13feed86cd 100644
--- a/src/calibre/gui2/tweak_book/text_search.py
+++ b/src/calibre/gui2/tweak_book/text_search.py
@@ -155,6 +155,15 @@ class TextSearch(QWidget):
 # }}}
 
 
+def file_matches_pattern(fname, pat):
+    root = current_container().parsed(fname)
+    if hasattr(root, 'xpath'):
+        raw = tostring(root, method='text', encoding='unicode', with_tail=True)
+    else:
+        raw = current_container().raw_data(fname)
+    return pat.search(raw) is not None
+
+
 def run_text_search(search, current_editor, current_editor_name, searchable_names, gui_parent, show_editor, edit_file):
     try:
         pat = get_search_regex(search)
@@ -176,12 +185,7 @@ def run_text_search(search, current_editor, current_editor_name, searchable_name
                     show_editor(fname)
                     return True
             else:
-                root = current_container().parsed(fname)
-                if hasattr(root, 'xpath'):
-                    raw = tostring(root, method='text', encoding='unicode', with_tail=True)
-                else:
-                    raw = current_container().raw_data(fname)
-                if pat.search(raw) is not None:
+                if file_matches_pattern(fname, pat):
                     edit_file(fname, syntax)
                     if editors[fname].find_text(pat, complete=True):
                         return True