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

from config import settings
from keyboards.client import referals_kb
from misc.utils import format_share_deep_link
from schemas.di import DependenciesDTO
from texts import GENERAL_PROCESSING, REFERALS_MENU

router = Router()


@router.callback_query(F.data == "referal")
async def referal_menu(
    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)
    referal_link = await create_start_link(cb.bot, f"ref_{user.id}", encode=True)
    share_link = format_share_deep_link(referal_link)

    await cb.message.edit_text(
        REFERALS_MENU.format(balance=user.balance, link=referal_link),
        reply_markup=referals_kb(share_link),
    )


@router.message(F.chat.id == settings.admin_group_id, Command("refpay"))
async def refpay(
    msg: Message,
    command: CommandObject,
    state: FSMContext,
    session: AsyncSession,
    deps: DependenciesDTO,
):
    await state.clear()

    await msg.delete()
    status_msg = await msg.answer(GENERAL_PROCESSING)
    user = await deps.user_service.fetch_user_by_thread(session, msg.message_thread_id)
    referal = await deps.user_repository.get_user_by_id(session, user.referal_id)
    if not referal:
        await status_msg.edit_text("❌ У пользователя нет реферала.")
        return

    amount = command.args

    if not amount.isdigit():
        await status_msg.edit_text(
            "❌ Укажите в аргументах количество средств, которые необходимо перевести рефералу пользователя."
        )
        return

    amount = int(amount)
    await deps.user_repository.increase_user_balance(session, referal.id, amount)
    await status_msg.edit_text(f"<b>💸 Баланс реферала ({referal.id}) пополнен на {amount}₽</b>")