X-Git-Url: http://www.project-moonshot.org/gitweb/?a=blobdiff_plain;f=xmltooling%2Futil%2FXMLHelper.cpp;h=d0fa6c2efe3fd6bd4b26f5e0659b4115a93d39b7;hb=5cb314df178f78c6fa7b9826c2c5a5298ec7a473;hp=7d61fdc83585121192a2a1f0ffdba1e8aa3df68c;hpb=dd11b998f20c8d80c9e7ecda0d8983610c7c9589;p=shibboleth%2Fcpp-xmltooling.git

diff --git a/xmltooling/util/XMLHelper.cpp b/xmltooling/util/XMLHelper.cpp
index 7d61fdc..d0fa6c2 100644
--- a/xmltooling/util/XMLHelper.cpp
+++ b/xmltooling/util/XMLHelper.cpp
@@ -1,221 +1,293 @@
-/*
- *  Copyright 2001-2006 Internet2
- * 
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/**
- * XMLHelper.cpp
- * 
- * A helper class for working with W3C DOM objects. 
- */
-
-#include "internal.h"
-#include "exceptions.h"
-#include "util/XMLHelper.h"
-#include "util/XMLConstants.h"
-
-#include <xercesc/framework/MemBufFormatTarget.hpp>
-#include <xercesc/util/XMLUniDefs.hpp>
-
-using namespace xmltooling;
-
-static const XMLCh type[]={chLatin_t, chLatin_y, chLatin_p, chLatin_e, chNull };
-    
-bool XMLHelper::hasXSIType(const DOMElement* e)
-{
-    if (e) {
-        if (e->hasAttributeNS(XMLConstants::XSI_NS, type)) {
-            return true;
-        }
-    }
-
-    return false;
-}
-
-QName* XMLHelper::getXSIType(const DOMElement* e)
-{
-    DOMAttr* attribute = e->getAttributeNodeNS(XMLConstants::XSI_NS, type);
-    if (attribute) {
-        const XMLCh* attributeValue = attribute->getTextContent();
-        if (attributeValue && *attributeValue) {
-            int i;
-            if ((i=XMLString::indexOf(attributeValue,chColon))>0) {
-                XMLCh* prefix=new XMLCh[i+1];
-                XMLString::subString(prefix,attributeValue,0,i);
-                prefix[i]=chNull;
-                QName* ret=new QName(e->lookupNamespaceURI(prefix), attributeValue + i + 1, prefix);
-                delete[] prefix;
-                return ret;
-            }
-            else {
-                return new QName(e->lookupNamespaceURI(&chNull), attributeValue);
-            }
-        }
-    }
-
-    return NULL;
-}
-
-DOMAttr* XMLHelper::getIdAttribute(const DOMElement* domElement)
-{
-    if(!domElement->hasAttributes()) {
-        return NULL;
-    }
-    
-    DOMNamedNodeMap* attributes = domElement->getAttributes();
-    DOMAttr* attribute;
-    for(XMLSize_t i = 0; i < attributes->getLength(); i++) {
-        attribute = static_cast<DOMAttr*>(attributes->item(i));
-        if(attribute->isId()) {
-            return attribute;
-        }
-    }
-    
-    return NULL;
-}
-
-QName* XMLHelper::getNodeQName(const DOMNode* domNode)
-{
-    if (domNode)
-        return new QName(domNode->getNamespaceURI(), domNode->getLocalName(), domNode->getPrefix());
-    return NULL; 
-}
-
-QName* XMLHelper::getAttributeValueAsQName(const DOMAttr* attribute)
-{
-    if (!attribute)
-        return NULL;
-    
-    int i;
-    const XMLCh* attributeValue=attribute->getTextContent();
-    if (attributeValue && (i=XMLString::indexOf(attributeValue,chColon))>0) {
-        XMLCh* prefix=new XMLCh[i+1];
-        XMLString::subString(prefix,attributeValue,0,i);
-        prefix[i]=chNull;
-        QName* ret=new QName(attribute->lookupNamespaceURI(prefix), attributeValue + i + 1, prefix);
-        delete[] prefix;
-        return ret;
-    }
-    
-    return new QName(attribute->lookupNamespaceURI(NULL), attributeValue);
-}
-
-DOMElement* XMLHelper::appendChildElement(DOMElement* parentElement, DOMElement* childElement)
-{
-    DOMDocument* parentDocument = parentElement->getOwnerDocument();
-    if (childElement->getOwnerDocument() != parentDocument) {
-        childElement = static_cast<DOMElement*>(parentDocument->importNode(childElement, true));
-    }
-
-    parentElement->appendChild(childElement);
-    return childElement;
-}
-
-const XMLCh* XMLHelper::getTextContent(const DOMElement* e)
-{
-    DOMNode* child=e->getFirstChild();
-    while (child) {
-        if (child->getNodeType()==DOMNode::TEXT_NODE)
-            return child->getNodeValue();
-        child=child->getNextSibling();
-    }
-    return NULL;
-}
-
-DOMElement* XMLHelper::getFirstChildElement(const DOMNode* n)
-{
-    DOMNode* child = n->getFirstChild();
-    while (child && child->getNodeType() != DOMNode::ELEMENT_NODE)
-        child = child->getNextSibling();
-    if (child)
-        return static_cast<DOMElement*>(child);
-    return NULL;
-}    
-
-DOMElement* XMLHelper::getLastChildElement(const DOMNode* n)
-{
-    DOMNode* child = n->getLastChild();
-    while (child && child->getNodeType() != DOMNode::ELEMENT_NODE)
-        child = child->getPreviousSibling();
-    if (child)
-        return static_cast<DOMElement*>(child);
-    return NULL;
-}    
-
-DOMElement* XMLHelper::getFirstChildElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
-{
-    DOMElement* e = getFirstChildElement(n);
-    while (e && !isNodeNamed(e, ns, localName))
-        e = getNextSiblingElement(e);
-    return e;
-}
-
-DOMElement* XMLHelper::getLastChildElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
-{
-    DOMElement* e = getLastChildElement(n);
-    while (e && !isNodeNamed(e, ns, localName))
-        e = getPreviousSiblingElement(e);
-    return e;
-}
-
-DOMElement* XMLHelper::getNextSiblingElement(const DOMNode* n)
-{
-    DOMNode* sib = n->getNextSibling();
-    while (sib && sib->getNodeType() != DOMNode::ELEMENT_NODE)
-        sib = sib->getNextSibling();
-    if (sib)
-        return static_cast<DOMElement*>(sib);
-    return NULL;
-}
-
-DOMElement* XMLHelper::getPreviousSiblingElement(const DOMNode* n)
-{
-    DOMNode* sib = n->getPreviousSibling();
-    while (sib && sib->getNodeType() != DOMNode::ELEMENT_NODE)
-        sib = sib->getPreviousSibling();
-    if (sib)
-        return static_cast<DOMElement*>(sib);
-    return NULL;
-}
-
-DOMElement* XMLHelper::getNextSiblingElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
-{
-    DOMElement* e = getNextSiblingElement(n);
-    while (e && !isNodeNamed(e, ns, localName))
-        e = getNextSiblingElement(e);
-    return e;
-}
-
-DOMElement* XMLHelper::getPreviousSiblingElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
-{
-    DOMElement* e = getPreviousSiblingElement(n);
-    while (e && !isNodeNamed(e, ns, localName))
-        e = getPreviousSiblingElement(e);
-    return e;
-}
-
-void XMLHelper::serialize(const DOMElement* e, std::string& buf)
-{
-    static const XMLCh impltype[] = { chLatin_L, chLatin_S, chNull };
-    static const XMLCh UTF8[]={ chLatin_U, chLatin_T, chLatin_F, chDigit_8, chNull };
-    DOMImplementation* impl=DOMImplementationRegistry::getDOMImplementation(impltype);
-    DOMWriter* serializer=(static_cast<DOMImplementationLS*>(impl))->createDOMWriter();
-    XercesJanitor<DOMWriter> janitor(serializer);
-    serializer->setEncoding(UTF8);
-    MemBufFormatTarget target;
-    if (!serializer->writeNode(&target,*e))
-        throw XMLParserException("unable to serialize XML");
-    buf.erase();
-    buf.append(reinterpret_cast<const char*>(target.getRawBuffer()),target.getLen());
-}
+/*
+ *  Copyright 2001-2007 Internet2
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * XMLHelper.cpp
+ * 
+ * A helper class for working with W3C DOM objects. 
+ */
+
+#include "internal.h"
+#include "exceptions.h"
+#include "util/XMLHelper.h"
+#include "util/XMLConstants.h"
+
+#include <xercesc/framework/MemBufFormatTarget.hpp>
+#include <xercesc/util/XMLUniDefs.hpp>
+
+using namespace xmltooling;
+using std::ostream;
+using std::list;
+
+static const XMLCh type[]={chLatin_t, chLatin_y, chLatin_p, chLatin_e, chNull };
+    
+bool XMLHelper::hasXSIType(const DOMElement* e)
+{
+    if (e) {
+        if (e->hasAttributeNS(xmlconstants::XSI_NS, type)) {
+            return true;
+        }
+    }
+
+    return false;
+}
+
+QName* XMLHelper::getXSIType(const DOMElement* e)
+{
+    DOMAttr* attribute = e->getAttributeNodeNS(xmlconstants::XSI_NS, type);
+    if (attribute) {
+        const XMLCh* attributeValue = attribute->getTextContent();
+        if (attributeValue && *attributeValue) {
+            int i;
+            if ((i=XMLString::indexOf(attributeValue,chColon))>0) {
+                XMLCh* prefix=new XMLCh[i+1];
+                XMLString::subString(prefix,attributeValue,0,i);
+                prefix[i]=chNull;
+                QName* ret=new QName(e->lookupNamespaceURI(prefix), attributeValue + i + 1, prefix);
+                delete[] prefix;
+                return ret;
+            }
+            else {
+                return new QName(e->lookupNamespaceURI(&chNull), attributeValue);
+            }
+        }
+    }
+
+    return NULL;
+}
+
+DOMAttr* XMLHelper::getIdAttribute(const DOMElement* domElement)
+{
+    if(!domElement->hasAttributes()) {
+        return NULL;
+    }
+    
+    DOMNamedNodeMap* attributes = domElement->getAttributes();
+    DOMAttr* attribute;
+    for(XMLSize_t i = 0; i < attributes->getLength(); i++) {
+        attribute = static_cast<DOMAttr*>(attributes->item(i));
+        if(attribute->isId()) {
+            return attribute;
+        }
+    }
+    
+    return NULL;
+}
+
+const XMLObject* XMLHelper::getXMLObjectById(const XMLObject& tree, const XMLCh* id)
+{
+    if (XMLString::equals(id, tree.getXMLID()))
+        return &tree;
+    
+    const XMLObject* ret;
+    const list<XMLObject*>& children = tree.getOrderedChildren();
+    for (list<XMLObject*>::const_iterator i=children.begin(); i!=children.end(); ++i) {
+        if (*i) {
+            ret = getXMLObjectById(*(*i), id);
+            if (ret)
+                return ret;
+        }
+    }
+    
+    return NULL;
+}
+
+QName* XMLHelper::getNodeQName(const DOMNode* domNode)
+{
+    if (domNode)
+        return new QName(domNode->getNamespaceURI(), domNode->getLocalName(), domNode->getPrefix());
+    return NULL; 
+}
+
+QName* XMLHelper::getAttributeValueAsQName(const DOMAttr* attribute)
+{
+    if (!attribute)
+        return NULL;
+    
+    int i;
+    const XMLCh* attributeValue=attribute->getTextContent();
+    if (attributeValue && (i=XMLString::indexOf(attributeValue,chColon))>0) {
+        XMLCh* prefix=new XMLCh[i+1];
+        XMLString::subString(prefix,attributeValue,0,i);
+        prefix[i]=chNull;
+        QName* ret=new QName(attribute->lookupNamespaceURI(prefix), attributeValue + i + 1, prefix);
+        delete[] prefix;
+        return ret;
+    }
+    
+    return new QName(attribute->lookupNamespaceURI(NULL), attributeValue);
+}
+
+DOMElement* XMLHelper::appendChildElement(DOMElement* parentElement, DOMElement* childElement)
+{
+    DOMDocument* parentDocument = parentElement->getOwnerDocument();
+    if (childElement->getOwnerDocument() != parentDocument) {
+        childElement = static_cast<DOMElement*>(parentDocument->importNode(childElement, true));
+    }
+
+    parentElement->appendChild(childElement);
+    return childElement;
+}
+
+const XMLCh* XMLHelper::getTextContent(const DOMElement* e)
+{
+    DOMNode* child=e->getFirstChild();
+    while (child) {
+        if (child->getNodeType()==DOMNode::TEXT_NODE)
+            return child->getNodeValue();
+        child=child->getNextSibling();
+    }
+    return NULL;
+}
+
+DOMElement* XMLHelper::getFirstChildElement(const DOMNode* n, const XMLCh* localName)
+{
+    DOMNode* child = n->getFirstChild();
+    while (child && child->getNodeType() != DOMNode::ELEMENT_NODE)
+        child = child->getNextSibling();
+    if (child && localName) {
+        if (!XMLString::equals(localName,child->getLocalName()))
+            return getNextSiblingElement(child, localName);
+    }
+    return static_cast<DOMElement*>(child);
+}    
+
+DOMElement* XMLHelper::getLastChildElement(const DOMNode* n, const XMLCh* localName)
+{
+    DOMNode* child = n->getLastChild();
+    while (child && child->getNodeType() != DOMNode::ELEMENT_NODE)
+        child = child->getPreviousSibling();
+    if (child && localName) {
+        if (!XMLString::equals(localName,child->getLocalName()))
+            return getPreviousSiblingElement(child, localName);
+    }
+    return static_cast<DOMElement*>(child);
+}    
+
+DOMElement* XMLHelper::getFirstChildElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
+{
+    DOMElement* e = getFirstChildElement(n, localName);
+    while (e && !XMLString::equals(e->getNamespaceURI(),ns))
+        e = getNextSiblingElement(e, localName);
+    return e;
+}
+
+DOMElement* XMLHelper::getLastChildElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
+{
+    DOMElement* e = getLastChildElement(n, localName);
+    while (e && !XMLString::equals(e->getNamespaceURI(),ns))
+        e = getPreviousSiblingElement(e, localName);
+    return e;
+}
+
+DOMElement* XMLHelper::getNextSiblingElement(const DOMNode* n, const XMLCh* localName)
+{
+    DOMNode* sib = n->getNextSibling();
+    while (sib && sib->getNodeType() != DOMNode::ELEMENT_NODE)
+        sib = sib->getNextSibling();
+    if (sib && localName) {
+        if (!XMLString::equals(localName,sib->getLocalName()))
+            return getNextSiblingElement(sib, localName);
+    }   
+    return static_cast<DOMElement*>(sib);
+}
+
+DOMElement* XMLHelper::getPreviousSiblingElement(const DOMNode* n, const XMLCh* localName)
+{
+    DOMNode* sib = n->getPreviousSibling();
+    while (sib && sib->getNodeType() != DOMNode::ELEMENT_NODE)
+        sib = sib->getPreviousSibling();
+    if (sib && localName) {
+        if (!XMLString::equals(localName,sib->getLocalName()))
+            return getPreviousSiblingElement(sib, localName);
+    }   
+    return static_cast<DOMElement*>(sib);
+}
+
+DOMElement* XMLHelper::getNextSiblingElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
+{
+    DOMElement* e = getNextSiblingElement(n, localName);
+    while (e && !XMLString::equals(e->getNamespaceURI(),ns))
+        e = getNextSiblingElement(e, localName);
+    return e;
+}
+
+DOMElement* XMLHelper::getPreviousSiblingElement(const DOMNode* n, const XMLCh* ns, const XMLCh* localName)
+{
+    DOMElement* e = getPreviousSiblingElement(n, localName);
+    while (e && !XMLString::equals(e->getNamespaceURI(),ns))
+        e = getPreviousSiblingElement(e, localName);
+    return e;
+}
+
+void XMLHelper::serialize(const DOMNode* n, std::string& buf)
+{
+    static const XMLCh impltype[] = { chLatin_L, chLatin_S, chNull };
+    static const XMLCh UTF8[]={ chLatin_U, chLatin_T, chLatin_F, chDigit_8, chNull };
+    DOMImplementation* impl=DOMImplementationRegistry::getDOMImplementation(impltype);
+    DOMWriter* serializer=(static_cast<DOMImplementationLS*>(impl))->createDOMWriter();
+    XercesJanitor<DOMWriter> janitor(serializer);
+    serializer->setEncoding(UTF8);
+    MemBufFormatTarget target;
+    if (!serializer->writeNode(&target,*n))
+        throw XMLParserException("unable to serialize XML");
+    buf.erase();
+    buf.append(reinterpret_cast<const char*>(target.getRawBuffer()),target.getLen());
+}
+
+namespace {
+    class StreamFormatTarget : public XMLFormatTarget
+    {
+    public:
+        StreamFormatTarget(std::ostream& out) : m_out(out) {}
+        ~StreamFormatTarget() {}
+
+        void writeChars(const XMLByte *const toWrite, const unsigned int count, XMLFormatter *const formatter) {
+            m_out.write(reinterpret_cast<const char*>(toWrite),count);
+        }
+
+        void flush() {
+            m_out.flush();
+        }
+
+    private:
+        std::ostream& m_out;
+    };
+};
+
+ostream& XMLHelper::serialize(const DOMNode* n, ostream& out)
+{
+    static const XMLCh impltype[] = { chLatin_L, chLatin_S, chNull };
+    static const XMLCh UTF8[]={ chLatin_U, chLatin_T, chLatin_F, chDigit_8, chNull };
+    DOMImplementation* impl=DOMImplementationRegistry::getDOMImplementation(impltype);
+    DOMWriter* serializer=(static_cast<DOMImplementationLS*>(impl))->createDOMWriter();
+    XercesJanitor<DOMWriter> janitor(serializer);
+    serializer->setEncoding(UTF8);
+    StreamFormatTarget target(out);
+    if (!serializer->writeNode(&target,*n))
+        throw XMLParserException("unable to serialize XML");
+    return out;
+}
+
+ostream& xmltooling::operator<<(ostream& ostr, const DOMNode& node)
+{
+    return XMLHelper::serialize(&node, ostr);
+}
+
+ostream& xmltooling::operator<<(ostream& ostr, const XMLObject& obj)
+{
+    return ostr << *(obj.marshall());
+}