com.kmwllc.lucille.APIApplication Maven / Gradle / Ivy

Go to download
package com.kmwllc.lucille;

import com.kmwllc.lucille.auth.BasicAuthenticator;
import com.kmwllc.lucille.config.AuthConfiguration.AuthType;
import com.kmwllc.lucille.config.LucilleAPIConfiguration;
import com.kmwllc.lucille.core.RunnerManager;
import com.kmwllc.lucille.endpoints.LivenessResource;
import com.kmwllc.lucille.endpoints.LucilleResource;
import com.kmwllc.lucille.endpoints.ReadinessResource;
import io.dropwizard.auth.AuthDynamicFeature;
import io.dropwizard.auth.AuthValueFactoryProvider;
import io.dropwizard.auth.PrincipalImpl;
import io.dropwizard.auth.basic.BasicCredentialAuthFilter;
import io.dropwizard.core.Application;
import io.dropwizard.core.setup.Bootstrap;
import io.dropwizard.core.setup.Environment;
import jdk.jfr.Experimental;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Main Class for the Lucille API.
 */
@Experimental
public class APIApplication extends Application {

  public static final Logger log = LoggerFactory.getLogger(APIApplication.class);

  @Override
  public String getName() {
    return "lucille-api";
  }

  // Turn off the default Dropwizard Logging
  @Override
  protected void bootstrapLogging() {}

  @Override
  public void initialize(Bootstrap bootstrap) {}

  @Override
  public void run(LucilleAPIConfiguration config, Environment env) throws Exception {
    RunnerManager runnerManager = RunnerManager.getInstance();

    // Enable Basic Auth
    if (config.getAuthConfig().getType().equals(AuthType.BASIC_AUTH)) {
      env.jersey().register(new AuthDynamicFeature(
          new BasicCredentialAuthFilter.Builder()
              .setAuthenticator(new BasicAuthenticator(config.getAuthConfig().getPassword()))
              .buildAuthFilter()
      ));
    } else {
      throw new Exception("No Auth configured for the Lucille Admin API.");
    }

    // Register our 3 Resources
    env.jersey().register(new LucilleResource(runnerManager));
    env.jersey().register(new LivenessResource());
    env.jersey().register(new ReadinessResource());
    env.jersey().register(new AuthValueFactoryProvider.Binder<>(PrincipalImpl.class));
  }

  public static void main(String[] args) throws Exception {
    new APIApplication().run(args);
  }
}