Add the ability to store secret dev-only env vars in .env.secret.toml (#15557)

Max Brunsfeld and Marshall created 1 year ago

Release Notes:

- N/A

---------

Co-authored-by: Marshall <[e-mail]>

Change summary

.gitignore                      |  3 +++
crates/collab/src/bin/dotenv.rs | 16 ++--------------
crates/collab/src/env.rs        | 32 ++++++++++++++++++++++++--------
3 files changed, 29 insertions(+), 22 deletions(-)

Detailed changes

.gitignore

@@ -29,3 +29,6 @@ DerivedData/
 .vscode
 .wrangler
 .flatpak-builder
+
+# Don't commit any secrets to the repo.
+.env.secret.toml

crates/collab/src/bin/dotenv.rs

@@ -1,20 +1,8 @@
-use anyhow::anyhow;
-use std::fs;
+use collab::env::get_dotenv_vars;
 
 fn main() -> anyhow::Result<()> {
-    let env: toml::map::Map<String, toml::Value> = toml::de::from_str(
-        &fs::read_to_string("./.env.toml").map_err(|_| anyhow!("no .env.toml file found"))?,
-    )?;
-
-    for (key, value) in env {
-        let value = match value {
-            toml::Value::String(value) => value,
-            toml::Value::Integer(value) => value.to_string(),
-            toml::Value::Float(value) => value.to_string(),
-            _ => panic!("unsupported TOML value in .env.toml for key {}", key),
-        };
+    for (key, value) in get_dotenv_vars()? {
         println!("export {}=\"{}\"", key, value);
     }
-
     Ok(())
 }

crates/collab/src/env.rs

@@ -1,12 +1,29 @@
-use anyhow::anyhow;
+use anyhow::{anyhow, Result};
 use std::fs;
 
-pub fn load_dotenv() -> anyhow::Result<()> {
-    let env: toml::map::Map<String, toml::Value> = toml::de::from_str(
-        &fs::read_to_string("./crates/collab/.env.toml")
-            .map_err(|_| anyhow!("no .env.toml file found"))?,
-    )?;
+pub fn get_dotenv_vars() -> Result<Vec<(String, String)>> {
+    let mut vars = Vec::new();
+    let env_content = fs::read_to_string("./crates/collab/.env.toml")
+        .map_err(|_| anyhow!("no .env.toml file found"))?;
 
+    add_vars(env_content, &mut vars)?;
+
+    if let Ok(secret_content) = fs::read_to_string("./crates/collab/.env.secret.toml") {
+        add_vars(secret_content, &mut vars)?;
+    }
+
+    Ok(vars)
+}
+
+pub fn load_dotenv() -> Result<()> {
+    for (key, value) in get_dotenv_vars()? {
+        std::env::set_var(key, value);
+    }
+    Ok(())
+}
+
+fn add_vars(env_content: String, vars: &mut Vec<(String, String)>) -> Result<()> {
+    let env: toml::map::Map<String, toml::Value> = toml::de::from_str(&env_content)?;
     for (key, value) in env {
         let value = match value {
             toml::Value::String(value) => value,
@@ -14,8 +31,7 @@ pub fn load_dotenv() -> anyhow::Result<()> {
             toml::Value::Float(value) => value.to_string(),
             _ => panic!("unsupported TOML value in .env.toml for key {}", key),
         };
-        std::env::set_var(key, value);
+        vars.push((key, value));
     }
-
     Ok(())
 }