org.babyfish.lang.bytecode.ASMTreeUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of babyfish-instrument-maven-plugin Show documentation
This tool project supports some maven plugins of BabyFish, it shouldn't used by the user directly.
The newest version!
/*
 * BabyFish, Object Model Framework for Java and JPA.
 * https://github.com/babyfish-ct/babyfish
 *
 * Copyright (c) 2008-2016, Tao Chen
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * Please visit "http://opensource.org/licenses/LGPL-3.0" to know more.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 */
package org.babyfish.lang.bytecode;

import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.function.Consumer;

import org.babyfish.org.objectweb.asm.Type;
import org.babyfish.org.objectweb.asm.tree.AnnotationNode;
import org.babyfish.org.objectweb.asm.tree.ClassNode;
import org.babyfish.org.objectweb.asm.tree.FieldNode;
import org.babyfish.org.objectweb.asm.tree.MethodNode;

/**
 * @author Tao Chen
 */
public class ASMTreeUtils {

    @Deprecated
    protected ASMTreeUtils() {
        throw new UnsupportedOperationException();
    }
    
    public static List getVisibleAnnotations(Object node) {
        List list = null;
        if (node instanceof ClassNode) {
            list = ((ClassNode)node).visibleAnnotations;
        } else if (node instanceof FieldNode) {
            list = ((FieldNode)node).visibleAnnotations;
        } else if (node instanceof MethodNode) {
            list = ((MethodNode)node).visibleAnnotations;
        } else {
            throw new IllegalArgumentException(
                    "node must be instance of \""
                    + ClassNode.class.getName()
                    + "\", \""
                    + MethodNode.class.getName()
                    + "\" or \""
                    + FieldNode.class.getName()
                    + "\""
            );
        }
        if (list == null) {
            return Collections.emptyList();
        }
        return list;
    }
    
    public static List getInvisibleAnnotations(Object node) {
        List list = null;
        if (node instanceof ClassNode) {
            list = ((ClassNode)node).invisibleAnnotations;
        } else if (node instanceof FieldNode) {
            list = ((FieldNode)node).invisibleAnnotations;
        } else if (node instanceof MethodNode) {
            list = ((MethodNode)node).invisibleAnnotations;
        } else {
            throw new IllegalArgumentException(
                    "node must be instance of \""
                    + ClassNode.class.getName()
                    + "\", \""
                    + MethodNode.class.getName()
                    + "\" or \""
                    + FieldNode.class.getName()
                    + "\""
            );
        }
        if (list == null) {
            return Collections.emptyList();
        }
        return list;
    }
    
    public static void consumeAnnotationNodes(Object asmTreeNode, Consumer consumer) {
        if (asmTreeNode instanceof ClassNode) {
            consumeAnnotationNodes((ClassNode)asmTreeNode, consumer);
        } else if (asmTreeNode instanceof FieldNode) {
            consumeAnnotationNodes((FieldNode)asmTreeNode, consumer);
        } else if (asmTreeNode instanceof MethodNode) {
            consumeAnnotationNodes((MethodNode)asmTreeNode, consumer);
        } else {
            throw new IllegalArgumentException(
                    "\"asmTreeNode\" must be instance of \""
                    + ClassNode.class.getName()
                    + "\", \""
                    + FieldNode.class.getName()
                    + "\" or \""
                    + MethodNode.class.getName()
                    + "\""
            );
        }
    }
    
    public static void consumeAnnotationNodes(ClassNode classNode, Consumer consumer) {
        if (classNode.visibleAnnotations != null) {
            for (AnnotationNode annotationNode : classNode.visibleAnnotations) {
                consumer.accept(annotationNode);
            }
        }
        if (classNode.invisibleAnnotations != null) {
            for (AnnotationNode annotationNode : classNode.invisibleAnnotations) {
                consumer.accept(annotationNode);
            }
        }
    }
    
    public static void consumeAnnotationNodes(FieldNode fieldNode, Consumer consumer) {
        if (fieldNode.visibleAnnotations != null) {
            for (AnnotationNode annotationNode : fieldNode.visibleAnnotations) {
                consumer.accept(annotationNode);
            }
        }
        if (fieldNode.invisibleAnnotations != null) {
            for (AnnotationNode annotationNode : fieldNode.invisibleAnnotations) {
                consumer.accept(annotationNode);
            }
        }
    }
    
    public static void consumeAnnotationNodes(MethodNode methodNode, Consumer consumer) {
        if (methodNode.visibleAnnotations != null) {
            for (AnnotationNode annotationNode : methodNode.visibleAnnotations) {
                consumer.accept(annotationNode);
            }
        }
        if (methodNode.invisibleAnnotations != null) {
            for (AnnotationNode annotationNode : methodNode.invisibleAnnotations) {
                consumer.accept(annotationNode);
            }
        }
    }
    
    public static AnnotationNode getAnnotationNode(Object node, Class annotationType) {
        String desc = Type.getDescriptor(annotationType);
        for (AnnotationNode annotationNode : getVisibleAnnotations(node)) {
            if (annotationNode.desc.equals(desc)) {
                return annotationNode;
            }
        }
        for (AnnotationNode annotationNode : getInvisibleAnnotations(node)) {
            if (annotationNode.desc.equals(desc)) {
                return annotationNode;
            }
        }
        return null;
    }
    
    @SuppressWarnings("unchecked")
    public static  T getAnnotationValue(AnnotationNode annotationNode, String name) {
        if (annotationNode.values != null) {
            Iterator