change to land.chipmunk.chayapak because yes
This commit is contained in:
@@ -1,161 +0,0 @@
|
||||
package me.chayapak1.chomens_bot.plugins;
|
||||
|
||||
import lombok.Getter;
|
||||
import me.chayapak1.chomens_bot.command.Command;
|
||||
import me.chayapak1.chomens_bot.command.CommandContext;
|
||||
import me.chayapak1.chomens_bot.commands.*;
|
||||
import me.chayapak1.chomens_bot.util.ElementUtilities;
|
||||
import net.dv8tion.jda.api.entities.Role;
|
||||
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
|
||||
import net.kyori.adventure.text.Component;
|
||||
import net.kyori.adventure.text.event.HoverEvent;
|
||||
import net.kyori.adventure.text.format.NamedTextColor;
|
||||
import org.apache.commons.lang3.exception.ExceptionUtils;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
|
||||
public class CommandHandlerPlugin {
|
||||
@Getter private final List<Command> commands = new ArrayList<>();
|
||||
|
||||
public CommandHandlerPlugin () {
|
||||
registerCommand(new CommandBlockCommand());
|
||||
registerCommand(new CowsayCommand());
|
||||
registerCommand(new EchoCommand());
|
||||
registerCommand(new CreatorCommand());
|
||||
registerCommand(new DiscordCommand());
|
||||
registerCommand(new HelpCommand());
|
||||
registerCommand(new TestCommand());
|
||||
registerCommand(new ThrowCommand());
|
||||
registerCommand(new ValidateCommand());
|
||||
registerCommand(new MusicCommand());
|
||||
registerCommand(new RandomTeleportCommand());
|
||||
registerCommand(new BotVisibilityCommand());
|
||||
registerCommand(new TPSBarCommand());
|
||||
registerCommand(new NetMessageCommand());
|
||||
registerCommand(new RefillCoreCommand());
|
||||
registerCommand(new WikipediaCommand());
|
||||
registerCommand(new UrbanCommand());
|
||||
registerCommand(new ClearChatCommand());
|
||||
registerCommand(new ListCommand());
|
||||
registerCommand(new ServerEvalCommand());
|
||||
registerCommand(new UUIDCommand());
|
||||
registerCommand(new TimeCommand());
|
||||
}
|
||||
|
||||
public void registerCommand (Command command) {
|
||||
commands.add(command);
|
||||
}
|
||||
|
||||
public Component executeCommand (String input, CommandContext context, boolean discord, String hash, String ownerHash, MessageReceivedEvent event) {
|
||||
final String[] splitInput = input.split("\\s+");
|
||||
|
||||
final String commandName = splitInput[0];
|
||||
|
||||
final Command command = ElementUtilities.findCommand(commands, commandName);
|
||||
|
||||
if (command == null) return Component.text("Unknown command: " + commandName).color(NamedTextColor.RED);
|
||||
|
||||
final int trustLevel = command.trustLevel();
|
||||
|
||||
final String[] fullArgs = Arrays.copyOfRange(splitInput, 1, splitInput.length);
|
||||
final int longestUsageIndex = getLongestUsageIndex(command.usage());
|
||||
final String usage = command.usage().get(longestUsageIndex);
|
||||
final int minimumArgs = getMinimumArgs(usage, discord, command.trustLevel());
|
||||
final int maximumArgs = getMaximumArgs(usage, discord, command.trustLevel());
|
||||
if (fullArgs.length < minimumArgs) return Component.text("Excepted minimum of " + minimumArgs + " argument(s), got " + fullArgs.length).color(NamedTextColor.RED);
|
||||
if (fullArgs.length > maximumArgs && !usage.contains("{")) return Component.text("Too much arguments, expected " + maximumArgs + " max").color(NamedTextColor.RED);
|
||||
|
||||
String userHash = "";
|
||||
if (trustLevel > 0 && splitInput.length >= 2) userHash = splitInput[1];
|
||||
|
||||
final String[] args = Arrays.copyOfRange(splitInput, (trustLevel > 0 && !discord) ? 2 : 1, splitInput.length);
|
||||
|
||||
// fix shit random messy code
|
||||
if (command.trustLevel() > 0) {
|
||||
if (discord) {
|
||||
final List<Role> roles = event.getMember().getRoles();
|
||||
|
||||
if (
|
||||
command.trustLevel() == 1 &&
|
||||
roles.stream().noneMatch(role -> role.getName().equalsIgnoreCase("Trusted")) &&
|
||||
roles.stream().noneMatch(role -> role.getName().equalsIgnoreCase("Host"))
|
||||
) return Component.text("You're not in the trusted role!").color(NamedTextColor.RED);
|
||||
|
||||
if (
|
||||
command.trustLevel() == 2 &&
|
||||
roles.stream().noneMatch(role -> role.getName().equalsIgnoreCase("Host"))
|
||||
) return Component.text("You're not in the host role!").color(NamedTextColor.RED);
|
||||
} else {
|
||||
if (
|
||||
command.trustLevel() == 1 &&
|
||||
!userHash.equals(hash) &&
|
||||
!userHash.equals(ownerHash)
|
||||
) return Component.text("Invalid hash").color(NamedTextColor.RED);
|
||||
|
||||
if (
|
||||
command.trustLevel() == 2 &&
|
||||
!userHash.equals(ownerHash)
|
||||
) return Component.text("Invalid OwnerHash").color(NamedTextColor.RED);
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
return command.execute(context, args, fullArgs);
|
||||
} catch (Exception exception) {
|
||||
exception.printStackTrace();
|
||||
|
||||
final String stackTrace = ExceptionUtils.getStackTrace(exception);
|
||||
if (!discord) {
|
||||
return Component
|
||||
.text("An error occurred while trying to execute the command, hover here for more details", NamedTextColor.RED)
|
||||
.hoverEvent(
|
||||
HoverEvent.showText(
|
||||
Component
|
||||
.text(stackTrace)
|
||||
.color(NamedTextColor.RED)
|
||||
)
|
||||
);
|
||||
} else {
|
||||
return Component.text(stackTrace).color(NamedTextColor.RED);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private static int getLongestUsageIndex(List<String> usages) {
|
||||
int longestIndex = 0;
|
||||
int maxLength = 0;
|
||||
for (int i = 0; i < usages.size(); i++) {
|
||||
String[] args = usages.get(i).split("\\s+");
|
||||
if (args.length > maxLength) {
|
||||
longestIndex = i;
|
||||
maxLength = args.length;
|
||||
}
|
||||
}
|
||||
return longestIndex;
|
||||
}
|
||||
|
||||
private static int getMinimumArgs(String usage, boolean discord, int trustLevel) {
|
||||
int count = 0;
|
||||
for (int i = 0; i < usage.length(); i++) {
|
||||
if (usage.charAt(i) == '<') {
|
||||
count++;
|
||||
}
|
||||
}
|
||||
if (usage.contains("<hash>")) count--; // bad fix?
|
||||
if ((discord && trustLevel > 0)) count--;
|
||||
return count;
|
||||
}
|
||||
|
||||
private static int getMaximumArgs(String usage, boolean discord, int trustLevel) {
|
||||
int count = 0;
|
||||
for (int i = 0; i < usage.length(); i++) {
|
||||
if (usage.charAt(i) == '<' || usage.charAt(i) == '[') {
|
||||
count++;
|
||||
}
|
||||
}
|
||||
if (discord && trustLevel > 0) count++;
|
||||
return count;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user