ai-shit

frontend/src/asset_manager.ts

@@ -1,5 +1,3 @@
-import type { AxiosInstance } from 'axios';
-
 export type Identifier = string | number;
 
 type Nullable<T> = T | null;
@@ -205,20 +203,20 @@ export const resolveDocumentAssetUrl = (
 };
 
 class AssetManager {
-  api: AxiosInstance | null;
+  fetchAsset: ((id: Identifier) => Promise<AssetLike | null>) | null;
   assetPresignTtlMs: number;
   assetCache: Map<Identifier, AssetLike>;
   assetInflight: Map<string, Promise<AssetLike | null>>;
 
-  constructor({ api, assetPresignTtlMs }: { api: AxiosInstance | null; assetPresignTtlMs: number }) {
-    this.api = api;
+  constructor({ fetchAsset, assetPresignTtlMs }: { fetchAsset: ((id: Identifier) => Promise<AssetLike | null>) | null; assetPresignTtlMs: number }) {
+    this.fetchAsset = fetchAsset;
     this.assetPresignTtlMs = assetPresignTtlMs;
     this.assetCache = new Map();
     this.assetInflight = new Map();
   }
 
-  setApi(api: AxiosInstance | null) {
-    this.api = api;
+  setFetchAsset(fetchAsset: ((id: Identifier) => Promise<AssetLike | null>) | null) {
+    this.fetchAsset = fetchAsset;
   }
 
   rememberAsset(entry?: Nullable<AssetLike>) {
@@ -271,18 +269,16 @@ class AssetManager {
       return this.assetInflight.get(inflightKey);
     }
 
-    if (!this.api) {
-      return Promise.reject(new Error('AssetManager API client is not configured.'));
+    if (!this.fetchAsset) {
+      return Promise.reject(new Error('AssetManager fetcher is not configured.'));
     }
 
-    const request: Promise<AssetLike | null> = this.api
-      .get(`/assets/${asset.id}`)
-      .then(({ data }) => {
+    const request: Promise<AssetLike | null> = this.fetchAsset(asset.id)
+      .then((data) => {
+        if (!data) return null;
         const cachedEntry = this.assetCache.get(asset.id) || baseAsset;
         const combined = { ...cachedEntry, ...asset, ...data };
-        const expires_at =
-          resolveAssetExpiresAt(data)
-          ?? resolveAssetExpiresAt(combined);
+        const expires_at = resolveAssetExpiresAt(combined);
         const entry = {
           ...combined,
           url: resolveAssetUrl(combined),