@cn-shell/aws-utils/dist/aws-s3.js

A Cloud Native extension for AWS

"use strict";
var __createBinding =
  (this && this.__createBinding) ||
  (Object.create
    ? function(o, m, k, k2) {
        if (k2 === undefined) k2 = k;
        Object.defineProperty(o, k2, {
          enumerable: true,
          get: function() {
            return m[k];
          },
        });
      }
    : function(o, m, k, k2) {
        if (k2 === undefined) k2 = k;
        o[k2] = m[k];
      });
var __setModuleDefault =
  (this && this.__setModuleDefault) ||
  (Object.create
    ? function(o, v) {
        Object.defineProperty(o, "default", { enumerable: true, value: v });
      }
    : function(o, v) {
        o["default"] = v;
      });
var __importStar =
  (this && this.__importStar) ||
  function(mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null)
      for (var k in mod)
        if (k !== "default" && Object.hasOwnProperty.call(mod, k))
          __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
  };
var __importDefault =
  (this && this.__importDefault) ||
  function(mod) {
    return mod && mod.__esModule ? mod : { default: mod };
  };
Object.defineProperty(exports, "__esModule", { value: true });
exports.Bucket = void 0;
// imports here
const Aws = __importStar(require("./aws-base"));
const s3_1 = __importDefault(require("aws-sdk/clients/s3"));
// import AWS from "aws-sdk/global";
// AWS.config.logger = console;
// S3 consts here
const S3_API_VER = "2006-03-01";
// Class Bucket here
class Bucket extends Aws.Base {
  // Constructor here
  constructor(name, opts) {
    super(name, opts);
    this._bucket = opts.bucket;
    this.info("S3 Bucket name: %s", this._bucket);
    // Create AWS service objects
    this._s3 = new s3_1.default({
      region: this._region,
      apiVersion: S3_API_VER,
    });
  }
  // Public and Private methods here
  async start() {
    return true;
  }
  async stop() {}
  async healthCheck() {
    return true;
  }
  async listFiles(directory) {
    let success = true;
    let data = await this._s3
      .listObjectsV2({
        Bucket: this._bucket,
        Prefix: `${directory}/`,
      })
      .promise()
      .catch(e => {
        this.error(
          "listFiles (bucket: %s) Error: (%s: %s). Prefix was (%s)",
          this._bucket,
          e.code,
          e,
          directory,
        );
        success = false;
      });
    let files = [];
    if (success && data.Contents !== undefined) {
      // NOTE: Will include a "/" if this is a folder
      for (let content of data.Contents) {
        if (content.Key !== undefined) {
          // This will create an array with 2 elements:
          //  - 1st is the dir
          //  - 2nd is the key
          let parts = content.Key.split(`${directory}/`);
          if (parts.length === 2 && parts[1].length > 0 && parts[1] !== "/") {
            files.push(parts[1]);
          }
        }
      }
    }
    return files;
  }
  async presignFileRequest(operation, directory, file, expires, contentType) {
    let url = await this._s3
      .getSignedUrlPromise(operation, {
        Bucket: this._bucket,
        Key: `${directory}/${file}`,
        Expires: expires,
        ContentType: contentType,
      })
      .catch(e => {
        this.error(
          "presignFileRequest (bucket: %s) Error: (%s: %s). File/Operation  (%s/%s)",
          this._bucket,
          e.code,
          e,
          file,
          operation,
        );
      });
    if (typeof url === "string") {
      return url;
    }
    return undefined;
  }
  async deleteFile(file) {
    let success = true;
    await this._s3
      .deleteObject({
        Bucket: this._bucket,
        Key: file,
      })
      .promise()
      .catch(e => {
        this.error(
          "deleteFile (bucket: %s) Error: (%s: %s). File was (%s)",
          this._bucket,
          e.code,
          e,
          file,
        );
        success = false;
      });
    return success;
  }
}
exports.Bucket = Bucket;