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

#include "nsIDNSListener.h"
#include "nsIDNSRecord.h"
#include "nsProxyRelease.h"
#include "nsThreadUtils.h"

class nsIEventTarget;
class nsICancelable;

namespace mozilla {
namespace net {

#define DNS_LISTENER_PROXY_IID \
  {0x8f172ca3, 0x7a7f, 0x4941, {0xa7, 0x0b, 0xbc, 0x72, 0x80, 0x2e, 0x9d, 0x9b}}

class DNSListenerProxy final : public nsIDNSListener {
 public:
  DNSListenerProxy(nsIDNSListener* aListener, nsIEventTarget* aTargetThread)
      // We want to make sure that |aListener| is only accessed on the target
      // thread.
      : mListener(aListener),
        mTargetThread(aTargetThread),
        mListenerAddress(reinterpret_cast<uintptr_t>(aListener)) {
    MOZ_ASSERT(mListener);
    MOZ_ASSERT(mTargetThread);
  }

  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSIDNSLISTENER
  NS_INLINE_DECL_STATIC_IID(DNS_LISTENER_PROXY_IID)

  uintptr_t GetOriginalListenerAddress() const { return mListenerAddress; }

 private:
  ~DNSListenerProxy() {
    NS_ProxyRelease("DNSListenerProxy::mListener", mTargetThread,
                    mListener.forget());
  }

  nsCOMPtr<nsIDNSListener> mListener;
  nsCOMPtr<nsIEventTarget> mTargetThread;
  uintptr_t mListenerAddress;
};

}  // namespace net
}  // namespace mozilla

#endif  // DNSListenerProxy_h_