/* 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 RedirectChannelRegistrar_h__
#define RedirectChannelRegistrar_h__

#include "nsIRedirectChannelRegistrar.h"

#include "nsIChannel.h"
#include "nsIParentChannel.h"
#include "nsInterfaceHashtable.h"
#include "mozilla/Attributes.h"
#include "mozilla/Mutex.h"

namespace mozilla {
namespace net {

class RedirectChannelRegistrar final : public nsIRedirectChannelRegistrar
{
  NS_DECL_ISUPPORTS
  NS_DECL_NSIREDIRECTCHANNELREGISTRAR

  RedirectChannelRegistrar();

private:
  ~RedirectChannelRegistrar() = default;

public:
  // Singleton accessor
  static already_AddRefed<nsIRedirectChannelRegistrar> GetOrCreate();

  // Cleanup the singleton instance on shutdown
  static void Shutdown();

protected:
  typedef nsInterfaceHashtable<nsUint32HashKey, nsIChannel>
          ChannelHashtable;
  typedef nsInterfaceHashtable<nsUint32HashKey, nsIParentChannel>
          ParentChannelHashtable;

  ChannelHashtable mRealChannels;
  ParentChannelHashtable mParentChannels;
  uint32_t mId;
  Mutex mLock;
};

} // namespace net
} // namespace mozilla

#endif