chrome/browser/devtools/protocol/storage_handler.h

// 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_DEVTOOLS_PROTOCOL_STORAGE_HANDLER_H_
#define CHROME_BROWSER_DEVTOOLS_PROTOCOL_STORAGE_HANDLER_H_

#include <memory>
#include <string>
#include <vector>

#include "base/memory/weak_ptr.h"
#include "base/scoped_observation.h"
#include "chrome/browser/devtools/protocol/protocol.h"
#include "chrome/browser/devtools/protocol/storage.h"
#include "chrome/browser/private_verification_tokens/private_verification_tokens_service.h"

namespace content {
class WebContents;
}  // namespace content

class StorageHandler : public protocol::Storage::Backend,
                       public PrivateVerificationTokensService::Observer {
 public:
  StorageHandler(content::WebContents* web_contents,
                 protocol::UberDispatcher* dispatcher);

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

  ~StorageHandler() override;

 private:
  protocol::Response Disable() override;

  void RunBounceTrackingMitigations(
      std::unique_ptr<RunBounceTrackingMitigationsCallback> callback) override;

  void GetPrivateVerificationTokens(
      std::unique_ptr<GetPrivateVerificationTokensCallback> callback) override;
  void ClearPrivateVerificationTokens(
      const std::string& in_issuerOrigin,
      std::unique_ptr<ClearPrivateVerificationTokensCallback> callback)
      override;
  void DeletePrivateVerificationToken(
      const std::string& in_tokenId,
      std::unique_ptr<DeletePrivateVerificationTokenCallback> callback)
      override;
  protocol::Response SetPrivateVerificationTokensTracking(bool enable) override;

  // PrivateVerificationTokensService::Observer:
  void OnTokensStored() override;
  void OnTokensDeleted() override;
  void OnShutdown() override;

  static void GotDeletedSites(
      std::unique_ptr<RunBounceTrackingMitigationsCallback> callback,
      const std::vector<std::string>& sites);

  base::WeakPtr<content::WebContents> web_contents_;
  std::unique_ptr<protocol::Storage::Frontend> frontend_;
  base::ScopedObservation<PrivateVerificationTokensService,
                          PrivateVerificationTokensService::Observer>
      pvt_observation_{this};
};

#endif  // CHROME_BROWSER_DEVTOOLS_PROTOCOL_STORAGE_HANDLER_H_