webterm-agent 0.2.5

Lightweight agent component for Webterm that enables secure browser-based terminal access
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

use crate::models::agent_error::AgentError;
use crate::models::connection_manager::ConnectionManager;
use crate::models::relay::Relay;
use crate::models::socket_reader::{SocketReader, SocketSubscriber};
use crate::models::socket_writer::{SocketPublisher, SocketWriter};
use futures::StreamExt;
use std::sync::Arc;
use tokio_tungstenite::connect_async;
use tracing::debug;

pub struct RelayConnection {
    writer: SocketWriter,
    reader: SocketReader,
}

impl RelayConnection {
    pub async fn new(
        relay: Arc<Relay>,
        nonce: String,
        cm: Arc<ConnectionManager>,
    ) -> Result<Self, AgentError> {
        debug!("Connecting to relay: {}", relay.websocket_url(None));

        let socket = connect_async(relay.websocket_url(Some(nonce))).await?;
        let (socket, _) = socket;
        let (relay_writer, relay_reader) = socket.split();

        Ok(Self {
            writer: SocketWriter::new(relay_writer, cm.clone()),
            reader: SocketReader::new(relay_reader, cm),
        })
    }

    pub async fn publisher(&self) -> SocketPublisher {
        self.writer.publisher()
    }

    pub async fn subscriber(&self) -> SocketSubscriber {
        self.reader.subscriber()
    }
}