mirrors/krata
1
0
Fork 0
mirror of https://github.com/edera-dev/krata.git synced 2025-08-03 13:11:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

krata: implement auto-exit handling

Browse Source
This commit is contained in:
Alex Zenla
2024-03-13 11:34:52 +00:00
parent 2ec619c0c3
commit 13bea70c0d
3 changed files with 83 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
crates/kratactl/bin/control.rs
View File

@ -52,6 +52,10 @@ async fn main() -> Result<()> {
let args = ControllerArgs::parse();
let mut client = ControlClientProvider::dial(args.connection.parse()?).await?;
let events = client
.watch_events(WatchEventsRequest {})
.await?
.into_inner();
match args.command {
Commands::Launch {
@ -82,9 +86,12 @@ async fn main() -> Result<()> {
};
println!("launched guest: {}", guest.id);
if attach {
let input = StdioConsoleStream::stdin_stream(guest.id).await;
let input = StdioConsoleStream::stdin_stream(guest.id.clone()).await;
let output = client.console_data(input).await?.into_inner();
let exit_hook_task =
StdioConsoleStream::guest_exit_hook(guest.id.clone(), events).await?;
StdioConsoleStream::stdout(output).await?;
exit_hook_task.abort();
}
}
@ -99,9 +106,11 @@ async fn main() -> Result<()> {
}
Commands::Console { guest } => {
let input = StdioConsoleStream::stdin_stream(guest).await;
let input = StdioConsoleStream::stdin_stream(guest.clone()).await;
let output = client.console_data(input).await?.into_inner();
let exit_hook_task = StdioConsoleStream::guest_exit_hook(guest.clone(), events).await?;
StdioConsoleStream::stdout(output).await?;
exit_hook_task.abort();
}
Commands::List { .. } => {