112 lines
3.3 KiB
JavaScript
112 lines
3.3 KiB
JavaScript
const fs = require('fs'),
|
|
path = require('path'),
|
|
express = require('express'),
|
|
cp = require("child_process"),
|
|
ffmpeg = require("ffmpeg-static")
|
|
|
|
const PORT = process.env.PORT || 8080
|
|
const app = express()
|
|
|
|
const MAX_FILESIZE = process.env.MAX_FILESIZE || 500
|
|
|
|
var formats = {
|
|
"mkv": "matroska"
|
|
}
|
|
|
|
app.get("/convert", async (req, res) => {
|
|
var file = req.query.file || ""
|
|
var format = req.query.format
|
|
var filePath = path.join(__dirname, 'downloads', file)
|
|
if (fs.existsSync(filePath)) {
|
|
res.setHeader('Content-Disposition', `attachment; filename="test.${format}"`);
|
|
const ffmpegProcess = cp.spawn(ffmpeg, [
|
|
'-i', filePath,
|
|
'-f', formats[format] || format,
|
|
'-movflags', 'frag_keyframe+empty_moov',
|
|
'-vcodec', 'copy' ,
|
|
'-'
|
|
], {
|
|
stdio: [
|
|
'pipe', 'pipe', 'pipe', 'pipe', 'pipe',
|
|
],
|
|
})
|
|
|
|
ffmpegProcess.stderr.setEncoding('utf-8')
|
|
ffmpegProcess.stderr.on('data', (data) => {
|
|
console.log(data)
|
|
})
|
|
// These are debugging lines to watch FFMPEG output :3
|
|
|
|
ffmpegProcess.stdio[1].pipe(res)
|
|
.on('close', () => {
|
|
fs.rmSync(filePath)
|
|
})
|
|
}
|
|
})
|
|
|
|
app.get("/download", async (req, res) => {
|
|
const url = req.query.url
|
|
const format = req.query.format || 'mp4'
|
|
const quality = req.query.quality || 'highest'
|
|
const defin = req.query.definition || 'hd'
|
|
const preset = 'medium'
|
|
|
|
res.setHeader("X-Accel-Buffering", "no")
|
|
res.setHeader("Content-Type", "text/html")
|
|
|
|
var downloadHTML = fs.readFileSync(path.join(__dirname, 'resources/downloading.html')).toString()
|
|
|
|
res.write(downloadHTML.substring(0, downloadHTML.indexOf("{CONTENT}")))
|
|
|
|
var fileName = Math.round(Math.random() * 100_000_000_000_000).toString() + '.' + 'webm'
|
|
var filePath = path.join(__dirname, 'downloads', fileName)
|
|
|
|
var ytdlpProcess = cp.spawn('yt-dlp', [
|
|
url,
|
|
'-o', filePath,
|
|
'--max-filesize', MAX_FILESIZE + 'm'
|
|
])
|
|
|
|
var lastDownload = 0
|
|
|
|
ytdlpProcess.stderr.setEncoding('utf-8')
|
|
ytdlpProcess.stderr.on('data', (data) => {
|
|
console.log(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(`<p>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!)</p>`)
|
|
}
|
|
else if (data.includes("[download]")) {
|
|
res.write(`<style>#downloading${lastDownload}{ display: none; }</style>`)
|
|
lastDownload += 1
|
|
res.write(`<p id="downloading${lastDownload}">` + data.substring(12) + `</p>`)
|
|
}
|
|
}
|
|
|
|
|
|
})
|
|
|
|
ytdlpProcess.on('close', () => {
|
|
if (fs.existsSync(filePath)) {
|
|
res.write(`<iframe src="/convert?file=${fileName}&format=${format}"></iframe>"`)
|
|
res.write(downloadHTML.substring(downloadHTML.indexOf("{CONTENT}") + 9), () => {res.end()})
|
|
} else {
|
|
res.write("<p>An error has occured!!! We're not exactly sure what the error is, but we cant seem to find the download file.</p>")
|
|
console.log(filePath)
|
|
}
|
|
})
|
|
|
|
})
|
|
|
|
app.use(express.static(path.join(__dirname, 'static')))
|
|
|
|
app.listen(PORT, function () {
|
|
console.log("Hosted on port " + PORT)
|
|
}) |