Logan
1acb0c49af
- Agent CLI: 多轮对话、SSH 长连接、流式推理、Evolver 按用户隔离、历史侧栏 - 终端审计: 回放播放器、命令书签、筛选分页导出、命令输出统计、录制落盘修复 - 修复 session_commands 表迁移、Kill 竞态导致录制丢失、无录制会话回放提示 - 审计历史页只读（移除删除），移除回放弹框录制搜索
108 lines
3.1 KiB
Go
108 lines
3.1 KiB
Go
package controller
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"ezops/model"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// SessionCommandHandler 会话命令审计处理器 // Logan.Li
|
|
type SessionCommandHandler struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewSessionCommandHandler 创建会话命令审计处理器
|
|
func NewSessionCommandHandler(db *gorm.DB) *SessionCommandHandler {
|
|
return &SessionCommandHandler{db: db}
|
|
}
|
|
|
|
// ListSessionCommands 会话命令列表
|
|
// GET /api/v1/session-commands?session_id=&username=&page=&page_size=
|
|
func (h *SessionCommandHandler) ListSessionCommands(c *gin.Context) {
|
|
var req model.SessionCommandQuery
|
|
if err := c.ShouldBindQuery(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, model.Response{Code: 400, Message: err.Error()})
|
|
return
|
|
}
|
|
if req.Page <= 0 {
|
|
req.Page = 1
|
|
}
|
|
if req.PageSize <= 0 {
|
|
req.PageSize = 20
|
|
}
|
|
|
|
query := h.db.Model(&model.SessionCommand{})
|
|
if req.SessionID > 0 {
|
|
query = query.Where("session_id = ?", req.SessionID)
|
|
}
|
|
if req.Username != "" {
|
|
query = query.Where("username LIKE ?", "%"+req.Username+"%")
|
|
}
|
|
if req.ServerID > 0 {
|
|
query = query.Where("server_id = ?", req.ServerID)
|
|
}
|
|
if req.Command != "" {
|
|
query = query.Where("command LIKE ?", "%"+req.Command+"%")
|
|
}
|
|
if req.StartDate != "" {
|
|
if t, err := time.ParseInLocation("2006-01-02", req.StartDate, time.Local); err == nil {
|
|
query = query.Where("timestamp >= ?", t)
|
|
}
|
|
}
|
|
if req.EndDate != "" {
|
|
if t, err := time.ParseInLocation("2006-01-02", req.EndDate, time.Local); err == nil {
|
|
query = query.Where("timestamp < ?", t.Add(24*time.Hour))
|
|
}
|
|
}
|
|
|
|
var total int64
|
|
if err := query.Count(&total).Error; err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.Response{Code: 500, Message: "查询失败"})
|
|
return
|
|
}
|
|
|
|
// 联表带出会话录制大小，前端可判断能否回放 // Logan.Li
|
|
var items []model.SessionCommandItem
|
|
listQuery := h.db.Table("sys_session_commands AS sc").
|
|
Select("sc.*, COALESCE(ts.record_size, 0) AS session_record_size").
|
|
Joins("LEFT JOIN sys_terminal_sessions AS ts ON ts.id = sc.session_id")
|
|
if req.SessionID > 0 {
|
|
listQuery = listQuery.Where("sc.session_id = ?", req.SessionID)
|
|
}
|
|
if req.Username != "" {
|
|
listQuery = listQuery.Where("sc.username LIKE ?", "%"+req.Username+"%")
|
|
}
|
|
if req.ServerID > 0 {
|
|
listQuery = listQuery.Where("sc.server_id = ?", req.ServerID)
|
|
}
|
|
if req.Command != "" {
|
|
listQuery = listQuery.Where("sc.command LIKE ?", "%"+req.Command+"%")
|
|
}
|
|
if req.StartDate != "" {
|
|
if t, err := time.ParseInLocation("2006-01-02", req.StartDate, time.Local); err == nil {
|
|
listQuery = listQuery.Where("sc.timestamp >= ?", t)
|
|
}
|
|
}
|
|
if req.EndDate != "" {
|
|
if t, err := time.ParseInLocation("2006-01-02", req.EndDate, time.Local); err == nil {
|
|
listQuery = listQuery.Where("sc.timestamp < ?", t.Add(24*time.Hour))
|
|
}
|
|
}
|
|
if err := listQuery.Order("sc.id DESC").
|
|
Offset((req.Page - 1) * req.PageSize).
|
|
Limit(req.PageSize).
|
|
Scan(&items).Error; err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.Response{Code: 500, Message: "查询失败"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.Response{Code: 200, Data: model.ListResponse{
|
|
List: items,
|
|
Total: total,
|
|
}})
|
|
}
|