This commit is contained in:
parent
600173771e
commit
18c2d9dc18
1 changed files with 85 additions and 0 deletions
85
commands/misc/dl.js
Normal file
85
commands/misc/dl.js
Normal file
|
@ -0,0 +1,85 @@
|
||||||
|
const { execFileSync } = require('child_process');
|
||||||
|
const path = require('path');
|
||||||
|
const fs = require('fs')
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
name: 'dl',
|
||||||
|
description: 'Download a video',
|
||||||
|
moreHelp: [
|
||||||
|
"Usage: <prefix>dl <url>"
|
||||||
|
],
|
||||||
|
async execute({message, args}) {
|
||||||
|
const downloadsDir = path.resolve(process.cwd(), 'data', 'downloads', Date.now().toString());
|
||||||
|
const cookieFilepath = path.resolve(process.cwd(), 'data', 'cookies.txt')
|
||||||
|
if(!fs.existsSync(cookieFilepath)) {
|
||||||
|
message.channel.send("Some dependencies are needed for the command to work properly. Please let the bot's owner know.")
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
fs.mkdirSync(downloadsDir, {recursive: true});
|
||||||
|
|
||||||
|
let url;
|
||||||
|
|
||||||
|
if(args.length > 0){
|
||||||
|
if(args[0].charAt(0) === '<' && args[0].charAt(args[0].length - 1) === '>'){
|
||||||
|
args[0] = args[0].slice(1, args[0].length - 1)
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
url = new URL(args[0]);
|
||||||
|
url = url.href;
|
||||||
|
} catch (error) {
|
||||||
|
this.cleanUp(downloadsDir);
|
||||||
|
message.channel.send("Could not parse the provided argument as a URL.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
this.cleanUp(downloadsDir);
|
||||||
|
return message.channel.send("You have to provide a URL in an argument.")
|
||||||
|
}
|
||||||
|
|
||||||
|
const originalMessage = await message.channel.send("Downloading video...")
|
||||||
|
|
||||||
|
if(this.executeCommand("yt-dlp", [url, "-P", downloadsDir, "--cookies", cookieFilepath]).error){
|
||||||
|
originalMessage.edit("An error occurred when downloading the video.");
|
||||||
|
this.cleanUp(downloadsDir);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
let files = fs.readdirSync(downloadsDir);
|
||||||
|
if(files.length < 1) {
|
||||||
|
this.cleanUp(downloadsDir);
|
||||||
|
originalMessage.edit("Something went wrong when downloading the video.")
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const filename = files[0];
|
||||||
|
|
||||||
|
await originalMessage.edit({
|
||||||
|
content: null,
|
||||||
|
files: [{
|
||||||
|
attachment: path.resolve(downloadsDir, filename)
|
||||||
|
}]})
|
||||||
|
|
||||||
|
this.cleanUp(downloadsDir);
|
||||||
|
|
||||||
|
|
||||||
|
},
|
||||||
|
|
||||||
|
cleanUp(downloadsDir){
|
||||||
|
fs.rmSync(downloadsDir, {force: true, recursive: true});
|
||||||
|
},
|
||||||
|
|
||||||
|
executeCommand(command, commandArgs, verbose=false) {
|
||||||
|
if (typeof command !== 'string' || !Array.isArray(commandArgs)) return { error: true };
|
||||||
|
console.log("Executing:", command, commandArgs.join(" "));
|
||||||
|
try {
|
||||||
|
const output = execFileSync(command, commandArgs, {encoding: 'utf8'})
|
||||||
|
if (output.length != 0 && verbose)
|
||||||
|
console.log(output)
|
||||||
|
} catch (error) {
|
||||||
|
console.error(`Error executing ${command} command:`, error);
|
||||||
|
return { error: true };
|
||||||
|
}
|
||||||
|
return { error: false };
|
||||||
|
},
|
||||||
|
|
||||||
|
}
|
Loading…
Add table
Add a link
Reference in a new issue