package com.mangofactory.swagger.readers.operation.parameter;
import com.google.common.base.Splitter;
import com.mangofactory.swagger.readers.Command;
import com.mangofactory.swagger.scanners.RequestMappingContext;
import com.wordnik.swagger.annotations.ApiParam;
import com.wordnik.swagger.model.AllowableListValues;
import com.wordnik.swagger.model.AllowableRangeValues;
import com.wordnik.swagger.model.AllowableValues;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.MethodParameter;
import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import static com.google.common.collect.Lists.*;
import static com.mangofactory.swagger.ScalaUtils.*;
import static org.apache.commons.lang.StringUtils.*;
public class ParameterAllowableReader implements Command<RequestMappingContext> {
private static final Logger log = LoggerFactory.getLogger(ParameterAllowableReader.class);
@Override
public void execute(RequestMappingContext context) {
MethodParameter methodParameter = (MethodParameter) context.get("methodParameter");
AllowableValues allowableValues = null;
String allowableValueString = findAnnotatedAllowableValues(methodParameter);
if (allowableValueString != null && !"".equals(allowableValueString)) {
allowableValues = ParameterAllowableReader.allowableValueFromString(allowableValueString);
} else {
if (methodParameter.getParameterType().isEnum()) {
Object[] enumConstants = methodParameter.getParameterType().getEnumConstants();
List<String> enumNames = new ArrayList<String>();
for (Object o : enumConstants) {
enumNames.add(o.toString());
}
allowableValues = new AllowableListValues(toScalaList(newArrayList(enumNames)), "LIST");
}
if (methodParameter.getParameterType().isArray()
&& methodParameter.getParameterType().getComponentType().isEnum()) {
Object[] enumConstants = methodParameter.getParameterType().getComponentType().getEnumConstants();
List<String> enumNames = new ArrayList<String>();
for (Object o : enumConstants) {
enumNames.add(o.toString());
}
allowableValues = new AllowableListValues(toScalaList(newArrayList(enumNames)), "LIST");
}
}
context.put("allowableValues", allowableValues);
}
public static AllowableValues allowableValueFromString(String allowableValueString) {
AllowableValues allowableValues = null;
allowableValueString = allowableValueString.trim().replaceAll(" ", "");
if (allowableValueString.startsWith("range[")) {
allowableValueString = allowableValueString.replaceAll("range\\[", "").replaceAll("]", "");
Iterable<String> split = Splitter.on(',').trimResults().omitEmptyStrings().split(allowableValueString);
List<String> ranges = newArrayList(split);
allowableValues = new AllowableRangeValues(ranges.get(0), ranges.get(1));
} else if (allowableValueString.contains(",")) {
Iterable<String> split = Splitter.on(',').trimResults().omitEmptyStrings().split(allowableValueString);
allowableValues = new AllowableListValues(toScalaList(newArrayList(split)), "LIST");
} else if (!isBlank(allowableValueString)) {
List<String> singleVal = Arrays.asList(allowableValueString.trim());
allowableValues = new AllowableListValues(toScalaList(singleVal), "LIST");
}
return allowableValues;
}
private String findAnnotatedAllowableValues(MethodParameter methodParameter) {
Annotation[] methodAnnotations = methodParameter.getParameterAnnotations();
if (null != methodAnnotations) {
for (Annotation annotation : methodAnnotations) {
if (annotation instanceof ApiParam) {
return ((ApiParam) annotation).allowableValues();
}
}
}
return null;
}
}