Merge pull request #138 from irreal/master

swyxio · web-flow · commit bdd84895049b · 2019-04-18T15:04:38.000-04:00
Inject client context into functions
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -31,6 +31,7 @@
     "http-proxy": "^1.17.0",
     "inquirer": "^6.2.2",
     "inquirer-autocomplete-prompt": "^1.0.1",
+    "jwt-decode": "^2.2.0",
     "netlify": "2.4.1",
     "netlify-cli-logo": "^1.0.0",
     "node-fetch": "^2.3.0",
diff --git a/src/utils/serve-functions.js b/src/utils/serve-functions.js
@@ -6,6 +6,7 @@ const queryString = require("querystring");
 const path = require("path");
 const getPort = require("get-port");
 const chokidar = require("chokidar");
+const jwtDecode = require("jwt-decode");
 // const chalk = require("chalk");
 const {
   NETLIFYDEVLOG,
@@ -83,6 +84,26 @@ function promiseCallback(promise, callback) {
 //   return path.join(functionPath, `${path.basename(functionPath)}.js`);
 // }
 
+function buildClientContext(headers) {
+  // inject a client context based on auth header, ported over from netlify-lambda (https://github.com/netlify/netlify-lambda/pull/57)
+  if (!headers.authorization) return;
+
+  const parts = headers.authorization.split(" ");
+  if (parts.length !== 2 || parts[0] !== "Bearer") return;
+
+  try {
+    return {
+      identity: {
+        url: "NETLIFY_LAMBDA_LOCALLY_EMULATED_IDENTITY_URL",
+        token: "NETLIFY_LAMBDA_LOCALLY_EMULATED_IDENTITY_TOKEN"
+      },
+      user: jwtDecode(parts[1])
+    };
+  } catch (_) {
+    // Ignore errors - bearer token is not a JWT, probably not intended for us
+  }
+}
+
 function createHandler(dir) {
   const functions = {};
   fs.readdirSync(dir).forEach(file => {
@@ -170,7 +191,11 @@ function createHandler(dir) {
     };
 
     const callback = createCallback(response);
-    const promise = handler.handler(lambdaRequest, {}, callback);
+    const promise = handler.handler(
+      lambdaRequest,
+      { clientContext: buildClientContext(request.headers) || {} },
+      callback
+    );
     promiseCallback(promise, callback);
   };
 }
