From 69dbfd7cb043d1cdb95f5d08ea1ef1f0a334ce77 Mon Sep 17 00:00:00 2001
From: Yagiz Nizipli <yagiz@nizipli.com>
Date: Thu, 16 May 2024 21:13:24 -0400
Subject: [PATCH] cli: add lifecycle event env to `node --run`

---
 doc/api/cli.md          | 11 +++++++++++
 src/node_task_runner.cc |  7 ++++++-
 src/node_task_runner.h  |  1 +
 3 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/doc/api/cli.md b/doc/api/cli.md
index 10ade9da2bd4482..07a0c7b6f6241a0 100644
--- a/doc/api/cli.md
+++ b/doc/api/cli.md
@@ -1847,6 +1847,10 @@ Modules preloaded with `--require` will run before modules preloaded with `--imp
 
 <!-- YAML
 added: v22.0.0
+changes:
+  - version: REPLACEME
+    pr-url: https://github.com/nodejs/node/pull/53032
+    description: `NODE_LIFECYCLE_EVENT` environment variable is added.
 -->
 
 > Stability: 1.1 - Active development
@@ -1887,6 +1891,13 @@ are:
 * Running `pre` or `post` scripts in addition to the specified script.
 * Defining package manager-specific environment variables.
 
+#### Environment variables
+
+The following environment variables are set when running a script with `--run`:
+
+* `NODE_LIFECYCLE_EVENT`: The name of the script being run. For example, if
+  `--run` is used to run `test`, the value of this variable will be `test`.
+
 ### `--secure-heap=n`
 
 <!-- YAML
diff --git a/src/node_task_runner.cc b/src/node_task_runner.cc
index 4b7d541340a79d5..a2c8f01806aecff 100644
--- a/src/node_task_runner.cc
+++ b/src/node_task_runner.cc
@@ -12,6 +12,7 @@ static constexpr const char* bin_path = "/node_modules/.bin";
 #endif  // _WIN32
 
 ProcessRunner::ProcessRunner(std::shared_ptr<InitializationResultImpl> result,
+                             std::string_view script_name,
                              std::string_view command,
                              const PositionalArgs& positional_args) {
   memset(&options_, 0, sizeof(uv_process_options_t));
@@ -85,6 +86,10 @@ ProcessRunner::ProcessRunner(std::shared_ptr<InitializationResultImpl> result,
   }
   uv_os_free_environ(env_items, env_count);
 
+  // Add NODE_LIFECYCLE_EVENT environment variable to the environment
+  // to indicate which script is being run.
+  env_vars_.push_back("NODE_LIFECYCLE_EVENT=" + std::string(script_name));
+
   // Use the stored reference on the instance.
   options_.file = file_.c_str();
 
@@ -276,7 +281,7 @@ void RunTask(std::shared_ptr<InitializationResultImpl> result,
     return;
   }
 
-  auto runner = ProcessRunner(result, command, positional_args);
+  auto runner = ProcessRunner(result, command_id, command, positional_args);
   runner.Run();
 }
 
diff --git a/src/node_task_runner.h b/src/node_task_runner.h
index 3d8973db98ab422..c4016f1429c5671 100644
--- a/src/node_task_runner.h
+++ b/src/node_task_runner.h
@@ -52,6 +52,7 @@ class ProcessRunner {
 };
 
 void RunTask(std::shared_ptr<InitializationResultImpl> result,
+             std::string_view script_name,
              std::string_view command_id,
              const PositionalArgs& positional_args);
 PositionalArgs GetPositionalArgs(const std::vector<std::string>& args);