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

#ifndef dom_NodeUbiReporting_h
#define dom_NodeUbiReporting_h

#include "Attr.h"
#include "Document.h"
#include "nsIContent.h"
#include "nsIContentInlines.h"
#include "nsINode.h"
#include "js/UbiNode.h"

/*
 * This file defines specializations of JS::ubi::Concrete for DOM nodes
 * so that the JS memory tools, which operate on the UbiNode graph, can
 * define subclasses of JS::ubi::Base that represent DOM nodes and
 * yield the outgoing edges in a DOM node graph for reporting.
 */

namespace JS::ubi {

// The DOM node base class.
// This is an abstract class and therefore does not require a concreteTypeName.
template <>
class Concrete<nsINode> : public Base {
 protected:
  explicit Concrete(nsINode* ptr) : Base(ptr) {}

 public:
  static void construct(void* storage, nsINode* ptr);
  Size size(mozilla::MallocSizeOf mallocSizeOf) const override;
  js::UniquePtr<EdgeRange> edges(JSContext* cx, bool wantNames) const override;

  nsINode& get() const { return *static_cast<nsINode*>(ptr); }
  CoarseType coarseType() const final { return CoarseType::DOMNode; }
  const char16_t* descriptiveTypeName() const override;
};

template <>
class Concrete<nsIContent> : public Concrete<nsINode> {
 protected:
  explicit Concrete(nsIContent* ptr) : Concrete<nsINode>(ptr) {}

 public:
  static void construct(void* storage, nsIContent* ptr) {
    new (storage) Concrete(ptr);
  }
  const char16_t* typeName() const override { return concreteTypeName; };
  static const char16_t concreteTypeName[];
};

template <>
class Concrete<mozilla::dom::Document> : public Concrete<nsINode> {
 protected:
  explicit Concrete(mozilla::dom::Document* ptr) : Concrete<nsINode>(ptr) {}

 public:
  static void construct(void* storage, mozilla::dom::Document* ptr) {
    new (storage) Concrete(ptr);
  }
  Size size(mozilla::MallocSizeOf mallocSizeOf) const override;

  mozilla::dom::Document& getDoc() const {
    return *static_cast<mozilla::dom::Document*>(ptr);
  }
  const char16_t* typeName() const override { return concreteTypeName; };
  static const char16_t concreteTypeName[];
};

template <>
class Concrete<mozilla::dom::Attr> : public Concrete<nsINode> {
 protected:
  explicit Concrete(mozilla::dom::Attr* ptr) : Concrete<nsINode>(ptr) {}

 public:
  static void construct(void* storage, mozilla::dom::Attr* ptr) {
    new (storage) Concrete(ptr);
  }
  const char16_t* typeName() const override { return concreteTypeName; };
  static const char16_t concreteTypeName[];
};

}  // namespace JS::ubi

#endif