Simplify process access to environment variables

Author: djc

src/bin/rustup-init.rs

@@ -27,7 +27,7 @@ use rustup::cli::rustup_mode;
 #[cfg(windows)]
 use rustup::cli::self_update;
 use rustup::cli::setup_mode;
-use rustup::currentprocess::{process, varsource::VarSource, with_runtime, OSProcess};
+use rustup::currentprocess::{process, with_runtime, OSProcess};
 use rustup::env_var::RUST_RECURSION_COUNT_MAX;
 use rustup::is_proxyable_tools;
 use rustup::utils::utils::{self, ExitCode};

src/cli/common.rs

@@ -14,7 +14,7 @@ use once_cell::sync::Lazy;
 
 use super::self_update;
 use crate::cli::download_tracker::DownloadTracker;
-use crate::currentprocess::{process, terminalsource, varsource::VarSource};
+use crate::currentprocess::{process, terminalsource};
 use crate::dist::dist::{TargetTriple, ToolchainDesc};
 use crate::dist::manifest::ComponentStatus;
 use crate::install::UpdateStatus;

src/cli/log.rs

@@ -1,7 +1,7 @@
 use std::fmt;
 use std::io::Write;
 
-use crate::currentprocess::{process, terminalsource, varsource::VarSource};
+use crate::currentprocess::{process, terminalsource};
 
 macro_rules! warn {
     ( $ ( $ arg : tt ) * ) => ( $crate::cli::log::warn_fmt ( format_args ! ( $ ( $ arg ) * ) ) )

src/cli/rustup_mode.rs

@@ -1488,8 +1488,6 @@ async fn man(
     command: &str,
     toolchain: Option<PartialToolchainDesc>,
 ) -> Result<utils::ExitCode> {
-    use crate::currentprocess::varsource::VarSource;
-
     let toolchain = Toolchain::from_partial(toolchain, cfg).await?;
     let mut path = toolchain.path().to_path_buf();
     path.push("share");

src/cli/self_update.rs

@@ -66,7 +66,7 @@ use crate::{
         markdown::md,
     },
     config::Cfg,
-    currentprocess::{process, varsource::VarSource},
+    currentprocess::process,
     dist::dist::{self, PartialToolchainDesc, Profile, TargetTriple, ToolchainDesc},
     install::UpdateStatus,
     toolchain::{

src/cli/self_update/shell.rs

@@ -29,7 +29,7 @@ use std::path::PathBuf;
 use anyhow::{bail, Result};
 
 use super::utils;
-use crate::currentprocess::{process, varsource::VarSource};
+use crate::currentprocess::process;
 
 pub(crate) type Shell = Box<dyn UnixShell>;
 

src/cli/self_update/unix.rs

@@ -5,7 +5,7 @@ use anyhow::{bail, Context, Result};
 
 use super::install_bins;
 use super::shell;
-use crate::currentprocess::{process, varsource::VarSource};
+use crate::currentprocess::process;
 use crate::utils::utils;
 use crate::utils::Notification;
 

src/cli/self_update/windows.rs

@@ -12,7 +12,7 @@ use super::super::errors::*;
 use super::common;
 use super::{install_bins, InstallOpts};
 use crate::cli::download_tracker::DownloadTracker;
-use crate::currentprocess::{process, varsource::VarSource};
+use crate::currentprocess::process;
 use crate::dist::dist::TargetTriple;
 use crate::utils::utils;
 use crate::utils::Notification;

src/config.rs

@@ -13,7 +13,7 @@ use tokio_stream::StreamExt;
 
 use crate::{
     cli::self_update::SelfUpdateMode,
-    currentprocess::{process, varsource::VarSource},
+    currentprocess::process,
     dist::{
         dist::{self, PartialToolchainDesc, Profile, ToolchainDesc},
         download::DownloadCfg,

src/currentprocess.rs

@@ -23,10 +23,8 @@ pub mod cwdsource;
 pub mod filesource;
 mod homethunk;
 pub mod terminalsource;
-pub mod varsource;
 
 use cwdsource::*;
-use varsource::*;
 
 /// An abstraction for the current process.
 ///
@@ -63,11 +61,11 @@ use varsource::*;
 /// methods are in performance critical loops (except perhaps progress bars -
 /// and even there we should be doing debouncing and managing update rates).
 #[enum_dispatch]
-pub trait CurrentProcess: CurrentDirSource + VarSource + Debug {}
+pub trait CurrentProcess: CurrentDirSource + Debug {}
 
 /// Allows concrete types for the currentprocess abstraction.
 #[derive(Clone, Debug)]
-#[enum_dispatch(CurrentProcess, CurrentDirSource, VarSource)]
+#[enum_dispatch(CurrentProcess, CurrentDirSource)]
 pub enum Process {
     OSProcess(OSProcess),
     #[cfg(feature = "test")]
@@ -88,6 +86,25 @@ impl Process {
             .map(String::from)
     }
 
+    pub fn var(&self, key: &str) -> Result<String, env::VarError> {
+        match self {
+            Process::OSProcess(_) => env::var(key),
+            #[cfg(feature = "test")]
+            Process::TestProcess(p) => match p.vars.get(key) {
+                Some(val) => Ok(val.to_owned()),
+                None => Err(env::VarError::NotPresent),
+            },
+        }
+    }
+
+    pub(crate) fn var_os(&self, key: &str) -> Option<OsString> {
+        match self {
+            Process::OSProcess(_) => env::var_os(key),
+            #[cfg(feature = "test")]
+            Process::TestProcess(p) => p.vars.get(key).map(OsString::from),
+        }
+    }
+
     pub(crate) fn args(&self) -> Box<dyn Iterator<Item = String> + '_> {
         match self {
             Process::OSProcess(_) => Box::new(env::args()),