blob: b6eb610f21f3bf1a1af09fd4a31e89090bd8615f (
plain)
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
|
use std::time::Duration;
use regex::Regex;
use tokio::{
io::{AsyncBufReadExt, BufReader},
time::timeout,
};
const OBTAIN_PTY_TIMEOUT: Duration = Duration::from_secs(5);
pub struct NoPty;
/// Extract pty path from stdout
pub async fn find_pty(
re: Regex,
process: &mut tokio::process::Child,
log_level: log::Level,
log_prefix: &str,
) -> Result<String, NoPty> {
let stdout = process.stdout.take().unwrap();
let stdout_reader = BufReader::new(stdout);
let (pty_path, reader) = timeout(OBTAIN_PTY_TIMEOUT, async {
let mut lines = stdout_reader.lines();
while let Ok(Some(line)) = lines.next_line().await {
log::log!(log_level, "{log_prefix}{line}");
if let Some(path) = re.captures(&line).and_then(|cap| cap.get(1)) {
return Ok((path.as_str().to_owned(), lines.into_inner()));
}
}
Err(NoPty)
})
.await
.map_err(|_| NoPty)??;
process.stdout.replace(reader.into_inner());
Ok(pty_path)
}
|
