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::load_balancer::v2::l7policy::rule::set;
use openstack_types::load_balancer::v2::l7policy::rule::response::set::RuleResponse;
#[derive(Args)]
#[command(about = "Update a L7 Rule")]
pub struct RuleCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
rule: Rule,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_l7policy_id",
value_name = "L7POLICY_ID"
)]
l7policy_id: String,
#[arg(
help_heading = "Path parameters",
id = "path_param_id",
value_name = "ID"
)]
id: String,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum CompareType {
Contains,
EndsWith,
EqualTo,
Regex,
StartsWith,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum Type {
Cookie,
FileType,
Header,
HostName,
Path,
SslConnHasCert,
SslDnField,
SslVerifyResult,
}
#[derive(Args, Clone)]
struct Rule {
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
admin_state_up: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
compare_type: Option<CompareType>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
invert: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
key: Option<String>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long)]
tags: Option<Vec<String>>,
#[arg(help_heading = "Body parameters", long)]
_type: Option<Type>,
#[arg(help_heading = "Body parameters", long)]
value: Option<String>,
}
impl RuleCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Rule");
let op = OutputProcessor::from_args(
parsed_args,
Some("load-balancer.l7policy/rule"),
Some("set"),
);
op.validate_args(parsed_args)?;
let mut ep_builder = set::Request::builder();
ep_builder.l7policy_id(&self.path.l7policy_id);
ep_builder.id(&self.path.id);
let args = &self.rule;
let mut rule_builder = set::RuleBuilder::default();
if let Some(val) = &args.admin_state_up {
rule_builder.admin_state_up(*val);
}
if let Some(val) = &args.compare_type {
let tmp = match val {
CompareType::Contains => set::CompareType::Contains,
CompareType::EndsWith => set::CompareType::EndsWith,
CompareType::EqualTo => set::CompareType::EqualTo,
CompareType::Regex => set::CompareType::Regex,
CompareType::StartsWith => set::CompareType::StartsWith,
};
rule_builder.compare_type(tmp);
}
if let Some(val) = &args.invert {
rule_builder.invert(*val);
}
if let Some(val) = &args.key {
rule_builder.key(val);
}
if let Some(val) = &args.tags {
rule_builder.tags(val.iter().map(Into::into).collect::<Vec<_>>());
}
if let Some(val) = &args._type {
let tmp = match val {
Type::Cookie => set::Type::Cookie,
Type::FileType => set::Type::FileType,
Type::Header => set::Type::Header,
Type::HostName => set::Type::HostName,
Type::Path => set::Type::Path,
Type::SslConnHasCert => set::Type::SslConnHasCert,
Type::SslDnField => set::Type::SslDnField,
Type::SslVerifyResult => set::Type::SslVerifyResult,
};
rule_builder._type(tmp);
}
if let Some(val) = &args.value {
rule_builder.value(val);
}
ep_builder.rule(
rule_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::<RuleResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}