dom/xslt/base/txNamespaceMap.cpp

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "txNamespaceMap.h"
   7 #include "nsGkAtoms.h"
   8 #include "txXPathNode.h"
   9
  10 txNamespaceMap::txNamespaceMap() = default;
  11
  12 txNamespaceMap::txNamespaceMap(const txNamespaceMap& aOther)
  13     : mPrefixes(aOther.mPrefixes.Clone()),
  14       mNamespaces(aOther.mNamespaces.Clone()) {}
  15
  16 nsresult txNamespaceMap::mapNamespace(nsAtom* aPrefix,
  17                                       const nsAString& aNamespaceURI) {
  18   nsAtom* prefix = aPrefix == nsGkAtoms::_empty ? nullptr : aPrefix;
  19
  20   int32_t nsId;
  21   if (prefix && aNamespaceURI.IsEmpty()) {
  22     // Remove the mapping
  23     int32_t index = mPrefixes.IndexOf(prefix);
  24     if (index >= 0) {
  25       mPrefixes.RemoveElementAt(index);
  26       mNamespaces.RemoveElementAt(index);
  27     }
  28
  29     return NS_OK;
  30   }
  31
  32   if (aNamespaceURI.IsEmpty()) {
  33     // Set default to empty namespace
  34     nsId = kNameSpaceID_None;
  35   } else {
  36     nsId = txNamespaceManager::getNamespaceID(aNamespaceURI);
  37     NS_ENSURE_FALSE(nsId == kNameSpaceID_Unknown, NS_ERROR_FAILURE);
  38   }
  39
  40   // Check if the mapping already exists
  41   int32_t index = mPrefixes.IndexOf(prefix);
  42   if (index >= 0) {
  43     mNamespaces.ElementAt(index) = nsId;
  44
  45     return NS_OK;
  46   }
  47
  48   // New mapping
  49   mPrefixes.AppendElement(prefix);
  50   mNamespaces.AppendElement(nsId);
  51
  52   return NS_OK;
  53 }
  54
  55 int32_t txNamespaceMap::lookupNamespace(nsAtom* aPrefix) {
  56   if (aPrefix == nsGkAtoms::xml) {
  57     return kNameSpaceID_XML;
  58   }
  59
  60   nsAtom* prefix = aPrefix == nsGkAtoms::_empty ? 0 : aPrefix;
  61
  62   int32_t index = mPrefixes.IndexOf(prefix);
  63   if (index >= 0) {
  64     return mNamespaces.SafeElementAt(index, kNameSpaceID_Unknown);
  65   }
  66
  67   if (!prefix) {
  68     return kNameSpaceID_None;
  69   }
  70
  71   return kNameSpaceID_Unknown;
  72 }
  73
  74 int32_t txNamespaceMap::lookupNamespaceWithDefault(const nsAString& aPrefix) {
  75   RefPtr<nsAtom> prefix = NS_Atomize(aPrefix);
  76   if (prefix != nsGkAtoms::_poundDefault) {
  77     return lookupNamespace(prefix);
  78   }
  79
  80   return lookupNamespace(nullptr);
  81 }