// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/android/early_trace_event_binding.h"

#include <stdint.h>

#include "base/android/jni_string.h"
#include "base/android/trace_event_binding.h"
#include "base/time/time.h"
#include "base/trace_event/trace_event.h"
#include "base/tracing_buildflags.h"

// Must come after all headers that specialize FromJniType() / ToJniType().
#include "base/tasks_jni/EarlyTraceEvent_jni.h"

namespace base {
namespace android {

static void JNI_EarlyTraceEvent_RecordEarlyBeginEvent(JNIEnv* env,
                                                      std::string& name,
                                                      jlong time_ns,
                                                      jint thread_id,
                                                      jlong thread_time_ms) {
  auto t = perfetto::ThreadTrack::ForThread(thread_id);
  TRACE_EVENT_BEGIN(internal::kJavaTraceCategory, perfetto::DynamicString{name},
                    t, TimeTicks::FromJavaNanoTime(time_ns));
}

static void JNI_EarlyTraceEvent_RecordEarlyEndEvent(JNIEnv* env,
                                                    std::string& name,
                                                    jlong time_ns,
                                                    jint thread_id,
                                                    jlong thread_time_ms) {
  auto t = perfetto::ThreadTrack::ForThread(thread_id);
  TRACE_EVENT_END(internal::kJavaTraceCategory, t,
                  TimeTicks::FromJavaNanoTime(time_ns));
}

static void JNI_EarlyTraceEvent_RecordEarlyToplevelBeginEvent(JNIEnv* env,
                                                              std::string& name,
                                                              jlong time_ns,
                                                              jint thread_id) {
  static const unsigned char* category_group_enabled =
      TRACE_EVENT_API_GET_CATEGORY_GROUP_ENABLED(
          internal::kToplevelTraceCategory);
  trace_event_internal::AddTraceEventWithThreadIdAndTimestamps(
      TRACE_EVENT_PHASE_BEGIN, category_group_enabled, name.c_str(),
      trace_event_internal::kNoId, PlatformThreadId(thread_id),
      TimeTicks::FromJavaNanoTime(time_ns),
      TRACE_EVENT_FLAG_JAVA_STRING_LITERALS | TRACE_EVENT_FLAG_COPY);
}

static void JNI_EarlyTraceEvent_RecordEarlyToplevelEndEvent(JNIEnv* env,
                                                            std::string& name,
                                                            jlong time_ns,
                                                            jint thread_id) {
  static const unsigned char* category_group_enabled =
      TRACE_EVENT_API_GET_CATEGORY_GROUP_ENABLED(
          internal::kToplevelTraceCategory);
  trace_event_internal::AddTraceEventWithThreadIdAndTimestamps(
      TRACE_EVENT_PHASE_END, category_group_enabled, name.c_str(),
      trace_event_internal::kNoId, PlatformThreadId(thread_id),
      TimeTicks::FromJavaNanoTime(time_ns),
      TRACE_EVENT_FLAG_JAVA_STRING_LITERALS | TRACE_EVENT_FLAG_COPY);
}

static void JNI_EarlyTraceEvent_RecordEarlyAsyncBeginEvent(JNIEnv* env,
                                                           std::string& name,
                                                           jlong id,
                                                           jlong time_ns) {
  TRACE_EVENT_BEGIN(
      internal::kJavaTraceCategory, nullptr,
      perfetto::Track(static_cast<uint64_t>(id)),
      TimeTicks::FromJavaNanoTime(time_ns),
      [&](::perfetto::EventContext& ctx) { ctx.event()->set_name(name); });
}

static void JNI_EarlyTraceEvent_RecordEarlyAsyncEndEvent(JNIEnv* env,
                                                         jlong id,
                                                         jlong time_ns) {
  TRACE_EVENT_END(internal::kJavaTraceCategory,
                  perfetto::Track(static_cast<uint64_t>(id)));
}

bool GetBackgroundStartupTracingFlagFromJava() {
  JNIEnv* env = jni_zero::AttachCurrentThread();
  return base::android::Java_EarlyTraceEvent_getBackgroundStartupTracingFlag(
      env);
}

void SetBackgroundStartupTracingFlag(bool enabled) {
  JNIEnv* env = jni_zero::AttachCurrentThread();
  base::android::Java_EarlyTraceEvent_setBackgroundStartupTracingFlag(env,
                                                                      enabled);
}

}  // namespace android
}  // namespace base

DEFINE_JNI(EarlyTraceEvent)