from __future__ import annotations

from app.slots.base import BaseSlot
from app.slots.coming_soon import ComingSoonSlot
from app.slots.forest_fortune import ForestFortuneSlot

_SLOTS: list[BaseSlot] = [
    ComingSoonSlot(
        slug="raven-reels",
        title="Вороньи Барабаны",
        description="Слоты у врат Вырия. Скоро откроются.",
    ),
    ComingSoonSlot(
        slug="golden-gate",
        title="Золотые Врата",
        description="Портал удачи. Пока запечатан.",
    ),
    ForestFortuneSlot(),
]

_BY_SLUG: dict[str, BaseSlot] = {slot.slug: slot for slot in _SLOTS}


def list_slots() -> list[BaseSlot]:
    return list(_SLOTS)


def get_slot(slug: str) -> BaseSlot | None:
    return _BY_SLUG.get(slug)