// Licensed to the Software Freedom Conservancy (SFC) under one

// or more contributor license agreements. See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership. The SFC licenses this file

// to you under the Apache License, Version 2.0 (the

// "License"); you may not use this file except in compliance

// with the License. You may obtain a copy of the License at

//

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing,

// software distributed under the License is distributed on an

// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

// KIND, either express or implied. See the License for the

// specific language governing permissions and limitations

// under the License.

package org.openqa.selenium.grid;

import org.openqa.selenium.cli.CliCommand;

import org.openqa.selenium.cli.WrappedPrintWriter;

import org.openqa.selenium.grid.config.Role;

import java.io.File;

import java.io.PrintStream;

import java.io.PrintWriter;

import java.util.Collections;

import java.util.Comparator;

import java.util.ServiceLoader;

import java.util.Set;

import java.util.TreeSet;

import static java.util.Comparator.comparing;

public class Main {

private final PrintStream out;

private final PrintStream err;

private final String[] args;

public static void main(String[] args) {

new Main(System.out, System.err, args).go();

}

Main(PrintStream out, PrintStream err, String[] args) {

// It's not private to make it visible for tests

this.out = out;

this.err = err;

this.args = args;

}

void go() {

// It's not private to make it visible for tests

if (args.length == 0) {

showHelp(Main.class.getClassLoader());

} else {

launch(args, Main.class.getClassLoader());

}

}

private static Set<CliCommand> loadCommands(ClassLoader loader) {

Set<CliCommand> commands = new TreeSet<>(comparing(CliCommand::getName));

ServiceLoader.load(CliCommand.class, loader).forEach(commands::add);

return commands;

}

private void showHelp(ClassLoader loader) {

new Help(loadCommands(loader)).configure(out, err).run();

}

private void launch(String[] args, ClassLoader loader) {

String commandName = args[0];

String[] remainingArgs = new String[args.length - 1];

System.arraycopy(args, 1, remainingArgs, 0, args.length - 1);

Set<CliCommand> commands = loadCommands(loader);

CliCommand command = commands.parallelStream()

.filter(cmd -> commandName.equals(cmd.getName()))

.findFirst()

.orElse(new Help(commands));

command.configure(out, err, remainingArgs).run();

}

private static class Help implements CliCommand {

private final Set<CliCommand> commands;

public Help(Set<CliCommand> commands) {

this.commands = commands;

}

@Override

public String getName() {

return "Selenium Server commands";

}

@Override

public Set<Role> getConfigurableRoles() {

return Collections.emptySet();

}

@Override

public Set<Object> getFlagObjects() {

return Collections.emptySet();

}

@Override

public String getDescription() {

return "A list of all the commands available. To use one, run `java -jar selenium.jar " +

"commandName`.";

}

@Override

public Executable configure(PrintStream out, PrintStream err, String... args) {

return () -> {

int longest = commands.stream()

.filter(CliCommand::isShown)

.map(CliCommand::getName)

.max(Comparator.comparingInt(String::length))

.map(String::length)

.orElse(0) + 2; // two space padding either side

PrintWriter outWriter = new WrappedPrintWriter(out, 72, 0);

outWriter.append(getName()).append("\n\n");

outWriter.append(getDescription()).append("\n").append("\n");

int indent = Math.min(longest + 2, 25);

String format = " %-" + longest + "s";

PrintWriter indented = new WrappedPrintWriter(out, 72, indent);

commands.stream()

.filter(CliCommand::isShown)

.forEach(cmd -> indented.format(format, cmd.getName())

.append(cmd.getDescription())

.append("\n"));

outWriter.write("\nFor each command, run with `--help` for command-specific help\n");

outWriter.write("\nUse the `--ext` flag before the command name to specify an additional " +

"classpath to use with the server (for example, to provide additional " +

"commands, or to provide additional driver implementations). For example:\n");

outWriter.write(String.format(

"%n java -jar selenium.jar --ext example.jar%sdir standalone --port 1234",

File.pathSeparator));

out.println("\n");

};

}

}

}