package gateway

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/modules"

	"go_dreamfactory/lego/base"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/lego/sys/log"
)

/*
模块名:gateway
描述:提供客户端网关路由服务 管理用户socket对象 以及分发用户消息
开发:李伟
*/
func NewModule() core.IModule {
	m := new(Gateway)
	return m
}

type Gateway struct {
	modules.ModuleBase
	service   base.IRPCXService
	wsService *WSServiceComp //websocket 服务组件 提供websocket服务监听
	agentMgr  *AgentMgrComp  //用户代理对象管理组件 管理用户socekt对象
	configure *configureComp
}

//模块名
func (this *Gateway) GetType() core.M_Modules {
	return comm.ModuleGate
}

//模块自定义参数
func (this *Gateway) NewOptions() (options core.IModuleOptions) {
	return new(Options)
}

//提供服务对象获取接口
func (this *Gateway) Service() base.IRPCXService {
	return this.service
}

//模块初始化函数
func (this *Gateway) Init(service core.IService, module core.IModule, options core.IModuleOptions) (err error) {
	err = this.ModuleBase.Init(service, module, options)
	this.service = service.(base.IRPCXService)
	return
}

//模块启动函数 注册rpc服务接口提供用户相关的rpc接口服务
func (this *Gateway) Start() (err error) {
	//注册用户绑定uid接口 登录成功后触发
	this.service.RegisterFunctionName(string(comm.Rpc_GatewayAgentBind), this.agentMgr.Bind)
	//注册用户解绑uid接口 登出或则切换账号是触发
	this.service.RegisterFunctionName(string(comm.Rpc_GatewayAgentUnBind), this.agentMgr.UnBind)
	//向用户发送消息接口
	this.service.RegisterFunctionName(string(comm.Rpc_GatewayAgentSendMsg), this.agentMgr.SendMsgToAgent)
	//向多个用户对象发送消息接口
	this.service.RegisterFunctionName(string(comm.Rpc_GatewaySendBatchMsg), this.agentMgr.SendMsgToAgents)
	//向所有用户发送消息接口
	this.service.RegisterFunctionName(string(comm.Rpc_GatewaySendRadioMsg), this.agentMgr.SendMsgToAllAgent)
	//关闭用户socket连接接口
	this.service.RegisterFunctionName(string(comm.Rpc_GatewayAgentClose), this.agentMgr.CloseAgent)
	err = this.ModuleBase.Start()
	return
}

//装备组件
func (this *Gateway) OnInstallComp() {
	this.ModuleBase.OnInstallComp()
	this.agentMgr = this.RegisterComp(new(AgentMgrComp)).(*AgentMgrComp)
	this.wsService = this.RegisterComp(new(WSServiceComp)).(*WSServiceComp)
	this.configure = this.RegisterComp(new(configureComp)).(*configureComp)
}

//有新的连接对象进入
func (this *Gateway) Connect(a IAgent) {
	log.Debugf("[Module.Gateway] have new connect:Ip[%s] SessionId:[%s]", a.IP(), a.SessionId())
	this.agentMgr.Connect(a)
}

//有用户断开连接
func (this *Gateway) DisConnect(a IAgent) {
	log.Debugf("[Module.Gateway] have disConnect:Ip[%s] SessionId:[%s] uid:[%s]", a.IP(), a.SessionId(), a.UserId())
	this.agentMgr.DisConnect(a)
}

//读取消息分发规则
func (this *Gateway) GetMsgDistribRule(mtype, stype string) (rule string, ok bool) {
	return this.configure.GetMsgDistribRule(mtype, stype)
}