Add support for generating bucket CID

filebase · Sep 17, 2024 · f117ce6 · f117ce6
1 parent 1b6fa88
commit f117ce6
Show file tree

Hide file tree

Showing 3 changed files with 80 additions and 1 deletion.
diff --git a/src/bucketManager.js b/src/bucketManager.js
@@ -3,7 +3,7 @@ import {
   DeleteBucketCommand,
   GetBucketAclCommand,
   ListBucketsCommand,
-  PutBucketAclCommand,
+  PutBucketAclCommand, PutBucketTaggingCommand,
   S3Client,
 } from "@aws-sdk/client-s3";
 
@@ -130,6 +130,39 @@ class BucketManager {
       });
     return !(typeof readPermission !== "undefined");
   }
+
+  /**
+   * @summary Generates the IPFS Directory/Folder CID for a given bucket
+   * @param {string} name - The name of the bucket to use.
+   * @returns {Promise<boolean>} A promise that resolves with the CID of the new directory/folder
+   */
+  async generateCid(name) {
+    const command = new PutBucketTaggingCommand({
+      Bucket: name,
+      Tagging: {
+        TagSet: [
+          {
+            Key: "generateBucketCid",
+            Value: "true"
+          }
+        ]
+      }
+    });
+
+    let cid = false;
+    command.middlewareStack.add(
+      (next) => async (args) => {
+        const response = await next(args);
+
+        // Get cid from headers
+        cid = response.response.headers["x-amz-meta-cid"];
+        return response;
+      }
+    );
+
+    await this.#client.send(command);
+    return cid;
+  }
 }
 
 export default BucketManager;
diff --git a/test/bucketManager.spec.cjs b/test/bucketManager.spec.cjs
@@ -30,6 +30,29 @@ test("create bucket", async () => {
   }
 });
 
+test("generate bucket cid", async () => {
+  // Initialize BucketManager
+  const bucketManager = new BucketManager(
+    process.env.TEST_S3_KEY || process.env.TEST_KEY,
+    process.env.TEST_S3_SECRET || process.env.TEST_SECRET,
+  );
+
+  // Create bucket `create-bucket-test-pass`
+  const bucketNameToGenerate = `${TEST_PREFIX}-generate-bucket-test-pass`;
+  await bucketManager.create(bucketNameToGenerate);
+
+  try {
+    // Generate bucket CID
+    const generatedCid = await bucketManager.generateCid(bucketNameToGenerate);
+
+    // Assert new bucket exists
+    assert.equal(generatedCid, "bafybeiczsscdsbs7ffqz55asqdf3smv6klcw3gofszvwlyarci47bgf354");
+  } finally {
+    // Delete new bucket
+    await bucketManager.delete(bucketNameToGenerate);
+  }
+});
+
 test("list buckets", async () => {
   const testBucketName = `${TEST_PREFIX}-list-bucket-test-pass`,
     bucketManager = new BucketManager(

diff --git a/test/bucketManager.spec.mjs b/test/bucketManager.spec.mjs
@@ -30,6 +30,29 @@ test("create bucket", async () => {
   }
 });
 
+test("generate bucket cid", async () => {
+  // Initialize BucketManager
+  const bucketManager = new BucketManager(
+    process.env.TEST_S3_KEY || process.env.TEST_KEY,
+    process.env.TEST_S3_SECRET || process.env.TEST_SECRET,
+  );
+
+  // Create bucket `create-bucket-test-pass`
+  const bucketNameToGenerate = `${TEST_PREFIX}-generate-bucket-test-pass`;
+  await bucketManager.create(bucketNameToGenerate);
+
+  try {
+    // Generate bucket CID
+    const generatedCid = await bucketManager.generateCid(bucketNameToGenerate);
+
+    // Assert new bucket exists
+    assert.equal(generatedCid, "bafybeiczsscdsbs7ffqz55asqdf3smv6klcw3gofszvwlyarci47bgf354");
+  } finally {
+    // Delete new bucket
+    await bucketManager.delete(bucketNameToGenerate);
+  }
+});
+
 test("list buckets", async () => {
   const testBucketName = `${TEST_PREFIX}-list-bucket-test-pass`,
     bucketManager = new BucketManager(