// opsxdb-mcp：stdio MCP Server，按本机策略暴露数据库工具。
//
// 配置：
//   ~/.opsxdb/mcp.json          — allowed_connection_ids；mode；每次工具调用重新读取
//   ~/.opsxdb/connections.json  — 桌面双写的连接列表（密码 ox1. 混淆）
//   ~/.opsxdb/logs/mcp-audit.jsonl
//
// Cursor 示例见 docs/mcp-cursor.example.json
package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

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

	"opsxdb/server/internal/aitools"
)

func main() {
	s := server.NewMCPServer(
		"opsxdb",
		"0.1.0",
		server.WithToolCapabilities(true),
	)

	for _, spec := range aitools.ToolSpecs() {
		raw, _ := json.Marshal(spec.InputSchema())
		t := mcp.NewToolWithRawSchema(spec.Name, spec.Description, raw)
		toolName := spec.Name
		s.AddTool(t, func(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
			out, err := callTool(ctx, toolName, req.GetArguments(), aitools.DefaultRuntime)
			if err != nil {
				return mcp.NewToolResultError(err.Error()), nil
			}
			return mcp.NewToolResultText(aitools.ResultJSON(out)), nil
		})
	}

	if err := server.ServeStdio(s); err != nil {
		fmt.Fprintf(os.Stderr, "opsxdb-mcp: %v\n", err)
		os.Exit(1)
	}
}

// 每次调用使用新的策略快照；读取失败直接拒绝，不继续用上一次的权限。
func callTool(ctx context.Context, tool string, args map[string]any, load func() (*aitools.Runtime, error)) (any, error) {
	rt, err := load()
	if err != nil {
		return nil, fmt.Errorf("load policy: %w", err)
	}
	rt.Source = "mcp"
	return rt.Call(ctx, tool, args)
}