join leave group event

Author: lz1998

main.go

@@ -53,6 +53,8 @@ func main() {
 	plugin.AddGroupMessagePlugin(plugins.ReportGroupMessage)
 	plugin.AddMemberJoinGroupPlugin(plugins.ReportMemberJoin)
 	plugin.AddMemberLeaveGroupPlugin(plugins.ReportMemberLeave)
+	plugin.AddJoinGroupPlugin(plugins.ReportJoinGroup)
+	plugin.AddLeaveGroupPlugin(plugins.ReportLeaveGroup)
 
 	plugin.Serve(cli)
 	log.Infof("插件加载完成")

pkg/plugin/plugin.go

@@ -11,6 +11,8 @@ type (
 	GroupMessagePlugin     = func(*client.QQClient, *message.GroupMessage) int32
 	MemberJoinGroupPlugin  = func(*client.QQClient, *client.MemberJoinGroupEvent) int32
 	MemberLeaveGroupPlugin = func(*client.QQClient, *client.MemberLeaveGroupEvent) int32
+	JoinGroupPlugin        = func(*client.QQClient, *client.GroupInfo) int32
+	LeaveGroupPlugin       = func(*client.QQClient, *client.GroupLeaveEvent) int32
 )
 
 const (
@@ -22,12 +24,16 @@ var PrivateMessagePluginList = make([]PrivateMessagePlugin, 0)
 var GroupMessagePluginList = make([]GroupMessagePlugin, 0)
 var MemberJoinGroupPluginList = make([]MemberJoinGroupPlugin, 0)
 var MemberLeaveGroupPluginList = make([]MemberLeaveGroupPlugin, 0)
+var JoinGroupPluginList = make([]JoinGroupPlugin, 0)
+var LeaveGroupPluginList = make([]LeaveGroupPlugin, 0)
 
 func Serve(cli *client.QQClient) {
 	cli.OnPrivateMessage(handlePrivateMessage)
 	cli.OnGroupMessage(handleGroupMessage)
 	cli.OnGroupMemberJoined(handleMemberJoinGroup)
 	cli.OnGroupMemberLeaved(handleMemberLeaveGroup)
+	cli.OnJoinGroup(handleJoinGroup)
+	cli.OnLeaveGroup(handleLeaveGroup)
 }
 
 // 添加私聊消息插件
@@ -50,6 +56,16 @@ func AddMemberLeaveGroupPlugin(plugin MemberLeaveGroupPlugin) {
 	MemberLeaveGroupPluginList = append(MemberLeaveGroupPluginList, plugin)
 }
 
+// 添加机器人进群插件
+func AddJoinGroupPlugin(plugin JoinGroupPlugin) {
+	JoinGroupPluginList = append(JoinGroupPluginList, plugin)
+}
+
+// 添加机器人离开群插件
+func AddLeaveGroupPlugin(plugin LeaveGroupPlugin) {
+	LeaveGroupPluginList = append(LeaveGroupPluginList, plugin)
+}
+
 func handlePrivateMessage(cli *client.QQClient, event *message.PrivateMessage) {
 	SafeGo(func() {
 		for _, plugin := range PrivateMessagePluginList {
@@ -90,6 +106,26 @@ func handleMemberLeaveGroup(cli *client.QQClient, event *client.MemberLeaveGroup
 	})
 }
 
+func handleJoinGroup(cli *client.QQClient, event *client.GroupInfo) {
+	SafeGo(func() {
+		for _, plugin := range JoinGroupPluginList {
+			if result := plugin(cli, event); result == MessageBlock {
+				break
+			}
+		}
+	})
+}
+
+func handleLeaveGroup(cli *client.QQClient, event *client.GroupLeaveEvent) {
+	SafeGo(func() {
+		for _, plugin := range LeaveGroupPluginList {
+			if result := plugin(cli, event); result == MessageBlock {
+				break
+			}
+		}
+	})
+}
+
 func SafeGo(fn func()) {
 	go func() {
 		defer func() {

service/plugins/report.go

@@ -1,12 +1,13 @@
 package plugins
 
 import (
+	"time"
+
 	"github.com/Mrs4s/MiraiGo/client"
 	"github.com/Mrs4s/MiraiGo/message"
 	"github.com/ProtobufBot/Go-Mirai-Client/pkg/plugin"
 	"github.com/ProtobufBot/Go-Mirai-Client/proto_gen/onebot"
 	"github.com/ProtobufBot/Go-Mirai-Client/service/bot"
-	"time"
 )
 
 func ReportPrivateMessage(cli *client.QQClient, event *message.PrivateMessage) int32 {
@@ -71,6 +72,7 @@ func ReportMemberJoin(cli *client.QQClient, event *client.MemberJoinGroupEvent)
 			SubType:    "approve",
 			GroupId:    event.Group.Code,
 			UserId:     event.Member.Uin,
+			OperatorId: 0,
 		},
 	}
 	bot.HandleEventFrame(cli, eventProto)
@@ -103,3 +105,50 @@ func ReportMemberLeave(cli *client.QQClient, event *client.MemberLeaveGroupEvent
 	bot.HandleEventFrame(cli, eventProto)
 	return plugin.MessageIgnore
 }
+
+func ReportJoinGroup(cli *client.QQClient, event *client.GroupInfo) int32 {
+	eventProto := &onebot.Frame{
+		FrameType: onebot.Frame_TGroupIncreaseNoticeEvent,
+	}
+	eventProto.Data = &onebot.Frame_GroupIncreaseNoticeEvent{
+		GroupIncreaseNoticeEvent: &onebot.GroupIncreaseNoticeEvent{
+			Time:       time.Now().Unix(),
+			SelfId:     cli.Uin,
+			PostType:   "message",
+			NoticeType: "group_increase",
+			SubType:    "approve",
+			GroupId:    event.Code,
+			UserId:     cli.Uin,
+			OperatorId: 0,
+		},
+	}
+	bot.HandleEventFrame(cli, eventProto)
+	return plugin.MessageIgnore
+}
+
+func ReportLeaveGroup(cli *client.QQClient, event *client.GroupLeaveEvent) int32 {
+	eventProto := &onebot.Frame{
+		FrameType: onebot.Frame_TGroupDecreaseNoticeEvent,
+	}
+	subType := "leave"
+	var operatorId int64 = 0
+	if event.Operator != nil {
+		subType = "kick"
+		operatorId = event.Operator.Uin
+	}
+
+	eventProto.Data = &onebot.Frame_GroupDecreaseNoticeEvent{
+		GroupDecreaseNoticeEvent: &onebot.GroupDecreaseNoticeEvent{
+			Time:       time.Now().Unix(),
+			SelfId:     cli.Uin,
+			PostType:   "message",
+			NoticeType: "group_decrease",
+			SubType:    subType,
+			GroupId:    event.Group.Code,
+			UserId:     cli.Uin,
+			OperatorId: operatorId,
+		},
+	}
+	bot.HandleEventFrame(cli, eventProto)
+	return plugin.MessageIgnore
+}