from fastapi import HTTPException, status

from app.models.auth import CognitoUser

MOCK_USERS = {"testuser": {"username": "testuser", "roles": ["admin"]}}


def mock_get_user_from_token(token: str) -> CognitoUser:
    """
    Mock version of get_user_from_token for local testing
    Accepts 'testuser' as a valid token and returns admin user
    """
    if token == "testuser":
        return CognitoUser(**MOCK_USERS["testuser"])
    raise HTTPException(
        status_code=status.HTTP_401_UNAUTHORIZED,
        detail="Invalid mock token - use 'testuser'",
    )


def mock_initiate_auth(username: str, password: str) -> dict:
    """
    Mock version of initiate_auth for local testing
    Accepts any username/password and returns a mock token
    """
    return {"AccessToken": "testuser", "ExpiresIn": 3600, "TokenType": "Bearer"}