Date: Fri, 27 Sep 2024 01:41:24 +0800
Subject: [PATCH 8/9] =?UTF-8?q?refact=EF=BC=9A=E5=87=86=E5=A4=87=E5=BC=80?=
=?UTF-8?q?=E5=A7=8B=E5=AE=8C=E5=96=84=E9=80=82=E9=85=8D=E5=99=A8=E8=AE=BE?=
=?UTF-8?q?=E8=AE=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
.gitignore | 1 +
adapters/lagrange/event_publisher.go | 3 +-
adapters/lagrange/lagrange.go | 79 +++++++++++++++++++++++++---
ice/database_manager.go | 6 +--
ice/node.go | 28 ++++++++--
main.go | 7 ++-
middlewares.go | 6 +++
7 files changed, 115 insertions(+), 15 deletions(-)
create mode 100644 middlewares.go
diff --git a/.gitignore b/.gitignore
index 7397e03..6e5e379 100644
--- a/.gitignore
+++ b/.gitignore
@@ -30,3 +30,4 @@ sig.bin
qrcode.png
signature.bin
*.toml
+*.db
diff --git a/adapters/lagrange/event_publisher.go b/adapters/lagrange/event_publisher.go
index 80bf98e..48fada5 100644
--- a/adapters/lagrange/event_publisher.go
+++ b/adapters/lagrange/event_publisher.go
@@ -2,6 +2,7 @@ package lagrange
import (
"github.com/Iceinu-Project/Iceinu/ice"
+ "github.com/Iceinu-Project/Iceinu/models/satori"
"github.com/LagrangeDev/LagrangeGo/client"
"github.com/LagrangeDev/LagrangeGo/message"
"time"
@@ -16,7 +17,7 @@ func EventsBinder() {
Target: ice.GetMasterNodeId(),
Timestamp: time.Now().Unix(),
Summary: "",
- Event: nil,
+ Event: &satori.EventSatori{},
})
})
}
diff --git a/adapters/lagrange/lagrange.go b/adapters/lagrange/lagrange.go
index 563a811..97ef4c8 100644
--- a/adapters/lagrange/lagrange.go
+++ b/adapters/lagrange/lagrange.go
@@ -2,8 +2,11 @@ package lagrange
import (
"github.com/Iceinu-Project/Iceinu/adapters"
+ "github.com/Iceinu-Project/Iceinu/log"
"github.com/LagrangeDev/LagrangeGo/client"
"github.com/LagrangeDev/LagrangeGo/client/auth"
+ "github.com/sirupsen/logrus"
+ "os"
)
// InfosLagrangeAdapter LagrangeGo适配器元信息
@@ -33,18 +36,56 @@ func (AdapterLagrangeGo) Init() error {
qqClientInstance := client.NewClient(uint32(AdapterLagrangeConf.Lagrange.Account), appInfo, AdapterLagrangeConf.Lagrange.SignServer)
qqClientInstance.SetLogger(GetProtocolLogger())
qqClientInstance.UseDevice(deviceInfo)
- //TODO implement me
- panic("implement me")
+
+ // 尝试读取签名文件
+ data, err := os.ReadFile("signature.bin")
+ if err != nil {
+ logrus.Warnln("读取签名文件时发生错误:", err)
+ } else {
+ sig, err := auth.UnmarshalSigInfo(data, true)
+ if err != nil {
+ logrus.Warnln("加载签名文件时发生错误:", err)
+ } else {
+ qqClientInstance.UseSig(sig)
+ }
+ }
+ // 保存Client实例
+ Client = qqClientInstance
+ return nil
}
func (AdapterLagrangeGo) SubscribeEvents() error {
- //TODO implement me
- panic("implement me")
+ EventsBinder()
+ return nil
}
func (AdapterLagrangeGo) Start() error {
- //TODO implement me
- panic("implement me")
+ // 在函数结束时释放Client并尝试保存签名
+ defer Client.Release()
+ defer SaveSignature()
+ // 登录
+ err := Login()
+ if err != nil {
+ return err
+ }
+ // 尝试刷新client的所有缓存
+ err = Client.RefreshFriendCache()
+ if err != nil {
+ return err
+ }
+ err = Client.RefreshAllGroupsInfo()
+ if err != nil {
+ return err
+ }
+ err = Client.RefreshAllGroupMembersCache()
+ if err != nil {
+ return err
+ }
+ err = Client.RefreshAllRkeyInfoCache()
+ if err != nil {
+ return err
+ }
+ return nil
}
func (AdapterLagrangeGo) GetAdapterInfo() *adapters.AdapterInfo {
@@ -55,3 +96,29 @@ func (AdapterLagrangeGo) GetUserTree() *adapters.UserTree {
//TODO implement me
panic("implement me")
}
+
+// Login 登录
+func Login() error {
+ // 声明 err 变量并进行错误处理
+ err := Client.Login("", "qrcode.png")
+ if err != nil {
+ log.Error("登录时发生错误:", err)
+ return err
+ }
+ return nil
+}
+
+// SaveSignature 保存sign信息
+func SaveSignature() {
+ data, err := Client.Sig().Marshal()
+ if err != nil {
+ log.Error("生成签名文件时发生错误err:", err)
+ return
+ }
+ err = os.WriteFile("signature.bin", data, 0644)
+ if err != nil {
+ log.Error("写入签名文件时发生错误 err:", err)
+ return
+ }
+ log.Info("签名已被写入签名文件")
+}
diff --git a/ice/database_manager.go b/ice/database_manager.go
index d88a2f3..b58e27a 100644
--- a/ice/database_manager.go
+++ b/ice/database_manager.go
@@ -35,9 +35,9 @@ func InitLocalDatabase() {
if count == 0 {
log.Infof("检测到数据库中没有节点数据，正在初始化节点数据...")
// 如果是第一次初始化，插入一个节点数据
- NodeId = GenerateNodeId()
+ SelfNodeId = GenerateNodeId()
db.Create(&IceinuNodeData{
- NodeId: NodeId,
+ NodeId: SelfNodeId,
AdapterModel: "satori",
})
} else {
@@ -45,7 +45,7 @@ func InitLocalDatabase() {
var nodeData IceinuNodeData
db.First(&nodeData)
// 更新全局NodeId
- NodeId = nodeData.NodeId
+ SetSelfNodeId(nodeData.NodeId)
}
// 设置数据库连接池参数
sqlDB, err := db.DB()
diff --git a/ice/node.go b/ice/node.go
index bf99c36..482b016 100644
--- a/ice/node.go
+++ b/ice/node.go
@@ -1,15 +1,35 @@
package ice
-import "github.com/google/uuid"
+import (
+ "github.com/Iceinu-Project/Iceinu/config"
+ "github.com/google/uuid"
+)
-var NodeId string
+var SelfNodeId string
+var MasterNodeId string
func GetSelfNodeId() string {
- return "0"
+ return SelfNodeId
}
func GetMasterNodeId() string {
- return "0"
+ if !config.IceConf.Node.IsEnableNode {
+ return GetSelfNodeId()
+ }
+ if config.IceConf.Node.IsMaster {
+ return GetSelfNodeId()
+ }
+ return MasterNodeId
+}
+
+// SetMasterNodeId 设置主节点ID
+func SetMasterNodeId(masterNodeId string) {
+ MasterNodeId = masterNodeId
+}
+
+// SetSelfNodeId 设置自身节点ID
+func SetSelfNodeId(selfNodeId string) {
+ SelfNodeId = selfNodeId
}
// GenerateNodeId 生成一个新的节点ID，节点ID是一个字符串形式的UUID
diff --git a/main.go b/main.go
index 972a330..c6ac7cc 100644
--- a/main.go
+++ b/main.go
@@ -27,7 +27,12 @@ func main() {
gradient.RGBToANSI(255, 255, 255) +
gradient.GradientBackgroundText(" 通用的模块化 Go 聊天机器人框架 ", "#00d2ff", "#3a7bd5") +
gradient.Reset)
- log.Infof("当前版本: " + gradient.Cyan + "β0.1.3")
+ log.Infof("当前版本: " + gradient.Cyan + "β0.1.4")
// 初始化数据库连接
ice.InitLocalDatabase()
+ // 如果启用集群/分布式模式，则尝试和主节点建立连接
+ // 输出节点Id和集群模式
+ log.Info("当前节点模式: " + gradient.DarkGray + config.IceConf.Node.Mode)
+ log.Info("Node Id: " + gradient.DarkGray + ice.GetSelfNodeId())
+ log.Info("MasterNode Id: " + gradient.DarkGray + ice.GetMasterNodeId())
}
diff --git a/middlewares.go b/middlewares.go
new file mode 100644
index 0000000..4939abf
--- /dev/null
+++ b/middlewares.go
@@ -0,0 +1,6 @@
+package main
+
+// SetCustomMiddleWare 设置自定义中间件
+func SetCustomMiddleWare() {
+
+}
From 3eedc1d6233564a522a14587e82b6314c12c6b67 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kyoku=E2=AD=90?=
<144906395+KyokuKong@users.noreply.github.com>
Date: Sat, 28 Sep 2024 19:55:32 +0800
Subject: [PATCH 9/9] =?UTF-8?q?refact=EF=BC=9A=E5=9F=BA=E6=9C=AC=E9=87=8D?=
=?UTF-8?q?=E6=9E=84=E4=BA=86=E9=80=82=E9=85=8D=E5=99=A8=E5=92=8C=E4=BA=8B?=
=?UTF-8?q?=E4=BB=B6=E6=80=BB=E7=BA=BF=E7=9A=84=E8=AE=BE=E8=AE=A1=EF=BC=8C?=
=?UTF-8?q?=E5=87=86=E5=A4=87=E5=90=88=E5=B9=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
adapters/lagrange/event_publisher.go | 137 ++++++++++-
adapters/lagrange/lagrange.go | 64 +++--
adapters/lagrange/lagrange_config.go | 4 +-
adapters/lagrange/protocol_logger.go | 3 +-
adapters/lagrange/subscribe_manager.go | 312 +++++++++++++++++++++++++
adapters/lagrange/tools.go | 174 ++++++++++++++
cache/iceinu_cache.go | 14 --
ice/event.go | 89 ++++++-
ice/event_maker.go | 200 ++++++++++++++++
ice/plugin_manager.go | 20 ++
log_format.go | 2 +-
main.go | 26 ++-
middlewares.go | 35 +++
models/satori/satori_elements.go | 58 +++++
models/satori/satori_model.go | 14 +-
models/satori/tools.go | 125 +++++++++-
16 files changed, 1201 insertions(+), 76 deletions(-)
create mode 100644 adapters/lagrange/subscribe_manager.go
create mode 100644 adapters/lagrange/tools.go
delete mode 100644 cache/iceinu_cache.go
create mode 100644 ice/event_maker.go
create mode 100644 ice/plugin_manager.go
diff --git a/adapters/lagrange/event_publisher.go b/adapters/lagrange/event_publisher.go
index 48fada5..678fa04 100644
--- a/adapters/lagrange/event_publisher.go
+++ b/adapters/lagrange/event_publisher.go
@@ -5,19 +5,144 @@ import (
"github.com/Iceinu-Project/Iceinu/models/satori"
"github.com/LagrangeDev/LagrangeGo/client"
"github.com/LagrangeDev/LagrangeGo/message"
+ "strconv"
"time"
)
-func EventsBinder() {
- // 绑定私聊消息事件
- Client.PrivateMessageEvent.Subscribe(func(client *client.QQClient, event *message.PrivateMessage) {
+func BindEvents() {
+ // 注册私聊事件
+ Manager.RegisterPrivateMessageHandler(func(client *client.QQClient, event *message.PrivateMessage) {
+ // 尝试从适配器的缓存中获取自身信息
+ selfInfo := GetSelfInfoInCache(client)
+ // 尝试从适配器的缓存中获取好友信息
+ friendInfo := GetFriendsDataInCache(client)
ice.Publish(&ice.IceinuEvent{
- Type: 5,
+ Type: 10,
From: ice.GetSelfNodeId(),
Target: ice.GetMasterNodeId(),
Timestamp: time.Now().Unix(),
- Summary: "",
- Event: &satori.EventSatori{},
+ Summary: "PrivateMessageEvent",
+ Event: &satori.EventSatori{
+ Id: uint64(event.Id),
+ Type: "PrivateMessageEvent",
+ Platform: "QQNT",
+ SelfId: strconv.Itoa(int(client.Uin)),
+ Timestamp: int64(event.Time),
+ Argv: nil,
+ Button: nil,
+ Channel: &satori.Channel{
+ Id: strconv.Itoa(int(event.Sender.Uin)),
+ Type: 1,
+ Name: event.Sender.Uid,
+ ParentId: "",
+ },
+ Group: &satori.Group{
+ Id: "",
+ Name: "",
+ Avatar: "",
+ Maxcount: 0,
+ MemberCount: 0,
+ },
+ Login: nil,
+ Member: nil,
+ Message: &satori.Message{
+ Id: strconv.Itoa(int(event.InternalId)),
+ Content: event.ToString(),
+ Channel: nil,
+ Group: nil,
+ Member: nil,
+ User: nil,
+ CreatedAt: int64(event.Time),
+ UpdatedAt: int64(event.Time),
+ MessageElements: ToSatoriElements(event.Elements),
+ },
+ Operator: &satori.User{
+ Id: strconv.Itoa(int(event.Sender.Uin)),
+ Name: event.Sender.CardName,
+ Nickname: event.Sender.Nickname,
+ Avatar: friendInfo[event.Sender.Uin].Avatar,
+ IsBot: false,
+ },
+ Role: nil,
+ User: &satori.User{
+ Id: strconv.Itoa(int(event.Self)),
+ Name: client.NickName(),
+ Nickname: client.NickName(),
+ Avatar: selfInfo.Avatar,
+ IsBot: false,
+ },
+ },
+ })
+ // 将LagrangeGo的消息存入消息缓存
+ err := Cache.Set(strconv.Itoa(int(event.InternalId)), event)
+ if err != nil {
+ return
+ }
+ })
+ // 注册群聊事件
+ Manager.RegisterGroupMessageHandler(func(client *client.QQClient, event *message.GroupMessage) {
+ // 尝试从LagrangeGo的内置缓存中获取群信息
+ groupInfo := client.GetCachedGroupInfo(event.GroupUin)
+ // 尝试从适配器的缓存中获取自身信息
+ selfInfo := GetSelfInfoInCache(client)
+ // 尝试从适配器的缓存中获取群成员映射
+ groupMemberData := GetGroupMembersDataInCache(client, event.GroupUin)
+ ice.Publish(&ice.IceinuEvent{
+ Type: 10,
+ From: ice.GetSelfNodeId(),
+ Target: ice.GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "GroupMessageEvent",
+ Event: &satori.EventSatori{
+ Id: uint64(event.Id),
+ Type: "GroupMessageEvent",
+ Platform: "QQNT",
+ SelfId: strconv.Itoa(int(client.Uin)),
+ Timestamp: int64(event.Time),
+ Argv: nil,
+ Button: nil,
+ Channel: &satori.Channel{
+ Id: strconv.Itoa(int(event.GroupUin)),
+ Type: 0,
+ Name: event.GroupName,
+ ParentId: "",
+ },
+ Group: &satori.Group{
+ Id: strconv.Itoa(int(event.GroupUin)),
+ Name: event.GroupName,
+ Avatar: groupInfo.Avatar,
+ Maxcount: groupInfo.MaxMember,
+ MemberCount: groupInfo.MemberCount,
+ },
+ Login: nil,
+ Member: nil,
+ Message: &satori.Message{
+ Id: strconv.Itoa(int(event.InternalId)),
+ Content: event.ToString(),
+ Channel: nil,
+ Group: nil,
+ Member: nil,
+ User: nil,
+ CreatedAt: int64(event.Time),
+ UpdatedAt: int64(event.Time),
+ MessageElements: ToSatoriElements(event.Elements),
+ },
+ Operator: &satori.User{
+ Id: strconv.Itoa(int(event.Sender.Uin)),
+ Name: event.Sender.CardName,
+ Nickname: event.Sender.Nickname,
+ Avatar: groupMemberData[event.Sender.Uin].Avatar,
+ IsBot: false,
+ },
+ Role: nil,
+ User: &satori.User{
+ Id: strconv.Itoa(int(client.Uin)),
+ Name: client.NickName(),
+ Nickname: client.NickName(),
+ Avatar: selfInfo.Avatar,
+ IsBot: false,
+ },
+ },
})
})
}
diff --git a/adapters/lagrange/lagrange.go b/adapters/lagrange/lagrange.go
index 97ef4c8..5ba9795 100644
--- a/adapters/lagrange/lagrange.go
+++ b/adapters/lagrange/lagrange.go
@@ -2,11 +2,15 @@ package lagrange
import (
"github.com/Iceinu-Project/Iceinu/adapters"
+ "github.com/Iceinu-Project/Iceinu/cache"
+ "github.com/Iceinu-Project/Iceinu/ice"
"github.com/Iceinu-Project/Iceinu/log"
"github.com/LagrangeDev/LagrangeGo/client"
"github.com/LagrangeDev/LagrangeGo/client/auth"
- "github.com/sirupsen/logrus"
"os"
+ "os/signal"
+ "strconv"
+ "syscall"
)
// InfosLagrangeAdapter LagrangeGo适配器元信息
@@ -24,12 +28,18 @@ var InfosLagrangeAdapter = adapters.AdapterInfo{
// Client LagrangeGo客户端实例
var Client *client.QQClient
+// Cache 消息缓存
+var Cache *cache.IceCacheManager
+
// AdapterLagrangeGo LagrangeGo适配器
type AdapterLagrangeGo struct{}
-func (AdapterLagrangeGo) Init() error {
+func (a *AdapterLagrangeGo) Init() error {
// 读取配置文件
AdapterConfigInit()
+ // 初始化消息缓存
+ log.Debug("正在初始化LagrangeGo的消息缓存...")
+ Cache = cache.NewIceCacheManager(AdapterLagrangeConf.CacheSize, AdapterLagrangeConf.CacheExpire)
// 日志输出
appInfo := auth.AppList["linux"]["3.2.10-25765"]
deviceInfo := auth.NewDeviceInfo(AdapterLagrangeConf.Lagrange.Account)
@@ -40,11 +50,11 @@ func (AdapterLagrangeGo) Init() error {
// 尝试读取签名文件
data, err := os.ReadFile("signature.bin")
if err != nil {
- logrus.Warnln("读取签名文件时发生错误:", err)
+ log.Warn("读取签名文件时发生错误:", err)
} else {
sig, err := auth.UnmarshalSigInfo(data, true)
if err != nil {
- logrus.Warnln("加载签名文件时发生错误:", err)
+ log.Warn("加载签名文件时发生错误:", err)
} else {
qqClientInstance.UseSig(sig)
}
@@ -54,45 +64,45 @@ func (AdapterLagrangeGo) Init() error {
return nil
}
-func (AdapterLagrangeGo) SubscribeEvents() error {
- EventsBinder()
+func (a *AdapterLagrangeGo) SubscribeEvents() error {
+ BindEvents()
return nil
}
-func (AdapterLagrangeGo) Start() error {
+func (a *AdapterLagrangeGo) Start() error {
// 在函数结束时释放Client并尝试保存签名
defer Client.Release()
defer SaveSignature()
- // 登录
- err := Login()
- if err != nil {
- return err
- }
- // 尝试刷新client的所有缓存
- err = Client.RefreshFriendCache()
- if err != nil {
- return err
- }
- err = Client.RefreshAllGroupsInfo()
+ // 事件订阅
+ err := a.SubscribeEvents()
if err != nil {
return err
}
- err = Client.RefreshAllGroupMembersCache()
+ SetAllSubscribes()
+ // 登录
+ err = Login()
if err != nil {
return err
}
- err = Client.RefreshAllRkeyInfoCache()
- if err != nil {
- return err
+ // 推送适配器连接事件
+ ice.MakeAdapterConnectEvent(InfosLagrangeAdapter.Name, InfosLagrangeAdapter.Model, strconv.Itoa(int(Client.Uin)), Client.NickName())
+
+ // 主协程关闭通道
+ mc := make(chan os.Signal, 2)
+ signal.Notify(mc, os.Interrupt, syscall.SIGTERM)
+ for {
+ switch <-mc {
+ case os.Interrupt, syscall.SIGTERM:
+ return nil
+ }
}
- return nil
}
-func (AdapterLagrangeGo) GetAdapterInfo() *adapters.AdapterInfo {
+func (a *AdapterLagrangeGo) GetAdapterInfo() *adapters.AdapterInfo {
return &InfosLagrangeAdapter
}
-func (AdapterLagrangeGo) GetUserTree() *adapters.UserTree {
+func (a *AdapterLagrangeGo) GetUserTree() *adapters.UserTree {
//TODO implement me
panic("implement me")
}
@@ -122,3 +132,7 @@ func SaveSignature() {
}
log.Info("签名已被写入签名文件")
}
+
+func GetAdapter() adapters.IceinuAdapter {
+ return &AdapterLagrangeGo{}
+}
diff --git a/adapters/lagrange/lagrange_config.go b/adapters/lagrange/lagrange_config.go
index 9c2499b..d832f26 100644
--- a/adapters/lagrange/lagrange_config.go
+++ b/adapters/lagrange/lagrange_config.go
@@ -9,8 +9,8 @@ var AdapterLagrangeConf *AdapterConfig
// AdapterConfig 适配器配置
type AdapterConfig struct {
- CacheSize int `toml:"message_cache_size"` // 消息缓存大小
- CacheExpire int `toml:"message_cache_expire"` // 消息缓存过期时间
+ CacheSize uint32 `toml:"message_cache_size"` // 消息缓存大小
+ CacheExpire uint32 `toml:"message_cache_expire"` // 消息缓存过期时间
Lagrange LgrConfig `toml:"lagrange"`
}
diff --git a/adapters/lagrange/protocol_logger.go b/adapters/lagrange/protocol_logger.go
index 9e950df..e0f6cd0 100644
--- a/adapters/lagrange/protocol_logger.go
+++ b/adapters/lagrange/protocol_logger.go
@@ -2,6 +2,7 @@ package lagrange
import (
"fmt"
+ gradient "github.com/Iceinu-Project/IceGradient"
"github.com/Iceinu-Project/Iceinu/log"
"os"
"path"
@@ -13,7 +14,7 @@ type ProtocolLogger struct{}
var dumpspath = "dump"
-const fromProtocol = "LGR | "
+const fromProtocol = "LGR | " + gradient.DarkGray
func (p ProtocolLogger) Info(format string, arg ...any) {
log.Infof(fromProtocol+format, arg...)
diff --git a/adapters/lagrange/subscribe_manager.go b/adapters/lagrange/subscribe_manager.go
new file mode 100644
index 0000000..2436768
--- /dev/null
+++ b/adapters/lagrange/subscribe_manager.go
@@ -0,0 +1,312 @@
+package lagrange
+
+import (
+ "github.com/LagrangeDev/LagrangeGo/client"
+ "github.com/LagrangeDev/LagrangeGo/client/event"
+ "github.com/LagrangeDev/LagrangeGo/message"
+)
+
+// 定义各个消息类型的处理函数
+
+type PrivateMessageHandler func(client *client.QQClient, message *message.PrivateMessage)
+type GroupMessageHandler func(client *client.QQClient, message *message.GroupMessage)
+type TempMessageHandler func(client *client.QQClient, message *message.TempMessage)
+
+type SelfPrivateMessageHandler func(client *client.QQClient, message *message.PrivateMessage)
+type SelfGroupMessageHandler func(client *client.QQClient, message *message.GroupMessage)
+type SelfTempMessageHandler func(client *client.QQClient, message *message.TempMessage)
+
+// 定义各个事件类型的处理函数
+
+type GroupJoinEventHandler func(client *client.QQClient, event *event.GroupMemberIncrease) // bot进群
+type GroupLeaveEventHandler func(client *client.QQClient, event *event.GroupMemberDecrease) // bot退群
+
+type GroupInvitedEventHandler func(client *client.QQClient, event *event.GroupInvite) // bot被邀请进群
+type GroupMemberJoinRequestEventHandler func(client *client.QQClient, event *event.GroupMemberJoinRequest) // 加群邀请
+type GroupMemberJoinEventHandler func(client *client.QQClient, event *event.GroupMemberIncrease) // 成员加群
+type GroupMemberLeaveEventHandler func(client *client.QQClient, event *event.GroupMemberDecrease) // 成员退群
+type GroupMuteEventHandler func(client *client.QQClient, event *event.GroupMute) // 禁言事件
+type GroupRecallEventHandler func(client *client.QQClient, event *event.GroupRecall)
+type GroupMemberPermissionChangedEventHandler func(client *client.QQClient, event *event.GroupMemberPermissionChanged)
+type GroupNameUpdatedEventHandler func(client *client.QQClient, event *event.GroupNameUpdated) // 群更名
+type MemberSpecialTitleUpdatedEventHandler func(client *client.QQClient, event *event.MemberSpecialTitleUpdated) // 群成员特殊头衔更新
+type NewFriendRequestHandler func(client *client.QQClient, event *event.NewFriendRequest) // 新朋友请求
+type FriendRecallEventHandler func(client *client.QQClient, event *event.FriendRecall) // 好友消息撤回
+type RenameEventHandler func(client *client.QQClient, event *event.Rename) // 好友昵称更改
+type FriendNotifyEventHandler func(client *client.QQClient, event event.INotifyEvent) // 好友通知
+type GroupNotifyEventHandler func(client *client.QQClient, event event.INotifyEvent) // 群通知
+
+// SubscribeManager 订阅管理器实现，用于向LagrangeGo注册各类消息/事件处理函数
+type SubscribeManager struct {
+ privateMessageHandlers []PrivateMessageHandler
+ groupMessageHandlers []GroupMessageHandler
+ tempMessageHandlers []TempMessageHandler
+ selfPrivateMessageHandlers []SelfPrivateMessageHandler
+ selfGroupMessageHandlers []SelfGroupMessageHandler
+ selfTempMessageHandlers []SelfTempMessageHandler
+ groupJoinEventHandlers []GroupJoinEventHandler
+ groupLeaveEventHandlers []GroupLeaveEventHandler
+ groupInviteEventHandlers []GroupInvitedEventHandler
+ groupMemberJoinRequestEventHandlers []GroupMemberJoinRequestEventHandler
+ groupMemberJoinEventHandlers []GroupMemberJoinEventHandler
+ groupMemberLeaveEventHandlers []GroupMemberLeaveEventHandler
+ groupMuteEventHandlers []GroupMuteEventHandler
+ groupRecallEventHandlers []GroupRecallEventHandler
+ groupMemberPermissionChangedEventHandlers []GroupMemberPermissionChangedEventHandler
+ groupNameUpdatedEventHandlers []GroupNameUpdatedEventHandler
+ memberSpecialTitleUpdatedEventHandlers []MemberSpecialTitleUpdatedEventHandler
+ newFriendRequestHandlers []NewFriendRequestHandler
+ friendRecallEventHandlers []FriendRecallEventHandler
+ renameEventHandlers []RenameEventHandler
+ friendNotifyEventHandlers []FriendNotifyEventHandler
+ groupNotifyEventHandlers []GroupNotifyEventHandler
+}
+
+// Manager 全局 SubscribeManager 实例
+var Manager = &SubscribeManager{}
+
+// RegisterPrivateMessageHandler 注册私聊消息处理函数
+func (sm *SubscribeManager) RegisterPrivateMessageHandler(handler PrivateMessageHandler) {
+ sm.privateMessageHandlers = append(sm.privateMessageHandlers, handler)
+}
+
+// RegisterGroupMessageHandler 注册群消息处理函数
+func (sm *SubscribeManager) RegisterGroupMessageHandler(handler GroupMessageHandler) {
+ sm.groupMessageHandlers = append(sm.groupMessageHandlers, handler)
+}
+
+// RegisterTempMessageHandler 注册临时消息处理函数
+func (sm *SubscribeManager) RegisterTempMessageHandler(handler TempMessageHandler) {
+ sm.tempMessageHandlers = append(sm.tempMessageHandlers, handler)
+}
+
+// RegisterSelfPrivateMessageHandler 注册自己的私聊消息处理函数
+func (sm *SubscribeManager) RegisterSelfPrivateMessageHandler(handler SelfPrivateMessageHandler) {
+ sm.selfPrivateMessageHandlers = append(sm.selfPrivateMessageHandlers, handler)
+}
+
+// RegisterSelfGroupMessageHandler 注册自己的群消息处理函数
+func (sm *SubscribeManager) RegisterSelfGroupMessageHandler(handler SelfGroupMessageHandler) {
+ sm.selfGroupMessageHandlers = append(sm.selfGroupMessageHandlers, handler)
+}
+
+// RegisterSelfTempMessageHandler 注册自己的临时消息处理函数
+func (sm *SubscribeManager) RegisterSelfTempMessageHandler(handler SelfTempMessageHandler) {
+ sm.selfTempMessageHandlers = append(sm.selfTempMessageHandlers, handler)
+}
+
+// RegisterGroupJoinEventHandler 注册群成员加入事件处理函数
+func (sm *SubscribeManager) RegisterGroupJoinEventHandler(handler GroupJoinEventHandler) {
+ sm.groupJoinEventHandlers = append(sm.groupJoinEventHandlers, handler)
+}
+
+// RegisterGroupLeaveEventHandler 注册群成员离开事件处理函数
+func (sm *SubscribeManager) RegisterGroupLeaveEventHandler(handler GroupLeaveEventHandler) {
+ sm.groupLeaveEventHandlers = append(sm.groupLeaveEventHandlers, handler)
+}
+
+// RegisterGroupInviteEventHandler 注册群邀请事件处理函数
+func (sm *SubscribeManager) RegisterGroupInviteEventHandler(handler GroupInvitedEventHandler) {
+ sm.groupInviteEventHandlers = append(sm.groupInviteEventHandlers, handler)
+
+}
+
+// RegisterGroupMemberJoinRequestEventHandler 注册群成员加群邀请事件处理函数
+func (sm *SubscribeManager) RegisterGroupMemberJoinRequestEventHandler(handler GroupMemberJoinRequestEventHandler) {
+ sm.groupMemberJoinRequestEventHandlers = append(sm.groupMemberJoinRequestEventHandlers, handler)
+}
+
+// RegisterGroupMemberJoinEventHandler 注册群成员加入事件处理函数
+func (sm *SubscribeManager) RegisterGroupMemberJoinEventHandler(handler GroupMemberJoinEventHandler) {
+ sm.groupMemberJoinEventHandlers = append(sm.groupMemberJoinEventHandlers, handler)
+}
+
+// RegisterGroupMemberLeaveEventHandler 注册群成员离开事件处理函数
+func (sm *SubscribeManager) RegisterGroupMemberLeaveEventHandler(handler GroupMemberLeaveEventHandler) {
+ sm.groupMemberLeaveEventHandlers = append(sm.groupMemberLeaveEventHandlers, handler)
+}
+
+// RegisterGroupMuteEventHandler 注册群成员禁言事件处理函数
+func (sm *SubscribeManager) RegisterGroupMuteEventHandler(handler GroupMuteEventHandler) {
+ sm.groupMuteEventHandlers = append(sm.groupMuteEventHandlers, handler)
+}
+
+// RegisterGroupRecallEventHandler 注册群消息撤回事件处理函数
+func (sm *SubscribeManager) RegisterGroupRecallEventHandler(handler GroupRecallEventHandler) {
+ sm.groupRecallEventHandlers = append(sm.groupRecallEventHandlers, handler)
+}
+
+// RegisterGroupMemberPermissionChangedEventHandler 注册群成员权限变更事件处理函数
+func (sm *SubscribeManager) RegisterGroupMemberPermissionChangedEventHandler(handler GroupMemberPermissionChangedEventHandler) {
+ sm.groupMemberPermissionChangedEventHandlers = append(sm.groupMemberPermissionChangedEventHandlers, handler)
+}
+
+// RegisterGroupNameUpdatedEventHandler 注册群名称变更事件处理函数
+func (sm *SubscribeManager) RegisterGroupNameUpdatedEventHandler(handler GroupNameUpdatedEventHandler) {
+ sm.groupNameUpdatedEventHandlers = append(sm.groupNameUpdatedEventHandlers, handler)
+}
+
+// RegisterMemberSpecialTitleUpdatedEventHandler 注册群成员特殊头衔变更事件处理函数
+func (sm *SubscribeManager) RegisterMemberSpecialTitleUpdatedEventHandler(handler MemberSpecialTitleUpdatedEventHandler) {
+ sm.memberSpecialTitleUpdatedEventHandlers = append(sm.memberSpecialTitleUpdatedEventHandlers, handler)
+}
+
+// RegisterNewFriendRequestHandler 注册新朋友请求处理函数
+func (sm *SubscribeManager) RegisterNewFriendRequestHandler(handler NewFriendRequestHandler) {
+ sm.newFriendRequestHandlers = append(sm.newFriendRequestHandlers, handler)
+}
+
+// RegisterFriendRecallEventHandler 注册好友消息撤回事件处理函数
+func (sm *SubscribeManager) RegisterFriendRecallEventHandler(handler FriendRecallEventHandler) {
+ sm.friendRecallEventHandlers = append(sm.friendRecallEventHandlers, handler)
+}
+
+// RegisterRenameEventHandler 注册好友昵称更改事件处理函数
+func (sm *SubscribeManager) RegisterRenameEventHandler(handler RenameEventHandler) {
+ sm.renameEventHandlers = append(sm.renameEventHandlers, handler)
+}
+
+// RegisterFriendNotifyEventHandler 注册好友通知事件处理函数
+func (sm *SubscribeManager) RegisterFriendNotifyEventHandler(handler FriendNotifyEventHandler) {
+ sm.friendNotifyEventHandlers = append(sm.friendNotifyEventHandlers, handler)
+}
+
+// RegisterGroupNotifyEventHandler 注册群通知事件处理函数
+func (sm *SubscribeManager) RegisterGroupNotifyEventHandler(handler GroupNotifyEventHandler) {
+ sm.groupNotifyEventHandlers = append(sm.groupNotifyEventHandlers, handler)
+}
+
+// SetAllSubscribes 设置所有订阅处理
+func SetAllSubscribes() {
+ Client.PrivateMessageEvent.Subscribe(func(client *client.QQClient, event *message.PrivateMessage) {
+ for _, handler := range Manager.privateMessageHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupMessageEvent.Subscribe(func(client *client.QQClient, event *message.GroupMessage) {
+ for _, handler := range Manager.groupMessageHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.TempMessageEvent.Subscribe(func(client *client.QQClient, event *message.TempMessage) {
+ for _, handler := range Manager.tempMessageHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.SelfPrivateMessageEvent.Subscribe(func(client *client.QQClient, event *message.PrivateMessage) {
+ for _, handler := range Manager.selfPrivateMessageHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.SelfGroupMessageEvent.Subscribe(func(client *client.QQClient, event *message.GroupMessage) {
+ for _, handler := range Manager.selfGroupMessageHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.SelfTempMessageEvent.Subscribe(func(client *client.QQClient, event *message.TempMessage) {
+ for _, handler := range Manager.selfTempMessageHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupJoinEvent.Subscribe(func(client *client.QQClient, event *event.GroupMemberIncrease) {
+ for _, handler := range Manager.groupJoinEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupLeaveEvent.Subscribe(func(client *client.QQClient, event *event.GroupMemberDecrease) {
+ for _, handler := range Manager.groupLeaveEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupInvitedEvent.Subscribe(func(client *client.QQClient, event *event.GroupInvite) {
+ for _, handler := range Manager.groupInviteEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupMemberJoinRequestEvent.Subscribe(func(client *client.QQClient, event *event.GroupMemberJoinRequest) {
+ for _, handler := range Manager.groupMemberJoinRequestEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupMemberJoinEvent.Subscribe(func(client *client.QQClient, event *event.GroupMemberIncrease) {
+ for _, handler := range Manager.groupMemberJoinEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupMemberLeaveEvent.Subscribe(func(client *client.QQClient, event *event.GroupMemberDecrease) {
+ for _, handler := range Manager.groupMemberLeaveEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupMuteEvent.Subscribe(func(client *client.QQClient, event *event.GroupMute) {
+ for _, handler := range Manager.groupMuteEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupRecallEvent.Subscribe(func(client *client.QQClient, event *event.GroupRecall) {
+ for _, handler := range Manager.groupRecallEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupMemberPermissionChangedEvent.Subscribe(func(client *client.QQClient, event *event.GroupMemberPermissionChanged) {
+ for _, handler := range Manager.groupMemberPermissionChangedEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupNameUpdatedEvent.Subscribe(func(client *client.QQClient, event *event.GroupNameUpdated) {
+ for _, handler := range Manager.groupNameUpdatedEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.MemberSpecialTitleUpdatedEvent.Subscribe(func(client *client.QQClient, event *event.MemberSpecialTitleUpdated) {
+ for _, handler := range Manager.memberSpecialTitleUpdatedEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.NewFriendRequestEvent.Subscribe(func(client *client.QQClient, event *event.NewFriendRequest) {
+ for _, handler := range Manager.newFriendRequestHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.FriendRecallEvent.Subscribe(func(client *client.QQClient, event *event.FriendRecall) {
+ for _, handler := range Manager.friendRecallEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.RenameEvent.Subscribe(func(client *client.QQClient, event *event.Rename) {
+ for _, handler := range Manager.renameEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.FriendNotifyEvent.Subscribe(func(client *client.QQClient, event event.INotifyEvent) {
+ for _, handler := range Manager.friendNotifyEventHandlers {
+ handler(client, event)
+ }
+ })
+
+ Client.GroupNotifyEvent.Subscribe(func(client *client.QQClient, event event.INotifyEvent) {
+ for _, handler := range Manager.groupNotifyEventHandlers {
+ handler(client, event)
+ }
+ })
+}
diff --git a/adapters/lagrange/tools.go b/adapters/lagrange/tools.go
new file mode 100644
index 0000000..ada0b32
--- /dev/null
+++ b/adapters/lagrange/tools.go
@@ -0,0 +1,174 @@
+package lagrange
+
+import (
+ "fmt"
+ "github.com/Iceinu-Project/Iceinu/log"
+ "github.com/Iceinu-Project/Iceinu/models/satori"
+ "github.com/LagrangeDev/LagrangeGo/client"
+ "github.com/LagrangeDev/LagrangeGo/client/entity"
+ "github.com/LagrangeDev/LagrangeGo/message"
+ "strconv"
+ "strings"
+)
+
+// ToSatoriElements 将LagrangeGo的消息元素切片转换为Satori的消息元素切片
+func ToSatoriElements(elements []message.IMessageElement) *[]satori.ElementSatori {
+ // 创建存储Satori消息切片的变量
+ var result []satori.ElementSatori
+ // 遍历传入的元素
+ for _, ele := range elements {
+ // 通过消息元素的Type方法来确定消息类型
+ switch ele.Type() {
+ // 将元素依次对应转换并传入
+ case message.Text:
+ ele := ele.(*message.TextElement)
+ // 检测文本中是否包含换行符
+ if strings.Contains(ele.Content, "\n") {
+ // 如果包含换行符，进行拆分和处理
+ textParts := strings.Split(ele.Content, "\n")
+ for i, part := range textParts {
+ // 将每段文本添加到 result
+ result = append(result, &satori.TextElement{Text: part})
+ // 如果不是最后一段文本，则插入 BrElement
+ if i < len(textParts)-1 {
+ result = append(result, &satori.BrElement{})
+ }
+ }
+ } else {
+ // 如果不包含换行符，直接添加文本元素
+ result = append(result, &satori.TextElement{Text: ele.Content})
+ }
+ case message.At:
+ ele := ele.(*message.AtElement)
+ result = append(result, &satori.AtElement{
+ Id: strconv.Itoa(int(ele.TargetUin)),
+ Name: ele.Display,
+ Role: "",
+ Type: strconv.Itoa(int(ele.Type())),
+ })
+ case message.Face:
+ ele := ele.(*message.FaceElement)
+ result = append(result, &satori.FaceElement{
+ Id: ele.FaceID,
+ })
+ case message.Voice:
+ ele := ele.(*message.VoiceElement)
+ result = append(result, &satori.AudioElement{
+ Src: ele.Url,
+ Title: ele.Name,
+ Duration: ele.Size,
+ Poster: "",
+ Cache: false,
+ Timeout: 0,
+ })
+ case message.Image:
+ ele := ele.(*message.ImageElement)
+ result = append(result, &satori.ImgElement{
+ Src: ele.Url,
+ Width: ele.Width,
+ Height: ele.Height,
+ Title: ele.ImageId,
+ Cache: false,
+ Timeout: 0,
+ })
+ case message.File:
+ ele := ele.(*message.FileElement)
+ result = append(result, &satori.FileElement{
+ Src: ele.FileUrl,
+ Title: ele.FileName,
+ Poster: "",
+ Cache: false,
+ Timeout: 0,
+ })
+ case message.Reply:
+ ele := ele.(*message.ReplyElement)
+ result = append(result, &satori.QuoteElement{
+ Id: strconv.Itoa(int(ele.SenderUin)),
+ Name: ele.SenderUid,
+ ChannelId: strconv.Itoa(int(ele.GroupUin)),
+ GroupId: strconv.Itoa(int(ele.GroupUin)),
+ Timestamp: int64(ele.Time),
+ Elements: ToSatoriElements(ele.Elements),
+ })
+ case message.Forward:
+ ele := ele.(*message.ForwardMessage)
+ result = append(result, &satori.MessageElement{
+ Forward: true,
+ Elements: UnzipNodes(ele.Nodes),
+ })
+
+ default:
+ result = append(result, &satori.UnsupportedElement{Type: strconv.Itoa(int(ele.Type()))})
+
+ }
+ }
+ return &result
+}
+
+func UnzipNodes(n []*message.ForwardNode) *[]satori.ElementSatori {
+ var result []satori.ElementSatori
+ for _, node := range n {
+ result = append(result, &satori.NodeElement{
+ GroupId: node.GroupId,
+ SenderId: node.SenderId,
+ SenderName: node.SenderName,
+ Time: node.Time,
+ Message: ToSatoriElements(node.Message),
+ })
+ }
+ return &result
+}
+
+// GetGroupMembersDataInCache 从缓存中获取群成员映射，如果缓存中没有则拉取并存入缓存
+func GetGroupMembersDataInCache(client *client.QQClient, groupId uint32) map[uint32]*entity.GroupMember {
+ // 尝试在内置缓存中查找群成员映射
+ var groupMemberMap map[uint32]*entity.GroupMember
+ err := Cache.Get(fmt.Sprintf("group_member_data_%d", groupId), &groupMemberMap)
+ if err != nil {
+ // 如果缓存中没有群成员映射，则创建一个
+ groupMembersData, err := client.GetGroupMembersData(groupId)
+ if err != nil {
+ log.Warn("无法获取群成员列表数据:", err)
+ }
+ err = Cache.Set(fmt.Sprintf("group_member_data_%d", groupId), groupMembersData)
+ if err != nil {
+ log.Warn("无法缓存群成员映射数据:", err)
+ }
+ groupMemberMap = groupMembersData
+ log.Debugf("%v群成员缓存数据更新完成，共%d个成员", groupId, len(groupMemberMap))
+ }
+ return groupMemberMap
+}
+
+// GetFriendsDataInCache 从缓存中获取好友映射，如果缓存中没有则拉取并存入缓存
+func GetFriendsDataInCache(client *client.QQClient) map[uint32]*entity.Friend {
+ // 尝试在内置缓存中查找好友映射
+ var friendMap map[uint32]*entity.Friend
+ err := Cache.Get("friend_data", &friendMap)
+ if err != nil {
+ // 如果缓存中没有好友映射，则创建一个
+ friendData, err := client.GetFriendsData()
+ if err != nil {
+ log.Warn("无法获取好友列表数据:", err)
+ }
+ err = Cache.Set("friend_data", friendData)
+ if err != nil {
+ log.Warn("无法缓存好友映射数据:", err)
+ }
+ friendMap = friendData
+ log.Debugf("好友缓存数据更新完成，共%d个好友", len(friendMap))
+ }
+ return friendMap
+}
+
+// GetSelfInfoInCache 从缓存中获取自身信息，如果缓存中没有则拉取并存入缓存
+func GetSelfInfoInCache(client *client.QQClient) *entity.Friend {
+ // 尝试在内置缓存中查找自身信息
+ friendData := GetFriendsDataInCache(client)
+ selfInfo, ok := friendData[client.Uin]
+ if !ok {
+ log.Warn("无法获取自身信息")
+ return nil
+ }
+ return selfInfo
+}
diff --git a/cache/iceinu_cache.go b/cache/iceinu_cache.go
deleted file mode 100644
index 4e8c49b..0000000
--- a/cache/iceinu_cache.go
+++ /dev/null
@@ -1,14 +0,0 @@
-package cache
-
-import (
- "github.com/Iceinu-Project/Iceinu/config"
- "github.com/Iceinu-Project/Iceinu/log"
-)
-
-var IceCache *IceCacheManager
-
-// 初始化缓存管理器
-func init() {
- log.Debugf("正在初始化内置缓存管理器...")
- IceCache = NewIceCacheManager(config.IceConf.Cache.MaxCacheSize, config.IceConf.Cache.CacheExpire)
-}
diff --git a/ice/event.go b/ice/event.go
index bf91068..e256065 100644
--- a/ice/event.go
+++ b/ice/event.go
@@ -1,5 +1,7 @@
package ice
+import "github.com/Iceinu-Project/Iceinu/adapters"
+
// IceinuEvent Iceinu全局事件结构体
//
// 默认的全局事件总线结构体，在直接使用Iceinu框架进行二次开发时使用这个事件结构
@@ -9,17 +11,31 @@ package ice
// 事件种类：
// 0：WebSocket心跳事件
//
-// 1：节点连接事件
+// 1：节点请求建立WebSocket连接事件
+//
+// 2：节点成功连接事件
+//
+// 3：节点断开连接事件
+//
+// 4：节点推送数据事件（主节点到子节点）
+//
+// 5：节点用户推送事件（子节点到主节点）
+//
+// 6：节点请求数据事件（子节点到主节点）
+//
+// 7：节点请求数据事件（主节点到子节点）
//
-// 2：节点断开事件
+// 8：适配器连接事件
//
-// 3：节点更新推送事件
+// 9：适配器断开连接事件
//
-// 4：节点更新请求事件
+// 10：消息接收事件
//
-// 5：消息接受事件（从子节点到主节点）
+// 11：消息发送事件
//
-// 6：消息处理事件（从主节点到子节点）
+// 12：节点失活事件（子节点到主节点）
+//
+// 13：节点重新激活事件（子节点到主节点）
type IceinuEvent struct {
Type uint8 `json:"type"`
From string `json:"from"` // 消息事件来源节点ID
@@ -29,18 +45,71 @@ type IceinuEvent struct {
Event interface{} `json:"event"` // 事件内容，用于承载不同类型的消息，使用时需要进行断言
}
+// WebSocketHeartbeatEvent 0：WebSocket心跳事件结构体
type WebSocketHeartbeatEvent struct {
OK bool `json:"ok"`
}
-type NodeConnectEvent struct {
+// NodeConnectRequestEvent 1：节点请求建立WebSocket连接事件结构体
+type NodeConnectRequestEvent struct {
+ Mode string `json:"mode"` // 组网模式
+ AdapterModel string `json:"adapter_model"` // 适配器模型
+ PluginVerifier string `json:"plugin_verifier"` // 插件校验值
+}
+
+// NodeConnectedEvent 2：节点成功连接事件结构体
+type NodeConnectedEvent struct {
+ OK bool `json:"ok"`
+}
+
+// NodeDisconnectedEvent 3：节点断开连接事件结构体
+type NodeDisconnectedEvent struct {
+ OK bool `json:"ok"`
}
-type NodeDisconnectEvent struct {
+// NodePushDataEvent 4：节点推送数据事件（主节点到子节点）结构体
+type NodePushDataEvent struct {
+ Data interface{} `json:"data"` // 推送的数据
}
-type NodeUpdatePushEvent struct {
+// NodeUserPushEvent 5：节点用户推送事件（子节点到主节点）结构体
+type NodeUserPushEvent struct {
+ UserTree adapters.UserTree `json:"user_tree"` // 用户树
}
-type NodeUpdateRequestEvent struct {
+// NodeRequestDataEvent 6：节点请求数据事件（子节点到主节点）结构体
+type NodeRequestDataEvent struct {
+ DataType string `json:"data_type"` // 请求的数据类型
+ Key string `json:"key"` // 请求的数据键
+ Query string `json:"query"` // 请求的查询内容
+}
+
+// RequestNodeDataEvent 7：节点请求数据事件（主节点到子节点）结构体
+type RequestNodeDataEvent struct {
+ DataType string `json:"data_type"` // 请求的数据类型
+ Key string `json:"key"` // 请求的数据键
+ Query string `json:"query"` // 请求的查询内容
+}
+
+// AdapterConnectEvent 8：适配器连接事件结构体
+type AdapterConnectEvent struct {
+ AdapterType string `json:"adapter_type"` // 适配器类型
+ AdapterModel string `json:"adapter_model"` // 适配器模型
+ UserId string `json:"user_id"` // 用户ID
+ UserName string `json:"user_name"` // 用户名称
+}
+
+// AdapterDisconnectEvent 9：适配器断开连接事件结构体
+type AdapterDisconnectEvent struct {
+ OK bool `json:"ok"`
+}
+
+// NodeDeactiveEvent 12：节点失活事件（子节点到主节点）结构体
+type NodeDeactiveEvent struct {
+ OK bool `json:"ok"`
+}
+
+// NodeReactiveEvent 13：节点重新激活事件（子节点到主节点）结构体
+type NodeReactiveEvent struct {
+ OK bool `json:"ok"`
}
diff --git a/ice/event_maker.go b/ice/event_maker.go
new file mode 100644
index 0000000..5420baa
--- /dev/null
+++ b/ice/event_maker.go
@@ -0,0 +1,200 @@
+package ice
+
+import "time"
+
+// MakeWebsocketHeartbeatEvent 创建一个WebSocket心跳事件
+func MakeWebsocketHeartbeatEvent() *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 0,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "WebSocketHeartbeatEvent",
+ Event: &WebSocketHeartbeatEvent{
+ OK: true,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodeConnectRequestEvent 创建一个节点请求建立WebSocket连接事件
+func MakeNodeConnectRequestEvent(mode string, adapterModel string, pluginVerifier string) *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 1,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodeConnectRequestEvent",
+ Event: &NodeConnectRequestEvent{
+ Mode: mode,
+ AdapterModel: adapterModel,
+ PluginVerifier: pluginVerifier,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodeConnectedEvent 创建一个节点成功连接事件
+func MakeNodeConnectedEvent() *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 2,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodeConnectedEvent",
+ Event: &NodeConnectedEvent{
+ OK: true,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodeDisconnectedEvent 创建一个节点断开连接事件
+func MakeNodeDisconnectedEvent() *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 3,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodeDisconnectedEvent",
+ Event: &NodeDisconnectedEvent{
+ OK: true,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodePushDataEvent 创建一个节点推送数据事件
+func MakeNodePushDataEvent(data interface{}) *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 4,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodePushDataEvent",
+ Event: data,
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodeUserPushEvent 创建一个节点用户推送事件
+func MakeNodeUserPushEvent(data interface{}) *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 5,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodeUserPushEvent",
+ Event: data,
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodeRequestDataEvent 创建一个节点请求数据事件
+func MakeNodeRequestDataEvent(dataType string, key string, query string) *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 6,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodeRequestDataEvent",
+ Event: &NodeRequestDataEvent{
+ DataType: dataType,
+ Key: key,
+ Query: query,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeRequestNodeDataEvent 创建一个请求节点数据事件
+func MakeRequestNodeDataEvent(dataType string, key string, query string) *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 7,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "RequestNodeDataEvent",
+ Event: &RequestNodeDataEvent{
+ DataType: dataType,
+ Key: key,
+ Query: query,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodeDeactiveEvent 创建一个节点失活事件
+func MakeNodeDeactiveEvent() *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 12,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodeActiveEvent",
+ Event: &NodeDeactiveEvent{
+ OK: true,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeNodeReactiveEvent 创建一个节点重新激活事件
+func MakeNodeReactiveEvent() *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 13,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "NodeActiveEvent",
+ Event: &NodeReactiveEvent{
+ OK: true,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeAdapterConnectEvent 创建一个适配器连接事件
+func MakeAdapterConnectEvent(adapterType string, model string, userId string, userName string) *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 8,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "AdapterConnectEvent",
+ Event: &AdapterConnectEvent{
+ AdapterType: adapterType,
+ AdapterModel: model,
+ UserId: userId,
+ UserName: userName,
+ },
+ }
+ Publish(e)
+ return e
+}
+
+// MakeAdapterDisconnectEvent 创建一个适配器断开连接事件
+func MakeAdapterDisconnectEvent() *IceinuEvent {
+ e := &IceinuEvent{
+ Type: 9,
+ From: GetSelfNodeId(),
+ Target: GetMasterNodeId(),
+ Timestamp: time.Now().Unix(),
+ Summary: "AdapterDisconnectEvent",
+ Event: &AdapterDisconnectEvent{
+ OK: true,
+ },
+ }
+ Publish(e)
+ return e
+}
diff --git a/ice/plugin_manager.go b/ice/plugin_manager.go
new file mode 100644
index 0000000..dabb44e
--- /dev/null
+++ b/ice/plugin_manager.go
@@ -0,0 +1,20 @@
+package ice
+
+var PluginVerifier string
+
+// GetPluginVerifier 获取插件的校验值，用于确认各个实例之间的插件版本是否一致
+//
+// 校验是可选的，在氷犬的消息事件系统中实例间的插件是否相同实际上并不影响组网，但是如果插件不同可能会导致一些意想不到的问题
+func GetPluginVerifier(isVerify bool) string {
+ if !isVerify {
+ PluginVerifier = "pass"
+ } else {
+ VerifyPlugin()
+ }
+ return PluginVerifier
+}
+
+// VerifyPlugin 校验插件是否一致，暂时是空函数
+func VerifyPlugin() {
+
+}
diff --git a/log_format.go b/log_format.go
index 8642106..e655eba 100644
--- a/log_format.go
+++ b/log_format.go
@@ -16,7 +16,7 @@ func (f *LogFormatter) Format(entry *logrus.Entry) ([]byte, error) {
var levelText string
switch entry.Level {
case logrus.DebugLevel, logrus.TraceLevel:
- levelColor = gradient.DarkGreen
+ levelColor = gradient.LightGreen
textColor = gradient.DarkGreen
levelText = "🚧DEBUG"
case logrus.InfoLevel:
diff --git a/main.go b/main.go
index c6ac7cc..59ba194 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,8 @@ package main
import (
"github.com/Iceinu-Project/IceGradient"
+ "github.com/Iceinu-Project/Iceinu/adapters"
+ "github.com/Iceinu-Project/Iceinu/adapters/lagrange"
"github.com/Iceinu-Project/Iceinu/config"
"github.com/Iceinu-Project/Iceinu/ice"
"github.com/Iceinu-Project/Iceinu/log"
@@ -18,21 +20,31 @@ func main() {
// 初始化内置配置文件读取
config.IceConfigInit()
- // 设置日志级别
- log.SetLevel(config.IceConf.LogLevel)
- log.Debugf("调试模式已启用")
// 输出欢迎日志
log.Infof("欢迎使用🧊" + gradient.Bold +
gradient.GradientText("氷犬 Iceinu Bot", "#00d2ff", "#3a7bd5") + gradient.DarkGray + " | " +
gradient.RGBToANSI(255, 255, 255) +
gradient.GradientBackgroundText(" 通用的模块化 Go 聊天机器人框架 ", "#00d2ff", "#3a7bd5") +
gradient.Reset)
- log.Infof("当前版本: " + gradient.Cyan + "β0.1.4")
+ log.Infof("当前版本: " + gradient.Cyan + "Re:β 0.2.4")
// 初始化数据库连接
ice.InitLocalDatabase()
+ // 设置日志级别
+ log.SetLevel(config.IceConf.LogLevel)
+ log.Debugf("调试模式已启用")
+ // 设置自定义中间件
+ SetCustomMiddleWare()
// 如果启用集群/分布式模式，则尝试和主节点建立连接
// 输出节点Id和集群模式
- log.Info("当前节点模式: " + gradient.DarkGray + config.IceConf.Node.Mode)
- log.Info("Node Id: " + gradient.DarkGray + ice.GetSelfNodeId())
- log.Info("MasterNode Id: " + gradient.DarkGray + ice.GetMasterNodeId())
+ log.Info("当前节点模式: " + gradient.Cyan + config.IceConf.Node.Mode)
+ log.Info("Node Id: " + gradient.Cyan + ice.GetSelfNodeId())
+ log.Info("MasterNode Id: " + gradient.Cyan + ice.GetMasterNodeId())
+ // 启动适配器
+ var adapter adapters.IceinuAdapter
+ adapter = lagrange.GetAdapter()
+ err := adapter.Init()
+ if err != nil {
+ return
+ }
+ adapter.Start()
}
diff --git a/middlewares.go b/middlewares.go
index 4939abf..79e43b2 100644
--- a/middlewares.go
+++ b/middlewares.go
@@ -1,6 +1,41 @@
package main
+import (
+ "github.com/Iceinu-Project/Iceinu/ice"
+ "github.com/Iceinu-Project/Iceinu/log"
+ "github.com/Iceinu-Project/Iceinu/models/satori"
+)
+
// SetCustomMiddleWare 设置自定义中间件
func SetCustomMiddleWare() {
+ // 这个函数中定义了一系列事件总线的中间件，你可以直接参考这些中间件以及氷犬的在线文档来编写中间件
+
+ ice.UseGlobalPublishMiddleware(func(event *ice.IceinuEvent, next func(event *ice.IceinuEvent)) {
+ // 默认的事件监听中间件示例，它会在任意事件发布时运行
+ log.Debugf("接收到来自节点 %s 的 %s 事件", event.From, event.Summary)
+ next(event)
+ })
+
+ ice.UseTypePublishMiddleware(8, func(event *ice.IceinuEvent, next func(event *ice.IceinuEvent)) {
+ // 适配器连接事件监听中间件，它会在适配器连接事件发布时运行
+ // 提取事件里的参数
+ adapterConnectEvent := event.Event.(*ice.AdapterConnectEvent)
+ log.Infof("来自节点 %s 的%s连接成功，机器人Id %s，机器人名称 %s", event.From, adapterConnectEvent.AdapterType, adapterConnectEvent.UserId, adapterConnectEvent.UserName)
+ next(event)
+ })
+
+ ice.UseSummaryPublishMiddleware("PrivateMessageEvent", func(event *ice.IceinuEvent, next func(event *ice.IceinuEvent)) {
+ // Satori私聊消息事件监听中间件，它会在私聊消息事件发布时运行
+ privateMessageEvent := event.Event.(*satori.EventSatori)
+ log.Infof("[%s][%s][私聊]%s@%s：%s", privateMessageEvent.Platform, privateMessageEvent.User.Name, privateMessageEvent.Operator.Id, privateMessageEvent.Operator.Nickname, satori.ElementsToSatori(*privateMessageEvent.Message.MessageElements))
+ next(event)
+ })
+
+ ice.UseSummaryPublishMiddleware("GroupMessageEvent", func(event *ice.IceinuEvent, next func(event *ice.IceinuEvent)) {
+ // Satori群聊消息事件监听中间件，它会在群聊消息事件发布时运行
+ groupMessageEvent := event.Event.(*satori.EventSatori)
+ log.Infof("[%s][%s][群聊][%s]%s@%s：%s", groupMessageEvent.Platform, groupMessageEvent.User.Name, groupMessageEvent.Group.Name, groupMessageEvent.Operator.Id, groupMessageEvent.Operator.Nickname, satori.ElementsToSatori(*groupMessageEvent.Message.MessageElements))
+ next(event)
+ })
}
diff --git a/models/satori/satori_elements.go b/models/satori/satori_elements.go
index 7bab417..cb875ae 100644
--- a/models/satori/satori_elements.go
+++ b/models/satori/satori_elements.go
@@ -1,10 +1,14 @@
package satori
+// 参考：https://satori.js.org/zh-CN/protocol/elements.html
+// 在Satori的标准元素基础上扩展了一部分元素，方便直接使用
+
// ElementSatori Satori标准事件元素接口
type ElementSatori interface {
GetType() string
}
+// TextElement 文本消息元素
type TextElement struct {
Text string
}
@@ -13,6 +17,7 @@ func (t *TextElement) GetType() string {
return "text"
}
+// AtElement 提及用户消息元素
type AtElement struct {
Id string
Name string
@@ -24,6 +29,7 @@ func (a *AtElement) GetType() string {
return "at"
}
+// SharpElement 提及频道消息元素
type SharpElement struct {
Id string
Name string
@@ -33,6 +39,7 @@ func (s *SharpElement) GetType() string {
return "sharp"
}
+// AElement 超链接消息元素
type AElement struct {
Href string
}
@@ -41,6 +48,7 @@ func (a *AElement) GetType() string {
return "a"
}
+// ImgElement 图片消息元素
type ImgElement struct {
Src string
Title string
@@ -54,6 +62,7 @@ func (i *ImgElement) GetType() string {
return "img"
}
+// AudioElement 音频消息元素
type AudioElement struct {
Src string
Title string
@@ -67,6 +76,7 @@ func (a *AudioElement) GetType() string {
return "audio"
}
+// VideoElement 视频消息元素
type VideoElement struct {
Src string
Title string
@@ -82,6 +92,7 @@ func (v *VideoElement) GetType() string {
return "video"
}
+// FileElement 文件消息元素
type FileElement struct {
Src string
Title string
@@ -94,6 +105,7 @@ func (f *FileElement) GetType() string {
return "file"
}
+// StrongElement 加粗消息元素
type StrongElement struct {
Elements *[]ElementSatori
}
@@ -102,6 +114,7 @@ func (s *StrongElement) GetType() string {
return "strong"
}
+// EmElement 斜体消息元素
type EmElement struct {
Elements *[]ElementSatori
}
@@ -110,6 +123,7 @@ func (e *EmElement) GetType() string {
return "em"
}
+// InsElement 下划线消息元素
type InsElement struct {
Elements *[]ElementSatori
}
@@ -118,6 +132,7 @@ func (i *InsElement) GetType() string {
return "ins"
}
+// DelElement 删除线消息元素
type DelElement struct {
Elements *[]ElementSatori
}
@@ -126,6 +141,7 @@ func (d *DelElement) GetType() string {
return "del"
}
+// SpoilerElement 剧透消息元素
type SpoilerElement struct {
Elements *[]ElementSatori
}
@@ -134,6 +150,7 @@ func (s *SpoilerElement) GetType() string {
return "spoiler"
}
+// CodeElement 代码消息元素
type CodeElement struct {
Elements *[]ElementSatori
}
@@ -142,6 +159,7 @@ func (c *CodeElement) GetType() string {
return "code"
}
+// SupElement 上标消息元素
type SupElement struct {
Elements *[]ElementSatori
}
@@ -150,6 +168,7 @@ func (s *SupElement) GetType() string {
return "sup"
}
+// SubElement 下标消息元素
type SubElement struct {
Elements *[]ElementSatori
}
@@ -158,6 +177,7 @@ func (s *SubElement) GetType() string {
return "sub"
}
+// BrElement 换行消息元素
type BrElement struct {
}
@@ -165,6 +185,7 @@ func (b *BrElement) GetType() string {
return "br"
}
+// HrElement 分割线消息元素
type HrElement struct {
}
@@ -172,6 +193,7 @@ func (h *HrElement) GetType() string {
return "hr"
}
+// PElement 段落消息元素
type PElement struct {
Elements *[]ElementSatori
}
@@ -180,6 +202,7 @@ func (p *PElement) GetType() string {
return "p"
}
+// MessageElement 消息元素
type MessageElement struct {
Id string
Forward bool
@@ -190,6 +213,7 @@ func (m *MessageElement) GetType() string {
return "message"
}
+// QuoteElement 引用消息元素
type QuoteElement struct {
Id string
Name string
@@ -203,6 +227,7 @@ func (q *QuoteElement) GetType() string {
return "quote"
}
+// AuthorElement 作者消息元素
type AuthorElement struct {
Id string
Name string
@@ -213,6 +238,7 @@ func (a *AuthorElement) GetType() string {
return "author"
}
+// ButtonElement 按钮消息元素
type ButtonElement struct {
Id string
Type string
@@ -224,3 +250,35 @@ type ButtonElement struct {
func (b *ButtonElement) GetType() string {
return "button"
}
+
+// FaceElement 表情消息元素
+type FaceElement struct {
+ Id uint16
+ IsLargeFace bool
+}
+
+func (f *FaceElement) GetType() string {
+ return "face"
+}
+
+// NodeElement 节点消息元素
+type NodeElement struct {
+ GroupId int64
+ SenderId int64
+ SenderName string
+ Time int32
+ Message *[]ElementSatori
+}
+
+func (n *NodeElement) GetType() string {
+ return "node"
+}
+
+// UnsupportedElement 未支持的消息元素
+type UnsupportedElement struct {
+ Type string
+}
+
+func (u *UnsupportedElement) GetType() string {
+ return "unsupported"
+}
diff --git a/models/satori/satori_model.go b/models/satori/satori_model.go
index c74b048..c421290 100644
--- a/models/satori/satori_model.go
+++ b/models/satori/satori_model.go
@@ -1,7 +1,5 @@
package satori
-import "time"
-
// EventSatori 基于Satori标准设计的事件接收结构体，出于方便使用进行了部分修改和拓展
type EventSatori struct {
Id uint64 `json:"id,omitempty"` // 事件ID
@@ -40,10 +38,10 @@ type Group struct {
// GroupMember 群组成员结构体
type GroupMember struct {
- User *User `json:"user,omitempty"`
- Nickname string `json:"nickname,omitempty"`
- Avatar string `json:"avatar,omitempty"`
- JoinedAt time.Time `json:"joinedAt"`
+ User *User `json:"user,omitempty"`
+ Nickname string `json:"nickname,omitempty"`
+ Avatar string `json:"avatar,omitempty"`
+ JoinedAt int64 `json:"joinedAt"`
}
// GroupRole 群组角色结构体
@@ -82,8 +80,8 @@ type Message struct {
Group *Group `json:"group,omitempty"`
Member *GroupMember `json:"member,omitempty"`
User *User `json:"user,omitempty"`
- CreatedAt time.Time `json:"createdAt"`
- UpdatedAt time.Time `json:"updatedAt"`
+ CreatedAt int64 `json:"createdAt"`
+ UpdatedAt int64 `json:"updatedAt"`
MessageElements *[]ElementSatori `json:"messageElements,omitempty"`
}
diff --git a/models/satori/tools.go b/models/satori/tools.go
index 194b9c5..7a06dfb 100644
--- a/models/satori/tools.go
+++ b/models/satori/tools.go
@@ -1,11 +1,132 @@
package satori
+import (
+ "strconv"
+ "strings"
+)
+
// ParseSatori 从Satori标准XHTML消息中解析出Satori标准事件元素
func ParseSatori(content string) *[]ElementSatori {
return nil
}
// ElementsToSatori 将Satori标准事件元素转换为Satori标准XHTML消息字符串
-func ElementsToSatori(elements *[]ElementSatori) string {
- return ""
+func ElementsToSatori(elements []ElementSatori) string {
+ var sb strings.Builder
+
+ for _, element := range elements {
+ switch e := element.(type) {
+ case *TextElement:
+ sb.WriteString(escapeXML(e.Text))
+ case *AtElement:
+ sb.WriteString(``)
+ case *SharpElement:
+ sb.WriteString(``)
+ case *AElement:
+ sb.WriteString(``)
+ case *ImgElement:
+ sb.WriteString(` + ` "` + escapeXML(e.Title) + `")
`)
+ case *AudioElement:
+ sb.WriteString(``)
+ case *VideoElement:
+ sb.WriteString(``)
+ case *FileElement:
+ sb.WriteString(``)
+ case *StrongElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *EmElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *InsElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *DelElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *SpoilerElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *CodeElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *SupElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *SubElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *BrElement:
+ sb.WriteString(`
`)
+ case *HrElement:
+ sb.WriteString(`
`)
+ case *PElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(`
`)
+ case *MessageElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(``)
+ case *QuoteElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Elements))
+ sb.WriteString(`
`)
+ case *AuthorElement:
+ sb.WriteString(``)
+ case *ButtonElement:
+ sb.WriteString(``)
+ case *FaceElement:
+ sb.WriteString(``)
+ case *NodeElement:
+ sb.WriteString(``)
+ sb.WriteString(ElementsToSatori(*e.Message))
+ sb.WriteString(``)
+ case *UnsupportedElement:
+ sb.WriteString(``)
+ }
+ }
+
+ return sb.String()
+}
+
+// escapeXML 转义特殊字符
+func escapeXML(s string) string {
+ return strings.ReplaceAll(strings.ReplaceAll(strings.ReplaceAll(strings.ReplaceAll(s, "&", "&"), "\"", """), "<", "<"), ">", ">")
+}
+
+// boolToString 将布尔值转换为字符串
+func boolToString(b bool) string {
+ if b {
+ return "true"
+ }
+ return "false"
+}
+
+// uint32ToString 将 uint32 转换为字符串
+func uint32ToString(u uint32) string {
+ return strconv.FormatUint(uint64(u), 10)
+}
+
+// int64ToString 将 int64 转换为字符串
+func int64ToString(i int64) string {
+ return strconv.FormatInt(i, 10)
+}
+
+// int32ToString 将 int32 转换为字符串
+func int32ToString(i int32) string {
+ return strconv.FormatInt(int64(i), 10)
+}
+
+// uint16ToString 将 uint16 转换为字符串
+func uint16ToString(u uint16) string {
+ return strconv.FormatUint(uint64(u), 10)
}