From 1de23bf8b67f5f56695d627ba9db81faeb88eadd Mon Sep 17 00:00:00 2001
From: Andrea <andrea@avris.it>
Date: Fri, 1 Apr 2022 00:04:08 +0200
Subject: [PATCH] #269 reduce S3 usage - clean up unused cards

---
 server/cleanupImages.js | 69 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)

diff --git a/server/cleanupImages.js b/server/cleanupImages.js
index 1a9da071..10b86c28 100644
--- a/server/cleanupImages.js
+++ b/server/cleanupImages.js
@@ -54,10 +54,32 @@ async function cleanup() {
         }
     }
 
+    const cards = {};
+    for (let row of await db.all(`
+        SELECT card
+        FROM profiles
+        WHERE card is not null AND card != ''
+    `)) {
+        cards[row.card.match('https://pronouns-page.s3.eu-west-1.amazonaws.com/card/[^/]+/.+-([^-]+)\.png')[1]] = true;
+    }
+    for (let row of await db.all(`
+        SELECT cardDark
+        FROM profiles
+        WHERE cardDark is not null AND cardDark != ''
+    `)) {
+        const m = row.cardDark.match('https://pronouns-page.s3.eu-west-1.amazonaws.com/card/[^/]+/.+-([^-]+)-dark\.png');
+        if (!m) {
+            console.error(row.cardDark);
+            continue;
+        }
+        cards[m[1]] = true;
+    }
+
     console.log('Avatars: ' + Object.keys(avatars).length);
     console.log('Flags: ' + Object.keys(flags).length);
     console.log('Sources: ' + Object.keys(sources).length);
     console.log('Terms: ' + Object.keys(terms).length);
+    console.log('Cards: ' + Object.keys(cards).length);
 
     await db.close();
 
@@ -132,6 +154,53 @@ async function cleanup() {
         }
     }
 
+    console.log('--- Cards ---');
+    marker = undefined;
+    while (true) {
+        console.log('Making a request');
+        const objects = await s3.listObjects({
+            Prefix: `card/`,
+            MaxKeys: chunkSize,
+            Marker: marker,
+        }).promise();
+
+        const toRemove = [];
+
+        for (let object of objects.Contents) {
+            overall++;
+            marker = object.Key;
+
+            if (object.LastModified > new Date() - 60*60*1000) {
+                fresh++;
+                continue;
+            }
+
+            const id = object.Key.endsWith('-dark.png')
+                ? object.Key.match('card/[^/]+/.+-([^-]+)-dark\.png')[1]
+                : object.Key.match('card/[^/]+/.+-([^-]+)\.png')[1];
+
+            if (!cards[id]) {
+                console.log(`REMOVING: ${object.Key}`);
+                toRemove.push({Key: object.Key});
+                removed += 1;
+                removedSize += object.Size;
+            }
+        }
+
+        if (execute && toRemove.length) {
+            console.log('--- Removal request ---');
+            await s3.deleteObjects({
+                Delete: {
+                    Objects: toRemove,
+                }
+            }).promise();
+        }
+
+        if (objects.Contents.length < chunkSize) {
+            break;
+        }
+    }
+
     console.log('--- Summary ---');
     console.log('Overall: ' + overall);
     console.log('Fresh: ' + fresh);