From 1b7600a2f615a6be17e424b4428f54f10ec4faeb Mon Sep 17 00:00:00 2001
From: Adrian Bonpin <adrianbonpin@gmail.com>
Date: Sat, 16 May 2026 17:36:06 +0800
Subject: [PATCH] feat: support removing screenshots during entry edit

---
 lib/api/performance.ts | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/lib/api/performance.ts b/lib/api/performance.ts
index c323ceb..e7c7fa2 100644
--- a/lib/api/performance.ts
+++ b/lib/api/performance.ts
@@ -402,6 +402,38 @@ export const performanceVerifyRoutes = new Elysia({
         .where(eq(performanceEntries.id, params.id))
         .returning()
 
+      // ── Remove screenshots marked for deletion ────────────────────────
+      const removedIds: string[] = Array.isArray(payload.removedScreenshotIds)
+        ? payload.removedScreenshotIds
+        : []
+
+      if (removedIds.length > 0) {
+        // Fetch storage keys before deleting rows
+        const toRemove = await db
+          .select({ id: entryScreenshots.id, storageKey: entryScreenshots.storageKey })
+          .from(entryScreenshots)
+          .where(
+            and(
+              eq(entryScreenshots.entryId, params.id),
+              sql`${entryScreenshots.id} = ANY(${removedIds})`,
+            ),
+          )
+
+        for (const ss of toRemove) {
+          try { await deleteObject(ss.storageKey) } catch { /* best-effort */ }
+          try {
+            await db.delete(storageObjects).where(eq(storageObjects.key, ss.storageKey))
+          } catch { /* best-effort */ }
+        }
+
+        // Delete the screenshot rows
+        if (toRemove.length > 0) {
+          await db.delete(entryScreenshots).where(
+            sql`${entryScreenshots.id} = ANY(${toRemove.map((s) => s.id)})`,
+          )
+        }
+      }
+
       // ── Atomic screenshot replacement ─────────────────────────────────
       if (screenshotFiles.length > 0) {
         // a. Upload all new screenshots to R2 FIRST (before deleting old ones)