diff options
Diffstat (limited to 'src/eval.rs')
-rw-r--r-- | src/eval.rs | 85 |
1 files changed, 85 insertions, 0 deletions
diff --git a/src/eval.rs b/src/eval.rs new file mode 100644 index 0000000..a46045a --- /dev/null +++ b/src/eval.rs @@ -0,0 +1,85 @@ +use futures::stream::Stream; +use snafu::{ResultExt, Snafu}; + +#[derive(Debug, Snafu)] +pub enum Error { + #[snafu(display("failed to parse command line '{}': {}", line, source))] + ParserError { + line: String, + source: crate::parser::Error, + }, + + #[snafu(display("failed to find command `{}`: {}", cmd, source))] + CommandError { + cmd: String, + source: crate::process::Error, + }, + + #[snafu(display("failed to run builtin command `{}`: {}", cmd, source))] + BuiltinExecution { + cmd: String, + source: crate::builtins::Error, + }, + + #[snafu(display("failed to run executable `{}`: {}", cmd, source))] + ProcessExecution { + cmd: String, + source: crate::process::Error, + }, +} + +pub fn eval(line: &str) -> Result<Eval, Error> { + Eval::new(line) +} + +pub enum CommandEvent { + Output(Vec<u8>), + ProcessExit(std::process::ExitStatus), + BuiltinExit, +} + +pub struct Eval { + stream: Box< + dyn futures::stream::Stream<Item = CommandEvent, Error = Error> + + Send, + >, +} + +impl Eval { + fn new(line: &str) -> Result<Self, Error> { + let (cmd, args) = + crate::parser::parse(line).context(ParserError { line })?; + let builtin_stream = crate::builtins::exec(&cmd, &args); + let stream: Box< + dyn futures::stream::Stream<Item = CommandEvent, Error = Error> + + Send, + > = if let Ok(s) = builtin_stream { + Box::new(s.map_err(move |e| Error::BuiltinExecution { + cmd: cmd.clone(), + source: e, + })) + } else { + let process_stream = crate::process::spawn(&cmd, &args); + match process_stream { + Ok(s) => { + Box::new(s.map_err(move |e| Error::ProcessExecution { + cmd: cmd.clone(), + source: e, + })) + } + Err(e) => return Err(e).context(CommandError { cmd }), + } + }; + Ok(Eval { stream }) + } +} + +#[must_use = "streams do nothing unless polled"] +impl futures::stream::Stream for Eval { + type Item = CommandEvent; + type Error = Error; + + fn poll(&mut self) -> futures::Poll<Option<Self::Item>, Self::Error> { + self.stream.poll() + } +} |