feat(users): users profile pictures

rivet-gg · Jun 10, 2024 · eaf3316 · eaf3316
1 parent d7a5892
commit eaf3316
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 4 deletions.
diff --git a/modules/uploads/public.ts b/modules/uploads/public.ts
@@ -0,0 +1,4 @@
+export const __CANONICAL_MODULE_NAME = "uploads";
+
+export type { UploadSize } from "./utils/data_size.ts";
+export { getBytes } from "./utils/data_size.ts";
diff --git a/modules/users/config.ts b/modules/users/config.ts
@@ -1,5 +1,7 @@
+import { Module } from "./module.gen.ts";
+
 export interface Config {
-    maxProfilePictureBytes: number;
+    maxProfilePictureSize: Module.uploads.UploadSize;
     allowedMimes?: string[];
 }
 

diff --git a/modules/users/scripts/prepare_profile_picture.ts b/modules/users/scripts/prepare_profile_picture.ts
@@ -1,4 +1,4 @@
-import { ScriptContext, RuntimeError } from "../module.gen.ts";
+import { ScriptContext, RuntimeError, Module } from "../module.gen.ts";
 import { DEFAULT_MIME_TYPES } from "../config.ts";
 
 export interface Request {
@@ -30,7 +30,8 @@ export async function run(
 	}
 
 	// Ensure the file is within the maximum configured size for a PFP
-	if (BigInt(req.contentLength) > ctx.userConfig.maxProfilePictureBytes) { 
+	const maxBytes = Module.uploads.getBytes(ctx.userConfig.maxProfilePictureSize);
+	if (BigInt(req.contentLength) > maxBytes) { 
 		throw new RuntimeError(
 			"file_too_large",
 			{ cause: `File is too large (${req.contentLength} bytes)` },

diff --git a/tests/basic/backend.json b/tests/basic/backend.json
@@ -22,7 +22,7 @@
         "users": {
             "registry": "local",
             "config": {
-                "maxProfilePictureBytes": 1048576
+                "maxProfilePictureSize": "1mib"
             }
         },
         "uploads": {