-
Notifications
You must be signed in to change notification settings - Fork 191
/
EnumOptionHandler.java
58 lines (51 loc) · 1.7 KB
/
EnumOptionHandler.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package org.kohsuke.args4j.spi;
import org.kohsuke.args4j.CmdLineException;
import org.kohsuke.args4j.CmdLineParser;
import org.kohsuke.args4j.OptionDef;
/**
* {@link Enum} {@link OptionHandler}.
*
* @author Kohsuke Kawaguchi
*/
public class EnumOptionHandler<T extends Enum<T>> extends OptionHandler<T> {
private final Class<T> enumType;
public EnumOptionHandler(CmdLineParser parser, OptionDef option, Setter<? super T> setter, Class<T> enumType) {
super(parser, option, setter);
this.enumType = enumType;
}
@Override
public int parseArguments(Parameters params) throws CmdLineException {
String s = params.getParameter(0).replaceAll("-", "_");
T value = null;
for( T o : enumType.getEnumConstants() )
if(o.name().equalsIgnoreCase(s)) {
value = o;
break;
}
if(value==null) {
if (option.isArgument()) {
throw new CmdLineException(owner, Messages.ILLEGAL_OPERAND, option.toString(), s);
} else {
throw new CmdLineException(owner, Messages.ILLEGAL_OPERAND, params.getParameter(-1),s);
}
}
setter.addValue(value);
return 1;
}
/*
* Returns all values of an enum type split by pipe.
* <tt>[ one | two | three]</tt>
* @see org.kohsuke.args4j.spi.OptionHandler#getDefaultMetaVariable()
*/
@Override
public String getDefaultMetaVariable() {
StringBuffer rv = new StringBuffer();
rv.append("[");
for (T t : enumType.getEnumConstants()) {
rv.append(t).append(" | ");
}
rv.delete(rv.length()-3, rv.length());
rv.append("]");
return rv.toString();
}
}