wng-lib 0.1.3

use crate::error;
use crate::errors::*;
use std::io::Write;
use std::path::Path;
use std::{fs, fs::File};

pub fn add_dep(dependency: &str, messages: bool) -> crate::Result<()> {
    if !Path::new("project.toml").exists() {
        return Err(error!("Project.toml not found in the current directory."));
    }

    let mut tomlized: toml::Value = toml::from_str(&fs::read_to_string("project.toml")?)?;

    if !tomlized.as_table().unwrap().contains_key("project") {
        return Err(error!("Missing key in project.toml: `project`"));
    } else if !tomlized["project"]
        .as_table()
        .unwrap()
        .contains_key("dependencies")
    {
        return Err(error!("Missing key in .wng.config: `project.dependencies`"));
    }

    let deps = match tomlized["project"]["dependencies"].as_array_mut() {
        Some(array) => array,
        None => return Err(error!("Cannot find field `dependencies` in `project.toml`")),
    };

    let mut already = false;
    for dep in deps.clone() {
        if let toml::Value::String(s) = dep {
            if s == dependency {
                already = true;
                break;
            }
        }
    }

    if !already {
        deps.push(toml::Value::String(dependency.to_owned()));
    } else {
        if messages {
            println!("Dependency `{}` already is in `project.toml`.", dependency);
        }
        return Ok(());
    }

    let mut file = File::create("./project.toml")?;

    file.write_all(toml::to_string(&tomlized)?.as_bytes())?;

    if messages {
        println!("[+] Successfully added dependency `{}`", dependency);
    }

    Ok(())
}

pub fn remove_dep(dependency: &str, messages: bool) -> crate::Result<()> {
    if !Path::new("project.toml").exists() {
        return Err(error!("Project.toml not found in the current directory."));
    }

    let mut tomlized: toml::Value = toml::from_str(&fs::read_to_string("project.toml")?)?;

    if !tomlized.as_table().unwrap().contains_key("project") {
        return Err(error!("Missing key in project.toml: `project`"));
    } else if !tomlized["project"]
        .as_table()
        .unwrap()
        .contains_key("dependencies")
    {
        return Err(error!("Missing key in .wng.config: `project.dependencies`"));
    }

    let deps = match tomlized["project"]["dependencies"].as_array_mut() {
        Some(array) => array,
        None => return Err(error!("Cannot find field `dependencies` in `project.toml`")),
    };

    for i in 0..deps.len() {
        if let toml::Value::String(s) = &deps[i] {
            if s == dependency {
                deps.remove(i);
                break;
            }
        }
    }

    let mut file = File::create("./project.toml")?;

    file.write_all(toml::to_string(&tomlized)?.as_bytes())?;

    if messages {
        println!("[+] Successfully added dependency `{}`", dependency);
    }

    Ok(())
}