from typing import Literal

from beanie import Document


class Invite(Document):
    code: str
    channel_id: int
    mode: Literal["instant", "approval"]
    used: bool = False

    class Settings:
        name = "invites"

    @classmethod
    async def get_by_code(cls, code: str) -> "Invite | None":
        return await cls.find_one(cls.code == code, cls.used == False)  # noqa: E712