chore: adjust load_env() to add .flaskenv and NODE_ENV support

src/main.rs

@@ -166,7 +166,7 @@ fn main() {
     if matches.contains_id("tasks") {
         // load .env for tasks
         if !no_dotenv {
-            dotenv().ok();
+            load_env();
         }
         // inject polyglot for tasks
         polyglot::inject_languages();
@@ -480,6 +480,16 @@ fn format_description(description: &str) -> String {
     return short_desc;
 }
 
+fn load_env() {
+    dotenv().ok();
+    if env::current_dir().unwrap().join(".flaskenv").exists() {
+        dotenv::from_filename(".flaskenv").ok();
+    }
+    if let Some(node_env) = env::var("NODE_ENV") {
+        dotenv::from_filename(format!(".env.{}", node_env)).ok();
+    }
+}
+
 #[cfg(target_family = "unix")]
 fn set_executable(path: &str) {
     use std::os::unix::fs::PermissionsExt;