mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-07-13 21:02:44 +08:00
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
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("扫描目标"),
|
|
),
|
|
),
|
|
toolHandler.StartScan,
|
|
)
|
|
s.AddTool(
|
|
mcp.NewTool("GetScanResults",
|
|
mcp.WithDescription("获取扫描结果"),
|
|
),
|
|
toolHandler.GetScanResults,
|
|
)
|
|
return s
|
|
|
|
}
|