noprefix now auto-adds prefix if file exists.
All checks were successful
continuous-integration/appveyor/branch AppVeyor build succeeded
All checks were successful
continuous-integration/appveyor/branch AppVeyor build succeeded
This commit is contained in:
parent
2b31fce045
commit
9a0fca4a22
3 changed files with 68 additions and 20 deletions
|
@ -6,13 +6,10 @@ import config from '../config.mjs'
|
||||||
let lastDateString = ''
|
let lastDateString = ''
|
||||||
let incrementor = 1
|
let incrementor = 1
|
||||||
|
|
||||||
export function uploadFile(ctx, siteName, noprefix = false) {
|
function getPrefix() {
|
||||||
return new Promise((res, rej) => {
|
|
||||||
const date = new Date()
|
const date = new Date()
|
||||||
|
|
||||||
let prefix = ''
|
let prefix = ''
|
||||||
|
|
||||||
if (!noprefix) {
|
|
||||||
// Generate 'YYYYMMDD_HHMMSS_' prefix
|
// Generate 'YYYYMMDD_HHMMSS_' prefix
|
||||||
prefix = date
|
prefix = date
|
||||||
.toISOString()
|
.toISOString()
|
||||||
|
@ -27,8 +24,16 @@ export function uploadFile(ctx, siteName, noprefix = false) {
|
||||||
incrementor++
|
incrementor++
|
||||||
} else {
|
} else {
|
||||||
lastDateString = prefix
|
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()
|
var form = new formidable.IncomingForm()
|
||||||
form.uploadDir = `./public/${siteName}`
|
form.uploadDir = `./public/${siteName}`
|
||||||
|
@ -46,6 +51,10 @@ export function uploadFile(ctx, siteName, noprefix = false) {
|
||||||
})
|
})
|
||||||
ctx.req.body = fields
|
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) {
|
fs.rename(files.file.path, `./public/${siteName}/${prefix}${file.name}`, function(err) {
|
||||||
if (err) return rej(err)
|
if (err) return rej(err)
|
||||||
file.path = `./public/${siteName}/${prefix}${file.name}`
|
file.path = `./public/${siteName}/${prefix}${file.name}`
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
{
|
{
|
||||||
"name": "storage-upload",
|
"name": "storage-upload",
|
||||||
"version": "2.1.3",
|
"version": "2.1.4",
|
||||||
"description": "",
|
"description": "",
|
||||||
"main": "index.js",
|
"main": "index.js",
|
||||||
"scripts": {
|
"scripts": {
|
||||||
|
|
|
@ -183,6 +183,45 @@ t.describe('Media (API)', () => {
|
||||||
assert.strictEqual(img.height, 700)
|
assert.strictEqual(img.height, 700)
|
||||||
assert.strictEqual(img.format, 'png')
|
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() {
|
t.timeout(10000).describe('POST /media/resize', function temp() {
|
||||||
|
|
Loading…
Reference in a new issue