ChomeNS/chomens-bot-java
3
1
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

some requests thing

Browse source
This commit is contained in:
Chayapak 2023-08-13 11:45:48 +07:00
parent 6167af7340
commit 9917a7eb1c
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/main/java/land/chipmunk/chayapak/chomens_bot/commands/UrbanCommand.java
View file

@ -5,6 +5,7 @@ import com.google.gson.JsonArray;
import com.google.gson.JsonElement; import com.google.gson.JsonElement;
import com.google.gson.JsonObject; import com.google.gson.JsonObject;
import land.chipmunk.chayapak.chomens_bot.Bot; 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.Command;
import land.chipmunk.chayapak.chomens_bot.command.CommandContext; import land.chipmunk.chayapak.chomens_bot.command.CommandContext;
import land.chipmunk.chayapak.chomens_bot.command.DiscordCommandContext; 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.URL;
import java.net.URLEncoder; import java.net.URLEncoder;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.util.concurrent.TimeUnit;
public class UrbanCommand extends Command { public class UrbanCommand extends Command {
public int requestsPer500MS = 0;
public UrbanCommand () { public UrbanCommand () {
super( super(
"urban", "urban",
@ -30,9 +34,13 @@ public class UrbanCommand extends Command {
TrustLevel.PUBLIC, TrustLevel.PUBLIC,
false false
); );
Main.executor.scheduleAtFixedRate(() -> requestsPer500MS = 0, 0, 500, TimeUnit.MILLISECONDS);
} }
public Component execute (CommandContext context, String[] args, String[] fullArgs) { 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 Bot bot = context.bot;
final boolean discord = context instanceof DiscordCommandContext; final boolean discord = context instanceof DiscordCommandContext;
@ -131,6 +139,8 @@ public class UrbanCommand extends Command {
} }
}); });
requestsPer500MS++;
return null; return null;
} }
} }