-
Notifications
You must be signed in to change notification settings - Fork 727
/
UnnecessarilyVisible.java
107 lines (98 loc) · 4.43 KB
/
UnnecessarilyVisible.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
* Copyright 2020 The Error Prone 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 com.google.errorprone.bugpatterns;
import static com.google.common.collect.ImmutableSet.toImmutableSet;
import static com.google.common.collect.Sets.immutableEnumSet;
import static com.google.common.collect.Sets.intersection;
import static com.google.errorprone.BugPattern.SeverityLevel.WARNING;
import static com.google.errorprone.fixes.SuggestedFixes.removeModifiers;
import static com.google.errorprone.matchers.Description.NO_MATCH;
import static com.google.errorprone.util.ASTHelpers.annotationsAmong;
import static com.google.errorprone.util.ASTHelpers.findSuperMethod;
import static com.google.errorprone.util.ASTHelpers.getSymbol;
import static com.google.errorprone.util.ASTHelpers.hasDirectAnnotationWithSimpleName;
import com.google.common.collect.ImmutableSet;
import com.google.errorprone.BugPattern;
import com.google.errorprone.VisitorState;
import com.google.errorprone.bugpatterns.BugChecker.MethodTreeMatcher;
import com.google.errorprone.fixes.SuggestedFix;
import com.google.errorprone.matchers.Description;
import com.google.errorprone.suppliers.Supplier;
import com.sun.source.tree.MethodTree;
import com.sun.tools.javac.code.Symbol.MethodSymbol;
import com.sun.tools.javac.util.Name;
import java.util.Set;
import java.util.stream.Stream;
import javax.lang.model.element.Modifier;
/** Suggests restricting the visibility of methods which should only be called by a framework. */
@BugPattern(
altNames = "RestrictInjectVisibility",
summary =
"Some methods (such as those annotated with @Inject or @Provides) are only intended to be"
+ " called by a framework, and so should have default visibility.",
severity = WARNING)
public final class UnnecessarilyVisible extends BugChecker implements MethodTreeMatcher {
private static final ImmutableSet<Modifier> VISIBILITY_MODIFIERS =
immutableEnumSet(Modifier.PROTECTED, Modifier.PUBLIC);
private static final Supplier<ImmutableSet<Name>> FRAMEWORK_ANNOTATIONS =
VisitorState.memoize(
s ->
Stream.of(
"com.google.inject.Inject",
"com.google.inject.Provides",
"com.google.inject.multibindings.ProvidesIntoMap",
"com.google.inject.multibindings.ProvidesIntoSet",
"dagger.Provides",
"javax.inject.Inject")
.map(s::getName)
.collect(toImmutableSet()));
private static final Supplier<ImmutableSet<Name>> INJECT_ANNOTATIONS =
VisitorState.memoize(
s ->
Stream.of("com.google.inject.Inject", "javax.inject.Inject")
.map(s::getName)
.collect(toImmutableSet()));
private static final String VISIBLE_FOR_TESTING_CAVEAT =
" If this is only for testing purposes, consider annotating the element with"
+ " @VisibleForTesting.";
@Override
public Description matchMethod(MethodTree tree, VisitorState state) {
MethodSymbol symbol = getSymbol(tree);
if (annotationsAmong(symbol, FRAMEWORK_ANNOTATIONS.get(state), state).isEmpty()) {
return NO_MATCH;
}
if (findSuperMethod(symbol, state.getTypes()).isPresent()) {
return NO_MATCH;
}
if (hasDirectAnnotationWithSimpleName(tree, "VisibleForTesting")) {
return NO_MATCH;
}
Set<Modifier> badModifiers = intersection(tree.getModifiers().getFlags(), VISIBILITY_MODIFIERS);
if (badModifiers.isEmpty()) {
return NO_MATCH;
}
return buildDescription(tree)
.addFix(
removeModifiers(tree.getModifiers(), state, badModifiers)
.orElse(SuggestedFix.emptyFix()))
.setMessage(
message()
+ (annotationsAmong(symbol, INJECT_ANNOTATIONS.get(state), state).isEmpty()
? ""
: VISIBLE_FOR_TESTING_CAVEAT))
.build();
}
}