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

use clap::Args;
use cloud_terrastodon_azure::AzureTenantArgument;
use cloud_terrastodon_azure::AzureTenantArgumentExt;
use cloud_terrastodon_azure::Scope;
use cloud_terrastodon_azure::fetch_all_resources;
use eyre::Result;
use eyre::bail;
use std::io::Write;
use tracing::info;

/// Arguments for showing a single Azure resource by id or by name.
#[derive(Args, Debug, Clone)]
pub struct AzureResourceShowArgs {
    /// Tracked tenant id or alias to query. Defaults to the active Azure CLI tenant.
    #[arg(long, default_value_t)]
    pub tenant: AzureTenantArgument<'static>,

    /// Resource id (scope) or resource name.
    pub resource: String,
}

impl AzureResourceShowArgs {
    pub async fn invoke(self) -> Result<()> {
        let tenant_id = self.tenant.resolve().await?;
        info!(needle = %self.resource, "Fetching all Azure resources");
        let resources = fetch_all_resources(tenant_id).await?;
        info!(count = resources.len(), "Fetched Azure resources");

        if let Some(resource) = resources
            .iter()
            .find(|resource| resource.id.expanded_form() == self.resource)
        {
            let stdout = std::io::stdout();
            let mut handle = stdout.lock();
            serde_json::to_writer_pretty(&mut handle, resource)?;
            handle.write_all(b"\n")?;
            return Ok(());
        }

        let mut named_matches = resources
            .into_iter()
            .filter(|resource| {
                resource.name == self.resource
                    || resource
                        .display_name
                        .as_ref()
                        .map(|x| x == &self.resource)
                        .unwrap_or(false)
            })
            .collect::<Vec<_>>();

        match named_matches.len() {
            0 => {
                bail!("No resource found matching '{}'.", self.resource);
            }
            1 => {
                let stdout = std::io::stdout();
                let mut handle = stdout.lock();
                serde_json::to_writer_pretty(&mut handle, &named_matches.remove(0))?;
                handle.write_all(b"\n")?;
                Ok(())
            }
            _ => {
                named_matches.sort_by_key(|a| a.id.expanded_form());
                let ids = named_matches
                    .iter()
                    .map(|resource| resource.id.expanded_form())
                    .collect::<Vec<_>>()
                    .join("\n  ");
                bail!(
                    "Multiple resources matched '{}'. Use a full resource id.\n  {}",
                    self.resource,
                    ids
                );
            }
        }
    }
}