sim2h 0.0.52-alpha2

use crate::websocket::{streams::WebsocketStreamState, BaseStream, TransportResult};
use holochain_tracing_macros::newrelic_autotrace;

/// Represents an individual connection
#[derive(Debug)]
pub struct WssInfo<T: std::io::Read + std::io::Write + std::fmt::Debug> {
    pub(in crate::websocket) request_id: String,
    pub(in crate::websocket) url: url::Url,
    pub(in crate::websocket) last_msg: std::time::Instant,
    pub(in crate::websocket) stateful_socket: WebsocketStreamState<T>,
}

#[newrelic_autotrace(SIM2H)]
impl<T: std::io::Read + std::io::Write + std::fmt::Debug> WssInfo<T> {
    pub fn close(&mut self) -> TransportResult<()> {
        if let WebsocketStreamState::ReadyWss(socket) = &mut self.stateful_socket {
            socket.write_message(tungstenite::Message::Close(None))?;
            socket.close(None)?;
            socket.write_pending()?;
        }
        self.stateful_socket = WebsocketStreamState::None;
        Ok(())
    }

    pub fn new(url: url::Url, socket: BaseStream<T>, is_server: bool) -> Self {
        #[allow(clippy::match_bool)]
        WssInfo {
            // TODO set a request id
            request_id: "".to_string(),
            url,
            last_msg: std::time::Instant::now(),
            stateful_socket: match is_server {
                false => WebsocketStreamState::Connecting(socket),
                true => WebsocketStreamState::ConnectingSrv(socket),
            },
        }
    }

    pub fn client(url: url::Url, socket: BaseStream<T>) -> Self {
        Self::new(url, socket, false)
    }

    pub fn server(url: url::Url, socket: BaseStream<T>) -> Self {
        Self::new(url, socket, true)
    }
}