♻️ Refactor logger in mcp

issue #64

Author: OpenWaygate

cmd/activity/list.go

@@ -3,12 +3,14 @@ package activity
 import (
 	"bytes"
 	"context"
+	"io"
+	"log/slog"
+
 	"github.com/eat-pray-ai/yutu/cmd"
 	"github.com/eat-pray-ai/yutu/pkg/activity"
 	"github.com/eat-pray-ai/yutu/pkg/utils"
 	"github.com/mark3labs/mcp-go/mcp"
 	"github.com/spf13/cobra"
-	"io"
 )
 
 var defaultParts = []string{"id", "snippet", "contentDetails"}
@@ -122,11 +124,22 @@ func listHandler(
 	output, _ = args["output"].(string)
 	jpath, _ = args["jsonpath"].(string)
 
+	slog.InfoContext(ctx, "activity list started")
+
 	var writer bytes.Buffer
 	err := list(&writer)
 	if err != nil {
+		slog.ErrorContext(
+			ctx, "activity list failed",
+			"error", err,
+			"args", args,
+		)
 		return mcp.NewToolResultError(err.Error()), err
 	}
+	slog.InfoContext(
+		ctx, "activity list completed successfully",
+		"resultSize", writer.Len(),
+	)
 	return mcp.NewToolResultText(writer.String()), nil
 }
 

cmd/constants.go

@@ -1,5 +1,10 @@
 package cmd
 
+import (
+	"log/slog"
+	"os"
+)
+
 const (
 	PartsUsage  = "Comma separated parts"
 	MRUsage     = "The maximum number of items that should be returned"
@@ -8,3 +13,17 @@ const (
 	JPUsage     = "JSONPath expression to filter the output"
 	JsonMIME    = "application/json"
 )
+
+var Logger *slog.Logger
+
+func init() {
+	Logger = slog.New(
+		slog.NewTextHandler(
+			os.Stdout, &slog.HandlerOptions{
+				Level: slog.LevelInfo,
+			},
+		),
+	)
+
+	slog.SetDefault(Logger)
+}

cmd/mcp.go

@@ -1,11 +1,14 @@
 package cmd
 
 import (
+	"context"
 	"fmt"
+	"log/slog"
+	"os"
+	"time"
+
 	"github.com/mark3labs/mcp-go/server"
 	"github.com/spf13/cobra"
-	"log"
-	"time"
 )
 
 const (
@@ -23,6 +26,7 @@ var (
 var MCP = server.NewMCPServer(
 	"yutu", Version,
 	server.WithToolCapabilities(true),
+	server.WithResourceCapabilities(true, true),
 	server.WithLogging(),
 	server.WithRecovery(),
 )
@@ -32,10 +36,15 @@ var mcpCmd = &cobra.Command{
 	Short: mcpShort,
 	Long:  mcpLong,
 	Run: func(cmd *cobra.Command, args []string) {
+		ctx := context.Background()
 		var err error
 		interval := 13 * time.Second
 		addr := fmt.Sprintf(":%d", port)
-		message := fmt.Sprintf("%s server listening on %s", mode, addr)
+		slog.InfoContext(
+			ctx, "starting MCP server",
+			"mode", mode,
+			"version", Version,
+		)
 
 		switch mode {
 		case "stdio":
@@ -45,12 +54,29 @@ var mcpCmd = &cobra.Command{
 				MCP,
 				server.WithHeartbeatInterval(interval),
 			)
-			log.Printf("%s/%s\n", message, "mcp")
+
+			slog.InfoContext(
+				ctx, "http server configuration",
+				"url", fmt.Sprintf("http://localhost:%d/mcp", port),
+				"heartbeat_interval", interval,
+			)
 			err = httpServer.Start(addr)
+		default:
+			slog.ErrorContext(
+				ctx, "invalid server mode",
+				"mode", mode,
+				"valid_modes", []string{"stdio", "http"},
+			)
+			os.Exit(1)
 		}
 
 		if err != nil {
-			log.Printf("Server error: %v\n", err)
+			slog.ErrorContext(
+				ctx, "starting server failed",
+				"error", err,
+				"mode", mode,
+			)
+			os.Exit(1)
 		}
 	},
 }