blob: a5e7e71158b37d992150ff21183c6c0bafdab8da (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
from discord.ext.commands import Cog, Context, command
from arthur.apis.netcup.ssh import rce_as_a_service
from arthur.bot import KingArthur
class RemoteCommands(Cog):
"""We love RCE."""
def __init__(self, bot: KingArthur) -> None:
self.bot = bot
@command(name="rce")
async def rce(self, ctx: Context, *, command: str) -> None:
"""Ed is the standard text editor."""
if not ctx.guild:
return
if not await ctx.bot.is_owner(ctx.author):
await ctx.message.add_reaction("❌")
return
response = await rce_as_a_service(command)
if not response.stderr and not response.stdout:
await ctx.send(f"Successfully ran with no output! {response.returncode = }")
return
await ctx.send(
f"{response.returncode = }\n"
f"stderr:```{response.stderr or '~Empty~'}```\n"
f"stdout:```{response.stdout or '~Empty~'}```\n"
)
async def setup(bot: KingArthur) -> None:
"""Add cog to bot."""
await bot.add_cog(RemoteCommands(bot))
|
