From 7d4dcc3e4bb35c12bf53c5b58025237fa95b321b Mon Sep 17 00:00:00 2001
From: Nathan Adams <dinnerbone@dinnerbone.com>
Date: Wed, 28 Jun 2017 11:08:26 +0200
Subject: [PATCH] Allow changing of a source after creation of a contextbuilder

---
 .../mojang/brigadier/context/CommandContextBuilder.java    | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java b/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java
index 066451e..3ed4aa9 100644
--- a/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java
+++ b/src/main/java/com/mojang/brigadier/context/CommandContextBuilder.java
@@ -9,13 +9,18 @@ import java.util.Map;
 public class CommandContextBuilder<S> {
     private final Map<String, ParsedArgument<?>> arguments = Maps.newHashMap();
     private final Map<CommandNode<S>, String> nodes = Maps.newLinkedHashMap();
-    private final S source;
+    private S source;
     private Command<S> command;
 
     public CommandContextBuilder(S source) {
         this.source = source;
     }
 
+    public CommandContextBuilder<S> withSource(S source) {
+        this.source = source;
+        return this;
+    }
+
     public S getSource() {
         return source;
     }