Replace aiortc with libdatachannel direct pipeline (#1083)

* fix stdin close error

* Refactor stream endpoint, fix encoder thread safety and improve tests

* fix and improve test

* remove not None

* use Enum

* use Enum and add todo

* remove poll

tests/stream_helper.py

@@ -1,21 +1,58 @@
-from aiortc import RTCPeerConnection, VideoStreamTrack
+import ctypes
+import os
+import threading
+import time
+from typing import Optional
 
-from facefusion.types import RtcOfferSet
+from starlette.testclient import TestClient
+
+from facefusion import rtc
+from facefusion.types import RtcSdpOffer
 
 
-async def create_rtc_offer() -> RtcOfferSet:
-	rtc_connection = RTCPeerConnection()
-	rtc_connection.addTrack(VideoStreamTrack())
-	rtc_offer = await rtc_connection.createOffer()
+def create_sdp_offer() -> Optional[RtcSdpOffer]:
+	rtc_library = rtc.create_static_rtc_library()
+	peer_connection = rtc.create_peer_connection(disable_auto_negotiation = True)
 
-	await rtc_connection.setLocalDescription(rtc_offer)
+	media_video = os.linesep.join(
+	[
+		'm=video 9 UDP/TLS/RTP/SAVPF 96',
+		'a=rtpmap:96 VP8/90000',
+		'a=recvonly',
+		'a=mid:0',
+		''
+	]).encode()
+	media_audio = os.linesep.join(
+	[
+		'm=audio 9 UDP/TLS/RTP/SAVPF 111',
+		'a=rtpmap:111 opus/48000/2',
+		'a=recvonly',
+		'a=mid:1',
+		''
+	]).encode()
 
-	rtc_offer_set : RtcOfferSet =\
-	{
-		'sdp': rtc_connection.localDescription.sdp,
-		'type': rtc_connection.localDescription.type
-	}
+	rtc_library.rtcAddTrack(peer_connection, media_video)
+	rtc_library.rtcAddTrack(peer_connection, media_audio)
+	rtc_library.rtcSetLocalDescription(peer_connection, b'offer')
 
-	await rtc_connection.close()
+	buffer_size = 16384
+	buffer_string = ctypes.create_string_buffer(buffer_size)
+	wait_limit = time.monotonic() + 5
 
-	return rtc_offer_set
+	while time.monotonic() < wait_limit:
+		if rtc_library.rtcGetLocalDescription(peer_connection, buffer_string, buffer_size) > 0:
+			sdp = buffer_string.value.decode()
+			rtc_library.rtcDeletePeerConnection(peer_connection)
+			return sdp
+		time.sleep(0.05)
+
+	rtc_library.rtcDeletePeerConnection(peer_connection)
+	return None
+
+
+def open_websocket_stream(test_client : TestClient, subprotocols : list[str], source_content : bytes, ready_event : threading.Event, stop_event : threading.Event) -> None:
+	with test_client.websocket_connect('/stream', subprotocols = subprotocols) as websocket:
+		websocket.send_bytes(source_content)
+		websocket.receive_text()
+		ready_event.set()
+		stop_event.wait()