package model

import (
	"fmt"
	"gorm.io/gorm"
	"strconv"
	"strings"
	"time"
)

type Device struct {
	gorm.Model
	Type        string  `json:"type"`      // 设备类型
	Vendor      string  `json:"vendor"`    // 厂商
	SN          string  `json:"device_sn"` // 编码
	Lng         float64 `json:"lng"`
	Lat         float64 `json:"lat"`
	Battery     float64 `json:"battery"`
	High        float64 `json:"high"`
	GpsType     string  `json:"gps_type"`
	LocateState string  `json:"locate_state"`
	Mac         string  `json:"mac"`
	Ip          string  `json:"ip"`
	IsOutBound  int     `json:"isOutBound" gorm:"default:1"` // 0 没出界 1 有出界
	BoundId     uint    `json:"bound_id"`                    // 所处的电子围栏
}

func (obj *Device) FindAll() []Device {
	list := []Device{}
	err := DB.Table("devices").Find(&list).Error
	if err != nil {
		return nil
	}

	return list
}

func (obj *Device) FindByBoundID(bound_id uint) ([]Device, int64) {
	list := []Device{}
	var total int64
	err := DB.Table("devices").Where("bound_id = ?", bound_id).Find(&list).Error
	err = DB.Table("devices").Where("bound_id = ?", bound_id).Count(&total).Error
	if err != nil {
		return nil, 0
	}
	return list, total
}

func (obj *Device) Update() (err error) {
	if isExistDevice(obj.SN) {
		var err error

		// 坐标为0时不更新位置
		dev := Device{Battery: obj.Battery}
		dev.UpdatedAt = time.Now()

		if obj.Lat == 0 {
			err = DB.Table("devices").Where("sn = ?", obj.SN).Select("battery").Updates(dev).Error
		} else {
			dev.High = obj.High
			dev.Lat = obj.Lat
			dev.Lng = obj.Lng

			err = DB.Table("devices").Where("sn = ?", obj.SN).Select("lng", "lat", "battery", "high").Updates(dev).Error
		}

		if err != nil {
			fmt.Println("修改数据出错")
			return err
		}
	} else {
		err := DB.Table("devices").Create(&obj).Error
		if err != nil {
			return err
		}
		// fmt.Println("新增安全帽数据")
	}

	hasZone, zoneId := CheckInBound(obj)
	if hasZone {
		if zoneId > 0 {
			DB.Table("devices").Where("sn=?", obj.SN).Update(Device{BoundId: zoneId, IsOutBound: 0})
		} else {
			DB.Table("devices").Where("sn=?", obj.SN).Update(Device{IsOutBound: 1})
		}
	}
	return nil
}

func isExistDevice(sn string) bool {
	var count int64
	DB.Table("devices").Where("sn = ?", sn).Count(&count)
	return count > 0
}

func CheckInBound(dv *Device) (bool, uint) {
	obj := DrawZone{}
	zones := obj.FindAll()
	if zones == nil {
		return false, 0
	}
	pointToCheck := Point{Lng: dv.Lng, Lat: dv.Lat}
	for i := 0; i < len(zones); i++ {
		zone := zones[i]
		arr := strings.Split(zone.Dots, "&")
		pointArr := []Point{}
		for i := 0; i < len(arr); i++ {
			str := arr[i]
			subArr := strings.Split(str, ",")
			f1, _ := strconv.ParseFloat(subArr[0], 64)
			f2, _ := strconv.ParseFloat(subArr[1], 64)
			newPoint := Point{Lng: f1, Lat: f2}
			pointArr = append(pointArr, newPoint)
		}
		if IsPointInPolygon(pointToCheck, pointArr) {
			return true, zone.ID
		}
	}
	return true, 0
}