102 lines
3.7 KiB
JavaScript
102 lines
3.7 KiB
JavaScript
import { SlashCommandBuilder } from 'discord.js';
|
|
import { exec } from 'child_process';
|
|
import { promisify } from 'util';
|
|
const execAsync = promisify(exec);
|
|
|
|
// Wrap Git errors
|
|
class GitError extends Error {
|
|
constructor(message) {
|
|
super(message);
|
|
this.name = 'GitError';
|
|
}
|
|
}
|
|
|
|
// Run `git <args>` and return trimmed output or throw
|
|
async function runGit(args) {
|
|
try {
|
|
const { stdout, stderr } = await execAsync(`git ${args.join(' ')}`);
|
|
const out = stdout.trim() || stderr.trim();
|
|
return out || '(no output)';
|
|
} catch (err) {
|
|
const msg = err.stderr?.trim() || err.message;
|
|
throw new GitError(msg);
|
|
}
|
|
}
|
|
|
|
// Wrap content in Markdown code block
|
|
function formatCodeBlock(content, lang = '') {
|
|
const fence = '```';
|
|
return lang
|
|
? `${fence}${lang}\n${content}\n${fence}`
|
|
: `${fence}\n${content}\n${fence}`;
|
|
}
|
|
|
|
// Split string into chunks of at most chunkSize
|
|
function chunkString(str, chunkSize) {
|
|
const chunks = [];
|
|
for (let i = 0; i < str.length; i += chunkSize) {
|
|
chunks.push(str.slice(i, i + chunkSize));
|
|
}
|
|
return chunks;
|
|
}
|
|
|
|
// Single /git command: run arbitrary git <args>
|
|
export const commands = [
|
|
{
|
|
data: new SlashCommandBuilder()
|
|
.setName('git')
|
|
.setDescription('Run an arbitrary git command (owner only)')
|
|
.addStringOption(opt =>
|
|
opt.setName('args')
|
|
.setDescription('Arguments to pass to git')
|
|
.setRequired(true))
|
|
.addBooleanOption(opt =>
|
|
opt.setName('ephemeral')
|
|
.setDescription('Make the reply ephemeral')
|
|
.setRequired(false)),
|
|
async execute(interaction, client) {
|
|
const ownerId = client.config.owner;
|
|
if (interaction.user.id !== ownerId) {
|
|
return interaction.reply({ content: 'Only the bot owner can run git commands.', ephemeral: true });
|
|
}
|
|
const raw = interaction.options.getString('args');
|
|
// Disallow semicolons to prevent command chaining
|
|
if (raw.includes(';')) {
|
|
return interaction.reply({ content: 'Semicolons are not allowed in git arguments.', ephemeral: true });
|
|
}
|
|
const ephemeral = interaction.options.getBoolean('ephemeral') ?? true;
|
|
const args = raw.match(/(?:[^\s"]+|"[^"]*")+/g)
|
|
.map(s => s.replace(/^"(.+)"$/, '$1'));
|
|
|
|
try {
|
|
// Log the exact git command being executed
|
|
const cmdStr = args.join(' ');
|
|
client.logger.warn(`Executing git command: git ${cmdStr}`);
|
|
const output = await runGit(args);
|
|
// Prepend the git command as a header; keep it intact when chunking
|
|
const header = `git ${cmdStr}\n`;
|
|
// Discord message limit ~2000; reserve for code fences
|
|
const maxContent = 1990;
|
|
// Calculate how much output can fit after the header in the first chunk
|
|
const firstChunkSize = Math.max(0, maxContent - header.length);
|
|
// Split the raw output into chunks
|
|
const outputChunks = chunkString(output, firstChunkSize);
|
|
// Send first block with header + first output chunk
|
|
const firstBlock = header + (outputChunks[0] || '');
|
|
await interaction.reply({ content: formatCodeBlock(firstBlock), ephemeral });
|
|
// Send any remaining blocks without the header
|
|
for (let i = 1; i < outputChunks.length; i++) {
|
|
await interaction.followUp({ content: formatCodeBlock(outputChunks[i]), ephemeral });
|
|
}
|
|
} catch (err) {
|
|
const msg = err instanceof GitError ? err.message : String(err);
|
|
await interaction.reply({ content: `Error: ${msg}`, ephemeral: true });
|
|
}
|
|
}
|
|
}
|
|
];
|
|
|
|
// No special init logic
|
|
export async function init(client) {
|
|
client.logger.warn('Git utilities module loaded - dangerous module, use with caution');
|
|
} |