use of net.starlark.java.eval.NoneType in project copybara by google.
the class ModuleLoader method processStarlarkMethod.
private DocFunction processStarlarkMethod(Method method, StarlarkMethod annotation, @Nullable String prefix) {
Type[] genericParameterTypes = method.getGenericParameterTypes();
Param[] starlarkParams = annotation.parameters();
if (genericParameterTypes.length < starlarkParams.length) {
throw new IllegalStateException(String.format("Missing java parameters for: %s\n" + "%s\n" + "%s", method, Arrays.toString(genericParameterTypes), Arrays.toString(starlarkParams)));
}
ImmutableList.Builder<DocParam> params = ImmutableList.builder();
Map<String, DocDefault> docDefaultsMap = stream(method.getAnnotationsByType(DocDefault.class)).collect(Collectors.toMap(DocDefault::field, identity(), (f, v) -> v));
for (int i = 0; i < starlarkParams.length; i++) {
Type parameterType = genericParameterTypes[i];
Param starlarkParam = starlarkParams[i];
// Compute the list of names of allowed types (e.g. string or bool or NoneType).
List<String> allowedTypeNames = new ArrayList<>();
if (starlarkParam.allowedTypes().length > 0) {
for (ParamType param : starlarkParam.allowedTypes()) {
allowedTypeNames.add(skylarkTypeName(param.type()) + (param.generic1() != Object.class ? " of " + skylarkTypeName(param.generic1()) : ""));
}
} else {
// Otherwise use the type of the parameter variable itself.
allowedTypeNames.add(skylarkTypeName(parameterType));
}
DocDefault fieldInfo = docDefaultsMap.get(starlarkParam.name());
if (fieldInfo != null && fieldInfo.allowedTypes().length > 0) {
allowedTypeNames = Arrays.asList(fieldInfo.allowedTypes());
}
params.add(new DocParam(starlarkParam.name(), fieldInfo != null ? fieldInfo.value() : emptyToNull(starlarkParam.defaultValue()), allowedTypeNames, starlarkParam.doc()));
}
String returnType = method.getGenericReturnType().equals(NoneType.class) || method.getGenericReturnType().equals(void.class) ? null : skylarkTypeName(method.getGenericReturnType());
return new DocFunction(prefix != null ? prefix + "." + annotation.name() : annotation.name(), annotation.doc(), returnType, params.build(), generateFlagsInfo(method), stream(method.getAnnotationsByType(Example.class)).map(DocExample::new).collect(toImmutableList()));
}
Aggregations