import logging
from enum import StrEnum

logger = logging.getLogger(__name__)


class GraphicType(StrEnum):
    STICKER = "STICKER"
    ANIMATION = "ANIMATION"
    VIDEO = "VIDEO"


class Response(StrEnum):
    TEXT = "TEXT"
    GRAPHIC = "GRAPHIC"


class ChatData:
    def __init__(
        self,
        id: str,
        chat_type: str,
        name: str,
        message_period: int = 10,
        answer_chance: float = 0.5,
        restricted: bool = False,
        silenced: bool = False,
    ) -> None:
        self.id = id
        self.type = chat_type
        self.name = name
        self.message_period = message_period
        self.answer_chance = answer_chance
        self.restricted = restricted
        self.silenced = silenced

    @staticmethod
    def from_sql_row(row: tuple[str, str, str, int, float, int, int]):
        logger.debug(f"From SQL row: {repr(row)}")
        return ChatData(
            row[0],
            row[1],
            row[2],
            int(row[3]),
            float(row[4]),
            int(row[5]) == 1,
            int(row[6]) == 1,
        )