const { Command } = require ( 'discord-akairo' ) ;
const YTPGenerator = require ( 'ytpplus-node' ) ;
const os = require ( 'os' ) ;
const fs = require ( 'fs' ) ;
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 : 'string'
} ,
{
id : 'max' ,
type : 'string'
}
] ,
description : {
content : 'Generate random ytp | --add with a link or attachment to add a video to the pool, only .mp4 work | --pool to see how many vid there is currently in the pool | --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' ,
usage : '(OPTIONAL) | [Minimum length of clip] [Max length of clip]' ,
examples : [ '5 10' ]
}
} ) ;
}
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 . add ) {
let loadingmsg = await message . channel . send ( 'Downloading <a:loadingmin:527579785212329984>' ) ;
let Attachment = ( message . attachments ) . array ( ) ;
let url = args . link ;
// Get attachment link
if ( Attachment [ 0 ] && ! args . link ) {
url = Attachment [ 0 ] . url ;
}
if ( url ) {
return downloader ( url , [ '--format=mp4' ] , ` ./asset/ytp/userVid/ ${ message . id } .mp4 ` )
. catch ( ( err ) => {
loadingmsg . delete ( ) ;
return message . channel . send ( err , { code : true } ) ;
} )
. then ( 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 {
const body = { hash : hash , link : args . link , 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 = [ ] ;
let files = fs . readdirSync ( './asset/ytp/userVid/' ) ;
// Count number of total vid
files . 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 ( ) * files . length ) ;
let vid = ` ./asset/ytp/userVid/ ${ files [ random ] } ` ;
if ( files [ 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 <a:loadingmin:527579785212329984> \n Some 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 ( ) ;
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 occured! please try again. If this happens alot, you should report this to the developers.' ) ;
} ) ;
} ) ;
}
}
module . exports = ytpCommand ;