@middy/sts 3.0.0-alpha.0 → 3.0.0-alpha.4

package/index.js

@@ -0,0 +1,71 @@
+import { canPrefetch, createPrefetchClient, createClient, getCache, getInternal, processCache, modifyCache } from '@middy/util';
+import STS from 'aws-sdk/clients/sts.js';
+const defaults = {
+  AwsClient: STS,
+  awsClientOptions: {},
+  awsClientCapture: undefined,
+  fetchData: {},
+  disablePrefetch: false,
+  cacheKey: 'sts',
+  cacheExpiry: -1,
+  setToContext: false
+};
+
+const stsMiddleware = (opts = {}) => {
+  const options = { ...defaults,
+    ...opts
+  };
+
+  const fetch = (request, cachedValues = {}) => {
+    const values = {};
+
+    for (const internalKey of Object.keys(options.fetchData)) {
+      if (cachedValues[internalKey]) continue;
+      const assumeRoleOptions = options.fetchData[internalKey];
+      assumeRoleOptions.RoleSessionName ?? (assumeRoleOptions.RoleSessionName = 'middy-sts-session-' + Math.ceil(Math.random() * 99999));
+      values[internalKey] = client.assumeRole(assumeRoleOptions).promise().then(resp => ({
+        accessKeyId: resp.Credentials.AccessKeyId,
+        secretAccessKey: resp.Credentials.SecretAccessKey,
+        sessionToken: resp.Credentials.SessionToken
+      })).catch(e => {
+        const value = getCache(options.cacheKey).value ?? {};
+        value[internalKey] = undefined;
+        modifyCache(options.cacheKey, value);
+        throw e;
+      });
+    }
+
+    return values;
+  };
+
+  let prefetch, client;
+
+  if (canPrefetch(options)) {
+    client = createPrefetchClient(options);
+    prefetch = processCache(options, fetch);
+  }
+
+  const stsMiddlewareBefore = async request => {
+    if (!client) {
+      client = await createClient(options, request);
+    }
+
+    const {
+      value
+    } = prefetch ?? processCache(options, fetch, request);
+    Object.assign(request.internal, value);
+
+    if (options.setToContext) {
+      const data = await getInternal(Object.keys(options.fetchData), request);
+      if (options.setToContext) Object.assign(request.context, data);
+    }
+
+    prefetch = null;
+  };
+
+  return {
+    before: stsMiddlewareBefore
+  };
+};
+
+export default stsMiddleware;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@middy/sts",
-  "version": "3.0.0-alpha.0",
+  "version": "3.0.0-alpha.4",
   "description": "STS (Security Token Service) credentials middleware for the middy framework",
   "type": "module",
   "engines": {
@@ -13,11 +13,13 @@
   "exports": "./index.js",
   "types": "index.d.ts",
   "files": [
+    "index.js",
     "index.d.ts"
   ],
   "scripts": {
     "test": "npm run test:unit",
-    "test:unit": "ava"
+    "test:unit": "ava",
+    "test:benchmark": "node __benchmarks__/index.js"
   },
   "license": "MIT",
   "keywords": [
@@ -46,12 +48,12 @@
   },
   "homepage": "https://github.com/middyjs/middy#readme",
   "dependencies": {
-    "@middy/util": "^3.0.0-alpha.0"
+    "@middy/util": "^3.0.0-alpha.4"
   },
   "devDependencies": {
-    "@middy/core": "^3.0.0-alpha.0",
+    "@middy/core": "^3.0.0-alpha.4",
     "aws-sdk": "^2.939.0",
     "aws-xray-sdk": "^3.3.3"
   },
-  "gitHead": "c533f62841c8a39d061d7b94f30ba178f002c8db"
+  "gitHead": "d4bea7f4e21f6a9bbb1f6f6908361169598b9e53"
 }