FranklinWH-Toolbox/service/cmd/mcp/tools/storage/file.go

package storage

import (
	"context"
	"fmt"

	"github.com/mark3labs/mcp-go/mcp"
	"github.com/mark3labs/mcp-go/server"

	"fwh-toolbox-server/cmd/mcp/tools/helper"
	"fwh-toolbox-server/utils/filestore"
)

// mcpFileStore MCP 专用的文件存储实例
var mcpFileStore *filestore.FileStore

// InitFileStore 初始化 MCP FileStore（由 main.go 调用）
func InitFileStore(fs *filestore.FileStore) {
	mcpFileStore = fs
}

// GetFileStore 获取 MCP FileStore 实例
func GetFileStore() *filestore.FileStore {
	return mcpFileStore
}

func registerFileTools(s *server.MCPServer) {
	s.AddTool(
		mcp.NewTool("storage_file_read",
			mcp.WithDescription("读取用户上传的文件内容。用户通过前端上传的文件存储在 /uploads/ 路径下。"),
			mcp.WithString("path", mcp.Required(), mcp.Description("文件路径，格式为 /uploads/{filename}")),
		),
		handleFileRead,
	)

	s.AddTool(
		mcp.NewTool("storage_file_write",
			mcp.WithDescription("将文本内容写入文件存储（MinIO）。可用于保存分析结果、生成报告等。文件路径会自动添加 /uploads/ 前缀。"),
			mcp.WithString("path", mcp.Required(), mcp.Description("文件路径，格式为 /uploads/{filename}（如 /uploads/report.csv）")),
			mcp.WithString("content", mcp.Required(), mcp.Description("要写入的文本内容")),
			mcp.WithString("userId", mcp.Description("用户标识，用于文件路径隔离（尽量填写，不传则使用默认值 mcp-agent）")),
			mcp.WithString("sessionId", mcp.Description("会话标识，用于文件路径隔离（尽量填写，不传则使用默认值 mcp-session）")),
		),
		handleFileWrite,
	)

	s.AddTool(
		mcp.NewTool("storage_file_list",
			mcp.WithDescription("列出当前会话中用户上传的所有文件，返回文件名、路径、大小等信息。"),
		),
		handleFileList,
	)
}

func handleFileRead(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
	path, err := req.RequireString("path")
	if err != nil {
		return helper.ParamError("path"), nil
	}

	if mcpFileStore == nil {
		return helper.ServiceError("文件存储", fmt.Errorf("未初始化")), nil
	}

	if !mcpFileStore.HasFile(path) {
		return helper.EmptyResult(fmt.Sprintf("文件不存在: %s", path)), nil
	}

	content, err := mcpFileStore.ReadFile(path)
	if err != nil {
		return helper.ServiceError("读取文件", err), nil
	}

	return helper.TextResult(content), nil
}

func handleFileList(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
	if mcpFileStore == nil {
		return helper.ServiceError("文件存储", fmt.Errorf("未初始化")), nil
	}

	files := mcpFileStore.GetFiles()
	if len(files) == 0 {
		return helper.EmptyResult("当前没有上传的文件"), nil
	}

	return helper.JSONResult(files), nil
}

func handleFileWrite(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
	path, err := req.RequireString("path")
	if err != nil {
		return helper.ParamError("path"), nil
	}
	content, err := req.RequireString("content")
	if err != nil {
		return helper.ParamError("content"), nil
	}

	if mcpFileStore == nil {
		return helper.ServiceError("文件存储", fmt.Errorf("未初始化")), nil
	}

	userId := req.GetString("userId", "mcp-agent")
	sessionId := req.GetString("sessionId", "mcp-session")

	record, err := mcpFileStore.WriteFile(userId, sessionId, path, content)
	if err != nil {
		return helper.ServiceError("写入文件", err), nil
	}

	return helper.JSONResult(map[string]any{
		"message":  "文件写入成功",
		"path":     record.Path,
		"fileName": record.FileName,
		"size":     record.Size,
		"key":      record.Key,
	}), nil
}