use clap::Args;
use eyre::OptionExt;
use tracing::info;
use openstack_sdk::AsyncOpenStack;
use crate::Cli;
use crate::OpenStackCliError;
use crate::output::OutputProcessor;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::find_by_name;
use openstack_sdk::api::identity::v3::project::find as find_project;
use openstack_sdk::api::load_balancer::v2::healthmonitor::list;
use openstack_sdk::api::{Pagination, paged};
use openstack_types::load_balancer::v2::healthmonitor::response::list::HealthmonitorResponse;
use tracing::warn;
#[derive(Args)]
#[command(about = "List Health Monitors")]
pub struct HealthmonitorsCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[arg(long, default_value_t = 10000)]
max_items: usize,
}
#[derive(Args)]
struct QueryParameters {
#[arg(help_heading = "Query parameters", long, value_parser = ["HTTP","HTTPS","PING","SCTP","TCP","TLS-HELLO","UDP-CONNECT"])]
_type: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Query parameters", long)]
admin_state_up: Option<bool>,
#[arg(help_heading = "Query parameters", long)]
created_at: Option<String>,
#[arg(help_heading = "Query parameters", long)]
delay: Option<i32>,
#[arg(help_heading = "Query parameters", long)]
description: Option<String>,
#[arg(help_heading = "Query parameters", long)]
expected_codes: Option<String>,
#[arg(help_heading = "Query parameters", long, value_parser = ["CONNECT","DELETE","GET","HEAD","OPTIONS","PATCH","POST","PUT","TRACE"])]
http_method: Option<String>,
#[arg(help_heading = "Query parameters", long)]
id: Option<String>,
#[arg(
help_heading = "Query parameters",
long("page-size"),
visible_alias("limit")
)]
limit: Option<i32>,
#[arg(help_heading = "Query parameters", long)]
marker: Option<String>,
#[arg(help_heading = "Query parameters", long)]
max_retries: Option<i32>,
#[arg(help_heading = "Query parameters", long)]
max_retries_down: Option<i32>,
#[arg(help_heading = "Query parameters", long)]
name: Option<String>,
#[arg(help_heading = "Query parameters", long)]
not_tags: Option<String>,
#[arg(help_heading = "Query parameters", long)]
not_tags_any: Option<String>,
#[arg(help_heading = "Query parameters", long, value_parser = ["DEGRADED","DRAINING","ERROR","NO_MONITOR","OFFLINE","ONLINE"])]
operating_status: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Query parameters", long)]
page_reverse: Option<bool>,
#[arg(help_heading = "Query parameters", long)]
pool_id: Option<String>,
#[command(flatten)]
project: ProjectInput,
#[arg(help_heading = "Query parameters", long, value_parser = ["ACTIVE","DELETED","ERROR","PENDING_CREATE","PENDING_DELETE","PENDING_UPDATE"])]
provisioning_status: Option<String>,
#[arg(help_heading = "Query parameters", long)]
tags: Option<String>,
#[arg(help_heading = "Query parameters", long)]
tags_any: Option<String>,
#[arg(help_heading = "Query parameters", long)]
timeout: Option<i32>,
#[arg(help_heading = "Query parameters", long)]
updated_at: Option<String>,
#[arg(help_heading = "Query parameters", long)]
url_path: Option<String>,
}
#[derive(Args)]
#[group(required = false, multiple = false)]
struct ProjectInput {
#[arg(long, help_heading = "Path parameters", value_name = "PROJECT_NAME")]
project_name: Option<String>,
#[arg(long, help_heading = "Path parameters", value_name = "PROJECT_ID")]
project_id: Option<String>,
#[arg(long, help_heading = "Path parameters", action = clap::ArgAction::SetTrue)]
current_project: bool,
}
#[derive(Args)]
struct PathParameters {}
impl HealthmonitorsCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("List Healthmonitors");
let op = OutputProcessor::from_args(
parsed_args,
Some("load-balancer.healthmonitor"),
Some("list"),
);
op.validate_args(parsed_args)?;
let mut ep_builder = list::Request::builder();
if let Some(val) = &self.query.admin_state_up {
ep_builder.admin_state_up(*val);
}
if let Some(val) = &self.query.created_at {
ep_builder.created_at(val);
}
if let Some(val) = &self.query.delay {
ep_builder.delay(*val);
}
if let Some(val) = &self.query.description {
ep_builder.description(val);
}
if let Some(val) = &self.query.expected_codes {
ep_builder.expected_codes(val);
}
if let Some(val) = &self.query.http_method {
ep_builder.http_method(val);
}
if let Some(val) = &self.query.id {
ep_builder.id(val);
}
if let Some(val) = &self.query.limit {
ep_builder.limit(*val);
}
if let Some(val) = &self.query.marker {
ep_builder.marker(val);
}
if let Some(val) = &self.query.max_retries {
ep_builder.max_retries(*val);
}
if let Some(val) = &self.query.max_retries_down {
ep_builder.max_retries_down(*val);
}
if let Some(val) = &self.query.name {
ep_builder.name(val);
}
if let Some(val) = &self.query.not_tags {
ep_builder.not_tags(val);
}
if let Some(val) = &self.query.not_tags_any {
ep_builder.not_tags_any(val);
}
if let Some(val) = &self.query.operating_status {
ep_builder.operating_status(val);
}
if let Some(val) = &self.query.page_reverse {
ep_builder.page_reverse(*val);
}
if let Some(val) = &self.query.pool_id {
ep_builder.pool_id(val);
}
if let Some(id) = &self.query.project.project_id {
ep_builder.project_id(id);
} else if let Some(name) = &self.query.project.project_name {
let mut sub_find_builder = find_project::Request::builder();
warn!(
"Querying project by name (because of `--project-name` parameter passed) may not be definite. This may fail in which case parameter `--project-id` should be used instead."
);
sub_find_builder.id(name);
let find_ep = sub_find_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let find_data: serde_json::Value = find_by_name(find_ep).query_async(client).await?;
match find_data.get("id") {
Some(val) => match val.as_str() {
Some(id_str) => {
ep_builder.project_id(id_str.to_owned());
}
None => {
return Err(OpenStackCliError::ResourceAttributeNotString(
serde_json::to_string(&val)?,
));
}
},
None => {
return Err(OpenStackCliError::ResourceAttributeMissing(
"id".to_string(),
));
}
};
} else if self.query.project.current_project {
ep_builder.project_id(
client
.get_auth_info()
.ok_or_eyre("Cannot determine current authentication information")?
.token
.user
.id,
);
}
if let Some(val) = &self.query.provisioning_status {
ep_builder.provisioning_status(val);
}
if let Some(val) = &self.query.tags {
ep_builder.tags(val);
}
if let Some(val) = &self.query.tags_any {
ep_builder.tags_any(val);
}
if let Some(val) = &self.query.timeout {
ep_builder.timeout(*val);
}
if let Some(val) = &self.query._type {
ep_builder._type(val);
}
if let Some(val) = &self.query.updated_at {
ep_builder.updated_at(val);
}
if let Some(val) = &self.query.url_path {
ep_builder.url_path(val);
}
let ep = ep_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let data: Vec<serde_json::Value> = paged(ep, Pagination::Limit(self.max_items))
.query_async(client)
.await?;
op.output_list::<HealthmonitorResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}