From 380da5479d9864eab699755cd5023353d33adfb4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Allen=20Zhang=20=28=E5=BC=A0=E6=B6=9B=29?=
 <wr_zhang25@163.com>
Date: Tue, 22 Oct 2024 17:13:20 +0800
Subject: [PATCH] feat: update

---
 plugins/swc-plugin-canyon/package.json |  2 +-
 plugins/swc-plugin-canyon/src/lib.rs   | 22 ++++++++++++++++++++--
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/plugins/swc-plugin-canyon/package.json b/plugins/swc-plugin-canyon/package.json
index 67307e3c..f082a99b 100644
--- a/plugins/swc-plugin-canyon/package.json
+++ b/plugins/swc-plugin-canyon/package.json
@@ -1,6 +1,6 @@
 {
   "name": "swc-plugin-canyon",
-  "version": "0.0.2-alpha.21",
+  "version": "0.0.2-alpha.22",
   "description": "A SWC plugin cooperates with istanbul to report the coverage",
   "main": "swc_plugin_canyon.wasm",
   "scripts": {
diff --git a/plugins/swc-plugin-canyon/src/lib.rs b/plugins/swc-plugin-canyon/src/lib.rs
index 94be5857..8c82e5ee 100644
--- a/plugins/swc-plugin-canyon/src/lib.rs
+++ b/plugins/swc-plugin-canyon/src/lib.rs
@@ -9,6 +9,7 @@ use swc_core::ecma::{
 use swc_core::ecma::ast::{AssignTarget};
 use swc_core::ecma::visit::VisitMutWith;
 use swc_core::plugin::{plugin_transform, proxies::TransformPluginProgramMetadata};
+use swc_core::plugin::metadata::TransformPluginMetadataContextKind;
 
 pub struct TransformVisitor {
     injected: bool,
@@ -99,7 +100,15 @@ impl VisitMut for TransformVisitor {
 }
 
 #[plugin_transform]
-pub fn process_transform(program: Program, _metadata: TransformPluginProgramMetadata) -> Program {
+pub fn process_transform(program: Program, metadata: TransformPluginProgramMetadata) -> Program {
+    // 使用TransformPluginProgramMetadata获取环境变量
+    let env = metadata.get_context(&TransformPluginMetadataContextKind::Env).unwrap_or("-".to_string());
+    let filename = metadata.get_context(&TransformPluginMetadataContextKind::Filename).unwrap_or("-".to_string());
+    let cwd = metadata.get_context(&TransformPluginMetadataContextKind::Cwd).unwrap_or("-".to_string());
+    println!("env: {}", env);
+    println!("filename: {}", filename);
+    println!("cwd: {}", cwd);
+    println!("看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我看我");
     program.fold_with(&mut as_folder(TransformVisitor::new()))
 }
 
@@ -110,5 +119,14 @@ test_inline!(
     // 输入代码
     r#"console.log("transform");"#,
     // 经插件转换后的输出代码
-    r#"console.log("transform"); window.__canyon__ = {dsn: "-",reporter:"-"};"#
+    r#"console.log("transform");
+(new Function('return this')()).__canyon__ = {
+    dsn: "-",
+    reporter: "-",
+    instrumentCwd: "/",
+    branch: "-",
+    sha: "-",
+    projectID: "-"
+};
+"#
 );