Tweak

1 files changed, 38 insertions, 0 deletions

diff --git a/JPratt/src/main/java/bjc/pratt/commands/BranchInitialCommand.java b/JPratt/src/main/java/bjc/pratt/commands/BranchInitialCommand.java
new file mode 100644
index 0000000..3f3093e
--- /dev/null
+++ b/JPratt/src/main/java/bjc/pratt/commands/BranchInitialCommand.java
@@ -0,0 +1,38 @@
+package bjc.pratt.commands;
+
+import java.util.Map;
+
+import bjc.pratt.ParserContext;
+import bjc.pratt.tokens.Token;
+import bjc.utils.parserutils.ParserException;
+
+/**
+ * Represents a initial command that has a number of 'sub-commands' in the way that Go/Git CLI does.
+ * 
+ * @author bjcul
+ *
+ * @param <K> Token key type
+ * @param <V> Token value type
+ * @param <C> Parser context type
+ */
+public class BranchInitialCommand<K, V, C> implements InitialCommand<K, V, C> {
+	private Map<K, InitialCommand<K, V, C>> comMap;
+	
+	/**
+	 * Create a new branch initial command
+	 * 
+	 * @param mep The map containing the commands
+	 */
+	public BranchInitialCommand(Map<K, InitialCommand<K, V, C>> mep) {
+		this.comMap = mep;
+	}
+	
+	@Override
+	public CommandResult<K, V> denote(Token<K, V> operator, ParserContext<K, V, C> ctx) throws ParserException {
+		Token<K, V> curToken = ctx.tokens.current();
+		ctx.tokens.expect(comMap.keySet());
+		
+		return comMap.get(curToken.getKey()).denote(curToken, ctx);
+	}
+
+}