新插件event, 好友申请群聊邀请事件处理

README.md

@@ -556,6 +556,14 @@ print("run[CQ:image,file="+j["img"]+"]")
 
   - [x] xxx疫情
 
+</details>
+<details>
+  <summary>城市疫情查询</summary>
+
+  `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/event"`
+
+  - [x] [开启|关闭]自动同意[申请|邀请|主人]
+
 </details>
 <details>
   <summary>渲染任意文字到图片</summary>

main.go

@@ -76,6 +76,7 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/drift_bottle"  // 漂流瓶
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/emojimix"      // 合成emoji
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/epidemic"      // 城市疫情查询
+	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/event"         // 好友申请群聊邀请事件处理
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/font"          // 渲染任意文字到图片
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/fortune"       // 运势
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/funny"         // 笑话

plugin/event/event.go

@@ -0,0 +1,143 @@
+// Package event 好友申请以及群聊邀请事件处理
+package event
+
+import (
+	"strconv"
+	"time"
+
+	ctrl "github.com/FloatTech/zbpctrl"
+	"github.com/FloatTech/zbputils/control"
+	"github.com/sirupsen/logrus"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+func init() {
+	engine := control.Register("event", &ctrl.Options[*zero.Ctx]{
+		DisableOnDefault: false,
+		Help: "好友申请以及群聊邀请事件处理，默认发送给主人列表第一位\n" +
+			" - [开启|关闭]自动同意[申请|邀请|主人]",
+	})
+	engine.OnRequest().SetBlock(false).
+		Handle(func(ctx *zero.Ctx) {
+			c, ok := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
+			if ok {
+				su := zero.BotConfig.SuperUsers[0]
+				now := time.Unix(ctx.Event.Time, 0).Format("2006-01-02 15:04:05")
+				flag := ctx.Event.Flag
+				comment := ctx.Event.Comment
+				userid := ctx.Event.UserID
+				username := ctx.CardOrNickName(userid)
+				data := c.GetData(-su)
+				switch ctx.Event.RequestType {
+				case "friend":
+					logrus.Info("[event]收到来自[", username, "](", userid, ")的好友申请")
+					if data&1 == 1 || data&0x20 == 0x20 && zero.SuperUserPermission(ctx) {
+						ctx.SetFriendAddRequest(flag, true, "")
+						ctx.SendPrivateForwardMessage(su, message.Message{message.CustomNode(username, userid,
+							"已自动同意在"+now+"收到来自"+
+								"\n用户:["+username+"]("+strconv.FormatInt(userid, 10)+")"+
+								"\n的好友请求:"+comment+
+								"\nflag:"+flag)})
+						return
+					}
+					ctx.SendPrivateForwardMessage(su,
+						message.Message{message.CustomNode(username, userid,
+							"在"+now+"收到来自"+
+								"\n用户:["+username+"]("+strconv.FormatInt(userid, 10)+")"+
+								"\n的好友请求:"+comment+
+								"\n请在下方复制flag并在前面加上:"+
+								"\n同意/拒绝申请，来决定同意还是拒绝"),
+							message.CustomNode(username, userid, flag)})
+				case "group":
+					if ctx.Event.SubType != "invite" {
+						return
+					}
+					groupid := ctx.Event.GroupID
+					groupname := ctx.GetGroupInfo(groupid, true).Name
+					logrus.Info("[event]收到来自[", username, "](", userid, ")的群聊邀请，群:[", groupname, "](", groupid, ")")
+					if data&0x10 == 0x10 || data&0x20 == 0x20 && zero.SuperUserPermission(ctx) {
+						ctx.SetGroupAddRequest(flag, "invite", true, "")
+						ctx.SendPrivateForwardMessage(su, message.Message{message.CustomNode(username, userid,
+							"已自动同意在"+now+"收到来自"+
+								"\n用户:["+username+"]("+strconv.FormatInt(userid, 10)+")的群聊邀请"+
+								"\n群聊:["+groupname+"]("+strconv.FormatInt(groupid, 10)+")"+
+								"\n验证信息:\n"+comment+
+								"\nflag:"+flag)})
+						return
+					}
+					ctx.SendPrivateForwardMessage(su,
+						message.Message{message.CustomNode(username, userid,
+							"在"+now+"收到来自"+
+								"\n用户:["+username+"]("+strconv.FormatInt(userid, 10)+")的群聊邀请"+
+								"\n群聊:["+groupname+"]("+strconv.FormatInt(groupid, 10)+")"+
+								"\n验证信息:\n"+comment+
+								"\n请在下方复制flag并在前面加上:"+
+								"\n同意/拒绝邀请，来决定同意还是拒绝"),
+							message.CustomNode(username, userid, flag)})
+				}
+			}
+		})
+	engine.OnRegex(`^(同意|拒绝)(申请|邀请)\s*(\d+)\s*(.*)$`, zero.SuperUserPermission, zero.OnlyPrivate).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			su := zero.BotConfig.SuperUsers[0]
+			cmd := ctx.State["regex_matched"].([]string)[1]
+			org := ctx.State["regex_matched"].([]string)[2]
+			flag := ctx.State["regex_matched"].([]string)[3]
+			other := ctx.State["regex_matched"].([]string)[4]
+			switch cmd {
+			case "同意":
+				switch org {
+				case "申请":
+					ctx.SetFriendAddRequest(flag, true, other)
+					ctx.SendPrivateMessage(su, message.Text("已", cmd, org))
+				case "邀请":
+					ctx.SetGroupAddRequest(flag, "invite", true, "")
+					ctx.SendPrivateMessage(su, message.Text("已", cmd, org))
+				}
+			case "拒绝":
+				switch org {
+				case "申请":
+					ctx.SetFriendAddRequest(flag, false, "")
+					ctx.SendPrivateMessage(su, message.Text("已", cmd, org))
+				case "邀请":
+					ctx.SetGroupAddRequest(flag, "invite", false, other)
+					ctx.SendPrivateMessage(su, message.Text("已", cmd, org))
+				}
+			}
+		})
+	engine.OnRegex(`^(开启|关闭)自动同意(申请|邀请|主人)$`, zero.SuperUserPermission, zero.OnlyPrivate).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			c := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
+			su := zero.BotConfig.SuperUsers[0]
+			option := ctx.State["regex_matched"].([]string)[1]
+			from := ctx.State["regex_matched"].([]string)[2]
+			data := c.GetData(-su)
+			switch option {
+			case "开启":
+				switch from {
+				case "申请":
+					data |= 1
+				case "邀请":
+					data |= 0x10
+				case "主人":
+					data |= 0x20
+				}
+			case "关闭":
+				switch from {
+				case "申请":
+					data &= 0x7fffffff_fffffffe
+				case "邀请":
+					data &= 0x7fffffff_ffffffef
+				case "主人":
+					data &= 0x7fffffff_ffffffdf
+				}
+			}
+			err := c.SetData(-su, data)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			ctx.SendChain(message.Text("已设置自动同意" + from + "为" + option))
+		})
+}