eclipsecon08/org.eclipse.dltk.examples.python.part4/src/org/eclipse/dltk/examples/python/internal/search/ExamplePythonMatchLocationParser.java

package org.eclipse.dltk.examples.python.internal.search;

import org.eclipse.dltk.ast.ASTNode;
import org.eclipse.dltk.ast.declarations.FieldDeclaration;
import org.eclipse.dltk.ast.expressions.CallExpression;
import org.eclipse.dltk.ast.expressions.Expression;
import org.eclipse.dltk.ast.references.SimpleReference;
import org.eclipse.dltk.ast.references.VariableReference;
import org.eclipse.dltk.ast.statements.Statement;
import org.eclipse.dltk.core.search.matching.MatchLocator;
import org.eclipse.dltk.core.search.matching.MatchLocatorParser;
import org.eclipse.dltk.core.search.matching.PatternLocator;
import org.eclipse.dltk.python.parser.ast.expressions.Assignment;
import org.eclipse.dltk.python.parser.ast.expressions.ExtendedVariableReference;

public class ExamplePythonMatchLocationParser extends MatchLocatorParser {
	protected ExamplePythonMatchLocationParser(MatchLocator locator) {
		super(locator);
	}

	protected void processStatement(ASTNode node, PatternLocator locator) {
		super.processStatement(node, locator);
		if (node instanceof ExtendedVariableReference) {
			ExtendedVariableReference ref = (ExtendedVariableReference) node;
			int expressionCount = ref.getExpressionCount();
			for (int i = 0; i < expressionCount; i++) {
				Expression e = ref.getExpression(i);
				if (ref.isCall(i) && e instanceof VariableReference) {
					CallExpression call = new CallExpression(null,
							((VariableReference) e).getName(), null);
					call.setStart(e.sourceStart());
					call.setEnd(e.sourceEnd());
					locator.match(call, getNodeSet());
				} else if (e instanceof VariableReference) {
					locator.match(node, getNodeSet());
				}
			}
		}
		if (node instanceof VariableReference) {
			locator.match((VariableReference) node, getNodeSet());
		}
		if (node instanceof Assignment) {
			Assignment as = (Assignment) node;
			Statement varref = as.getLeft();
			if (varref instanceof VariableReference) {
				VariableReference vref = (VariableReference) varref;
				locator.match(
						new FieldDeclaration(vref.getName(),
								vref.sourceStart(), vref.sourceEnd(), vref
										.sourceStart(), vref.sourceEnd()),
						getNodeSet());
			}
			if (varref instanceof ExtendedVariableReference) {
				ExtendedVariableReference extvar = (ExtendedVariableReference) varref;
				if (extvar.getExpressionCount() == 2) {
					VariableReference vref = (VariableReference) extvar
							.getExpression(1);
					locator.match(
							new FieldDeclaration(vref.getName(), vref
									.sourceStart(), vref.sourceEnd(), vref
									.sourceStart(), vref.sourceEnd()),
							getNodeSet());
				}
			}
		}
	}
}