feat: add logging

Author: 35C4n0r

lib/httpapi/server.go

@@ -226,7 +226,7 @@ func NewServer(ctx context.Context, config ServerConfig) (*Server, error) {
 	humaConfig.Info.Description = "HTTP API for Claude Code, Goose, and Aider.\n\nhttps://github.com/coder/agentapi"
 	api := humachi.New(router, humaConfig)
 	formatMessage := func(message string, userInput string) string {
-		return mf.FormatAgentMessage(config.AgentType, message, userInput)
+		return mf.FormatAgentMessage(config.AgentType, message, userInput, logger)
 	}
 
 	isAgentReadyForInitialPrompt := func(message string) bool {

lib/msgfmt/message_box.go

@@ -1,6 +1,7 @@
 package msgfmt
 
 import (
+	"log/slog"
 	"strings"
 )
 
@@ -101,7 +102,7 @@ func removeAmpMessageBox(msg string) string {
 	return formattedMsg
 }
 
-func removeClaudeReportTaskToolCall(msg string) string {
+func removeClaudeReportTaskToolCall(msg string, logger *slog.Logger) string {
 	// Remove all tool calls that start with `● coder - coder_report_task (MCP)` till we encounter the next line starting with ●
 	lines := strings.Split(msg, "\n")
 
@@ -144,6 +145,11 @@ func removeClaudeReportTaskToolCall(msg string) string {
 	for i := len(toolCallIdxs) - 1; i >= 0; i-- {
 		idxPair := toolCallIdxs[i]
 		start, end := idxPair[0], idxPair[1]
+
+		// Capture the tool call content before removing it
+		toolCallContent := strings.Join(lines[start:end], "\n")
+		logger.Info("Removing tool call", "content", toolCallContent)
+
 		lines = append(lines[:start], lines[end:]...)
 	}
 

lib/msgfmt/msgfmt.go

@@ -1,6 +1,7 @@
 package msgfmt
 
 import (
+	"log/slog"
 	"strings"
 )
 
@@ -254,10 +255,10 @@ func formatGenericMessage(message string, userInput string, agentType AgentType)
 	return message
 }
 
-func formatClaudeMessage(message string, userInput string) string {
+func formatClaudeMessage(message string, userInput string, logger *slog.Logger) string {
 	message = RemoveUserInput(message, userInput, AgentTypeClaude)
 	message = removeMessageBox(message)
-	message = removeClaudeReportTaskToolCall(message)
+	message = removeClaudeReportTaskToolCall(message, logger)
 	message = trimEmptyLines(message)
 	return message
 }
@@ -283,10 +284,10 @@ func formatAmpMessage(message string, userInput string) string {
 	return message
 }
 
-func FormatAgentMessage(agentType AgentType, message string, userInput string) string {
+func FormatAgentMessage(agentType AgentType, message string, userInput string, logger *slog.Logger) string {
 	switch agentType {
 	case AgentTypeClaude:
-		return formatClaudeMessage(message, userInput)
+		return formatClaudeMessage(message, userInput, logger)
 	case AgentTypeGoose:
 		return formatGenericMessage(message, userInput, agentType)
 	case AgentTypeAider:

lib/msgfmt/msgfmt_test.go

@@ -2,6 +2,7 @@ package msgfmt
 
 import (
 	"embed"
+	"log/slog"
 	"path"
 	"strings"
 	"testing"
@@ -233,7 +234,7 @@ func TestFormatAgentMessage(t *testing.T) {
 					assert.NoError(t, err)
 					expected, err := testdataDir.ReadFile(path.Join(dir, string(agentType), c.Name(), "expected.txt"))
 					assert.NoError(t, err)
-					assert.Equal(t, string(expected), FormatAgentMessage(agentType, string(msg), string(userInput)))
+					assert.Equal(t, string(expected), FormatAgentMessage(agentType, string(msg), string(userInput), slog.Default()))
 				})
 			}
 		})