@arbidocs/blocks 0.3.112 → 0.3.114

package/dist/index.cjs

@@ -1451,14 +1451,35 @@ function useAssetResolverActive() {
 function isAssetRef(src) {
   return !!src && src.startsWith(ASSET_REF_PREFIX);
 }
+function useAssetImage(docId) {
+  const arbi = react.useArbi();
+  const { data: thumbs } = react.useThumbnails([docId]);
+  const thumb = thumbs?.[docId];
+  const { data: objectUrl } = reactQuery.useQuery({
+    queryKey: ["arbi", "asset-image", docId],
+    queryFn: async () => {
+      const res = await arbi.documents.download(docId);
+      const blob = await res.blob();
+      return URL.createObjectURL(blob);
+    },
+    // Only download when the thumbnail is genuinely absent (not merely loading).
+    enabled: !!docId && thumbs != null && !thumb && arbi.hasWorkspace,
+    staleTime: 10 * 6e4
+  });
+  react$1.useEffect(() => {
+    return () => {
+      if (objectUrl) URL.revokeObjectURL(objectUrl);
+    };
+  }, [objectUrl]);
+  return thumb ?? objectUrl;
+}
 function AssetThumb({
   docId,
   alt,
   className,
   testId
 }) {
-  const { data } = react.useThumbnails([docId]);
-  const src = data?.[docId];
+  const src = useAssetImage(docId);
   return /* @__PURE__ */ jsxRuntime.jsx("img", { src: src ?? void 0, alt: alt ?? "", className, "data-testid": testId });
 }
 var ASPECT = {
@@ -2865,7 +2886,9 @@ function ImageFieldControl({
   const [busy, setBusy] = react$1.useState(false);
   const arbi = react.useArbi();
   const gen = useImageGen();
+  const queryClient = reactQuery.useQueryClient();
   const tp = (...q) => tid(`${testIdPrefix}-image`, name, ...q);
+  const refreshAssets = () => queryClient.invalidateQueries({ queryKey: ["arbi", "documents", workspaceId] });
   const { data: docs } = react.useDocuments(workspaceId);
   const imageDocs = react$1.useMemo(
     () => (docs ?? []).filter(
@@ -2886,7 +2909,10 @@ function ImageFieldControl({
       await arbi.selectWorkspace(workspaceId);
       const res = await arbi.documents.uploadFile(file, file.name, { folder });
       const id = res.doc_ext_ids?.[0];
-      if (id) onChange(asAssetRef(id));
+      if (id) {
+        onChange(asAssetRef(id));
+        void refreshAssets();
+      }
     } finally {
       setBusy(false);
     }
@@ -2894,7 +2920,10 @@ function ImageFieldControl({
   const onGenerate = async () => {
     if (!prompt.trim() || gen.isGenerating) return;
     const id = await gen.generate(prompt);
-    if (id) onChange(asAssetRef(id));
+    if (id) {
+      onChange(asAssetRef(id));
+      void refreshAssets();
+    }
   };
   const MODES = [
     { key: "asset", label: "Assets", icon: lucideReact.ImagePlus },