[grid] Allow cli commands to be hidden

SeleniumHQ · Jan 16, 2020 · d5cf59a · d5cf59a
1 parent cc37627
commit d5cf59a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/java/server/src/org/openqa/selenium/cli/CliCommand.java b/java/server/src/org/openqa/selenium/cli/CliCommand.java
@@ -25,6 +25,10 @@ public interface CliCommand {
 
   Executable configure(String... args);
 
+  default boolean isShown() {
+    return true;
+  }
+
   interface Executable {
 
     void run() throws Exception;

diff --git a/java/server/src/org/openqa/selenium/grid/Main.java b/java/server/src/org/openqa/selenium/grid/Main.java
@@ -146,6 +146,7 @@ public String getDescription() {
     public Executable configure(String... args) {
       return () -> {
         int longest = commands.stream()
+                          .filter(CliCommand::isShown)
                           .map(CliCommand::getName)
                           .max(Comparator.comparingInt(String::length))
                           .map(String::length)
@@ -159,7 +160,9 @@ public Executable configure(String... args) {
         String format = "  %-" + longest + "s";
 
         PrintWriter indented = new WrappedPrintWriter(System.out, 72, indent);
-        commands.forEach(cmd -> indented.format(format, cmd.getName())
+        commands.stream()
+          .filter(CliCommand::isShown)
+          .forEach(cmd -> indented.format(format, cmd.getName())
             .append(cmd.getDescription())
             .append("\n"));