AnnotatedFinder.java

package com.simpligility.maven.plugins.android.asm;

import java.util.concurrent.atomic.AtomicBoolean;
import org.objectweb.asm.AnnotationVisitor;
import org.objectweb.asm.Attribute;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.FieldVisitor;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.Type;

/**
 * Finds classes annotated with a set of annotations.
 *
 * @author secondsun@gmail.com
 */
public class AnnotatedFinder extends ClassVisitor 
{

    private static final String TEST_RUNNER = "Lorg/junit/runner/RunWith;";
    
    public AnnotatedFinder( String[] parentPackages ) 
    {
        super( Opcodes.ASM4 );
    }

    private final AtomicBoolean isDescendantFound = new AtomicBoolean( false );

    @Override
    public void visit( int version, int access, String name, String signature, String superName, String[] interfaces )
    {
    }

    private void flagAsFound()
    {
        isDescendantFound.set( true );
    }

    /**
     * Returns whether a match was found.
     *
     * @return <code>true</code> is a match was found, <code>false</code> otherwise.
     */
    public boolean isDescendantFound()
    {
        return isDescendantFound.get();
    }

    @Override
    public void visitSource( String source, String debug )
    {
    }

    @Override
    public void visitOuterClass( String owner, String name, String desc )
    {
    }

    @Override
    public AnnotationVisitor visitAnnotation( String desc, boolean visible )
    {
        if ( TEST_RUNNER.equals( desc ) ) 
        {
            return new AnnotationVisitor( Opcodes.ASM4 ) 
            {

                @Override
                public void visit( String name, Object value ) 
                {
                    if ( value instanceof Type ) 
                    {
                        if ( ( ( Type ) value ).getClassName().contains( "AndroidJUnit4" ) ) 
                        {
                            flagAsFound();
                        }
                    }
                }
                
            };
        } 
        return null;
    }

    @Override
    public void visitAttribute( Attribute attr )
    {
    }

    @Override
    public void visitInnerClass( String name, String outerName, String innerName, int access )
    {
    }

    @Override
    public FieldVisitor visitField( int access, String name, String desc, String signature, Object value )
    {
        return null;
    }

    @Override
    public MethodVisitor visitMethod( int access, String name, String desc, String signature, String[] exceptions )
    {
        return null;
    }

    @Override
    public void visitEnd()
    {
    }
    
}