From 6cba58999eafbad6e70d6203ee2c454577990698 Mon Sep 17 00:00:00 2001
From: Tristan Yang <yanggc888@163.com>
Date: Sat, 8 Jul 2023 08:28:47 +0800
Subject: [PATCH] feat: add image download in markdown

---
 src/components/ImagePreview.tsx | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/components/ImagePreview.tsx b/src/components/ImagePreview.tsx
index c60afd61..a693d2a3 100644
--- a/src/components/ImagePreview.tsx
+++ b/src/components/ImagePreview.tsx
@@ -35,7 +35,13 @@ const ImagePreview = ({ container, context = "chat" }: Props) => {
         const originUrl = `${urlObj.origin}${urlObj.pathname}?file_path=${urlObj.searchParams.get(
           "file_path"
         )}`;
-        const data = { originUrl };
+        const thumbnail = `${urlObj.origin}${urlObj.pathname}?file_path=${urlObj.searchParams.get(
+          "file_path"
+        )}&thumbnail=true`;
+        const downloadLink = `${urlObj.origin}${
+          urlObj.pathname
+        }?file_path=${urlObj.searchParams.get("file_path")}&download=true`;
+        const data = { originUrl, downloadLink, thumbnail };
         setPreviewImage(data);
       }
     };
@@ -48,11 +54,7 @@ const ImagePreview = ({ container, context = "chat" }: Props) => {
     };
   }, [container, context]);
   return previewImage ? (
-    <ImagePreviewModal
-      download={context == "chat"}
-      data={previewImage}
-      closeModal={closePreviewModal}
-    />
+    <ImagePreviewModal download={true} data={previewImage} closeModal={closePreviewModal} />
   ) : null;
 };