preservetube
/
backend
mirror of https://github.com/preservetube/backend.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

make videos longer than 30 minutes 360p (sorry)

This commit is contained in:
localhost 2024-03-29 15:32:12 +01:00
parent b02cc7bd63
commit 42ae296490
3 changed files with 48 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
controller/websocket.js
View File

@ -73,7 +73,7 @@ exports.save = async (ws, req) => {
async function startDownloading() {
ws.send('INFO - Spawning yt-dlp!')
const download = await ytdlp.downloadVideo(`https://www.youtube.com/watch?v=${id}`, ws)
const download = await ytdlp.downloadVideo(`https://www.youtube.com/watch?v=${id}`, ws, id)
if (download.fail) {
await redis.del(id)
ws.send(`DATA - ${download.message}`)
@ -170,7 +170,7 @@ exports.playlist = async (ws, req) => {
ws.send(`INFO - Downloading ${video.title}<br><br>`)
await redis.set(id, 'downloading')
const download = await ytdlp.downloadVideo('https://www.youtube.com' + video.url, ws)
const download = await ytdlp.downloadVideo('https://www.youtube.com' + video.url, ws, id)
if (download.fail) {
ws.send(`DATA - ${download.message}`)
await redis.del(id)
@ -269,7 +269,7 @@ exports.channel = async (ws, req) => {
ws.send(`INFO - Downloading ${video.title}<br><br>`)
await redis.set(id, 'downloading')
const download = await ytdlp.downloadVideo('https://www.youtube.com' + video.url, ws)
const download = await ytdlp.downloadVideo('https://www.youtube.com' + video.url, ws, id)
if (download.fail) {
ws.send(`DATA - ${download.message}`)
await redis.del(id)

2
utils/metadata.js
View File

@ -30,7 +30,7 @@ async function getVideoMetadata(id) {
for (let retries = 0; retries < maxRetries; retries++) {
try {
const instance = await getInstance()
const response = await fetch(`${instance}/api/v1/videos/${id}?fields=videoId,title,descriptionHtml,videoThumbnails,published,authorId,error&pretty=1`, {
const response = await fetch(`${instance}/api/v1/videos/${id}?fields=videoId,title,descriptionHtml,videoThumbnails,published,authorId,lengthSeconds,error&pretty=1`, {
headers: {
'User-Agent': 'Mozilla/5.0 (compatible; PreserveTube/0.0; +https://preservetube.com)'
}

47
utils/ytdlp.js
View File

@ -1,9 +1,19 @@
const child_process = require('child_process')
const DOMPurify = require('isomorphic-dompurify')
const metadata = require('./metadata.js')
async function downloadVideo(url, ws) {
return new Promise((resolve, reject) => {
const child = child_process.spawn("../yt-dlp", ["--proxy", "socks5://gluetun:1080", url], {cwd: 'videos', shell: false})
async function downloadVideo(url, ws, id) {
return new Promise(async (resolve, reject) => {
const args = ['--proxy', 'socks5://gluetun:1080', url]
const video = await metadata.getVideoMetadata(id)
if (video.lengthSeconds > 1500) {
const formats = await getFormats(url, ws)
if (!formats.fail && formats.includes('18 mp4')) {
args.push('-f 18')
}
}
const child = child_process.spawn('../yt-dlp', args, {cwd: 'videos', shell: false})
// https://github.com/yt-dlp/yt-dlp/blob/cc8d8441524ec3442d7c0d3f8f33f15b66aa06f3/README.md?plain=1#L1500
child.stdout.on("data", data => {
@ -34,4 +44,35 @@ async function downloadVideo(url, ws) {
})
}
async function getFormats(url, ws) {
return new Promise((resolve, reject) => {
const child = child_process.spawn("../yt-dlp", [url, "-F"], {cwd: 'videos', shell: false})
let outputs = ''
child.stdout.on("data", data => {
const msg = data.toString().trim()
if (!msg) return
outputs = outputs + msg
})
child.stderr.on("data", data => {
const msg = data.toString().trim()
if (!msg) return
if (ws) ws.send(`DATA - ${DOMPurify.sanitize(msg)}`)
})
child.on("close", async (code, signal) => {
if (code == 2 || code == 1) { // https://github.com/yt-dlp/yt-dlp/issues/4262
reject({
fail: true
})
} else {
resolve(outputs)
}
})
})
}
module.exports = { downloadVideo }