package net.minecraft.gametest.framework; import com.mojang.brigadier.Message; 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 java.util.Arrays; import java.util.Collection; import java.util.Optional; import java.util.concurrent.CompletableFuture; import java.util.stream.Stream; import net.minecraft.commands.CommandSourceStack; import net.minecraft.commands.SharedSuggestionProvider; import net.minecraft.network.chat.Component; public class TestFunctionArgument implements ArgumentType { private static final Collection EXAMPLES = Arrays.asList("techtests.piston", "techtests"); public TestFunction parse(StringReader reader) throws CommandSyntaxException { String string = reader.readUnquotedString(); Optional optional = GameTestRegistry.findTestFunction(string); if (optional.isPresent()) { return (TestFunction)optional.get(); } else { Message message = Component.literal("No such test: " + string); throw new CommandSyntaxException(new SimpleCommandExceptionType(message), message); } } public static TestFunctionArgument testFunctionArgument() { return new TestFunctionArgument(); } public static TestFunction getTestFunction(CommandContext context, String argument) { return context.getArgument(argument, TestFunction.class); } @Override public CompletableFuture listSuggestions(CommandContext commandContext, SuggestionsBuilder suggestionsBuilder) { return suggestTestFunction(commandContext, suggestionsBuilder); } public static CompletableFuture suggestTestFunction(CommandContext context, SuggestionsBuilder builder) { Stream stream = GameTestRegistry.getAllTestFunctions().stream().map(TestFunction::testName); return SharedSuggestionProvider.suggest(stream, builder); } @Override public Collection getExamples() { return EXAMPLES; } }