diff options
Diffstat (limited to 'packages/meshbay-node/src/meshbay_node/hub_client.py')
| -rw-r--r-- | packages/meshbay-node/src/meshbay_node/hub_client.py | 19 |
1 files changed, 16 insertions, 3 deletions
diff --git a/packages/meshbay-node/src/meshbay_node/hub_client.py b/packages/meshbay-node/src/meshbay_node/hub_client.py index b345187..92c39db 100644 --- a/packages/meshbay-node/src/meshbay_node/hub_client.py +++ b/packages/meshbay-node/src/meshbay_node/hub_client.py @@ -229,12 +229,24 @@ class HubClient: except Exception: pass + async def update_ws_groups(self, group_ids: list[str]) -> None: + """Tell the hub about changed group list without dropping the connection.""" + ws = self._ws + if ws: + try: + await ws.send(json.dumps({ + "type": "update_groups", + "group_ids": group_ids, + })) + except Exception: + pass + async def maintain_ws( self, on_incoming: Any = None, on_revocation: Any = None, on_webrtc_offer: Any = None, - group_ids: list[str] | None = None, + group_ids: list[str] | None = None, # static list or callable returning one ) -> None: """ Maintain a persistent WebSocket connection to the hub. @@ -267,8 +279,9 @@ class HubClient: "token": self._session.access_token, "node_id": self._session.node_id, } - if group_ids: - auth_msg["group_ids"] = group_ids + gids = group_ids() if callable(group_ids) else group_ids + if gids: + auth_msg["group_ids"] = gids await ws.send(json.dumps(auth_msg)) # Bounded: a hub that accepts the socket and then says nothing # — which is what it does for a few seconds while restarting — |