QQBot/src/main/java/com/yutou/qqbot/models/setu/GetSeTu.java

package com.yutou.qqbot.models.setu;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.interfaces.DownloadInterface;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.HttpTools;
import com.yutou.qqbot.utlis.Log;
import com.yutou.qqbot.utlis.RedisTools;
import com.yutou.qqbot.utlis.StringUtils;
import net.mamoe.mirai.event.events.MessageEvent;
import net.mamoe.mirai.message.data.At;
import net.mamoe.mirai.message.data.MessageChainBuilder;
import redis.clients.jedis.Jedis;

import java.io.File;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class GetSeTu extends Model {
    @Override
    public boolean isUserPublic() {
        return true;
    }

    @Override
    public String[] getUsePowers() {
        return new String[0];
    }

    @Override
    public String getModelName() {
        return "随机二次元图";
    }

    @Override
    public void onTime(Long qq, String time) {
        super.onTime(qq, time);
        if ("00:00:00".equals(time)) {
            del();
        } else if (RedisTools.get("Setu_time", "23:59:00").equals(time)) {
            send();
        }
    }

    private void del() {
        Jedis redis = RedisTools.getRedis();
        redis.select(RedisTools.QQBOT_USER);
        Set<String> keys = redis.keys("Setu_*");
        for (String key : keys) {
            redis.del(key);
        }
        redis.close();
    }

    private void send() {
        Jedis redis = RedisTools.getRedis();
        redis.select(RedisTools.QQBOT_USER);
        Set<String> keys = redis.keys("Setu_*");
        Log.i("进入涩图打印:" + keys.size());
        redis.close();
        for (String key : keys) {
            if ("Setu_time".equals(key)) {
                continue;
            }
            String ret = RedisTools.get(key);
            MessageChainBuilder builder = null;
            Log.i(key, ret);
            if (!StringUtils.isEmpty(ret)) {
                JSONArray array = JSONArray.parseArray(ret);
                builder = new MessageChainBuilder();
                builder.append("各位老色胚们晚上好，现在公布本群涩图数据!");
                builder.append("\n");
                builder.append("本群今日共查询:");
                builder.append(String.valueOf(array.size()));
                builder.append("次!\n");
                builder.append("接下来公布大伙的XP系统：");
                builder.append("\n");
                Map<Long, List<String>> map = new HashMap<>();
                for (Object o : array) {
                    JSONObject item = (JSONObject) o;
                    List<String> tags = map.getOrDefault(item.getLong("user"), new ArrayList<>());
                    tags.add(item.getString("key"));
                    map.put(item.getLong("user"), tags);
                }
                Long maxQQ = -1L;
                int max = -1;
                for (Long qq : map.keySet()) {
                    builder.append("------------------\n");
                    List<String> list = map.get(qq);
                    if (list.size() > max) {
                        max = list.size();
                        maxQQ = qq;
                    }
                    builder.append(new At(qq))
                            .append(":\n");
                    for (String tag : list) {
                        builder.append(tag).append("\n");
                    }
                    builder.append("------------------\n");
                }
                builder.append("那么今日最色批的就是:");
                builder.append(new At(maxQQ));
                builder.append("\n");
                builder.append("如果炸群了请找他(\n");
                builder.append("明天见~");
            }
            Log.i("打印色图统计：" + builder);
            if (builder != null) {
                QQBotManager.getInstance().sendMessage(Long.parseLong(key.replace("Setu_", "")), builder);
            }
        }
    }

    @Override
    public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
        super.onMessage(qq, event, isGroup);
        boolean isRun = false;
        boolean isR18 = false;
        if (msg.contains("来点") && msg.contains("色图")) {
            isRun = true;
        } else if (msg.contains("来点") && msg.contains("涩图")) {
            isRun = true;
            isR18 = true;
            msg = msg.replace("涩图", "色图");
        }
        if (isRun) {
            Log.i(event.getSource().getFromId() + " > " + msg);
            Pattern pattern = Pattern.compile("来点(.*?)色图");
            Matcher matcher = pattern.matcher(msg);
            String key = null;
            if (matcher.find()) {
                key = matcher.group(1);
            }
            if (!getSeTu("tag", key, isR18, false, qq, event)) {
                if (!getSeTu("tag", key, isR18, true, qq, event)) {
                    if (!getSeTu("keyword", key, isR18, true, qq, event)) {
                        QQBotManager.getInstance().sendMessage(qq, "找不到喵~ 不如随便来点吧~");
                        getSeTu("tag", null, false, false, qq, event);
                    }
                }
            }

            JSONObject data = new JSONObject();
            data.put("user", event.getSource().getFromId());
            data.put("group", qq);
            data.put("key", key);

            String redis = RedisTools.get("Setu_" + qq);
            JSONArray array = new JSONArray();
            if (!StringUtils.isEmpty(redis)) {
                array = JSONArray.parseArray(redis);
            }
            array.add(data);
            RedisTools.set("Setu_" + qq, array.toString());
        }
    }

    private boolean getSeTu(String model, String key, boolean r18, boolean fuzzyR18, Long qq, MessageEvent event) {
        String url = "https://api.lolicon.app/setu/v2?r18=0&size=regular";
        if (r18) {
            url = "https://api.lolicon.app/setu/v2?r18=1&size=regular";
        }
        if (!StringUtils.isEmpty(key)) {
            if ("tag".equals(model)) {
                String[] keys = key.split(" ");
                StringBuilder keyBuilder = new StringBuilder();
                for (String _key : keys) {
                    keyBuilder.append(URLEncoder.encode(_key, StandardCharsets.UTF_8)).append("&tag=");
                }
                key = keyBuilder.toString();
                key = key.substring(0, key.length() - 5);
            } else {
                key = URLEncoder.encode(key, StandardCharsets.UTF_8);
            }
            if (r18) {
                url = "https://api.lolicon.app/setu/v2?" + model + "=" + key + "&r18=1&size=regular";
            } else {
                url = "https://api.lolicon.app/setu/v2?" + model + "=" + key + "&r18=0&size=regular";
            }
        }
        if (fuzzyR18) {
            url = url.replace("&r18=0", "&r18=2").replace("&r18=1", "&r18=2");
        }
        System.out.println("url = " + url);
        String ret = HttpTools.get(url);
        JSONObject json = JSONObject.parseObject(ret);
        if (json.getJSONArray("data").size() == 0) {
            return false;
        }
        JSONObject item = json.getJSONArray("data").getJSONObject(0);
        StringBuilder builder = new StringBuilder();
        builder.append(item.getString("title"));
        builder.append("\n");
        builder.append("P站ID:");
        builder.append(item.getInteger("pid"));
        builder.append("\n");
        builder.append("R18:");
        if (item.getBoolean("r18")) {
            builder.append("YES!");
        } else {
            builder.append("NO~");
        }
        builder.append("\n");
        builder.append("tags:");
        for (Object tags : item.getJSONArray("tags")) {
            builder.append(tags).append("、");
        }
        HttpTools.download(item.getJSONObject("urls").getString("regular"),
                System.currentTimeMillis() + "_setu.jpg",
                true
                , new DownloadInterface() {
                    @Override
                    public void onDownload(File file) {
                        super.onDownload(file);
                        QQBotManager.getInstance().sendMessage(file, qq, event.getMessage(), builder.toString());
                    }

                    @Override
                    public void onError(Exception e) {
                        super.onError(e);
                        QQBotManager.getInstance().sendMessage(qq, "获取失败喵~");
                        e.printStackTrace();
                    }
                });
        return true;
    }

    public static void main(String[] args) {
        String msg = "来点1色图";
        Pattern pattern = Pattern.compile("来点(.*?)色图");
        Matcher matcher = pattern.matcher(msg);
        String key = null;
        if (matcher.find()) {
            key = matcher.group(1);
        }
        System.out.println("key = " + key);
    }
}