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

完成基本功能转移

Browse Source
This commit is contained in:
Yutou
2024-05-04 17:26:27 +08:00
parent 3b52742ac1
commit 5a7382d02c
89 changed files with 2642 additions and 604 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,19 +3,19 @@ package com.yutou.qqbot.models.setu;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.yutou.napcat.QQDatabase;
import com.yutou.napcat.event.MessageEvent;
import com.yutou.napcat.handle.*;
import com.yutou.napcat.model.SourceFrom;
import com.yutou.qqbot.Annotations.UseModel;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.data.MessageChainBuilder;
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.MessageReceipt;
import net.mamoe.mirai.message.data.At;
import net.mamoe.mirai.message.data.MessageChainBuilder;
import net.mamoe.mirai.message.data.QuoteReply;
import redis.clients.jedis.Jedis;
import java.io.File;
@@ -111,7 +111,7 @@ public class GetSeTu extends Model {
builder.append("明天见~");
}
if (builder != null) {
QQBotManager.getInstance().sendMessage(group, builder);
QQBotManager.getInstance().sendMessage(QQDatabase.checkFriend(group), group, builder.toString());
}
}
@@ -218,41 +218,27 @@ public class GetSeTu extends Model {
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);
builder.append("\n看不到图?点这里:http://setu.cnmglz.com/setu/").append(file.getName());
QQBotManager.getInstance().sendMessage(file, qq, event.getMessage(), "");
MessageChainBuilder chain = new MessageChainBuilder();
chain.append(new QuoteReply(event.getMessage()));
chain.append(builder.toString());
MessageReceipt<?> message = QQBotManager.getInstance().sendMessage(qq, chain);
Log.i(getModelName(), message);
}
@Override
public void onError(Exception e) {
super.onError(e);
getSeTu(model, tmpKey, r18, fuzzyR18, qq, event, reset - 1);
e.printStackTrace();
}
});
builder.append("\n看不到图?点这里:").append(item.getJSONObject("urls").getString("regular"));
QQBotManager.getInstance().sendMessage(false, qq,
new Image(item.getJSONObject("urls").getString("regular")),
new Reply(event.getMessageId())
);
QQBotManager.getInstance().sendMessage(false, qq,
new Reply(event.getMessageId()),
new Text(builder.toString())
);
return true;
}
public static void main(String[] args) {
String msg = "来点1色图";
String msg = "来点色图";
Pattern pattern = Pattern.compile("来点(.*?)色图");
Matcher matcher = pattern.matcher(msg);
String key = null;
if (matcher.find()) {
key = matcher.group(1);
}
System.out.println("key = " + key);
MessageEvent event = new MessageEvent();
SourceFrom sourceFrom = new SourceFrom();
sourceFrom.setUserId(583819556L);
event.setSource(sourceFrom);
event.setRawMessage(msg);
new GetSeTu().onMessage(583819556L, event, false);
}
}

184
src/main/java/com/yutou/qqbot/models/setu/QQSetu.java
View File

@@ -4,17 +4,25 @@ package com.yutou.qqbot.models.setu;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.yutou.napcat.QQDatabase;
import com.yutou.napcat.enums.MessageEnum;
import com.yutou.napcat.event.GroupMessageEvent;
import com.yutou.napcat.handle.At;
import com.yutou.napcat.handle.BaseHandle;
import com.yutou.napcat.handle.Image;
import com.yutou.napcat.handle.Text;
import com.yutou.napcat.http.NapCatApi;
import com.yutou.napcat.model.MessageBean;
import com.yutou.okhttp.HttpBody;
import com.yutou.qqbot.Annotations.UseModel;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.data.MessageChainBuilder;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.AppTools;
import com.yutou.qqbot.utlis.Log;
import com.yutou.qqbot.utlis.RedisTools;
import net.mamoe.mirai.event.events.GroupMessageEvent;
import net.mamoe.mirai.event.events.MessageEvent;
import net.mamoe.mirai.message.data.At;
import net.mamoe.mirai.message.data.Image;
import net.mamoe.mirai.message.data.MessageChainBuilder;
import com.yutou.napcat.event.MessageEvent;
import com.yutou.qqbot.utlis.StringUtils;
import java.io.IOException;
import java.net.HttpURLConnection;
@@ -25,88 +33,91 @@ import java.util.*;
@UseModel
public class QQSetu extends Model {
private int db_print =1;//统计结果
private int db_user =3;//当次数据
private int db_print = 1;//统计结果
private int db_user = 3;//当次数据
private long group;
private Timer timer;
private static final Map<String, Float> setuScore = new HashMap<>();
public void printTodaySetu() {
String redisKey= AppTools.getToDayTime() + "_setu";
public void printTodaySetu() throws Exception {
String redisKey = AppTools.getToDayTime() + "_setu";
Log.i("今日涩图 redisKey = " + redisKey);
String js = RedisTools.get(redisKey, db_print);
if (js != null) {
JSONObject json = JSON.parseObject(js);
if(json.containsKey("isPrint")&&json.getBooleanValue("isPrint")){
if (json.containsKey("isPrint") && json.getBooleanValue("isPrint")) {
return;
}
Map<String,Float> groupAverage=new HashMap<>();
Map<String,String> groupImage=new HashMap<>();
JSONObject setu=null;
Map<String, Float> groupAverage = new HashMap<>();
Map<String, String> groupImage = new HashMap<>();
JSONObject setu = null;
for (String id : json.keySet()) {
String group=json.getJSONObject(id).getJSONObject("info").getLong("group")+"";
if(groupAverage.containsKey(group)){
if(groupAverage.get(group)<=json.getJSONObject(id).getFloat("average")){
groupAverage.put(group,json.getJSONObject(id).getFloat("average"));
groupImage.put(group,id);
String group = json.getJSONObject(id).getJSONObject("info").getLong("group") + "";
if (groupAverage.containsKey(group)) {
if (groupAverage.get(group) <= json.getJSONObject(id).getFloat("average")) {
groupAverage.put(group, json.getJSONObject(id).getFloat("average"));
groupImage.put(group, id);
}
}else{
groupAverage.put(group,json.getJSONObject(id).getFloat("average"));
groupImage.put(group,id);
} else {
groupAverage.put(group, json.getJSONObject(id).getFloat("average"));
groupImage.put(group, id);
}
}
for (String id : groupImage.keySet()) {
setu=json.getJSONObject(groupImage.get(id));
if(setu!=null){
json.put("isPrint",true);
RedisTools.set(db_print,redisKey,json.toJSONString());
JSONObject info=setu.getJSONObject("info");
JSONObject score=setu.getJSONObject("score");
MessageChainBuilder builder = new MessageChainBuilder();
builder.append(Image.fromId(info.getString("id")));
builder.append("本日最佳涩图由").append(new At(info.getLong("sourQQ"))).append("提供\n");
builder.append("获得分数 ").append(String.valueOf(setu.getFloat("average"))).append("\n");
builder.append("共有 ").append(String.valueOf(score.getIntValue("userNumber"))).append(" 人参与投票");
QQBotManager.getInstance().sendMessage(info.getLong("group"),builder);
Log.i("今日涩图:"+builder.toString());
setu = json.getJSONObject(groupImage.get(id));
if (setu != null) {
json.put("isPrint", true);
RedisTools.set(db_print, redisKey, json.toJSONString());
JSONObject info = setu.getJSONObject("info");
JSONObject score = setu.getJSONObject("score");
HttpBody<MessageBean> body = NapCatApi.getMessageApi().getMessage(info.getLong("id")).execute().body();
List<BaseHandle<?>> sendList = new ArrayList<>();
if (body == null) {
sendList.add(new Text("[图片获取失败]"));
return;
}
MessageEvent handle = MessageEvent.parseHandleHttp(body.getSrc());
Image image = handle.findType(Image.class);
if (image == null) {
sendList.add(new Text("[图片获取失败]"));
} else {
sendList.add(image);
}
//builder.append(Image.fromId(info.getString("id")));
sendList.add(new Text(
"本日最佳涩图由", false
));
sendList.add(new At(handle.getSource().getUserId()));
sendList.add(new Text("提供", false));
sendList.add(new Text("获得分数 ", false));
sendList.add(new Text(String.valueOf(setu.getFloat("average")), false));
sendList.add(new Text("共有 ", false));
sendList.add(new Text(String.valueOf(score.getIntValue("userNumber")), false));
sendList.add(new Text(" 人参与投票", false));
QQBotManager.getInstance().sendMessage(handle.isUser(), handle.getGroupId(), sendList);
}
}
}else {
} else {
Log.i("今日没有涩图");
}
}
private boolean isSetu(Image image) {
String url = Image.queryUrl(image);
try {
HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection();
int length = connection.getContentLength();
connection.disconnect();
if (length > 50000) {
return true;
}
} catch (IOException e) {
e.printStackTrace();
return false;
}
return false;
private boolean isSetu(Long length) {
return length > 50000;
}
private void setuBuilder(Image image, GroupMessageEvent event) {
if (!isSetu(image)) {
private void setuBuilder(Image image, MessageEvent event) {
if (StringUtils.isEmpty(image.getData().getFileSize()) || !isSetu(Long.parseLong(image.getData().getFileSize()))) {
return;
}
if (RedisTools.get(event.getGroup().getId()+"setu", db_user) != null) {
printSetu(event.getGroup().getId());
if (RedisTools.get(event.getGroupId() + "setu", db_user) != null) {
printSetu(event.getGroupId());
}
setuScore.clear();
JSONObject json = new JSONObject();
json.put("id", image.getImageId());
json.put("sourName", event.getSenderName());
json.put("sourQQ", event.getSender().getId());
json.put("group", event.getGroup().getId());
RedisTools.set(event.getGroup().getId()+"setu", json.toJSONString(),6*60, db_user);
json.put("id", event.getMessageId());
RedisTools.set(event.getGroupId() + "setu", json.toJSONString(), 6 * 60, db_user);
if (timer != null) {
timer.cancel();
timer = null;
@@ -119,7 +130,7 @@ public class QQSetu extends Model {
timer.schedule(new TimerTask() {
@Override
public void run() {
if(!setuScore.isEmpty()){
if (!setuScore.isEmpty()) {
printSetu(group);
}
timer.cancel();
@@ -129,7 +140,7 @@ public class QQSetu extends Model {
}
private void printSetu(long group) {
JSONObject jt = JSON.parseObject(RedisTools.get(group+"setu", db_user));
JSONObject jt = JSON.parseObject(RedisTools.get(group + "setu", db_user));
String id = jt.getString("id");
float average = 0;
float max = 0;
@@ -137,7 +148,7 @@ public class QQSetu extends Model {
float length = 0;
String maxName = "";
String minName = "";
if(setuScore.size()<=1){
if (setuScore.size() <= 1) {
return;
}
for (String name : setuScore.keySet()) {
@@ -161,8 +172,8 @@ public class QQSetu extends Model {
score.put("userNumber", setuScore.size());
average = average / setuScore.size();
String builder = "涩图评分:" + average +"\n "+
"其中最高分由:" + maxName + " 给与：" + max +"\n "+
String builder = "涩图评分:" + average + "\n " +
"其中最高分由:" + maxName + " 给与：" + max + "\n " +
"其中最低分由:" + minName + " 给与:" + min;
QQBotManager.getInstance().sendMessage(group, builder);
String st = RedisTools.get(AppTools.getToDayTime() + "_setu", db_print);
@@ -186,33 +197,34 @@ public class QQSetu extends Model {
json.put(id, item);
RedisTools.set(db_print, AppTools.getToDayTime() + "_setu", json.toJSONString());
}
RedisTools.remove(group+"setu",db_user);
RedisTools.remove(group + "setu", db_user);
setuScore.clear();
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup){
if(!isGroup){
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
if (!isGroup) {
return;
}
group=qq;
String msg=event.getMessage().contentToString();
if ("[图片]".equals(msg.trim())) {
Image image = (Image) event.getMessage().stream().filter(Image.class::isInstance).findFirst().orElse(null);
if (image != null) {
setuBuilder(image, (GroupMessageEvent) event);
return;
}
Image image = event.hasType(MessageEnum.IMAGE) ? event.findType(Image.class) : null;
if (image != null) {
setuBuilder(image, event);
return;
}
Text text = event.hasType(MessageEnum.TEXT) ? event.findType(Text.class) : null;
if (text == null) {
return;
}
try {
if(msg.trim().length()>3){
if (text.toString().trim().length() > 3) {
return;
}
float i = Float.parseFloat(msg.trim());
float i = Float.parseFloat(text.toString().trim());
if (i > 0 && i <= 10) {
String name = event.getSenderName();
String sender=event.getSender().getId()+"";
String name = event.getSource().getNickname();
String sender = event.getSource().getUserId() + "";
if (!setuScore.containsKey(name)) {
setuScore.put(name+"|"+sender, i);
setuScore.put(name + "|" + sender, i);
}
}
} catch (Exception ignored) {
@@ -220,10 +232,14 @@ public class QQSetu extends Model {
}
@Override
public void onTime(Long qq,String time) {
super.onTime(qq,time);
if("23:59:00".equals(time)){
printTodaySetu();
public void onTime(Long qq, String time) {
super.onTime(qq, time);
if ("23:59:00".equals(time)) {
try {
printTodaySetu();
} catch (Exception e) {
e.printStackTrace();
}
}
}