view src/command.rs @ 41:63e77a9de3ab

...
author AnaTofuZ <anatofuz@gmail.com>
date Sun, 22 Nov 2020 16:48:05 +0900
parents d3055f6c6fb7
children 2787ada7650b
line wrap: on
line source

use super::user;
use super::virsh;
use super::xml;

use std::io;

pub fn list(user_name: &str) {
    let (ldump_msg, vm_list_strs) = virsh::get_vm_list(user_name);
    println!("{}", ldump_msg.info_msg);
    println!("{}", ldump_msg.border_line);
    for vm_info in vm_list_strs {
        println!("{}", vm_info);
    }
}

pub fn templates() -> Result<(), io::Error> {
    let templates_list = virsh::get_template_list()?;

    for name in templates_list {
        println!("{}", name);
    }

    Ok(())
}

pub fn define(user_detail: &user::UserDetail, vm_name: &str) {
    let vm_name = generate_vm_name(&user_detail.user.name, &vm_name);
    let builder = xml::Builder::new(&user_detail, &vm_name).finalize();
    if let Ok(xml_path) = builder.generate() {
        virsh::command_require_vm_name(&xml_path, "define")
    }
}

pub fn define_from_template(user_detail: &user::UserDetail, vm_name: &str, template: &str) {
    let vm_name = generate_vm_name(&user_detail.user.name, &vm_name);
    println!("{}", vm_name);
    let backing_path = match virsh::get_backing_path(template) {
        Some(path) => path,
        None => {
            println!("not found {}", template);
            return;
        }
    };
    let builder = xml::Builder::new(&user_detail, &vm_name)
        .backing(&backing_path)
        .finalize();
    let xml_path = match builder.generate() {
        Ok(xml_path) => xml_path,
        Err(e) => {
            println!("failed generate xml: {}", e);
            return;
        }
    };
    virsh::command_require_vm_name(&xml_path, "define");
    virsh::generate_qemu_from_template(&vm_name, "hoge");
}

pub fn start(user_name: &str, vm_name: &str) {
    virsh::command_require_vm_name(&get_vm_name(user_name, vm_name), "start");
}

pub fn common(user_name: &str, vm_name: &str, command: &str) {
    exec_cmd_from_name_or_id(user_name, vm_name, command)
}

pub fn dumpvncpasswd(_user_name: &str, _vm_name: &str) {}

fn exec_cmd_from_name_or_id(user_name: &str, name_or_id: &str, command: &str) {
    if name_or_id.parse::<u8>().is_ok() {
        let vm_name_or_none = get_vm_name_from_id(user_name, &name_or_id);
        if let Some(vm_name) = vm_name_or_none {
            return virsh::command_require_vm_name(&vm_name, command);
        }
    }
    virsh::command_require_vm_name(&get_vm_name(user_name, name_or_id), command);
}

fn get_vm_name_from_id(user_name: &str, arg_id: &str) -> Option<String> {
    let (_, vm_list_strs) = virsh::get_vm_list(user_name);
    for vm in vm_list_strs {
        let mut splitted = vm.split_whitespace();
        let vm_id = splitted.next().unwrap();
        if (vm_id != "-") && (vm_id == arg_id) {
            let vm_name = splitted.next().unwrap();
            return Some(String::from(vm_name));
        }
    }
    None
}

fn get_vm_name(user_name: &str, vm_name: &str) -> String {
    let vm_name = match vm_name.parse::<u8>() {
        Ok(_) => format!("{}-{}", user_name, vm_name),
        Err(_) => vm_name.to_string(),
    };

    if vm_name.starts_with(user_name) {
        return vm_name;
    }
    format!("{}-{}", user_name, vm_name)
}

fn generate_vm_name(user_name: &str, vm_name: &str) -> String {
    if vm_name.starts_with(user_name) {
        return vm_name.to_string();
    }
    format!("{}-{}", user_name, vm_name)
}