msgfmt: more robust input box removal

Author: hugodutka

lib/msgfmt/claude.go

@@ -21,16 +21,21 @@ func removeClaudeMessageBox(msg string) string {
 		}
 	}
 
-	if strings.HasPrefix(trimmedLastLine(), "? for shortcuts") {
-		popLine()
-	}
-	msgBoxEdge := "───────────────"
-	if strings.Contains(trimmedLastLine(), msgBoxEdge) {
-		popLine()
+	// The ">" symbol is often used to indicate the user input line.
+	// We remove all lines including and after the last ">" symbol
+	// in the message.
+	greaterThanLineIdx := -1
+	for i := len(lines) - 1; i >= max(len(lines)-6, 0); i-- {
+		if strings.Contains(lines[i], ">") {
+			greaterThanLineIdx = i
+			break
+		}
 	}
-	if strings.Contains(trimmedLastLine(), ">") {
-		popLine()
+	if greaterThanLineIdx >= 0 {
+		lines = lines[:greaterThanLineIdx]
 	}
+
+	msgBoxEdge := "───────────────"
 	if strings.Contains(trimmedLastLine(), msgBoxEdge) {
 		popLine()
 	}

lib/msgfmt/testdata/format/claude/auto-accept-edits/expected.txt

@@ -0,0 +1,14 @@
+╭────────────────────────────────────────────╮                                  
+│ ✻ Welcome to Claude Code research preview! │                                  
+│                                            │                                  
+│   /help for help                           │                                  
+│                                            │                                  
+│   cwd: /Users/hugodutka/dev/agentapi      │                                  
+╰────────────────────────────────────────────╯                                  
+                                                                                
+ Tips for getting started:                                                      
+                                                                                
+ 1. Run /init to create a CLAUDE.md file with instructions for Claude           
+ 2. Run /terminal-setup to set up terminal integration                          
+ 3. Use Claude to help with file analysis, editing, bash commands and git       
+ 4. Be as specific as you would with another engineer for the best results      

lib/msgfmt/testdata/format/claude/auto-accept-edits/msg.txt

@@ -0,0 +1,20 @@
+╭────────────────────────────────────────────╮                                  
+│ ✻ Welcome to Claude Code research preview! │                                  
+│                                            │                                  
+│   /help for help                           │                                  
+│                                            │                                  
+│   cwd: /Users/hugodutka/dev/agentapi      │                                  
+╰────────────────────────────────────────────╯                                  
+                                                                                
+ Tips for getting started:                                                      
+                                                                                
+ 1. Run /init to create a CLAUDE.md file with instructions for Claude           
+ 2. Run /terminal-setup to set up terminal integration                          
+ 3. Use Claude to help with file analysis, editing, bash commands and git       
+ 4. Be as specific as you would with another engineer for the best results      
+                                                                                
+╭──────────────────────────────────────────────────────────────────────────────╮
+│ >                                                                            │
+╰──────────────────────────────────────────────────────────────────────────────╯
+  ⏵⏵ auto-accept edits on (shift+tab to       ✓ Update installed • Restart to   
+  toggle)                                     apply