// 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 CHROME_BROWSER_BROWSING_DATA_CHROME_BROWSING_DATA_MODEL_DELEGATE_H_
#define CHROME_BROWSER_BROWSING_DATA_CHROME_BROWSING_DATA_MODEL_DELEGATE_H_

#include "base/containers/flat_map.h"
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/profiles/profile.h"
#include "components/browsing_data/content/browsing_data_model.h"
#include "components/webid/federated_identity_data_model.h"
#include "content/public/browser/render_frame_host.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"

namespace content {
class StoragePartition;
}  // namespace content

class ChromeBrowsingDataModelDelegate final
    : public BrowsingDataModel::Delegate {
 public:
  // Storage types which are represented by the model. Some types have
  // incomplete implementations, and are marked as such.
  // TODO(crbug.com/40205603): Complete implementations for all browsing data.
  enum class StorageType {
    kTopics = static_cast<int>(BrowsingDataModel::StorageType::kLastType) +
              1,      // Not fetched from disk.
    kIsolatedWebApp,  // Not yet deletable.
    kMediaDeviceSalt,
    kFederatedIdentity,

    kFirstType = kTopics,
    kLastType = kFederatedIdentity,
  };

  static void BrowsingDataAccessed(content::RenderFrameHost* rfh,
                                   const BrowsingDataModel::DataKey& data_key,
                                   StorageType storage_type,
                                   bool blocked);

  static std::unique_ptr<ChromeBrowsingDataModelDelegate> CreateForProfile(
      Profile* profile);

  static std::unique_ptr<ChromeBrowsingDataModelDelegate>
  CreateForStoragePartition(Profile* profile,
                            content::StoragePartition* storage_partition);

  ~ChromeBrowsingDataModelDelegate() override;

  // BrowsingDataModel::Delegate:
  void GetAllDataKeys(
      base::OnceCallback<void(std::vector<DelegateEntry>)> callback) override;
  void RemoveDataKey(const BrowsingDataModel::DataKey& data_key,
                     BrowsingDataModel::StorageTypeSet storage_types,
                     base::OnceClosure callback) override;
  std::optional<BrowsingDataModel::DataOwner> GetDataOwner(
      const BrowsingDataModel::DataKey& data_key,
      BrowsingDataModel::StorageType storage_type) const override;
  std::optional<bool> IsStorageTypeCookieLike(
      BrowsingDataModel::StorageType storage_type) const override;
  std::optional<bool> IsBlockedByThirdPartyCookieBlocking(
      const BrowsingDataModel::DataKey& data_key,
      BrowsingDataModel::StorageType storage_type) const override;
  bool IsCookieDeletionDisabled(const GURL& url) override;
  base::WeakPtr<BrowsingDataModel::Delegate> AsWeakPtr() override;

 private:
  ChromeBrowsingDataModelDelegate(Profile* profile,
                                  content::StoragePartition* storage_partition);
  void GetAllMediaDeviceSaltDataKeys(
      base::OnceCallback<void(std::vector<DelegateEntry>)> callback,
      std::vector<DelegateEntry> entries);
  void GotAllMediaDeviceSaltDataKeys(
      base::OnceCallback<void(std::vector<DelegateEntry>)> callback,
      std::vector<DelegateEntry> entries,
      std::vector<blink::StorageKey> storage_keys);
  void RemoveMediaDeviceSalt(const blink::StorageKey& storage_key,
                             base::OnceClosure callback);
  void GetAllFederatedIdentityDataKeys(
      base::OnceCallback<void(std::vector<DelegateEntry>)> callback,
      std::vector<DelegateEntry> entries);
  void GotAllFederatedIdentityDataKeys(
      base::OnceCallback<void(std::vector<DelegateEntry>)> callback,
      std::vector<DelegateEntry> entries,
      std::vector<webid::FederatedIdentityDataModel::DataKey> data_keys);
  void RemoveFederatedIdentityData(
      const webid::FederatedIdentityDataModel::DataKey& data_key,
      base::OnceClosure callback);

  const raw_ptr<Profile, DanglingUntriaged> profile_;
  const raw_ptr<content::StoragePartition, DanglingUntriaged>
      storage_partition_;
  base::WeakPtrFactory<ChromeBrowsingDataModelDelegate> weak_ptr_factory_{this};
};

#endif  // CHROME_BROWSER_BROWSING_DATA_CHROME_BROWSING_DATA_MODEL_DELEGATE_H_