diff options
Diffstat (limited to 'java/dagger/internal/codegen/writing/ModuleProxies.java')
-rw-r--r-- | java/dagger/internal/codegen/writing/ModuleProxies.java | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/java/dagger/internal/codegen/writing/ModuleProxies.java b/java/dagger/internal/codegen/writing/ModuleProxies.java index 6d174f824..fcd9b56e5 100644 --- a/java/dagger/internal/codegen/writing/ModuleProxies.java +++ b/java/dagger/internal/codegen/writing/ModuleProxies.java @@ -27,7 +27,6 @@ import static javax.lang.model.element.Modifier.PUBLIC; import static javax.lang.model.element.Modifier.STATIC; import static javax.lang.model.util.ElementFilter.constructorsIn; -import com.google.common.collect.ImmutableList; import com.squareup.javapoet.ClassName; import com.squareup.javapoet.CodeBlock; import com.squareup.javapoet.TypeSpec; @@ -79,20 +78,25 @@ public final class ModuleProxies { } @Override + public ClassName nameGeneratedType(TypeElement moduleElement) { + return moduleProxies.constructorProxyTypeName(moduleElement); + } + + @Override public Element originatingElement(TypeElement moduleElement) { return moduleElement; } @Override - public ImmutableList<TypeSpec.Builder> topLevelTypes(TypeElement moduleElement) { + public Optional<TypeSpec.Builder> write(TypeElement moduleElement) { ModuleKind.checkIsModule(moduleElement, metadataUtil); return moduleProxies.nonPublicNullaryConstructor(moduleElement).isPresent() - ? ImmutableList.of(buildProxy(moduleElement)) - : ImmutableList.of(); + ? Optional.of(buildProxy(moduleElement)) + : Optional.empty(); } private TypeSpec.Builder buildProxy(TypeElement moduleElement) { - return classBuilder(moduleProxies.constructorProxyTypeName(moduleElement)) + return classBuilder(nameGeneratedType(moduleElement)) .addModifiers(PUBLIC, FINAL) .addMethod(constructorBuilder().addModifiers(PRIVATE).build()) .addMethod( |