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

#include <reactperflogger/fusebox/FuseboxTracer.h>

namespace facebook::react::jsinspector_modern {

bool TracingAgent::handleRequest(const cdp::PreparsedRequest& req) {
  if (req.method == "Tracing.start") {
    // @cdp Tracing.start support is experimental.
    if (FuseboxTracer::getFuseboxTracer().startTracing()) {
      frontendChannel_(cdp::jsonResult(req.id));
    } else {
      frontendChannel_(cdp::jsonError(
          req.id,
          cdp::ErrorCode::InternalError,
          "Tracing session already started"));
    }

    return true;
  } else if (req.method == "Tracing.end") {
    // @cdp Tracing.end support is experimental.
    bool firstChunk = true;
    auto id = req.id;
    bool wasStopped = FuseboxTracer::getFuseboxTracer().stopTracing(
        [this, firstChunk, id](const folly::dynamic& eventsChunk) {
          if (firstChunk) {
            frontendChannel_(cdp::jsonResult(id));
          }
          frontendChannel_(cdp::jsonNotification(
              "Tracing.dataCollected",
              folly::dynamic::object("value", eventsChunk)));
        });

    if (!wasStopped) {
      frontendChannel_(cdp::jsonError(
          req.id,
          cdp::ErrorCode::InternalError,
          "Tracing session not started"));
    } else {
      frontendChannel_(cdp::jsonNotification(
          "Tracing.tracingComplete",
          folly::dynamic::object("dataLossOccurred", false)));
    }

    frontendChannel_(cdp::jsonResult(req.id));
    return true;
  }

  return false;
}

} // namespace facebook::react::jsinspector_modern