use clap::Args;
use eyre::{OptionExt, WrapErr};
use tracing::info;
use openstack_cli_core::cli::CliArgs;
use openstack_cli_core::error::OpenStackCliError;
use openstack_cli_core::output::OutputProcessor;
use openstack_sdk::AsyncOpenStack;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::find;
use openstack_sdk::api::network::v2::network::find;
use openstack_sdk::api::network::v2::network::set;
use openstack_types::network::v2::network::response;
use serde_json::Value;
#[derive(Args)]
#[command(about = "Update network")]
pub struct NetworkCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
network: Network,
}
#[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 Network {
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
admin_state_up: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
description: Option<String>,
#[arg(help_heading = "Body parameters", long)]
dns_domain: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
is_default: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
mtu: Option<u32>,
#[arg(help_heading = "Body parameters", long)]
name: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
port_security_enabled: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
provider_network_type: Option<String>,
#[arg(help_heading = "Body parameters", long)]
provider_physical_network: Option<String>,
#[arg(help_heading = "Body parameters", long)]
provider_segmentation_id: Option<String>,
#[arg(help_heading = "Body parameters", long)]
qos_policy_id: Option<String>,
#[arg(help_heading = "Body parameters", long, action = clap::ArgAction::SetTrue, conflicts_with = "qos_policy_id")]
no_qos_policy_id: bool,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
router_external: Option<bool>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="JSON", value_parser=openstack_cli_core::common::parse_json)]
segments: Option<Vec<Value>>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
shared: Option<bool>,
}
impl NetworkCommand {
pub async fn take_action<C: CliArgs>(
&self,
parsed_args: &C,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Network");
let op = OutputProcessor::from_args(parsed_args, Some("network.network"), Some("set"));
op.validate_args(parsed_args)?;
let mut find_builder = find::Request::builder();
find_builder.id(&self.path.id);
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::Request::builder();
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.network;
let mut network_builder = set::NetworkBuilder::default();
if let Some(val) = &args.admin_state_up {
network_builder.admin_state_up(*val);
}
if let Some(val) = &args.description {
network_builder.description(val);
}
if let Some(val) = &args.dns_domain {
network_builder.dns_domain(val);
}
if let Some(val) = &args.is_default {
network_builder.is_default(*val);
}
if let Some(val) = &args.mtu {
network_builder.mtu(*val);
}
if let Some(val) = &args.name {
network_builder.name(val);
}
if let Some(val) = &args.port_security_enabled {
network_builder.port_security_enabled(*val);
}
if let Some(val) = &args.provider_network_type {
network_builder.provider_network_type(val);
}
if let Some(val) = &args.provider_physical_network {
network_builder.provider_physical_network(val);
}
if let Some(val) = &args.provider_segmentation_id {
network_builder.provider_segmentation_id(val);
}
if let Some(val) = &args.qos_policy_id {
network_builder.qos_policy_id(Some(val.into()));
} else if args.no_qos_policy_id {
network_builder.qos_policy_id(None);
}
if let Some(val) = &args.router_external {
network_builder.router_external(*val);
}
if let Some(val) = &args.segments {
let segments_builder: Vec<set::Segments> = val
.iter()
.flat_map(|v| serde_json::from_value::<set::Segments>(v.to_owned()))
.collect::<Vec<set::Segments>>();
network_builder.segments(segments_builder);
}
if let Some(val) = &args.shared {
network_builder.shared(*val);
}
ep_builder.network(
network_builder
.build()
.wrap_err("error preparing the request data")?,
);
let ep = ep_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let data: serde_json::Value = ep.query_async(client).await?;
op.output_single::<response::set::NetworkResponse>(data.clone())?;
op.show_command_hint()?;
Ok(())
}
}