添加在机器人在线状态中显示服务器状态的功能

Fixes #95
Xujiayao · Jul 4, 2024 · abd788a · abd788a
1 parent 7b9d9f7
commit abd788a
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 5 deletions.
diff --git a/src/main/java/com/xujiayao/discord_mc_chat/Config.java b/src/main/java/com/xujiayao/discord_mc_chat/Config.java
@@ -23,6 +23,8 @@ public static class Generic {
 		public String language = "en_us";
 
 		public String botToken = "";
+
+		public boolean showServerStatusInBotStatus = true;
 		public String botPlayingActivity = "Minecraft (%onlinePlayerCount%/%maxPlayerCount%)";
 		public String botListeningActivity = "";
 

diff --git a/src/main/java/com/xujiayao/discord_mc_chat/Main.java b/src/main/java/com/xujiayao/discord_mc_chat/Main.java
@@ -3,6 +3,7 @@
 import com.xujiayao.discord_mc_chat.minecraft.MinecraftEventListener;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.JDABuilder;
+import net.dv8tion.jda.api.OnlineStatus;
 import net.dv8tion.jda.api.entities.Webhook;
 import net.dv8tion.jda.api.entities.channel.concrete.TextChannel;
 import net.dv8tion.jda.api.requests.GatewayIntent;
@@ -88,6 +89,7 @@ public void onInitializeServer() {
 					.setMemberCachePolicy(MemberCachePolicy.ALL)
 					.enableIntents(GatewayIntent.GUILD_MEMBERS, GatewayIntent.MESSAGE_CONTENT)
 					.addEventListeners(new DiscordEventListener())
+					.setStatus(CONFIG.generic.showServerStatusInBotStatus ? OnlineStatus.DO_NOT_DISTURB : OnlineStatus.ONLINE)
 					.build();
 
 			JDA.awaitReady();
@@ -174,7 +176,7 @@ public void onInitializeServer() {
 
 			SERVER = server;
 
-			Utils.setBotActivity();
+			Utils.setBotPresence();
 
 			Utils.initCheckUpdateTimer();
 
@@ -191,6 +193,12 @@ public void onInitializeServer() {
 			}
 		});
 
+		ServerLifecycleEvents.SERVER_STOPPING.register(server -> {
+			if (CONFIG.generic.showServerStatusInBotStatus) {
+				JDA.getPresence().setStatus(OnlineStatus.DO_NOT_DISTURB);
+			}
+		});
+
 		ServerLifecycleEvents.SERVER_STOPPED.register(server -> {
 			MSPT_MONITOR_TIMER.cancel();
 			CHANNEL_TOPIC_MONITOR_TIMER.cancel();

diff --git a/src/main/java/com/xujiayao/discord_mc_chat/minecraft/MinecraftEventListener.java b/src/main/java/com/xujiayao/discord_mc_chat/minecraft/MinecraftEventListener.java
@@ -265,7 +265,7 @@ public static void init() {
 		});
 
 		MinecraftEvents.PLAYER_JOIN.register(player -> {
-			Utils.setBotActivity();
+			Utils.setBotPresence();
 
 			if (CONFIG.generic.announcePlayerJoinLeave) {
 				CHANNEL.sendMessage(Translations.translateMessage("message.joinServer")
@@ -278,7 +278,7 @@ public static void init() {
 		});
 
 		MinecraftEvents.PLAYER_QUIT.register(player -> {
-			Utils.setBotActivity();
+			Utils.setBotPresence();
 
 			if (CONFIG.generic.announcePlayerJoinLeave) {
 				CHANNEL.sendMessage(Translations.translateMessage("message.leftServer")

diff --git a/src/main/java/com/xujiayao/discord_mc_chat/utils/Utils.java b/src/main/java/com/xujiayao/discord_mc_chat/utils/Utils.java
@@ -6,6 +6,7 @@
 import com.google.gson.JsonObject;
 import com.mojang.authlib.GameProfile;
 import com.xujiayao.discord_mc_chat.multi_server.MultiServer;
+import net.dv8tion.jda.api.OnlineStatus;
 import net.dv8tion.jda.api.entities.Activity;
 import net.dv8tion.jda.api.entities.Member;
 import net.dv8tion.jda.api.entities.Role;
@@ -273,7 +274,7 @@ public static String reload() {
 			ConfigManager.init(true);
 			Translations.init();
 
-			Utils.setBotActivity();
+			Utils.setBotPresence();
 
 			CHANNEL = JDA.getTextChannelById(CONFIG.generic.channelId);
 			if (CHANNEL == null) {
@@ -517,11 +518,18 @@ public static String getStatsCommandMessage(String type, String name, boolean ha
 		return message.toString();
 	}
 
-	public static void setBotActivity() {
+	public static void setBotPresence() {
 		if (SERVER == null) {
 			// Bot is registered before official server start
 			return;
 		}
+
+		if (CONFIG.generic.showServerStatusInBotStatus && SERVER.getPlayerCount() == 0) {
+			JDA.getPresence().setStatus(OnlineStatus.IDLE);
+		} else {
+			JDA.getPresence().setStatus(OnlineStatus.ONLINE);
+		}
+
 		if (!CONFIG.generic.botPlayingActivity.isEmpty()) {
 			JDA.getPresence().setActivity(Activity.playing(CONFIG.generic.botPlayingActivity
 					.replace("%onlinePlayerCount%", Integer.toString(SERVER.getPlayerCount()))