io.github.xfournet.jconfig.cli.JConfigCli Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jconfig-cli Show documentation
JConfig CLI
There is a newer version: 0.8.0
package io.github.xfournet.jconfig.cli;

import java.nio.file.Paths;
import java.util.*;
import com.beust.jcommander.JCommander;
import com.beust.jcommander.ParameterException;
import com.beust.jcommander.Parameters;
import io.github.xfournet.jconfig.JConfig;
import io.github.xfournet.jconfig.cli.command.ApplyCommand;
import io.github.xfournet.jconfig.cli.command.DiffCommand;
import io.github.xfournet.jconfig.cli.command.FilterCommand;
import io.github.xfournet.jconfig.cli.command.HelpCommand;
import io.github.xfournet.jconfig.cli.command.MergeCommand;
import io.github.xfournet.jconfig.cli.command.RemoveCommand;
import io.github.xfournet.jconfig.cli.command.SetCommand;

import static io.github.xfournet.jconfig.JConfigBuilder.jConfigBuilder;

@SuppressWarnings("WeakerAccess")
public class JConfigCli {

    private final String m_programName;
    private final List m_commands;
    private final JConfig m_jConfig;

    public JConfigCli(String programName, List commands, JConfig jConfig) {
        m_programName = programName;
        m_commands = commands;
        m_jConfig = jConfig;
    }

    public boolean run(String[] args) {
        Map commandTable = new LinkedHashMap<>();

        for (Command command : m_commands) {
            addCommand(commandTable, command);
        }

        HelpCommand helpCommand = new HelpCommand();
        addCommand(commandTable, helpCommand);

        JCommander jc = new JCommander();
        jc.setCaseSensitiveOptions(false);
        jc.setColumnSize(160);
        jc.setProgramName(m_programName);
        commandTable.values().forEach(jc::addCommand);

        String error = null;
        try {
            jc.parse(args);
        } catch (ParameterException e) {
            error = e.getMessage();
        }

        String parsedCommand = jc.getParsedCommand();
        if (parsedCommand != null) {
            parsedCommand = parsedCommand.toLowerCase(Locale.ROOT);
        }

        Command command = commandTable.get(parsedCommand);
        if (command == null) {
            command = helpCommand;
        }

        if (error != null) {
            System.err.printf("Error: %s%n", error);
            if (!helpCommand.equals(command)) {
                return false;
            }
        }

        try {
            command.execute(new CommandContextImpl(jc, m_jConfig));
        } catch (JConfigException e) {
            System.err.printf("%s: %s%n", m_programName, e.getMessage());
            return false;
        }

        return error == null;
    }

    private void addCommand(Map commandTable, Command command) {
        Parameters parametersAnnotation = command.getClass().getAnnotation(Parameters.class);
        if (parametersAnnotation == null) {
            throw new IllegalArgumentException("Command has no @Parameters annotation: " + command);
        }

        for (String commandName : parametersAnnotation.commandNames()) {
            commandName = commandName.toLowerCase(Locale.ROOT);
            Command existingCommand = commandTable.put(commandName, command);
            if (existingCommand != null) {
                throw new IllegalStateException("Two commands are registered with same name: " + commandName);
            }
        }
    }

    private final class CommandContextImpl implements CommandContext {
        private final JCommander m_jCommander;
        private final JConfig m_jConfig;

        CommandContextImpl(JCommander jCommander, JConfig jConfig) {
            m_jCommander = jCommander;
            m_jConfig = jConfig;
        }

        @Override
        public JCommander getJCommander() {
            return m_jCommander;
        }

        @Override
        public JConfig getJConfig() {
            return m_jConfig;
        }
    }

    public static void main(String[] args) {
        JConfig jConfig = jConfigBuilder().build(Paths.get(""));
        if (!new JConfigCli("jconfig", defaultCommands(), jConfig).run(args)) {
            System.exit(1);
        }
    }

    public static List defaultCommands() {
        return Arrays.asList(new ApplyCommand(), new DiffCommand(), new FilterCommand(), new MergeCommand(), new RemoveCommand(), new SetCommand());
    }
}