/*******************************************************************************
 *  Copyright (c) 2012 Google, Inc.
 *  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:
 *  Google, Inc. - initial API and implementation
 *******************************************************************************/
package com.windowtester.eclipse.ui.convert.rule;

import java.util.List;

import org.eclipse.jdt.core.dom.ASTNode;
import org.eclipse.jdt.core.dom.ClassInstanceCreation;
import org.eclipse.jdt.core.dom.SimpleType;
import org.eclipse.jdt.core.dom.Type;

import com.windowtester.eclipse.ui.convert.WTAPIAbstractVisitor;
import com.windowtester.eclipse.ui.convert.WTConvertAPIContext;
import com.windowtester.runtime.swt.locator.SWTWidgetLocator;

/**
 * Replace some SWTWidgetLocator instances with more specific locator instances.</br> For
 * example,
 * 
 * <pre>
 * new SWTWidgetLocator(Shell.class, "Java - Eclipse SDK")
 * </pre>
 * 
 * should be replaced with
 * 
 * <pre>
 * new ShellLocator("Java - Eclipse SDK")
 * </pre>
 */
public class WTReplaceSWTWidgetLocatorRule extends WTAPIAbstractVisitor
{
	private final Class<?> locatorClass;
	private final Class<?> widgetClass;

	public WTReplaceSWTWidgetLocatorRule(Class<?> widgetClass, Class<?> locatorClass) {
		super(null);
		this.widgetClass = widgetClass;
		this.locatorClass = locatorClass;
		WTConvertAPIContext.addCommonType(widgetClass);
	}

	/**
	 * Detect specific SWTWidgetLocator constructor references and replace them with more
	 * specific locator instances.
	 */
	@SuppressWarnings("unchecked")
	public void endVisit(ClassInstanceCreation node) {
		super.endVisit(node);
		String typeName = getNodeType(node);
		if (!SWTWidgetLocator.class.getName().equals(typeName))
			return;
		List<ASTNode> arguments = node.arguments();
		if (arguments.size() != 2)
			return;
		if (!(widgetClass.getName() + ".class").equals(getNodeType(arguments.get(0))))
			return;
		final Type type = node.getType();
		if (!type.isSimpleType())
			return;
		context.addImport(locatorClass.getName(), false);
		context.setTypeName((SimpleType) type, locatorClass.getSimpleName());
		context.remove(arguments, 0);
	}
}