Reuse the FC env variable if present

1 files changed, 8 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index be71598..b3195c7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,7 @@
 use structopt::StructOpt;
 use toml::Value;
 use std::path::{Path, PathBuf};
+use std::env;
 
 #[derive(Debug, StructOpt)]
 struct Cli {
@@ -67,10 +68,16 @@ add_executable(p1 ../main.f90 {})
     };
     args.extend(vec!["-B", "build", "."]);
     println!("[+] cmake {:?}", args);
+    let env_fc = env::var("FC");
+    let fc : String = if env_fc.is_err() {
+        "gfortran".to_string()
+    } else {
+        env_fc.unwrap()
+    };
     let output = std::process::Command::new("cmake")
                            .args(&args)
                            .current_dir(target_dir)
-                           .env("FC", "gfortran")
+                           .env("FC", fc)
                            .output().unwrap();
     println!("status: {}", output.status);
     println!("stdout: {}", String::from_utf8_lossy(&output.stdout));