Logan.LiandCursor
2fd1624bec
收紧 SSH 命令护栏、权限校验与远程执行会话，并同步主流大模型供应商列表。 Co-authored-by: Cursor <cursoragent@cursor.com>
94 lines
3.3 KiB
Go
94 lines
3.3 KiB
Go
// agent_cli_api.go - Agent CLI 会话历史 REST 接口 // Logan.Li
|
|
// 供前端展示历史任务列表与回放某次任务的完整推理过程。
|
|
package controller
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"ezops/middleware"
|
|
"ezops/model"
|
|
"ezops/pkg/agentcli"
|
|
)
|
|
|
|
// ListSessions 会话列表（按当前用户隔离） // Logan.Li
|
|
// GET /api/v1/aiops/agent-cli/sessions?mode=&server_id=&cluster_id=&namespace=&page=&page_size=
|
|
func (h *AgentCLIHandler) ListSessions(c *gin.Context) {
|
|
uid := middleware.GetUserID(c)
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
if page < 1 {
|
|
page = 1
|
|
}
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
|
|
if pageSize <= 0 || pageSize > 100 {
|
|
pageSize = 20
|
|
}
|
|
|
|
q := h.db.Model(&model.AIRun{}).Where("user_id = ? AND feature = ?", uid, "agent_cli")
|
|
if mode := c.Query("mode"); mode != "" {
|
|
mode = agentcli.NormalizeRunMode(mode)
|
|
if mode == agentcli.RunModeKubernetes {
|
|
q = q.Where("mode IN ?", []string{agentcli.RunModeKubernetes, "k8s"})
|
|
} else {
|
|
q = q.Where("mode = ?", mode)
|
|
}
|
|
}
|
|
if sid := c.Query("server_id"); sid != "" {
|
|
q = q.Where("subject_type = ? AND subject_id = ?", "host", sid)
|
|
}
|
|
if cid := c.Query("cluster_id"); cid != "" {
|
|
q = q.Where("subject_id = ?", cid)
|
|
}
|
|
if ns := c.Query("namespace"); ns != "" {
|
|
q = q.Where("subject_scope = ?", ns)
|
|
}
|
|
|
|
var total int64
|
|
q.Count(&total)
|
|
var sessions []model.AIRun
|
|
if err := q.Order("created_at DESC").Offset((page - 1) * pageSize).Limit(pageSize).Find(&sessions).Error; err != nil {
|
|
c.JSON(http.StatusOK, model.Response{Code: 500, Message: "查询失败: " + err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, model.Response{Code: 200, Data: gin.H{"list": sessions, "total": total}})
|
|
}
|
|
|
|
// GetSession 会话详情（含完整推理消息，供回放） // Logan.Li
|
|
// GET /api/v1/aiops/agent-cli/sessions/:id
|
|
func (h *AgentCLIHandler) GetSession(c *gin.Context) {
|
|
uid := middleware.GetUserID(c)
|
|
id := c.Param("id")
|
|
var sess model.AIRun
|
|
if err := h.db.Where("user_id = ?", uid).First(&sess, id).Error; err != nil {
|
|
c.JSON(http.StatusOK, model.Response{Code: 404, Message: "会话不存在"})
|
|
return
|
|
}
|
|
var msgs []model.AIRunMessage
|
|
h.db.Where("run_id = ?", sess.RunID).Order("seq ASC, id ASC").Find(&msgs)
|
|
c.JSON(http.StatusOK, model.Response{Code: 200, Data: gin.H{"session": sess, "messages": msgs}})
|
|
}
|
|
|
|
// DeleteSession 删除会话及其消息 // Logan.Li
|
|
// DELETE /api/v1/aiops/agent-cli/sessions/:id
|
|
func (h *AgentCLIHandler) DeleteSession(c *gin.Context) {
|
|
uid := middleware.GetUserID(c)
|
|
id := c.Param("id")
|
|
var sess model.AIRun
|
|
if err := h.db.Where("user_id = ?", uid).First(&sess, id).Error; err != nil {
|
|
c.JSON(http.StatusOK, model.Response{Code: 404, Message: "会话不存在"})
|
|
return
|
|
}
|
|
if err := h.db.Where("run_id = ?", sess.RunID).Delete(&model.AIRunMessage{}).Error; err != nil {
|
|
c.JSON(http.StatusOK, model.Response{Code: 500, Message: "删除消息失败: " + err.Error()})
|
|
return
|
|
}
|
|
// 会话历史可删除，但远程执行 Span 与命令审计必须独立保留。 // Logan.Li
|
|
if err := h.db.Delete(&model.AIRun{}, id).Error; err != nil {
|
|
c.JSON(http.StatusOK, model.Response{Code: 500, Message: "删除会话失败: " + err.Error()})
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, model.Response{Code: 200, Message: "已删除"})
|
|
}
|