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

aaa

This commit is contained in:
localhost 2024-06-29 11:31:01 +02:00
parent b83f44240d
commit 777b371fff
2 changed files with 31 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
controller/websocket.js
View File

@ -58,7 +58,7 @@ exports.save = async (ws, req) => {
if (await redis.get(id) != 'downloading') { if (await redis.get(id) != 'downloading') {
await redis.set(id, 'downloading', 'EX', 300) await redis.set(id, 'downloading', 'EX', 300)
const confirm = await captcha.checkCaptcha(msg) const confirm = true //await captcha.checkCaptcha(msg)
if (confirm) startDownloading() if (confirm) startDownloading()
else { else {

35
utils/ytdlp.js
View File

@ -36,9 +36,21 @@ async function downloadVideo(url, ws, id) {
let speed = 0 let speed = 0
const alreadyPrecentages = [] const alreadyPrecentages = []
console.log(downloadJson.url[1])
const target = Array.isArray(downloadJson.url) ? downloadJson.url[0] : downloadJson.url const target = Array.isArray(downloadJson.url) ? downloadJson.url[0] : downloadJson.url
const download = wget.download(target, `./videos/${id}.mp4`) const download = wget.download(target, `./videos/${id}.mp4`, {
proxy: {
protocol: 'http',
host: 'gluetun',
port: '8888',
headers: {
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36',
accept: '*/*',
origin: 'https://www.youtube.com',
referer: 'https://www.youtube.com',
DNT: '?1'
}
}
})
download.on('start', fileSize => { download.on('start', fileSize => {
size = fileSize size = fileSize
@ -77,7 +89,20 @@ async function downloadVideo(url, ws, id) {
if (Array.isArray(downloadJson.url)) { // this means video is separate if (Array.isArray(downloadJson.url)) { // this means video is separate
ws.send(`DATA - Downloading the video has finished, downloading audio.`); ws.send(`DATA - Downloading the video has finished, downloading audio.`);
const audioDownload = wget.download(downloadJson.url[1], `./videos/${id}_audio.mp4`) const audioDownload = wget.download(downloadJson.url[1], `./videos/${id}_audio.mp4`, {
proxy: {
protocol: 'http',
host: 'gluetun',
port: '8888',
headers: {
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36',
accept: '*/*',
origin: 'https://www.youtube.com',
referer: 'https://www.youtube.com',
DNT: '?1'
}
}
})
audioDownload.on('end', async (audioOutput) => { audioDownload.on('end', async (audioOutput) => {
if (audioOutput !== 'Finished writing to disk') { if (audioOutput !== 'Finished writing to disk') {
return resolve({ fail: true }); return resolve({ fail: true });
@ -88,8 +113,8 @@ async function downloadVideo(url, ws, id) {
await mergeIt(`./videos/${id}_audio.mp4`, `./videos/${id}_video.mp4`, `./videos/${id}.mp4`); await mergeIt(`./videos/${id}_audio.mp4`, `./videos/${id}_video.mp4`, `./videos/${id}.mp4`);
ws.send(`DATA - Merging succeeded.`) ws.send(`DATA - Merging succeeded.`)
fs.rmdirSync(`./videos/${id}_audio.mp4`) fs.rmSync(`./videos/${id}_audio.mp4`)
fs.rmdirSync(`./videos/${id}_video.mp4`) fs.rmSync(`./videos/${id}_video.mp4`)
return resolve({ fail: false }); return resolve({ fail: false });
} catch (error) { } catch (error) {