OpenFrame/lib/r2-cleanup.ts

import { DeleteObjectCommand } from '@aws-sdk/client-s3';
import { r2Client, R2_BUCKET_NAME } from '@/lib/r2';
import { db } from '@/lib/db';
import { runWithConcurrency } from '@/lib/async-pool';

/** The path prefix for images served by the upload API. */
const IMAGE_PATH_PREFIX = '/api/upload/image/';
/** The path prefix for audio URLs served by the upload API. */
const AUDIO_PATH_PREFIX = '/api/upload/audio/';
const CLEANUP_DELETE_CONCURRENCY = 5;

/**
 * Extract the R2 object key from a media URL.
 * Uses string parsing instead of regex to avoid ReDoS risk on untrusted input.
 */
function mediaUrlToKey(url: string): string | null {
    if (url.includes(AUDIO_PATH_PREFIX)) {
        const filename = url.slice(url.indexOf(AUDIO_PATH_PREFIX) + AUDIO_PATH_PREFIX.length);
        return filename ? `voice/${filename}` : null;
    } else if (url.includes(IMAGE_PATH_PREFIX)) {
        const filename = url.slice(url.indexOf(IMAGE_PATH_PREFIX) + IMAGE_PATH_PREFIX.length);
        return filename ? `images/${filename}` : null;
    }
    return null;
}

/**
 * Delete a list of media files from R2 (best-effort, logs failures).
 */
async function deleteMediaFiles(mediaUrls: string[]) {
    const mediaKeys = [...new Set(mediaUrls.map(mediaUrlToKey).filter((key): key is string => Boolean(key)))];

    await runWithConcurrency(mediaKeys, CLEANUP_DELETE_CONCURRENCY, async (key) => {
        try {
            await r2Client.send(
                new DeleteObjectCommand({ Bucket: R2_BUCKET_NAME, Key: key })
            );
        } catch (err) {
            console.error(`Failed to delete media from R2 (key: ${key}):`, err);
        }
    });
}

/**
 * Collect all media URLs from comments under a given video (all versions).
 */
export async function collectVideoMediaUrls(videoId: string): Promise<string[]> {
    const [comments, assets] = await Promise.all([
        db.comment.findMany({
            where: {
                OR: [{ voiceUrl: { not: null } }, { imageUrl: { not: null } }],
                version: { videoParentId: videoId },
            },
            select: { voiceUrl: true, imageUrl: true },
        }),
        db.videoAsset.findMany({
            where: {
                videoId,
                provider: 'R2_IMAGE',
            },
            select: { sourceUrl: true },
        }),
    ]);
    const urls: string[] = [];
    comments.forEach(c => {
        if (c.voiceUrl) urls.push(c.voiceUrl);
        if (c.imageUrl) urls.push(c.imageUrl);
    });
    assets.forEach((asset) => {
        if (asset.sourceUrl) urls.push(asset.sourceUrl);
    });
    return urls;
}

/**
 * Collect all media URLs from comments under all videos in a project.
 */
export async function collectProjectMediaUrls(projectId: string): Promise<string[]> {
    const [comments, assets] = await Promise.all([
        db.comment.findMany({
            where: {
                OR: [{ voiceUrl: { not: null } }, { imageUrl: { not: null } }],
                version: { video: { projectId } },
            },
            select: { voiceUrl: true, imageUrl: true },
        }),
        db.videoAsset.findMany({
            where: {
                provider: 'R2_IMAGE',
                video: { projectId },
            },
            select: { sourceUrl: true },
        }),
    ]);
    const urls: string[] = [];
    comments.forEach(c => {
        if (c.voiceUrl) urls.push(c.voiceUrl);
        if (c.imageUrl) urls.push(c.imageUrl);
    });
    assets.forEach((asset) => {
        if (asset.sourceUrl) urls.push(asset.sourceUrl);
    });
    return urls;
}

/**
 * Collect all media URLs from comments under all projects in a workspace.
 */
export async function collectWorkspaceMediaUrls(workspaceId: string): Promise<string[]> {
    const [comments, assets] = await Promise.all([
        db.comment.findMany({
            where: {
                OR: [{ voiceUrl: { not: null } }, { imageUrl: { not: null } }],
                version: { video: { project: { workspaceId } } },
            },
            select: { voiceUrl: true, imageUrl: true },
        }),
        db.videoAsset.findMany({
            where: {
                provider: 'R2_IMAGE',
                video: { project: { workspaceId } },
            },
            select: { sourceUrl: true },
        }),
    ]);
    const urls: string[] = [];
    comments.forEach(c => {
        if (c.voiceUrl) urls.push(c.voiceUrl);
        if (c.imageUrl) urls.push(c.imageUrl);
    });
    assets.forEach((asset) => {
        if (asset.sourceUrl) urls.push(asset.sourceUrl);
    });
    return urls;
}

/**
 * Delete all media files for a video from R2.
 * Call BEFORE deleting the video from the database (cascade would remove comment rows).
 */
export async function cleanupVideoMediaFiles(videoId: string) {
    const urls = await collectVideoMediaUrls(videoId);
    await deleteMediaFiles(urls);
}

/**
 * Delete all media files for a project from R2.
 * Call BEFORE deleting the project from the database.
 */
export async function cleanupProjectMediaFiles(projectId: string) {
    const urls = await collectProjectMediaUrls(projectId);
    await deleteMediaFiles(urls);
}

/**
 * Delete all media files for a workspace from R2.
 * Call BEFORE deleting the workspace from the database.
 */
export async function cleanupWorkspaceMediaFiles(workspaceId: string) {
    const urls = await collectWorkspaceMediaUrls(workspaceId);
    await deleteMediaFiles(urls);
}