Function with_retry_and_hook

crates/opencode-provider/src/retry.rs:182–221

Same as [`with_retry`] but calls `hook(attempt, &error, delay_ms)` before each retry sleep, giving the caller a chance to log or update UI.

(
    config: &RetryConfig,
    mut f: F,
    mut hook: H,
)

/// Same as [`with_retry`] but calls `hook(attempt, &error, delay_ms)` before
/// each retry sleep, giving the caller a chance to log or update UI.
pub async fn with_retry_and_hook<F, Fut, T, E, H>(
    config: &RetryConfig,
    mut f: F,
    mut hook: H,
) -> Result<T, E>
where
    F: FnMut() -> Fut,
    Fut: Future<Output = Result<T, E>>,
    E: IsRetryable + std::fmt::Debug,
    H: FnMut(u32, &E, u64),
{
    let mut attempt: u32 = 0;

    loop {
        attempt += 1;
        match f().await {
            Ok(val) => return Ok(val),
            Err(e) => {
                if attempt >= config.max_attempts {
                    return Err(e);
                }
                match e.is_retryable() {
                    Some(msg) => {
                        let delay_ms = delay(attempt, None);
                        hook(attempt, &e, delay_ms);
                        warn!(
                            attempt,
                            max = config.max_attempts,
                            delay_ms,
                            reason = %msg,
                            "retrying after transient error"
                        );
                        tokio::time::sleep(std::time::Duration::from_millis(delay_ms)).await;
                    }
                    None => return Err(e),
                }
            }
        }
    }
}

// ---------------------------------------------------------------------------
// Tests