use cloud_terrastodon_azure_types::prelude::ConsentType;
use cloud_terrastodon_azure_types::prelude::EntraServicePrincipalId;
use cloud_terrastodon_azure_types::prelude::EntraUserId;
use cloud_terrastodon_azure_types::prelude::OAuth2PermissionGrant;
use cloud_terrastodon_azure_types::prelude::OAuth2PermissionGrantId;
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 std::path::PathBuf;
use std::sync::LazyLock;
use tracing::info;
pub static FETCH_OAUTH2_PERMISSION_GRANTS_CACHE_DIR: LazyLock<PathBuf> =
LazyLock::new(|| PathBuf::from_iter(["ms", "graph", "GET", "oauth2PermissionGrants"]));
pub struct OAuth2PermissionGrantListRequest;
pub fn fetch_oauth2_permission_grants() -> OAuth2PermissionGrantListRequest {
OAuth2PermissionGrantListRequest
}
#[async_trait]
impl cloud_terrastodon_command::CacheableCommand for OAuth2PermissionGrantListRequest {
type Output = Vec<OAuth2PermissionGrant>;
fn cache_key(&self) -> CacheKey {
CacheKey::new(FETCH_OAUTH2_PERMISSION_GRANTS_CACHE_DIR.to_path_buf())
}
async fn run(self) -> eyre::Result<Self::Output> {
let url = "https://graph.microsoft.com/v1.0/oauth2PermissionGrants";
let mut cmd = CommandBuilder::new(CommandKind::AzureCLI);
cmd.cache(self.cache_key());
cmd.arg("rest");
cmd.args(["--method", "GET"]);
cmd.args(["--url", url]);
#[derive(Debug, Deserialize)]
struct Response {
value: Vec<OAuth2PermissionGrant>,
}
let resp = cmd.run::<Response>().await?;
Ok(resp.value)
}
}
cloud_terrastodon_command::impl_cacheable_into_future!(OAuth2PermissionGrantListRequest);
pub async fn create_oauth2_permission_grant(
resource_id: EntraServicePrincipalId,
client_id: EntraServicePrincipalId,
user_id: EntraUserId,
scope: String,
) -> eyre::Result<OAuth2PermissionGrant> {
info!(
"Creating OAuth2 permission grant for {} for {}",
scope, user_id
);
let url = "https://graph.microsoft.com/v1.0/oauth2PermissionGrants";
let body = OAuth2PermissionGrant {
resource_id,
client_id,
consent_type: ConsentType::Principal,
id: OAuth2PermissionGrantId("".to_string()),
principal_id: Some(user_id),
scope,
};
let mut cmd = CommandBuilder::new(CommandKind::AzureCLI);
cmd.args(["rest", "--method", "POST"]);
cmd.args(["--url", url]);
cmd.arg("--body");
cmd.azure_file_arg("body.json", serde_json::to_string_pretty(&body)?);
cmd.run().await
}
#[cfg(test)]
mod tests {
use super::*;
#[tokio::test]
async fn it_works() -> eyre::Result<()> {
let found = fetch_oauth2_permission_grants().await?;
for row in found {
println!("- {row:?}");
}
Ok(())
}
}