Implement describe() for JCommander
This commit is contained in:
2
pom.xml
2
pom.xml
@@ -15,7 +15,7 @@
|
||||
|
||||
<properties>
|
||||
<jline.version>3.3.0</jline.version>
|
||||
<assertj.version>3.2.0</assertj.version>
|
||||
<assertj.version>3.8.0</assertj.version>
|
||||
<java.version>1.8</java.version>
|
||||
</properties>
|
||||
|
||||
|
||||
@@ -16,7 +16,9 @@
|
||||
|
||||
package org.springframework.shell2;
|
||||
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
|
||||
import org.springframework.core.MethodParameter;
|
||||
@@ -61,7 +63,7 @@ public class ParameterDescription {
|
||||
/**
|
||||
* The list of 'keys' that can be used to specify this parameter, if any.
|
||||
*/
|
||||
private List<String> keys;
|
||||
private List<String> keys = Collections.emptyList();
|
||||
|
||||
/**
|
||||
* Depending on the {@link ParameterResolver}, whether keys are mandatory to identify this parameter.
|
||||
@@ -102,7 +104,7 @@ public class ParameterDescription {
|
||||
}
|
||||
|
||||
public ParameterDescription defaultValue(String defaultValue) {
|
||||
this.defaultValue = Optional.of(defaultValue);
|
||||
this.defaultValue = Optional.ofNullable(defaultValue);
|
||||
return this;
|
||||
}
|
||||
|
||||
@@ -145,10 +147,30 @@ public class ParameterDescription {
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return String.format("%s %s", keys().iterator().next(), formal());
|
||||
return String.format("%s %s", keys.isEmpty() ? "" : keys().iterator().next(), formal());
|
||||
}
|
||||
|
||||
public MethodParameter parameter() {
|
||||
return parameter;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean equals(Object o) {
|
||||
if (this == o) return true;
|
||||
if (o == null || getClass() != o.getClass()) return false;
|
||||
ParameterDescription that = (ParameterDescription) o;
|
||||
return mandatoryKey == that.mandatoryKey &&
|
||||
Objects.equals(parameter, that.parameter) &&
|
||||
Objects.equals(type, that.type) &&
|
||||
Objects.equals(formal, that.formal) &&
|
||||
Objects.equals(defaultValue, that.defaultValue) &&
|
||||
Objects.equals(defaultValueWhenFlag, that.defaultValueWhenFlag) &&
|
||||
Objects.equals(keys, that.keys) &&
|
||||
Objects.equals(help, that.help);
|
||||
}
|
||||
|
||||
@Override
|
||||
public int hashCode() {
|
||||
return Objects.hash(parameter, type, formal, defaultValue, defaultValueWhenFlag, keys, mandatoryKey, help);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -20,6 +20,8 @@ import java.lang.reflect.Constructor;
|
||||
import java.lang.reflect.Executable;
|
||||
import java.lang.reflect.Method;
|
||||
import java.lang.reflect.Parameter;
|
||||
import java.util.stream.IntStream;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
import org.springframework.core.DefaultParameterNameDiscoverer;
|
||||
import org.springframework.core.MethodParameter;
|
||||
@@ -76,4 +78,9 @@ public class Utils {
|
||||
return methodParameter;
|
||||
}
|
||||
|
||||
public static Stream<MethodParameter> createMethodParameters(Executable executable) {
|
||||
return IntStream.range(0, executable.getParameterCount())
|
||||
.mapToObj(i -> createMethodParameter(executable, i));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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<>();
|
||||
|
||||
@@ -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);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -238,18 +238,10 @@ public class Help {
|
||||
}
|
||||
|
||||
private List<ParameterDescription> getParameterDescriptions(MethodTarget methodTarget) {
|
||||
Parameter[] parameters = methodTarget.getMethod().getParameters();
|
||||
List<ParameterDescription> parameterDescriptions = new ArrayList<>();
|
||||
for (int i = 0, parametersLength = parameters.length; i < parametersLength; i++) {
|
||||
for (ParameterResolver parameterResolver : parameterResolvers) {
|
||||
MethodParameter methodParameter = Utils.createMethodParameter(methodTarget.getMethod(), i);
|
||||
if (parameterResolver.supports(methodParameter)) {
|
||||
parameterDescriptions.add(parameterResolver.describe(methodParameter).findFirst().get());
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
return parameterDescriptions;
|
||||
return Utils.createMethodParameters(methodTarget.getMethod())
|
||||
.flatMap(mp -> parameterResolvers.stream().filter(pr -> pr.supports(mp)).limit(1L).flatMap(pr -> pr.describe(mp)))
|
||||
.collect(Collectors.toList());
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user