bundles/org.eclipse.wst.jsdt.web.ui/src/org/eclipse/wst/jsdt/web/ui/views/contentoutline/JFaceNodeAdapterForJs.java

package org.eclipse.wst.jsdt.web.ui.views.contentoutline;

import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.Position;
import org.eclipse.swt.graphics.Image;
import org.eclipse.wst.html.ui.internal.contentoutline.JFaceNodeAdapterForHTML;
import org.eclipse.wst.jsdt.core.IJavaElement;
import org.eclipse.wst.jsdt.core.JavaModelException;
import org.eclipse.wst.jsdt.internal.core.SourceRefElement;
import org.eclipse.wst.jsdt.ui.JavaElementLabelProvider;
import org.eclipse.wst.jsdt.ui.StandardJavaElementContentProvider;
import org.eclipse.wst.jsdt.web.core.internal.Logger;
import org.eclipse.wst.jsdt.web.core.internal.java.IJSPTranslation;
import org.eclipse.wst.jsdt.web.core.internal.java.JSPTranslation;
import org.eclipse.wst.jsdt.web.core.internal.java.JSPTranslationAdapter;
import org.eclipse.wst.sse.core.StructuredModelManager;
import org.eclipse.wst.sse.core.internal.model.FactoryRegistry;
import org.eclipse.wst.sse.core.internal.provisional.IModelManager;
import org.eclipse.wst.sse.core.internal.provisional.INodeAdapter;
import org.eclipse.wst.sse.core.internal.provisional.IStructuredModel;
import org.eclipse.wst.sse.core.internal.provisional.text.IStructuredDocument;
import org.eclipse.wst.sse.ui.internal.contentoutline.IJFaceNodeAdapter;
import org.eclipse.wst.xml.core.internal.document.NodeImpl;
import org.eclipse.wst.xml.core.internal.provisional.document.IDOMDocument;
import org.eclipse.wst.xml.core.internal.provisional.document.IDOMModel;
import org.eclipse.wst.xml.ui.internal.contentoutline.JFaceNodeAdapterFactory;
import org.w3c.dom.Node;

public class JFaceNodeAdapterForJs extends JFaceNodeAdapterForHTML {
	public JFaceNodeAdapterForJs(JFaceNodeAdapterFactory adapterFactory) {
		super(adapterFactory);
	}
	
	public Object[] getChildren(Object object) {
		if (object instanceof IJavaElement) return getJavaElementProvider().getChildren(object);
		Node node = (Node) object;
		if (isJSElementParent(node)) {
			Object[] results = getJSElementsFromNode(node.getFirstChild());
			return results;
		}
		return super.getChildren(object);
	}
	
	public Object[] getElements(Object object) {
		if (object instanceof IJavaElement) return getJavaElementProvider().getElements(object);
		return super.getElements(object);
	}
	
	public String getLabelText(Object node) {
		if (node instanceof JsJfaceNode) return getJavaElementLabelProvider().getText(((JsJfaceNode) node).getJsElement());
		if (node instanceof IJavaElement) return getJavaElementLabelProvider().getText((IJavaElement) node);
		return super.getLabelText(node);
	}
	
	public Image getLabelImage(Object node) {
		if (node instanceof JsJfaceNode) return getJavaElementLabelProvider().getImage(((JsJfaceNode) node).getJsElement());
		if (node instanceof IJavaElement) return getJavaElementLabelProvider().getImage((IJavaElement) node);
		return super.getLabelImage(node);
	}
	
	public Object getParent(Object element) {
		if (element instanceof IJavaElement) return getJavaElementProvider().getParent(element);
		return super.getParent(element);
	}
	
	public boolean hasChildren(Object object) {
		if (object instanceof IJavaElement) return getJavaElementProvider().hasChildren(object);
		Node node = (Node) object;
		if (isJSElementParent(node)) {
			Object[] nodes = getJSElementsFromNode(node.getFirstChild());
			boolean hasElements = (nodes != null && nodes.length > 0);
			return hasElements;
		}
		return super.hasChildren(object);
	}
	
	private boolean isJSElementParent(Node node) {
		return (node.hasChildNodes() && node.getNodeName().equalsIgnoreCase("script"));
	}
	
	private boolean isJSElement(Object object) {
		if (object instanceof IJavaElement) return true;
		Node node = (Node) object;
		Node parent = node.getParentNode();
		if (parent != null && parent.getNodeName().equalsIgnoreCase("script") && node.getNodeType() == Node.TEXT_NODE) {
			return true;
		}
		return false;
	}
	
	private Object[] getJSElementsFromNode(Node node) {
		
			IJavaElement[] result = getJavaElementsInRange(node);
			if (result == null) return null;
			Object[] newResults = new Object[result.length];
			for (int i = 0; i < result.length; i++) {
				int htmllength = 0;
				int htmloffset = 0;
				Position position = null;
				try {
					htmllength = ((SourceRefElement) (result[i])).getSourceRange().getLength();
					htmloffset = getTranslation(node).getJspOffset(((SourceRefElement) (result[i])).getSourceRange().getOffset());
					position = new Position(htmloffset, htmllength);
				} catch (JavaModelException e) {
					e.printStackTrace();
				}
				newResults[i] = getJsNode(node.getParentNode(), (IJavaElement) result[i], position);
			}
			return newResults;
		
	}
	
	private JSPTranslation getTranslation(Node node) {
	
			IStructuredModel model = null;
			IModelManager modelManager = StructuredModelManager.getModelManager();
			IDOMDocument xmlDoc = null;
			try {
				if (modelManager != null) {
					IStructuredDocument doc = ((NodeImpl) node).getStructuredDocument();
					model = modelManager.getExistingModelForRead(doc);
					
					// model = modelManager.getModelForRead(doc);
				}
				IDOMModel domModel = (IDOMModel) model;
				xmlDoc = domModel.getDocument();
			} catch (Exception e) {
				Logger.logException(e);
			} finally {
				if (model != null) {
					// model.changedModel();
					model.releaseFromRead();
				}
			}
			
			if (xmlDoc == null) return null;
			JSPTranslationAdapter translationAdapter = (JSPTranslationAdapter)xmlDoc.getAdapterFor(IJSPTranslation.class);
			return translationAdapter.getJSPTranslation();
	}
	
	private IJavaElement[] getJavaElementsInRange(Node node) {
		int startOffset = 0;
		int endOffset = 0;
		int type = node.getNodeType();
		IJavaElement[] result = null;
		if (node.getNodeType() == Node.TEXT_NODE && (node instanceof NodeImpl)) {
			startOffset = ((NodeImpl) node).getStartOffset();
			endOffset = ((NodeImpl) node).getEndOffset();
			result = getTranslation(node).getAllElementsInJsRange(startOffset, endOffset);
		}
		return result;
	}
	
	private Object getJsNode(Node parent, IJavaElement root, Position position) {
		JsJfaceNode instance = new JsJfaceNode(parent, root, position);
		// ((JsJfaceNode)instance).setAdapterRegistry(registry);
		INodeAdapter adapter = ((JsJfaceNode) instance).getAdapterFor(IJFaceNodeAdapter.class);
		if (!(adapter instanceof JFaceNodeAdapterForJs)) {
			((JsJfaceNode) instance).removeAdapter(adapter);
			((JsJfaceNode) instance).addAdapter(this);
		}
		return instance;
	}
	
	private StandardJavaElementContentProvider getJavaElementProvider() {
		return new StandardJavaElementContentProvider(true);
	}
	
	private JavaElementLabelProvider getJavaElementLabelProvider() {
		return new JavaElementLabelProvider();
	}
}