report temp message

Author: lz1998

pkg/plugin/plugin.go

@@ -9,6 +9,7 @@ import (
 type (
 	PrivateMessagePlugin        = func(*client.QQClient, *message.PrivateMessage) int32
 	GroupMessagePlugin          = func(*client.QQClient, *message.GroupMessage) int32
+	TempMessagePlugin           = func(*client.QQClient, *message.TempMessage) int32
 	MemberJoinGroupPlugin       = func(*client.QQClient, *client.MemberJoinGroupEvent) int32
 	MemberLeaveGroupPlugin      = func(*client.QQClient, *client.MemberLeaveGroupEvent) int32
 	JoinGroupPlugin             = func(*client.QQClient, *client.GroupInfo) int32
@@ -28,6 +29,7 @@ const (
 
 var PrivateMessagePluginList = make([]PrivateMessagePlugin, 0)
 var GroupMessagePluginList = make([]GroupMessagePlugin, 0)
+var TempMessagePluginList = make([]TempMessagePlugin, 0)
 var MemberJoinGroupPluginList = make([]MemberJoinGroupPlugin, 0)
 var MemberLeaveGroupPluginList = make([]MemberLeaveGroupPlugin, 0)
 var JoinGroupPluginList = make([]JoinGroupPlugin, 0)
@@ -42,6 +44,7 @@ var NewFriendAddedPluginList = make([]NewFriendAddedPlugin, 0)
 func Serve(cli *client.QQClient) {
 	cli.OnPrivateMessage(handlePrivateMessage)
 	cli.OnGroupMessage(handleGroupMessage)
+	cli.OnTempMessage(handleTempMessage)
 	cli.OnGroupMemberJoined(handleMemberJoinGroup)
 	cli.OnGroupMemberLeaved(handleMemberLeaveGroup)
 	cli.OnJoinGroup(handleJoinGroup)
@@ -64,6 +67,11 @@ func AddGroupMessagePlugin(plugin GroupMessagePlugin) {
 	GroupMessagePluginList = append(GroupMessagePluginList, plugin)
 }
 
+// 添加临时消息插件
+func AddTempMessagePlugin(plugin TempMessagePlugin) {
+	TempMessagePluginList = append(TempMessagePluginList, plugin)
+}
+
 // 添加群成员加入插件
 func AddMemberJoinGroupPlugin(plugin MemberJoinGroupPlugin) {
 	MemberJoinGroupPluginList = append(MemberJoinGroupPluginList, plugin)
@@ -134,6 +142,16 @@ func handleGroupMessage(cli *client.QQClient, event *message.GroupMessage) {
 	})
 }
 
+func handleTempMessage(cli *client.QQClient, event *message.TempMessage) {
+	SafeGo(func() {
+		for _, plugin := range TempMessagePluginList {
+			if result := plugin(cli, event); result == MessageBlock {
+				break
+			}
+		}
+	})
+}
+
 func handleMemberJoinGroup(cli *client.QQClient, event *client.MemberJoinGroupEvent) {
 	SafeGo(func() {
 		for _, plugin := range MemberJoinGroupPluginList {

service/handler/bot.go

@@ -133,6 +133,7 @@ func CreateBotImpl(uin int64, password string) {
 	log.Infof("加载上报插件 Report")
 	plugin.AddPrivateMessagePlugin(plugins.ReportPrivateMessage)
 	plugin.AddGroupMessagePlugin(plugins.ReportGroupMessage)
+	plugin.AddTempMessagePlugin(plugins.ReportTempMessage)
 	plugin.AddMemberJoinGroupPlugin(plugins.ReportMemberJoin)
 	plugin.AddMemberLeaveGroupPlugin(plugins.ReportMemberLeave)
 	plugin.AddJoinGroupPlugin(plugins.ReportJoinGroup)

service/plugins/report.go

@@ -97,6 +97,32 @@ func ReportGroupMessage(cli *client.QQClient, event *message.GroupMessage) int32
 	return plugin.MessageIgnore
 }
 
+func ReportTempMessage(cli *client.QQClient, event *message.TempMessage) int32 {
+	// TODO 撤回？
+	eventProto := &onebot.Frame{
+		FrameType: onebot.Frame_TPrivateMessageEvent,
+	}
+	eventProto.Data = &onebot.Frame_PrivateMessageEvent{
+		PrivateMessageEvent: &onebot.PrivateMessageEvent{
+			Time:        time.Now().Unix(),
+			SelfId:      cli.Uin,
+			PostType:    "message",
+			MessageType: "private",
+			SubType:     "group",
+			MessageId:   event.Id,
+			UserId:      event.Sender.Uin,
+			Message:     bot.MiraiMsgToProtoMsg(event.Elements),
+			RawMessage:  bot.MiraiMsgToRawMsg(event.Elements),
+			Sender: &onebot.PrivateMessageEvent_Sender{
+				UserId:   event.Sender.Uin,
+				Nickname: event.Sender.Nickname,
+			},
+		},
+	}
+	bot.HandleEventFrame(cli, eventProto)
+	return plugin.MessageIgnore
+}
+
 func ReportMemberJoin(cli *client.QQClient, event *client.MemberJoinGroupEvent) int32 {
 	eventProto := &onebot.Frame{
 		FrameType: onebot.Frame_TGroupIncreaseNoticeEvent,