package saft_hat

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"io/ioutil"
	"net/http"
	"strconv"
	"strings"
	"time"
)

// 获取设备状态的请求结构
type DeviceStatusReq struct {
	g.Meta `path:"/device/status" method:"post" tags:"安全帽相关" summary:"获取安全帽状态"`
	Data   []string `json:"data" dc:"设备号"`
}

// 获取设备状态
func (h Hat) DeviceStatus(ctx context.Context, req *DeviceStatusReq) (res *DeviceStatusRes, err error) {
	return StatusCheck(ctx, req.Data)
}

type DeviceStatusRes struct {
	Code int                `json:"code"` // 响应代码
	Data []DeviceStatusInfo `json:"data"` // 设备数据
	Msg  string             `json:"msg"`  // 响应消息
}

// 定义数组中每个对象的结构
type DeviceStatusInfo struct {
	IMEI   string `json:"imei"`
	Status int    `json:"status"`
}

// 检查设备的状态情况
func StatusCheck(ctx context.Context, devNums []string) (res *DeviceStatusRes, err error) {
	res = new(DeviceStatusRes)
	data := strings.Join(devNums, ",")
	timestamp := time.Now().Unix()

	// 准备生成签名的参数
	params := map[string]string{
		"appid":     appid,
		"data":      data,
		"secret":    secret,
		"timestamp": strconv.FormatInt(timestamp, 10),
	}

	// 生成签名
	sign := GenerateSignature(params)

	// 构造请求体
	payload := map[string]interface{}{
		"appid":     appid,
		"data":      data,
		"secret":    secret,
		"sign":      sign,
		"timestamp": timestamp,
	}
	payloadBytes, err := json.Marshal(payload)
	if err != nil {
		return res, err
	}

	// 发送POST请求
	resp, err := http.Post("https://www.loctp.com/api/crm/v1/getStatus", "application/json", bytes.NewBuffer(payloadBytes))

	if err != nil {
		return res, err
	}
	defer resp.Body.Close()

	// 检查响应
	respBody, err := ioutil.ReadAll(resp.Body)
	fmt.Println(string(respBody))
	if err != nil {
		return res, err
	}

	err = json.Unmarshal(respBody, res)
	if err != nil {
		return res, err
	}

	return res, nil
}

// 心跳检测
func HeartCheck() {
	ticker := time.NewTicker(60 * time.Second)
	defer ticker.Stop() // 确保ticker被适当释放

	for {
		select {
		case <-ticker.C:
			var devices []Device
			err := g.Model("device").Fields("dev_num").Scan(&devices)
			if err != nil {
				fmt.Println("获取设备列表出错:", err)
				continue // 如果获取失败，则跳过本次循环
			}

			if len(devices) == 0 {
				fmt.Println("没有找到任何设备")
				continue // 如果没有设备，则跳过本次循环
			}

			// 构建devNums切片
			var devNums []string
			for _, device := range devices {
				devNums = append(devNums, device.DevNum)
			}
			res, err := StatusCheck(context.Background(), devNums)
			if err != nil {
				fmt.Println("检查设备状态出错:", err)
				continue // 如果检查失败，则跳过本次循环
			}

			for _, deviceStatus := range res.Data {
				// 更新 device 表中的 status 字段
				_, err := g.Model("device").Data("status", deviceStatus.Status).Where("dev_num", deviceStatus.IMEI).Update()
				if err != nil {
					fmt.Printf("更新设备 %s 状态出错: %v\n", deviceStatus.IMEI, err)
					continue // 如果更新失败，则跳过本次设备的更新
				}
			}
		}
	}
}