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

#include <functional>

#include "mozilla/Mutex.h"
#include "nsISupportsImpl.h"
#include "nsTArray.h"

class WebTransportSessionEventListener;

namespace mozilla::net {

class WebTransportStreamBase;

class WebTransportSessionBase {
 public:
  NS_INLINE_DECL_PURE_VIRTUAL_REFCOUNTING

  WebTransportSessionBase() = default;

  void SetWebTransportSessionEventListener(
      WebTransportSessionEventListener* listener);

  virtual uint64_t GetStreamId() const = 0;
  virtual void CloseSession(uint32_t aStatus, const nsACString& aReason) = 0;
  virtual void GetMaxDatagramSize() = 0;
  virtual nsresult ExportKeyingMaterial(const nsTArray<uint8_t>& aLabel,
                                        const nsTArray<uint8_t>& aContext,
                                        nsTArray<uint8_t>& aKeyingMaterial) = 0;
  virtual void GetNegotiatedProtocol(nsACString& aProtocol) = 0;
  virtual void SendDatagram(nsTArray<uint8_t>&& aData, uint64_t aTrackingId,
                            uint64_t aSendGroupId, int64_t aSendOrder) = 0;
  virtual nsresult RegisterSendGroup(uint64_t aGroupId) = 0;
  virtual void CreateOutgoingBidirectionalStream(
      std::function<void(Result<RefPtr<WebTransportStreamBase>, nsresult>&&)>&&
          aCallback) = 0;
  virtual void CreateOutgoingUnidirectionalStream(
      std::function<void(Result<RefPtr<WebTransportStreamBase>, nsresult>&&)>&&
          aCallback) = 0;
  virtual void StartReading() {}

 protected:
  virtual ~WebTransportSessionBase() = default;

  already_AddRefed<WebTransportSessionEventListener> GetListener();
  already_AddRefed<WebTransportSessionEventListener> TakeListener();

  Mutex mListenerLock{"WebTransportSessionBase::mListenerLock"};
  RefPtr<WebTransportSessionEventListener> mListener
      MOZ_GUARDED_BY(mListenerLock);
};

}  // namespace mozilla::net

#endif  // mozilla_net_WebTransportSessionBase_h