/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- */

/* vim: set ts=8 sts=4 et sw=4 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 DomainPolicy_h__

#define DomainPolicy_h__

#include "nsIDomainPolicy.h"

#include "nsTHashtable.h"

#include "nsURIHashKey.h"

namespace mozilla {

namespace ipc {

class URIParams;

} // namespace ipc

enum DomainSetChangeType{

    ACTIVATE_POLICY,

    DEACTIVATE_POLICY,

    ADD_DOMAIN,

    REMOVE_DOMAIN,

    CLEAR_DOMAINS

};

enum DomainSetType{

    NO_TYPE,

    BLACKLIST,

    SUPER_BLACKLIST,

    WHITELIST,

    SUPER_WHITELIST

};

class DomainSet final : public nsIDomainSet

{

public:

    NS_DECL_ISUPPORTS

    NS_DECL_NSIDOMAINSET

    explicit DomainSet(DomainSetType aType)

        : mType(aType)

    {}

    void CloneSet(InfallibleTArray<mozilla::ipc::URIParams>* aDomains);

protected:

    virtual ~DomainSet() {}

    nsTHashtable<nsURIHashKey> mHashTable;

    DomainSetType mType;

};

class DomainPolicy final : public nsIDomainPolicy

{

public:

    NS_DECL_ISUPPORTS

    NS_DECL_NSIDOMAINPOLICY

    DomainPolicy();

private:

    virtual ~DomainPolicy();

    RefPtr<DomainSet> mBlacklist;

    RefPtr<DomainSet> mSuperBlacklist;

    RefPtr<DomainSet> mWhitelist;

    RefPtr<DomainSet> mSuperWhitelist;

};

} /* namespace mozilla */

#endif /* DomainPolicy_h__ */