tests/testserver.py

@@ -0,0 +1,124 @@
+import time
+import json
+from typing import Literal
+from websockets.sync.server import serve, ServerConnection
+from websockets.exceptions import ConnectionClosedError
+from threading import Thread
+from pydantic import BaseModel, Field
+
+
+class MeetingPermissions(BaseModel):
+    canToggleMute: bool = False
+    canToggleVideo: bool = False
+    canToggleHand: bool = False
+    canToggleBlur: bool = False
+    canToggleRecord: bool = False
+    canLeave: bool = False
+    canReact: bool = False
+    canToggleShareTray: bool = False
+    canToggleChat: bool = False
+    canStopSharing: bool = False
+    canPair: bool = False
+
+
+class MeetingState(BaseModel):
+    isMuted: bool = False
+    isCameraOn: bool = False
+    isHandRaised: bool = False
+    isInMeeting: bool = False
+    isRecordingOn: bool = False
+    isBackgroundBlurred: bool = False
+    hasUnreadMessages: bool = False
+
+
+class MeetingUpdate(BaseModel):
+    meetingState: MeetingState = Field(default_factory=MeetingState)
+    meetingPermissions: MeetingPermissions = Field(default_factory=MeetingPermissions)
+
+
+class TeamsMsg(BaseModel):
+    apiVersion: Literal["1.0.0"] = "1.0.0"
+    meetingUpdate: MeetingUpdate = Field(default_factory=MeetingUpdate)
+
+
+class TeamsSim:
+    def __init__(self) -> None:
+        self.meeting = TeamsMsg()
+        self.rx_thread = Thread(target=self.rx_thread_run)
+        self.ws: ServerConnection | None = None
+        self.active: bool = True
+
+        # {
+        #     "apiVersion": "1.0.0",
+        #     "service": "toggle-mute",
+        #     "action": "toggle-mute",
+        #     "manufacturer": "Elgato",
+        #     "device": "NkTeamsCtrl",
+        #     "timestamp": time.time_ns() // 1000,
+        # }
+
+    def rx_thread_run(self):
+        assert self.ws is not None
+
+        while self.active:
+            try:
+                raw_msg = self.ws.recv()
+            except ConnectionClosedError:
+                self.ws = None
+                self.active = False
+                break
+            msg = json.loads(raw_msg)
+            print(f">{msg}")
+            match msg.get("action", ""):
+                case "toggle-mute":
+                    self.meeting.meetingUpdate.meetingState.isMuted ^= True
+                case "toggle-video":
+                    self.meeting.meetingUpdate.meetingState.isCameraOn ^= True
+                case "toggle-hand":
+                    self.meeting.meetingUpdate.meetingState.isHandRaised ^= True
+                case "leave-call":
+                    self.meeting.meetingUpdate.meetingState.isInMeeting = False
+                case _:
+                    continue
+            self.send_status()
+
+    def send_status(self) -> None:
+        if self.ws is None:
+            return
+        msg = self.meeting.json()
+        print(f"<{msg}")
+        self.ws.send(msg)
+
+    def handler(self, websocket: ServerConnection):
+        if self.rx_thread.is_alive():
+            self.active = False
+            if self.ws is not None:
+                self.ws.close_socket()
+            self.rx_thread.join()
+            self.rx_thread = Thread(target=self.rx_thread_run)
+        self.ws = websocket
+        self.active = True
+        self.rx_thread.start()
+
+        while True:
+            time.sleep(0.5)
+            if self.meeting.meetingUpdate.meetingState.isInMeeting is False:
+                print('### not in meeting, "reconnecting" in 5')
+                time.sleep(5)
+                self.meeting.meetingUpdate.meetingState.isInMeeting = True
+                self.send_status()
+            time.sleep(5)
+            self.meeting.meetingUpdate.meetingState.hasUnreadMessages = True
+            self.send_status()
+            time.sleep(5)
+            self.meeting.meetingUpdate.meetingState.hasUnreadMessages = False
+            self.send_status()
+
+    def run(self):
+        with serve(self.handler, host="localhost", port=8124) as server:
+            server.serve_forever()
+
+
+if __name__ == "__main__":
+    srv = TeamsSim()
+    srv.run()