feat: Add per user storage quota

2 files changed, 6 insertions, 0 deletions

diff --git a/packages/trpc/routers/admin.ts b/packages/trpc/routers/admin.ts
index f46e17f2..5e169857 100644
--- a/packages/trpc/routers/admin.ts
+++ b/packages/trpc/routers/admin.ts
@@ -351,6 +351,10 @@ export const adminAppRouter = router({
         updateData.bookmarkQuota = input.bookmarkQuota;
       }
 
+      if (input.storageQuota !== undefined) {
+        updateData.storageQuota = input.storageQuota;
+      }
+
       if (Object.keys(updateData).length === 0) {
         throw new TRPCError({
           code: "BAD_REQUEST",
diff --git a/packages/trpc/routers/users.ts b/packages/trpc/routers/users.ts
index ea5e6944..3c52b321 100644
--- a/packages/trpc/routers/users.ts
+++ b/packages/trpc/routers/users.ts
@@ -148,6 +148,7 @@ export const usersAppRouter = router({
             role: z.enum(["user", "admin"]).nullable(),
             localUser: z.boolean(),
             bookmarkQuota: z.number().nullable(),
+            storageQuota: z.number().nullable(),
           }),
         ),
       }),
@@ -161,6 +162,7 @@ export const usersAppRouter = router({
           role: users.role,
           password: users.password,
           bookmarkQuota: users.bookmarkQuota,
+          storageQuota: users.storageQuota,
         })
         .from(users);