use clap::Args;
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::dns::v2::zone::find as find_zone;
use openstack_sdk::api::dns::v2::zone::recordset::find;
use openstack_sdk::api::dns::v2::zone::recordset::set;
use openstack_sdk::api::find;
use openstack_sdk::api::find_by_name;
use openstack_types::dns::v2::zone::recordset::response::set::RecordsetResponse;
use tracing::warn;
#[derive(Args)]
#[command(about = "Update a Recordset")]
pub struct RecordsetCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[arg(help_heading = "Body parameters", long)]
description: Option<String>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long)]
records: Option<Vec<String>>,
#[arg(help_heading = "Body parameters", long)]
ttl: Option<i32>,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_id",
value_name = "ID"
)]
id: String,
#[command(flatten)]
zone: ZoneInput,
}
#[derive(Args)]
#[group(required = true, multiple = false)]
struct ZoneInput {
#[arg(long, help_heading = "Path parameters", value_name = "ZONE_NAME")]
zone_name: Option<String>,
#[arg(long, help_heading = "Path parameters", value_name = "ZONE_ID")]
zone_id: Option<String>,
}
impl RecordsetCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Recordset");
let op = OutputProcessor::from_args(parsed_args, Some("dns.zone/recordset"), Some("set"));
op.validate_args(parsed_args)?;
let mut find_builder = find::Request::builder();
find_builder.id(&self.path.id);
if let Some(id) = &self.path.zone.zone_id {
find_builder.zone_id(id);
} else if let Some(name) = &self.path.zone.zone_name {
let mut sub_find_builder = find_zone::Request::builder();
warn!(
"Querying zone by name (because of `--zone-name` parameter passed) may not be definite. This may fail in which case parameter `--zone-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) => {
find_builder.zone_id(id_str.to_owned());
}
None => {
return Err(OpenStackCliError::ResourceAttributeNotString(
serde_json::to_string(&val)?,
));
}
},
None => {
return Err(OpenStackCliError::ResourceAttributeMissing(
"id".to_string(),
));
}
};
}
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());
if let Some(id) = &self.path.zone.zone_id {
ep_builder.zone_id(id);
} else if let Some(name) = &self.path.zone.zone_name {
let mut sub_find_builder = find_zone::Request::builder();
warn!(
"Querying zone by name (because of `--zone-name` parameter passed) may not be definite. This may fail in which case parameter `--zone-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.zone_id(id_str.to_owned());
}
None => {
return Err(OpenStackCliError::ResourceAttributeNotString(
serde_json::to_string(&val)?,
));
}
},
None => {
return Err(OpenStackCliError::ResourceAttributeMissing(
"id".to_string(),
));
}
};
}
if let Some(arg) = &self.description {
ep_builder.description(arg);
}
if let Some(arg) = &self.records {
ep_builder.records(arg.iter().map(Into::into).collect::<Vec<_>>());
}
if let Some(arg) = &self.ttl {
ep_builder.ttl(*arg);
}
let ep = ep_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let data = ep.query_async(client).await?;
op.output_single::<RecordsetResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}