use std::io::Write;
use std::{fmt, io, path};
use console::style;
use crate::error::mapped_to_io_other_err;
use crate::Extension;
pub fn load_or_prompt<T>(
directory: impl AsRef<path::Path>,
filename: impl fmt::Display,
extension: impl Into<Extension>,
) -> io::Result<T>
where
T: fmt::Debug,
T: serde::Serialize + serde::de::DeserializeOwned,
T: lexa_prompt::Prompt,
{
let extension = extension.into();
let fullpath = path::Path::new(directory.as_ref()).join(format!("{filename}.{extension}"));
if !fullpath.exists() {
return save(fullpath, extension);
}
crate::load(directory, filename, extension).or_else(|_| save(fullpath, extension))
}
fn save<T>(fullpath: impl AsRef<path::Path>, extension: Extension) -> io::Result<T>
where
T: fmt::Debug,
T: serde::Serialize + serde::de::DeserializeOwned,
T: lexa_prompt::Prompt,
{
log::warn!(
"Les données du fichier « {} » sont manquantes ou corrompues.",
style(fullpath.as_ref().display()).yellow()
);
println!();
println!(
"Création du formulaire pour le fichier « {} »...",
style(fullpath.as_ref().display()).yellow()
);
println!();
let data = T::prompt().map_err(|err| io::Error::new(io::ErrorKind::InvalidData, err.to_string()))?;
log::debug!(" Les données du formulaire sont\n{:#?}", &data);
if !lexa_prompt::confirm(format!(
"Sauvegarder le résultat dans « {} »",
style(fullpath.as_ref().display()).yellow()
)) {
return Ok(data);
}
let mut fd = std::fs::File::create(fullpath)?;
let content = match extension {
| Extension::JSON => serde_json::to_string_pretty(&data).map_err(mapped_to_io_other_err)?,
| Extension::TOML => serde_toml::to_string_pretty(&data).map_err(mapped_to_io_other_err)?,
| Extension::YAML => serde_yaml::to_string(&data).map_err(mapped_to_io_other_err)?,
| _ => unreachable!(),
};
write!(fd, "{}", content)?;
Ok(data)
}