From a2fc4e90a835c154c44ee448e24ca6f94452446a Mon Sep 17 00:00:00 2001
From: Vipul Cariappa <vipulcariappa@gmail.com>
Date: Thu, 3 Oct 2024 20:26:40 +0530
Subject: [PATCH] ability to use arrow keys to navigate through previous
 commands (#37)

---
 idd.py | 83 +++++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 77 insertions(+), 6 deletions(-)

diff --git a/idd.py b/idd.py
index db1d075..d8c83a5 100755
--- a/idd.py
+++ b/idd.py
@@ -2,6 +2,7 @@
 import sys
 
 from textual import on
+from textual import events
 from textual.app import App, ComposeResult
 from textual.reactive import Reactive
 from textual.widgets import Input
@@ -50,6 +51,12 @@ def __init__(self, disable_asm=False, disable_registers=False):
         super().__init__()
         self.disable_asm = disable_asm
         self.disable_registers = disable_registers
+        self.common_history = [""]
+        self.common_history_index = 0
+        self.base_history = [""]
+        self.base_history_index = 0
+        self.regressed_history = [""]
+        self.regressed_history_index = 0
 
     async def set_command_result(self, version) -> None:
         state = Debugger.get_state(version)
@@ -282,9 +289,20 @@ async def execute_debugger_command(self, event: Input.Changed) -> None:
                 self.diff_area1.append([self.parallel_command_bar.value])
                 self.diff_area2.append([self.parallel_command_bar.value])
 
-                await self.set_common_command_result(result)
+                # append to history
+                self.common_history.append(self.parallel_command_bar.value)
+                self.common_history_index = len(self.common_history)
+            
+            else:
+                # execute last command from history
+                result = Debugger.run_parallel_command(self.common_history[-1])
+
+                self.diff_area1.append([self.parallel_command_bar.value])
+                self.diff_area2.append([self.parallel_command_bar.value])
+
+            await self.set_common_command_result(result)
 
-                self.parallel_command_bar.value = ""
+            self.parallel_command_bar.value = ""
 
         elif event.control.id == 'base-command-bar':
             if self.base_command_bar.value != "":
@@ -292,9 +310,20 @@ async def execute_debugger_command(self, event: Input.Changed) -> None:
                 self.diff_area1.append([self.base_command_bar.value])
                 self.diff_area1.append(result)
 
-                await self.set_command_result("base")
+                # append to history
+                self.base_history.append(self.base_command_bar.value)
+                self.base_history_index = len(self.base_history)
+            
+            else:
+                # execute last command from history
+                result = Debugger.run_single_command(self.base_history[-1], "base")
+                
+                self.diff_area1.append([self.base_command_bar.value])
+                self.diff_area1.append(result)
+
+            await self.set_command_result("base")
 
-                self.base_command_bar.value = ""
+            self.base_command_bar.value = ""
 
         elif event.control.id == 'regressed-command-bar':
             if self.regressed_command_bar.value != "":
@@ -302,9 +331,51 @@ async def execute_debugger_command(self, event: Input.Changed) -> None:
                 self.diff_area2.append([self.regressed_command_bar.value])
                 self.diff_area2.append(result)
 
-                await self.set_command_result("regressed")
+                # append to history
+                self.regressed_history.append(self.regressed_command_bar.value)
+                self.regressed_history_index = len(self.regressed_history)
+            
+            else:
+                # execute last command from history
+                result = Debugger.run_single_command(self.regressed_history[-1], "regressed")
+                self.diff_area2.append([self.regressed_command_bar.value])
+                self.diff_area2.append(result)
+
+            await self.set_command_result("regressed")
+    
+            self.regressed_command_bar.value = ""
+
+    async def on_key(self, event: events.Key) -> None:
+        if self.focused.id == "parallel-command-bar":
+            if event.key == "up":
+                self.common_history_index = (self.common_history_index - 1) % len(self.common_history)
+            elif event.key == "down":
+                self.common_history_index = (self.common_history_index + 1) % len(self.common_history)
+            else:
+                return
+            
+            self.parallel_command_bar.value = self.common_history[self.common_history_index]
+
+        elif self.focused.id == "base-command-bar":
+            if event.key == "up":
+                self.base_history_index = (self.base_history_index - 1) % len(self.base_history)
+            elif event.key == "down":
+                self.base_history_index = (self.base_history_index + 1) % len(self.base_history)
+            else:
+                return
+            
+            self.base_command_bar.value = self.base_history[self.base_history_index]
+
+        elif self.focused.id == "regressed-command-bar":
+            if event.key == "up":
+                self.regressed_history_index = (self.regressed_history_index - 1) % len(self.regressed_history)
+            elif event.key == "down":
+                self.regressed_history_index = (self.regressed_history_index + 1) % len(self.regressed_history)
+            else:
+                return
+            
+            self.regressed_command_bar.value = self.regressed_history[self.regressed_history_index]
 
-                self.regressed_command_bar.value = ""
 
 if __name__ == "__main__":
     Debugger = None