io.lightlink.excel.CopyingTemplateHandler.java Source code

Introduction

Here is the source code for io.lightlink.excel.CopyingTemplateHandler.java

Source

package io.lightlink.excel;

/*
 * #%L
 * LightLink Core
 * %%
 * Copyright (C) 2015 - 2016 Vitaliy Shevchuk
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/lgpl-3.0.html>.
 * #L%
 */

import org.apache.commons.lang.StringEscapeUtils;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;

import java.io.IOException;
import java.io.Writer;
import java.util.Map;

public class CopyingTemplateHandler extends DefaultHandler {
    protected Writer out;
    StringBuffer textBuffer = new StringBuffer();

    public CopyingTemplateHandler(Writer out) {
        this.out = out;
    }

    public void startDocument() throws SAXException {
        emit("<?xml version='1.0' encoding='UTF-8'?>");
    }

    public void endDocument() throws SAXException {

        try {
            out.flush();
        } catch (IOException e) {
            throw new SAXException("I/O error", e);
        }
    }

    public void characters(char[] buf, int offset, int len) throws SAXException {
        String s = new String(buf, offset, len);

        if (textBuffer == null) {
            textBuffer = new StringBuffer(s);
        } else {
            textBuffer.append(s);
        }
    }

    protected void emit(String s) {
        try {
            out.write(s);
            out.flush();
        } catch (IOException e) {
            throw new RuntimeException(e.toString(), e);
        }
    }

    public void startElement(String namespaceURI, String sName, // simple name
            String qName, // qualified name
            Attributes attrs) throws SAXException {
        echoText();

        String eName = sName; // element name

        if ("".equals(eName)) {
            eName = qName; // not namespaceAware
        }

        printElementStart(eName, attrs);

    }

    public void endElement(String namespaceURI, String sName, // simple name
            String qName // qualified name
    ) throws SAXException {

        String eName = sName; // element name

        if ("".equals(eName)) {
            eName = qName; // not namespaceAware
        }

        echoText();
        emit("</" + eName + ">\n");

        textBuffer.setLength(0);

    }

    protected void printElementStart(String eName, Attributes attrs) throws SAXException {
        emit("\n<" + eName);

        if (attrs != null) {
            for (int i = 0; i < attrs.getLength(); i++) {
                String aName = attrs.getLocalName(i); // Attr name

                if ("".equals(aName)) {
                    aName = attrs.getQName(i);
                }

                emit(" " + aName + "=\"" + attrs.getValue(i) + "\"");
            }
        }

        emit(">");
    }

    protected void printEmptyElement(String eName, Map<String, Object> attrs) {
        printElementNameAndAttributes(eName, attrs);
        emit("/>");

    }

    protected void printElementStart(String eName, Map<String, Object> attrs) {
        printElementNameAndAttributes(eName, attrs);
        emit(">");
    }

    protected void printElementNameAndAttributes(String eName, Map<String, Object> attrs) {
        emit("\n<" + eName);
        for (Map.Entry<String, Object> entry : attrs.entrySet()) {
            String key = entry.getKey();
            Object value = entry.getValue();
            if (!key.startsWith("!"))
                emit(" " + key + "=\"" + value + "\"");
        }
    }

    protected void echoText() throws SAXException {
        String text = textBuffer.toString();
        emit(StringEscapeUtils.escapeXml(text));
        textBuffer.setLength(0);
    }
}