170 lines
4.4 KiB
JavaScript
170 lines
4.4 KiB
JavaScript
'use strict'
|
|
|
|
/* global wiki */
|
|
|
|
const path = require('path')
|
|
const Promise = require('bluebird')
|
|
const fs = Promise.promisifyAll(require('fs-extra'))
|
|
const multer = require('multer')
|
|
const os = require('os')
|
|
const _ = require('lodash')
|
|
|
|
/**
|
|
* Local Disk Storage
|
|
*/
|
|
module.exports = {
|
|
|
|
_uploadsPath: './repo/uploads',
|
|
_uploadsThumbsPath: './data/thumbs',
|
|
|
|
uploadImgHandler: null,
|
|
|
|
/**
|
|
* Initialize Local Data Storage model
|
|
*/
|
|
init () {
|
|
this._uploadsPath = path.resolve(wiki.ROOTPATH, wiki.config.paths.repo, 'uploads')
|
|
this._uploadsThumbsPath = path.resolve(wiki.ROOTPATH, wiki.config.paths.data, 'thumbs')
|
|
|
|
this.createBaseDirectories()
|
|
// this.initMulter()
|
|
|
|
return this
|
|
},
|
|
|
|
/**
|
|
* Init Multer upload handlers
|
|
*/
|
|
initMulter () {
|
|
let maxFileSizes = {
|
|
// img: wiki.config.uploads.maxImageFileSize * 1024 * 1024,
|
|
// file: wiki.config.uploads.maxOtherFileSize * 1024 * 1024
|
|
img: 3 * 1024 * 1024,
|
|
file: 10 * 1024 * 1024
|
|
}
|
|
|
|
// -> IMAGES
|
|
|
|
this.uploadImgHandler = multer({
|
|
storage: multer.diskStorage({
|
|
destination: (req, f, cb) => {
|
|
cb(null, path.resolve(wiki.ROOTPATH, wiki.config.paths.data, 'temp-upload'))
|
|
}
|
|
}),
|
|
fileFilter: (req, f, cb) => {
|
|
// -> Check filesize
|
|
|
|
if (f.size > maxFileSizes.img) {
|
|
return cb(null, false)
|
|
}
|
|
|
|
// -> Check MIME type (quick check only)
|
|
|
|
if (!_.includes(['image/png', 'image/jpeg', 'image/gif', 'image/webp'], f.mimetype)) {
|
|
return cb(null, false)
|
|
}
|
|
|
|
cb(null, true)
|
|
}
|
|
}).array('imgfile', 20)
|
|
|
|
// -> FILES
|
|
|
|
this.uploadFileHandler = multer({
|
|
storage: multer.diskStorage({
|
|
destination: (req, f, cb) => {
|
|
cb(null, path.resolve(wiki.ROOTPATH, wiki.config.paths.data, 'temp-upload'))
|
|
}
|
|
}),
|
|
fileFilter: (req, f, cb) => {
|
|
// -> Check filesize
|
|
|
|
if (f.size > maxFileSizes.file) {
|
|
return cb(null, false)
|
|
}
|
|
|
|
cb(null, true)
|
|
}
|
|
}).array('binfile', 20)
|
|
|
|
return true
|
|
},
|
|
|
|
/**
|
|
* Creates a base directories (Synchronous).
|
|
*/
|
|
createBaseDirectories () {
|
|
try {
|
|
fs.ensureDirSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.data))
|
|
fs.emptyDirSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.data))
|
|
fs.ensureDirSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.data, './cache'))
|
|
fs.ensureDirSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.data, './thumbs'))
|
|
fs.ensureDirSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.data, './temp-upload'))
|
|
|
|
if (os.type() !== 'Windows_NT') {
|
|
fs.chmodSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.data, './temp-upload'), '755')
|
|
}
|
|
|
|
fs.ensureDirSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.repo))
|
|
fs.ensureDirSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.repo, './uploads'))
|
|
|
|
if (os.type() !== 'Windows_NT') {
|
|
fs.chmodSync(path.resolve(wiki.ROOTPATH, wiki.config.paths.repo, './uploads'), '755')
|
|
}
|
|
} catch (err) {
|
|
wiki.logger.error(err)
|
|
}
|
|
|
|
wiki.logger.info('Disk Data Paths: OK')
|
|
},
|
|
|
|
/**
|
|
* Gets the uploads path.
|
|
*
|
|
* @return {String} The uploads path.
|
|
*/
|
|
getUploadsPath () {
|
|
return this._uploadsPath
|
|
},
|
|
|
|
/**
|
|
* Gets the thumbnails folder path.
|
|
*
|
|
* @return {String} The thumbs path.
|
|
*/
|
|
getThumbsPath () {
|
|
return this._uploadsThumbsPath
|
|
},
|
|
|
|
/**
|
|
* Check if filename is valid and unique
|
|
*
|
|
* @param {String} f The filename
|
|
* @param {String} fld The containing folder
|
|
* @param {boolean} isImage Indicates if image
|
|
* @return {Promise<String>} Promise of the accepted filename
|
|
*/
|
|
validateUploadsFilename (f, fld, isImage) {
|
|
let fObj = path.parse(f)
|
|
let fname = _.chain(fObj.name).trim().toLower().kebabCase().value().replace(new RegExp('[^a-z0-9-' + wiki.data.regex.cjk + wiki.data.regex.arabic + ']', 'g'), '')
|
|
let fext = _.toLower(fObj.ext)
|
|
|
|
if (isImage && !_.includes(['.jpg', '.jpeg', '.png', '.gif', '.webp'], fext)) {
|
|
fext = '.png'
|
|
}
|
|
|
|
f = fname + fext
|
|
let fpath = path.resolve(this._uploadsPath, fld, f)
|
|
|
|
return fs.statAsync(fpath).then((s) => {
|
|
throw new Error(wiki.lang.t('errors:fileexists', { path: f }))
|
|
}).catch((err) => {
|
|
if (err.code === 'ENOENT') {
|
|
return f
|
|
}
|
|
throw err
|
|
})
|
|
}
|
|
|
|
}
|