fix: update message box formatting detection for claude (#169)

35C4n0r · web-flow · commit 93ac955ad132 · 2026-01-14T15:29:15.000+05:30
diff --git a/lib/msgfmt/message_box.go b/lib/msgfmt/message_box.go
@@ -28,7 +28,7 @@ func findGreaterThanMessageBox(lines []string) int {
 func findGenericSlimMessageBox(lines []string) int {
 	for i := len(lines) - 3; i >= max(len(lines)-9, 0); i-- {
 		if strings.Contains(lines[i], "───────────────") &&
-			(strings.Contains(lines[i+1], "|") || strings.Contains(lines[i+1], "│")) &&
+			(strings.Contains(lines[i+1], "|") || strings.Contains(lines[i+1], "│") || strings.Contains(lines[i+1], "❯")) &&
 			strings.Contains(lines[i+2], "───────────────") {
 			return i
 		}
diff --git a/lib/msgfmt/testdata/format/claude/auto-accept-edits/msg.txt b/lib/msgfmt/testdata/format/claude/auto-accept-edits/msg.txt
@@ -13,8 +13,8 @@
  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                             
diff --git a/lib/msgfmt/testdata/format/claude/first_message/msg.txt b/lib/msgfmt/testdata/format/claude/first_message/msg.txt
@@ -13,7 +13,7 @@
  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      
                                                                                 
-╭──────────────────────────────────────────────────────────────────────────────╮
-│ > Try "refactor handler.go"                                                  │
-╰──────────────────────────────────────────────────────────────────────────────╯
+───────────────────────────────────────────────────────────────────
+❯
+───────────────────────────────────────────────────────────────────
   ? for shortcuts                                                               
diff --git a/lib/msgfmt/testdata/format/claude/multi-line-input/msg.txt b/lib/msgfmt/testdata/format/claude/multi-line-input/msg.txt
@@ -26,5 +26,6 @@
 ⏺ This code is from                                                             
   /Users/hugodutka/dev/agentapi/chat/src/components/ChatInterface.tsx          
                                                                                 
-╭──────────────────────────────────────────────────────────────────────────────╮
-│ >                                                                            │
+───────────────────────────────────────────────────────────────────
+❯
+───────────────────────────────────────────────────────────────────
diff --git a/lib/msgfmt/testdata/format/claude/remove-task-tool-call/msg.txt b/lib/msgfmt/testdata/format/claude/remove-task-tool-call/msg.txt
@@ -73,8 +73,8 @@
      }
 
 
-──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────
-> 
-──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────
+───────────────────────────────────────────────────────────────────
+❯
+───────────────────────────────────────────────────────────────────
   ? for shortcuts                                                                                                                                                                            ⧉ In message_box.go
 
diff --git a/lib/msgfmt/testdata/format/claude/second_message/msg.txt b/lib/msgfmt/testdata/format/claude/second_message/msg.txt
@@ -17,7 +17,7 @@
                                                                                 
 ⏺ I'm doing well! How can I help you with your coding project today?            
                                                                                 
-╭──────────────────────────────────────────────────────────────────────────────╮
-│ >                                                                            │
-╰──────────────────────────────────────────────────────────────────────────────╯
+───────────────────────────────────────────────────────────────────
+❯
+───────────────────────────────────────────────────────────────────
   ? for shortcuts                                                               

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ func findGreaterThanMessageBox(lines []string) int {`
`28`	`28`	`func findGenericSlimMessageBox(lines []string) int {`
`29`	`29`	`for i := len(lines) - 3; i >= max(len(lines)-9, 0); i-- {`
`30`	`30`	`if strings.Contains(lines[i], "───────────────") &&`
`31`		`- (strings.Contains(lines[i+1], "\|") \|\| strings.Contains(lines[i+1], "│")) &&`
	`31`	`+ (strings.Contains(lines[i+1], "\|") \|\| strings.Contains(lines[i+1], "│") \|\| strings.Contains(lines[i+1], "❯")) &&`
`32`	`32`	`strings.Contains(lines[i+2], "───────────────") {`
`33`	`33`	`return i`
`34`	`34`	`}`
Original file line number	Diff line number	Diff line change
`@@ -73,8 +73,8 @@`
`73`	`73`	`}`
`74`	`74`
`75`	`75`
`76`		-──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────
`77`		`->`
`78`		-──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────
	`76`	`+───────────────────────────────────────────────────────────────────`
	`77`	`+❯`
	`78`	`+───────────────────────────────────────────────────────────────────`
`79`	`79`	`? for shortcuts ⧉ In message_box.go`
`80`	`80`