noprefix now auto-adds prefix if file exists.
All checks were successful
continuous-integration/appveyor/branch AppVeyor build succeeded

This commit is contained in:
Jonatan Nilsson 2022-08-13 01:03:06 +00:00
parent 2b31fce045
commit 9a0fca4a22
3 changed files with 68 additions and 20 deletions

View file

@ -6,13 +6,10 @@ import config from '../config.mjs'
let lastDateString = ''
let incrementor = 1
export function uploadFile(ctx, siteName, noprefix = false) {
return new Promise((res, rej) => {
function getPrefix() {
const date = new Date()
let prefix = ''
if (!noprefix) {
// Generate 'YYYYMMDD_HHMMSS_' prefix
prefix = date
.toISOString()
@ -27,9 +24,17 @@ export function uploadFile(ctx, siteName, noprefix = false) {
incrementor++
} else {
lastDateString = prefix
incrementor = 1
}
return prefix
}
export function uploadFile(ctx, siteName, noprefix = false) {
return new Promise((res, rej) => {
let prefix = ''
var form = new formidable.IncomingForm()
form.uploadDir = `./public/${siteName}`
form.maxFileSize = config.get('fileSize')
@ -46,6 +51,10 @@ export function uploadFile(ctx, siteName, noprefix = false) {
})
ctx.req.body = fields
if (!noprefix || fs.existsSync(`./public/${siteName}/${prefix}${file.name}`)) {
prefix = getPrefix()
}
fs.rename(files.file.path, `./public/${siteName}/${prefix}${file.name}`, function(err) {
if (err) return rej(err)
file.path = `./public/${siteName}/${prefix}${file.name}`

View file

@ -1,6 +1,6 @@
{
"name": "storage-upload",
"version": "2.1.3",
"version": "2.1.4",
"description": "",
"main": "index.js",
"scripts": {

View file

@ -183,6 +183,45 @@ t.describe('Media (API)', () => {
assert.strictEqual(img.height, 700)
assert.strictEqual(img.format, 'png')
})
t.test('should upload and create file with prefix if exists', async () => {
let token = encode(null, { iss: 'development' }, secret)
await fs.unlink(resolve(`../../public/development/test.png`)).catch(function() {})
let datas = []
datas.push(await client.upload(
`/media/noprefix?token=${token}`,
resolve('test.png')
))
datas.push(await client.upload(
`/media/noprefix?token=${token}`,
resolve('test.png')
))
testFiles.push(datas[0].path)
testFiles.push(datas[1].path)
let prefix = new Date().toISOString().replace(/-/g,'').split('T')[0]
let first = datas[0]
let second = datas[1]
assert.notStrictEqual(first.filename, second.filename)
assert.notOk(first.filename.startsWith(currYear))
assert.notOk(first.filename.startsWith(prefix))
assert.ok(second.filename.startsWith(currYear))
assert.ok(second.filename.startsWith(prefix))
let stats = await Promise.all([
fs.stat(resolve('test.png')),
fs.stat(resolve('../../public/development/test.png')),
fs.stat(resolve('../../public/development/' + second.filename)),
])
assert.strictEqual(stats[0].size, stats[1].size)
assert.strictEqual(stats[0].size, stats[2].size)
})
})
t.timeout(10000).describe('POST /media/resize', function temp() {