refactor: Start tracking bookmark assets in the assets table

1 files changed, 19 insertions, 0 deletions

diff --git a/packages/trpc/lib/attachments.ts b/packages/trpc/lib/attachments.ts
index 6fe1ef40..175947f8 100644
--- a/packages/trpc/lib/attachments.ts
+++ b/packages/trpc/lib/attachments.ts
@@ -8,6 +8,8 @@ export function mapDBAssetTypeToUserType(assetType: AssetTypes): ZAssetType {
     [AssetTypes.LINK_SCREENSHOT]: "screenshot",
     [AssetTypes.LINK_FULL_PAGE_ARCHIVE]: "fullPageArchive",
     [AssetTypes.LINK_BANNER_IMAGE]: "bannerImage",
+    [AssetTypes.BOOKMARK_ASSET]: "bookmarkAsset",
+    [AssetTypes.UNKNOWN]: "bannerImage",
   };
   return map[assetType];
 }
@@ -19,6 +21,8 @@ export function mapSchemaAssetTypeToDB(
     screenshot: AssetTypes.LINK_SCREENSHOT,
     fullPageArchive: AssetTypes.LINK_FULL_PAGE_ARCHIVE,
     bannerImage: AssetTypes.LINK_BANNER_IMAGE,
+    bookmarkAsset: AssetTypes.BOOKMARK_ASSET,
+    unknown: AssetTypes.UNKNOWN,
   };
   return map[assetType];
 }
@@ -28,6 +32,8 @@ export function humanFriendlyNameForAssertType(type: ZAssetType) {
     screenshot: "Screenshot",
     fullPageArchive: "Full Page Archive",
     bannerImage: "Banner Image",
+    bookmarkAsset: "Bookmark Asset",
+    unknown: "Unknown",
   };
   return map[type];
 }
@@ -37,6 +43,19 @@ export function isAllowedToAttachAsset(type: ZAssetType) {
     screenshot: true,
     fullPageArchive: false,
     bannerImage: true,
+    bookmarkAsset: false,
+    unknown: false,
+  };
+  return map[type];
+}
+
+export function isAllowedToDetachAsset(type: ZAssetType) {
+  const map: Record<ZAssetType, boolean> = {
+    screenshot: true,
+    fullPageArchive: true,
+    bannerImage: true,
+    bookmarkAsset: false,
+    unknown: false,
   };
   return map[type];
 }