even more review stuff

This commit is contained in:
steven-omaha
2023-01-31 16:42:16 +01:00
parent 71a977e60d
commit 8f4864c573
6 changed files with 114 additions and 4 deletions
+80 -1
View File
@@ -1,5 +1,9 @@
use std::collections::HashMap;
use std::fmt::Debug;
use std::fs::{read_to_string, File};
use std::io::Write;
use std::process::Command;
use std::rc::Rc;
use std::{collections::HashSet, process::ExitStatus};
use anyhow::{Context, Result};
@@ -12,18 +16,83 @@ pub(in crate::backend) type Text = &'static str;
pub(crate) trait Backend: Debug {
fn get_binary(&self) -> Text;
fn get_section(&self) -> Text;
fn get_switches_info(&self) -> Switches;
fn get_switches_install(&self) -> Switches;
fn get_switches_remove(&self) -> Switches;
fn get_managed_packages(&self) -> &HashSet<Package>;
fn get_switches_make_dependency(&self) -> Switches;
fn load(&mut self, groups: &HashSet<Group>);
fn get_managed_packages(&self) -> &HashSet<Package>;
/// Get all packages that are installed in the system.
fn get_all_installed_packages(&self) -> Result<HashSet<Package>>;
/// Get all packages that were installed in the system explicitly.
fn get_explicitly_installed_packages(&self) -> Result<HashSet<Package>>;
fn assign_group(&self, to_assign: Vec<(Package, Rc<Group>)>) {
let mut group_package_map = HashMap::new();
for (p, group) in to_assign {
if !group_package_map.contains_key(&group) {
group_package_map.insert(group.clone(), vec![]);
}
let inner = group_package_map.get_mut(&group).unwrap();
inner.push(p);
}
for vecs in group_package_map.values_mut() {
vecs.sort();
}
for (group, packages) in group_package_map {
let group_file_content = read_to_string(&group.path).unwrap();
let old_length = dbg!(group_file_content.len());
let mut must_write_section_header = false;
let mut lines = group_file_content.lines();
lines.find(|line| line.contains(&format!("[{}]", self.get_section())));
lines.next().unwrap();
let start_of_section = if let Some(start_of_section) = group_file_content.find() {
start_of_section
} else {
must_write_section_header = true;
old_length
};
dbg!(&start_of_section);
let mut new_file_content = group_file_content
.get(..start_of_section)
.unwrap()
.to_owned();
dbg!(&new_file_content);
todo!();
if dbg!(must_write_section_header) {
new_file_content.push_str(&format!("\n[{}]", self.get_section()));
}
for package in packages {
new_file_content.push_str(&format!("\n{package}"));
}
new_file_content.push('\n');
if old_length > start_of_section {
new_file_content.push_str(group_file_content.get(old_length..).unwrap());
}
let mut file = File::create(&group.path).unwrap();
write!(file, "{new_file_content}").unwrap();
}
}
/// Install the specified packages.
fn install_packages(&self, packages: &[Package]) -> Result<ExitStatus> {
let mut cmd = Command::new(self.get_binary());
@@ -35,6 +104,16 @@ pub(crate) trait Backend: Debug {
.with_context(|| format!("running command {cmd:?}"))
}
fn make_dependency(&self, packages: &[Package]) -> Result<ExitStatus> {
let mut cmd = Command::new(self.get_binary());
cmd.args(self.get_switches_make_dependency());
for p in packages {
cmd.arg(format!("{p}"));
}
cmd.status()
.with_context(|| format!("running command [{cmd:?}]"))
}
/// Remove the specified packages.
fn remove_packages(&self, packages: &[Package]) -> Result<ExitStatus> {
let mut cmd = Command::new(self.get_binary());