detect-targets 0.1.85

use crate::detect_targets;

use std::sync::Arc;

use tokio::sync::SetOnce;

#[derive(Debug)]
enum DesiredTargetsInner {
    AutoDetect(Arc<SetOnce<Vec<String>>>),
    Initialized(Vec<String>),
}

#[derive(Debug)]
pub struct DesiredTargets(DesiredTargetsInner);

impl DesiredTargets {
    fn initialized(targets: Vec<String>) -> Self {
        Self(DesiredTargetsInner::Initialized(targets))
    }

    fn auto_detect() -> Self {
        let arc = Arc::new(SetOnce::new());

        let set_once = arc.clone();
        tokio::spawn(async move {
            set_once.set(detect_targets().await).unwrap();
        });

        Self(DesiredTargetsInner::AutoDetect(arc))
    }

    pub async fn get(&self) -> &[String] {
        use DesiredTargetsInner::*;

        match &self.0 {
            Initialized(targets) => targets,
            AutoDetect(set_once) => set_once.wait().await,
        }
    }

    /// If `DesiredTargets` is provided with a list of desired targets instead
    /// of detecting the targets, then this function would return `Some`.
    pub fn get_initialized(&self) -> Option<&[String]> {
        use DesiredTargetsInner::*;

        match &self.0 {
            Initialized(targets) => Some(targets),
            AutoDetect(..) => None,
        }
    }
}

/// If opts_targets is `Some`, then it will be used.
/// Otherwise, call `detect_targets` using `tokio::spawn` to detect targets.
///
/// Since `detect_targets` internally spawns a process and wait for it,
/// it's pretty costy, it is recommended to run this fn ASAP and
/// reuse the result.
pub fn get_desired_targets(opts_targets: Option<Vec<String>>) -> DesiredTargets {
    if let Some(targets) = opts_targets {
        DesiredTargets::initialized(targets)
    } else {
        DesiredTargets::auto_detect()
    }
}