use anyhow::{Context, Error, bail};
use crate::common;
use crate::common::app_context::AppContext;
use crate::http_client::{MantaClient, OpenApiResultExt};
use crate::openapi_client::types::{AddNodesToGroupRequest, Group};
use crate::output::action_result;
pub struct ExecParams<'a> {
pub label: &'a str,
pub description: Option<&'a str>,
pub hosts_expression: Option<&'a str>,
pub assume_yes: bool,
pub dry_run: bool,
pub output: Option<&'a str>,
}
pub async fn exec(
ctx: &AppContext<'_>,
auth_token: &str,
p: ExecParams<'_>,
) -> Result<(), Error> {
let grp = Group {
label: p.label.to_string(),
description: p.description.map(String::from),
tags: None,
members: None,
exclusive_group: Some("false".to_string()),
};
let add_nodes_req = p.hosts_expression.map(|expr| AddNodesToGroupRequest {
hosts_expression: expr.to_string(),
});
if p.dry_run {
let preview = match &add_nodes_req {
Some(req) => serde_json::json!({
"create_group": &grp,
"add_nodes_to_group": req,
}),
None => serde_json::json!({ "create_group": &grp }),
};
return action_result::print_with_data(
"Would create group (and add nodes if provided):",
&preview,
p.output,
);
}
if !common::confirm::confirm(
&format!(
"This operation will create the group below:\n{}\nPlease confirm to proceed",
serde_json::to_string_pretty(&grp)
.context("Failed to serialize group")?
),
p.assume_yes,
) {
bail!("Operation cancelled by user");
}
let client = MantaClient::from_app_ctx(ctx, Some(auth_token))?;
client
.openapi
.create_group(client.site_name(), &grp)
.await
.into_anyhow()?;
if let Some(req) = add_nodes_req {
client
.openapi
.add_nodes_to_group(p.label, client.site_name(), &req)
.await
.into_anyhow()?;
}
action_result::print(&format!("Group '{}' created", p.label), p.output)?;
Ok(())
}
#[cfg(test)]
mod tests {
#[test]
fn accepts_dry_run() {
let result = crate::build::build_cli().try_get_matches_from([
"manta",
"add",
"group",
"--label",
"compute",
"--dry-run",
]);
assert!(
result.is_ok(),
"expected --dry-run to parse on `add group`: {result:?}"
);
}
#[test]
fn accepts_dry_run_short_alias() {
let result = crate::build::build_cli().try_get_matches_from([
"manta", "add", "group", "--label", "compute", "-d",
]);
assert!(
result.is_ok(),
"expected -d short alias to parse on `add group`: {result:?}"
);
}
#[test]
fn description_short_alias_is_capital_d() {
let result = crate::build::build_cli().try_get_matches_from([
"manta",
"add",
"group",
"--label",
"compute",
"-D",
"ops cluster",
]);
assert!(
result.is_ok(),
"expected -D to parse as --description on `add group`: {result:?}"
);
}
}