krecik/
api.rs

1use crate::*;
2use rayon::prelude::*;
3
4
5/**
6   Public library API for Krecik remote-checks functionality
7**/
8/// Return checks from path, excluding remotes
9#[instrument]
10pub fn all_checks_but_remotes() -> Vec<Check> {
11    list_all_checks_from(CHECKS_DIR)
12        .par_iter()
13        .filter_map(|check_path| {
14            trace!("check_path: {check_path}");
15            if !check_path.contains(REMOTE_CHECKS_DIR) && !check_path.contains(TESTS_DIR) {
16                // select only valid Check, just ignore any malformed ones
17                read_single_check(check_path)
18            } else {
19                None
20            }
21        })
22        .collect()
23}
24
25
26/// Return remote domain+pages checks via mapper
27#[instrument]
28pub fn all_checks_pongo_merged() -> Vec<Check> {
29    list_all_checks_from(&format!("{}/{}", CHECKS_DIR, REMOTE_CHECKS_DIR))
30        .into_iter()
31        .map(|pongo_mapper| {
32            let mapper = read_pongo_mapper(&pongo_mapper);
33            let all_pongo_checks = get_pongo_checks(&mapper.url);
34            let domain_checks = all_pongo_checks
35                .clone()
36                .into_par_iter()
37                .flat_map(|check| collect_pongo_domains(&check))
38                .collect();
39            let pongo_checks = all_pongo_checks
40                .into_par_iter()
41                .flat_map(|check| collect_pongo_hosts(&check, &mapper))
42                .collect();
43
44            Check {
45                pages: Some(pongo_checks),
46                domains: Some(domain_checks),
47                notifier: mapper.notifier,
48            }
49        })
50        .collect()
51}