com.github.antelopeframework.validation.FileSuffixValidator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of antelope-commons Show documentation

The newest version!

package com.github.antelopeframework.validation;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;

import org.apache.commons.lang3.StringUtils;
import org.springframework.web.multipart.MultipartFile;

public class FileSuffixValidator implements ConstraintValidator {
	
	String[] suffixs;
	
	@Override
	public void initialize(FileSuffix parameters) {
		suffixs = StringUtils.split(parameters.suffix(), ",");
	}

	@Override
	public boolean isValid(MultipartFile value, ConstraintValidatorContext context) {
		if (value == null) {
			return true;
		}
		
		String fileName = value.getOriginalFilename();
		if (!StringUtils.endsWithAny(fileName, suffixs)) {
			return false;
		}
		
		return true;
	}
}