use prometeu_hal::telemetry::CertificationConfig;

pub fn load_cap_config(path: &str) -> Option<CertificationConfig> {
    let content = std::fs::read_to_string(path).ok()?;
    let mut config = CertificationConfig { enabled: true, ..Default::default() };

    for line in content.lines() {
        let line = line.trim();
        if line.is_empty() || line.starts_with('#') {
            continue;
        }
        let parts: Vec<&str> = line.split('=').collect();
        if parts.len() != 2 {
            continue;
        }
        let key = parts[0].trim();
        let val = parts[1].trim();

        match key {
            "cycles_budget" => config.cycles_budget_per_frame = val.parse().ok(),
            "max_syscalls" => config.max_syscalls_per_frame = val.parse().ok(),
            "max_host_cpu_us" => config.max_host_cpu_us_per_frame = val.parse().ok(),
            _ => {}
        }
    }

    Some(config)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_load_cap_config() {
        let content = "cycles_budget=500\nmax_syscalls=10\n# comment\nmax_host_cpu_us=2000";
        let path = "test_cap.cfg";
        std::fs::write(path, content).unwrap();

        let config = load_cap_config(path).unwrap();
        assert!(config.enabled);
        assert_eq!(config.cycles_budget_per_frame, Some(500));
        assert_eq!(config.max_syscalls_per_frame, Some(10));
        assert_eq!(config.max_host_cpu_us_per_frame, Some(2000));

        std::fs::remove_file(path).unwrap();
    }

    #[test]
    fn test_load_cap_config_not_found() {
        let config = load_cap_config("non_existent.cfg");
        assert!(config.is_none());
    }
}