Allow stack to be chosen from a modal wizard if not provided

Author: mansurpasha

src/index.js

@@ -15,7 +15,8 @@ import {
   checkLogsForErrors,
 } from "./services/processEventLogs";
 import { getLogEvents } from "./services/awsCloudwatchLogs";
-import { clargsWizardModal } from "./modals/clargsWizardModal";
+import { regionWizardModal } from "./modals/regionWizardModal";
+import { stackWizardModal } from "./modals/stackWizardModal";
 
 const blessed = require("blessed");
 const contrib = require("blessed-contrib");
@@ -86,13 +87,6 @@ if (program.sam) {
   }
 }
 
-if (!program.stackName) {
-  console.error(
-    "error: required option '-n, --stack-name <stackName>' not specified"
-  );
-  process.exit(1);
-}
-
 AWS.config.credentials = getAWSCredentials();
 
 let cloudformation;
@@ -492,15 +486,33 @@ class Main {
   }
 }
 
+function promptStackName() {
+  const stackTable = stackWizardModal(screen, program, cloudformation);
+  stackTable.key(["enter"], () => {
+    program.stackName = stackTable.ritems[stackTable.selected];
+    new Main().render();
+  });
+}
+
+function promptRegion() {
+  const regionTable = regionWizardModal(screen, program);
+  regionTable.key(["enter"], () => {
+    program.region = regionTable.ritems[regionTable.selected];
+    AWS.config.region = program.region;
+    updateAWSServices();
+    if (!program.stackName) {
+      promptStackName();
+    } else {
+      new Main().render();
+    }
+  });
+}
+
 function startTool() {
   if (!program.region) {
-    const regionTable = clargsWizardModal(screen, program);
-    regionTable.key(["enter"], () => {
-      program.region = regionTable.ritems[regionTable.selected];
-      AWS.config.region = program.region;
-      updateAWSServices();
-      new Main().render();
-    });
+    promptRegion();
+  } else if (!program.stackName) {
+    promptStackName();
   } else {
     new Main().render();
   }

src/modals/clargsWizardModal.js renamed to src/modals/regionWizardModal.js

@@ -9,7 +9,7 @@ function updateRegionTable(table) {
   table.setItems(regions);
 }
 
-const clargsWizardModal = (screen, program) => {
+const regionWizardModal = (screen, program) => {
   let region = "";
   const wizardLayout = new ModalLayout(screen, 112, 27, true);
 
@@ -18,7 +18,7 @@ const clargsWizardModal = (screen, program) => {
     program.region = region;
   };
 
-  new ModalTitle(wizardLayout, 110, "Select your stack/region");
+  new ModalTitle(wizardLayout, 110, "Select your region");
 
   const regionTable = new InteractiveList(wizardLayout, 110, 20, "Regions");
 
@@ -41,4 +41,4 @@ const clargsWizardModal = (screen, program) => {
   return regionTable;
 };
 
-module.exports = { clargsWizardModal };
+module.exports = { regionWizardModal };

src/modals/stackWizardModal.js

@@ -0,0 +1,57 @@
+import { Box } from "../components/box";
+import { ModalLayout } from "../components/modalLayout";
+import { ModalTitle } from "../components/modalTitle";
+import { InteractiveList } from "../components/interactiveList";
+
+function updateStackTable(table, stacks) {
+  table.setItems(stacks);
+}
+
+function updateStackNames(api, table, screen) {
+  api.describeStacks({}, (err, data) => {
+    if (err) {
+      console.error(err);
+    } else {
+      const stacks = [];
+      data.Stacks.forEach((stack) => {
+        stacks.push(stack.StackName);
+      });
+      updateStackTable(table, stacks);
+      screen.render();
+    }
+  });
+}
+
+const stackWizardModal = (screen, program, cloudformation) => {
+  let stack = "";
+  const wizardLayout = new ModalLayout(screen, 112, 27, true);
+
+  const closeModal = () => {
+    program.stackName = stack;
+    wizardLayout.destroy();
+  };
+
+  new ModalTitle(wizardLayout, 110, "Select your stack");
+
+  const stackTable = new InteractiveList(wizardLayout, 110, 20, "Stacks");
+
+  new Box(
+    wizardLayout,
+    110,
+    4,
+    "Arrow keys to navigate | ENTER to select stack"
+  );
+
+  updateStackNames(cloudformation, stackTable, screen);
+  stackTable.focus();
+
+  stackTable.key(["enter"], () => {
+    stack = stackTable.ritems[stackTable.selected];
+    closeModal();
+  });
+
+  screen.render();
+  return stackTable;
+};
+
+module.exports = { stackWizardModal };