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

#include "nsCOMPtr.h"
#include "nsIStreamListener.h"
#include "nsIRequestObserver.h"
#include "nsIThreadRetargetableStreamListener.h"
#include "mozilla/Attributes.h"

namespace mozilla {
namespace net {

// Wrapper class to make replacement of nsHttpChannel's listener
// from JavaScript possible. It is workaround for bug 433711 and 682305.
class nsStreamListenerWrapper final : public nsIStreamListener
                                    , public nsIThreadRetargetableStreamListener
{
public:
  explicit nsStreamListenerWrapper(nsIStreamListener *listener)
    : mListener(listener)
  {
    MOZ_ASSERT(mListener, "no stream listener specified");
  }

  NS_DECL_THREADSAFE_ISUPPORTS
  NS_FORWARD_SAFE_NSIREQUESTOBSERVER(mListener)
  NS_FORWARD_SAFE_NSISTREAMLISTENER(mListener)
  NS_DECL_NSITHREADRETARGETABLESTREAMLISTENER

private:
  ~nsStreamListenerWrapper() {}
  nsCOMPtr<nsIStreamListener> mListener;
};

} // namespace net
} // namespace mozilla

#endif // nsStreamListenerWrapper_h__