use clap::Args;
use eyre::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::identity::v3::user::find;
use openstack_sdk::api::identity::v3::user::set;
use openstack_types::identity::v3::user::response::set::UserResponse;
use serde_json::Value;
#[derive(Args)]
#[command(about = "Update user")]
pub struct UserCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
user: User,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_id",
value_name = "ID"
)]
id: String,
}
#[derive(Args, Clone)]
#[group(required = false, multiple = true)]
struct Options {
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
ignore_change_password_upon_first_use: Option<bool>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
ignore_lockout_failure_attempts: Option<bool>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
ignore_password_expiry: Option<bool>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
ignore_user_inactivity: Option<bool>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
lock_password: Option<bool>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
multi_factor_auth_enabled: Option<bool>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="[String] as JSON", value_parser=crate::common::parse_json)]
multi_factor_auth_rules: Option<Vec<Vec<String>>>,
}
#[derive(Args, Clone)]
struct User {
#[arg(help_heading = "Body parameters", long)]
default_project_id: Option<String>,
#[arg(help_heading = "Body parameters", long, action = clap::ArgAction::SetTrue, conflicts_with = "default_project_id")]
no_default_project_id: bool,
#[arg(help_heading = "Body parameters", long)]
description: Option<String>,
#[arg(help_heading = "Body parameters", long, action = clap::ArgAction::SetTrue, conflicts_with = "description")]
no_description: bool,
#[arg(help_heading = "Body parameters", long)]
domain_id: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
enabled: Option<bool>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="JSON", value_parser=crate::common::parse_json)]
federated: Option<Vec<Value>>,
#[arg(help_heading = "Body parameters", long)]
name: Option<String>,
#[command(flatten)]
options: Option<Options>,
#[arg(help_heading = "Body parameters", long)]
password: Option<String>,
#[arg(help_heading = "Body parameters", long, action = clap::ArgAction::SetTrue, conflicts_with = "password")]
no_password: bool,
}
impl UserCommand {
pub async fn take_action(
&self,
parsed_args: &Cli,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set User");
let op = OutputProcessor::from_args(parsed_args, Some("identity.user"), Some("set"));
op.validate_args(parsed_args)?;
let mut find_builder = find::Request::builder();
find_builder.id(&self.path.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 = 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());
let args = &self.user;
let mut user_builder = set::UserBuilder::default();
if let Some(val) = &args.default_project_id {
user_builder.default_project_id(Some(val.into()));
} else if args.no_default_project_id {
user_builder.default_project_id(None);
}
if let Some(val) = &args.description {
user_builder.description(Some(val.into()));
} else if args.no_description {
user_builder.description(None);
}
if let Some(val) = &args.domain_id {
user_builder.domain_id(val);
}
if let Some(val) = &args.enabled {
user_builder.enabled(*val);
}
if let Some(val) = &args.federated {
let federated_builder: Vec<set::Federated> = val
.iter()
.flat_map(|v| serde_json::from_value::<set::Federated>(v.to_owned()))
.collect::<Vec<set::Federated>>();
user_builder.federated(federated_builder);
}
if let Some(val) = &args.name {
user_builder.name(val);
}
if let Some(val) = &args.options {
let mut options_builder = set::OptionsBuilder::default();
if let Some(val) = &val.ignore_change_password_upon_first_use {
options_builder.ignore_change_password_upon_first_use(*val);
}
if let Some(val) = &val.ignore_lockout_failure_attempts {
options_builder.ignore_lockout_failure_attempts(*val);
}
if let Some(val) = &val.ignore_password_expiry {
options_builder.ignore_password_expiry(*val);
}
if let Some(val) = &val.ignore_user_inactivity {
options_builder.ignore_user_inactivity(*val);
}
if let Some(val) = &val.lock_password {
options_builder.lock_password(*val);
}
if let Some(val) = &val.multi_factor_auth_enabled {
options_builder.multi_factor_auth_enabled(*val);
}
if let Some(val) = &val.multi_factor_auth_rules {
options_builder.multi_factor_auth_rules(val.iter());
}
user_builder.options(
options_builder
.build()
.wrap_err("error preparing the request data")?,
);
}
if let Some(val) = &args.password {
user_builder.password(Some(val.into()));
} else if args.no_password {
user_builder.password(None);
}
ep_builder.user(
user_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::<UserResponse>(data)?;
op.show_command_hint()?;
Ok(())
}
}