1

2026-05-25 17:08:18 +08:00
parent df501f6151
commit f1259b71b5
9178 changed files with 1626125 additions and 0 deletions
@@ -0,0 +1,65 @@
+import * as React from 'react';
+import { COMMON_PROPS } from "../common";
+/**
+ * Merge props provided `items` or context collected images
+ */
+export default function usePreviewItems(items) {
+  // Context collection image data
+  const [images, setImages] = React.useState({});
+  const registerImage = React.useCallback((id, data) => {
+    setImages(imgs => ({
+      ...imgs,
+      [id]: data
+    }));
+    return () => {
+      setImages(imgs => {
+        const cloneImgs = {
+          ...imgs
+        };
+        delete cloneImgs[id];
+        return cloneImgs;
+      });
+    };
+  }, []);
+
+  // items
+  const mergedItems = React.useMemo(() => {
+    // use `items` first
+    if (items) {
+      return items.map(item => {
+        if (typeof item === 'string') {
+          return {
+            data: {
+              src: item
+            }
+          };
+        }
+        const data = {};
+        Object.keys(item).forEach(key => {
+          if (['src', ...COMMON_PROPS].includes(key)) {
+            data[key] = item[key];
+          }
+        });
+        return {
+          data
+        };
+      });
+    }
+
+    // use registered images secondly
+    return Object.keys(images).reduce((total, id) => {
+      const {
+        canPreview,
+        data
+      } = images[id];
+      if (canPreview) {
+        total.push({
+          data,
+          id
+        });
+      }
+      return total;
+    }, []);
+  }, [items, images]);
+  return [mergedItems, registerImage, !!items];
+}