/* 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/. */

#include "SVGElementFactory.h"

#include "mozilla/StaticPtr.h"
#include "mozilla/dom/Element.h"
#include "mozilla/dom/FromParser.h"
#include "mozilla/dom/NodeInfo.h"
#include "nsGkAtoms.h"
#include "nsHashKeys.h"
#include "nsIContent.h"
#include "nsTHashMap.h"

using namespace mozilla;
using namespace mozilla::dom;

// Hash table that maps nsAtom* SVG tags to a SVGContentCreatorFunction.
using TagAtomTable =
    nsTHashMap<nsPtrHashKey<nsAtom>, SVGContentCreatorFunction>;
StaticAutoPtr<TagAtomTable> sTagAtomTable;

#define SVG_TAG(_tag, _classname)                                         \
  nsresult NS_NewSVG##_classname##Element(                                \
      nsIContent** aResult,                                               \
      already_AddRefed<mozilla::dom::NodeInfo> aNodeInfo);                \
                                                                          \
  nsresult NS_NewSVG##_classname##Element(                                \
      nsIContent** aResult,                                               \
      already_AddRefed<mozilla::dom::NodeInfo> aNodeInfo,                 \
      FromParser aFromParser) {                                           \
    return NS_NewSVG##_classname##Element(aResult, std::move(aNodeInfo)); \
  }

#define SVG_FROM_PARSER_TAG(_tag, _classname)

#include "SVGTagList.inc"
#undef SVG_TAG
#undef SVG_FROM_PARSER_TAG

nsresult NS_NewSVGElement(Element** aResult,
                          already_AddRefed<mozilla::dom::NodeInfo> aNodeInfo);

enum SVGTag {
#define SVG_TAG(_tag, _classname) eSVGTag_##_tag,
#define SVG_FROM_PARSER_TAG(_tag, _classname) eSVGTag_##_tag,
#include "SVGTagList.inc"
#undef SVG_TAG
#undef SVG_FROM_PARSER_TAG
  eSVGTag_Count
};

void SVGElementFactory::Init() {
  sTagAtomTable = new TagAtomTable(64);

#define SVG_TAG(_tag, _classname) \
  sTagAtomTable->InsertOrUpdate(  \
      nsGkAtoms::_tag,            \
      SVGContentCreatorFunction(NS_NewSVG##_classname##Element));
#define SVG_FROM_PARSER_TAG(_tag, _classname) \
  sTagAtomTable->InsertOrUpdate(              \
      nsGkAtoms::_tag,                        \
      SVGContentCreatorFunction(NS_NewSVG##_classname##Element));
#include "SVGTagList.inc"
#undef SVG_TAG
#undef SVG_FROM_PARSER_TAG
}

void SVGElementFactory::Shutdown() { sTagAtomTable = nullptr; }

nsresult NS_NewSVGElement(Element** aResult,
                          already_AddRefed<mozilla::dom::NodeInfo> aNodeInfo,
                          FromParser aFromParser) {
  NS_ASSERTION(sTagAtomTable, "no lookup table, needs SVGElementFactory::Init");

  RefPtr<mozilla::dom::NodeInfo> ni = aNodeInfo;
  nsAtom* name = ni->NameAtom();

  NS_ASSERTION(
      ni->NamespaceEquals(kNameSpaceID_SVG),
      "Trying to create SVG elements that aren't in the SVG namespace");

  SVGContentCreatorFunction cb = sTagAtomTable->Get(name);
  if (cb) {
    nsCOMPtr<nsIContent> content;
    nsresult rv = cb(getter_AddRefs(content), ni.forget(), aFromParser);
    *aResult = content.forget().take()->AsElement();
    return rv;
  }

  // if we don't know what to create, just create a standard svg element:
  return NS_NewSVGElement(aResult, ni.forget());
}

nsresult NS_NewSVGUnknownElement(
    nsIContent** aResult, already_AddRefed<mozilla::dom::NodeInfo> aNodeInfo,
    FromParser aFromParser) {
  RefPtr<mozilla::dom::NodeInfo> ni = aNodeInfo;
  nsCOMPtr<Element> element;
  nsresult rv = NS_NewSVGElement(getter_AddRefs(element), ni.forget());
  element.forget(aResult);
  return rv;
}