Browse Source
Extract and refactor inner-classes from the GroovyCompiler to improve code readability.pull/85/merge
Phillip Webb
12 years ago
4 changed files with 278 additions and 188 deletions
@ -0,0 +1,73 @@
@@ -0,0 +1,73 @@
|
||||
/* |
||||
* Copyright 2012-2013 the original author or 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 org.springframework.boot.cli.compiler; |
||||
|
||||
import groovy.lang.GroovyClassLoader; |
||||
|
||||
import org.codehaus.groovy.ast.ASTNode; |
||||
import org.codehaus.groovy.ast.ClassNode; |
||||
import org.codehaus.groovy.ast.ModuleNode; |
||||
import org.codehaus.groovy.control.SourceUnit; |
||||
import org.codehaus.groovy.transform.ASTTransformation; |
||||
|
||||
/** |
||||
* {@link ASTTransformation} to apply |
||||
* {@link CompilerAutoConfiguration#applyDependencies(DependencyCustomizer) dependency |
||||
* auto-configuration}. |
||||
* |
||||
* @author Phillip Webb |
||||
* @author Dave Syer |
||||
* @author Andy Wilkinson |
||||
*/ |
||||
class DependencyAutoConfigurationTransformation implements ASTTransformation { |
||||
|
||||
private final GroovyClassLoader loader; |
||||
|
||||
private final ArtifactCoordinatesResolver coordinatesResolver; |
||||
|
||||
private final Iterable<CompilerAutoConfiguration> compilerAutoConfigurations; |
||||
|
||||
DependencyAutoConfigurationTransformation(GroovyClassLoader loader, |
||||
ArtifactCoordinatesResolver coordinatesResolver, |
||||
Iterable<CompilerAutoConfiguration> compilerAutoConfigurations) { |
||||
this.loader = loader; |
||||
this.coordinatesResolver = coordinatesResolver; |
||||
this.compilerAutoConfigurations = compilerAutoConfigurations; |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void visit(ASTNode[] nodes, SourceUnit source) { |
||||
for (ASTNode astNode : nodes) { |
||||
if (astNode instanceof ModuleNode) { |
||||
visitModule((ModuleNode) astNode); |
||||
} |
||||
} |
||||
} |
||||
|
||||
private void visitModule(ModuleNode module) { |
||||
DependencyCustomizer dependencies = new DependencyCustomizer(this.loader, module, |
||||
this.coordinatesResolver); |
||||
for (ClassNode classNode : module.getClasses()) { |
||||
for (CompilerAutoConfiguration autoConfiguration : this.compilerAutoConfigurations) { |
||||
if (autoConfiguration.matches(classNode)) { |
||||
autoConfiguration.applyDependencies(dependencies); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,141 @@
@@ -0,0 +1,141 @@
|
||||
/* |
||||
* Copyright 2012-2013 the original author or 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 org.springframework.boot.cli.compiler; |
||||
|
||||
import groovy.lang.Grab; |
||||
|
||||
import java.util.Arrays; |
||||
import java.util.Collections; |
||||
import java.util.HashSet; |
||||
import java.util.Set; |
||||
|
||||
import org.codehaus.groovy.ast.ASTNode; |
||||
import org.codehaus.groovy.ast.AnnotatedNode; |
||||
import org.codehaus.groovy.ast.AnnotationNode; |
||||
import org.codehaus.groovy.ast.ClassCodeVisitorSupport; |
||||
import org.codehaus.groovy.ast.ClassNode; |
||||
import org.codehaus.groovy.ast.ImportNode; |
||||
import org.codehaus.groovy.ast.ModuleNode; |
||||
import org.codehaus.groovy.ast.expr.ConstantExpression; |
||||
import org.codehaus.groovy.ast.expr.Expression; |
||||
import org.codehaus.groovy.control.SourceUnit; |
||||
import org.codehaus.groovy.transform.ASTTransformation; |
||||
|
||||
/** |
||||
* {@link ASTTransformation} to resolve {@link Grab} artifact coordinates. |
||||
* @author Andy Wilkinson |
||||
* @author Phillip Webb |
||||
*/ |
||||
class ResolveDependencyCoordinatesTransformation implements ASTTransformation { |
||||
|
||||
private static final Set<String> GRAB_ANNOTATION_NAMES = Collections |
||||
.unmodifiableSet(new HashSet<String>(Arrays.asList(Grab.class.getName(), |
||||
Grab.class.getSimpleName()))); |
||||
|
||||
private final ArtifactCoordinatesResolver coordinatesResolver; |
||||
|
||||
ResolveDependencyCoordinatesTransformation( |
||||
ArtifactCoordinatesResolver coordinatesResolver) { |
||||
this.coordinatesResolver = coordinatesResolver; |
||||
} |
||||
|
||||
@Override |
||||
public void visit(ASTNode[] nodes, SourceUnit source) { |
||||
ClassVisitor classVisitor = new ClassVisitor(source); |
||||
for (ASTNode node : nodes) { |
||||
if (node instanceof ModuleNode) { |
||||
ModuleNode module = (ModuleNode) node; |
||||
for (ImportNode importNode : module.getImports()) { |
||||
visitAnnotatedNode(importNode); |
||||
} |
||||
for (ClassNode classNode : module.getClasses()) { |
||||
visitAnnotatedNode(classNode); |
||||
classNode.visitContents(classVisitor); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
private void visitAnnotatedNode(AnnotatedNode annotatedNode) { |
||||
for (AnnotationNode annotationNode : annotatedNode.getAnnotations()) { |
||||
if (GRAB_ANNOTATION_NAMES.contains(annotationNode.getClassNode().getName())) { |
||||
transformGrabAnnotation(annotationNode); |
||||
} |
||||
} |
||||
} |
||||
|
||||
private void transformGrabAnnotation(AnnotationNode grabAnnotation) { |
||||
grabAnnotation.setMember("initClass", new ConstantExpression(false)); |
||||
String value = getValue(grabAnnotation); |
||||
if (value != null && !isConvenienceForm(value)) { |
||||
applyGroupAndVersion(grabAnnotation, value); |
||||
} |
||||
} |
||||
|
||||
private String getValue(AnnotationNode annotation) { |
||||
Expression expression = annotation.getMember("value"); |
||||
if (expression instanceof ConstantExpression) { |
||||
Object value = ((ConstantExpression) expression).getValue(); |
||||
return (value instanceof String ? (String) value : null); |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
private boolean isConvenienceForm(String value) { |
||||
return value.contains(":") || value.contains("#"); |
||||
} |
||||
|
||||
private void applyGroupAndVersion(AnnotationNode annotation, String module) { |
||||
if (module != null) { |
||||
setMember(annotation, "module", module); |
||||
} |
||||
else { |
||||
Expression expression = annotation.getMembers().get("module"); |
||||
module = (String) ((ConstantExpression) expression).getValue(); |
||||
} |
||||
if (annotation.getMember("group") == null) { |
||||
setMember(annotation, "group", this.coordinatesResolver.getGroupId(module)); |
||||
} |
||||
if (annotation.getMember("version") == null) { |
||||
setMember(annotation, "version", this.coordinatesResolver.getVersion(module)); |
||||
} |
||||
} |
||||
|
||||
private void setMember(AnnotationNode annotation, String name, String value) { |
||||
ConstantExpression expression = new ConstantExpression(value); |
||||
annotation.setMember(name, expression); |
||||
} |
||||
|
||||
private class ClassVisitor extends ClassCodeVisitorSupport { |
||||
|
||||
private final SourceUnit source; |
||||
|
||||
public ClassVisitor(SourceUnit source) { |
||||
this.source = source; |
||||
} |
||||
|
||||
@Override |
||||
protected SourceUnit getSourceUnit() { |
||||
return this.source; |
||||
} |
||||
|
||||
@Override |
||||
public void visitAnnotations(AnnotatedNode node) { |
||||
visitAnnotatedNode(node); |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue