remowt-client 0.1.1

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

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 uuid::Uuid;

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) -> Result<RemowtShell> {
		let sock = self
			.runtime_dir()
			.join(format!("remowt-shell-{}.sock", Uuid::new_v4()));

		let forwarded = self.forward_socket(&sock).await?;
		let client: PtyClient<BifConfig> = self.endpoints();
		let id = client
			.open_shell(sock, 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.rpc.remote(Address::Agent),
		})
	}
}