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

56 lines
1.3 KiB
JavaScript

const fs = require('fs')
const uuid = require('node-uuid')
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')
module.exports = {
promises: {
writeStream,
deleteFile
},
writeStream: callbackify(writeStream),
deleteFile: callbackify(deleteFile)
}
const pipeline = promisify(Stream.pipeline)
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 = uuid.v1()
}
key = key.replace(/\//g, '-')
return path.join(Settings.path.uploadFolder, key)
}