update

.env.example

@@ -0,0 +1,4 @@
+HOST="0.0.0.0"
+PORT="3000"
+MONGODB_URL="mongodb://user:pass@localhost/filesharing?authSource=admin"
+UPLOAD_SIZE_LIMIT=10485760

config.js

@@ -1,9 +0,0 @@
-require('dotenv').config()
-module.exports = {
-    SERVER_HOST: process.env.HOST || 'localhost',
-    SERVER_PORT: process.env.PORT || '3000',
-    MONGODB_URL: process.env.MONGDB || 'mongodb://localhost/filesharing',
-    UPLOAD_SIZE_LIMIT: process.env.UPLOAD_SIZE_LIMIT || 102400,
-    UPLOAD_TEMP_PATH: process.env.UPLOAD_TEMP_PATH || '/tmp',
-    UPLOAD_SAVE_PATH: process.env.UPLOAD_SAVE_PATH || 'upload',
-}

package.json

@@ -15,9 +15,11 @@
     "author": "Zhao Xin <7176466@qq.com>",
     "license": "MIT",
     "dependencies": {
-        "bcrypt": "^5.0.1",
+        "bcryptjs": "^2.4.3",
+        "cron": "^2.1.0",
         "dotenv": "^16.0.1",
         "express": "^4.18.1",
+        "filesize": "^9.0.11",
         "md5-file": "^5.0.0",
         "moment": "^2.29.4",
         "mongoose": "^6.5.3",

public/style.css

@@ -6,7 +6,7 @@
 
 body {
     user-select: none;
-    max-width: 360px;
+    max-width: 480px;
     margin: 0 auto;
 }
 
@@ -30,5 +30,19 @@ form button {
 
 .small {
     color: gray;
-    font-size: small;
+    font-size: xx-small;
+}
+
+audio,
+video {
+    width: 480px;
+}
+
+img {
+    width: 480px;
+    object-fit: cover;
+}
+
+::marker {
+    content: '💾 ';
 }

server.js

@@ -1,33 +1,62 @@
-const { SERVER_HOST, SERVER_PORT, MONGODB_URL, UPLOAD_TEMP_PATH, UPLOAD_SAVE_PATH, UPLOAD_SIZE_LIMIT } = require('./config')
+require('dotenv').config()
+const HOST = process.env.HOST || 'localhost'
+const PORT = process.env.PORT || '3000'
+const MONGODB_URL = process.env.MONGODB_URL || 'mongodb+srv://test:It6E1HC5p5K83bS8@test.ekqr7.azure.mongodb.net/test?retryWrites=true&w=majority'
+const UPLOAD_SIZE_LIMIT = process.env.UPLOAD_SIZE_LIMIT || 102400
+const UPLOAD_TEMP_PATH = process.env.UPLOAD_TEMP_PATH || '/tmp'
+const UPLOAD_SAVE_PATH = process.env.UPLOAD_SAVE_PATH || 'upload'
+
 const fs = require('fs')
 const path = require('path')
-const multer = require('multer')
-const bcrypt = require('bcrypt')
 const md5file = require('md5-file')
-const express = require('express')
+const filesize = require('filesize')
+const { hash, compare } = require('bcryptjs')
+
 const mongoose = require('mongoose')
+mongoose.connect(MONGODB_URL, { useNewUrlParser: true }).catch(error => console.error(error))
+mongoose.connection.on('connected', () => console.info('mongodb is connected'))
+mongoose.connection.on('error', (error) => console.error(error))
 const Filesharing = require('./models/Filesharing')
-mongoose.connect(MONGODB_URL, (error) => {
-    if (error) {
-        console.error(error)
-    } else {
-        console.info('mongodb connected successfully')
-    }
-})
+
+const CronJob = require('cron').CronJob
+const job = new CronJob(
+    '0 * * * * *',
+    async () => {
+        const files = await Filesharing.find({ createdAt: { $lt: new Date(Date.now() - 24 * 3600 * 1000) } })
+        for (let file of files) {
+            const id = file.id
+            await Filesharing.findByIdAndDelete(id)
+            console.log(`删除共享 ${id}`)
+            const ref_number = await Filesharing.countDocuments({ md5: file.md5 })
+            if (ref_number == 0) {
+                const file_path = path.join(__dirname, UPLOAD_SAVE_PATH, file.md5)
+                fs.unlinkSync(file_path)
+                console.log(`删除文件 ${file_path}`)
+            }
+        }
+    },
+    null,
+    true
+)
+
+const express = require('express')
 const app = express()
-app.set('view engine', 'pug')
 app.use(express.static('public'))
-app.use(express.urlencoded({ extended: true }))
+app.set('view engine', 'pug')
+app.use(express.urlencoded({ extended: false }))
 app.use(express.json())
 app.locals.moment = require('moment')
 app.locals.moment.locale('zh-cn')
+app.locals.filesize = filesize.partial({ base: 2, standard: 'jedec' })
 app.get('/', async (req, res) => {
-    const filesharing = await Filesharing.find().sort({ createdAt: -1 })
-    res.render('index', { filesharing })
+    const files = await Filesharing.find().sort({ createdAt: -1 })
+    res.render('index', { title: '文件分享', files })
 })
 app.get('/upload', async (req, res) => {
-    res.render('upload')
+    res.render('upload', { title: '文件上传' })
 })
+
+const multer = require('multer')
 const upload = multer({ dest: UPLOAD_TEMP_PATH, limits: { fileSize: UPLOAD_SIZE_LIMIT } })
 app.post('/upload', upload.single('file'), async (req, res) => {
     const file_temp_path = path.join(UPLOAD_TEMP_PATH, req.file.filename)
@@ -38,36 +67,37 @@ app.post('/upload', upload.single('file'), async (req, res) => {
     }
     fs.unlinkSync(file_temp_path) // 删除临时文件
     // 写入数据库
-    const filesharing = new Filesharing({
+    const file = await Filesharing.create({
         md5,
         size: req.file.size,
         encoding: req.file.encoding,
         mimetype: req.file.mimetype,
         // 解决了multer将中文文件名错误编码的问题
         filename: Buffer.from(req.file.originalname, 'latin1').toString('utf8'),
-        password: req.body.password ? await bcrypt.hash(req.body.password, 16) : '',
+        password: req.body.password ? await hash(req.body.password, 16) : '',
     })
-    const file = await filesharing.save()
-    res.status(201).render('upload', { file })
+    console.log(`上传文件 ${file.filename}`)
+    res.status(201).render('upload', { title: '文件上传', file })
 })
-app.get('/file/:id', async (req, res) => {
+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)
         if (!file) throw '试图下载不存在的文件'
-        res.render('download', { file })
+        res.render('download', { title: '文件下载', file })
     } catch (error) {
         console.error(error)
         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)
         if (!file) throw '试图下载不存在的文件'
-        if (file.password == '' || (await bcrypt.compare(req.body.password, file.password)) == true) {
-            await file.update({ $inc: { downloads: 1 } })
+        if (file.password == '' || (await compare(req.body.password, file.password)) == true) {
+            await Filesharing.findByIdAndUpdate(req.params.id, { $inc: { downloads: 1 } })
+            console.log(`下载文件 ${file.filename}`)
             res.status(200).download(path.join(UPLOAD_SAVE_PATH, file.md5), file.filename)
         } else {
             res.sendStatus(401)
@@ -77,6 +107,18 @@ app.post('/file/:id', async (req, res) => {
         res.sendStatus(404)
     }
 })
-const server = app.listen(SERVER_PORT, SERVER_HOST, () => {
-    console.info(`server is running at http://${SERVER_HOST}:${SERVER_PORT}`)
+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.listen(PORT, HOST, () => {
+    console.info(`server is running at http://${HOST}:${PORT}`)
 })

views/download.pug

@@ -1,30 +1,32 @@
-doctype html
-html(lang='zh')
-    head
-        meta(charset='UTF-8')
-        meta(http-equiv='X-UA-Compatible' content='IE=edge')
-        meta(name='viewport' content='width=device-width, initial-scale=1.0')
-        link(rel='icon' href='/favicon.ico' type='image/x-icon')
-        link(rel='stylesheet' href='/style.css')
-        title #{file.filename} | 文件下载
-    body
-        h1 文件下载
-        p
-            a(href='/') 返回首页
-        form(method='POST')
-            label 文件
-            label= file.filename
-            label 类型
-            label= file.mimetype
-            label 大小
-            label= file.size
-                |   字节
-            label 上传
-            label= moment(file.createdAt).fromNow()
-            label 热度
-            label= file.downloads
-                |   次下载
-            if file.password
-                label(for='password_input') 密码
-                input(id='password_input' name='password' type='password' autocomplete='off'  required)
-            button(id='download_button' type='submit') 下载
+extends layout
+block main
+    p
+        a(href='/') 返回首页
+    form(method='POST')
+        label 文件
+        label= file.filename
+        label 类型
+        label= file.mimetype
+        label 大小
+        label= filesize(file.size)
+        label 上传
+        label= moment(file.createdAt).fromNow()
+        label 热度
+        label= file.downloads
+            |   次下载
+        if file.password
+            label(for='password_input') 密码
+            input(id='password_input' name='password' type='password' autocomplete='off'  required)
+        button(id='download_button' type='submit') 下载
+    if !file.password
+        - const mimetype = file.mimetype.split('/')[0]
+        - const src = '/file/' + file.md5
+        if mimetype == 'image'
+            p 预览
+            img(src=src)
+        if mimetype == 'audio'
+            p 预览
+            audio(controls src=src)
+        if mimetype == 'video'
+            p 预览
+            video(controls src=src)

views/index.pug

@@ -1,23 +1,14 @@
-doctype html
-html(lang='zh')
-    head
-        meta(charset='UTF-8')
-        meta(http-equiv='X-UA-Compatible' content='IE=edge')
-        meta(name='viewport' content='width=device-width, initial-scale=1.0')
-        link(rel='icon' href='/favicon.ico' type='image/x-icon')
-        link(rel='stylesheet' href='/style.css')
-        title 文件分享
-    body
-        h1 文件分享
-        p
-            a(href='/upload') 我要上传
-        ul
-            if filesharing.length==0
-                li 暂无文件
-            else
-            for file in filesharing
-                li
-                    a(href=`/file/${file.id}` target='_blank')= file.filename
-                    if file.password
-                        span   ㊙️
-                    span.small   ( #{moment(file.createdAt).fromNow()}, #{file.downloads} 次下载 )
+extends layout
+block main
+    p
+        a(href='upload') 我要上传
+    ul
+        if files.length==0
+            li 暂无文件
+        else
+        for file in files
+            li
+                a(href=`download/${file.id}`)= file.filename
+                if file.password
+                    span  ㊙️
+                span.small  (#{moment(file.createdAt).fromNow()}, #{file.downloads} 次下载)

views/layout.pug

@@ -0,0 +1,13 @@
+doctype html
+html(lang='zh')
+    head
+        meta(charset='UTF-8')
+        meta(http-equiv='X-UA-Compatible' content='IE=edge')
+        meta(name='viewport' content='width=device-width, initial-scale=1.0')
+        link(rel='icon' href='/favicon.ico' type='image/x-icon')
+        link(rel='stylesheet' href='/style.css')
+        title #{title}
+    body
+        .container
+            h1 #{title}
+            block main

views/upload.pug

@@ -1,28 +1,19 @@
-doctype html
-html(lang='zh')
-    head
-        meta(charset='UTF-8')
-        meta(http-equiv='X-UA-Compatible' content='IE=edge')
-        meta(name='viewport' content='width=device-width, initial-scale=1.0')
-        link(rel='icon' href='/favicon.ico' type='image/x-icon')
-        link(rel='stylesheet' href='/style.css')
-        title 文件上传
-    body
-        h1 文件上传
-        p
-            a(href='/') 返回首页
-        if file
-            section
-                p
-                    | 文件  
-                    var #{file.filename}  
-                    | 分享成功！
-                p 链接  
-                    - const url = '/file/' + file.id
-                    a(href=url)= url
-        form(method='POST' action='/upload' enctype='multipart/form-data')
-            label(for='file_input') 文件
-            input(id='file_input' name='file' type='file' required)
-            label(for='password_input') 密码
-            input(id='password_input' name='password' type='password' autocomplete='off')
-            button(id='share_button' type='submit') 分享
+extends layout
+block main
+    p
+        a(href='/') 返回首页
+    if file
+        section
+            p
+                | 文件  
+                var #{file.filename}  
+                | 分享成功！
+            p 链接  
+                - const url = 'download/' + file.id
+                a(href=url)= url
+    form(method='POST' action='upload' enctype='multipart/form-data')
+        label(for='file_input') 文件
+        input(id='file_input' name='file' type='file' required)
+        label(for='password_input') 密码
+        input(id='password_input' name='password' type='password' autocomplete='off')
+        button(id='share_button' type='submit') 分享