|
4 | 4 |
|
5 | 5 | package com.icerockdev.validation |
6 | 6 |
|
| 7 | +import org.hibernate.validator.internal.constraintvalidators.AbstractEmailValidator |
| 8 | +import org.hibernate.validator.internal.util.logging.LoggerFactory |
| 9 | +import java.lang.invoke.MethodHandles |
| 10 | +import java.util.regex.Pattern.compile |
| 11 | +import java.util.regex.PatternSyntaxException |
7 | 12 | import javax.validation.Constraint |
| 13 | +import javax.validation.ConstraintValidatorContext |
8 | 14 | import javax.validation.Payload |
9 | | -import javax.validation.constraints.Email |
10 | 15 | import javax.validation.constraints.Pattern |
| 16 | +import kotlin.annotation.AnnotationTarget.ANNOTATION_CLASS |
| 17 | +import kotlin.annotation.AnnotationTarget.CONSTRUCTOR |
| 18 | +import kotlin.annotation.AnnotationTarget.FIELD |
| 19 | +import kotlin.annotation.AnnotationTarget.FUNCTION |
| 20 | +import kotlin.annotation.AnnotationTarget.TYPE_PARAMETER |
| 21 | +import kotlin.annotation.AnnotationTarget.VALUE_PARAMETER |
11 | 22 | import kotlin.reflect.KClass |
12 | 23 |
|
13 | | -@Target(AnnotationTarget.FIELD) |
14 | | -@Email |
15 | | -@Pattern(regexp = ".+@.+\\..+", message = "{javax.validation.constraints.Email.message}") |
16 | | -@Constraint(validatedBy = []) |
17 | | -@MustBeDocumented |
| 24 | +const val DEFAULT_EMAIL_REGEXP = ".+@.+\\..+" |
| 25 | + |
| 26 | +/** |
| 27 | + * Annotation to validate email |
| 28 | + * |
| 29 | + * Example, validate field |
| 30 | + * @field:StrictEmail(message = "Invalid email") |
| 31 | + */ |
| 32 | +@Constraint(validatedBy = [StrictEmailValidator::class]) |
| 33 | +@Target(allowedTargets = [FUNCTION, FIELD, ANNOTATION_CLASS, CONSTRUCTOR, VALUE_PARAMETER, TYPE_PARAMETER]) |
| 34 | +@kotlin.annotation.Retention(AnnotationRetention.RUNTIME) |
18 | 35 | annotation class StrictEmail( |
19 | 36 | val message: String = "{javax.validation.constraints.Email.message}", |
20 | 37 | val groups: Array<KClass<*>> = [], |
21 | | - val payload: Array<KClass<out Payload>> = [] |
| 38 | + val payload: Array<KClass<out Payload>> = [], |
| 39 | + val regexp: String = DEFAULT_EMAIL_REGEXP, |
| 40 | + val flags: Array<Pattern.Flag> = [] |
22 | 41 | ) |
| 42 | + |
| 43 | +class StrictEmailValidator : AbstractEmailValidator<StrictEmail?>() { |
| 44 | + private var pattern: java.util.regex.Pattern = compile(DEFAULT_EMAIL_REGEXP, 0) |
| 45 | + override fun initialize(constraintAnnotation: StrictEmail?) { |
| 46 | + super.initialize(constraintAnnotation) |
| 47 | + val flags: Array<Pattern.Flag> = constraintAnnotation?.flags ?: arrayOf() |
| 48 | + var intFlag = 0 |
| 49 | + for (flag in flags) { |
| 50 | + intFlag = intFlag.or(flag.value) |
| 51 | + } |
| 52 | + |
| 53 | + if (DEFAULT_EMAIL_REGEXP != constraintAnnotation?.regexp || constraintAnnotation.flags.isNotEmpty()) { |
| 54 | + pattern = try { |
| 55 | + compile(constraintAnnotation?.regexp ?: DEFAULT_EMAIL_REGEXP, intFlag) |
| 56 | + } catch (e: PatternSyntaxException) { |
| 57 | + throw logger.getInvalidRegularExpressionException(e) |
| 58 | + } |
| 59 | + } |
| 60 | + } |
| 61 | + |
| 62 | + override fun isValid(value: CharSequence, context: ConstraintValidatorContext): Boolean { |
| 63 | + val isValid = super.isValid(value, context) |
| 64 | + if (!isValid) { |
| 65 | + return isValid |
| 66 | + } |
| 67 | + val m = pattern.matcher(value) |
| 68 | + return m.matches() |
| 69 | + } |
| 70 | + |
| 71 | + companion object { |
| 72 | + private val logger = LoggerFactory.make(MethodHandles.lookup()) |
| 73 | + } |
| 74 | +} |
0 commit comments