Add push method for Cleanup struct, slightly simplifying code in the

installer
This commit is contained in:
Nathan Fisher 2023-04-19 07:35:38 -04:00
parent bd5090069d
commit 1d595e0f1e
3 changed files with 25 additions and 11 deletions

View File

@ -5,12 +5,29 @@ use {
std::{error::Error, sync::mpsc::Sender},
};
#[derive(Debug)]
#[derive(Debug, Default)]
pub struct Cleanup {
hooks: Vec<Hooks>,
}
impl Cleanup {
pub fn new() -> Self {
Self::default()
}
pub fn push(&mut self, hook: Hooks) {
match hook {
Hooks::Info(_) | Hooks::User(_, _) | Hooks::Group(_, _) | Hooks::Pinstall(_) => {
self.hooks.push(hook)
}
Hooks::Man | Hooks::GlibSchema => {
if !self.hooks.contains(&hook) {
self.hooks.push(hook);
}
}
}
}
pub fn run(&self, sender: Sender<InstallMessage>) -> Result<(), Box<dyn Error>> {
self.hooks
.par_iter()

View File

@ -5,7 +5,7 @@ mod cli;
use {
clap::ArgMatches,
cli::cli,
hpk::{CreationError, Creator, Dependency, InstallMessage, Installer, Message, Specs, Version},
hpk::{Cleanup, CreationError, Creator, Dependency, InstallMessage, Installer, Message, Specs, Version},
indicatif::{ProgressBar, ProgressStyle},
ron::ser::{to_writer_pretty, PrettyConfig},
std::{
@ -136,7 +136,7 @@ fn install_local<P: AsRef<OsStr> + fmt::Display>(
}
}
});
let mut hooks = vec![];
let mut hooks = Cleanup::new();
installer.install(&mut hooks, sender)?;
match handle.join() {
Ok(package) => {

View File

@ -3,6 +3,7 @@ pub use error::Error;
use {
crate::{
Cleanup,
tar::{Archive, Node},
Entry, Group, Hooks, Package, Pinstall, User,
},
@ -86,7 +87,7 @@ impl<T: io::Read> Installer<T> {
pub fn install(
self,
hooks: &mut Vec<Hooks>,
hooks: &mut Cleanup,
sender: Sender<InstallMessage>,
) -> Result<Package, Error> {
let reader = Decoder::new(self.reader)?;
@ -136,9 +137,7 @@ impl<T: io::Read> Installer<T> {
if let Some(s) = node.header.prefix() {
if s.contains("/share/man/") {
let mut h = hooks.lock().unwrap();
if !h.contains(&Hooks::Man) {
h.push(Hooks::Man);
}
} else if s.contains("/share/info") {
hooks
.lock()
@ -146,11 +145,9 @@ impl<T: io::Read> Installer<T> {
.push(Hooks::Info(fpath.to_str().unwrap().to_string()));
} else if s.contains("/share/glib-2.0/schemas") {
let mut h = hooks.lock().unwrap();
if !h.contains(&Hooks::GlibSchema) {
h.push(Hooks::GlibSchema);
}
}
}
// Match up a package entry with a tar node
let entry = package
.plist
@ -242,7 +239,7 @@ fn pop_appstream(archive: &mut Archive, db_pkgdir: &Path) -> Result<(), Error> {
fn pop_pinstall(
archive: &mut Archive,
hooks: &mut Vec<Hooks>,
hooks: &mut Cleanup,
pkgname: &str,
root: &Path,
) -> Result<(), Error> {