package com.atlassian.asap.core.server.jersey;
   
   import com.atlassian.asap.api.Jwt;
   import org.glassfish.hk2.api.Factory;
   import org.glassfish.hk2.api.InjectionResolver;
   import org.glassfish.hk2.api.ServiceLocator;
   import org.glassfish.hk2.api.TypeLiteral;
   import org.glassfish.hk2.utilities.binding.AbstractBinder;
   import org.glassfish.jersey.server.internal.inject.AbstractContainerRequestValueFactory;
  import org.glassfish.jersey.server.internal.inject.AbstractValueFactoryProvider;
  import org.glassfish.jersey.server.internal.inject.MultivaluedParameterExtractorProvider;
  import org.glassfish.jersey.server.internal.inject.ParamInjectionResolver;
  import org.glassfish.jersey.server.model.Parameter;
  import org.glassfish.jersey.server.spi.internal.ValueFactoryProvider;
  
  import javax.inject.Inject;
  import javax.inject.Singleton;
  
  import static com.atlassian.asap.core.server.filter.AbstractRequestAuthenticationFilter.AUTHENTIC_JWT_REQUEST_ATTRIBUTE;
  
  /**
   * JwtParamBinder can be registered with Jersey to provide a JwtParam to Jwt token binding on resource methods.  Requires
   * the {@link AuthorizationRequestFilter} to be installed.
   */
  public class JwtParamBinder extends AbstractBinder {
  
      @Override
      protected void configure() {
          bind(JwtParamInjectResolver.class)
                  .to(new TypeLiteral<InjectionResolver<JwtParam>>() {
                  })
                  .in(Singleton.class);
          bind(JwtValueProvider.class)
                  .to(ValueFactoryProvider.class)
                  .in(Singleton.class);
      }
  
      private static class JwtParamInjectResolver extends ParamInjectionResolver<JwtParam> {
          @SuppressWarnings("checkstyle:RedundantModifier")
          public JwtParamInjectResolver() {
              super(JwtValueProvider.class);
          }
      }
  
      private static class JwtValueFactory extends AbstractContainerRequestValueFactory<Jwt> {
          @Override
          public Jwt provide() {
              Object obj = getContainerRequest().getProperty(AUTHENTIC_JWT_REQUEST_ATTRIBUTE);
              return obj instanceof Jwt ? (Jwt) obj : null;
          }
  
          @Override
          public void dispose(Jwt instance) {
          }
      }
  
      private static class JwtValueProvider extends AbstractValueFactoryProvider {
          @SuppressWarnings("checkstyle:RedundantModifier")
          @Inject
          public JwtValueProvider(MultivaluedParameterExtractorProvider extractorProvider, ServiceLocator locator) {
              super(extractorProvider, locator, Parameter.Source.UNKNOWN);
          }
  
          @Override
          protected Factory<?> createValueFactory(Parameter parameter) {
              if (!parameter.isAnnotationPresent(JwtParam.class) || !Jwt.class.equals(parameter.getRawType())) {
                  return null;
              }
              return new JwtValueFactory();
          }
      }
  }