50 lines
1.2 KiB
JavaScript
50 lines
1.2 KiB
JavaScript
const ytdl = require('ytdl-core'),
|
|
fs = require('fs'),
|
|
path = require('path'),
|
|
express = require('express'),
|
|
ffmpeg = require('fluent-ffmpeg'),
|
|
bodyParser = require('body-parser')
|
|
|
|
const PORT = process.env.PORT || 8080
|
|
const app = express()
|
|
|
|
process.on('uncaughtException', (err, origin) => {
|
|
fs.writeSync(
|
|
process.stderr.fd,
|
|
`Caught exception: ${err}\n` +
|
|
`Exception origin: ${origin}`,
|
|
);
|
|
});
|
|
|
|
app.use(bodyParser.urlencoded({ extended: false }))
|
|
|
|
app.get("/download", (req, res) => {
|
|
const filename = req.query.filename
|
|
const url = req.query.url
|
|
const format = req.query.format
|
|
const quality = req.query.quality
|
|
|
|
if (!ytdl.validateURL(url)) {
|
|
res.send("Invalid URL!")
|
|
return
|
|
}
|
|
|
|
if (format != "mp4" && format != "webm") {
|
|
res.header('Content-Disposition', `attachment; filename="${filename}.${format};"`);
|
|
|
|
var proc = new ffmpeg({ source: ytdl(url, { 'quality': quality }) })
|
|
.format(format)
|
|
.pipe(res)
|
|
|
|
} else {
|
|
res.header('Content-Disposition', `attachment; filename="${filename}.${format}"`);
|
|
ytdl(url, { 'format': format, 'quality': quality }).pipe(res);
|
|
}
|
|
|
|
});
|
|
|
|
app.use(express.static(path.join(__dirname, 'static')))
|
|
|
|
app.listen(PORT, function () {
|
|
console.log("Hosted on port " + PORT)
|
|
}) |