use cloud_terrastodon_azure_devops_types::AzureDevOpsOrganizationUrl;
use cloud_terrastodon_azure_devops_types::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::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?;
assert!(
!entitlements.is_empty(),
"Expected at least one Azure DevOps user entitlement"
);
assert!(
entitlements.iter().all(|entitlement| {
!entitlement.user.display_name.is_empty()
&& !entitlement.user.unique_name.is_empty()
}),
"Expected sampled Azure DevOps user entitlements to include user identity data"
);
Ok(())
}
}