feat(plugin/runner): Support pluginEnvVars (#10318)

**Description:**

> Currently, `WasiEnvBuilder` clones a lot to pass environment variables
to Wasm. I think it should get environment variables from the host
dynamically. Any plan to support it?

_Originally posted by @kdy1 in
[#9668](#9668 (comment)

I think this is possible, we can manually add envs to `WasiEnvBuilder`
by `add_env`. So I finsh a rough implementation. More suggestions are
welcome.

.swcrc
```json
{
    "jsc": {
        "experimental": {
            "plugins": [["./my_first_plugin.wasm", {}]],
            "pluginEnvVars": ["BUILD_REPOSITORY_NAME"]
        }
    }
}
```

plugin.rs
```rs
#[plugin_transform]
pub fn process_transform(program: Program, _metadata: TransformPluginProgramMetadata) -> Program {
    for env in std::env::vars() {
        println!("{}: {}", env.0, env.1);
    }
    program
}
```

**Related issue:**

 - Closes #9668

Author: CPunisher

.changeset/forty-spoons-hug.md

@@ -0,0 +1,5 @@
+---
+swc_plugin_runner: major
+---
+
+feat(plugin/runner): Support pluginEnvVars

crates/swc/src/config/mod.rs

@@ -608,6 +608,7 @@ impl Options {
 
                 Box::new(crate::plugin::plugins(
                     experimental.plugins,
+                    experimental.plugin_env_vars,
                     transform_metadata_context,
                     comments.cloned(),
                     cm.clone(),
@@ -1202,6 +1203,8 @@ pub struct JscExperimental {
     /// This requires cargo feature `plugin`.
     #[serde(default)]
     pub plugins: Option<Vec<PluginConfig>>,
+    #[serde(default)]
+    pub plugin_env_vars: Option<Vec<Atom>>,
     /// If true, keeps import assertions in the output.
     #[serde(default, alias = "keepImportAssertions")]
     pub keep_import_attributes: BoolConfig<false>,

crates/swc/src/plugin.rs

@@ -9,6 +9,7 @@
 use std::path::PathBuf;
 
 use anyhow::{Context, Result};
+use atoms::Atom;
 use common::FileName;
 use serde::{Deserialize, Serialize};
 use swc_common::errors::HANDLER;
@@ -34,13 +35,15 @@ pub struct PluginConfig(pub String, pub serde_json::Value);
 
 pub fn plugins(
     configured_plugins: Option<Vec<PluginConfig>>,
+    plugin_env_vars: Option<Vec<Atom>>,
     metadata_context: std::sync::Arc<swc_common::plugin::metadata::TransformPluginMetadataContext>,
     comments: Option<swc_common::comments::SingleThreadedComments>,
     source_map: std::sync::Arc<swc_common::SourceMap>,
     unresolved_mark: swc_common::Mark,
 ) -> impl Pass {
     fold_pass(RustPlugins {
         plugins: configured_plugins,
+        plugin_env_vars: plugin_env_vars.map(std::sync::Arc::new),
         metadata_context,
         comments,
         source_map,
@@ -50,6 +53,7 @@ pub fn plugins(
 
 struct RustPlugins {
     plugins: Option<Vec<PluginConfig>>,
+    plugin_env_vars: Option<std::sync::Arc<Vec<Atom>>>,
     metadata_context: std::sync::Arc<swc_common::plugin::metadata::TransformPluginMetadataContext>,
     comments: Option<swc_common::comments::SingleThreadedComments>,
     source_map: std::sync::Arc<swc_common::SourceMap>,
@@ -125,11 +129,13 @@ impl RustPlugins {
                                 .get_fs_cache_root()
                                 .map(std::path::PathBuf::from),
                         );
+
                         let mut transform_plugin_executor =
                             swc_plugin_runner::create_plugin_transform_executor(
                                 &self.source_map,
                                 &self.unresolved_mark,
                                 &self.metadata_context,
+                                self.plugin_env_vars.clone(),
                                 plugin_module_bytes,
                                 Some(p.1),
                                 runtime,

crates/swc/src/wasm_analysis.rs

@@ -149,6 +149,7 @@ impl Compiler {
                         &self.cm,
                         &unresolved_mark,
                         transform_metadata_context,
+                        None,
                         plugin_module_bytes,
                         Some(p.1.clone()),
                         runtime,

crates/swc_plugin_runner/Cargo.toml

@@ -64,6 +64,7 @@ virtual-fs   = { workspace = true }
 wasmer       = { workspace = true }
 wasmer-wasix = { workspace = true }
 
+swc_atoms = { version = "5.0.0", path = '../swc_atoms' }
 swc_common = { version = "8.0.1", path = "../swc_common", features = [
   "concurrent",
 ] }
@@ -80,7 +81,6 @@ wasmer-compiler-cranelift = { version = "=5.0.5-rc1", default-features = false }
 codspeed-criterion-compat = { workspace = true }
 criterion                 = { workspace = true }
 
-swc_atoms = { version = "5.0.0", path = '../swc_atoms' }
 swc_css_ast = { version = "8.0.0", path = "../swc_css_ast", features = [
   "rkyv-impl",
 ] }

crates/swc_plugin_runner/benches/ecma_invoke.rs

@@ -95,6 +95,7 @@ fn bench_transform(b: &mut Bencher, plugin_dir: &Path) {
                             "development".to_string(),
                             None,
                         )),
+                        None,
                         Box::new(plugin_module.clone()),
                         None,
                         None,

crates/swc_plugin_runner/src/lib.rs

@@ -25,6 +25,7 @@ pub fn create_plugin_transform_executor(
     source_map: &Arc<SourceMap>,
     unresolved_mark: &swc_common::Mark,
     metadata_context: &Arc<TransformPluginMetadataContext>,
+    plugin_env_vars: Option<Arc<Vec<swc_atoms::Atom>>>,
     plugin_module: Box<dyn PluginModuleBytes>,
     plugin_config: Option<serde_json::Value>,
     runtime: Option<Arc<dyn wasmer_wasix::Runtime + Send + Sync>>,
@@ -34,6 +35,7 @@ pub fn create_plugin_transform_executor(
         source_map,
         unresolved_mark,
         metadata_context,
+        plugin_env_vars,
         plugin_config,
         runtime,
     )
@@ -44,6 +46,7 @@ pub fn create_plugin_transform_executor(
     source_map: &Arc<SourceMap>,
     unresolved_mark: &swc_common::Mark,
     metadata_context: &Arc<TransformPluginMetadataContext>,
+    plugin_env_vars: Option<Arc<Vec<swc_atoms::Atom>>>,
     plugin_module: Box<dyn PluginModuleBytes>,
     plugin_config: Option<serde_json::Value>,
     runtime: Option<()>,

crates/swc_plugin_runner/src/transform_executor.rs

@@ -176,6 +176,7 @@ pub struct TransformExecutor {
     source_map: Arc<SourceMap>,
     unresolved_mark: swc_common::Mark,
     metadata_context: Arc<TransformPluginMetadataContext>,
+    plugin_env_vars: Option<Arc<Vec<swc_atoms::Atom>>>,
     plugin_config: Option<serde_json::Value>,
     module_bytes: Box<dyn PluginModuleBytes>,
     runtime: Option<Arc<dyn Runtime + Send + Sync>>,
@@ -192,6 +193,7 @@ impl TransformExecutor {
         source_map: &Arc<SourceMap>,
         unresolved_mark: &swc_common::Mark,
         metadata_context: &Arc<TransformPluginMetadataContext>,
+        plugin_env_vars: Option<Arc<Vec<swc_atoms::Atom>>>,
         plugin_config: Option<serde_json::Value>,
         runtime: Option<Arc<dyn Runtime + Send + Sync>>,
     ) -> Self {
@@ -208,6 +210,7 @@ impl TransformExecutor {
             source_map: source_map.clone(),
             unresolved_mark: *unresolved_mark,
             metadata_context: metadata_context.clone(),
+            plugin_env_vars,
             plugin_config,
             module_bytes,
             runtime,
@@ -289,14 +292,22 @@ impl TransformExecutor {
             // - should we support this?
             // - can we limit to allowlisted input / output only?
             // - should there be a top-level config from .swcrc to manually override this?
-            let wasi_env_builder = if let Ok(cwd) = env::current_dir() {
+            let mut wasi_env_builder = if let Ok(cwd) = env::current_dir() {
                 builder
                     .fs(default_fs_backing())
                     .map_dirs(vec![("/cwd".to_string(), cwd)].drain(..))?
             } else {
                 builder
             };
 
+            if let Some(env_vars) = self.plugin_env_vars.as_ref() {
+                for env in env_vars.iter() {
+                    if let Ok(value) = env::var(env.as_str()) {
+                        wasi_env_builder.add_env(env.as_str(), value);
+                    }
+                }
+            }
+
             //create the `WasiEnv`
             let mut wasi_env = wasi_env_builder.finalize(&mut store)?;
 

crates/swc_plugin_runner/tests/css_rkyv.rs

@@ -107,6 +107,7 @@ fn invoke(input: PathBuf) {
                     "development".to_string(),
                     Some(experimental_metadata),
                 )),
+                None,
                 Box::new(PLUGIN_BYTES.clone()),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,
@@ -159,6 +160,7 @@ fn invoke(input: PathBuf) {
                     "development".to_string(),
                     Some(experimental_metadata.clone()),
                 )),
+                None,
                 Box::new(PLUGIN_BYTES.clone()),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,
@@ -177,6 +179,7 @@ fn invoke(input: PathBuf) {
                     "development".to_string(),
                     Some(experimental_metadata),
                 )),
+                None,
                 Box::new(PLUGIN_BYTES.clone()),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,

crates/swc_plugin_runner/tests/ecma_integration.rs

@@ -140,6 +140,7 @@ fn internal() {
                     "development".to_string(),
                     Some(experimental_metadata),
                 )),
+                None,
                 Box::new(PLUGIN_BYTES.clone()),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,
@@ -214,6 +215,7 @@ fn internal() {
                             "development".to_string(),
                             Some(experimental_metadata),
                         )),
+                        None,
                         Box::new(PLUGIN_BYTES.clone()),
                         Some(json!({ "pluginConfig": "testValue" })),
                         None,

crates/swc_plugin_runner/tests/ecma_rkyv.rs

@@ -116,6 +116,7 @@ fn internal(input: PathBuf) {
                     "development".to_string(),
                     Some(experimental_metadata),
                 )),
+                None,
                 Box::new(PLUGIN_BYTES.clone()),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,
@@ -174,6 +175,7 @@ fn internal(input: PathBuf) {
                     "development".to_string(),
                     Some(experimental_metadata.clone()),
                 )),
+                None,
                 Box::new(PLUGIN_BYTES.clone()),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,
@@ -192,6 +194,7 @@ fn internal(input: PathBuf) {
                     "development".to_string(),
                     Some(experimental_metadata),
                 )),
+                None,
                 Box::new(PLUGIN_BYTES.clone()),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,

crates/swc_plugin_runner/tests/issues.rs

@@ -116,6 +116,7 @@ fn issue_6404() -> Result<(), Error> {
                     "development".to_string(),
                     Some(experimental_metadata),
                 )),
+                None,
                 Box::new(plugin_module),
                 Some(json!({ "pluginConfig": "testValue" })),
                 None,