content/browser/browsing_data/same_site_data_remover_impl.h

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

#ifndef CONTENT_BROWSER_BROWSING_DATA_SAME_SITE_DATA_REMOVER_IMPL_H_
#define CONTENT_BROWSER_BROWSING_DATA_SAME_SITE_DATA_REMOVER_IMPL_H_

#include <set>
#include <string>

#include "base/callback.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/storage_partition.h"

namespace net {
class CookieStore;
}

namespace content {

class CONTENT_EXPORT SameSiteDataRemoverImpl {
 public:
  explicit SameSiteDataRemoverImpl(BrowserContext* browser_context);
  ~SameSiteDataRemoverImpl();

  // Returns a set containing domains associated with deleted SameSite=None
  // cookies.
  const std::set<std::string>& GetDeletedDomainsForTesting();

  // Deletes cookies with SameSite attribute value NO_RESTRICTION
  // from the CookieStore and stores domains of deleted cookies in the
  // same_site_none_domains_ vector. The closure is called after cookies
  // have been deleted.
  void DeleteSameSiteNoneCookies(base::OnceClosure closure);

  // Clears additional storage for domains that have cookies with SameSite value
  // NO_RESTRICTION from the StoragePartition. Storage is cleared based on the
  // domains of the cookies deleted in DeleteSameSiteNoneCookies().
  //
  // This is called safely on an instance which is destroyed after the function
  // call since it's not needed for the function execution.
  void ClearStoragePartitionData(base::OnceClosure closure);

  // Clears storage for domains in the provided set.
  void ClearStoragePartitionForOrigins(
      base::OnceClosure closure,
      StoragePartition::OriginMatcherFunction origin_matcher);

  // For testing purposes only.
  void OverrideStoragePartitionForTesting(StoragePartition* storage_partition);

 private:
  BrowserContext* browser_context_;
  StoragePartition* storage_partition_;
  std::set<std::string> same_site_none_domains_;

  DISALLOW_COPY_AND_ASSIGN(SameSiteDataRemoverImpl);
};

}  // namespace content

#endif  // CONTENT_BROWSER_BROWSING_DATA_SAME_SITE_DATA_REMOVER_IMPL_H_