94 lines
3.6 KiB
Java
94 lines
3.6 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.writing;
|
|
|
|
import static androidx.room.compiler.processing.compat.XConverters.toJavac;
|
|
|
|
import com.squareup.javapoet.ClassName;
|
|
import com.squareup.javapoet.CodeBlock;
|
|
import dagger.assisted.Assisted;
|
|
import dagger.assisted.AssistedFactory;
|
|
import dagger.assisted.AssistedInject;
|
|
import dagger.internal.Preconditions;
|
|
import dagger.internal.codegen.binding.BindingGraph;
|
|
import dagger.internal.codegen.binding.ComponentRequirement;
|
|
import dagger.internal.codegen.binding.ProvisionBinding;
|
|
import dagger.internal.codegen.compileroption.CompilerOptions;
|
|
import dagger.internal.codegen.javapoet.Expression;
|
|
|
|
/** A binding expression for component provision methods. */
|
|
final class ComponentProvisionRequestRepresentation extends RequestRepresentation {
|
|
private final ProvisionBinding binding;
|
|
private final BindingGraph bindingGraph;
|
|
private final ComponentRequirementExpressions componentRequirementExpressions;
|
|
private final CompilerOptions compilerOptions;
|
|
private final boolean isExperimentalMergedMode;
|
|
|
|
@AssistedInject
|
|
ComponentProvisionRequestRepresentation(
|
|
@Assisted ProvisionBinding binding,
|
|
BindingGraph bindingGraph,
|
|
ComponentImplementation componentImplementation,
|
|
ComponentRequirementExpressions componentRequirementExpressions,
|
|
CompilerOptions compilerOptions) {
|
|
this.binding = binding;
|
|
this.bindingGraph = bindingGraph;
|
|
this.componentRequirementExpressions = componentRequirementExpressions;
|
|
this.compilerOptions = compilerOptions;
|
|
this.isExperimentalMergedMode =
|
|
componentImplementation.compilerMode().isExperimentalMergedMode();
|
|
}
|
|
|
|
@Override
|
|
Expression getDependencyExpression(ClassName requestingClass) {
|
|
CodeBlock componentDependency =
|
|
isExperimentalMergedMode
|
|
? CodeBlock.of("(($T) dependencies[0])", componentRequirement().type().getTypeName())
|
|
: getComponentRequirementExpression(requestingClass);
|
|
CodeBlock invocation =
|
|
CodeBlock.of(
|
|
"$L.$L()",
|
|
componentDependency,
|
|
toJavac(binding.bindingElement().get()).getSimpleName());
|
|
return Expression.create(
|
|
binding.contributedPrimitiveType().orElse(binding.key().type().xprocessing()),
|
|
maybeCheckForNull(binding, compilerOptions, invocation));
|
|
}
|
|
|
|
CodeBlock getComponentRequirementExpression(ClassName requestingClass) {
|
|
return componentRequirementExpressions.getExpression(componentRequirement(), requestingClass);
|
|
}
|
|
|
|
private ComponentRequirement componentRequirement() {
|
|
return bindingGraph
|
|
.componentDescriptor()
|
|
.getDependencyThatDefinesMethod(binding.bindingElement().get());
|
|
}
|
|
|
|
static CodeBlock maybeCheckForNull(
|
|
ProvisionBinding binding, CompilerOptions compilerOptions, CodeBlock invocation) {
|
|
return binding.shouldCheckForNull(compilerOptions)
|
|
? CodeBlock.of("$T.checkNotNullFromComponent($L)", Preconditions.class, invocation)
|
|
: invocation;
|
|
}
|
|
|
|
@AssistedFactory
|
|
static interface Factory {
|
|
ComponentProvisionRequestRepresentation create(ProvisionBinding binding);
|
|
}
|
|
}
|