1 files changed, 25 insertions, 0 deletions
diff --git a/packages/trpc/models/assets.ts b/packages/trpc/models/assets.ts
index ad114341..f97cfffb 100644
--- a/packages/trpc/models/assets.ts
+++ b/packages/trpc/models/assets.ts
@@ -4,7 +4,11 @@ import { z } from "zod";
 
 import { assets } from "@karakeep/db/schema";
 import { deleteAsset } from "@karakeep/shared/assetdb";
+import serverConfig from "@karakeep/shared/config";
+import { createSignedToken } from "@karakeep/shared/signedTokens";
+import { zAssetSignedTokenSchema } from "@karakeep/shared/types/assets";
 import { zAssetTypesSchema } from "@karakeep/shared/types/bookmarks";
+import { getAssetUrl } from "@karakeep/shared/utils/assetUtils";
 
 import { AuthedContext } from "..";
 import {
@@ -254,4 +258,25 @@ export class Asset {
       });
     }
   }
+
+  getUrl() {
+    return getAssetUrl(this.asset.id);
+  }
+
+  static getPublicSignedAssetUrl(
+    assetId: string,
+    assetOwnerId: string,
+    expireAt: number,
+  ) {
+    const payload: z.infer<typeof zAssetSignedTokenSchema> = {
+      assetId,
+      userId: assetOwnerId,
+    };
+    const signedToken = createSignedToken(
+      payload,
+      serverConfig.signingSecret(),
+      expireAt,
+    );
+    return `${serverConfig.publicApiUrl}/public/assets/${assetId}?token=${signedToken}`;
+  }
 }