Source code for dibug.commands.shell

from subprocess import run

from discord import Message

from dibug.classes.command import DibugCommand
from dibug.classes.embed import DibugEmbed


class ShellCommand(DibugCommand):
    aliases = ["shell", "sh"]

    async def execute(self, msg: Message, args: str) -> None:
        if not args:
            await msg.reply("Missing code")
            return

        res = await msg.reply("Running...")

        if args.startswith("```sh") and args.endswith("```"):
            args = args[6:-3]

        result = run(args, capture_output=True, shell=True, text=True)

        if result.stderr:
            embed = (
                DibugEmbed("Shell", True)
                .chunked_fields(
                    "Input",
                    args,
                    "sh",
                )
                .chunked_fields(
                    "Error",
                    result.stderr,
                    "sh",
                )
            )

            await res.edit(content=None, embed=embed)
            return

        embed = (
            DibugEmbed("Shell")
            .chunked_fields(
                "Input",
                args,
                "sh",
            )
            .chunked_fields(
                "Output",
                result.stdout,
                "ansi",
            )
        )

        await res.edit(content=None, embed=embed)