2e9ba46f19
- Updated requirements.txt to use art-common@11aa056 with l2_server field - All routers now import UserContext from artdag_common - Removed duplicate UserContext from auth_service.py - dependencies.py sets l2_server from settings on user context Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
142 lines
3.7 KiB
Python
142 lines
3.7 KiB
Python
"""
|
|
FastAPI dependency injection container.
|
|
|
|
Provides shared resources and services to route handlers.
|
|
"""
|
|
|
|
from functools import lru_cache
|
|
from typing import Optional
|
|
import asyncio
|
|
|
|
from fastapi import Request, Depends, HTTPException
|
|
from jinja2 import Environment
|
|
|
|
from artdag_common.middleware.auth import UserContext, get_user_from_cookie, get_user_from_header
|
|
|
|
from .config import settings
|
|
|
|
|
|
# Lazy imports to avoid circular dependencies
|
|
_redis_client = None
|
|
_cache_manager = None
|
|
_database = None
|
|
|
|
|
|
def get_redis_client():
|
|
"""Get the Redis client singleton."""
|
|
global _redis_client
|
|
if _redis_client is None:
|
|
import redis
|
|
_redis_client = redis.from_url(settings.redis_url, decode_responses=True)
|
|
return _redis_client
|
|
|
|
|
|
def get_cache_manager():
|
|
"""Get the cache manager singleton."""
|
|
global _cache_manager
|
|
if _cache_manager is None:
|
|
from cache_manager import get_cache_manager as _get_cache_manager
|
|
_cache_manager = _get_cache_manager()
|
|
return _cache_manager
|
|
|
|
|
|
def get_database():
|
|
"""Get the database singleton."""
|
|
global _database
|
|
if _database is None:
|
|
import database
|
|
_database = database
|
|
return _database
|
|
|
|
|
|
def get_templates(request: Request) -> Environment:
|
|
"""Get the Jinja2 environment from app state."""
|
|
return request.app.state.templates
|
|
|
|
|
|
async def get_current_user(request: Request) -> Optional[UserContext]:
|
|
"""
|
|
Get the current user from request (cookie or header).
|
|
|
|
This is a permissive dependency - returns None if not authenticated.
|
|
Use require_auth for routes that require authentication.
|
|
"""
|
|
# Try header first (API clients)
|
|
ctx = get_user_from_header(request)
|
|
if ctx:
|
|
# Add l2_server from settings
|
|
ctx.l2_server = settings.l2_server
|
|
return ctx
|
|
|
|
# Fall back to cookie (browser)
|
|
ctx = get_user_from_cookie(request)
|
|
if ctx:
|
|
ctx.l2_server = settings.l2_server
|
|
return ctx
|
|
|
|
|
|
async def require_auth(request: Request) -> UserContext:
|
|
"""
|
|
Require authentication for a route.
|
|
|
|
Raises:
|
|
HTTPException 401 if not authenticated
|
|
HTTPException 302 redirect to login for HTML requests
|
|
"""
|
|
ctx = await get_current_user(request)
|
|
if ctx is None:
|
|
# Check if HTML request for redirect
|
|
accept = request.headers.get("accept", "")
|
|
if "text/html" in accept:
|
|
raise HTTPException(
|
|
status_code=302,
|
|
headers={"Location": "/login"}
|
|
)
|
|
raise HTTPException(status_code=401, detail="Authentication required")
|
|
return ctx
|
|
|
|
|
|
async def get_user_context_from_cookie(request: Request) -> Optional[UserContext]:
|
|
"""
|
|
Legacy compatibility: get user from cookie.
|
|
|
|
Validates token with L2 server if configured.
|
|
"""
|
|
ctx = get_user_from_cookie(request)
|
|
if ctx is None:
|
|
return None
|
|
|
|
# If L2 server configured, could validate token here
|
|
# For now, trust the cookie
|
|
return ctx
|
|
|
|
|
|
# Service dependencies (lazy loading)
|
|
|
|
def get_run_service():
|
|
"""Get the run service."""
|
|
from .services.run_service import RunService
|
|
return RunService(
|
|
database=get_database(),
|
|
redis=get_redis_client(),
|
|
cache=get_cache_manager(),
|
|
)
|
|
|
|
|
|
def get_recipe_service():
|
|
"""Get the recipe service."""
|
|
from .services.recipe_service import RecipeService
|
|
return RecipeService(
|
|
redis=get_redis_client(), # Kept for API compatibility, not used
|
|
cache=get_cache_manager(),
|
|
)
|
|
|
|
|
|
def get_cache_service():
|
|
"""Get the cache service."""
|
|
from .services.cache_service import CacheService
|
|
return CacheService(
|
|
cache_manager=get_cache_manager(),
|
|
database=get_database(),
|
|
)
|