var fs = require('fs') var knox = require('knox') var uuid = require('node-uuid') var crypto = require('crypto') var s3 var acceptableuploadTypes = { 'image/gif': 'gif', 'image/jpeg': 'jpg', 'image/jpg': 'jpg', 'image/png': 'png', 'application/pdf': 'pdf', } module.exports = {} module.exports.init = function (opt){ if (opt.local) { s3 = { local: opt.local } } else { s3 = knox.createClient({ key: opt.key, secret: opt.secret, bucket: opt.bucket, }) } } module.exports.put = function (opt) { var filename var err var now = new Date() var file = opt.file // console.log("Received", file.originalname, file.mimetype) var types = opt.types || acceptableuploadTypes var extension = types[file.mimetype] if (opt.filename) { filename = opt.filename } else if (opt.preserveFilename) { filename = file.originalname } else { filename = uuid.v1() + "." + extension; } var remote_path = "/" + opt.dirname + "/" + filename if (! extension) { err = "Unacceptable filetype." } else if (opt.maxSize && file.size > opt.maxSize) { err = "File too large. Uploads can be a maximum of " + opt.maxSize + " bytes." } if (err) { console.error(">>>", err) opt.unacceptable && opt.unacceptable(err) return } opt.acceptable && opt.acceptable(err) // console.log("upload >", remote_path) if (s3.local) { var hash = crypto.createHash("sha256").update(filename, "utf8").digest("hex") var hash_path = hash.substr(0, 2) fs.mkdir(s3.local.localPath + hash_path, function(error){ if (error && error.code !== "EEXIST") { console.error("error creating directory") console.error(error) opt.unacceptable && opt.unacceptable(error) return } var localPath = (s3.local.localPath + hash_path + "/" + filename).replace(/\/\//g, "/") var remotePath = (s3.local.remotePath + hash_path + "/" + filename).replace(/\/\//g, "/") fs.writeFile(localPath, file.buffer, {}, function(error){ if (error) { console.error("error saving file") console.error(error) opt.unacceptable && opt.unacceptable(error) return } opt.success && opt.success(remotePath) }); }) } else { s3.putBuffer(file.buffer, remote_path, { 'Content-Length': file.size, 'Content-Type': file.mimetype, 'x-amz-acl': 'public-read' }, function(err, s3res) { if (err || s3res.statusCode !== 200) { console.error(err); if (s3res && s3res.resume) { s3res.resume() } return; } var file_url = s3res.url || s3res.req.url opt.success && opt.success(file_url) }).on('error', function(err, s3res){ console.error(err) s3res && s3res.resume && s3res.resume() }) } }