'use strict'
const youtubedl = require ( 'youtube-dl' )
const fs = require ( 'fs' )
const ffmpeg = require ( 'fluent-ffmpeg' )
const { version } = require ( '../../../package.json' ) ;
const Ws = use ( 'Ws' ) ;
const Antl = use ( 'Antl' ) ;
let viewCounter = 0 ;
let files = [ ] ;
let day ;
let month ;
let announcementArray ;
let announcement ;
function formatBytes ( bytes , decimals = 2 ) { // https://stackoverflow.com/a/18650828
if ( bytes === 0 ) return '0 Bytes' ;
const k = 1024 ;
const dm = decimals < 0 ? 0 : decimals ;
const sizes = [ 'Bytes' , 'KB' , 'MB' , 'GB' , 'TB' , 'PB' , 'EB' , 'ZB' , 'YB' ] ;
const i = Math . floor ( Math . log ( bytes ) / Math . log ( k ) ) ;
return parseFloat ( ( bytes / Math . pow ( k , i ) ) . toFixed ( dm ) ) + ' ' + sizes [ i ] ;
}
class DownloadController {
async index ( { view , request , locale } ) {
viewCounter ++ ;
// Coudln't find a cleaner way to make it change with the browser locale
announcementArray = [ Antl . forLocale ( locale ) . formatMessage ( 'announcement.1' ) , Antl . forLocale ( locale ) . formatMessage ( 'announcement.2' ) , Antl . forLocale ( locale ) . formatMessage ( 'announcement.3' ) , Antl . forLocale ( locale ) . formatMessage ( 'announcement.4' ) , Antl . forLocale ( locale ) . formatMessage ( 'announcement.5' ) , Antl . forLocale ( locale ) . formatMessage ( 'announcement.6' ) , 'Playlist download is experimental' ] ;
// Get random announcement
announcement = announcementArray [ Math . floor ( Math . random ( ) * announcementArray . length ) ] ;
// Get date for some event
let today = new Date ( ) ;
day = today . getDay ( ) ;
month = today . getMonth ( ) ;
// If legacy link return
if ( request . url ( ) == '/legacy' ) return view . render ( 'legacy' , { version : version , viewCounter : viewCounter , day : day , month : month , announcement : announcement } ) ;
files = [ ] ;
let file = [ ] ;
for ( let f of fs . readdirSync ( './public/uploads' ) ) {
file . push ( f )
}
// get the 5 most recent files
file = file . sort ( ( a , b ) => {
if ( ( a || b ) . endsWith ( '.mp4' ) || ( a || b ) . endsWith ( '.webm' ) || ( a || b ) . endsWith ( '.mp3' ) || ( a || b ) . endsWith ( '.flac' ) || ( a || b ) . endsWith ( '.zip' ) ) {
let time1 = fs . statSync ( ` ./public/uploads/ ${ b } ` ) . ctime ;
let time2 = fs . statSync ( ` ./public/uploads/ ${ a } ` ) . ctime ;
if ( time1 < time2 ) return - 1 ;
if ( time1 > time2 ) return 1 ;
}
return 0 ;
} ) . slice ( 0 , 5 )
// Save space by deleting file that doesn't appear in the recent feed
for ( let f of fs . readdirSync ( './public/uploads' ) ) {
if ( ! file . includes ( f ) && ( f != 'hidden' && f != '.keep' && f != 'playlist' ) ) {
fs . unlinkSync ( ` ./public/uploads/ ${ f } ` ) ;
}
}
for ( let f of file ) {
if ( f . endsWith ( '.mp4' ) || f . endsWith ( '.webm' ) ) {
// Send file name, file size in MB relative path for the file
let fileInfo = formatBytes ( fs . statSync ( ` ./public/uploads/ ${ f } ` ) . size ) . split ( ' ' ) ;
files . push ( { name : f . split ( '.' ) . slice ( 0 , - 1 ) . join ( '.' ) , size : fileInfo [ 0 ] , unit : fileInfo [ 1 ] , date : fs . statSync ( ` ./public/uploads/ ${ f } ` ) . ctime , location : ` uploads/ ${ f } ` , ext : f . split ( '.' ) . pop ( ) , img : '' } ) ;
} else if ( f . endsWith ( '.mp3' ) || f . endsWith ( '.flac' ) ) {
// Send file name, file size in MB relative path for the file and relative path of music.png
let fileInfo = formatBytes ( fs . statSync ( ` ./public/uploads/ ${ f } ` ) . size ) . split ( ' ' ) ;
files . push ( { name : f . split ( '.' ) . slice ( 0 , - 1 ) . join ( '.' ) , size : fileInfo [ 0 ] , unit : fileInfo [ 1 ] , date : fs . statSync ( ` ./public/uploads/ ${ f } ` ) . ctime , location : ` uploads/ ${ f } ` , ext : f . split ( '.' ) . pop ( ) , img : ` /asset/music.png ` } ) ;
}
}
return view . render ( 'index' , { version : version , viewCounter : viewCounter , file : files , day : day , month : month , announcement : announcement } ) ;
}
async download ( { view , request , response } ) {
const ws = Ws . getChannel ( 'progress' ) . topic ( 'progress' ) ;
// To be honest i forgot what it does, but i think i need it
response . implicitEnd = false
let option , DLFile
// Get form input
let data = {
url : request . input ( 'URL' ) ,
quality : request . input ( 'quality' ) ,
format : request . input ( 'format' ) ,
alt : request . input ( 'alt' ) ,
feed : request . input ( 'feed' )
}
if ( ! data . url ) {
if ( ws ) {
ws . socket . emit ( 'error' , 'bruh moment, you didin\'t input a link.' ) ;
}
return ;
}
// If alt download ( Quality settings and file format option doesn't work here )
if ( data . alt ) {
let altFolder ;
if ( data . feed == 'on' ) {
altFolder = './public/uploads/hidden/alt.mp4' ;
} else {
altFolder = './public/uploads/alt.mp4'
}
if ( fs . existsSync ( altFolder ) ) {
fs . unlink ( altFolder , ( err ) => {
if ( err ) ;
} ) ;
}
return youtubedl . exec ( data . url , [ '--format=mp4' , '-o' , altFolder ] , { } , function ( err , output ) {
if ( err ) {
console . error ( err ) ;
if ( ws ) {
ws . socket . emit ( 'error' , err . toString ( ) ) ;
}
return ;
}
console . log ( altFolder . slice ( 17 ) )
if ( ws ) {
ws . socket . emit ( 'end' , altFolder . slice ( 17 ) ) ;
}
} ) ;
} else {
if ( data . url . match ( /^.*(youtu.be\/|list=)([^#\&\?]*).*/ ) ) {
playlistDownload ( data ) ;
} else {
videoDownload ( data ) ;
}
}
function videoDownload ( data ) {
// Download as mp4 if possible
let video = youtubedl ( data . url , [ '-f' , data . quality ] ) ;
video . on ( 'error' , function ( err ) {
console . error ( err ) ;
if ( ws ) {
ws . socket . emit ( 'error' , err . toString ( ) ) ;
}
} )
let ext ;
let size = 0
video . on ( 'info' , function ( info ) {
size = info . size
// Set file name
ext = info . ext ;
let title = info . title . slice ( 0 , 50 ) ;
DLFile = ` ${ title . replace ( /\s/g , '_' ) } . ${ ext } ` ;
DLFile = DLFile . replace ( /[()]|[/]|[\\]|[?]|[!]|[#]/g , '_' ) ;
// If no title use the ID
if ( title == '_' ) title = ` _ ${ info . id } ` ;
// If user want to hide from the feed
if ( data . feed == 'on' )
DLFile = ` hidden/ ${ DLFile } ` ;
video . pipe ( fs . createWriteStream ( ` ./public/uploads/ ${ DLFile } ` ) ) ;
} ) ;
let pos = 0
video . on ( 'data' , function data ( chunk ) {
pos += chunk . length
// `size` should not be 0 here.
if ( size ) {
let percent = ( pos / size * 100 ) . toFixed ( 2 )
if ( ws ) {
ws . socket . emit ( 'progress' , percent ) ;
}
}
} )
video . on ( 'end' , function ( ) {
if ( data . format == 'mp3' || data . format == 'flac' ) {
// If user requested an audio format, convert it
ffmpeg ( ` ./public/uploads/ ${ DLFile } ` )
. noVideo ( )
. audioChannels ( '2' )
. audioFrequency ( '44100' )
. audioBitrate ( '320k' )
. format ( data . format )
. save ( ` ./public/uploads/ ${ DLFile . replace ( ` . ${ ext } ` , ` . ${ data . format } ` ) } ` )
. on ( 'progress' , ( progress ) => {
ws . socket . emit ( progress . percent )
} )
. on ( 'end' , ( ) => {
fs . unlinkSync ( ` ./public/uploads/ ${ DLFile } ` ) ;
if ( ws ) {
ws . socket . emit ( 'end' , ` ./public/uploads/ ${ DLFile . replace ( ` . ${ ext } ` , ` . ${ data . format } ` ) } ` ) ;
}
} ) ;
} else {
if ( ws ) {
ws . socket . emit ( 'end' , ` ./public/uploads/ ${ DLFile } ` ) ;
}
}
} ) ;
}
function playlistDownload ( data ) {
const video = youtubedl ( data . url )
video . on ( 'error' , function error ( err ) {
console . error ( err ) ;
if ( ws ) {
ws . socket . emit ( 'error' , err . toString ( ) ) ;
}
} ) ;
let ext ;
let size = 0
video . on ( 'info' , function ( info ) {
console . log ( info ) ;
size = info . size
// Set file name
ext = info . ext ;
let title = info . title . slice ( 0 , 50 ) ;
DLFile = ` ${ title . replace ( /\s/g , '_' ) } . ${ ext } ` ;
DLFile = DLFile . replace ( /[()]|[/]|[\\]|[?]|[!]|[#]/g , '_' ) ;
// If no title use the ID
if ( title == '_' ) title = ` _ ${ info . id } ` ;
// If user want to hide from the feed
if ( data . feed == 'on' )
DLFile = ` hidden/playlist/ ${ DLFile } ` ;
video . pipe ( fs . createWriteStream ( ` ./public/uploads/playlist/ ${ DLFile } ` ) ) ;
} ) ;
let pos = 0
video . on ( 'data' , function data ( chunk ) {
pos += chunk . length
// `size` should not be 0 here.
if ( size ) {
let percent = ( pos / size * 100 ) . toFixed ( 2 )
process . stdout . cursorTo ( 0 )
process . stdout . clearLine ( 1 )
if ( ws ) {
ws . socket . emit ( 'progress' , percent ) ;
}
}
} ) ;
video . on ( 'end' , function ( ) {
if ( data . format == 'mp3' || data . format == 'flac' ) {
// If user requested an audio format, convert it
ffmpeg ( ` ./public/uploads/playlist/ ${ DLFile } ` )
. noVideo ( )
. audioChannels ( '2' )
. audioFrequency ( '44100' )
. audioBitrate ( '320k' )
. format ( data . format )
. save ( ` ./public/uploads/playlist/ ${ DLFile . replace ( ` . ${ ext } ` , ` . ${ data . format } ` ) } ` )
. on ( 'progress' , ( progress ) => {
ws . socket . emit ( progress . percent )
} )
. on ( 'end' , ( ) => {
fs . unlinkSync ( ` ./public/uploads/playlist/ ${ DLFile } ` ) ;
if ( ws ) {
ws . socket . emit ( 'end' , ` ./public/uploads/playlist/ ${ DLFile . replace ( ` . ${ ext } ` , ` . ${ data . format } ` ) } ` ) ;
}
} ) ;
} else {
if ( ws ) {
ws . socket . emit ( 'end' , ` ./public/uploads/playlist/ ${ DLFile } ` ) ;
}
}
} ) ;
video . on ( 'next' , playlistDownload ) ;
}
}
}
module . exports = DownloadController