Simplify process access to argument iterator

Author: djc

src/cli/common.rs

@@ -15,7 +15,6 @@ use once_cell::sync::Lazy;
 use super::self_update;
 use crate::cli::download_tracker::DownloadTracker;
 use crate::currentprocess::{
-    argsource::ArgSource,
     filesource::{StdinSource, StdoutSource},
     process, terminalsource,
     varsource::VarSource,

src/cli/proxy_mode.rs

@@ -7,7 +7,7 @@ use crate::{
     cli::{common::set_globals, job, self_update},
     command::run_command_for_dir,
     config::Cfg,
-    currentprocess::{argsource::ArgSource, process},
+    currentprocess::process,
     toolchain::names::{LocalToolchainName, ResolvableLocalToolchainName},
     utils::utils,
 };
@@ -18,7 +18,8 @@ pub async fn main(arg0: &str) -> Result<ExitStatus> {
 
     let _setup = job::setup();
 
-    let mut args = process().args_os().skip(1);
+    let process = process();
+    let mut args = process.args_os().skip(1);
 
     // Check for a + toolchain specifier
     let arg1 = args.next();

src/cli/rustup_mode.rs

@@ -24,7 +24,6 @@ use crate::{
     command,
     config::{new_toolchain_with_reason, ActiveReason, Cfg},
     currentprocess::{
-        argsource::ArgSource,
         filesource::{StderrSource, StdoutSource},
         process,
         terminalsource::{self, ColorableTerminal},
@@ -1513,7 +1512,8 @@ async fn man(
 
 fn set_auto_self_update(cfg: &mut Cfg, auto_self_update_mode: &str) -> Result<utils::ExitCode> {
     if self_update::NEVER_SELF_UPDATE {
-        let mut args = process().args_os();
+        let process = process();
+        let mut args = process.args_os();
         let arg0 = args.next().map(PathBuf::from);
         let arg0 = arg0
             .as_ref()

src/currentprocess.rs

@@ -20,14 +20,12 @@ use home::env as home;
 #[cfg(feature = "test")]
 use rand::{thread_rng, Rng};
 
-pub mod argsource;
 pub mod cwdsource;
 pub mod filesource;
 mod homethunk;
 pub mod terminalsource;
 pub mod varsource;
 
-use argsource::*;
 use cwdsource::*;
 use filesource::*;
 use varsource::*;
@@ -69,7 +67,6 @@ use varsource::*;
 #[enum_dispatch]
 pub trait CurrentProcess:
     home::Env
-    + ArgSource
     + CurrentDirSource
     + VarSource
     + StdoutSource
@@ -84,7 +81,6 @@ pub trait CurrentProcess:
 #[derive(Clone, Debug)]
 #[enum_dispatch(
     CurrentProcess,
-    ArgSource,
     CurrentDirSource,
     VarSource,
     StdoutSource,
@@ -111,6 +107,22 @@ impl Process {
             .and_then(std::ffi::OsStr::to_str)
             .map(String::from)
     }
+
+    pub(crate) fn args(&self) -> Box<dyn Iterator<Item = String> + '_> {
+        match self {
+            Process::OSProcess(_) => Box::new(env::args()),
+            #[cfg(feature = "test")]
+            Process::TestProcess(p) => Box::new(p.args.iter().cloned()),
+        }
+    }
+
+    pub(crate) fn args_os(&self) -> Box<dyn Iterator<Item = OsString> + '_> {
+        match self {
+            Process::OSProcess(_) => Box::new(env::args_os()),
+            #[cfg(feature = "test")]
+            Process::TestProcess(p) => Box::new(p.args.iter().map(OsString::from)),
+        }
+    }
 }
 
 /// Obtain the current instance of CurrentProcess