MissKatyPyro/misskaty/plugins/mediainfo.py

"""
 * @author        yasir <yasiramunandar@gmail.com>
 * @date          2022-12-01 09:12:27
 * @lastModified  2022-12-01 09:32:31
 * @projectName   MissKatyPyro
 * Copyright @YasirPedia All rights reserved
"""
import io
from os import remove as osremove
import time
import subprocess
from pyrogram import filters
from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
from misskaty.vars import COMMAND_HANDLER
from utils import get_file_id
from misskaty import app
from misskaty.helper.media_helper import post_to_telegraph, runcmd
from misskaty.core.decorator.errors import capture_err
from misskaty.helper.pyro_progress import (
    progress_for_pyrogram,
)


@app.on_message(filters.command(["mediainfo"], COMMAND_HANDLER))
@capture_err
async def mediainfo(client, message):
    if message.reply_to_message and message.reply_to_message.media:
        process = await message.reply_text(
            "`Sedang memproses, lama waktu tergantung ukuran file kamu...`", quote=True
        )
        file_info = get_file_id(message.reply_to_message)
        if file_info is None:
            await process.edit_text("Balas ke format media yang valid")
            return
        c_time = time.time()
        # file_path = safe_filename(await reply.download())
        file_path = await client.download_media(
            message=message.reply_to_message,
            progress=progress_for_pyrogram,
            progress_args=("trying to download, sabar yakk..", process, c_time),
        )
        output_ = await runcmd(f'mediainfo "{file_path}"')
        out = output_[0] if len(output_) != 0 else None
        body_text = f"""
    <img src='https://telegra.ph/file/72c99bbc89bbe4e178cc9.jpg' />
    <h2>JSON</h2>
    <pre>{file_info}.type</pre>
    <br>
    <h2>DETAILS</h2>
    <pre>{out or 'Not Supported'}</pre>
    """
        title = "MissKaty Bot Mediainfo"
        text_ = file_info.message_type
        link = post_to_telegraph(title, body_text)
        markup = InlineKeyboardMarkup([[InlineKeyboardButton(text=text_, url=link)]])
        await message.reply("ℹ️ <b>MEDIA INFO</b>", reply_markup=markup, quote=True)
        await process.delete()
        try:
            osremove(file_path)
        except Exception:
            pass
    else:
        try:
            link = message.text.split(" ", maxsplit=1)[1]
            if link.startswith("https://file.yasirweb.my.id"):
                link = link.replace(
                    "https://file.yasirweb.my.id", "https://file.yasiraris.workers.dev"
                )
            if link.startswith("https://link.yasirweb.my.id"):
                link = link.replace(
                    "https://link.yasirweb.my.id", "https://yasirrobot.herokuapp.com"
                )
            process = await message.reply_text("`Mohon tunggu sejenak...`")
            try:
                output = subprocess.check_output(["mediainfo", f"{link}"]).decode(
                    "utf-8"
                )
            except Exception:
                return await process.edit(
                    "Sepertinya link yang kamu kirim tidak valid, pastikan direct link dan bisa di download."
                )
            title = "MissKaty Bot Mediainfo"
            body_text = f"""
                         <img src='https://telegra.ph/file/72c99bbc89bbe4e178cc9.jpg' />
                         <pre>{output}</pre>
                         """
            tgraph = post_to_telegraph(title, body_text)
            # siteurl = "https://spaceb.in/api/v1/documents/"
            # response = await http.post(siteurl, data={"content": output, "extension": 'txt'} )
            # response = response.json()
            # spacebin = "https://spaceb.in/"+response['payload']['id']
            markup = InlineKeyboardMarkup(
                [[InlineKeyboardButton(text="💬 Telegraph", url=tgraph)]]
            )
            with io.BytesIO(str.encode(output)) as out_file:
                out_file.name = "MissKaty_Mediainfo.txt"
                await message.reply_document(
                    out_file,
                    caption=f"Hasil mediainfo anda..\n\n<b>Request by:</b> {message.from_user.mention}",
                    reply_markup=markup,
                )
                await process.delete()
        except IndexError:
            return await message.reply_text(
                "Gunakan command /mediainfo [link], atau reply telegram media dengan /mediainfo."
            )