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

package com.yutou.qqbot.models.setu;

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.StringUtils;
import net.mamoe.mirai.event.events.MessageEvent;
import net.mamoe.mirai.message.data.MessageChainBuilder;

import java.io.File;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;

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 onMessage(Long qq, MessageEvent event, boolean isGroup) {
        super.onMessage(qq, event, isGroup);
        if (msg.startsWith("来点")&&(msg.endsWith("色图")||msg.endsWith("涩图"))) {
            Log.i(event.getSource().getFromId()+" > "+msg);
            String key=msg.replace("来点","").replace("色图","").replace("涩图","");
            String url="https://api.lolicon.app/setu/v2?r18=2";
            if(!StringUtils.isEmpty(key)){
                url="https://api.lolicon.app/setu/v2?tag="+ URLEncoder.encode(key, StandardCharsets.UTF_8)+"&r18=2";
            }
            String ret = HttpTools.get(url);
            JSONObject json = JSONObject.parseObject(ret);
            System.out.println("json = " + json);
            if(json.getJSONArray("data").size()==0){
                QQBotManager.getInstance().sendMessage(qq,"找不到喵~");
                return;
            }
            JSONObject item=json.getJSONArray("data").getJSONObject(0);
            StringBuilder builder=new StringBuilder();
            builder.append(item.getString("title"));
            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("original"),
                    System.currentTimeMillis()+"_setu.jpg",
                    true
                    , new DownloadInterface() {
                        @Override
                        public void onDownload(File file) {
                            super.onDownload(file);
                            QQBotManager.getInstance().sendMessage(file, qq, builder.toString());
                        }

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

    public static void main(String[] args) {
        QQBotManager.getInstance();
        try {
            Thread.sleep(5000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        String ret = HttpTools.get("https://api.ixiaowai.cn/api/api.php?return=json");
        JSONObject json = JSONObject.parseObject(ret);
        HttpTools.download(json.getString("imgurl"),
                "setu.jpg", new DownloadInterface() {
                    @Override
                    public void onDownload(File file) {
                        super.onDownload(file);
                        QQBotManager.getInstance().sendMessage(file, 583819556L, "");
                    }
                });
    }
}