Function start_tunnel_proxy

crates/openshell-cli/src/edge_tunnel.rs:62–95

Start the local tunnel proxy. Returns an [`EdgeTunnelProxy`] with the local address to connect to. The proxy runs as a background tokio task.

(
    gateway_endpoint: &str,
    edge_token: &str,
)

/// Returns an [`EdgeTunnelProxy`] with the local address to connect to.
/// The proxy runs as a background tokio task.
pub async fn start_tunnel_proxy(
    gateway_endpoint: &str,
    edge_token: &str,
) -> Result<EdgeTunnelProxy> {
    let listener = TcpListener::bind("127.0.0.1:0").await.into_diagnostic()?;
    let local_addr = listener.local_addr().into_diagnostic()?;

    // Convert the gateway endpoint to a WebSocket URL.
    // https://foo.com -> wss://foo.com
    // http://foo.com  -> ws://foo.com  (edge proxy over plain HTTP, uncommon but handled)
    let ws_url = format!(
        "{}/_ws_tunnel",
        gateway_endpoint
            .replacen("https://", "wss://", 1)
            .replacen("http://", "ws://", 1)
            .trim_end_matches('/')
    );

    let config = Arc::new(TunnelConfig {
        ws_url,
        edge_token: edge_token.to_string(),
    });

    debug!(
        local_addr = %local_addr,
        gateway = %gateway_endpoint,
        "starting edge tunnel proxy"
    );

    // Spawn the accept loop.
    tokio::spawn(accept_loop(listener, config));

    Ok(EdgeTunnelProxy { local_addr })
}

/// Accept loop: for each incoming TCP connection, spawn a handler that
/// opens a WebSocket to the edge and pipes bytes bidirectionally.