nyla.solutions.global.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.global Show documentation
Nyla Solutions Global Java API provides support for basic application utilities (application configuration, data encryption, debugger and text processing).
The newest version!
package nyla.solutions.global.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 nyla.solutions.global.exception.SystemException;
import nyla.solutions.global.io.IO;

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

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;
		  }

		}

}