package controllers

import (
	"fmt"

	"github.com/engigu/baihu-panel/internal/database"
	"github.com/engigu/baihu-panel/internal/models"
	"github.com/engigu/baihu-panel/internal/services/tasks"
	"github.com/engigu/baihu-panel/internal/utils"

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

type LogWSController struct{}

func NewLogWSController() *LogWSController {
	return &LogWSController{}
}

func (lc *LogWSController) StreamLog(c *gin.Context) {
	logIDStr := c.Query("log_id")
	if logIDStr == "" {
		return
	}

	logID := logIDStr

	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		return
	}
	defer conn.Close()

	// 1. 检查数据库中是否已结束
	var taskLog models.TaskLog
	if err := database.DB.Where("id = ?", logID).First(&taskLog).Error; err == nil {
		if taskLog.Status != "running" {
			// 已结束，读取库内日志
			content, err := utils.DecompressFromBase64(taskLog.Output)
			if err != nil {
				conn.WriteMessage(websocket.TextMessage, []byte("解压日志失败: "+err.Error()))
				return
			}
			conn.WriteMessage(websocket.TextMessage, []byte(content))
			return
		}
	}

	// 2. 未结束或未找到记录，尝试从 TinyLogManager 获取
	tl := tasks.GetActiveLog(logID)
	if tl == nil {
		conn.WriteMessage(websocket.TextMessage, []byte("未找到正在运行的任务日志"))
		return
	}

	// 发送系统提示
	conn.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("[System] 连接成功，正在监听日志... (LogID: %s)\n", logID)))

	// 发送最后 100 行
	lastLines, err := tl.ReadLastLines(100)
	if err == nil && len(lastLines) > 0 {
		conn.WriteMessage(websocket.TextMessage, lastLines)
	}

	// 订阅实时更新
	sub := tl.Subscribe()
	defer tl.Unsubscribe(sub)

	// 推送更新
	for {
		select {
		case data, ok := <-sub:
			if !ok {
				// 任务结束，尝试刷新最后一次库内完整内容
				var finalLog models.TaskLog
				if err := database.DB.Where("id = ?", logID).First(&finalLog).Error; err == nil {
					content, _ := utils.DecompressFromBase64(finalLog.Output)
					if content != "" {
						conn.WriteMessage(websocket.TextMessage, []byte("\n--- 任务已结束 ---\n"))
						// 这里可以选择性再推一次完整版，或直接退出
					}
				}
				return
			}
			if err := conn.WriteMessage(websocket.TextMessage, data); err != nil {
				return
			}
		case <-c.Request.Context().Done():
			return
		}
	}
}