Add a (b)ack option to 'Is this a valid secret?' Closes Issue #63

detect_secrets/core/audit.py

@@ -14,6 +14,7 @@
 from .color import BashColor
 from .color import Color
 from .potential_secret import PotentialSecret
+from .bidirectional_iterator import BidirectionalIterator
 
 
 class SecretNotFoundOnSpecifiedLineError(Exception):
@@ -29,7 +30,8 @@ def audit_baseline(baseline_filename):
 
     current_secret_index = 0
     results = defaultdict(list)
-    for filename, secret, total in _secret_generator(original_baseline):
+    secret_iterator = BidirectionalIterator(list(_secret_generator(original_baseline)))
+    for filename, secret, total in secret_iterator:
         _clear_screen()
 
         if 'is_secret' not in secret:
@@ -43,7 +45,7 @@ def audit_baseline(baseline_filename):
                     total,
                     original_baseline['plugins_used'],
                 )
-                decision = _get_user_decision()
+                decision = _get_user_decision(can_step_back=secret_iterator.can_step_back())
             except SecretNotFoundOnSpecifiedLineError:
                 decision = _get_user_decision(prompt_secret_decision=False)
         else:
@@ -54,6 +56,9 @@ def audit_baseline(baseline_filename):
         if decision == 'q':
             print('Quitting...')
             break
+
+        if decision == 'b':
+            secret_iterator.step_back_on_next_iteration()
 
         _handle_user_decision(decision, secret)
         results[filename].append(secret)
@@ -174,14 +179,16 @@ def _print_context(filename, secret, count, total, plugin_settings):   # pragma:
         raise error_obj
 
 
-def _get_user_decision(prompt_secret_decision=True):
+def _get_user_decision(prompt_secret_decision=True, can_step_back=False):
     """
     :type prompt_secret_decision: bool
     :param prompt_secret_decision: if False, won't ask to label secret.
     """
     allowable_user_input = ['s', 'q']
     if prompt_secret_decision:
         allowable_user_input.extend(['y', 'n'])
+    if can_step_back:
+        allowable_user_input.append('b')
 
     user_input = None
     while user_input not in allowable_user_input:
@@ -192,6 +199,8 @@ def _get_user_decision(prompt_secret_decision=True):
             user_input_string = 'Is this a valid secret? (y)es, (n)o, '
         else:
             user_input_string = 'What would you like to do? '
+        if 'b' in allowable_user_input:
+            user_input_string += '(b)ack, '
         user_input_string += '(s)kip, (q)uit: '
 
         user_input = input(user_input_string)

detect_secrets/core/bidirectional_iterator.py

@@ -0,0 +1,31 @@
+class BidirectionalIterator(object):
+    def __init__(self, collection):
+        self.collection = collection
+        self.index = -1 #starts on -1, as index is increased _before_ getting result
+        self.step_back_once = False 
+
+    def __next__(self):
+        try:
+            if self.step_back_once:
+                self.index -= 1
+                self.step_back_once = False 
+                if self.index < 0:
+                    raise StopIteration
+            else:
+                self.index += 1
+            result = self.collection[self.index]
+        except IndexError:
+            raise StopIteration
+        return result
+
+    def next(self):
+        self.__next__()
+
+    def step_back_on_next_iteration(self):
+        self.step_back_once = True
+
+    def can_step_back(self):
+        return self.index > 0
+
+    def __iter__(self):
+        return self