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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
/*
* Copyright (C) 2008, Shawn O. Pearce <spearce@spearce.org>
*
* (Taken from JGit org.eclipse.jgit.pgm.opt.CmdLineParser.)
*
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* - Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* - Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* - Neither the name of the Git Development Community nor the names of its
* contributors may be used to endorse or promote products derived from this
* software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
package com.google.gerrit.util.cli;
import com.google.inject.Inject;
import com.google.inject.Injector;
import com.google.inject.Key;
import com.google.inject.assistedinject.Assisted;
import org.kohsuke.args4j.Argument;
import org.kohsuke.args4j.CmdLineException;
import org.kohsuke.args4j.IllegalAnnotationError;
import org.kohsuke.args4j.Option;
import org.kohsuke.args4j.OptionDef;
import org.kohsuke.args4j.spi.OptionHandler;
import org.kohsuke.args4j.spi.Setter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.ResourceBundle;
/**
* Extended command line parser which handles --foo=value arguments.
* <p>
* The args4j package does not natively handle --foo=value and instead prefers
* to see --foo value on the command line. Many users are used to the GNU style
* --foo=value long option, so we convert from the GNU style format to the
* args4j style format prior to invoking args4j for parsing.
*/
public class CmdLineParser {
public interface Factory {
CmdLineParser create(Object bean);
}
private final Injector injector;
private final MyParser parser;
/**
* Creates a new command line owner that parses arguments/options and set them
* into the given object.
*
* @param bean instance of a class annotated by
* {@link org.kohsuke.args4j.Option} and
* {@link org.kohsuke.args4j.Argument}. this object will receive
* values.
*
* @throws IllegalAnnotationError if the option bean class is using args4j
* annotations incorrectly.
*/
@Inject
public CmdLineParser(final Injector injector, @Assisted final Object bean)
throws IllegalAnnotationError {
this.injector = injector;
this.parser = new MyParser(bean);
}
public void addArgument(Setter<?> setter, Argument a) {
parser.addArgument(setter, a);
}
public void addOption(Setter<?> setter, Option o) {
parser.addOption(setter, o);
}
public void printSingleLineUsage(Writer w, ResourceBundle rb) {
parser.printSingleLineUsage(w, rb);
}
public void printUsage(Writer out, ResourceBundle rb) {
parser.printUsage(out, rb);
}
public void parseArgument(final String... args) throws CmdLineException {
final ArrayList<String> tmp = new ArrayList<String>(args.length);
for (int argi = 0; argi < args.length; argi++) {
final String str = args[argi];
if (str.equals("--")) {
while (argi < args.length)
tmp.add(args[argi++]);
break;
}
if (str.startsWith("--")) {
final int eq = str.indexOf('=');
if (eq > 0) {
tmp.add(str.substring(0, eq));
tmp.add(str.substring(eq + 1));
continue;
}
}
tmp.add(str);
}
parser.parseArgument(tmp.toArray(new String[tmp.size()]));
}
private class MyParser extends org.kohsuke.args4j.CmdLineParser {
MyParser(final Object bean) {
super(bean);
}
@SuppressWarnings("unchecked")
@Override
protected OptionHandler createOptionHandler(final OptionDef option,
final Setter setter) {
if (isHandlerSpecified(option) || isEnum(setter) || isPrimitive(setter)) {
return super.createOptionHandler(option, setter);
}
final Key<OptionHandlerFactory<?>> key =
OptionHandlerUtil.keyFor(setter.getType());
Injector i = injector;
while (i != null) {
if (i.getBindings().containsKey(key)) {
return i.getInstance(key).create(this, option, setter);
}
i = i.getParent();
}
return super.createOptionHandler(option, setter);
}
private boolean isHandlerSpecified(final OptionDef option) {
return option.handler() != OptionHandler.class;
}
@SuppressWarnings("unchecked")
private boolean isEnum(final Setter setter) {
return Enum.class.isAssignableFrom(setter.getType());
}
@SuppressWarnings("unchecked")
private boolean isPrimitive(final Setter setter) {
return setter.getType().isPrimitive();
}
}
}
|
