refactor: Start tracking bookmark assets in the assets table

2 files changed, 14 insertions, 0 deletions

diff --git a/packages/shared/assetdb.ts b/packages/shared/assetdb.ts
index dd464139..4edfa1ec 100644
--- a/packages/shared/assetdb.ts
+++ b/packages/shared/assetdb.ts
@@ -120,6 +120,18 @@ export async function readAsset({
   return { asset, metadata };
 }
 
+export async function getAssetSize({
+  userId,
+  assetId,
+}: {
+  userId: string;
+  assetId: string;
+}) {
+  const assetDir = getAssetDir(userId, assetId);
+  const stat = await fs.promises.stat(path.join(assetDir, "asset.bin"));
+  return stat.size;
+}
+
 export async function deleteAsset({
   userId,
   assetId,
diff --git a/packages/shared/types/bookmarks.ts b/packages/shared/types/bookmarks.ts
index 86bbbc1a..f4b4fd4a 100644
--- a/packages/shared/types/bookmarks.ts
+++ b/packages/shared/types/bookmarks.ts
@@ -15,6 +15,8 @@ export const zAssetTypesSchema = z.enum([
   "screenshot",
   "bannerImage",
   "fullPageArchive",
+  "bookmarkAsset",
+  "unknown",
 ]);
 export type ZAssetType = z.infer<typeof zAssetTypesSchema>;