rct-s3-uploader/router.ts

React s3 uploader

import express, { RequestHandler } from 'express'
import aws from 'aws-sdk'

import { v4 as uuidv4 } from 'uuid'
import { TOptions, Ts3Options } from './types'

const checkTrailingSlash = (path: string) =>
  !path.endsWith('/') ? path + '/' : path

export default (options: TOptions, middleWare?: RequestHandler[]) => {
  const S3_BUCKET = options.bucket
  // eslint-disable-next-line @typescript-eslint/no-unused-vars
  const getFileKeyDir = options.getFileKeyDir || ((req: Request) => '')

  if (!S3_BUCKET) throw new Error('S3_BUCKET is required')

  let getS3 = options.getS3
  if (!getS3) {
    const s3Options: Ts3Options = {}
    if (options.region) s3Options.region = options.region
    if (options.signatureVersion) s3Options.signatureVersion = options.signatureVersion

    getS3 = () => new aws.S3(s3Options)
  }

  if (options.uniquePrefix === undefined) options.uniquePrefix = true

  const router = express.Router()

  const tempRedirect = (req: any, res: any) => {
    const params = {
      Bucked: S3_BUCKET,
      Key: checkTrailingSlash(getFileKeyDir(req)) + req.params[0]
    }
    const s3 = (!!getS3 && getS3()) as aws.S3
    s3.getSignedUrl('getObject', params, (err, url) => {
      res.redirect(url)
    })
  }

  router.get(/\/img\/(.*)/, ...(middleWare || []), (req, res) => tempRedirect(req, res))

  router.get(/\/uploads\/(.*)/, ...(middleWare || []), (req, res) => tempRedirect(req, res))

  router.get('/sign', ...(middleWare || []), (req, res) => {
    const filename = (req.query.path || '') + (options.uniquePrefix
      ? uuidv4() + '_' : '') + req.query.objectName
    const mimeType = req.query.contentType
    const fileKey = checkTrailingSlash(getFileKeyDir(req as unknown as Request)) + filename

    if (options.headers) res.set(options.headers)

    const s3 = (!!getS3 && getS3()) as aws.S3
    const params = {
      Bucked: S3_BUCKET,
      Key: fileKey,
      Expires: options.signatureExpires || 60,
      ContentType: mimeType,
      ACL: options.ACL || 'private',
    }

    s3.getSignedUrl('putObject', params, (err, data) => {
      if (err) return res.status(500).send('Cannot create S3 signed URL')
      res.json({
        signedUrl: data,
        publicUrl: '/s3/uploads/' + filename,
        filename,
        fileKey,
      })
    })
  })

  return router
}