LocalFileWriter.coffee 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. fs = require("fs")
  2. uuid = require('node-uuid')
  3. path = require("path")
  4. _ = require("underscore")
  5. logger = require("logger-sharelatex")
  6. metrics = require("metrics-sharelatex")
  7. Settings = require("settings-sharelatex")
  8. module.exports =
  9. writeStream: (stream, key, callback)->
  10. timer = new metrics.Timer("writingFile")
  11. callback = _.once callback
  12. fsPath = @_getPath(key)
  13. logger.log fsPath:fsPath, "writing file locally"
  14. writeStream = fs.createWriteStream(fsPath)
  15. stream.pipe writeStream
  16. writeStream.on "finish", ->
  17. timer.done()
  18. logger.log fsPath:fsPath, "finished writing file locally"
  19. callback(null, fsPath)
  20. writeStream.on "error", (err)->
  21. logger.err err:err, fsPath:fsPath, "problem writing file locally, with write stream"
  22. callback err
  23. stream.on "error", (err)->
  24. logger.log err:err, fsPath:fsPath, "problem writing file locally, with read stream"
  25. callback err
  26. deleteFile: (fsPath, callback)->
  27. logger.log fsPath:fsPath, "removing local temp file"
  28. fs.unlink fsPath, callback
  29. _getPath : (key)->
  30. if !key?
  31. key = uuid.v1()
  32. key = key.replace(/\//g,"-")
  33. path.join(Settings.path.uploadFolder, key)