package server

import (
	"fmt"

	"github.com/mark3labs/mcp-go/mcp"
	"github.com/mark3labs/mcp-go/server"
	"github.com/shadow1ng/fscan/Common"
	"github.com/shadow1ng/fscan/Remote/service"
)

func StartMcpServer(transport string) {
	mcpServer := NewFscanMCPServer()

	if transport == "sse" {
		Common.LogSuccess("🚀 启动 MCP SSE 服务器，监听地址: http://localhost:8080")
		sseServer := server.NewSSEServer(mcpServer, server.WithBaseURL("http://localhost:8080"))
		if err := sseServer.Start(":8080"); err != nil {
			Common.LogError(fmt.Sprintf("❌ 启动 SSE 服务器失败: %v", err))
			panic(err)
		}
	} else {
		Common.LogSuccess("🚀 启动 MCP Stdio 服务器（标准输入输出模式）")
		if err := server.ServeStdio(mcpServer); err != nil {
			Common.LogError(fmt.Sprintf("❌ 启动 Stdio 服务器失败: %v", err))
			panic(err)
		}
	}
}

func NewFscanMCPServer() *server.MCPServer {
	s := server.NewMCPServer(
		"Fscan MCP",
		"1.0.0",
	)
	toolHandler := service.NewFscanMCPTool()
	// 添加提示词

	// 添加工具处理器
	s.AddTool(
		mcp.NewTool("StartScan",
			mcp.WithDescription("启动端口和服务扫描任务，适用于安全评估或资产排查场景。"),
			mcp.WithString("target",
				mcp.Required(),
				mcp.Description("待扫描的目标地址，支持IP、域名或CIDR格式（如192.168.1.1、example.com、10.0.0.0/24）。"),
			),
		),
		toolHandler.StartScan,
	)
	s.AddTool(
		mcp.NewTool("GetScanResults",
			mcp.WithDescription("获取当前扫描任务的执行进度和已完成部分的结果。若扫描尚未完成，也会返回当前阶段的中间结果，供用户分析或决策使用。"),
		),
		toolHandler.GetScanResults,
	)
	return s

}