Project rename1.2.1

1 files changed, 0 insertions, 247 deletions

diff --git a/src/house.rs b/src/house.rs
deleted file mode 100644
index abd0eef..0000000
--- a/src/house.rs
+++ /dev/null
@@ -1,247 +0,0 @@
-/*!
-  * `House`s are where [crate::waifu::Waifu]s live (the physical hypervisors that
-  * libvirtd connects to
-  */
-use crate::waifu::*;
-use crate::errors::Error;
-use crate::van::Van;
-use serde::{Serialize, Deserialize};
-use virt::{connect::Connect, domain::Domain};
-use std::process::{ExitStatus};
-use std::os::unix::process::ExitStatusExt;
-use rand::Rng;
-
-use tokio::{process::Command};
-
-#[derive(Serialize, Deserialize, Debug)]
-pub enum Address {
-    IP(String),
-    Domain(String)
-}
-
-impl ToString for Address {
-    fn to_string(&self) -> String {
-        match self {
-            Address::IP(s) => s.clone(),
-            Address::Domain(s) => s.clone(),
-        }
-    }
-}
-
-/// Defines a "house" where waifus live
-#[derive(Debug, Serialize, Deserialize)]
-pub struct House {
-    /// Hostname
-    pub name: String,
-    /// FQDN or IP address, a way to talk to the house
-    pub address: String,
-
-    /// Whether or not the House is local (same machine) or remote (networked machine)
-    pub remote: bool,
-
-    /// Connection to the House, if available
-    #[serde(skip)]
-    con: Option<Connect>,
-}
-
-impl House { 
-    /// Creates a new House based on a libvirtd connect URL
-    ///
-    /// Example:
-    /// ```
-    /// use waifulib::house::House;
-    /// let mut h: House = House::new("test:///default".to_string()).unwrap();
-    /// ```
-    pub fn new(url: String) -> Result<Self, Error> {
-        let c = Connect::open(&url.clone())?;
-
-        let remote = if url.contains("qemu:///") || url.contains("localhost") || url.contains("127.0.0.1") {
-            true
-        } else {
-            false
-        };
-
-        // If the connection succeeds, we've got one!
-        Ok(Self {
-            name: c.get_hostname()?,
-            address: c.get_uri()?,
-            remote,
-            con: Some(c),
-        })
-    }
-
-    /// Lists the "inhabitants" of the House (the [Waifu]s on the machine
-    ///
-    /// ```
-    /// use waifulib::house::House;
-    /// let mut h: House = House::new("test:///default".to_string()).unwrap();
-    ///
-    /// assert_eq!(h.inhabitants().unwrap().len(), 1);
-    /// ```
-    pub fn inhabitants(&mut self) -> Result<Vec<Waifu>, Error> {
-        match &self.con {
-            Some(c) => {
-                let domains = c.list_all_domains(0)?;
-
-                let mut waifus: Vec<Waifu> = Vec::new();
-
-                for d in domains.iter() {
-                    waifus.push(d.clone().try_into()?);
-                }
-                Ok(waifus)
-            },
-            None => {
-                return Err(Error::Connection("Domain connection was None".to_string()));
-            }
-        }
-    }
-
-    /// Introduces a new Waifu into the House, taking care of everything needed to make the VM run
-    /// fine
-    ///
-    /// If the installation image doesn't exist in the default libvirtd pool, this will fail with
-    /// [`Error::MissingImage`][crate::errors::Error::MissingImage].
-    ///
-    /// ```
-    /// use waifulib::house::House
-    ///
-    /// let mut h: House = House::new("test:///default".to_string()).unwrap();
-    ///
-    /// h.introduce("test-2", 1024, 1, 20000, "test.iso").unwrap();
-    /// ```
-    pub async fn introduce(&mut self, name: String, max_mem: Memory, max_cpus: CpuCount, disk_size_mb: u64, van: Van) -> Result<Waifu, Error> {
-        // Check for image on host machine
-        
-        if self.remote {
-            let mut output = Command::new("ssh")
-                .args([
-                      "-oStrictHostKeyChecking=accept-new",
-                      &self.address.clone(),
-                      "stat",
-                      &format!("/var/lib/libvirt/images/{}", van.make_pretty_name().clone())
-                ])
-                .output()
-                .await?;
-
-            if output.status != ExitStatus::from_raw(0) {
-               return Err(Error::MissingImage(van.name.clone()));
-            }
-
-            // Allocate VM disk
-            output = Command::new("ssh")
-                .args([
-                      "-oStrictHostKeyChecking=accept-new",
-                      &self.address.clone(),
-                      "qemu-img",
-                      "create",
-                      "-f",
-                      "qcow2",
-                      &format!("/var/lib/libvirt/images/{}.qcow2", name.clone()),
-                      &format!("{}M", disk_size_mb)
-                ])
-                .output()
-                .await?;
-
-            if output.status != ExitStatus::from_raw(0) {
-                return Err(Error::Allocation(String::from_utf8(output.stdout).unwrap()));
-            }
-        } else {
-            // It's local
-            let mut output = Command::new("stat")
-                .args([
-                      &format!("/var/lib/libvirt/images/{}", van.make_pretty_name().clone()),
-                ])
-                .output()
-                .await?;
-
-            if output.status != ExitStatus::from_raw(0) {
-                return Err(Error::MissingImage(van.name.clone()));
-            }
-
-            output = Command::new("qemu-img")
-                .args([
-                      "create",
-                      "-f",
-                      "qcow2",
-                      &format!("/var/lib/libvirt/images/{}.qcow2", name.clone()),
-                      &format!("{}M", disk_size_mb)
-                ])
-                .output()
-                .await?;
-
-            if output.status != ExitStatus::from_raw(0) {
-                return Err(Error::Allocation(String::from_utf8(output.stdout).unwrap()));
-            }
-        }
-
-        let uuid = uuid::Uuid::new_v4();
-        
-        // Let's get that XML ready
-        let mut buf: Vec<u8> = vec![];
-
-        crate::templates::vm_xml(
-            &mut buf,
-            name.clone(),
-            uuid.to_string(),
-            random_mac().clone(),
-            true,
-            max_mem.0,
-            max_cpus.0,
-            "blank".to_string()
-            )?;
-
-        let buf = String::from_utf8(buf).unwrap();
-
-        println!("{}", buf);
-
-        let dom: Domain = match &self.con {
-                Some(c) => {
-                    let dom = Domain::define_xml(&c, &buf)?;
-
-                    dom.create()?;
-
-                    dom
-                },
-                None => {
-                    return Err(Error::Connection("Connection was None".to_string()));
-                }
-            };
-
-        dom.try_into()
-    }
-
-}
-
-fn random_mac() -> String {
-    let mut addr = rand::thread_rng().gen::<[u8; 6]>();
-    addr[0] = (addr[0] | 2) & 0xfe;
-    mac_address::MacAddress::new(addr).to_string()
-}
-
-#[cfg(test)]
-mod test {
-    use super::*;
-    /// Kind of a stupid test, but just a sanity check to make sure that [ToString] impl up above
-    /// is still working
-    #[test]
-    fn addr_to_string() {
-        let a: Address = Address::IP("127.0.0.1".to_string());
-        let d: Address = Address::Domain("example.com".to_string());
-
-        assert_eq!(String::from("127.0.0.1"), a.to_string());
-        assert_eq!("example.com".to_string(), d.to_string());
-    }
-
-    #[test]
-    fn connect_to_house() {
-        let _: House = House::new("test:///default".to_string()).unwrap();
-    }
-
-    #[test]
-    fn list_inhabitants() {
-        let mut h: House = House::new("test:///default".to_string()).unwrap();
-
-        assert_eq!(h.inhabitants().unwrap().len(), 1);        
-    }
-
-}