package main

import (
	"context"
	"fmt"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/mark3labs/mcp-go/mcp"
	"github.com/mark3labs/mcp-go/server"
)

func main() {
	// 1. Create MCP Server
	mcpSrv := server.NewMCPServer(
		"GoFrame MCP Demo",
		"1.0.0",
	)

	// 2. Add an addition tool
	tool := mcp.NewTool("add",
		mcp.WithDescription("Add two numbers"),
		mcp.WithNumber("a", mcp.Required(), mcp.Description("First number")),
		mcp.WithNumber("b", mcp.Required(), mcp.Description("Second number")),
	)

	mcpSrv.AddTool(tool, func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
		a, err := request.RequireFloat("a")
		if err != nil {
			return mcp.NewToolResultError("参数a错误: " + err.Error()), nil
		}
		b, err := request.RequireFloat("b")
		if err != nil {
			return mcp.NewToolResultError("参数b错误: " + err.Error()), nil
		}
		result := a + b
		return mcp.NewToolResultText(fmt.Sprintf("结果: %v + %v = %v", a, b, result)), nil
	})

	// 3. Create SSE Server (Transport Layer)
	sseSrv := server.NewSSEServer(mcpSrv, server.WithBaseURL("http://localhost:8080"))

	// 4. Start GoFrame Server
	s := g.Server()
	s.SetPort(8080)

	// 5. Bind MCP Handlers using ghttp.WrapH
	// /sse endpoint for the initial connection
	// The SSE server provides SSEHandler and MessageHandler which return http.Handler
	s.BindHandler("/sse", ghttp.WrapH(sseSrv.SSEHandler()))
	// /message endpoint for client messages (POST)
	s.BindHandler("/message", ghttp.WrapH(sseSrv.MessageHandler()))

	s.Run()
}