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

README.md

@@ -1 +1,3 @@
-# 文件分享服务器
+# 文件分享
+
+一个文件分享服务应用

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/data',
-    TEMP_PATH: '/tmp',
-    SAVE_PATH: 'upload',
-    UPLOAD_SIZE_LIMIT: 1024,
-}

models/Filesharing.js

@@ -1,5 +1,5 @@
 const mongoose = require('mongoose')
-const FileshareSchema = new mongoose.Schema({
+const Filesharing = new mongoose.Schema({
     md5: { type: String, require: true },
     size: { type: Number, require: true },
     encoding: { type: String, require: true },
@@ -9,4 +9,4 @@ const FileshareSchema = new mongoose.Schema({
     downloads: { type: Number, default: 0 },
     createdAt: { type: Date, default: Date.now },
 })
-module.exports = mongoose.model('Fileshare', FileshareSchema)
+module.exports = mongoose.model('Filesharing', Filesharing)

package.json

@@ -1,7 +1,7 @@
 {
-    "name": "fileshare",
+    "name": "filesharing",
     "version": "0.1.0",
-    "description": "文件分享服务器",
+    "description": "文件分享",
     "main": "server.js",
     "scripts": {
         "start": "nodemon",
@@ -9,16 +9,19 @@
     },
     "repository": {
         "type": "git",
-        "url": "ssh://git@nas:2222/zhaoxin/fileshare.git"
+        "url": "ssh://git@nas:2222/zhaoxin/filesharing.git"
     },
     "keywords": [],
     "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",
         "multer": "^1.4.5-lts.1",
         "pug": "^3.0.2"

public/client.js

@@ -1 +0,0 @@
-// file_input.value = null

public/style.css

@@ -6,12 +6,13 @@
 
 body {
     user-select: none;
-    max-width: 360px;
+    max-width: 480px;
     margin: 0 auto;
 }
 
 p,
 h1,
+li,
 form {
     margin: 1rem auto;
 }
@@ -26,3 +27,22 @@ form button {
     grid-column: span 2;
     cursor: pointer;
 }
+
+.small {
+    color: gray;
+    font-size: xx-small;
+}
+
+audio,
+video {
+    width: 480px;
+}
+
+img {
+    width: 480px;
+    object-fit: cover;
+}
+
+::marker {
+    content: '💾 ';
+}

server.js

@@ -1,70 +1,124 @@
-const { SERVER_HOST, SERVER_PORT, MONGODB_URL, TEMP_PATH, 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')
-const Fileshare = require('./models/fileshare')
-mongoose.connect(MONGODB_URL, (error) => {
-    if (error) {
-        console.error(error)
-    } else {
-        console.info('mongodb connected successfully')
-    }
-})
+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')
+
+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.get('/', (req, res) => {
-    res.render('upload')
+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 files = await Filesharing.find().sort({ createdAt: -1 })
+    res.render('index', { title: '文件分享', files })
 })
-const upload = multer({ dest: TEMP_PATH, limits: { fileSize: UPLOAD_SIZE_LIMIT } })
+app.get('/upload', async (req, res) => {
+    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(TEMP_PATH, req.file.filename)
+    const file_temp_path = path.join(UPLOAD_TEMP_PATH, req.file.filename)
     const md5 = await md5file(file_temp_path)
-    const file_save_path = path.join(__dirname, SAVE_PATH, md5)
+    const file_save_path = path.join(__dirname, UPLOAD_SAVE_PATH, md5)
     if (!fs.existsSync(file_save_path)) {
-        fs.cpSync(file_temp_path, file_save_path) // 复制文件到UPLOAD_PATH
+        fs.cpSync(file_temp_path, file_save_path) // 复制临时文件到UPLOAD_PATH
     }
     fs.unlinkSync(file_temp_path) // 删除临时文件
-    let file = {
+    // 写入数据库
+    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) : '',
-    }
-    // 写入数据库
-    console.log(req.file)
-    console.log(file)
-    const fileshare = new Fileshare(file)
-    file = await fileshare.save()
-    console.log(file)
-    res.status(201).render('upload', { file })
+        password: req.body.password ? await hash(req.body.password, 16) : '',
+    })
+    console.log(`上传文件 ${file.filename}`)
+    res.status(201).render('upload', { title: '文件上传', file })
 })
-app.get('/file/:id', async (req, res) => {
-    const file = await Fileshare.findById(req.params.id)
-    res.render('download', { file })
-})
-app.post('/file/:id', async (req, res) => {
-    const file = await Fileshare.findById(req.params.id)
-    if (file) {
-        if (file.password == '' || (await bcrypt.compare(req.body.password, file.password)) == true) {
-            await file.update({ $inc: { downloads: 1 } })
-            res.status(200).download(path.join(SAVE_PATH, file.md5), file.filename)
-        } else {
-            res.sendStatus(401)
-        }
-    } else {
+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', { title: '文件下载', file })
+    } catch (error) {
+        console.error(error)
         res.sendStatus(404)
     }
 })
-const server = app.listen(SERVER_PORT, SERVER_HOST, () => {
-    console.info(`server is running at http://${SERVER_HOST}:${SERVER_PORT}`)
+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 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)
+        }
+    } catch (error) {
+        console.error(error)
+        res.sendStatus(404)
+    }
+})
+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,27 +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 下载文件
-    body
-        h1 下载文件
-        form(method='POST')
-            label 文件
-            label= file.filename
-            label 大小
-            label= file.size
-                |   字节
-            label 类型
-            label= file.mimetype
-            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') 下载
-    script(src='/client.js')
+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

@@ -0,0 +1,14 @@
+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,27 +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 分享文件
-        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') 分享
-        script(src='/client.js')
+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') 分享