use clap::Args;
use eyre::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 json_patch::diff;
use openstack_cli_core::common::parse_key_val;
use openstack_sdk::api::QueryAsync;
use openstack_sdk::api::find;
use openstack_sdk::api::image::v2::image::find;
use openstack_sdk::api::image::v2::image::patch;
use openstack_types::image::v2::image::response::patch::*;
use serde_json::Value;
use serde_json::json;
#[derive(Args)]
#[command(about = "Update image")]
pub struct ImageCommand {
#[command(flatten)]
query: QueryParameters,
#[command(flatten)]
path: PathParameters,
#[arg(help_heading = "Body parameters", long)]
container_format: Option<String>,
#[arg(help_heading = "Body parameters", long)]
disk_format: Option<String>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long, value_name="JSON", value_parser=openstack_cli_core::common::parse_json)]
locations: Option<Vec<Value>>,
#[arg(help_heading = "Body parameters", long)]
min_disk: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
min_ram: Option<i32>,
#[arg(help_heading = "Body parameters", long)]
name: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
os_hidden: Option<bool>,
#[arg(help_heading = "Body parameters", long)]
owner: Option<String>,
#[arg(action=clap::ArgAction::Set, help_heading = "Body parameters", long)]
protected: Option<bool>,
#[arg(action=clap::ArgAction::Append, help_heading = "Body parameters", long)]
tags: Option<Vec<String>>,
#[arg(help_heading = "Body parameters", long)]
visibility: Option<Visibility>,
#[arg(long="property", value_name="key=value", value_parser=parse_key_val::<String, String>)]
#[arg(help_heading = "Body parameters")]
properties: Option<Vec<(String, String)>>,
}
#[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 Visibility {
Community,
Private,
Public,
Shared,
}
impl ImageCommand {
pub async fn take_action<C: CliArgs>(
&self,
parsed_args: &C,
client: &mut AsyncOpenStack,
) -> Result<(), OpenStackCliError> {
info!("Set Image");
let op = OutputProcessor::from_args(parsed_args, Some("image.image"), Some("patch"));
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 resource_id = find_data["id"]
.as_str()
.ok_or_else(|| eyre::eyre!("resource ID must be a string"))?
.to_string();
let data: ImageResponse = serde_json::from_value(find_data)?;
let mut new = data.clone();
if let Some(val) = &self.container_format {
new.container_format = Some(val.into());
}
if let Some(val) = &self.disk_format {
new.disk_format = Some(val.into());
}
if let Some(val) = &self.locations {
new.locations = Some(serde_json::from_value(val.to_owned().into())?);
}
if let Some(val) = &self.min_disk {
new.min_disk = Some(*val);
}
if let Some(val) = &self.min_ram {
new.min_ram = Some(*val);
}
if let Some(val) = &self.name {
new.name = Some(val.into());
}
if let Some(val) = &self.os_hidden {
new.os_hidden = Some(*val);
}
if let Some(val) = &self.owner {
new.owner = Some(val.into());
}
if let Some(val) = &self.protected {
new.protected = Some(*val);
}
if let Some(val) = &self.tags {
new.tags = Some(serde_json::from_value(val.to_owned().into())?);
}
if let Some(val) = &self.visibility {
let tmp = match val {
Visibility::Community => "community",
Visibility::Private => "private",
Visibility::Public => "public",
Visibility::Shared => "shared",
};
new.visibility = Some(
tmp.parse()
.map_err(|_| eyre::eyre!("unsupported value for visibility"))?,
);
}
let curr_json =
serde_json::to_value(&data).wrap_err("current state must be a valid json object")?;
let mut new_json =
serde_json::to_value(&new).wrap_err("new state must be a valid json object")?;
if let Some(properties) = &self.properties {
for (key, val) in properties {
new_json[key] = json!(val);
}
}
let patch = diff(&curr_json, &new_json);
let mut patch_ep_builder = patch::Request::builder();
patch_ep_builder.id(&resource_id);
patch_ep_builder.patch(patch);
let patch_ep = patch_ep_builder
.build()
.map_err(|x| OpenStackCliError::EndpointBuild(x.to_string()))?;
let new_data = patch_ep.query_async(client).await?;
op.output_single::<ImageResponse>(new_data)?;
op.show_command_hint()?;
Ok(())
}
}