[jboss-cvs] Picketlink SVN: r538 - federation/trunk/picketlink-fed-core/src/main/java/org/picketlink/identity/federation/core/util.

[jboss-cvs-commits at lists.jboss.org]

Author: anil.saldhana at jboss.com
Date: 2010-11-08 12:06:33 -0500 (Mon, 08 Nov 2010)
New Revision: 538

Modified:
   federation/trunk/picketlink-fed-core/src/main/java/org/picketlink/identity/federation/core/util/StaxUtil.java
Log:
add stax writing dom element

Modified: federation/trunk/picketlink-fed-core/src/main/java/org/picketlink/identity/federation/core/util/StaxUtil.java
===================================================================
--- federation/trunk/picketlink-fed-core/src/main/java/org/picketlink/identity/federation/core/util/StaxUtil.java	2010-11-08 15:03:15 UTC (rev 537)
+++ federation/trunk/picketlink-fed-core/src/main/java/org/picketlink/identity/federation/core/util/StaxUtil.java	2010-11-08 17:06:33 UTC (rev 538)
@@ -22,6 +22,7 @@
 package org.picketlink.identity.federation.core.util;
 
 import java.io.OutputStream;
+import java.util.Stack;
 
 import javax.xml.namespace.QName;
 import javax.xml.stream.XMLEventWriter;
@@ -30,6 +31,11 @@
 import javax.xml.stream.XMLStreamWriter;
 
 import org.picketlink.identity.federation.core.exceptions.ProcessingException;
+import org.w3c.dom.Attr;
+import org.w3c.dom.DOMException;
+import org.w3c.dom.Element;
+import org.w3c.dom.NamedNodeMap;
+import org.w3c.dom.Node;
 
 /**
  * Utility class that deals with StAX
@@ -38,6 +44,8 @@
  */
 public class StaxUtil
 { 
+   private static ThreadLocal<Stack<String>> registeredNSStack = new ThreadLocal<Stack<String>>();
+   
    /**
     * Flush the stream writer
     * @param writer
@@ -206,6 +214,122 @@
       }
    }
    
+   /**
+    * Write a DOM Node to the stream
+    * @param writer
+    * @param node
+    * @throws ProcessingException
+    */
+   public static void writeDOMNode( XMLStreamWriter writer, Node node ) throws ProcessingException
+   {
+      try
+      {
+         short nodeType = node.getNodeType();
+         
+         switch( nodeType ) 
+         {
+            case Node.ELEMENT_NODE:
+               writeDOMElement( writer, (Element) node);
+               break;
+            case Node.TEXT_NODE: 
+               writer.writeCharacters(node.getNodeValue());
+               break;
+            case Node.COMMENT_NODE:
+               writer.writeComment(node.getNodeValue());
+               break;  
+            case Node.CDATA_SECTION_NODE:
+               writer.writeCData(node.getNodeValue());
+               break; 
+            default: 
+               //Don't care
+         }
+      }
+      catch (DOMException e)
+      {
+         throw new ProcessingException( e );
+      }
+      catch (XMLStreamException e)
+      {
+         throw new ProcessingException( e );
+      }      
+   }
+
+   
+   /**
+    * Write DOM Element to the stream
+    * @param writer
+    * @param domElement
+    * @throws ProcessingException
+    */
+   public static void writeDOMElement( XMLStreamWriter writer, Element domElement ) throws ProcessingException
+   {
+      if( registeredNSStack.get() == null )
+      {
+         registeredNSStack.set( new Stack<String>() );
+      }
+      String domElementPrefix = domElement.getPrefix();
+      
+      if (domElementPrefix == null) 
+      {
+          domElementPrefix = "";
+      }
+      
+      String domElementNS = domElement.getNamespaceURI();
+      if (domElementNS == null) 
+      {
+          domElementNS = "";
+      }
+      
+      writeStartElement(writer, domElementPrefix, domElement.getLocalName(), domElementNS);
+
+      
+      //Should we register namespace
+      if( domElementPrefix != "" && !registeredNSStack.get().contains(domElementNS) )
+      {
+         writeNameSpace(writer, domElementPrefix, domElementNS ); 
+         registeredNSStack.get().push( domElementNS );
+      }
+
+      // Deal with Attributes
+      NamedNodeMap attrs = domElement.getAttributes();
+      for (int i = 0, len = attrs.getLength(); i < len; ++i) 
+      {
+          Attr attr = (Attr) attrs.item(i);
+          String attributePrefix = attr.getPrefix();
+          String attribLocalName = attr.getLocalName();
+          String attribValue = attr.getValue();
+
+          if (attributePrefix == null || attributePrefix.length() == 0) 
+          { 
+             if ( "xmlns".equals( attribLocalName )) 
+              {
+                 writeDefaultNameSpace( writer, attribValue );
+              } 
+              else 
+              {
+                 writeAttribute( writer, attribLocalName, attribValue );
+              }
+          } 
+          else 
+          {
+              if ( "xmlns".equals( attributePrefix )) 
+              {
+                 writeNameSpace( writer, attribLocalName, attribValue); 
+              } 
+              else 
+              {
+                 writeAttribute( writer, new QName( attr.getNamespaceURI(), attribLocalName, attributePrefix ), attribValue);
+              }
+          }
+      }
+
+      for ( Node child = domElement.getFirstChild(); child != null; child = child.getNextSibling() ) 
+      {
+          writeDOMNode( writer, child);
+      }
+
+      writeEndElement(writer);
+   }
     
    
    /**