Simplify process access to stdin

Author: djc

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::{filesource::StdinSource, terminalsource, varsource::VarSource};
+use crate::currentprocess::{terminalsource, varsource::VarSource};
 use crate::dist::dist::{TargetTriple, ToolchainDesc};
 use crate::install::UpdateStatus;
 use crate::utils::notifications as util_notifications;

src/currentprocess.rs

@@ -26,7 +26,6 @@ pub mod terminalsource;
 pub mod varsource;
 
 use cwdsource::*;
-use filesource::*;
 use varsource::*;
 
 /// An abstraction for the current process.
@@ -64,20 +63,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:
-    home::Env + CurrentDirSource + VarSource + StdinSource + ProcessSource + Debug
-{
-}
+pub trait CurrentProcess: home::Env + CurrentDirSource + VarSource + ProcessSource + Debug {}
 
 /// Allows concrete types for the currentprocess abstraction.
 #[derive(Clone, Debug)]
-#[enum_dispatch(
-    CurrentProcess,
-    CurrentDirSource,
-    VarSource,
-    StdinSource,
-    ProcessSource
-)]
+#[enum_dispatch(CurrentProcess, CurrentDirSource, VarSource, ProcessSource)]
 pub enum Process {
     OSProcess(OSProcess),
     #[cfg(feature = "test")]
@@ -114,6 +104,14 @@ impl Process {
         }
     }
 
+    pub(crate) fn stdin(&self) -> Box<dyn filesource::Stdin> {
+        match self {
+            Process::OSProcess(_) => Box::new(io::stdin()),
+            #[cfg(feature = "test")]
+            Process::TestProcess(p) => Box::new(filesource::TestStdin(p.stdin.clone())),
+        }
+    }
+
     pub(crate) fn stdout(&self) -> Box<dyn filesource::Writer> {
         match self {
             Process::OSProcess(_) => Box::new(io::stdout()),
@@ -229,9 +227,9 @@ pub struct TestProcess {
     pub args: Vec<String>,
     pub vars: HashMap<String, String>,
     pub id: u64,
-    pub stdin: TestStdinInner,
-    pub stdout: TestWriterInner,
-    pub stderr: TestWriterInner,
+    pub stdin: filesource::TestStdinInner,
+    pub stdout: filesource::TestWriterInner,
+    pub stderr: filesource::TestWriterInner,
 }
 
 #[cfg(feature = "test")]

src/currentprocess/filesource.rs

@@ -1,8 +1,6 @@
 use std::io::{self, BufRead, Cursor, Read, Result, Write};
 use std::sync::{Arc, Mutex, MutexGuard};
 
-use enum_dispatch::enum_dispatch;
-
 use crate::currentprocess::process;
 
 use super::terminalsource::{ColorableTerminal, StreamSelector};
@@ -16,12 +14,6 @@ pub trait Stdin {
 /// Stand-in for std::io::StdinLock
 pub trait StdinLock: Read + BufRead {}
 
-/// Stand-in for std::io::stdin
-#[enum_dispatch]
-pub trait StdinSource {
-    fn stdin(&self) -> Box<dyn Stdin>;
-}
-
 // ----------------- OS support for stdin -----------------
 
 impl StdinLock for io::StdinLock<'_> {}
@@ -36,12 +28,6 @@ impl Stdin for io::Stdin {
     }
 }
 
-impl StdinSource for super::OSProcess {
-    fn stdin(&self) -> Box<dyn Stdin> {
-        Box::new(io::stdin())
-    }
-}
-
 // ----------------------- test support for stdin ------------------
 
 struct TestStdinLock<'a> {
@@ -67,7 +53,7 @@ impl BufRead for TestStdinLock<'_> {
 
 pub(crate) type TestStdinInner = Arc<Mutex<Cursor<String>>>;
 
-struct TestStdin(TestStdinInner);
+pub(super) struct TestStdin(pub(super) TestStdinInner);
 
 impl Stdin for TestStdin {
     fn lock(&self) -> Box<dyn StdinLock + '_> {
@@ -80,13 +66,6 @@ impl Stdin for TestStdin {
     }
 }
 
-#[cfg(feature = "test")]
-impl StdinSource for super::TestProcess {
-    fn stdin(&self) -> Box<dyn Stdin> {
-        Box::new(TestStdin(self.stdin.clone()))
-    }
-}
-
 // -------------- stdout -------------------------------
 
 /// This is a stand-in for [`std::io::StdoutLock`] and [`std::io::StderrLock`].