comty/packages/server/classes/StorageClient/index.js

const Minio = require("minio")
import path from "path"

export const generateDefaultBucketPolicy = (payload) => {
    const { bucketName } = payload

    if (!bucketName) {
        throw new Error("bucketName is required")
    }

    return {
        Version: "2012-10-17",
        Statement: [
            {
                Action: [
                    "s3:GetObject"
                ],
                Effect: "Allow",
                Principal: {
                    AWS: [
                        "*"
                    ]
                },
                Resource: [
                    `arn:aws:s3:::${bucketName}/*`
                ],
                Sid: ""
            }
        ]
    }
}

export class StorageClient extends Minio.Client {
    constructor(options) {
        super(options)

        this.defaultBucket = String(options.defaultBucket)
        this.defaultRegion = String(options.defaultRegion)
    }

    composeRemoteURL = (key, extraKey) => {
        let _path = path.join(this.defaultBucket, key)

        if (typeof extraKey === "string") {
            _path = path.join(_path, extraKey)
        }

        return `${this.protocol}//${this.host}:${this.port}/${_path}`
    }

    setDefaultBucketPolicy = async (bucketName) => {
        const policy = generateDefaultBucketPolicy({ bucketName })

        return this.setBucketPolicy(bucketName, JSON.stringify(policy))
    }

    initialize = async () => {
        console.log("🔌 Checking if storage client have default bucket...")

        try {
            const bucketExists = await this.bucketExists(this.defaultBucket)

            if (!bucketExists) {
                console.warn("🪣 Default bucket not exists! Creating new bucket...")

                await this.makeBucket(this.defaultBucket, "s3")

                // set default bucket policy
                await this.setDefaultBucketPolicy(this.defaultBucket)
            }
        } catch (error) {
            console.error(`Failed to check if default bucket exists or create default bucket >`, error)
        }

        try {
            // check if default bucket policy exists
            const bucketPolicy = await this.getBucketPolicy(this.defaultBucket).catch(() => {
                return null
            })

            if (!bucketPolicy) {
                // set default bucket policy
                await this.setDefaultBucketPolicy(this.defaultBucket)
            }
        } catch (error) {
            console.error(`Failed to get or set default bucket policy >`, error)
        }

        console.log("✅ Storage client is ready.")
    }
}

export const createStorageClientInstance = (options) => {
    return new StorageClient({
        endPoint: process.env.S3_ENDPOINT,
        port: Number(process.env.S3_PORT),
        useSSL: ToBoolean(process.env.S3_USE_SSL),
        accessKey: process.env.S3_ACCESS_KEY,
        secretKey: process.env.S3_SECRET_KEY,
        defaultBucket: process.env.S3_BUCKET,
        defaultRegion: process.env.S3_REGION,
        ...options,
    })
}

export default createStorageClientInstance