1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
use snafu::futures01::StreamExt as _;
use snafu::ResultExt as _;
#[derive(Debug, snafu::Snafu)]
pub enum Error {
#[snafu(display("failed to parse command line '{}': {}", line, source))]
Parser {
line: String,
source: crate::parser::Error,
},
#[snafu(display("failed to find command `{}`: {}", cmd, source))]
Command {
cmd: String,
source: tokio_pty_process_stream::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: tokio_pty_process_stream::Error,
},
}
#[allow(dead_code)]
pub type Result<T> = std::result::Result<T, Error>;
pub fn eval(line: &str) -> Eval {
Eval::new(line)
}
pub struct Eval {
line: String,
stream: Option<
Box<
dyn futures::stream::Stream<
Item = tokio_pty_process_stream::Event,
Error = Error,
> + Send,
>,
>,
manage_screen: bool,
}
impl Eval {
pub fn new(line: &str) -> Self {
Self {
line: line.to_string(),
stream: None,
manage_screen: true,
}
}
pub fn set_raw(mut self, raw: bool) -> Self {
self.manage_screen = raw;
self
}
}
#[must_use = "streams do nothing unless polled"]
impl futures::stream::Stream for Eval {
type Item = tokio_pty_process_stream::Event;
type Error = Error;
fn poll(&mut self) -> futures::Poll<Option<Self::Item>, Self::Error> {
if self.stream.is_none() {
let line = self.line.as_ref();
let (cmd, args) =
crate::parser::parse(line).context(Parser { line })?;
let builtin_stream = crate::builtins::Builtin::new(&cmd, &args);
let stream: Box<
dyn futures::stream::Stream<
Item = tokio_pty_process_stream::Event,
Error = Error,
> + Send,
> = if let Ok(s) = builtin_stream {
Box::new(s.context(BuiltinExecution { cmd }))
} else {
let input = crate::async_stdin::Stdin::new();
let process = tokio_pty_process_stream::ResizingProcess::new(
tokio_pty_process_stream::Process::new(
&cmd, &args, input,
),
);
Box::new(process.context(ProcessExecution { cmd }))
};
self.stream = Some(stream);
}
if let Some(ref mut stream) = &mut self.stream {
stream.poll()
} else {
unreachable!()
}
}
}
|
