feature: Download images and screenshots

2 files changed, 21 insertions, 1 deletions

diff --git a/packages/shared-react/utils/assetUtils.ts b/packages/shared-react/utils/assetUtils.ts
new file mode 100644
index 00000000..119451d9
--- /dev/null
+++ b/packages/shared-react/utils/assetUtils.ts
@@ -0,0 +1,3 @@
+export function getAssetUrl(assetId: string) {
+  return `/api/assets/${assetId}`;
+}
diff --git a/apps/web/lib/bookmarkUtils.tsx b/packages/shared-react/utils/bookmarkUtils.ts
index 475ba383..da199a40 100644
--- a/apps/web/lib/bookmarkUtils.tsx
+++ b/packages/shared-react/utils/bookmarkUtils.ts
@@ -1,7 +1,24 @@
-import type { ZBookmark } from "@hoarder/shared/types/bookmarks";
+import type {
+  ZBookmark,
+  ZBookmarkedLink,
+} from "@hoarder/shared/types/bookmarks";
+
+import { getAssetUrl } from "./assetUtils";
 
 const MAX_LOADING_MSEC = 30 * 1000;
 
+export function getBookmarkLinkImageUrl(bookmark: ZBookmarkedLink) {
+  if (bookmark.imageAssetId) {
+    return { url: getAssetUrl(bookmark.imageAssetId), localAsset: true };
+  }
+  if (bookmark.screenshotAssetId) {
+    return { url: getAssetUrl(bookmark.screenshotAssetId), localAsset: true };
+  }
+  return bookmark.imageUrl
+    ? { url: bookmark.imageUrl, localAsset: false }
+    : null;
+}
+
 export function isBookmarkStillCrawling(bookmark: ZBookmark) {
   return (
     bookmark.content.type == "link" &&