package admin
import (
"context"
"github.com/weibaohui/k8m/internal/dao"
"github.com/weibaohui/k8m/pkg/comm/utils"
"github.com/weibaohui/k8m/pkg/comm/utils/amis"
"github.com/weibaohui/k8m/pkg/plugins/modules/mcp_runtime/models"
"github.com/weibaohui/k8m/pkg/plugins/modules/mcp_runtime/service"
"github.com/weibaohui/k8m/pkg/response"
"k8s.io/klog/v2"
)
type ServerController struct {
}
// @Summary 获取MCP服务器列表
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/mcp/list [get]
func (m *ServerController) List(c *response.Context) {
params := dao.BuildParams(c)
var mcpServer models.MCPServerConfig
list, count, err := mcpServer.List(params)
amis.WriteJsonListTotalWithError(c, count, list, err)
}
// @Summary 连接指定MCP服务器
// @Security BearerAuth
// @Param name path string true "MCP服务器名称"
// @Success 200 {object} string
// @Router /admin/mcp/connect/{name} [post]
func (m *ServerController) Connect(c *response.Context) {
name := c.Param("name")
ctx := amis.GetContextWithUser(c)
err := service.McpService().Host().ConnectServer(ctx, name)
amis.WriteJsonErrorOrOK(c, err)
}
// @Summary 删除MCP服务器
// @Security BearerAuth
// @Param request body object true "删除请求体包含IDs数组"
// @Success 200 {object} string
// @Router /admin/mcp/delete [post]
func (m *ServerController) Delete(c *response.Context) {
var req struct {
IDs []int `json:"ids"`
}
if err := c.ShouldBindJSON(&req); err != nil {
amis.WriteJsonError(c, err)
return
}
var servers []models.MCPServerConfig
dao.DB().Where("id in?", req.IDs).Find(&servers)
// 删除
dao.DB().Where("id in ?", req.IDs).Delete(&models.MCPServerConfig{})
for _, server := range servers {
service.McpService().RemoveServer(server)
}
amis.WriteJsonOK(c)
}
// @Summary 创建或更新MCP服务器
// @Security BearerAuth
// @Param request body models.MCPServerConfig true "MCP服务器配置信息"
// @Success 200 {object} string
// @Router /admin/mcp/save [post]
func (m *ServerController) Save(c *response.Context) {
params := dao.BuildParams(c)
var entity models.MCPServerConfig
if err := c.ShouldBindJSON(&entity); err != nil {
amis.WriteJsonError(c, err)
return
}
err := entity.Save(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
ctx := amis.GetContextWithUser(c)
service.McpService().UpdateServer(ctx, entity)
removeTools(entity)
addTools(ctx, params, entity)
amis.WriteJsonErrorOrOK(c, err)
}
// @Summary 快速更新MCP服务器状态
// @Security BearerAuth
// @Param id path int true "MCP服务器ID"
// @Param status path string true "服务器状态(true/false)"
// @Success 200 {object} string
// @Router /admin/mcp/save/id/{id}/status/{status} [post]
func (m *ServerController) QuickSave(c *response.Context) {
id := c.Param("id")
status := c.Param("status")
params := dao.BuildParams(c)
var entity models.MCPServerConfig
err := dao.DB().Where("id = ?", id).First(&entity).Error
if err != nil {
amis.WriteJsonError(c, err)
return
}
if status == "true" {
entity.Enabled = true
} else {
entity.Enabled = false
}
err = entity.Save(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
removeTools(entity)
ctx := amis.GetContextWithUser(c)
service.McpService().UpdateServer(ctx, entity)
if status == "true" {
addTools(ctx, params, entity)
}
amis.WriteJsonErrorOrOK(c, err)
}
// @Summary 获取MCP服务器日志列表
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/mcp/log/list [get]
func (m *ServerController) MCPLogList(c *response.Context) {
params := dao.BuildParams(c)
var tool models.MCPToolLog
list, count, err := tool.List(params)
amis.WriteJsonListTotalWithError(c, count, list, err)
}
func addTools(ctx context.Context, params *dao.Params, entity models.MCPServerConfig) bool {
// 获取Tools列表
if tools, err := service.McpService().GetTools(ctx, entity); err == nil {
for _, tool := range tools {
mt := models.MCPTool{
ServerName: entity.Name,
Name: tool.Name,
Description: tool.Description,
InputSchema: utils.ToJSON(tool.InputSchema),
Enabled: true,
}
err = mt.Save(params)
if err != nil {
klog.V(6).Infof("保存工具失败:[%s/%s] %v\n", entity.Name, tool.Name, err)
return true
}
}
}
return false
}
func removeTools(entity models.MCPServerConfig) {
dao.DB().Where("server_name = ?", entity.Name).Delete(&models.MCPTool{})
}
