Fixed commands with a trailing whitespace

SoniEx2 · Nov 8, 2017 · 6e61fef · 6e61fef
1 parent 2c39925
commit 6e61fef
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/build.gradle b/build.gradle
@@ -3,7 +3,7 @@ import groovy.io.FileType
 apply plugin: 'java-library'
 apply plugin: 'maven'
 
-version = '0.1.1'
+version = '0.1.2'
 group = 'com.mojang'
 
 task wrapper(type: Wrapper) {

diff --git a/src/main/java/com/mojang/brigadier/CommandDispatcher.java b/src/main/java/com/mojang/brigadier/CommandDispatcher.java
@@ -153,7 +153,9 @@ private ParseResults<S> parseNodes(final CommandNode<S> node, final StringReader
 
             context.withCommand(child.getCommand());
             if (reader.canRead()) {
-                reader.skip();
+                if (reader.canRead(2)) {
+                    reader.skip();
+                }
                 if (child.getRedirect() != null) {
                     final CommandContextBuilder<S> childContext = new CommandContextBuilder<>(this, source, reader.getCursor());
                     childContext.withNode(child.getRedirect(), new StringRange(reader.getCursor(), reader.getCursor()));

diff --git a/src/test/java/com/mojang/brigadier/CommandDispatcherTest.java b/src/test/java/com/mojang/brigadier/CommandDispatcherTest.java
@@ -173,6 +173,16 @@ public void testExecuteSubcommand() throws Exception {
         verify(subCommand).run(any(CommandContext.class));
     }
 
+    @SuppressWarnings("unchecked")
+    @Test
+    public void testParseIncomplete() throws Exception {
+        subject.register(literal("foo").then(literal("bar").executes(command)));
+
+        final ParseResults<Object> parse = subject.parse("foo ", source);
+        assertThat(parse.getReader().getRemaining(), equalTo(" "));
+        assertThat(parse.getContext().getNodes().size(), is(1));
+    }
+
     @SuppressWarnings("unchecked")
     @Test
     public void testExecuteAmbiguiousParentSubcommand() throws Exception {