From c798673069e1b37f8ffdb12d0fe9742c57bd2285 Mon Sep 17 00:00:00 2001
From: verse91 <versedev.store@proton.me>
Date: Sun, 19 Apr 2026 09:59:53 +0700
Subject: [PATCH] fix(wrapper): up/down arrow hide menu

---
 root/wrapper.go | 41 ++++++++++++++++++++++-------------------
 1 file changed, 22 insertions(+), 19 deletions(-)

diff --git a/root/wrapper.go b/root/wrapper.go
index f82554d..896a161 100644
--- a/root/wrapper.go
+++ b/root/wrapper.go
@@ -47,7 +47,7 @@ func runWrapper() {
 	c.ExtraFiles[10] = w
 	c.Env = adapter.GetEnv(10, os.Getpid())
 
-	ptmx, err := pty.Start(c) 
+	ptmx, err := pty.Start(c)
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "[IRIS] failed to start PTY: %v\n", err)
 		return
@@ -199,38 +199,41 @@ func runWrapper() {
 
 				if b == '\033' {
 					// handle escape sequences like arrow keys and functional shortcuts
-					if i+2 < n && inputSlice[i+1] == '[' && inputSlice[i+2] == 'Z' {
+					if i+2 < n && (inputSlice[i+1] == '[' || inputSlice[i+1] == 'O') {
 						// shift tab: hide/unhide menu dropdown
-						intercepted = true
-						suggestionsEnabled = !suggestionsEnabled
-						if !suggestionsEnabled {
-							os.Stdout.Write([]byte(overlay.ClearAndDisable()))
-						} else {
-							shouldOverlayDraw = true
+						if inputSlice[i+1] == '[' && inputSlice[i+2] == 'Z' {
+							intercepted = true
+							suggestionsEnabled = !suggestionsEnabled
+							if !suggestionsEnabled {
+								os.Stdout.Write([]byte(overlay.ClearAndDisable()))
+							} else {
+								shouldOverlayDraw = true
+							}
+							i += 2
+							continue
 						}
-						i += 2
-						continue
-					}
 
-					ptmx.Write([]byte{b})
-					if i+2 < n && inputSlice[i+1] == '[' {
-						if overlay.Visible {
-							switch inputSlice[i+2] {
-							case 'A': // up arrow
+						if overlay.Visible && (inputSlice[i+2] == 'A' || inputSlice[i+2] == 'B') {
+							intercepted = true
+							if inputSlice[i+2] == 'A' { // up arrow
 								overlay.Cursor--
 								if overlay.Cursor < 0 {
 									overlay.Cursor = 0
 								}
-								os.Stdout.Write([]byte(overlay.Render()))
-							case 'B': // down arrow
+							} else { // down arrow
 								overlay.Cursor++
 								if overlay.Cursor >= len(overlay.Items) {
 									overlay.Cursor = len(overlay.Items) - 1
 								}
-								os.Stdout.Write([]byte(overlay.Render()))
 							}
+							os.Stdout.Write([]byte(overlay.Render()))
+							i += 2
+							continue
 						}
 					}
+
+					// forward escape sequence to pty if not intercepted
+					ptmx.Write([]byte{b})
 					// skip remaining bytes of the escape sequence to avoid misinterpretation
 					for j := i + 1; j < n; j++ {
 						char := inputSlice[j]