use clap::Args;
use eyre::WrapErr;
use tracing::info;
use openstack_sdk::AsyncOpenStack;
use crate::Cli;
use crate::OpenStackCliError;
use crate::output::OutputProcessor;
use clap::ValueEnum;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::find;
use openstack_sdk::api::load_balancer::v2::pool::find;
use openstack_sdk::api::load_balancer::v2::pool::set;
use openstack_types::load_balancer::v2::pool::response::set::PoolResponse;
#[derive(Args)]
#[command(about = "Update a Pool")]
pub struct PoolCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
pool: Pool,
}
#[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 LbAlgorithm {
LeastConnections,
RoundRobin,
SourceIp,
SourceIpPort,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum Type {
AppCookie,
HttpCookie,
SourceIp,
}
#[derive(Args, Clone)]
#[group(required = false, multiple = true)]
struct SessionPersistence {
#[arg(help_heading = "Body parameters", long)]
cookie_name: Option<String>,
#[arg(help_heading = "Body parameters", long)]
persistence_granularity: Option<String>,
#[arg(help_heading = "Body parameters", long)]
persistence_timeout: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
_type: Option<Type>,
}
#[derive(Args, Clone)]
struct Pool {
#[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)]
alpn_protocols: Option<Vec<String>>,
#[arg(help_heading = "Body parameters", long)]
ca_tls_container_ref: Option<String>,
#[arg(help_heading = "Body parameters", long)]
crl_container_ref: Option<String>,
#[arg(help_heading = "Body parameters", long)]
description: Option<String>,
#[arg(help_heading = "Body parameters", long)]
lb_algorithm: Option<LbAlgorithm>,
#[arg(help_heading = "Body parameters", long)]
name: Option<String>,
#[command(flatten)]
session_persistence: Option<SessionPersistence>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long)]
tags: Option<Vec<String>>,
#[arg(help_heading = "Body parameters", long)]
tls_ciphers: Option<String>,
#[arg(help_heading = "Body parameters", long)]
tls_container_ref: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
tls_enabled: Option<bool>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long)]
tls_versions: Option<Vec<String>>,
}
impl PoolCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Pool");
let op = OutputProcessor::from_args(parsed_args, Some("load-balancer.pool"), 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.pool;
let mut pool_builder = set::PoolBuilder::default();
if let Some(val) = &args.admin_state_up {
pool_builder.admin_state_up(*val);
}
if let Some(val) = &args.alpn_protocols {
pool_builder.alpn_protocols(val.iter().map(Into::into).collect::<Vec<_>>());
}
if let Some(val) = &args.ca_tls_container_ref {
pool_builder.ca_tls_container_ref(val);
}
if let Some(val) = &args.crl_container_ref {
pool_builder.crl_container_ref(val);
}
if let Some(val) = &args.description {
pool_builder.description(val);
}
if let Some(val) = &args.lb_algorithm {
let tmp = match val {
LbAlgorithm::LeastConnections => set::LbAlgorithm::LeastConnections,
LbAlgorithm::RoundRobin => set::LbAlgorithm::RoundRobin,
LbAlgorithm::SourceIp => set::LbAlgorithm::SourceIp,
LbAlgorithm::SourceIpPort => set::LbAlgorithm::SourceIpPort,
};
pool_builder.lb_algorithm(tmp);
}
if let Some(val) = &args.name {
pool_builder.name(val);
}
if let Some(val) = &args.session_persistence {
let mut session_persistence_builder = set::SessionPersistenceBuilder::default();
if let Some(val) = &val.cookie_name {
session_persistence_builder.cookie_name(val);
}
if let Some(val) = &val.persistence_granularity {
session_persistence_builder.persistence_granularity(val);
}
if let Some(val) = &val.persistence_timeout {
session_persistence_builder.persistence_timeout(*val);
}
if let Some(val) = &val._type {
let tmp = match val {
Type::AppCookie => set::Type::AppCookie,
Type::HttpCookie => set::Type::HttpCookie,
Type::SourceIp => set::Type::SourceIp,
};
session_persistence_builder._type(tmp);
}
pool_builder.session_persistence(
session_persistence_builder
.build()
.wrap_err("error preparing the request data")?,
);
}
if let Some(val) = &args.tags {
pool_builder.tags(val.iter().map(Into::into).collect::<Vec<_>>());
}
if let Some(val) = &args.tls_ciphers {
pool_builder.tls_ciphers(val);
}
if let Some(val) = &args.tls_container_ref {
pool_builder.tls_container_ref(val);
}
if let Some(val) = &args.tls_enabled {
pool_builder.tls_enabled(*val);
}
if let Some(val) = &args.tls_versions {
pool_builder.tls_versions(val.iter().map(Into::into).collect::<Vec<_>>());
}
ep_builder.pool(
pool_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::<PoolResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}