pika/internal/handler/ssh_login_handler.go

package handler

import (
	"context"
	"net/http"

	"github.com/dushixiang/pika/internal/service"
	"github.com/go-orz/orz"
	"github.com/labstack/echo/v4"
	"go.uber.org/zap"
)

// SSHLoginHandler SSH登录处理器
type SSHLoginHandler struct {
	logger  *zap.Logger
	service *service.SSHLoginService
}

// NewSSHLoginHandler 创建处理器
func NewSSHLoginHandler(logger *zap.Logger, service *service.SSHLoginService) *SSHLoginHandler {
	return &SSHLoginHandler{
		logger:  logger,
		service: service,
	}
}

// GetConfig 获取SSH登录监控配置
// GET /api/agents/:id/ssh-login/config
func (h *SSHLoginHandler) GetConfig(c echo.Context) error {
	agentID := c.Param("id")
	if agentID == "" {
		return c.JSON(http.StatusBadRequest, map[string]string{
			"message": "探针ID不能为空",
		})
	}

	config, err := h.service.GetConfig(agentID)
	if err != nil {
		h.logger.Error("获取SSH登录监控配置失败", zap.Error(err))
		return c.JSON(http.StatusInternalServerError, map[string]string{
			"message": "获取配置失败",
		})
	}

	if config == nil {
		// 返回默认配置
		return c.JSON(http.StatusOK, map[string]interface{}{
			"enabled": false,
		})
	}

	return c.JSON(http.StatusOK, config)
}

// UpdateConfig 更新SSH登录监控配置
// POST /api/agents/:id/ssh-login/config
func (h *SSHLoginHandler) UpdateConfig(c echo.Context) error {
	agentID := c.Param("id")
	if agentID == "" {
		return c.JSON(http.StatusBadRequest, map[string]string{
			"message": "探针ID不能为空",
		})
	}

	var req struct {
		Enabled bool `json:"enabled"`
	}

	if err := c.Bind(&req); err != nil {
		return c.JSON(http.StatusBadRequest, map[string]string{
			"message": "请求参数错误",
		})
	}

	config, configSent, err := h.service.UpdateConfig(c.Request().Context(), agentID, req.Enabled)
	if err != nil {
		h.logger.Error("更新SSH登录监控配置失败", zap.Error(err))
		return c.JSON(http.StatusInternalServerError, map[string]string{
			"message": "更新配置失败",
		})
	}

	// 根据下发状态返回不同的消息
	message := "配置已保存"
	if configSent {
		message = "配置已保存并成功下发到探针"
	} else {
		message = "配置已保存，将在探针下次连接时生效"
	}

	return c.JSON(http.StatusOK, map[string]interface{}{
		"message":    message,
		"config":     config,
		"configSent": configSent, // 告知前端配置是否成功下发
	})
}

// ListEvents 查询SSH登录事件
// GET /api/agents/:id/ssh-login/events
func (h *SSHLoginHandler) ListEvents(c echo.Context) error {
	agentID := c.Param("id")

	// 获取分页参数
	pageReq := orz.GetPageRequest(c)
	builder := orz.NewPageBuilder(h.service.SSHLoginEventRepo.Repository).
		PageRequest(pageReq).
		Equal("agentId", agentID).
		Equal("username", c.QueryParam("username")).
		Equal("ip", c.QueryParam("ip")).
		Equal("status", c.QueryParam("status"))

	ctx := context.Background()
	page, err := builder.Execute(ctx)
	if err != nil {
		return err
	}

	return orz.Ok(c, page)
}

// GetEvent 获取单个SSH登录事件
// GET /api/ssh-login/events/:id
func (h *SSHLoginHandler) GetEvent(c echo.Context) error {
	eventID := c.Param("id")
	if eventID == "" {
		return c.JSON(http.StatusBadRequest, map[string]string{
			"message": "事件ID不能为空",
		})
	}

	ctx := context.Background()
	event, exists, err := h.service.SSHLoginEventRepo.FindByIdExists(ctx, eventID)
	if err != nil {
		h.logger.Error("获取SSH登录事件失败", zap.Error(err))
		return c.JSON(http.StatusInternalServerError, map[string]string{
			"message": "获取事件失败",
		})
	}

	if !exists {
		return c.JSON(http.StatusNotFound, map[string]string{
			"message": "事件不存在",
		})
	}

	return c.JSON(http.StatusOK, event)
}

// DeleteEvents 删除探针的所有SSH登录事件
// DELETE /api/agents/:id/ssh-login/events
func (h *SSHLoginHandler) DeleteEvents(c echo.Context) error {
	agentID := c.Param("id")

	ctx := context.Background()
	if err := h.service.DeleteEventsByAgentID(ctx, agentID); err != nil {
		h.logger.Error("删除SSH登录事件失败", zap.Error(err))
		return c.JSON(http.StatusInternalServerError, map[string]string{
			"message": "删除失败",
		})
	}

	return orz.Ok(c, orz.Map{})
}