feat: list active streams with pre-constructed URLs for various protocols

Author: rambo

src/rmmtxauthz/web/userdirect.py

@@ -1,12 +1,15 @@
 """APIs usable directly by the user with mTLS"""
 
+from typing import Dict, Any, Sequence
 import logging
 
 from fastapi import APIRouter, Depends, Request
 from libpvarki.middleware import MTLSHeader
 
 from ..db.user import User
 from ..schema.userdirect import UserCredentials
+from ..mediamtx import MediaMTXControl
+from ..config import RMMTXSettings
 
 LOGGER = logging.getLogger(__name__)
 
@@ -24,3 +27,16 @@ async def get_credentials(request: Request) -> UserCredentials:
     """Get my MediaMTX credentials"""
     user = await User.by_username(get_callsign(request))
     return UserCredentials(username=user.username, password=user.mtxpassword)
+
+
+@userrouter.get("/streams")
+async def get_streams(request: Request) -> Sequence[Dict[str, Any]]:
+    """Get streams"""
+    user = await User.by_username(get_callsign(request))
+    conf = RMMTXSettings.singleton()
+    if conf.mtx_address == "__REQUEST_HOSTNAME__":
+        LOGGER.warning("Setting RMMTX_MTX_ADDRESS from the request header")
+        conf.mtx_address = request.headers.get("host", "__REQUEST_HOSTNAME__:1234").split(":", 1)[0]
+        LOGGER.info("Setting RMMTX_MTX_ADDRESS is now: {}".format(conf.mtx_address))
+    streams = await MediaMTXControl.singleton().get_paths(insert_credentials=f"{user.username}:{user.mtxpassword}@")
+    return streams