/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef mozilla_UrlClassifierExceptionList_h
#define mozilla_UrlClassifierExceptionList_h

#include "nsHashKeys.h"
#include "nsTHashMap.h"
#include "nsIUrlClassifierExceptionList.h"
#include "nsISupports.h"
#include "nsTArray.h"
#include "nsString.h"

namespace mozilla::net {

/**
 * @see nsIUrlClassifierExceptionList
 */
class UrlClassifierExceptionList final : public nsIUrlClassifierExceptionList {
 public:
  NS_DECL_ISUPPORTS
  NS_DECL_NSIURLCLASSIFIEREXCEPTIONLIST

  UrlClassifierExceptionList() = default;

 private:
  ~UrlClassifierExceptionList() = default;

  // A list of exception entries
  using ExceptionEntryArray =
      nsTArray<RefPtr<nsIUrlClassifierExceptionListEntry>>;

  // A map from (schemeless) site to a list of exception entries.
  using SiteToEntries = nsTHashMap<nsCStringHashKey, ExceptionEntryArray>;

  // Helper method to check if any exception in the array matches the given
  // load.
  static bool ExceptionListMatchesLoad(ExceptionEntryArray* aExceptions,
                                       nsIURI* aURI, nsIURI* aTopLevelURI,
                                       bool aIsPrivateBrowsing);

  // Helper method to extract the schemeless site from a URL pattern.
  NS_IMETHODIMP GetSchemelessSiteFromUrlPattern(const nsACString& aUrlPattern,
                                                nsACString& aSite);

  // The feature this exception list is for, e.g. "tracking-protection".
  nsCString mFeature;

  // A two stage hash map to store the (top level) site-specific exception
  // entries.
  // The outer hash map key is the top level (schemeless) site.
  // The inner hash map key is the (schemeless) site of the load to be checked.
  nsTHashMap<nsCStringHashKey, SiteToEntries> mExceptions;

  // A map of exception list entries which apply across all top level sites.
  // The hash map key is the (schemeless) site of the load to be checked.
  nsTHashMap<nsCStringHashKey, ExceptionEntryArray> mGlobalExceptions;
};

}  // namespace mozilla::net

#endif