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

涩图模块：搜索不到后会尝试搜索非18、模糊搜索

Browse Source
This commit is contained in:
Yutousama 2022-04-18 22:49:27 +08:00
parent 086d773ac0
commit 19a9fb0d1b
2 changed files with 70 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/com/yutou/qqbot/QQBotApplication.java
View File

@ -6,7 +6,7 @@ import org.springframework.boot.autoconfigure.SpringBootApplication;
@SpringBootApplication
public class QQBotApplication {
public static final String version="QQBot v.1.2.18";
public static final String version="QQBot v.1.2.19";
public static void main(String[] args) {
System.out.println("version = " + version);
SpringApplication.run(QQBotApplication.class, args);

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

@ -47,70 +47,83 @@ public class GetSeTu extends Model {
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);
Matcher matcher = pattern.matcher(msg);
String key = null;
if (matcher.find()) {
key = matcher.group(1);
}
String url = "https://api.lolicon.app/setu/v2?r18=0&size=regular";
if (isR18) {
url = "https://api.lolicon.app/setu/v2?r18=1&size=regular";
}
if (!StringUtils.isEmpty(key)) {
if (isR18) {
url = "https://api.lolicon.app/setu/v2?tag=" + URLEncoder.encode(key, StandardCharsets.UTF_8) + "&r18=1&size=regular";
} else {
url = "https://api.lolicon.app/setu/v2?tag=" + URLEncoder.encode(key, StandardCharsets.UTF_8) + "&r18=0&size=regular";
if (!getSeTu("tag", key, isR18, false, qq, event)) {
if (!getSeTu("tag", key, isR18, true, qq, event)) {
if (!getSeTu("dec", key, isR18, true, qq, event)) {
QQBotManager.getInstance().sendMessage(qq, "找不到喵~");
}
}
}
String ret = HttpTools.get(url);
JSONObject json = JSONObject.parseObject(ret);
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("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();
}
});
}
}
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 (r18) {
url = "https://api.lolicon.app/setu/v2?" + model + "=" + URLEncoder.encode(key, StandardCharsets.UTF_8) + "&r18=1&size=regular";
} else {
url = "https://api.lolicon.app/setu/v2?" + model + "=" + URLEncoder.encode(key, StandardCharsets.UTF_8) + "&r18=0&size=regular";
}
}
if (fuzzyR18) {
url = url.replace("&r18=0", "&r18=2").replace("&r18=1", "&r18=2");
}
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("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色图";
String msg = "来点1色图";
Pattern pattern = Pattern.compile("来点(.*?)色图");
Matcher matcher=pattern.matcher(msg);
String key=null;
if(matcher.find()){
key=matcher.group(1);
Matcher matcher = pattern.matcher(msg);
String key = null;
if (matcher.find()) {
key = matcher.group(1);
}
System.out.println("key = " + key);
}