use crate::commands::Command;
use crate::env::Environment;
use crate::error::Error;
use crate::image::ImageStore;
use crate::instance::{InstanceStore, PortForward};
use crate::model::DataSize;
use crate::view::Console;
use clap::Parser;
#[derive(Parser)]
pub struct InstanceModifyCommand {
instance: String,
#[clap(short, long)]
cpus: Option<u16>,
#[clap(short, long)]
mem: Option<DataSize>,
#[clap(short, long)]
disk: Option<DataSize>,
#[clap(short, long)]
port: Vec<PortForward>,
#[clap(short = 'P', long)]
rm_port: Vec<PortForward>,
}
impl Command for InstanceModifyCommand {
fn run(
&self,
_console: &mut dyn Console,
_env: &Environment,
_image_store: &dyn ImageStore,
instance_store: &dyn InstanceStore,
) -> Result<(), Error> {
let mut instance = instance_store.load(&self.instance)?;
if let Some(cpus) = &self.cpus {
instance.cpus = *cpus;
}
if let Some(mem) = &self.mem {
instance.mem = mem.get_bytes() as u64;
}
if let Some(disk) = &self.disk {
instance_store.resize(&mut instance, disk.get_bytes() as u64)?;
}
instance.hostfwd.append(&mut self.port.clone());
instance.hostfwd.retain(|p| !self.rm_port.contains(p));
instance_store.store(&instance)?;
Result::Ok(())
}
}