import { PutObjectCommand, S3Client } from '@aws-sdk/client-s3';
import { Buffer } from 'node:buffer';
import { v4 as uuidv4 } from 'uuid';

class AWSUtil {
    constructor() {
        this.s3 = new S3Client({ region: "us-east-2" });
    }

    async uploadFile(base64, ACL = "public-read") {
        const base64Data = Buffer.from(base64.replace(/^data:image\/\w+;base64,/, ""), 'base64');
        const type = base64.split(';')[0].split('/')[1];
        const uuid = uuidv4();

        const params = {
            Bucket: process.env.AWS_S3_BUCKET,
            Key: `${uuid}.${type}`,
            Body: base64Data,
            ContentEncoding: 'base64', // required
            ContentType: `image/${type}` // required
        };

        // Create an object and upload it to the Amazon S3 bucket.
        try {
            const results = await this.s3.send(new PutObjectCommand(params));
            console.log(
                "Successfully created " +
                params.Key +
                " and uploaded it to " +
                params.Bucket +
                "/" +
                params.Key
            );
        } catch (err) {
            console.log("Error", err);
        }

        return `https://tattletires.s3.us-east-2.amazonaws.com/${params.Key}`
    }

    async deleteFile(Location) {
        const params = {
            Bucket: process.env.AWS_S3_BUCKET,
            Key: Location.split("s3.amazonaws.com/").pop()
        };

        const data = await this.s3.deleteObject(params).promise();

        return data;
    }
}

export default AWSUtil;