// Copyright 2018 The Chromium Authors
// 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_ADAPTERS_ICE_TRANSPORT_ADAPTER_IMPL_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_ICE_TRANSPORT_ADAPTER_IMPL_H_

#include <vector>

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "third_party/blink/renderer/modules/peerconnection/adapters/ice_transport_adapter.h"
#include "third_party/webrtc/api/ice_transport_interface.h"

namespace blink {

// IceTransportAdapter implementation backed by the WebRTC PortAllocator /
// P2PTransportChannel.
class IceTransportAdapterImpl final : public IceTransportAdapter {
 public:
  // Create an IceTransportAdapter for an existing |ice_transport_channel|
  // object.
  IceTransportAdapterImpl(Delegate* delegate,
                          webrtc::scoped_refptr<webrtc::IceTransportInterface>
                              ice_transport_channel);

  ~IceTransportAdapterImpl() override;

  // IceTransportAdapter overrides.
  void StartGathering(
      const webrtc::IceParameters& local_parameters,
      const webrtc::ServerAddresses& stun_servers,
      const std::vector<webrtc::RelayServerConfig>& turn_servers,
      IceTransportPolicy policy) override;
  void Start(
      const webrtc::IceParameters& remote_parameters,
      webrtc::IceRole role,
      const Vector<webrtc::Candidate>& initial_remote_candidates) override;
  void HandleRemoteRestart(
      const webrtc::IceParameters& new_remote_parameters) override;
  void AddRemoteCandidate(const webrtc::Candidate& candidate) override;

 private:
  webrtc::IceTransportInternal* ice_transport_channel() {
    return ice_transport_channel_->internal();
  }
  void SetupIceTransportChannel();
  // Callbacks from P2PTransportChannel.
  void OnGatheringStateChanged(webrtc::IceTransportInternal* transport);
  void OnCandidateGathered(webrtc::IceTransportInternal* transport,
                           const webrtc::Candidate& candidate);
  void OnStateChanged(webrtc::IceTransportInternal* transport);
  void OnNetworkRouteChanged(
      std::optional<webrtc::NetworkRoute> new_network_route);
  void OnRoleConflict(webrtc::IceTransportInternal* transport);

  const raw_ptr<Delegate> delegate_;
  webrtc::scoped_refptr<webrtc::IceTransportInterface> ice_transport_channel_;
  base::WeakPtrFactory<IceTransportAdapterImpl> weak_factory_{this};
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_ICE_TRANSPORT_ADAPTER_IMPL_H_