org.aspectj/org.aspectj.ajdt.core/src/main/java/org/aspectj/ajdt/internal/compiler/ast/AddAtAspectJAnnotationsVisitor.java

/* *******************************************************************
 * Copyright (c) 2005 IBM Corporation Ltd
 * All rights reserved. 
 * This program and the accompanying materials are made available 
 * under the terms of the Eclipse Public License v1.0 
 * which accompanies this distribution and is available at 
 * http://www.eclipse.org/legal/epl-v10.html 
 *  
 * Contributors: 
 *     Adrian Colyer  initial implementation 
 * ******************************************************************/
package org.aspectj.ajdt.internal.compiler.ast;

import org.aspectj.org.eclipse.jdt.internal.compiler.ASTVisitor;
import org.aspectj.org.eclipse.jdt.internal.compiler.ast.CompilationUnitDeclaration;
import org.aspectj.org.eclipse.jdt.internal.compiler.ast.MethodDeclaration;
import org.aspectj.org.eclipse.jdt.internal.compiler.ast.TypeDeclaration;
import org.aspectj.org.eclipse.jdt.internal.compiler.lookup.BlockScope;
import org.aspectj.org.eclipse.jdt.internal.compiler.lookup.ClassScope;
import org.aspectj.org.eclipse.jdt.internal.compiler.lookup.CompilationUnitScope;

/**
 * Adds runtime visible annotations to code-style aspect declarations so that the MAP can provide aspect information at runtime.
 * 
 * Done: - AspectDeclaration - AdviceDeclaration - PointcutDeclaration
 * 
 * To Do: - DeclareDeclaration - Inter-Type Declaration
 */
public class AddAtAspectJAnnotationsVisitor extends ASTVisitor {

	private boolean makeReflectable;
	// private CompilationUnitDeclaration unit;

	public AddAtAspectJAnnotationsVisitor(CompilationUnitDeclaration unit, boolean makeReflectable) {
		// this.unit = unit;
		this.makeReflectable= makeReflectable;
	}

	public boolean visit(TypeDeclaration localTypeDeclaration, BlockScope scope) {
		if (localTypeDeclaration instanceof AspectDeclaration) {
			((AspectDeclaration) localTypeDeclaration).addAtAspectJAnnotations();
		}
		return true;
	}

	public boolean visit(TypeDeclaration memberTypeDeclaration, ClassScope scope) {
		if (memberTypeDeclaration instanceof AspectDeclaration) {
			((AspectDeclaration) memberTypeDeclaration).addAtAspectJAnnotations();
		}
		return true;
	}

	public boolean visit(TypeDeclaration typeDeclaration, CompilationUnitScope scope) {
		if (typeDeclaration instanceof AspectDeclaration) {
			((AspectDeclaration) typeDeclaration).addAtAspectJAnnotations();
		}
		return true;
	}

	public boolean visit(MethodDeclaration methodDeclaration, ClassScope scope) {
		if (methodDeclaration instanceof AdviceDeclaration) {
			((AdviceDeclaration) methodDeclaration).addAtAspectJAnnotations();
		} else if (methodDeclaration instanceof PointcutDeclaration) {
			((PointcutDeclaration) methodDeclaration).addAtAspectJAnnotations();
		} else if (methodDeclaration instanceof DeclareDeclaration) {
			((DeclareDeclaration) methodDeclaration).addAtAspectJAnnotations();
		} else if (methodDeclaration instanceof InterTypeDeclaration) {
			if (makeReflectable) {
				((InterTypeDeclaration) methodDeclaration).addAtAspectJAnnotations();
			}
		}
		return false;
	}

}