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

#ifndef CONTENT_PUBLIC_BROWSER_PRIVATE_AGGREGATION_DATA_MODEL_H_
#define CONTENT_PUBLIC_BROWSER_PRIVATE_AGGREGATION_DATA_MODEL_H_

#include <set>

#include "base/functional/callback_forward.h"
#include "content/common/content_export.h"
#include "url/origin.h"

namespace content {

class CONTENT_EXPORT PrivateAggregationDataModel {
 public:
  // TODO(tquintanilla): Consider switching to site-based structure.
  class CONTENT_EXPORT DataKey {
   public:
    explicit DataKey(url::Origin reporting_origin);

    DataKey(const DataKey&);
    DataKey(DataKey&&);

    DataKey& operator=(const DataKey&);
    DataKey& operator=(DataKey&&);

    ~DataKey();

    const url::Origin& reporting_origin() const { return reporting_origin_; }

    bool operator<(const DataKey&) const;

    bool operator==(const DataKey&) const;

   private:
    url::Origin reporting_origin_;
  };

  virtual ~PrivateAggregationDataModel() = default;

  virtual void GetAllDataKeys(
      base::OnceCallback<void(std::set<DataKey>)> callback) = 0;

  virtual void RemovePendingDataKey(const DataKey& data_key,
                                    base::OnceClosure callback) = 0;
};

}  // namespace content

#endif  // CONTENT_PUBLIC_BROWSER_PRIVATE_AGGREGATION_DATA_MODEL_H_