hk 1.44.2

A tool for managing git hooks
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use std::collections::HashMap;

use crate::Result;
use tokio::sync::watch;

pub struct StepDepends {
    depends: HashMap<String, (watch::Sender<bool>, watch::Receiver<bool>)>,
}

impl StepDepends {
    pub fn new(names: &[&str]) -> Self {
        StepDepends {
            depends: names
                .iter()
                .map(|name| (name.to_string(), watch::channel(false)))
                .collect(),
        }
    }

    pub fn is_done(&self, step: &str) -> bool {
        let Some((_tx, rx)) = self.depends.get(step) else {
            return true;
        };
        *rx.clone().borrow_and_update()
    }

    pub async fn wait_for(&self, step: &str) -> Result<()> {
        let Some((_tx, rx)) = self.depends.get(step) else {
            return Ok(());
        };
        let mut rx = rx.clone();
        while !*rx.borrow_and_update() {
            rx.changed().await?;
        }
        Ok(())
    }

    pub fn mark_done(&self, step: &str) -> Result<()> {
        let (tx, _rx) = self.depends.get(step).unwrap();
        tx.send(true)?;
        Ok(())
    }
}