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

// fwh-agent-mcp：新 AI 助手的业务数据 MCP 服务（Streamable HTTP）。
//
//	fwh-agent-mcp -set=readonly -listen=127.0.0.1:8091   # 只读工具集
//	fwh-agent-mcp -set=ops      -listen=127.0.0.1:8092   # 写工具集
//
// 配置读取独立的 ./config.agent-mcp.env（-config 可指定），不读主服务的 config.env；AGENT_MCP_TOKEN 必填。
package main

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

	"go.uber.org/zap"

	"fwh-toolbox-server/agentmcp"
)

func main() {
	set := flag.String("set", agentmcp.SetReadOnly, "工具集：readonly | ops")
	listen := flag.String("listen", "", "监听地址，默认取 AGENT_MCP_LISTEN（127.0.0.1:8091）")
	configFile := flag.String("config", agentmcp.DefaultConfigFile, "配置文件路径")
	flag.Parse()

	// 配置文件缺失时直接退出：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))
	}

	s, tasks, err := agentmcp.NewServer(*set, log, settings.MaxConcurrency)
	if err != nil {
		log.Fatal("[agent-mcp] 创建 MCP Server 失败", zap.Error(err))
	}

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

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