Add some features.

SwaggyMacro · Mar 24, 2022 · 8415189 · 8415189
1 parent 8bd4410
commit 8415189
Show file tree

Hide file tree

Showing 3 changed files with 73 additions and 0 deletions.
diff --git a/src/main/java/com/julym/bot/BotRunner.java b/src/main/java/com/julym/bot/BotRunner.java
@@ -309,6 +309,62 @@ public void commandManager(GroupMessageEvent event, MessageChain chain){
                 SendMessage(event, "该指令仅限机器人主人使用！");
             }
         }
+
+        if (chain.contentToString().replace("：",":").indexOf("!NSFW查看白名单列表") == 0) {
+            if (this.Master.equals("")){
+                SendMessage(event, "请先使用 \"!NSFW设置主人:QQ号\" 设置机器人主人QQ！ ");
+            }else if(this.Master.equals(Long.toString(event.getSender().getId()))){
+                event.getSubject().sendMessage(new MessageChainBuilder().
+                        append(new At(event.getSender().getId())).
+                        append(this.nsfwConfig._whiteList.getAll()).
+                        build());
+            }else{
+                SendMessage(event, "该指令仅限机器人主人使用！");
+            }
+        }
+
+        if (chain.contentToString().replace("：",":").indexOf("!NSFW查看群列表") == 0) {
+            if (this.Master.equals("")){
+                SendMessage(event, "请先使用 \"!NSFW设置主人:QQ号\" 设置机器人主人QQ！ ");
+            }else if(this.Master.equals(Long.toString(event.getSender().getId()))){
+                event.getSubject().sendMessage(new MessageChainBuilder().
+                        append(new At(event.getSender().getId())).
+                        append(this.nsfwConfig._groups.getAll()).
+                        build());
+            }else{
+                SendMessage(event, "该指令仅限机器人主人使用！");
+            }
+        }
+
+        if (chain.contentToString().replace("：",":").indexOf("!NSFW清空群列表") == 0) {
+            if (this.Master.equals("")){
+                SendMessage(event, "请先使用 \"!NSFW设置主人:QQ号\" 设置机器人主人QQ！ ");
+            }else if(this.Master.equals(Long.toString(event.getSender().getId()))){
+                this.nsfwConfig._groups.clear();
+                this.nsfwConfig.SaveConfig();
+                event.getSubject().sendMessage(new MessageChainBuilder().
+                        append(new At(event.getSender().getId())).
+                        append("清除成功!").
+                        build());
+            }else{
+                SendMessage(event, "该指令仅限机器人主人使用！");
+            }
+        }
+
+        if (chain.contentToString().replace("：",":").indexOf("!NSFW清空白名单") == 0) {
+            if (this.Master.equals("")){
+                SendMessage(event, "请先使用 \"!NSFW设置主人:QQ号\" 设置机器人主人QQ！ ");
+            }else if(this.Master.equals(Long.toString(event.getSender().getId()))){
+                this.nsfwConfig._whiteList.clear();
+                this.nsfwConfig.SaveConfig();
+                event.getSubject().sendMessage(new MessageChainBuilder().
+                        append(new At(event.getSender().getId())).
+                        append("清除成功!").
+                        build());
+            }else{
+                SendMessage(event, "该指令仅限机器人主人使用！");
+            }
+        }
     }
 
     public void receiveMessage(GroupMessageEvent event){

diff --git a/src/main/java/com/julym/config/NsfwConfig.java b/src/main/java/com/julym/config/NsfwConfig.java
@@ -102,6 +102,14 @@ public boolean isWhite(String num){
             return false;
         }
 
+        public void clear(){
+            this.lists.clear();
+        }
+
+        public String getAll(){
+            return this.lists.toString();
+        }
+
     }
 
     public static class master{
@@ -171,6 +179,14 @@ public boolean hasGroup(String num){
             return false;
         }
 
+        public void clear(){
+            this.group.clear();
+        }
+
+        public String getAll(){
+            return this.group.toString();
+        }
+
     }
 
     public static class reply{

diff --git a/src/main/java/com/julym/util/AntiModel.java b/src/main/java/com/julym/util/AntiModel.java
@@ -10,6 +10,7 @@
 import java.awt.*;
 import java.awt.image.BufferedImage;
 import java.awt.image.PixelGrabber;
+import java.io.File;
 import java.io.IOException;
 import java.util.Map;
 import java.util.Objects;