add pass-through method

Author: tmacro

lib/auth/AuthInfo.ts

@@ -52,6 +52,16 @@ export type AccountCanonicalInfoResults = {
     },
 };
 
+export type RateLimitConfig = {
+    RequestsPerSecond?: { Limit: number };
+}
+
+export type AccountRateLimitConfigResults = {
+    message: {
+        body: RateLimitConfig,
+    },
+};
+
 /**
  * Class containing requester's information received from Vault
  * @param {object} info from Vault including arn, canonicalID,

lib/auth/Vault.ts

@@ -6,6 +6,7 @@ import AuthInfo, {
     AuthorizationResults,
     AuthV4Results,
     AccountCanonicalInfo,
+    RateLimitConfig,
 } from './AuthInfo';
 import { ArsenalCallback } from '../types';
 import RequestContext from '../policyEvaluator/RequestContext';
@@ -409,6 +410,41 @@ export default class Vault {
         });
     }
 
+    /**
+     * A getter for account canonical IDs given a list of account IDs
+     * @param canonicalId - account canonicalId
+     * @param log - log object
+     * @param callback - callback function
+     * @returns callback with either error or an object from Vault
+     * containing the rate limit config for the account
+     */
+    getAccountLimitsByCanonicalId(
+        canonicalId: string,
+        log: RequestLogger,
+        callback: ArsenalCallback<RateLimitConfig | undefined>,
+    ) {
+        if (typeof this.client.getAccountLimitsByCanonicalIds !== 'function') {
+            callback(null, undefined);
+            return;
+        }
+
+        log.trace('getting rate limit config from Vault based on canonicalId');
+        const options = {
+            reqUid: log.getSerializedUids(),
+            logger: log,
+        };
+        this.client.getAccountLimitsByCanonicalIds([canonicalId], options, (err, res) => {
+            if (err) {
+                log.debug('received error message from vault', {
+                    error: err,
+                    canonicalId,
+                });
+                return callback(err);
+            }
+            return callback(null, res.message.body[canonicalId]);
+        });
+    }
+
     /** checkPolicies -- call Vault to evaluate policies
      * @param {object} requestContextParams - parameters needed to construct
      * requestContext in Vault