@middy/rds-signer 3.0.0-alpha.0 → 3.0.0-alpha.4

package/index.js

@@ -0,0 +1,74 @@
+import { canPrefetch, getInternal, processCache, getCache, modifyCache } from '@middy/util';
+import RDS from 'aws-sdk/clients/rds.js';
+const defaults = {
+  AwsClient: RDS.Signer,
+  awsClientOptions: {},
+  fetchData: {},
+  disablePrefetch: false,
+  cacheKey: 'rds-signer',
+  cacheExpiry: -1,
+  setToContext: false
+};
+
+const rdsSignerMiddleware = (opts = {}) => {
+  const options = { ...defaults,
+    ...opts
+  };
+
+  const fetch = (request, cachedValues = {}) => {
+    const values = {};
+
+    for (const internalKey of Object.keys(options.fetchData)) {
+      if (cachedValues[internalKey]) continue;
+      const client = new options.AwsClient({ ...options.awsClientOptions,
+        ...options.fetchData[internalKey]
+      });
+      values[internalKey] = new Promise((resolve, reject) => {
+        client.getAuthToken({}, (e, token) => {
+          if (e) {
+            reject(e);
+          }
+
+          if (!token.includes('X-Amz-Security-Token=')) {
+            reject(new Error('X-Amz-Security-Token Missing'));
+          }
+
+          resolve(token);
+        });
+      }).catch(e => {
+        const value = getCache(options.cacheKey).value ?? {};
+        value[internalKey] = undefined;
+        modifyCache(options.cacheKey, value);
+        throw e;
+      });
+    }
+
+    return values;
+  };
+
+  let prefetch;
+
+  if (canPrefetch(options)) {
+    prefetch = processCache(options, fetch);
+  }
+
+  const rdsSignerMiddlewareBefore = async 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);
+      Object.assign(request.context, data);
+    }
+
+    prefetch = null;
+  };
+
+  return {
+    before: rdsSignerMiddlewareBefore
+  };
+};
+
+export default rdsSignerMiddleware;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@middy/rds-signer",
-  "version": "3.0.0-alpha.0",
+  "version": "3.0.0-alpha.4",
   "description": "RDS (Relational Database 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,13 +48,13 @@
   },
   "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",
     "@types/node": "^17.0.0",
     "aws-sdk": "^2.939.0",
     "aws-xray-sdk": "^3.3.3"
   },
-  "gitHead": "c533f62841c8a39d061d7b94f30ba178f002c8db"
+  "gitHead": "d4bea7f4e21f6a9bbb1f6f6908361169598b9e53"
 }