mirror of
https://github.com/fumiama/NanoBot.git
synced 2026-06-05 10:40:24 +08:00
394 lines
13 KiB
Go
394 lines
13 KiB
Go
package nano
|
||
|
||
import (
|
||
"encoding/base64"
|
||
"encoding/json"
|
||
"reflect"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/pkg/errors"
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
var (
|
||
ErrEmptyMessagePost = errors.New("empty message post")
|
||
)
|
||
|
||
// Message 消息对象
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#%E6%B6%88%E6%81%AF%E5%AF%B9%E8%B1%A1-message
|
||
type Message struct {
|
||
ID string `json:"id"`
|
||
ChannelID string `json:"channel_id"`
|
||
GuildID string `json:"guild_id"`
|
||
GroupOpenID string `json:"group_openid"`
|
||
FileUUID string `json:"file_uuid"`
|
||
FileInfo string `json:"file_info"`
|
||
Content string `json:"content"`
|
||
Timestamp time.Time `json:"timestamp"`
|
||
EditedTimestamp time.Time `json:"edited_timestamp"`
|
||
FileInfoTTL int `json:"ttl"`
|
||
MentionEveryone bool `json:"mention_everyone"`
|
||
Author *User `json:"author"`
|
||
Attachments []MessageAttachment `json:"attachments"`
|
||
Embeds []MessageEmbed `json:"embeds"`
|
||
Member *Member `json:"member"`
|
||
Ark *MessageArk `json:"ark"`
|
||
SeqInChannel string `json:"seq_in_channel"`
|
||
MessageReference *MessageReference `json:"message_reference"`
|
||
SrcGuildID string `json:"src_guild_id"`
|
||
Data *struct {
|
||
MessageAudit *MessageAudited `json:"message_audit,omitempty"`
|
||
} `json:"data,omitempty"`
|
||
}
|
||
|
||
// "=> ｷﾞﾙﾄﾞ:", ctx.Message.GuildID+", 频道:", ctx.Message.ChannelID+", 用户:", ctx.Message.Author.Username+"("+ctx.Message.Author.ID+"), 内容:", ctx.Message.Content
|
||
func (m *Message) String() string {
|
||
sb := strings.Builder{}
|
||
if m.SeqInChannel != "" {
|
||
sb.WriteByte('[')
|
||
sb.WriteString(m.SeqInChannel)
|
||
sb.WriteByte(']')
|
||
}
|
||
sb.WriteString(m.ID)
|
||
sb.WriteString(" ｷﾞﾙﾄﾞ: ")
|
||
sb.WriteString(m.GuildID)
|
||
if m.SrcGuildID != "" {
|
||
sb.WriteString(", 元ｷﾞﾙﾄﾞ: ")
|
||
sb.WriteString(m.SrcGuildID)
|
||
}
|
||
sb.WriteString(", 频道: ")
|
||
sb.WriteString(m.ChannelID)
|
||
if m.Author != nil {
|
||
sb.WriteString(", 用户: ")
|
||
sb.WriteString(m.Author.Username)
|
||
sb.WriteByte('(')
|
||
sb.WriteString(m.Author.ID)
|
||
sb.WriteByte(')')
|
||
if m.Author.Bot {
|
||
sb.WriteString("(机器人)")
|
||
}
|
||
} else {
|
||
sb.WriteString(", 用户: 未知")
|
||
}
|
||
if m.Content == "" {
|
||
sb.WriteString(", 无文本")
|
||
} else {
|
||
sb.WriteString(", 文本: ")
|
||
if m.MentionEveryone {
|
||
sb.WriteString("[@全体]")
|
||
}
|
||
sb.WriteString(m.Content)
|
||
}
|
||
if len(m.Attachments) > 0 {
|
||
sb.WriteString(", 附加: ")
|
||
for _, a := range m.Attachments {
|
||
sb.WriteString("<ID:")
|
||
sb.WriteString(a.ID)
|
||
sb.WriteString(",URL:")
|
||
sb.WriteString(a.URL)
|
||
sb.WriteByte('>')
|
||
}
|
||
}
|
||
if len(m.Embeds) > 0 {
|
||
for _, e := range m.Embeds {
|
||
sb.WriteString(", 嵌入: <标题:")
|
||
sb.WriteString(e.Title)
|
||
sb.WriteString(",提示:")
|
||
sb.WriteString(e.Prompt)
|
||
sb.WriteByte('>')
|
||
}
|
||
}
|
||
if m.Ark != nil {
|
||
sb.WriteString(", 模版: ")
|
||
sb.WriteString(strconv.Itoa(m.Ark.TemplateID))
|
||
}
|
||
if m.MessageReference != nil {
|
||
sb.WriteString(", 回复: ")
|
||
sb.WriteString(m.MessageReference.MessageID)
|
||
}
|
||
return sb.String()
|
||
}
|
||
|
||
// MessageEmbed https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messageembed
|
||
type MessageEmbed struct {
|
||
Title string `json:"title"`
|
||
Prompt string `json:"prompt"`
|
||
Thumbnail *MessageEmbedThumbnail `json:"thumbnail"`
|
||
Fields []MessageEmbedField `json:"fields"`
|
||
}
|
||
|
||
// MessageEmbedThumbnail https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messageembedthumbnail
|
||
type MessageEmbedThumbnail struct {
|
||
URL string `json:"url"`
|
||
}
|
||
|
||
// MessageEmbedField https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messageembedfield
|
||
type MessageEmbedField struct {
|
||
Name string `json:"name"`
|
||
}
|
||
|
||
// MessageAttachment https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messageattachment
|
||
type MessageAttachment struct {
|
||
ContentType string `json:"content_type,omitempty"`
|
||
Filename string `json:"filename,omitempty"`
|
||
Height int `json:"height,omitempty"`
|
||
ID string `json:"id,omitempty"`
|
||
Size int `json:"size,omitempty"`
|
||
URL string `json:"url,omitempty"`
|
||
Width int `json:"width,omitempty"`
|
||
}
|
||
|
||
// MessageArk https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messageark
|
||
type MessageArk struct {
|
||
TemplateID int `json:"template_id"`
|
||
KV []MessageArkKV `json:"kv"`
|
||
}
|
||
|
||
// MessageArkKV https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messagearkkv
|
||
type MessageArkKV struct {
|
||
Key string `json:"key"`
|
||
Value string `json:"value"`
|
||
Obj []MessageArkObj `json:"obj"`
|
||
}
|
||
|
||
// MessageArkObj https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messagearkobj
|
||
type MessageArkObj struct {
|
||
ObjKV []MessageArkObjKV `json:"obj_kv"`
|
||
}
|
||
|
||
// MessageArkObjKV https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messagearkobjkv
|
||
type MessageArkObjKV struct {
|
||
Key string `json:"key"`
|
||
Value string `json:"value"`
|
||
}
|
||
|
||
// MessageReference https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messagereference
|
||
type MessageReference struct {
|
||
MessageID string `json:"message_id"`
|
||
IgnoreGetMessageError bool `json:"ignore_get_message_error"`
|
||
}
|
||
|
||
// MessageDelete https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#messagedelete
|
||
type MessageDelete struct {
|
||
Message *Message `json:"message"`
|
||
OpUser *User `json:"op_user"`
|
||
}
|
||
|
||
func (mdl *MessageDelete) String() string {
|
||
sb := strings.Builder{}
|
||
sb.WriteString("用户ID ")
|
||
sb.WriteString(mdl.OpUser.ID)
|
||
sb.WriteString(" 删除了消息: ")
|
||
sb.WriteString(mdl.Message.ID)
|
||
sb.WriteString(" ｷﾞﾙﾄﾞ: ")
|
||
sb.WriteString(mdl.Message.GuildID)
|
||
if mdl.Message.SrcGuildID != "" {
|
||
sb.WriteString(", 元ｷﾞﾙﾄﾞ: ")
|
||
sb.WriteString(mdl.Message.SrcGuildID)
|
||
}
|
||
sb.WriteString(", 频道: ")
|
||
sb.WriteString(mdl.Message.ChannelID)
|
||
if mdl.Message.Member.User != nil {
|
||
sb.WriteString(", 用户: ")
|
||
sb.WriteString(mdl.Message.Member.User.Username)
|
||
sb.WriteByte('(')
|
||
sb.WriteString(mdl.Message.Member.User.ID)
|
||
sb.WriteByte(')')
|
||
if mdl.Message.Member.User.Bot {
|
||
sb.WriteString("(机器人)")
|
||
}
|
||
} else {
|
||
sb.WriteString(", 用户: 未知")
|
||
}
|
||
return sb.String()
|
||
}
|
||
|
||
// MessageAudited 消息审核对象
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/message/model.html#%E6%B6%88%E6%81%AF%E5%AE%A1%E6%A0%B8%E5%AF%B9%E8%B1%A1-messageaudited
|
||
type MessageAudited struct {
|
||
AuditID string `json:"audit_id"`
|
||
AuditTime time.Time `json:"audit_time"`
|
||
ChannelID string `json:"channel_id"`
|
||
CreateTime time.Time `json:"create_time"`
|
||
GuildID string `json:"guild_id"`
|
||
MessageID string `json:"message_id"`
|
||
}
|
||
|
||
// GetMessageFromChannel 获取子频道 channel_id 下的消息 message_id 的详情
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/message/get_message_of_id.html
|
||
func (bot *Bot) GetMessageFromChannel(messageid, channelid string) (*Message, error) {
|
||
return bot.getOpenAPIofMessage("/channels/" + channelid + "/messages/" + messageid)
|
||
}
|
||
|
||
// MessagePost 发送消息所需参数
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/message/post_messages.html#%E9%80%9A%E7%94%A8%E5%8F%82%E6%95%B0
|
||
type MessagePost struct {
|
||
// https://bot.q.qq.com/wiki/develop/api-231017/server-inter/message/send-receive/send.html
|
||
Type MessageType `json:"msg_type"`
|
||
Seq int `json:"msg_seq,omitempty"` // 回复消息的序号，与 msg_id 联合使用，避免相同消息id回复重复发送，不填默认是1。相同的 msg_id + msg_seq 重复发送会失败。
|
||
Content string `json:"content,omitempty"`
|
||
Embed *MessageEmbed `json:"embed,omitempty"` // https://bot.q.qq.com/wiki/develop/api/openapi/message/template/embed_message.html
|
||
Ark *MessageArk `json:"ark,omitempty"` // https://bot.q.qq.com/wiki/develop/api/openapi/message/message_template.html
|
||
MessageReference *MessageReference `json:"message_reference,omitempty"`
|
||
Image string `json:"image,omitempty"`
|
||
ImageFile string `json:"-"` // ImageFile 为图片路径 file:/// or base64:// or base16384:// , 与 Image, ImageBytes 参数二选一, 优先 ImageBytes
|
||
ImageBytes []byte `json:"-"` // ImageBytes 图片数据
|
||
ReplyMessageID string `json:"msg_id,omitempty"`
|
||
ReplyEventID string `json:"event_id,omitempty"`
|
||
Markdown *MessageMarkdown `json:"markdown,omitempty"`
|
||
KeyBoard *MessageKeyboard `json:"keyboard,omitempty"`
|
||
Media *MessageMedia `json:"media,omitempty"`
|
||
}
|
||
|
||
func (mp *MessagePost) String() string {
|
||
sb := strings.Builder{}
|
||
if mp.Seq > 0 {
|
||
sb.WriteString("[v2:")
|
||
sb.WriteString(mp.Type.String())
|
||
sb.WriteString("#")
|
||
sb.WriteString(strconv.Itoa(mp.Seq))
|
||
sb.WriteString("]")
|
||
}
|
||
if mp.Content == "" {
|
||
sb.WriteString("无文本")
|
||
} else {
|
||
sb.WriteString("文本: ")
|
||
sb.WriteString(mp.Content)
|
||
}
|
||
if mp.ReplyMessageID != "" {
|
||
sb.WriteString(", 回应消息: ")
|
||
sb.WriteString(mp.ReplyMessageID)
|
||
}
|
||
if mp.ReplyEventID != "" {
|
||
sb.WriteString(", 回应事件: ")
|
||
sb.WriteString(mp.ReplyEventID)
|
||
}
|
||
if mp.Embed != nil {
|
||
sb.WriteString(", 嵌入: <标题:")
|
||
sb.WriteString(mp.Embed.Title)
|
||
sb.WriteString(",提示:")
|
||
sb.WriteString(mp.Embed.Prompt)
|
||
sb.WriteByte('>')
|
||
}
|
||
if mp.Ark != nil {
|
||
sb.WriteString(", 模版: ")
|
||
sb.WriteString(strconv.Itoa(mp.Ark.TemplateID))
|
||
}
|
||
if mp.MessageReference != nil {
|
||
sb.WriteString(", 回复: ")
|
||
sb.WriteString(mp.MessageReference.MessageID)
|
||
}
|
||
if mp.Image != "" {
|
||
sb.WriteString(", 图片URL: ")
|
||
sb.WriteString(mp.Image)
|
||
}
|
||
if mp.ImageFile != "" {
|
||
sb.WriteString(", 图片内容: ")
|
||
x := mp.ImageFile
|
||
if len(x) > 64 {
|
||
x = x[:64] + "..."
|
||
}
|
||
sb.WriteString(x)
|
||
}
|
||
if len(mp.ImageBytes) > 0 {
|
||
sb.WriteString(", 图片大小: ")
|
||
sb.WriteString(strconv.Itoa(len(mp.ImageBytes)))
|
||
}
|
||
if mp.Markdown != nil {
|
||
sb.WriteString(", MD模版: ")
|
||
sb.WriteString(strconv.Itoa(mp.Markdown.TemplateID))
|
||
}
|
||
if mp.KeyBoard != nil {
|
||
sb.WriteString(", KB模版: ")
|
||
sb.WriteString(mp.KeyBoard.ID)
|
||
}
|
||
if mp.Media != nil {
|
||
sb.WriteString(", 富媒体: ")
|
||
data, err := base64.StdEncoding.DecodeString(mp.Media.FileInfo)
|
||
if err == nil {
|
||
sb.Write(data)
|
||
} else {
|
||
sb.WriteString(mp.Media.FileInfo)
|
||
}
|
||
}
|
||
return sb.String()
|
||
}
|
||
|
||
func (bot *Bot) postMessageTo(ep string, content *MessagePost) (*Message, error) {
|
||
if len(content.ImageBytes) == 0 && content.ImageFile == "" {
|
||
return bot.postOpenAPIofMessage(ep, "", WriteBodyFromJSON(content))
|
||
}
|
||
x := reflect.ValueOf(content).Elem()
|
||
t := x.Type()
|
||
msg := []any{}
|
||
for i := 0; i < x.NumField(); i++ {
|
||
xi := x.Field(i)
|
||
if xi.IsZero() {
|
||
continue
|
||
}
|
||
tag, _, _ := strings.Cut(t.Field(i).Tag.Get("json"), ",")
|
||
if tag == "-" {
|
||
tag = "file_image"
|
||
}
|
||
msg = append(msg, tag)
|
||
if xi.Kind() == reflect.Pointer && xi.Elem().Kind() == reflect.Struct {
|
||
data, err := json.Marshal(xi.Interface())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
msg = append(msg, data)
|
||
continue
|
||
}
|
||
msg = append(msg, xi.Interface()) // []byte or string
|
||
}
|
||
if len(msg) < 2 {
|
||
return nil, ErrEmptyMessagePost
|
||
}
|
||
body, contenttype, err := WriteBodyByMultipartFormData(msg...)
|
||
if err != nil {
|
||
return nil, errors.Wrap(err, getThisFuncName())
|
||
}
|
||
return bot.postOpenAPIofMessage(ep, contenttype, body)
|
||
}
|
||
|
||
// PostMessageToChannel 向 channel_id 指定的子频道发送消息
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/message/post_messages.html
|
||
func (bot *Bot) PostMessageToChannel(id string, content *MessagePost) (*Message, error) {
|
||
logrus.Infoln(getLogHeader(), "<= [公]频道:", id+",", content)
|
||
return bot.postMessageTo("/channels/"+id+"/messages", content)
|
||
}
|
||
|
||
// DeleteMessageInChannel 回子频道 channel_id 下的消息 message_id
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/message/delete_message.html
|
||
func (bot *Bot) DeleteMessageInChannel(channelid, messageid string, hidetip bool) error {
|
||
logrus.Infoln(getLogHeader(), "<x 频道:", channelid+", 消息:", messageid)
|
||
return bot.DeleteOpenAPI(WriteHTTPQueryIfNotNil("/channels/"+channelid+"/messages/"+messageid,
|
||
"hidetip", hidetip,
|
||
), "", nil)
|
||
}
|
||
|
||
// MessageSetting 频道消息频率设置对象
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/setting/model.html
|
||
type MessageSetting struct {
|
||
DisableCreateDm bool `json:"disable_create_dm"`
|
||
DisablePushMsg bool `json:"disable_push_msg"`
|
||
ChannelIDs []string `json:"channel_ids"`
|
||
ChannelPushMaxNum uint32 `json:"channel_push_max_num"`
|
||
}
|
||
|
||
// GetGuildMessageSetting 获取机器人在频道 guild_id 内的消息频率设置
|
||
//
|
||
// https://bot.q.qq.com/wiki/develop/api/openapi/setting/message_setting.html
|
||
func (bot *Bot) GetGuildMessageSetting(id string) (*MessageSetting, error) {
|
||
return bot.getOpenAPIofMessageSetting("/guilds/" + id + "/message/setting")
|
||
}
|