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

#include "nsIBinaryHttp.h"
#include "nsIHttpChannel.h"
#include "nsIHttpChannelInternal.h"
#include "nsIObliviousHttp.h"
#include "nsIObliviousHttpChannel.h"
#include "nsIStreamListener.h"
#include "nsITimedChannel.h"
#include "nsIUploadChannel2.h"
#include "nsTHashMap.h"

namespace mozilla::net {

class ObliviousHttpChannel final : public nsIObliviousHttpChannel,
                                   public nsIHttpChannelInternal,
                                   public nsIStreamListener,
                                   public nsIUploadChannel2,
                                   public nsITimedChannel {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSICHANNEL
  NS_DECL_NSIHTTPCHANNEL
  NS_DECL_NSIOBLIVIOUSHTTPCHANNEL
  NS_DECL_NSIREQUESTOBSERVER
  NS_DECL_NSISTREAMLISTENER
  NS_DECL_NSIUPLOADCHANNEL2

  ObliviousHttpChannel(nsIURI* targetURI,
                       const nsTArray<uint8_t>& encodedConfig,
                       nsIHttpChannel* innerChannel);

  NS_FORWARD_SAFE_NSIREQUEST(mInnerChannel)
  NS_FORWARD_SAFE_NSIIDENTCHANNEL(mInnerChannel)
  NS_FORWARD_SAFE_NSIHTTPCHANNELINTERNAL(mInnerChannelInternal)
  NS_FORWARD_SAFE_NSITIMEDCHANNEL(mInnerChannelTimed)

 protected:
  ~ObliviousHttpChannel();

  nsresult ProcessOnStopRequest();
  void EmitOnDataAvailable(nsIStreamListener*);

  nsCOMPtr<nsIURI> mTargetURI;
  nsTArray<uint8_t> mEncodedConfig;

  nsCString mMethod{"GET"_ns};
  nsCString mContentType;
  nsTHashMap<nsCStringHashKey, nsCString> mHeaders;
  nsTArray<uint8_t> mContent;

  nsCOMPtr<nsIHttpChannel> mInnerChannel;
  nsCOMPtr<nsIHttpChannelInternal> mInnerChannelInternal;
  nsCOMPtr<nsITimedChannel> mInnerChannelTimed;
  nsCOMPtr<nsIObliviousHttpClientRequest> mEncapsulatedRequest;
  nsTArray<uint8_t> mRawResponse;
  nsCOMPtr<nsIBinaryHttpResponse> mBinaryHttpResponse;

  nsCOMPtr<nsIStreamListener> mStreamListener;
};

}  // namespace mozilla::net

#endif  // mozilla_net_ObliviousHttpChannel_h