feat: Add support for singlefile extension uploads. #172

2 files changed, 13 insertions, 1 deletions

diff --git a/packages/shared/assetdb.ts b/packages/shared/assetdb.ts
index 2ef69279..d2b9eebf 100644
--- a/packages/shared/assetdb.ts
+++ b/packages/shared/assetdb.ts
@@ -25,6 +25,13 @@ export const IMAGE_ASSET_TYPES: Set<string> = new Set<string>([
 // The assets that we allow the users to upload
 export const SUPPORTED_UPLOAD_ASSET_TYPES: Set<string> = new Set<string>([
   ...IMAGE_ASSET_TYPES,
+  ASSET_TYPES.TEXT_HTML,
+  ASSET_TYPES.APPLICATION_PDF,
+]);
+
+// The assets that we allow as a bookmark of type asset
+export const SUPPORTED_BOOKMARK_ASSET_TYPES: Set<string> = new Set<string>([
+  ...IMAGE_ASSET_TYPES,
   ASSET_TYPES.APPLICATION_PDF,
 ]);
 
diff --git a/packages/shared/types/bookmarks.ts b/packages/shared/types/bookmarks.ts
index a1e39280..0a414ff9 100644
--- a/packages/shared/types/bookmarks.ts
+++ b/packages/shared/types/bookmarks.ts
@@ -18,6 +18,7 @@ export const zAssetTypesSchema = z.enum([
   "fullPageArchive",
   "video",
   "bookmarkAsset",
+  "precrawledArchive",
   "unknown",
 ]);
 export type ZAssetType = z.infer<typeof zAssetTypesSchema>;
@@ -126,7 +127,11 @@ export const zNewBookmarkRequestSchema = z
   })
   .and(
     z.discriminatedUnion("type", [
-      z.object({ type: z.literal(BookmarkTypes.LINK), url: z.string().url() }),
+      z.object({
+        type: z.literal(BookmarkTypes.LINK),
+        url: z.string().url(),
+        precrawledArchiveId: z.string().optional(),
+      }),
       z.object({
         type: z.literal(BookmarkTypes.TEXT),
         text: z.string(),