✨ register interceptor as extension

Author: CorentinDoue

package.json

@@ -13,7 +13,7 @@
     "cdk": "cdk",
     "deploy": "pnpm build:layer && cdk deploy",
     "start": "cdk watch",
-    "build:layer": "./node_modules/.bin/esbuild  ./src/layers/interceptorExtension/interceptor.ts --bundle --outfile='./src/layers/interceptorExtension/interceptor.js' --platform=node"
+    "build:layer": "./node_modules/.bin/esbuild  ./src/layers/interceptorExtension/interceptor.ts --bundle --outfile='./src/layers/interceptorExtension/interceptor.js' --platform=node --main-fields=module,main"
   },
   "devDependencies": {
     "@swarmion/integration-tests": "^0.27.0",
@@ -30,6 +30,7 @@
     "@aws-cdk/aws-apigatewayv2-integrations-alpha": "^2.77.0-alpha.0",
     "aws-cdk-lib": "^2.77.0",
     "constructs": "^10.2.9",
+    "lambda-extension-service": "^0.0.1",
     "msw": "^1.2.1",
     "node-fetch": "^3.3.1",
     "prettier": "^2.8.8",

pnpm-lock.yaml

@@ -1,11 +1,18 @@
 import { rest } from "msw";
 import { setupServer } from "msw/node";
+import { EventTypes, ExtensionAPIService } from "lambda-extension-service";
 
 console.log("Executing interceptor extension code...");
 
 const server = setupServer(
   rest.all("*", async (req, res, ctx) => {
     const url = req.url.toString();
+    if (
+      process.env.AWS_LAMBDA_RUNTIME_API &&
+      url.includes(process.env.AWS_LAMBDA_RUNTIME_API)
+    ) {
+      return req.passthrough();
+    }
     const method = req.method;
     console.log(`request intercepted: ${method} ${url}`);
 
@@ -19,3 +26,21 @@ const server = setupServer(
 );
 
 server.listen({ onUnhandledRequest: "bypass" });
+
+const main = async () => {
+  const extensionApiService = new ExtensionAPIService({
+    extensionName: "interceptor-extension",
+  });
+  await extensionApiService.register([EventTypes.Invoke]); // EventTypes.Shutdown is not supported for internal extensions
+  extensionApiService.onTelemetryEvent((event) =>
+    console.log("Telemetry event received: ", JSON.stringify(event))
+  );
+  await extensionApiService.registerTelemetry();
+
+  while (true) {
+    const event = await extensionApiService.next();
+    console.log("Next lambda event received: ", JSON.stringify(event));
+  }
+};
+
+main().catch((error) => console.error(error));