SubCommandArgumentParser.java revision 99faa045b6241c1d2843cce1b7a9d9c97055beae
0N/A/*
2362N/A * CDDL HEADER START
0N/A *
0N/A * The contents of this file are subject to the terms of the
0N/A * Common Development and Distribution License, Version 1.0 only
0N/A * (the "License"). You may not use this file except in compliance
2362N/A * with the License.
0N/A *
2362N/A * You can obtain a copy of the license at
0N/A * trunk/opends/resource/legal-notices/OpenDS.LICENSE
0N/A * or https://OpenDS.dev.java.net/OpenDS.LICENSE.
0N/A * See the License for the specific language governing permissions
0N/A * and limitations under the License.
0N/A *
0N/A * When distributing Covered Code, include this CDDL HEADER in each
0N/A * file and include the License file at
0N/A * trunk/opends/resource/legal-notices/OpenDS.LICENSE. If applicable,
0N/A * add the following below this CDDL HEADER, with the fields enclosed
0N/A * by brackets "[]" replaced with your own identifying information:
0N/A * Portions Copyright [yyyy] [name of copyright owner]
2362N/A *
2362N/A * CDDL HEADER END
2362N/A *
0N/A *
0N/A * Portions Copyright 2006-2007 Sun Microsystems, Inc.
0N/A */
0N/Apackage org.opends.server.util.args;
0N/A
0N/A
0N/A
0N/Aimport java.io.FileInputStream;
0N/Aimport java.io.IOException;
0N/Aimport java.io.OutputStream;
0N/Aimport java.util.HashMap;
0N/Aimport java.util.LinkedList;
0N/Aimport java.util.Properties;
0N/A
0N/Aimport static org.opends.server.messages.MessageHandler.*;
0N/Aimport static org.opends.server.messages.UtilityMessages.*;
0N/Aimport static org.opends.server.util.ServerConstants.*;
0N/Aimport static org.opends.server.util.StaticUtils.*;
0N/A
0N/A
0N/A
0N/A/**
0N/A * This class defines a variant of the argument parser that can be used with
0N/A * applications that use subcommands to customize their behavior and that have a
0N/A * different set of options per subcommand (e.g, "cvs checkout" takes different
0N/A * options than "cvs commit"). This parser also has the ability to use global
0N/A * options that will always be applicable regardless of the subcommand in
0N/A * addition to the subcommand-specific arguments. There must not be any
0N/A * conflicts between the global options and the option for any subcommand, but
0N/A * it is allowed to re-use subcommand-specific options for different purposes
0N/A * between different subcommands.
0N/A */
0N/Apublic class SubCommandArgumentParser
0N/A{
0N/A // The argument that will be used to trigger the display of usage information.
0N/A private Argument usageArgument;
0N/A
0N/A // Indicates whether subcommand and long argument names should be treated in a
0N/A // case-sensitive manner.
0N/A private boolean longArgumentsCaseSensitive;
0N/A
0N/A // Indicates whether the usage information has been displayed.
0N/A private boolean usageDisplayed;
0N/A
0N/A // The set of global arguments defined for this parser, referenced by short
0N/A // ID.
0N/A private HashMap<Character,Argument> globalShortIDMap;
0N/A
0N/A // The set of global arguments defined for this parser, referenced by
0N/A // argument name.
0N/A private HashMap<String,Argument> globalArgumentMap;
0N/A
0N/A // The set of global arguments defined for this parser, referenced by long
0N/A // ID.
0N/A private HashMap<String,Argument> globalLongIDMap;
0N/A
0N/A // The set of subcommands defined for this parser, referenced by subcommand
0N/A // name.
0N/A private HashMap<String,SubCommand> subCommands;
0N/A
0N/A // The total set of global arguments defined for this parser.
0N/A private LinkedList<Argument> globalArgumentList;
0N/A
0N/A // The output stream to which usage information should be printed.
0N/A private OutputStream usageOutputStream;
0N/A
0N/A // The fully-qualified name of the Java class that should be invoked to launch
0N/A // the program with which this argument parser is associated.
0N/A private String mainClassName;
0N/A
0N/A // A human-readable description for the tool, which will be included when
0N/A // displaying usage information.
0N/A private String toolDescription;
0N/A
0N/A // The raw set of command-line arguments that were provided.
0N/A private String[] rawArguments;
0N/A
0N/A // The subcommand requested by the user as part of the command-line arguments.
0N/A private SubCommand subCommand;
0N/A
0N/A
0N/A
0N/A /**
0N/A * Creates a new instance of this subcommand argument parser with no
0N/A * arguments.
0N/A *
0N/A * @param mainClassName The fully-qualified name of the Java
0N/A * class that should be invoked to launch
0N/A * the program with which this argument
0N/A * parser is associated.
0N/A * @param toolDescription A human-readable description for the
0N/A * tool, which will be included when
0N/A * displaying usage information.
0N/A * @param longArgumentsCaseSensitive Indicates whether subcommand and long
0N/A * argument names should be treated in a
0N/A * case-sensitive manner.
0N/A */
0N/A public SubCommandArgumentParser(String mainClassName, String toolDescription,
0N/A boolean longArgumentsCaseSensitive)
0N/A {
0N/A this.mainClassName = mainClassName;
0N/A this.toolDescription = toolDescription;
0N/A this.longArgumentsCaseSensitive = longArgumentsCaseSensitive;
0N/A
0N/A globalArgumentList = new LinkedList<Argument>();
0N/A globalArgumentMap = new HashMap<String,Argument>();
0N/A globalShortIDMap = new HashMap<Character,Argument>();
0N/A globalLongIDMap = new HashMap<String,Argument>();
0N/A subCommands = new HashMap<String,SubCommand>();
0N/A usageDisplayed = false;
0N/A rawArguments = null;
0N/A subCommand = null;
0N/A usageArgument = null;
0N/A usageOutputStream = null;
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Retrieves the fully-qualified name of the Java class that should be invoked
0N/A * to launch the program with which this argument parser is associated.
0N/A *
0N/A * @return The fully-qualified name of the Java class that should be invoked
0N/A * to launch the program with which this argument parser is
0N/A * associated.
0N/A */
0N/A public String getMainClassName()
0N/A {
0N/A return mainClassName;
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Retrieves a human-readable description for this tool, which should be
0N/A * included at the top of the command-line usage information.
0N/A *
0N/A * @return A human-readable description for this tool, or {@code null} if
0N/A * none is available.
0N/A */
0N/A public String getToolDescription()
0N/A {
0N/A return toolDescription;
0N/A }
0N/A
0N/A
0N/A
0N/A /**
993N/A * Indicates whether subcommand names and long argument strings should be
0N/A * treated in a case-sensitive manner.
0N/A *
0N/A * @return <CODE>true</CODE> if subcommand names and long argument strings
0N/A * should be treated in a case-sensitive manner, or
0N/A * <CODE>false</CODE> if they should not.
0N/A */
0N/A public boolean longArgumentsCaseSensitive()
0N/A {
0N/A return longArgumentsCaseSensitive;
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Retrieves the list of all global arguments that have been defined for this
0N/A * argument parser.
0N/A *
0N/A * @return The list of all global arguments that have been defined for this
0N/A * argument parser.
993N/A */
0N/A public LinkedList<Argument> getGlobalArgumentList()
0N/A {
0N/A return globalArgumentList;
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Indicates whether this argument parser contains a global argument with the
0N/A * specified name.
993N/A *
0N/A * @param argumentName The name for which to make the determination.
0N/A *
0N/A * @return <CODE>true</CODE> if a global argument exists with the specified
0N/A * name, or <CODE>false</CODE> if not.
0N/A */
0N/A public boolean hasGlobalArgument(String argumentName)
0N/A {
0N/A return globalArgumentMap.containsKey(argumentName);
0N/A }
0N/A
993N/A
0N/A
0N/A /**
0N/A * Retrieves the global argument with the specified name.
0N/A *
0N/A * @param name The name of the global argument to retrieve.
0N/A *
993N/A * @return The global argument with the specified name, or <CODE>null</CODE>
0N/A * if there is no such argument.
0N/A */
0N/A public Argument getGlobalArgument(String name)
993N/A {
0N/A return globalArgumentMap.get(name);
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Retrieves the set of global arguments mapped by the short identifier that
0N/A * may be used to reference them. Note that arguments that do not have a
0N/A * short identifier will not be present in this list.
0N/A *
0N/A * @return The set of global arguments mapped by the short identifier that
0N/A * may be used to reference them.
0N/A */
0N/A public HashMap<Character,Argument> getGlobalArgumentsByShortID()
0N/A {
0N/A return globalShortIDMap;
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Indicates whether this argument parser has a global argument with the
0N/A * specified short ID.
0N/A *
0N/A * @param shortID The short ID character for which to make the
0N/A * determination.
0N/A *
0N/A * @return <CODE>true</CODE> if a global argument exists with the specified
0N/A * short ID, or <CODE>false</CODE> if not.
0N/A */
0N/A public boolean hasGlobalArgumentWithShortID(Character shortID)
0N/A {
0N/A return globalShortIDMap.containsKey(shortID);
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Retrieves the global argument with the specified short identifier.
0N/A *
0N/A * @param shortID The short identifier for the global argument to retrieve.
0N/A *
0N/A * @return The global argument with the specified short identifier, or
0N/A * <CODE>null</CODE> if there is no such argument.
0N/A */
0N/A public Argument getGlobalArgumentForShortID(Character shortID)
0N/A {
0N/A return globalShortIDMap.get(shortID);
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Retrieves the set of global arguments mapped by the long identifier that
0N/A * may be used to reference them. Note that arguments that do not have a long
0N/A * identifier will not be present in this list.
0N/A *
0N/A * @return The set of global arguments mapped by the long identifier that may
0N/A * be used to reference them.
0N/A */
0N/A public HashMap<String,Argument> getGlobalArgumentsByLongID()
0N/A {
0N/A return globalLongIDMap;
0N/A }
0N/A
0N/A
0N/A
0N/A /**
0N/A * Indicates whether this argument parser has a global argument with the
0N/A * specified long ID.
0N/A *
0N/A * @param longID The long ID string for which to make the determination.
0N/A *
0N/A * @return <CODE>true</CODE> if a global argument exists with the specified
0N/A * long ID, or <CODE>false</CODE> if not.
0N/A */
0N/A public boolean hasGlobalArgumentWithLongID(String longID)
0N/A {
0N/A return globalLongIDMap.containsKey(longID);
0N/A }
/**
* Retrieves the global argument with the specified long identifier.
*
* @param longID The long identifier for the global argument to retrieve.
*
* @return The global argument with the specified long identifier, or
* <CODE>null</CODE> if there is no such argument.
*/
public Argument getGlobalArgumentForLongID(String longID)
{
return globalLongIDMap.get(longID);
}
/**
* Retrieves the set of subcommands defined for this argument parser,
* referenced by subcommand name.
*
* @return The set of subcommands defined for this argument parser,
* referenced by subcommand name.
*/
public HashMap<String,SubCommand> getSubCommands()
{
return subCommands;
}
/**
* Indicates whether this argument parser has a subcommand with the specified
* name.
*
* @param name The subcommand name for which to make the determination.
*
* @return <CODE>true</CODE> if this argument parser has a subcommand with
* the specified name, or <CODE>false</CODE> if it does not.
*/
public boolean hasSubCommand(String name)
{
return subCommands.containsKey(name);
}
/**
* Retrieves the subcommand with the specified name.
*
* @param name The name of the subcommand to retrieve.
*
* @return The subcommand with the specified name, or <CODE>null</CODE> if no
* such subcommand is defined.
*/
public SubCommand getSubCommand(String name)
{
return subCommands.get(name);
}
/**
* Retrieves the subcommand that was selected in the set of command-line
* arguments.
*
* @return The subcommand that was selected in the set of command-line
* arguments, or <CODE>null</CODE> if none was selected.
*/
public SubCommand getSubCommand()
{
return subCommand;
}
/**
* Retrieves the raw set of arguments that were provided.
*
* @return The raw set of arguments that were provided, or <CODE>null</CODE>
* if the argument list has not yet been parsed.
*/
public String[] getRawArguments()
{
return rawArguments;
}
/**
* Adds the provided argument to the set of global arguments handled by this
* parser.
*
* @param argument The argument to be added.
*
* @throws ArgumentException If the provided argument conflicts with another
* global or subcommand argument that has already
* been defined.
*/
public void addGlobalArgument(Argument argument)
throws ArgumentException
{
String argumentName = argument.getName();
if (globalArgumentMap.containsKey(argumentName))
{
int msgID = MSGID_SUBCMDPARSER_DUPLICATE_GLOBAL_ARG_NAME;
String message = getMessage(msgID, argumentName);
throw new ArgumentException(msgID, message);
}
for (SubCommand s : subCommands.values())
{
if (s.getArgumentForName(argumentName) != null)
{
int msgID = MSGID_SUBCMDPARSER_GLOBAL_ARG_NAME_SUBCMD_CONFLICT;
String message = getMessage(msgID, argumentName, s.getName());
throw new ArgumentException(msgID, message);
}
}
Character shortID = argument.getShortIdentifier();
if (shortID != null)
{
if (globalShortIDMap.containsKey(shortID))
{
String name = globalShortIDMap.get(shortID).getName();
int msgID = MSGID_SUBCMDPARSER_DUPLICATE_GLOBAL_ARG_SHORT_ID;
String message = getMessage(msgID, String.valueOf(shortID),
argumentName, name);
throw new ArgumentException(msgID, message);
}
for (SubCommand s : subCommands.values())
{
if (s.getArgument(shortID) != null)
{
String cmdName = s.getName();
String name = s.getArgument(shortID).getName();
int msgID = MSGID_SUBCMDPARSER_GLOBAL_ARG_SHORT_ID_CONFLICT;
String message = getMessage(msgID, String.valueOf(shortID),
argumentName, name, cmdName);
throw new ArgumentException(msgID, message);
}
}
}
String longID = argument.getLongIdentifier();
if (longID != null)
{
if (! longArgumentsCaseSensitive)
{
longID = toLowerCase(longID);
}
if (globalLongIDMap.containsKey(longID))
{
String name = globalLongIDMap.get(longID).getName();
int msgID = MSGID_SUBCMDPARSER_DUPLICATE_GLOBAL_ARG_LONG_ID;
String message = getMessage(msgID, longID, argumentName, name);
throw new ArgumentException(msgID, message);
}
for (SubCommand s : subCommands.values())
{
if (s.getArgument(longID) != null)
{
String cmdName = s.getName();
String name = s.getArgument(longID).getName();
int msgID = MSGID_SUBCMDPARSER_GLOBAL_ARG_LONG_ID_CONFLICT;
String message = getMessage(msgID, longID, argumentName, name,
cmdName);
throw new ArgumentException(msgID, message);
}
}
}
if (shortID != null)
{
globalShortIDMap.put(shortID, argument);
}
if (longID != null)
{
globalLongIDMap.put(longID, argument);
}
globalArgumentList.add(argument);
}
/**
* Sets the provided argument as one which will automatically trigger the
* output of usage information if it is provided on the command line and no
* further argument validation will be performed. Note that the caller will
* still need to add this argument to the parser with the
* <CODE>addArgument</CODE> method, and the argument should not be required
* and should not take a value. Also, the caller will still need to check
* for the presence of the usage argument after calling
* <CODE>parseArguments</CODE> to know that no further processing will be
* required.
*
* @param argument The argument whose presence should automatically
* trigger the display of usage information.
*/
public void setUsageArgument(Argument argument)
{
usageArgument = argument;
usageOutputStream = System.out;
}
/**
* Sets the provided argument as one which will automatically trigger the
* output of usage information if it is provided on the command line and no
* further argument validation will be performed. Note that the caller will
* still need to add this argument to the parser with the
* <CODE>addArgument</CODE> method, and the argument should not be required
* and should not take a value. Also, the caller will still need to check
* for the presence of the usage argument after calling
* <CODE>parseArguments</CODE> to know that no further processing will be
* required.
*
* @param argument The argument whose presence should automatically
* trigger the display of usage information.
* @param outputStream The output stream to which the usage information
* should be written.
*/
public void setUsageArgument(Argument argument, OutputStream outputStream)
{
usageArgument = argument;
usageOutputStream = outputStream;
}
/**
* Adds the provided subcommand to this argument parser. This is only
* intended for use by the <CODE>SubCommand</CODE> constructor and does not
* do any validation of its own to ensure that there are no conflicts with the
* subcommand or any of its arguments.
*
* @param subCommand The subcommand to add to this argument parser.
*/
void addSubCommand(SubCommand subCommand)
{
subCommands.put(toLowerCase(subCommand.getName()), subCommand);
}
/**
* Parses the provided set of arguments and updates the information associated
* with this parser accordingly.
*
* @param rawArguments The raw set of arguments to parse.
*
* @throws ArgumentException If a problem was encountered while parsing the
* provided arguments.
*/
public void parseArguments(String[] rawArguments)
throws ArgumentException
{
parseArguments(rawArguments, null);
}
/**
* Parses the provided set of arguments and updates the information associated
* with this parser accordingly. Default values for unspecified arguments
* may be read from the specified properties file.
*
* @param rawArguments The set of raw arguments to parse.
* @param propertiesFile The path to the properties file to use to
* obtain default values for unspecified
* properties.
* @param requirePropertiesFile Indicates whether the parsing should fail if
* the provided properties file does not exist
* or is not accessible.
*
* @throws ArgumentException If a problem was encountered while parsing the
* provided arguments or interacting with the
* properties file.
*/
public void parseArguments(String[] rawArguments, String propertiesFile,
boolean requirePropertiesFile)
throws ArgumentException
{
this.rawArguments = rawArguments;
Properties argumentProperties = null;
try
{
Properties p = new Properties();
FileInputStream fis = new FileInputStream(propertiesFile);
p.load(fis);
fis.close();
argumentProperties = p;
}
catch (Exception e)
{
if (requirePropertiesFile)
{
int msgID = MSGID_SUBCMDPARSER_CANNOT_READ_PROPERTIES_FILE;
String message = getMessage(msgID, String.valueOf(propertiesFile),
getExceptionMessage(e));
throw new ArgumentException(msgID, message, e);
}
}
parseArguments(rawArguments, argumentProperties);
}
/**
* Parses the provided set of arguments and updates the information associated
* with this parser accordingly. Default values for unspecified arguments may
* be read from the specified properties if any are provided.
*
* @param rawArguments The set of raw arguments to parse.
* @param argumentProperties A set of properties that may be used to provide
* default values for arguments not included in
* the given raw arguments.
*
* @throws ArgumentException If a problem was encountered while parsing the
* provided arguments.
*/
public void parseArguments(String[] rawArguments,
Properties argumentProperties)
throws ArgumentException
{
this.rawArguments = rawArguments;
int numArguments = rawArguments.length;
for (int i=0; i < numArguments; i++)
{
String arg = rawArguments[i];
if (arg.equals("--"))
{
// This is not legal because we don't allow unnamed trailing arguments
// in this parser.
int msgID = MSGID_SUBCMDPARSER_LONG_ARG_WITHOUT_NAME;
String message = getMessage(msgID, arg);
throw new ArgumentException(msgID, message);
}
else if (arg.startsWith("--"))
{
// This indicates that we are using the long name to reference the
// argument. It may be in any of the following forms:
// --name
// --name value
// --name=value
String argName = arg.substring(2);
String argValue = null;
int equalPos = argName.indexOf('=');
if (equalPos < 0)
{
// This is fine. The value is not part of the argument name token.
}
else if (equalPos == 0)
{
// The argument starts with "--=", which is not acceptable.
int msgID = MSGID_SUBCMDPARSER_LONG_ARG_WITHOUT_NAME;
String message = getMessage(msgID, arg);
throw new ArgumentException(msgID, message);
}
else
{
// The argument is in the form --name=value, so parse them both out.
argValue = argName.substring(equalPos+1);
argName = argName.substring(0, equalPos);
}
// If we're not case-sensitive, then convert the name to lowercase.
if (! longArgumentsCaseSensitive)
{
argName = toLowerCase(argName);
}
// See if the specified name references a global argument. If not, then
// see if it references a subcommand argument.
Argument a = globalLongIDMap.get(argName);
if (a == null)
{
if (subCommand == null)
{
if (argName.equals("help"))
{
// "--help" will always be interpreted as requesting usage
// information.
try
{
getUsage(usageOutputStream);
} catch (Exception e) {}
return;
}
else
{
// There is no such global argument.
int msgID = MSGID_SUBCMDPARSER_NO_GLOBAL_ARGUMENT_FOR_LONG_ID;
String message = getMessage(msgID, argName);
throw new ArgumentException(msgID, message);
}
}
else
{
a = subCommand.getArgument(argName);
if (a == null)
{
if (argName.equals("help"))
{
// "--help" will always be interpreted as requesting usage
// information.
try
{
getUsage(usageOutputStream);
} catch (Exception e) {}
return;
}
else
{
// There is no such global or subcommand argument.
int msgID = MSGID_SUBCMDPARSER_NO_ARGUMENT_FOR_LONG_ID;
String message = getMessage(msgID, argName);
throw new ArgumentException(msgID, message);
}
}
}
}
a.setPresent(true);
// If this is the usage argument, then immediately stop and print
// usage information.
if ((usageArgument != null) &&
usageArgument.getName().equals(a.getName()))
{
try
{
getUsage(usageOutputStream);
} catch (Exception e) {}
return;
}
// See if the argument takes a value. If so, then make sure one was
// provided. If not, then make sure none was provided.
if (a.needsValue())
{
if (argValue == null)
{
if ((i+1) == numArguments)
{
int msgID = MSGID_SUBCMDPARSER_NO_VALUE_FOR_ARGUMENT_WITH_LONG_ID;
String message = getMessage(msgID, argName);
throw new ArgumentException(msgID, message);
}
argValue = rawArguments[++i];
}
StringBuilder invalidReason = new StringBuilder();
if (! a.valueIsAcceptable(argValue, invalidReason))
{
int msgID = MSGID_SUBCMDPARSER_VALUE_UNACCEPTABLE_FOR_LONG_ID;
String message = getMessage(msgID, argValue, argName,
invalidReason.toString());
throw new ArgumentException(msgID, message);
}
// If the argument already has a value, then make sure it is
// acceptable to have more than one.
if (a.hasValue() && (! a.isMultiValued()))
{
int msgID = MSGID_SUBCMDPARSER_NOT_MULTIVALUED_FOR_LONG_ID;
String message = getMessage(msgID, argName);
throw new ArgumentException(msgID, message);
}
a.addValue(argValue);
}
else
{
if (argValue != null)
{
int msgID = MSGID_SUBCMDPARSER_ARG_FOR_LONG_ID_DOESNT_TAKE_VALUE;
String message = getMessage(msgID, argName);
throw new ArgumentException(msgID, message);
}
}
}
else if (arg.startsWith("-"))
{
// This indicates that we are using the 1-character name to reference
// the argument. It may be in any of the following forms:
// -n
// -nvalue
// -n value
if (arg.equals("-"))
{
int msgID = MSGID_SUBCMDPARSER_INVALID_DASH_AS_ARGUMENT;
String message = getMessage(msgID);
throw new ArgumentException(msgID, message);
}
char argCharacter = arg.charAt(1);
String argValue;
if (arg.length() > 2)
{
argValue = arg.substring(2);
}
else
{
argValue = null;
}
// Get the argument with the specified short ID. It may be either a
// global argument or a subcommand-specific argument.
Argument a = globalShortIDMap.get(argCharacter);
if (a == null)
{
if (subCommand == null)
{
if (argCharacter == '?')
{
// "-?" will always be interpreted as requesting usage.
try
{
getUsage(usageOutputStream);
} catch (Exception e) {}
return;
}
else
{
// There is no such argument registered.
int msgID = MSGID_SUBCMDPARSER_NO_GLOBAL_ARGUMENT_FOR_SHORT_ID;
String message = getMessage(msgID, String.valueOf(argCharacter));
throw new ArgumentException(msgID, message);
}
}
else
{
a = subCommand.getArgument(argCharacter);
if (a == null)
{
if (argCharacter == '?')
{
// "-?" will always be interpreted as requesting usage.
try
{
getUsage(usageOutputStream);
} catch (Exception e) {}
return;
}
else
{
// There is no such argument registered.
int msgID = MSGID_SUBCMDPARSER_NO_ARGUMENT_FOR_SHORT_ID;
String message = getMessage(msgID,
String.valueOf(argCharacter));
throw new ArgumentException(msgID, message);
}
}
}
}
a.setPresent(true);
// If this is the usage argument, then immediately stop and print
// usage information.
if ((usageArgument != null) &&
usageArgument.getName().equals(a.getName()))
{
try
{
getUsage(usageOutputStream);
} catch (Exception e) {}
return;
}
// See if the argument takes a value. If so, then make sure one was
// provided. If not, then make sure none was provided.
if (a.needsValue())
{
if (argValue == null)
{
if ((i+1) == numArguments)
{
int msgID =
MSGID_SUBCMDPARSER_NO_VALUE_FOR_ARGUMENT_WITH_SHORT_ID;
String message = getMessage(msgID, String.valueOf(argCharacter));
throw new ArgumentException(msgID, message);
}
argValue = rawArguments[++i];
}
StringBuilder invalidReason = new StringBuilder();
if (! a.valueIsAcceptable(argValue, invalidReason))
{
int msgID = MSGID_SUBCMDPARSER_VALUE_UNACCEPTABLE_FOR_SHORT_ID;
String message = getMessage(msgID, argValue,
String.valueOf(argCharacter),
invalidReason.toString());
throw new ArgumentException(msgID, message);
}
// If the argument already has a value, then make sure it is
// acceptable to have more than one.
if (a.hasValue() && (! a.isMultiValued()))
{
int msgID = MSGID_SUBCMDPARSER_NOT_MULTIVALUED_FOR_SHORT_ID;
String message = getMessage(msgID, String.valueOf(argCharacter));
throw new ArgumentException(msgID, message);
}
a.addValue(argValue);
}
else
{
if (argValue != null)
{
// If we've gotten here, then it means that we're in a scenario like
// "-abc" where "a" is a valid argument that doesn't take a value.
// However, this could still be valid if all remaining characters in
// the value are also valid argument characters that don't take
// values.
int valueLength = argValue.length();
for (int j=0; j < valueLength; j++)
{
char c = argValue.charAt(j);
Argument b = globalShortIDMap.get(c);
if (b == null)
{
if (subCommand == null)
{
int msgID =
MSGID_SUBCMDPARSER_NO_GLOBAL_ARGUMENT_FOR_SHORT_ID;
String message = getMessage(msgID,
String.valueOf(argCharacter));
throw new ArgumentException(msgID, message);
}
else
{
b = subCommand.getArgument(c);
if (b == null)
{
int msgID = MSGID_SUBCMDPARSER_NO_ARGUMENT_FOR_SHORT_ID;
String message = getMessage(msgID,
String.valueOf(argCharacter));
throw new ArgumentException(msgID, message);
}
}
}
if (b.needsValue())
{
// This means we're in a scenario like "-abc" where b is a
// valid argument that takes a value. We don't support that.
int msgID = MSGID_SUBCMDPARSER_CANT_MIX_ARGS_WITH_VALUES;
String message = getMessage(msgID, String.valueOf(argCharacter),
argValue, String.valueOf(c));
throw new ArgumentException(msgID, message);
}
else
{
b.setPresent(true);
// If this is the usage argument, then immediately stop and
// print usage information.
if ((usageArgument != null) &&
usageArgument.getName().equals(b.getName()))
{
try
{
getUsage(usageOutputStream);
} catch (Exception e) {}
return;
}
}
}
}
}
}
else
{
// It's not a short or long identifier, so check to see if it is a
// subcommand name. If not, then it's invalid. If so, then make sure
// that it was the only subcommand provided.
String nameToCheck = arg;
if (! longArgumentsCaseSensitive)
{
nameToCheck = toLowerCase(arg);
}
SubCommand sc = subCommands.get(nameToCheck);
if (sc == null)
{
int msgID = MSGID_SUBCMDPARSER_INVALID_ARGUMENT;
String message = getMessage(msgID, arg);
throw new ArgumentException(msgID, message);
}
else if (subCommand == null)
{
subCommand = sc;
}
else
{
int msgID = MSGID_SUBCMDPARSER_MULTIPLE_SUBCOMMANDS;
String message = getMessage(msgID, arg, subCommand.getName());
throw new ArgumentException(msgID, message);
}
}
}
// Iterate through all the global arguments and make sure that they have
// values or a suitable default is available.
for (Argument a : globalArgumentList)
{
if ((! a.isPresent()) && a.needsValue())
{
// See if there is a default value in the properties that can be used.
boolean valueSet = false;
if ((argumentProperties != null) && (a.getPropertyName() != null))
{
String value = argumentProperties.getProperty(a.getPropertyName());
if (value != null)
{
a.addValue(value);
valueSet = true;
}
}
// If there is still no value, then see if the argument defines a
// default.
if ((! valueSet) && (a.getDefaultValue() != null))
{
a.addValue(a.getDefaultValue());
valueSet = true;
}
// If there is still no value and the argument is required, then that's
// a problem.
if ((! valueSet) && a.isRequired())
{
int msgID = MSGID_SUBCMDPARSER_NO_VALUE_FOR_REQUIRED_ARG;
String message = getMessage(msgID, a.getName());
throw new ArgumentException(msgID, message);
}
}
}
// Iterate through all the subcommand-specific arguments and make sure that
// they have values or a suitable default is available.
if (subCommand != null)
{
for (Argument a : subCommand.getArguments())
{
if ((! a.isPresent()) && a.needsValue())
{
// See if there is a default value in the properties that can be used.
boolean valueSet = false;
if ((argumentProperties != null) && (a.getPropertyName() != null))
{
String value = argumentProperties.getProperty(a.getPropertyName());
if (value != null)
{
a.addValue(value);
valueSet = true;
}
}
// If there is still no value, then see if the argument defines a
// default.
if ((! valueSet) && (a.getDefaultValue() != null))
{
a.addValue(a.getDefaultValue());
valueSet = true;
}
// If there is still no value and the argument is required, then
// that's a problem.
if ((! valueSet) && a.isRequired())
{
int msgID = MSGID_SUBCMDPARSER_NO_VALUE_FOR_REQUIRED_ARG;
String message = getMessage(msgID, a.getName());
throw new ArgumentException(msgID, message);
}
}
}
}
}
/**
* Appends complete usage information to the provided buffer. It will include
* information about global options as well as all subcommand-specific
* options. The output will be somewhat compressed in that it will not
* include any of the descriptions for any of the arguments.
*
* @param buffer The buffer to which the usage information should be
* appended.
*/
public void getFullUsage(StringBuilder buffer)
{
usageDisplayed = true;
if ((toolDescription != null) && (toolDescription.length() > 0))
{
buffer.append(wrapText(toolDescription, 79));
buffer.append(EOL);
}
String scriptName = System.getProperty(PROPERTY_SCRIPT_NAME);
if ((scriptName == null) || (scriptName.length() == 0))
{
buffer.append("Usage: java ");
buffer.append(mainClassName);
}
else
{
buffer.append("Usage: ");
buffer.append(scriptName);
}
buffer.append(" {subcommand} {options}");
buffer.append(EOL);
buffer.append(EOL);
buffer.append("Available subcommands:");
buffer.append(EOL);
int indentNb = 0;
for (SubCommand sc : subCommands.values())
{
if (sc.getName().length() > indentNb )
{
indentNb = sc.getName().length();
}
}
indentNb++;
for (SubCommand sc : subCommands.values())
{
buffer.append(" " + sc.getName());
for (int i=0; i < indentNb - sc.getName().length() ; i++)
{
buffer.append(" ");
}
buffer.append(sc.getDescription());
buffer.append(EOL);
}
buffer.append(EOL);
buffer.append("The accepted value for global options are:");
buffer.append(EOL);
for (Argument a : globalArgumentList)
{
if (a.isHidden())
{
continue;
}
String value;
if (a.needsValue())
{
String valuePlaceholder = a.getValuePlaceholder();
if (a == null)
{
value = " {value}";
}
else
{
value = " " + valuePlaceholder;
}
}
else
{
value = "";
}
Character shortIDChar = a.getShortIdentifier();
if (shortIDChar != null)
{
buffer.append(" -");
buffer.append(shortIDChar);
buffer.append(value);
String longIDString = a.getLongIdentifier();
if (longIDString != null)
{
buffer.append(", --");
buffer.append(longIDString);
buffer.append(value);
}
}
else
{
String longIDString = a.getLongIdentifier();
if (longIDString != null)
{
buffer.append(" --");
buffer.append(longIDString);
buffer.append(value);
}
}
buffer.append(EOL);
indentAndWrap(" ", a.getDescription(), buffer);
}
buffer.append(EOL);
}
/**
* Appends usage information for the specified subcommand to the provided
* buffer.
*
* @param buffer The buffer to which the usage information should be
* appended.
* @param subCommand The subcommand for which to display the usage
* information.
*/
public void getSubCommandUsage(StringBuilder buffer, SubCommand subCommand)
{
usageDisplayed = true;
String scriptName = System.getProperty(PROPERTY_SCRIPT_NAME);
String printName;
if ((scriptName == null) || (scriptName.length() == 0))
{
buffer.append("Usage: java ");
buffer.append(mainClassName);
printName = "java " + mainClassName;
}
else
{
buffer.append("Usage: ");
buffer.append(scriptName);
printName = scriptName;
}
buffer.append(" ");
buffer.append(subCommand.getName());
buffer.append(" {options}");
buffer.append(EOL);
buffer.append(subCommand.getDescription());
buffer.append(EOL);
if ( ! globalArgumentList.isEmpty())
{
buffer.append(EOL);
buffer.append("Global Options:");
buffer.append(EOL);
buffer.append(" See \"" + printName + " --help\".");
buffer.append(EOL);
}
if ( ! subCommand.getArguments().isEmpty() )
{
buffer.append(EOL);
buffer.append("SubCommand Options:");
buffer.append(EOL);
}
for (Argument a : subCommand.getArguments())
{
// If this argument is hidden, then skip it.
if (a.isHidden())
{
continue;
}
// Write a line with the short and/or long identifiers that may be used
// for the argument.
Character shortID = a.getShortIdentifier();
if (shortID != null)
{
int currentLength = buffer.length();
buffer.append(" -");
buffer.append(shortID.charValue());
if (a.needsValue())
{
buffer.append(" ");
buffer.append(a.getValuePlaceholder());
}
String longID = a.getLongIdentifier();
if (longID != null)
{
StringBuilder newBuffer = new StringBuilder();
newBuffer.append(", --");
newBuffer.append(longID);
if (a.needsValue())
{
newBuffer.append(" ");
newBuffer.append(a.getValuePlaceholder());
}
int lineLength = (buffer.length() - currentLength) + 2 +
newBuffer.length();
if (lineLength > 80)
{
buffer.append(EOL);
buffer.append(newBuffer.toString());
}
else
{
buffer.append(" ");
buffer.append(newBuffer.toString());
}
}
buffer.append(EOL);
}
else
{
String longID = a.getLongIdentifier();
if (longID != null)
{
buffer.append(" --");
buffer.append(longID);
if (a.needsValue())
{
buffer.append(" ");
buffer.append(a.getValuePlaceholder());
}
buffer.append(EOL);
}
}
indentAndWrap(" ", a.getDescription(), buffer);
if (a.isRequired())
{
buffer.append(" This argument is mandatory.");
}
buffer.append(EOL);
}
}
/**
* Retrieves a string containing usage information based on the defined
* arguments.
*
* @return A string containing usage information based on the defined
* arguments.
*/
public String getUsage()
{
StringBuilder buffer = new StringBuilder();
if (subCommand == null)
{
getFullUsage(buffer);
}
else
{
getSubCommandUsage(buffer, subCommand);
}
return buffer.toString();
}
/**
* Writes usage information based on the defined arguments to the provided
* output stream.
*
* @param outputStream The output stream to which the usage information
* should be written.
*
* @throws IOException If a problem occurs while attempting to write the
* usage information to the provided output stream.
*/
public void getUsage(OutputStream outputStream)
throws IOException
{
StringBuilder buffer = new StringBuilder();
if (subCommand == null)
{
getFullUsage(buffer);
}
else
{
getSubCommandUsage(buffer, subCommand);
}
outputStream.write(getBytes(buffer.toString()));
}
/**
* Indicates whether the usage information has been displayed to the end user
* either by an explicit argument like "-H" or "--help", or by a built-in
* argument like "-?".
*
* @return {@code true} if the usage information has been displayed, or
* {@code false} if not.
*/
public boolean usageDisplayed()
{
return usageDisplayed;
}
/**
* Write one or more lines with the description of the argument. We will
* indent the description five characters and try our best to wrap at or
* before column 79 so it will be friendly to 80-column displays.
*/
private void indentAndWrap(String indent, String text, StringBuilder buffer)
{
int actualSize = 80 - indent.length();
if (text.length() <= actualSize)
{
buffer.append(indent);
buffer.append(text);
buffer.append(EOL);
}
else
{
String s = text;
while (s.length() > actualSize)
{
int spacePos = s.lastIndexOf(' ', actualSize);
if (spacePos > 0)
{
buffer.append(indent);
buffer.append(s.substring(0, spacePos).trim());
s = s.substring(spacePos + 1).trim();
buffer.append(EOL);
}
else
{
// There are no spaces in the first actualSize -1 columns. See
// if there is one after that point. If so, then break there.
// If not, then don't break at all.
spacePos = s.indexOf(' ');
if (spacePos > 0)
{
buffer.append(indent);
buffer.append(s.substring(0, spacePos).trim());
s = s.substring(spacePos + 1).trim();
buffer.append(EOL);
}
else
{
buffer.append(indent);
buffer.append(s);
s = "";
buffer.append(EOL);
}
}
}
if (s.length() > 0)
{
buffer.append(indent);
buffer.append(s);
buffer.append(EOL);
}
}
}
}