use clap::Args;
use eyre::{OptionExt, WrapErr};
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;
use openstack_sdk::api::load_balancer::v2::pool::member::find;
use openstack_sdk::api::load_balancer::v2::pool::member::replace;
use openstack_types::load_balancer::v2::pool::member::response::replace::MemberResponse;
use serde_json::Value;
#[derive(Args)]
#[command(about = "Batch Update Members")]
pub struct MemberCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="JSON", value_parser=crate::common::parse_json)]
members: Vec<Value>,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_pool_id",
value_name = "POOL_ID"
)]
pool_id: String,
}
impl MemberCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Member");
let op = OutputProcessor::from_args(
parsed_args,
Some("load-balancer.pool/member"),
Some("replace"),
);
op.validate_args(parsed_args)?;
let mut find_builder = find::Request::builder();
find_builder.pool_id(&self.path.pool_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 = replace::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.pool_id(resource_id.clone());
let members_builder: Vec<replace::Members> = self
.members
.iter()
.flat_map(|v| serde_json::from_value::<replace::Members>(v.to_owned()))
.collect::<Vec<replace::Members>>();
ep_builder.members(members_builder);
let ep = ep_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let data = ep.query_async(client).await?;
op.output_single::<MemberResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}