use std::fs::File;
use std::io::Write;
use std::path::{Path, PathBuf};
use crate::archiver::{ArchiveEntry, ToteArchiver};
use crate::{Error, Result};
pub(super) struct Archiver {}
impl ToteArchiver for Archiver {
fn perform(
&self,
file: File,
targets: &[PathBuf],
config: &crate::ArchiveConfig,
) -> Result<Vec<ArchiveEntry>> {
let mut builder = ar::Builder::new(file);
let mut errs = vec![];
let mut entries = vec![];
for tp in targets {
for entry in config.iter(tp) {
let path = entry.into_path();
if !path.is_file() {
continue;
}
entries.push(ArchiveEntry::from(&path));
let dest_path = config.path_in_archive(&path);
if let Err(e) = process_file(&mut builder, &path, &dest_path) {
errs.push(e);
}
}
}
Error::error_or(entries, errs)
}
fn enable(&self) -> bool {
true
}
}
fn process_file<W: Write>(
builder: &mut ar::Builder<W>,
target: &Path,
dest_path: &Path,
) -> Result<()> {
let mut file = File::open(target).map_err(Error::IO)?;
let name = dest_path.to_string_lossy().into_owned();
builder
.append_file(name.as_bytes(), &mut file)
.map_err(Error::IO)
}
#[cfg(test)]
mod tests {
use crate::archiver::test_support::targets;
use std::path::PathBuf;
fn run_test<F>(f: F)
where
F: FnOnce() -> PathBuf,
{
let result = std::panic::catch_unwind(std::panic::AssertUnwindSafe(f));
match result {
Ok(path) => teardown(path),
Err(err) => std::panic::resume_unwind(err),
}
}
#[test]
fn test_tar() {
run_test(|| {
let config = crate::ArchiveConfig::builder()
.dest("results/test.ar")
.overwrite(true)
.build();
let v = targets();
let result = crate::archive(&v, &config);
let path = PathBuf::from("results/test.ar");
if let Err(e) = result {
panic!("{:?}", e);
}
assert!(result.is_ok());
assert!(path.exists());
path
});
}
fn teardown(path: PathBuf) {
let _ = std::fs::remove_file(path);
}
}