diff --git a/index.js b/index.js index 8736507..0c9fe4a 100644 --- a/index.js +++ b/index.js @@ -3,6 +3,7 @@ const fs = require('fs'), express = require('express'), cp = require("child_process"), ffmpeg = require("ffmpeg-static") +const { stringify } = require('querystring') const PORT = process.env.PORT || 8080 const app = express() @@ -54,91 +55,163 @@ app.get("/convert", async (req, res) => { app.get("/download", async (req, res) => { const url = req.query.url const format = req.query.format || 'mp4' - const quality = req.query.quality || 'highest' + const quality = req.query.quality || '720' res.setHeader("X-Accel-Buffering", "no") - res.setHeader("Content-Type", "text/html") - var downloadHTML = fs.readFileSync(path.join(__dirname, 'resources/downloading.html')).toString() + var fileName = Math.ceil(Math.random() * 100_000_000_000).toString() + console.log(fileName) + var videoName = cp.spawnSync('yt-dlp', ['-S', 'res:' + quality, '--get-filename', url]).stdout.toString() - res.write(downloadHTML.substring(0, downloadHTML.indexOf("{CONTENT}"))) + videoName = videoName.substring(0, videoName.lastIndexOf('.')) + videoName = videoName.substring(0, videoName.lastIndexOf('[')) - var fileName = cp.spawnSync('yt-dlp', ['-S', 'res:' + quality,'--get-filename', url]).stdout.toString() + if (!["mp3", "wav", "ogx"].includes(format) && Number(quality) > 720) { + res.setHeader("Content-Type", "text/html") - if (!["mp3", "wav", "ogx"].includes(format)) { - var filePath = path.join(__dirname, 'downloads', fileName).trim() + var downloadHTML = fs.readFileSync(path.join(__dirname, 'resources/downloading.html')).toString() + + res.write(downloadHTML.substring(0, downloadHTML.indexOf("{CONTENT}"))) + + var filePath = path.join(__dirname, 'downloads', fileName) var ytdlpProcess = cp.spawn('yt-dlp', [ url, - '-o', filePath.substring(0, filePath.lastIndexOf('.')), + '-o', filePath, '--max-filesize', MAX_FILESIZE + 'm', '-S', 'res:' + quality, '--no-playlist', ]) - } else { - fileName = fileName.substring(0, fileName.lastIndexOf(".")) + '.mp3' - var filePath = path.join(__dirname, 'downloads', fileName).trim() + + var lastDownload = 0 + + ytdlpProcess.stderr.setEncoding('utf-8') + ytdlpProcess.stderr.on('data', (data) => { + console.log(data) + res.write(`

` + data + `

`) + }) + + var debounce = false + + ytdlpProcess.stdout.setEncoding('utf-8') + ytdlpProcess.stdout.on('data', (data) => { + if (!debounce) { + if (data.includes("max-filesize")) { + debounce = true + res.write(`

Uh oh! The video you're trying to download is too large for this server's current settings ${MAX_FILESIZE}. Please try another server? (Visit main page and go to the codeberg for a list of instances!)

`) + } + else if (data.includes("[download]")) { + res.write(``) + lastDownload += 1 + res.write(`

` + data.substring(12) + `

`) + } + } + + + }) + + var exited = false + + ytdlpProcess.on('close', () => { + + var files = fs.readdirSync(path.join(__dirname, 'downloads')) + for (let index = 0; index < files.length; index++) { + const file = files[index]; + console.log(file) + if (file.includes(fileName)) { + fileName = file + filePath = path.join(__dirname, 'downloads', fileName) + } + } + + if (exited) { + if (fs.existsSync(filePath)) { + fs.rmSync(filePath) + } + } + else if (fs.existsSync(filePath)) { + res.write(`"`) + res.write(downloadHTML.substring(downloadHTML.indexOf("{CONTENT}") + 9) + ``, () => { res.end() }) + } else { + res.write("

An error has occured!!! We're not exactly sure what the error is, but we cant seem to find the download file. Double check the URL, and if the URL is fine, then file an issue on codeberg.

") + } + }) + + res.on("error", () => { + if (fs.existsSync(filePath)) { + fs.rmSync(filePath) + } + exited = true + }) + } else if (["mp3", "wav", "ogx"].includes(format)) { + res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(videoName)}.${format}"`); var ytdlpProcess = cp.spawn('yt-dlp', [ url, '-x', - '-o', filePath.substring(0, filePath.lastIndexOf('.')), + '-o', '-', + '--max-filesize', MAX_FILESIZE + 'm', + '--audio-format', 'mp3', + '--no-playlist', + ]) + + const ffmpegProcess = cp.spawn(ffmpeg, [ + '-i', 'pipe:3', + '-f', formats[format] || format, + '-movflags', 'frag_keyframe+empty_moov', + '-c:a', 'libmp3lame', + '-preset','ultrafast', + '-crf', '23', + '-loglevel', 'error', + '-' + ], { + stdio: [ + 'pipe', 'pipe', 'pipe', 'pipe', 'pipe', + ], + }) + + ytdlpProcess.stdout.pipe(ffmpegProcess.stdio[3]) + + ytdlpProcess.stderr.setEncoding('utf-8') + ytdlpProcess.stderr.on('data', (data) => { + console.log(data) + }) + + ffmpegProcess.stdio[1].pipe(res) + } else { + res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(videoName)}.${format}"`); + + var ytdlpProcess = cp.spawn('yt-dlp', [ + url, + '-o', '-', '--max-filesize', MAX_FILESIZE + 'm', '--no-playlist', - '--audio-format', 'mp3', ]) + + const ffmpegProcess = cp.spawn(ffmpeg, [ + '-i', 'pipe:3', + '-f', formats[format] || format, + '-movflags', 'frag_keyframe+empty_moov', + '-c:a', 'libmp3lame', + '-preset','ultrafast', + '-crf', '23', + '-loglevel', 'error', + '-' + ], { + stdio: [ + 'pipe', 'pipe', 'pipe', 'pipe', 'pipe', + ], + }) + + ytdlpProcess.stdout.pipe(ffmpegProcess.stdio[3]) + + ytdlpProcess.stderr.setEncoding('utf-8') + ytdlpProcess.stderr.on('data', (data) => { + console.log(data) + }) + + ffmpegProcess.stdio[1].pipe(res) } - var lastDownload = 0 - - ytdlpProcess.stderr.setEncoding('utf-8') - ytdlpProcess.stderr.on('data', (data) => { - console.log(data) - res.write(`

` + data + `

`) - }) - - var debounce = false - - ytdlpProcess.stdout.setEncoding('utf-8') - ytdlpProcess.stdout.on('data', (data) => { - if (!debounce) { - if (data.includes("max-filesize")) { - debounce = true - res.write(`

Uh oh! The video you're trying to download is too large for this server's current settings ${MAX_FILESIZE}. Please try another server? (Visit main page and go to the codeberg for a list of instances!)

`) - } - else if (data.includes("[download]")) { - res.write(``) - lastDownload += 1 - res.write(`

` + data.substring(12) + `

`) - } - } - - - }) - - var exited = false - - ytdlpProcess.on('close', () => { - console.log(filePath) - if (exited) { - if (fs.existsSync(filePath)) { - fs.rmSync(filePath) - } - } - else if (fs.existsSync(filePath)) { - res.write(`"`) - res.write(downloadHTML.substring(downloadHTML.indexOf("{CONTENT}") + 9) + ``, () => { res.end() }) - } else { - res.write("

An error has occured!!! We're not exactly sure what the error is, but we cant seem to find the download file. Double check the URL, and if the URL is fine, then file an issue on codeberg.

") - } - }) - - res.on("error", () => { - if (fs.existsSync(filePath)) { - fs.rmSync(filePath) - } - exited = true - }) - }) app.use(express.static(path.join(__dirname, 'static')))