FranklinWH-Toolbox/service/cmd/mcp/main.go

package main

import (
	"fmt"
	"log"
	"os"

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

	"fwh-toolbox-server/cache"
	"fwh-toolbox-server/config"
	"fwh-toolbox-server/database"
	"fwh-toolbox-server/logger"
	"fwh-toolbox-server/repositories"
	"fwh-toolbox-server/utils/filestore"

	"fwh-toolbox-server/cmd/mcp/tools"
	"fwh-toolbox-server/cmd/mcp/tools/storage"
)

const (
	serverName    = "fwh-tools"
	serverVersion = "2.0.0"
)

func main() {
	// 加载配置（复用现有 config）
	cfg, err := config.LoadConfig()
	if err != nil {
		log.Fatalf("加载配置失败: %v", err)
	}

	// 初始化 logger
	logger.Init(cfg.App.AppEnv)
	defer logger.Log.Sync()

	// 初始化基础设施（与 HTTP Server 相同）
	if err := database.Init(&cfg.App); err != nil {
		log.Fatalf("数据库初始化失败: %v", err)
	}
	cache.Init(cfg.App.RedisAddr)
	repositories.PlaneWebHttpInit(&cfg.PlaneWebApi)

	// 初始化 MCP FileStore
	mcpFS := filestore.NewFileStore("mcp")
	storage.InitFileStore(mcpFS)

	// 创建 MCP Server
	s := server.NewMCPServer(
		serverName,
		serverVersion,
		server.WithToolCapabilities(false),
		server.WithRecovery(),
		server.WithInstructions(serverInstructions),
	)

	// 注册所有 Tools（唯一入口）
	tools.RegisterAll(s)

	// 启动 stdio 通信
	fmt.Fprintln(os.Stderr, "[fwh-mcp-server] started (stdio) v"+serverVersion)
	if err := server.ServeStdio(s); err != nil {
		log.Fatalf("[fwh-mcp-server] error: %v", err)
	}
}

// serverInstructions MCP Server 使用说明，帮助 AI 理解工具分类和使用优先级
const serverInstructions = `FranklinWH 工具箱 MCP Server — 内部工具平台的 AI 接口层。

## 工具分类（按访问层级）

### service_* — 业务逻辑层（优先使用）
调用后端 Service 层函数，包含业务校验、数据聚合和格式化。
适用场景：常规的业务数据查询和操作。
- service_site_*: 站点设备管理（查详情、搜索、功率流）
- service_sim_*: SIM 卡管理（查详情、搜索）
- service_monitor_*: 运维监控（告警统计、停机统计）
- service_feedback_*: 反馈协作（创建/查询 Plane Issue）

### data_* — 数据直查层（灵活查询）
直接读取数据库或缓存，跳过业务逻辑层。仅限只读操作。
适用场景：当 service 层工具无法满足查询需求时（自定义 filter、聚合统计等）。
- data_mongo_*: MongoDB 集合查询
- data_redis_*: Redis 缓存查看

### storage_* — 对象存储层
读取 MinIO 对象存储或用户上传的文件。仅限只读操作。
- storage_file_*: 用户上传文件读取
- storage_minio_*: MinIO bucket 对象浏览和读取

### util_* — 辅助工具
纯函数工具，不涉及数据持久化。
- util_parse_sn: 解析设备 SN 序列号含义

## 使用原则
1. 优先使用 service_* 工具，它们包含完整的业务逻辑
2. 仅当 service 层无法满足时才使用 data_* 直查
3. 所有 data_* 和 storage_* 工具仅支持只读操作
4. 查询 MongoDB 时注意 collection 白名单限制`