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

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

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.jface.text.BadLocationException;
import org.eclipse.jface.text.Document;
import org.eclipse.jface.text.IDocument;
import org.eclipse.text.edits.MalformedTreeException;
import org.eclipse.text.edits.MultiTextEdit;
import org.eclipse.wst.jsdt.core.ICompilationUnit;
import org.eclipse.wst.jsdt.core.IMember;
import org.eclipse.wst.jsdt.web.core.internal.java.DocumentChangeListenerToTextEdit;
import org.eclipse.wst.jsdt.web.core.internal.java.JsTranslation;
import org.eclipse.wst.jsdt.web.ui.views.contentoutline.JsJfaceNode;
import org.eclipse.wst.sse.core.StructuredModelManager;
import org.eclipse.wst.sse.core.internal.provisional.IModelManager;
import org.eclipse.wst.sse.core.internal.provisional.IStructuredModel;
import org.eclipse.wst.sse.core.internal.provisional.text.IStructuredDocument;

/**
 * @author childsb
 * 
 */
public class AddJavaDocStubOperation extends org.eclipse.wst.jsdt.internal.corext.codemanipulation.AddJavaDocStubOperation {
	private IDocument copy;
	/**
	 * @param members
	 */
	private JsJfaceNode node;
	private DocumentChangeListenerToTextEdit textEditListener;;
	
	public AddJavaDocStubOperation(IMember[] members, JsJfaceNode node) {
		super(members);
		this.node = node;
	}
	
	protected void applyChanges() {
		IModelManager modelManager = StructuredModelManager.getModelManager();
		IStructuredModel model = null;
		IStructuredDocument doc = node.getStructuredDocument();
		try {
			MultiTextEdit edits = textEditListener.getTextEdits();
			model = modelManager.getExistingModelForEdit(doc);
			model.aboutToChangeModel();
			model.beginRecording(this, "Generate JsDoc", "Generate JsDoc");
			edits.apply(doc);
		} catch (MalformedTreeException ex) {
			// TODO Auto-generated catch block
			ex.printStackTrace();
		} catch (BadLocationException ex) {
			// TODO Auto-generated catch block
			ex.printStackTrace();
		} finally {
			if (model != null) {
				model.endRecording(this);
				model.changedModel();
				model.releaseFromEdit();
			}
		}
	}
	
	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.wst.jsdt.internal.corext.codemanipulation.AddJavaDocStubOperation#getDocument(org.eclipse.wst.jsdt.core.ICompilationUnit,
	 *      org.eclipse.core.runtime.IProgressMonitor)
	 */
	
	protected IDocument getDocument(ICompilationUnit cu, IProgressMonitor monitor) throws CoreException {
		return getJavaDocumentFromNode();
	}
	
	protected IDocument getJavaDocumentFromNode() {
		if (copy == null) {
			JsTranslation tran = node.getTranslation();
			copy = new Document(tran.getJsText());
			textEditListener = new DocumentChangeListenerToTextEdit();
			copy.addDocumentListener(textEditListener);
		}
		return copy;
	}
	
	
	public void run(IProgressMonitor monitor) throws CoreException, OperationCanceledException {
		super.run(monitor);
		applyChanges();
		/* need to apply the text edits back to the original doc */
	}
}