zhaoxin/filesharing
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
赵鑫 2022-09-07 00:01:56 +08:00
parent b1b2b29689
commit 1a23273708
9 changed files with 138 additions and 1013 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.env.example
View File

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

9
config.js
View File

@ -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',
}

912
package-lock.json generated
View File

File diff suppressed because it is too large Load Diff

2
package.json
View File

@ -15,7 +15,7 @@
"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",

56
server.js
View File

@ -1,21 +1,24 @@
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 filesize = require('filesize')
const express = require('express')
const { hash, compare } = require('bcryptjs')
const mongoose = require('mongoose')
const CronJob = require('cron').CronJob
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 () => {
@ -35,21 +38,25 @@ const job = new CronJob(
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.filesize = filesize.partial({ base: 2, standard: 'jedec' })
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)
@ -67,17 +74,17 @@ app.post('/upload', upload.single('file'), async (req, res) => {
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) : '',
})
console.log(`上传文件 ${file.filename}`)
res.status(201).render('upload', { file })
res.status(201).render('upload', { title: '文件上传', file })
})
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)
@ -88,7 +95,7 @@ app.post('/download/:id', async (req, res) => {
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) {
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)
@ -111,6 +118,7 @@ app.get('/file/:md5', 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.listen(PORT, HOST, () => {
console.info(`server is running at http://${HOST}:${PORT}`)
})

17
views/download.pug
View File

@ -1,16 +1,7 @@
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='/share/favicon.ico' type='image/x-icon')
link(rel='stylesheet' href='/share/style.css')
title #{file.filename} | 文件下载
body
h1 文件下载
extends layout
block main
p
a(href='/share') 返回首页
a(href='/') 返回首页
form(method='POST')
label 文件
label= file.filename
@ -29,7 +20,7 @@ html(lang='zh')
button(id='download_button' type='submit') 下载
if !file.password
- const mimetype = file.mimetype.split('/')[0]
- const src = '/share/file/' + file.md5
- const src = '/file/' + file.md5
if mimetype == 'image'
p 预览
img(src=src)

17
views/index.pug
View File

@ -1,21 +1,12 @@
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='/share/favicon.ico' type='image/x-icon')
link(rel='stylesheet' href='/share/style.css')
title 文件分享
body
h1 文件分享
extends layout
block main
p
a(href='upload') 我要上传
ul
if filesharing.length==0
if files.length==0
li 暂无文件
else
for file in filesharing
for file in files
li
a(href=`download/${file.id}`)= file.filename
if file.password

13
views/layout.pug Normal file
View File

@ -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

15
views/upload.pug
View File

@ -1,16 +1,7 @@
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='/share/favicon.ico' type='image/x-icon')
link(rel='stylesheet' href='/share/style.css')
title 文件上传
body
h1 文件上传
extends layout
block main
p
a(href='/share') 返回首页
a(href='/') 返回首页
if file
section
p