minium.tools.fs.web.method.support.AntPathHandlerMethodArgumentResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of minium-tools-fs Show documentation
There is a newer version: 1.4.1
package minium.tools.fs.web.method.support;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.Arrays;

import javax.servlet.http.HttpServletRequest;

import org.springframework.core.MethodParameter;
import org.springframework.stereotype.Component;
import org.springframework.util.AntPathMatcher;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.context.request.ServletWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;
import org.springframework.web.servlet.HandlerMapping;

import com.google.common.base.Charsets;
import com.google.common.base.Function;
import com.google.common.base.Joiner;
import com.google.common.collect.Collections2;


@Component
public class AntPathHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {

	private static final String URL_PATH_SEPARATOR = "/";

	@Override
	public Object resolveArgument(MethodParameter methodParameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest,
			WebDataBinderFactory binderFactory) {
		HttpServletRequest request = ((ServletWebRequest) webRequest).getRequest();

		AntPath antMatcherPath = methodParameter.getParameterAnnotation(AntPath.class);
		if (!antMatcherPath.value().isEmpty()) {
			// first, we try to get the path value from the corresponding
			// parameter
			String param = antMatcherPath.value();
			String parameter = request.getParameter(param);
			if (parameter != null) {
				return parameter;
			}
		}

		// http://stackoverflow.com/questions/3686808/spring-3-requestmapping-get-path-value
		String pattern = (String) request.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE);
		String path = (String) request.getAttribute(HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE);

		String extractedPaths = new AntPathMatcher().extractPathWithinPattern(pattern, path);
		return decode(extractedPaths);
	}

	@Override
	public boolean supportsParameter(MethodParameter methodParameter) {
		AntPath antMatcherPath = methodParameter.getParameterAnnotation(AntPath.class);
		return antMatcherPath != null;
	}

    String decode(String extractedPaths) {
    	return Joiner.on(URL_PATH_SEPARATOR).join(Collections2.transform(Arrays.asList(extractedPaths.split(URL_PATH_SEPARATOR)), new Function() {
    		@Override
    		public String apply(String input) {
    			try {
    				return URLDecoder.decode(input, Charsets.UTF_8.name());
    			} catch (UnsupportedEncodingException e) {
    				return input;
    			}
    		}
    	}));
    }
}