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

lower ratelimit, avoid crashing on video error

This commit is contained in:
localhost 2026-03-07 10:55:36 +01:00
parent 7429a9603a
commit 4afe15a715
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/router/websocket.ts
View File

@ -14,7 +14,7 @@ import { parseSlop } from '@/utils/slop';
const app = new Elysia() const app = new Elysia()
const videoIds: Record<string, string> = {} const videoIds: Record<string, string> = {}
const MB_LIMIT = 500 const MB_LIMIT = 350
const saveKey = (videoId: string) => `save:${videoId}` const saveKey = (videoId: string) => `save:${videoId}`
const checkMbLimit = async (hash: string, mb?: number): Promise<boolean> => { const checkMbLimit = async (hash: string, mb?: number): Promise<boolean> => {
@ -218,7 +218,10 @@ app.ws('/savechannel', {
videoIds[ws.id] = `downloading-${channelId}`; videoIds[ws.id] = `downloading-${channelId}`;
const videos = await getChannelVideos(channelId); const videos = await getChannelVideos(channelId);
const hash = Bun.hash(getRateLimitKey(ws.data.headers['cf-connecting-ip'] || '0.0.0.0')) if (!Array.isArray(videos)) {
await cleanup(ws, channelId);
return sendError(ws, 'Unable to retrieve channel videos from YouTube. Please try again later.');
}
for (const video of videos.slice(0, 5)) { for (const video of videos.slice(0, 5)) {
if (!video || (await redis.get(saveKey(video.video_id))) || (await redis.get(`blacklist:${video.video_id}`))) continue; if (!video || (await redis.get(saveKey(video.video_id))) || (await redis.get(`blacklist:${video.video_id}`))) continue;
@ -277,4 +280,4 @@ app.ws('/savechannel', {
}) })
app.onError(error) app.onError(error)
export default app export default app