nyla.solutions.xml.DomJsoupHtmlStragety Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nyla.solutions.xml Show documentation
Xml implementation wrapper
The newest version!
package nyla.solutions.xml;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;

import org.jsoup.Jsoup;
import org.w3c.dom.Document;
import org.w3c.dom.Node;

import nyla.solutions.core.exception.SystemException;
import nyla.solutions.core.io.IO;

public class DomJsoupHtmlStragety implements DomStrategy {

	@Override
	public Document toDocument(String aXML) {
		
		return DOMBuilder.jsoup2DOM(Jsoup.parse(aXML));
	}

	@Override
	public Document toDocument(InputStream inputStream) {
		try {
			
			// TODO Auto-generated method stub
			return toDocument(IO.readText(inputStream, true));
		} catch (IOException e) {
			throw new SystemException(e);
		}
	}

	@Override
	public Document toDocument(URL url) {

		try {
			return DOMBuilder.jsoup2DOM(Jsoup.parse(url, timeoutMillis));
		} catch (IOException e) {
			throw new SystemException("Cannot open URL:"+url,e);
		}
	}
	//--------------------------------------
	private int timeoutMillis = 1000*60;
	public final static class DOMBuilder {

		  /**
		   * Restrict instantiation
		   */
		  private DOMBuilder() {}

		   /**
		   * Returns a W3C DOM that exposes the same content as the supplied Jsoup document into a W3C DOM.
		   * @param jsoupDocument The Jsoup document to convert.
		   * @return A W3C Document.
		   */
		  public static Document jsoup2DOM(org.jsoup.nodes.Document jsoupDocument) {
		    
		    Document document = null;
		    
		    try {
		      
		      /* Obtain the document builder for the configured XML parser. */
		      DocumentBuilderFactory docBuilderFactory = DocumentBuilderFactory.newInstance();
		      DocumentBuilder docBuilder = docBuilderFactory.newDocumentBuilder();
		      
		      /* Create a document to contain the content. */
		      document = docBuilder.newDocument();
		      createDOM(jsoupDocument, document, document, new HashMap());
		      
		    } catch (ParserConfigurationException pce) {
		      throw new RuntimeException(pce);
		    }
		    
		    return document;
		  }
		  
		  /**
		   * The internal helper that copies content from the specified Jsoup Node into a W3C {@link Node}.
		   * @param node The Jsoup node containing the content to copy to the specified W3C {@link Node}.
		   * @param out The W3C {@link Node} that receives the DOM content.
		   */
		  private static void createDOM(org.jsoup.nodes.Node node, Node out, Document doc, Map ns) {
		         
		    if (node instanceof org.jsoup.nodes.Document) {
		      
		      org.jsoup.nodes.Document d = ((org.jsoup.nodes.Document) node);
		      for (org.jsoup.nodes.Node n : d.childNodes()) {
		        createDOM(n, out,doc,ns);
		      }
		      
		    } else if (node instanceof org.jsoup.nodes.Element) {
		      
		      org.jsoup.nodes.Element e = ((org.jsoup.nodes.Element) node);
		      org.w3c.dom.Element _e = doc.createElement(e.tagName());
	
		      out.appendChild(_e);
		      org.jsoup.nodes.Attributes atts = e.attributes();
		      
		      for(org.jsoup.nodes.Attribute a : atts){
		        String attName = a.getKey();
		        //omit xhtml namespace
		        if (attName.equals("xmlns")) {
		          continue;
		        }
		        String attPrefix = getNSPrefix(attName);
		        if (attPrefix != null) {
		          if (attPrefix.equals("xmlns")) {
		            ns.put(getLocalName(attName), a.getValue());
		          }
		          else if (!attPrefix.equals("xml")) {
		            String namespace = ns.get(attPrefix);
		            if (namespace == null) {
		              //fix attribute names looking like qnames
		              attName = attName.replace(':','_');
		            }
		          }
		        }
		        _e.setAttribute(attName, a.getValue());
		      }
		      
		      for (org.jsoup.nodes.Node n : e.childNodes()) {
		        createDOM(n, _e, doc,ns);
		      }
		      
		    } else if (node instanceof org.jsoup.nodes.TextNode) {
		      
		      org.jsoup.nodes.TextNode t = ((org.jsoup.nodes.TextNode) node);
		      if (!(out instanceof Document)) {
		        out.appendChild(doc.createTextNode(t.text()));
		      }
		    }
		  }
		  
		  // some hacks for handling namespace in jsoup2DOM conversion
		  private static String getNSPrefix(String name) {
		    if (name != null) {
		      int pos = name.indexOf(':');
		      if (pos > 0) {
		        return name.substring(0,pos);
		      }
		    }
		    return null;
		  }
		  
		  private static String getLocalName(String name) {
		    if (name != null) {
		      int pos = name.lastIndexOf(':');
		      if (pos > 0) {
		        return name.substring(pos+1);
		      }
		    }
		    return name;
		  }

		}

}