Implement describe() for JCommander

This commit is contained in:
Eric Bottard
2017-06-22 18:57:06 +02:00
parent e49ee885ef
commit a277c53d8d
7 changed files with 88 additions and 24 deletions

View File

@@ -16,6 +16,8 @@
package org.springframework.shell2.jcommander;
import static org.springframework.shell2.Utils.unCamelify;
import java.lang.annotation.Annotation;
import java.util.Arrays;
import java.util.Collection;
@@ -34,6 +36,7 @@ import org.springframework.shell2.CompletionContext;
import org.springframework.shell2.CompletionProposal;
import org.springframework.shell2.ParameterDescription;
import org.springframework.shell2.ParameterResolver;
import org.springframework.shell2.Utils;
import org.springframework.stereotype.Component;
import org.springframework.util.ReflectionUtils;
@@ -74,17 +77,33 @@ public class JCommanderParameterResolver implements ParameterResolver {
@Override
public Object resolve(MethodParameter methodParameter, List<String> words) {
Object pojo = BeanUtils.instantiateClass(methodParameter.getParameterType());
JCommander jCommander = new JCommander();
jCommander.addObject(pojo);
jCommander.setAcceptUnknownOptions(true);
JCommander jCommander = createJCommander(methodParameter);
jCommander.parse(words.toArray(new String[words.size()]));
return pojo;
return jCommander.getObjects().get(0);
}
private JCommander createJCommander(MethodParameter methodParameter) {
Object pojo = BeanUtils.instantiateClass(methodParameter.getParameterType());
JCommander jCommander = new JCommander(pojo);
jCommander.setAcceptUnknownOptions(true);
return jCommander;
}
@Override
public Stream<ParameterDescription> describe(MethodParameter parameter) {
throw new UnsupportedOperationException();
JCommander jCommander = createJCommander(parameter);
com.beust.jcommander.ParameterDescription mainParameter = jCommander.getMainParameter();
return Stream.concat(
jCommander.getParameters().stream(),
mainParameter != null ? Stream.of(mainParameter) : Stream.empty()
)
.map(j -> new ParameterDescription(parameter, unCamelify(j.getParameterized().getType().getSimpleName()))
.keys(Arrays.asList(j.getParameter().names()))
.help(j.getDescription())
.mandatoryKey(!j.equals(mainParameter))
// Not ideal as this does not take reverse-conversion into account, but just toString()
.defaultValue(j.getDefault() == null ? "" : String.valueOf(j.getDefault()))
);
}
@Override

View File

@@ -29,11 +29,11 @@ import com.beust.jcommander.Parameter;
*/
public class FieldCollins {
@Parameter(names = "--name")
@Parameter(names = {"--name", "-n"}, description = "what's in a name?")
private String name;
@Parameter(names = "-level")
private int level;
private int level = 3;
@Parameter(description = "rest")
private List<String> rest = new ArrayList<>();

View File

@@ -17,13 +17,18 @@
package org.springframework.shell2.jcommander;
import static java.util.Arrays.asList;
import static java.util.Collections.singletonList;
import static org.assertj.core.api.Assertions.assertThat;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Collections;
import java.util.stream.Stream;
import org.junit.Test;
import org.springframework.core.MethodParameter;
import org.springframework.shell2.ParameterDescription;
import org.springframework.shell2.Utils;
import org.springframework.util.ReflectionUtils;
@@ -61,4 +66,23 @@ public class JCommanderParameterResolverTest {
assertThat(resolved.getLevel()).isEqualTo(2);
assertThat(resolved.getRest()).containsOnlyOnce("something-else", "yet-something-else");
}
@Test
public void testDescribe() {
MethodParameter methodParameter = Utils.createMethodParameter(COMMAND_METHOD, 0);
Stream<ParameterDescription> desciptions = resolver.describe(methodParameter);
ParameterDescription name = new ParameterDescription(methodParameter, "string")
.keys(Arrays.asList("--name", "-n"))
.help("what's in a name?")
.defaultValue("");
ParameterDescription level = new ParameterDescription(methodParameter, "int")
.keys(singletonList("-level"))
.defaultValue("3");
ParameterDescription rest = new ParameterDescription(methodParameter, "list")
.defaultValue("[]")
.mandatoryKey(false)
.help("rest");
assertThat(desciptions).contains(name, level, rest);
}
}