// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "peridot/bin/basemgr/cobalt/cobalt.h"

#include <fuchsia/cobalt/cpp/fidl.h>
#include <lib/cobalt/cpp/deprecated_cobalt_logger.h>

#include "peridot/bin/basemgr/cobalt/basemgr_metrics_registry.cb.h"
#include "src/lib/cobalt/cpp/cobalt_logger.h"

namespace modular {
namespace {

cobalt::CobaltLogger* g_cobalt_logger = nullptr;

}  // namespace

fit::deferred_action<fit::closure> InitializeCobalt(async_dispatcher_t* dispatcher,
                                                    sys::ComponentContext* context) {
  FXL_DCHECK(!g_cobalt_logger) << "Cobalt has already been initialized.";

  std::unique_ptr<cobalt::CobaltLogger> cobalt_logger =
      cobalt::NewCobaltLoggerFromProjectName(dispatcher, context, cobalt_registry::kProjectName);

  g_cobalt_logger = cobalt_logger.get();
  return fit::defer<fit::closure>(
      [cobalt_logger = std::move(cobalt_logger)] { g_cobalt_logger = nullptr; });
}

void ReportEvent(cobalt_registry::ModularLifetimeEventsMetricDimensionEventType event) {
  if (g_cobalt_logger) {
    g_cobalt_logger->LogEvent(
        static_cast<uint32_t>(cobalt_registry::kModularLifetimeEventsMetricId),
        static_cast<uint32_t>(event));
  }
}

void ReportModuleLaunchTime(std::string module_url, zx::duration time) {
  if (g_cobalt_logger) {
    g_cobalt_logger->LogElapsedTime(
        static_cast<uint32_t>(cobalt_registry::kModuleLaunchTimeMetricId), 0, module_url, time);
  }
}

void ReportStoryLaunchTime(zx::duration time) {
  if (g_cobalt_logger) {
    g_cobalt_logger->LogElapsedTime(
        static_cast<uint32_t>(cobalt_registry::kStoryLaunchTimeMetricId), 0, "", time);
  }
}

void ReportSessionAgentEvent(const std::string& url,
                             cobalt_registry::SessionAgentEventsMetricDimensionEventType event) {
  if (g_cobalt_logger) {
    g_cobalt_logger->LogEventCount(
        static_cast<uint32_t>(cobalt_registry::kSessionAgentEventsMetricId),
        static_cast<uint32_t>(event), url /* component */,
        zx::duration(0) /* period_duration_micros */, 1 /* count */);
  }
}

}  // namespace modular