MENU = "MENU"
ATTENTE_PHOTO_FIN = "ATTENTE_PHOTO_FIN"
ATTENTE_CONTENU_SAV = "ATTENTE_CONTENU_SAV"
ATTENTE_CONTENU_MATERIEL = "ATTENTE_CONTENU_MATERIEL"
SAISIE_MANUELLE_CHANTIER = "SAISIE_MANUELLE_CHANTIER"
ATTENTE_FAQ = "ATTENTE_FAQ"

_states: dict[int, str] = {}
_data: dict[int, dict] = {}


def get_state(user_id: int) -> str:
    return _states.get(user_id, MENU)


def set_state(user_id: int, state: str) -> None:
    _states[user_id] = state


def get_data(user_id: int, key: str, default=None):
    return _data.get(user_id, {}).get(key, default)


def set_data(user_id: int, key: str, value) -> None:
    _data.setdefault(user_id, {})[key] = value


def clear_data(user_id: int) -> None:
    _data.pop(user_id, None)