package models

import (
	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/logger.git"
	"encoding/json"
	"regexp"
)

type CameraPolygon struct {
	Id            string `gorm:"primary_key;column:id" json:"id"`
	CameraId      string `gorm:"column:camera_id" json:"camera_id"`      //摄像机id
	Name          string `gorm:"column:name" json:"name"`           //形状名称
	Polygon       string `gorm:"column:polygon" json:"polygon"`        //形状结构定义
	DirectionLine string `gorm:"column:direction_line" json:"direction_line"` //方向线
	Type 		  string `gorm:"column:type" json:"type"`			 //类型，["line","rect","polygon"]
	DefenceState  int    `gorm:"column:defence_state;default:0" json:"defence_state"` //布撤防状态
}

const (
	TYPE_LINE = "line" //线
	TYPE_RECT = "rect" //矩形
	TYPE_POLYGON = "polygon" //面
	CAMERAPOLYGON_AREA_ALL = "全部区域"
)

//坐标定义
//type Point struct {
//	X float32 `json:"x"`
//	Y float32 `json:"y"`
//}

func (CameraPolygon) TableName() string {
	return "camera_polygon"
}

func (cp *CameraPolygon) FindAll() (polygons []CameraPolygon) {
	if err := db.Table("camera_polygon").Scan(&polygons).Error;err !=nil {
		return nil
	}
	return polygons
}

func (cp *CameraPolygon) FindAllMap() map[string]CameraPolygon {
	m := make(map[string]CameraPolygon,0)
	polygons := cp.FindAll()
	if polygons !=nil {
		for _,p :=range polygons {
			m[p.Id] = p
		}
	}
	return m
}

func (cp *CameraPolygon) SelectById(id string) (model CameraPolygon,flag bool) {
	exist := db.Table("camera_polygon").First(&model,"id=?",id).RecordNotFound()
	return model,!exist
}

func (cp *CameraPolygon) FindAllByCameraId(cameraId string) (polygons []CameraPolygon, err error) {
	if err := db.Table("camera_polygon").Where("camera_id = ?", cameraId).Scan(&polygons).Error;err !=nil {
		return nil, err
	}
	return polygons,nil
}

//FindByCameraId 根据摄像机id查找在规则中使用的多边形
func (cp *CameraPolygon) FindRulePolygonsByCameraId(cameraId string) (polygons []CameraPolygon, err error) {
	if err := db.Table("camera_polygon").Where("camera_id = ? and type !=?", cameraId, TYPE_LINE).Scan(&polygons).Error;err !=nil {
		return nil, err
	}
	return polygons,nil
}

//保存摄像机区域
func (cp *CameraPolygon) Insert() (result bool, err error) {
	if err = db.Table("camera_polygon").Save(&cp).Error; err != nil {
		return false, err
	}
	return true, nil
}

//更新摄像机区域
func (cp *CameraPolygon) Update() (bool,error) {
	result := db.Exec("update camera_polygon set camera_id=?,name=?,polygon=?,direction_line=?,type=?,defence_state=? where id=?",cp.CameraId,cp.Name,cp.Polygon,cp.DirectionLine,cp.Type,cp.DefenceState,cp.Id)
	if result.Error != nil {
		return false, result.Error
	}
	return result.RowsAffected > 0, nil
}

//判断摄像机有没有重名的区域
func (cp CameraPolygon) Exist(cameraId string, name string) (model CameraPolygon, exist bool) {
	exist = db.Table("camera_polygon").First(&model, "camera_id=? and name=?", cameraId, name).RecordNotFound()
	return model, !exist
}

//UnInstall 删除摄像机区域
func (cp *CameraPolygon) Delete(id string,name string) bool {
	if err := db.Table("camera_polygon").Where("id=?",id).Delete(&CameraPolygon{}).Error; err != nil {
		return false
	}
	var crg CameraRuleGroup
	groups := crg.FindGroupByPolygonId(id)
	if groups != nil {
		for _,g :=range groups {
			//将groupText中的已删除的多边形名称替换为NULL
			reg := regexp.MustCompile(`(`+name+`于)`)
			newGText := reg.ReplaceAllString(g.GroupText, `<span style="color:RGB(255,0,0);">NULL</span>于`)
			crg.UpdateText(g.Id, newGText)
		}
	}
	err := db.Exec("update camera_task_args set polygon_id='' where polygon_id=?",id).Error
	if err !=nil {
		logger.Debug("rm polygon_id in camera_task_args err:",err)
	}
	return true
}

func (cp *CameraPolygon) UpdateDefenceStateByPolygonId(polygonId string,state int) bool {
	result := db.Exec("update camera_polygon set defence_state=? where id=?",state,polygonId)
	if result.Error!=nil{
		return false
	}
	if result.RowsAffected>0{
		return true
	} else {
		return false
	}
}

func (cp *CameraPolygon) UpdateDefenceStateByCameraId(cameraId string,state int) bool {
	result := db.Exec("update camera_polygon set defence_state=? where camera_id=?",state,cameraId)
	if result.Error!=nil{
		return false
	}
	if result.RowsAffected>0{
		return true
	} else {
		return false
	}
}