package controllers

import (
	"opsxloki/dto"
	"opsxloki/models"
	"opsxloki/services"
	"opsxloki/utils"

	"github.com/gin-gonic/gin"
)

type CollectorController struct {
	service *services.CollectorService
}

func NewCollectorController() *CollectorController {
	return &CollectorController{service: services.NewCollectorService()}
}

func (ctrl *CollectorController) Create(c *gin.Context) {
	var req dto.CreateCollectorRequest
	if !utils.BindAndValidate(c, &req) {
		return
	}
	collector, err := ctrl.service.Create(&services.CreateCollectorRequest{
		Name: req.Name, Type: req.Type, LokiDeploymentID: req.LokiDeploymentID,
		HostID: req.HostID, ClusterID: req.ClusterID, TargetDeployment: req.TargetDeployment,
		Namespace: req.Namespace, LogPath: req.LogPath, Labels: req.Labels,
	})
	if err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, models.SanitizedJSON(collector))
}

func (ctrl *CollectorController) Get(c *gin.Context) {
	id, err := utils.IntParam(c, "id")
	if err != nil {
		utils.BadRequest(c, "ID 无效")
		return
	}
	collector, err := ctrl.service.Get(uint(id))
	if err != nil {
		utils.NotFound(c, "采集器不存在")
		return
	}
	utils.Success(c, models.SanitizedJSON(collector))
}

func (ctrl *CollectorController) List(c *gin.Context) {
	page, pageSize := utils.Pagination(c)
	collectors, total, err := ctrl.service.List(page, pageSize)
	if err != nil {
		utils.Error(c, 500, err.Error())
		return
	}
	utils.Success(c, utils.PageData{
		List: models.SanitizedJSON(collectors), Total: total, Page: page, PageSize: pageSize,
	})
}

func (ctrl *CollectorController) Update(c *gin.Context) {
	id, err := utils.IntParam(c, "id")
	if err != nil {
		utils.BadRequest(c, "ID 无效")
		return
	}
	var updates map[string]interface{}
	if !utils.BindAndValidate(c, &updates) {
		return
	}
	if err := ctrl.service.Update(uint(id), updates); err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, gin.H{"message": "更新成功"})
}

func (ctrl *CollectorController) Delete(c *gin.Context) {
	id, err := utils.IntParam(c, "id")
	if err != nil {
		utils.BadRequest(c, "ID 无效")
		return
	}
	if err := ctrl.service.Delete(uint(id)); err != nil {
		utils.Error(c, 400, err.Error())
		return
	}
	utils.Success(c, gin.H{"message": "删除成功"})
}

func (ctrl *CollectorController) Install(c *gin.Context) {
	id, err := utils.IntParam(c, "id")
	if err != nil {
		utils.BadRequest(c, "ID 无效")
		return
	}
	go ctrl.service.Install(uint(id))
	utils.Success(c, gin.H{"message": "已开始部署采集器"})
}

func (ctrl *CollectorController) Uninstall(c *gin.Context) {
	id, err := utils.IntParam(c, "id")
	if err != nil {
		utils.BadRequest(c, "ID 无效")
		return
	}
	go ctrl.service.Uninstall(uint(id))
	utils.Success(c, gin.H{"message": "已开始卸载采集器"})
}