新插件event, 好友申请群聊邀请事件处理 (#403)

README.md

@@ -556,6 +556,18 @@ print("run[CQ:image,file="+j["img"]+"]")
 
   - [x] xxx疫情
 
+</details>
+<details>
+  <summary>好友申请及群聊邀请事件处理</summary>
+
+  `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/event"`
+
+  - [x] [开启|关闭]自动同意[申请|邀请|主人]
+
+  - [x] [同意|拒绝][申请|邀请][flag]
+
+  - flag跟随事件一起发送, 默认同意主人的事件
+
 </details>
 <details>
   <summary>渲染任意文字到图片</summary>

main.go

@@ -77,6 +77,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,155 @@
+// Package event 好友申请以及群聊邀请事件处理
+package event
+
+import (
+	"strconv"
+	"time"
+
+	ctrl "github.com/FloatTech/zbpctrl"
+	"github.com/FloatTech/zbputils/control"
+	base14 "github.com/fumiama/go-base16384"
+	"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" +
+			" - [开启|关闭]自动同意[申请|邀请|主人]\n" +
+			" - [同意|拒绝][申请|邀请][flag]\n" +
+			"flag跟随事件一起发送, 默认同意主人的事件",
+	})
+	engine.On("request/group/invite").SetBlock(true).
+		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
+				es := base14.EncodeString(flag)
+				comment := ctx.Event.Comment
+				userid := ctx.Event.UserID
+				username := ctx.CardOrNickName(userid)
+				data := c.GetData(-su)
+				groupid := ctx.Event.GroupID
+				groupname := ctx.GetGroupInfo(groupid, true).Name
+				logrus.Info("[event]收到来自[", username, "](", userid, ")的群聊邀请，群:[", groupname, "](", groupid, ")")
+				if data&2 == 2 || data&1 != 1 && 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:"+es)})
+					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, es)})
+			}
+		})
+	engine.On("request/friend").SetBlock(true).
+		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
+				es := base14.EncodeString(flag)
+				comment := ctx.Event.Comment
+				userid := ctx.Event.UserID
+				username := ctx.CardOrNickName(userid)
+				data := c.GetData(-su)
+				logrus.Info("[event]收到来自[", username, "](", userid, ")的好友申请")
+				if data&4 == 4 || data&1 != 1 && 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:"+es)})
+					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, es)})
+			}
+		})
+	engine.OnRegex(`^(同意|拒绝)(申请|邀请)\s*([一-踀]+[㴁-㴆]?)\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]
+			es := base14.DecodeString(flag)
+			switch cmd {
+			case "同意":
+				switch org {
+				case "申请":
+					ctx.SetFriendAddRequest(es, true, other)
+					ctx.SendPrivateMessage(su, message.Text("已", cmd, org))
+				case "邀请":
+					ctx.SetGroupAddRequest(es, "invite", true, "")
+					ctx.SendPrivateMessage(su, message.Text("已", cmd, org))
+				}
+			case "拒绝":
+				switch org {
+				case "申请":
+					ctx.SetFriendAddRequest(es, false, "")
+					ctx.SendPrivateMessage(su, message.Text("已", cmd, org))
+				case "邀请":
+					ctx.SetGroupAddRequest(es, "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 |= 4
+				case "邀请":
+					data |= 2
+				case "主人":
+					data &= 7
+				}
+			case "关闭":
+				switch from {
+				case "申请":
+					data &= 3
+				case "邀请":
+					data &= 5
+				case "主人":
+					data |= 1
+				}
+			}
+			err := c.SetData(-su, data)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			ctx.SendChain(message.Text("已设置自动同意" + from + "为" + option))
+		})
+}

plugin/hyaku/main.go

@@ -21,7 +21,7 @@ import (
 
 const bed = "https://gitcode.net/u011570312/OguraHyakuninIsshu/-/raw/master/"
 
-// nolint: asciicheck
+//nolint: asciicheck
 type line struct {
 	番号, 歌人, 上の句, 下の句, 上の句ひらがな, 下の句ひらがな string
 }

plugin/qqwife/command.go

@@ -14,7 +14,7 @@ import (
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
 
-	//反并发
+	// 反并发
 	"github.com/wdvxdr1123/ZeroBot/extension/single"
 	// 数据库
 	sql "github.com/FloatTech/sqlite"
@@ -28,8 +28,8 @@ import (
 	"github.com/FloatTech/zbputils/img/text"
 )
 
-//nolint: asciicheck
 // nolint: asciicheck
+//nolint: asciicheck
 var (
 	民政局 = &婚姻登记{
 		db: &sql.Sqlite{},