io.jooby.guice.JoobyModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jooby-guice Show documentation
There is a newer version: 3.5.3
/*
 * Jooby https://jooby.io
 * Apache License Version 2.0 https://jooby.io/LICENSE.txt
 * Copyright 2014 Edgar Espina
 */
package io.jooby.guice;

import java.lang.reflect.Type;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import com.google.inject.AbstractModule;
import com.google.inject.Key;
import com.google.inject.binder.LinkedBindingBuilder;
import com.google.inject.name.Names;
import com.google.inject.util.Types;
import com.typesafe.config.Config;
import com.typesafe.config.ConfigObject;
import edu.umd.cs.findbugs.annotations.NonNull;
import io.jooby.Environment;
import io.jooby.Jooby;
import io.jooby.ServiceKey;
import io.jooby.ServiceRegistry;
import jakarta.inject.Provider;

/**
 * Exposes Jooby objects to Guice. This module exposes {@link Environment}, {@link Config} and
 * application services.
 *
 * @author edgar
 * @since 2.0.0
 */
public class JoobyModule extends AbstractModule {
  private Jooby application;

  /**
   * Creates a new Jooby Module.
   *
   * @param application Jooby application.
   */
  public JoobyModule(@NonNull Jooby application) {
    this.application = application;
  }

  @Override
  protected void configure() {
    configureEnv(application.getEnvironment());
    configureResources(application.getServices());
  }

  private void configureResources(ServiceRegistry registry) {
    // bind the available resources as well, supporting the name annotations that may be set
    for (Map.Entry, Provider> entry : registry.entrySet()) {
      ServiceKey key = entry.getKey();
      Provider provider = entry.getValue();
      LinkedBindingBuilder binding;
      if (key.getName() != null) {
        binding = bind(key.getType()).annotatedWith(Names.named(key.getName()));
      } else {
        binding = bind(key.getType());
      }
      binding.toProvider(provider);
    }
  }

  /*package*/ void configureEnv(Environment env) {
    Config config = env.getConfig();
    // root nodes
    traverse("", config.root());

    // configuration properties
    for (var entry : config.entrySet()) {
      var name = entry.getKey();
      var named = Names.named(name);
      var value = entry.getValue().unwrapped();

      if (value instanceof List values) {
        componentType(values)
            .forEach(
                componentType -> {
                  Type listType = Types.listOf(componentType);
                  Key key = Key.get(listType, Names.named(name));
                  bind(key).toInstance(values);
                });
        value = values.stream().map(Object::toString).collect(Collectors.joining(","));
      }
      bindConstant().annotatedWith(named).to(value.toString());
    }
  }

  /*package*/ void traverse(String p, ConfigObject root) {
    root.forEach(
        (n, v) -> {
          if (v instanceof ConfigObject child) {
            var path = p + n;
            var named = Names.named(path);
            bind(Config.class).annotatedWith(named).toProvider(child::toConfig);
            traverse(path + ".", child);
          }
        });
  }

  private Stream componentType(List values) {
    if (values.isEmpty()) {
      // For empty list we generates a binding for primitive wrappers.
      return Stream.of(
          String.class,
          Integer.class,
          Long.class,
          Float.class,
          Double.class,
          Boolean.class,
          Object.class);
    }
    return Stream.of(values.get(0).getClass());
  }
}