member join/leave

Author: lz1998

main.go

@@ -51,6 +51,8 @@ func main() {
 	log.Infof("加载上报插件 Report")
 	plugin.AddPrivateMessagePlugin(plugins.ReportPrivateMessage)
 	plugin.AddGroupMessagePlugin(plugins.ReportGroupMessage)
+	plugin.AddMemberJoinGroupPlugin(plugins.ReportMemberJoin)
+	plugin.AddMemberLeaveGroupPlugin(plugins.ReportMemberLeave)
 
 	plugin.Serve(cli)
 	log.Infof("插件加载完成")

pkg/plugin/plugin.go

@@ -7,8 +7,10 @@ import (
 )
 
 type (
-	PrivateMessagePlugin = func(*client.QQClient, *message.PrivateMessage) int32
-	GroupMessagePlugin   = func(*client.QQClient, *message.GroupMessage) int32
+	PrivateMessagePlugin   = func(*client.QQClient, *message.PrivateMessage) int32
+	GroupMessagePlugin     = func(*client.QQClient, *message.GroupMessage) int32
+	MemberJoinGroupPlugin  = func(*client.QQClient, *client.MemberJoinGroupEvent) int32
+	MemberLeaveGroupPlugin = func(*client.QQClient, *client.MemberLeaveGroupEvent) int32
 )
 
 const (
@@ -18,11 +20,14 @@ const (
 
 var PrivateMessagePluginList = make([]PrivateMessagePlugin, 0)
 var GroupMessagePluginList = make([]GroupMessagePlugin, 0)
+var MemberJoinGroupPluginList = make([]MemberJoinGroupPlugin, 0)
+var MemberLeaveGroupPluginList = make([]MemberLeaveGroupPlugin, 0)
 
 func Serve(cli *client.QQClient) {
 	cli.OnPrivateMessage(handlePrivateMessage)
 	cli.OnGroupMessage(handleGroupMessage)
-
+	cli.OnGroupMemberJoined(handleMemberJoinGroup)
+	cli.OnGroupMemberLeaved(handleMemberLeaveGroup)
 }
 
 // 添加私聊消息插件
@@ -35,20 +40,50 @@ func AddGroupMessagePlugin(plugin GroupMessagePlugin) {
 	GroupMessagePluginList = append(GroupMessagePluginList, plugin)
 }
 
-func handlePrivateMessage(cli *client.QQClient, msg *message.PrivateMessage) {
+// 添加群成员加入插件
+func AddMemberJoinGroupPlugin(plugin MemberJoinGroupPlugin) {
+	MemberJoinGroupPluginList = append(MemberJoinGroupPluginList, plugin)
+}
+
+// 添加群成员离开插件
+func AddMemberLeaveGroupPlugin(plugin MemberLeaveGroupPlugin) {
+	MemberLeaveGroupPluginList = append(MemberLeaveGroupPluginList, plugin)
+}
+
+func handlePrivateMessage(cli *client.QQClient, event *message.PrivateMessage) {
 	SafeGo(func() {
 		for _, plugin := range PrivateMessagePluginList {
-			if result := plugin(cli, msg); result == MessageBlock {
+			if result := plugin(cli, event); result == MessageBlock {
 				break
 			}
 		}
 	})
 }
 
-func handleGroupMessage(cli *client.QQClient, msg *message.GroupMessage) {
+func handleGroupMessage(cli *client.QQClient, event *message.GroupMessage) {
 	SafeGo(func() {
 		for _, plugin := range GroupMessagePluginList {
-			if result := plugin(cli, msg); result == MessageBlock {
+			if result := plugin(cli, event); result == MessageBlock {
+				break
+			}
+		}
+	})
+}
+
+func handleMemberJoinGroup(cli *client.QQClient, event *client.MemberJoinGroupEvent) {
+	SafeGo(func() {
+		for _, plugin := range MemberJoinGroupPluginList {
+			if result := plugin(cli, event); result == MessageBlock {
+				break
+			}
+		}
+	})
+}
+
+func handleMemberLeaveGroup(cli *client.QQClient, event *client.MemberLeaveGroupEvent) {
+	SafeGo(func() {
+		for _, plugin := range MemberLeaveGroupPluginList {
+			if result := plugin(cli, event); result == MessageBlock {
 				break
 			}
 		}

service/plugins/report.go

@@ -57,3 +57,49 @@ func ReportGroupMessage(cli *client.QQClient, event *message.GroupMessage) int32
 	bot.HandleEventFrame(cli, eventProto)
 	return plugin.MessageIgnore
 }
+
+func ReportMemberJoin(cli *client.QQClient, event *client.MemberJoinGroupEvent) 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.Group.Code,
+			UserId:     event.Member.Uin,
+		},
+	}
+	bot.HandleEventFrame(cli, eventProto)
+	return plugin.MessageIgnore
+}
+
+func ReportMemberLeave(cli *client.QQClient, event *client.MemberLeaveGroupEvent) 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:     event.Member.Uin,
+			OperatorId: operatorId,
+		},
+	}
+	bot.HandleEventFrame(cli, eventProto)
+	return plugin.MessageIgnore
+}