use clap::Args;
use eyre::{OptionExt, WrapErr};
use tracing::info;
use openstack_sdk::AsyncOpenStack;
use crate::Cli;
use crate::OpenStackCliError;
use crate::output::OutputProcessor;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::compute::v2::aggregate::find;
use openstack_sdk::api::compute::v2::aggregate::set_20;
use openstack_sdk::api::find;
use openstack_types::compute::v2::aggregate::response::set::AggregateResponse;
#[derive(Args)]
#[command(about = "Update Aggregate (microversion = 2.0)")]
pub struct AggregateCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
aggregate: Aggregate,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_id",
value_name = "ID"
)]
id: String,
}
#[derive(Args, Clone)]
struct Aggregate {
#[arg(help_heading = "Body parameters", long)]
availability_zone: Option<String>,
#[arg(help_heading = "Body parameters", long, action = clap::ArgAction::SetTrue, conflicts_with = "availability_zone")]
no_availability_zone: bool,
#[arg(help_heading = "Body parameters", long)]
name: Option<String>,
}
impl AggregateCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Aggregate");
let op = OutputProcessor::from_args(parsed_args, Some("compute.aggregate"), Some("set"));
op.validate_args(parsed_args)?;
let mut find_builder = find::Request::builder();
find_builder.id(&self.path.id);
find_builder.header(
http::header::HeaderName::from_static("openstack-api-version"),
http::header::HeaderValue::from_static("compute 2.0"),
);
let find_ep = find_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let find_data: serde_json::Value = find(find_ep).query_async(client).await?;
let mut ep_builder = set_20::Request::builder();
ep_builder.header(
http::header::HeaderName::from_static("openstack-api-version"),
http::header::HeaderValue::from_static("compute 2.0"),
);
let resource_id = find_data["id"]
.as_str()
.ok_or_else(|| eyre::eyre!("resource ID must be a string"))?
.to_string();
ep_builder.id(resource_id.clone());
let args = &self.aggregate;
let mut aggregate_builder = set_20::AggregateBuilder::default();
if let Some(val) = &args.availability_zone {
aggregate_builder.availability_zone(Some(val.into()));
} else if args.no_availability_zone {
aggregate_builder.availability_zone(None);
}
if let Some(val) = &args.name {
aggregate_builder.name(val);
}
ep_builder.aggregate(
aggregate_builder
.build()
.wrap_err("error preparing the request data")?,
);
let ep = ep_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let data = ep.query_async(client).await?;
op.output_single::<AggregateResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}