RegistryServiceConfiguration.java

package org.gbif.checklistbank.config;

import com.google.common.base.Strings;
import org.gbif.checklistbank.registry.FileRegistryModule;
import org.gbif.registry.ws.client.guice.RegistryWsClientModule;
import org.gbif.ws.client.guice.AnonymousAuthModule;
import org.gbif.ws.client.guice.SingleUserAuthModule;

import java.io.File;
import java.util.List;
import java.util.Properties;
import javax.validation.Valid;
import javax.validation.constraints.NotNull;

import com.beust.jcommander.ParametersDelegate;
import com.google.common.collect.Lists;
import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.Module;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Configuration needed to anonymously connect to the registry webservices.
 */
@SuppressWarnings("PublicField")
public class RegistryServiceConfiguration {
  private static final Logger LOG = LoggerFactory.getLogger(RegistryServiceConfiguration.class);

  @ParametersDelegate
  @Valid
  @NotNull
  public String wsUrl = "http://api.gbif.org/v1";

  public String user;

  public String password;

  public Injector createRegistryInjector(Module ... otherModules) {
    List<Module> modules = Lists.newArrayList(otherModules);
    modules.addAll(guiceModules());

    Injector injClient = Guice.createInjector(modules);
    return injClient;
  }

  public List<Module> guiceModules() {
    // check if wsUrl is a http(s) protocol
    if (wsUrl.startsWith("http")) {
      LOG.info("Connecting to registry services at {}", wsUrl);
      Properties props = new Properties();
      props.put("registry.ws.url", wsUrl);

      List<Module> modules = Lists.newArrayList();
      if (!Strings.isNullOrEmpty(user) && !Strings.isNullOrEmpty(password)) {
        modules.add( new SingleUserAuthModule(user, password) );
      } else {
        modules.add( new AnonymousAuthModule() );
      }
      modules.add(new RegistryWsClientModule(props));
      return modules;

    } else {
      // if not we take it as a file name and use an über simple file based registry implementation
      LOG.info("Loading registry datasets from {}", wsUrl);
      return Lists.newArrayList(new FileRegistryModule(new File(wsUrl)));
    }
  }

}