cloud_terrastodon_azure 0.35.1

Helpers for interacting with Azure 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

use cloud_terrastodon_azure_types::AzureLocationName;
use cloud_terrastodon_azure_types::ComputeSkuName;
use cloud_terrastodon_azure_types::Price;
use cloud_terrastodon_command::CacheKey;
use cloud_terrastodon_command::CacheableCommand;
use cloud_terrastodon_command::CommandBuilder;
use cloud_terrastodon_command::CommandKind;
use cloud_terrastodon_command::async_trait;
use compact_str::CompactString;
use std::path::PathBuf;

pub struct ComputeSkuPricesRequest {
    pub location: AzureLocationName,
    pub sku: ComputeSkuName,
}

pub fn fetch_compute_sku_prices(
    location: AzureLocationName,
    sku: ComputeSkuName,
) -> ComputeSkuPricesRequest {
    ComputeSkuPricesRequest { location, sku }
}

#[async_trait]
impl CacheableCommand for ComputeSkuPricesRequest {
    type Output = Vec<Price>;

    fn cache_key(&self) -> CacheKey {
        CacheKey::new(PathBuf::from_iter([
            "az",
            "vm",
            "list-sku-pricings",
            self.location.to_string().as_ref(),
            self.sku.as_ref(),
        ]))
    }

    async fn run(self) -> eyre::Result<Self::Output> {
        let url = format!(
            "https://prices.azure.com/api/retail/prices?{filter}&meterRegion='primary'&currencyCode='CAD'",
            filter = format_args!(
                "$filter=serviceName eq '{service_name}' and tolower(armRegionName) eq tolower('{location}') and armSkuName eq '{sku}'",
                service_name = "Virtual Machines",
                location = self.location,
                sku = self.sku
            )
        );
        let mut cmd = CommandBuilder::new(CommandKind::AzureCLI);
        cmd.args(["rest", "--method", "GET", "--url"]);
        cmd.azure_file_arg("url.txt", url);
        cmd.cache(self.cache_key());

        #[derive(serde::Deserialize)]
        #[serde(deny_unknown_fields)]
        #[serde(rename_all = "PascalCase")]
        #[allow(dead_code)]
        struct Response {
            billing_currency: CompactString,
            count: usize,
            customer_entity_id: CompactString,
            customer_entity_type: CompactString,
            items: Vec<Price>,
            next_page_link: Option<String>,
        }
        let rtn = cmd.run::<Response>().await?.items;
        Ok(rtn)
    }
}

cloud_terrastodon_command::impl_cacheable_into_future!(ComputeSkuPricesRequest);

#[cfg(test)]
mod test {
    use crate::fetch_compute_sku_prices;
    use cloud_terrastodon_azure_types::AzureLocationName;
    use cloud_terrastodon_azure_types::ComputeSkuName;

    #[tokio::test]
    pub async fn it_works() -> eyre::Result<()> {
        let sku = ComputeSkuName::try_new("Standard_D2s_v5")?;
        let location = AzureLocationName::CanadaCentral;
        let prices = fetch_compute_sku_prices(location, sku).await?;
        assert!(!prices.is_empty()); // idk why failing, add CLI for sku browse
        assert!(prices.iter().any(|price| price.unit_price > 0.0));
        Ok(())
    }
}