implement section

This commit is contained in:
timeshifter
2023-01-06 14:14:21 +01:00
parent cb5c53c1c5
commit 0ed6359293
3 changed files with 69 additions and 12 deletions
+30 -12
View File
@@ -1,36 +1,33 @@
use std::collections::HashSet; use std::collections::HashSet;
use std::fs::File; use std::fs::{read_to_string, File};
use std::hash::Hash; use std::hash::Hash;
use std::io::{BufRead, BufReader}; use std::io::{BufRead, BufReader};
use std::path::Path;
use anyhow::{anyhow, Context, Result}; use anyhow::{anyhow, Context, Result};
use crate::section::Section;
use crate::Package; use crate::Package;
#[derive(Debug)] #[derive(Debug)]
pub struct Group { pub struct Group {
pub name: String, pub name: String,
pub packages: HashSet<Package>, pub sections: HashSet<Section>,
} }
impl Group { impl Group {
pub fn load_from_dir() -> Result<HashSet<Self>> { pub fn load_from_dir() -> Result<HashSet<Self>> {
let mut result = HashSet::new(); let mut result = HashSet::new();
let path = crate::path::get_pacdef_group_dir().context("getting pacdef group dir")?; let path = crate::path::get_pacdef_group_dir().context("getting pacdef group dir")?;
for entry in path.read_dir().context("reading group dir")? { for entry in path.read_dir().context("reading group dir")? {
let file = entry.context("getting a file")?; let file = entry.context("getting a file")?;
let name = file.file_name(); let name = file.file_name();
let f = File::open(file.path()).context("reading the file")?;
let reader = BufReader::new(f);
let packages = Package::from_lines(reader.lines()); let group = Group::try_from(name)?;
result.insert(Group { result.insert(group);
name: name
.into_string()
.map_err(|e| anyhow!("could not get group name, {e:?}"))?,
packages,
});
} }
Ok(result) Ok(result)
} }
} }
@@ -58,10 +55,31 @@ impl Hash for Group {
impl PartialEq for Group { impl PartialEq for Group {
fn eq(&self, other: &Self) -> bool { fn eq(&self, other: &Self) -> bool {
self.name == other.name && self.packages == other.packages self.name == other.name
} }
} }
impl Eq for Group { impl Eq for Group {
fn assert_receiver_is_total_eq(&self) {} fn assert_receiver_is_total_eq(&self) {}
} }
impl<P> From<P> for Group
where
P: AsRef<Path>,
{
fn from(p: P) -> Self {
let path = p.as_ref();
let content = read_to_string(path).unwrap();
let name = path.file_name().unwrap().to_string_lossy().to_string();
let mut lines = content.lines().peekable();
let mut sections = HashSet::new();
while lines.peek().is_some() {
let section = Section::from_lines(&mut lines);
sections.insert(section);
}
Self { name, sections }
}
}
+1
View File
@@ -7,6 +7,7 @@ mod env;
mod group; mod group;
mod package; mod package;
mod path; mod path;
mod section;
mod ui; mod ui;
pub use crate::core::Pacdef; pub use crate::core::Pacdef;
+38
View File
@@ -0,0 +1,38 @@
use std::{collections::HashSet, hash::Hash};
use crate::Package;
#[derive(Debug, Eq)]
pub struct Section {
pub name: String,
pub packages: HashSet<Package>,
}
impl Section {
pub(crate) fn new(name: String, packages: HashSet<Package>) -> Self {
Self { name, packages }
}
pub fn from_lines<'a>(iter: &mut impl Iterator<Item = &'a str>) -> Self {
let name = iter.find(|line| line.starts_with('[')).unwrap().to_string();
let packages = iter
.take_while(|line| !line.starts_with('['))
.filter(|line| !line.contains(char::is_alphabetic))
.map(Package::from)
.collect();
Self::new(name, packages)
}
}
impl Hash for Section {
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.name.hash(state);
}
}
impl PartialEq for Section {
fn eq(&self, other: &Self) -> bool {
self.name == other.name
}
}