private boolean doesCommandExist(String commandName, int arity) {
for (ShellCommand cmd : commandTable) {
if (cmd.canBeDenotedBy(commandName) && cmd.getArity() == arity) {
return true;
}
}
return false;
}
public List<ShellCommand> commandsByName(String discriminator) {
List<ShellCommand> collectedTable = new ArrayList<ShellCommand>();
// collection
for (ShellCommand cs : commandTable) {
if (cs.canBeDenotedBy(discriminator)) {
collectedTable.add(cs);
}
}
return collectedTable;
}
public ShellCommand lookupCommand(String discriminator, List<Token> tokens) throws CLIException {
List<ShellCommand> collectedTable = commandsByName(discriminator);
// reduction
List<ShellCommand> reducedTable = new ArrayList<ShellCommand>();
for (ShellCommand cs : collectedTable) {
if (cs.getMethod().getParameterTypes().length == tokens.size() - 1
|| (cs.getMethod().isVarArgs()
&& (cs.getMethod().getParameterTypes().length <= tokens.size() - 1))) {
reducedTable.add(cs);
}
}
// selection
if (collectedTable.size() == 0) {
throw CLIException.createCommandNotFound(discriminator);
} else if (reducedTable.size() == 0) {
throw CLIException.createCommandNotFoundForArgNum(discriminator, tokens.size() - 1);
} else if (reducedTable.size() > 1) {
throw CLIException.createAmbiguousCommandExc(discriminator, tokens.size() - 1);
} else {
return reducedTable.get(0);
}
}
public void addMethod(Method method, Object handler, String prefix) {
Command annotation = method.getAnnotation(Command.class);
assert method != null;
String name;
String autoAbbrev = null;
if (annotation != null && annotation.name() != null && !annotation.name().equals("")) {
name = annotation.name();
} else {
CommandNamer.NamingInfo autoNames = namer.nameCommand(method);
name = autoNames.commandName;
for (String abbr : autoNames.possibleAbbreviations) {
if (!doesCommandExist(prefix + abbr, method.getParameterTypes().length)) {
autoAbbrev = abbr;
break;
}
}
}
ShellCommand command = new ShellCommand(handler, method, prefix, name);
if (annotation != null && annotation.abbrev() != null && !annotation.abbrev().equals("")) {
command.setAbbreviation(annotation.abbrev());
} else {
command.setAbbreviation(autoAbbrev);
}
if (annotation != null
&& annotation.description() != null
&& !annotation.description().equals("")) {
command.setDescription(annotation.description());
}
if (annotation != null && annotation.header() != null && !annotation.header().equals("")) {
command.setHeader(annotation.header());
}
commandTable.add(command);
}
