com.g2forge.alexandria.command.IStructuredCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ax-command Show documentation

Library for command line programming.

There is a newer version: 0.0.18

package com.g2forge.alexandria.command;

import java.io.InputStream;
import java.io.PrintStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.g2forge.alexandria.command.exit.Exit;
import com.g2forge.alexandria.command.exit.IExit;
import com.g2forge.alexandria.java.core.marker.ICommand;

public interface IStructuredCommand extends ICommand {
	public static class SubCommandBuilder {
		protected final Map map = new HashMap<>();

		public SubCommandBuilder add(IStandardCommand subcommand, String... names) {
			for (String name : names) {
				if (map.put(name, subcommand) != null) throw new IllegalArgumentException(String.format("Command named \"%1$s\" was already defined!", name));
			}
			return this;
		}

		public IStandardCommand build() {
			return invocation -> {
				final List arguments = invocation.getArguments();

				final String name = arguments.get(0);
				final IStandardCommand subcommand = map.get(name);
				if (subcommand == null) {
					invocation.io.getStandardError().println(String.format("Unrecognized command \"%1$s\"!", name));
					return FAIL;
				} else {
					final CommandInvocation subinvocation = new CommandInvocation<>(arguments.subList(1, arguments.size()), invocation.getIo(), invocation.getWorking());
					return subcommand.invoke(subinvocation);
				}
			};
		}
	}

	public static final IExit SUCCESS = new Exit(0);

	public static final IExit FAIL = new Exit(1);
}