Guillaume Chau
38 lines
971 B
JavaScript
38 lines
971 B
JavaScript
import { createWriteStream } from 'fs'
|
|
import { resolve } from 'path'
|
|
import { sync } from 'mkdirp'
|
|
import { generate } from 'shortid'
|
|
import { db } from './db'
|
|
|
|
const uploadDir = resolve(__dirname, '../../live/uploads')
|
|
|
|
// Ensure upload directory exists
|
|
sync(uploadDir)
|
|
|
|
const storeUpload = async ({ stream, filename }) => {
|
|
const id = generate()
|
|
const file = `${id}-${filename}`
|
|
const path = `${uploadDir}/${file}`
|
|
const urlPath = `files/${file}`
|
|
|
|
return new Promise((resolve, reject) =>
|
|
stream
|
|
.pipe(createWriteStream(path))
|
|
.on('finish', () => resolve({ id, path: urlPath }))
|
|
.on('error', reject)
|
|
)
|
|
}
|
|
|
|
const recordFile = file =>
|
|
db
|
|
.get('uploads')
|
|
.push(file)
|
|
.last()
|
|
.write()
|
|
|
|
export async function processUpload (file) {
|
|
const { stream, filename, mimetype, encoding } = await file
|
|
const { id, path } = await storeUpload({ stream, filename })
|
|
return recordFile({ id, filename, mimetype, encoding, path })
|
|
}
|