mirror of
https://github.com/LamGC/ContentGrabbingJi.git
synced 2025-04-30 06:37:36 +00:00
LamGC
1f48b3ffdc
[Change] RandomRankingArtworksSender 调整异常捕获日志输出; [Change] MiraiMessageEvent, MiraiMessageSender, SpringCQMessageEvent 适配更改;
100 lines
4.3 KiB
Java
100 lines
4.3 KiB
Java
package net.lamgc.cgj.bot;
|
||
|
||
import com.google.gson.JsonObject;
|
||
import net.lamgc.cgj.bot.message.MessageSender;
|
||
import net.lamgc.cgj.pixiv.PixivDownload;
|
||
import net.lamgc.cgj.pixiv.PixivURL;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
|
||
import java.io.IOException;
|
||
import java.util.Calendar;
|
||
import java.util.Date;
|
||
import java.util.List;
|
||
import java.util.Random;
|
||
|
||
/**
|
||
* 推荐作品发送器
|
||
*/
|
||
public class RandomRankingArtworksSender extends AutoSender {
|
||
|
||
private final Logger log;
|
||
private final int rankingStart;
|
||
private final int rankingStop;
|
||
private final PixivURL.RankingMode mode;
|
||
private final PixivURL.RankingContentType contentType;
|
||
private final PixivDownload.PageQuality quality;
|
||
|
||
/**
|
||
* 构造一个推荐作品发送器
|
||
* @param messageSender 消息发送器
|
||
* @param rankingStart 排行榜开始范围(从1开始, 名次)，如传入0或负数则为默认值，默认为1
|
||
* @param rankingStop 排名榜结束范围(包括该名次)，如传入0或负数则为默认值，默认为150
|
||
* @param mode 排行榜模式
|
||
* @param contentType 排行榜内容类型
|
||
* @param quality 图片质量, 详见{@link PixivDownload.PageQuality}
|
||
* @throws IndexOutOfBoundsException 当 rankingStart > rankingStop时抛出
|
||
*/
|
||
public RandomRankingArtworksSender(MessageSender messageSender, int rankingStart, int rankingStop, PixivURL.RankingMode mode, PixivURL.RankingContentType contentType, PixivDownload.PageQuality quality) {
|
||
super(messageSender);
|
||
this.mode = mode;
|
||
this.contentType = contentType;
|
||
log = LoggerFactory.getLogger(this.toString());
|
||
this.rankingStart = rankingStart > 0 ? rankingStart : 1;
|
||
this.rankingStop = rankingStop > 0 ? rankingStop : 150;
|
||
if(this.rankingStart > this.rankingStop) {
|
||
throw new IndexOutOfBoundsException("rankingStart=" + this.rankingStart + ", rankingStop=" + this.rankingStop);
|
||
}
|
||
this.quality = quality == null ? PixivDownload.PageQuality.REGULAR : quality;
|
||
}
|
||
|
||
@Override
|
||
public void send() {
|
||
Date queryDate = new Date();
|
||
Calendar calendar = Calendar.getInstance();
|
||
calendar.setTime(queryDate);
|
||
if(calendar.get(Calendar.HOUR_OF_DAY) < 12) {
|
||
calendar.add(Calendar.DAY_OF_YEAR, -2);
|
||
} else {
|
||
calendar.add(Calendar.DAY_OF_YEAR, -1);
|
||
}
|
||
queryDate = calendar.getTime();
|
||
|
||
int selectRanking = rankingStart + new Random().nextInt(rankingStop - rankingStart + 1);
|
||
try {
|
||
List<JsonObject> rankingList = BotCommandProcess.getRankingInfoByCache(
|
||
contentType,
|
||
mode,
|
||
queryDate,
|
||
selectRanking,
|
||
1, false);
|
||
|
||
log.debug("RankingResult.size: {}", rankingList.size());
|
||
if(rankingList.size() != 1) {
|
||
log.error("排行榜选取失败!(获取到了多个结果)");
|
||
return;
|
||
}
|
||
|
||
JsonObject rankingInfo = rankingList.get(0);
|
||
int illustId = rankingInfo.get("illust_id").getAsInt();
|
||
if(BotCommandProcess.isNoSafe(illustId, SettingProperties.getProperties(SettingProperties.GLOBAL), false)) {
|
||
log.warn("作品为r18作品, 取消本次发送.");
|
||
return;
|
||
} else if(BotCommandProcess.isReported(illustId)) {
|
||
log.warn("作品Id {} 被报告, 正在等待审核, 跳过该作品.", illustId);
|
||
return;
|
||
}
|
||
|
||
StringBuilder message = new StringBuilder();
|
||
message.append("#美图推送 - 今日排行榜 第 ").append(rankingInfo.get("rank").getAsInt()).append(" 名\n");
|
||
message.append("标题：").append(rankingInfo.get("title").getAsString()).append("(").append(illustId).append(")\n");
|
||
message.append("作者：").append(rankingInfo.get("user_name").getAsString()).append("\n");
|
||
message.append(BotCommandProcess.getImageById(0, illustId, quality, 1));
|
||
message.append("\n如有不当作品，可使用\".cgj report -id ").append(illustId).append("\"向色图姬反馈。");
|
||
getMessageSender().sendMessage(message.toString());
|
||
} catch (Exception e) {
|
||
log.error("发送随机作品时发生异常", e);
|
||
}
|
||
}
|
||
}
|