java/dagger/internal/codegen/javac/JavacPluginModule.java

/*
 * Copyright (C) 2017 The Dagger Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package dagger.internal.codegen.javac;

import androidx.room.compiler.processing.XMessager;
import androidx.room.compiler.processing.XProcessingEnv;
import androidx.room.compiler.processing.compat.XConverters;
import com.sun.tools.javac.model.JavacElements;
import com.sun.tools.javac.model.JavacTypes;
import com.sun.tools.javac.util.Context;
import dagger.Binds;
import dagger.Module;
import dagger.Provides;
import dagger.internal.codegen.binding.BindingGraphFactory;
import dagger.internal.codegen.binding.ComponentDescriptorFactory;
import dagger.internal.codegen.compileroption.CompilerOptions;
import dagger.internal.codegen.langmodel.DaggerElements;
import dagger.internal.codegen.langmodel.DaggerTypes;
import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.util.Elements; // ALLOW_TYPES_ELEMENTS
import javax.lang.model.util.Types; // ALLOW_TYPES_ELEMENTS

/**
 * A module that provides a {@link BindingGraphFactory} and {@link ComponentDescriptorFactory} for
 * use in {@code javac} plugins. Requires a binding for the {@code javac} {@link Context}.
 */
@Module(includes = JavacPluginModule.BindsModule.class)
public final class JavacPluginModule {
  @Module
  interface BindsModule {
    @Binds
    CompilerOptions compilerOptions(JavacPluginCompilerOptions compilerOptions);
  }

  private final XProcessingEnv processingEnv;

  public JavacPluginModule(Context context) {
    this(JavacElements.instance(context), JavacTypes.instance(context));
  }

  public JavacPluginModule(Elements elements, Types types) {
    this.processingEnv =
        XProcessingEnv.create(new JavacPluginProcessingEnvironment(elements, types));
  }

  @Provides
  XMessager messager() {
    return processingEnv.getMessager();
  }

  @Provides
  DaggerElements daggerElements() {
    ProcessingEnvironment env = XConverters.toJavac(processingEnv);
    return new DaggerElements(env.getElementUtils(), env.getTypeUtils()); // ALLOW_TYPES_ELEMENTS
  }

  @Provides
  DaggerTypes daggerTypes(DaggerElements elements) {
    ProcessingEnvironment env = XConverters.toJavac(processingEnv);
    return new DaggerTypes(env.getTypeUtils(), elements); // ALLOW_TYPES_ELEMENTS
  }

  @Provides
  XProcessingEnv xProcessingEnv() {
    return processingEnv;
  }
}