skippyall
52 lines
2.2 KiB
Java
52 lines
2.2 KiB
Java
package io.github.skippyall.minions.command;
|
|
|
|
import com.mojang.brigadier.context.CommandContext;
|
|
import com.mojang.brigadier.exceptions.CommandSyntaxException;
|
|
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
|
|
import com.mojang.brigadier.suggestion.SuggestionProvider;
|
|
import com.mojang.brigadier.suggestion.Suggestions;
|
|
import com.mojang.brigadier.suggestion.SuggestionsBuilder;
|
|
import io.github.skippyall.minions.minion.MinionData;
|
|
import io.github.skippyall.minions.minion.MinionPersistentState;
|
|
import io.github.skippyall.minions.minion.MinionProfileUtils;
|
|
import net.minecraft.server.MinecraftServer;
|
|
import net.minecraft.server.command.ServerCommandSource;
|
|
import net.minecraft.text.Text;
|
|
|
|
import java.util.Optional;
|
|
import java.util.UUID;
|
|
import java.util.concurrent.CompletableFuture;
|
|
|
|
public class MinionArgument {
|
|
public static final SimpleCommandExceptionType MINION_NOT_PRESENT = new SimpleCommandExceptionType(Text.translatable("minions.command.minion.not_present"));
|
|
|
|
public static final MinionSuggestionProvider SUGGESTION_PROVIDER = new MinionSuggestionProvider();
|
|
|
|
public static MinionData parse(MinecraftServer server, String argument) throws CommandSyntaxException {
|
|
Optional<MinionData> data = Optional.empty();
|
|
if(argument.startsWith(MinionProfileUtils.PREFIX)) {
|
|
data = MinionPersistentState.get(server).getMinionWithName(argument);
|
|
} else {
|
|
try {
|
|
data = Optional.ofNullable(MinionPersistentState.get(server).getMinionData(UUID.fromString(argument)));
|
|
} catch (IllegalArgumentException ignored) {}
|
|
}
|
|
|
|
if(data.isEmpty()) {
|
|
throw MINION_NOT_PRESENT.create();
|
|
}
|
|
return data.get();
|
|
}
|
|
|
|
public static class MinionSuggestionProvider implements SuggestionProvider<ServerCommandSource> {
|
|
@Override
|
|
public CompletableFuture<Suggestions> getSuggestions(CommandContext<ServerCommandSource> context, SuggestionsBuilder builder) throws CommandSyntaxException {
|
|
for (MinionData data : MinionPersistentState.get(context.getSource().getServer()).getMinionDataList()) {
|
|
builder.suggest(data.name());
|
|
}
|
|
|
|
return builder.buildFuture();
|
|
}
|
|
}
|
|
}
|