typescript

frontend/src/desktop/hooks/usePreviewMetadata.ts

@@ -1,8 +1,32 @@
 import { useEffect, useState } from 'react';
 import { createAssetView } from '../../asset_manager';
 
-const usePreviewMetadata = (documents, getDocumentAsset, ensureAssetUrl) => {
-  const [metadataMap, setMetadataMap] = useState(() => new Map());
+interface DocumentLike {
+  id?: string | number;
+  current_version?: unknown;
+  tags?: unknown;
+}
+
+interface AssetLike {
+  id?: string | number;
+  [key: string]: unknown;
+}
+
+interface PreviewMetadataEntry {
+  docId: string;
+  width: number;
+  height: number;
+}
+
+type GetDocumentAsset = (doc: DocumentLike, type: string) => AssetLike | null | undefined;
+type EnsureAssetUrl = (docId: string | number, asset: AssetLike, options?: { start?: number; limit?: number }) => Promise<AssetLike | null | undefined>;
+
+const usePreviewMetadata = (
+  documents: DocumentLike[] | null | undefined,
+  getDocumentAsset?: GetDocumentAsset,
+  ensureAssetUrl?: EnsureAssetUrl,
+) => {
+  const [metadataMap, setMetadataMap] = useState<Map<string, PreviewMetadataEntry>>(() => new Map());
 
   useEffect(() => {
     let cancelled = false;
@@ -14,19 +38,19 @@ const usePreviewMetadata = (documents, getDocumentAsset, ensureAssetUrl) => {
       };
     }
 
-    const fetchMetadataForDoc = async (doc) => {
+    const fetchMetadataForDoc = async (doc: DocumentLike) => {
       if (!doc?.id) {
         return null;
       }
 
       const docId = String(doc.id);
-      const resolveAsset = (type) => getDocumentAsset?.(doc, type) ?? null;
+      const resolveAsset = (type: string) => getDocumentAsset?.(doc, type) ?? null;
 
       let asset = resolveAsset('preview') || resolveAsset('thumbnail');
       let view = createAssetView(asset);
       let metadata = view.getPrimaryMetadata();
 
-      const hasDimensions = (meta) =>
+      const hasDimensions = (meta: { width?: number | string; height?: number | string } | null | undefined) =>
         Number.isFinite(Number(meta?.width)) &&
         Number.isFinite(Number(meta?.height)) &&
         Number(meta.width) > 0 &&