375 lines
12 KiB
TypeScript
375 lines
12 KiB
TypeScript
import express from 'express';
|
|
import { Innertube } from 'youtubei.js'
|
|
import * as cheerio from 'cheerio';
|
|
import * as fs from 'node:fs'
|
|
|
|
import { EnabledTrackTypes } from 'googlevideo/utils';
|
|
import {
|
|
createOutputStream,
|
|
createStreamSink,
|
|
createSabrStream,
|
|
type DownloadOutput
|
|
} from './utils/sabr-stream-factory.js';
|
|
import type { SabrPlaybackOptions } from 'googlevideo/sabr-stream';
|
|
import { getVideoStreams, downloadStream } from './utils/companion';
|
|
|
|
const ffmpeg = require('fluent-ffmpeg')
|
|
const ffmpegStatic = require('ffmpeg-static')
|
|
|
|
const app = express();
|
|
require('express-ws')(app)
|
|
ffmpeg.setFfmpegPath(ffmpegStatic)
|
|
|
|
const maxRetries = 5
|
|
const platforms = ['IOS', 'ANDROID', 'YTSTUDIO_ANDROID', 'YTMUSIC_ANDROID']
|
|
|
|
app.get('/health', async (req, res) => {
|
|
try {
|
|
const urls = ['/video/sRMMwpDTs5k', '/channel/UCRijo3ddMTht_IHyNSNXpNQ', '/videos/UCRijo3ddMTht_IHyNSNXpNQ']
|
|
|
|
const results = await Promise.all(urls.map(async (url) => {
|
|
const response = await fetch(`http://localhost:8008${url}`);
|
|
const jsonData: any = await response.json();
|
|
const status = jsonData.error ? 'unhealthy' : 'healthy';
|
|
return { url, status };
|
|
}));
|
|
|
|
console.log('Health check results:', results);
|
|
|
|
const isHealthy = results.every(result => result.status === 'healthy');
|
|
if (isHealthy) {
|
|
res.status(200).json({ message: 'All endpoints are healthy', results });
|
|
} else {
|
|
res.status(500).json({ error: 'Health check failed', results });
|
|
switchIps()
|
|
}
|
|
} catch (error:any) {
|
|
console.error('Health check failed:', error.message);
|
|
switchIps()
|
|
res.status(500).json({ error: 'Health check failed', results: [], errorMessage: error.message });
|
|
}
|
|
})
|
|
|
|
app.get('/video/:id', async (req, res) => {
|
|
let error = ''
|
|
|
|
for (let retries = 0; retries < maxRetries; retries++) {
|
|
try {
|
|
const platform = platforms[retries % platforms.length];
|
|
const yt = await Innertube.create();
|
|
const info = await yt.getInfo(req.params.id, { // @ts-ignore
|
|
client: platform
|
|
});
|
|
|
|
if (!info) {
|
|
error = 'ErrorCantConnectToServiceAPI'
|
|
continue;
|
|
}
|
|
if (info.playability_status!.status !== 'OK') {
|
|
error = 'ErrorYTUnavailable'
|
|
continue;
|
|
}
|
|
if (info.basic_info.is_live) {
|
|
error = 'ErrorLiveVideo'
|
|
continue;
|
|
}
|
|
if (info.basic_info.title == 'Video Not Available') {
|
|
error = 'YoutubeIsFuckingWithMe'
|
|
continue;
|
|
}
|
|
return res.json(info)
|
|
} catch (error) {
|
|
continue
|
|
}
|
|
}
|
|
|
|
res.json({ error: error || 'ErrorUnknown' })
|
|
})
|
|
|
|
app.get('/channel/:id', async (req, res) => {
|
|
let error = ''
|
|
|
|
for (let retries = 0; retries < maxRetries; retries++) {
|
|
try {
|
|
const yt = await Innertube.create();
|
|
const info = await yt.getChannel(req.params.id);
|
|
|
|
if (!info) {
|
|
error = 'ErrorCantConnectToServiceAPI'
|
|
continue;
|
|
}
|
|
return res.json(info)
|
|
} catch (error) {
|
|
continue
|
|
}
|
|
}
|
|
|
|
res.json({ error: error || 'ErrorUnknown' })
|
|
})
|
|
|
|
app.get('/videos/:id', async (req, res) => {
|
|
try {
|
|
const videos = [];
|
|
const yt = await Innertube.create();
|
|
const channel = await yt.getChannel(req.params.id);
|
|
let json = await channel.getVideos();
|
|
|
|
videos.push(...json.videos);
|
|
|
|
while (json.has_continuation && videos.length < 60) {
|
|
json = await getNextPage(json);
|
|
videos.push(...json.videos);
|
|
}
|
|
|
|
return res.json(videos)
|
|
} catch (e) {
|
|
res.json(false)
|
|
}
|
|
|
|
async function getNextPage(json: any) {
|
|
const page = await json.getContinuation();
|
|
return page;
|
|
}
|
|
})
|
|
|
|
// @ts-ignore
|
|
app.ws('/download/:id', async (ws, req) => {
|
|
const config = await Bun.file('config.json').json()
|
|
const yt = await Innertube.create();
|
|
let quality = '480p'
|
|
|
|
const info = await yt.getInfo(req.params.id, {
|
|
client: 'TV_SIMPLY'
|
|
});
|
|
if (!info || info.basic_info.duration == undefined) {
|
|
ws.send('Unable to retrieve video info from YouTube. Please try again later.');
|
|
return ws.close()
|
|
}
|
|
|
|
if (info.basic_info.duration >= 900) quality = '360p'
|
|
quality = getVideoQuality(info, quality)
|
|
|
|
if (info.playability_status?.status !== 'OK') {
|
|
ws.send(`This video is not available for download (${info.playability_status?.status} ${info.playability_status?.reason}).`);
|
|
return ws.close()
|
|
} else if (info.basic_info.is_live) {
|
|
ws.send('This video is live, and cannot be downloaded.');
|
|
return ws.close()
|
|
} else if (info.basic_info.id != req.params.id) {
|
|
ws.send('This video is not available for download. Youtube is serving a different video.');
|
|
return ws.close()
|
|
}
|
|
|
|
let audioOutputStream: DownloadOutput | undefined;
|
|
let videoOutputStream: DownloadOutput | undefined;
|
|
|
|
if (config.useCompanion) {
|
|
const streamOptions = {
|
|
videoQuality: quality,
|
|
audioQuality: 'AUDIO_QUALITY_LOW'
|
|
};
|
|
const { streamResults, error } = await getVideoStreams(req.params.id, info.streaming_data!.adaptive_formats, streamOptions);
|
|
if (streamResults == false) {
|
|
ws.send(error)
|
|
return ws.close()
|
|
}
|
|
|
|
const { videoStreamUrl, audioStreamUrl, selectedFormats } = streamResults;
|
|
|
|
const videoSizeTotal = (selectedFormats.audioFormat.content_length || 0)
|
|
+ (selectedFormats.videoFormat.content_length || 0)
|
|
|
|
if (videoSizeTotal > (1_048_576 * 150) && !config.whitelist.includes(req.params.id)) {
|
|
ws.send('Is this content considered high risk? If so, please email me at admin@preservetube.com.');
|
|
ws.send('This video is too large, and unfortunately, Preservetube does not have unlimited storage.');
|
|
return ws.close()
|
|
} else if (!selectedFormats.videoFormat.content_length) {
|
|
ws.send('Youtube isn\'t giving us enough information to be able to tell if we can process this video.')
|
|
ws.send('Please try again later.')
|
|
return ws.close()
|
|
}
|
|
|
|
const audioOutputStream = createOutputStream(req.params.id, selectedFormats.audioFormat.mime_type!);
|
|
const videoOutputStream = createOutputStream(req.params.id, selectedFormats.videoFormat.mime_type!);
|
|
|
|
await Promise.all([
|
|
downloadStream(videoStreamUrl, selectedFormats.videoFormat, videoOutputStream.stream, ws, 'video'),
|
|
downloadStream(audioStreamUrl, selectedFormats.audioFormat, audioOutputStream.stream, ws, 'audio')
|
|
]);
|
|
} else {
|
|
const streamOptions: SabrPlaybackOptions = {
|
|
videoQuality: quality,
|
|
audioQuality: 'AUDIO_QUALITY_LOW',
|
|
enabledTrackTypes: EnabledTrackTypes.VIDEO_AND_AUDIO
|
|
};
|
|
const { streamResults } = await createSabrStream(req.params.id, streamOptions);
|
|
const { videoStream, audioStream, selectedFormats } = streamResults;
|
|
|
|
const config = await Bun.file('config.json').json()
|
|
const videoSizeTotal = (selectedFormats.audioFormat.contentLength || 0)
|
|
+ (selectedFormats.videoFormat.contentLength || 0)
|
|
|
|
if (videoSizeTotal > (1_048_576 * 150) && !config.whitelist.includes(req.params.id)) {
|
|
ws.send('Is this content considered high risk? If so, please email me at admin@preservetube.com.');
|
|
ws.send('This video is too large, and unfortunately, Preservetube does not have unlimited storage.');
|
|
return ws.close()
|
|
} else if (!selectedFormats.videoFormat.contentLength) {
|
|
ws.send('Youtube isn\'t giving us enough information to be able to tell if we can process this video.')
|
|
ws.send('Please try again later.')
|
|
return ws.close()
|
|
}
|
|
|
|
const audioOutputStream = createOutputStream(req.params.id, selectedFormats.audioFormat.mimeType!);
|
|
const videoOutputStream = createOutputStream(req.params.id, selectedFormats.videoFormat.mimeType!);
|
|
|
|
await Promise.all([
|
|
videoStream.pipeTo(createStreamSink(selectedFormats.videoFormat, videoOutputStream.stream, ws, 'video')),
|
|
audioStream.pipeTo(createStreamSink(selectedFormats.audioFormat, audioOutputStream.stream, ws, 'audio'))
|
|
]);
|
|
}
|
|
|
|
if (audioOutputStream == undefined || videoOutputStream == undefined) {
|
|
ws.send('This should not happen. Please report it via admin@preservetube.com.')
|
|
return ws.close()
|
|
}
|
|
|
|
ws.send('Downloaded video and audio. Merging them together.')
|
|
|
|
await mergeIt(audioOutputStream.filePath, videoOutputStream.filePath, `./output/${req.params.id}.mp4`, ws)
|
|
await cleanupTempFiles([ audioOutputStream.filePath, videoOutputStream.filePath ]);
|
|
|
|
ws.send('done')
|
|
ws.close()
|
|
});
|
|
|
|
app.get('/getWebpageJson', async (req, res) => {
|
|
if (!req.query.url) return res.send('no url')
|
|
|
|
const ytRes = await fetch(req.query.url as string);
|
|
if (!ytRes.ok) return res.status(500).send('failed to fetch youtube url')
|
|
|
|
const html = await ytRes.text();
|
|
const $ = cheerio.load(html);
|
|
let found: string | null = null;
|
|
|
|
$('script').each((_, el) => {
|
|
const scriptContent = $(el).html();
|
|
if (scriptContent && scriptContent.includes('var ytInitialData = ')) {
|
|
const jsonStr = scriptContent.split('var ytInitialData = ')[1]?.split("};")[0];
|
|
if (jsonStr) {
|
|
found = jsonStr + "}";
|
|
return false;
|
|
}
|
|
}
|
|
});
|
|
|
|
if (!found) return res.status(500).send('failed to find youtube json')
|
|
try {
|
|
res.json(JSON.parse(found))
|
|
} catch (_) {
|
|
res.status(500).send('failed to parse youtube json')
|
|
}
|
|
})
|
|
|
|
function getVideoQuality(json: any, quality: string) {
|
|
const adaptiveFormats = json['streaming_data']['adaptive_formats'];
|
|
let video = adaptiveFormats.find((f: any) => f.quality_label === quality && !f.has_audio);
|
|
|
|
if (!video) {
|
|
const target = parseInt(quality);
|
|
video = adaptiveFormats // find the quality thats closest to the one we wanted
|
|
.filter((f: any) => !f.has_audio && f.quality_label)
|
|
.reduce((prev: any, curr: any) => {
|
|
const currDiff = Math.abs(parseInt(curr.quality_label) - target);
|
|
const prevDiff = prev ? Math.abs(parseInt(prev.quality_label) - target) : Infinity;
|
|
return currDiff < prevDiff ? curr : prev;
|
|
}, null);
|
|
}
|
|
|
|
return video ? video.quality_label : null;
|
|
}
|
|
|
|
function mergeIt(audioPath: string, videoPath: string, outputPath: string, ws: any) {
|
|
return new Promise((resolve, reject) => {
|
|
ffmpeg()
|
|
.input(videoPath)
|
|
.input(audioPath)
|
|
.outputOptions([ '-c:v copy', '-c:a copy', '-map 0:v:0', '-map 1:a:0', '-movflags +faststart' ])
|
|
.on('progress', (progress:any) => {
|
|
if (progress.percent) {
|
|
ws.send(`[merging] ${progress.precent}% done`)
|
|
}
|
|
})
|
|
.on('end', () => {
|
|
resolve(outputPath);
|
|
})
|
|
.on('error', (err:any) => {
|
|
reject(new Error(`Error merging files: ${err.message}`));
|
|
})
|
|
.save(outputPath);
|
|
});
|
|
}
|
|
|
|
async function cleanupTempFiles(files: string[]) {
|
|
for (const file of files) {
|
|
try {
|
|
fs.unlinkSync(file);
|
|
} catch (error) {
|
|
console.warn(`Failed to delete temp file ${file}:`, error);
|
|
}
|
|
}
|
|
}
|
|
|
|
async function switchIps() {
|
|
const currentIp: any = await (await fetch('http://localhost:8000/v1/publicip/ip', {
|
|
headers: {
|
|
'X-API-Key': '64d1781e469965c1cdad611b0c05d313'
|
|
}
|
|
})).json()
|
|
const currentDate = new Date()
|
|
|
|
console.log(`starting switching ips. ${currentIp.public_ip}, ${currentIp.city}, ${currentIp.region}, ${currentIp.organization}`)
|
|
|
|
const s = await fetch('http://localhost:8000/v1/vpn/status', {
|
|
method: 'PUT',
|
|
headers: {
|
|
'X-API-Key': '64d1781e469965c1cdad611b0c05d313',
|
|
'Content-Type': 'application/json'
|
|
},
|
|
body: JSON.stringify({ status: 'stopped' })
|
|
})
|
|
console.log(`stopped vpn - ${await s.text()}`)
|
|
|
|
const r = await fetch('http://localhost:8000/v1/vpn/status', {
|
|
method: 'PUT',
|
|
headers: {
|
|
'X-API-Key': '64d1781e469965c1cdad611b0c05d313',
|
|
'Content-Type': 'application/json'
|
|
},
|
|
body: JSON.stringify({ status: 'running' })
|
|
})
|
|
console.log(`turned on vpn - ${await r.text()}`)
|
|
|
|
await new Promise((resolve, reject) => {
|
|
const intervalId = setInterval(async () => {
|
|
const newIp: any = await (await fetch('http://localhost:8000/v1/publicip/ip', {
|
|
headers: {
|
|
'X-API-Key': '64d1781e469965c1cdad611b0c05d313',
|
|
},
|
|
})).json();
|
|
|
|
if (newIp.public_ip !== '') {
|
|
console.log(`finished switching ips. ${newIp.public_ip}, ${newIp.city}, ${newIp.region}, ${newIp.organization}. took ${(new Date().getTime() - currentDate.getTime()) / 1000}s`)
|
|
clearInterval(intervalId);
|
|
resolve('done')
|
|
}
|
|
}, 500);
|
|
})
|
|
}
|
|
|
|
setInterval(switchIps, 30 * 60000) // 30 minutes
|
|
|
|
app.listen(8008, () => {
|
|
console.log('the metadata server is up.')
|
|
// switchIps()
|
|
}) |