1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package com.puppycrawl.tools.checkstyle.checks.indentation;
21
22 import java.lang.reflect.Constructor;
23 import java.util.HashMap;
24 import java.util.Map;
25 import java.util.Set;
26
27 import com.puppycrawl.tools.checkstyle.api.DetailAST;
28 import com.puppycrawl.tools.checkstyle.api.TokenTypes;
29 import com.puppycrawl.tools.checkstyle.utils.CommonUtil;
30
31
32
33
34
35 public class HandlerFactory {
36
37
38
39
40 private final Map<Integer, Constructor<?>> typeHandlers = new HashMap<>();
41
42
43 private final Map<DetailAST, AbstractExpressionHandler> createdHandlers = new HashMap<>();
44
45
46
47
48
49
50
51
52 public HandlerFactory() {
53 register(TokenTypes.CASE_GROUP, CaseHandler.class);
54 register(TokenTypes.LITERAL_SWITCH, SwitchHandler.class);
55 register(TokenTypes.SLIST, SlistHandler.class);
56 register(TokenTypes.PACKAGE_DEF, PackageDefHandler.class);
57 register(TokenTypes.LITERAL_ELSE, ElseHandler.class);
58 register(TokenTypes.LITERAL_IF, IfHandler.class);
59 register(TokenTypes.LITERAL_TRY, TryHandler.class);
60 register(TokenTypes.LITERAL_CATCH, CatchHandler.class);
61 register(TokenTypes.LITERAL_FINALLY, FinallyHandler.class);
62 register(TokenTypes.LITERAL_DO, DoWhileHandler.class);
63 register(TokenTypes.LITERAL_WHILE, WhileHandler.class);
64 register(TokenTypes.LITERAL_FOR, ForHandler.class);
65 register(TokenTypes.METHOD_DEF, MethodDefHandler.class);
66 register(TokenTypes.CTOR_DEF, MethodDefHandler.class);
67 register(TokenTypes.CLASS_DEF, ClassDefHandler.class);
68 register(TokenTypes.ENUM_DEF, ClassDefHandler.class);
69 register(TokenTypes.OBJBLOCK, ObjectBlockHandler.class);
70 register(TokenTypes.INTERFACE_DEF, ClassDefHandler.class);
71 register(TokenTypes.IMPORT, ImportHandler.class);
72 register(TokenTypes.MODULE_IMPORT, ImportHandler.class);
73 register(TokenTypes.ARRAY_INIT, ArrayInitHandler.class);
74 register(TokenTypes.ANNOTATION_ARRAY_INIT, AnnotationArrayInitHandler.class);
75 register(TokenTypes.METHOD_CALL, MethodCallHandler.class);
76 register(TokenTypes.CTOR_CALL, MethodCallHandler.class);
77 register(TokenTypes.SUPER_CTOR_CALL, MethodCallHandler.class);
78 register(TokenTypes.LABELED_STAT, LabelHandler.class);
79 register(TokenTypes.STATIC_INIT, StaticInitHandler.class);
80 register(TokenTypes.INSTANCE_INIT, SlistHandler.class);
81 register(TokenTypes.VARIABLE_DEF, MemberDefHandler.class);
82 register(TokenTypes.LITERAL_NEW, NewHandler.class);
83 register(TokenTypes.INDEX_OP, IndexHandler.class);
84 register(TokenTypes.LITERAL_SYNCHRONIZED, SynchronizedHandler.class);
85 register(TokenTypes.LAMBDA, LambdaHandler.class);
86 register(TokenTypes.ANNOTATION_DEF, ClassDefHandler.class);
87 register(TokenTypes.ANNOTATION_FIELD_DEF, MethodDefHandler.class);
88 register(TokenTypes.SWITCH_RULE, SwitchRuleHandler.class);
89 register(TokenTypes.LITERAL_YIELD, YieldHandler.class);
90 register(TokenTypes.RECORD_DEF, ClassDefHandler.class);
91 register(TokenTypes.COMPACT_CTOR_DEF, MethodDefHandler.class);
92 }
93
94
95
96
97
98
99
100
101
102
103 private <T> void register(int type, Class<T> handlerClass) {
104 final Constructor<T> ctor = CommonUtil.getConstructor(handlerClass,
105 IndentationCheck.class,
106
107 DetailAST.class,
108
109 AbstractExpressionHandler.class
110 );
111 typeHandlers.put(type, ctor);
112 }
113
114
115
116
117
118
119
120 public boolean isHandledType(int type) {
121 final Set<Integer> typeSet = typeHandlers.keySet();
122 return typeSet.contains(type);
123 }
124
125
126
127
128
129
130 public int[] getHandledTypes() {
131 final Set<Integer> typeSet = typeHandlers.keySet();
132 final int[] types = new int[typeSet.size()];
133 int index = 0;
134 for (final Integer val : typeSet) {
135 types[index] = val;
136 index++;
137 }
138
139 return types;
140 }
141
142
143
144
145
146
147
148
149
150
151 public AbstractExpressionHandler getHandler(IndentationCheck indentCheck,
152 DetailAST ast, AbstractExpressionHandler parent) {
153 final AbstractExpressionHandler resultHandler;
154 final AbstractExpressionHandler handler =
155 createdHandlers.get(ast);
156 if (handler != null) {
157 resultHandler = handler;
158 }
159 else if (ast.getType() == TokenTypes.METHOD_CALL) {
160 resultHandler = createMethodCallHandler(indentCheck, ast, parent);
161 }
162 else {
163 final Constructor<?> handlerCtor = typeHandlers.get(ast.getType());
164 resultHandler = (AbstractExpressionHandler) CommonUtil.invokeConstructor(
165 handlerCtor, indentCheck, ast, parent);
166 }
167 return resultHandler;
168 }
169
170
171
172
173
174
175
176
177
178
179 private AbstractExpressionHandler createMethodCallHandler(IndentationCheck indentCheck,
180 DetailAST ast, AbstractExpressionHandler parent) {
181 DetailAST astNode = ast.getFirstChild();
182 while (astNode.getType() == TokenTypes.DOT) {
183 astNode = astNode.getFirstChild();
184 }
185 AbstractExpressionHandler theParent = parent;
186 if (isHandledType(astNode.getType())) {
187 theParent = getHandler(indentCheck, astNode, theParent);
188 createdHandlers.put(astNode, theParent);
189 }
190 return new MethodCallHandler(indentCheck, ast, theParent);
191 }
192
193
194 public void clearCreatedHandlers() {
195 createdHandlers.clear();
196 }
197
198 }