es.prodevelop.pui9.data.converters.MultipartFileConverter Maven / Gradle / Ivy

Go to download
package es.prodevelop.pui9.data.converters;

import java.io.IOException;
import java.util.Collections;
import java.util.Set;

import org.apache.commons.io.FilenameUtils;
import org.springframework.core.convert.TypeDescriptor;
import org.springframework.core.convert.converter.GenericConverter;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import es.prodevelop.pui9.file.PuiDocumentDefinition;

/**
 * This class allows to set a @RequestParam parameter in your controllers
 * indicating that the type of this parameter is a {@link MultipartFile}.
 * Automatically, the value is converted into a Date using the
 * {@link PuiDocumentDefinition} class
 * 
 * @author Marc Gil - [email protected]
 */
@Component
public class MultipartFileConverter implements GenericConverter {

	@Override
	public Set getConvertibleTypes() {
		return Collections.singleton(new ConvertiblePair(MultipartFile.class, PuiDocumentDefinition.class));
	}

	@Override
	public Object convert(Object source, TypeDescriptor sourceType, TypeDescriptor targetType) {
		if (source == null) {
			return null;
		}

		MultipartFile mf = (MultipartFile) source;
		PuiDocumentDefinition pdd = new PuiDocumentDefinition();
		try {
			pdd.setInputStream(mf.getInputStream());
		} catch (IOException e) {
			// do nothing
		}
		pdd.setOriginalFileName(mf.getOriginalFilename());
		pdd.setFileName(FilenameUtils.getBaseName(mf.getOriginalFilename()));
		pdd.setFileExtension(FilenameUtils.getExtension(mf.getOriginalFilename()).toLowerCase());
		pdd.setFileSize(mf.getSize());

		return pdd;
	}

}