use crate::error::Error;
use crate::instance::{InstanceDao, InstanceStore, PortForward};
use crate::util;
use clap::Parser;
#[derive(Parser)]
pub struct InstanceModifyCommand {
instance: String,
#[clap(short, long)]
cpus: Option<u16>,
#[clap(short, long)]
mem: Option<String>,
#[clap(short, long)]
disk: Option<String>,
#[clap(short, long)]
port: Vec<PortForward>,
#[clap(short = 'P', long)]
rm_port: Vec<PortForward>,
}
impl InstanceModifyCommand {
pub fn run(&self, instance_dao: &InstanceDao) -> Result<(), Error> {
let mut instance = instance_dao.load(&self.instance)?;
if let Some(cpus) = &self.cpus {
instance.cpus = *cpus;
}
if let Some(mem) = &self.mem {
instance.mem = util::human_readable_to_bytes(mem)?;
}
if let Some(disk) = &self.disk {
instance_dao.resize(&mut instance, util::human_readable_to_bytes(disk)?)?;
}
instance.hostfwd.append(&mut self.port.clone());
instance.hostfwd.retain(|p| !self.rm_port.contains(p));
instance_dao.store(&instance)?;
Result::Ok(())
}
}