| /******************************************************************************* |
| * Copyright (c) 2007, 2008 IBM Corporation and others. |
| * 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: |
| * IBM Corporation - initial API and implementation |
| *******************************************************************************/ |
| package org.eclipse.wst.xml.core.internal.formatter; |
| |
| import java.io.IOException; |
| |
| import org.eclipse.core.resources.IFile; |
| import org.eclipse.core.runtime.CoreException; |
| import org.eclipse.core.runtime.IProgressMonitor; |
| import org.eclipse.core.runtime.NullProgressMonitor; |
| import org.eclipse.jface.text.DocumentRewriteSession; |
| import org.eclipse.jface.text.DocumentRewriteSessionType; |
| import org.eclipse.jface.text.IDocument; |
| import org.eclipse.jface.text.IDocumentExtension4; |
| import org.eclipse.text.edits.TextEdit; |
| import org.eclipse.wst.sse.core.StructuredModelManager; |
| import org.eclipse.wst.sse.core.internal.format.IStructuredFormatProcessor; |
| import org.eclipse.wst.sse.core.internal.provisional.IStructuredModel; |
| import org.eclipse.wst.sse.core.internal.provisional.text.IStructuredDocument; |
| import org.eclipse.wst.xml.core.internal.Logger; |
| import org.eclipse.wst.xml.core.internal.provisional.document.IDOMNode; |
| import org.w3c.dom.Node; |
| |
| /** |
| * This is a wrapper for the new xml formatter so that it can be still |
| * considered an IStruturedFormatProcessor |
| */ |
| public class XMLFormatterFormatProcessor implements IStructuredFormatProcessor { |
| private DefaultXMLPartitionFormatter fFormatter = null; |
| private IProgressMonitor fProgressMonitor = null; |
| |
| public void formatDocument(IDocument document, int start, int length) throws IOException, CoreException { |
| if (document instanceof IStructuredDocument) { |
| IStructuredModel model = StructuredModelManager.getModelManager().getModelForEdit((IStructuredDocument) document); |
| if (model != null) { |
| try { |
| formatModel(model, start, length); |
| } |
| finally { |
| model.releaseFromEdit(); |
| } |
| } |
| } |
| } |
| |
| public void formatFile(IFile file) throws IOException, CoreException { |
| if (file == null) |
| return; |
| |
| IStructuredModel structuredModel = null; |
| // OutputStream outputStream = null; |
| try { |
| // setup structuredModel |
| // Note: We are getting model for edit. Will save model if model |
| // changed. |
| structuredModel = StructuredModelManager.getModelManager().getModelForEdit(file); |
| |
| // format |
| formatModel(structuredModel); |
| |
| // save model if needed |
| if (!structuredModel.isSharedForEdit() && structuredModel.isSaveNeeded()) |
| structuredModel.save(); |
| } |
| finally { |
| // ensureClosed(outputStream, null); |
| // release from model manager |
| if (structuredModel != null) { |
| structuredModel.releaseFromEdit(); |
| } |
| |
| } |
| } |
| |
| public void formatModel(IStructuredModel structuredModel) { |
| int start = 0; |
| int length = structuredModel.getStructuredDocument().getLength(); |
| |
| formatModel(structuredModel, start, length); |
| } |
| |
| public void formatModel(IStructuredModel structuredModel, int start, int length) { |
| if(fProgressMonitor != null) |
| fProgressMonitor.beginTask("", 2); |
| IStructuredDocument structuredDocument = structuredModel.getStructuredDocument(); |
| DocumentRewriteSession activeRewriteSession = ((IDocumentExtension4) structuredDocument).getActiveRewriteSession(); |
| boolean startedRewriteSession = false; |
| if (activeRewriteSession == null) { |
| activeRewriteSession = ((IDocumentExtension4) structuredDocument).startRewriteSession(DocumentRewriteSessionType.UNRESTRICTED); |
| startedRewriteSession = true; |
| } |
| getFormatter().setProgressMonitor(new NullProgressMonitor() { |
| public boolean isCanceled() { |
| return fProgressMonitor != null && fProgressMonitor.isCanceled(); |
| } |
| }); |
| TextEdit edit = getFormatter().format(structuredModel, start, length); |
| if(fProgressMonitor != null) |
| fProgressMonitor.worked(1); |
| |
| try { |
| structuredModel.aboutToChangeModel(); |
| edit.apply(structuredDocument); |
| if(fProgressMonitor != null) |
| fProgressMonitor.worked(1); |
| } |
| catch (Exception e) { |
| Logger.log(Logger.WARNING_DEBUG, e.getMessage(), e); |
| } |
| finally { |
| if (startedRewriteSession && activeRewriteSession != null) { |
| ((IDocumentExtension4) structuredDocument).stopRewriteSession(activeRewriteSession); |
| } |
| structuredModel.changedModel(); |
| if(fProgressMonitor != null) |
| fProgressMonitor.done(); |
| } |
| } |
| |
| public void formatNode(Node node) { |
| if (node instanceof IDOMNode) { |
| IDOMNode domNode = (IDOMNode) node; |
| formatModel(domNode.getModel(), domNode.getStartOffset(), domNode.getLength()); |
| } |
| } |
| |
| public void setProgressMonitor(IProgressMonitor monitor) { |
| fProgressMonitor = monitor; |
| } |
| |
| private DefaultXMLPartitionFormatter getFormatter() { |
| if (fFormatter == null) { |
| fFormatter = new DefaultXMLPartitionFormatter(); |
| } |
| return fFormatter; |
| } |
| |
| } |