com.enonic.cms.core.xslt.XsltProcessorHelper.java Source code

Introduction

Here is the source code for com.enonic.cms.core.xslt.XsltProcessorHelper.java

Source

/*
 * Copyright 2000-2011 Enonic AS
 * http://www.enonic.com/license
 */
package com.enonic.cms.core.xslt;

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

import javax.servlet.http.HttpServletResponse;
import javax.xml.transform.Result;
import javax.xml.transform.Source;
import javax.xml.transform.URIResolver;
import javax.xml.transform.dom.DOMResult;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.jdom.transform.JDOMResult;
import org.jdom.transform.JDOMSource;

import com.google.common.io.Closeables;

public final class XsltProcessorHelper {
    private final XsltProcessorManager manager;
    private XsltProcessor processor;
    private Source input;

    public XsltProcessorHelper() {
        this(XsltProcessorManagerAccessor.getProcessorManager());
    }

    public XsltProcessorHelper(final XsltProcessorManager manager) {
        this.manager = manager;
    }

    public XsltProcessorHelper stylesheet(final Source xsl, final URIResolver resolver) {
        this.processor = this.manager.createProcessor(xsl, resolver);
        return this;
    }

    public XsltProcessorHelper input(final Source input) {
        this.input = input;
        return this;
    }

    public XsltProcessorHelper input(final org.jdom.Document input) {
        return input(new JDOMSource(input));
    }

    public XsltProcessorHelper input(final org.w3c.dom.Document input) {
        return input(new DOMSource(input));
    }

    public XsltProcessorHelper param(final String key, final Object value) {
        return param(key, value, false);
    }

    public XsltProcessorHelper param(final String key, final Object value, boolean convertToString) {
        this.processor.setParameter(key, convertToString ? value.toString() : value);
        return this;
    }

    public XsltProcessorHelper params(final Map<?, ?> map) {
        return params(map, false);
    }

    public XsltProcessorHelper params(final Map<?, ?> map, boolean convertToString) {
        if (map != null) {
            for (final Object key : map.keySet()) {
                param((String) key, map.get(key), convertToString);
            }
        }

        return this;
    }

    public String process() {
        final StringWriter out = new StringWriter();

        try {
            process(out);
        } finally {
            Closeables.closeQuietly(out);
        }

        return out.toString();
    }

    public org.w3c.dom.Document processDom() {
        final DOMResult result = new DOMResult();
        process(result);
        return (org.w3c.dom.Document) result.getNode();
    }

    public org.jdom.Document processJDom() {
        final JDOMResult result = new JDOMResult();
        process(result);
        return result.getDocument();
    }

    public void process(final Result result) {
        final XsltProcessor processor = processor();
        processor.process(this.input, result);
    }

    public void process(final Writer writer) {
        final StreamResult result = new StreamResult(writer);
        process(result);
    }

    public void process(final HttpServletResponse response) {
        try {
            response.setContentType(this.processor.getContentType());
            process(response.getWriter());
        } catch (IOException e) {
            throw new XsltProcessorException(e);
        }
    }

    public XsltProcessor processor() {
        return this.processor;
    }
}