org.giavacms.base.common.util.HtmlUtils.java Source code

Introduction

Here is the source code for org.giavacms.base.common.util.HtmlUtils.java

Source

/*
 * Copyright 2013 GiavaCms.org.
 *
 * Licensed under the Eclipse Public License version 1.0, available at
 * http://www.eclipse.org/legal/epl-v10.html
 */
package org.giavacms.base.common.util;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;

import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerConfigurationException;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;

import org.apache.commons.lang.StringEscapeUtils;
import org.jboss.logging.Logger;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.tidy.Tidy;

public class HtmlUtils {

    protected static Logger logger = Logger.getLogger(HtmlUtils.class.getName());

    public static String encode(String toEncode) {
        try {
            return URLEncoder.encode(toEncode, "UTF-8");
        } catch (UnsupportedEncodingException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return "";
    }

    public static String normalizeHtml(String code) {
        if (code == null) {
            code = "";
        }
        Tidy tidy = new Tidy();
        tidy.setXHTML(true);
        tidy.setTidyMark(false);
        tidy.setDocType("omit");
        tidy.setPrintBodyOnly(true);
        tidy.setInputEncoding("UTF-8");
        tidy.setShowErrors(0);
        tidy.setShowWarnings(false);
        tidy.setIndentContent(true);
        InputStream is;
        String content = "";
        OutputStream arg1 = new ByteArrayOutputStream();
        try {
            is = new ByteArrayInputStream(code.getBytes("UTF-8"));

            tidy.parse(is, arg1);
            // logger.info("*****************PRIMA: ");
            // logger.info(arg1.toString());
            // logger.info("*******************DOPO: ");
            // StringEscapeUtils.unescapeHtml(arg0)
            content = StringEscapeUtils.unescapeHtml(arg1.toString());
            content = handleAmpersand(content);
            // logger.info(content);

        } catch (UnsupportedEncodingException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return content;
    }

    public static String prettyHtml(String code) {
        if (code == null) {
            code = "";
        }
        Tidy tidy = new Tidy();
        tidy.setXHTML(true);
        tidy.setTidyMark(false);
        tidy.setDocType("omit");
        tidy.setPrintBodyOnly(true);
        tidy.setInputEncoding("UTF-8");
        tidy.setShowErrors(0);
        tidy.setShowWarnings(false);
        tidy.setIndentContent(true);
        // Convert HTML to DOM
        Document htmlDOM = tidy.parseDOM(new ByteArrayInputStream(code.getBytes()), null);
        Node body = htmlDOM.getElementsByTagName("body").item(0);
        // Pretty Print
        OutputStream out = new ByteArrayOutputStream();
        tidy.pprint(body, out);

        return out.toString();
    }

    public static String prettyXml(String code) {
        if (code == null) {
            code = "";
        }

        try {
            TransformerFactory transformerFactory = TransformerFactory.newInstance();
            transformerFactory.setAttribute("indent-number", 4);
            Transformer transformer;
            transformer = transformerFactory.newTransformer();
            transformer.setOutputProperty(OutputKeys.INDENT, "yes");
            StreamResult result = new StreamResult(new StringWriter());
            StreamSource source = new StreamSource(new StringReader(code));
            transformer.transform(source, result);
            return result.getWriter().toString();
        } catch (TransformerConfigurationException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (TransformerException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        return "";
    }

    private static String handleAmpersand(String content) {
        return (content == null) ? null : content.replaceAll("&", "&");
    }

}