Search code examples
javavalidationjerseyjax-rsbean-validation

Custom Validation over FormDataContentDisposition while uploading image as Multipart form data


I want to validate the file extension of image being uploaded using multipart form data in Jersey.
I have created a custom annotation ImageExtensionValidator:

package io.gupshup.dashboard.annotations;

import static java.lang.annotation.ElementType.FIELD;
import static java.lang.annotation.ElementType.PARAMETER;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

import java.lang.annotation.Retention;
import java.lang.annotation.Target;
import java.util.regex.Pattern;

import javax.validation.Constraint;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import javax.validation.Payload;

@Retention(RUNTIME)
@Target({ FIELD, PARAMETER })
@Constraint(validatedBy = { ImageExtensionValidator.Validator.class })
public @interface ImageExtensionValidator {

    String message() default "Please Enter valid Image";

    String url() default "";

    Class<?>[] groups() default {};

    Class<? extends Payload>[] payload() default {};

    class Validator implements ConstraintValidator<ImageExtensionValidator, String> {

        private String constraintMessage;
        private String fileName;
        Pattern pattern = Pattern.compile("\\.(?:jpg|gif|png|jpeg|JPG|GIF|PNG|JPEG)");

        @Override
        public void initialize(ImageExtensionValidator constraintAnnotation) {
            constraintMessage = constraintAnnotation.message();
            fileName = constraintAnnotation.url();
        }

        @Override
        public boolean isValid(String value, ConstraintValidatorContext context) {

            if (value == null || value.trim().isEmpty()) {
                return true;
            } else if (pattern.matcher(value.trim()).matches()) {
                return true;
            } else {
                context.disableDefaultConstraintViolation();
                context.buildConstraintViolationWithTemplate(constraintMessage).addConstraintViolation();
                return false;
            }

        }
    }
}

And using it like:

@FormDataParam("logo")
private InputStream logoStream;

@ImageExtensionValidator
@FormDataParam("logo")
private FormDataContentDisposition logoDetails;

I am getting the following error while using this:

javax.validation.UnexpectedTypeException: HV000030: No validator could be found for type: org.glassfish.jersey.media.multipart.FormDataContentDisposition.

I know there are other ways to do it. But something like this will make my code neater. Thanks in advance.


Solution

  • After Paul's comment I realized a mistake in my ImageExtensionValidator code. The class should be defined as:

    class Validator implements ConstraintValidator<ImageExtensionValidator,
                                                   FormDataContentDisposition> {}