mirror of
https://github.com/wez/wezterm.git
synced 2024-11-27 02:25:28 +03:00
e6421d1b72
This breaking API change allows us to explicitly generate EOF when the taken writer is dropped. The examples have been updated to show how to manage read, write and waiting without deadlock for both linux and windows. Need to confirm that this is still good on macOS, but my confidence is high. I've also removed ssh2 support from this crate as part of this change. We haven't used it directly in wezterm in a long while and removing it from here means that there is slightly less code to keep compiling over and over. refs: https://github.com/wez/wezterm/discussions/2392 refs: https://github.com/wez/wezterm/issues/1396
74 lines
2.5 KiB
Rust
74 lines
2.5 KiB
Rust
use anyhow::anyhow;
|
|
use futures::prelude::*;
|
|
use portable_pty::{native_pty_system, CommandBuilder, PtySize};
|
|
|
|
// This example shows how to use the `smol` crate to use portable_pty
|
|
// in an asynchronous application.
|
|
|
|
fn main() -> anyhow::Result<()> {
|
|
smol::block_on(async {
|
|
let pty_system = native_pty_system();
|
|
|
|
let pair = pty_system.openpty(PtySize {
|
|
rows: 24,
|
|
cols: 80,
|
|
pixel_width: 0,
|
|
pixel_height: 0,
|
|
})?;
|
|
|
|
let cmd = CommandBuilder::new("whoami");
|
|
|
|
// Move the slave to another thread to block and spawn a
|
|
// command.
|
|
// Note that this implicitly drops slave and closes out
|
|
// file handles which is important to avoid deadlock
|
|
// when waiting for the child process!
|
|
let slave = pair.slave;
|
|
let mut child = smol::unblock(move || slave.spawn_command(cmd)).await?;
|
|
|
|
{
|
|
// Obtain the writer.
|
|
// When the writer is dropped, EOF will be sent to
|
|
// the program that was spawned.
|
|
// It is important to take the writer even if you don't
|
|
// send anything to its stdin so that EOF can be
|
|
// generated, otherwise you risk deadlocking yourself.
|
|
let writer = pair.master.take_writer()?;
|
|
|
|
// Explicitly generate EOF
|
|
drop(writer);
|
|
}
|
|
|
|
println!(
|
|
"child status: {:?}",
|
|
smol::unblock(move || child
|
|
.wait()
|
|
.map_err(|e| anyhow!("waiting for child: {}", e)))
|
|
.await?
|
|
);
|
|
|
|
let reader = pair.master.try_clone_reader()?;
|
|
|
|
// Take care to drop the master after our processes are
|
|
// done, as some platforms get unhappy if it is dropped
|
|
// sooner than that.
|
|
drop(pair.master);
|
|
|
|
let mut lines = smol::io::BufReader::new(smol::Unblock::new(reader)).lines();
|
|
while let Some(line) = lines.next().await {
|
|
let line = line.map_err(|e| anyhow!("problem reading line: {}", e))?;
|
|
// We print with escapes escaped because the windows conpty
|
|
// implementation synthesizes title change escape sequences
|
|
// in the output stream and it can be confusing to see those
|
|
// printed out raw in another terminal.
|
|
print!("output: len={} ", line.len());
|
|
for c in line.escape_debug() {
|
|
print!("{}", c);
|
|
}
|
|
println!();
|
|
}
|
|
|
|
Ok(())
|
|
})
|
|
}
|