openstack_cli_load_balancer/v2/quota/
set.rs1use clap::Args;
23use eyre::{OptionExt, WrapErr};
24use tracing::info;
25
26use openstack_cli_core::cli::CliArgs;
27use openstack_cli_core::error::OpenStackCliError;
28use openstack_cli_core::output::OutputProcessor;
29use openstack_sdk::AsyncOpenStack;
30
31use eyre::eyre;
32use openstack_sdk::api::QueryAsync;
33use openstack_sdk::api::find_by_name;
34use openstack_sdk::api::identity::v3::project::find as find_project;
35use openstack_sdk::api::load_balancer::v2::quota::set;
36use openstack_types::load_balancer::v2::quota::response;
37use tracing::warn;
38
39#[derive(Args)]
54#[command(about = "Update a Quota")]
55pub struct QuotaCommand {
56 #[command(flatten)]
58 query: QueryParameters,
59
60 #[command(flatten)]
62 path: PathParameters,
63
64 #[command(flatten)]
66 quota: Quota,
67}
68
69#[derive(Args)]
71struct QueryParameters {}
72
73#[derive(Args)]
75struct PathParameters {
76 #[command(flatten)]
78 project: ProjectInput,
79}
80
81#[derive(Args)]
83#[group(required = true, multiple = false)]
84struct ProjectInput {
85 #[arg(long, help_heading = "Path parameters", value_name = "PROJECT_NAME")]
87 project_name: Option<String>,
88 #[arg(long, help_heading = "Path parameters", value_name = "PROJECT_ID")]
90 project_id: Option<String>,
91 #[arg(long, help_heading = "Path parameters", action = clap::ArgAction::SetTrue)]
93 current_project: bool,
94}
95#[derive(Args, Clone)]
97struct Quota {
98 #[arg(help_heading = "Body parameters", long)]
99 health_monitor: Option<i32>,
100
101 #[arg(help_heading = "Body parameters", long)]
105 healthmonitor: Option<i32>,
106
107 #[arg(help_heading = "Body parameters", long)]
110 l7policy: Option<i32>,
111
112 #[arg(help_heading = "Body parameters", long)]
115 l7rule: Option<i32>,
116
117 #[arg(help_heading = "Body parameters", long)]
120 listener: Option<i32>,
121
122 #[arg(help_heading = "Body parameters", long)]
123 load_balancer: Option<i32>,
124
125 #[arg(help_heading = "Body parameters", long)]
129 loadbalancer: Option<i32>,
130
131 #[arg(help_heading = "Body parameters", long)]
134 member: Option<i32>,
135
136 #[arg(help_heading = "Body parameters", long)]
139 pool: Option<i32>,
140}
141
142impl QuotaCommand {
143 pub async fn take_action<C: CliArgs>(
145 &self,
146 parsed_args: &C,
147 client: &mut AsyncOpenStack,
148 ) -> Result<(), OpenStackCliError> {
149 info!("Set Quota");
150
151 let op = OutputProcessor::from_args(parsed_args, Some("load-balancer.quota"), Some("set"));
152 op.validate_args(parsed_args)?;
153
154 let mut ep_builder = set::Request::builder();
155
156 if let Some(id) = &self.path.project.project_id {
158 ep_builder.project_id(id);
160 } else if let Some(name) = &self.path.project.project_name {
161 let mut sub_find_builder = find_project::Request::builder();
163 warn!(
164 "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."
165 );
166
167 sub_find_builder.id(name);
168 let find_ep = sub_find_builder
169 .build()
170 .map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
171 let find_data: serde_json::Value = find_by_name(find_ep).query_async(client).await?;
172 match find_data.get("id") {
174 Some(val) => match val.as_str() {
175 Some(id_str) => {
176 ep_builder.project_id(id_str.to_owned());
177 }
178 None => {
179 return Err(OpenStackCliError::ResourceAttributeNotString(
180 serde_json::to_string(&val)?,
181 ));
182 }
183 },
184 None => {
185 return Err(OpenStackCliError::ResourceAttributeMissing(
186 "id".to_string(),
187 ));
188 }
189 };
190 } else if self.path.project.current_project {
191 let token = client
192 .get_auth_info()
193 .ok_or_eyre("Cannot determine current authentication information")?
194 .token;
195 if let Some(project) = token.project {
196 ep_builder.project_id(
197 project
198 .id
199 .ok_or_eyre("Project ID is missing in the project auth info")?,
200 );
201 } else {
202 return Err(eyre!("Current project information can not be identified").into());
203 }
204 }
205
206 let args = &self.quota;
209 let mut quota_builder = set::QuotaBuilder::default();
210 if let Some(val) = &args.health_monitor {
211 quota_builder.health_monitor(*val);
212 }
213
214 if let Some(val) = &args.healthmonitor {
215 quota_builder.healthmonitor(*val);
216 }
217
218 if let Some(val) = &args.l7policy {
219 quota_builder.l7policy(*val);
220 }
221
222 if let Some(val) = &args.l7rule {
223 quota_builder.l7rule(*val);
224 }
225
226 if let Some(val) = &args.listener {
227 quota_builder.listener(*val);
228 }
229
230 if let Some(val) = &args.load_balancer {
231 quota_builder.load_balancer(*val);
232 }
233
234 if let Some(val) = &args.loadbalancer {
235 quota_builder.loadbalancer(*val);
236 }
237
238 if let Some(val) = &args.member {
239 quota_builder.member(*val);
240 }
241
242 if let Some(val) = &args.pool {
243 quota_builder.pool(*val);
244 }
245
246 ep_builder.quota(
247 quota_builder
248 .build()
249 .wrap_err("error preparing the request data")?,
250 );
251
252 let ep = ep_builder
253 .build()
254 .map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
255
256 let data: serde_json::Value = ep.query_async(client).await?;
257
258 op.output_single::<response::set::QuotaResponse>(data.clone())?;
259 op.show_command_hint()?;
261 Ok(())
262 }
263}