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::healthmonitor::create;
use openstack_types::load_balancer::v2::healthmonitor::response::create::HealthmonitorResponse;
#[derive(Args)]
#[command(about = "Create Health Monitor")]
pub struct HealthmonitorCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
healthmonitor: Healthmonitor,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum HttpMethod {
Connect,
Delete,
Get,
Head,
Options,
Patch,
Post,
Put,
Trace,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum Type {
Http,
Https,
Ping,
Sctp,
Tcp,
TlsHello,
UdpConnect,
}
#[derive(Args, Clone)]
struct Healthmonitor {
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
admin_state_up: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
delay: i32,
#[arg(help_heading = "Body parameters", long)]
domain_name: Option<String>,
#[arg(help_heading = "Body parameters", long)]
expected_codes: Option<String>,
#[arg(help_heading = "Body parameters", long)]
http_method: Option<HttpMethod>,
#[arg(help_heading = "Body parameters", long)]
http_version: Option<f32>,
#[arg(help_heading = "Body parameters", long)]
max_retries: i32,
#[arg(help_heading = "Body parameters", long)]
max_retries_down: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
name: Option<String>,
#[arg(help_heading = "Body parameters", long)]
pool_id: String,
#[arg(help_heading = "Body parameters", long)]
project_id: Option<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>,
#[arg(help_heading = "Body parameters", long)]
timeout: i32,
#[arg(help_heading = "Body parameters", long)]
_type: Type,
#[arg(help_heading = "Body parameters", long)]
url_path: Option<String>,
}
impl HealthmonitorCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Create Healthmonitor");
let op = OutputProcessor::from_args(
parsed_args,
Some("load-balancer.healthmonitor"),
Some("create"),
);
op.validate_args(parsed_args)?;
let mut ep_builder = create::Request::builder();
let args = &self.healthmonitor;
let mut healthmonitor_builder = create::HealthmonitorBuilder::default();
if let Some(val) = &args.admin_state_up {
healthmonitor_builder.admin_state_up(*val);
}
healthmonitor_builder.delay(args.delay);
if let Some(val) = &args.domain_name {
healthmonitor_builder.domain_name(val);
}
if let Some(val) = &args.expected_codes {
healthmonitor_builder.expected_codes(val);
}
if let Some(val) = &args.http_method {
let tmp = match val {
HttpMethod::Connect => create::HttpMethod::Connect,
HttpMethod::Delete => create::HttpMethod::Delete,
HttpMethod::Get => create::HttpMethod::Get,
HttpMethod::Head => create::HttpMethod::Head,
HttpMethod::Options => create::HttpMethod::Options,
HttpMethod::Patch => create::HttpMethod::Patch,
HttpMethod::Post => create::HttpMethod::Post,
HttpMethod::Put => create::HttpMethod::Put,
HttpMethod::Trace => create::HttpMethod::Trace,
};
healthmonitor_builder.http_method(tmp);
}
if let Some(val) = &args.http_version {
healthmonitor_builder.http_version(*val);
}
healthmonitor_builder.max_retries(args.max_retries);
if let Some(val) = &args.max_retries_down {
healthmonitor_builder.max_retries_down(*val);
}
if let Some(val) = &args.name {
healthmonitor_builder.name(val);
}
healthmonitor_builder.pool_id(&args.pool_id);
if let Some(val) = &args.project_id {
healthmonitor_builder.project_id(val);
}
if let Some(val) = &args.tags {
healthmonitor_builder.tags(val.iter().map(Into::into).collect::<Vec<_>>());
}
if let Some(val) = &args.tenant_id {
healthmonitor_builder.tenant_id(val);
}
healthmonitor_builder.timeout(args.timeout);
let tmp = match &args._type {
Type::Http => create::Type::Http,
Type::Https => create::Type::Https,
Type::Ping => create::Type::Ping,
Type::Sctp => create::Type::Sctp,
Type::Tcp => create::Type::Tcp,
Type::TlsHello => create::Type::TlsHello,
Type::UdpConnect => create::Type::UdpConnect,
};
healthmonitor_builder._type(tmp);
if let Some(val) = &args.url_path {
healthmonitor_builder.url_path(val);
}
ep_builder.healthmonitor(
healthmonitor_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::<HealthmonitorResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}