From 45966d1d1e56f54ad614f50a5c7d9c7574ff2e45 Mon Sep 17 00:00:00 2001
From: Nathan Adams <dinnerbone@dinnerbone.com>
Date: Mon, 13 Nov 2017 14:38:38 +0100
Subject: [PATCH] Changed custom suggestions to be nullable

---
 .../builder/RequiredArgumentBuilder.java       |  7 +------
 .../brigadier/tree/ArgumentCommandNode.java    | 18 +++++++++++++-----
 2 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/mojang/brigadier/builder/RequiredArgumentBuilder.java b/src/main/java/com/mojang/brigadier/builder/RequiredArgumentBuilder.java
index 694f256e..6e619d8b 100644
--- a/src/main/java/com/mojang/brigadier/builder/RequiredArgumentBuilder.java
+++ b/src/main/java/com/mojang/brigadier/builder/RequiredArgumentBuilder.java
@@ -5,11 +5,6 @@
 import com.mojang.brigadier.tree.ArgumentCommandNode;
 import com.mojang.brigadier.tree.CommandNode;
 
-import java.util.Collection;
-import java.util.Collections;
-import java.util.concurrent.CompletableFuture;
-import java.util.function.Function;
-
 public class RequiredArgumentBuilder<S, T> extends ArgumentBuilder<S, RequiredArgumentBuilder<S, T>> {
     private final String name;
     private final ArgumentType<T> type;
@@ -30,7 +25,7 @@ public RequiredArgumentBuilder<S, T> suggests(final CommandSuggestions.Provider<
     }
 
     public CommandSuggestions.Provider<S> getSuggestionsProvider() {
-        return suggestionsProvider == null ? type::listSuggestions : suggestionsProvider;
+        return suggestionsProvider;
     }
 
     @Override
diff --git a/src/main/java/com/mojang/brigadier/tree/ArgumentCommandNode.java b/src/main/java/com/mojang/brigadier/tree/ArgumentCommandNode.java
index 3fa97bc6..20f8a788 100644
--- a/src/main/java/com/mojang/brigadier/tree/ArgumentCommandNode.java
+++ b/src/main/java/com/mojang/brigadier/tree/ArgumentCommandNode.java
@@ -21,13 +21,13 @@ public class ArgumentCommandNode<S, T> extends CommandNode<S> {
 
     private final String name;
     private final ArgumentType<T> type;
-    private final CommandSuggestions.Provider<S> suggestionsProvider;
+    private final CommandSuggestions.Provider<S> customSuggestions;
 
-    public ArgumentCommandNode(final String name, final ArgumentType<T> type, final Command<S> command, final Predicate<S> requirement, final CommandNode<S> redirect, final RedirectModifier<S> modifier, final CommandSuggestions.Provider<S> suggestionsProvider) {
+    public ArgumentCommandNode(final String name, final ArgumentType<T> type, final Command<S> command, final Predicate<S> requirement, final CommandNode<S> redirect, final RedirectModifier<S> modifier, final CommandSuggestions.Provider<S> customSuggestions) {
         super(command, requirement, redirect, modifier);
         this.name = name;
         this.type = type;
-        this.suggestionsProvider = suggestionsProvider;
+        this.customSuggestions = customSuggestions;
     }
 
     public ArgumentType<T> getType() {
@@ -44,6 +44,10 @@ public String getUsageText() {
         return USAGE_ARGUMENT_OPEN + name + USAGE_ARGUMENT_CLOSE;
     }
 
+    public CommandSuggestions.Provider<S> getCustomSuggestions() {
+        return customSuggestions;
+    }
+
     @Override
     public void parse(final StringReader reader, final CommandContextBuilder<S> contextBuilder) throws CommandSyntaxException {
         final int start = reader.getCursor();
@@ -56,7 +60,11 @@ public void parse(final StringReader reader, final CommandContextBuilder<S> cont
 
     @Override
     public CompletableFuture<Collection<String>> listSuggestions(final CommandContext<S> context, final String command) {
-        return suggestionsProvider.getSuggestions(context, command);
+        if (customSuggestions == null) {
+            return type.listSuggestions(context, command);
+        } else {
+            return customSuggestions.getSuggestions(context, command);
+        }
     }
 
     @Override
@@ -64,7 +72,7 @@ public RequiredArgumentBuilder<S, T> createBuilder() {
         final RequiredArgumentBuilder<S, T> builder = RequiredArgumentBuilder.argument(name, type);
         builder.requires(getRequirement());
         builder.redirect(getRedirect(), getRedirectModifier());
-        builder.suggests(suggestionsProvider);
+        builder.suggests(customSuggestions);
         if (getCommand() != null) {
             builder.executes(getCommand());
         }