overleaf/services/filestore/app/coffee/S3PersistorManager.coffee

156 lines
6.1 KiB
CoffeeScript
Raw Normal View History

http = require('http')
http.globalAgent.maxSockets = 300
https = require('https')
https.globalAgent.maxSockets = 300
2014-02-14 11:39:05 -05:00
settings = require("settings-sharelatex")
request = require("request")
logger = require("logger-sharelatex")
fs = require("fs")
knox = require("knox")
path = require("path")
LocalFileWriter = require("./LocalFileWriter")
Errors = require("./Errors")
2014-02-14 11:39:05 -05:00
_ = require("underscore")
thirtySeconds = 30 * 1000
buildDefaultOptions = (bucketName, method, key)->
return {
aws:
key: settings.filestore.s3.key
secret: settings.filestore.s3.secret
2014-02-14 11:39:05 -05:00
bucket: bucketName
method: method
timeout: thirtySeconds
uri:"https://#{bucketName}.s3.amazonaws.com/#{key}"
}
module.exports =
sendFile: (bucketName, key, fsPath, callback)->
2014-02-14 11:39:05 -05:00
s3Client = knox.createClient
key: settings.filestore.s3.key
secret: settings.filestore.s3.secret
2014-02-14 11:39:05 -05:00
bucket: bucketName
putEventEmiter = s3Client.putFile fsPath, key, (err, res)->
if err?
logger.err err:err, bucketName:bucketName, key:key, fsPath:fsPath,"something went wrong uploading file to s3"
return callback(err)
2014-02-17 11:31:33 -05:00
if !res?
logger.err err:err, res:res, bucketName:bucketName, key:key, fsPath:fsPath, "no response from s3 put file"
return callback("no response from put file")
2014-02-14 11:39:05 -05:00
if res.statusCode != 200
logger.err bucketName:bucketName, key:key, fsPath:fsPath, "non 200 response from s3 putting file"
return callback("non 200 response from s3 on put file")
2014-06-02 12:25:40 -04:00
LocalFileWriter.deleteFile fsPath, (err)->
logger.log res:res, bucketName:bucketName, key:key, fsPath:fsPath,"file uploaded to s3"
callback(err)
2014-02-14 11:39:05 -05:00
putEventEmiter.on "error", (err)->
logger.err err:err, bucketName:bucketName, key:key, fsPath:fsPath, "error emmited on put of file"
callback err
sendStream: (bucketName, key, readStream, callback)->
2014-02-14 11:39:05 -05:00
logger.log bucketName:bucketName, key:key, "sending file to s3"
readStream.on "error", (err)->
logger.err bucketName:bucketName, key:key, "error on stream to send to s3"
LocalFileWriter.writeStream readStream, null, (err, fsPath)=>
if err?
logger.err bucketName:bucketName, key:key, fsPath:fsPath, err:err, "something went wrong writing stream to disk"
return callback(err)
@sendFile bucketName, key, fsPath, callback
# opts may be {start: Number, end: Number}
getFileStream: (bucketName, key, opts, callback = (err, res)->)->
opts = opts || {}
headers = {}
if opts.start? and opts.end?
headers['Range'] = "bytes=#{opts.start}-#{opts.end}"
callback = _.once callback
2014-02-14 11:39:05 -05:00
logger.log bucketName:bucketName, key:key, "getting file from s3"
s3Client = knox.createClient
key: opts.credentials?.auth_key || settings.filestore.s3.key
secret: opts.credentials?.auth_secret || settings.filestore.s3.secret
bucket: bucketName
s3Stream = s3Client.get(key, headers)
s3Stream.end()
s3Stream.on 'response', (res) ->
if res.statusCode == 404
logger.log bucketName:bucketName, key:key, "file not found in s3"
return callback new Errors.NotFoundError("File not found in S3: #{bucketName}:#{key}"), null
if res.statusCode not in [200, 206]
logger.log bucketName:bucketName, key:key, "error getting file from s3: #{res.statusCode}"
return callback new Error("Got non-200 response from S3: #{res.statusCode}"), null
callback null, res
s3Stream.on 'error', (err) ->
2014-02-17 11:31:33 -05:00
logger.err err:err, bucketName:bucketName, key:key, "error getting file stream from s3"
callback err
2014-02-14 11:39:05 -05:00
copyFile: (bucketName, sourceKey, destKey, callback)->
logger.log bucketName:bucketName, sourceKey:sourceKey, destKey:destKey, "copying file in s3"
s3Client = knox.createClient
key: settings.filestore.s3.key
secret: settings.filestore.s3.secret
2014-02-14 11:39:05 -05:00
bucket: bucketName
s3Client.copyFile sourceKey, destKey, (err)->
if err?
2014-02-17 11:31:33 -05:00
logger.err err:err, bucketName:bucketName, sourceKey:sourceKey, destKey:destKey, "something went wrong copying file in aws"
2014-02-14 11:39:05 -05:00
callback(err)
deleteFile: (bucketName, key, callback)->
logger.log bucketName:bucketName, key:key, "delete file in s3"
options = buildDefaultOptions(bucketName, "delete", key)
request options, (err, res)->
if err?
2014-02-17 11:31:33 -05:00
logger.err err:err, res:res, bucketName:bucketName, key:key, "something went wrong deleting file in aws"
2014-02-14 11:39:05 -05:00
callback(err)
deleteDirectory: (bucketName, key, _callback)->
2014-06-02 12:25:40 -04:00
# deleteMultiple can call the callback multiple times so protect against this.
callback = (args...) ->
_callback(args...)
2014-06-02 12:25:40 -04:00
_callback = () ->
logger.log key: key, bucketName: bucketName, "deleting directory"
2014-02-14 11:39:05 -05:00
s3Client = knox.createClient
key: settings.filestore.s3.key
secret: settings.filestore.s3.secret
2014-02-14 11:39:05 -05:00
bucket: bucketName
s3Client.list prefix:key, (err, data)->
2015-03-31 05:35:33 -04:00
if err?
logger.err err:err, bucketName:bucketName, key:key, "something went wrong listing prefix in aws"
return callback(err)
2014-02-14 11:39:05 -05:00
keys = _.map data.Contents, (entry)->
return entry.Key
s3Client.deleteMultiple keys, callback
checkIfFileExists:(bucketName, key, callback)->
logger.log bucketName:bucketName, key:key, "checking if file exists in s3"
options = buildDefaultOptions(bucketName, "head", key)
request options, (err, res)->
if err?
2014-02-19 10:24:29 -05:00
logger.err err:err, res:res, bucketName:bucketName, key:key, "something went wrong checking file in aws"
return callback(err)
if !res?
logger.err err:err, res:res, bucketName:bucketName, key:key, "no response object returned when checking if file exists"
err = new Error("no response from s3 #{bucketName} #{key}")
return callback(err)
2014-02-14 11:39:05 -05:00
exists = res.statusCode == 200
logger.log bucketName:bucketName, key:key, exists:exists, "checked if file exsists in s3"
callback(err, exists)
2016-03-12 02:35:49 -05:00
directorySize:(bucketName, key, callback)->
2016-03-12 02:35:49 -05:00
logger.log bucketName:bucketName, key:key, "get project size in s3"
2016-03-12 09:08:07 -05:00
s3Client = knox.createClient
key: settings.filestore.s3.key
secret: settings.filestore.s3.secret
bucket: bucketName
s3Client.list prefix:key, (err, data)->
if err?
logger.err err:err, bucketName:bucketName, key:key, "something went wrong listing prefix in aws"
return callback(err)
totalSize = 0
_.each data.Contents, (entry)->
totalSize += entry.Size
logger.log totalSize:totalSize, "total size"
callback null, totalSize