2 files changed, 15 insertions, 6 deletions
diff --git a/packages/shared/assetdb.ts b/packages/shared/assetdb.ts
index 4cea06b0..fb625af8 100644
--- a/packages/shared/assetdb.ts
+++ b/packages/shared/assetdb.ts
@@ -6,18 +6,26 @@ import serverConfig from "./config";
 
 const ROOT_PATH = path.join(serverConfig.dataDir, "assets");
 
+export const enum ASSET_TYPES {
+  IMAGE_JPEG = "image/jpeg",
+  IMAGE_PNG = "image/png",
+  IMAGE_WEBP = "image/webp",
+  APPLICATION_PDF = "application/pdf",
+  TEXT_HTML = "text/html",
+}
+
 // The assets that we allow the users to upload
-export const SUPPORTED_UPLOAD_ASSET_TYPES = new Set([
-  "image/jpeg",
-  "image/png",
-  "image/webp",
-  "application/pdf",
+export const SUPPORTED_UPLOAD_ASSET_TYPES: Set<string> = new Set<string>([
+  ASSET_TYPES.IMAGE_JPEG,
+  ASSET_TYPES.IMAGE_PNG,
+  ASSET_TYPES.IMAGE_WEBP,
+  ASSET_TYPES.APPLICATION_PDF,
 ]);
 
 // The assets that we support saving in the asset db
 export const SUPPORTED_ASSET_TYPES = new Set([
   ...SUPPORTED_UPLOAD_ASSET_TYPES,
-  "text/html",
+  ASSET_TYPES.TEXT_HTML,
 ]);
 
 function getAssetDir(userId: string, assetId: string) {
diff --git a/packages/shared/types/bookmarks.ts b/packages/shared/types/bookmarks.ts
index 06cd632e..c9e3e1a5 100644
--- a/packages/shared/types/bookmarks.ts
+++ b/packages/shared/types/bookmarks.ts
@@ -30,6 +30,7 @@ export const zBookmarkedAssetSchema = z.object({
   assetType: z.enum(["image", "pdf"]),
   assetId: z.string(),
   fileName: z.string().nullish(),
+  sourceUrl: z.string().nullish(),
 });
 export type ZBookmarkedAsset = z.infer<typeof zBookmarkedAssetSchema>;