update theme to support download

Author: davidwtf

theme/layout/index.tsx

@@ -54,6 +54,81 @@ export default () => {
     window.parent.postMessage(window.location.href, "*");
   }, []);
 
+  // Handle file download links
+  useEffect(() => {
+    const DOWNLOAD_EXTENSIONS = ['.ipynb', '.zip', '.tgz', '.tar.gz', '.sh', '.py', '.sql'];
+
+    const getPathname = (href: string): string => {
+      try {
+        return new URL(href, window.location.origin).pathname;
+      } catch {
+        return href.split('?')[0].split('#')[0];
+      }
+    };
+
+    const shouldDownload = (pathname: string): boolean => {
+      const lowerPath = pathname.toLowerCase();
+      return DOWNLOAD_EXTENSIONS.some(ext => lowerPath.endsWith(ext));
+    };
+
+    const downloadFile = (url: string, filename: string) => {
+      const isSameOrigin = url.startsWith('/') || url.startsWith(window.location.origin);
+
+      if (isSameOrigin) {
+        fetch(url)
+          .then(res => res.ok ? res.blob() : Promise.reject(new Error('Failed to fetch')))
+          .then(blob => {
+            const blobUrl = window.URL.createObjectURL(blob);
+            const a = document.createElement('a');
+            a.href = blobUrl;
+            a.download = filename;
+            a.style.display = 'none';
+            document.body.appendChild(a);
+            a.click();
+            setTimeout(() => {
+              window.URL.revokeObjectURL(blobUrl);
+              document.body.removeChild(a);
+            }, 100);
+          })
+          .catch(() => console.warn('Failed to download file:', url));
+      } else {
+        const a = document.createElement('a');
+        a.href = url;
+        a.download = filename;
+        a.style.display = 'none';
+        document.body.appendChild(a);
+        a.click();
+        setTimeout(() => document.body.removeChild(a), 100);
+      }
+    };
+
+    const handleDownloadLinks = () => {
+      document.querySelectorAll('a[href]:not([data-download-handled])').forEach((link) => {
+        const href = link.getAttribute('href');
+        if (!href) return;
+
+        const pathname = getPathname(href);
+        if (!shouldDownload(pathname)) return;
+
+        link.setAttribute('data-download-handled', 'true');
+        link.addEventListener('click', (e: Event) => {
+          e.preventDefault();
+          e.stopPropagation();
+          e.stopImmediatePropagation();
+          downloadFile(href, pathname.split('/').pop() || 'download');
+        }, { capture: true });
+      });
+    };
+
+    handleDownloadLinks();
+    const observer = new MutationObserver(() => setTimeout(handleDownloadLinks, 100));
+    observer.observe(document.querySelector('.rspress-doc-content') || document.body, {
+      childList: true,
+      subtree: true
+    });
+    return () => observer.disconnect();
+  }, []);
+
   return (
     <Layout
       uiSwitch={uiSwitch}