use clap::Args;
use eyre::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 clap::ValueEnum;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::network::v2::floatingip::port_forwarding::create;
use openstack_types::network::v2::floatingip::port_forwarding::response;
#[derive(Args)]
#[command(about = "Create port forwarding")]
pub struct PortForwardingCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
port_forwarding: PortForwarding,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_floatingip_id",
value_name = "FLOATINGIP_ID"
)]
floatingip_id: String,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum Protocol {
Dccp,
Icmp,
Ipv6Icmp,
Sctp,
Tcp,
Udp,
}
#[derive(Args, Clone)]
struct PortForwarding {
#[arg(help_heading = "Body parameters", long)]
description: Option<String>,
#[arg(help_heading = "Body parameters", long)]
external_port: Option<Option<f32>>,
#[arg(help_heading = "Body parameters", long)]
external_port_range: Option<f32>,
#[arg(help_heading = "Body parameters", long)]
internal_ip_address: Option<String>,
#[arg(help_heading = "Body parameters", long)]
internal_port: Option<Option<f32>>,
#[arg(help_heading = "Body parameters", long)]
internal_port_id: Option<String>,
#[arg(help_heading = "Body parameters", long)]
internal_port_range: Option<f32>,
#[arg(help_heading = "Body parameters", long)]
project_id: Option<String>,
#[arg(help_heading = "Body parameters", long)]
protocol: Option<Protocol>,
}
impl PortForwardingCommand {
pub async fn take_action<C: CliArgs>(
&self,
parsed_args: &C,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Create PortForwarding");
let op = OutputProcessor::from_args(
parsed_args,
Some("network.floatingip/port_forwarding"),
Some("create"),
);
op.validate_args(parsed_args)?;
let mut ep_builder = create::Request::builder();
ep_builder.floatingip_id(&self.path.floatingip_id);
let args = &self.port_forwarding;
let mut port_forwarding_builder = create::PortForwardingBuilder::default();
if let Some(val) = &args.description {
port_forwarding_builder.description(val);
}
if let Some(val) = &args.external_port {
port_forwarding_builder.external_port(*val);
}
if let Some(val) = &args.external_port_range {
port_forwarding_builder.external_port_range(*val);
}
if let Some(val) = &args.internal_ip_address {
port_forwarding_builder.internal_ip_address(val);
}
if let Some(val) = &args.internal_port {
port_forwarding_builder.internal_port(*val);
}
if let Some(val) = &args.internal_port_id {
port_forwarding_builder.internal_port_id(val);
}
if let Some(val) = &args.internal_port_range {
port_forwarding_builder.internal_port_range(*val);
}
if let Some(val) = &args.project_id {
port_forwarding_builder.project_id(val);
}
if let Some(val) = &args.protocol {
let tmp = match val {
Protocol::Dccp => create::Protocol::Dccp,
Protocol::Icmp => create::Protocol::Icmp,
Protocol::Ipv6Icmp => create::Protocol::Ipv6Icmp,
Protocol::Sctp => create::Protocol::Sctp,
Protocol::Tcp => create::Protocol::Tcp,
Protocol::Udp => create::Protocol::Udp,
};
port_forwarding_builder.protocol(tmp);
}
ep_builder.port_forwarding(
port_forwarding_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::create::PortForwardingResponse>(data.clone())?;
op.show_command_hint()?;
Ok(())
}
}