BIG BIG BIG BIG BIG UPDATES

Co-authored-by: Copilot <[email protected]>

backend/app/api/editor/router.py

@@ -0,0 +1,62 @@
+from core.security import (
+    InvalidTokenError,
+    authenticate_token,
+    require_authenticated_user,
+)
+from fastapi import APIRouter, Depends, HTTPException, WebSocket, status
+from services.editor import python_lsp
+from starlette.websockets import WebSocketDisconnect
+
+router = APIRouter(prefix="/proxy/editor/python", tags=["Editor"])
+
+
+@router.get("/session", dependencies=[Depends(require_authenticated_user)])
+async def python_editor_session():
+    if python_lsp.lsp_binary() is None:
+        raise HTTPException(
+            status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
+            detail="basedpyright-langserver is not installed",
+        )
+
+    session = python_lsp.get_session()
+    return {
+        "root_uri": session.root_uri,
+        "file_uri": session.file_uri,
+    }
+
+
+async def authenticate_editor_socket(websocket: WebSocket) -> None:
+    token = websocket.query_params.get("token")
+    if token is None:
+        authorization = websocket.headers.get("authorization", "")
+        scheme, _, value = authorization.partition(" ")
+        if scheme.lower() == "bearer" and value:
+            token = value
+
+    if not token:
+        raise InvalidTokenError
+
+    await authenticate_token(token)
+
+
+@router.websocket("/ws")
+async def python_editor_socket(websocket: WebSocket):
+    try:
+        await authenticate_editor_socket(websocket)
+    except InvalidTokenError:
+        await websocket.close(code=4401)
+        return
+
+    if python_lsp.lsp_binary() is None:
+        await websocket.close(code=1011, reason="basedpyright unavailable")
+        return
+
+    session = python_lsp.get_session()
+    await websocket.accept()
+
+    try:
+        await python_lsp.bridge(websocket, session)
+    except WebSocketDisconnect:
+        return
+    except RuntimeError:
+        await websocket.close(code=1011, reason="language server failed")