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

// fwh-agent-mcp：新 AI 助手的业务数据 MCP 服务（Streamable HTTP）。
//
//	fwh-agent-mcp -set=readonly,ops -listen=127.0.0.1:8091
//
// 一个进程、一个端口加载多个工具集，每个工具集挂在 /mcp/<工具集>（只读工具集另挂 /mcp 兼容旧地址）；
// -set 省略时加载全部工具集。
//
// 配置读取独立的 ./config.agent-mcp.env（-config 可指定），不读主服务的 config.env；AGENT_MCP_TOKEN 必填，
// AGENT_MCP_TOKEN_<工具集大写>（如 AGENT_MCP_TOKEN_OPS）可为单个工具集指定单独的令牌。
package main

import (
	"context"
	"flag"
	"fmt"
	"os"
	"os/signal"
	"strings"
	"syscall"

	"go.uber.org/zap"

	"fwh-toolbox-server/agentmcp"
)

func main() {
	setFlag := flag.String("set", strings.Join(agentmcp.AllSets, ","), "工具集，逗号分隔："+strings.Join(agentmcp.AllSets, ","))
	listen := flag.String("listen", "", "监听地址，默认取 AGENT_MCP_LISTEN（127.0.0.1:8091）")
	configFile := flag.String("config", agentmcp.DefaultConfigFile, "配置文件路径")
	flag.Parse()

	sets, err := agentmcp.ParseSets(*setFlag)
	if err != nil {
		fmt.Fprintf(os.Stderr, "[fwh-agent-mcp] -set 无效：%v\n", err)
		os.Exit(1)
	}

	// 配置文件缺失时直接退出：config.LoadConfig 读不到文件会静默使用默认值，连错库也不易察觉
	settings, err := agentmcp.LoadSettings(*configFile)
	if err != nil {
		fmt.Fprintf(os.Stderr, "[fwh-agent-mcp] %v，拒绝启动（示例见 deploy/config.agent-mcp.env.example）\n", err)
		os.Exit(1)
	}
	if settings.Token == "" {
		fmt.Fprintln(os.Stderr, "[fwh-agent-mcp] AGENT_MCP_TOKEN 未配置，拒绝启动")
		os.Exit(1)
	}
	addr := settings.Listen
	if *listen != "" {
		addr = *listen
	}

	log := agentmcp.InitLogger(settings.LogPath)
	defer func() { _ = log.Sync() }()

	if err := agentmcp.InitInfra(*configFile); err != nil {
		log.Fatal("[agent-mcp] 初始化失败", zap.Error(err))
	}

	endpoints := make([]agentmcp.Endpoint, 0, len(sets))
	for _, set := range sets {
		s, tasks, err := agentmcp.NewServer(set, log, settings.MaxConcurrency)
		if err != nil {
			log.Fatal("[agent-mcp] 创建 MCP Server 失败", zap.String("set", set), zap.Error(err))
		}
		endpoints = append(endpoints, agentmcp.Endpoint{Set: set, Server: s, Tasks: tasks, Token: settings.TokenFor(set)})
		log.Info("[agent-mcp] 加载工具集",
			zap.String("set", set),
			zap.String("path", agentmcp.EndpointPath(set)),
			zap.Bool("ownToken", settings.SetTokens[set] != ""))
	}

	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
	defer stop()

	log.Info("[agent-mcp] 启动", zap.Strings("sets", sets), zap.String("version", agentmcp.ServerVersion), zap.Int("maxConcurrency", settings.MaxConcurrency))
	if err := agentmcp.Serve(ctx, addr, agentmcp.Handler(endpoints, log), log); err != nil {
		log.Fatal("[agent-mcp] HTTP 服务异常退出", zap.Error(err))
	}
	log.Info("[agent-mcp] 已退出")
}