package controllers

import (
	"fmt"
	"time"

	"github.com/gin-gonic/gin"

	"lokistack/services"
	"lokistack/utils"
)

// SystemController 系统级运维操作：备份 / 恢复 / 巡检。
type SystemController struct{}

func NewSystemController() *SystemController { return &SystemController{} }

// Backup 导出当前平台数据（tar.gz 下载）。
//
// 内容：数据文件 + kubeconfigs/ + known_hosts/ + manifest.json（含密钥指纹），
// 不含明文密钥。恢复必须使用同一把 ENCRYPTION_KEY。
func (ctrl *SystemController) Backup(c *gin.Context) {
	name := fmt.Sprintf("lokistack-backup-%s.tar.gz", time.Now().Format("20060102-150405"))
	c.Header("Content-Type", "application/gzip")
	c.Header("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, name))

	if _, err := services.CreateBackup(c.Writer); err != nil {
		// 已经写了响应头，无法再改状态码，只能把错误写进 body 尾部
		_, _ = c.Writer.WriteString(fmt.Sprintf("\n\n# 备份失败: %v", err))
		return
	}
}

// Restore 上传备份包：校验 + 暂存，**不直接覆盖**线上数据库。
// 返回暂存路径与密钥指纹比对结果，由管理员确认后再调用 RestoreApply。
func (ctrl *SystemController) Restore(c *gin.Context) {
	file, err := c.FormFile("file")
	if err != nil {
		utils.BadRequest(c, "请上传备份文件（form-data 字段名 file）")
		return
	}
	if file.Size > 128<<20 {
		utils.BadRequest(c, "备份文件超过 128MB，请用 run.sh restore 在服务器上操作")
		return
	}
	f, err := file.Open()
	if err != nil {
		utils.InternalServerError(c, fmt.Sprintf("读取上传文件失败: %v", err))
		return
	}
	defer f.Close()

	res, err := services.ValidateAndStageBackup(f)
	if err != nil {
		utils.BadRequest(c, fmt.Sprintf("备份校验失败: %v", err))
		return
	}
	utils.Success(c, res)
}

// RestoreApply 应用已暂存的备份。应用后需重启平台进程才生效。
func (ctrl *SystemController) RestoreApply(c *gin.Context) {
	var req struct {
		Confirm bool `json:"confirm"`
	}
	_ = c.ShouldBindJSON(&req)
	if !req.Confirm {
		utils.BadRequest(c, "恢复会覆盖当前全部数据，请传 confirm=true 确认")
		return
	}
	path, err := services.ApplyStagedBackup()
	if err != nil {
		utils.BadRequest(c, err.Error())
		return
	}
	utils.Success(c, gin.H{
		"message": "备份已应用，请重启平台使新数据生效（kubectl -n log rollout restart deploy/lokistack）",
		"dbPath":  path,
	})
}

// Backups 列出服务器上已有的备份文件（run.sh backup 产出）。
func (ctrl *SystemController) Backups(c *gin.Context) {
	utils.Success(c, gin.H{"list": services.ListBackups()})
}

// Health 平台自检：一键诊断用。
// 聚合「能不能用」的判断，避免排障时在多个页面之间跳。
func (ctrl *SystemController) Health(c *gin.Context) {
	utils.Success(c, services.SelfCheck())
}