data-entry-app/backend/app/api/auth.py

from fastapi import APIRouter, Depends, HTTPException, status
from pydantic import BaseModel
from sqlalchemy import select
from sqlalchemy.orm import Session

from app.api.deps import AuthSession, require_admin_session, require_client_session
from app.core.config import settings
from app.core.security import issue_token
from app.db.session import get_db
from app.models.client_access import ClientAccount

router = APIRouter(prefix="/api/auth", tags=["auth"])


class LoginRequest(BaseModel):
    email: str
    password: str


class SessionResponse(BaseModel):
    name: str
    email: str
    role: str
    tenant_id: str | None = None
    token: str


def _build_session_response(*, name: str, email: str, role: str, tenant_id: str | None = None) -> SessionResponse:
    token = issue_token({"name": name, "email": email, "role": role, "tenant_id": tenant_id})
    return SessionResponse(name=name, email=email, role=role, tenant_id=tenant_id, token=token)


@router.post("/client/login", response_model=SessionResponse)
def client_login(payload: LoginRequest, db: Session = Depends(get_db)):
    if payload.email.strip().lower() != settings.client_email.lower() or payload.password != settings.client_password:
        raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid client email or password")

    client_account = db.scalar(select(ClientAccount).where(ClientAccount.tenant_id == settings.client_tenant_id))
    if client_account is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Client account is not configured")

    return _build_session_response(
        name=settings.client_name,
        email=settings.client_email,
        role="client",
        tenant_id=client_account.tenant_id,
    )


@router.post("/admin/login", response_model=SessionResponse)
def admin_login(payload: LoginRequest):
    if payload.email.strip().lower() != settings.admin_email.lower() or payload.password != settings.admin_password:
        raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid admin email or password")

    return _build_session_response(name=settings.admin_name, email=settings.admin_email, role="admin")


@router.get("/client/session", response_model=SessionResponse)
def read_client_session(session: AuthSession = Depends(require_client_session)):
    return _build_session_response(name=session.name, email=session.email, role=session.role, tenant_id=session.tenant_id)


@router.get("/admin/session", response_model=SessionResponse)
def read_admin_session(session: AuthSession = Depends(require_admin_session)):
    return _build_session_response(name=session.name, email=session.email, role=session.role)