extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-04-30 16:14:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Performance concerns

Browse Source
This commit is contained in:
Sam Hedin 2020-06-14 14:56:11 +02:00
parent 97bad0e4de
commit 309f5e02a0
1 changed files with 24 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
crates/nu-cli/src/env/directory_specific_environment.rs vendored
View File

@ -9,7 +9,7 @@ pub struct DirectorySpecificEnvironment {
allowed_directories: IndexSet<PathBuf>, allowed_directories: IndexSet<PathBuf>,
//Directory -> Env key. If an environment var has been added from a .nu in a directory, we track it here so we can remove it when the user leaves the directory. //Directory -> Env key. If an environment var has been added from a .nu in a directory, we track it here so we can remove it when the user leaves the directory.
added_env_vars: IndexMap<PathBuf, IndexSet<String>>, added_env_vars: IndexMap<PathBuf, Vec<String>>,
//Directory -> (env_key, value). If a .nu overwrites some existing environment variables, they are added here so that they can be restored later. //Directory -> (env_key, value). If a .nu overwrites some existing environment variables, they are added here so that they can be restored later.
overwritten_env_values: IndexMap<PathBuf, IndexMap<String, OsString>>, overwritten_env_values: IndexMap<PathBuf, IndexMap<String, OsString>>,
@ -84,7 +84,7 @@ impl DirectorySpecificEnvironment {
while let Some(wdir) = working_dir { while let Some(wdir) = working_dir {
if self.allowed_directories.contains(wdir) { if self.allowed_directories.contains(wdir) {
let toml_doc = std::fs::read_to_string(wdir.join(".nu-env").as_path()) let toml_doc = std::fs::read_to_string(wdir.join(".nu-env").as_path())
.unwrap_or_else(|_| r#"[env]"#.to_string()) .unwrap_or_else(|_| "[env]".to_string())
.parse::<toml::Value>()?; .parse::<toml::Value>()?;
toml_doc toml_doc
@ -101,13 +101,14 @@ impl DirectorySpecificEnvironment {
if let Some(val) = std::env::var_os(k) { if let Some(val) = std::env::var_os(k) {
self.overwritten_env_values self.overwritten_env_values
.entry(wdir.to_path_buf()) .entry(wdir.to_path_buf())
.or_insert(IndexMap::new()) .or_insert_with(|| IndexMap::new())
.insert(k.clone(), val); .insert(k.clone(), val);
} else { //Otherwise, we just track that we added it here } else {
//Otherwise, we just track that we added it here
self.added_env_vars self.added_env_vars
.entry(wdir.to_path_buf()) .entry(wdir.to_path_buf())
.or_insert(IndexSet::new()) .or_insert_with(|| vec![])
.insert(k.clone()); .push(k.clone());
} }
} }
}); });
@ -126,29 +127,24 @@ impl DirectorySpecificEnvironment {
pub fn env_vars_to_delete(&mut self) -> Result<Vec<String>> { pub fn env_vars_to_delete(&mut self) -> Result<Vec<String>> {
let current_dir = std::env::current_dir()?; let current_dir = std::env::current_dir()?;
let mut new_added_env_vars: IndexMap<PathBuf, IndexSet<String>> = IndexMap::new(); let mut new_added_env_vars = IndexMap::new();
//Gather up all environment variables that should be deleted. let mut working_dir = Some(current_dir.as_path());
//If we are not in a directory or one of its subdirectories, mark the env_vals it maps to for removal.
let vars_to_delete = self.added_env_vars.iter().fold(
Vec::new(),
|mut vars_to_delete, (directory, env_vars)| {
let mut working_dir = Some(current_dir.as_path());
while let Some(wdir) = working_dir { while let Some(wdir) = working_dir {
if wdir == directory { if let Some(vars_added_by_this_directory) = self.added_env_vars.get(wdir) {
if let Some(vars_added_by_this_directory) = self.added_env_vars.get(wdir) { //If we are still in a directory, we should continue to track the vars it added. new_added_env_vars.insert(wdir.to_path_buf(), vars_added_by_this_directory.clone());
new_added_env_vars.insert(wdir.to_path_buf(), vars_added_by_this_directory.clone()); //If we are still in a directory, we should continue to track the vars it added.
} }
return vars_to_delete; working_dir = working_dir.expect("Root directory has no parent").parent();
} else { }
working_dir = working_dir.expect("Root directory has no parent").parent();
} //Gather up all environment variables that should be deleted.
} let mut vars_to_delete = vec![];
//only delete vars from directories we are not in for (dir, added_keys) in &self.added_env_vars {
vars_to_delete.extend(env_vars.clone()); if !new_added_env_vars.contains_key(dir) {
vars_to_delete vars_to_delete.extend(added_keys.clone());
}, }
); }
self.added_env_vars = new_added_env_vars; self.added_env_vars = new_added_env_vars;
Ok(vars_to_delete) Ok(vars_to_delete)
} }