Task_Reporter/app/modules/auth/middleware.py

"""Authentication middleware for protected routes

自動處理：
1. Token 驗證
2. 3 天不活動逾時檢查
3. AD token 自動刷新（5 分鐘內過期時）
4. 重試計數器管理（最多 3 次）
"""
from fastapi import Request, status
from fastapi.responses import JSONResponse
from datetime import datetime, timedelta
from app.core.database import SessionLocal
from app.core.config import get_settings
from app.modules.auth.services.session_service import session_service
from app.modules.auth.services.encryption import encryption_service
from app.modules.auth.services.ad_client import ad_auth_service
import logging

settings = get_settings()
logger = logging.getLogger(__name__)


class AuthMiddleware:
    """Authentication middleware"""

    async def __call__(self, request: Request, call_next):
        """Process request through authentication checks"""

        # Skip auth for non-API routes (frontend), login/logout, and docs
        path = request.url.path
        if not path.startswith("/api") or path in ["/api/auth/login", "/api/auth/logout", "/api/health"]:
            return await call_next(request)
        if path in ["/docs", "/openapi.json", "/redoc"]:
            return await call_next(request)

        # Extract token from Authorization header
        authorization = request.headers.get("Authorization")
        if not authorization or not authorization.startswith("Bearer "):
            return JSONResponse(
                status_code=status.HTTP_401_UNAUTHORIZED,
                content={"detail": "Authentication required"}
            )

        internal_token = authorization.replace("Bearer ", "")

        # Get database session
        db = SessionLocal()
        try:
            # Query session
            user_session = session_service.get_session_by_token(db, internal_token)
            if not user_session:
                return JSONResponse(
                    status_code=status.HTTP_401_UNAUTHORIZED,
                    content={"detail": "Invalid or expired token"}
                )

            # Check 3-day inactivity timeout
            inactivity_limit = datetime.utcnow() - timedelta(days=settings.SESSION_INACTIVITY_DAYS)
            if user_session.last_activity < inactivity_limit:
                session_service.delete_session(db, user_session.id)
                return JSONResponse(
                    status_code=status.HTTP_401_UNAUTHORIZED,
                    content={"detail": "Session expired due to inactivity. Please login again."}
                )

            # Check if refresh attempts exceeded
            if user_session.refresh_attempt_count >= settings.MAX_REFRESH_ATTEMPTS:
                session_service.delete_session(db, user_session.id)
                return JSONResponse(
                    status_code=status.HTTP_401_UNAUTHORIZED,
                    content={"detail": "Session expired due to authentication failures. Please login again."}
                )

            # Check if AD token needs refresh (< 5 minutes until expiry)
            time_until_expiry = user_session.ad_token_expires_at - datetime.utcnow()
            if time_until_expiry < timedelta(minutes=settings.TOKEN_REFRESH_THRESHOLD_MINUTES):
                # Auto-refresh AD token
                refresh_error = await self._refresh_ad_token(db, user_session)
                if refresh_error:
                    return refresh_error

            # Update last_activity
            session_service.update_activity(db, user_session.id)

            # Attach user info to request state
            request.state.user = {
                "id": user_session.id,
                "username": user_session.username,
                "display_name": user_session.display_name,
            }

        finally:
            db.close()

        return await call_next(request)

    async def _refresh_ad_token(self, db, user_session):
        """Auto-refresh AD token using stored encrypted password

        Returns:
            JSONResponse on error, None on success
        """
        try:
            # Decrypt password
            password = encryption_service.decrypt_password(user_session.encrypted_password)

            # Re-authenticate with AD API
            ad_result = await ad_auth_service.authenticate(user_session.username, password)

            # Update session with new token
            session_service.update_ad_token(
                db, user_session.id, ad_result["token"], ad_result["expires_at"]
            )

            logger.info(f"AD token refreshed successfully for user: {user_session.username}")
            return None  # Success

        except (ValueError, ConnectionError) as e:
            # Refresh failed, increment counter
            new_count = session_service.increment_refresh_attempts(db, user_session.id)

            logger.warning(
                f"AD token refresh failed for user {user_session.username}. "
                f"Attempt {new_count}/{settings.MAX_REFRESH_ATTEMPTS}"
            )

            # If reached max attempts, delete session
            if new_count >= settings.MAX_REFRESH_ATTEMPTS:
                session_service.delete_session(db, user_session.id)
                logger.error(
                    f"Session terminated for {user_session.username} after {new_count} failed refresh attempts"
                )
                return JSONResponse(
                    status_code=status.HTTP_401_UNAUTHORIZED,
                    content={"detail": "Session terminated. Your password may have been changed. Please login again."}
                )
            else:
                return JSONResponse(
                    status_code=status.HTTP_401_UNAUTHORIZED,
                    content={"detail": "Token refresh failed. Please try again or re-login if issue persists."}
                )


auth_middleware = AuthMiddleware()