codimd/lib/imageRouter/s3.ts

import * as fs from "fs";
import * as path from "path";

import {S3Client} from "@aws-sdk/client-s3-node/S3Client";
import {PutObjectCommand, PutObjectInput} from "@aws-sdk/client-s3-node/commands/PutObjectCommand";

import * as config from "../config";
import {getImageMimeType} from "../utils";
import * as logger from "../logger";


const credentials = {
  accessKeyId: config.s3.accessKeyId,
  secretAccessKey: config.s3.secretAccessKey
}

const s3 = new S3Client({
  credentials,
  region: config.s3.region,
  endpoint: config.s3.endpoint
})

export function uploadImage(imagePath, callback) {
  if (!imagePath || typeof imagePath !== 'string') {
    callback(new Error('Image path is missing or wrong'), null)
    return
  }

  if (!callback || typeof callback !== 'function') {
    logger.error('Callback has to be a function')
    return
  }

  fs.readFile(imagePath, function (err, buffer) {
    if (err) {
      callback(err, null)
      return
    }
    const params: PutObjectInput = {
      Bucket: config.s3bucket,
      Key: path.join('uploads', path.basename(imagePath)),
      Body: buffer,
      ACL: 'public-read'
    }
    const mimeType = getImageMimeType(imagePath)
    if (mimeType) {
      params.ContentType = mimeType
    }

    const command = new PutObjectCommand(params)

    s3.send(command).then(data => {
      let s3Endpoint = 's3.amazonaws.com'
      if (config.s3.endpoint) {
        s3Endpoint = config.s3.endpoint
      } else 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}`)
    }).catch(err => {
      if (err) {
        callback(new Error(err), null)
      }
    })
  })
}