Add ratelimit for /hub command

Author: UserNugget

src/main/java/net/elytrium/velocitytools/Settings.java

@@ -81,6 +81,11 @@ public static class COMMANDS {
     public SEND SEND = new SEND();
     public HUB HUB = new HUB();
 
+    public String RATELIMITED = "&cPlease wait before next usage!";
+
+    @Comment(@CommentValue("Ratelimit delay in milliseconds"))
+    public int RATELIMIT_DELAY = 1000;
+
     public static class ALERT {
 
       public boolean ENABLED = true;

src/main/java/net/elytrium/velocitytools/VelocityTools.java

@@ -24,6 +24,8 @@
 import com.velocitypowered.api.plugin.annotation.DataDirectory;
 import com.velocitypowered.api.proxy.ProxyServer;
 import com.velocitypowered.proxy.protocol.StateRegistry;
+import com.velocitypowered.proxy.util.ratelimit.Ratelimiter;
+import com.velocitypowered.proxy.util.ratelimit.Ratelimiters;
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import java.io.IOException;
 import java.nio.file.Files;
@@ -66,6 +68,8 @@ public class VelocityTools {
   private static Logger LOGGER;
   @MonotonicNonNull
   private static Serializer SERIALIZER;
+  @MonotonicNonNull
+  private static Ratelimiter RATELIMITER;
 
   private final ProxyServer server;
   private final Path dataDirectory;
@@ -137,6 +141,8 @@ public void reload() {
       setSerializer(new Serializer(serializer));
     }
 
+    setRatelimiter(Ratelimiters.createWithMilliseconds(Settings.IMP.COMMANDS.RATELIMIT_DELAY));
+
     List<String> aliases = Settings.IMP.COMMANDS.HUB.ALIASES;
     aliases.forEach(alias -> this.server.getCommandManager().unregister(alias));
     if (Settings.IMP.COMMANDS.HUB.ENABLED && !aliases.isEmpty()) {
@@ -188,11 +194,19 @@ private static void setSerializer(Serializer serializer) {
     SERIALIZER = serializer;
   }
 
+  private static void setRatelimiter(Ratelimiter ratelimiter) {
+    RATELIMITER = ratelimiter;
+  }
+
   public static Logger getLogger() {
     return LOGGER;
   }
 
   public static Serializer getSerializer() {
     return SERIALIZER;
   }
+
+  public static Ratelimiter getRatelimiter() {
+    return RATELIMITER;
+  }
 }

src/main/java/net/elytrium/velocitytools/commands/HubCommand.java

@@ -28,7 +28,7 @@
 import net.elytrium.velocitytools.VelocityTools;
 import net.kyori.adventure.text.Component;
 
-public class HubCommand implements SimpleCommand {
+public class HubCommand extends RatelimitedCommand {
 
   private final ProxyServer server;
   private final List<String> servers;
@@ -49,15 +49,12 @@ public HubCommand(ProxyServer server) {
   }
 
   @Override
-  public void execute(SimpleCommand.Invocation invocation) {
-    CommandSource source = invocation.source();
-    if (!(source instanceof Player)) {
+  protected void execute(CommandSource source, String[] args) {
+    if (!(source instanceof Player player)) {
       source.sendMessage(CommandMessages.PLAYERS_ONLY);
       return;
     }
 
-    Player player = (Player) source;
-
     String serverName;
     int serversSize = this.servers.size();
     if (serversSize > 1) {

src/main/java/net/elytrium/velocitytools/commands/RatelimitedCommand.java

@@ -0,0 +1,47 @@
+/*
+ * Copyright (C) 2021 - 2023 Elytrium
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package net.elytrium.velocitytools.commands;
+
+import com.velocitypowered.api.command.CommandSource;
+import com.velocitypowered.api.command.SimpleCommand;
+import com.velocitypowered.api.proxy.Player;
+import net.elytrium.velocitytools.Settings;
+import net.elytrium.velocitytools.VelocityTools;
+import net.kyori.adventure.text.Component;
+
+public abstract class RatelimitedCommand implements SimpleCommand {
+
+  private final Component ratelimited;
+
+  public RatelimitedCommand() {
+    this.ratelimited = VelocityTools.getSerializer().deserialize(Settings.IMP.COMMANDS.RATELIMITED);
+  }
+
+  @Override
+  public final void execute(SimpleCommand.Invocation invocation) {
+    CommandSource source = invocation.source();
+    if (source instanceof Player player && !VelocityTools.getRatelimiter().attempt(player.getRemoteAddress().getAddress())) {
+      source.sendMessage(this.ratelimited);
+      return;
+    }
+
+    this.execute(source, invocation.arguments());
+  }
+
+  protected abstract void execute(CommandSource source, String[] args);
+}