Compare commits
2 commits
d1955da813
...
aebff1a12a
Author | SHA1 | Date | |
---|---|---|---|
aebff1a12a | |||
a57cd87fff |
1 changed files with 254 additions and 38 deletions
|
@ -4,19 +4,25 @@ import com.google.common.collect.ImmutableList;
|
|||
import com.pengrad.telegrambot.Callback;
|
||||
import com.pengrad.telegrambot.TelegramBot;
|
||||
import com.pengrad.telegrambot.UpdatesListener;
|
||||
import com.pengrad.telegrambot.model.Message;
|
||||
import com.pengrad.telegrambot.model.Update;
|
||||
import com.pengrad.telegrambot.model.User;
|
||||
import com.pengrad.telegrambot.request.EditMessageText;
|
||||
import com.pengrad.telegrambot.request.SendMessage;
|
||||
import com.pengrad.telegrambot.response.BaseResponse;
|
||||
import com.pengrad.telegrambot.response.SendResponse;
|
||||
import com.velocitypowered.api.event.Subscribe;
|
||||
import com.velocitypowered.api.event.connection.DisconnectEvent;
|
||||
import com.velocitypowered.api.event.player.ServerConnectedEvent;
|
||||
import com.velocitypowered.api.event.player.ServerPostConnectEvent;
|
||||
import com.velocitypowered.api.event.proxy.ProxyShutdownEvent;
|
||||
import com.velocitypowered.api.proxy.Player;
|
||||
import com.velocitypowered.api.proxy.ProxyServer;
|
||||
import com.velocitypowered.api.proxy.server.RegisteredServer;
|
||||
import de.strifel.VTools.VTools;
|
||||
import net.kyori.adventure.text.Component;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
import org.yaml.snakeyaml.Yaml;
|
||||
|
||||
import java.io.IOException;
|
||||
|
@ -27,6 +33,8 @@ import java.nio.file.Path;
|
|||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.LinkedBlockingQueue;
|
||||
import java.util.function.BiConsumer;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
public class TGBridge {
|
||||
|
@ -36,11 +44,14 @@ public class TGBridge {
|
|||
|
||||
private TelegramBot bot;
|
||||
private String TOKEN = "";
|
||||
private long BOT_ID = -1;
|
||||
private long CHAT_ID = 0L;
|
||||
|
||||
private int ONLINE_STATUS_MESSAGE_ID = -1;
|
||||
|
||||
private long backoffSec = 1L;
|
||||
|
||||
public TGBridge(VTools plugin) {
|
||||
public TGBridge(@NotNull VTools plugin) {
|
||||
INSTANCE = this;
|
||||
this.plugin = plugin;
|
||||
this.server = plugin.getServer();
|
||||
|
@ -49,6 +60,7 @@ public class TGBridge {
|
|||
public void register() {
|
||||
server.getEventManager().register(plugin, this);
|
||||
botInit();
|
||||
initUpdateThread();
|
||||
}
|
||||
|
||||
private void loadConfig() {
|
||||
|
@ -67,6 +79,7 @@ public class TGBridge {
|
|||
synchronized (this) {
|
||||
this.CHAT_ID = Long.parseLong(config.getOrDefault("chat_id", "0"));
|
||||
this.TOKEN = config.getOrDefault("token", "");
|
||||
this.BOT_ID = Long.parseLong(TOKEN.split(":")[0]);
|
||||
}
|
||||
} catch (Exception e) {
|
||||
plugin.logger.error("parsing config", e);
|
||||
|
@ -89,44 +102,47 @@ public class TGBridge {
|
|||
) {
|
||||
if (update.message().text() != null && !update.message().text().isEmpty()) {
|
||||
String msg = update.message().text();
|
||||
if (msg.equals("/list")) {
|
||||
ArrayList<String> out = new ArrayList<>();
|
||||
String fmt = server.getAllPlayers().size() > 1 ? "%d players are currently connected to the proxy." : "%d player is currently connected to the proxy.";
|
||||
out.add(String.format(fmt, server.getAllPlayers().size()));
|
||||
List<RegisteredServer> servers = new ArrayList<>(server.getAllServers());
|
||||
for (RegisteredServer server : servers) {
|
||||
List<Player> onServer = ImmutableList.copyOf(server.getPlayersConnected());
|
||||
if (!onServer.isEmpty()) {
|
||||
out.add(String.format("[%s] (%d): %s",
|
||||
server.getServerInfo().getName(),
|
||||
onServer.size(),
|
||||
onServer.stream().map(Player::getUsername).collect(Collectors.joining(", ")))
|
||||
switch (msg) {
|
||||
case "/list":
|
||||
outbound(genOnlineStatus());
|
||||
break;
|
||||
case "/setpin":
|
||||
Message replyTo = update.message().replyToMessage();
|
||||
if (replyTo == null || replyTo.from() == null || replyTo.from().id() != BOT_ID) {
|
||||
outbound("must reply a message that from the bot");
|
||||
} else {
|
||||
int messageId = replyTo.messageId();
|
||||
ONLINE_STATUS_MESSAGE_ID = messageId > 0 ? messageId : ONLINE_STATUS_MESSAGE_ID;
|
||||
outbound("done");
|
||||
}
|
||||
break;
|
||||
case "/genpin":
|
||||
outbound(genOnlineStatus(),
|
||||
(sendMessage, sendResponse) -> {
|
||||
if (!sendResponse.isOk()) {
|
||||
plugin.logger.error(String.format("sendMessage error %d: %s", sendResponse.errorCode(), sendResponse.description()));
|
||||
} else {
|
||||
int messageId = sendResponse.message().messageId();
|
||||
ONLINE_STATUS_MESSAGE_ID = messageId > 0 ? messageId : ONLINE_STATUS_MESSAGE_ID;
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
}
|
||||
}
|
||||
outbound(String.join("\n", out));
|
||||
break;
|
||||
}
|
||||
tgInbound(update.message().from(), msg);
|
||||
}
|
||||
else if (update.message().sticker() != null) {
|
||||
} else if (update.message().sticker() != null) {
|
||||
tgInbound(update.message().from(), "[sticker]");
|
||||
}
|
||||
else if (update.message().photo() != null) {
|
||||
} else if (update.message().photo() != null) {
|
||||
tgInbound(update.message().from(), "[photo]");
|
||||
}
|
||||
else if (update.message().audio() != null) {
|
||||
} else if (update.message().audio() != null) {
|
||||
tgInbound(update.message().from(), "[audio]");
|
||||
}
|
||||
else if (update.message().voice() != null) {
|
||||
} else if (update.message().voice() != null) {
|
||||
tgInbound(update.message().from(), "[voice]");
|
||||
}
|
||||
else if (update.message().document() != null) {
|
||||
} else if (update.message().document() != null) {
|
||||
tgInbound(update.message().from(), "[document]");
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception e) {
|
||||
} catch (Exception e) {
|
||||
plugin.logger.error("handling update", e);
|
||||
}
|
||||
}
|
||||
|
@ -142,6 +158,24 @@ public class TGBridge {
|
|||
});
|
||||
}
|
||||
|
||||
private String genOnlineStatus() {
|
||||
ArrayList<String> out = new ArrayList<>();
|
||||
String fmt = server.getAllPlayers().size() > 1 ? "%d players are currently connected to the proxy." : "%d player is currently connected to the proxy.";
|
||||
out.add(String.format(fmt, server.getAllPlayers().size()));
|
||||
List<RegisteredServer> registeredServers = new ArrayList<>(server.getAllServers());
|
||||
for (RegisteredServer registeredServer : registeredServers) {
|
||||
List<Player> onServer = ImmutableList.copyOf(registeredServer.getPlayersConnected());
|
||||
if (!onServer.isEmpty()) {
|
||||
out.add(String.format("[%s] (%d): %s",
|
||||
registeredServer.getServerInfo().getName(),
|
||||
onServer.size(),
|
||||
onServer.stream().map(Player::getUsername).collect(Collectors.joining(", ")))
|
||||
);
|
||||
}
|
||||
}
|
||||
return String.join("\n", out);
|
||||
}
|
||||
|
||||
protected void tgInbound(User user, String content) {
|
||||
inbound(String.format("[tg] <%s> %s", user.lastName() == null ? user.firstName() : String.format("%s %s", user.firstName(), user.lastName()), content));
|
||||
}
|
||||
|
@ -155,6 +189,10 @@ public class TGBridge {
|
|||
}
|
||||
|
||||
protected void outbound(String content) {
|
||||
outbound(content, null);
|
||||
}
|
||||
|
||||
protected void outbound(String content, @Nullable BiConsumer<SendMessage, SendResponse> onResponse) {
|
||||
if (bot == null) return;
|
||||
if (content.length() > 4000) {
|
||||
content = content.substring(0, 4000);
|
||||
|
@ -162,9 +200,13 @@ public class TGBridge {
|
|||
bot.execute(new SendMessage(CHAT_ID, content), new Callback<SendMessage, SendResponse>() {
|
||||
@Override
|
||||
public void onResponse(SendMessage sendMessage, SendResponse sendResponse) {
|
||||
if (onResponse == null) {
|
||||
if (!sendResponse.isOk()) {
|
||||
plugin.logger.error(String.format("sendMessage error %d: %s", sendResponse.errorCode(), sendResponse.description()));
|
||||
}
|
||||
} else {
|
||||
onResponse.accept(sendMessage, sendResponse);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
|
@ -179,21 +221,195 @@ public class TGBridge {
|
|||
if (bot == null) return;
|
||||
bot.removeGetUpdatesListener();
|
||||
bot.shutdown();
|
||||
PROXY_SHUT_DOWN = true;
|
||||
}
|
||||
|
||||
@Subscribe
|
||||
public void onServerConnected(ServerConnectedEvent event) {
|
||||
if (event.getPreviousServer().isEmpty()) {
|
||||
if (!event.getPlayer().hasPermission("vtools.globalchat.bypassbridge.join")) {
|
||||
outbound(String.format("%s joined the proxy", event.getPlayer().getUsername()));
|
||||
joinLeftAnnounce(String.format("%s joined the proxy", event.getPlayer().getUsername()));
|
||||
}
|
||||
}
|
||||
updateRequests.add(new UpdateRequest());
|
||||
}
|
||||
|
||||
@Subscribe
|
||||
public void onDisconnect(DisconnectEvent event) {
|
||||
if (!event.getPlayer().hasPermission("vtools.globalchat.bypassbridge.join")) {
|
||||
outbound(String.format("%s left the proxy", event.getPlayer().getUsername()));
|
||||
joinLeftAnnounce(String.format("%s left the proxy", event.getPlayer().getUsername()));
|
||||
}
|
||||
updateRequests.add(new UpdateRequest(2));
|
||||
}
|
||||
|
||||
@Subscribe
|
||||
public void onServerPostConnect(ServerPostConnectEvent event){
|
||||
updateRequests.add(new UpdateRequest());
|
||||
}
|
||||
|
||||
|
||||
|
||||
private boolean PROXY_SHUT_DOWN = false;
|
||||
|
||||
private static class UpdateRequest {
|
||||
int updateTimes;
|
||||
|
||||
UpdateRequest() {
|
||||
this(1);
|
||||
}
|
||||
|
||||
UpdateRequest(int updateTimes) {
|
||||
this.updateTimes = updateTimes;
|
||||
}
|
||||
}
|
||||
|
||||
private LinkedBlockingQueue<UpdateRequest> updateRequests = new LinkedBlockingQueue<>();
|
||||
|
||||
private void initUpdateThread() {
|
||||
new Thread(() -> {
|
||||
while (true) {
|
||||
if (PROXY_SHUT_DOWN) {
|
||||
setOnlineStatusNotAvailable();
|
||||
return;
|
||||
}
|
||||
int maxiterationNum = 0;
|
||||
UpdateRequest oldestRequest = null;
|
||||
try {
|
||||
oldestRequest = updateRequests.take();
|
||||
} catch (InterruptedException ignored) {}
|
||||
if (oldestRequest == null) continue;
|
||||
maxiterationNum = Math.max(maxiterationNum, oldestRequest.updateTimes - 1);
|
||||
while (!updateRequests.isEmpty()){
|
||||
try {
|
||||
maxiterationNum = Math.max(maxiterationNum, updateRequests.take().updateTimes - 1);
|
||||
} catch (InterruptedException ignored) {}
|
||||
}
|
||||
|
||||
if (!updateOnlineStatus()) {
|
||||
updateRequests.add(oldestRequest); // 更新失败 回去吧您内
|
||||
}
|
||||
if(maxiterationNum>0){
|
||||
updateRequests.add(new UpdateRequest(maxiterationNum));
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
new Thread(()->{
|
||||
while (true) {
|
||||
if (PROXY_SHUT_DOWN) {
|
||||
return;
|
||||
}
|
||||
String oldestMessage = null;
|
||||
try {
|
||||
oldestMessage = announceQueue.take();
|
||||
} catch (InterruptedException ignored) {}
|
||||
|
||||
if(!currentAnnounce.isValid()){
|
||||
currentAnnounce = new JoinLeftAnnounceMessage(oldestMessage);
|
||||
continue;
|
||||
}
|
||||
ArrayList<String> messages = new ArrayList<>(announceQueue.size() + 1);
|
||||
messages.add(oldestMessage);
|
||||
while (!announceQueue.isEmpty()) {
|
||||
try {
|
||||
messages.add(announceQueue.take());
|
||||
} catch (InterruptedException ignored) {}
|
||||
}
|
||||
currentAnnounce.addLines(messages);
|
||||
}
|
||||
}).start();
|
||||
}
|
||||
|
||||
protected boolean updateOnlineStatus() {
|
||||
return editOnlineStatusMessage(genOnlineStatus());
|
||||
}
|
||||
|
||||
protected boolean setOnlineStatusNotAvailable() {
|
||||
return editOnlineStatusMessage("(proxy already shutdown)");
|
||||
}
|
||||
|
||||
protected boolean editOnlineStatusMessage(String text) {
|
||||
if (ONLINE_STATUS_MESSAGE_ID < 1) {
|
||||
return true;
|
||||
}
|
||||
BaseResponse response;
|
||||
try {
|
||||
response = bot.execute(new EditMessageText(CHAT_ID, ONLINE_STATUS_MESSAGE_ID, text));
|
||||
} catch (RuntimeException e) {return false;}
|
||||
return response != null && response.isOk();
|
||||
}
|
||||
|
||||
private class JoinLeftAnnounceMessage {
|
||||
private int messageId;
|
||||
private long time;
|
||||
|
||||
private StringBuilder text;
|
||||
|
||||
boolean isValid() {
|
||||
if (messageId < 1) {
|
||||
return false;
|
||||
}
|
||||
long dt = System.currentTimeMillis() - time;
|
||||
return dt <= 60_000 && dt >= 0;
|
||||
}
|
||||
|
||||
protected JoinLeftAnnounceMessage(String firstMessage) {
|
||||
text = new StringBuilder(firstMessage);
|
||||
time = System.currentTimeMillis();
|
||||
sendAnnounceMessage();
|
||||
}
|
||||
|
||||
private void sendAnnounceMessage() {
|
||||
if (bot == null) {
|
||||
messageId = -1;
|
||||
return;
|
||||
}
|
||||
SendResponse response;
|
||||
try {
|
||||
response = bot.execute(new SendMessage(CHAT_ID, text.toString()));
|
||||
|
||||
} catch (RuntimeException e) {
|
||||
messageId = -1;
|
||||
return;
|
||||
}
|
||||
if(response.isOk() == false){
|
||||
messageId = -1;
|
||||
return;
|
||||
}
|
||||
messageId = response.message().messageId();
|
||||
}
|
||||
|
||||
protected JoinLeftAnnounceMessage(){
|
||||
messageId = 0;
|
||||
time = 0;
|
||||
text = new StringBuilder();
|
||||
} //dummy
|
||||
private void addLines(List<String> messages) {
|
||||
for (String message : messages) {
|
||||
text.append('\n').append(message);
|
||||
}
|
||||
updateAnnounceMessage();
|
||||
}
|
||||
|
||||
private void updateAnnounceMessage() {
|
||||
if(!isValid()){
|
||||
plugin.logger.error("message should only push to a valid object");
|
||||
return;
|
||||
}
|
||||
if (bot == null) {
|
||||
messageId = -1;
|
||||
return;
|
||||
}
|
||||
try {
|
||||
bot.execute(new EditMessageText(CHAT_ID, messageId, text.toString()));
|
||||
} catch (RuntimeException ignored) {}
|
||||
}
|
||||
}
|
||||
|
||||
private JoinLeftAnnounceMessage currentAnnounce = new JoinLeftAnnounceMessage();
|
||||
private LinkedBlockingQueue<String> announceQueue = new LinkedBlockingQueue<>();
|
||||
|
||||
private void joinLeftAnnounce(String message){
|
||||
announceQueue.add(message);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue