112 lines
3.9 KiB
Java
112 lines
3.9 KiB
Java
/*
|
|
* Copyright (C) 2016 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.validation;
|
|
|
|
import static dagger.internal.codegen.base.FrameworkTypes.isFrameworkType;
|
|
import static dagger.internal.codegen.validation.BindingElementValidator.AllowsMultibindings.NO_MULTIBINDINGS;
|
|
import static dagger.internal.codegen.validation.BindingElementValidator.AllowsScoping.NO_SCOPING;
|
|
import static dagger.internal.codegen.validation.BindingMethodValidator.Abstractness.MUST_BE_ABSTRACT;
|
|
import static dagger.internal.codegen.validation.BindingMethodValidator.ExceptionSuperclass.NO_EXCEPTIONS;
|
|
|
|
import com.google.auto.common.MoreTypes;
|
|
import com.google.common.collect.ImmutableSet;
|
|
import dagger.Module;
|
|
import dagger.internal.codegen.base.MapType;
|
|
import dagger.internal.codegen.base.SetType;
|
|
import dagger.internal.codegen.binding.InjectionAnnotations;
|
|
import dagger.internal.codegen.kotlin.KotlinMetadataUtil;
|
|
import dagger.internal.codegen.langmodel.DaggerElements;
|
|
import dagger.internal.codegen.langmodel.DaggerTypes;
|
|
import dagger.multibindings.Multibinds;
|
|
import dagger.producers.ProducerModule;
|
|
import javax.inject.Inject;
|
|
import javax.lang.model.element.ExecutableElement;
|
|
import javax.lang.model.type.TypeMirror;
|
|
|
|
/** A validator for {@link Multibinds} methods. */
|
|
class MultibindsMethodValidator extends BindingMethodValidator {
|
|
|
|
/** Creates a validator for {@link Multibinds @Multibinds} methods. */
|
|
@Inject
|
|
MultibindsMethodValidator(
|
|
DaggerElements elements,
|
|
DaggerTypes types,
|
|
KotlinMetadataUtil kotlinMetadataUtil,
|
|
DependencyRequestValidator dependencyRequestValidator,
|
|
InjectionAnnotations injectionAnnotations) {
|
|
super(
|
|
elements,
|
|
types,
|
|
kotlinMetadataUtil,
|
|
Multibinds.class,
|
|
ImmutableSet.of(Module.class, ProducerModule.class),
|
|
dependencyRequestValidator,
|
|
MUST_BE_ABSTRACT,
|
|
NO_EXCEPTIONS,
|
|
NO_MULTIBINDINGS,
|
|
NO_SCOPING,
|
|
injectionAnnotations);
|
|
}
|
|
|
|
@Override
|
|
protected ElementValidator elementValidator(ExecutableElement element) {
|
|
return new Validator(element);
|
|
}
|
|
|
|
private class Validator extends MethodValidator {
|
|
Validator(ExecutableElement element) {
|
|
super(element);
|
|
}
|
|
|
|
@Override
|
|
protected void checkParameters() {
|
|
if (!element.getParameters().isEmpty()) {
|
|
report.addError(bindingMethods("cannot have parameters"));
|
|
}
|
|
}
|
|
|
|
/** Adds an error unless the method returns a {@code Map<K, V>} or {@code Set<T>}. */
|
|
@Override
|
|
protected void checkType() {
|
|
if (!isPlainMap(element.getReturnType())
|
|
&& !isPlainSet(element.getReturnType())) {
|
|
report.addError(bindingMethods("must return Map<K, V> or Set<T>"));
|
|
}
|
|
}
|
|
|
|
private boolean isPlainMap(TypeMirror returnType) {
|
|
if (!MapType.isMap(returnType)) {
|
|
return false;
|
|
}
|
|
MapType mapType = MapType.from(returnType);
|
|
return !mapType.isRawType()
|
|
&& MoreTypes.isType(mapType.valueType()) // No wildcards.
|
|
&& !isFrameworkType(mapType.valueType());
|
|
}
|
|
|
|
private boolean isPlainSet(TypeMirror returnType) {
|
|
if (!SetType.isSet(returnType)) {
|
|
return false;
|
|
}
|
|
SetType setType = SetType.from(returnType);
|
|
return !setType.isRawType()
|
|
&& MoreTypes.isType(setType.elementType()) // No wildcards.
|
|
&& !isFrameworkType(setType.elementType());
|
|
}
|
|
}
|
|
}
|