mirror of
https://github.com/yusufipk/OpenFrame.git
synced 2026-09-11 17:46:06 +00:00
feat: implement R2 audio file management and rate limiting enhancements
- Add R2 client setup and audio upload functionality in lib/r2.ts. - Create audio file cleanup functions in lib/r2-cleanup.ts to delete voice files associated with videos, projects, and workspaces. - Enhance rate limiting in lib/rate-limit.ts with new action-specific limits and improved IP validation. - Introduce a unified rate limit check function that returns a 429 response when limits are exceeded. - Update package.json to include the AWS SDK for S3.
This commit is contained in:
@@ -0,0 +1,104 @@
|
||||
import { DeleteObjectCommand } from '@aws-sdk/client-s3';
|
||||
import { r2Client, R2_BUCKET_NAME } from '@/lib/r2';
|
||||
import { db } from '@/lib/db';
|
||||
|
||||
/** The path prefix for audio URLs served by the upload API. */
|
||||
const AUDIO_PATH_PREFIX = '/api/upload/audio/';
|
||||
|
||||
/**
|
||||
* Extract the R2 object key from a voice URL like /api/upload/audio/filename.webm.
|
||||
* Uses string parsing instead of regex to avoid ReDoS risk on untrusted input.
|
||||
*/
|
||||
function voiceUrlToKey(url: string): string | null {
|
||||
const idx = url.indexOf(AUDIO_PATH_PREFIX);
|
||||
if (idx === -1) return null;
|
||||
const filename = url.slice(idx + AUDIO_PATH_PREFIX.length);
|
||||
return filename ? `voice/${filename}` : null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete a list of voice files from R2 (best-effort, logs failures).
|
||||
*/
|
||||
async function deleteVoiceFiles(voiceUrls: string[]) {
|
||||
for (const url of voiceUrls) {
|
||||
try {
|
||||
const key = voiceUrlToKey(url);
|
||||
if (key) {
|
||||
await r2Client.send(
|
||||
new DeleteObjectCommand({ Bucket: R2_BUCKET_NAME, Key: key })
|
||||
);
|
||||
}
|
||||
} catch (err) {
|
||||
console.error('Failed to delete audio from R2:', err);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Collect all voice URLs from comments under a given video (all versions).
|
||||
*/
|
||||
export async function collectVideoVoiceUrls(videoId: string): Promise<string[]> {
|
||||
const comments = await db.comment.findMany({
|
||||
where: {
|
||||
voiceUrl: { not: null },
|
||||
version: { videoParentId: videoId },
|
||||
},
|
||||
select: { voiceUrl: true },
|
||||
});
|
||||
return comments.map((c) => c.voiceUrl).filter(Boolean) as string[];
|
||||
}
|
||||
|
||||
/**
|
||||
* Collect all voice URLs from comments under all videos in a project.
|
||||
*/
|
||||
export async function collectProjectVoiceUrls(projectId: string): Promise<string[]> {
|
||||
const comments = await db.comment.findMany({
|
||||
where: {
|
||||
voiceUrl: { not: null },
|
||||
version: { video: { projectId } },
|
||||
},
|
||||
select: { voiceUrl: true },
|
||||
});
|
||||
return comments.map((c) => c.voiceUrl).filter(Boolean) as string[];
|
||||
}
|
||||
|
||||
/**
|
||||
* Collect all voice URLs from comments under all projects in a workspace.
|
||||
*/
|
||||
export async function collectWorkspaceVoiceUrls(workspaceId: string): Promise<string[]> {
|
||||
const comments = await db.comment.findMany({
|
||||
where: {
|
||||
voiceUrl: { not: null },
|
||||
version: { video: { project: { workspaceId } } },
|
||||
},
|
||||
select: { voiceUrl: true },
|
||||
});
|
||||
return comments.map((c) => c.voiceUrl).filter(Boolean) as string[];
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete all voice files for a video from R2.
|
||||
* Call BEFORE deleting the video from the database (cascade would remove comment rows).
|
||||
*/
|
||||
export async function cleanupVideoVoiceFiles(videoId: string) {
|
||||
const urls = await collectVideoVoiceUrls(videoId);
|
||||
await deleteVoiceFiles(urls);
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete all voice files for a project from R2.
|
||||
* Call BEFORE deleting the project from the database.
|
||||
*/
|
||||
export async function cleanupProjectVoiceFiles(projectId: string) {
|
||||
const urls = await collectProjectVoiceUrls(projectId);
|
||||
await deleteVoiceFiles(urls);
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete all voice files for a workspace from R2.
|
||||
* Call BEFORE deleting the workspace from the database.
|
||||
*/
|
||||
export async function cleanupWorkspaceVoiceFiles(workspaceId: string) {
|
||||
const urls = await collectWorkspaceVoiceUrls(workspaceId);
|
||||
await deleteVoiceFiles(urls);
|
||||
}
|
||||
Reference in New Issue
Block a user