org.articleEditor.articleKit.DocxEditorKit.java Source code

Introduction

Here is the source code for org.articleEditor.articleKit.DocxEditorKit.java

Source

/*
 * Copyright 2013 Japplis.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.articleEditor.articleKit;

import java.io.*;
import java.nio.charset.StandardCharsets;
import javax.swing.text.BadLocationException;
import javax.swing.text.Document;
import javax.swing.text.StyledEditorKit;
import javax.swing.text.ViewFactory;
import org.apache.poi.xwpf.usermodel.XWPFDocument;

import org.articleEditor.insertContent.POIDocxReader;
import org.articleEditor.view.DocxViewFactory;

/**
 * This is the implementation of editing functionality.
 *
 * @author Stanislav Lapitsky
 */
public class DocxEditorKit extends StyledEditorKit {

    /**
     * Create a copy of the editor kit. This allows an implementation to serve as a prototype for others, so that they
     * can be quickly created.
     *
     * @return the copy
     */
    @Override
    public Object clone() {
        return new DocxEditorKit();
    }

    /**
     * Get the MIME type of the data that this kit represents support for. This kit supports the type.
     *
     * @return the type
     */
    @Override
    public String getContentType() {
        return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
    }

    /**
     * Insert content from the given stream which is expected to be in a format appropriate for this kind of content
     * handler.
     *
     * @param in The stream to read from
     * @param doc The destination for the insertion.
     * @param pos The location in the document to place the content.
     * @exception IOException on any I/O error
     * @exception BadLocationException if pos represents an invalid location within the document.
     */
    @Override
    public void read(InputStream in, Document doc, int pos) throws IOException, BadLocationException {
        POIDocxReader db = new POIDocxReader(doc);
        db.read(in, pos);
    }

    /**
     * Insert content from the given stream, which will be treated as plain text.
     *
     * @param in The stream to read from
     * @param doc The destination for the insertion.
     * @param pos The location in the document to place the content.
     * @exception IOException on any I/O error
     * @exception BadLocationException if pos represents an invalid location within the document.
     */
    @Override
    public void read(Reader in, Document doc, int pos) throws IOException, BadLocationException {

        BufferedReader br = new BufferedReader(in);

        String s = br.readLine();
        StringBuilder sb = new StringBuilder();
        while (s != null) {
            sb.append(s).append("\n");
            s = br.readLine();
        }
        // System.out.println(sb.toString());
        read(new ByteArrayInputStream(sb.toString().getBytes(StandardCharsets.UTF_8)), doc, pos);
    }

    /**
     * Write content from a document to the given stream in a format appropriate for this kind of content handler.
     *
     * @param out The stream to write to
     * @param doc The source for the write.
     * @param pos The location in the document to fetch the content.
     * @param len The amount to write out.
     * @exception IOException on any I/O error
     * @exception BadLocationException if pos represents an invalid location within the document.
     */
    @Override
    public void write(OutputStream out, Document doc, int pos, int len) throws IOException, BadLocationException {
        XWPFDocument poiDocument = (XWPFDocument) doc.getProperty("XWPFDocument");
        poiDocument.write(out);
    }

    /**
     * Write content from a document to the given stream as plain text.
     *
     * @param out The stream to write to
     * @param doc The source for the write.
     * @param pos The location in the document to fetch the content.
     * @param len The amount to write out.
     * @exception IOException on any I/O error
     * @exception BadLocationException if pos represents an invalid location within the document.
     */
    @Override
    public void write(Writer out, Document doc, int pos, int len) throws IOException, BadLocationException {

        throw new BadLocationException("Not implemented!", 0);
    }

    /**
     * Write content from a document to the given file as plain text.
     *
     * @param   fileName ?out? The name of source file to write to
     * @param   doc The source for the write.
     * @exception   IOException on any I/O error
     * @exception   BadLocationException if pos represents an invalid location within the document.
     */
    public void write(String fileName, Document doc) throws IOException, BadLocationException {
        XWPFDocument poiDocument = (XWPFDocument) doc.getProperty("XWPFDocument");
        try (FileOutputStream out = new FileOutputStream(fileName)) {
            poiDocument.write(out);
        }
    }

    /**
     * Create an uninitialized text storage model that is appropriate for this type of editor.
     *
     * @return the model
     */
    @Override
    public Document createDefaultDocument() {
        DocxDocument doc = new DocxDocument();
        return doc;
    }

    /**
     * Fetch a factory that is suitable for producing views of any models that are produced by this kit.
     *
     * @return the factory
     */
    @Override
    public ViewFactory getViewFactory() {
        return new DocxViewFactory();
    }

}