remowt-client 0.1.9

russh-based client connection to a remowt agent
Documentation 
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

use anyhow::{anyhow, Result};
use bifrostlink::declarative::RemoteEndpoints as _;
use bifrostlink::Remote;
use remowt_endpoints::pty::{PtyClient, ShellId};
use remowt_link_shared::{Address, BifConfig};

use crate::forwarded::RemowtStream;
use crate::Remowt;

pub struct RemowtShell {
	pub id: ShellId,
	pub stream: RemowtStream,
	remote: Remote<BifConfig>,
}
impl RemowtShell {
	pub fn resizer(&self) -> RemowtShellResizer {
		RemowtShellResizer {
			remote: self.remote.clone(),
			id: self.id,
		}
	}
}

#[derive(Clone)]
pub struct RemowtShellResizer {
	remote: Remote<BifConfig>,
	id: ShellId,
}

impl RemowtShellResizer {
	pub async fn resize(&self, cols: u16, rows: u16) -> Result<()> {
		PtyClient::wrap(self.remote.clone())
			.resize(self.id, cols, rows)
			.await?
			.map_err(|e| anyhow!("failed to resize remote shell: {e}"))
	}
}

impl Remowt {
	pub async fn open_shell(
		&self,
		term: &str,
		cols: u16,
		rows: u16,
		escalate: bool,
	) -> Result<RemowtShell> {
		let (forwarded, path) = self.bind_runtime_unix("shell").await?;
		let client: PtyClient<BifConfig> = if escalate {
			self.run0_endpoints().await?
		} else {
			self.endpoints()
		};
		let id = client
			.open_shell(path, term.to_owned(), cols, rows)
			.await?
			.map_err(|e| anyhow!("agent failed to open shell: {e}"))?;
		let stream = forwarded.accept().await?;

		Ok(RemowtShell {
			id,
			stream,
			remote: self.0.rpc.remote(Address::Agent),
		})
	}
}