cloud_terrastodon_azure_devops 0.35.1

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

use cloud_terrastodon_azure_devops_types::AzureDevOpsAgentPool;
use cloud_terrastodon_azure_devops_types::AzureDevOpsOrganizationUrl;
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 AzureDevOpsAgentPoolListRequest<'a> {
    pub org_url: &'a AzureDevOpsOrganizationUrl,
}

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

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

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

    async fn run(self) -> eyre::Result<Self::Output> {
        debug!("Fetching Azure DevOps agent pools");
        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", "distributedtask"]);
        cmd.args(["--resource", "pools"]);
        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<AzureDevOpsAgentPool>,
        }

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

        debug!("Found {} Azure DevOps agent pools", resp.count);

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

        Ok(pools)
    }
}

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

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

    #[tokio::test]
    pub async fn it_works() -> eyre::Result<()> {
        let org_url = get_default_organization_url().await?;
        let pools = fetch_azure_devops_agent_pools(&org_url).await?;
        assert!(
            !pools.is_empty(),
            "Expected at least one Azure DevOps agent pool"
        );
        assert!(
            pools.iter().all(|pool| !pool.name.as_ref().is_empty()),
            "Expected sampled Azure DevOps agent pools to have names"
        );

        Ok(())
    }
}