bundles/org.eclipse.wst.jsdt.web.ui/src/org/eclipse/wst/jsdt/web/ui/actions/JsElementActionProxy.java

/**
 * 
 */
package org.eclipse.wst.jsdt.web.ui.actions;

import java.util.ArrayList;
import java.util.Iterator;

import org.eclipse.jface.action.IAction;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.wst.jsdt.core.IJavaElement;

/**
 * @author childsb
 *
 */
public class JsElementActionProxy extends SimpleJSDTActionProxy {
	
	
	
	public Object[] getRunArgs(IAction action) {
		/* Needs to return an array of IJavaElements.  Since its one arg of type IJavaElement[] need to
		 * put into an object array
		 */
		return new Object[] {getJsElementsFromSelection(getCurrentSelection())};	
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.wst.jsdt.web.ui.actions.SimpleJSDTActionProxy#getRunArgTypes()
	 */
	public Class[] getRunArgTypes() {
		return new Class[] {(new IJavaElement[0] ).getClass()};
	}

	/* Util method to get all the java elements in a selection */
	public static IJavaElement[] getJsElementsFromSelection(ISelection selection) {
		if(selection==null) return new IJavaElement[0];
		ArrayList elements = new ArrayList();
		
		if(selection instanceof IStructuredSelection) {
			Iterator itt = ((IStructuredSelection)selection).iterator();
			while(itt.hasNext()) {
				Object element = itt.next();
				if(element instanceof IJavaElement) {
					elements.add(element);
				}
				if(element instanceof IJavaWebNode) {
					elements.add(  ((IJavaWebNode)element).getJavaElement()        );
				}
			}
			return (IJavaElement[])elements.toArray(new IJavaElement[elements.size()]);
		}
		return new IJavaElement[0];
	}

}