chore: allow function based config

libp2p · Jun 14, 2024 · 1e052de · 1e052de
1 parent 03044e0
commit 1e052de
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/packages/transport-webrtc/src/private-to-public/listener.ts b/packages/transport-webrtc/src/private-to-public/listener.ts
@@ -34,7 +34,7 @@ export interface WebRTCDirectListenerInit {
   certificates?: TransportCertificate[]
   maxInboundStreams?: number
   dataChannel?: DataChannelOptions
-  rtcConfiguration?: RTCConfiguration
+  rtcConfiguration?: RTCConfiguration | (() => RTCConfiguration | Promise<RTCConfiguration>)
   useLibjuice?: boolean
 }
 

diff --git a/packages/transport-webrtc/src/private-to-public/transport.ts b/packages/transport-webrtc/src/private-to-public/transport.ts
@@ -49,7 +49,7 @@ export interface WebRTCMetrics {
 }
 
 export interface WebRTCTransportDirectInit {
-  rtcConfiguration?: RTCConfiguration
+  rtcConfiguration?: RTCConfiguration | (() => RTCConfiguration | Promise<RTCConfiguration>)
   dataChannel?: DataChannelOptions
   certificates?: TransportCertificate[]
   useLibjuice?: boolean

diff --git a/packages/transport-webrtc/src/private-to-public/utils/get-rtcpeerconnection.ts b/packages/transport-webrtc/src/private-to-public/utils/get-rtcpeerconnection.ts
@@ -78,7 +78,7 @@ export class DirectRTCPeerConnection extends RTCPeerConnection {
   }
 }
 
-export async function createDialerRTCPeerConnection (name: string, ufrag: string, rtcConfiguration?: RTCConfiguration, certificate?: TransportCertificate): Promise<DirectRTCPeerConnection> {
+export async function createDialerRTCPeerConnection (name: string, ufrag: string, rtcConfiguration?: RTCConfiguration | (() => RTCConfiguration | Promise<RTCConfiguration>), certificate?: TransportCertificate): Promise<DirectRTCPeerConnection> {
   if (certificate == null) {
     const keyPair = await crypto.subtle.generateKey({
       name: 'ECDSA',
@@ -90,14 +90,16 @@ export async function createDialerRTCPeerConnection (name: string, ufrag: string
     })
   }
 
+  const rtcConfig = typeof rtcConfiguration === 'function' ? await rtcConfiguration() : rtcConfiguration
+
   // https://github.com/libp2p/specs/blob/master/webrtc/webrtc-direct.md#browser-to-public-server
   const peerConnection = new PeerConnection(name, {
     disableFingerprintVerification: true,
     disableAutoNegotiation: true,
     certificatePemFile: certificate.pem,
     keyPemFile: certificate.privateKey,
     maxMessageSize: 16384,
-    iceServers: toLibdatachannelIceServers(rtcConfiguration?.iceServers) ?? DEFAULT_STUN_SERVERS
+    iceServers: toLibdatachannelIceServers(rtcConfig?.iceServers) ?? DEFAULT_STUN_SERVERS
   })
 
   return new DirectRTCPeerConnection({