feat: add maintenance mode

2026-06-11 00:19:54 +08:00
parent c0289c55d3
commit 113ae24afe
14 changed files with 490 additions and 3 deletions
@@ -4,6 +4,8 @@ import cn.infstar.essentialsC.admin.AdminModeManager;
 import cn.infstar.essentialsC.commands.BaseCommand;
 import cn.infstar.essentialsC.commands.CommandRegistry;
 import cn.infstar.essentialsC.commands.HelpCommand;
+import cn.infstar.essentialsC.maintenance.MaintenanceListener;
+import cn.infstar.essentialsC.maintenance.MaintenanceManager;
 import cn.infstar.essentialsC.tpsbar.TpsBarService;
 import org.bukkit.Bukkit;
 import org.bukkit.command.Command;
@@ -18,6 +20,7 @@ public final class EssentialsC extends JavaPlugin {
    private static LangManager langManager;
    private ModuleManager moduleManager;
    private AdminModeManager adminModeManager;
+    private MaintenanceManager maintenanceManager;
    private TpsBarService tpsBarManager;

    @Override
@@ -30,6 +33,11 @@ public final class EssentialsC extends JavaPlugin {
            getServer().getPluginManager().registerEvents(adminModeManager, this);
        }

+        if (moduleManager.isEnabled(ModuleManager.MAINTENANCE)) {
+            maintenanceManager = new MaintenanceManager(this);
+            getServer().getPluginManager().registerEvents(new MaintenanceListener(this), this);
+        }
+
        if (moduleManager.isEnabled(ModuleManager.TPSBAR)) {
            tpsBarManager = createOptionalService("cn.infstar.essentialsC.tpsbar.TpsBarManager", TpsBarService.class);
        }
@@ -51,6 +59,9 @@ public final class EssentialsC extends JavaPlugin {
        if (adminModeManager != null) {
            adminModeManager.shutdown();
        }
+        if (maintenanceManager != null) {
+            maintenanceManager.shutdown();
+        }
        getLogger().info("EssentialsC 已禁用。");
    }

@@ -66,6 +77,10 @@ public final class EssentialsC extends JavaPlugin {
        return moduleManager;
    }

+    public MaintenanceManager getMaintenanceManager() {
+        return maintenanceManager;
+    }
+
    public TpsBarService getTpsBarManager() {
        return tpsBarManager;
    }
@@ -151,6 +166,7 @@ public final class EssentialsC extends JavaPlugin {
    }

    private void registerCommandWithAliases(org.bukkit.command.CommandMap commandMap, String name, BaseCommand executor, String... aliases) {
+        String fallbackPrefix = getName().toLowerCase(java.util.Locale.ROOT);
        Command command = new Command(name) {
            @Override
            public boolean execute(CommandSender sender, String commandLabel, String[] args) {
@@ -171,7 +187,7 @@ public final class EssentialsC extends JavaPlugin {
        };

        command.setPermission(executor.getPermission());
-        commandMap.register("", command);
+        commandMap.register(fallbackPrefix, command);

        for (String alias : aliases) {
            Command aliasCmd = new Command(alias) {
@@ -193,7 +209,7 @@ public final class EssentialsC extends JavaPlugin {
                }
            };
            aliasCmd.setPermission(executor.getPermission());
-            commandMap.register("", aliasCmd);
+            commandMap.register(fallbackPrefix, aliasCmd);
        }
    }
 }
@@ -19,6 +19,7 @@ public final class ModuleManager {
    public static final String TPSBAR = "tpsbar";
    public static final String JEI_SYNC = "jei-sync";
    public static final String MOB_DROPS = "mob-drops";
+    public static final String MAINTENANCE = "maintenance";

    private static final Map<String, Boolean> DEFAULT_MODULES = new LinkedHashMap<>();

@@ -29,6 +30,7 @@ public final class ModuleManager {
        DEFAULT_MODULES.put(TPSBAR, true);
        DEFAULT_MODULES.put(JEI_SYNC, true);
        DEFAULT_MODULES.put(MOB_DROPS, false);
+        DEFAULT_MODULES.put(MAINTENANCE, true);
    }

    private final JavaPlugin plugin;
@@ -42,6 +42,7 @@ public final class CommandRegistry {
        register("repair", "essentialsc.command.repair", ModuleManager.PLAYER, "cn.infstar.essentialsC.commands.RepairCommand", "rep");
        register("tpsbar", "essentialsc.command.tpsbar", ModuleManager.TPSBAR, "cn.infstar.essentialsC.commands.TpsBarCommand");
        register("mobdrops", "essentialsc.mobdrops.enderman", ModuleManager.MOB_DROPS, "cn.infstar.essentialsC.commands.MobDropCommand");
+        register("maintenance", "essentialsc.command.maintenance", ModuleManager.MAINTENANCE, "cn.infstar.essentialsC.commands.MaintenanceCommand", "maint");
        registerSubCommand("admin", "essentialsc.command.admin", ModuleManager.ADMIN_MODE, "cn.infstar.essentialsC.commands.AdminCommand");
    }

@@ -37,6 +37,9 @@ public class HelpCommand extends BaseCommand implements TabCompleter {
            EssentialsC.getLangManager().reload();
            plugin.getModuleManager().reload();
            CommandRegistry.clearCache();
+            if (plugin.getMaintenanceManager() != null && plugin.getModuleManager().isEnabled(ModuleManager.MAINTENANCE)) {
+                plugin.getMaintenanceManager().reload();
+            }
            TpsBarService tpsBarService = plugin.getTpsBarManager();
            if (tpsBarService != null) {
                if (plugin.getModuleManager().isEnabled(ModuleManager.TPSBAR)) {
@@ -65,6 +68,9 @@ public class HelpCommand extends BaseCommand implements TabCompleter {
                EssentialsC.getLangManager().reload();
                plugin.getModuleManager().reload();
                CommandRegistry.clearCache();
+                if (plugin.getMaintenanceManager() != null && plugin.getModuleManager().isEnabled(ModuleManager.MAINTENANCE)) {
+                    plugin.getMaintenanceManager().reload();
+                }
                TpsBarService tpsBarService = plugin.getTpsBarManager();
                if (tpsBarService != null) {
                    if (plugin.getModuleManager().isEnabled(ModuleManager.TPSBAR)) {
@@ -201,6 +207,10 @@ public class HelpCommand extends BaseCommand implements TabCompleter {
            otherCommands.append(lang.getString("help.commands.tpsbar")).append("\n");
            hasOtherCommands = true;
        }
+        if (CommandRegistry.isAvailable("maintenance") && player.hasPermission("essentialsc.command.maintenance")) {
+            otherCommands.append(lang.getString("help.commands.maintenance")).append("\n");
+            hasOtherCommands = true;
+        }

        if (hasOtherCommands) {
            sendPrefixed(player, lang.getString("help.section-other"));
@@ -272,6 +282,8 @@ public class HelpCommand extends BaseCommand implements TabCompleter {
                {"repair", "essentialsc.command.repair"},
                {"rep", "essentialsc.command.repair"},
                {"tpsbar", "essentialsc.command.tpsbar"},
+                {"maintenance", "essentialsc.command.maintenance"},
+                {"maint", "essentialsc.command.maintenance"},
                {"mobdrops", "essentialsc.mobdrops.enderman"},
                {"admin", "essentialsc.command.admin"},
                {"version", null},
@@ -324,6 +336,11 @@ public class HelpCommand extends BaseCommand implements TabCompleter {
                }
                return players;
            }
+
+            if ((subCmd.equals("maintenance") || subCmd.equals("maint"))
+                && sender.hasPermission("essentialsc.command.maintenance")) {
+                return completeMaintenanceArgs(args[1]);
+            }
        }

        return new ArrayList<>();
@@ -339,4 +356,15 @@ public class HelpCommand extends BaseCommand implements TabCompleter {
        }
        return completions;
    }
+
+    private List<String> completeMaintenanceArgs(String partialInput) {
+        List<String> completions = new ArrayList<>();
+        String partial = partialInput.toLowerCase();
+        for (String option : List.of("on", "off", "status", "reload")) {
+            if (option.startsWith(partial)) {
+                completions.add(option);
+            }
+        }
+        return completions;
+    }
 }
@@ -0,0 +1,86 @@
+package cn.infstar.essentialsC.commands;
+
+import cn.infstar.essentialsC.maintenance.MaintenanceManager;
+import org.bukkit.command.Command;
+import org.bukkit.command.CommandSender;
+import org.bukkit.command.TabCompleter;
+import org.bukkit.entity.Player;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class MaintenanceCommand extends BaseCommand implements TabCompleter {
+
+    public MaintenanceCommand() {
+        super("essentialsc.command.maintenance");
+    }
+
+    @Override
+    protected boolean execute(Player player, String[] args) {
+        return executeCommand(player, args);
+    }
+
+    @Override
+    protected boolean executeConsole(CommandSender sender, String[] args) {
+        return executeCommand(sender, args);
+    }
+
+    private boolean executeCommand(CommandSender sender, String[] args) {
+        MaintenanceManager maintenanceManager = plugin.getMaintenanceManager();
+        if (maintenanceManager == null) {
+            sender.sendMessage(getLang().getPrefixedString("messages.module-disabled"));
+            return true;
+        }
+
+        if (args.length == 0 || args[0].equalsIgnoreCase("status")) {
+            sendStatus(sender, maintenanceManager);
+            return true;
+        }
+
+        switch (args[0].toLowerCase()) {
+            case "on", "enable", "enabled" -> {
+                maintenanceManager.setEnabled(true);
+                sender.sendMessage(getLang().getPrefixedString("maintenance.messages.enabled"));
+                return true;
+            }
+            case "off", "disable", "disabled" -> {
+                maintenanceManager.setEnabled(false);
+                sender.sendMessage(getLang().getPrefixedString("maintenance.messages.disabled"));
+                return true;
+            }
+            case "reload" -> {
+                maintenanceManager.reload();
+                sender.sendMessage(getLang().getPrefixedString("maintenance.messages.reloaded"));
+                return true;
+            }
+            default -> {
+                sender.sendMessage(getLang().getPrefixedString("maintenance.messages.usage"));
+                return true;
+            }
+        }
+    }
+
+    private void sendStatus(CommandSender sender, MaintenanceManager maintenanceManager) {
+        String status = getLang().getString(maintenanceManager.isEnabled()
+            ? "maintenance.status.enabled"
+            : "maintenance.status.disabled");
+        sender.sendMessage(getLang().getPrefixedString("maintenance.messages.status",
+            java.util.Map.of("status", status)));
+    }
+
+    @Override
+    public List<String> onTabComplete(CommandSender sender, Command command, String label, String[] args) {
+        if (args.length != 1 || !sender.hasPermission(getPermission())) {
+            return List.of();
+        }
+
+        String partial = args[0].toLowerCase();
+        List<String> completions = new ArrayList<>();
+        for (String option : List.of("on", "off", "status", "reload")) {
+            if (option.startsWith(partial)) {
+                completions.add(option);
+            }
+        }
+        return completions;
+    }
+}
@@ -0,0 +1,66 @@
+package cn.infstar.essentialsC.maintenance;
+
+import cn.infstar.essentialsC.EssentialsC;
+import cn.infstar.essentialsC.ModuleManager;
+import org.bukkit.event.EventHandler;
+import org.bukkit.event.Listener;
+import org.bukkit.event.player.PlayerLoginEvent;
+import org.bukkit.event.player.PlayerJoinEvent;
+import org.bukkit.event.player.PlayerQuitEvent;
+import org.bukkit.event.server.ServerListPingEvent;
+
+public final class MaintenanceListener implements Listener {
+
+    private final EssentialsC plugin;
+
+    public MaintenanceListener(EssentialsC plugin) {
+        this.plugin = plugin;
+    }
+
+    @EventHandler
+    public void onServerListPing(ServerListPingEvent event) {
+        MaintenanceManager maintenanceManager = plugin.getMaintenanceManager();
+        if (!plugin.getModuleManager().isEnabled(ModuleManager.MAINTENANCE)
+            || maintenanceManager == null
+            || !maintenanceManager.isEnabled()
+            || !maintenanceManager.isMotdEnabled()) {
+            return;
+        }
+
+        event.setMotd(maintenanceManager.getMotd());
+    }
+
+    @EventHandler
+    public void onPlayerLogin(PlayerLoginEvent event) {
+        MaintenanceManager maintenanceManager = plugin.getMaintenanceManager();
+        if (!plugin.getModuleManager().isEnabled(ModuleManager.MAINTENANCE)
+            || maintenanceManager == null
+            || !maintenanceManager.isEnabled()) {
+            return;
+        }
+
+        if (event.getPlayer().hasPermission(maintenanceManager.getBypassPermission())) {
+            return;
+        }
+
+        event.disallow(PlayerLoginEvent.Result.KICK_OTHER, maintenanceManager.getKickMessage());
+    }
+
+    @EventHandler
+    public void onPlayerJoin(PlayerJoinEvent event) {
+        MaintenanceManager maintenanceManager = plugin.getMaintenanceManager();
+        if (!plugin.getModuleManager().isEnabled(ModuleManager.MAINTENANCE) || maintenanceManager == null) {
+            return;
+        }
+
+        maintenanceManager.addBossBarPlayer(event.getPlayer());
+    }
+
+    @EventHandler
+    public void onPlayerQuit(PlayerQuitEvent event) {
+        MaintenanceManager maintenanceManager = plugin.getMaintenanceManager();
+        if (maintenanceManager != null) {
+            maintenanceManager.removeBossBarPlayer(event.getPlayer());
+        }
+    }
+}
@@ -0,0 +1,178 @@
+package cn.infstar.essentialsC.maintenance;
+
+import cn.infstar.essentialsC.EssentialsC;
+import org.bukkit.ChatColor;
+import org.bukkit.boss.BarColor;
+import org.bukkit.boss.BarStyle;
+import org.bukkit.boss.BossBar;
+import org.bukkit.configuration.file.FileConfiguration;
+import org.bukkit.configuration.file.YamlConfiguration;
+import org.bukkit.entity.Player;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.List;
+import java.util.Locale;
+
+public final class MaintenanceManager {
+
+    private static final int CURRENT_CONFIG_VERSION = 1;
+
+    private final EssentialsC plugin;
+    private final File configFile;
+    private FileConfiguration config;
+    private BossBar bossBar;
+
+    public MaintenanceManager(EssentialsC plugin) {
+        this.plugin = plugin;
+        this.configFile = new File(plugin.getDataFolder(), "maintenance.yml");
+        reload();
+    }
+
+    public void reload() {
+        if (!configFile.exists()) {
+            plugin.saveResource("maintenance.yml", false);
+        }
+
+        config = YamlConfiguration.loadConfiguration(configFile);
+        config.addDefault("config-version", CURRENT_CONFIG_VERSION);
+        config.addDefault("enabled", false);
+        config.addDefault("bypass-permission", "essentialsc.maintenance.bypass");
+        config.addDefault("motd.enabled", true);
+        config.addDefault("motd.lines", List.of("&c服务器维护中", "&7请稍后再试"));
+        config.addDefault("kick-message", List.of("&c服务器正在维护", "&7请稍后再进入"));
+        config.addDefault("bossbar.enabled", true);
+        config.addDefault("bossbar.title", "&c服务器正在维护");
+        config.addDefault("bossbar.color", "RED");
+        config.addDefault("bossbar.style", "SOLID");
+        config.addDefault("bossbar.progress", 1.0D);
+        config.options().copyDefaults(true);
+        save();
+        refreshBossBar();
+    }
+
+    public boolean isEnabled() {
+        return config.getBoolean("enabled", false);
+    }
+
+    public void setEnabled(boolean enabled) {
+        config.set("enabled", enabled);
+        save();
+        refreshBossBar();
+    }
+
+    public String getBypassPermission() {
+        return config.getString("bypass-permission", "essentialsc.maintenance.bypass");
+    }
+
+    public boolean isMotdEnabled() {
+        return config.getBoolean("motd.enabled", true);
+    }
+
+    public String getMotd() {
+        return colorizeLines(config.getStringList("motd.lines"));
+    }
+
+    public String getKickMessage() {
+        return colorizeLines(config.getStringList("kick-message"));
+    }
+
+    public void refreshBossBar() {
+        clearBossBar();
+        if (!isEnabled() || !isBossBarEnabled()) {
+            return;
+        }
+
+        bossBar = plugin.getServer().createBossBar(
+            colorize(config.getString("bossbar.title", "&c服务器正在维护")),
+            readBossBarColor(),
+            readBossBarStyle()
+        );
+        bossBar.setProgress(readBossBarProgress());
+        bossBar.setVisible(true);
+
+        for (Player player : plugin.getServer().getOnlinePlayers()) {
+            addBossBarPlayer(player);
+        }
+    }
+
+    public void addBossBarPlayer(Player player) {
+        if (bossBar == null || player == null || !player.isOnline()) {
+            return;
+        }
+        if (player.hasPermission(getBypassPermission())) {
+            bossBar.removePlayer(player);
+            return;
+        }
+        bossBar.addPlayer(player);
+    }
+
+    public void removeBossBarPlayer(Player player) {
+        if (bossBar != null && player != null) {
+            bossBar.removePlayer(player);
+        }
+    }
+
+    public void shutdown() {
+        clearBossBar();
+    }
+
+    private void clearBossBar() {
+        if (bossBar == null) {
+            return;
+        }
+
+        bossBar.removeAll();
+        bossBar = null;
+    }
+
+    private boolean isBossBarEnabled() {
+        return config.getBoolean("bossbar.enabled", true);
+    }
+
+    private BarColor readBossBarColor() {
+        try {
+            return BarColor.valueOf(config.getString("bossbar.color", "RED").toUpperCase(Locale.ROOT));
+        } catch (IllegalArgumentException ignored) {
+            return BarColor.RED;
+        }
+    }
+
+    private BarStyle readBossBarStyle() {
+        try {
+            return BarStyle.valueOf(config.getString("bossbar.style", "SOLID").toUpperCase(Locale.ROOT));
+        } catch (IllegalArgumentException ignored) {
+            return BarStyle.SOLID;
+        }
+    }
+
+    private double readBossBarProgress() {
+        double progress = config.getDouble("bossbar.progress", 1.0D);
+        if (!Double.isFinite(progress)) {
+            return 1.0D;
+        }
+        return Math.max(0.0D, Math.min(1.0D, progress));
+    }
+
+    private String colorizeLines(List<String> lines) {
+        if (lines == null || lines.isEmpty()) {
+            return "";
+        }
+
+        return String.join("\n", lines.stream()
+            .map(this::colorize)
+            .toList());
+    }
+
+    private String colorize(String text) {
+        return ChatColor.translateAlternateColorCodes('&', text == null ? "" : text);
+    }
+
+    private void save() {
+        try {
+            config.save(configFile);
+        } catch (IOException e) {
+            plugin.getLogger().warning("保存 maintenance.yml 失败: " + e.getMessage());
+        }
+    }
+}