"""FastAPI dependencies: auth, db, tenant context, RBAC.""" from __future__ import annotations import logging import uuid from typing import Any import redis.asyncio as aioredis from fastapi import Depends, HTTPException, Request, status from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncSession from app.config import get_settings from app.core.auth import get_redis, get_session_data, refresh_session_ttl from app.core.db import get_db, set_tenant_context, set_user_context from app.models.guest_user import GuestUser logger = logging.getLogger(__name__) # Known write-permission modules — used by require_write() to check # specific permissions instead of broad wildcards like *:write _WRITE_PERMISSIONS = [ "contacts:write", "contacts:create", "users:write", "roles:write", "audit:write", "attachments:write", "workflows:write", "sequences:write", "addresses:write", "taxes:write", "currencies:write", "notifications:write", "import_export:write", "user_preferences:write", ] async def get_redis_dep() -> aioredis.Redis: """FastAPI dependency for Redis client.""" return get_redis() async def get_current_guest( request: Request, redis: aioredis.Redis = Depends(get_redis_dep), ) -> dict[str, Any]: """Get the current guest user from guest session cookie. Returns session data dict with guest_user_id, tenant_id, email, name. Used for guest-specific endpoints (guest login, guest contacts). """ session_id = request.cookies.get("guest_session") if not session_id: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail={"detail": "Not authenticated", "code": "not_authenticated"}, ) import json raw = await redis.get(f"guest_session:{session_id}") if raw is None: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail={"detail": "Session expired or invalid", "code": "session_invalid"}, ) session_data = json.loads(raw) # Extend TTL on each request (sliding session) await redis.expire(f"guest_session:{session_id}", 1800) return session_data async def get_current_user( request: Request, db: AsyncSession = Depends(get_db), redis: aioredis.Redis = Depends(get_redis_dep), ) -> dict[str, Any]: """Get the current authenticated user from session cookie. Returns session data dict with user_id, tenant_id, email, name, role, and resolved permissions from Redis cache. """ settings = get_settings() session_id = request.cookies.get(settings.session_cookie_name) if not session_id: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail={"detail": "Not authenticated", "code": "not_authenticated"}, ) session_data = await get_session_data(redis, session_id) if session_data is None: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail={"detail": "Session expired or invalid", "code": "session_invalid"}, ) # Sliding session: extend TTL on each authenticated request await refresh_session_ttl(redis, session_id) if not session_data.get("is_active", True): raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail={"detail": "User account deactivated", "code": "user_inactive"}, ) # Set RLS tenant context tenant_id = uuid.UUID(session_data["tenant_id"]) await set_tenant_context(db, tenant_id) # Set RLS user context for row-level security user_id = uuid.UUID(session_data["user_id"]) from app.models.group import UserGroup groups_q = await db.execute( select(UserGroup.group_id) .where(UserGroup.user_id == user_id) .where(UserGroup.tenant_id == tenant_id) ) group_ids = [row[0] for row in groups_q] is_admin = session_data.get("is_system_admin", False) await set_user_context(db, user_id, group_ids, is_admin) # Check membership status (P1.7: suspended membership should not be usable) from app.models.user import UserTenant membership_q = await db.execute( select(UserTenant.status) .where(UserTenant.user_id == user_id) .where(UserTenant.tenant_id == tenant_id) ) membership_status = membership_q.scalar_one_or_none() if membership_status is not None and membership_status != "active": raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={"detail": f"Mitgliedschaft ist {membership_status}, Zugriff verweigert", "code": "membership_suspended"}, ) # Load resolved permissions from cache (or DB on miss) from app.core.permissions import get_cached_permissions user_id = uuid.UUID(session_data["user_id"]) resolved = await get_cached_permissions(db, redis, user_id, tenant_id) session_data["permissions"] = resolved.get("permissions", []) session_data["denied_permissions"] = resolved.get("denied", []) session_data["field_permissions"] = resolved.get("field_permissions", {}) session_data["is_system_admin"] = resolved.get("is_system_admin", False) return session_data async def get_current_user_bearer( request: Request, db: AsyncSession = Depends(get_db), ) -> dict[str, Any]: """Get the current user from a Bearer API token. Alternative to session-based auth for programmatic access (MCP, API clients). Returns the same dict shape as get_current_user. """ auth_header = request.headers.get("Authorization", "") if not auth_header.startswith("Bearer "): raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail={"detail": "Bearer token required", "code": "not_authenticated"}, ) token = auth_header[7:] # Strip "Bearer " from app.core.api_token import verify_api_token user_data = await verify_api_token(db, token) if user_data is None: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail={"detail": "Invalid or expired token", "code": "token_invalid"}, ) # Set RLS tenant context tenant_id = uuid.UUID(user_data["tenant_id"]) await set_tenant_context(db, tenant_id) # Set RLS user context user_id = uuid.UUID(user_data["user_id"]) from app.models.group import UserGroup groups_q = await db.execute( select(UserGroup.group_id) .where(UserGroup.user_id == user_id) .where(UserGroup.tenant_id == tenant_id) ) group_ids = [row[0] for row in groups_q] is_admin = user_data.get("is_system_admin", False) await set_user_context(db, user_id, group_ids, is_admin) # Load resolved permissions from app.core.permissions import get_cached_permissions redis = get_redis() resolved = await get_cached_permissions(db, redis, user_id, tenant_id) user_data["permissions"] = resolved.get("permissions", []) user_data["denied_permissions"] = resolved.get("denied", []) user_data["field_permissions"] = resolved.get("field_permissions", {}) user_data["is_system_admin"] = resolved.get("is_system_admin", False) return user_data async def get_current_user_or_bearer( request: Request, db: AsyncSession = Depends(get_db), redis: aioredis.Redis = Depends(get_redis_dep), ) -> dict[str, Any]: """Get current user from session cookie OR Bearer token. Tries session auth first, falls back to Bearer token. Used by MCP routes that accept both auth methods. """ auth_header = request.headers.get("Authorization", "") if auth_header.startswith("Bearer "): return await get_current_user_bearer(request, db) return await get_current_user(request, db, redis) async def require_admin( current_user: dict[str, Any] = Depends(get_current_user), ) -> dict[str, Any]: """Require admin role (legacy + new permission system). Legacy role string 'admin' is deprecated — log a warning when used. New system uses is_system_admin or *:* permission. """ if current_user.get("is_system_admin"): return current_user # Legacy role string fallback — deprecated if current_user.get("role") == "admin": logger.warning( "Legacy role string 'admin' used for user=%s — deprecated, " "migrate to is_system_admin or *:* permission", current_user.get("user_id"), ) return current_user # New permission system check from app.core.permissions import check_permission if check_permission(current_user, "*:*"): return current_user raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={"detail": "Admin access required", "code": "forbidden"}, ) async def require_write( current_user: dict[str, Any] = Depends(get_current_user), ) -> dict[str, Any]: """Require write permission (admin, editor, or custom role with write perms). Legacy role strings 'admin'/'editor' are deprecated — log a warning when used. New system checks specific module:write permissions instead of broad wildcards. """ if current_user.get("is_system_admin"): return current_user # Legacy role string fallback — deprecated role = current_user.get("role", "viewer") if role in ("admin", "editor"): logger.warning( "Legacy role string '%s' used for user=%s in require_write — deprecated, " "migrate to specific module:write permissions", role, current_user.get("user_id"), ) return current_user # Check via permission system for specific write permissions from app.core.permissions import check_permission for perm in _WRITE_PERMISSIONS: if check_permission(current_user, perm): return current_user raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={"detail": "Write access required", "code": "forbidden"}, ) def require_permission(permission: str): """FastAPI dependency factory: require a specific permission. Usage: @router.get("/contacts", dependencies=[Depends(require_permission("contacts:read"))]) """ async def _check( current_user: dict[str, Any] = Depends(get_current_user), ) -> dict[str, Any]: if current_user.get("is_system_admin"): return current_user from app.core.permissions import check_permission if check_permission(current_user, permission): return current_user raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={ "detail": f"Permission '{permission}' required", "code": "forbidden", }, ) return _check def require_field_access(module: str, field: str, default: str = "read"): """FastAPI dependency factory: require field-level access. Usage: @router.get("/contacts/{id}", dependencies=[Depends(require_field_access("contacts", "annual_revenue"))]) """ async def _check( current_user: dict[str, Any] = Depends(get_current_user), ) -> dict[str, Any]: if current_user.get("is_system_admin"): return current_user from app.core.permissions import check_field_access access = check_field_access(current_user, module, field, default) if access == "hidden": raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={ "detail": f"Field '{field}' is hidden", "code": "field_hidden", }, ) return current_user return _check async def get_tenant_id( current_user: dict[str, Any] = Depends(get_current_user), ) -> uuid.UUID: """Extract tenant_id from current user session.""" return uuid.UUID(current_user["tenant_id"]) async def get_current_user_id( current_user: dict[str, Any] = Depends(get_current_user), ) -> uuid.UUID: """Extract user_id from current user session.""" return uuid.UUID(current_user["user_id"]) def require_active_plugin(plugin_name: str): """FastAPI dependency factory: require that a plugin is active. Checks both global activation (permission registry) and per-tenant activation (tenant_plugin_activation table). Uses Redis cache for per-tenant check to avoid DB query on every request. Cache key: plugin-activation:{tenant_id}:{plugin_name} TTL: 60 seconds. Invalidated on activate/deactivate. Returns 403 if the plugin is not active. Fails closed (503) on errors. """ async def _check() -> None: from app.core.permission_registry import get_permission_registry try: registry = get_permission_registry() if not registry.is_plugin_active(plugin_name): raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={ "detail": f"Plugin '{plugin_name}' is not active", "code": "plugin_inactive", }, ) # Per-tenant activation check with Redis cache from app.core.redis import get_redis from app.core.db import async_session_maker from sqlalchemy import text import json redis = get_redis() # Get tenant_id from current session context async with async_session_maker() as db: result = await db.execute( text("SELECT current_setting('app.current_tenant_id', true)::uuid") ) tenant_id = result.scalar() if tenant_id is not None and redis is not None: cache_key = f"plugin-activation:{tenant_id}:{plugin_name}" cached = await redis.get(cache_key) if cached is not None: is_active = json.loads(cached) if not is_active: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={ "detail": f"Plugin '{plugin_name}' is not active for this tenant", "code": "plugin_inactive_tenant", }, ) return # Cache hit — plugin is active for this tenant # Cache miss — query DB async with async_session_maker() as db: result = await db.execute( text(""" SELECT is_active FROM tenant_plugin_activation WHERE plugin_name = :name AND tenant_id = :tid """), {"name": plugin_name, "tid": tenant_id}, ) row = result.first() if row is not None: is_active = row[0] # Cache the result (60s TTL) await redis.setex(cache_key, 60, json.dumps(is_active)) if not is_active: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={ "detail": f"Plugin '{plugin_name}' is not active for this tenant", "code": "plugin_inactive_tenant", }, ) else: # No entry = default active (backward compatible) await redis.setex(cache_key, 60, json.dumps(True)) else: # No Redis or no tenant_id — fallback to DB query without cache async with async_session_maker() as db: result = await db.execute( text(""" SELECT is_active FROM tenant_plugin_activation WHERE plugin_name = :name AND tenant_id = current_setting('app.current_tenant_id', true)::uuid """), {"name": plugin_name}, ) row = result.first() if row is not None and not row[0]: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail={ "detail": f"Plugin '{plugin_name}' is not active for this tenant", "code": "plugin_inactive_tenant", }, ) except HTTPException: raise except Exception as exc: # Fail-closed: if registry check fails, deny access (P1.2 fix) logger.error("Plugin activation check failed for '%s': %s", plugin_name, exc) raise HTTPException( status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail={"detail": f"Plugin activation check failed", "code": "plugin_check_error"}, ) return _check