feat: adding --watch command to ascii-art diagram (#46)

Co-authored-by: ljacobsson <lars@mathem.se>

Author: ljacobsson

commands/asciiart/index.js

@@ -3,6 +3,7 @@ const mxGenerator = require("../../graph/MxGenerator");
 const templateParser = require("../../shared/templateParser");
 const filterConfig = require("../../resources/FilterConfig");
 const mxGraphToAsciiArt = require("./mxgraph-to-asciiart");
+const fs = require("fs");
 program
   .command("ascii-art")
   .alias("a")
@@ -17,17 +18,41 @@ program
   )
   .option("-co, --cdk-output [outputPath]", "CDK synth output path", `cdk.out`)
   .option("-s, --skip-synth", "Skips CDK synth", false)
+  .option(
+    "-w, --watch",
+    "Watch for changes in template and rerender diagram on change",
+    false
+  )
   .option(
     "-e, --exclude-types [excludeTypes...]",
     "List of resource types to exclude when using CI mode"
   )
   .description("Generates an ascii-art diagram from a CloudFormation template")
   .action(async (cmd) => {
+    const xml = await render(cmd);
+    if (cmd.watch) {
+      fs.watchFile(cmd.templateFile, (a, b) => {
+        render(cmd);
+      });
+    }
+  });
+async function render(cmd) {
+  try {
     const template = templateParser.get(cmd).template;
+
+    if (!template.Resources) {
+      console.log("Can't find resources");
+      return;
+    }
     filterConfig.resourceNamesToInclude = Object.keys(template.Resources);
     filterConfig.resourceTypesToInclude = Object.keys(template.Resources).map(
       (p) => template.Resources[p].Type
     );
+    mxGenerator.reset();
     const xml = await mxGenerator.renderTemplate(template);
     mxGraphToAsciiArt.render(xml);
-  });
+    return xml;
+  } catch (err) {
+    console.log(err.message);
+  }
+}

graph/ui/icons.js

@@ -11,6 +11,14 @@ const icons = {
     icon: "mxgraph.aws4.api_gateway",
     serviceType: "networkingcontentdelivery",
   },
+  "AWS::Serverless::HttpApi": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::Serverless::StateMachine": {
+    icon: "mxgraph.aws4.step_functions",
+    serviceType: "applicationintegration",
+  },
   "AWS::Lambda::Permission": {
     icon: "mxgraph.aws4.policy",
     serviceType: "securityidentitycomplicance",
@@ -94,8 +102,59 @@ const icons = {
   "AWS::WAFv2::WebACLAssociation": {
     icon: "mxgraph.aws4.waf",
     serviceType: "securityidentitycomplicance",
-  }
-  
+  },
+  "AWS::ApiGatewayV2::Api": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::ApiGatewayManagedOverrides": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::ApiMapping": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::Authorizer": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::Deployment": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::DomainName": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::Integration": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::IntegrationResponse": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::Model": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::Route": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::RouteResponse": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::Stage": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
+  "AWS::ApiGatewayV2::VpcLink": {
+    icon: "mxgraph.aws4.api_gateway",
+    serviceType: "networkingcontentdelivery",
+  },
 };
 
 const colors = {
@@ -244,7 +303,7 @@ const serviceTranslation = {
   kms: "key_management_service",
   stepfunctions: "step_functions",
   events: "eventbridge",
-  logs: "cloudwatch"
+  logs: "cloudwatch",
 };
 
 function getIcon(type) {

resources/IconMap.js

@@ -31,6 +31,7 @@ function get(cmd) {
     template = fromCDK(cmd);
   }
   templateCache.templates["root"] = template;
+  if (!template.Resources) return { isJson, template };
   Object.keys(template.Resources)
     .filter((p) => template.Resources[p].Type === "AWS::CloudFormation::Stack")
     .map((p) => {