mirror of
https://github.com/hedgedoc/hedgedoc.git
synced 2024-12-01 18:24:51 -05:00
1756e76dc3
This should make the imageRouter more modular and easier to extent. Also a lot of code duplication was removed which should simplify maintenance in future. In the new setup we only need to provide a new module file which exports a function called `uploadImage` and takes a filePath and a callback as argument. The callback itself takes an error and an url as parameter. This eliminates the need of a try-catch-block around the statement and re-enabled the optimization in NodeJS. Signed-off-by: Sheogorath <sheogorath@shivering-isles.com>
50 lines
1.3 KiB
JavaScript
50 lines
1.3 KiB
JavaScript
'use strict'
|
|
const fs = require('fs')
|
|
const path = require('path')
|
|
|
|
const config = require('../../config')
|
|
const {getImageMimeType} = require('../../utils')
|
|
|
|
const AWS = require('aws-sdk')
|
|
const awsConfig = new AWS.Config(config.s3)
|
|
const s3 = new AWS.S3(awsConfig)
|
|
|
|
exports.uploadImage = function (imagePath, callback) {
|
|
if (!imagePath || typeof imagePath !== 'string') {
|
|
callback(new Error('Image path is missing or wrong'), null)
|
|
return
|
|
}
|
|
|
|
if (!callback || typeof callback !== 'function') {
|
|
callback(new Error('Callback has to be a function'), null)
|
|
return
|
|
}
|
|
|
|
fs.readFile(imagePath, function (err, buffer) {
|
|
if (err) {
|
|
callback(new Error(err), null)
|
|
return
|
|
}
|
|
let params = {
|
|
Bucket: config.s3bucket,
|
|
Key: path.join('uploads', path.basename(imagePath)),
|
|
Body: buffer
|
|
}
|
|
|
|
const mimeType = getImageMimeType(imagePath)
|
|
if (mimeType) { params.ContentType = mimeType }
|
|
|
|
s3.putObject(params, function (err, data) {
|
|
if (err) {
|
|
callback(new Error(err), null)
|
|
return
|
|
}
|
|
|
|
let s3Endpoint = 's3.amazonaws.com'
|
|
if (config.s3.region && config.s3.region !== 'us-east-1') {
|
|
s3Endpoint = `s3-${config.s3.region}.amazonaws.com`
|
|
}
|
|
callback(null, `https://${s3Endpoint}/${config.s3bucket}/${params.Key}`)
|
|
})
|
|
})
|
|
}
|