import string
import time
from typing import Any, NamedTuple

import factory.fuzzy
from faker import Faker

from constants import BotStagesEnum
from core.bot.models.chat_gpt import ChatGpt
from tests.integration.factories.utils import BaseModelFactory

faker = Faker("ru_RU")


class User(NamedTuple):
    id: int
    is_bot: bool
    first_name: str | None
    last_name: str | None
    username: str | None
    language_code: str


class BotUserFactory(factory.Factory):
    id = factory.Sequence(lambda n: 1000 + n)
    is_bot = False
    first_name = factory.Faker("first_name")
    last_name = factory.Faker("last_name")
    username = faker.profile(fields=["username"])["username"]
    language_code = "ru"

    class Meta:
        model = User


class Chat(NamedTuple):
    id: int
    first_name: str | None
    last_name: str | None
    username: str
    type: str


class BotChatFactory(factory.Factory):
    id = factory.Sequence(lambda n: 1 + n)
    first_name = factory.Faker("first_name")
    last_name = factory.Faker("last_name")
    username = faker.profile(fields=["username"])["username"]
    type = "private"

    class Meta:
        model = Chat


class ChatGptModelFactory(BaseModelFactory):
    id = factory.Sequence(lambda n: n + 1)
    model = factory.Faker("word")
    priority = factory.Faker("random_int", min=0, max=42)

    class Meta:
        model = ChatGpt


class BotInfoFactory(factory.DictFactory):
    token = factory.Faker(
        "bothify", text="#########:??????????????????????????-#????????#?", letters=string.ascii_letters
    )  # example: 579694714:AAFpK8w6zkkUrD4xSeYwF3MO8e-4Grmcy7c
    payment_provider_token = factory.Faker(
        "bothify", text="#########:TEST:????????????????", letters=string.ascii_letters
    )  # example: 579694714:TEST:K8w6zkkUrD4xSeYw
    chat_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
    super_group_id = factory.Faker("random_int", min=-(10**12) - 10**9, max=-(10**12))  # -1001838004577
    forum_group_id = factory.Faker("random_int", min=-(10**12) - 10**9, max=-(10**12))
    channel_name = factory.Faker("name")
    channel_id = factory.LazyAttribute(lambda obj: f"@{obj.channel_name}")
    name = factory.Faker("name")
    fake_username = factory.Faker("name")
    username = factory.LazyAttribute(lambda obj: "_".join(f"@{obj.fake_username}".split(" ")))  # @Peter_Parker

    class Meta:
        exclude = ("channel_name", "fake_username")


class BotEntitleFactory(factory.DictFactory):
    type = "bot_command"
    offset = 0
    length = 42


class BotMessageFactory(factory.DictFactory):
    message_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
    chat = factory.LazyFunction(lambda: BotChatFactory()._asdict())
    date = time.time()
    text = factory.Faker("text")
    entities = factory.LazyFunction(lambda: [BotEntitleFactory()])
    voice = None

    @classmethod
    def create_instance(cls, **kwargs: Any) -> dict[str, Any]:
        return {**cls.build(**kwargs), "from": BotUserFactory()._asdict()}


class BotUpdateFactory(factory.DictFactory):
    update_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
    message = factory.LazyFunction(lambda: BotMessageFactory.create_instance())


class CallBackFactory(factory.DictFactory):
    id = factory.Faker("bothify", text="###################")
    chat_instance = factory.Faker("bothify", text="###################")
    message = factory.LazyFunction(lambda: BotMessageFactory.create_instance())
    data = factory.fuzzy.FuzzyChoice(BotStagesEnum)

    @classmethod
    def create_instance(cls, **kwargs: Any) -> dict[str, Any]:
        return {**cls.build(**kwargs), "from": BotUserFactory()._asdict()}


class BotCallBackQueryFactory(factory.DictFactory):
    update_id = factory.Faker("random_int", min=10**8, max=10**9 - 1)
    callback_query = factory.LazyFunction(lambda: BotMessageFactory.create_instance())


class BotVoiceFactory(factory.DictFactory):
    duration = factory.Faker("random_int", min=1, max=700)
    file_id = factory.Faker(
        "lexify", text="????????????????????????????????????????????????????????????????????????", locale="en_US"
    )
    file_size = factory.Faker("random_int")
    file_unique_id = factory.Faker("lexify", text="???????????????", locale="en_US")
    mime_type = "audio/ogg"