// Copyright 2020 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_DIRECT_SOCKETS_TCP_SOCKET_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_DIRECT_SOCKETS_TCP_SOCKET_H_

#include "base/optional.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "mojo/public/cpp/system/data_pipe.h"
#include "services/network/public/mojom/tcp_socket.mojom-blink.h"
#include "third_party/blink/renderer/bindings/core/v8/script_promise_resolver.h"
#include "third_party/blink/renderer/modules/direct_sockets/tcp_readable_stream_wrapper.h"
#include "third_party/blink/renderer/modules/direct_sockets/tcp_writable_stream_wrapper.h"
#include "third_party/blink/renderer/modules/modules_export.h"
#include "third_party/blink/renderer/platform/bindings/script_wrappable.h"
#include "third_party/blink/renderer/platform/heap/heap.h"
#include "third_party/blink/renderer/platform/heap/member.h"
#include "third_party/blink/renderer/platform/scheduler/public/frame_or_worker_scheduler.h"

namespace net {
class IPEndPoint;
}  // namespace net

namespace blink {

class MODULES_EXPORT TCPSocket final
    : public ScriptWrappable,
      public network::mojom::blink::SocketObserver {
  DEFINE_WRAPPERTYPEINFO();

 public:
  explicit TCPSocket(ScriptPromiseResolver&);
  ~TCPSocket() override;

  TCPSocket(const TCPSocket&) = delete;
  TCPSocket& operator=(const TCPSocket&) = delete;

  // Called by NavigatorSocket when initiating a connection:
  mojo::PendingReceiver<network::mojom::blink::TCPConnectedSocket>
  GetTCPSocketReceiver();
  mojo::PendingRemote<network::mojom::blink::SocketObserver>
  GetTCPSocketObserver();
  void Init(int32_t result,
            const base::Optional<net::IPEndPoint>& local_addr,
            const base::Optional<net::IPEndPoint>& peer_addr,
            mojo::ScopedDataPipeConsumerHandle receive_stream,
            mojo::ScopedDataPipeProducerHandle send_stream);

  // Web-exposed function
  ScriptPromise close(ScriptState*, ExceptionState&);

  ReadableStream* readable() const;
  WritableStream* writable() const;
  String remoteAddress() const;
  uint16_t remotePort() const;

  // network::mojom::blink::SocketObserver:
  void OnReadError(int32_t net_error) override;
  void OnWriteError(int32_t net_error) override;

  // ScriptWrappable:
  void Trace(Visitor* visitor) const override;

 private:
  void OnSocketObserverConnectionError();

  void OnReadableStreamAbort();
  void OnWritableStreamAbort();

  void DoClose(bool is_local_close);
  void ResetReadableStream();
  void ResetWritableStream();

  Member<ScriptPromiseResolver> resolver_;
  FrameOrWorkerScheduler::SchedulingAffectingFeatureHandle
      feature_handle_for_scheduler_;

  mojo::Remote<network::mojom::blink::TCPConnectedSocket> tcp_socket_;
  mojo::Receiver<network::mojom::blink::SocketObserver>
      socket_observer_receiver_{this};

  Member<TCPReadableStreamWrapper> tcp_readable_stream_wrapper_;
  Member<TCPWritableStreamWrapper> tcp_writable_stream_wrapper_;
  base::Optional<net::IPEndPoint> local_addr_;
  base::Optional<net::IPEndPoint> peer_addr_;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_DIRECT_SOCKETS_TCP_SOCKET_H_