org.codehaus.groovy.ast.tools.GeneralUtils Maven / Gradle / Ivy
Show all versions of groovy Show documentation
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.codehaus.groovy.ast.tools;
import groovy.lang.MetaProperty;
import org.codehaus.groovy.ast.ASTNode;
import org.codehaus.groovy.ast.AnnotatedNode;
import org.codehaus.groovy.ast.AnnotationNode;
import org.codehaus.groovy.ast.ClassHelper;
import org.codehaus.groovy.ast.ClassNode;
import org.codehaus.groovy.ast.FieldNode;
import org.codehaus.groovy.ast.MethodNode;
import org.codehaus.groovy.ast.PackageNode;
import org.codehaus.groovy.ast.Parameter;
import org.codehaus.groovy.ast.PropertyNode;
import org.codehaus.groovy.ast.Variable;
import org.codehaus.groovy.ast.VariableScope;
import org.codehaus.groovy.ast.expr.ArgumentListExpression;
import org.codehaus.groovy.ast.expr.AttributeExpression;
import org.codehaus.groovy.ast.expr.BinaryExpression;
import org.codehaus.groovy.ast.expr.BooleanExpression;
import org.codehaus.groovy.ast.expr.CastExpression;
import org.codehaus.groovy.ast.expr.ClassExpression;
import org.codehaus.groovy.ast.expr.ClosureExpression;
import org.codehaus.groovy.ast.expr.ConstantExpression;
import org.codehaus.groovy.ast.expr.ConstructorCallExpression;
import org.codehaus.groovy.ast.expr.DeclarationExpression;
import org.codehaus.groovy.ast.expr.ElvisOperatorExpression;
import org.codehaus.groovy.ast.expr.Expression;
import org.codehaus.groovy.ast.expr.FieldExpression;
import org.codehaus.groovy.ast.expr.LambdaExpression;
import org.codehaus.groovy.ast.expr.ListExpression;
import org.codehaus.groovy.ast.expr.MapEntryExpression;
import org.codehaus.groovy.ast.expr.MapExpression;
import org.codehaus.groovy.ast.expr.MethodCallExpression;
import org.codehaus.groovy.ast.expr.NotExpression;
import org.codehaus.groovy.ast.expr.PropertyExpression;
import org.codehaus.groovy.ast.expr.StaticMethodCallExpression;
import org.codehaus.groovy.ast.expr.TernaryExpression;
import org.codehaus.groovy.ast.expr.VariableExpression;
import org.codehaus.groovy.ast.stmt.BlockStatement;
import org.codehaus.groovy.ast.stmt.CaseStatement;
import org.codehaus.groovy.ast.stmt.CatchStatement;
import org.codehaus.groovy.ast.stmt.EmptyStatement;
import org.codehaus.groovy.ast.stmt.ExpressionStatement;
import org.codehaus.groovy.ast.stmt.ForStatement;
import org.codehaus.groovy.ast.stmt.IfStatement;
import org.codehaus.groovy.ast.stmt.ReturnStatement;
import org.codehaus.groovy.ast.stmt.Statement;
import org.codehaus.groovy.ast.stmt.SwitchStatement;
import org.codehaus.groovy.ast.stmt.SynchronizedStatement;
import org.codehaus.groovy.ast.stmt.ThrowStatement;
import org.codehaus.groovy.ast.stmt.TryCatchStatement;
import org.codehaus.groovy.classgen.BytecodeExpression;
import org.codehaus.groovy.control.io.ReaderSource;
import org.codehaus.groovy.runtime.GeneratedClosure;
import org.codehaus.groovy.syntax.Token;
import org.codehaus.groovy.syntax.Types;
import org.codehaus.groovy.transform.AbstractASTTransformation;
import org.objectweb.asm.MethodVisitor;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.function.Consumer;
import static org.codehaus.groovy.antlr.PrimitiveHelper.getDefaultValueForPrimitive;
import static org.codehaus.groovy.ast.ClassHelper.isObjectType;
import static org.codehaus.groovy.ast.tools.GenericsUtils.parameterizeType;
/**
* Handy methods when working with the Groovy AST
*/
public class GeneralUtils {
public static final Token ASSIGN = Token.newSymbol(Types.ASSIGN, -1, -1);
public static final Token EQ = Token.newSymbol(Types.COMPARE_EQUAL, -1, -1);
public static final Token NE = Token.newSymbol(Types.COMPARE_NOT_EQUAL, -1, -1);
public static final Token NOT_IDENTICAL = Token.newSymbol(Types.COMPARE_NOT_IDENTICAL, -1, -1);
public static final Token LT = Token.newSymbol(Types.COMPARE_LESS_THAN, -1, -1);
public static final Token AND = Token.newSymbol(Types.LOGICAL_AND, -1, -1);
public static final Token OR = Token.newSymbol(Types.LOGICAL_OR, -1, -1);
public static final Token CMP = Token.newSymbol(Types.COMPARE_TO, -1, -1);
public static final Token INSTANCEOF = Token.newSymbol(Types.KEYWORD_INSTANCEOF, -1, -1);
public static final Token PLUS = Token.newSymbol(Types.PLUS, -1, -1);
private static final Token INDEX = Token.newSymbol("[", -1, -1);
public static BinaryExpression andX(final Expression lhv, final Expression rhv) {
return binX(lhv, AND, rhv);
}
public static ArgumentListExpression args(final Expression... expressions) {
List list = new ArrayList<>(expressions.length);
Collections.addAll(list, expressions);
return args(list);
}
public static ArgumentListExpression args(final List expressions) {
return new ArgumentListExpression(expressions);
}
public static ArgumentListExpression args(final Parameter... parameters) {
return new ArgumentListExpression(parameters);
}
public static ArgumentListExpression args(final String... names) {
return args(Arrays.stream(names).map(GeneralUtils::varX).toArray(Expression[]::new));
}
public static CastExpression asX(final ClassNode type, final Expression expression) {
return CastExpression.asExpression(type, expression);
}
public static Statement assignS(final Expression target, final Expression value) {
return stmt(assignX(target, value));
}
public static Statement assignNullS(final Expression target) {
return assignS(target, ConstantExpression.EMPTY_EXPRESSION);
}
public static Expression assignX(final Expression target, final Expression value) {
return binX(target, ASSIGN, value);
}
public static Expression attrX(final Expression oe, final Expression prop) {
return new AttributeExpression(oe, prop);
}
@Deprecated
public static Expression attrX(final Expression oe, final String propName) {
return new AttributeExpression(oe, constX(propName));
}
public static BinaryExpression binX(final Expression left, final Token token, final Expression right) {
return new BinaryExpression(left, token, right);
}
public static BlockStatement block(final VariableScope scope, final Statement... stmts) {
BlockStatement block = new BlockStatement();
block.setVariableScope(scope);
for (Statement stmt : stmts) {
block.addStatement(stmt);
}
return block;
}
public static BlockStatement block(final VariableScope scope, final List stmts) {
BlockStatement block = new BlockStatement();
block.setVariableScope(scope);
for (Statement stmt : stmts) {
block.addStatement(stmt);
}
return block;
}
public static BlockStatement block(final Statement... stmts) {
BlockStatement block = new BlockStatement();
for (Statement stmt : stmts) {
block.addStatement(stmt);
}
return block;
}
public static BooleanExpression boolX(final Expression expr) {
return new BooleanExpression(expr);
}
public static BytecodeExpression bytecodeX(final Consumer writer) {
return new BytecodeExpression() {
@Override
public void visit(final MethodVisitor visitor) {
writer.accept(visitor);
}
};
}
public static BytecodeExpression bytecodeX(final ClassNode type, final Consumer writer) {
BytecodeExpression expression = bytecodeX(writer);
expression.setType(type);
return expression;
}
public static MethodCallExpression callSuperX(final String methodName) {
return callSuperX(methodName, MethodCallExpression.NO_ARGUMENTS);
}
public static MethodCallExpression callSuperX(final String methodName, final Expression args) {
return callX(varX("super"), methodName, args);
}
public static MethodCallExpression callThisX(final String methodName) {
return callThisX(methodName, MethodCallExpression.NO_ARGUMENTS);
}
public static MethodCallExpression callThisX(final String methodName, final Expression args) {
return callX(varX("this"), methodName, args);
}
public static MethodCallExpression callX(final Expression receiver, final String methodName) {
return callX(receiver, methodName, MethodCallExpression.NO_ARGUMENTS);
}
public static MethodCallExpression callX(final Expression receiver, final String methodName, final Expression args) {
return new MethodCallExpression(receiver, methodName, args);
}
public static MethodCallExpression callX(final Expression receiver, final Expression method, final Expression args) {
return new MethodCallExpression(receiver, method, args);
}
public static StaticMethodCallExpression callX(final ClassNode receiver, final String methodName) {
return callX(receiver, methodName, MethodCallExpression.NO_ARGUMENTS);
}
public static StaticMethodCallExpression callX(final ClassNode receiver, final String methodName, final Expression args) {
return new StaticMethodCallExpression(receiver, methodName, args);
}
public static CaseStatement caseS(final Expression expression, Statement code) {
return new CaseStatement(expression, code);
}
public static CastExpression castX(final ClassNode type, final Expression expression) {
return new CastExpression(type, expression);
}
public static CastExpression castX(final ClassNode type, final Expression expression, final boolean ignoreAutoboxing) {
return new CastExpression(type, expression, ignoreAutoboxing);
}
public static CatchStatement catchS(final Parameter variable, final Statement code) {
return new CatchStatement(variable, code);
}
public static ClassExpression classX(final ClassNode clazz) {
return new ClassExpression(clazz);
}
public static ClassExpression classX(final Class clazz) {
return classX(ClassHelper.make(clazz).getPlainNodeReference());
}
public static ClosureExpression closureX(final Parameter[] params, final Statement code) {
return new ClosureExpression(params, code);
}
public static ClosureExpression closureX(final Statement code) {
return closureX(Parameter.EMPTY_ARRAY, code);
}
/**
* Builds a lambda expression
*
* @param params lambda parameters
* @param code lambda code
* @return the lambda expression
*/
public static LambdaExpression lambdaX(final Parameter[] params, final Statement code) {
return new LambdaExpression(params, code);
}
/**
* Builds a lambda expression with no parameters
*
* @param code lambda code
* @return the lambda expression
*/
public static LambdaExpression lambdaX(final Statement code) {
return lambdaX(Parameter.EMPTY_ARRAY, code);
}
/**
* Builds a binary expression that compares two values.
*
* @param lhv expression for the value to compare from
* @param rhv expression for the value to compare to
* @return the expression comparing two values
*/
public static BinaryExpression cmpX(final Expression lhv, final Expression rhv) {
return binX(lhv, CMP, rhv);
}
public static ConstantExpression constX(final Object val) {
return new ConstantExpression(val);
}
public static ConstantExpression constX(final Object val, final boolean keepPrimitive) {
return new ConstantExpression(val, keepPrimitive);
}
public static ConstructorCallExpression ctorX(final ClassNode type, final Expression args) {
return new ConstructorCallExpression(type, args);
}
public static ConstructorCallExpression ctorX(final ClassNode type) {
return ctorX(type, ArgumentListExpression.EMPTY_ARGUMENTS);
}
public static Statement ctorSuperS(final Expression args) {
return stmt(ctorSuperX(args));
}
public static ConstructorCallExpression ctorSuperX(Expression args) {
return ctorX(ClassNode.SUPER, args);
}
public static Statement ctorThisS(final Expression args) {
return stmt(ctorThisX(args));
}
public static ConstructorCallExpression ctorThisX(Expression args) {
return ctorX(ClassNode.THIS, args);
}
public static Statement ctorSuperS() {
return stmt(ctorSuperX());
}
public static ConstructorCallExpression ctorSuperX() {
return ctorX(ClassNode.SUPER);
}
public static Statement ctorThisS() {
return stmt(ctorThisX());
}
public static ConstructorCallExpression ctorThisX() {
return ctorX(ClassNode.THIS);
}
public static Statement declS(final Expression target, final Expression init) {
return stmt(declX(target, init));
}
public static DeclarationExpression declX(final Expression target, final Expression init) {
return new DeclarationExpression(target, ASSIGN, init);
}
/**
* Returns a constant expression with the default value for the given type
* (i.e., {@code false} for boolean, {@code 0} for numbers or {@code null}.
*
* @since 4.0.0
*/
public static ConstantExpression defaultValueX(final ClassNode type) {
return Optional.ofNullable((ConstantExpression) getDefaultValueForPrimitive(type)).orElse(nullX());
}
public static ElvisOperatorExpression elvisX(final Expression base, final Expression otherwise) {
return new ElvisOperatorExpression(base, otherwise);
}
public static MapEntryExpression entryX(final Expression key, final Expression value) {
return new MapEntryExpression(key, value);
}
public static BinaryExpression eqX(final Expression left, final Expression right) {
return binX(left, EQ, right);
}
public static BooleanExpression equalsNullX(final Expression expr) {
return boolX(eqX(nullX(), expr));
}
public static FieldExpression fieldX(final FieldNode fieldNode) {
return new FieldExpression(fieldNode);
}
public static FieldExpression fieldX(final ClassNode owner, final String fieldName) {
return new FieldExpression(owner.getField(fieldName));
}
public static Expression findArg(final String argName) {
return propX(varX("args"), argName);
}
public static ForStatement forS(Parameter variable, Expression collectionExpression, Statement loopS) {
return new ForStatement(variable, collectionExpression, loopS);
}
public static List getAllMethods(final ClassNode type) {
ClassNode node = type;
List result = new ArrayList<>();
while (node != null) {
result.addAll(node.getMethods());
node = node.getSuperClass();
}
return result;
}
public static List getAllProperties(final ClassNode type) {
ClassNode node = type;
List result = new ArrayList<>();
while (node != null) {
result.addAll(node.getProperties());
node = node.getSuperClass();
}
return result;
}
public static List getInstanceNonPropertyFields(final ClassNode cNode) {
List result = new ArrayList<>();
for (FieldNode fNode : cNode.getFields()) {
if (!fNode.isStatic() && cNode.getProperty(fNode.getName()) == null) {
result.add(fNode);
}
}
return result;
}
public static List getInstanceNonPropertyFieldNames(final ClassNode cNode) {
List fList = getInstanceNonPropertyFields(cNode);
List result = new ArrayList<>(fList.size());
for (FieldNode fNode : fList) {
result.add(fNode.getName());
}
return result;
}
public static List getInstanceProperties(final ClassNode cNode) {
List result = new ArrayList<>();
for (PropertyNode pNode : cNode.getProperties()) {
if (!pNode.isStatic()) {
result.add(pNode);
}
}
return result;
}
public static List getInstancePropertyNames(final ClassNode cNode) {
List pList = BeanUtils.getAllProperties(cNode, false, false, true);
List result = new ArrayList<>(pList.size());
for (PropertyNode pNode : pList) {
result.add(pNode.getName());
}
return result;
}
public static List getInstancePropertyFields(final ClassNode cNode) {
List result = new ArrayList<>();
for (PropertyNode pNode : cNode.getProperties()) {
if (!pNode.isStatic()) {
result.add(pNode.getField());
}
}
return result;
}
private static void addAllInterfaces(final Set result, final ClassNode source) {
for (ClassNode in : source.getInterfaces()) {
in = parameterizeType(source, in);
if (result.add(in))
addAllInterfaces(result, in);
}
ClassNode sc = source.redirect().getUnresolvedSuperClass(false);
if (sc != null && !isObjectType(sc)) {
addAllInterfaces(result, parameterizeType(source, sc));
}
}
public static Set getInterfacesAndSuperInterfaces(final ClassNode cNode) {
Set result = new LinkedHashSet<>();
if (cNode.isInterface()) result.add(cNode);
addAllInterfaces(result, cNode);
return result;
}
public static List getSuperNonPropertyFields(final ClassNode cNode) {
List result;
if (isObjectType(cNode)) {
result = new ArrayList<>();
} else {
result = getSuperNonPropertyFields(cNode.getSuperClass());
}
for (FieldNode fNode : cNode.getFields()) {
if (!fNode.isStatic() && cNode.getProperty(fNode.getName()) == null) {
result.add(fNode);
}
}
return result;
}
public static List getSuperPropertyFields(final ClassNode cNode) {
List result;
if (isObjectType(cNode)) {
result = new ArrayList<>();
} else {
result = getSuperPropertyFields(cNode.getSuperClass());
}
for (PropertyNode pNode : cNode.getProperties()) {
if (!pNode.isStatic()) {
result.add(pNode.getField());
}
}
return result;
}
public static List getAllProperties(final Set names, final ClassNode cNode, final boolean includeProperties, final boolean includeFields, final boolean includePseudoGetters, final boolean includePseudoSetters, final boolean traverseSuperClasses, final boolean skipReadonly) {
return getAllProperties(names, cNode, cNode, includeProperties, includeFields, includePseudoGetters, includePseudoSetters, traverseSuperClasses, skipReadonly);
}
public static List getAllProperties(final Set names, final ClassNode origType, final ClassNode cNode, final boolean includeProperties, final boolean includeFields, final boolean includePseudoGetters, final boolean includePseudoSetters, final boolean traverseSuperClasses, final boolean skipReadonly) {
return getAllProperties(names, origType, cNode, includeProperties, includeFields, includePseudoGetters, includePseudoSetters, traverseSuperClasses, skipReadonly, false, false, false);
}
public static List getAllProperties(final Set names, final ClassNode origType, final ClassNode cNode, final boolean includeProperties,
final boolean includeFields, final boolean includePseudoGetters, final boolean includePseudoSetters,
final boolean traverseSuperClasses, final boolean skipReadonly, final boolean reverse, final boolean allNames, final boolean includeStatic) {
List result = new ArrayList<>();
if (!(isObjectType(cNode)) && traverseSuperClasses && !reverse) {
result.addAll(getAllProperties(names, origType, cNode.getSuperClass(), includeProperties, includeFields, includePseudoGetters, includePseudoSetters, true, skipReadonly));
}
if (includeProperties) {
for (PropertyNode pNode : cNode.getProperties()) {
if ((!pNode.isStatic() || includeStatic) && !names.contains(pNode.getName())) {
result.add(pNode);
names.add(pNode.getName());
}
}
if (includePseudoGetters || includePseudoSetters) {
BeanUtils.addPseudoProperties(origType, cNode, result, names, includeStatic, includePseudoGetters, includePseudoSetters, traverseSuperClasses);
}
}
if (includeFields) {
for (FieldNode fNode : cNode.getFields()) {
if ((fNode.isStatic() && !includeStatic) || fNode.isSynthetic() || cNode.getProperty(fNode.getName()) != null || names.contains(fNode.getName())) {
continue;
}
// internal field
if (fNode.getName().contains("$") && !allNames) {
continue;
}
if (fNode.isPrivate() && !cNode.equals(origType)) {
continue;
}
if (fNode.isFinal() && fNode.getInitialExpression() != null && skipReadonly) {
continue;
}
result.add(new PropertyNode(fNode, fNode.getModifiers(), null, null));
names.add(fNode.getName());
}
}
if (!(isObjectType(cNode)) && traverseSuperClasses && reverse) {
result.addAll(getAllProperties(names, origType, cNode.getSuperClass(), includeProperties, includeFields, includePseudoGetters, includePseudoSetters, true, skipReadonly));
}
return result;
}
/**
* This method is similar to {@link #propX(Expression, Expression)} but will make sure that if the property
* being accessed is defined inside the classnode provided as a parameter, then a getter call is generated
* instead of a field access.
* @param annotatedNode the class node where the property node is accessed from
* @param pNode the property being accessed
* @return a method call expression or a property expression
*/
public static Expression getterThisX(final ClassNode annotatedNode, final PropertyNode pNode) {
ClassNode owner = pNode.getDeclaringClass();
if (annotatedNode.equals(owner)) {
return callThisX(pNode.getGetterNameOrDefault());
}
return propX(varX("this"), pNode.getName());
}
/**
* This method is similar to {@link #propX(Expression, Expression)} but will make sure that if the property
* being accessed is defined inside the classnode provided as a parameter, then a getter call is generated
* instead of a field access.
* @param annotatedNode the class node where the property node is accessed from
* @param receiver the object having the property
* @param pNode the property being accessed
* @return a method call expression or a property expression
*/
public static Expression getterX(final ClassNode annotatedNode, final Expression receiver, final PropertyNode pNode) {
ClassNode owner = pNode.getDeclaringClass();
if (annotatedNode.equals(owner)) {
return callX(receiver, pNode.getGetterNameOrDefault());
}
return propX(receiver, pNode.getName());
}
public static BinaryExpression hasClassX(final Expression instance, final ClassNode cNode) {
return eqX(classX(cNode), callX(instance, "getClass"));
}
public static BinaryExpression hasEqualFieldX(final FieldNode fNode, final Expression other) {
return eqX(varX(fNode), propX(other, fNode.getName()));
}
public static BinaryExpression hasEqualPropertyX(final ClassNode cNode, final PropertyNode pNode, final VariableExpression other) {
return eqX(getterThisX(cNode, pNode), getterX(other.getOriginType(), other, pNode));
}
@Deprecated
public static BinaryExpression hasEqualPropertyX(final PropertyNode pNode, final Expression other) {
String getterName = pNode.getGetterNameOrDefault();
return eqX(callThisX(getterName), callX(other, getterName));
}
public static BooleanExpression hasSameFieldX(final FieldNode fNode, final Expression other) {
return sameX(varX(fNode), propX(other, fNode.getName()));
}
public static BooleanExpression hasSamePropertyX(final PropertyNode pNode, final Expression other) {
ClassNode cNode = pNode.getDeclaringClass();
return sameX(getterThisX(cNode, pNode), getterX(cNode, other, pNode));
}
@Deprecated
public static Statement ifElseS$$bridge(final Expression cond, final Statement thenStmt, final Statement elseStmt) {
return ifElseS(cond, thenStmt, elseStmt);
}
public static IfStatement ifElseS(final Expression cond, final Statement thenStmt, final Statement elseStmt) {
return new IfStatement(
cond instanceof BooleanExpression ? (BooleanExpression) cond : boolX(cond),
thenStmt,
elseStmt
);
}
@Deprecated
public static Statement ifS$$bridge(final Expression cond, final Expression trueExpr) {
return ifS(cond, trueExpr);
}
public static IfStatement ifS(final Expression cond, final Expression trueExpr) {
return ifElseS(cond, stmt(trueExpr), EmptyStatement.INSTANCE);
}
@Deprecated
public static Statement ifS$$bridge(final Expression cond, final Statement trueStmt) {
return ifS(cond, trueStmt);
}
public static IfStatement ifS(final Expression cond, final Statement trueStmt) {
return ifElseS(cond, trueStmt, EmptyStatement.INSTANCE);
}
public static Expression indexX(final Expression target, final Expression value) {
return binX(target, INDEX, value);
}
public static BooleanExpression isInstanceOfX(final Expression expr, final ClassNode type) {
return boolX(binX(expr, INSTANCEOF, classX(type)));
}
/**
* @since 4.0.8
*/
public static BooleanExpression isNullOrInstanceOfX(final Expression expr, final ClassNode type) {
return boolX(orX(binX(nullX(), Token.newSymbol(Types.COMPARE_IDENTICAL, -1, -1), expr), binX(expr, INSTANCEOF, classX(type))));
}
/**
* Alias for {@link #equalsNullX(Expression)}
*/
public static BooleanExpression isNullX(final Expression expr) {
return equalsNullX(expr);
}
public static BooleanExpression isOneX(final Expression expr) {
return boolX(binX(expr, EQ, constX(1)));
}
public static BooleanExpression isTrueX(final Expression argExpr) {
return boolX(binX(argExpr, EQ, constX(Boolean.TRUE)));
}
public static BooleanExpression isZeroX(final Expression expr) {
return boolX(binX(expr, EQ, constX(0)));
}
public static ListExpression listX(final List args) {
return new ListExpression(args);
}
public static ListExpression list2args(final List args) {
ListExpression result = new ListExpression();
for (Object o : args) {
result.addExpression(constX(o));
}
return result;
}
public static ListExpression classList2args(final List args) {
ListExpression result = new ListExpression();
for (Object o : args) {
result.addExpression(classX(ClassHelper.make(o.toString())));
}
return result;
}
public static VariableExpression localVarX(final String name) {
VariableExpression result = varX(name);
result.setAccessedVariable(result);
return result;
}
public static VariableExpression localVarX(final String name, final ClassNode type) {
VariableExpression result = varX(name, type);
result.setAccessedVariable(result);
return result;
}
public static BinaryExpression ltX(final Expression lhv, final Expression rhv) {
return binX(lhv, LT, rhv);
}
public static MapEntryExpression mapEntryX(final Expression keyExpr, final Expression valueExpr) {
return new MapEntryExpression(keyExpr, valueExpr);
}
public static MapEntryExpression mapEntryX(final String key, final Expression valueExpr) {
return new MapEntryExpression(constX(key), valueExpr);
}
public static MapExpression mapX(final List expressions) {
return new MapExpression(expressions);
}
public static BinaryExpression neX(final Expression lhv, final Expression rhv) {
return binX(lhv, NE, rhv);
}
public static BinaryExpression notIdenticalX(final Expression lhv, final Expression rhv) {
return binX(lhv, NOT_IDENTICAL, rhv);
}
public static BooleanExpression notNullX(final Expression expr) {
return boolX(binX(nullX(), NE, expr));
}
public static NotExpression notX(final Expression expr) {
return new NotExpression(expr instanceof BooleanExpression ? expr : boolX(expr));
}
public static ConstantExpression nullX() {
return constX(null);
}
public static BinaryExpression orX(final Expression lhv, final Expression rhv) {
return binX(lhv, OR, rhv);
}
public static Parameter param(final ClassNode type, final String name) {
return param(type, name, null);
}
public static Parameter param(final ClassNode type, final String name, final Expression initialExpression) {
Parameter param = new Parameter(type, name);
if (initialExpression != null) {
param.setInitialExpression(initialExpression);
}
return param;
}
public static Parameter[] params(final Parameter... params) {
return (params != null ? params : Parameter.EMPTY_ARRAY);
}
public static BinaryExpression plusX(final Expression lhv, final Expression rhv) {
return binX(lhv, PLUS, rhv);
}
public static PropertyExpression propX(final Expression owner, final String property) {
return new PropertyExpression(owner, property);
}
@Deprecated
public static Expression propX$$bridge(final Expression owner, final String property) {
return propX(owner, property);
}
public static PropertyExpression propX(final Expression owner, final Expression property) {
return new PropertyExpression(owner, property);
}
@Deprecated
public static Expression propX$$bridge(final Expression owner, final Expression property) {
return propX(owner, property);
}
public static PropertyExpression propX(final Expression owner, final Expression property, final boolean safe) {
return new PropertyExpression(owner, property, safe);
}
public static Statement returnS(final Expression expr) {
return new ReturnStatement(expr);
}
public static Statement safeExpression(final Expression fieldExpr, final Expression expression) {
return new IfStatement(isNullX(fieldExpr), stmt(fieldExpr), stmt(expression));
}
public static BooleanExpression sameX(final Expression self, final Expression other) {
return boolX(callX(self, "is", args(other)));
}
public static Statement stmt(final Expression expr) {
return new ExpressionStatement(expr);
}
public static SwitchStatement switchS(final Expression expr) {
return new SwitchStatement(expr);
}
public static SwitchStatement switchS(final Expression expr, final Statement defaultStatement) {
return new SwitchStatement(expr, defaultStatement);
}
public static SwitchStatement switchS(final Expression expr, final List caseStatements, final Statement defaultStatement) {
return new SwitchStatement(expr, caseStatements, defaultStatement);
}
public static TernaryExpression ternaryX(final Expression cond, final Expression trueExpr, final Expression elseExpr) {
return new TernaryExpression(
cond instanceof BooleanExpression ? (BooleanExpression) cond : boolX(cond),
trueExpr,
elseExpr);
}
public static PropertyExpression thisPropX(final boolean implicit, final String property) {
PropertyExpression pexp = propX(varX("this"), property);
pexp.setImplicitThis(implicit);
return pexp;
}
public static ThrowStatement throwS(final Expression expr) {
return new ThrowStatement(expr);
}
public static TryCatchStatement tryCatchS(final Statement tryStatement) {
return tryCatchS(tryStatement, EmptyStatement.INSTANCE);
}
public static TryCatchStatement tryCatchS(final Statement tryStatement, final Statement finallyStatement) {
return new TryCatchStatement(tryStatement, finallyStatement);
}
public static TryCatchStatement tryCatchS(final Statement tryStatement, final Statement finallyStatement, final CatchStatement... catchStatements) {
TryCatchStatement result = new TryCatchStatement(tryStatement, finallyStatement);
for (CatchStatement catchStatement : catchStatements) {
result.addCatch(catchStatement);
}
return result;
}
public static VariableExpression varX(final String name) {
return new VariableExpression(name);
}
public static VariableExpression varX(final Variable variable) {
return new VariableExpression(variable);
}
public static VariableExpression varX(final String name, final ClassNode type) {
return new VariableExpression(name, type);
}
//--------------------------------------------------------------------------
public static Parameter[] cloneParams(final Parameter[] parameters) {
return Arrays.stream(parameters).map(p -> param(p.getOriginType(), p.getName())).toArray(Parameter[]::new);
}
/**
* Copies all candidateAnnotations with retention policy {@link java.lang.annotation.RetentionPolicy#RUNTIME}
* and {@link java.lang.annotation.RetentionPolicy#CLASS}.
*
* Annotations with {@link org.codehaus.groovy.runtime.GeneratedClosure} members are not supported at present.
*/
public static void copyAnnotatedNodeAnnotations(final AnnotatedNode annotatedNode, final List copied, final List notCopied) {
copyAnnotatedNodeAnnotations(annotatedNode, copied, notCopied, true);
}
/**
* Copies all candidateAnnotations with retention policy {@link java.lang.annotation.RetentionPolicy#RUNTIME}
* and {@link java.lang.annotation.RetentionPolicy#CLASS}.
* {@link groovy.transform.Generated} annotations will be copied if {@code includeGenerated} is true.
*
* Annotations with {@link org.codehaus.groovy.runtime.GeneratedClosure} members are not supported at present.
*/
public static void copyAnnotatedNodeAnnotations(final AnnotatedNode annotatedNode, final List copied, final List notCopied, final boolean includeGenerated) {
List annotationList = annotatedNode.getAnnotations();
for (AnnotationNode annotation : annotationList) {
List annotations = annotation.getClassNode().getAnnotations(AbstractASTTransformation.RETENTION_CLASSNODE);
if (annotations.isEmpty()) continue;
if (hasClosureMember(annotation)) {
notCopied.add(annotation);
continue;
}
if (!includeGenerated && annotation.getClassNode().getName().equals("groovy.transform.Generated")) {
continue;
}
AnnotationNode retentionPolicyAnnotation = annotations.get(0);
Expression valueExpression = retentionPolicyAnnotation.getMember("value");
if (!(valueExpression instanceof PropertyExpression)) continue;
PropertyExpression propertyExpression = (PropertyExpression) valueExpression;
boolean processAnnotation = propertyExpression.getProperty() instanceof ConstantExpression
&& ("RUNTIME".equals(((ConstantExpression) (propertyExpression.getProperty())).getValue())
|| "CLASS".equals(((ConstantExpression) (propertyExpression.getProperty())).getValue()));
if (processAnnotation) {
AnnotationNode newAnnotation = new AnnotationNode(annotation.getClassNode());
for (Map.Entry member : annotation.getMembers().entrySet()) {
newAnnotation.addMember(member.getKey(), member.getValue());
}
newAnnotation.setSourcePosition(annotatedNode);
copied.add(newAnnotation);
}
}
}
public static Statement createConstructorStatementDefault(final FieldNode fNode) {
final String name = fNode.getName();
final ClassNode fType = fNode.getType();
final Expression fieldExpr = propX(varX("this"), name);
Expression initExpr = fNode.getInitialValueExpression();
Statement assignInit;
if (initExpr == null || (initExpr instanceof ConstantExpression && ((ConstantExpression) initExpr).isNullExpression())) {
if (ClassHelper.isPrimitiveType(fType)) {
assignInit = EmptyStatement.INSTANCE;
} else {
assignInit = assignS(fieldExpr, ConstantExpression.EMPTY_EXPRESSION);
}
} else {
assignInit = assignS(fieldExpr, initExpr);
}
fNode.setInitialValueExpression(null);
Expression value = findArg(name);
return ifElseS(isNullX(value), assignInit, assignS(fieldExpr, castX(fType, value)));
}
/**
* Generally preferred to use {@link PropertyNode#getGetterNameOrDefault()} directly.
*/
public static String getGetterName(final PropertyNode pNode) {
return pNode.getGetterNameOrDefault();
}
/**
* WARNING: Avoid this method unless just the name and type are available.
* Use {@link #getGetterName(PropertyNode)} if the propertyNode is available.
*/
public static String getGetterName(final String name, final Class type) {
return MetaProperty.getGetterName(name, type);
}
/**
* WARNING: Avoid this method unless just the name is available.
* Use {@link #getGetterName(PropertyNode)} if the propertyNode is available.
* Use {@link #getGetterName(String, Class)} if the type is available.
*/
public static String getGetterName(final String name) {
return MetaProperty.getGetterName(name, Object.class);
}
public static String getSetterName(final String name) {
return MetaProperty.getSetterName(name);
}
/**
* Converts an expression into the String source. Only some specific expressions like closure expression
* support this.
*
* @param readerSource a source
* @param expression an expression. Can't be null
* @return the source the closure was created from
* @throws java.lang.IllegalArgumentException when expression is null
* @throws java.lang.Exception when closure can't be read from source
*/
public static String convertASTToSource(final ReaderSource readerSource, final ASTNode expression) throws Exception {
if (expression == null) throw new IllegalArgumentException("Null: expression");
StringBuilder result = new StringBuilder();
for (int x = expression.getLineNumber(), y = expression.getLastLineNumber(); x <= y; x += 1) {
String line = readerSource.getLine(x, null);
if (line == null) {
throw new Exception(
"Error calculating source code for expression. Trying to read line " + x + " from " + readerSource.getClass()
);
}
if (x == expression.getLastLineNumber()) {
line = line.substring(0, expression.getLastColumnNumber() - 1);
}
if (x == expression.getLineNumber()) {
line = line.substring(expression.getColumnNumber() - 1);
}
//restoring line breaks is important b/c of lack of semicolons
result.append(line).append('\n');
}
String source = result.toString().trim();
return source;
}
public static boolean copyStatementsWithSuperAdjustment(final ClosureExpression pre, final BlockStatement body) {
Statement preCode = pre.getCode();
boolean changed = false;
if (preCode instanceof BlockStatement) {
BlockStatement block = (BlockStatement) preCode;
List statements = block.getStatements();
for (int i = 0, n = statements.size(); i < n; i += 1) {
Statement statement = statements.get(i);
// adjust the first statement if it's a super call
if (i == 0 && statement instanceof ExpressionStatement) {
ExpressionStatement es = (ExpressionStatement) statement;
Expression preExp = es.getExpression();
if (preExp instanceof MethodCallExpression) {
MethodCallExpression mce = (MethodCallExpression) preExp;
String name = mce.getMethodAsString();
if ("super".equals(name)) {
es.setExpression(new ConstructorCallExpression(ClassNode.SUPER, mce.getArguments()));
changed = true;
}
}
}
body.addStatement(statement);
}
}
return changed;
}
private static boolean hasClosureMember(final AnnotationNode annotation) {
Map members = annotation.getMembers();
for (Map.Entry member : members.entrySet()) {
if (member.getValue() instanceof ClosureExpression) return true;
if (member.getValue() instanceof ClassExpression) {
ClassExpression classExpression = (ClassExpression) member.getValue();
Class typeClass = classExpression.getType().isResolved() ? classExpression.getType().redirect().getTypeClass() : null;
if (typeClass != null && GeneratedClosure.class.isAssignableFrom(typeClass)) return true;
}
}
return false;
}
public static boolean hasDeclaredMethod(final ClassNode cNode, final String name, final int argsCount) {
List methods = cNode.getDeclaredMethods(name);
for (MethodNode method : methods) {
Parameter[] params = method.getParameters();
if (params != null && params.length == argsCount) {
return true;
}
}
return false;
}
public static MethodNode findDeclaredMethod(final ClassNode cNode, final String name, final int argsCount) {
// TODO ignore bridge methods?
List methods = cNode.getDeclaredMethods(name);
for (MethodNode method : methods) {
Parameter[] params = method.getParameters();
if (params != null && params.length == argsCount) {
return method;
}
}
return null;
}
public static boolean inSamePackage(final ClassNode first, final ClassNode second) {
PackageNode firstPackage = first.getPackage();
PackageNode secondPackage = second.getPackage();
return ((firstPackage == null && secondPackage == null)
|| firstPackage != null && secondPackage != null && firstPackage.getName().equals(secondPackage.getName()));
}
public static boolean inSamePackage(final Class first, final Class second) {
Package firstPackage = first.getPackage();
Package secondPackage = second.getPackage();
return ((firstPackage == null && secondPackage == null)
|| firstPackage != null && secondPackage != null && firstPackage.getName().equals(secondPackage.getName()));
}
public static boolean isDefaultVisibility(final int modifiers) {
return (modifiers & (Modifier.PRIVATE | Modifier.PUBLIC | Modifier.PROTECTED)) == 0;
}
public static boolean isOrImplements(final ClassNode type, final ClassNode interfaceType) {
return type.equals(interfaceType) || type.implementsInterface(interfaceType);
}
/**
* @deprecated use MethodNodeUtils#methodDescriptorWithoutReturnType(MethodNode) instead
*/
@Deprecated
public static String makeDescriptorWithoutReturnType(final MethodNode mn) {
StringBuilder sb = new StringBuilder();
sb.append(mn.getName()).append(':');
for (Parameter p : mn.getParameters()) {
sb.append(p.getType()).append(',');
}
return sb.toString();
}
public static boolean maybeFallsThrough(final Statement statement) {
if (statement.isEmpty()) return true;
if (statement instanceof ReturnStatement) {
return false;
} else if (statement instanceof ThrowStatement) {
return false;
} else if (statement instanceof BlockStatement) {
List list = ((BlockStatement) statement).getStatements();
final int last = list.size() - 1;
if (!maybeFallsThrough(list.get(last))) return false;
for (int i = 0; i < last; i += 1)
if (!maybeFallsThrough(list.get(i))) return false;
} else if (statement instanceof IfStatement) {
return maybeFallsThrough(((IfStatement) statement).getElseBlock())
|| maybeFallsThrough(((IfStatement) statement).getIfBlock());
} else if (statement instanceof SwitchStatement) {
// TODO
} else if (statement instanceof TryCatchStatement) {
TryCatchStatement tryCatch = (TryCatchStatement) statement;
if (!maybeFallsThrough(tryCatch.getFinallyStatement())) return false;
for (CatchStatement cs : tryCatch.getCatchStatements())
if (maybeFallsThrough(cs.getCode())) return true;
return maybeFallsThrough(tryCatch.getTryStatement());
} else if (statement instanceof SynchronizedStatement) {
return maybeFallsThrough(((SynchronizedStatement) statement).getCode());
}
return true;
}
}