From 013ca67c151b51575151424084f6358522b83579 Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Wed, 24 Dec 2025 13:58:37 +0200
Subject: refactor: move assets to their own model (#2301)

* refactor: move assets to their own model

* move asset privacy checks to the model
---
 packages/trpc/routers/bookmarks.ts | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

(limited to 'packages/trpc/routers/bookmarks.ts')

diff --git a/packages/trpc/routers/bookmarks.ts b/packages/trpc/routers/bookmarks.ts
index 65d401e2..a9d0df38 100644
--- a/packages/trpc/routers/bookmarks.ts
+++ b/packages/trpc/routers/bookmarks.ts
@@ -48,9 +48,9 @@ import { normalizeTagName } from "@karakeep/shared/utils/tag";
 import type { AuthedContext } from "../index";
 import { authedProcedure, createRateLimitMiddleware, router } from "../index";
 import { getBookmarkIdsFromMatcher } from "../lib/search";
+import { Asset } from "../models/assets";
 import { BareBookmark, Bookmark } from "../models/bookmarks";
 import { ImportSession } from "../models/importSessions";
-import { ensureAssetOwnership } from "./assets";
 
 export const ensureBookmarkOwnership = experimental_trpcMiddleware<{
   ctx: AuthedContext;
@@ -178,10 +178,7 @@ export const bookmarksAppRouter = router({
                   .returning()
               )[0];
               if (input.precrawledArchiveId) {
-                await ensureAssetOwnership({
-                  ctx,
-                  assetId: input.precrawledArchiveId,
-                });
+                await Asset.ensureOwnership(ctx, input.precrawledArchiveId);
                 await tx
                   .update(assets)
                   .set({
@@ -232,13 +229,13 @@ export const bookmarksAppRouter = router({
                   sourceUrl: null,
                 })
                 .returning();
-              const uploadedAsset = await ensureAssetOwnership({
-                ctx,
-                assetId: input.assetId,
-              });
+              const uploadedAsset = await Asset.fromId(ctx, input.assetId);
+              uploadedAsset.ensureOwnership();
               if (
-                !uploadedAsset.contentType ||
-                !SUPPORTED_BOOKMARK_ASSET_TYPES.has(uploadedAsset.contentType)
+                !uploadedAsset.asset.contentType ||
+                !SUPPORTED_BOOKMARK_ASSET_TYPES.has(
+                  uploadedAsset.asset.contentType,
+                )
               ) {
                 throw new TRPCError({
                   code: "BAD_REQUEST",
-- 
cgit v1.3-1-g0d28