use clap::Args;
use eyre::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::network::v2::quota::set;
use openstack_types::network::v2::quota::response::set::QuotaResponse;
#[derive(Args)]
#[command(about = "Update quota for a project")]
pub struct QuotaCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
quota: Quota,
}
#[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 Quota {
#[arg(help_heading = "Body parameters", long)]
floatingip: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
network: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
port: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
project_id: Option<String>,
#[arg(help_heading = "Body parameters", long)]
rbac_policy: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
router: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
security_group: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
security_group_rule: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
subnet: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
subnetpool: Option<i32>,
}
impl QuotaCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Quota");
let op = OutputProcessor::from_args(parsed_args, Some("network.quota"), Some("set"));
op.validate_args(parsed_args)?;
let mut ep_builder = set::Request::builder();
ep_builder.id(&self.path.id);
let args = &self.quota;
let mut quota_builder = set::QuotaBuilder::default();
if let Some(val) = &args.floatingip {
quota_builder.floatingip(*val);
}
if let Some(val) = &args.network {
quota_builder.network(*val);
}
if let Some(val) = &args.port {
quota_builder.port(*val);
}
if let Some(val) = &args.project_id {
quota_builder.project_id(val);
}
if let Some(val) = &args.rbac_policy {
quota_builder.rbac_policy(*val);
}
if let Some(val) = &args.router {
quota_builder.router(*val);
}
if let Some(val) = &args.security_group {
quota_builder.security_group(*val);
}
if let Some(val) = &args.security_group_rule {
quota_builder.security_group_rule(*val);
}
if let Some(val) = &args.subnet {
quota_builder.subnet(*val);
}
if let Some(val) = &args.subnetpool {
quota_builder.subnetpool(*val);
}
ep_builder.quota(
quota_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::<QuotaResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}