from fastapi import HTTPException
from fastapi.responses import RedirectResponse
from sqlalchemy.orm import Session
from starlette.requests import Request

from web.models.user import User, UserStatusEnum


def get_session_user_id(request: Request) -> int | None:
    return request.session.get("user_id")


def get_current_user(request: Request, db: Session) -> User:
    user_id = get_session_user_id(request)
    if not user_id:
        raise HTTPException(status_code=307, headers={"Location": "/login"})
    user = db.get(User, user_id)
    if not user or user.status == UserStatusEnum.suspended:
        request.session.clear()
        raise HTTPException(status_code=307, headers={"Location": "/login"})
    return user


def login_redirect() -> RedirectResponse:
    return RedirectResponse("/login", status_code=303)