cloud_terrastodon_rest 0.36.0

In-process REST helpers and request objects for Cloud Terrastodon
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 crate::RestService;
use cloud_terrastodon_azure_types::AzureTenantId;
use cloud_terrastodon_azure_types::SubscriptionId;
use eyre::Result;
use reqwest::Url;
use std::future::Future;

pub async fn infer_tenant_id_for_request<F, Fut>(
    service: RestService,
    url: &Url,
    resolve_subscription_tenant: F,
) -> Result<Option<AzureTenantId>>
where
    F: FnOnce(SubscriptionId) -> Fut,
    Fut: Future<Output = Result<AzureTenantId>>,
{
    if service != RestService::AzureResourceManager {
        return Ok(None);
    }

    let Some(subscription_id) = extract_arm_subscription_id(url) else {
        return Ok(None);
    };

    Ok(Some(resolve_subscription_tenant(subscription_id).await?))
}

pub fn extract_arm_subscription_id(url: &Url) -> Option<SubscriptionId> {
    let mut segments = url.path_segments()?;
    let first = segments.next()?;
    if !first.eq_ignore_ascii_case("subscriptions") {
        return None;
    }
    let subscription = segments.next()?;
    subscription.parse().ok()
}

#[cfg(test)]
mod tests {
    use super::RestService;
    use super::extract_arm_subscription_id;
    use super::infer_tenant_id_for_request;
    use cloud_terrastodon_azure_types::AzureTenantId;
    use cloud_terrastodon_azure_types::SubscriptionId;
    use reqwest::Url;

    #[test]
    fn extracts_subscription_id_from_arm_subscription_url() {
        let url = Url::parse(
            "https://management.azure.com/subscriptions/11111111-1111-1111-1111-111111111111/locations?api-version=2022-12-01",
        )
        .unwrap();
        let expected = "11111111-1111-1111-1111-111111111111"
            .parse::<SubscriptionId>()
            .unwrap();
        assert_eq!(extract_arm_subscription_id(&url), Some(expected));
    }

    #[tokio::test]
    async fn skips_tenant_inference_for_non_arm_services() -> eyre::Result<()> {
        let url = Url::parse("https://graph.microsoft.com/v1.0/organization").unwrap();
        let tenant =
            infer_tenant_id_for_request(RestService::MicrosoftGraph, &url, |_| async move {
                eyre::bail!("resolver should not be called")
            })
            .await?;
        assert_eq!(tenant, None);
        Ok(())
    }

    #[tokio::test]
    async fn infers_tenant_for_arm_subscription_urls() -> eyre::Result<()> {
        let url = Url::parse(
            "https://management.azure.com/subscriptions/11111111-1111-1111-1111-111111111111/locations?api-version=2022-12-01",
        )
        .unwrap();
        let expected = "22222222-2222-2222-2222-222222222222"
            .parse::<AzureTenantId>()
            .unwrap();
        let tenant = infer_tenant_id_for_request(
            RestService::AzureResourceManager,
            &url,
            |_subscription_id| async move { Ok(expected) },
        )
        .await?;
        assert_eq!(tenant, Some(expected));
        Ok(())
    }
}