openstack_cli 0.13.5

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// SPDX-License-Identifier: Apache-2.0
//
// WARNING: This file is automatically generated from OpenAPI schema using
// `openstack-codegenerator`.

//! Set Quota command
//!
//! Wraps invoking of the `v2/lbaas/quotas/{project_id}` with `PUT` method

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 eyre::eyre;
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::quota::set;
use openstack_types::load_balancer::v2::quota::response::set::QuotaResponse;
use tracing::warn;

/// Updates a quota for a project.
///
/// If the request is valid, the service returns the `Accepted (202)` response
/// code.
///
/// This operation returns the updated quota object.
///
/// If the quota is specified as `null` the quota will use the deployment
/// default quota settings.
///
/// Specifying a quota of `-1` means the quota is unlimited.
///
/// Specifying a quota of `0` means the project cannot create any of the
/// resource.
#[derive(Args)]
#[command(about = "Update a Quota")]
pub struct QuotaCommand {
    /// Request Query parameters
    #[command(flatten)]
    query: QueryParameters,

    /// Path parameters
    #[command(flatten)]
    path: PathParameters,

    /// Individual quota definitions.
    #[command(flatten)]
    quota: Quota,
}

/// Query parameters
#[derive(Args)]
struct QueryParameters {}

/// Path parameters
#[derive(Args)]
struct PathParameters {
    /// Project resource for which the operation should be performed.
    #[command(flatten)]
    project: ProjectInput,
}

/// Project input select group
#[derive(Args)]
#[group(required = true, multiple = false)]
struct ProjectInput {
    /// Project Name.
    #[arg(long, help_heading = "Path parameters", value_name = "PROJECT_NAME")]
    project_name: Option<String>,
    /// Project ID.
    #[arg(long, help_heading = "Path parameters", value_name = "PROJECT_ID")]
    project_id: Option<String>,
    /// Current project.
    #[arg(long, help_heading = "Path parameters", action = clap::ArgAction::SetTrue)]
    current_project: bool,
}
/// Quota Body data
#[derive(Args, Clone)]
struct Quota {
    #[arg(help_heading = "Body parameters", long)]
    health_monitor: Option<i32>,

    /// The configured health monitor quota limit. A setting of `null` means it
    /// is using the deployment default quota. A setting of `-1` means
    /// unlimited.
    #[arg(help_heading = "Body parameters", long)]
    healthmonitor: Option<i32>,

    /// The configured l7policy quota limit. A setting of `null` means it is
    /// using the deployment default quota. A setting of `-1` means unlimited.
    #[arg(help_heading = "Body parameters", long)]
    l7policy: Option<i32>,

    /// The configured l7rule quota limit. A setting of `null` means it is
    /// using the deployment default quota. A setting of `-1` means unlimited.
    #[arg(help_heading = "Body parameters", long)]
    l7rule: Option<i32>,

    /// The configured listener quota limit. A setting of `null` means it is
    /// using the deployment default quota. A setting of `-1` means unlimited.
    #[arg(help_heading = "Body parameters", long)]
    listener: Option<i32>,

    #[arg(help_heading = "Body parameters", long)]
    load_balancer: Option<i32>,

    /// The configured load balancer quota limit. A setting of `null` means it
    /// is using the deployment default quota. A setting of `-1` means
    /// unlimited.
    #[arg(help_heading = "Body parameters", long)]
    loadbalancer: Option<i32>,

    /// The configured member quota limit. A setting of `null` means it is
    /// using the deployment default quota. A setting of `-1` means unlimited.
    #[arg(help_heading = "Body parameters", long)]
    member: Option<i32>,

    /// The configured pool quota limit. A setting of `null` means it is using
    /// the deployment default quota. A setting of `-1` means unlimited.
    #[arg(help_heading = "Body parameters", long)]
    pool: Option<i32>,
}

impl QuotaCommand {
    /// Perform command action
    pub async fn take_action(
        &self,
        parsed_args: &Cli,
        client: &mut AsyncOpenStack,
    ) -> Result<(), OpenStackCliError> {
        info!("Set Quota");

        let op = OutputProcessor::from_args(parsed_args, Some("load-balancer.quota"), Some("set"));
        op.validate_args(parsed_args)?;

        let mut ep_builder = set::Request::builder();

        // Process path parameter `project_id`
        if let Some(id) = &self.path.project.project_id {
            // project_id is passed. No need to lookup
            ep_builder.project_id(id);
        } else if let Some(name) = &self.path.project.project_name {
            // project_name is passed. Need to lookup resource
            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?;
            // Try to extract resource id
            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.path.project.current_project {
            let token = client
                .get_auth_info()
                .ok_or_eyre("Cannot determine current authentication information")?
                .token;
            if let Some(project) = token.project {
                ep_builder.project_id(
                    project
                        .id
                        .ok_or_eyre("Project ID is missing in the project auth info")?,
                );
            } else {
                return Err(eyre!("Current project information can not be identified").into());
            }
        }

        // Set body parameters
        // Set Request.quota data
        let args = &self.quota;
        let mut quota_builder = set::QuotaBuilder::default();
        if let Some(val) = &args.health_monitor {
            quota_builder.health_monitor(*val);
        }

        if let Some(val) = &args.healthmonitor {
            quota_builder.healthmonitor(*val);
        }

        if let Some(val) = &args.l7policy {
            quota_builder.l7policy(*val);
        }

        if let Some(val) = &args.l7rule {
            quota_builder.l7rule(*val);
        }

        if let Some(val) = &args.listener {
            quota_builder.listener(*val);
        }

        if let Some(val) = &args.load_balancer {
            quota_builder.load_balancer(*val);
        }

        if let Some(val) = &args.loadbalancer {
            quota_builder.loadbalancer(*val);
        }

        if let Some(val) = &args.member {
            quota_builder.member(*val);
        }

        if let Some(val) = &args.pool {
            quota_builder.pool(*val);
        }

        ep_builder.quota(
            quota_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::<QuotaResponse>(data)?;
        // Show command specific hints
        op.show_command_hint()?;
        Ok(())
    }
}