Function discover_targets

packages/server/src/webkit.rs:102–129

(
    udid: &str,
    http_origin: Option<&str>,
)

}

pub async fn discover_targets(
    udid: &str,
    http_origin: Option<&str>,
) -> Result<WebKitTargetDiscovery, AppError> {
    let monitor = webkit_discovery_monitor(udid);
    monitor.clone().ensure_started();

    let deadline = Instant::now() + WEBKIT_DISCOVERY_TIMEOUT;
    loop {
        let discovery = monitor.discovery(http_origin).await;
        if !discovery.targets.is_empty() || Instant::now() >= deadline {
            return Ok(discovery);
        }

        let Some(remaining) = deadline.checked_duration_since(Instant::now()) else {
            return Ok(discovery);
        };
        if timeout(
            remaining.min(Duration::from_millis(250)),
            monitor.notify.notified(),
        )
        .await
        .is_err()
        {
            continue;
        }
    }
}

fn webkit_discovery_monitor(udid: &str) -> Arc<WebKitDiscoveryMonitor> {
    let monitors = WEBKIT_DISCOVERY_MONITORS.get_or_init(|| Mutex::new(HashMap::new()));