88 lines
3.4 KiB
Java
88 lines
3.4 KiB
Java
/*
|
|
* Copyright (C) 2015 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.common.base.Preconditions.checkNotNull;
|
|
import static dagger.internal.codegen.binding.MapKeys.getMapKeyExpression;
|
|
import static dagger.internal.codegen.binding.SourceFiles.mapFactoryClassName;
|
|
|
|
import com.google.common.collect.ImmutableSet;
|
|
import com.squareup.javapoet.CodeBlock;
|
|
import dagger.internal.codegen.base.MapType;
|
|
import dagger.internal.codegen.binding.BindingGraph;
|
|
import dagger.internal.codegen.binding.ContributionBinding;
|
|
import dagger.internal.codegen.langmodel.DaggerElements;
|
|
import dagger.model.DependencyRequest;
|
|
import dagger.producers.Produced;
|
|
import dagger.producers.Producer;
|
|
import javax.inject.Provider;
|
|
import javax.lang.model.type.TypeMirror;
|
|
|
|
/** A factory creation expression for a multibound map. */
|
|
final class MapFactoryCreationExpression extends MultibindingFactoryCreationExpression {
|
|
|
|
private final ComponentImplementation componentImplementation;
|
|
private final BindingGraph graph;
|
|
private final ContributionBinding binding;
|
|
private final DaggerElements elements;
|
|
|
|
MapFactoryCreationExpression(
|
|
ContributionBinding binding,
|
|
ComponentImplementation componentImplementation,
|
|
ComponentBindingExpressions componentBindingExpressions,
|
|
BindingGraph graph,
|
|
DaggerElements elements) {
|
|
super(binding, componentImplementation, componentBindingExpressions);
|
|
this.binding = checkNotNull(binding);
|
|
this.componentImplementation = checkNotNull(componentImplementation);
|
|
this.graph = checkNotNull(graph);
|
|
this.elements = checkNotNull(elements);
|
|
}
|
|
|
|
@Override
|
|
public CodeBlock creationExpression() {
|
|
CodeBlock.Builder builder = CodeBlock.builder().add("$T.", mapFactoryClassName(binding));
|
|
if (!useRawType()) {
|
|
MapType mapType = MapType.from(binding.key().type());
|
|
// TODO(ronshapiro): either inline this into mapFactoryClassName, or add a
|
|
// mapType.unwrappedValueType() method that doesn't require a framework type
|
|
TypeMirror valueType = mapType.valueType();
|
|
for (Class<?> frameworkClass :
|
|
ImmutableSet.of(Provider.class, Producer.class, Produced.class)) {
|
|
if (mapType.valuesAreTypeOf(frameworkClass)) {
|
|
valueType = mapType.unwrappedValueType(frameworkClass);
|
|
break;
|
|
}
|
|
}
|
|
builder.add("<$T, $T>", mapType.keyType(), valueType);
|
|
}
|
|
|
|
builder.add("builder($L)", binding.dependencies().size());
|
|
|
|
for (DependencyRequest dependency : binding.dependencies()) {
|
|
ContributionBinding contributionBinding = graph.contributionBinding(dependency.key());
|
|
builder.add(
|
|
".put($L, $L)",
|
|
getMapKeyExpression(contributionBinding, componentImplementation.name(), elements),
|
|
multibindingDependencyExpression(dependency));
|
|
}
|
|
builder.add(".build()");
|
|
|
|
return builder.build();
|
|
}
|
|
}
|