2019-12-16 10:42:31 +00:00
|
|
|
const fs = require('fs')
|
|
|
|
const uuid = require('node-uuid')
|
|
|
|
const path = require('path')
|
2019-12-18 15:40:30 +00:00
|
|
|
const Stream = require('stream')
|
|
|
|
const { callbackify, promisify } = require('util')
|
2019-12-16 10:42:31 +00:00
|
|
|
const metrics = require('metrics-sharelatex')
|
|
|
|
const Settings = require('settings-sharelatex')
|
2019-12-18 15:40:30 +00:00
|
|
|
const { WriteError } = require('./Errors')
|
2014-02-14 16:39:05 +00:00
|
|
|
|
2019-12-16 10:42:31 +00:00
|
|
|
module.exports = {
|
2019-12-18 15:40:30 +00:00
|
|
|
promises: {
|
|
|
|
writeStream,
|
|
|
|
deleteFile
|
2019-12-16 10:42:31 +00:00
|
|
|
},
|
2019-12-18 15:40:30 +00:00
|
|
|
writeStream: callbackify(writeStream),
|
|
|
|
deleteFile: callbackify(deleteFile)
|
|
|
|
}
|
2014-02-14 16:39:05 +00:00
|
|
|
|
2019-12-18 15:40:30 +00:00
|
|
|
const pipeline = promisify(Stream.pipeline)
|
2014-02-14 16:39:05 +00:00
|
|
|
|
2019-12-18 15:40:30 +00:00
|
|
|
async function writeStream(stream, key) {
|
|
|
|
const timer = new metrics.Timer('writingFile')
|
|
|
|
const fsPath = _getPath(key)
|
|
|
|
|
|
|
|
const writeStream = fs.createWriteStream(fsPath)
|
|
|
|
try {
|
|
|
|
await pipeline(stream, writeStream)
|
|
|
|
timer.done()
|
|
|
|
return fsPath
|
|
|
|
} catch (err) {
|
2020-01-06 14:43:23 +00:00
|
|
|
await deleteFile(fsPath)
|
|
|
|
|
2019-12-18 15:40:30 +00:00
|
|
|
throw new WriteError({
|
|
|
|
message: 'problem writing file locally',
|
|
|
|
info: { err, fsPath }
|
|
|
|
}).withCause(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
async function deleteFile(fsPath) {
|
|
|
|
if (!fsPath) {
|
|
|
|
return
|
|
|
|
}
|
2020-01-06 14:43:23 +00:00
|
|
|
try {
|
|
|
|
await promisify(fs.unlink)(fsPath)
|
|
|
|
} catch (err) {
|
|
|
|
if (err.code !== 'ENOENT') {
|
|
|
|
throw new WriteError({
|
|
|
|
message: 'failed to delete file',
|
|
|
|
info: { fsPath }
|
|
|
|
}).withCause(err)
|
|
|
|
}
|
|
|
|
}
|
2019-12-18 15:40:30 +00:00
|
|
|
}
|
2018-11-07 16:23:55 +00:00
|
|
|
|
2019-12-18 15:40:30 +00:00
|
|
|
function _getPath(key) {
|
|
|
|
if (key == null) {
|
|
|
|
key = uuid.v1()
|
2019-12-16 10:42:31 +00:00
|
|
|
}
|
2019-12-18 15:40:30 +00:00
|
|
|
key = key.replace(/\//g, '-')
|
|
|
|
return path.join(Settings.path.uploadFolder, key)
|
2019-12-16 10:42:31 +00:00
|
|
|
}
|