package org.glassfish.pfl.tf.tools.enhancer;
import org.glassfish.pfl.basic.tools.file.FileWrapper;
import org.glassfish.pfl.basic.tools.file.Scanner;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.glassfish.pfl.tf.timer.spi.TimingInfoProcessor;
import org.glassfish.pfl.tf.spi.Util;
import org.glassfish.pfl.tf.spi.annotation.MethodMonitorGroup;
import org.glassfish.pfl.objectweb.asm.AnnotationVisitor;
import org.glassfish.pfl.objectweb.asm.ClassReader;
import org.glassfish.pfl.objectweb.asm.ClassVisitor;
import org.glassfish.pfl.objectweb.asm.FieldVisitor;
import org.glassfish.pfl.objectweb.asm.MethodVisitor;
import org.glassfish.pfl.objectweb.asm.Opcodes;
import org.glassfish.pfl.objectweb.asm.Type;
import org.glassfish.pfl.objectweb.asm.commons.EmptyVisitor;
public class AnnotationScannerAction implements Scanner.Action {
private static Class<MethodMonitorGroup> MMG_CLASS =
MethodMonitorGroup.class ;
private static String MMG_DESCRIPTOR =
Type.getType(MMG_CLASS).getDescriptor() ;
private final Util util ;
private final TimingInfoProcessor tip ;
private Set<String> annotationNames = new HashSet<String>() ;
private String currentClass ;
AnnotationScannerAction(Util util, TimingInfoProcessor tip) {
this.util = util ;
this.tip = tip ;
}
public Set<String> getAnnotationNames() {
return annotationNames ;
}
private class AnnoScanner extends EmptyVisitor {
private boolean visitingAnnotation = false ;
private String annotationValueName = null ;
private String timerGroupDescription ;
private String timerGroupName ;
private List<Type> timerGroupMembers ;
@Override
public void visit( int version, int access, String name, String signature,
String superName, String[] interfaces ) {
util.info( 3, "Visiting class " + name ) ;
if ((access & Opcodes.ACC_ANNOTATION) == Opcodes.ACC_ANNOTATION) {
util.info( 2, "\t(Annotation)") ;
currentClass = name ;
}
}
private String getGroupName( String desc ) {
String result = desc ;
final int index = desc.lastIndexOf('/') ;
if (index >= 0) {
result = desc.substring( index + 1 ) ;
}
return result ;
}
@Override
public AnnotationVisitor visitAnnotation( String desc,
boolean visible ) {
util.info( 3, "\tVisiting annotation " + desc ) ;
if (desc.equals( MMG_DESCRIPTOR )) {
annotationNames.add( currentClass );
visitingAnnotation = true ;
timerGroupName = getGroupName( currentClass ) ;
timerGroupDescription = "TimerGroup for Annotation "
+ timerGroupName ;
timerGroupMembers = new ArrayList<Type>() ;
return this ;
}
return null ;
}
@Override
public void visit( String name, Object value ) {
if (name == null) {
if (annotationValueName.equals( "value" )) {
timerGroupMembers.add( (Type)value ) ;
}
} else if(name.equals("description")) {
if (value instanceof String) {
timerGroupDescription = (String)value ;
}
} else if (name.equals( "value")) {
if (value instanceof Type[]) {
timerGroupMembers = Arrays.asList( (Type[])value ) ;
}
}
}
@Override
public AnnotationVisitor visitArray( String name ) {
annotationValueName = name ;
return this ;
}
@Override
public void visitEnd() {
if (visitingAnnotation) {
visitingAnnotation = false ;
tip.addTimerGroup( timerGroupName, timerGroupDescription ) ;
for (Type type : timerGroupMembers) {
String name = type.getClassName() ;
final int index = name.lastIndexOf('.') ;
if (index >= 0) {
name = name.substring(index + 1) ;
}
tip.contains( name ) ;
}
}
}
@Override
public MethodVisitor visitMethod( final int access, final String name,
final String desc, final String signature,
final String[] exceptions ) {
return null ;
}
@Override
public FieldVisitor visitField( final int access, final String name,
final String desc, final String signature,
final Object value ) {
return null ;
}
}
public boolean evaluate(FileWrapper fw) {
try {
byte[] inputData = fw.readAll();
ClassReader cr = new ClassReader( inputData ) ;
ClassVisitor as = new AnnoScanner() ;
cr.accept( as, 0 );
} catch (IOException ex) {
return true ;
}
return true ;
}
}