// Copyright 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_RTC_QUIC_STREAM_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_RTC_QUIC_STREAM_H_

#include "third_party/blink/renderer/core/dom/context_lifecycle_observer.h"
#include "third_party/blink/renderer/modules/event_target_modules.h"
#include "third_party/blink/renderer/modules/modules_export.h"
#include "third_party/blink/renderer/modules/peerconnection/adapters/quic_stream_proxy.h"

namespace blink {

class RTCQuicTransport;

enum class RTCQuicStreamState { kNew, kOpening, kOpen, kClosing, kClosed };

class MODULES_EXPORT RTCQuicStream final : public EventTargetWithInlineData,
                                           public ContextClient,
                                           public QuicStreamProxy::Delegate {
  DEFINE_WRAPPERTYPEINFO();

 public:
  RTCQuicStream(ExecutionContext* context,
                RTCQuicTransport* transport,
                QuicStreamProxy* stream_proxy);
  ~RTCQuicStream() override;

  // Called from the RTCQuicTransport when it is being stopped.
  void Stop();
  bool IsClosed() const { return state_ == RTCQuicStreamState::kClosed; }

  // rtc_quic_stream.idl
  RTCQuicTransport* transport() const;
  String state() const;
  uint32_t readBufferedAmount() const;
  uint32_t writeBufferedAmount() const;
  void finish();
  void reset();
  DEFINE_ATTRIBUTE_EVENT_LISTENER(statechange);

  // EventTarget overrides.
  const AtomicString& InterfaceName() const override;
  ExecutionContext* GetExecutionContext() const override;

  // For garbage collection.
  void Trace(blink::Visitor* visitor) override;

 private:
  // Closes the stream. This will change the state to kClosed and deregister it
  // from the RTCQuicTransport. The QuicStreamProxy can no longer be used after
  // this point.
  void Close();

  // QuicStreamProxy::Delegate overrides.
  void OnRemoteReset() override;
  void OnRemoteFinish() override;

  Member<RTCQuicTransport> transport_;
  RTCQuicStreamState state_ = RTCQuicStreamState::kOpen;
  bool readable_ = true;
  bool writeable_ = true;
  uint32_t read_buffered_amount_ = 0;
  uint32_t write_buffered_amount_ = 0;
  QuicStreamProxy* proxy_;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_RTC_QUIC_STREAM_H_