-
Notifications
You must be signed in to change notification settings - Fork 638
/
NoArgsConstructorProcessor.java
89 lines (73 loc) · 3.57 KB
/
NoArgsConstructorProcessor.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
package de.plushnikov.intellij.plugin.processor.clazz.constructor;
import com.intellij.psi.PsiAnnotation;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiField;
import com.intellij.psi.PsiMethod;
import de.plushnikov.intellij.plugin.problem.ProblemBuilder;
import de.plushnikov.intellij.plugin.processor.LombokPsiElementUsage;
import de.plushnikov.intellij.plugin.util.LombokProcessorUtil;
import de.plushnikov.intellij.plugin.util.PsiAnnotationUtil;
import de.plushnikov.intellij.plugin.util.PsiClassUtil;
import lombok.NoArgsConstructor;
import org.jetbrains.annotations.NotNull;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
/**
* @author Plushnikov Michail
*/
public class NoArgsConstructorProcessor extends AbstractConstructorClassProcessor {
public NoArgsConstructorProcessor() {
super(NoArgsConstructor.class, PsiMethod.class);
}
@Override
protected boolean validate(@NotNull PsiAnnotation psiAnnotation, @NotNull PsiClass psiClass, @NotNull ProblemBuilder builder) {
boolean result;
result = super.validate(psiAnnotation, psiClass, builder);
final String staticConstructorName = getStaticConstructorName(psiAnnotation);
result &= validateIsConstructorNotDefined(psiClass, staticConstructorName, Collections.emptyList(), builder);
return result;
}
@NotNull
public Collection<PsiMethod> createNoArgsConstructor(@NotNull PsiClass psiClass, @NotNull String methodVisibility, @NotNull PsiAnnotation psiAnnotation) {
final boolean forceConstructorWithJavaDefaults = isForceConstructor(psiAnnotation);
return createNoArgsConstructor(psiClass, methodVisibility, psiAnnotation, forceConstructorWithJavaDefaults);
}
@NotNull
public Collection<PsiMethod> createNoArgsConstructor(@NotNull PsiClass psiClass, @NotNull String methodVisibility, @NotNull PsiAnnotation psiAnnotation, boolean withJavaDefaults) {
final Collection<PsiField> params = getConstructorFields(psiClass, withJavaDefaults);
return createConstructorMethod(psiClass, methodVisibility, psiAnnotation, withJavaDefaults, params);
}
private boolean isForceConstructor(@NotNull PsiAnnotation psiAnnotation) {
return PsiAnnotationUtil.getBooleanAnnotationValue(psiAnnotation, "force", false);
}
protected void generatePsiElements(@NotNull PsiClass psiClass, @NotNull PsiAnnotation psiAnnotation, @NotNull List<? super PsiElement> target) {
final String methodVisibility = LombokProcessorUtil.getAccessVisibility(psiAnnotation);
if (null != methodVisibility) {
target.addAll(createNoArgsConstructor(psiClass, methodVisibility, psiAnnotation));
}
}
@Override
public LombokPsiElementUsage checkFieldUsage(@NotNull PsiField psiField, @NotNull PsiAnnotation psiAnnotation) {
final PsiClass containingClass = psiField.getContainingClass();
if (null != containingClass) {
final boolean forceConstructorWithJavaDefaults = isForceConstructor(psiAnnotation);
final Collection<PsiField> params = getConstructorFields(containingClass, forceConstructorWithJavaDefaults);
if (PsiClassUtil.getNames(params).contains(psiField.getName())) {
return LombokPsiElementUsage.WRITE;
}
}
return LombokPsiElementUsage.NONE;
}
@NotNull
private Collection<PsiField> getConstructorFields(PsiClass containingClass, boolean forceConstructorWithJavaDefaults) {
Collection<PsiField> params;
if (forceConstructorWithJavaDefaults) {
params = getRequiredFields(containingClass);
} else {
params = Collections.emptyList();
}
return params;
}
}