blob: 7cebc10db8762f3a9ee25bc3ae85c13e8ca11edb [file] [log] [blame]
/*
* 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 com.google.auto.common.MoreElements.asExecutable;
import static com.google.auto.common.MoreElements.asType;
import static com.google.auto.common.MoreElements.asVariable;
import static com.google.common.base.CaseFormat.LOWER_CAMEL;
import static com.google.common.base.CaseFormat.UPPER_CAMEL;
import static com.google.common.base.Preconditions.checkArgument;
import static com.squareup.javapoet.MethodSpec.methodBuilder;
import static dagger.internal.codegen.base.RequestKinds.requestTypeName;
import static dagger.internal.codegen.binding.ConfigurationAnnotations.getNullableType;
import static dagger.internal.codegen.binding.SourceFiles.generatedClassNameForBinding;
import static dagger.internal.codegen.binding.SourceFiles.memberInjectedFieldSignatureForVariable;
import static dagger.internal.codegen.binding.SourceFiles.membersInjectorNameForType;
import static dagger.internal.codegen.binding.SourceFiles.protectAgainstKeywords;
import static dagger.internal.codegen.extension.DaggerStreams.toImmutableMap;
import static dagger.internal.codegen.javapoet.CodeBlocks.makeParametersCodeBlock;
import static dagger.internal.codegen.javapoet.CodeBlocks.toConcatenatedCodeBlock;
import static dagger.internal.codegen.javapoet.CodeBlocks.toParametersCodeBlock;
import static dagger.internal.codegen.javapoet.TypeNames.rawTypeName;
import static dagger.internal.codegen.langmodel.Accessibility.isElementAccessibleFrom;
import static dagger.internal.codegen.langmodel.Accessibility.isRawTypeAccessible;
import static dagger.internal.codegen.langmodel.Accessibility.isRawTypePubliclyAccessible;
import static dagger.internal.codegen.langmodel.Accessibility.isTypeAccessibleFrom;
import static java.util.stream.Collectors.toList;
import static javax.lang.model.element.Modifier.PUBLIC;
import static javax.lang.model.element.Modifier.STATIC;
import static javax.lang.model.type.TypeKind.VOID;
import com.google.auto.common.MoreElements;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.squareup.javapoet.AnnotationSpec;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.CodeBlock;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterSpec;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeVariableName;
import dagger.internal.Preconditions;
import dagger.internal.codegen.base.UniqueNameSet;
import dagger.internal.codegen.binding.AssistedInjectionAnnotations;
import dagger.internal.codegen.binding.MembersInjectionBinding.InjectionSite;
import dagger.internal.codegen.binding.ProvisionBinding;
import dagger.internal.codegen.compileroption.CompilerOptions;
import dagger.internal.codegen.extension.DaggerCollectors;
import dagger.internal.codegen.javapoet.CodeBlocks;
import dagger.internal.codegen.javapoet.TypeNames;
import dagger.internal.codegen.kotlin.KotlinMetadataUtil;
import dagger.internal.codegen.langmodel.DaggerTypes;
import dagger.model.DependencyRequest;
import dagger.model.RequestKind;
import java.util.List;
import java.util.Optional;
import java.util.function.Function;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.AnnotationMirror;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.Parameterizable;
import javax.lang.model.element.TypeElement;
import javax.lang.model.element.VariableElement;
import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror;
/** Convenience methods for creating and invoking {@link InjectionMethod}s. */
final class InjectionMethods {
/**
* A method that returns an object from a {@code @Provides} method or an {@code @Inject}ed
* constructor. Its parameters match the dependency requests for constructor and members
* injection.
*
* <p>For {@code @Provides} methods named "foo", the method name is "proxyFoo". For example:
*
* <pre><code>
* abstract class FooModule {
* {@literal @Provides} static Foo provideFoo(Bar bar, Baz baz) { … }
* }
*
* public static proxyProvideFoo(Bar bar, Baz baz) { … }
* </code></pre>
*
* <p>For {@code @Inject}ed constructors, the method name is "newFoo". For example:
*
* <pre><code>
* class Foo {
* {@literal @Inject} Foo(Bar bar) {}
* }
*
* public static Foo newFoo(Bar bar) { … }
* </code></pre>
*/
static final class ProvisionMethod {
// These names are already defined in factories and shouldn't be used for the proxy method name.
private static final ImmutableSet<String> BANNED_PROXY_NAMES = ImmutableSet.of("get", "create");
/**
* Returns a method that invokes the binding's {@linkplain ProvisionBinding#bindingElement()
* constructor} and injects the instance's members.
*/
static MethodSpec create(
ProvisionBinding binding,
CompilerOptions compilerOptions,
KotlinMetadataUtil metadataUtil) {
ExecutableElement element = asExecutable(binding.bindingElement().get());
switch (element.getKind()) {
case CONSTRUCTOR:
return constructorProxy(element);
case METHOD:
return methodProxy(
element,
methodName(element),
InstanceCastPolicy.IGNORE,
CheckNotNullPolicy.get(binding, compilerOptions),
metadataUtil);
default:
throw new AssertionError(element);
}
}
/**
* Invokes the injection method for {@code binding}, with the dependencies transformed with the
* {@code dependencyUsage} function.
*/
static CodeBlock invoke(
ProvisionBinding binding,
Function<DependencyRequest, CodeBlock> dependencyUsage,
ClassName requestingClass,
Optional<CodeBlock> moduleReference,
CompilerOptions compilerOptions,
KotlinMetadataUtil metadataUtil) {
ImmutableList.Builder<CodeBlock> arguments = ImmutableList.builder();
moduleReference.ifPresent(arguments::add);
invokeArguments(binding, dependencyUsage, requestingClass).forEach(arguments::add);
ClassName enclosingClass = generatedClassNameForBinding(binding);
MethodSpec methodSpec = create(binding, compilerOptions, metadataUtil);
return invokeMethod(methodSpec, arguments.build(), enclosingClass, requestingClass);
}
static ImmutableList<CodeBlock> invokeArguments(
ProvisionBinding binding,
Function<DependencyRequest, CodeBlock> dependencyUsage,
ClassName requestingClass) {
ImmutableMap<VariableElement, DependencyRequest> dependencyRequestMap =
binding.provisionDependencies().stream()
.collect(
toImmutableMap(
request -> MoreElements.asVariable(request.requestElement().get()),
request -> request));
ImmutableList.Builder<CodeBlock> arguments = ImmutableList.builder();
for (VariableElement parameter :
asExecutable(binding.bindingElement().get()).getParameters()) {
if (AssistedInjectionAnnotations.isAssistedParameter(parameter)) {
arguments.add(CodeBlock.of("$L", parameter.getSimpleName()));
} else if (dependencyRequestMap.containsKey(parameter)) {
DependencyRequest request = dependencyRequestMap.get(parameter);
arguments.add(
injectionMethodArgument(request, dependencyUsage.apply(request), requestingClass));
} else {
throw new AssertionError("Unexpected parameter: " + parameter);
}
}
return arguments.build();
}
private static MethodSpec constructorProxy(ExecutableElement constructor) {
TypeElement enclosingType = MoreElements.asType(constructor.getEnclosingElement());
MethodSpec.Builder builder =
methodBuilder(methodName(constructor))
.addModifiers(PUBLIC, STATIC)
.varargs(constructor.isVarArgs())
.returns(TypeName.get(enclosingType.asType()));
copyTypeParameters(builder, enclosingType);
copyThrows(builder, constructor);
CodeBlock arguments =
copyParameters(builder, new UniqueNameSet(), constructor.getParameters());
return builder.addStatement("return new $T($L)", enclosingType, arguments).build();
}
/**
* Returns {@code true} if injecting an instance of {@code binding} from {@code callingPackage}
* requires the use of an injection method.
*/
static boolean requiresInjectionMethod(
ProvisionBinding binding, CompilerOptions compilerOptions, ClassName requestingClass) {
ExecutableElement method = MoreElements.asExecutable(binding.bindingElement().get());
return !binding.injectionSites().isEmpty()
|| binding.shouldCheckForNull(compilerOptions)
|| !isElementAccessibleFrom(method, requestingClass.packageName())
// This check should be removable once we drop support for -source 7
|| method.getParameters().stream()
.map(VariableElement::asType)
.anyMatch(type -> !isRawTypeAccessible(type, requestingClass.packageName()));
}
/**
* Returns the name of the {@code static} method that wraps {@code method}. For methods that are
* associated with {@code @Inject} constructors, the method will also inject all {@link
* InjectionSite}s.
*/
private static String methodName(ExecutableElement method) {
switch (method.getKind()) {
case CONSTRUCTOR:
return "newInstance";
case METHOD:
String methodName = method.getSimpleName().toString();
return BANNED_PROXY_NAMES.contains(methodName)
? "proxy" + LOWER_CAMEL.to(UPPER_CAMEL, methodName)
: methodName;
default:
throw new AssertionError(method);
}
}
}
/**
* A static method that injects one member of an instance of a type. Its first parameter is an
* instance of the type to be injected. The remaining parameters match the dependency requests for
* the injection site.
*
* <p>Example:
*
* <pre><code>
* class Foo {
* {@literal @Inject} Bar bar;
* {@literal @Inject} void setThings(Baz baz, Qux qux) {}
* }
*
* public static injectBar(Foo instance, Bar bar) { … }
* public static injectSetThings(Foo instance, Baz baz, Qux qux) { … }
* </code></pre>
*/
static final class InjectionSiteMethod {
/**
* When a type has an inaccessible member from a supertype (e.g. an @Inject field in a parent
* that's in a different package), a method in the supertype's package must be generated to give
* the subclass's members injector a way to inject it. Each potentially inaccessible member
* receives its own method, as the subclass may need to inject them in a different order from
* the parent class.
*/
static MethodSpec create(InjectionSite injectionSite, KotlinMetadataUtil metadataUtil) {
String methodName = methodName(injectionSite);
switch (injectionSite.kind()) {
case METHOD:
return methodProxy(
asExecutable(injectionSite.element()),
methodName,
InstanceCastPolicy.CAST_IF_NOT_PUBLIC,
CheckNotNullPolicy.IGNORE,
metadataUtil);
case FIELD:
Optional<AnnotationMirror> qualifier =
injectionSite.dependencies().stream()
// methods for fields have a single dependency request
.collect(DaggerCollectors.onlyElement())
.key()
.qualifier();
return fieldProxy(asVariable(injectionSite.element()), methodName, qualifier);
}
throw new AssertionError(injectionSite);
}
/**
* Invokes each of the injection methods for {@code injectionSites}, with the dependencies
* transformed using the {@code dependencyUsage} function.
*
* @param instanceType the type of the {@code instance} parameter
*/
static CodeBlock invokeAll(
ImmutableSet<InjectionSite> injectionSites,
ClassName generatedTypeName,
CodeBlock instanceCodeBlock,
TypeMirror instanceType,
Function<DependencyRequest, CodeBlock> dependencyUsage,
DaggerTypes types,
KotlinMetadataUtil metadataUtil) {
return injectionSites.stream()
.map(
injectionSite -> {
TypeMirror injectSiteType =
types.erasure(injectionSite.element().getEnclosingElement().asType());
// If instance has been declared as Object because it is not accessible from the
// component, but the injectionSite is in a supertype of instanceType that is
// publicly accessible, the InjectionSiteMethod will request the actual type and not
// Object as the first parameter. If so, cast to the supertype which is accessible
// from within generatedTypeName
CodeBlock maybeCastedInstance =
!types.isSubtype(instanceType, injectSiteType)
&& isTypeAccessibleFrom(injectSiteType, generatedTypeName.packageName())
? CodeBlock.of("($T) $L", injectSiteType, instanceCodeBlock)
: instanceCodeBlock;
return CodeBlock.of(
"$L;",
invoke(
injectionSite,
generatedTypeName,
maybeCastedInstance,
dependencyUsage,
metadataUtil));
})
.collect(toConcatenatedCodeBlock());
}
/**
* Invokes the injection method for {@code injectionSite}, with the dependencies transformed
* using the {@code dependencyUsage} function.
*/
private static CodeBlock invoke(
InjectionSite injectionSite,
ClassName generatedTypeName,
CodeBlock instanceCodeBlock,
Function<DependencyRequest, CodeBlock> dependencyUsage,
KotlinMetadataUtil metadataUtil) {
ImmutableList.Builder<CodeBlock> arguments = ImmutableList.builder();
arguments.add(instanceCodeBlock);
if (!injectionSite.dependencies().isEmpty()) {
arguments.addAll(
injectionSite.dependencies().stream().map(dependencyUsage).collect(toList()));
}
ClassName enclosingClass =
membersInjectorNameForType(asType(injectionSite.element().getEnclosingElement()));
MethodSpec methodSpec = create(injectionSite, metadataUtil);
return invokeMethod(methodSpec, arguments.build(), enclosingClass, generatedTypeName);
}
/*
* TODO(ronshapiro): this isn't perfect, as collisions could still exist. Some examples:
*
* - @Inject void members() {} will generate a method that conflicts with the instance
* method `injectMembers(T)`
* - Adding the index could conflict with another member:
* @Inject void a(Object o) {}
* @Inject void a(String s) {}
* @Inject void a1(String s) {}
*
* Here, Method a(String) will add the suffix "1", which will conflict with the method
* generated for a1(String)
* - Members named "members" or "methods" could also conflict with the {@code static} injection
* method.
*/
private static String methodName(InjectionSite injectionSite) {
int index = injectionSite.indexAmongAtInjectMembersWithSameSimpleName();
String indexString = index == 0 ? "" : String.valueOf(index + 1);
return "inject"
+ LOWER_CAMEL.to(UPPER_CAMEL, injectionSite.element().getSimpleName().toString())
+ indexString;
}
}
private static CodeBlock injectionMethodArgument(
DependencyRequest dependency, CodeBlock argument, ClassName generatedTypeName) {
TypeMirror keyType = dependency.key().type();
CodeBlock.Builder codeBlock = CodeBlock.builder();
if (!isRawTypeAccessible(keyType, generatedTypeName.packageName())
&& isTypeAccessibleFrom(keyType, generatedTypeName.packageName())) {
if (!dependency.kind().equals(RequestKind.INSTANCE)) {
TypeName usageTypeName = accessibleType(dependency);
codeBlock.add("($T) ($T)", usageTypeName, rawTypeName(usageTypeName));
} else if (dependency.requestElement().get().asType().getKind().equals(TypeKind.TYPEVAR)) {
codeBlock.add("($T)", keyType);
}
}
return codeBlock.add(argument).build();
}
/**
* Returns the parameter type for {@code dependency}. If the raw type is not accessible, returns
* {@link Object}.
*/
private static TypeName accessibleType(DependencyRequest dependency) {
TypeName typeName = requestTypeName(dependency.kind(), accessibleType(dependency.key().type()));
return dependency
.requestElement()
.map(element -> element.asType().getKind().isPrimitive())
.orElse(false)
? typeName.unbox()
: typeName;
}
/**
* Returns the accessible type for {@code type}. If the raw type is not accessible, returns {@link
* Object}.
*/
private static TypeName accessibleType(TypeMirror type) {
return isRawTypePubliclyAccessible(type) ? TypeName.get(type) : TypeName.OBJECT;
}
private enum InstanceCastPolicy {
CAST_IF_NOT_PUBLIC, IGNORE;
boolean useObjectType(TypeMirror instanceType) {
return this == CAST_IF_NOT_PUBLIC && !isRawTypePubliclyAccessible(instanceType);
}
}
private enum CheckNotNullPolicy {
IGNORE, CHECK_FOR_NULL;
CodeBlock checkForNull(CodeBlock maybeNull) {
return this.equals(IGNORE)
? maybeNull
: CodeBlock.of("$T.checkNotNullFromProvides($L)", Preconditions.class, maybeNull);
}
static CheckNotNullPolicy get(ProvisionBinding binding, CompilerOptions compilerOptions) {
return binding.shouldCheckForNull(compilerOptions) ? CHECK_FOR_NULL : IGNORE;
}
}
private static MethodSpec methodProxy(
ExecutableElement method,
String methodName,
InstanceCastPolicy instanceCastPolicy,
CheckNotNullPolicy checkNotNullPolicy,
KotlinMetadataUtil metadataUtil) {
MethodSpec.Builder builder =
methodBuilder(methodName).addModifiers(PUBLIC, STATIC).varargs(method.isVarArgs());
TypeElement enclosingType = asType(method.getEnclosingElement());
boolean isMethodInKotlinObject = metadataUtil.isObjectClass(enclosingType);
boolean isMethodInKotlinCompanionObject = metadataUtil.isCompanionObjectClass(enclosingType);
UniqueNameSet parameterNameSet = new UniqueNameSet();
CodeBlock instance;
if (isMethodInKotlinCompanionObject || method.getModifiers().contains(STATIC)) {
instance = CodeBlock.of("$T", rawTypeName(TypeName.get(enclosingType.asType())));
} else if (isMethodInKotlinObject) {
// Call through the singleton instance.
// See: https://kotlinlang.org/docs/reference/java-to-kotlin-interop.html#static-methods
instance = CodeBlock.of("$T.INSTANCE", rawTypeName(TypeName.get(enclosingType.asType())));
} else {
copyTypeParameters(builder, enclosingType);
boolean useObject = instanceCastPolicy.useObjectType(enclosingType.asType());
instance = copyInstance(builder, parameterNameSet, enclosingType.asType(), useObject);
}
CodeBlock arguments = copyParameters(builder, parameterNameSet, method.getParameters());
CodeBlock invocation =
checkNotNullPolicy.checkForNull(
CodeBlock.of("$L.$L($L)", instance, method.getSimpleName(), arguments));
copyTypeParameters(builder, method);
copyThrows(builder, method);
if (method.getReturnType().getKind().equals(VOID)) {
return builder.addStatement("$L", invocation).build();
} else {
getNullableType(method)
.ifPresent(annotation -> CodeBlocks.addAnnotation(builder, annotation));
return builder
.returns(TypeName.get(method.getReturnType()))
.addStatement("return $L", invocation).build();
}
}
private static MethodSpec fieldProxy(
VariableElement field, String methodName, Optional<AnnotationMirror> qualifierAnnotation) {
MethodSpec.Builder builder =
methodBuilder(methodName)
.addModifiers(PUBLIC, STATIC)
.addAnnotation(
AnnotationSpec.builder(TypeNames.INJECTED_FIELD_SIGNATURE)
.addMember("value", "$S", memberInjectedFieldSignatureForVariable(field))
.build());
qualifierAnnotation.map(AnnotationSpec::get).ifPresent(builder::addAnnotation);
TypeElement enclosingType = asType(field.getEnclosingElement());
copyTypeParameters(builder, enclosingType);
boolean useObject = !isRawTypePubliclyAccessible(enclosingType.asType());
UniqueNameSet parameterNameSet = new UniqueNameSet();
CodeBlock instance = copyInstance(builder, parameterNameSet, enclosingType.asType(), useObject);
CodeBlock argument = copyParameters(builder, parameterNameSet, ImmutableList.of(field));
return builder.addStatement("$L.$L = $L", instance, field.getSimpleName(), argument).build();
}
private static CodeBlock invokeMethod(
MethodSpec methodSpec,
ImmutableList<CodeBlock> parameters,
ClassName enclosingClass,
ClassName requestingClass) {
checkArgument(methodSpec.parameters.size() == parameters.size());
CodeBlock parameterBlock = makeParametersCodeBlock(parameters);
return enclosingClass.equals(requestingClass)
? CodeBlock.of("$L($L)", methodSpec.name, parameterBlock)
: CodeBlock.of("$T.$L($L)", enclosingClass, methodSpec.name, parameterBlock);
}
private static void copyTypeParameters(
MethodSpec.Builder methodBuilder, Parameterizable element) {
element.getTypeParameters().stream()
.map(TypeVariableName::get)
.forEach(methodBuilder::addTypeVariable);
}
private static void copyThrows(MethodSpec.Builder methodBuilder, ExecutableElement method) {
method.getThrownTypes().stream().map(TypeName::get).forEach(methodBuilder::addException);
}
private static CodeBlock copyParameters(
MethodSpec.Builder methodBuilder,
UniqueNameSet parameterNameSet,
List<? extends VariableElement> parameters) {
return parameters.stream()
.map(
parameter -> {
String name =
parameterNameSet.getUniqueName(validJavaName(parameter.getSimpleName()));
TypeMirror type = parameter.asType();
boolean useObject = !isRawTypePubliclyAccessible(type);
return copyParameter(methodBuilder, type, name, useObject);
})
.collect(toParametersCodeBlock());
}
private static CodeBlock copyParameter(
MethodSpec.Builder methodBuilder, TypeMirror type, String name, boolean useObject) {
TypeName typeName = useObject ? TypeName.OBJECT : TypeName.get(type);
methodBuilder.addParameter(ParameterSpec.builder(typeName, name).build());
return useObject ? CodeBlock.of("($T) $L", type, name) : CodeBlock.of("$L", name);
}
private static CodeBlock copyInstance(
MethodSpec.Builder methodBuilder,
UniqueNameSet parameterNameSet,
TypeMirror type,
boolean useObject) {
CodeBlock instance =
copyParameter(methodBuilder, type, parameterNameSet.getUniqueName("instance"), useObject);
// If we had to cast the instance add an extra parenthesis incase we're calling a method on it.
return useObject ? CodeBlock.of("($L)", instance) : instance;
}
private static String validJavaName(CharSequence name) {
if (SourceVersion.isIdentifier(name)) {
return protectAgainstKeywords(name.toString());
}
StringBuilder newName = new StringBuilder(name.length());
char firstChar = name.charAt(0);
if (!Character.isJavaIdentifierStart(firstChar)) {
newName.append('_');
}
name.chars().forEach(c -> newName.append(Character.isJavaIdentifierPart(c) ? c : '_'));
return newName.toString();
}
}