refactor: integrate CoderAPI into authentication and workspace management

- Added a new dependency function to provide a CoderAPI instance for use in routers.
- Updated auth_router to utilize CoderAPI in the callback endpoint, enhancing authentication flow.
- Modified workspace_router to incorporate CoderAPI in workspace state retrieval and management functions, improving workspace operations.
- Streamlined dependency management across routers for better modularity and code organization.

Author: atyrode

src/backend/dependencies.py

@@ -2,10 +2,11 @@
 from typing import Optional, Dict, Any
 from uuid import UUID
 
-from fastapi import Request, HTTPException
+from fastapi import Request, HTTPException, Depends
 
 from config import get_session, is_token_expired, refresh_token
 from database.service import UserService
+from coder import CoderAPI
 
 class UserSession:
     """
@@ -130,3 +131,9 @@ def _handle_auth_error(self, detail: str, status_code: int = 401) -> Optional[No
 require_auth = AuthDependency(auto_error=True)
 optional_auth = AuthDependency(auto_error=False)
 require_admin = AuthDependency(auto_error=True, require_admin=True)
+
+def get_coder_api():
+    """
+    Dependency that provides a CoderAPI instance.
+    """
+    return CoderAPI()

src/backend/routers/auth_router.py

@@ -1,15 +1,15 @@
 import secrets
 import jwt
 import httpx
-from fastapi import APIRouter, Request, HTTPException
+from fastapi import APIRouter, Request, HTTPException, Depends
 from fastapi.responses import RedirectResponse, FileResponse, JSONResponse
 import os
 
 from config import get_auth_url, get_token_url, OIDC_CONFIG, set_session, delete_session, STATIC_DIR, get_session
+from dependencies import get_coder_api
 from coder import CoderAPI
 
 auth_router = APIRouter()
-coder_api = CoderAPI()
 
 @auth_router.get("/login")
 async def login(request: Request, kc_idp_hint: str = None, popup: str = None):
@@ -31,7 +31,12 @@ async def login(request: Request, kc_idp_hint: str = None, popup: str = None):
     return response
 
 @auth_router.get("/callback")
-async def callback(request: Request, code: str, state: str = "default"):
+async def callback(
+    request: Request, 
+    code: str, 
+    state: str = "default",
+    coder_api: CoderAPI = Depends(get_coder_api)
+):
     session_id = request.cookies.get('session_id')
     if not session_id:
         raise HTTPException(status_code=400, detail="No session")

src/backend/routers/workspace_router.py

@@ -4,11 +4,10 @@
 from fastapi import APIRouter, Depends, HTTPException
 from fastapi.responses import JSONResponse
 
-from dependencies import UserSession, require_auth
+from dependencies import UserSession, require_auth, get_coder_api
 from coder import CoderAPI
 
 workspace_router = APIRouter()
-coder_api = CoderAPI()
 
 class WorkspaceState(BaseModel):
     id: str
@@ -19,7 +18,10 @@ class WorkspaceState(BaseModel):
     agent: str
 
 @workspace_router.get("/state", response_model=WorkspaceState)
-async def get_workspace_state(user: UserSession = Depends(require_auth)):
+async def get_workspace_state(
+    user: UserSession = Depends(require_auth),
+    coder_api: CoderAPI = Depends(get_coder_api)
+):
     """
     Get the current state of the user's workspace
     """
@@ -52,12 +54,15 @@ async def get_workspace_state(user: UserSession = Depends(require_auth)):
 
 
 @workspace_router.post("/start")
-async def start_workspace(user: UserSession = Depends(require_auth)):
+async def start_workspace(
+    user: UserSession = Depends(require_auth),
+    coder_api: CoderAPI = Depends(get_coder_api)
+):
     """
     Start a workspace for the authenticated user
     """
 
-    workspace: WorkspaceState = await get_workspace_state(user)
+    workspace: WorkspaceState = await get_workspace_state(user, coder_api)
 
     try:
         response = coder_api.start_workspace(workspace.id)
@@ -67,15 +72,18 @@ async def start_workspace(user: UserSession = Depends(require_auth)):
 
 
 @workspace_router.post("/stop")
-async def stop_workspace(user: UserSession = Depends(require_auth)):
+async def stop_workspace(
+    user: UserSession = Depends(require_auth),
+    coder_api: CoderAPI = Depends(get_coder_api)
+):
     """
     Stop a workspace for the authenticated user
     """
 
-    workspace: WorkspaceState = await get_workspace_state(user)
+    workspace: WorkspaceState = await get_workspace_state(user, coder_api)
 
     try:
         response = coder_api.stop_workspace(workspace.id)
         return JSONResponse(content=response)
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))