cmdlineoptions/CmdOptions.java

206 lines
6.2 KiB
Java
Raw Normal View History

2013-12-12 20:08:52 +01:00
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
public class CmdOptions {
private static CmdOptions instance;
public static class Option {
public String name;
public String cmdOption;
public String longCmdOption;
public String description;
public String defaultParameter;
public String[] possibleParams;
public String[] alternatives;
public boolean set;
public ArrayList<String> values;
public Option() {
values = new ArrayList<String>();
}
public String toString() {
String ret = name + " (" + longCmdOption + (cmdOption != null ? "/" + cmdOption : "") + ")"
+ (defaultParameter != null ? ": default=" + defaultParameter : "") + (description != null ? "\n\t\t" + description : "");
if (possibleParams != null) {
boolean start = true;
ret += "\n\t\t(Possible parameters: ";
for (String s : possibleParams) {
ret += (start ? "" : ", ") + s;
start = false;
}
ret += ")";
}
if (set) {
ret += "\n\t\t--> current Setting: ";
if (values.size() > 0) {
boolean start = true;
for (String s : values) {
ret += (start ? "" : ",") + s;
start = false;
}
} else {
ret += "true";
}
// ret += "\n";
}
return ret;
}
}
private HashMap<String, Option> options;
private CmdOptions() {
options = new HashMap<String, Option>();
this.setOption("help", "--help", "-h", null, "Show all possible options and their parameters.", null, new String[] { "-?" });
}
public static CmdOptions i() {
if (instance == null) {
instance = new CmdOptions();
}
return instance;
}
public String[] getOption(String name) {
if (options.get(name).values.size() > 0)
return options.get(name).values.toArray(new String[0]);
else if (options.get(name).defaultParameter != null)
return new String[] { options.get(name).defaultParameter };
return null;
}
public Integer[] getOptionAsInt(String name) {
if (options.get(name).values.size() > 0) {
ArrayList<Integer> list = new ArrayList<Integer>();
for (String o : options.get(name).values) {
list.add(Integer.parseInt(o));
}
return list.toArray(new Integer[0]);
} else if (options.get(name).defaultParameter != null)
return new Integer[] { Integer.parseInt(options.get(name).defaultParameter) };
return null;
}
public Double[] getOptionAsDouble(String name) {
if (options.get(name).values.size() > 0) {
ArrayList<Double> list = new ArrayList<Double>();
for (String o : options.get(name).values) {
list.add(Double.parseDouble(o));
}
return list.toArray(new Double[0]);
} else if (options.get(name).defaultParameter != null)
return new Double[] { Double.parseDouble(options.get(name).defaultParameter) };
return null;
}
public boolean isSet(String option) {
return options.get(option).set;
}
public boolean isSet(String option, String parameter) {
for (String o : options.get(option).values) {
if (o.equals(parameter)) {
return true;
}
}
return false;
}
public void setOption(String name, String longcmd, String cmd, String defaultparam, String description) {
this.setOption(name, longcmd, cmd, defaultparam, description, null, null);
}
public void setOption(String name, String longcmd, String cmd, String defaultparam, String description, String[] possibleParams, String[] alternatives) {
Option o = new Option();
o.cmdOption = cmd;
o.name = name;
o.longCmdOption = longcmd;
o.defaultParameter = defaultparam;
o.description = description;
o.alternatives = alternatives;
o.possibleParams = possibleParams;
options.put(name, o);
}
public String toString(boolean help) {
StringBuilder b = new StringBuilder();
if (help) {
b.append("Possible options:\n");
}
b.append("-options\n");
Option[] vars = options.values().toArray(new Option[0]);
Arrays.sort(vars, new Comparator<Option>() {
@Override
public int compare(Option o1, Option o2) {
return o1.name.compareTo(o2.name);
}
});
for (Option o : vars) {
b.append("\t").append(o.toString()).append("\n");
}
b.append("/options\n");
return b.toString();
}
public void parse(String[] args) {
// now parse
if (args.length > 0) {
int l = 0, i = 0;
String arg = null;
// iterate through all options
for (; i < args.length; ++i) {
arg = args[i];
Option o = null;
// search for correct option
for (Option op : options.values()) {
if (arg.equals(op.longCmdOption) || op.cmdOption != null && arg.equals(op.cmdOption) || op.alternatives != null
&& Arrays.asList(op.alternatives).contains(arg)) {
o = op;
break;
}
}
// if it is unknown
if (o == null) {
System.err.println("Unrecognized option '" + arg + "'");
continue;
}
o.set = true;
// now iterate through the parameters
int j = i + 1;
while (args.length > j && !args[j].startsWith("-")) {
if (o.possibleParams != null) {
if (Arrays.asList(o.possibleParams).contains(args[j]))
o.values.add(args[j]);
else
System.err.println("Parameter \"" + args[j] + "\" for Option \"" + o.name + "\" not allowed!");
} else {
o.values.add(args[j]);
}
++j;
}
i = j - 1;
}
}
if (options.get("help").set) {
System.out.println(this.toString(true));
System.exit(0);
}
// set default for that options that aren't set
for (Option o : options.values()) {
if (!o.set && o.defaultParameter != null && o.defaultParameter.equals("")) {
System.err.println(o.name + " (" + o.cmdOption + "): has no default parameter and has to be set on commandline!");
System.exit(1);
}
if (!o.set && o.defaultParameter != null)
o.values.add(o.defaultParameter);
}
System.out.println(this.toString(false));
}
}