overleaf/services/filestore/app/js/LocalFileWriter.js

57 lines
1.3 KiB
JavaScript
Raw Normal View History

const fs = require('fs')
const crypto = require('crypto')
const path = require('path')
const Stream = require('stream')
const { callbackify, promisify } = require('util')
const metrics = require('@overleaf/metrics')
const Settings = require('@overleaf/settings')
const { WriteError } = require('./Errors')
2014-02-14 11:39:05 -05:00
module.exports = {
promises: {
writeStream,
2021-07-13 07:04:46 -04:00
deleteFile,
},
writeStream: callbackify(writeStream),
2021-07-13 07:04:46 -04:00
deleteFile: callbackify(deleteFile),
}
2014-02-14 11:39:05 -05:00
const pipeline = promisify(Stream.pipeline)
2014-02-14 11:39:05 -05: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) {
await deleteFile(fsPath)
throw new WriteError('problem writing file locally', { fsPath }, err)
}
}
async function deleteFile(fsPath) {
if (!fsPath) {
return
}
try {
await promisify(fs.unlink)(fsPath)
} catch (err) {
if (err.code !== 'ENOENT') {
throw new WriteError('failed to delete file', { fsPath }, err)
}
}
}
function _getPath(key) {
if (key == null) {
key = crypto.randomUUID()
}
key = key.replace(/\//g, '-')
return path.join(Settings.path.uploadFolder, key)
}