agents-kit 0.0.2

use std::net::SocketAddr;
use std::time::Duration;

use crate::KitError;
use crate::error::{DeadlineKind, ErrorDetail, ErrorPhase};

use super::jsonrpc::{CodexRuntimeBuilder, CodexTransportRuntime};

pub(crate) struct CodexTcp;

impl CodexTcp {
    pub(crate) async fn connect(
        address: SocketAddr,
        timeout: Duration,
    ) -> Result<CodexTransportRuntime, KitError> {
        let endpoint = format!("ws://{address}/");
        let (stream, _) = tokio::time::timeout(
            timeout,
            tokio_tungstenite::connect_async_with_config(
                endpoint.as_str(),
                Some(CodexRuntimeBuilder::websocket_config()),
                false,
            ),
        )
        .await
        .map_err(|_| {
            KitError::timeout("TCP connection deadline elapsed")
                .in_phase(ErrorPhase::ClientSetup)
                .with_detail(ErrorDetail::DeadlineElapsed {
                    kind: DeadlineKind::Lifecycle,
                    timeout: Some(timeout),
                })
        })?
        .map_err(|_| {
            KitError::integration("TCP connection failed")
                .in_phase(ErrorPhase::ClientSetup)
                .with_detail(ErrorDetail::TransportFailure)
        })?;
        Ok(CodexRuntimeBuilder::websocket(endpoint, stream))
    }
}