forked from ProjectWIND/ProjectWIND
384 lines
12 KiB
Go
384 lines
12 KiB
Go
package core
|
||
|
||
import (
|
||
"ProjectWIND/LOG"
|
||
"ProjectWIND/wba"
|
||
"crypto/rand"
|
||
"encoding/json"
|
||
"fmt"
|
||
)
|
||
|
||
type apiInfo struct{}
|
||
|
||
/*
|
||
关于API的说明：
|
||
|
||
1.所有API请求按照OneBot11标准，使用JSON格式进行数据交换。api命名为由原文档中蛇形命名法改为双驼峰命名法。
|
||
|
||
2.无响应的API请求使用ws协议处理，有响应的API需添加echo字段。
|
||
|
||
3.wind会从配置文件中读取API请求的url，请确保正确填写。
|
||
*/
|
||
|
||
//1.无响应API,使用ws协议处理
|
||
|
||
// SendMsg 发送消息(自动判断消息类型)
|
||
func (a *apiInfo) SendMsg(msg wba.MessageEventInfo, message string, autoEscape bool) {
|
||
// 构建发送消息的JSON数据
|
||
var messageData wba.APIRequestInfo
|
||
|
||
messageType := msg.MessageType
|
||
|
||
messageData.Action = "send_msg"
|
||
switch messageType {
|
||
case "private":
|
||
{
|
||
messageData.Params.UserId = msg.UserId
|
||
break
|
||
}
|
||
case "group":
|
||
{
|
||
messageData.Params.GroupId = msg.GroupId
|
||
break
|
||
}
|
||
default:
|
||
{
|
||
LOG.ERROR("发送消息(SendMsg)时，消息类型错误: %v", messageType)
|
||
}
|
||
}
|
||
messageData.Params.Message = message
|
||
messageData.Params.AutoEscape = autoEscape
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("发送消息时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
// 发送消息
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("发送消息时，发送失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("发送消息(SendMsg)(至：%v-%v:%v-%v):%v", msg.MessageType, msg.GroupId, msg.UserId, msg.Sender.Nickname, message)
|
||
return
|
||
}
|
||
|
||
// SendPrivateMsg 发送私聊消息
|
||
func (a *apiInfo) SendPrivateMsg(msg wba.MessageEventInfo, message string, autoEscape bool) {
|
||
// 构建发送消息的JSON数据
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "send_private_msg"
|
||
messageData.Params.UserId = msg.UserId
|
||
messageData.Params.Message = message
|
||
messageData.Params.AutoEscape = autoEscape
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("发送消息(SendPrivateMsg)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
// 发送消息
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("发送消息(SendPrivateMsg)时，发送失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("发送消息(SendPrivateMsg)(至：%v-%v:%v-%v):%v", msg.MessageType, msg.GroupId, msg.UserId, msg.Sender.Nickname, message)
|
||
return
|
||
}
|
||
|
||
// SendGroupMsg 发送群消息
|
||
func (a *apiInfo) SendGroupMsg(msg wba.MessageEventInfo, message string, autoEscape bool) {
|
||
// 构建发送消息的JSON数据
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "send_group_msg"
|
||
messageData.Params.GroupId = msg.GroupId
|
||
messageData.Params.Message = message
|
||
messageData.Params.AutoEscape = autoEscape
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("发送消息(SendGroupMsg)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
// 发送消息
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("发送消息(SendGroupMsg)时，发送失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("发送消息(SendGroupMsg)(至：%v-%v:%v-%v):%v", msg.MessageType, msg.GroupId, msg.UserId, msg.Sender.Nickname, message)
|
||
return
|
||
}
|
||
|
||
// DeleteMsg 撤回消息
|
||
func (a *apiInfo) DeleteMsg(msg wba.MessageEventInfo) {
|
||
// 构建删除消息的JSON数据
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "delete_msg"
|
||
messageData.Params.MessageId = msg.MessageId
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("撤回消息(DeleteMsg)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("撤回消息(DeleteMsg)时，发送失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("撤回消息(DeleteMsg):[id:%v]%v", msg.MessageId, msg.RawMessage)
|
||
return
|
||
}
|
||
|
||
// SendLike 发送赞
|
||
func (a *apiInfo) SendLike(userId int64, times int) {
|
||
// 构建发送赞的JSON数据
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "send_like"
|
||
messageData.Params.UserId = userId
|
||
messageData.Params.Times = times
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("发送赞(SendLike)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("发送赞(SendLike)时，发送失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("发送赞(SendLike)(至：%v):%v", userId, times)
|
||
return
|
||
}
|
||
|
||
// SetGroupKick 将指定用户移出群聊(需要群主或管理员权限)
|
||
func (a *apiInfo) SetGroupKick(groupId int64, userId int64, rejectAddRequest bool) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_kick"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.UserId = userId
|
||
messageData.Params.RejectAddRequest = rejectAddRequest
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("移出群聊(SetGroupKick)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("移出群聊(SetGroupKick)时，发送失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("移出群聊(SetGroupKick)(从：%v-%v):%v", groupId, userId, rejectAddRequest)
|
||
return
|
||
}
|
||
|
||
// SetGroupBan 将指定用户禁言(需要群主或管理员权限)
|
||
func (a *apiInfo) SetGroupBan(groupId int64, userId int64, duration int32) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_ban"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.UserId = userId
|
||
messageData.Params.Duration = duration
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("禁言群成员(SetGroupBan)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("禁言群成员(SetGroupBan)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("禁言群成员(SetGroupBan)(在：%v-%v):%v", groupId, userId, duration)
|
||
return
|
||
}
|
||
|
||
// SetGroupWholeBan 设置全员禁言(需要群主或管理员权限)
|
||
func (a *apiInfo) SetGroupWholeBan(groupId int64, enable bool) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_whole_ban"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.Enable = enable
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("设置全员禁言(SetGroupWholeBan)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("设置全员禁言(SetGroupWholeBan)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("设置全员禁言(SetGroupWholeBan)(在：%v):%v", groupId, enable)
|
||
return
|
||
}
|
||
|
||
// SetGroupAdmin 设置群管理员(需要群主权限)
|
||
func (a *apiInfo) SetGroupAdmin(groupId int64, userId int64, enable bool) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_admin"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.UserId = userId
|
||
messageData.Params.Enable = enable
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("设置群管理员(SetGroupAdmin)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("设置群管理员(SetGroupAdmin)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("设置群管理员(SetGroupAdmin)(在：%v-%v):%v", groupId, userId, enable)
|
||
return
|
||
}
|
||
|
||
// SetGroupCard 设置群名片(需要群主或管理员权限)
|
||
func (a *apiInfo) SetGroupCard(groupId int64, userId int64, card string) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_card"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.UserId = userId
|
||
messageData.Params.Card = card
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("设置群名片(SetGroupCard)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("设置群名片(SetGroupCard)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("设置群名片(SetGroupCard)(在：%v-%v):%v", groupId, userId, card)
|
||
return
|
||
}
|
||
|
||
// SetGroupName 设置群名称(可能需要群主或管理员权限)
|
||
func (a *apiInfo) SetGroupName(groupId int64, groupName string) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_name"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.GroupName = groupName
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("设置群名称(SetGroupName)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("设置群名称(SetGroupName)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("设置群名称(SetGroupName)(在：%v):%v", groupId, groupName)
|
||
return
|
||
}
|
||
|
||
// SetGroupLeave 退出群聊
|
||
func (a *apiInfo) SetGroupLeave(groupId int64, isDismiss bool) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_leave"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.IsDismiss = isDismiss
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("退出群聊(SetGroupLeave)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("退出群聊(SetGroupLeave)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("退出群聊(SetGroupLeave)(在：%v):%v", groupId, isDismiss)
|
||
return
|
||
}
|
||
|
||
// SetGroupSpecialTitle 设置群专属头衔(需要群主权限)
|
||
func (a *apiInfo) SetGroupSpecialTitle(groupId int64, userId int64, specialTitle string, duration int32) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_special_title"
|
||
messageData.Params.GroupId = groupId
|
||
messageData.Params.UserId = userId
|
||
messageData.Params.SpecialTitle = specialTitle
|
||
messageData.Params.Duration = duration
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("设置群特殊头衔(SetGroupSpecialTitle)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("设置群特殊头衔(SetGroupSpecialTitle)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("设置群特殊头衔(SetGroupSpecialTitle)(在：%v-%v):%v-%v", groupId, userId, specialTitle, duration)
|
||
return
|
||
}
|
||
|
||
// SetFriendAddRequest 处理加好友请求
|
||
func (a *apiInfo) SetFriendAddRequest(flag string, approve bool, remark string) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_friend_add_request"
|
||
messageData.Params.Flag = flag
|
||
messageData.Params.Approve = approve
|
||
messageData.Params.Remark = remark
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("处理加好友请求(SetFriendAddRequest)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("处理加好友请求(SetFriendAddRequest)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("处理加好友请求(SetFriendAddRequest)(在：%v):%v-%v-%v", flag, approve, remark)
|
||
return
|
||
}
|
||
|
||
// SetGroupAddRequest 处理加群请求/邀请
|
||
func (a *apiInfo) SetGroupAddRequest(flag string, subType string, approve bool, reason string) {
|
||
var messageData wba.APIRequestInfo
|
||
messageData.Action = "set_group_add_request"
|
||
messageData.Params.Flag = flag
|
||
messageData.Params.SubType = subType
|
||
messageData.Params.Approve = approve
|
||
messageData.Params.Reason = reason
|
||
messageJson, err := json.Marshal(messageData)
|
||
if err != nil {
|
||
LOG.ERROR("处理加群请求/邀请(SetGroupAddRequest)时，构建JSON数据失败: %v", err)
|
||
return
|
||
}
|
||
err = wsAPI(messageJson)
|
||
if err != nil {
|
||
LOG.ERROR("处理加群请求/邀请(SetGroupAddRequest)时，执行失败: %v", err)
|
||
return
|
||
}
|
||
LOG.INFO("处理加群请求/邀请(SetGroupAddRequest)(在：%v-%v-%v):%v", flag, subType, approve, reason)
|
||
return
|
||
}
|
||
|
||
// 2.有响应API，需添加echo字段
|
||
|
||
func (a *apiInfo) GetLoginInfo(flag string, approve bool) {
|
||
|
||
}
|
||
|
||
var AppApi apiInfo
|
||
|
||
func GenerateUUID() (string, error) {
|
||
uuid := make([]byte, 16)
|
||
_, err := rand.Read(uuid)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
// 设置UUID版本号（版本4），将第6字节的高4位设置为0100
|
||
uuid[6] = (uuid[6] & 0x0F) | 0x40
|
||
// 设置UUID变体（RFC 4122规范定义的变体），将第8字节的高4位设置为10
|
||
uuid[8] = (uuid[8] & 0x3F) | 0x80
|
||
|
||
return fmt.Sprintf("%x-%x-%x-%x-%x", uuid[0:4], uuid[4:6], uuid[6:8], uuid[8:10], uuid[10:]), nil
|
||
}
|