feat: use regex instead

Author: 35C4n0r

lib/msgfmt/message_box.go

@@ -1,9 +1,12 @@
 package msgfmt
 
 import (
+	"regexp"
 	"strings"
 )
 
+var genericSlimMessageBoxPattern = regexp.MustCompile(`(?m)^.*─{15,}.*\n.*[|│❯].*\n(?:.*\n)?.*─{15,}.*`)
+
 // Usually something like
 // ───────────────
 // >
@@ -26,16 +29,25 @@ 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], "❯")) {
-			if (i+2 < len(lines) && strings.Contains(lines[i+2], "───────────────")) ||
-				(i+3 < len(lines) && strings.Contains(lines[i+3], "───────────────")) {
-				return i
-			}
-		}
+	// genericSlimMessageBoxPattern matches a message box pattern like:
+	// ───────────────
+	// | or │ or ❯
+	//
+	// (optional line(s))
+	// ───────────────
+
+	// Search within the last ~9 lines for the message box pattern
+	startIdx := max(len(lines)-9, 0)
+	searchText := strings.Join(lines[startIdx:], "\n")
+
+	loc := genericSlimMessageBoxPattern.FindStringIndex(searchText)
+	if loc == nil {
+		return -1
 	}
-	return -1
+
+	// Count newlines before the match to find the line number
+	linesBeforeMatch := strings.Count(searchText[:loc[0]], "\n")
+	return startIdx + linesBeforeMatch
 }
 
 func removeMessageBox(msg string) string {