package models

import (
	"fmt"
	"strconv"
	"strings"
	"vamicro/config"

	"basic.com/valib/logger.git"
)

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 Task struct {
	TaskId   int64  `db:"task_id" json:"taskId"`
	TaskName string `db:"task_name" json:"taskName"`
}

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

}