forge-core-utils 0.8.7-rc.26

use std::{future::Future, sync::OnceLock};

use tokio::runtime::{Builder, Handle, Runtime, RuntimeFlavor};

fn rt() -> &'static Runtime {
    static RT: OnceLock<Runtime> = OnceLock::new();
    RT.get_or_init(|| {
        Builder::new_multi_thread()
            .enable_all()
            .build()
            .expect("failed to build global Tokio runtime")
    })
}

/// Run an async future from sync code safely.
/// If already inside a Tokio runtime, it will use that runtime.
pub fn block_on<F, T>(fut: F) -> T
where
    F: Future<Output = T> + Send,
    T: Send,
{
    match Handle::try_current() {
        // Already inside a Tokio runtime
        Ok(h) => match h.runtime_flavor() {
            // Use block_in_place so other tasks keep running.
            RuntimeFlavor::MultiThread => tokio::task::block_in_place(|| rt().block_on(fut)),
            // Spawn a new thread to avoid freezing a single-thread runtime.
            RuntimeFlavor::CurrentThread | _ => std::thread::scope(|s| {
                s.spawn(|| rt().block_on(fut))
                    .join()
                    .expect("thread panicked")
            }),
        },
        // Outside Tokio: block normally.
        Err(_) => rt().block_on(fut),
    }
}