1 files changed, 23 insertions, 0 deletions

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();
+    }
 }