cloud_terrastodon_entrypoint 0.35.0

Main entrypoint for the Cloud Terrastodon 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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use cloud_terrastodon_azure::AzureTenantId;
use cloud_terrastodon_azure::fetch_all_entra_users;
use cloud_terrastodon_azure::fetch_all_policy_assignments;
use cloud_terrastodon_azure::fetch_all_policy_definitions;
use cloud_terrastodon_azure::fetch_all_policy_set_definitions;
use cloud_terrastodon_azure::fetch_all_resource_groups;
use cloud_terrastodon_azure::fetch_all_role_assignments;
use eyre::Result;
use indicatif::ProgressBar;
use tokio::task::JoinSet;
pub async fn populate_cache(tenant_id: AzureTenantId) -> Result<()> {
    let mut work: JoinSet<(&str, bool)> = JoinSet::new();
    work.spawn(async move {
        (
            "fetch_all_policy_assignments",
            fetch_all_policy_assignments(tenant_id).await.is_ok(),
        )
    });
    work.spawn(async move {
        (
            "fetch_all_policy_definitions",
            fetch_all_policy_definitions(tenant_id).await.is_ok(),
        )
    });
    work.spawn(async move {
        (
            "fetch_all_policy_set_definitions",
            fetch_all_policy_set_definitions(tenant_id).await.is_ok(),
        )
    });
    work.spawn(async move {
        (
            "fetch_all_resource_groups",
            fetch_all_resource_groups(tenant_id).await.is_ok(),
        )
    });
    work.spawn(async move {
        (
            "fetch_all_role_assignments",
            fetch_all_role_assignments(tenant_id).await.is_ok(),
        )
    });
    work.spawn(async move {
        (
            "fetch_all_users",
            fetch_all_entra_users(tenant_id).await.is_ok(),
        )
    });
    let pb = ProgressBar::new(work.len() as u64);
    // pb.set_style(
    //     ProgressStyle::default_bar()
    //         .template("[{wide_bar} {pos}/{len} {msg}")?,
    // );
    pb.tick();
    while let Some(x) = work.join_next().await {
        let (operation, success) = x?;
        pb.inc(1);
        let msg = format!(
            "{} {}",
            operation,
            if success { "succeeded" } else { "failed" }
        );
        // pb.set_message(msg);
        pb.println(msg);
    }
    pb.finish_with_message("Cache population complete");
    Ok(())
}

#[cfg(test)]
mod test {
    use super::*;
    use cloud_terrastodon_azure::get_test_tenant_id;

    #[test_log::test(tokio::test)]
    #[ignore]
    async fn it_works() -> Result<()> {
        populate_cache(get_test_tenant_id().await?).await?;
        Ok(())
    }
}