JavaWeb/QQBot
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

涩图模块：区分色图和涩图关键词

Browse Source
This commit is contained in:
Yutousama 2022-04-17 15:25:51 +08:00
parent 4e56d31e43
commit 10fd1d6863
1 changed files with 29 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/main/java/com/yutou/qqbot/models/setu/GetSeTu.java
View File

@ -33,28 +33,42 @@ public class GetSeTu extends Model {
@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";
boolean isRun = false;
boolean isR18 = false;
if (msg.contains("来点") && msg.contains("色图")) {
isRun = true;
} else if (msg.contains("来点") && msg.contains("涩图")) {
isRun = true;
isR18 = true;
}
if (isRun) {
Log.i(event.getSource().getFromId() + " > " + msg);
String key = msg.replace("来点", "").replace("色图", "").replace("涩图", "");
String url = "https://api.lolicon.app/setu/v2?r18=0";
if (isR18) {
url = "https://api.lolicon.app/setu/v2?r18=1";
}
if (!StringUtils.isEmpty(key)) {
if (isR18) {
url = "https://api.lolicon.app/setu/v2?tag=" + URLEncoder.encode(key, StandardCharsets.UTF_8) + "&r18=1";
} else {
url = "https://api.lolicon.app/setu/v2?tag=" + URLEncoder.encode(key, StandardCharsets.UTF_8) + "&r18=0";
}
}
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,"找不到喵~");
if (json.getJSONArray("data").size() == 0) {
QQBotManager.getInstance().sendMessage(qq, "找不到喵~");
return;
}
JSONObject item=json.getJSONArray("data").getJSONObject(0);
StringBuilder builder=new StringBuilder();
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")){
if (item.getBoolean("r18")) {
builder.append("YES!");
}else{
} else {
builder.append("NO~");
}
builder.append("\n");
@ -63,7 +77,7 @@ public class GetSeTu extends Model {
builder.append(tags).append("、");
}
HttpTools.download(item.getJSONObject("urls").getString("original"),
System.currentTimeMillis()+"_setu.jpg",
System.currentTimeMillis() + "_setu.jpg",
true
, new DownloadInterface() {
@Override
@ -75,7 +89,7 @@ public class GetSeTu extends Model {
@Override
public void onError(Exception e) {
super.onError(e);
QQBotManager.getInstance().sendMessage(qq,"获取失败喵~");
QQBotManager.getInstance().sendMessage(qq, "获取失败喵~");
}
});
}