use clap::Args;
use eyre::{OptionExt, WrapErr};
use tracing::info;
use openstack_cli_core::cli::CliArgs;
use openstack_cli_core::error::OpenStackCliError;
use openstack_cli_core::output::OutputProcessor;
use openstack_sdk::AsyncOpenStack;
use clap::ValueEnum;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::block_storage::v3::volume::os_volume_upload_image_31;
#[derive(Args)]
pub struct VolumeCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[command(flatten)]
os_volume_upload_image: OsVolumeUploadImage,
}
#[derive(Args)]
struct QueryParameters {}
#[derive(Args)]
struct PathParameters {
#[arg(
help_heading = "Path parameters",
id = "path_param_id",
value_name = "ID"
)]
id: String,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum DiskFormat {
Ploop,
Qcow2,
Raw,
Vdi,
Vhd,
Vhdx,
Vmdk,
}
#[derive(Clone, Eq, Ord, PartialEq, PartialOrd, ValueEnum)]
enum Visibility {
Community,
Private,
Public,
Shared,
}
#[derive(Args, Clone)]
struct OsVolumeUploadImage {
#[arg(help_heading = "Body parameters", long)]
container_format: Option<String>,
#[arg(help_heading = "Body parameters", long, action = clap::ArgAction::SetTrue, conflicts_with = "container_format")]
no_container_format: bool,
#[arg(help_heading = "Body parameters", long)]
disk_format: Option<DiskFormat>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
force: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
image_name: String,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
protected: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
visibility: Option<Visibility>,
}
impl VolumeCommand {
pub async fn take_action<C: CliArgs>(
&self,
parsed_args: &C,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Action Volume");
let op = OutputProcessor::from_args(
parsed_args,
Some("block-storage.volume"),
Some("os_volume_upload_image"),
);
op.validate_args(parsed_args)?;
let mut ep_builder = os_volume_upload_image_31::Request::builder();
ep_builder.header(
http::header::HeaderName::from_static("openstack-api-version"),
http::header::HeaderValue::from_static("volume 3.1"),
);
ep_builder.id(&self.path.id);
let args = &self.os_volume_upload_image;
let mut os_volume_upload_image_builder =
os_volume_upload_image_31::OsVolumeUploadImageBuilder::default();
if let Some(val) = &args.container_format {
os_volume_upload_image_builder.container_format(Some(val.into()));
} else if args.no_container_format {
os_volume_upload_image_builder.container_format(None);
}
if let Some(val) = &args.disk_format {
let tmp = match val {
DiskFormat::Ploop => os_volume_upload_image_31::DiskFormat::Ploop,
DiskFormat::Qcow2 => os_volume_upload_image_31::DiskFormat::Qcow2,
DiskFormat::Raw => os_volume_upload_image_31::DiskFormat::Raw,
DiskFormat::Vdi => os_volume_upload_image_31::DiskFormat::Vdi,
DiskFormat::Vhd => os_volume_upload_image_31::DiskFormat::Vhd,
DiskFormat::Vhdx => os_volume_upload_image_31::DiskFormat::Vhdx,
DiskFormat::Vmdk => os_volume_upload_image_31::DiskFormat::Vmdk,
};
os_volume_upload_image_builder.disk_format(tmp);
}
if let Some(val) = &args.force {
os_volume_upload_image_builder.force(*val);
}
os_volume_upload_image_builder.image_name(&args.image_name);
if let Some(val) = &args.protected {
os_volume_upload_image_builder.protected(*val);
}
if let Some(val) = &args.visibility {
let tmp = match val {
Visibility::Community => os_volume_upload_image_31::Visibility::Community,
Visibility::Private => os_volume_upload_image_31::Visibility::Private,
Visibility::Public => os_volume_upload_image_31::Visibility::Public,
Visibility::Shared => os_volume_upload_image_31::Visibility::Shared,
};
os_volume_upload_image_builder.visibility(tmp);
}
ep_builder.os_volume_upload_image(
os_volume_upload_image_builder
.build()
.wrap_err("error preparing the request data")?,
);
let ep = ep_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
openstack_sdk::api::ignore(ep).query_async(client).await?;
op.show_command_hint()?;
Ok(())
}
}