436 lines
18 KiB
Java
436 lines
18 KiB
Java
package com.yutou.bilibili.QQBot;
|
||
|
||
import com.alibaba.fastjson.JSONObject;
|
||
import com.yutou.bilibili.BiliBili.Controllers.RealTimeDataController;
|
||
import com.yutou.bilibili.BiliBili.Live;
|
||
import com.yutou.bilibili.BiliBili.LiveUtils;
|
||
import com.yutou.bilibili.BiliBili.Tools.SaveLive;
|
||
import com.yutou.bilibili.BilibiliApplication;
|
||
import com.yutou.bilibili.Tools.*;
|
||
import com.yutou.bilibili.interfaces.DownloadInterface;
|
||
import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo;
|
||
import net.mamoe.mirai.Bot;
|
||
import net.mamoe.mirai.BotFactory;
|
||
import net.mamoe.mirai.event.GlobalEventChannel;
|
||
import net.mamoe.mirai.event.events.GroupMessageEvent;
|
||
import net.mamoe.mirai.message.data.Image;
|
||
import net.mamoe.mirai.message.data.MessageChainBuilder;
|
||
import net.mamoe.mirai.utils.BotConfiguration;
|
||
import net.mamoe.mirai.utils.ExternalResource;
|
||
import org.springframework.beans.BeansException;
|
||
import org.springframework.context.ApplicationContext;
|
||
import org.springframework.context.ApplicationContextAware;
|
||
|
||
import javax.annotation.Resource;
|
||
import java.io.File;
|
||
import java.lang.reflect.Field;
|
||
import java.text.SimpleDateFormat;
|
||
import java.util.ArrayList;
|
||
import java.util.Date;
|
||
import java.util.List;
|
||
import java.util.Objects;
|
||
import java.util.function.Consumer;
|
||
import java.util.regex.Matcher;
|
||
import java.util.regex.Pattern;
|
||
|
||
public class QQBotManager implements ApplicationContextAware {
|
||
private static ApplicationContext applicationContext;
|
||
|
||
@Override
|
||
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
|
||
if (QQBotManager.applicationContext == null)
|
||
QQBotManager.applicationContext = applicationContext;
|
||
}
|
||
|
||
private <T> T getBean(Class<T> tClass) {
|
||
return applicationContext.getBean(tClass);
|
||
}
|
||
|
||
private static class QQCommands {
|
||
private final static String QQ_HELP = "!help";
|
||
private final static String QQ_SYSTEM_LOG = "!log";
|
||
private final static String QQ_SYSTEM_STOP = "!stopliveservice";
|
||
private final static String QQ_GET_IP = "!ip";
|
||
private final static String QQ_GET_VERSION = "!version";
|
||
private final static String QQ_LIVE_LIST = "!列表";
|
||
private final static String QQ_LIVE_SAVE = "!录播列表";
|
||
private final static String QQ_LIVE_TO_DAY_DATE = "!今日数据";
|
||
private final static String QQ_LIVE_DATE = "!数据";
|
||
private final static String QQ_LIVE_USE_SAVE = "!启动录播";
|
||
}
|
||
|
||
private static QQBotManager botManager = null;
|
||
private Bot bot;
|
||
private static long qqGroup;
|
||
private boolean isLogin = false;
|
||
private static boolean isInit = false;
|
||
|
||
@Resource
|
||
RealTimeDataController realTimeDataController;
|
||
|
||
|
||
private QQBotManager() {
|
||
qqGroup = Long.parseLong((String) ConfigTools.load(ConfigTools.CONFIG, "qq_group"));
|
||
}
|
||
public void stop(){
|
||
if(bot!=null){
|
||
bot.close();
|
||
bot=null;
|
||
}
|
||
}
|
||
public void init() {
|
||
if (!((boolean) ConfigTools.load(ConfigTools.CONFIG, "qq_bot"))) {
|
||
return;
|
||
}
|
||
new Thread(new Runnable() {
|
||
@Override
|
||
public void run() {
|
||
long qq = Long.parseLong((String) ConfigTools.load(ConfigTools.CONFIG, "qq_number"));
|
||
String password = (String) ConfigTools.load(ConfigTools.CONFIG, "qq_password");
|
||
bot = BotFactory.INSTANCE.newBot(qq, password, new BotConfiguration() {
|
||
{
|
||
setProtocol(MiraiProtocol.ANDROID_PAD);
|
||
fileBasedDeviceInfo("qq_bot_devices_info.json");
|
||
if (!((boolean) ConfigTools.load(ConfigTools.CONFIG, "qq_debug"))) {
|
||
noBotLog();
|
||
noNetworkLog();
|
||
}
|
||
}
|
||
});
|
||
//Events.registerEvents(bot, new MessageListener());
|
||
GlobalEventChannel.INSTANCE.subscribeAlways(GroupMessageEvent.class, new MessageListener());
|
||
Log.i("准备登陆");
|
||
bot.login();
|
||
Log.i("登陆成功");
|
||
isLogin = true;
|
||
isInit = true;
|
||
new Thread(new Runnable() {
|
||
@Override
|
||
public void run() {
|
||
try {
|
||
Thread.sleep(1000);
|
||
} catch (InterruptedException e) {
|
||
com.yutou.bilibili.Tools.Log.e(e);
|
||
}
|
||
String str = sendMessage("姬妻酱上线拉~");
|
||
Log.i(str);
|
||
|
||
}
|
||
}).start();
|
||
bot.join();
|
||
|
||
}
|
||
}).start();
|
||
|
||
}
|
||
|
||
public static QQBotManager getInstance() {
|
||
if (botManager == null && !isInit) {
|
||
botManager = new QQBotManager();
|
||
}
|
||
return botManager;
|
||
}
|
||
|
||
public boolean isLogin() {
|
||
return isLogin;
|
||
}
|
||
|
||
private Image getImage(File file) {
|
||
if (bot != null) {
|
||
return Objects.requireNonNull(bot.getGroup(qqGroup)).uploadImage(ExternalResource.create(file));
|
||
}
|
||
return null;
|
||
}
|
||
|
||
private String getNotLoginQQ() {
|
||
return "没有登录QQ";
|
||
}
|
||
|
||
|
||
public String sendMessage(String text) {
|
||
if (bot != null) {
|
||
try {
|
||
return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(text).toString();
|
||
} catch (Exception e) {
|
||
com.yutou.bilibili.Tools.Log.e(e);
|
||
}
|
||
}
|
||
return getNotLoginQQ();
|
||
}
|
||
|
||
public String sendMessage(Long group, String text) {
|
||
if (bot != null) {
|
||
try {
|
||
return Objects.requireNonNull(bot.getGroup(group)).sendMessage(text).toString();
|
||
} catch (Exception e) {
|
||
com.yutou.bilibili.Tools.Log.e(e);
|
||
}
|
||
}
|
||
return getNotLoginQQ();
|
||
}
|
||
|
||
public void sendMessage(Long group, MessageChainBuilder builder) {
|
||
if (bot != null) {
|
||
Objects.requireNonNull(bot.getGroup(group)).sendMessage(builder.asMessageChain());
|
||
}
|
||
}
|
||
|
||
|
||
public String sendMessage(File imageFile, String text) {
|
||
try {
|
||
if (bot != null) {
|
||
Image image = getImage(imageFile);
|
||
MessageChainBuilder builder = new MessageChainBuilder();
|
||
if (image != null) {
|
||
builder.append(image);
|
||
}
|
||
builder.append(text);
|
||
return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(builder.asMessageChain()).toString();
|
||
}
|
||
} catch (Exception e) {
|
||
com.yutou.bilibili.Tools.Log.e(e);
|
||
}
|
||
|
||
return getNotLoginQQ();
|
||
}
|
||
|
||
public String sendMessage(List<File> imgs, String text) {
|
||
if (bot != null) {
|
||
MessageChainBuilder builder = new MessageChainBuilder();
|
||
for (File img : imgs) {
|
||
builder.append(Objects.requireNonNull(getImage(img)));
|
||
}
|
||
builder.append(text);
|
||
return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(builder.asMessageChain()).toString();
|
||
}
|
||
return getNotLoginQQ();
|
||
}
|
||
|
||
private static List<String> getImages(String str) {
|
||
List<String> list = new ArrayList<>();
|
||
String regex = "<img(.*?)/img>";
|
||
Pattern pattern = Pattern.compile(regex);
|
||
Matcher matcher = pattern.matcher(str);
|
||
while (matcher.find()) {
|
||
list.add(matcher.group().replace("<img", "")
|
||
.replace("/img>", "")
|
||
.trim());
|
||
}
|
||
return list;
|
||
}
|
||
|
||
public static void main(String[] args) {
|
||
getInstance();
|
||
}
|
||
|
||
private static class MessageListener implements Consumer<GroupMessageEvent> {
|
||
|
||
|
||
@Override
|
||
public void accept(GroupMessageEvent event) {
|
||
String msg = event.getMessage().contentToString();
|
||
if (qqGroup == event.getGroup().getId()) {
|
||
myGroup(msg);
|
||
}
|
||
}
|
||
|
||
private void myGroup(String msg) {
|
||
msg = msg.replace("！", "!").toLowerCase();
|
||
msg=msg.trim();
|
||
StringBuilder builder = new StringBuilder();
|
||
JSONObject json;
|
||
String[] cmd = new String[0];
|
||
int _roomId = 0;
|
||
switch (msg) {
|
||
case QQCommands.QQ_GET_IP:
|
||
json = JSONObject.parseObject(HttpTools.get("https://api.asilu.com/ip/"));
|
||
String ip = json.getString("ip");
|
||
getInstance().sendMessage("服务器IP:" + ip);
|
||
break;
|
||
case QQCommands.QQ_GET_VERSION:
|
||
sendVersion();
|
||
break;
|
||
case QQCommands.QQ_SYSTEM_STOP:
|
||
getInstance().sendMessage("正在停止服务");
|
||
System.out.println("结束进程");
|
||
System.exit(0);
|
||
break;
|
||
case QQCommands.QQ_LIVE_LIST:
|
||
builder.append("当前正在记录数据的直播间:");
|
||
builder.append("\n");
|
||
for (Live live : Live.lives) {
|
||
JSONObject liveJson = LiveUtils.LiveInfoManager.getInstance().getInfo(live.getInfo().getRoomid());
|
||
if (LiveUtils.isLivePlayer(live.getInfo().getRoomid())) {
|
||
builder.append("【直播中】");
|
||
}
|
||
builder
|
||
.append("名字:").append(live.geData().getName()).append(" ")
|
||
.append(" roomId:").append(live.geData().getRoomid()).append(" ");
|
||
if (liveJson != null) {
|
||
builder.append("标题:").append(liveJson.getJSONObject("data").getJSONObject("room_info").getString("title")).append("\n");
|
||
} else {
|
||
builder.append("\n");
|
||
}
|
||
}
|
||
getInstance().sendMessage(builder.toString());
|
||
break;
|
||
case QQCommands.QQ_LIVE_SAVE:
|
||
builder.append("当前正在录制的直播间:");
|
||
builder.append("\n");
|
||
for (String roomId : SaveLive.getInstance().getLiveList()) {
|
||
BilibiliUpInfo data = new BilibiliUpInfo();
|
||
data.setRoomid(Integer.parseInt(roomId));
|
||
Live live = LiveUtils.liveContains(data);
|
||
if (live != null) {
|
||
File file = SaveLive.getInstance().getLiveFile(Integer.parseInt(roomId));
|
||
builder
|
||
.append("名字:").append(live.geData().getName()).append(" ")
|
||
.append("文件大小(字节):").append(file.length()).append(" ")
|
||
.append("roomId:").append(live.geData().getRoomid()).append("\n");
|
||
}
|
||
|
||
}
|
||
getInstance().sendMessage(builder.toString());
|
||
break;
|
||
case QQCommands.QQ_HELP:
|
||
for (Field field : QQCommands.class.getDeclaredFields()) {
|
||
try {
|
||
field.setAccessible(true);
|
||
builder.append(field.get(null)).append("\n");
|
||
} catch (IllegalAccessException e) {
|
||
com.yutou.bilibili.Tools.Log.e(e);
|
||
}
|
||
}
|
||
getInstance().sendMessage(builder.toString());
|
||
break;
|
||
default:
|
||
if (msg.startsWith(QQCommands.QQ_LIVE_TO_DAY_DATE)) {
|
||
try {
|
||
cmd = msg.split(" ");
|
||
_roomId = Integer.parseInt(cmd[1]);
|
||
QQBotManager.getInstance().sendMessage("请稍等，正在查询...");
|
||
sendGiftData(_roomId, null, null);
|
||
} catch (Exception e) {
|
||
getInstance().sendMessage("参数错误。\n使用方式: " + QQCommands.QQ_LIVE_TO_DAY_DATE + "+空格+roomId");
|
||
}
|
||
} else if (msg.startsWith(QQCommands.QQ_LIVE_DATE)) {
|
||
try {
|
||
cmd = msg.split(" ");
|
||
_roomId = Integer.parseInt(cmd[1]);
|
||
Date startTime = new SimpleDateFormat("yyyy-MM-dd_HH:mm:ss").parse(cmd[2]);
|
||
Date endTime = new SimpleDateFormat("yyyy-MM-dd_HH:mm:ss").parse(cmd[3]);
|
||
QQBotManager.getInstance().sendMessage("请稍等，正在查询...");
|
||
sendGiftData(_roomId, startTime, endTime);
|
||
} catch (Exception e) {
|
||
getInstance().sendMessage("参数错误。" +
|
||
"\n使用方式: " + QQCommands.QQ_LIVE_DATE + "+空格+roomId+起始时间+空格+结束时间" +
|
||
"\n时间格式：年-月-日_时:分:秒 例:2021-4-1_12:00:00" +
|
||
"\n时分秒不可省略");
|
||
}
|
||
} else if (msg.startsWith(QQCommands.QQ_LIVE_USE_SAVE)) {
|
||
try {
|
||
cmd = msg.split(" ");
|
||
_roomId = Integer.parseInt(cmd[1]);
|
||
if (SaveLive.getInstance().checkLive(_roomId)) {
|
||
SaveLive.getInstance().stop(_roomId);
|
||
}
|
||
SaveLive.getInstance().startLive(_roomId);
|
||
getInstance().sendMessage("已启动" + _roomId + "的录播");
|
||
} catch (Exception e) {
|
||
getInstance().sendMessage("参数错误。\n使用方式: " + QQCommands.QQ_LIVE_USE_SAVE + "+空格+roomId");
|
||
}
|
||
} else if (msg.startsWith(QQCommands.QQ_SYSTEM_LOG)) {
|
||
try {
|
||
cmd = msg.split(" ");
|
||
switch (cmd[1]){
|
||
case "true":
|
||
RedisTools.set(1,"live-log","true");
|
||
getInstance().sendMessage("日志设为 true");
|
||
break;
|
||
case "false":
|
||
RedisTools.set(1,"live-log","false");
|
||
getInstance().sendMessage("日志设为 false");
|
||
break;
|
||
default:
|
||
getInstance().sendMessage("设置错误,"+QQCommands.QQ_SYSTEM_LOG +" true|false");
|
||
}
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
}
|
||
}else {
|
||
System.out.println("未知指令:"+msg);
|
||
}
|
||
}
|
||
}
|
||
|
||
public void sendGiftData(int roomId, Date startTime, Date endTime) {
|
||
StringBuilder builder = new StringBuilder();
|
||
if (startTime == null) {
|
||
startTime = AppTools.getToDayStartTime();
|
||
}
|
||
if (endTime == null) {
|
||
endTime = AppTools.getToDayNowTime();
|
||
}
|
||
if (getInstance().realTimeDataController == null)
|
||
getInstance().realTimeDataController = getInstance().getBean(RealTimeDataController.class);
|
||
JSONObject json = getInstance().realTimeDataController.queryToDayLiveData(roomId, startTime, endTime);
|
||
System.out.println(json);
|
||
builder.append("当前人气:").append(json.getJSONObject("data").getInteger("popular")).append("\n");
|
||
builder.append("普通观众入场:").append(json.getJSONObject("data").getInteger("userLength")).append("\n");
|
||
builder.append("舰长入场:").append(json.getJSONObject("data").getInteger("vipLength")).append("\n");
|
||
int price = 0;
|
||
for (Object o : json.getJSONObject("data").getJSONArray("price")) {
|
||
price += ((JSONObject) o).getInteger("price");
|
||
}
|
||
builder.append("金瓜子:").append(price).append(" 抽成后:").append((price / 2) / 1000).append("￥").append("\n");
|
||
builder.append("礼物收益情况:").append("\n");
|
||
for (Object o : json.getJSONObject("data").getJSONArray("gift")) {
|
||
builder.append(((JSONObject) o).getString("giftName")).append(":").append(((JSONObject) o).getInteger("size")).append("\n");
|
||
}
|
||
getInstance().sendMessage(builder.toString());
|
||
}
|
||
|
||
private List<File> files;
|
||
private int index = 0;
|
||
|
||
private void sendImagesMsg(List<String> imgs, String text) {
|
||
files = new ArrayList<>();
|
||
index = 0;
|
||
if (imgs.size() == 0) {
|
||
getInstance().sendMessage(text);
|
||
return;
|
||
}
|
||
for (String img : imgs) {
|
||
Tools.download(img, new DownloadInterface() {
|
||
@Override
|
||
public void onDownload(File file) {
|
||
super.onDownload(file);
|
||
files.add(file);
|
||
send(imgs.size(), text);
|
||
}
|
||
|
||
@Override
|
||
public void onError(Exception e) {
|
||
super.onError(e);
|
||
index++;
|
||
send(imgs.size(), text);
|
||
}
|
||
});
|
||
}
|
||
}
|
||
|
||
private void send(int size, String text) {
|
||
if ((files.size() + index) == size) {
|
||
String str = getInstance().sendMessage(files, text);
|
||
Log.i("str = " + str);
|
||
}
|
||
}
|
||
|
||
private void sendVersion() {
|
||
String msg = "软件版本：" + BilibiliApplication.version;
|
||
QQBotManager.getInstance().sendMessage(msg);
|
||
}
|
||
|
||
}
|
||
}
|