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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
import {
ArgType,
BushCommand,
clientSendAndPermCheck,
codeblock,
colors,
emojis,
formatError,
type CommandMessage,
type SlashMessage
} from '#lib';
import assert from 'assert';
import chalk from 'chalk';
import { exec } from 'child_process';
import { ApplicationCommandOptionType, cleanCodeBlockContent, EmbedBuilder } from 'discord.js';
import { promisify } from 'util';
assert(chalk);
const sh = promisify(exec);
const clean = (text: string | any) => {
chalk.toString;
if (typeof text === 'string') {
return (text = cleanCodeBlockContent(text));
} else return text;
};
export default class ShCommand extends BushCommand {
public constructor() {
super('sh', {
aliases: ['sh', 'shell', 'cmd'],
category: 'dev',
description: 'Run shell commands.',
usage: ['sh <command>'],
examples: ['sh git pull'],
args: [
{
id: 'command',
description: 'The content you would like to run as a shell command.',
type: 'string',
match: 'rest',
prompt: 'What would you like run',
retry: '{error} Invalid command to run.',
slashType: ApplicationCommandOptionType.String
}
],
ownerOnly: true,
clientPermissions: (m) => clientSendAndPermCheck(m),
userPermissions: []
});
}
public override async exec(message: CommandMessage | SlashMessage, args: { command: ArgType<'string'> }) {
if (!client.config.owners.includes(message.author.id))
return await message.util.reply(`${emojis.error} Only my developers can run this command.`);
const input = clean(args.command);
const embed = new EmbedBuilder()
.setColor(colors.gray)
.setFooter({ text: message.author.tag, iconURL: message.author.avatarURL() ?? undefined })
.setTimestamp()
.setTitle('Shell Command')
.addFields([
{ name: '📥 Input', value: await codeblock(input, 1024, 'sh', true) },
{ name: 'Running', value: emojis.loading }
]);
await message.util.reply({ embeds: [embed] });
/* const pattern = [
'[\\u001B\\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[-a-zA-Z\\d\\/#&.:=?%@~_]*)*)?\\u0007)',
'(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PR-TZcf-ntqry=><~]))'
].join('|');
function strip(abc: string): string {
return abc.replace(new RegExp(pattern, 'g'), '');
} */
try {
const output = await sh(args.command, { env: { ...process.env, FORCE_COLOR: 'true' } });
const stdout = /* strip( */ clean(output.stdout); /* ) */
const stderr = /* strip( */ clean(output.stderr); /* ) */
embed.setTitle(`${emojis.successFull} Executed command successfully.`).setColor(colors.success).spliceFields(1, 1);
if (stdout) embed.addFields([{ name: '📤 stdout', value: await codeblock(stdout, 1024, 'ansi', true) }]);
if (stderr) embed.addFields([{ name: '📤 stderr', value: await codeblock(stderr, 1024, 'ansi', true) }]);
} catch (e) {
embed.setTitle(`${emojis.errorFull} An error occurred while executing.`).setColor(colors.error).spliceFields(1, 1);
embed.addFields([{ name: '📤 Output', value: await codeblock(formatError(e, true), 1024, 'ansi', true) }]);
}
await message.util.edit({ embeds: [embed] });
}
}
|
