com.browserup.bup.rest.validation.PortWithExistingProxyConstraint Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of browserup-proxy-rest Show documentation

BrowserUp Proxy - REST

There is a newer version: 3.0.0

package com.browserup.bup.rest.validation;

import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;

import javax.validation.Constraint;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import javax.validation.Payload;
import javax.ws.rs.core.Context;

import com.browserup.bup.proxy.MitmProxyManager;

import com.browserup.bup.rest.validation.util.MessageSanitizer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = { PortWithExistingProxyConstraint.PortWithExistingProxyConstraintValidator.class })
public @interface PortWithExistingProxyConstraint {

  String message() default "";

  String paramName() default "port";

  Class[] groups() default {};

  Class[] payload() default {};

  class PortWithExistingProxyConstraintValidator
      implements ConstraintValidator {
    private static final Logger LOG = LoggerFactory.getLogger(PortWithExistingProxyConstraintValidator.class);
    private static final String PARAM_NAME = "proxy port";

    private final MitmProxyManager proxyManager;

    public PortWithExistingProxyConstraintValidator(@Context MitmProxyManager proxyManager) {
      this.proxyManager = proxyManager;
    }

    @Override
    public void initialize(PortWithExistingProxyConstraint constraintAnnotation) {
    }

    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
      if (proxyManager.get(value) != null) {
        return true;
      }

      String escapedValue = MessageSanitizer.escape(value.toString());
      String errorMessage = String.format("No proxy server found for specified port %s", escapedValue);
      LOG.warn(errorMessage);

      context.buildConstraintViolationWithTemplate(errorMessage).addPropertyNode(PARAM_NAME).addConstraintViolation();
      return false;
    }
  }
}