package service

import (
	"time"
	"vamicro/config"
	"vamicro/saas-service/service/nodeService"

	"basic.com/valib/go-aiot.git/aiotProto/aiot"
	"basic.com/valib/go-aiot.git/client"
	"basic.com/valib/logger.git"
	"go.uber.org/zap"
)

const DefaultNetRetry = 15 * time.Second

var CliSrv tcpCli

type tcpCli struct {
	Cli      *client.Client
	IsMaster bool
}

func StartSrv(log *zap.SugaredLogger) {
	// 初始化节点
	nodeList := make([]*aiot.DeviceNode, 0)
	if len(nodeService.Node.NodeLists) > 0 {
		for _, node := range nodeService.Node.NodeLists {
			deviceNode := &aiot.DeviceNode{}
			deviceNode.DeviceId = node.NodeId
			nodeList = append(nodeList, deviceNode)
		}
	}
	deviceRegister := &aiot.DeviceRegister{
		DeviceId:    nodeService.Node.NodeId,
		ServerName:  config.Server.ServerName,
		ClusterId:   nodeService.Node.ClusterId,
		ClusterName: nodeService.Node.ClusterName,
		VirtualIp:   nodeService.Node.VirtualIp,
		DeviceList:  nodeList,
	}
	CliSrv.Cli = client.NewClient(config.ClusterSet.SaasReportUrl, nodeService.Node.NodeId, deviceRegister, cliCon, log)
	CliSrv.IsMaster = nodeService.Node.IsMaster
	if nodeService.Node.IsMaster {
		CliSrv.Cli.InitClient()
		CliSrv.Cli.StartSrv()
	} else {
		CliSrv.Cli.SetState(client.StateDisconnected)
	}
}

func WatchSrv(log *zap.SugaredLogger) {
	ticker := time.NewTicker(DefaultNetRetry)
	for {
		select {
		case <-ticker.C:
			CliSrv.IsMaster = nodeService.Node.IsMaster
			if !nodeService.Node.IsMaster {
				// 非主节点
				logger.Debug("client is not master... ", nodeService.Node.NodeId)
				// 已连接，则强制断开
				if CliSrv.Cli != nil && !CliSrv.Cli.IsClosed() && CliSrv.Cli.GetState() != client.StateInit {
					CliSrv.Cli.Close()
				}
			} else {
				// 是主节点
				logger.Debug("client is is master connect status: ", CliSrv.Cli.GetState())
				// 启动服务
				if !CliSrv.Cli.IsConnected() {
					go StartSrv(log)
				}
			}
		}
	}
}