You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
2.2 KiB
Rust
62 lines
2.2 KiB
Rust
use crate::{err_unrec, inf};
|
|
use std::fs::File;
|
|
use std::io::{Error, Write};
|
|
use toml_edit::{value, Document};
|
|
use crate::mods::strs::{err_rec};
|
|
|
|
pub fn rem_pkg(pkgs: &Vec<String>) {
|
|
let file = format!("{}/.local/ame/aurPkgs.db", std::env::var("HOME").unwrap());
|
|
let database = String::new();
|
|
if std::path::Path::new(&file).exists() {
|
|
let database = std::fs::read_to_string(&file).expect("Can't Open Database");
|
|
} else {
|
|
err_rec(String::from("Database wasn't found, creating new one"));
|
|
let dbFile = File::create(&file);
|
|
let database = String::new();
|
|
}
|
|
|
|
let mut update_database = database;
|
|
for i in pkgs {
|
|
if update_database.contains(i) {
|
|
let results = raur::search(&i);
|
|
for res in &results {
|
|
let database_entry = format!(
|
|
"{} = {{ name = \"{}\", version = \"{}\"}}\n",
|
|
&res[0].name, &res[0].name, &res[0].version
|
|
);
|
|
update_database = format!("{}", update_database.replace(&database_entry, ""));
|
|
}
|
|
}
|
|
}
|
|
let file_as_path = File::create(std::path::Path::new(&file)).unwrap();
|
|
let db_update_res = write!(&file_as_path, "{}", update_database);
|
|
match db_update_res {
|
|
Ok(_) => inf(format!("Database update successful")),
|
|
Err(_) => err_unrec(format!("Couldn't update database")),
|
|
}
|
|
}
|
|
|
|
pub fn add_pkg(from_repo: bool, pkg: &str) -> Result<(), Error> {
|
|
let file = format!("{}/.local/ame/aurPkgs.db", std::env::var("HOME").unwrap());
|
|
let database = std::fs::read_to_string(&file).expect("cant open database");
|
|
let mut file_as_path = File::create(std::path::Path::new(&file))?;
|
|
|
|
let mut db_parsed = database.parse::<Document>().expect("invalid Database");
|
|
if from_repo == false {
|
|
let results = raur::search(&pkg);
|
|
for res in &results {
|
|
for r in res {
|
|
db_parsed[&r.name]["name"] = value(&r.name);
|
|
db_parsed[&r.name]["version"] = value(&r.version);
|
|
}
|
|
}
|
|
} else {
|
|
db_parsed[&pkg]["name"] = value(pkg);
|
|
db_parsed[&pkg]["version"] = value(pkg);
|
|
}
|
|
file_as_path
|
|
.write_all(format!("{}", db_parsed).as_bytes())
|
|
.unwrap();
|
|
Ok(())
|
|
}
|