FileStoreHandler.coffee 3.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. logger = require("logger-sharelatex")
  2. fs = require("fs")
  3. request = require("request")
  4. settings = require("settings-sharelatex")
  5. oneMinInMs = 60 * 1000
  6. fiveMinsInMs = oneMinInMs * 5
  7. module.exports = FileStoreHandler =
  8. uploadFileFromDisk: (project_id, file_id, fsPath, callback)->
  9. fs.lstat fsPath, (err, stat)->
  10. if err?
  11. logger.err err:err, project_id:project_id, file_id:file_id, fsPath:fsPath, "error stating file"
  12. callback(err)
  13. if !stat?
  14. logger.err project_id:project_id, file_id:file_id, fsPath:fsPath, "stat is not available, can not check file from disk"
  15. return callback(new Error("error getting stat, not available"))
  16. if !stat.isFile()
  17. logger.log project_id:project_id, file_id:file_id, fsPath:fsPath, "tried to upload symlink, not contining"
  18. return callback(new Error("can not upload symlink"))
  19. _cb = callback
  20. callback = (err) ->
  21. callback = -> # avoid double callbacks
  22. _cb(err)
  23. logger.log project_id:project_id, file_id:file_id, fsPath:fsPath, "uploading file from disk"
  24. readStream = fs.createReadStream(fsPath)
  25. readStream.on "error", (err)->
  26. logger.err err:err, project_id:project_id, file_id:file_id, fsPath:fsPath, "something went wrong on the read stream of uploadFileFromDisk"
  27. callback err
  28. readStream.on "open", () ->
  29. opts =
  30. method: "post"
  31. uri: FileStoreHandler._buildUrl(project_id, file_id)
  32. timeout:fiveMinsInMs
  33. writeStream = request(opts)
  34. writeStream.on "error", (err)->
  35. logger.err err:err, project_id:project_id, file_id:file_id, fsPath:fsPath, "something went wrong on the write stream of uploadFileFromDisk"
  36. callback err
  37. writeStream.on 'response', (response) ->
  38. if response.statusCode not in [200, 201]
  39. err = new Error("non-ok response from filestore for upload: #{response.statusCode}")
  40. logger.err {err, statusCode: response.statusCode}, "error uploading to filestore"
  41. callback(err)
  42. else
  43. callback(null)
  44. readStream.pipe writeStream
  45. getFileStream: (project_id, file_id, query, callback)->
  46. logger.log project_id:project_id, file_id:file_id, query:query, "getting file stream from file store"
  47. queryString = ""
  48. if query? and query["format"]?
  49. queryString = "?format=#{query['format']}"
  50. opts =
  51. method : "get"
  52. uri: "#{@_buildUrl(project_id, file_id)}#{queryString}"
  53. timeout:fiveMinsInMs
  54. headers: {}
  55. if query? and query['range']?
  56. rangeText = query['range']
  57. if rangeText && rangeText.match? && rangeText.match(/\d+-\d+/)
  58. opts.headers['range'] = "bytes=#{query['range']}"
  59. readStream = request(opts)
  60. readStream.on "error", (err) ->
  61. logger.err {err, project_id, file_id, query}, "error in file stream"
  62. callback(null, readStream)
  63. deleteFile: (project_id, file_id, callback)->
  64. logger.log project_id:project_id, file_id:file_id, "telling file store to delete file"
  65. opts =
  66. method : "delete"
  67. uri: @_buildUrl(project_id, file_id)
  68. timeout:fiveMinsInMs
  69. request opts, (err, response)->
  70. if err?
  71. logger.err err:err, project_id:project_id, file_id:file_id, "something went wrong deleting file from filestore"
  72. callback(err)
  73. copyFile: (oldProject_id, oldFile_id, newProject_id, newFile_id, callback)->
  74. logger.log oldProject_id:oldProject_id, oldFile_id:oldFile_id, newProject_id:newProject_id, newFile_id:newFile_id, "telling filestore to copy a file"
  75. opts =
  76. method : "put"
  77. json:
  78. source:
  79. project_id:oldProject_id
  80. file_id:oldFile_id
  81. uri: @_buildUrl(newProject_id, newFile_id)
  82. timeout:fiveMinsInMs
  83. request opts, (err)->
  84. if err?
  85. logger.err err:err, oldProject_id:oldProject_id, oldFile_id:oldFile_id, newProject_id:newProject_id, newFile_id:newFile_id, "something went wrong telling filestore api to copy file"
  86. callback(err)
  87. _buildUrl: (project_id, file_id)->
  88. return "#{settings.apis.filestore.url}/project/#{project_id}/file/#{file_id}"