use std::io::Write;
use std::{fmt, io, path};
use console::style;
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: impl Into<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()
)) {
let extension = extension.into();
let mut fd = std::fs::File::create(fullpath)?;
let content = match extension.to_string().parse::<Extension>() {
| Ok(Extension::JSON) => {
serde_json::to_string_pretty(&data)
.map_err(|err| io::Error::new(io::ErrorKind::Other, err))?
}
| Ok(Extension::TOML) => {
serde_toml::to_string_pretty(&data)
.map_err(|err| io::Error::new(io::ErrorKind::Other, err))?
}
| Ok(Extension::YAML) => {
serde_yaml::to_string(&data)
.map_err(|err| io::Error::new(io::ErrorKind::Other, err))?
}
| _ => unimplemented!(),
};
write!(fd, "{}", content)?;
}
Ok(data)
}