From f05e3658c4c70bd7f92101f281c49b61ed745271 Mon Sep 17 00:00:00 2001
From: Zhang Minghan <zmh13054618081@dingtalk.com>
Date: Sat, 23 Sep 2023 18:58:12 +0800
Subject: [PATCH] fix code suffix

---
 app/src/routes/Generation.tsx | 4 ++--
 generation/api.go             | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/app/src/routes/Generation.tsx b/app/src/routes/Generation.tsx
index 70b1c95..f55c6b1 100644
--- a/app/src/routes/Generation.tsx
+++ b/app/src/routes/Generation.tsx
@@ -90,11 +90,11 @@ function Wrapper({ onSend }: WrapperProps) {
             {
               hash.length > 0 &&
                 <div className={`hash-box`}>
-                  <a className={`download-box`} target={`_blank`} href={`${rest_api}/generation/download/tar?hash=${hash}`}>
+                  <a className={`download-box`} href={`${rest_api}/generation/download/tar?hash=${hash}`}>
                     <FileDown className={`h-6 w-6`} />
                     <p>{ t('generate.download', { name: "tar.gz"}) }</p>
                   </a>
-                  <a className={`download-box`} target={`_blank`} href={`${rest_api}/generation/download/zip?hash=${hash}`}>
+                  <a className={`download-box`} href={`${rest_api}/generation/download/zip?hash=${hash}`}>
                     <FileDown className={`h-6 w-6`} />
                     <p>{ t('generate.download', { name: "zip"}) }</p>
                   </a>
diff --git a/generation/api.go b/generation/api.go
index ccdfbe7..c7cf81c 100644
--- a/generation/api.go
+++ b/generation/api.go
@@ -23,11 +23,13 @@ type WebsocketGenerationForm struct {
 
 func ProjectTarDownloadAPI(c *gin.Context) {
 	hash := strings.TrimSpace(c.Query("hash"))
+	c.Writer.Header().Add("Content-Disposition", "attachment; filename=code.tar.gz")
 	c.File(fmt.Sprintf("generation/data/out/%s.tar.gz", hash))
 }
 
 func ProjectZipDownloadAPI(c *gin.Context) {
 	hash := strings.TrimSpace(c.Query("hash"))
+	c.Writer.Header().Add("Content-Disposition", "attachment; filename=code.zip")
 	c.File(fmt.Sprintf("generation/data/out/%s.zip", hash))
 }