package main

import (
	"context"
	"flag"
	"os"
	"os/signal"
	"syscall"
	"vamicro/config"
	"vamicro/gb28181-service/cache"
	"vamicro/gb28181-service/controllers"
	"vamicro/gb28181-service/models"

	"basic.com/valib/bhomeclient.git"
	"basic.com/valib/bhomedbapi.git"
	"basic.com/valib/logger.git"
	vaversion "basic.com/valib/version.git"
)

var (
	procName = "gb28181-service"
	proc     = &bhomeclient.ProcInfo{
		Name: procName, //进程名称
		ID:   procName, //进程id
		Info: "",       //进程的描述信息，用于区分同一进程名称下多个进程
	}
	env = flag.String("e", "pro", "")
)

func init() {
	flag.Parse()
	vaversion.Usage()

	config.Init(*env)
	// 日志初始化
	var logFile = config.LogConf.Path + "vamicro-" + procName + ".log"
	logger.InitLogger(logFile, config.LogConf.Level, config.LogConf.MaxSize, config.LogConf.MaxBackups, config.LogConf.MaxAge)
	logger.Info("log init success !")
}

func main() {
	flag.Parse()

	models.Init()
	defer models.CloseDB()
	ctx, cancel := context.WithCancel(context.Background())
	fm, pubTopics := initFuncMap()
	var reg = &bhomeclient.RegisterInfo{
		Proc:     *proc,
		Channel:  nil,
		PubTopic: pubTopics,
		SubTopic: []string{bhomeclient.Proc_Scene_Service, bhomeclient.Proc_System_Service},
	}

	q := make(chan os.Signal, 1)
	signal.Notify(q, os.Interrupt, os.Kill, syscall.SIGTERM)

	ms, err := bhomeclient.NewMicroNode(ctx, q, config.Server.AnalyServerId, reg, nil)
	if err != nil {
		return
	}
	bhomedbapi.InitLog(nil)
	bhomedbapi.InitGetNetNode(ms.GetLocalNetNodeByTopic)
	bhomedbapi.InitDoReq(ms.RequestOnly)

	go dealSubMsg(ctx, ms)

	go ms.StartServer(fm)
	cache.InitDataCache()

	<-q
	ms.DeRegister()
	cancel()
	ms.Free()
}

//处理订阅消息
func dealSubMsg(ctx context.Context, ms *bhomeclient.MicroNode) {

	for {
		select {
		case <-ctx.Done():
			return
		case msg := <-ms.SubCh:
			logger.Debug("recv sub msg:", msg)

			cache.UpdateBySub(msg)
		}
	}
}

const urlPrefix = "/data/api-v"

func initFuncMap() (map[string]bhomeclient.MicroFunc, []string) {
	funcMap := make(map[string]bhomeclient.MicroFunc)
	gc := new(controllers.Gb28181Controller)
	ac := new(controllers.AreaController)

	funcMap[urlPrefix+"/gb28181/tree"] = gc.CameraGb28181Tree
	funcMap[urlPrefix+"/gb28181/camera/showAll"] = gc.GetAllCameras
	funcMap[urlPrefix+"/gb28181/camera/update"] = gc.CameraUpdate
	funcMap[urlPrefix+"/gb28181/findAreaByParentId"] = gc.FindAreaByParentId
	funcMap[urlPrefix+"/gb28181/area/update"] = ac.AreaUpdate
	funcMap[urlPrefix+"/gb28181/camera/updateCameraArea"] = ac.UpdateCameraArea

	funcMap[urlPrefix+"/gb28181/camera/show"] = gc.CameraSel
	funcMap[urlPrefix+"/gb28181/camera/getCamerasByRunType"] = gc.GetCamerasByRunType
	funcMap[urlPrefix+"/gb28181/camera/updateRunEnable"] = gc.UpdateRunEnable
	funcMap[urlPrefix+"/gb28181/camera/changeRunType"] = gc.ChangeRunType
	funcMap[urlPrefix+"/gb28181/camera/updateIsRunningState"] = gc.UpdateIsRunningState
	funcMap[urlPrefix+"/gb28181/camera/updateIsRunningAll"] = gc.UpdateIsRunningAll
	funcMap[urlPrefix+"/gb28181/camera/updateSnapshotUrl"] = gc.UpdateSnapshot

	funcMap[urlPrefix+"/gb28181/camera/findOnMap"] = gc.FindOnMap
	funcMap[urlPrefix+"/gb28181/camera/updatePos"] = gc.UpdatePos
	funcMap[urlPrefix+"/gb28181/camera/nodeCamera"] = gc.NodeCamera
	funcMap[urlPrefix+"/gb28181/camera/coord"] = gc.Coord
	funcMap[urlPrefix+"/gb28181/camera/updateCoord"] = gc.UpdateCoord

	funcMap[urlPrefix+"/gb28181/newGbId"] = gc.NewGbId
	funcMap[urlPrefix+"/gb28181/configEdit"] = gc.Gb28181ConfigEdit
	funcMap[urlPrefix+"/gb28181/saasConfigEdit"] = gc.SaasGb28181ConfigEdit
	funcMap[urlPrefix+"/gb28181/configShow"] = gc.Gb28181ConfigShow
	funcMap[urlPrefix+"/gb28181/getAllSubServer"] = gc.GetAllSubServer
	funcMap[urlPrefix+"/gb28181/treeDelete"] = gc.Gb28181TreeDelete
	funcMap[urlPrefix+"/gb28181/treeRefresh"] = gc.Gb28181TreeRefresh
	funcMap[urlPrefix+"/gb28181/getAllChannel"] = gc.GetAllChannel
	funcMap[urlPrefix+"/gb28181/saveGb28181CamTree"] = gc.SaveGb28181CamTree
	funcMap[urlPrefix+"/gb28181/camera/capture"] = gc.Capture
	funcMap[urlPrefix+"/gb28181/treeSelected"] = gc.TreeSelected
	funcMap[urlPrefix+"/gb28181/getSyncCameraData"] = gc.GetSyncCameraData
	funcMap[urlPrefix+"/gb28181/saveNotExistOrUpdate"] = gc.SaveNotExistOrUpdate
	funcMap[urlPrefix+"/gb28181/getVssTable"] = gc.GetGb28181VssTable

	var pubTopics []string
	for key, _ := range funcMap {
		pubTopics = append(pubTopics, key)
	}
	return funcMap, pubTopics
}