FileFieldArgumentResolver.java

package com.github.dynamicextensionsalfresco.webscripts.arguments;

import java.lang.annotation.Annotation;

import com.github.dynamicextensionsalfresco.webscripts.AnnotationWebScriptRequest;
import com.github.dynamicextensionsalfresco.webscripts.WebScriptUtil;
import com.github.dynamicextensionsalfresco.webscripts.annotations.FileField;
import com.github.dynamicextensionsalfresco.webscripts.annotations.Header;

import org.springframework.extensions.webscripts.WebScriptRequest;
import org.springframework.extensions.webscripts.WebScriptResponse;
import org.springframework.extensions.webscripts.servlet.FormData.FormField;
import org.springframework.extensions.webscripts.servlet.WebScriptServletRequest;
import org.springframework.util.StringUtils;

/**
 * {@link ArgumentResolver} that maps String parameters annotated with {@link Header}.
 * 
 * @author Laurens Fridael
 * 
 */
public class FileFieldArgumentResolver implements ArgumentResolver<FormField, FileField> {

	@Override
	public boolean supports(final Class<?> argumentType, final Class<? extends Annotation> annotationType) {
		return FormField.class.equals(argumentType) && FileField.class.equals(annotationType);
	}

	@Override
	public FormField resolveArgument(final Class<?> argumentType, final FileField fileField, final String name,
			WebScriptRequest request, final WebScriptResponse response) {
		if (request instanceof AnnotationWebScriptRequest) {
			request = ((AnnotationWebScriptRequest) request).getWebScriptRequest();
		}
		FormField formField = null;
		final WebScriptServletRequest req = WebScriptUtil.extractWebScriptServletRequest(request);
		if (req != null) {
			String parameterName = fileField.value();
			if (StringUtils.hasText(parameterName) == false) {
				parameterName = name;
			}
			formField = req.getFileField(parameterName);
		}
		return formField;
	}
}