core/executors.py

#    This file is part of the AutoAnime distribution.
#    Copyright (c) 2025 Kaif_00z
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, version 3.
#
#    This program is distributed in the hope that it will be useful, but
#    WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
#    General Public License for more details.
#
# License can be found in <
# https://github.com/kaif-00z/AutoAnimeBot/blob/main/LICENSE > .

# if you are using this following code then don't forgot to give proper
# credit to t.me/kAiF_00z (github.com/kaif-00z)

import asyncio
import os
import secrets
import shutil
from glob import glob
from traceback import format_exc

from telethon import Button

from core.bot import LOGS, Bot, Var
from database import DataBase
from functions.info import AnimeInfo
from functions.tools import Tools
from libs.logger import Reporter


class Executors:
    def __init__(
        self,
        bot: Bot,
        dB: DataBase,
        configurations: dict,
        input_file: str,
        info: AnimeInfo,
        reporter: Reporter,
    ):
        self.is_original = configurations.get("original_upload")
        self.is_button = configurations.get("button_upload")
        self.anime_info = info
        self.bot = bot
        self.input_file = input_file
        self.tools = Tools()
        self.db = dB
        self.reporter = reporter
        self.msg_id = None
        self.output_file = None

    async def execute(self):
        try:
            rename = await self.anime_info.rename(self.is_original)
            self.output_file = f"encode/{rename}"
            thumb = await self.tools.cover_dl((await self.anime_info.get_poster()))
            if self.is_original:
                await self.reporter.started_renaming()
                succ, out = await self.tools.rename_file(
                    self.input_file, self.output_file
                )
                if not succ:
                    return False, out
            else:
                _log_msg = await self.reporter.started_compressing()
                succ, _new_msg = await self.tools.compress(
                    self.input_file, self.output_file, _log_msg
                )
                if not succ:
                    return False, _new_msg
                self.reporter.msg = _new_msg
            await self.reporter.started_uploading()
            if self.is_button:
                msg = await self.bot.upload_anime(
                    self.output_file, rename, thumb or "thumb.jpg", is_button=True
                )
                btn = Button.url(
                    f"{self.anime_info.data.get('video_resolution')}",
                    url=f"https://t.me/{((await self.bot.get_me()).username)}?start={msg.id}",
                )
                self.msg_id = msg.id
                return True, btn
            msg = await self.bot.upload_anime(
                self.output_file, rename, thumb or "thumb.jpg"
            )
            self.msg_id = msg.id
            return True, []
        except BaseException:
            await self.reporter.report_error(str(format_exc()), log=True)
            return False, str(format_exc())

    def run_further_work(self):
        asyncio.run(self.further_work())

    async def further_work(self):
        try:
            if self.msg_id:
                await self.reporter.started_gen_ss()
                msg = await self.bot.get_messages(
                    Var.BACKUP_CHANNEL if self.is_button else Var.MAIN_CHANNEL,
                    ids=self.msg_id,
                )
                btn = [
                    [],
                ]
                link_info = await self.tools.mediainfo(self.output_file, self.bot)
                if link_info:
                    btn.append(
                        [
                            Button.url(
                                "📜 MediaInfo",
                                url=link_info,
                            )
                        ]
                    )
                    await msg.edit(buttons=btn)
                _hash = secrets.token_hex(nbytes=7)
                ss_path, sp_path = await self.tools.gen_ss_sam(_hash, self.output_file)
                if ss_path and sp_path:
                    ss = await self.bot.send_message(
                        Var.CLOUD_CHANNEL,
                        file=glob(f"{ss_path}/*") or ["assest/poster_not_found.jpg"],
                    )
                    sp = await self.bot.send_message(
                        Var.CLOUD_CHANNEL,
                        file=sp_path,
                        thumb="thumb.jpg",
                        force_document=True,
                    )
                    await self.db.store_items(_hash, [[i.id for i in ss], [sp.id]])
                    btn.append(
                        [
                            Button.url(
                                "📺 Sample & ScreenShots",
                                url=f"https://t.me/{((await self.bot.get_me()).username)}?start={_hash}",
                            )
                        ]
                    )
                    await msg.edit(buttons=btn)
                    await self.reporter.all_done()
                    try:
                        shutil.rmtree(_hash)
                        os.remove(sp_path)
                        os.remove(self.input_file)
                        os.remove(self.output_file)
                    except BaseException:
                        LOGS.error(str(format_exc()))
        except BaseException:
            await self.reporter.report_error(str(format_exc()), log=True)