org.mycore.common.content.MCRSourceContent Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mycore-base Show documentation
There is a newer version: 2024.05
/*
 * This file is part of ***  M y C o R e  ***
 * See http://www.mycore.de/ for details.
 *
 * MyCoRe is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * MyCoRe 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 Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MyCoRe.  If not, see .
 */

package org.mycore.common.content;

import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;

import javax.xml.transform.Source;
import javax.xml.transform.TransformerException;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.sax.SAXSource;
import javax.xml.transform.stream.StreamSource;

import org.jdom2.Document;
import org.jdom2.Element;
import org.jdom2.transform.JDOMSource;
import org.mycore.common.MCRException;
import org.mycore.common.xml.MCRURIResolver;
import org.w3c.dom.Node;

/**
 * @author Thomas Scheffler (yagee)
 */
public class MCRSourceContent extends MCRWrappedContent {
    private static final MCRURIResolver URI_RESOLVER = MCRURIResolver.instance();

    private Source source;

    public MCRSourceContent(Source source) {
        if (source == null) {
            throw new NullPointerException("Source cannot be null");
        }
        this.source = source;
        MCRContent baseContent = null;
        if (source instanceof JDOMSource) {
            JDOMSource src = (JDOMSource) source;
            Document xml = src.getDocument();
            if (xml != null) {
                baseContent = new MCRJDOMContent(xml);
            } else {
                for (Object node : src.getNodes()) {
                    if (node instanceof Element) {
                        Element element = (Element) node;
                        Document doc = element.getDocument();
                        if (doc == null) {
                            baseContent = new MCRJDOMContent(element);
                        } else {
                            if (doc.getRootElement() == element) {
                                baseContent = new MCRJDOMContent(doc);
                            } else {
                                baseContent = new MCRJDOMContent(element.clone());
                            }
                        }
                        break;
                    } else if (node instanceof Document) {
                        baseContent = new MCRJDOMContent((Document) node);
                        break;
                    }
                }
            }
        } else if (source instanceof SAXSource) {
            SAXSource src = (SAXSource) source;
            baseContent = new MCRSAXContent(src.getXMLReader(), src.getInputSource());
        } else if (source instanceof DOMSource) {
            Node node = ((DOMSource) source).getNode();
            baseContent = new MCRDOMContent(node.getOwnerDocument());
        } else if (source instanceof StreamSource) {
            InputStream inputStream = ((StreamSource) source).getInputStream();
            if (inputStream != null) {
                baseContent = new MCRStreamContent(inputStream);
            } else {
                try {
                    URL url = new URL(source.getSystemId());
                    baseContent = new MCRURLContent(url);
                } catch (MalformedURLException e) {
                    throw new MCRException("Could not create instance of MCRURLContent for SYSTEMID: "
                        + source.getSystemId(), e);
                }
            }
        }
        if (baseContent == null) {
            throw new MCRException("Could not get MCRContent from " + source.getClass().getCanonicalName()
                + ", systemId:" + source.getSystemId());
        }
        baseContent.setSystemId(getSystemId());
        this.setBaseContent(baseContent);
    }

    /**
     * Build instance of MCRSourceContent by resolving via {@link MCRURIResolver}
     * 
     * @throws TransformerException
     *             thrown by {@link MCRURIResolver#resolve(String, String)}
     */
    public static MCRSourceContent getInstance(String uri) throws TransformerException {
        Source source = URI_RESOLVER.resolve(uri, null);
        if (source == null) {
            return null;
        }
        return new MCRSourceContent(source);
    }

    @Override
    public String getSystemId() {
        return source.getSystemId();
    }

    @Override
    public Source getSource() throws IOException {
        return source;
    }
}