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

#ifndef COMPONENTS_TRACING_COMMON_BACKGROUND_TRACING_METRICS_PROVIDER_H_
#define COMPONENTS_TRACING_COMMON_BACKGROUND_TRACING_METRICS_PROVIDER_H_

#include <vector>

#include "base/functional/callback.h"
#include "components/metrics/metrics_provider.h"
#include "components/tracing/tracing_export.h"
#include "third_party/metrics_proto/system_profile.pb.h"
#include "third_party/metrics_proto/trace_log.pb.h"

namespace tracing {

// Provides trace log metrics collected using BackgroundTracingManager to UMA
// proto. Background tracing uploads metrics of larger size compared to UMA
// histograms and it is better to upload them as independent metrics rather
// than part of UMA histograms log. The background tracing manager will make
// sure the traces are small when uploading over data.
class TRACING_EXPORT BackgroundTracingMetricsProvider
    : public metrics::MetricsProvider {
 public:
  static base::RepeatingCallback<void(metrics::SystemProfileProto&)>
  GetSystemProfileMetricsRecorder();

  BackgroundTracingMetricsProvider();
  ~BackgroundTracingMetricsProvider() override;

  BackgroundTracingMetricsProvider(const BackgroundTracingMetricsProvider&) =
      delete;
  BackgroundTracingMetricsProvider& operator=(
      const BackgroundTracingMetricsProvider&) = delete;

  void RecordSystemProfileMetrics(
      metrics::SystemProfileProto& system_profile_proto);

  // metrics::MetricsProvider:
  void AsyncInit(base::OnceClosure done_callback) override;
  bool HasIndependentMetrics() override;
  void ProvideIndependentMetrics(
      base::OnceClosure serialize_log_callback,
      base::OnceCallback<void(bool)> done_callback,
      metrics::ChromeUserMetricsExtension* uma_proto,
      base::HistogramSnapshotManager* snapshot_manager) override;

 protected:
  // Embedders can override this to do any additional processing of the log
  // before it is sent. This includes processing of the trace itself (e.g.
  // compression).
  virtual base::OnceCallback<bool(metrics::ChromeUserMetricsExtension*,
                                  std::string&&)>
  GetEmbedderMetricsProvider();

  virtual void RecordCoreSystemProfileMetrics(
      metrics::SystemProfileProto& system_profile_proto) = 0;

  // Writes |serialized_trace| into |logs|'s |raw_data| field.
  static void SetTrace(metrics::TraceLog* log, std::string&& serialized_trace);

  std::vector<std::unique_ptr<metrics::MetricsProvider>>
      system_profile_providers_;

  base::WeakPtrFactory<BackgroundTracingMetricsProvider> weak_factory_{this};
};

}  // namespace tracing

#endif  // COMPONENTS_TRACING_COMMON_BACKGROUND_TRACING_METRICS_PROVIDER_H_