73 lines
3.1 KiB
Java
73 lines
3.1 KiB
Java
/*
|
|
*
|
|
* MIT License
|
|
*
|
|
* Copyright (c) 2020 TerraForged
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*/
|
|
|
|
package com.terraforged.mod.command.arg;
|
|
|
|
import com.mojang.brigadier.StringReader;
|
|
import com.mojang.brigadier.arguments.ArgumentType;
|
|
import com.mojang.brigadier.context.CommandContext;
|
|
import com.mojang.brigadier.exceptions.CommandSyntaxException;
|
|
import com.mojang.brigadier.exceptions.SimpleCommandExceptionType;
|
|
import com.mojang.brigadier.suggestion.Suggestions;
|
|
import com.mojang.brigadier.suggestion.SuggestionsBuilder;
|
|
import net.minecraft.command.ISuggestionProvider;
|
|
import net.minecraft.util.ResourceLocation;
|
|
import net.minecraft.util.registry.Registry;
|
|
import net.minecraft.util.text.StringTextComponent;
|
|
import net.minecraft.world.biome.Biome;
|
|
import net.minecraftforge.registries.ForgeRegistries;
|
|
|
|
import java.util.concurrent.CompletableFuture;
|
|
|
|
public class BiomeArgType implements ArgumentType<Biome> {
|
|
|
|
@Override
|
|
public Biome parse(StringReader reader) throws CommandSyntaxException {
|
|
ResourceLocation resourcelocation = ResourceLocation.read(reader);
|
|
return Registry.BIOME.getValue(resourcelocation)
|
|
.orElseThrow(() -> createException("Invalid biome", "%s is not a valid biome", resourcelocation));
|
|
}
|
|
|
|
@Override
|
|
public <S> CompletableFuture<Suggestions> listSuggestions(CommandContext<S> context, SuggestionsBuilder suggestions) {
|
|
return ISuggestionProvider.func_212476_a(ForgeRegistries.BIOMES.getValues().stream().map(Biome::getRegistryName), suggestions);
|
|
}
|
|
|
|
private static CommandSyntaxException createException(String type, String message, Object... args) {
|
|
return new CommandSyntaxException(
|
|
new SimpleCommandExceptionType(new StringTextComponent(type)),
|
|
new StringTextComponent(String.format(message, args))
|
|
);
|
|
}
|
|
|
|
public static ArgumentType<Biome> biome() {
|
|
return new BiomeArgType();
|
|
}
|
|
|
|
public static <S> Biome getBiome(CommandContext<S> context, String name) {
|
|
return context.getArgument(name, Biome.class);
|
|
}
|
|
}
|