001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041:
042: package org.netbeans.modules.e2e.wsdl.wsdl2java;
043:
044: import java.io.IOException;
045: import java.io.StringWriter;
046: import java.io.Writer;
047: import javax.swing.text.StyledDocument;
048: import org.openide.cookies.EditorCookie;
049: import org.openide.cookies.SaveCookie;
050: import org.openide.filesystems.FileObject;
051: import org.openide.loaders.DataObject;
052: import org.openide.text.IndentEngine;
053:
054: /**
055: *
056: * @author Michal Skvor
057: */
058: public class OutputFileFormatter {
059:
060: private FileObject fileObject;
061: private DataObject dataObject;
062: private StyledDocument styledDocument;
063: private IndentEngine indentEngine;
064: private Writer writer;
065: private StringWriter stringWriter;
066:
067: /** Creates a new instance of OutputFileFormatter */
068: public OutputFileFormatter(FileObject fileObject) throws Exception {
069:
070: // System.err.println(" >>> Generating file - " + fileObject.getPath());
071:
072: this .fileObject = fileObject;
073:
074: dataObject = DataObject.find(fileObject);
075: EditorCookie editorCookie = dataObject
076: .getCookie(EditorCookie.class);
077:
078: styledDocument = editorCookie.openDocument();
079: styledDocument.remove(0, styledDocument.getLength());
080: indentEngine = IndentEngine.find(styledDocument);
081:
082: stringWriter = new StringWriter(4096);
083: writer = indentEngine.createWriter(styledDocument, 0,
084: stringWriter);
085: }
086:
087: /**
088: * Appends to the end of well indented generated file
089: *
090: * @param data string to be written
091: */
092: public void write(String data) {
093: try {
094: writer.write(data);
095: } catch (IOException ex) {
096: java.util.logging.Logger.getLogger("global")
097: .log(java.util.logging.Level.SEVERE,
098: ex.getMessage(), ex);
099: }
100: }
101:
102: /**
103: * Closes file generation and flushes all caches
104: */
105: public void close() {
106: try {
107: writer.flush();
108: writer.close();
109:
110: styledDocument.insertString(0, stringWriter.getBuffer()
111: .toString(), null);
112:
113: SaveCookie save = dataObject.getCookie(SaveCookie.class);
114: if (save != null) {
115: save.save();
116: }
117:
118: } catch (Exception ex) {
119: java.util.logging.Logger.getLogger("global")
120: .log(java.util.logging.Level.SEVERE,
121: ex.getMessage(), ex);
122: }
123:
124: }
125: }
|