repokit 5.0.0

A knowledgebase for your repository - wrapped in a CLI
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

use std::{
    process::exit,
    sync::{LazyLock, Mutex, MutexGuard},
};

pub struct PostProcessor {
    tasks: Vec<Box<dyn Fn() + Send + 'static>>,
}

impl PostProcessor {
    pub fn new() -> Self {
        PostProcessor { tasks: Vec::new() }
    }

    pub fn get() -> MutexGuard<'static, PostProcessor> {
        REPOKIT_POST_PROCESSOR.lock().unwrap()
    }

    pub fn register_task<F>(&mut self, task: F)
    where
        F: Fn() + Send + 'static,
    {
        self.tasks.push(Box::new(task));
    }

    pub fn flush(&mut self) {
        for task in self.tasks.iter() {
            task();
        }
        self.tasks.clear();
        self.tasks.shrink_to_fit();
        exit(0);
    }
}

static REPOKIT_POST_PROCESSOR: LazyLock<Mutex<PostProcessor>> =
    LazyLock::new(|| Mutex::new(PostProcessor::new()));