package httpreader

import (
	"fmt"
	"net"
	"net/http"
	"time"

	"FireLeave_tool/logger"
)

// CheckHTTPService 检查HTTP服务是否可用
func CheckHTTPService(port int, timeout time.Duration) bool {
	client := &http.Client{
		Timeout: timeout,
	}

	url := fmt.Sprintf("http://127.0.0.1:%d/health", port)
	resp, err := client.Get(url)
	if err != nil {
		logger.Logger.Printf("HTTP服务健康检查失败 (Port=%d): %v", port, err)
		return false
	}
	defer resp.Body.Close()

	return resp.StatusCode == http.StatusOK
}

// CheckPortAvailable 检查端口是否可用
func CheckPortAvailable(port int) bool {
	address := fmt.Sprintf(":%d", port)
	listener, err := net.Listen("tcp", address)
	if err != nil {
		return false
	}
	listener.Close()
	return true
}

// WaitForHTTPService 等待HTTP服务启动
func WaitForHTTPService(port int, timeout time.Duration) bool {
	deadline := time.Now().Add(timeout)
	logger.Logger.Printf("等待HTTP服务启动 (Port=%d, Timeout=%v)", port, timeout)

	attempt := 0
	for time.Now().Before(deadline) {
		attempt++

		if CheckHTTPService(port, 2*time.Second) {
			logger.Logger.Printf("HTTP服务启动成功 (Port=%d, 尝试次数=%d)", port, attempt)
			return true
		}

		if attempt == 1 || attempt%5 == 0 {
			logger.Logger.Printf("等待HTTP服务... (Port=%d, 尝试次数=%d)", port, attempt)
		}

		time.Sleep(1 * time.Second)
	}

	logger.Logger.Printf("HTTP服务启动超时 (Port=%d, 总尝试次数=%d)", port, attempt)
	return false
}

// ValidateInferenceResult 验证推理结果数据
func ValidateInferenceResult(result *InferenceResult) error {
	if result.Serial == "" {
		return fmt.Errorf("serial字段不能为空")
	}
	if result.At <= 0 {
		return fmt.Errorf("at时间戳无效")
	}

	// 验证参数数据
	for i, param := range result.Params {
		if param.Number < 0 {
			return fmt.Errorf("参数[%d]的number不能为负数", i)
		}
		if param.ClassIdx < 0 {
			return fmt.Errorf("参数[%d]的class_idx不能为负数", i)
		}
	}

	return nil
}

// GetPersonCountFromResult 从推理结果中获取人员数量
func GetPersonCountFromResult(result *InferenceResult) int {
	if result == nil || len(result.Params) == 0 {
		return 0
	}

	// 查找人员类别的参数（假设 class_idx=1 表示人员）
	for _, param := range result.Params {
		if param.ClassIdx == 1 { // 人员类别
			return param.Number
		}
	}

	return 0
}

// CreateTestInferenceResult 创建测试用的推理结果（用于调试）
func CreateTestInferenceResult(deviceUUID string, personCount int) InferenceResult {
	return InferenceResult{
		Serial: deviceUUID,
		At:     time.Now().Unix(),
		Type:   1,
		Params: []struct {
			ClassIdx int    `json:"class_idx"`
			Name     string `json:"name"`
			Number   int    `json:"number"`
		}{
			{
				ClassIdx: 1,
				Name:     "person",
				Number:   personCount,
			},
		},
	}
}