michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: /*
michael@0:  * A class for managing namespace IDs and mapping back and forth
michael@0:  * between namespace IDs and namespace URIs.
michael@0:  */
michael@0: 
michael@0: #include "nsNameSpaceManager.h"
michael@0: 
michael@0: #include "nscore.h"
michael@0: #include "nsAutoPtr.h"
michael@0: #include "nsINodeInfo.h"
michael@0: #include "nsCOMArray.h"
michael@0: #include "nsContentCreatorFunctions.h"
michael@0: #include "nsString.h"
michael@0: #include "nsINodeInfo.h"
michael@0: #include "mozilla/ClearOnShutdown.h"
michael@0: #include "mozilla/dom/XBLChildrenElement.h"
michael@0: #include "mozilla/dom/Element.h"
michael@0: 
michael@0: using namespace mozilla;
michael@0: using namespace mozilla::dom;
michael@0: 
michael@0: #define kXMLNSNameSpaceURI "http://www.w3.org/2000/xmlns/"
michael@0: #define kXMLNameSpaceURI "http://www.w3.org/XML/1998/namespace"
michael@0: #define kXHTMLNameSpaceURI "http://www.w3.org/1999/xhtml"
michael@0: #define kXLinkNameSpaceURI "http://www.w3.org/1999/xlink"
michael@0: #define kXSLTNameSpaceURI "http://www.w3.org/1999/XSL/Transform"
michael@0: #define kXBLNameSpaceURI "http://www.mozilla.org/xbl"
michael@0: #define kMathMLNameSpaceURI "http://www.w3.org/1998/Math/MathML"
michael@0: #define kRDFNameSpaceURI "http://www.w3.org/1999/02/22-rdf-syntax-ns#"
michael@0: #define kXULNameSpaceURI "http://www.mozilla.org/keymaster/gatekeeper/there.is.only.xul"
michael@0: #define kSVGNameSpaceURI "http://www.w3.org/2000/svg"
michael@0: 
michael@0: StaticAutoPtr<nsNameSpaceManager> nsNameSpaceManager::sInstance;
michael@0: 
michael@0: /* static */ nsNameSpaceManager*
michael@0: nsNameSpaceManager::GetInstance() {
michael@0:   if (!sInstance) {
michael@0:     sInstance = new nsNameSpaceManager();
michael@0:     if (sInstance->Init()) {
michael@0:       ClearOnShutdown(&sInstance);
michael@0:     } else {
michael@0:       delete sInstance;
michael@0:       sInstance = nullptr;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   return sInstance;
michael@0: }
michael@0: 
michael@0: bool nsNameSpaceManager::Init()
michael@0: {
michael@0:   nsresult rv;
michael@0: #define REGISTER_NAMESPACE(uri, id) \
michael@0:   rv = AddNameSpace(NS_LITERAL_STRING(uri), id); \
michael@0:   NS_ENSURE_SUCCESS(rv, false)
michael@0: 
michael@0:   // Need to be ordered according to ID.
michael@0:   REGISTER_NAMESPACE(kXMLNSNameSpaceURI, kNameSpaceID_XMLNS);
michael@0:   REGISTER_NAMESPACE(kXMLNameSpaceURI, kNameSpaceID_XML);
michael@0:   REGISTER_NAMESPACE(kXHTMLNameSpaceURI, kNameSpaceID_XHTML);
michael@0:   REGISTER_NAMESPACE(kXLinkNameSpaceURI, kNameSpaceID_XLink);
michael@0:   REGISTER_NAMESPACE(kXSLTNameSpaceURI, kNameSpaceID_XSLT);
michael@0:   REGISTER_NAMESPACE(kXBLNameSpaceURI, kNameSpaceID_XBL);
michael@0:   REGISTER_NAMESPACE(kMathMLNameSpaceURI, kNameSpaceID_MathML);
michael@0:   REGISTER_NAMESPACE(kRDFNameSpaceURI, kNameSpaceID_RDF);
michael@0:   REGISTER_NAMESPACE(kXULNameSpaceURI, kNameSpaceID_XUL);
michael@0:   REGISTER_NAMESPACE(kSVGNameSpaceURI, kNameSpaceID_SVG);
michael@0: 
michael@0: #undef REGISTER_NAMESPACE
michael@0: 
michael@0:   return true;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: nsNameSpaceManager::RegisterNameSpace(const nsAString& aURI,
michael@0:                                       int32_t& aNameSpaceID)
michael@0: {
michael@0:   if (aURI.IsEmpty()) {
michael@0:     aNameSpaceID = kNameSpaceID_None; // xmlns="", see bug 75700 for details
michael@0: 
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   nsresult rv = NS_OK;
michael@0:   if (!mURIToIDTable.Get(&aURI, &aNameSpaceID)) {
michael@0:     aNameSpaceID = mURIArray.Length() + 1; // id is index + 1
michael@0: 
michael@0:     rv = AddNameSpace(aURI, aNameSpaceID);
michael@0:     if (NS_FAILED(rv)) {
michael@0:       aNameSpaceID = kNameSpaceID_Unknown;
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   NS_POSTCONDITION(aNameSpaceID >= -1, "Bogus namespace ID");
michael@0:   
michael@0:   return rv;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: nsNameSpaceManager::GetNameSpaceURI(int32_t aNameSpaceID, nsAString& aURI)
michael@0: {
michael@0:   NS_PRECONDITION(aNameSpaceID >= 0, "Bogus namespace ID");
michael@0:   
michael@0:   int32_t index = aNameSpaceID - 1; // id is index + 1
michael@0:   if (index < 0 || index >= int32_t(mURIArray.Length())) {
michael@0:     aURI.Truncate();
michael@0: 
michael@0:     return NS_ERROR_ILLEGAL_VALUE;
michael@0:   }
michael@0: 
michael@0:   aURI = *mURIArray.ElementAt(index);
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: int32_t
michael@0: nsNameSpaceManager::GetNameSpaceID(const nsAString& aURI)
michael@0: {
michael@0:   if (aURI.IsEmpty()) {
michael@0:     return kNameSpaceID_None; // xmlns="", see bug 75700 for details
michael@0:   }
michael@0: 
michael@0:   int32_t nameSpaceID;
michael@0: 
michael@0:   if (mURIToIDTable.Get(&aURI, &nameSpaceID)) {
michael@0:     NS_POSTCONDITION(nameSpaceID >= 0, "Bogus namespace ID");
michael@0:     return nameSpaceID;
michael@0:   }
michael@0: 
michael@0:   return kNameSpaceID_Unknown;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: NS_NewElement(Element** aResult,
michael@0:               already_AddRefed<nsINodeInfo>&& aNodeInfo,
michael@0:               FromParser aFromParser)
michael@0: {
michael@0:   nsCOMPtr<nsINodeInfo> ni = aNodeInfo;
michael@0:   int32_t ns = ni->NamespaceID();
michael@0:   if (ns == kNameSpaceID_XHTML) {
michael@0:     return NS_NewHTMLElement(aResult, ni.forget(), aFromParser);
michael@0:   }
michael@0: #ifdef MOZ_XUL
michael@0:   if (ns == kNameSpaceID_XUL) {
michael@0:     return NS_NewXULElement(aResult, ni.forget());
michael@0:   }
michael@0: #endif
michael@0:   if (ns == kNameSpaceID_MathML) {
michael@0:     return NS_NewMathMLElement(aResult, ni.forget());
michael@0:   }
michael@0:   if (ns == kNameSpaceID_SVG) {
michael@0:     return NS_NewSVGElement(aResult, ni.forget(), aFromParser);
michael@0:   }
michael@0:   if (ns == kNameSpaceID_XBL && ni->Equals(nsGkAtoms::children)) {
michael@0:     NS_ADDREF(*aResult = new XBLChildrenElement(ni.forget()));
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   return NS_NewXMLElement(aResult, ni.forget());
michael@0: }
michael@0: 
michael@0: bool
michael@0: nsNameSpaceManager::HasElementCreator(int32_t aNameSpaceID)
michael@0: {
michael@0:   return aNameSpaceID == kNameSpaceID_XHTML ||
michael@0: #ifdef MOZ_XUL
michael@0:          aNameSpaceID == kNameSpaceID_XUL ||
michael@0: #endif
michael@0:          aNameSpaceID == kNameSpaceID_MathML ||
michael@0:          aNameSpaceID == kNameSpaceID_SVG ||
michael@0:          false;
michael@0: }
michael@0: 
michael@0: nsresult nsNameSpaceManager::AddNameSpace(const nsAString& aURI,
michael@0:                                           const int32_t aNameSpaceID)
michael@0: {
michael@0:   if (aNameSpaceID < 0) {
michael@0:     // We've wrapped...  Can't do anything else here; just bail.
michael@0:     return NS_ERROR_OUT_OF_MEMORY;
michael@0:   }
michael@0:   
michael@0:   NS_ASSERTION(aNameSpaceID - 1 == (int32_t) mURIArray.Length(),
michael@0:                "BAD! AddNameSpace not called in right order!");
michael@0: 
michael@0:   nsString* uri = new nsString(aURI);
michael@0:   if (!uri || !mURIArray.AppendElement(uri)) {
michael@0:     delete uri;
michael@0:     return NS_ERROR_OUT_OF_MEMORY;
michael@0:   }
michael@0: 
michael@0:   mURIToIDTable.Put(uri, aNameSpaceID);
michael@0: 
michael@0:   return NS_OK;
michael@0: }