From fa6cea35cb6899f802d24951bf5e12639a251eb6 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Sat, 18 Nov 2017 15:16:05 +0100
Subject: Finish tree parsing of CLI args.

---
 .../crcs/ectester/common/cli/TreeCommandLine.java  | 23 ++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'src/cz/crcs/ectester/common/cli/TreeCommandLine.java')

diff --git a/src/cz/crcs/ectester/common/cli/TreeCommandLine.java b/src/cz/crcs/ectester/common/cli/TreeCommandLine.java
index ef6079e..e0927fa 100644
--- a/src/cz/crcs/ectester/common/cli/TreeCommandLine.java
+++ b/src/cz/crcs/ectester/common/cli/TreeCommandLine.java
@@ -5,6 +5,8 @@ import org.apache.commons.cli.Option;
 import org.apache.commons.cli.ParseException;
 
 import java.util.Iterator;
+import java.util.List;
+import java.util.Objects;
 import java.util.Properties;
 import java.util.function.BiFunction;
 
@@ -45,6 +47,10 @@ public class TreeCommandLine extends CommandLine {
         return next;
     }
 
+    public boolean isNext(String next) {
+        return Objects.equals(getNextName(), next);
+    }
+
     public CommandLine getThis() {
         return cli;
     }
@@ -131,4 +137,21 @@ public class TreeCommandLine extends CommandLine {
     public Option[] getOptions() {
         return cli.getOptions();
     }
+
+    public String getArg(int index) {
+        if (index < 0 || index >= cli.getArgs().length) {
+            return null;
+        }
+        return cli.getArgs()[index];
+    }
+
+    @Override
+    public String[] getArgs() {
+        return cli.getArgs();
+    }
+
+    @Override
+    public List<String> getArgList() {
+        return cli.getArgList();
+    }
 }
-- 
cgit v1.2.3-70-g09d2