Skip to content

Commit

Permalink
Merge branch 'main' into command_based_token_drop_env
Browse files Browse the repository at this point in the history
  • Loading branch information
aviramha committed Nov 19, 2022
2 parents 8a47137 + 61f2f32 commit 852fcc9
Show file tree
Hide file tree
Showing 8 changed files with 285 additions and 63 deletions.
20 changes: 20 additions & 0 deletions deny.toml
Expand Up @@ -83,6 +83,26 @@ multiple-versions = "deny"
name = "idna"
version = "0.2"

# waiting on hyper-rustls and below to bump its chain
[[bans.skip]]
name = "windows_i686_msvc"
version = "0.36"
[[bans.skip]]
name = "windows_aarch64_msvc"
version = "0.36"
[[bans.skip]]
name = "windows-sys"
version = "0.36"
[[bans.skip]]
name = "windows_i686_gnu"
version = "0.36"
[[bans.skip]]
name = "windows_x86_64_msvc"
version = "0.36"
[[bans.skip]]
name = "windows_x86_64_gnu"
version = "0.36"

[[bans.skip]]
# waiting for ahash/getrandom to bump wasi as we have two branches:
# ahash -> getrandom -> wasi old
Expand Down
6 changes: 6 additions & 0 deletions examples/Cargo.toml
Expand Up @@ -53,6 +53,7 @@ backoff = "0.4.0"
clap = { version = "4.0", default-features = false, features = ["std", "cargo", "derive"] }
edit = "0.1.3"
tokio-stream = { version = "0.1.9", features = ["net"] }
crossterm = {version = "0.25.0" }

[[example]]
name = "configmapgen_controller"
Expand Down Expand Up @@ -206,3 +207,8 @@ path = "custom_client_trace.rs"
[[example]]
name = "secret_syncer"
path = "secret_syncer.rs"

[[example]]
name = "pod_shell_crossterm"
path = "pod_shell_crossterm.rs"
required-features = ["ws"]
131 changes: 131 additions & 0 deletions examples/pod_shell_crossterm.rs
@@ -0,0 +1,131 @@
use futures::{channel::mpsc::Sender, SinkExt, StreamExt};
use k8s_openapi::api::core::v1::Pod;

use kube::{
api::{Api, AttachParams, AttachedProcess, DeleteParams, PostParams, ResourceExt, TerminalSize},
runtime::wait::{await_condition, conditions::is_pod_running},
Client,
};
#[cfg(unix)] use tokio::signal;
use tokio::{io::AsyncWriteExt, select};

#[cfg(unix)]
// Send the new terminal size to channel when it change
async fn handle_terminal_size(mut channel: Sender<TerminalSize>) -> Result<(), anyhow::Error> {
let (width, height) = crossterm::terminal::size()?;
channel.send(TerminalSize { height, width }).await?;

// create a stream to catch SIGWINCH signal
let mut sig = signal::unix::signal(signal::unix::SignalKind::window_change())?;
loop {
if sig.recv().await == None {
return Ok(());
}

let (width, height) = crossterm::terminal::size()?;
channel.send(TerminalSize { height, width }).await?;
}
}

#[cfg(windows)]
// We don't support window for terminal size change, we only send the initial size
async fn handle_terminal_size(mut channel: Sender<TerminalSize>) -> Result<(), anyhow::Error> {
let (width, height) = crossterm::terminal::size()?;
channel.send(TerminalSize { height, width }).await?;
let mut ctrl_c = tokio::signal::windows::ctrl_c()?;
ctrl_c.recv().await;
Ok(())
}


#[tokio::main]
async fn main() -> anyhow::Result<()> {
let client = Client::try_default().await?;

let pods: Api<Pod> = Api::default_namespaced(client);
let p: Pod = serde_json::from_value(serde_json::json!({
"apiVersion": "v1",
"kind": "Pod",
"metadata": { "name": "example" },
"spec": {
"containers": [{
"name": "example",
"image": "alpine",
// Do nothing
"command": ["tail", "-f", "/dev/null"],
}],
}
}))?;
// Create pod if don't exist
pods.create(&PostParams::default(), &p).await?;

// Wait until the pod is running, otherwise we get 500 error.
let running = await_condition(pods.clone(), "example", is_pod_running());
let _ = tokio::time::timeout(std::time::Duration::from_secs(15), running).await?;

{
// Here we we put the terminal in 'raw' mode to directly get the input from the user and sending it to the server and getting the result from the server to display directly.
// We also watch for change in your terminal size and send it to the server so that application that use the size work properly.
crossterm::terminal::enable_raw_mode()?;
let mut attached: AttachedProcess = pods
.exec(
"example",
vec!["sh"],
&AttachParams::default().stdin(true).tty(true).stderr(false),
)
.await?;

let mut stdin = tokio_util::io::ReaderStream::new(tokio::io::stdin());
let mut stdout = tokio::io::stdout();

let mut output = tokio_util::io::ReaderStream::new(attached.stdout().unwrap());
let mut input = attached.stdin().unwrap();

let term_tx = attached.terminal_size().unwrap();

let mut handle_terminal_size_handle = tokio::spawn(handle_terminal_size(term_tx));

loop {
select! {
message = stdin.next() => {
match message {
Some(Ok(message)) => {
input.write(&message).await?;
}
_ => {
break;
},
}
},
message = output.next() => {
match message {
Some(Ok(message)) => {
stdout.write(&message).await?;
stdout.flush().await?;
},
_ => {
break
},
}
},
result = &mut handle_terminal_size_handle => {
match result {
Ok(_) => println!("End of terminal size stream"),
Err(e) => println!("Error getting terminal size: {e:?}")
}
},
};
}
crossterm::terminal::disable_raw_mode()?;
}

// Delete it
pods.delete("example", &DeleteParams::default())
.await?
.map_left(|pdel| {
assert_eq!(pdel.name_any(), "example");
});

println!("");
Ok(())
}
2 changes: 1 addition & 1 deletion kube-client/Cargo.toml
Expand Up @@ -37,7 +37,7 @@ rustdoc-args = ["--cfg", "docsrs"]

[dependencies]
base64 = { version = "0.13.0", optional = true }
chrono = { version = "0.4.19", optional = true, default-features = false }
chrono = { version = "0.4.23", optional = true, default-features = false }
dirs = { package = "dirs-next", optional = true, version = "2.0.0" }
serde = { version = "1.0.130", features = ["derive"] }
serde_json = "1.0.68"
Expand Down
2 changes: 1 addition & 1 deletion kube-client/src/api/mod.rs
Expand Up @@ -5,7 +5,7 @@ mod core_methods;
#[cfg(feature = "ws")] mod remote_command;
use std::fmt::Debug;

#[cfg(feature = "ws")] pub use remote_command::AttachedProcess;
#[cfg(feature = "ws")] pub use remote_command::{AttachedProcess, TerminalSize};
#[cfg(feature = "ws")] mod portforward;
#[cfg(feature = "ws")] pub use portforward::Portforwarder;

Expand Down

0 comments on commit 852fcc9

Please sign in to comment.