cloud_terrastodon_azure_devops 0.34.0

Helpers for interacting with Azure DevOps for the Cloud Terrastodon project
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use cloud_terrastodon_azure_devops_types::prelude::AzureDevOpsOrganizationUrl;
use cloud_terrastodon_azure_devops_types::prelude::AzureDevOpsUserLicenseEntitlement;
use cloud_terrastodon_command::CacheKey;
use cloud_terrastodon_command::CommandBuilder;
use cloud_terrastodon_command::CommandKind;
use cloud_terrastodon_command::async_trait;
use serde::Deserialize;
use serde_json::Value;
use std::path::PathBuf;
use tracing::debug;

pub struct AzureDevOpsUserLicenseEntitlementsRequest<'a> {
    pub org_url: &'a AzureDevOpsOrganizationUrl,
}

pub fn fetch_azure_devops_user_license_entitlements<'a>(
    org_url: &'a AzureDevOpsOrganizationUrl,
) -> AzureDevOpsUserLicenseEntitlementsRequest<'a> {
    AzureDevOpsUserLicenseEntitlementsRequest { org_url }
}

#[async_trait]
impl<'a> cloud_terrastodon_command::CacheableCommand
    for AzureDevOpsUserLicenseEntitlementsRequest<'a>
{
    type Output = Vec<AzureDevOpsUserLicenseEntitlement>;

    fn cache_key(&self) -> CacheKey {
        CacheKey::new(PathBuf::from_iter([
            "az",
            "devops",
            self.org_url.organization_name.as_ref(),
            "license",
            "entitlement",
            "list",
        ]))
    }

    async fn run(self) -> eyre::Result<Self::Output> {
        debug!("Fetching Azure DevOps user entitlements");
        let mut cmd = CommandBuilder::new(CommandKind::AzureCLI);
        cmd.args(["devops", "invoke"]);
        let org = self.org_url.to_string();
        cmd.args(["--organization", org.as_str()]);
        cmd.args(["--area", "licensing"]);
        cmd.args(["--resource", "entitlements"]);
        cmd.args(["--api-version", "7.2-preview"]);
        cmd.args(["--encoding", "utf-8"]);
        cmd.cache(self.cache_key());

        #[derive(Deserialize)]
        struct InvokeResponse {
            continuation_token: Option<Value>,
            count: u32,
            value: Vec<AzureDevOpsUserLicenseEntitlement>,
        }

        let resp = cmd.run::<InvokeResponse>().await?;
        let entitlements = resp.value;

        debug!("Found {} Azure DevOps user entitlements", resp.count);

        if resp.continuation_token.is_some() {
            todo!("Add support for continuation token...");
        }

        Ok(entitlements)
    }
}

cloud_terrastodon_command::impl_cacheable_into_future!(AzureDevOpsUserLicenseEntitlementsRequest<'a>, 'a);

#[cfg(test)]
mod test {
    use super::*;
    use crate::prelude::get_default_organization_url;

    #[tokio::test]
    pub async fn it_works() -> eyre::Result<()> {
        let org_url = get_default_organization_url().await?;
        let entitlements = fetch_azure_devops_user_license_entitlements(&org_url).await?;
        println!("Found {} user entitlements", entitlements.len());
        for entitlement in entitlements.iter().take(5) {
            println!(
                "User: {} ({}) - License: {:?} - Status: {:?}",
                entitlement.user.display_name,
                entitlement.user.unique_name,
                entitlement.license,
                entitlement.status
            );
        }

        Ok(())
    }
}