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

#pragma once

#ifdef WITH_PERFETTO

#include <perfetto.h>

namespace facebook::react {

class HermesPerfettoDataSource : public perfetto::DataSource<HermesPerfettoDataSource> {
 public:
  void OnSetup(const SetupArgs &) override {}

  void OnStart(const StartArgs &) override;

  void OnFlush(const FlushArgs &) override;

  void OnStop(const StopArgs &a) override;

  static void RegisterDataSource()
  {
    perfetto::DataSourceDescriptor dsd;
    dsd.set_name("com.facebook.hermes.profiler");
    HermesPerfettoDataSource::Register(dsd);
  }

  constexpr static perfetto::BufferExhaustedPolicy kBufferExhaustedPolicy = perfetto::BufferExhaustedPolicy::kStall;
};

} // namespace facebook::react

PERFETTO_DECLARE_DATA_SOURCE_STATIC_MEMBERS(facebook::react::HermesPerfettoDataSource);

#endif // WITH_PERFETTO
