/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include <memory>

#include "NativePerformanceObserver.h"
#include "PerformanceEntryReporter.h"

#include <react/renderer/uimanager/UIManagerBinding.h>
#include <react/utils/CoreFeatures.h>

#include "Plugins.h"

std::shared_ptr<facebook::react::TurboModule>
NativePerformanceObserverModuleProvider(
    std::shared_ptr<facebook::react::CallInvoker> jsInvoker) {
  return std::make_shared<facebook::react::NativePerformanceObserver>(
      std::move(jsInvoker));
}

namespace facebook::react {

NativePerformanceObserver::NativePerformanceObserver(
    std::shared_ptr<CallInvoker> jsInvoker)
    : NativePerformanceObserverCxxSpec(std::move(jsInvoker)) {
  setEventLogger(&PerformanceEntryReporter::getInstance());
}

NativePerformanceObserver::~NativePerformanceObserver() {
  setEventLogger(nullptr);
}

void NativePerformanceObserver::startReporting(
    jsi::Runtime& rt,
    int32_t entryType) {
  PerformanceEntryReporter& reporter = PerformanceEntryReporter::getInstance();

  PerformanceEntryType entryTypeEnum =
      static_cast<PerformanceEntryType>(entryType);
  reporter.startReporting(entryTypeEnum);

  if (entryTypeEnum == PerformanceEntryType::EVENT &&
      CoreFeatures::enableReportEventPaintTime) {
    UIManagerBinding::getBinding(rt)->getUIManager().registerMountHook(
        reporter);
  }
}

void NativePerformanceObserver::stopReporting(
    jsi::Runtime& rt,
    int32_t entryType) {
  PerformanceEntryReporter& reporter = PerformanceEntryReporter::getInstance();

  PerformanceEntryType entryTypeEnum =
      static_cast<PerformanceEntryType>(entryType);
  reporter.stopReporting(entryTypeEnum);

  if (entryTypeEnum == PerformanceEntryType::EVENT &&
      CoreFeatures::enableReportEventPaintTime) {
    UIManagerBinding::getBinding(rt)->getUIManager().unregisterMountHook(
        reporter);
  }
}

void NativePerformanceObserver::setIsBuffered(
    jsi::Runtime& rt,
    std::vector<int32_t> entryTypes,
    bool isBuffered) {
  for (const int32_t entryType : entryTypes) {
    PerformanceEntryReporter::getInstance().setAlwaysLogged(
        static_cast<PerformanceEntryType>(entryType), isBuffered);
  }
}

GetPendingEntriesResult NativePerformanceObserver::popPendingEntries(
    jsi::Runtime& rt) {
  return PerformanceEntryReporter::getInstance().popPendingEntries();
}

void NativePerformanceObserver::setOnPerformanceEntryCallback(
    jsi::Runtime& rt,
    std::optional<AsyncCallback<>> callback) {
  PerformanceEntryReporter::getInstance().setReportingCallback(callback);
}

void NativePerformanceObserver::logRawEntry(
    jsi::Runtime& rt,
    RawPerformanceEntry entry) {
  PerformanceEntryReporter::getInstance().logEntry(entry);
}

std::vector<std::pair<std::string, uint32_t>>
NativePerformanceObserver::getEventCounts(jsi::Runtime& rt) {
  const auto& eventCounts =
      PerformanceEntryReporter::getInstance().getEventCounts();
  return std::vector<std::pair<std::string, uint32_t>>(
      eventCounts.begin(), eventCounts.end());
}

void NativePerformanceObserver::setDurationThreshold(
    jsi::Runtime& rt,
    int32_t entryType,
    double durationThreshold) {
  PerformanceEntryReporter::getInstance().setDurationThreshold(
      static_cast<PerformanceEntryType>(entryType), durationThreshold);
}

void NativePerformanceObserver::clearEntries(
    jsi::Runtime& rt,
    int32_t entryType,
    std::optional<std::string> entryName) {
  PerformanceEntryReporter::getInstance().clearEntries(
      static_cast<PerformanceEntryType>(entryType),
      entryName ? entryName->c_str() : nullptr);
}

std::vector<RawPerformanceEntry> NativePerformanceObserver::getEntries(
    jsi::Runtime& rt,
    std::optional<int32_t> entryType,
    std::optional<std::string> entryName) {
  return PerformanceEntryReporter::getInstance().getEntries(
      entryType ? static_cast<PerformanceEntryType>(*entryType)
                : PerformanceEntryType::UNDEFINED,
      entryName ? entryName->c_str() : nullptr);
}

std::vector<RawPerformanceEntryType>
NativePerformanceObserver::getSupportedPerformanceEntryTypes(jsi::Runtime& rt) {
  return {
      static_cast<RawPerformanceEntryType>(PerformanceEntryType::MARK),
      static_cast<RawPerformanceEntryType>(PerformanceEntryType::MEASURE),
      static_cast<RawPerformanceEntryType>(PerformanceEntryType::EVENT),
  };
}

} // namespace facebook::react