AssociationRemover.java

/* This file is part of Green.
 *
 * Copyright (C) 2005 The Research Foundation of State University of New York
 * All Rights Under Copyright Reserved, The Research Foundation of S.U.N.Y.
 * 
 * Green is free software, licensed under the terms of the Eclipse
 * Public License, version 1.0.  The license is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */

package edu.buffalo.cse.green.relationship.association;

import static org.eclipse.jdt.core.dom.ASTNode.FIELD_DECLARATION;

import java.util.AbstractList;
import java.util.ArrayList;
import java.util.List;

import org.eclipse.jdt.core.IField;
import org.eclipse.jdt.core.ILocalVariable;
import org.eclipse.jdt.core.dom.Assignment;
import org.eclipse.jdt.core.dom.Block;
import org.eclipse.jdt.core.dom.BodyDeclaration;
import org.eclipse.jdt.core.dom.ExpressionStatement;
import org.eclipse.jdt.core.dom.FieldDeclaration;
import org.eclipse.jdt.core.dom.IBinding;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.Name;
import org.eclipse.jdt.core.dom.SingleVariableDeclaration;
import org.eclipse.jdt.core.dom.VariableDeclarationFragment;

import edu.buffalo.cse.green.relationships.RelationshipRemover;
import edu.buffalo.cse.green.relationships.DeclarationInfoProvider;

/**
 * @see edu.buffalo.cse.green.relationship.RelationshipRemover
 * 
 * @author bcmartin
 */
public class AssociationRemover extends RelationshipRemover {
	private List<ExpressionStatement> lEXP;

	private List<IField> lFIE;

	private List<ILocalVariable> lPAR;

	private DeclarationInfoProvider _typeInfo;

	/**
	 * @see edu.buffalo.cse.green.relationships.RelationshipRemover#init()
	 */
	protected void init() {
		lEXP = new ArrayList<ExpressionStatement>();
		lFIE = new ArrayList<IField>();
		lPAR = new ArrayList<ILocalVariable>();
	}

	/**
	 * @see edu.buffalo.cse.green.relationships.RelationshipRemover#finish()
	 */
	protected void finish() {
		for (ExpressionStatement exp : lEXP) {
			Block block = (Block) exp.getParent();
			exp.delete();
			MethodDeclaration mdc = (MethodDeclaration) block.getParent();
			processAddInvocations(block);

			for (SingleVariableDeclaration varDec : (AbstractList<SingleVariableDeclaration>) mdc
					.parameters()) {
				IBinding binding = varDec.getName().resolveBinding();
				
				if (binding.getJavaElement().equals(
						lPAR.get(0))) {
					varDec.delete();
					break;
				}
			}
			
			lPAR.remove(0);
		}

		for (IField field : lFIE) {
			for (Object element : _typeInfo.bodyDeclarations()) {
				BodyDeclaration dec = (BodyDeclaration) element;
				
				if (dec.getNodeType() == FIELD_DECLARATION) {
					FieldDeclaration fieldDec = (FieldDeclaration) dec;
					
					List<VariableDeclarationFragment> fragments =
						(AbstractList<VariableDeclarationFragment>)
						fieldDec.fragments();

					for (VariableDeclarationFragment fragment : fragments) {
						IField dField = (IField) fragment.getName()
						.resolveBinding().getJavaElement();

						if (field.equals(dField)) {
							fieldDec.delete();
							break;
						}
					}
				}
			}
		}
	}

	/**
	 * @see edu.buffalo.cse.green.relationships.RelationshipVisitor#process(org.eclipse.jdt.core.dom.TypeDeclaration)
	 */
	protected boolean process(DeclarationInfoProvider node) {
		return true;
	}

	/**
	 * @see org.eclipse.jdt.core.dom.ASTVisitor#visit(org.eclipse.jdt.core.dom.ExpressionStatement)
	 */
	public boolean visit(ExpressionStatement node) {
		if (getMatcher().match(node, getRelationship().getFeatures().get(0))) {
			_typeInfo = getCurrentTypeInfo();
			lEXP.add(node);
			Assignment ass = (Assignment) node.getExpression();
			Name lName = (Name) ass.getLeftHandSide();
			Name rName = (Name) ass.getRightHandSide();

			lFIE.add((IField) lName.resolveBinding().getJavaElement());
			lPAR.add((ILocalVariable) rName.resolveBinding().getJavaElement());
		}

		return true;
	}
}