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

/* 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 nsURIHashKey_h__

#define nsURIHashKey_h__

#include "PLDHashTable.h"

#include "nsCOMPtr.h"

#include "nsIURI.h"

#include "nsHashKeys.h"

#include "mozilla/Move.h"

#include "mozilla/Unused.h"

/**

 * Hashtable key class to use with nsTHashtable/nsBaseHashtable

 */

class nsURIHashKey : public PLDHashEntryHdr

{

public:

    typedef nsIURI* KeyType;

    typedef const nsIURI* KeyTypePointer;

    explicit nsURIHashKey(const nsIURI* aKey) :

        mKey(const_cast<nsIURI*>(aKey)) { MOZ_COUNT_CTOR(nsURIHashKey); }

    nsURIHashKey(nsURIHashKey&& toMove)

        : PLDHashEntryHdr(std::move(toMove))

        , mKey(std::move(toMove.mKey))

    {

        MOZ_COUNT_CTOR(nsURIHashKey);

    }

    ~nsURIHashKey() { MOZ_COUNT_DTOR(nsURIHashKey); }

    nsIURI* GetKey() const { return mKey; }

    bool KeyEquals(const nsIURI* aKey) const {

        bool eq;

        if (!mKey) {

            return !aKey;

        }

        if (NS_SUCCEEDED(mKey->Equals(const_cast<nsIURI*>(aKey), &eq))) {

            return eq;

        }

        return false;

    }

    static const nsIURI* KeyToPointer(nsIURI* aKey) { return aKey; }

    static PLDHashNumber HashKey(const nsIURI* aKey) {

        if (!aKey) {

            // If the key is null, return hash for empty string.

            return mozilla::HashString(EmptyCString());

        }

        nsAutoCString spec;

        // If GetSpec() fails, ignoring the failure and proceeding with an

        // empty |spec| seems like the best thing to do.

        mozilla::Unused << const_cast<nsIURI*>(aKey)->GetSpec(spec);

        return mozilla::HashString(spec);

    }

    enum { ALLOW_MEMMOVE = true };

protected:

    nsCOMPtr<nsIURI> mKey;

};

#endif // nsURIHashKey_h__