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