use super::user;

use quick_xml::events::{BytesEnd, BytesStart, BytesText, Event};
use quick_xml::{Reader, Writer};
use rand::Rng;
use std::fs;
use std::fs::File;
use std::io::{BufReader, BufWriter, Error};
use std::path::Path;
use uuid::Uuid;

const CHARSET: &[u8] = b"ABCDEFGHIJKLMNOPQRSTUVWXYZ\
                        abcdefghijklmnopqrstuvwxyz\
                        0123456789)(*^%$#@!~";

const PASSWORD_LEN: usize = 30;

const DOMAIN_XMLNS_QEMU: (&str, &str) =
    ("xmlns:qemu", "http://libvirt.org/schemas/domain/qemu/1.0");

const IE_VIRSH_TEMPLATE_VM_NAME: &str = "ie-virsh-template";
const VNC_XML_TAG: &str = "graphics";

const ROOT_START_TAG: &str = "domain";

const QEMU_COMMAND_LINE_TAG: &str = "qemu:commandline";
const QEMU_ARG_TAG: &str = "qemu:arg";

const TEMPLATE_XML_FILE: &str = "/etc/libvirt/template.xml";

const LIBVIRT_XML_DIR: &str = "/etc/libvirt/qemu";
const QCOW2_PATH: &str = "/mnt/ie-virsh";

pub struct GenerateVMArg {
    vm_name: String,
    qcow2_path: String,
    xml_path: String,
    vnc_password: String,
    is_debug: bool,
    tcp_port: u64,
}

pub fn dump_vnc_passwd(user: user::UserDetail, _vm_name: &str) -> Result<String, Error> {
    let user_pass = user.getpass();
    let mut reader = Reader::from_reader(BufReader::new(File::open(get_xml_dir(&user_pass))?));
    let mut buf = Vec::new();
    loop {
        match reader.read_event(&mut buf) {
            Ok(Event::Eof) => break,
            Ok(Event::Empty(ref e)) if e.name() == VNC_XML_TAG.as_bytes() => {}
            Ok(_) => {}
            Err(e) => panic!("Error at position {}: {:?}", reader.buffer_position(), e),
        }
    }
    Ok(String::from("ok"))
}

fn get_xml_dir(user_path: &str) -> String {
    format!("{}/{}", LIBVIRT_XML_DIR, user_path)
}

impl GenerateVMArg {
    pub fn new(user_name: &str, vm_name: &str, is_debug: bool) -> GenerateVMArg {
        let year = user_name.chars().skip(1).take(2).collect::<String>();
        let affilication = if year.parse::<u8>().is_ok() {
            // /etc/libvirt/qemu/e19/e195729
            user_name.chars().take(3).collect::<String>()
        } else {
            "teacher".to_string()
        };

        let xml_dir = format!("{}/{}/{}", LIBVIRT_XML_DIR, affilication, user_name);
        let xml_path = format!("{}/{}.xml", xml_dir, vm_name);

        if !Path::new(&xml_dir).exists() {
            fs::create_dir_all(xml_dir).ok();
        }

        let qcow2_dir = format!("{}/{}/{}", QCOW2_PATH, affilication, user_name);

        let qcow2_path = format!("{}/{}.qcow2", qcow2_dir, vm_name);

        if !Path::new(&qcow2_dir).exists() {
            fs::create_dir_all(qcow2_dir).ok();
        }

        let pw = generate_pw();

        GenerateVMArg {
            vm_name: vm_name.to_string(),
            qcow2_path,
            xml_path,
            vnc_password: pw,
            is_debug,
            tcp_port: 90,
        }
    }

    pub fn generate(self) -> Result<String, Error> {
        let mut reader = Reader::from_reader(BufReader::new(File::open(TEMPLATE_XML_FILE)?));

        println!("generate xml :{}", self.xml_path);
        let mut writer = Writer::new(BufWriter::new(File::create(self.xml_path.clone()).unwrap()));
        let mut buf = Vec::new();
        loop {
            match reader.read_event(&mut buf) {
                Ok(Event::Start(ref e)) if e.name() == b"uuid" => {
                    writer
                        .write_event(Event::Start(e.clone()))
                        .expect("faild write event");
                    reader
                        .read_event(&mut Vec::new())
                        .expect("faild read event");
                    let vm_uuid = Uuid::new_v4().to_string();
                    let elem = BytesText::from_plain_str(&vm_uuid);
                    writer.write_event(Event::Text(elem)).unwrap();
                }

                Ok(Event::Start(ref e))
                    if (e.name() == ROOT_START_TAG.as_bytes() && self.is_debug) =>
                {
                    let mut elem = e.clone();
                    elem.push_attribute(DOMAIN_XMLNS_QEMU);
                    writer.write_event(Event::Start(elem)).unwrap();

                    let qemu_command_line_start =
                        BytesStart::borrowed_name(QEMU_COMMAND_LINE_TAG.as_bytes());
                    writer
                        .write_event(Event::Start(qemu_command_line_start))
                        .unwrap();

                    for value in ["-S", "-gdb"].iter() {
                        let mut qemu_elem = BytesStart::borrowed_name(QEMU_ARG_TAG.as_bytes());
                        let v: &str = &value;
                        qemu_elem.push_attribute(("value", v));
                        writer.write_event(Event::Empty(qemu_elem)).unwrap();
                    }

                    let mut qemu_elem = BytesStart::borrowed_name(QEMU_ARG_TAG.as_bytes());
                    let gdb_port: &str = &format!("tcp::{}", self.tcp_port);
                    qemu_elem.push_attribute(("value", gdb_port));
                    writer.write_event(Event::Empty(qemu_elem)).unwrap();

                    let qemu_command_line_end =
                        BytesEnd::borrowed(QEMU_COMMAND_LINE_TAG.as_bytes());
                    writer
                        .write_event(Event::End(qemu_command_line_end))
                        .unwrap();
                }

                Ok(Event::Empty(ref e)) if e.name() == VNC_XML_TAG.as_bytes() => {
                    let mut elem = e.clone();
                    let pw: &str = &self.vnc_password;
                    elem.push_attribute(("passwd", pw));
                    writer.write_event(Event::Empty(elem)).ok();
                }

                Ok(Event::Empty(ref e)) if (e.name() == b"source") => {
                    let mut elem = e.clone();
                    let is_qcow_file = elem
                        .attributes()
                        .find(|attr| attr.as_ref().unwrap().key == b"file");
                    if is_qcow_file.is_some() {
                        elem.clear_attributes();
                        let qcow2_path: &str = &self.qcow2_path;
                        elem.push_attribute(("file", qcow2_path));
                    }
                    writer.write_event(Event::Empty(elem)).ok();
                }

                Ok(Event::Text(ref e)) if e.escaped() == IE_VIRSH_TEMPLATE_VM_NAME.as_bytes() => {
                    let elem = BytesText::from_plain_str(&self.vm_name);
                    writer.write_event(Event::Text(elem)).unwrap();
                }
                Ok(Event::Eof) => break,
                // you can use either `e` or `&e` if you don't want to move the event
                Ok(e) => assert!(writer.write_event(&e).is_ok()),
                Err(e) => panic!("Error at position {}: {:?}", reader.buffer_position(), e),
            }
            buf.clear();
        }
        println!("generate xml : {}", self.xml_path);
        println!("vnc password : {}", self.vnc_password);
        Ok(self.xml_path)
    }
}

fn generate_pw() -> String {
    let mut rng = rand::thread_rng();

    (0..PASSWORD_LEN)
        .map(|_| {
            let idx = rng.gen_range(0, CHARSET.len());
            CHARSET[idx] as char
        })
        .collect()
}