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 crate::common::parse_key_val;
use clap::ValueEnum;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::find;
use openstack_sdk::api::network::v2::port::find;
use openstack_sdk::api::network::v2::port::set;
use openstack_types::network::v2::port::response::set::PortResponse;
use serde_json::Value;
#[derive(Args)]
#[command(about = "Update port")]
pub struct PortCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
port: Port,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_id",
value_name = "ID"
)]
id: String,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum BindingVnicType {
AcceleratorDirect,
AcceleratorDirectPhysical,
Baremetal,
Direct,
DirectPhysical,
Macvtap,
Normal,
RemoteManaged,
SmartNic,
Vdpa,
VirtioForwarder,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum DataPlaneStatus {
Active,
Down,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum NumaAffinityPolicy {
Legacy,
Preferred,
Required,
Socket,
}
#[derive(Args, Clone)]
struct Port {
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
admin_state_up: Option<bool>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="JSON", value_parser=crate::common::parse_json)]
allowed_address_pairs: Option<Vec<Value>>,
#[arg(help_heading = "Body parameters", long)]
binding_host_id: Option<String>,
#[arg(help_heading = "Body parameters", long, value_name="key=value", value_parser=parse_key_val::<String, Value>)]
binding_profile: Option<Vec<(String, Value)>>,
#[arg(help_heading = "Body parameters", long)]
binding_vnic_type: Option<BindingVnicType>,
#[arg(help_heading = "Body parameters", long)]
data_plane_status: Option<DataPlaneStatus>,
#[arg(help_heading = "Body parameters", long)]
description: Option<String>,
#[arg(help_heading = "Body parameters", long)]
device_id: Option<String>,
#[arg(help_heading = "Body parameters", long)]
device_owner: Option<String>,
#[arg(help_heading = "Body parameters", long)]
dns_domain: Option<String>,
#[arg(help_heading = "Body parameters", long)]
dns_name: Option<String>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="JSON", value_parser=crate::common::parse_json)]
extra_dhcp_opts: Option<Vec<Value>>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="JSON", value_parser=crate::common::parse_json)]
fixed_ips: Option<Vec<Value>>,
#[arg(help_heading = "Body parameters", long, value_name="key=value", value_parser=parse_key_val::<String, Value>)]
hints: Option<Vec<(String, Value)>>,
#[arg(help_heading = "Body parameters", long)]
mac_address: Option<String>,
#[arg(help_heading = "Body parameters", long)]
name: Option<String>,
#[arg(help_heading = "Body parameters", long)]
numa_affinity_policy: Option<NumaAffinityPolicy>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
port_security_enabled: Option<bool>,
#[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::Append, help_heading = "Body parameters", long)]
security_groups: Option<Vec<String>>,
}
impl PortCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Port");
let op = OutputProcessor::from_args(parsed_args, Some("network.port"), 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.port;
let mut port_builder = set::PortBuilder::default();
if let Some(val) = &args.admin_state_up {
port_builder.admin_state_up(*val);
}
if let Some(val) = &args.allowed_address_pairs {
let allowed_address_pairs_builder: Vec<set::AllowedAddressPairs> = val
.iter()
.flat_map(|v| serde_json::from_value::<set::AllowedAddressPairs>(v.to_owned()))
.collect::<Vec<set::AllowedAddressPairs>>();
port_builder.allowed_address_pairs(allowed_address_pairs_builder);
}
if let Some(val) = &args.binding_host_id {
port_builder.binding_host_id(val);
}
if let Some(val) = &args.binding_profile {
port_builder.binding_profile(val.iter().cloned());
}
if let Some(val) = &args.binding_vnic_type {
let tmp = match val {
BindingVnicType::AcceleratorDirect => set::BindingVnicType::AcceleratorDirect,
BindingVnicType::AcceleratorDirectPhysical => {
set::BindingVnicType::AcceleratorDirectPhysical
}
BindingVnicType::Baremetal => set::BindingVnicType::Baremetal,
BindingVnicType::Direct => set::BindingVnicType::Direct,
BindingVnicType::DirectPhysical => set::BindingVnicType::DirectPhysical,
BindingVnicType::Macvtap => set::BindingVnicType::Macvtap,
BindingVnicType::Normal => set::BindingVnicType::Normal,
BindingVnicType::RemoteManaged => set::BindingVnicType::RemoteManaged,
BindingVnicType::SmartNic => set::BindingVnicType::SmartNic,
BindingVnicType::Vdpa => set::BindingVnicType::Vdpa,
BindingVnicType::VirtioForwarder => set::BindingVnicType::VirtioForwarder,
};
port_builder.binding_vnic_type(tmp);
}
if let Some(val) = &args.data_plane_status {
let tmp = match val {
DataPlaneStatus::Active => set::DataPlaneStatus::Active,
DataPlaneStatus::Down => set::DataPlaneStatus::Down,
};
port_builder.data_plane_status(tmp);
}
if let Some(val) = &args.description {
port_builder.description(val);
}
if let Some(val) = &args.device_id {
port_builder.device_id(val);
}
if let Some(val) = &args.device_owner {
port_builder.device_owner(val);
}
if let Some(val) = &args.dns_domain {
port_builder.dns_domain(val);
}
if let Some(val) = &args.dns_name {
port_builder.dns_name(val);
}
if let Some(val) = &args.extra_dhcp_opts {
use std::collections::BTreeMap;
port_builder.extra_dhcp_opts(
val.iter()
.map(|v| {
v.as_object()
.ok_or_eyre("extra_dhcp_opts must be a valid json object")
.map(|obj| {
obj.into_iter()
.map(|(k, v)| (k.into(), v.clone()))
.collect::<BTreeMap<_, Value>>()
})
})
.collect::<Result<Vec<_>, _>>()?,
);
}
if let Some(val) = &args.fixed_ips {
let fixed_ips_builder: Vec<set::FixedIps> = val
.iter()
.flat_map(|v| serde_json::from_value::<set::FixedIps>(v.to_owned()))
.collect::<Vec<set::FixedIps>>();
port_builder.fixed_ips(fixed_ips_builder);
}
if let Some(val) = &args.hints {
port_builder.hints(val.iter().cloned());
}
if let Some(val) = &args.mac_address {
port_builder.mac_address(val);
}
if let Some(val) = &args.name {
port_builder.name(val);
}
if let Some(val) = &args.numa_affinity_policy {
let tmp = match val {
NumaAffinityPolicy::Legacy => set::NumaAffinityPolicy::Legacy,
NumaAffinityPolicy::Preferred => set::NumaAffinityPolicy::Preferred,
NumaAffinityPolicy::Required => set::NumaAffinityPolicy::Required,
NumaAffinityPolicy::Socket => set::NumaAffinityPolicy::Socket,
};
port_builder.numa_affinity_policy(tmp);
}
if let Some(val) = &args.port_security_enabled {
port_builder.port_security_enabled(*val);
}
if let Some(val) = &args.qos_policy_id {
port_builder.qos_policy_id(Some(val.into()));
} else if args.no_qos_policy_id {
port_builder.qos_policy_id(None);
}
if let Some(val) = &args.security_groups {
port_builder.security_groups(val.iter().map(Into::into).collect::<Vec<_>>());
}
ep_builder.port(
port_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::<PortResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}