Implement Pipeline.find_command

guilload · Jun 27, 2017 · 4fcd86e · 4fcd86e
1 parent a28e973
commit 4fcd86e
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/pipeline.rs b/src/pipeline.rs
@@ -55,17 +55,25 @@ impl Pipeline {
         Ok(())
     }
 
+    pub fn find_command(&mut self, pid: libc::pid_t) -> Option<&mut Command> {
+        self.commands.iter_mut().find(|c| c.pid == pid)
+    }
+
     pub fn is_completed(&self) -> bool {
         self.commands.iter().all(|c| c.is_completed())
     }
 
+    pub fn is_suspended(&self) -> bool {
+        self.commands.iter().all(|c| c.is_suspended())
+    }
+
     pub fn wait(&mut self) -> Result<(), nix::Error> {
-        while !self.is_completed() { // FIXME
+        while !self.is_completed() {
             let status = waitpid(-1 * self.pgid, Some(WUNTRACED))?;
 
             match status {
-                WaitStatus::Exited(pid, _) => self.commands.iter_mut().find(|c| c.pid == pid).unwrap().status(Status::Completed),  // FIXME
-                WaitStatus::Stopped(pid, _) => self.commands.iter_mut().find(|c| c.pid == pid).unwrap().status(Status::Suspended),
+                WaitStatus::Exited(pid, _) => self.find_command(pid).unwrap().status(Status::Completed),
+                WaitStatus::Stopped(pid, _) => self.find_command(pid).unwrap().status(Status::Suspended),
                 _ => panic!("unimplemented!"),
             };
         }