package service

import (
	"basic.com/valib/bhomeclient.git"
	"basic.com/valib/bhomedbapi.git"
	"basic.com/valib/serf.git/serf"
	"encoding/json"
	"errors"
	"github.com/satori/go.uuid"
	"time"
	"vamicro/config"
	"vamicro/system-service/models"
	sc "vamicro/system-service/serf"
	dvo "vamicro/devicemanage-service/vo"
	"vamicro/system-service/sys"
	"vamicro/system-service/vo"
)

func init() {
	sc.RegisterRpcHandles(
		DevAuthApply,
		RemoteCreateCluster,
		RemoteSearchCluster,
		RemoteGetSearchNodes,
		RemoteJoinCluster,
		RemoteReboot,
		RemoteUninstall,
		RemoteUpgrade,
		RemoteSysUpdate,
	)
}

func DevAuthApply(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	type applyArg struct {
		Key 		string 		`json:"key"`
		FromDevId 	string 		`json:"fromDevId"`
		FromIp 		string 		`json:"fromIp"`
	}
	var ret bhomeclient.Reply
	var retErr error
	var reqBody applyArg
	if err := json.Unmarshal(arg.Data, &reqBody); err == nil && reqBody.Key != "" && reqBody.FromIp != "" && reqBody.FromDevId != "" {
		//1.验证请求的key是否匹配
		var config models.AuthConfig
		ic, _ := config.Select()
		if ic > 0 {
			if config.AuthType == models.AuthType_Key && reqBody.Key != config.Password {
				ret.Msg = "申请密钥不匹配"
			}
		}

		//2.写入authDevice
		var da models.AuthDevice
		i, _ := da.FindByDevId(reqBody.FromDevId)
		if i == 0 { //未申请过
			tmp := models.AuthDevice{
				Id: uuid.NewV4().String(),
				DevId: reqBody.FromDevId,
				DevIp: reqBody.FromIp,
				ApplyKey: reqBody.Key,
				CreateTime: time.Now().Format("2006-01-02 15:04:05"),
			}
			if tmp.Insert() {
				ret.Success = true
				ret.Msg = "添加成功，待审核"
			} else {
				ret.Msg = "添加失败"
			}
		} else { //已申请过
			if da.Status == models.AuthStatus_Agreed {
				ret.Success = true
				ret.Msg = "已通过，无需重复申请"
			} else if da.Status == models.AuthStatus_AuthCanceled {
				ret.Msg = "已取消授权"
			} else {
				ret.Msg = "请等候审核"
			}
		}
	} else {
		ret.Msg = "参数有误"
	}
	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteCreateCluster(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	var param dvo.CreateClusterArg
	if err := json.Unmarshal(arg.Data, &param);err != nil {
		retErr = err
	}
	if len(param.Password) != 6 || param.ClusterName == "" {
		ret.Msg = "参数有误"
	} else {
		sv := NewClusterService(sc.GetBusHandle())
		b, newCluterId := sv.Create(param.ClusterName, param.Password, param.VirtualIp)
		if b {
			param.ClusterId = newCluterId
			ret.Data = param
		} else {
			retErr = errors.New("创建集群失败")
		}
	}

	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteSearchCluster(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	var param dvo.SearchClusterArg
	if err := json.Unmarshal(arg.Data, &param);err != nil {
		retErr = err
	}
	if len(param.Password) != 6 {
		retErr = errors.New("密码错误")
	} else {
		searchNum := uuid.NewV4().String()
		sv := NewClusterService(sc.GetBusHandle())
		retErr = sv.SearchByPwd(param.Password)
		if retErr !=nil {
			ret.Success = true
			ret.Msg = "搜索中..."
		} else {
			ret.Success = true
			ret.Data = searchNum
		}
	}

	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteGetSearchNodes(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	sv := NewClusterService(sc.GetBusHandle())
	nodes := sv.SearchNodes()
	if nodes !=nil && len(nodes)>0 {
		var nodeArr []interface{}
		for _,n :=range nodes {
			nodeArr = append(nodeArr, n)
		}

		ret.Data = nodeArr
	} else {
		ret.Data = []interface{}{}
	}
	ret.Success = true
	ret.Msg = "查询成功"
	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteJoinCluster(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	var param dvo.JoinClusterArg
	if err := json.Unmarshal(arg.Data, &param);err != nil {
		retErr = err
	} else {
		sv := NewClusterService(sc.GetBusHandle())
		cj := vo.ClusterJoinVo{
			ClusterId: param.ClusterId,
			Password: param.Password,
			NodeIps: param.NodeIps,
		}
		if b,err := sv.JoinCluster(&cj); b{
			ret.Success = true
			ret.Msg = "加入成功"
		} else {
			ret.Success = false
			retErr = err
		}
	}

	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteReboot(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	var param dvo.RebootArg
	if err := json.Unmarshal(arg.Data, &param);err != nil {
		retErr = err
	} else {
		if isOk, msg := sys.Reboot(); !isOk {
			ret.Msg = msg
		} else {
			ret.Success = true
			ret.Msg = "正在重启..."
		}
	}
	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteUninstall(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	var param dvo.UninstallArg
	if err := json.Unmarshal(arg.Data, &param);err != nil {
		retErr = err
	} else {
		if param.Type ==1 { //卸载算法
			api := bhomedbapi.SdkApi{}
			b, d := api.Delete(param.Id)
			ret.Success = b
			ret.Data = d
		} else if param.Type == 2 {  //卸载应用
			api := bhomedbapi.AppApi{}
			b, d := api.Delete(param.Id)
			ret.Success = b
			ret.Data = d
		}
	}

	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteUpgrade(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	var param dvo.UpgradeArg
	if err := json.Unmarshal(arg.Data, &param);err != nil {
		retErr = err
	} else {
		api := bhomedbapi.AppApi{}
		b, d := api.Upgrade(param.Id)
		ret.Success = b
		ret.Data = d
	}

	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}

func RemoteSysUpdate(arg sc.RpcParamTopic) ([]serf.NodeResponse, error) {
	var ret bhomeclient.Reply
	var retErr error
	var param dvo.SysUpdateArg
	if err := json.Unmarshal(arg.Data, &param);err != nil {
		retErr = err
	} else {
		//调用version-control的接口，执行升级操作
		ret.Msg = "build..."
	}

	nr := serf.NodeResponse{
		From: config.Server.AnalyServerId,
	}
	if data, err := json.Marshal(ret); err != nil {
		retErr = err
	} else {
		nr.Payload = data
	}

	return append([]serf.NodeResponse{}, nr), retErr
}