disallowed group

README.md

@@ -26,6 +26,18 @@ imports (
 // ...
 ```
 
+在全局配置文件中写入配置
+
+```yaml
+aimerneige:
+  fortune:
+    blacklist: # 黑名单用户
+      - 1781924496
+      - 2802340025
+    disallowed: # 关闭功能的群
+      - 546362685
+```
+
 ## LICENSE
 
 <a href="https://www.gnu.org/licenses/agpl-3.0.en.html">

fortune.go

@@ -17,6 +17,7 @@ import (
 var instance *fortune
 var logger = utils.GetModuleLogger("com.aimerneige.fortune")
 var blacklistUser []int64
+var disallowedList []int64
 
 type fortune struct {
 }
@@ -41,6 +42,10 @@ func (f *fortune) Init() {
 	for _, user := range blacklistUserSlice {
 		blacklistUser = append(blacklistUser, int64(user))
 	}
+	disallowedSlice := config.GlobalConfig.GetIntSlice("aimerneige.fortune.disallowed")
+	for _, groupCode := range disallowedList {
+		disallowedSlice = append(disallowedSlice, int(groupCode))
+	}
 }
 
 // PostInit 第二次初始化
@@ -52,6 +57,9 @@ func (f *fortune) PostInit() {
 // Serve 注册服务函数部分
 func (f *fortune) Serve(b *bot.Bot) {
 	b.OnGroupMessage(func(c *client.QQClient, msg *message.GroupMessage) {
+		if isDisallowed(msg.GroupCode) {
+			return
+		}
 		reply := getFortuneReply(msg.ToString(), msg.Sender.Uin)
 		c.SendGroupMessage(msg.GroupCode, reply)
 	})
@@ -165,3 +173,12 @@ func inBlacklist(userID int64) bool {
 	}
 	return false
 }
+
+func isDisallowed(groupCode int64) bool {
+	for _, v := range disallowedList {
+		if groupCode == v {
+			return true
+		}
+	}
+	return false
+}