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::network::v2::port::create;
use openstack_types::network::v2::port::response::create::PortResponse;
use serde_json::Value;
#[derive(Args)]
#[command(about = "Create port")]
pub struct PortCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
port: Port,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {}
#[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 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)]
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)]
device_profile: Option<String>,
#[arg(help_heading = "Body parameters", long, action = clap::ArgAction::SetTrue, conflicts_with = "device_profile")]
no_device_profile: bool,
#[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)]
network_id: 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(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
propagate_uplink_status: 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>>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long)]
tags: Option<Vec<String>>,
#[arg(help_heading = "Body parameters", long)]
tenant_id: Option<String>,
}
impl PortCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Create Port");
let op = OutputProcessor::from_args(parsed_args, Some("network.port"), Some("create"));
op.validate_args(parsed_args)?;
let mut ep_builder = create::Request::builder();
let args = &self.port;
let mut port_builder = create::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<create::AllowedAddressPairs> = val
.iter()
.flat_map(|v| serde_json::from_value::<create::AllowedAddressPairs>(v.to_owned()))
.collect::<Vec<create::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 => create::BindingVnicType::AcceleratorDirect,
BindingVnicType::AcceleratorDirectPhysical => {
create::BindingVnicType::AcceleratorDirectPhysical
}
BindingVnicType::Baremetal => create::BindingVnicType::Baremetal,
BindingVnicType::Direct => create::BindingVnicType::Direct,
BindingVnicType::DirectPhysical => create::BindingVnicType::DirectPhysical,
BindingVnicType::Macvtap => create::BindingVnicType::Macvtap,
BindingVnicType::Normal => create::BindingVnicType::Normal,
BindingVnicType::RemoteManaged => create::BindingVnicType::RemoteManaged,
BindingVnicType::SmartNic => create::BindingVnicType::SmartNic,
BindingVnicType::Vdpa => create::BindingVnicType::Vdpa,
BindingVnicType::VirtioForwarder => create::BindingVnicType::VirtioForwarder,
};
port_builder.binding_vnic_type(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.device_profile {
port_builder.device_profile(Some(val.into()));
} else if args.no_device_profile {
port_builder.device_profile(None);
}
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<create::FixedIps> = val
.iter()
.flat_map(|v| serde_json::from_value::<create::FixedIps>(v.to_owned()))
.collect::<Vec<create::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.network_id {
port_builder.network_id(val);
}
if let Some(val) = &args.numa_affinity_policy {
let tmp = match val {
NumaAffinityPolicy::Legacy => create::NumaAffinityPolicy::Legacy,
NumaAffinityPolicy::Preferred => create::NumaAffinityPolicy::Preferred,
NumaAffinityPolicy::Required => create::NumaAffinityPolicy::Required,
NumaAffinityPolicy::Socket => create::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.propagate_uplink_status {
port_builder.propagate_uplink_status(*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<_>>());
}
if let Some(val) = &args.tags {
port_builder.tags(val.iter().map(Into::into).collect::<Vec<_>>());
}
if let Some(val) = &args.tenant_id {
port_builder.tenant_id(val);
}
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(())
}
}