3 files changed, 11 insertions, 0 deletions
diff --git a/packages/trpc/lib/attachments.ts b/packages/trpc/lib/attachments.ts
index fb9e2079..f3170c22 100644
--- a/packages/trpc/lib/attachments.ts
+++ b/packages/trpc/lib/attachments.ts
@@ -9,6 +9,7 @@ import {
 export function mapDBAssetTypeToUserType(assetType: AssetTypes): ZAssetType {
   const map: Record<AssetTypes, z.infer<typeof zAssetTypesSchema>> = {
     [AssetTypes.LINK_SCREENSHOT]: "screenshot",
+    [AssetTypes.LINK_PDF]: "pdf",
     [AssetTypes.ASSET_SCREENSHOT]: "assetScreenshot",
     [AssetTypes.LINK_FULL_PAGE_ARCHIVE]: "fullPageArchive",
     [AssetTypes.LINK_PRECRAWLED_ARCHIVE]: "precrawledArchive",
@@ -29,6 +30,7 @@ export function mapSchemaAssetTypeToDB(
 ): AssetTypes {
   const map: Record<ZAssetType, AssetTypes> = {
     screenshot: AssetTypes.LINK_SCREENSHOT,
+    pdf: AssetTypes.LINK_PDF,
     assetScreenshot: AssetTypes.ASSET_SCREENSHOT,
     fullPageArchive: AssetTypes.LINK_FULL_PAGE_ARCHIVE,
     precrawledArchive: AssetTypes.LINK_PRECRAWLED_ARCHIVE,
@@ -46,6 +48,7 @@ export function mapSchemaAssetTypeToDB(
 export function humanFriendlyNameForAssertType(type: ZAssetType) {
   const map: Record<ZAssetType, string> = {
     screenshot: "Screenshot",
+    pdf: "PDF",
     assetScreenshot: "Asset Screenshot",
     fullPageArchive: "Full Page Archive",
     precrawledArchive: "Precrawled Archive",
@@ -63,6 +66,7 @@ export function humanFriendlyNameForAssertType(type: ZAssetType) {
 export function isAllowedToAttachAsset(type: ZAssetType) {
   const map: Record<ZAssetType, boolean> = {
     screenshot: true,
+    pdf: true,
     assetScreenshot: true,
     fullPageArchive: false,
     precrawledArchive: true,
@@ -80,6 +84,7 @@ export function isAllowedToAttachAsset(type: ZAssetType) {
 export function isAllowedToDetachAsset(type: ZAssetType) {
   const map: Record<ZAssetType, boolean> = {
     screenshot: true,
+    pdf: true,
     assetScreenshot: true,
     fullPageArchive: true,
     precrawledArchive: true,
diff --git a/packages/trpc/models/bookmarks.ts b/packages/trpc/models/bookmarks.ts
index 7ecbcfed..a8b30fc5 100644
--- a/packages/trpc/models/bookmarks.ts
+++ b/packages/trpc/models/bookmarks.ts
@@ -161,6 +161,7 @@ export class Bookmark extends BareBookmark {
         screenshotAssetId: assets.find(
           (a) => a.assetType == AssetTypes.LINK_SCREENSHOT,
         )?.id,
+        pdfAssetId: assets.find((a) => a.assetType == AssetTypes.LINK_PDF)?.id,
         fullPageArchiveAssetId: assets.find(
           (a) => a.assetType == AssetTypes.LINK_FULL_PAGE_ARCHIVE,
         )?.id,
@@ -525,6 +526,9 @@ export class Bookmark extends BareBookmark {
             if (row.assets.assetType == AssetTypes.LINK_SCREENSHOT) {
               content.screenshotAssetId = row.assets.id;
             }
+            if (row.assets.assetType == AssetTypes.LINK_PDF) {
+              content.pdfAssetId = row.assets.id;
+            }
             if (row.assets.assetType == AssetTypes.LINK_FULL_PAGE_ARCHIVE) {
               content.fullPageArchiveAssetId = row.assets.id;
             }
diff --git a/packages/trpc/routers/bookmarks.ts b/packages/trpc/routers/bookmarks.ts
index a9d0df38..fb9b4697 100644
--- a/packages/trpc/routers/bookmarks.ts
+++ b/packages/trpc/routers/bookmarks.ts
@@ -562,6 +562,7 @@ export const bookmarksAppRouter = router({
       z.object({
         bookmarkId: z.string(),
         archiveFullPage: z.boolean().optional().default(false),
+        storePdf: z.boolean().optional().default(false),
       }),
     )
     .use(ensureBookmarkOwnership)
@@ -577,6 +578,7 @@ export const bookmarksAppRouter = router({
         {
           bookmarkId: input.bookmarkId,
           archiveFullPage: input.archiveFullPage,
+          storePdf: input.storePdf,
         },
         {
           groupId: ctx.user.id,