use std::{
fs::{self},
path::Path,
};
use anyhow::{Context, Result};
use glob::{MatchOptions, Pattern};
use poe_data_tools::fs::{FS, FileSystem};
pub fn extract_files(fs: &mut FS, patterns: &[Pattern], output_folder: &Path) -> Result<()> {
let filenames = fs
.list()
.filter(|filename| {
patterns.iter().any(|pattern| {
pattern.matches_with(
filename,
MatchOptions {
require_literal_separator: true,
..Default::default()
},
)
})
})
.collect::<Vec<_>>();
fs.batch_read(&filenames)
.filter_map(|(path, res)| match res {
Ok(b) => Some((path, b)),
Err(e) => {
log::error!("Failed to extract file: {:?}: {:?}", path, e);
None
}
})
.map(|(filename, contents)| -> Result<_, anyhow::Error> {
let out_filename = output_folder.join(filename.as_ref());
fs::create_dir_all(out_filename.parent().unwrap()).with_context(|| {
format!(
"Failed to create folder: {:?}",
out_filename.parent().unwrap()
)
})?;
fs::write(&out_filename, &contents)
.with_context(|| format!("Failed to write file: {:?}", out_filename))?;
Ok(filename)
})
.for_each(|result| match result {
Ok(filename) => log::info!("Extracted file: {}", filename),
Err(e) => log::error!("Failed to extract file: {:?}", e),
});
Ok(())
}