const { Command } = require('discord-akairo'); const { proxy } = require('../../config.json'); const YTPGenerator = require('ytpplus-node'); const os = require('os'); const fs = require('fs'); const attachment = require('../../utils/attachment'); const downloader = require('../../utils/download'); const md5File = require('md5-file'); const ytpHash = require('../../models').ytpHash; const MAX_CLIPS = 20; class ytpCommand extends Command { constructor() { super('ytp', { aliases: ['ytp', 'ytpplus', 'ytp+'], category: 'fun', clientPermissions: ['ATTACH_FILES', 'SEND_MESSAGES'], args: [ { id: 'add', match: 'flag', flag: ['--add'] }, { id: 'pool', match: 'flag', flag: ['--pool'] }, { id: 'force', match: 'flag', flag: ['--force'] }, { id: 'randomSound', match: 'flag', flag: ['--randomSound'] }, { id: 'randomSoundMute', match: 'flag', flag: ['--randomSoundMute'] }, { id: 'reverse', match: 'flag', flag: ['--reverse'] }, { id: 'chorus', match: 'flag', flag: ['--chorus'] }, { id: 'vibrato', match: 'flag', flag: ['--vibrato'] }, { id: 'highPitch', match: 'flag', flag: ['--highPitch'] }, { id: 'lowPitch', match: 'flag', flag: ['--lowPitch'] }, { id: 'speedUp', match: 'flag', flag: ['--speedUp'] }, { id: 'slowDown', match: 'flag', flag: ['--slowDown'] }, { id: 'dance', match: 'flag', flag: ['--dance'] }, { id: 'squidward', match: 'flag', flag: ['--squidward'] }, { id: 'how', match: 'flag', flag: ['--how'] }, { id: 'debug', match: 'flag', flag: ['--debug'] }, { id: 'link', type: 'url', prompt: { start: 'Please send the URL of which video you want to download. Say `cancel` to stop the command', retry: 'Please send a valid URL of the video you want to download. Say `cancel` to stop the command', optional: true, }, unordered: true }, { id: 'max', type: 'string', unordered: true }, { id: 'proxy', match: 'option', flag: ['--proxy'], }, { id: 'listproxy', match: 'flag', flag: ['--listproxy', '--proxylist'] } ], description: { content: 'Generate random ytp\n--add with a link or attachment to add a video to the pool, only .mp4 work\n--pool to see how many vid there is currently in the pool\n--force to make the command work outside of nsfw channel BE AWARE THAT IT WON\'T CHANGE THE FINAL RESULT SO NSFW CAN STILL HAPPEN\n`--proxy #` to select a proxy, `--listproxy` to see a list of proxy', usage: '(OPTIONAL) | [Minimum length of clip] [Max length of clip]', examples: ['5 10', '--add https://www.youtube.com/watch?v=6n3pFFPSlW4', '--add https://www.youtube.com/watch?v=6n3pFFPSlW4 --proxy 1', '--listproxy'] } }); } async exec(message, args) { if (args.pool) { let mp4 = []; fs.readdirSync('./asset/ytp/userVid/').forEach(file => { if (file.endsWith('mp4')) { mp4.push(file); } }); return message.channel.send(`There is currently ${mp4.length} videos, you can add yours by doing \`\`${this.client.commandHandler.prefix[0]}ytp --add (link or attachment)\`\``); } if (args.listproxy) { let proxys = []; let i = 0; proxy.forEach(proxy => { i++; proxys.push(`[${i}] ${ proxy.hideip ? '[IP HIDDEN]' : proxy.ip.substring(0, proxy.ip.length - 5)} - ${proxy.country}`); }); const Embed = this.client.util.embed() .setColor(message.member ? message.member.displayHexColor : 'NAVY') .setTitle('List of available proxy') .setDescription(proxys.join('\n')) .setFooter('You can help me get more proxy by either donating to me or providing a proxy for me'); return message.channel.send(Embed); } if (args.add) { if (args.proxy && !args.proxyAuto) { // args.proxyAuto is only provided when the command is run after a error 429 args.proxy = args.proxy -1; if (!proxy[args.proxy]) args.proxy = 0; } let loadingmsg = await message.channel.send('Downloading '); let url; if (args.link) url = args.link.href; else url = await attachment(message); if (url) { let options = ['--format=mp4']; if (args.proxy) { options.push('--proxy'); options.push(proxy[args.proxy].ip); } return downloader(url, options, `./asset/ytp/userVid/${message.id}.mp4`) .on('error', (err) => { if (err.includes('HTTP Error 429: Too Many Requests')) { if (args.proxy != null) { args.proxy = args.proxy + 1; } else { args.proxy = 0; args.proxyAuto = true; } if (!proxy[args.proxy]) return message.channel.send('`HTTP Error 429: Too Many Requests.`\nThe website you tried to download from probably has the bot blocked along with its proxy'); loadingmsg.delete(); return this.client.commandHandler.runCommand(message, this.client.commandHandler.findCommand('ytp'), args); } if (err.includes('Error: status code 403')) return message.channel.send('`HTTP Error 403: Forbidden.`\nThe video you tried to download is not publicly available therefor the bot can\'t download it.'); return message.channel.send(err, { code: true }); }) .on('end', async output => { const hash = md5File.sync(output); const ytphash = await ytpHash.findOne({where: {hash: hash}}); if (ytphash) { fs.unlinkSync(output); loadingmsg.delete(); return message.reply('This video is a duplicate... Not adding.'); } else { let file = fs.statSync(output); let fileSize = file.size / 1000000.0; if (fileSize > 50) { fs.unlinkSync(output); loadingmsg.delete(); return message.reply('Video too big.. Not adding.'); } const body = {hash: hash, link: url, messageID: message.id}; await ytpHash.create(body); } let mp4 = []; fs.readdirSync('./asset/ytp/userVid/').forEach(file => { if (file.endsWith('mp4')) { mp4.push(file); } }); loadingmsg.delete(); return message.reply(`Video successfully added to the pool! There is now ${mp4.length} videos`); }); } else { loadingmsg.delete(); return message.channel.send('You need a valid video link!'); } } if (!message.channel.nsfw && !args.force) return message.channel.send(`Please execute this command in an NSFW channel ( Content might not be NSFW but since the video are user submitted better safe than sorry ) OR do \`\`${this.client.commandHandler.prefix[0]}ytp --force\`\` to make the command work outside of nsfw channel BE AWARE THAT IT WON'T CHANGE THE FINAL RESULT SO NSFW CAN STILL HAPPEN`); // Read userVid folder and select random vid and only take .mp4 let mp4 = []; let asset = []; // Count number of total vid fs.readdirSync('./asset/ytp/userVid/').forEach(file => { if (file.endsWith('mp4')) { mp4.push(file); } }); // Select random vid depending on the amount of MAX_CLIPS for (let i = 0; i < MAX_CLIPS; i++) { let random = Math.floor(Math.random() * mp4.length); let vid = `./asset/ytp/userVid/${mp4[random]}`; if (mp4[random].endsWith('mp4')) { if (!asset.includes(vid)) { asset.push(vid); } } } let loadingmsg = await message.channel.send(`Processing, this can take a ***long*** time, i'll ping you when i finished \nSome info: There are currently ${mp4.length} videos, you can add yours by doing \`\`${this.client.commandHandler.prefix[0]}ytp --add (link or attachment). Thanks for contributing!\`\``); let options = { debug: args.debug, MIN_STREAM_DURATION: args.link ? Math.floor(args.link) : null, MAX_STREAM_DURATION: args.link && args.max ? args.max : Math.floor((Math.random() * 3) + 1), // Random duration of video clip sources: './asset/ytp/sources/', sounds: './asset/ytp/sounds/', music: './asset/ytp/music/', resources: './asset/ytp/resources/', temp: os.tmpdir(), sourceList: asset, intro: args.force ? './asset/ytp/intro.mp4' : null, outro: './asset/ytp/outro.mp4', OUTPUT_FILE: `${os.tmpdir()}/${message.id}_YTP.mp4`, MAX_CLIPS: MAX_CLIPS, transitions: true, showFileNames: true, effects: { effect_RandomSound: !args.randomSound, effect_RandomSoundMute: !args.randomSoundMute, effect_Reverse: !args.reverse, effect_Chorus: !args.chorus, effect_Vibrato: !args.vibrato, effect_HighPitch: !args.highPitch, effect_LowPitch: !args.lowPitch, effect_SpeedUp: !args.speedUp, effect_SlowDown: !args.slowDown, effect_Dance: !args.dance, effect_Squidward: !args.squidward, effect_How: !args.how } }; new YTPGenerator().configurateAndGo(options) .then(() => { md5File(`${os.tmpdir()}/${message.id}_YTP.mp4`).then(async hash => { const body = {hash: hash, link: 'null', messageID: message.id}; await ytpHash.create(body); }); loadingmsg.delete(); return message.reply('Here is your YTP! Remember, it might contain nsfw, so be careful!', {files: [`${os.tmpdir()}/${message.id}_YTP.mp4`]}) .catch(err => { console.error(err); return message.channel.send('Whoops, look like the vid might be too big for discord, my bad, please try again'); }); }) .catch(err => { console.error(err); loadingmsg.delete(); if (!args.retry) args.retry = 0; else args.retry += 1; if (args.retry === 3) { return message.reply({files: [Math.random() < 0.5 ? './asset/ytp/error1.mp4' : './asset/ytp/error2.mp4']}) .catch(err => { // In case it can't send the video for some reason console.error(err); return message.channel.send('Oh no, an error has occurred! please try again. If this happens alot, you should report this to the developers.'); }); } return this.client.commandHandler.runCommand(message, this.client.commandHandler.findCommand('ytp'), args); }); } } module.exports = ytpCommand;