feat(chalice): EFS sessions (#651)

api/.gitignore

@@ -174,4 +174,5 @@ logs*.txt
 SUBNETS.json
 
 ./chalicelib/.configs
-README/*
+README/*
+.local

api/app.py

@@ -4,6 +4,7 @@ from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from decouple import config
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.middleware.gzip import GZipMiddleware
 from starlette.responses import StreamingResponse
 
 from chalicelib.utils import helper
@@ -14,7 +15,7 @@ from routers.crons import core_dynamic_crons
 from routers.subs import dashboard, insights, metrics, v1_api
 
 app = FastAPI(root_path="/api", docs_url=config("docs_url", default=""), redoc_url=config("redoc_url", default=""))
-
+app.add_middleware(GZipMiddleware, minimum_size=1000)
 
 @app.middleware('http')
 async def or_middleware(request: Request, call_next):

api/chalicelib/core/assist.py

@@ -1,6 +1,6 @@
 import requests
 from decouple import config
-
+from os.path import exists
 import schemas
 from chalicelib.core import projects
 
@@ -158,3 +158,11 @@ def autocomplete(project_id, q: str, key: str = None):
 def get_ice_servers():
     return config("iceServers") if config("iceServers", default=None) is not None \
                                    and len(config("iceServers")) > 0 else None
+
+
+def get_raw_mob_by_id(project_id, session_id):
+    path_to_file = config("FS_DIR") + "/" + str(session_id)
+
+    if exists(path_to_file):
+        return path_to_file
+    return None

api/env.default

@@ -47,4 +47,5 @@ sessions_region=us-east-1
 sourcemaps_bucket=sourcemaps
 sourcemaps_reader=http://127.0.0.1:9000/sourcemaps
 stage=default-foss
-version_number=1.4.0
+version_number=1.4.0
+FS_DIR=/mnt/efs

api/routers/core.py

@@ -2,6 +2,7 @@ from typing import Union, Optional
 
 from decouple import config
 from fastapi import Depends, Body, BackgroundTasks, HTTPException
+from fastapi.responses import FileResponse
 from starlette import status
 
 import schemas
@@ -183,8 +184,8 @@ def session_top_filter_values(projectId: int, context: schemas.CurrentContext =
 @app.get('/{projectId}/integrations', tags=["integrations"])
 def get_integrations_status(projectId: int, context: schemas.CurrentContext = Depends(OR_context)):
     data = integrations_global.get_global_integrations_status(tenant_id=context.tenant_id,
-                                                               user_id=context.user_id,
-                                                               project_id=projectId)
+                                                              user_id=context.user_id,
+                                                              project_id=projectId)
     return {"data": data}
 
 
@@ -895,6 +896,16 @@ def get_live_session(projectId: int, sessionId: str, background_tasks: Backgroun
     return {'data': data}
 
 
+@app.get('/{projectId}/assist/sessions/{sessionId}/replay', tags=["assist"])
+def get_live_session_replay_file(projectId: int, sessionId: str,
+                                 context: schemas.CurrentContext = Depends(OR_context)):
+    path = assist.get_raw_mob_by_id(project_id=projectId, session_id=sessionId)
+    if path is None:
+        return {"errors": ["Replay file not found"]}
+
+    return FileResponse(path=path, media_type="application/octet-stream")
+
+
 @app.post('/{projectId}/heatmaps/url', tags=["heatmaps"])
 def get_heatmaps_by_url(projectId: int, data: schemas.GetHeatmapPayloadSchema = Body(...),
                         context: schemas.CurrentContext = Depends(OR_context)):

ee/api/app.py

@@ -5,18 +5,20 @@ from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from decouple import config
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.middleware.gzip import GZipMiddleware
 from starlette import status
 from starlette.responses import StreamingResponse, JSONResponse
 
 from chalicelib.utils import helper
 from chalicelib.utils import pg_client
 from routers import core, core_dynamic, ee, saml
-from routers.subs import v1_api
 from routers.crons import core_crons
 from routers.crons import core_dynamic_crons
 from routers.subs import dashboard, insights, metrics, v1_api_ee
+from routers.subs import v1_api
 
 app = FastAPI(root_path="/api", docs_url=config("docs_url", default=""), redoc_url=config("redoc_url", default=""))
+app.add_middleware(GZipMiddleware, minimum_size=1000)
 
 
 @app.middleware('http')

ee/api/env.default

@@ -57,3 +57,4 @@ sourcemaps_bucket=sourcemaps
 sourcemaps_reader=http://127.0.0.1:9000/sourcemaps
 stage=default-ee
 version_number=1.0.0
+FS_DIR=/mnt/efs