feat: update logic

Author: 35C4n0r

lib/msgfmt/message_box.go

@@ -106,6 +106,7 @@ func removeClaudeReportTaskToolCall(msg string) string {
 	lines := strings.Split(msg, "\n")
 
 	toolCallStartIdx := -1
+	newLineAfterToolCallIdx := -1
 
 	// Store all tool call start and end indices [[start, end], ...]
 	var toolCallIdxs [][]int
@@ -121,7 +122,17 @@ func removeClaudeReportTaskToolCall(msg string) string {
 
 			// Reset to find the next tool call
 			toolCallStartIdx = -1
+			newLineAfterToolCallIdx = -1
 		}
+		if len(line) == 0 && toolCallStartIdx != -1 && newLineAfterToolCallIdx == -1 {
+			newLineAfterToolCallIdx = i
+		}
+	}
+
+	// Handle the case where the last tool call goes till the end of the message
+	// And a failsafe when the next message is not prefixed with ●
+	if toolCallStartIdx != -1 && newLineAfterToolCallIdx != -1 {
+		toolCallIdxs = append(toolCallIdxs, []int{toolCallStartIdx, newLineAfterToolCallIdx})
 	}
 
 	// If no tool calls found, return original message

lib/msgfmt/testdata/format/claude/remove-task-tool-call/msg.txt

@@ -54,4 +54,19 @@
   How to play:
   Open the HTML file in your web browser and use the arrow keys
    to move the snake. Collect the red food to grow and increase
-   your score!
+   your score!
+
+● coder - coder_report_task (MCP)(summary: "Snake game created
+                                 successfully at snake-game.html",
+                                 link: "file:///home/coder/snake-ga
+                                 me.html", state: "working")
+  ⎿  {
+       "message": "Thanks for reporting!"
+     }
+
+
+──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────
+> 
+──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────
+  ? for shortcuts                                                                                                                                                                            ⧉ In message_box.go
+