maint-5-2/tapestry-ioc/src/main/java/org/apache/tapestry5/ioc/internal/services/AnnotationMemberValueVisitor.java

// Copyright 2010 The Apache Software Foundation
//
// 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.apache.tapestry5.ioc.internal.services;

import java.lang.reflect.Method;
import java.util.List;

import javassist.bytecode.ConstPool;
import javassist.bytecode.annotation.Annotation;
import javassist.bytecode.annotation.AnnotationMemberValue;
import javassist.bytecode.annotation.ArrayMemberValue;
import javassist.bytecode.annotation.BooleanMemberValue;
import javassist.bytecode.annotation.ByteMemberValue;
import javassist.bytecode.annotation.CharMemberValue;
import javassist.bytecode.annotation.ClassMemberValue;
import javassist.bytecode.annotation.DoubleMemberValue;
import javassist.bytecode.annotation.EnumMemberValue;
import javassist.bytecode.annotation.FloatMemberValue;
import javassist.bytecode.annotation.IntegerMemberValue;
import javassist.bytecode.annotation.LongMemberValue;
import javassist.bytecode.annotation.MemberValue;
import javassist.bytecode.annotation.MemberValueVisitor;
import javassist.bytecode.annotation.ShortMemberValue;
import javassist.bytecode.annotation.StringMemberValue;

import org.apache.tapestry5.ioc.internal.util.CollectionFactory;

public class AnnotationMemberValueVisitor implements MemberValueVisitor
{
    private final ConstPool constPool;

    private final CtClassSource classSource;

    private final Object value;

    public AnnotationMemberValueVisitor(final ConstPool constPool, final CtClassSource classSource, final Object value)
    {
        this.constPool = constPool;
        this.classSource = classSource;
        this.value = value;
    }

    public void visitAnnotationMemberValue(final AnnotationMemberValue mb)
    {
        Class annotationType = getClass(value);
        
        final Method[] methods = annotationType.getDeclaredMethods();

        for (final Method method : methods)
        {
            try
            {
                final Object result = method.invoke(value);

                final MemberValue memberValue = Annotation.createMemberValue(
                        this.constPool,
                        this.classSource.toCtClass(result.getClass()));

                memberValue.accept(new AnnotationMemberValueVisitor(this.constPool, this.classSource, result));
                
                mb.getValue().addMemberValue(method.getName(), memberValue);
            }
            catch (Exception e)
            {
                throw new RuntimeException(e);
            }
        }
        
    }

    public void visitArrayMemberValue(final ArrayMemberValue mb)
    {
        final Object[] array = (Object[]) this.value;

        final List<MemberValue> members = CollectionFactory.newList();
        
        for (final Object object : array)
        {
            try
            {
                final MemberValue memberValue = Annotation.createMemberValue(
                        this.constPool,
                        this.classSource.toCtClass(getClass(object)));

                memberValue.accept(new AnnotationMemberValueVisitor(this.constPool, this.classSource, object));

                members.add(memberValue);
            }
            catch (final Exception e)
            {
                throw new RuntimeException(e);
            }
        }

        mb.setValue(members.toArray(new MemberValue[] {}));
    }
    
    private Class getClass(Object object)
    {
        if(object instanceof java.lang.annotation.Annotation)
        {
            return ((java.lang.annotation.Annotation)object).annotationType();
        }
        
        return object.getClass();
    }

    public void visitBooleanMemberValue(final BooleanMemberValue mb)
    {
        mb.setValue((Boolean) this.value);
    }

    public void visitByteMemberValue(final ByteMemberValue mb)
    {
        mb.setValue((Byte) this.value);
    }

    public void visitCharMemberValue(final CharMemberValue mb)
    {
        mb.setValue((Character) this.value);
    }

    public void visitDoubleMemberValue(final DoubleMemberValue mb)
    {
        mb.setValue((Double) this.value);
    }

    public void visitEnumMemberValue(final EnumMemberValue mb)
    {
        final Enum enumeration = (Enum) this.value;
        
        final Class type = enumeration.getDeclaringClass();
        mb.setType(type.getName());
        mb.setValue(enumeration.name());
    }

    public void visitFloatMemberValue(final FloatMemberValue mb)
    {
        mb.setValue((Float) this.value);
    }

    public void visitIntegerMemberValue(final IntegerMemberValue mb)
    {
        mb.setValue((Integer) this.value);
    }

    public void visitLongMemberValue(final LongMemberValue mb)
    {
        mb.setValue((Long) this.value);
    }

    public void visitShortMemberValue(final ShortMemberValue mb)
    {
        mb.setValue((Short) this.value);
    }

    public void visitStringMemberValue(final StringMemberValue mb)
    {
        mb.setValue((String) this.value);
    }

    public void visitClassMemberValue(final ClassMemberValue mb)
    {
        mb.setValue(((Class) this.value).getName());
    }

}