Merge pull request #12 from mhlabs/local-replays

Local replays

README.md

@@ -115,5 +115,8 @@ If the rule's logical ID is omitted such as `evb local --rule` the tool will par
 *Use cases*: 
 * You want to test the behaviour of an already deployed rule where you don't know the stack's name or where it doesn't belong to a stack.
 
+## Replaying archived events
+Add `--replay` option to command. This will guide you through a wizard to find the archive and set the time range to replay.
+
 ## Forward events to sam-local
 All `evb local` commands support a `--sam-local` flag. When used, events will be passed on to sam-local for more advanced debugging

index.js

@@ -11,7 +11,8 @@ const localPatternListener = require("./src/evb-local/listeners/localPatternList
 const arnListener = require("./src/evb-local/listeners/arnListener");
 const package = require('./package.json');
 const eventTester = require("./src/event-tester");
-const replayBuilder = require("./src/replay-builder");
+const archiveUtil = require("./src/archive-util");
+const inputUtil = require("./src/input-util");
 require("@mhlabs/aws-sdk-sso");
 
 program.version(package.version, "-v, --vers", "output the current version");
@@ -106,7 +107,7 @@ program
   .description("Starts a replay of events against a specific destination")
   .action(async (cmd) => {
     initAuth(cmd);
-    await replayBuilder.replay(cmd.eventbus, cmd.rulePrefix);
+    await archiveUtil.replay(cmd.eventbus, cmd.rulePrefix);
   });
 
 const ruleDefault = "choose from template";
@@ -132,9 +133,18 @@ program
   )
   .option("-p, --profile [profile]", "AWS profile to use")
   .option("--region [region]", "The AWS region to use. Falls back on AWS_REGION environment variable if not specified")
+  .option("--replay", "Presents a UI for selecting archive and time range to replay")
   .description("Initiates local consumption of a stack's EventBridge rules")
   .action(async (cmd) => {
     initAuth(cmd);
+    let replayConfig;
+    if(cmd.replay) {
+      const eventbus = await inputUtil.getEventBusName(new AWS.EventBridge());
+      replayConfig = await archiveUtil.getReplayConfig(eventbus);
+      replayConfig.EventBusName = eventbus;
+    }
+
+
     if (cmd.stackName) {
       await stackListener.init(cmd.stackName, cmd.compact, cmd.samLocal);
       return;
@@ -148,7 +158,8 @@ program
         cmd.rule === ruleDefault ? null : cmd.rule,
         cmd.templateFile,
         cmd.compact,
-        cmd.samLocal
+        cmd.samLocal,
+        replayConfig
       );
       return;
     }

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mhlabs/evb-cli",
-  "version": "1.1.19",
+  "version": "1.1.20",
   "description": "A package for building EventBridge/CloudWatch Events patterns",
   "main": "index.js",
   "scripts": {

sam/package.json

@@ -10,6 +10,7 @@
     ]
   },
   "dependencies": {
+    "aws-sdk": "^2.790.0",
     "node-cache": "^5.1.0",
     "uuid": "^8.0.0",
     "ws": "^7.2.5"

sam/src/Registration.js

@@ -8,13 +8,15 @@ const apigateway = new AWS.ApiGatewayManagementApi({
 });
 
 const dynamoDb = new AWS.DynamoDB.DocumentClient();
+const eventBridge = new AWS.EventBridge();
 
 async function handler(event, context) {
   console.log(event);
   const body = JSON.parse(event.body);
   const token = body.token;
   const localRule = body.localRule;
   const ruleArn = body.ruleArn;
+  const replaySettings = body.replaySettings;
   console.log("Rulearn", ruleArn);
   let ruleNames;
   if (localRule) {
@@ -47,9 +49,25 @@ async function handler(event, context) {
   await apigateway
     .postToConnection({
       ConnectionId: event.requestContext.connectionId,
-      Data: "Connected!",
+      Data: JSON.stringify({
+        Status:
+          "Connected!" +
+          (replaySettings
+            ? `\n\nReplay starting. This can take a few minutes. You can follow the progress here: https://${process.env.AWS_REGION}.console.aws.amazon.com/events/home?region=${process.env.AWS_REGION}#/replay/${replaySettings.ReplayName}`
+            : ""),
+        Rules: ruleNames,
+        EvbLocalRegistration: true,
+      }),
     })
     .promise();
+  if (replaySettings) {
+    replaySettings.Destination.FilterArns = ruleNames.map(
+      (p) =>
+        `arn:aws:events:${process.env.AWS_REGION}:${process.env.AccountId}:rule/${replaySettings.EventBusName}/${p}`
+    );
+    delete replaySettings.EventBusName;
+    const resp = await eventBridge.startReplay(replaySettings).promise();
+  }
 
   return { statusCode: 200 };
 }

sam/src/builders/ruleArnCreator.js

@@ -29,6 +29,11 @@ async function create(event) {
       .promise();
     const newRuleName = eventBridgeClient.getRuleName(busName);
     ruleNames.push(newRuleName);
+    if (body.replaySettings && ruleResponse.EventPattern) {
+      const pattern = JSON.parse(ruleResponse.EventPattern);
+      pattern["replay-name"] = [body.replaySettings.ReplayName];
+      ruleResponse.EventPattern = JSON.stringify(pattern);
+    }
     await eventBridgeClient.putRule(busName, ruleResponse, newRuleName);
     const targets = [];
     for (const target of ruleTargets.Targets) {

sam/src/builders/stackRuleCreator.js

@@ -26,6 +26,10 @@ async function create(event) {
         .promise();
       const newRuleName = eventBridgeClient.getRuleName(busName);
       ruleNames.push(newRuleName);
+      if (body.replayName) {
+        ruleResponse.EventPattern["replay-name"] =
+          body.replaySettings.ReplayName;
+      }
       await eventBridgeClient.putRule(busName, ruleResponse, newRuleName);
       const targets = [];
       for (const target of ruleTargets.Targets) {

sam/template.yml

@@ -12,7 +12,7 @@ Metadata:
     ReadmeUrl: README.md
     Labels: ['eventbridge', 'evb-cli', 'debugging']
     HomePageUrl: https://github.com/mhlabs/evb-cli#readme
-    SemanticVersion: 0.0.5
+    SemanticVersion: 0.0.6
     SourceCodeUrl: https://github.com/mhlabs/evb-cli
 Resources:
   OnConnectFunction:
@@ -117,6 +117,7 @@ Resources:
           - events:ListTargetsByRule
           - events:PutRule
           - events:PutTargets
+          - events:StartReplay
           Resource:
           - "*"
       - Version: '2012-10-17'