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

#include "nsIAsyncInputStream.h"
#include "nsIAsyncOutputStream.h"
#include "nsIWebTransportStream.h"
#include "nsCOMPtr.h"
#include "WebTransportStreamBase.h"

namespace mozilla::net {

class WebTransportStreamProxy final
    : public nsIWebTransportReceiveStream,
      public nsIWebTransportSendStream,
      public nsIWebTransportBidirectionalStream {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS

  explicit WebTransportStreamProxy(WebTransportStreamBase* aStream);

  NS_IMETHOD SendStopSending(uint8_t aError) override;
  NS_IMETHOD SendFin() override;
  NS_IMETHOD Reset(uint8_t aErrorCode) override;
  NS_IMETHOD GetSendStreamStats(
      nsIWebTransportStreamStatsCallback* aCallback) override;
  NS_IMETHOD GetReceiveStreamStats(
      nsIWebTransportStreamStatsCallback* aCallback) override;

  NS_IMETHOD GetHasReceivedFIN(bool* aHasReceivedFIN) override;

  NS_IMETHOD GetInputStream(nsIAsyncInputStream** aOut) override;
  NS_IMETHOD GetOutputStream(nsIAsyncOutputStream** aOut) override;

  NS_IMETHOD GetStreamId(uint64_t* aId) override;
  NS_IMETHOD SetSendOrder(Maybe<int64_t> aSendOrder) override;

 private:
  virtual ~WebTransportStreamProxy();

  class AsyncInputStreamWrapper : public nsIAsyncInputStream {
   public:
    NS_DECL_THREADSAFE_ISUPPORTS
    NS_DECL_NSIINPUTSTREAM
    NS_DECL_NSIASYNCINPUTSTREAM

    AsyncInputStreamWrapper(nsIAsyncInputStream* aStream,
                            WebTransportStreamBase* aWebTransportStream);

   private:
    virtual ~AsyncInputStreamWrapper();
    void MaybeCloseStream();

    nsCOMPtr<nsIAsyncInputStream> mStream;
    RefPtr<WebTransportStreamBase> mWebTransportStream;
  };

  class AsyncOutputStreamWrapper : public nsIAsyncOutputStream {
   public:
    NS_DECL_THREADSAFE_ISUPPORTS
    NS_DECL_NSIOUTPUTSTREAM
    NS_DECL_NSIASYNCOUTPUTSTREAM

    explicit AsyncOutputStreamWrapper(nsIAsyncOutputStream* aStream);

   private:
    virtual ~AsyncOutputStreamWrapper();

    nsCOMPtr<nsIAsyncOutputStream> mStream;
  };

  RefPtr<WebTransportStreamBase> mWebTransportStream;
  RefPtr<AsyncOutputStreamWrapper> mWriter;
  RefPtr<AsyncInputStreamWrapper> mReader;
};

}  // namespace mozilla::net

#endif