82 lines
2.8 KiB
Java
82 lines
2.8 KiB
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;
|
||
|
|
}
|
||
|
|
}
|