Rewrite set-env command to node.js

package.json

@@ -59,6 +59,7 @@
     "nx-cloud": "16.0.5",
     "prettier": "^2.6.2",
     "react-refresh": "^0.10.0",
+    "string-argv": "^0.3.2",
     "tailwindcss": "^3.3.2",
     "tailwindcss-animate": "^1.0.5",
     "ts-jest": "29.1.0",

packages/internal/core/custom-task-runner.js

@@ -34,10 +34,7 @@ async function loadStageEnv(shouldValidate = true) {
   }
 }
 
-const noValidateTasks = [
-      'tools:bootstrap-infra',
-      'infra-shared:bootstrap'
-  ]
+const noValidateTasks = ['tools:bootstrap-infra', 'infra-shared:bootstrap'];
 
 module.exports = async (...args) => {
   const taskId = args[0]?.[0]?.id;

packages/internal/core/package.json

@@ -4,6 +4,7 @@
   "devDependencies": {
     "env-cmd": "^10.1.0",
     "envalid": "^7.3.1",
-    "dotenv": "^16.1.4"
+    "dotenv": "^16.1.4",
+    "lookpath": "^1.2.2"
   }
 }

packages/internal/core/project.json

@@ -60,7 +60,7 @@
       "executor": "nx:run-commands",
       "options": {
         "color": true,
-        "commands": ["nx docker-compose:up", "nx run webapp:start"],
+        "commands": ["nx run core:docker-compose:up", "nx run webapp:start"],
         "parallel": false
       }
     },
@@ -70,6 +70,15 @@
       "options": {
         "lintFilePatterns": ["packages/internal/core/**/*.ts"]
       }
+    },
+    "set-env": {
+      "executor": "nx:run-commands",
+      "options": {
+        "color": true,
+        "commands": ["node ./scripts/set-env.js {args.env}"],
+        "cwd": "packages/internal/core",
+        "parallel": false
+      }
     }
   },
   "tags": []

packages/internal/core/scripts/lib/runCommand.js

@@ -0,0 +1,19 @@
+const { spawn } = require('node:child_process');
+
+function runCommand(command, args) {
+  return new Promise((resolve, reject) => {
+    const cmd = spawn(command, args, { stdio: 'inherit' });
+
+    cmd.on('close', (code) => {
+      if (code !== 0) {
+        reject(
+          new Error(`"${command} ${args.join(' ')}" failed with code ${code}`)
+        );
+      } else {
+        resolve();
+      }
+    });
+  });
+}
+
+module.exports = { runCommand };

packages/internal/core/scripts/set-env.js

@@ -0,0 +1,37 @@
+#!/usr/bin/env node
+
+const os = require('os');
+const { lookpath } = require('lookpath');
+
+const { runCommand } = require('./lib/runCommand');
+
+const ENV_STAGE = process.argv[2];
+const AWS_VAULT_PROFILE = process.env.AWS_VAULT_PROFILE;
+
+function getCmd() {
+  if (os.platform() === 'win32') {
+    return 'cmd.exe';
+  }
+  return process.env.SHELL;
+}
+
+(async () => {
+  try {
+    if (!ENV_STAGE || ENV_STAGE === 'undefined' || ENV_STAGE === 'local') {
+      console.error('Please set --env argument');
+      process.exit(1);
+    }
+
+    process.env.ENV_STAGE = ENV_STAGE;
+
+    const cmd = getCmd();
+    if (await lookpath('aws-vault')) {
+      await runCommand('aws-vault', ['exec', AWS_VAULT_PROFILE, '--', cmd]);
+    } else {
+      await runCommand(cmd, []);
+    }
+  } catch (err) {
+    console.error(err);
+    process.exit(1);
+  }
+})();