Enable profiling env var

brownsys · Jun 14, 2024 · 0a92416 · 0a92416
1 parent 2080a43
commit 0a92416
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/crates/paralegal-flow/src/lib.rs b/crates/paralegal-flow/src/lib.rs
@@ -391,6 +391,18 @@ impl rustc_plugin::RustcPlugin for DfppPlugin {
 
         let opts = Box::leak(Box::new(plugin_args));
 
+        const RERUN_VAR: &str = "RERUN_WITH_DEBUGGER";
+        if let Ok(debugger) = std::env::var(RERUN_VAR) {
+            println!("Restarting with debugger '{debugger}'");
+            let mut dsplit = debugger.split(' ');
+            let mut cmd = std::process::Command::new(dsplit.next().unwrap());
+            cmd.args(dsplit)
+                .args(std::env::args())
+                .env_remove(RERUN_VAR);
+            println!("{cmd:?}");
+            std::process::exit(cmd.status().unwrap().code().unwrap_or(0));
+        }
+
         compiler_args.extend([
             "--cfg".into(),
             "paralegal".into(),