added snippet comband
This commit is contained in:
st committed
1 parent
2d29b80b50
commit
623300ea4a
1 file changed
+245
@@ -0,0 +1,245 @@
|
|||||||
|
'use strict';
|
||||||
|
|
||||||
|
const fs = require('fs');
|
||||||
|
const fsp = require('fs/promises');
|
||||||
|
const os = require('os');
|
||||||
|
const path = require('path');
|
||||||
|
const { spawn } = require('child_process');
|
||||||
|
const { pipeline } = require('stream/promises');
|
||||||
|
const { Readable } = require('stream');
|
||||||
|
|
||||||
|
function isMp3Attachment(attachment) {
|
||||||
|
const name = (attachment.name || '').toLowerCase();
|
||||||
|
const contentType = (attachment.contentType || '').toLowerCase();
|
||||||
|
|
||||||
|
return name.endsWith('.mp3') || contentType.startsWith('audio/');
|
||||||
|
}
|
||||||
|
|
||||||
|
function isVideoAttachment(attachment) {
|
||||||
|
const name = (attachment.name || '').toLowerCase();
|
||||||
|
const contentType = (attachment.contentType || '').toLowerCase();
|
||||||
|
|
||||||
|
return (
|
||||||
|
name.endsWith('.mp4') ||
|
||||||
|
name.endsWith('.mov') ||
|
||||||
|
name.endsWith('.mkv') ||
|
||||||
|
name.endsWith('.webm') ||
|
||||||
|
name.endsWith('.m4v') ||
|
||||||
|
contentType.startsWith('video/')
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function runProcess(command, args) {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const child = spawn(command, args, { stdio: ['ignore', 'ignore', 'pipe'] });
|
||||||
|
let stderr = '';
|
||||||
|
|
||||||
|
child.stderr.on('data', (chunk) => {
|
||||||
|
stderr += chunk.toString();
|
||||||
|
});
|
||||||
|
|
||||||
|
child.on('error', reject);
|
||||||
|
child.on('close', (code) => {
|
||||||
|
if (code === 0) {
|
||||||
|
resolve();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
reject(new Error(stderr || `${command} exited with code ${code}`));
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function downloadAttachment(attachment, filePath) {
|
||||||
|
const response = await fetch(attachment.url);
|
||||||
|
|
||||||
|
if (!response.ok || !response.body) {
|
||||||
|
throw new Error(`failed to download ${attachment.name}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
await pipeline(Readable.fromWeb(response.body), fs.createWriteStream(filePath));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function getDuration(filePath) {
|
||||||
|
const probe = spawn('ffprobe', [
|
||||||
|
'-v', 'error',
|
||||||
|
'-show_entries', 'format=duration',
|
||||||
|
'-of', 'default=noprint_wrappers=1:nokey=1',
|
||||||
|
filePath,
|
||||||
|
], { stdio: ['ignore', 'pipe', 'pipe'] });
|
||||||
|
|
||||||
|
let stdout = '';
|
||||||
|
let stderr = '';
|
||||||
|
|
||||||
|
probe.stdout.on('data', (chunk) => {
|
||||||
|
stdout += chunk.toString();
|
||||||
|
});
|
||||||
|
|
||||||
|
probe.stderr.on('data', (chunk) => {
|
||||||
|
stderr += chunk.toString();
|
||||||
|
});
|
||||||
|
|
||||||
|
const exitCode = await new Promise((resolve, reject) => {
|
||||||
|
probe.on('error', reject);
|
||||||
|
probe.on('close', resolve);
|
||||||
|
});
|
||||||
|
|
||||||
|
if (exitCode !== 0) {
|
||||||
|
throw new Error(stderr || 'failed to read media duration');
|
||||||
|
}
|
||||||
|
|
||||||
|
const duration = Number.parseFloat(stdout.trim());
|
||||||
|
if (!Number.isFinite(duration) || duration <= 0) {
|
||||||
|
throw new Error('invalid media duration');
|
||||||
|
}
|
||||||
|
|
||||||
|
return duration;
|
||||||
|
}
|
||||||
|
|
||||||
|
function buildEmbed(client, message, description, fields = []) {
|
||||||
|
const embed = {
|
||||||
|
color: 0x8aa2b3,
|
||||||
|
title: '',
|
||||||
|
author: {
|
||||||
|
name: message.author.username,
|
||||||
|
icon_url: message.author.displayAvatarURL(),
|
||||||
|
},
|
||||||
|
description,
|
||||||
|
footer: {
|
||||||
|
text: `twitter @ ${message.guild?.name || 'dms'}`,
|
||||||
|
icon_url: client.user.displayAvatarURL(),
|
||||||
|
},
|
||||||
|
timestamp: new Date(),
|
||||||
|
};
|
||||||
|
|
||||||
|
if (fields.length > 0) {
|
||||||
|
embed.fields = fields;
|
||||||
|
}
|
||||||
|
|
||||||
|
return embed;
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
name: 'snippet',
|
||||||
|
description: 'mix a random mp3 segment with a random video segment and export a clipped video',
|
||||||
|
aliases: ['mix', 'clip'],
|
||||||
|
usage: 'snippet <seconds> [mp3 attachment] [video attachment]',
|
||||||
|
|
||||||
|
async execute(client, message, args = []) {
|
||||||
|
const requestedLength = Number.parseInt(args[0], 10);
|
||||||
|
|
||||||
|
if (!Number.isInteger(requestedLength) || requestedLength <= 0) {
|
||||||
|
return message.reply({
|
||||||
|
content: null,
|
||||||
|
embeds: [buildEmbed(
|
||||||
|
client,
|
||||||
|
message,
|
||||||
|
'<:no:1481811955592466492> provide a clip length in seconds, then attach one mp3 and one video file.',
|
||||||
|
)],
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const attachments = [...message.attachments.values()];
|
||||||
|
const mp3Attachment = attachments.find(isMp3Attachment);
|
||||||
|
const videoAttachment = attachments.find(isVideoAttachment);
|
||||||
|
|
||||||
|
if (!mp3Attachment || !videoAttachment) {
|
||||||
|
return message.reply({
|
||||||
|
content: null,
|
||||||
|
embeds: [buildEmbed(
|
||||||
|
client,
|
||||||
|
message,
|
||||||
|
'<:no:1481811955592466492> attach exactly one mp3 file and one video file to use this command.',
|
||||||
|
)],
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const sent = await message.reply({
|
||||||
|
content: null,
|
||||||
|
embeds: [buildEmbed(
|
||||||
|
client,
|
||||||
|
message,
|
||||||
|
'<a:loading_icon:1366587619890302998> building your clip...',
|
||||||
|
)],
|
||||||
|
});
|
||||||
|
|
||||||
|
const tempDir = await fsp.mkdtemp(path.join(os.tmpdir(), 'twitter-snippet-'));
|
||||||
|
const mp3Path = path.join(tempDir, mp3Attachment.name || 'input.mp3');
|
||||||
|
const videoPath = path.join(tempDir, videoAttachment.name || 'input.mp4');
|
||||||
|
const outputPath = path.join(tempDir, 'snippet.mp4');
|
||||||
|
|
||||||
|
try {
|
||||||
|
await Promise.all([
|
||||||
|
downloadAttachment(mp3Attachment, mp3Path),
|
||||||
|
downloadAttachment(videoAttachment, videoPath),
|
||||||
|
]);
|
||||||
|
|
||||||
|
const mp3Duration = await getDuration(mp3Path);
|
||||||
|
const videoDuration = await getDuration(videoPath);
|
||||||
|
|
||||||
|
if (requestedLength > mp3Duration || requestedLength > videoDuration) {
|
||||||
|
throw new Error('requested length is longer than one of the attached files');
|
||||||
|
}
|
||||||
|
|
||||||
|
const clipLength = requestedLength;
|
||||||
|
|
||||||
|
const videoStart = videoDuration === clipLength
|
||||||
|
? 0
|
||||||
|
: Math.random() * (videoDuration - clipLength);
|
||||||
|
const audioStart = mp3Duration === clipLength
|
||||||
|
? 0
|
||||||
|
: Math.random() * (mp3Duration - clipLength);
|
||||||
|
|
||||||
|
await runProcess('ffmpeg', [
|
||||||
|
'-y',
|
||||||
|
'-ss', videoStart.toFixed(3),
|
||||||
|
'-i', videoPath,
|
||||||
|
'-ss', audioStart.toFixed(3),
|
||||||
|
'-i', mp3Path,
|
||||||
|
'-t', clipLength.toFixed(3),
|
||||||
|
'-map', '0:v:0',
|
||||||
|
'-map', '1:a:0',
|
||||||
|
'-c:v', 'libx264',
|
||||||
|
'-preset', 'veryfast',
|
||||||
|
'-crf', '20',
|
||||||
|
'-c:a', 'aac',
|
||||||
|
'-b:a', '192k',
|
||||||
|
'-pix_fmt', 'yuv420p',
|
||||||
|
'-vf', 'scale=trunc(iw/2)*2:trunc(ih/2)*2',
|
||||||
|
'-movflags', '+faststart',
|
||||||
|
outputPath,
|
||||||
|
]);
|
||||||
|
|
||||||
|
const finalEmbed = buildEmbed(
|
||||||
|
client,
|
||||||
|
message,
|
||||||
|
'<:check:1481811973447356496> clip finished successfully.',
|
||||||
|
[
|
||||||
|
{ name: 'mp3 source', value: `**${mp3Attachment.name || 'unknown'}**`, inline: true },
|
||||||
|
{ name: 'video source', value: `**${videoAttachment.name || 'unknown'}**`, inline: true },
|
||||||
|
{ name: 'clip length', value: `**${clipLength.toFixed(2)}s**`, inline: true },
|
||||||
|
],
|
||||||
|
);
|
||||||
|
|
||||||
|
await sent.edit({
|
||||||
|
content: null,
|
||||||
|
embeds: [finalEmbed],
|
||||||
|
files: [{ attachment: outputPath, name: 'snippet.mp4' }],
|
||||||
|
});
|
||||||
|
} catch (err) {
|
||||||
|
await sent.edit({
|
||||||
|
content: null,
|
||||||
|
embeds: [buildEmbed(
|
||||||
|
client,
|
||||||
|
message,
|
||||||
|
'<:no:1481811955592466492> failed to build the clip.',
|
||||||
|
[
|
||||||
|
{ name: 'reason', value: `**${err.message || 'unknown error'}**` },
|
||||||
|
],
|
||||||
|
)],
|
||||||
|
}).catch(() => null);
|
||||||
|
} finally {
|
||||||
|
await fsp.rm(tempDir, { recursive: true, force: true }).catch(() => null);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
};
|
||||||
Reference in new issue
Block a user