package models

import (
	"basic.com/valib/logger.git"
	"fmt"
	"strconv"
	"strings"
	"vamicro/config"
)

const (
	Default_Layer = -9999 //摄像机添加进来的默认楼层
	TYPE_LOCAL_CAMERA = 0 //本地摄像机
	TYPE_GB28181_CAMERA = 1 //国标摄像机

	TYPE_RUNTYPE_VIDEO = -1 //单纯的监控，不做分析
	TYPE_RUNTYPE_POLL = 0 //轮询做任务
	TYPE_RUNTYPE_REALTIME = 1 //实时做任务

	Camera_Status_NoRule = 0  //未配规则
	Camera_Status_Wait = 1  //等待处理
	Camera_Status_Doing = 2  //处理中
	Camera_Status_Other = -1 //其他情况
)

type Camera struct {
	Id        			string       	`gorm:"primary_key;column:id;type:varchar(100);unique;" json:"id"`
	Name      			string       	`gorm:"column:name" json:"name"  bind:"required"`
	Alias     			string 	   		`gorm:"column:alias" json:"alias"` //摄像机的别名
	SnapshotUrl 		string     		`gorm:"column:snapshot_url" json:"snapshot_url"`//快照地址
	Type      			int          	`gorm:"column:type" json:"type" `
	Addr      			string       	`gorm:"column:addr" json:"addr"`
	Longitude 			float32      	`gorm:"column:longitude" json:"longitude"`
	Latitude  			float32      	`gorm:"column:latitude" json:"latitude"`
	Floor     			int		   		`gorm:"column:floor" json:"floor"`//楼层
	Rtsp      			string       	`gorm:"column:rtsp" json:"rtsp"`
	Ip        			string       	`gorm:"column:ip" json:"ip"`
	Port      			int          	`gorm:"column:port" json:"port"`
	Username  			string       	`gorm:"column:username" json:"username"`
	Password  			string       	`gorm:"column:password" json:"password"`
	Brand     			string       	`gorm:"column:brand" json:"brand"`
	Reserved  			string       	`gorm:"column:reserved" json:"reserved"`
	IsRunning 			bool 		   	`gorm:"column:is_running" json:"is_running"`//是否正在解码
	RunEnable 			bool 		   	`gorm:"column:run_enable" json:"run_enable"`//控制实时处理或轮询处理的开关
	RunType   			int 		   	`gorm:"column:run_type" json:"run_type"`//处理类型：0：轮询，1：实时,-1:无任务，不做分析或者分析任务被关了
	RunServerId 		string 	   		`gorm:"column:run_server_id" json:"run_server_id"`//当前正在处理的分析服务器id

	ResolutionWidth  	int   			`gorm:"column:resolution_width;default:0" json:"resolution_width"`//分辨率宽
	ResolutionHeight 	int   			`gorm:"column:resolution_height;default:0" json:"resolution_height"`//分辨率高

	VoiceEnable 		bool 	   		`gorm:"column:voiceEnable;default:0" json:"voiceEnable"`
	VoiceId 			string 	   		`gorm:"column:voiceId" json:"voiceId"`
}

type CameraTreeNode struct {
	Id     string `json:"id"`
	Areaid string   `json:"areaid"`
	Name   string `json:"name"`
	CameraType int `json:"cameratype"`//是本地还是国标
	Rtsp   string `json:"rtsp"`
}

func (camera *Camera) GetIdNameMap() (map[string]string) {
	cameraEMap :=make(map[string]string,0)
	allCams, err := camera.FindAll()
	if err ==nil && allCams !=nil && len(allCams) >0 {
		for _,model :=range allCams {
			cameraEMap[model.Id]=model.Name
		}
	}

	return cameraEMap
}

//cameraType (0:查全部,1:分析摄像机,2:监控摄像机)
//cameraName
func (camera *Camera) Find(cameraType int, cameraName string, camType int, isPlatform int) (camMenus []CameraTreeNode, err error) {
	logger.Debug("camType:",camType)
	cType := strconv.Itoa(camType)
	var sql = "select c.id,ca.areaid as areaid,case ifnull(c.alias,'') when '' then c.name else c.alias end as name,c.type as cameratype,c.rtsp from camera_area ca join cameras c on ca.cameraid=c.id where c.type="+cType+""

	if cameraName != "" {
		sql += " and c.name like '%" + cameraName + "%'"
	}
	if cameraType == 1 { //查分析摄像机
		sql += " and (c.run_type="+strconv.Itoa(TYPE_RUNTYPE_POLL)+" or c.run_type="+strconv.Itoa(TYPE_RUNTYPE_REALTIME)+")"
	} else if cameraType == 2 { //查监控摄像机
		sql += " and c.run_type="+strconv.Itoa(TYPE_RUNTYPE_VIDEO)+""
	} else if cameraType == 3 { //查联动摄像机
		sql += " and c.id in (select crga.camera_id from camera_rule_group_arg crga join camera_task_link ctl on crga.group_id=ctl.link_task_id)"
	}
	if camType == TYPE_LOCAL_CAMERA {//国标的始终显示整棵树，本地树本机查本机，平台查所有
		if isPlatform == 0 {//查本机
			sql += " and c.run_server_id='"+config.Server.AnalyServerId+"'"
		}
	}

	sql += " order by c.id asc"

	if err := db.Raw(sql).Scan(&camMenus).Error; err != nil {
		return nil, err
	}
	return camMenus, nil
}

func (cam *Camera) FindAllByServer(serverId string,cameraName string)(list []Camera,err error) {
	var sql = "select * from cameras where 1=1 "
	if serverId != "" {
		sql += " and (run_server_id='"+serverId+"' or (run_server_id ='' or run_server_id is NULL))"
	}
	if cameraName != "" {
		sql += " and name like '%"+cameraName+"%'"
	}
	if err := db.Raw(sql).Scan(&list).Error;err !=nil{
		return nil,err
	}
	return list,nil
}

func (camera *Camera) GetCamerasByRunType(runType int,cameraName string) (list []Camera,err error) {
	analyServerId := config.Server.AnalyServerId//当前分析服务器的id(analyServerId,在配置文件中)
	sql := "select * from cameras where run_type=? and run_server_id=? "
	if cameraName !=""{
		sql = sql+" and name like ?"
	}

	if err := db.Raw(sql,runType,analyServerId,"%"+cameraName+"%").Scan(&list).Error;err !=nil{
		return nil,err
	}
	return list,nil
}

func (camera *Camera) Insert() (err error) {
	tx := db.Table("cameras").Begin()

	if tx.Error != nil {
		return err
	}

	if err := tx.Create(&camera).Error; err != nil {
		tx.Rollback()
		return err
	}
	//新增摄像机成功,发布消息

	err = tx.Commit().Error
	return err
}

func (camera *Camera) SelectById(cameraId string) (rows int64, err error) {
	dbselect := db.Where("id = ?", cameraId).First(&camera)
	if dbselect.Error != nil || dbselect.RowsAffected == 0 {
		return 0, err
	}
	return dbselect.RowsAffected, nil
}

func (camera *Camera) FindAll() (cams []Camera, err error) {
	dberr := db.Table("cameras").Find(&cams)
	if dberr.Error != nil {
		return nil, err
	}
	return cams, nil
}

func (camera *Camera) FindAllMap() map[string]Camera {
	m := make(map[string]Camera,0)
	cams, err := camera.FindAll()
	if err ==nil && cams !=nil {
		for _,model :=range cams {
			m[model.Id] = model
		}
	}
	return m
}

func (camera *Camera) Update() (err error) {
	logger.Debug("camera:",camera)
	longitude := fmt.Sprintf("%3.4f",camera.Longitude)
	latitude := fmt.Sprintf("%3.4f",camera.Latitude)
	voiceEnable := 0
	if camera.VoiceEnable {
		voiceEnable = 1
	}
	sql := "update cameras set name='"+camera.Name+"',alias='"+camera.Alias+"',type="+strconv.Itoa(camera.Type)+",addr='"+camera.Addr+"',longitude="+longitude+",latitude="+latitude+",rtsp='"+camera.Rtsp+"',ip='"+camera.Ip+"',port="+strconv.Itoa(camera.Port)+",username='"+camera.Username+"',password='"+camera.Password+"',brand='"+camera.Brand+"',reserved='"+camera.Reserved+"',run_server_id='"+camera.RunServerId+"',resolution_width="+strconv.Itoa(camera.ResolutionWidth)+",resolution_height="+strconv.Itoa(camera.ResolutionHeight)+",voiceEnable="+strconv.Itoa(voiceEnable)+",voiceId='"+camera.VoiceId+"' where id='"+camera.Id+"'"

	if err := db.Exec(sql).Error; err != nil {
		return err
	}

	return nil
}

func (camera *Camera) UpdateRunEnable(cameraId string,runEnable bool) bool{
	result := db.Exec("update cameras set run_enable=? where id=?", runEnable, cameraId)
	if result.Error !=nil {
		return false
	}
	return result.RowsAffected>0
}

func (camera *Camera) ChangeRunType(cameraId string,runType int) bool{
	result := db.Exec("update cameras set run_type=? where id=?", runType, cameraId)
	if result.Error !=nil {
		return false
	}
	return result.RowsAffected>0
}

func (camera *Camera) UpdateIsRunningState(cameraId string,isRunning bool) bool{
	isRunningStr := "0"
	if isRunning {
		isRunningStr = "1"
	}
	result := db.Exec("update cameras set is_running="+isRunningStr+" where id='"+cameraId+"'")
	if result.Error !=nil {
		return false
	}
	return result.RowsAffected>0
}

func (camera *Camera) UpdateIsRunningAll(camIds []string) bool {
	analyServerId := config.Server.AnalyServerId
	sql := "update cameras set is_running=0 where run_server_id='"+analyServerId+"'"
	uIds := ""
	if camIds != nil {
		for _,id := range camIds {
			uIds = uIds + "'"+id+"',"
		}
	}
	uIds = strings.Trim(uIds, ",")
	if uIds != "" {
		sql += " and id not in ("+uIds+");update cameras set is_running=1 where run_server_id='"+analyServerId+"' and id in ("+uIds+");"
	}
	logger.Debug("UpdateIsRunningAll sql:", sql)
	result := db.Exec(sql)
	if result.Error !=nil {
		return false
	}
	return true
}

func (camera *Camera) UpdateSnapshot(cameraId string, snapshot string) bool{
	result := db.Exec("update cameras set snapshot_url=? where id=?", snapshot, cameraId)
	if result.Error !=nil {
		return false
	}
	return result.RowsAffected>0
}

func (camera *Camera) Delete(cid string) (int64, error) {
	var err error
	tx := GetDB().Begin()
	defer func() {
		if err !=nil && tx !=nil{
			tx.Rollback()
		}
	}()
	dbdel := tx.Exec("delete from cameras where id=?",cid)
	err = dbdel.Error
	if err != nil || dbdel.RowsAffected == 0 {
		return 0, err
	}
	if err = tx.Exec("delete from camera_polygon where camera_id=?",cid).Error;err !=nil{
		return 0,err
	}
	if err = tx.Exec("delete from camera_area where cameraId=?",cid).Error;err !=nil {
		return 0,err
	}
	if err = tx.Exec("delete from camera_sensor where camera_id=?",cid).Error;err !=nil {
		return 0,err
	}
	tx.Commit()
	//发布数据更改消息
	return dbdel.RowsAffected, nil

}