From 659aa9d1113c5a2c1de58f7d0d6245aee9430b26 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B5=B5=E9=91=AB?= <7176466@qq.com>
Date: Sat, 27 Aug 2022 00:46:04 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E6=96=87=E4=BB=B6=E9=A2=84?=
 =?UTF-8?q?=E8=A7=88=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 public/style.css   |  9 +++++++++
 server.js          | 15 +++++++++++++--
 views/download.pug |  9 +++++++++
 views/index.pug    |  2 +-
 views/upload.pug   |  2 +-
 5 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/public/style.css b/public/style.css
index 5405986..b7e900a 100644
--- a/public/style.css
+++ b/public/style.css
@@ -32,3 +32,12 @@ form button {
     color: gray;
     font-size: small;
 }
+
+audio {
+    width: 360px;
+}
+
+img {
+    width: 360px;
+    object-fit: cover;
+}
diff --git a/server.js b/server.js
index 5c18213..7b635f0 100644
--- a/server.js
+++ b/server.js
@@ -49,7 +49,18 @@ app.post('/upload', upload.single('file'), async (req, res) => {
     })
     res.status(201).render('upload', { file })
 })
-app.get('/file/:id', async (req, res) => {
+app.get('/file/:md5', async (req, res) => {
+    try {
+        const file = await Filesharing.findOne({ md5: req.params.md5 })
+        if (!file) throw '试图下载不存在的文件'
+        if (file.password != '') throw '访问的文件需要密码'
+        res.sendFile(path.join(__dirname, UPLOAD_SAVE_PATH, req.params.md5))
+    } catch (error) {
+        console.error(error)
+        res.sendStatus(404)
+    }
+})
+app.get('/download/:id', async (req, res) => {
     try {
         if (!req.params.id.match(/^[0-9a-fA-F]{24}$/)) throw '格式错误的ObjectId'
         const file = await Filesharing.findById(req.params.id)
@@ -60,7 +71,7 @@ app.get('/file/:id', async (req, res) => {
         res.sendStatus(404)
     }
 })
-app.post('/file/:id', async (req, res) => {
+app.post('/download/:id', async (req, res) => {
     try {
         if (!req.params.id.match(/^[0-9a-fA-F]{24}$/)) throw '格式错误的ObjectId'
         const file = await Filesharing.findById(req.params.id)
diff --git a/views/download.pug b/views/download.pug
index d05249f..281d675 100644
--- a/views/download.pug
+++ b/views/download.pug
@@ -28,3 +28,12 @@ html(lang='zh')
                 label(for='password_input') 密码
                 input(id='password_input' name='password' type='password' autocomplete='off'  required)
             button(id='download_button' type='submit') 下载
+        if !file.password
+            - var mimetype = file.mimetype.split('/')[0]
+            - var src = '/file/' + file.md5
+            if mimetype == 'image'
+                p 预览
+                img(src=src)
+            if mimetype == 'audio'
+                p 预览
+                audio(controls src=src)
diff --git a/views/index.pug b/views/index.pug
index 8973d6a..8d4e39c 100644
--- a/views/index.pug
+++ b/views/index.pug
@@ -17,7 +17,7 @@ html(lang='zh')
             else
             for file in filesharing
                 li
-                    a(href=`/file/${file.id}`)= file.filename
+                    a(href=`/download/${file.id}`)= file.filename
                     if file.password
                         span &nbsp; ㊙️
                     span.small &nbsp; ( #{moment(file.createdAt).fromNow()}, #{file.downloads} 次下载 )
diff --git a/views/upload.pug b/views/upload.pug
index 635e05b..c6e42ef 100644
--- a/views/upload.pug
+++ b/views/upload.pug
@@ -18,7 +18,7 @@ html(lang='zh')
                     var #{file.filename} &nbsp;
                     | 分享成功！
                 p 链接 &nbsp;
-                    - const url = '/file/' + file.id
+                    - const url = '/download/' + file.id
                     a(href=url)= url
         form(method='POST' action='/upload' enctype='multipart/form-data')
             label(for='file_input') 文件