Function prepare_webkit_target_for_attach

packages/server/src/webkit.rs:578–675

(
    inspector_reader: &mut R,
    inspector_writer: &mut W,
    connection_id: &str,
    app_id: &str,
    page_id: u64,
)

}

async fn prepare_webkit_target_for_attach<R, W>(
    inspector_reader: &mut R,
    inspector_writer: &mut W,
    connection_id: &str,
    app_id: &str,
    page_id: u64,
) -> Result<(), AppError>
where
    R: AsyncRead + Unpin,
    W: AsyncWrite + Unpin,
{
    send_forward_get_listing(inspector_writer, connection_id, app_id).await?;
    let deadline = Instant::now() + WEBKIT_TARGET_ATTACH_TIMEOUT;
    let mut released_connections = HashSet::new();

    loop {
        let Some(remaining) = deadline.checked_duration_since(Instant::now()) else {
            return Err(AppError::native(format!(
                "Timed out preparing WebKit target {app_id}/{page_id} for inspection."
            )));
        };
        let packet = timeout(
            remaining.min(Duration::from_millis(500)),
            read_packet(inspector_reader),
        )
        .await;
        let packet = match packet {
            Ok(Ok(packet)) => packet,
            Ok(Err(error)) => return Err(error),
            Err(_) => continue,
        };

        let message = parse_rpc_message(&packet)?;
        match message.selector.as_str() {
            "_rpc_applicationSentListing:" => {
                if string_value(&message.args, "WIRApplicationIdentifierKey").as_deref()
                    != Some(app_id)
                {
                    continue;
                }

                let Some(page) = parse_page_listing(&message.args)
                    .into_iter()
                    .find(|page| page.page_id == page_id)
                else {
                    return Err(AppError::not_found(format!(
                        "WebKit target {app_id}/{page_id} is no longer available."
                    )));
                };

                let Some(existing_connection_id) = page.connection_id else {
                    return Ok(());
                };
                if existing_connection_id == connection_id {
                    return Ok(());
                }
                if !released_connections.insert(existing_connection_id.clone()) {
                    send_forward_get_listing(inspector_writer, connection_id, app_id).await?;