Class BucketManager

storagetransfer/test/utils/bucket.js:28–254

const uuid = require('uuid');

class BucketManager {
  constructor() {
    this.client = new StorageTransferServiceClient();
    this.storage = new Storage();

    /**
     * The GCP Project ID. Cached after initial request
     */
    this._cachedProjectId = '';

    /**
     * @type {Bucket[]}
     */
    this.gcsBuckets = [];
    /**
     * @type {string[]}
     */
    this.blobStorageContainers = [];
    /**
     * @type {string[]}
     */
    this.s3Buckets = [];
  }

  setupBlobStorageFromConnectionString(connectionString = '') {
    this.blobStorage =
      AzureStorageBlob.BlobServiceClient.fromConnectionString(connectionString);
  }

  setupS3(options = {}) {
    this.s3 = new AWS.S3({apiVersion: '2006-03-01', ...options});
  }

  async getProjectId() {
    if (!this._cachedProjectId) {
      this._cachedProjectId = await this.storage.getProjectId();
    }

    return this._cachedProjectId;
  }

  /**
   * Generates a unique name for GCS and S3 buckets.
   *
   * @returns {string} Name of bucket
   */
  static generateBucketName() {
    return `nodejs-sts-samples-${uuid.v4()}`;
  }

  /**
   * Configures permissions for STS to read/write to the bucket.
   *
   * @param {Bucket} bucket
   * @param {number} waitForPropagation the time in milliseconds, if any, to wait
   *   for the policy to propigate (default 7 minutes). See for details:
   *   - https://cloud.google.com/iam/docs/faq#access_revoke
   *   - https://cloud.google.com/iam/docs/policies#structure