feat: more mcp tools (#101)

* feat: add support for -header in mcp client

* feat: add more mcp tools

* feat: script rules

Author: skynet2

.gitignore

@@ -41,3 +41,5 @@ build/.aws-sam/
 server
 
 bin/
+
+mcp-client

cmd/mcp-client/main.go

@@ -6,6 +6,7 @@ import (
 	"log"
 	"os"
 	"os/signal"
+	"strings"
 	"syscall"
 
 	"github.com/mark3labs/mcp-go/client"
@@ -14,9 +15,23 @@ import (
 	"github.com/mark3labs/mcp-go/server"
 )
 
+type headerFlags []string
+
+func (h *headerFlags) String() string {
+	return strings.Join(*h, ", ")
+}
+
+func (h *headerFlags) Set(value string) error {
+	*h = append(*h, value)
+	return nil
+}
+
 func main() {
+	var headers headerFlags
+
 	serverURL := flag.String("server", "http://localhost:8080/mcp/", "Go Money MCP server URL")
 	token := flag.String("token", "", "Service token for authentication (required)")
+	flag.Var(&headers, "header", "Additional HTTP header in 'Key: Value' format (can be specified multiple times)")
 	flag.Parse()
 
 	if *token == "" {
@@ -33,11 +48,21 @@ func main() {
 		cancel()
 	}()
 
+	httpHeaders := map[string]string{
+		"Authorization": "Bearer " + *token,
+	}
+
+	for _, h := range headers {
+		parts := strings.SplitN(h, ":", 2)
+		if len(parts) != 2 {
+			log.Fatalf("invalid header format %q, expected 'Key: Value'", h)
+		}
+		httpHeaders[strings.TrimSpace(parts[0])] = strings.TrimSpace(parts[1])
+	}
+
 	httpTransport, err := transport.NewStreamableHTTP(
 		*serverURL,
-		transport.WithHTTPHeaders(map[string]string{
-			"Authorization": "Bearer " + *token,
-		}),
+		transport.WithHTTPHeaders(httpHeaders),
 	)
 	if err != nil {
 		log.Fatalf("failed to create transport: %v", err)

cmd/server/main.go

@@ -104,26 +104,6 @@ func main() {
 		grpcServer.GetMux().Handle("/", handlers.SpaHandler(config.StaticFilesDirectory))
 	}
 
-	if !config.MCP.Disable {
-		logger.Info().Str("path", config.MCP.DocsDir).Msg("Reading mcp docs")
-		mcpDocs, mcpErr := gomoneyMcp.ReadDocsFromPath(config.MCP.DocsDir)
-		if mcpErr != nil {
-			logger.Fatal().Err(mcpErr).Msg("failed to read mcp docs")
-		}
-
-		if mcpDocs == "" {
-			logger.Fatal().Msg("mcp docs are empty")
-		}
-
-		mcpServer := gomoneyMcp.NewServer(&gomoneyMcp.ServerConfig{
-			DB:   database.GetDb(database.DbTypeReadonly),
-			Docs: mcpDocs,
-		})
-
-		grpcServer.GetMux().Handle("/mcp", middlewares.HTTPAuthMiddleware(jwtService, mcpServer.Handler()))
-		logger.Info().Msg("MCP server enabled at /mcp")
-	}
-
 	userService := users.NewService(&users.ServiceConfig{
 		JwtSvc: jwtService,
 	})
@@ -225,6 +205,29 @@ func main() {
 		AccountSvc:     accountSvc,
 	})
 
+	if !config.MCP.Disable {
+		logger.Info().Str("path", config.MCP.DocsDir).Msg("Reading mcp docs")
+		mcpDocs, mcpErr := gomoneyMcp.ReadDocsFromPath(config.MCP.DocsDir)
+		if mcpErr != nil {
+			logger.Fatal().Err(mcpErr).Msg("failed to read mcp docs")
+		}
+
+		if mcpDocs == "" {
+			logger.Fatal().Msg("mcp docs are empty")
+		}
+
+		mcpServer := gomoneyMcp.NewServer(&gomoneyMcp.ServerConfig{
+			DB:          database.GetDb(database.DbTypeMaster),
+			Docs:        mcpDocs,
+			CategorySvc: categoriesSvc,
+			RulesSvc:    rulesSvc,
+			DryRunSvc:   dryRunSvc,
+		})
+
+		grpcServer.GetMux().Handle("/mcp", middlewares.HTTPAuthMiddleware(jwtService, mcpServer.Handler()))
+		logger.Info().Msg("MCP server enabled at /mcp")
+	}
+
 	_ = handlers.NewTransactionApi(grpcServer, transactionSvc, applicableAccountSvc, mapper)
 	_ = handlers.NewTagsApi(grpcServer, tagSvc)
 	_ = handlers.NewRulesApi(grpcServer, &handlers.RulesApiConfig{

pkg/mcp/category_tool.go

@@ -0,0 +1,67 @@
+package mcp
+
+import (
+	"context"
+	"fmt"
+
+	categoriesv1 "buf.build/gen/go/xskydev/go-money-pb/protocolbuffers/go/gomoneypb/categories/v1"
+	gomoneypbv1 "buf.build/gen/go/xskydev/go-money-pb/protocolbuffers/go/gomoneypb/v1"
+	"github.com/ft-t/go-money/pkg/database"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+func (s *Server) handleCreateCategory(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	args := request.GetArguments()
+
+	name, ok := args["name"].(string)
+	if !ok || name == "" {
+		return mcp.NewToolResultError("name parameter is required"), nil
+	}
+
+	queryCtx, cancel := context.WithTimeout(ctx, queryTimeout)
+	defer cancel()
+
+	queryCtx = database.WithContext(queryCtx, s.db)
+
+	resp, err := s.cfg.CategorySvc.CreateCategory(queryCtx, &categoriesv1.CreateCategoryRequest{
+		Category: &gomoneypbv1.Category{
+			Name: name,
+		},
+	})
+	if err != nil {
+		return mcp.NewToolResultError(fmt.Sprintf("failed to create category: %v", err)), nil
+	}
+
+	return mcp.NewToolResultText(fmt.Sprintf("Category created with id %d", resp.Category.Id)), nil
+}
+
+func (s *Server) handleUpdateCategory(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	args := request.GetArguments()
+
+	categoryID, ok := args["id"].(float64)
+	if !ok {
+		return mcp.NewToolResultError("id parameter is required"), nil
+	}
+
+	name, ok := args["name"].(string)
+	if !ok || name == "" {
+		return mcp.NewToolResultError("name parameter is required"), nil
+	}
+
+	queryCtx, cancel := context.WithTimeout(ctx, queryTimeout)
+	defer cancel()
+
+	queryCtx = database.WithContext(queryCtx, s.db)
+
+	resp, err := s.cfg.CategorySvc.UpdateCategory(queryCtx, &categoriesv1.UpdateCategoryRequest{
+		Category: &gomoneypbv1.Category{
+			Id:   int32(categoryID),
+			Name: name,
+		},
+	})
+	if err != nil {
+		return mcp.NewToolResultError(fmt.Sprintf("failed to update category: %v", err)), nil
+	}
+
+	return mcp.NewToolResultText(fmt.Sprintf("Category %d updated to '%s'", resp.Category.Id, resp.Category.Name)), nil
+}