@windrun-huaiin/third-ui 5.12.0 → 5.12.1

package/dist/main/index.js

@@ -2771,59 +2771,48 @@ function Gallery({ sectionClassName, button }) {
   const galleryItems = t.raw("prompts");
   const defaultImgUrl = t.raw("defaultImgUrl");
   const [imageErrors, setImageErrors] = (0, import_react35.useState)(/* @__PURE__ */ new Set());
+  const [downloadingItems, setDownloadingItems] = (0, import_react35.useState)(/* @__PURE__ */ new Set());
+  const cdnProxyUrl = process.env.NEXT_PUBLIC_STYLE_CDN_PROXY_URL;
   const handleDownload = (item, index) => __async(null, null, function* () {
     var _a;
+    if (downloadingItems.has(index)) {
+      return;
+    }
+    setDownloadingItems((prev) => new Set(prev).add(index));
     try {
-      const response = yield fetch(item.url, {
-        method: "GET",
-        // CORS mode declaration
-        mode: "cors"
-      });
-      if (!response.ok) {
-        throw new Error(`HTTP error! status: ${response.status}`);
-      }
-      const blob = yield response.blob();
-      const url = window.URL.createObjectURL(blob);
-      const contentType = response.headers.get("content-type");
+      const originalUrl = new URL(item.url);
+      const filename = originalUrl.pathname.substring(1);
+      const proxyUrl = `${cdnProxyUrl}/${encodeURIComponent(filename)}`;
+      const urlExtension = (_a = item.url.split(".").pop()) == null ? void 0 : _a.toLowerCase();
       let extension = ".webp";
-      if (contentType) {
-        switch (contentType) {
-          case "image/jpeg":
-          case "image/jpg":
-            extension = ".jpg";
-            break;
-          case "image/png":
-            extension = ".png";
-            break;
-          case "image/gif":
-            extension = ".gif";
-            break;
-          case "image/webp":
-            extension = ".webp";
-            break;
-          case "image/svg+xml":
-            extension = ".svg";
-            break;
-          default:
-            const urlExtension = (_a = item.url.split(".").pop()) == null ? void 0 : _a.toLowerCase();
-            if (urlExtension && ["jpg", "jpeg", "png", "gif", "webp", "svg"].includes(urlExtension)) {
-              extension = `.${urlExtension}`;
-            }
-        }
+      if (urlExtension && ["jpg", "jpeg", "png", "gif", "webp", "svg"].includes(urlExtension)) {
+        extension = `.${urlExtension}`;
       }
       const downloadPrefix = t("downloadPrefix");
+      const response = yield fetch(proxyUrl);
+      if (!response.ok) {
+        throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+      }
+      const blob = yield response.blob();
+      const blobUrl = URL.createObjectURL(blob);
       const a = document.createElement("a");
-      a.href = url;
+      a.href = blobUrl;
       a.download = `${downloadPrefix}-${index + 1}${extension}`;
       a.style.display = "none";
       document.body.appendChild(a);
       a.click();
       setTimeout(() => {
-        window.URL.revokeObjectURL(url);
         document.body.removeChild(a);
+        URL.revokeObjectURL(blobUrl);
       }, 100);
     } catch (error) {
       console.error("Download failed:", error);
+    } finally {
+      setDownloadingItems((prev) => {
+        const newSet = new Set(prev);
+        newSet.delete(index);
+        return newSet;
+      });
     }
   });
   const handleImageError = (index) => {
@@ -2857,8 +2846,12 @@ function Gallery({ sectionClassName, button }) {
         "button",
         {
           onClick: () => handleDownload(item, index),
-          className: "bg-black/50 hover:bg-black/70 p-2 rounded-full text-white/80 hover:text-white transition-all duration-300",
-          children: /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(globalLucideIcons.Download, { className: "h-5 w-5 text-white" })
+          disabled: downloadingItems.has(index),
+          className: cn(
+            "p-2 rounded-full transition-all duration-300",
+            downloadingItems.has(index) ? "bg-black/30 text-white/50" : "bg-black/50 hover:bg-black/70 text-white/80 hover:text-white"
+          ),
+          children: downloadingItems.has(index) ? /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(globalLucideIcons.Loader2, { className: "h-5 w-5 text-white animate-spin" }) : /* @__PURE__ */ (0, import_jsx_runtime33.jsx)(globalLucideIcons.Download, { className: "h-5 w-5 text-white" })
         }
       ) })
     ] }, index)) }),