Add files via upload

internal/handler/config.go

@@ -6,6 +6,8 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"strconv"
+	"strings"
 	"sync"
 
 	"cyberstrike-ai/internal/config"
@@ -91,6 +93,99 @@ func (h *ConfigHandler) GetConfig(c *gin.Context) {
 	})
 }
 
+// GetToolsResponse 获取工具列表响应（分页）
+type GetToolsResponse struct {
+	Tools      []ToolConfigInfo `json:"tools"`
+	Total      int              `json:"total"`
+	Page       int              `json:"page"`
+	PageSize   int              `json:"page_size"`
+	TotalPages int              `json:"total_pages"`
+}
+
+// GetTools 获取工具列表（支持分页和搜索）
+func (h *ConfigHandler) GetTools(c *gin.Context) {
+	h.mu.RLock()
+	defer h.mu.RUnlock()
+
+	// 解析分页参数
+	page := 1
+	pageSize := 20
+	if pageStr := c.Query("page"); pageStr != "" {
+		if p, err := strconv.Atoi(pageStr); err == nil && p > 0 {
+			page = p
+		}
+	}
+	if pageSizeStr := c.Query("page_size"); pageSizeStr != "" {
+		if ps, err := strconv.Atoi(pageSizeStr); err == nil && ps > 0 && ps <= 100 {
+			pageSize = ps
+		}
+	}
+
+	// 解析搜索参数
+	searchTerm := c.Query("search")
+	searchTermLower := ""
+	if searchTerm != "" {
+		searchTermLower = strings.ToLower(searchTerm)
+	}
+
+	// 获取所有工具并应用搜索过滤
+	allTools := make([]ToolConfigInfo, 0, len(h.config.Security.Tools))
+	for _, tool := range h.config.Security.Tools {
+		toolInfo := ToolConfigInfo{
+			Name:        tool.Name,
+			Description: tool.ShortDescription,
+			Enabled:     tool.Enabled,
+		}
+		// 如果没有简短描述，使用详细描述的前100个字符
+		if toolInfo.Description == "" {
+			desc := tool.Description
+			if len(desc) > 100 {
+				desc = desc[:100] + "..."
+			}
+			toolInfo.Description = desc
+		}
+
+		// 如果有关键词，进行搜索过滤
+		if searchTermLower != "" {
+			nameLower := strings.ToLower(toolInfo.Name)
+			descLower := strings.ToLower(toolInfo.Description)
+			if !strings.Contains(nameLower, searchTermLower) && !strings.Contains(descLower, searchTermLower) {
+				continue // 不匹配，跳过
+			}
+		}
+
+		allTools = append(allTools, toolInfo)
+	}
+
+	total := len(allTools)
+	totalPages := (total + pageSize - 1) / pageSize
+	if totalPages == 0 {
+		totalPages = 1
+	}
+
+	// 计算分页范围
+	offset := (page - 1) * pageSize
+	end := offset + pageSize
+	if end > total {
+		end = total
+	}
+
+	var tools []ToolConfigInfo
+	if offset < total {
+		tools = allTools[offset:end]
+	} else {
+		tools = []ToolConfigInfo{}
+	}
+
+	c.JSON(http.StatusOK, GetToolsResponse{
+		Tools:      tools,
+		Total:      total,
+		Page:       page,
+		PageSize:   pageSize,
+		TotalPages: totalPages,
+	})
+}
+
 // UpdateConfigRequest 更新配置请求
 type UpdateConfigRequest struct {
 	OpenAI *config.OpenAIConfig `json:"openai,omitempty"`

internal/handler/monitor.go

@@ -2,6 +2,7 @@ package handler
 
 import (
 	"net/http"
+	"strconv"
 	"time"
 
 	"cyberstrike-ai/internal/database"
@@ -34,31 +35,96 @@ type MonitorResponse struct {
 	Executions []*mcp.ToolExecution      `json:"executions"`
 	Stats      map[string]*mcp.ToolStats `json:"stats"`
 	Timestamp  time.Time                  `json:"timestamp"`
+	Total      int                        `json:"total,omitempty"`
+	Page       int                        `json:"page,omitempty"`
+	PageSize   int                        `json:"page_size,omitempty"`
+	TotalPages int                        `json:"total_pages,omitempty"`
 }
 
 // Monitor 获取监控信息
 func (h *MonitorHandler) Monitor(c *gin.Context) {
-	executions := h.loadExecutions()
+	// 解析分页参数
+	page := 1
+	pageSize := 20
+	if pageStr := c.Query("page"); pageStr != "" {
+		if p, err := strconv.Atoi(pageStr); err == nil && p > 0 {
+			page = p
+		}
+	}
+	if pageSizeStr := c.Query("page_size"); pageSizeStr != "" {
+		if ps, err := strconv.Atoi(pageSizeStr); err == nil && ps > 0 && ps <= 100 {
+			pageSize = ps
+		}
+	}
+
+	executions, total := h.loadExecutionsWithPagination(page, pageSize)
 	stats := h.loadStats()
 
+	totalPages := (total + pageSize - 1) / pageSize
+	if totalPages == 0 {
+		totalPages = 1
+	}
+
 	c.JSON(http.StatusOK, MonitorResponse{
 		Executions: executions,
 		Stats:      stats,
 		Timestamp:  time.Now(),
+		Total:      total,
+		Page:       page,
+		PageSize:   pageSize,
+		TotalPages: totalPages,
 	})
 }
 
 func (h *MonitorHandler) loadExecutions() []*mcp.ToolExecution {
+	executions, _ := h.loadExecutionsWithPagination(1, 1000)
+	return executions
+}
+
+func (h *MonitorHandler) loadExecutionsWithPagination(page, pageSize int) ([]*mcp.ToolExecution, int) {
 	if h.db == nil {
-		return h.mcpServer.GetAllExecutions()
+		allExecutions := h.mcpServer.GetAllExecutions()
+		total := len(allExecutions)
+		offset := (page - 1) * pageSize
+		end := offset + pageSize
+		if end > total {
+			end = total
+		}
+		if offset >= total {
+			return []*mcp.ToolExecution{}, total
+		}
+		return allExecutions[offset:end], total
 	}
 
-	executions, err := h.db.LoadToolExecutions()
+	offset := (page - 1) * pageSize
+	executions, err := h.db.LoadToolExecutionsWithPagination(offset, pageSize)
 	if err != nil {
 		h.logger.Warn("从数据库加载执行记录失败，回退到内存数据", zap.Error(err))
-		return h.mcpServer.GetAllExecutions()
+		allExecutions := h.mcpServer.GetAllExecutions()
+		total := len(allExecutions)
+		offset := (page - 1) * pageSize
+		end := offset + pageSize
+		if end > total {
+			end = total
+		}
+		if offset >= total {
+			return []*mcp.ToolExecution{}, total
+		}
+		return allExecutions[offset:end], total
 	}
-	return executions
+
+	// 获取总数
+	total, err := h.db.CountToolExecutions()
+	if err != nil {
+		h.logger.Warn("获取执行记录总数失败", zap.Error(err))
+		// 回退：使用已加载的记录数估算
+		total = offset + len(executions)
+		if len(executions) == pageSize {
+			total = offset + len(executions) + 1
+		}
+	}
+
+	return executions, total
 }
 
 func (h *MonitorHandler) loadStats() map[string]*mcp.ToolStats {