@hiiretail/gcp-infra-generators 1.0.0 → 1.0.1

package/dist/generators/maintenance/manage-states/index.js

@@ -1,294 +1,213 @@
-var __defProp = Object.defineProperty;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-var __commonJS = (cb, mod) => function __require() {
-  return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
-};
-
-// src/validators.js
-var require_validators = __commonJS({
-  "src/validators.js"(exports2, module2) {
-    var path2 = require("path");
-    module2.exports = {
-      chain: /* @__PURE__ */ __name((input, ...validators) => {
-        let msg = "";
-        validators.every((validator) => {
-          msg = validator(input);
-          return msg === true;
-        });
-        return msg === true ? true : msg;
-      }, "chain"),
-      filename: /* @__PURE__ */ __name((input) => {
-        if (!input) {
-          return true;
-        }
-        return path2.basename(input) === input ? true : "Invalid filename";
-      }, "filename"),
-      maxLength: /* @__PURE__ */ __name((input, maxLength) => !input || input.length <= maxLength ? true : `Exceeds max \
-length: ${maxLength}`, "maxLength"),
-      required: /* @__PURE__ */ __name((input) => {
-        const msg = "Required";
-        if (Array.isArray(input)) {
-          return input.length > 0 ? true : msg;
-        }
-        if (input) {
-          return input.trim().length > 0 ? true : msg;
-        }
-        return msg;
-      }, "required")
-    };
-  }
-});
-
-// src/BaseGenerator.js
-var require_BaseGenerator = __commonJS({
-  "src/BaseGenerator.js"(exports2, module2) {
-    var Generator = require("yeoman-generator");
-    var path2 = require("path");
-    var fs2 = require("fs");
-    var inquirer2 = require("inquirer");
-    var { chain, required, filename } = require_validators();
-    module2.exports = class extends Generator {
-      constructor(args, opts) {
-        super(args, opts);
-        this.baseDir = path2.resolve(path2.join(__dirname, ".."));
-        this.destinationRoot(process.cwd());
-        const [command, generator] = opts.namespace.split(":").slice(-2);
-        this.generatorId = path2.join(command, generator);
-        this.sourceRoot(
-          path2.join(this.baseDir, "generators", this.generatorId, "templates")
-        );
-        this.copyDir = (templateDir, targetDir, answers = this.answers, skipIfExists = false) => {
-          if (skipIfExists && fs2.existsSync(targetDir)) {
-            return;
-          }
-          this.fs.copyTpl(
-            this.templatePath(`${templateDir}/**/*`),
-            this.destinationPath(targetDir),
-            answers
-          );
-        };
-        this.listSubDirectories = (parent) => fs2.readdirSync(parent).filter((f) => !f.startsWith(".")).filter((f) => fs2.
-        lstatSync(path2.join(parent, f)).isDirectory()).sort((a, b) => a.localeCompare(b));
-        this.kebabCase = (input) => input.replace(/\s|_/g, "-");
-        this.chooseOrCreatePrompts = (name, getChoicesDirectory) => [
-          {
-            when: /* @__PURE__ */ __name((answers) => fs2.existsSync(getChoicesDirectory(answers)), "when"),
-            type: "list",
-            name,
-            message: `Choose ${name}`,
-            store: true,
-            choices: /* @__PURE__ */ __name((answers) => [
-              ...this.listSubDirectories(getChoicesDirectory(answers)),
-              new inquirer2.Separator(),
-              `Create new ${name}`
-            ], "choices"),
-            validate: required,
-            filter: this.kebabCase
-          },
-          {
-            when: /* @__PURE__ */ __name((answers) => !fs2.existsSync(getChoicesDirectory(answers)) || answers[name] ===
-            `Create-new-${name}`, "when"),
-            type: "input",
-            name: `new-${name}`,
-            message: `New ${name} name`,
-            store: false,
-            validate: /* @__PURE__ */ __name((input) => chain(input, required, filename), "validate"),
-            filter: this.kebabCase
-          },
-          {
-            when: /* @__PURE__ */ __name((answers) => !fs2.existsSync(getChoicesDirectory(answers)) || answers[name] ===
-            "Create-new-tribe", "when"),
-            type: "input",
-            name: "costCenter",
-            message: "Please provide the Cost Center of the Tribe",
-            store: false,
-            validate: /* @__PURE__ */ __name((input) => chain(input, required, filename), "validate"),
-            filter: this.kebabCase
-          }
-        ];
-      }
-    };
-  }
-});
+const fg = require('glob');
+const fs = require('fs');
+const path = require('path');
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const Separator = require('enquirer-separator');
+const { Storage } = require('@google-cloud/storage');
+const BaseGenerator = require('../../../src/BaseGenerator');
 
-// generators/maintenance/manage-states/index.js
-var fg = require("glob");
-var fs = require("fs");
-var path = require("path");
-var chalk = require("chalk");
-var inquirer = require("inquirer");
-var Separator = require("enquirer-separator");
-var { Storage } = require("@google-cloud/storage");
-var BaseGenerator = require_BaseGenerator();
-var getBucketName = /* @__PURE__ */ __name((answers) => {
+const getBucketName = (answers) => {
   const dirName = path.basename(process.cwd());
-  let bucketName = "";
-  if (dirName === "tf-infra-gcp") {
-    bucketName = "platform-tf-admin-prod";
-  } else if (dirName.endsWith("-common")) {
-    bucketName = `tf-state-${dirName.replace("-common", "")}-${answers.env}`;
+  let bucketName = '';
+  if (dirName === 'tf-infra-gcp') {
+    bucketName = 'platform-tf-admin-prod';
+  } else if (dirName.endsWith('-common')) {
+    bucketName = `tf-state-${dirName.replace('-common', '')}-${answers.env}`;
   } else {
     throw new Error(
-      "Run this command from the root of tf-infra-gcp of -common repository"
+      'Run this command from the root of tf-infra-gcp of -common repository',
     );
   }
   return bucketName;
-}, "getBucketName");
-var downloadStateFiles = /* @__PURE__ */ __name((files, dest) => {
+};
+
+const downloadStateFiles = (files, dest) => {
   files.forEach((file) => {
     const destFile = path.join(dest, file.name);
     const modTime = new Date(file.metadata.updated);
     fs.mkdirSync(path.dirname(destFile), { recursive: true });
     file.download({ destination: destFile }, () => {
       console.log(
-        `gs://${file.bucket.name}/${file.name} [${chalk.magenta(file.metadata.size)}] (${chalk.yellow(file.metadata.updated.
-        slice(0, 10))})`
+        `gs://${file.bucket.name}/${file.name} [${chalk.magenta(file.metadata.size)}] (${chalk.yellow(file.metadata.updated.slice(0, 10))})`,
       );
       fs.utimesSync(destFile, modTime, modTime);
     });
   });
-}, "downloadStateFiles");
+};
+
 module.exports = class extends BaseGenerator {
   prompting() {
     const prompts = [
       {
-        type: "list",
-        name: "command",
-        message: "Select operation to do",
+        type: 'list',
+        name: 'command',
+        message: 'Select operation to do',
         store: true,
         choices: [
           {
-            name: "Search for orphan state files",
-            value: "searchOrphanStates"
+            name: 'Search for orphan state files',
+            value: 'searchOrphanStates',
           },
           {
-            name: "Download all state files",
-            value: "downloadAll"
-          }
-        ]
+            name: 'Download all state files',
+            value: 'downloadAll',
+          },
+        ],
       },
+
       {
-        type: "list",
-        name: "env",
-        message: "Select environment to work with",
+        type: 'list',
+        name: 'env',
+        message: 'Select environment to work with',
         store: true,
-        choices: ["staging", "prod"],
-        when: /* @__PURE__ */ __name(() => path.basename(process.cwd()).endsWith("-common"), "when")
+        choices: ['staging', 'prod'],
+        when: () => path.basename(process.cwd()).endsWith('-common'),
       },
+
       {
-        type: "checkbox",
-        name: "stateFiles",
-        message: "Next state files are orphan",
-        when: /* @__PURE__ */ __name(async (response) => response.command === "searchOrphanStates", "when"),
-        choices: /* @__PURE__ */ __name(async (response) => {
+        type: 'checkbox',
+        name: 'stateFiles',
+        message: 'Next state files are orphan',
+        when: async (response) => response.command === 'searchOrphanStates',
+
+        choices: async (response) => {
           const constExpectedStates = [];
           const choices = [];
-          if (path.basename(process.cwd()) === "tf-infra-gcp") {
+
+          if (path.basename(process.cwd()) === 'tf-infra-gcp') {
             const tgFiles = fg.sync(
-              "organization/extendaretail-com/**/terragrunt.hcl",
-              { dot: false }
+              'organization/extendaretail-com/**/terragrunt.hcl',
+              { dot: false },
             );
             tgFiles.forEach((el) => {
-              const genStateFile = el.replace(/^organization\//, "").replace("terragrunt.hcl", "default.tfstate");
+              const genStateFile = el
+                .replace(/^organization\//, '')
+                .replace('terragrunt.hcl', 'default.tfstate');
               constExpectedStates.push(genStateFile);
             });
           } else {
             const tgFiles = fg.sync(`infra/${response.env}/**/terragrunt.hcl`, {
-              dot: false
+              dot: false,
             });
             tgFiles.forEach((el) => {
-              const genStateFile = el.replace(/^infra\//, "").replace("terragrunt.hcl", "terraform.tfstate/default.tfsta\
-te");
+              const genStateFile = el
+                .replace(/^infra\//, '')
+                .replace('terragrunt.hcl', 'terraform.tfstate/default.tfstate');
               constExpectedStates.push(genStateFile);
             });
           }
+
           const storage = new Storage();
           const bucket = getBucketName(response);
-          const [files] = await storage.bucket(bucket).getFiles().catch((err) => {
-            console.log(
-              chalk.red("Error occured during requesting the bucket:")
-            );
-            console.log(err.message);
-            process.exit(1);
-          });
+          const [files] = await storage
+            .bucket(bucket)
+            .getFiles()
+            .catch((err) => {
+              console.log(
+                chalk.red('Error occured during requesting the bucket:'),
+              );
+              console.log(err.message);
+              process.exit(1);
+            });
           await Promise.all(
-            files.filter((file) => !constExpectedStates.includes(file.name)).map(async (file) => {
-              let name = `${file.name} [${chalk.magenta(file.metadata.size)}] (${chalk.yellow(file.metadata.updated.slice(
-              0, 10))})`;
-              if (file.name.endsWith(".tfstate")) {
-                const stateData = JSON.parse(await file.download());
-                if (stateData.resources.length === 0 || stateData.resources.every((i) => i.instances.length === 0)) {
-                  name = `${chalk.green("[empty]")} ${name}`;
+            files
+              .filter((file) => !constExpectedStates.includes(file.name))
+              .map(async (file) => {
+                let name = `${file.name} [${chalk.magenta(file.metadata.size)}] (${chalk.yellow(file.metadata.updated.slice(0, 10))})`;
+
+                if (file.name.endsWith('.tfstate')) {
+                  const stateData = JSON.parse(await file.download());
+                  if (
+                    stateData.resources.length === 0 ||
+                    stateData.resources.every((i) => i.instances.length === 0)
+                  ) {
+                    name = `${chalk.green('[empty]')} ${name}`;
+                  } else {
+                    name = `${chalk.red('[contains]')} ${name}`;
+                  }
+                } else if (file.name.endsWith('.tflock')) {
+                  name = `${chalk.red('[LOCK]')} ${name}`;
                 } else {
-                  name = `${chalk.red("[contains]")} ${name}`;
+                  name = `${chalk.yellowBright('[unknown]')} ${name}`;
                 }
-              } else if (file.name.endsWith(".tflock")) {
-                name = `${chalk.red("[LOCK]")} ${name}`;
-              } else {
-                name = `${chalk.yellowBright("[unknown]")} ${name}`;
-              }
-              choices.push({
-                name,
-                value: file
-              });
-            })
+
+                choices.push({
+                  name,
+                  value: file,
+                });
+              }),
           );
+
           if (choices.length === 0) {
             this.log(
               chalk.green(
-                "There are no orhan state files. All files have related terragrunt code in the repository"
-              )
+                'There are no orhan state files. All files have related terragrunt code in the repository',
+              ),
             );
             process.exit();
           }
-          choices.sort((a, b) => a.value.name > b.value.name ? 1 : -1);
-          choices.push(new Separator("------"));
+
+          choices.sort((a, b) => (a.value.name > b.value.name ? 1 : -1));
+          choices.push(new Separator('------'));
           return choices;
-        }, "choices")
+        },
       },
+
       {
-        type: "list",
-        name: "subCommand",
-        message: "What to do with selected files?",
-        choices: ["nothing", "Download", "Delete"],
-        default: "nothing",
-        when: /* @__PURE__ */ __name((response) => response.command === "searchOrphanStates" && response.stateFiles.length >
-        0, "when")
+        type: 'list',
+        name: 'subCommand',
+        message: 'What to do with selected files?',
+        choices: ['nothing', 'Download', 'Delete'],
+        default: 'nothing',
+        when: (response) =>
+          response.command === 'searchOrphanStates' &&
+          response.stateFiles.length > 0,
       },
+
       {
-        type: "input",
-        name: "savePath",
-        message: "Enter destination directory to save state files to",
-        default: /* @__PURE__ */ __name(async (response) => response.env ? `download-states-${response.env}` : "download\
--states", "default"),
-        when: /* @__PURE__ */ __name(async (response) => response.command === "downloadAll" || response.subCommand === "\
-Download", "when")
-      }
+        type: 'input',
+        name: 'savePath',
+        message: 'Enter destination directory to save state files to',
+        default: async (response) =>
+          response.env ? `download-states-${response.env}` : 'download-states',
+        when: async (response) =>
+          response.command === 'downloadAll' ||
+          response.subCommand === 'Download',
+      },
     ];
+
     return this.prompt(prompts).then((props) => {
       this.answers = props;
     });
   }
+
   writing() {
-    if (this.answers.command === "searchOrphanStates" && this.answers.subCommand === "Download") {
+    if (
+      this.answers.command === 'searchOrphanStates' &&
+      this.answers.subCommand === 'Download'
+    ) {
       downloadStateFiles(this.answers.stateFiles, this.answers.savePath);
     }
-    if (this.answers.command === "searchOrphanStates" && this.answers.subCommand === "Delete") {
-      this.log(chalk.red("Next files is going to be deleted:"));
+
+    if (
+      this.answers.command === 'searchOrphanStates' &&
+      this.answers.subCommand === 'Delete'
+    ) {
+      this.log(chalk.red('Next files is going to be deleted:'));
       this.answers.stateFiles.forEach((file) => {
         this.log(
-          `gs://${file.bucket.name}/${file.name} [${chalk.magenta(file.metadata.size)}] (${chalk.yellow(file.metadata.updated.
-          slice(0, 10))})`
+          `gs://${file.bucket.name}/${file.name} [${chalk.magenta(file.metadata.size)}] (${chalk.yellow(file.metadata.updated.slice(0, 10))})`,
         );
       });
+
       const deleteQuestion = [
         {
-          type: "confirm",
-          name: "confirm",
-          message: "Do you really want to delete them?",
-          default: false
-        }
+          type: 'confirm',
+          name: 'confirm',
+          message: 'Do you really want to delete them?',
+          default: false,
+        },
       ];
       inquirer.prompt(deleteQuestion).then((deleteAnswer) => {
         if (deleteAnswer.confirm === true) {
@@ -298,11 +217,15 @@ Download", "when")
         }
       });
     }
-    if (this.answers.command === "downloadAll") {
+
+    if (this.answers.command === 'downloadAll') {
       const storage = new Storage();
-      storage.bucket(getBucketName(this.answers)).getFiles().then(([files]) => {
-        downloadStateFiles(files, this.answers.savePath);
-      });
+      storage
+        .bucket(getBucketName(this.answers))
+        .getFiles()
+        .then(([files]) => {
+          downloadStateFiles(files, this.answers.savePath);
+        });
     }
   }
 };