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

/* eslint-disable
    handle-callback-err,
*/
// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
 * decaffeinate suggestions:
 * DS102: Remove unnecessary code created because of implicit returns
 * DS207: Consider shorter variations of null checks
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
const fs = require('fs')
const uuid = require('node-uuid')
const path = require('path')
const _ = require('underscore')
const logger = require('logger-sharelatex')
const metrics = require('metrics-sharelatex')
const Settings = require('settings-sharelatex')
const Errors = require('./Errors')

module.exports = {
  writeStream(stream, key, callback) {
    const timer = new metrics.Timer('writingFile')
    callback = _.once(callback)
    const fsPath = this._getPath(key)
    logger.log({ fsPath }, 'writing file locally')
    const writeStream = fs.createWriteStream(fsPath)
    writeStream.on('finish', function() {
      timer.done()
      logger.log({ fsPath }, 'finished writing file locally')
      return callback(null, fsPath)
    })
    writeStream.on('error', function(err) {
      logger.err(
        { err, fsPath },
        'problem writing file locally, with write stream'
      )
      return callback(err)
    })
    stream.on('error', function(err) {
      logger.log(
        { err, fsPath },
        'problem writing file locally, with read stream'
      )
      return callback(err)
    })
    return stream.pipe(writeStream)
  },

  getStream(fsPath, _callback) {
    if (_callback == null) {
      _callback = function(err, res) {}
    }
    const callback = _.once(_callback)
    const timer = new metrics.Timer('readingFile')
    logger.log({ fsPath }, 'reading file locally')
    const readStream = fs.createReadStream(fsPath)
    readStream.on('end', function() {
      timer.done()
      return logger.log({ fsPath }, 'finished reading file locally')
    })
    readStream.on('error', function(err) {
      logger.err(
        { err, fsPath },
        'problem reading file locally, with read stream'
      )
      if (err.code === 'ENOENT') {
        return callback(new Errors.NotFoundError(err.message), null)
      } else {
        return callback(err)
      }
    })
    return callback(null, readStream)
  },

  deleteFile(fsPath, callback) {
    if (fsPath == null || fsPath === '') {
      return callback()
    }
    logger.log({ fsPath }, 'removing local temp file')
    return fs.unlink(fsPath, callback)
  },

  _getPath(key) {
    if (key == null) {
      key = uuid.v1()
    }
    key = key.replace(/\//g, '-')
    return path.join(Settings.path.uploadFolder, key)
  }
}