feat: Implement admin dashboard with user management, statistics, and role-based access control.

2026-09-11 09:36:08 +00:00 · 2026-02-20 15:17:32 +03:00
parent 1bb1c8e574
commit 83eeffe5c0
10 changed files with 685 additions and 5 deletions
@@ -0,0 +1,81 @@
+import { unstable_cache } from 'next/cache';
+import { db } from '@/lib/db';
+import { r2Client, R2_BUCKET_NAME } from '@/lib/r2';
+import { ListObjectsV2Command } from '@aws-sdk/client-s3';
+
+// Cache for 10 minutes (600 seconds)
+export const getCachedTotalStorage = unstable_cache(
+    async () => {
+        let totalStorageBytes = 0;
+        try {
+            let isTruncated = true;
+            let continuationToken: string | undefined = undefined;
+
+            while (isTruncated) {
+                const commandParams: any = { Bucket: R2_BUCKET_NAME };
+                if (continuationToken) commandParams.ContinuationToken = continuationToken;
+
+                const data = await r2Client.send(new ListObjectsV2Command(commandParams));
+                if (data.Contents) {
+                    for (const item of data.Contents) {
+                        totalStorageBytes += item.Size || 0;
+                    }
+                }
+                isTruncated = data.IsTruncated ?? false;
+                continuationToken = data.NextContinuationToken;
+            }
+        } catch (err) {
+            console.error('Failed to fetch total storage stats:', err);
+            return -1;
+        }
+        return totalStorageBytes;
+    },
+    ['admin-total-storage'],
+    { revalidate: 600 }
+);
+
+export const getCachedUserVoiceStorage = unstable_cache(
+    async () => {
+        // Return a plain object so it maps cleanly out of unstable_cache across requests
+        const userStorage: Record<string, number> = {};
+        try {
+            const fileSizes = new Map<string, number>();
+            let isTruncated = true;
+            let continuationToken: string | undefined = undefined;
+
+            while (isTruncated) {
+                const commandParams: any = { Bucket: R2_BUCKET_NAME };
+                if (continuationToken) commandParams.ContinuationToken = continuationToken;
+
+                const data = await r2Client.send(new ListObjectsV2Command(commandParams));
+                if (data.Contents) {
+                    for (const item of data.Contents) {
+                        if (item.Key) fileSizes.set(item.Key, item.Size || 0);
+                    }
+                }
+                isTruncated = data.IsTruncated ?? false;
+                continuationToken = data.NextContinuationToken;
+            }
+
+            const voiceComments = await db.comment.findMany({
+                where: { voiceUrl: { not: null }, authorId: { not: null } },
+                select: { authorId: true, voiceUrl: true }
+            });
+
+            for (const comment of voiceComments) {
+                if (!comment.authorId || !comment.voiceUrl) continue;
+                const keyParts = comment.voiceUrl.split('/');
+                const filename = keyParts[keyParts.length - 1];
+                const r2Key = `voice/${filename}`;
+                const size = fileSizes.get(r2Key) || 0;
+
+                userStorage[comment.authorId] = (userStorage[comment.authorId] || 0) + size;
+            }
+        } catch (err) {
+            console.error('Failed to parse user storage:', err);
+        }
+        return userStorage;
+    },
+    ['admin-user-voice-storage'],
+    { revalidate: 600 }
+);