From 9917a7eb1c980982d35b2e4df849fcf986a541bb Mon Sep 17 00:00:00 2001
From: ChomeNS <95471003+ChomeNS@users.noreply.github.com>
Date: Sun, 13 Aug 2023 11:45:48 +0700
Subject: [PATCH] some requests thing

---
 .../chayapak/chomens_bot/commands/UrbanCommand.java    | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/main/java/land/chipmunk/chayapak/chomens_bot/commands/UrbanCommand.java b/src/main/java/land/chipmunk/chayapak/chomens_bot/commands/UrbanCommand.java
index 61a696f..82aab53 100644
--- a/src/main/java/land/chipmunk/chayapak/chomens_bot/commands/UrbanCommand.java
+++ b/src/main/java/land/chipmunk/chayapak/chomens_bot/commands/UrbanCommand.java
@@ -5,6 +5,7 @@ import com.google.gson.JsonArray;
 import com.google.gson.JsonElement;
 import com.google.gson.JsonObject;
 import land.chipmunk.chayapak.chomens_bot.Bot;
+import land.chipmunk.chayapak.chomens_bot.Main;
 import land.chipmunk.chayapak.chomens_bot.command.Command;
 import land.chipmunk.chayapak.chomens_bot.command.CommandContext;
 import land.chipmunk.chayapak.chomens_bot.command.DiscordCommandContext;
@@ -19,8 +20,11 @@ import net.kyori.adventure.text.format.TextDecoration;
 import java.net.URL;
 import java.net.URLEncoder;
 import java.nio.charset.StandardCharsets;
+import java.util.concurrent.TimeUnit;
 
 public class UrbanCommand extends Command {
+    public int requestsPer500MS = 0;
+
     public UrbanCommand () {
         super(
                 "urban",
@@ -30,9 +34,13 @@ public class UrbanCommand extends Command {
                 TrustLevel.PUBLIC,
                 false
         );
+
+        Main.executor.scheduleAtFixedRate(() -> requestsPer500MS = 0, 0, 500, TimeUnit.MILLISECONDS);
     }
 
     public Component execute (CommandContext context, String[] args, String[] fullArgs) {
+        if (requestsPer500MS > 10) return Component.text("Too many requests").color(NamedTextColor.RED);
+
         final Bot bot = context.bot;
 
         final boolean discord = context instanceof DiscordCommandContext;
@@ -131,6 +139,8 @@ public class UrbanCommand extends Command {
             }
         });
 
+        requestsPer500MS++;
+
         return null;
     }
 }