from aiogram import Router, F
from aiogram.filters import CommandStart, CommandObject, Command
from aiogram.types import CallbackQuery, Message
from aiogram.fsm.context import FSMContext
from sqlalchemy.ext.asyncio import AsyncSession

from misc.utils import format_subscription_link
from schemas.di import DependenciesDTO
from keyboards.client import main_menu, return_to_menu
from texts import GENERAL_PROCESSING, build_main_menu_text, FAQ_TEXT

router = Router()


@router.message(CommandStart(deep_link=True, deep_link_encoded=True))
async def fetch_referal(
    msg: Message,
    command: CommandObject,
    state: FSMContext,
    session: AsyncSession,
    deps: DependenciesDTO,
):
    await state.clear()

    data = command.args.split("_")
    if not data:
        await standard_start(msg, command, state)
        return

    referal = int(data[1])
    user = await deps.user_service.add_user(
        session, user_id=msg.from_user.id, referal=referal, rw_sdk=deps.rw_sdk
    )

    await msg.reply(
        build_main_menu_text(format_subscription_link(user.subscription_link)),
        reply_markup=main_menu,
    )


@router.message(Command(commands=["start", "cancel"]))
async def standard_start(
    msg: Message,
    state: FSMContext,
    session: AsyncSession,
    deps: DependenciesDTO,
):
    await state.clear()

    user = await deps.user_service.add_user(
        session, user_id=msg.from_user.id, rw_sdk=deps.rw_sdk
    )

    await msg.reply(
        build_main_menu_text(format_subscription_link(user.subscription_link)),
        reply_markup=main_menu,
    )


@router.callback_query(F.data == "menu:main")
async def main_menu_cb(
    cb: CallbackQuery, state: FSMContext, session: AsyncSession, deps: DependenciesDTO
):
    await state.clear()

    user = await deps.user_repository.get_user_by_id(session, cb.from_user.id)
    await cb.message.edit_text(
        build_main_menu_text(format_subscription_link(user.subscription_link)),
        reply_markup=main_menu,
    )


@router.callback_query(F.data == "update_link")
async def update_link(
    cb: CallbackQuery, state: FSMContext, session: AsyncSession, deps: DependenciesDTO
):
    await state.clear()
    await cb.message.edit_text(GENERAL_PROCESSING)

    user = await deps.user_service.update_user_link(
        session, cb.from_user.id, deps.rw_sdk
    )
    await cb.answer("✅")
    await cb.message.edit_text(
        build_main_menu_text(format_subscription_link(user.subscription_link)),
        reply_markup=main_menu,
    )


@router.callback_query(F.data == "faq")
async def faq(cb: CallbackQuery, state: FSMContext):
    await state.clear()

    await cb.message.edit_text(FAQ_TEXT, reply_markup=return_to_menu)