go-wxhelper/common/current/robot.go

package current

import (
	"go-wechat/model/dto"
	"go-wechat/plugin"
)

// robotInfo
// @description: 机器人信息
type robotInfo struct {
	info           dto.RobotUserInfo
	MessageHandler plugin.MessageHandler // 启用的插件
}

// 当前接入的机器人信息
var ri robotInfo

// SetRobotInfo
// @description: 设置机器人信息
// @param info
func SetRobotInfo(info dto.RobotUserInfo) {
	ri.info = info
}

// GetRobotInfo
// @description: 获取机器人信息
// @return dto.RobotUserInfo
func GetRobotInfo() dto.RobotUserInfo {
	return ri.info
}

// GetRobotMessageHandler
// @description: 获取机器人插件信息
// @return robotInfo
func GetRobotMessageHandler() plugin.MessageHandler {
	return ri.MessageHandler
}

// SetRobotMessageHandler
// @description: 设置机器人插件信息
// @param handler
func SetRobotMessageHandler(handler plugin.MessageHandler) {
	ri.MessageHandler = handler
}