package handler

import (
	"encoding/json"
	"net/http"
	"time"

	"go_rain_dtu/internal/dao"
	"go_rain_dtu/internal/tcp"
	"go_rain_dtu/pkg/logger"
)

type SensorHandler struct {
	dao *dao.SensorDAO
}

func NewSensorHandler(dao *dao.SensorDAO) *SensorHandler {
	return &SensorHandler{dao: dao}
}

func (h *SensorHandler) GetAggregatedData(w http.ResponseWriter, r *http.Request) {
	interval := r.URL.Query().Get("interval")
	if interval == "" {
		interval = "1hour"
	}

	endTime := time.Now()
	startTime := endTime.Add(-24 * time.Hour) // 默认显示24小时数据

	if startStr := r.URL.Query().Get("start"); startStr != "" {
		if t, err := time.Parse("2006-01-02T15:04", startStr); err == nil {
			startTime = t
		}
	}
	if endStr := r.URL.Query().Get("end"); endStr != "" {
		if t, err := time.Parse("2006-01-02T15:04", endStr); err == nil {
			endTime = t
		}
	}

	data, err := h.dao.GetAggregatedData(startTime, endTime, interval)
	if err != nil {
		logger.Logger.Printf("获取聚合数据失败: %v", err)
		http.Error(w, "服务器内部错误", http.StatusInternalServerError)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	// 使用自定义的编码器，设置时间格式
	encoder := json.NewEncoder(w)
	encoder.SetIndent("", "  ")
	if err := encoder.Encode(data); err != nil {
		logger.Logger.Printf("JSON编码失败: %v", err)
		http.Error(w, "服务器内部错误", http.StatusInternalServerError)
		return
	}
}

// 处理静态文件
func (h *SensorHandler) ServeStatic(w http.ResponseWriter, r *http.Request) {
	if r.URL.Path == "/" {
		http.ServeFile(w, r, "static/index.html")
		return
	}
	http.FileServer(http.Dir("static")).ServeHTTP(w, r)
}

// GetConnectionStatus 返回当前TCP连接状态
func (h *SensorHandler) GetConnectionStatus(w http.ResponseWriter, r *http.Request) {
	connected, ip, port := tcp.GetConnectionStatus()

	status := struct {
		Connected bool   `json:"connected"`
		IP        string `json:"ip"`
		Port      string `json:"port"`
		Count     int    `json:"count"` // 添加连接数量
	}{
		Connected: connected,
		IP:        ip,
		Port:      port,
		Count:     tcp.GetActiveConnectionCount(), // 获取活跃连接数
	}

	w.Header().Set("Content-Type", "application/json")
	if err := json.NewEncoder(w).Encode(status); err != nil {
		logger.Logger.Printf("JSON编码失败: %v", err)
		http.Error(w, "服务器内部错误", http.StatusInternalServerError)
	}
}

// GetLatestData 获取最新的聚合数据
func (h *SensorHandler) GetLatestData(w http.ResponseWriter, r *http.Request) {
	interval := r.URL.Query().Get("interval")
	if interval == "" {
		interval = "5min" // 默认使用更细粒度的时间间隔
	}

	// 默认查询最近1小时的数据
	endTime := time.Now()
	startTime := endTime.Add(-1 * time.Hour)

	// 如果提供了时间范围，则使用提供的范围
	if startStr := r.URL.Query().Get("start"); startStr != "" {
		if t, err := time.Parse(time.RFC3339, startStr); err == nil {
			startTime = t
		}
	}
	if endStr := r.URL.Query().Get("end"); endStr != "" {
		if t, err := time.Parse(time.RFC3339, endStr); err == nil {
			endTime = t
		}
	}

	// 查询数据库获取最新数据
	data, err := h.dao.GetAggregatedData(startTime, endTime, interval)
	if err != nil {
		logger.Logger.Printf("获取最新聚合数据失败: %v", err)
		http.Error(w, "服务器内部错误", http.StatusInternalServerError)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	encoder := json.NewEncoder(w)
	encoder.SetIndent("", "  ")
	if err := encoder.Encode(data); err != nil {
		logger.Logger.Printf("JSON编码失败: %v", err)
		http.Error(w, "服务器内部错误", http.StatusInternalServerError)
		return
	}
}