use crate::codegen::dumper::internal_config::{ConfigDumpContent, DumperInternalConfig};
use crate::codegen::generator::acc::Acc;
use crate::codegen::generator::misc::path_texts::PathTexts;
use crate::codegen::generator::misc::target::TargetOrCommon;
use crate::utils::file_utils::create_dir_all_and_write;
use crate::utils::path_utils::path_to_string;
use anyhow::Context;
use convert_case::{Case, Casing};
use log::debug;
use pathdiff::diff_paths;
use serde::Serialize;
use std::path::Path;
use strum::IntoEnumIterator;
pub(super) mod internal_config;
#[derive(Clone)]
pub(crate) struct Dumper<'a> {
config: &'a DumperInternalConfig,
content: Option<ConfigDumpContent>,
name_prefix: String,
}
impl<'a> Dumper<'a> {
pub(crate) fn new(config: &'a DumperInternalConfig) -> Self {
Self {
config,
content: None,
name_prefix: "".to_string(),
}
}
pub(crate) fn with_content(&self, content: ConfigDumpContent) -> Self {
Self {
content: Some(content),
..self.clone()
}
}
pub(crate) fn with_add_name_prefix(&self, add_name_prefix: &str) -> Self {
Self {
name_prefix: format!("{}{}", self.name_prefix, add_name_prefix),
..self.clone()
}
}
pub(crate) fn dump<T: Serialize>(&self, name: &str, data: &T) -> anyhow::Result<()> {
if !self.is_enabled() {
return Ok(());
}
self.dump_str(name, &serde_json::to_string_pretty(data)?)
}
pub(crate) fn dump_path_texts(
&self,
partial_name: &str,
path_texts: &PathTexts,
base_dir: &Path,
) -> anyhow::Result<()> {
if !self.is_enabled() {
return Ok(());
}
for path_text in path_texts.0.iter() {
self.dump_str(
&format!(
"{partial_name}/{}",
path_to_string(
&diff_paths(&path_text.path, base_dir).context("cannot diff path")?
)?
),
&path_text.text.all_code(),
)?;
}
Ok(())
}
pub(crate) fn dump_acc(
&self,
partial_name: &str,
extension: &str,
acc: &Acc<Option<String>>,
) -> anyhow::Result<()> {
if !self.is_enabled() {
return Ok(());
}
for target in TargetOrCommon::iter() {
self.dump_str(
&format!("{partial_name}/{target}.{extension}"),
&acc[target].clone().unwrap_or_default(),
)?;
}
Ok(())
}
pub(crate) fn dump_str(&self, partial_name: &str, text: &str) -> anyhow::Result<()> {
if !self.is_enabled() {
return Ok(());
}
let name = format!("{}{}", self.name_prefix, partial_name);
let path = (self.config.dump_directory)
.join(self.content.unwrap().to_string().to_case(Case::Snake))
.join(&name);
debug!("Dumping {name} into {path:?}");
create_dir_all_and_write(path, text)
}
fn is_enabled(&self) -> bool {
self.config.dump_contents.contains(&self.content.unwrap())
}
}