add --env option for cli to load env file

cenfun · Sep 22, 2024 · 152f807 · 152f807
1 parent a817069
commit 152f807
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/lib/cli.js b/lib/cli.js
@@ -65,8 +65,21 @@ const checkRegisterFeature = () => {
     return true;
 };
 
+const loadEnv = (cliOptions) => {
+    if (!cliOptions.env) {
+        return;
+    }
+    const envFile = cliOptions.env === true ? '.env' : cliOptions.env;
+    const loadEnvFile = process.loadEnvFile;
+    if (typeof loadEnvFile === 'function') {
+        loadEnvFile(envFile);
+    }
+};
+
 const initNodeOptions = async (cliOptions) => {
 
+    loadEnv(cliOptions);
+
     const supportRegister = checkRegisterFeature();
     if (!supportRegister) {
         return;
@@ -207,6 +220,8 @@ program
     .option('--import <module>', 'preload module at startup')
     .option('--require <module>', 'preload module at startup')
 
+    .option('--env [path]', 'env file (default: ".env")')
+
     .action((_command, cliOptions) => {
         const args = [].concat(program.args).concat(subArgv);
         if (args[0] === '--') {