1 /***
2 *
3 * Copyright 2004 James Strachan
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *
17 **/
18 package org.codehaus.groovy.syntax;
19
20 import org.codehaus.groovy.ast.ClassHelper;
21 import org.codehaus.groovy.ast.ClassNode;
22 import org.codehaus.groovy.ast.ModuleNode;
23 import org.codehaus.groovy.control.SourceUnit;
24
25 import java.util.ArrayList;
26 import java.util.HashMap;
27 import java.util.List;
28 import java.util.Map;
29
30 /***
31 * A common base class of AST helper methods which can be shared across the classic and new parsers
32 *
33 * @author Jochen Theodorou
34 * @author James Strachan
35 * @author Bob McWhirter
36 * @author Sam Pullara
37 * @author Chris Poirier
38 * @version $Revision: 1.10 $
39 */
40 public class ASTHelper {
41
42 private static final String[] EMPTY_STRING_ARRAY = new String[0];
43
44 /*** The SourceUnit controlling us */
45 private SourceUnit controller;
46
47 /*** Our ClassLoader, which provides information on external types */
48 private ClassLoader classLoader;
49
50 /*** Our imports, simple name => fully qualified name */
51 private Map imports;
52 protected ModuleNode output;
53
54 /**</package-summary/html">The package name in which the module sits *//package-summary.html">em>* The package name in which the module sits */
55 private String packageName/package-summary.html">ong> String packageName;
56
57
58 protected static HashMap resolutions = new HashMap();
59
60 private static String NOT_RESOLVED = new String();
61
62 /*** temporarily store the class names that the current modulenode contains */
63 private List newClasses = new ArrayList();
64
65 public ASTHelper(SourceUnit controller, ClassLoader classLoader) {
66 this();
67 this.controller = controller;
68 this.classLoader = classLoader;
69 }
70
71 public ASTHelper() {
72 imports = new HashMap();
73 }
74
75 public String getPackageName() {
76 return</strong> packageName;
77 }
78
79 public void setPackageName(String packageName) {/package-summary.html">ong> void setPackageName(String packageName) {
80 this.packageName = packageName;
81 if (packageName!=null && packageName.length()>0){
82 packageName+='.';
83 }
84 output.setPackageName(packageName);
85 }
86
87
88 /***
89 * Returns our class loader (as supplied on construction).
90 */
91 public ClassLoader getClassLoader() {
92 return classLoader;
93 }
94
95 public void setClassLoader(ClassLoader classLoader) {
96 this.classLoader = classLoader;
97 }
98
99 public SourceUnit getController() {
100 return controller;
101 }
102
103 public void setController(SourceUnit controller) {
104 this.controller = controller;
105 }
106
107 /***
108 * Returns a fully qualified name for any given potential type
109 * name. Returns null if no qualified name could be determined.
110 */
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258 /***
259 * Returns two names joined by a dot. If the base name is
260 * empty, returns the name unchanged.
261 */
262 public static String dot(String base, String name) {
263 if (base != null && base.length() > 0) {
264 return base + "." + name;
265 }
266
267 return name;
268 }
269
270 protected void makeModule() {
271 this.newClasses.clear();
272 this.output = new ModuleNode(controller);
273 resolutions.clear();
274 }
275
276 /***
277 * A synonym for <code>dot( base, "" )</code>.
278 */
279 protected String dot(String base) {
280 return dot(base, "");
281 }
282
283
284
285
286
287
288
289
290
291
292 protected void addNewClassName(String name) {
293 this.newClasses.add(name);
294 }
295
296 protected void importClass(ClassNode type, String name, String as) {
297 if (as==null) as=name;
298
299 output.addImport(as, type);
300 imports.put(as, type);
301 }
302
303 protected void importPackageWithStar(String importPackage) {
304 String[] classes = output.addImportPackage( dot(importPackage) );
305 for( int i = 0; i < classes.length; i++ )
306 {
307 imports.put( classes[i], dot(importPackage, classes[i]) );
308 }
309 }
310 }