移植之前版本所有功能

2021-12-07 21:59:41 +08:00
parent 06cb246af7
commit 697bf302ac
24 changed files with 1127 additions and 258 deletions
--- a/src/main/java/com/yutou/qqbot/models/Commands/Bangumi.java
+++ b/src/main/java/com/yutou/qqbot/models/Commands/Bangumi.java
@@ -0,0 +1,126 @@
+package com.yutou.qqbot.models.Commands;
+
+import com.yutou.qqbot.QQBotManager;
+import com.yutou.qqbot.interfaces.DownloadInterface;
+import com.yutou.qqbot.models.Model;
+import com.yutou.qqbot.utlis.BangumiTools;
+import com.yutou.qqbot.utlis.HttpTools;
+import com.yutou.qqbot.utlis.Log;
+import com.yutou.qqbot.utlis.RedisTools;
+import net.mamoe.mirai.event.events.MessageEvent;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public class Bangumi extends Model {
+
+
+    @Override
+    public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
+        super.onMessage(qq, event, isGroup);
+
+        if(!isGroupPower){
+            System.err.println(qq+" 没有权限");
+            return;
+        }
+        switch (msg){
+            case QQGroupCommands.QQ_BANGUMI_TODAY -> {
+                RedisTools.remove("reportToDayBangumi");
+                QQBotManager.getInstance().sendMessage(qq, BangumiTools.reportToDayBangumi());
+            }
+            case QQGroupCommands.QQ_BANGUMI_LIST -> {
+                QQBotManager.getInstance().sendMessage(qq,"获取中...");
+                QQBotManager.getInstance().sendMessage(qq,BangumiTools.reportBangumiList());
+            }default -> {
+                if(msg.startsWith(QQGroupCommands.QQ_BANGUMI_SUB)){
+                    subBanGumi(qq,msg);
+                }
+            }
+        }
+
+    }
+    private  void subBanGumi(long qq,String msg){
+        List<String> infos = null;
+        try {
+            int id = Integer.parseInt(msg.replace(QQGroupCommands.QQ_BANGUMI_SUB, "").trim());
+            infos = BangumiTools.reportBangumiInfo(id);
+        } catch (Exception e) {
+            String key = msg.replace(QQGroupCommands.QQ_BANGUMI_SUB, "").trim();
+            infos = BangumiTools.reportSearchBangumi(key);
+        }
+        for (String info : infos) {
+            List<String> imgs = new ArrayList<>();
+            if (info.contains("<img") && info.contains(" /img>")) {
+                imgs = getImages(info);
+                for (String img : imgs) {
+                    info = info.replace("<img " + img + " /img>", "");
+                }
+            }
+            sendImagesMsg(imgs,qq, info);
+        }
+    }
+    public static List<String> getImages(String str) {
+        List<String> list = new ArrayList<>();
+        String regex = "<img(.*?)/img>";
+        Pattern pattern = Pattern.compile(regex);
+        Matcher matcher = pattern.matcher(str);
+        while (matcher.find()) {
+            list.add(matcher.group().replace("<img", "")
+                    .replace("/img>", "")
+                    .trim());
+        }
+        return list;
+    }
+    private List<File> files;
+    private int index = 0;
+
+    private void sendImagesMsg(List<String> imgs,Long qq, String text) {
+        files = new ArrayList<>();
+        index = 0;
+        if (imgs.size() == 0) {
+            QQBotManager.getInstance().sendMessage(qq,text);
+            return;
+        }
+        for (String img : imgs) {
+            HttpTools.download(img,null, new DownloadInterface() {
+                @Override
+                public void onDownload(File file) {
+                    super.onDownload(file);
+                    files.add(file);
+                    send(imgs.size(),qq, text);
+                }
+
+                @Override
+                public void onError(Exception e) {
+                    super.onError(e);
+                    index++;
+                    send(imgs.size(),qq, text);
+                }
+            });
+        }
+    }
+
+    private void send(int size,Long qq, String text) {
+        if ((files.size() + index) == size) {
+            String str = QQBotManager.getInstance().sendMessage(files,qq, text);
+            Log.i("str = " + str);
+        }
+    }
+
+    @Override
+    public boolean isUserPublic() {
+        return true;
+    }
+
+    @Override
+    public String[] getUsePowers() {
+        return new String[]{
+                QQGroupCommands.QQ_BANGUMI_SUB,
+                QQGroupCommands.QQ_BANGUMI_LIST,
+                QQGroupCommands.QQ_BANGUMI_TODAY
+        };
+    }
+}