71 lines
2.0 KiB
JavaScript
71 lines
2.0 KiB
JavaScript
import AWS from 'aws-sdk'
|
|
import { MEDIA_URL } from '@/lib/constants'
|
|
|
|
const bucketRegion = 'us-east-1'
|
|
const Bucket = process.env.NEXT_PUBLIC_AWS_UPLOAD_BUCKET
|
|
|
|
AWS.config.update({
|
|
region: bucketRegion
|
|
})
|
|
|
|
const config = {
|
|
apiVersion: '2006-03-01',
|
|
s3ForcePathStyle: process.env.NODE_ENV === 'development'
|
|
}
|
|
|
|
export function createPresignedPost ({ key, type, size }) {
|
|
// for local development, we use the NEXT_PUBLIC_MEDIA_URL which
|
|
// is reachable from the host machine
|
|
if (process.env.NODE_ENV === 'development') {
|
|
config.endpoint = process.env.NEXT_PUBLIC_MEDIA_URL
|
|
}
|
|
|
|
const s3 = new AWS.S3(config)
|
|
return new Promise((resolve, reject) => {
|
|
s3.createPresignedPost({
|
|
Bucket,
|
|
Fields: { key },
|
|
Expires: 300,
|
|
Conditions: [
|
|
{ 'Content-Type': type },
|
|
{ 'Cache-Control': 'max-age=31536000' },
|
|
{ acl: 'public-read' },
|
|
['content-length-range', size, size]
|
|
]
|
|
}, (err, preSigned) => { err ? reject(err) : resolve(preSigned) })
|
|
})
|
|
}
|
|
|
|
export async function deleteObjects (keys) {
|
|
// for local development, we use the MEDIA_URL which
|
|
// is reachable from the container network
|
|
if (process.env.NODE_ENV === 'development') {
|
|
config.endpoint = MEDIA_URL
|
|
}
|
|
|
|
const s3 = new AWS.S3(config)
|
|
// max 1000 keys per request
|
|
// see https://docs.aws.amazon.com/cli/latest/reference/s3api/delete-objects.html
|
|
const batchSize = 1000
|
|
const deleted = []
|
|
for (let i = 0; i < keys.length; i += batchSize) {
|
|
const batch = keys.slice(i, i + batchSize)
|
|
await new Promise((resolve, reject) => {
|
|
const params = {
|
|
Bucket,
|
|
Delete: {
|
|
Objects: batch.map(key => ({ Key: String(key) }))
|
|
}
|
|
}
|
|
s3.deleteObjects(params, (err, data) => {
|
|
if (err) return reject(err)
|
|
const deleted = data.Deleted?.map(({ Key }) => parseInt(Key)) || []
|
|
resolve(deleted)
|
|
})
|
|
}).then((deleteConfirmed) => {
|
|
deleted.push(...deleteConfirmed)
|
|
}).catch(console.error)
|
|
}
|
|
return deleted
|
|
}
|