package cache

import (
	"basic.com/pubsub/cache.git/shardmap"
	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/bhomeclient.git"
	"basic.com/valib/bhomedbapi.git"
	"basic.com/valib/c_bhomebus.git/proto/source/bhome_msg"
	"basic.com/valib/logger.git"
	"encoding/json"
	"errors"
)

var cMap *shardmap.ShardMap

const (
	SERVER_KEY     = "SERVERINFO"     //服务器信息
	LOCALCAMERAS   = "LOCALCAMERAS"   //本地摄像机列表
	GB28181CAMERAS = "GB28181CAMERAS" //国标摄像机列表
	SCENERULES     = "SCENERULES"     //摄像机场景列表
)

func UpdateBySub(msg *bhome_msg.MsgPublish) {
	switch string(msg.Topic) {
	case bhomeclient.Proc_System_Service:
		initServerInfo()
	case bhomeclient.Proc_Camera_Service, bhomeclient.Proc_Gb28181_Service, bhomeclient.Proc_Scene_Service:
		var protoMsg protomsg.DbChangeMessage
		if err := json.Unmarshal(msg.Data, &protoMsg); err == nil {
			if protoMsg.Table == protomsg.TableChanged_T_Camera {
				// initLocalCameras()
			} else if protoMsg.Table == protomsg.TableChanged_T_Camera {
				// initGb28181Cameras()
			} else if protoMsg.Table == protomsg.TableChanged_T_CameraRule {
				initCameraRules()
			}
		}
	}
}

//缓存一些其他数据库的数据，并通过订阅消息实时更新
func InitDataCache() {
	cMap = shardmap.New(uint8(32))

	initServerInfo() //初始化服务器配置信息

	// 代码中不需要摄像机信息
	// initLocalCameras()
	// initGb28181Cameras()

	initCameraRules()
}

func initServerInfo() {
	var api bhomedbapi.SysSetApi
	b, s := api.GetServerInfo()
	logger.Debug("GetServerInfo b:", b, "s:", s)
	if b && s.ServerId != "" {
		cMap.Set(SERVER_KEY, s)
	}
}

func GetServerInfo() (conf protomsg.LocalConfig, err error) {
	config, b := cMap.Get(SERVER_KEY)
	if b {
		return config.(protomsg.LocalConfig), nil
	} else {
		initServerInfo()
		nc, nb := cMap.Get(SERVER_KEY)
		if nb {
			return nc.(protomsg.LocalConfig), nil
		} else {
			return conf, errors.New("conf not found")
		}
	}
}

func initLocalCameras() {
	var camApi bhomedbapi.CameraApi
	localCams := camApi.FindAll("", "", "", "") //获取所有本地摄像机
	cMap.Set(LOCALCAMERAS, localCams)
}

func GetLocalCameras() ([]protomsg.Camera, error) {
	cams, b := cMap.Get(LOCALCAMERAS)
	if b && cams != nil {
		return cams.([]protomsg.Camera), nil
	} else {
		initLocalCameras()
		nc, nb := cMap.Get(LOCALCAMERAS)
		if nb {
			return nc.([]protomsg.Camera), nil
		} else {
			return nil, errors.New("no cached localCameras")
		}
	}
}

func initGb28181Cameras() {
	var camApi bhomedbapi.Gb28181Api
	gbCams := camApi.FindAll("", "", "", "") //获取所有本地摄像机
	cMap.Set(GB28181CAMERAS, gbCams)
}

func GetGb28181Cameras() ([]protomsg.Camera, error) {
	cams, b := cMap.Get(GB28181CAMERAS)
	if b && cams != nil {
		return cams.([]protomsg.Camera), nil
	} else {
		initGb28181Cameras()
		nc, nb := cMap.Get(GB28181CAMERAS)
		if nb {
			return nc.([]protomsg.Camera), nil
		} else {
			return nil, errors.New("no cached gb28181Cameras")
		}
	}
}

func initCameraRules() {
	var api bhomedbapi.CameraRuleApi
	b, allCamRules := api.FindAll()
	logger.Debug("initCameraRules b:", b, " len(allCamRules):", len(allCamRules))
	if b {
		cMap.Set(SCENERULES, allCamRules)
	}
}

func GetCameraRules() ([]protomsg.CameraAndRules, error) {
	allRules, b := cMap.Get(SCENERULES)
	if b {
		return allRules.([]protomsg.CameraAndRules), nil
	} else {
		initCameraRules()
		nr, nb := cMap.Get(SCENERULES)
		if nb {
			return nr.([]protomsg.CameraAndRules), nil
		} else {
			return nil, errors.New("no CameraAndRules cached")
		}
	}
}