allow using the sdk with fishjam id

Author: czerwiukk

.tool-versions

@@ -0,0 +1,2 @@
+python 3.13.5
+poetry 2.1.3

examples/room_manager/room_service.py

@@ -24,14 +24,14 @@ class PeerAccess:
     room: Resource
     peer: Resource
     peer_token: str
-    url: str
 
 
 class RoomService:
     def __init__(self, args: Namespace, logger: Logger):
         self.fishjam_client = FishjamClient(
-            fishjam_url=args.fishjam_url,
+            fishjam_id=args.fishjam_id,
             management_token=args.management_token,
+            fishjam_url=args.fishjam_url,
         )
         self.url = args.fishjam_url.replace("http", "ws")
         self.room_name_to_room_id: dict[str, str] = {}
@@ -105,7 +105,6 @@ def __create_peer(self, room_name: str, peer_name: str) -> PeerAccess:
             room=Resource(id=room_id, name=room_name),
             peer=Resource(id=peer.id, name=peer_name),
             peer_token=token,
-            url=self.url,
         )
 
         self.peer_name_to_access[peer_name] = peer_access

fishjam/_ws_notifier.py

@@ -21,21 +21,26 @@
     ALLOWED_NOTIFICATIONS,
     AllowedNotification,
 )
+from fishjam.utils import get_fishjam_url
 
 
 class FishjamNotifier:
     """
     Allows for receiving WebSocket messages from Fishjam.
     """
 
-    def __init__(self, fishjam_url: str, management_token: str):
+    def __init__(
+        self,
+        fishjam_id: str,
+        management_token: str,
+        fishjam_url: str | None = None,
+    ):
         """
         Create FishjamNotifier instance, providing the fishjam url and management token.
         """
 
-        self._fishjam_url = (
-            f"{fishjam_url.replace('http', 'ws')}/socket/server/websocket"
-        )
+        websocket_url = get_fishjam_url(fishjam_id, fishjam_url).replace("http", "ws")
+        self._fishjam_url = f"{websocket_url}/socket/server/websocket"
         self._management_token: str = management_token
         self._websocket: client.WebSocketClientProtocol | None = None
         self._ready: bool = False

fishjam/api/_client.py

@@ -4,11 +4,17 @@
 from fishjam._openapi_client.models import Error
 from fishjam._openapi_client.types import Response
 from fishjam.errors import HTTPError
+from fishjam.utils import get_fishjam_url
 
 
 class Client:
-    def __init__(self, fishjam_url: str, management_token: str):
-        self.client = AuthenticatedClient(f"{fishjam_url}", token=management_token)
+    def __init__(
+        self, fishjam_id: str, management_token: str, *, fishjam_url: str | None = None
+    ):
+        self.client = AuthenticatedClient(
+            get_fishjam_url(fishjam_id, fishjam_url),
+            token=management_token,
+        )
 
     def _request(self, method, **kwargs):
         response = method.sync_detailed(client=self.client, **kwargs)

fishjam/api/_fishjam_client.py

@@ -82,11 +82,21 @@ class PeerOptions:
 class FishjamClient(Client):
     """Allows for managing rooms"""
 
-    def __init__(self, fishjam_url: str, management_token: str):
+    def __init__(
+        self,
+        fishjam_id: str,
+        management_token: str,
+        *,
+        fishjam_url: str | None = None,
+    ):
         """
         Create FishjamClient instance, providing the fishjam url and managment token.
         """
-        super().__init__(fishjam_url=fishjam_url, management_token=management_token)
+        super().__init__(
+            fishjam_id=fishjam_id,
+            management_token=management_token,
+            fishjam_url=fishjam_url,
+        )
 
     def create_peer(
         self, room_id: str, options: PeerOptions | None = None

fishjam/errors.py

@@ -61,3 +61,7 @@ class InternalServerError(HTTPError):
     def __init__(self, errors):
         """@private"""
         super().__init__(errors)
+
+
+class MissingFishjamIdError(Exception):
+    pass

fishjam/utils.py

@@ -0,0 +1,11 @@
+from fishjam.errors import MissingFishjamIdError
+
+
+def get_fishjam_url(fishjam_id: str | None, fishjam_url: str | None) -> str:
+    if not fishjam_url and not fishjam_id:
+        raise MissingFishjamIdError
+
+    if fishjam_url:
+        return fishjam_url
+
+    return f"https://fishjam.io/api/v1/connect/{fishjam_id}"