Add YouTube recoding

2023-11-07 14:13:14 +02:00
parent 54006008d3
commit 06c327933f
13 changed files with 245 additions and 15 deletions
--- a/bot/handlers/on_chosen/recode_cached.py
+++ b/bot/handlers/on_chosen/recode_cached.py
@@ -0,0 +1,101 @@
+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()