/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-

 * vim: set ts=8 sts=4 et sw=4 tw=99:

 * 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 "NodeUbiReporting.h"

#include "js/UbiNodeUtils.h"

#include "nsWindowSizes.h"

using JS::ubi::SimpleEdgeRange;

using JS::ubi::EdgeRange;

const char16_t JS::ubi::Concrete<nsIDocument>::concreteTypeName[] = u"nsIDocument";

const char16_t JS::ubi::Concrete<nsIContent>::concreteTypeName[] = u"nsIContent";

const char16_t JS::ubi::Concrete<Attr>::concreteTypeName[] = u"Attr";

void

JS::ubi::Concrete<nsINode>::construct(void* storage, nsINode* ptr)

{

  // nsINode is abstract, and all of its inherited instances have

  // an overridden function with instructions to construct ubi::Nodes.

  // We actually want to call that function and construct from those instances.

  ptr->ConstructUbiNode(storage);

}

js::UniquePtr<EdgeRange>

JS::ubi::Concrete<nsINode>::edges(JSContext* cx, bool wantNames) const

{

  AutoSuppressGCAnalysis suppress;

  auto range = js::MakeUnique<SimpleEdgeRange>();

  if (!range) {

    return nullptr;

  }

  if (get().GetParent()) {

    char16_t* edgeName = nullptr;

    if (wantNames) {

      edgeName = NS_xstrdup(u"Parent Node");

    }

    if (!range->addEdge(JS::ubi::Edge(edgeName, get().GetParent()))) {

      return nullptr;

    }

  }

  for (auto curr = get().GetFirstChild(); curr; curr = curr->GetNextSibling()) {

    char16_t* edgeName = nullptr;

    if (wantNames) {

      edgeName = NS_xstrdup(u"Child Node");

    }

    if (!range->addEdge(JS::ubi::Edge(edgeName, curr))) {

      return nullptr;

    }

  }

  return js::UniquePtr<EdgeRange>(range.release());

}

JS::ubi::Node::Size

JS::ubi::Concrete<nsINode>::size(mozilla::MallocSizeOf mallocSizeOf) const

{

  AutoSuppressGCAnalysis suppress;

  mozilla::SizeOfState sz(mallocSizeOf);

  nsWindowSizes wn(sz);

  size_t n = 0;

  get().AddSizeOfIncludingThis(wn, &n);

  return n;

}

const char16_t*

JS::ubi::Concrete<nsINode>::descriptiveTypeName() const

{

  return get().NodeName().get();

}

JS::ubi::Node::Size

JS::ubi::Concrete<nsIDocument>::size(mozilla::MallocSizeOf mallocSizeOf) const

{

  AutoSuppressGCAnalysis suppress;

  mozilla::SizeOfState sz(mallocSizeOf);

  nsWindowSizes wn(sz);

  getDoc().DocAddSizeOfIncludingThis(wn);

  return wn.getTotalSize();

}