/*
    * 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 <http://www.gnu.org/licenses/>.
   */
  
  package org.mycore.common.content;
  
  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;
  
  import javax.xml.transform.Source;
  import javax.xml.transform.Transformer;
  import javax.xml.transform.TransformerException;
  import javax.xml.transform.TransformerFactory;
  import javax.xml.transform.dom.DOMSource;
  import javax.xml.transform.sax.SAXSource;
  import javax.xml.transform.stream.StreamResult;
  import javax.xml.transform.stream.StreamSource;
  import java.io.ByteArrayOutputStream;
  import java.io.IOException;
  import java.io.InputStream;
  import java.net.MalformedURLException;
  import java.net.URL;
  
  import jakarta.xml.bind.util.JAXBSource;
  
  /**
   * @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 JAXBSource) {
              TransformerFactory transformerFactory = TransformerFactory.newDefaultInstance();
              try {
                  Transformer transformer = transformerFactory.newTransformer();
                  ByteArrayOutputStream bout = new ByteArrayOutputStream();
                  transformer.transform(source, new StreamResult(bout));
                  baseContent = new MCRByteContent(bout.toByteArray());
              } catch (TransformerException e) {
                  throw new MCRException("Error while resolving JAXBSource", e);
              }
          } 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;
     }
 }