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

#include "mozilla/HangAnnotations.h"

#include <vector>

#include "MainThreadUtils.h"
#include "mozilla/DebugOnly.h"
#include "nsXULAppAPI.h"
#include "mozilla/BackgroundHangMonitor.h"

namespace mozilla {

void
BackgroundHangAnnotations::AddAnnotation(const nsString& aName, const int32_t aData)
{
  nsAutoString dataString;
  dataString.AppendInt(aData);
  AppendElement(HangAnnotation(aName, dataString));
}

void
BackgroundHangAnnotations::AddAnnotation(const nsString& aName, const double aData)
{
  nsAutoString dataString;
  dataString.AppendFloat(aData);
  AppendElement(HangAnnotation(aName, dataString));
}

void
BackgroundHangAnnotations::AddAnnotation(const nsString& aName, const nsString& aData)
{
  AppendElement(HangAnnotation(aName, aData));
}

void
BackgroundHangAnnotations::AddAnnotation(const nsString& aName, const nsCString& aData)
{
  NS_ConvertUTF8toUTF16 dataString(aData);
  AppendElement(HangAnnotation(aName, dataString));
}

void
BackgroundHangAnnotations::AddAnnotation(const nsString& aName, const bool aData)
{
  if (aData) {
    AppendElement(HangAnnotation(aName, NS_LITERAL_STRING("true")));
  } else {
    AppendElement(HangAnnotation(aName, NS_LITERAL_STRING("false")));
  }
}

BackgroundHangAnnotators::BackgroundHangAnnotators()
  : mMutex("BackgroundHangAnnotators::mMutex")
{
  MOZ_COUNT_CTOR(BackgroundHangAnnotators);
}

BackgroundHangAnnotators::~BackgroundHangAnnotators()
{
  MOZ_ASSERT(mAnnotators.empty());
  MOZ_COUNT_DTOR(BackgroundHangAnnotators);
}

bool
BackgroundHangAnnotators::Register(BackgroundHangAnnotator& aAnnotator)
{
  MutexAutoLock lock(mMutex);
  auto result = mAnnotators.insert(&aAnnotator);
  return result.second;
}

bool
BackgroundHangAnnotators::Unregister(BackgroundHangAnnotator& aAnnotator)
{
  MutexAutoLock lock(mMutex);
  DebugOnly<std::set<BackgroundHangAnnotator*>::size_type> numErased;
  numErased = mAnnotators.erase(&aAnnotator);
  MOZ_ASSERT(numErased == 1);
  return mAnnotators.empty();
}

BackgroundHangAnnotations
BackgroundHangAnnotators::GatherAnnotations()
{
  BackgroundHangAnnotations annotations;
  { // Scope for lock
    MutexAutoLock lock(mMutex);
    for (std::set<BackgroundHangAnnotator*>::iterator i = mAnnotators.begin(),
         e = mAnnotators.end();
         i != e; ++i) {
      (*i)->AnnotateHang(annotations);
    }
  }
  return annotations;
}

} // namespace mozilla
