// 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_SEGMENTATION_PLATFORM_INTERNAL_DATABASE_UKM_DATABASE_IMPL_H_
#define COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_DATABASE_UKM_DATABASE_IMPL_H_

#include <cstdint>
#include <memory>
#include <vector>

#include "base/files/file_path.h"
#include "base/threading/sequence_bound.h"
#include "base/time/time.h"
#include "components/segmentation_platform/internal/database/ukm_database.h"
#include "components/segmentation_platform/internal/database/ukm_database_backend.h"
#include "components/segmentation_platform/internal/database/ukm_types.h"
#include "services/metrics/public/cpp/ukm_source_id.h"
#include "services/metrics/public/mojom/ukm_interface.mojom.h"
#include "url/gurl.h"

namespace segmentation_platform {

// Created and run on the main UI thread, and hosts the `UkmDatabaseBackend`
// which runs on a background thread.
class UkmDatabaseImpl : public UkmDatabase {
 public:
  explicit UkmDatabaseImpl(const base::FilePath& database_path, bool in_memory);
  ~UkmDatabaseImpl() override;

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

  void InitDatabase(SuccessCallback callback) override;
  void StoreUkmEntry(ukm::mojom::UkmEntryPtr ukm_entry) override;
  void UpdateUrlForUkmSource(ukm::SourceId source_id,
                             const GURL& url,
                             bool is_validated,
                             const std::string& profile_id) override;
  void OnUrlValidated(const GURL& url, const std::string& profile_id) override;
  void RemoveUrls(const std::vector<GURL>& urls, bool all_urls) override;
  void AddUmaMetric(const std::string& profile_id,
                    const UmaMetricEntry& row) override;
  void RunReadOnlyQueries(QueryList&& queries, QueryCallback callback) override;
  void CleanupOldEntries(base::Time ukm_time_limit,
                         base::Time uma_time_limit) override;
  void CleanupItems(const std::string& profile_id,
                    std::vector<CleanupItem> cleanup_items) override;
  void CommitTransactionForTesting() override;

 private:
  base::SequenceBound<UkmDatabaseBackend> backend_;
};

}  // namespace segmentation_platform

#endif  // COMPONENTS_SEGMENTATION_PLATFORM_INTERNAL_DATABASE_UKM_DATABASE_IMPL_H_