package org.glassfish.jersey.server.wadl.processor;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import javax.ws.rs.HttpMethod;
import javax.ws.rs.container.ContainerRequestContext;
import javax.ws.rs.core.Configuration;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.annotation.Priority;
import javax.inject.Inject;
import javax.inject.Provider;
import org.glassfish.jersey.process.Inflector;
import org.glassfish.jersey.server.ExtendedUriInfo;
import org.glassfish.jersey.server.model.ModelProcessor;
import org.glassfish.jersey.server.model.Resource;
import org.glassfish.jersey.server.model.ResourceModel;
import org.glassfish.jersey.server.model.internal.ModelProcessorUtil;
@Priority(Integer.MAX_VALUE)
public class OptionsMethodProcessor implements ModelProcessor {
private final List<ModelProcessorUtil.Method> methodList;
public OptionsMethodProcessor() {
methodList = new ArrayList<>();
methodList.add(new ModelProcessorUtil.Method(HttpMethod.OPTIONS, MediaType.WILDCARD_TYPE, MediaType.TEXT_PLAIN_TYPE,
PlainTextOptionsInflector.class));
methodList.add(new ModelProcessorUtil.Method(HttpMethod.OPTIONS, MediaType.WILDCARD_TYPE, MediaType.WILDCARD_TYPE,
GenericOptionsInflector.class));
}
private static class PlainTextOptionsInflector implements Inflector<ContainerRequestContext, Response> {
@Inject
private Provider<ExtendedUriInfo> extendedUriInfo;
@Override
public Response apply(ContainerRequestContext containerRequestContext) {
Set<String> allowedMethods = ModelProcessorUtil.getAllowedMethods(extendedUriInfo.get()
.getMatchedRuntimeResources().get(0));
final String allowedList = allowedMethods.toString();
final String optionsBody = allowedList.substring(1, allowedList.length() - 1);
return Response.ok(optionsBody, MediaType.TEXT_PLAIN_TYPE)
.allow(allowedMethods)
.build();
}
}
private static class GenericOptionsInflector implements Inflector<ContainerRequestContext, Response> {
@Inject
private Provider<ExtendedUriInfo> extendedUriInfo;
@Override
public Response apply(ContainerRequestContext containerRequestContext) {
final Set<String> allowedMethods = ModelProcessorUtil.getAllowedMethods(
(extendedUriInfo.get().getMatchedRuntimeResources().get(0)));
return Response.ok()
.allow(allowedMethods)
.header(HttpHeaders.CONTENT_LENGTH, "0")
.type(containerRequestContext.getAcceptableMediaTypes().get(0))
.build();
}
}
@Override
public ResourceModel processResourceModel(ResourceModel resourceModel, Configuration configuration) {
return ModelProcessorUtil.enhanceResourceModel(resourceModel, false, methodList, true).build();
}
@Override
public ResourceModel processSubResource(ResourceModel subResourceModel, Configuration configuration) {
return ModelProcessorUtil.enhanceResourceModel(subResourceModel, true, methodList, true).build();
}
}