Function copy_ws_to_writer

crates/openshell-server/src/ws_tunnel.rs:127–159

Copy bytes from WebSocket binary/text frames into an async writer.

(
    mut ws_source: futures::stream::SplitStream<axum::extract::ws::WebSocket>,
    mut writer: W,
)

/// Copy bytes from WebSocket binary/text frames into an async writer.
async fn copy_ws_to_writer<W>(
    mut ws_source: futures::stream::SplitStream<axum::extract::ws::WebSocket>,
    mut writer: W,
) -> Result<(), Box<dyn std::error::Error + Send + Sync>>
where
    W: AsyncWrite + Unpin,
{
    while let Some(msg) = ws_source.next().await {
        match msg {
            Ok(Message::Binary(data)) => {
                if writer.write_all(&data).await.is_err() {
                    break;
                }
            }
            Ok(Message::Text(text)) => {
                // Some proxies send text frames — treat as binary.
                if writer.write_all(text.as_bytes()).await.is_err() {
                    break;
                }
            }
            Ok(Message::Close(_)) => break,
            Ok(Message::Ping(_) | Message::Pong(_)) => {
                // Handled automatically by axum's WebSocket.
            }
            Err(e) => {
                debug!(error = %e, "WS tunnel: ws read error");
                break;
            }
        }
    }
    let _ = writer.shutdown().await;
    Ok(())
}