add: demo server and client

2024-04-06 20:45:15 -07:00
parent 5a149bc922
commit 72effb5b2a
100 changed files with 11603 additions and 62 deletions
--- a/demo_server/demo_client/demo.py
+++ b/demo_server/demo_client/demo.py
@@ -0,0 +1,112 @@
+import time
+import asyncio
+from multiprocessing import Process
+from aiortc import RTCIceCandidate, RTCPeerConnection, RTCSessionDescription
+from openapi_client.rest import ApiException
+from pprint import pprint
+from aiortc import RTCPeerConnection, RTCSessionDescription
+from aiortc.sdp import candidate_from_sdp, candidate_to_sdp
+import uuid
+import openapi_client
+
+# Defining the host is optional and defaults to http://127.0.0.1:8080
+# See configuration.py for a list of all supported configuration parameters.
+configuration = openapi_client.Configuration(
+    host = "http://127.0.0.1:8080"
+)
+
+server_name="myserver"
+service_name="http1"
+
+
+
+def serve(peernet, knock, server_name):
+
+    async def server_internal():
+        pc = RTCPeerConnection()
+
+        # Define our handlers
+        done = False
+        session = "%s" % uuid.uuid4()
+
+        @pc.on("datachannel")
+        def on_datachannel(channel):
+            @channel.on("message")
+            def on_message(message):
+                # Process the message, and send a response
+                channel.send("pong" + message[4:])
+        
+        @pc.on("connectionstatechange")
+        async def on_connectionstatechange():
+            if pc.connectionState == "failed":
+                done = True
+                await pc.close()
+        
+        @pc.on("icecandidate")
+        async def on_icecandidate(candidate):
+            peernet.peer_net_service_create_ice_candidatge(session=knock.offer.name, ice_candidate={
+                "candidate": candidate_to_sdp(obj),
+                "sdpMid": obj.sdpMid,
+                "sdpLineIndex": obj.sdpMLineIndex,
+            })
+        
+        
+        # Add the remote offer
+        offer = RTCSessionDescription(sdp=knock.offer.sdp, type="offer")
+        await pc.setRemoteDescription(offer)
+        # Create an answer
+        answer = await pc.createAnswer()
+        await pc.setLocalDescription(answer)
+        # Send the answer
+        session_id = "%s" % uuid.uuid4()
+        knock.answer = {
+            "name": session_id,
+            "sdp": answer.sdp,
+            "sdpType": answer.type,
+        }
+        peernet.peer_net_service_update_knock(server_name, service_name, knock.name, knock2=knock)
+
+        while not done:
+            resp = peernet.peer_net_service_claim_ice_candidates(session_id)
+            if resp.ice_candidates != None:
+                time.sleep(1)
+                continue
+            for candidate in resp.ice_candidates:
+                ice_candidate = candidate_from_sdp(candidate.candidate)
+                ice_candidate.sdpMid = candidate.sdpMid
+                ice_candidate.sdpMLineIndex = candidate.sdpLineIndex
+                await pc.addIceCandidate(candidate)
+    coro = server_internal()
+
+    # run event loop
+    loop = asyncio.get_event_loop()
+    try:
+        loop.run_until_complete(coro)
+    except KeyboardInterrupt:
+        pass
+
+# Enter a context with an instance of the API client
+with openapi_client.ApiClient(configuration) as api_client:
+    # Create an instance of the API class
+    peernet = openapi_client.PeerNetServiceApi(api_client)
+
+    resp = peernet.peer_net_service_create_server(server={
+        "name": server_name,
+        "rooms": ["room1"],
+        "services": [
+            {
+                "name": service_name,
+                "protocol": "http",
+                "version": "1.0",
+            },
+        ],
+    })
+
+    while True:
+        print("Checking for knocks")
+        resp = peernet.peer_net_service_list_knocks(server=server_name, service=service_name)
+        for knock in resp.knocks:
+            print("Processing %s", knock)
+            p = Process(target=serve, args=(peernet, knock, server_name))
+            p.start()
+        time.sleep(5)