openreplay/ee/api/routers/subs/v1_api_ee.py

from fastapi import Depends, Body

import schemas
from chalicelib.utils import assist_helper
from chalicelib.core import projects
from or_dependencies import OR_context
from routers import core
from routers.base import get_routers

public_app, app, app_apikey = get_routers()


@app_apikey.get('/v1/assist/credentials', tags=["api"])
def get_assist_credentials():
    credentials = assist_helper.get_temporary_credentials()
    if "errors" in credentials:
        return credentials
    return {"data": credentials}


@app_apikey.get('/v1/{projectKey}/assist/sessions', tags=["api"])
def get_sessions_live(projectKey: str, userId: str = None, context: schemas.CurrentContext = Depends(OR_context)):
    projectId = projects.get_by_project_key(projectKey)
    if projectId is None:
        return {"errors": ["invalid projectKey"]}
    return core.get_sessions_live(projectId=projectId, userId=userId, context=context)


@app_apikey.post('/v1/{projectKey}/assist/sessions', tags=["api"])
def sessions_live(projectKey: str, data: schemas.LiveSessionsSearchPayloadSchema = Body(...),
                  context: schemas.CurrentContext = Depends(OR_context)):
    projectId = projects.get_by_project_key(projectKey)
    if projectId is None:
        return {"errors": ["invalid projectKey"]}
    return core.sessions_live(projectId=projectId, data=data, context=context)