@sudolabs-io/aws-ssm-cli/dist/client.js

Command line tool for AWS Systems Manager Parameter Store

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.addThrottleMiddleware = exports.delay = exports.createClient = void 0;
const client_ssm_1 = require("@aws-sdk/client-ssm");
function createClient({ region, accessKeyId, secretAccessKey } = {}) {
    const credentials = accessKeyId && secretAccessKey
        ? {
            accessKeyId,
            secretAccessKey,
        }
        : undefined;
    return new client_ssm_1.SSMClient({ region, credentials });
}
exports.createClient = createClient;
function delay(time) {
    return __awaiter(this, void 0, void 0, function* () {
        return new Promise((resolve) => setTimeout(resolve, time));
    });
}
exports.delay = delay;
function addThrottleMiddleware(client, { batchSize, wait }) {
    let currentBatchSize = 0;
    // https://aws.amazon.com/blogs/developer/middleware-stack-modular-aws-sdk-js/
    client.middlewareStack.add((next) => (args) => __awaiter(this, void 0, void 0, function* () {
        if (currentBatchSize > batchSize) {
            currentBatchSize = 0;
            yield delay(wait);
        }
        currentBatchSize++;
        return yield next(args);
    }), {
        step: 'initialize',
    });
    return client;
}
exports.addThrottleMiddleware = addThrottleMiddleware;