from aiogram import Router, Bot, F
from aiogram.types import (
    BufferedInputFile, InputMediaAudio,
    ChosenInlineResult,
)

from bot.modules.youtube.downloader import YouTubeBytestream

from bot.utils.config import config
from bot.modules.database import db
from bot.modules.settings import UserSettings

from io import BytesIO

router = Router()


@router.chosen_inline_result(
    F.result_id.startswith('spotc::') | F.result_id.startswith('ytc::')
)
async def on_cached_chosen(chosen_result: ChosenInlineResult, bot: Bot,
                           settings: UserSettings):
    if settings['recode_youtube'].value != 'yes':
        await bot.edit_message_reply_markup(
            inline_message_id=chosen_result.inline_message_id,
            reply_markup=None
        )
        return

    if (
        type(
            db.recoded.get(
                song_id := chosen_result.result_id
                .removeprefix('spotc::')
                .removeprefix('ytc::')
            )
        ) in [bool, type(None)]
    ):
        await bot.edit_message_reply_markup(
            inline_message_id=chosen_result.inline_message_id,
            reply_markup=None
        )
        return

    await bot.edit_message_caption(
        inline_message_id=chosen_result.inline_message_id,
        caption='🔄 Recoding...',
        reply_markup=None
    )

    message = await bot.forward_message(
        config.telegram.files_chat,
        config.telegram.files_chat,
        db.recoded[song_id]
    )

    song_io: BytesIO = await bot.download(  # type: ignore
        destination=BytesIO(),
        file=message.audio.file_id,
    )
    await message.delete()

    bytestream = YouTubeBytestream.from_bytestream(
        bytestream=song_io,
        filename=message.audio.file_name,
        duration=message.audio.duration,
    )

    await bytestream.rerender()

    audio = await bot.send_audio(
        chat_id=config.telegram.files_chat,
        audio=BufferedInputFile(
            file=bytestream.file,
            filename=bytestream.filename,
        ),
        thumbnail=BufferedInputFile(
            file=(await bot.download(message.audio.thumbnail.file_id)).read(),
            filename='thumbnail.jpg'
        ),
        performer=message.audio.performer,
        title=message.audio.title,
        duration=bytestream.duration,
    )

    await bot.edit_message_caption(
        inline_message_id=chosen_result.inline_message_id,
        caption='',
        reply_markup=None,
    )
    await bot.edit_message_media(
        inline_message_id=chosen_result.inline_message_id,
        media=InputMediaAudio(media=audio.audio.file_id)
    )

    if chosen_result.result_id.startswith('spotc::'):
        db.spotify[song_id] = audio.audio.file_id
    else:
        db.youtube[song_id] = audio.audio.file_id
    db.recoded[song_id] = True
    await db.occasionally_write()