gecko-dev/dom/base/nsXMLNameSpaceMap.cpp
2015-05-03 15:32:37 -04:00

111 lines
2.7 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
/*
* A class for keeping track of prefix-to-namespace-id mappings
*/
#include "nsXMLNameSpaceMap.h"
#include "nsContentUtils.h"
#include "nsGkAtoms.h"
#include "nsNameSpaceManager.h"
#include "mozilla/dom/NameSpaceConstants.h"
template <>
class nsDefaultComparator <nsNameSpaceEntry, nsIAtom*> {
public:
bool Equals(const nsNameSpaceEntry& aEntry, nsIAtom* const& aPrefix) const {
return aEntry.prefix == aPrefix;
}
};
template <>
class nsDefaultComparator <nsNameSpaceEntry, int32_t> {
public:
bool Equals(const nsNameSpaceEntry& aEntry, const int32_t& aNameSpace) const {
return aEntry.nameSpaceID == aNameSpace;
}
};
/* static */ nsXMLNameSpaceMap*
nsXMLNameSpaceMap::Create(bool aForXML)
{
nsXMLNameSpaceMap *map = new nsXMLNameSpaceMap();
NS_ENSURE_TRUE(map, nullptr);
if (aForXML) {
nsresult rv1 = map->AddPrefix(nsGkAtoms::xmlns,
kNameSpaceID_XMLNS);
nsresult rv2 = map->AddPrefix(nsGkAtoms::xml, kNameSpaceID_XML);
if (NS_FAILED(rv1) || NS_FAILED(rv2)) {
delete map;
map = nullptr;
}
}
return map;
}
nsXMLNameSpaceMap::nsXMLNameSpaceMap()
: mNameSpaces(4)
{
}
nsresult
nsXMLNameSpaceMap::AddPrefix(nsIAtom *aPrefix, int32_t aNameSpaceID)
{
if (!mNameSpaces.Contains(aPrefix) && !mNameSpaces.AppendElement(aPrefix)) {
return NS_ERROR_OUT_OF_MEMORY;
}
mNameSpaces[mNameSpaces.IndexOf(aPrefix)].nameSpaceID = aNameSpaceID;
return NS_OK;
}
nsresult
nsXMLNameSpaceMap::AddPrefix(nsIAtom *aPrefix, nsString &aURI)
{
int32_t id;
nsresult rv = nsContentUtils::NameSpaceManager()->RegisterNameSpace(aURI,
id);
NS_ENSURE_SUCCESS(rv, rv);
return AddPrefix(aPrefix, id);
}
int32_t
nsXMLNameSpaceMap::FindNameSpaceID(nsIAtom *aPrefix) const
{
size_t index = mNameSpaces.IndexOf(aPrefix);
if (index != mNameSpaces.NoIndex) {
return mNameSpaces[index].nameSpaceID;
}
// The default mapping for no prefix is no namespace. If a non-null prefix
// was specified and we didn't find it, we return an error.
return aPrefix ? kNameSpaceID_Unknown : kNameSpaceID_None;
}
nsIAtom*
nsXMLNameSpaceMap::FindPrefix(int32_t aNameSpaceID) const
{
size_t index = mNameSpaces.IndexOf(aNameSpaceID);
if (index != mNameSpaces.NoIndex) {
return mNameSpaces[index].prefix;
}
return nullptr;
}
void
nsXMLNameSpaceMap::Clear()
{
mNameSpaces.Clear();
}