From 899916c0fbc0e63db28811a422f4d2973b796172 Mon Sep 17 00:00:00 2001
From: Stephen Boddy <stephen.j.boddy@gmail.com>
Date: Sun, 21 Jun 2015 17:57:14 +0200
Subject: [PATCH] (trunk-1552) Merge search bar wrap toggle from Christophe
 Bourez

---
 terminatorlib/searchbar.py | 69 ++++++++++++++++++++++++++------------
 1 file changed, 48 insertions(+), 21 deletions(-)

diff --git a/terminatorlib/searchbar.py b/terminatorlib/searchbar.py
index d20c726f..d3088405 100755
--- a/terminatorlib/searchbar.py
+++ b/terminatorlib/searchbar.py
@@ -22,6 +22,7 @@ class Searchbar(Gtk.HBox):
     reslabel = None
     next = None
     prev = None
+    wrap = None
 
     vte = None
     config = None
@@ -78,16 +79,31 @@ class Searchbar(Gtk.HBox):
         self.prev.set_sensitive(False)
         self.prev.connect('clicked', self.prev_search)
 
+        # Wrap checkbox
+        self.wrap = Gtk.CheckButton(_('Wrap'))
+        self.wrap.show()
+        self.wrap.set_sensitive(True)
+        self.wrap.connect('toggled', self.wrap_toggled)
+
         self.pack_start(label, False, True, 0)
         self.pack_start(self.entry, True, True, 0)
         self.pack_start(self.reslabel, False, True, 0)
         self.pack_start(self.prev, False, False, 0)
         self.pack_start(self.next, False, False, 0)
+        self.pack_start(self.wrap, False, False, 0)
         self.pack_end(close, False, False, 0)
 
         self.hide()
         self.set_no_show_all(True)
 
+    def wrap_toggled(self, toggled):
+        if self.searchrow is None:
+            self.prev.set_sensitive(False)
+            self.next.set_sensitive(False)
+        elif toggled:
+            self.prev.set_sensitive(True)
+            self.next.set_sensitive(True)
+
     def get_vte(self):
         """Find our parent widget"""
         parent = self.get_parent()
@@ -100,6 +116,9 @@ class Searchbar(Gtk.HBox):
         key = Gdk.keyval_name(event.keyval)
         if key == 'Escape':
             self.end_search()
+        else:
+            self.prev.set_sensitive(False)
+            self.next.set_sensitive(False)
 
     def start_search(self):
         """Show ourselves"""
@@ -116,7 +135,7 @@ class Searchbar(Gtk.HBox):
             return
 
         if searchtext != self.searchstring:
-            self.searchrow = self.get_vte_buffer_range()[0]
+            self.searchrow = self.get_vte_buffer_range()[0] - 1
             self.searchstring = searchtext
             self.searchre = re.compile(searchtext)
 
@@ -128,46 +147,54 @@ class Searchbar(Gtk.HBox):
     def next_search(self, widget):
         """Search forwards and jump to the next result, if any"""
         startrow,endrow = self.get_vte_buffer_range()
+        found = startrow <= self.searchrow and self.searchrow < endrow
+        row = self.searchrow
         while True:
-            if self.searchrow >= endrow:
-                self.searchrow = startrow
-                self.reslabel.set_text(_('No more results'))
-                return
-            buffer = self.vte.get_text_range(self.searchrow, 0,
-                                             self.searchrow+1, 0,
-                                             self.search_character)
+            row += 1
+            if row >= endrow:
+                if found and self.wrap.get_active():
+                    row = startrow - 1
+                else:
+                    self.prev.set_sensitive(found)
+                    self.next.set_sensitive(False)
+                    self.reslabel.set_text(_('No more results'))
+                    return
+            buffer = self.vte.get_text_range(row, 0, row + 1, 0, self.search_character)
 
             buffer = buffer[0]
             buffer = buffer[:buffer.find('\n')]
             matches = self.searchre.search(buffer)
             if matches:
+                self.searchrow = row
+                self.prev.set_sensitive(True)
                 self.search_hit(self.searchrow)
-                self.searchrow += 1
                 return
-            self.searchrow += 1
 
-    # FIXME: There is an issue in switching search direction, probably because
-    # we increment/decrement self.searchrow after each search iteration
     def prev_search(self, widget):
         """Jump back to the previous search"""
         startrow,endrow = self.get_vte_buffer_range()
+        found = startrow <= self.searchrow and self.searchrow < endrow
+        row = self.searchrow
         while True:
-            if self.searchrow <= startrow:
-                self.searchrow = endrow
-                self.reslabel.set_text(_('No more results'))
-                return
-            buffer = self.vte.get_text_range(self.searchrow, 0,
-                                             self.searchrow+1, 0,
-                                             self.search_character)
+            row -= 1
+            if row <= startrow:
+                if found and self.wrap.get_active():
+                    row = endrow
+                else:
+                    self.next.set_sensitive(found)
+                    self.prev.set_sensitive(False)
+                    self.reslabel.set_text(_('No more results'))
+                    return
+            buffer = self.vte.get_text_range(row, 0, row + 1, 0, self.search_character)
 
             buffer = buffer[0]
             buffer = buffer[:buffer.find('\n')]
             matches = self.searchre.search(buffer)
             if matches:
+                self.searchrow = row
+                self.next.set_sensitive(True)
                 self.search_hit(self.searchrow)
-                self.searchrow -= 1
                 return
-            self.searchrow -= 1
 
     def search_hit(self, row):
         """Update the UI for a search hit"""