use indicatif::{MultiProgress, ProgressBar, ProgressStyle};
use mame_parser::file_handling::unpack_files;
use mame_parser::models::MameDataType;
use mame_parser::progress::{CallbackType, ProgressInfo, SharedProgressCallback};
use std::error::Error;
use std::path::Path;
use std::sync::Arc;
#[docify::export]
fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
let workspace_path = Path::new("playground");
let multi_progress = MultiProgress::new();
let progress_bars = Arc::new(
MameDataType::all_variants()
.iter()
.map(|&data_type| {
let progress_bar = multi_progress.add(ProgressBar::new(100));
progress_bar.set_style(
ProgressStyle::default_bar()
.template(&format!("{{spinner:.green}} [{{elapsed_precise}}] [{{bar:20.cyan/blue}}] {{pos}}/{{len}} ({{eta}}) {{msg}}"))
.progress_chars("#>-"),
);
(data_type, progress_bar)
})
.collect::<Vec<_>>(),
);
let shared_progress_callback: SharedProgressCallback = Arc::new(
move |data_type: MameDataType, progress_info: ProgressInfo| {
if let Some((_, progress_bar)) = progress_bars.iter().find(|(dt, _)| *dt == data_type) {
match progress_info.callback_type {
CallbackType::Progress => {
progress_bar.set_length(progress_info.total);
progress_bar.set_position(progress_info.progress);
}
CallbackType::Info => {
progress_bar.set_message(progress_info.message);
}
CallbackType::Finish => {
progress_bar.set_length(progress_info.total);
progress_bar.set_position(progress_info.progress);
progress_bar.finish_with_message(progress_info.message);
}
CallbackType::Error => {
progress_bar.finish_with_message(progress_info.message);
}
}
}
},
);
let handles = unpack_files(workspace_path, shared_progress_callback);
multi_progress.join().unwrap();
for handle in handles {
match handle.join().unwrap() {
Ok(path) => {
println!("Unpacked data file: {}", path.display());
}
Err(e) => {
eprintln!("Error during unpacking: {}", e);
}
}
}
Ok(())
}