From 9fa57aa1e7cb754b1f31d94b0881c0f2730c0693 Mon Sep 17 00:00:00 2001
From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com>
Date: Wed, 21 May 2025 08:32:25 -0400
Subject: [PATCH] test(config): verify fallback when no home

---
 libmarlin/src/config.rs       | 15 ++++++++------
 libmarlin/src/config_tests.rs | 39 +++++++++++++++++++++++++++++++++++
 2 files changed, 48 insertions(+), 6 deletions(-)

diff --git a/libmarlin/src/config.rs b/libmarlin/src/config.rs
index d0292a1..37ee117 100644
--- a/libmarlin/src/config.rs
+++ b/libmarlin/src/config.rs
@@ -35,12 +35,15 @@ impl Config {
         let digest = h.finish(); // 64-bit
         let file_name = format!("index_{digest:016x}.db");
 
-        if let Some(dirs) = ProjectDirs::from("io", "Marlin", "marlin") {
-            let dir = dirs.data_dir();
-            std::fs::create_dir_all(dir)?;
-            return Ok(Self {
-                db_path: dir.join(file_name),
-            });
+        // If HOME and XDG_DATA_HOME are missing we can't resolve an XDG path
+        if std::env::var_os("HOME").is_some() || std::env::var_os("XDG_DATA_HOME").is_some() {
+            if let Some(dirs) = ProjectDirs::from("io", "Marlin", "marlin") {
+                let dir = dirs.data_dir();
+                std::fs::create_dir_all(dir)?;
+                return Ok(Self {
+                    db_path: dir.join(file_name),
+                });
+            }
         }
 
         // 3) very last resort – workspace-relative DB
diff --git a/libmarlin/src/config_tests.rs b/libmarlin/src/config_tests.rs
index eac090f..2c79883 100644
--- a/libmarlin/src/config_tests.rs
+++ b/libmarlin/src/config_tests.rs
@@ -20,3 +20,42 @@ fn load_xdg_or_fallback() {
     let cfg = Config::load().unwrap();
     assert!(cfg.db_path.to_string_lossy().ends_with(".db"));
 }
+
+#[test]
+fn load_fallback_current_dir() {
+    // Save and clear HOME & XDG_DATA_HOME
+    let orig_home = env::var_os("HOME");
+    let orig_xdg = env::var_os("XDG_DATA_HOME");
+    env::remove_var("HOME");
+    env::remove_var("XDG_DATA_HOME");
+    env::remove_var("MARLIN_DB_PATH");
+
+    let cfg = Config::load().unwrap();
+
+    // Compute expected file name based on current directory hash
+    let cwd = env::current_dir().unwrap();
+    use std::collections::hash_map::DefaultHasher;
+    use std::hash::{Hash, Hasher};
+    let mut h = DefaultHasher::new();
+    cwd.hash(&mut h);
+    let digest = h.finish();
+    let expected_name = format!("index_{:016x}.db", digest);
+
+    assert_eq!(cfg.db_path, std::path::PathBuf::from(&expected_name));
+    assert!(cfg
+        .db_path
+        .file_name()
+        .unwrap()
+        .to_string_lossy()
+        .starts_with("index_"));
+
+    // Restore environment variables
+    match orig_home {
+        Some(val) => env::set_var("HOME", val),
+        None => env::remove_var("HOME"),
+    }
+    match orig_xdg {
+        Some(val) => env::set_var("XDG_DATA_HOME", val),
+        None => env::remove_var("XDG_DATA_HOME"),
+    }
+}