aboutsummaryrefslogtreecommitdiff
path: root/src/commands/dev/sh.ts
blob: 1b9ce229dc5385e3b7309ca427efe8b6f0fe50e8 (plain)
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
import {
	ArgType,
	BushCommand,
	clientSendAndPermCheck,
	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 (!this.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 this.client.utils.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 this.client.utils.codeblock(stdout, 1024, 'ansi', true) });
			if (stderr) embed.addFields({ name: '📤 stderr', value: await this.client.utils.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 this.client.utils.codeblock(formatError(e, true), 1024, 'ansi', true) });
		}
		await message.util.edit({ embeds: [embed] });
	}
}