mould-rs/src/format/env.rs

use super::{ConfigItem, FormatHandler, ItemStatus, ValueType, PathSegment};
use std::fs;
use std::io::Write;
use std::path::Path;

pub struct EnvHandler;

impl FormatHandler for EnvHandler {
    fn parse(&self, path: &Path) -> anyhow::Result<Vec<ConfigItem>> {
        let content = fs::read_to_string(path)?;
        let mut vars = Vec::new();

        for line in content.lines() {
            let line = line.trim();
            if line.is_empty() || line.starts_with('#') {
                continue; // Skip comments and empty lines
            }

            if let Some((key, val)) = line.split_once('=') {
                let parsed_val = val.trim().trim_matches('"').trim_matches('\'').to_string();
                let key_str = key.trim().to_string();
                vars.push(ConfigItem {
                    key: key_str.clone(),
                    path: vec![PathSegment::Key(key_str)],
                    value: Some(parsed_val.clone()),
                    template_value: Some(parsed_val.clone()),
                    default_value: Some(parsed_val),
                    depth: 0,
                    is_group: false,
                    status: ItemStatus::Present,
                    value_type: ValueType::String,
                });
            }
        }

        Ok(vars)
    }

    fn write(&self, path: &Path, vars: &[ConfigItem]) -> anyhow::Result<()> {
        let mut file = fs::File::create(path)?;
        for var in vars {
            if !var.is_group {
                let val = var.value.as_deref()
                    .or(var.template_value.as_deref())
                    .unwrap_or("");
                writeln!(file, "{}={}", var.key, val)?;
            }
        }
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::io::Write;
    use tempfile::NamedTempFile;

    #[test]
    fn test_parse_env_example() {
        let mut file = NamedTempFile::new().unwrap();
        writeln!(
            file,
            "# A comment\nKEY1=value1\nKEY2=\"value2\"\nKEY3='value3'\nEMPTY="
        )
        .unwrap();

        let handler = EnvHandler;
        let vars = handler.parse(file.path()).unwrap();
        assert_eq!(vars.len(), 4);
        assert_eq!(vars[0].key, "KEY1");
        assert_eq!(vars[0].value.as_deref(), Some("value1"));
        assert_eq!(vars[1].key, "KEY2");
        assert_eq!(vars[1].value.as_deref(), Some("value2"));
        assert_eq!(vars[2].key, "KEY3");
        assert_eq!(vars[2].value.as_deref(), Some("value3"));
        assert_eq!(vars[3].key, "EMPTY");
        assert_eq!(vars[3].value.as_deref(), Some(""));
    }

    #[test]
    fn test_merge_env() {
        let handler = EnvHandler;

        let mut env_file = NamedTempFile::new().unwrap();
        writeln!(env_file, "KEY1=custom1\nKEY3=custom3").unwrap();
        let mut vars = handler.parse(env_file.path()).unwrap(); // Active vars

        let mut example_file = NamedTempFile::new().unwrap();
        writeln!(example_file, "KEY1=default1\nKEY2=default2").unwrap();

        handler.merge(example_file.path(), &mut vars).unwrap(); // Merge template into active

        // Should preserve order of active, then append template
        assert_eq!(vars.len(), 3);
        
        // Active key that exists in template
        assert_eq!(vars[0].key, "KEY1");
        assert_eq!(vars[0].value.as_deref(), Some("custom1")); // Keeps active value
        assert_eq!(vars[0].template_value.as_deref(), Some("default1")); // Gets template default
        assert_eq!(vars[0].status, ItemStatus::Modified);

        // Active key that DOES NOT exist in template
        assert_eq!(vars[1].key, "KEY3");
        assert_eq!(vars[1].value.as_deref(), Some("custom3"));
        assert_eq!(vars[1].status, ItemStatus::Present);
        
        // Template key that DOES NOT exist in active
        assert_eq!(vars[2].key, "KEY2");
        assert_eq!(vars[2].value.as_deref(), None); // Missing from active
        assert_eq!(vars[2].template_value.as_deref(), Some("default2"));
        assert_eq!(vars[2].status, ItemStatus::MissingFromActive);
    }

    #[test]
    fn test_write_env() {
        let file = NamedTempFile::new().unwrap();
        let vars = vec![ConfigItem {
            key: "KEY1".to_string(),
            path: vec![PathSegment::Key("KEY1".to_string())],
            value: Some("value1".to_string()),
            template_value: None,
            default_value: None,
            depth: 0,
            is_group: false,
            status: ItemStatus::Present,
            value_type: ValueType::String,
        }];

        let handler = EnvHandler;
        handler.write(file.path(), &vars).unwrap();

        let content = fs::read_to_string(file.path()).unwrap();
        assert_eq!(content.trim(), "KEY1=value1");
    }
}