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

#include <string>
#include <vector>

#include "content/browser/btm/btm_utils.h"
#include "content/common/content_export.h"
#include "url/gurl.h"

namespace content {

// Filters a chain of URLs to the ones which accessed cookies.
//
// Intended for use by a WebContentsObserver which overrides OnCookiesAccessed
// and DidFinishNavigation.
class CONTENT_EXPORT CookieAccessFilter {
 public:
  CookieAccessFilter();
  ~CookieAccessFilter();

  // Record that `url` accessed cookies.
  void AddAccess(const GURL& url, CookieOperation op);

  // Clear `result` and fill it with the the type of cookie access for each URL.
  // `result` will have the same length as `urls`. Returns true iff every
  // previously-recorded cookie access was successfully matched to a URL in
  // `urls`. Otherwise returns false, and `result` is filled entirely with
  // kUnknown. (Note: this depends on the order of previous calls to
  // AddAccess()).
  bool Filter(const std::vector<GURL>& urls,
              std::vector<BtmDataAccessType>& result) const;

  // Returns true iff AddAccess() has never been called.
  bool is_empty() const { return accesses_.empty(); }

  // Returns a vector containing the URLs of added cookie accesses.
  //
  // TODO - crbug.com/406841434: Remove once we identify the source of
  // mismatched cookie accesses.
  std::vector<GURL> GetUrlsForDebuging() const;

 private:
  struct CookieAccess {
    GURL url;
    BtmDataAccessType type = BtmDataAccessType::kUnknown;
  };

  // We use a vector rather than a set of URLs because order can matter. If the
  // same URL appears twice in a redirect chain, we might be able to distinguish
  // between them.
  std::vector<CookieAccess> accesses_;
};

}  // namespace content

#endif  // CONTENT_BROWSER_BTM_COOKIE_ACCESS_FILTER_H_