mirror of
https://github.com/BobbyWibowo/lolisafe.git
synced 2024-12-13 16:06:21 +00:00
85 lines
2.0 KiB
JavaScript
85 lines
2.0 KiB
JavaScript
const { promisify } = require('util')
|
|
const fs = require('fs')
|
|
const path = require('path')
|
|
const config = require('./../config')
|
|
const logger = require('./../logger')
|
|
|
|
const self = {}
|
|
|
|
// Promisify these fs functions
|
|
const fsFuncs = [
|
|
'access',
|
|
'copyFile',
|
|
'lstat',
|
|
'mkdir',
|
|
'readdir',
|
|
'readFile',
|
|
'rename',
|
|
'rmdir',
|
|
'symlink',
|
|
'unlink',
|
|
'writeFile'
|
|
]
|
|
|
|
for (const fsFunc of fsFuncs) {
|
|
self[fsFunc] = promisify(fs[fsFunc])
|
|
}
|
|
|
|
self.uploads = path.resolve(config.uploads.folder)
|
|
self.chunks = config.uploads.chunksFolder
|
|
? path.resolve(config.uploads.chunksFolder)
|
|
: path.join(self.uploads, 'chunks')
|
|
self.thumbs = path.join(self.uploads, 'thumbs')
|
|
self.zips = path.join(self.uploads, 'zips')
|
|
|
|
self.thumbPlaceholder = path.resolve(config.uploads.generateThumbs.placeholder || 'public/images/unavailable.png')
|
|
|
|
self.logs = path.resolve(config.logsFolder)
|
|
|
|
self.customPages = path.resolve('pages/custom')
|
|
self.dist = process.env.NODE_ENV === 'development'
|
|
? path.resolve('dist-dev')
|
|
: path.resolve('dist')
|
|
self.public = path.resolve('public')
|
|
|
|
self.errorRoot = path.resolve(config.errorPages.rootDir)
|
|
|
|
const verify = [
|
|
self.uploads,
|
|
self.chunks,
|
|
self.thumbs,
|
|
self.zips,
|
|
self.logs,
|
|
self.customPages
|
|
]
|
|
|
|
self.init = async () => {
|
|
// Check & create directories
|
|
for (const p of verify) {
|
|
try {
|
|
await self.access(p)
|
|
} catch (err) {
|
|
if (err.code !== 'ENOENT') {
|
|
throw err
|
|
} else {
|
|
const mkdir = await self.mkdir(p)
|
|
if (mkdir) logger.log(`Created directory: ${p}`)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Purge any leftover in chunks directory
|
|
const uuidDirs = await self.readdir(self.chunks)
|
|
await Promise.all(uuidDirs.map(async uuid => {
|
|
const root = path.join(self.chunks, uuid)
|
|
const chunks = await self.readdir(root)
|
|
await Promise.all(chunks.map(chunk =>
|
|
self.unlink(path.join(root, chunk))
|
|
))
|
|
await self.rmdir(root)
|
|
}))
|
|
if (uuidDirs.length) logger.log(`Purged ${uuidDirs.length} unfinished chunks`)
|
|
}
|
|
|
|
module.exports = self
|