joan/Haha-Yes
1
0
Fork 0
forked from Supositware/Haha-Yes
Code Pull requests Activity

Some duplication protection

Browse source 
Signed-off-by: loicbersier <loic.bersier1@gmail.com>
This commit is contained in:
loicbersier 2020-05-13 08:47:11 +02:00
parent aed32cd2b9
commit f8803ec5a7
1 changed files with 22 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
commands/fun/ytp.js
View file

@ -3,6 +3,10 @@ const YTPGenerator = require('ytpplus-node');
const os = require('os'); const os = require('os');
const fs = require('fs'); const fs = require('fs');
const youtubedl = require('youtube-dl'); const youtubedl = require('youtube-dl');
const md5File = require('md5-file');
const ytpHash = require('../../models').ytpHash;
const MAX_CLIPS = 20; const MAX_CLIPS = 20;
@ -128,7 +132,7 @@ class ytpCommand extends Command {
} }
if (url) { if (url) {
return youtubedl.exec(url, ['--rm-cache-dir', '--no-playlist', '--max-filesize', '50m', '--format=mp4', '-o', `./asset/ytp/userVid/${message.id}.mp4`], {}, function(err, output) { return youtubedl.exec(url, ['--rm-cache-dir', '--no-playlist', '--max-filesize', '50m', '--format=mp4', '-o', `./asset/ytp/userVid/${message.id}.mp4`], {}, async function(err, output) {
console.log(output); console.log(output);
if (err) { if (err) {
console.error(err); console.error(err);
@ -140,7 +144,19 @@ class ytpCommand extends Command {
loadingmsg.delete(); loadingmsg.delete();
return message.channel.send(output[2]); return message.channel.send(output[2]);
} }
} }
const hash = md5File.sync(`./asset/ytp/userVid/${message.id}.mp4`);
const ytphash = await ytpHash.findOne({where: {hash: hash}});
if (ytphash) {
fs.unlinkSync(`./asset/ytp/userVid/${message.id}.mp4`);
return message.reply('This video is a duplicate... Not adding.');
} else {
const body = {hash: hash, link: args.link, messageID: message.id};
await ytpHash.create(body);
}
let mp4 = []; let mp4 = [];
fs.readdirSync('./asset/ytp/userVid/').forEach(file => { fs.readdirSync('./asset/ytp/userVid/').forEach(file => {
@ -219,6 +235,10 @@ class ytpCommand extends Command {
new YTPGenerator().configurateAndGo(options) new YTPGenerator().configurateAndGo(options)
.then(() => { .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(); loadingmsg.delete();
return message.reply('Here is your YTP! Remember, it might contain nsfw!', {files: [`${os.tmpdir()}/${message.id}_YTP.mp4`]}) return message.reply('Here is your YTP! Remember, it might contain nsfw!', {files: [`${os.tmpdir()}/${message.id}_YTP.mp4`]})
.catch(err => { .catch(err => {