From 1c4163e05071f31d8a214333b4c5c58581e975c0 Mon Sep 17 00:00:00 2001
From: root <root@DESKTOP-V9DGVF3.localdomain>
Date: Tue, 20 Jan 2026 22:15:19 +0400
Subject: [PATCH] fix: ensure path prefix search respects directory boundaries

Added a trailing slash to the SQL LIKE pattern to prevent partial directory matching (e.g., 'proj' matching 'project'). Added a regression test.
---
 src/core/db.rs         | 54 +++++++++++++++++++++++++++++++++++++++++-
 src/core/repro_test.rs | 48 +++++++++++++++++++++++++++++++++++++
 2 files changed, 101 insertions(+), 1 deletion(-)
 create mode 100644 src/core/repro_test.rs

diff --git a/src/core/db.rs b/src/core/db.rs
index 080e586..41e3379 100644
--- a/src/core/db.rs
+++ b/src/core/db.rs
@@ -159,7 +159,14 @@ impl Database {
         limit: usize,
     ) -> Result<Vec<SearchResult>> {
         let path_prefix_str = path_prefix.to_string_lossy();
-        let like_pattern = format!("{}%", path_prefix_str);
+        
+        // Ensure the prefix ends with a directory separator to avoid partial directory matching
+        // e.g. "/home/user/proj" should become "/home/user/proj/" to avoid matching "/home/user/project"
+        let like_pattern = if path_prefix_str.ends_with(std::path::MAIN_SEPARATOR) {
+            format!("{}%", path_prefix_str)
+        } else {
+            format!("{}{}%", path_prefix_str, std::path::MAIN_SEPARATOR)
+        };
 
         let mut stmt = self.conn.prepare(
             r"SELECT c.id, c.file_id, f.path, c.content, c.start_line, c.end_line, c.embedding
@@ -303,3 +310,48 @@ fn cosine_similarity(a: &[f32], b: &[f32]) -> f32 {
 
     (dot / (norm_a.sqrt() * norm_b.sqrt())) as f32
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use tempfile::tempdir;
+
+    #[test]
+    fn test_path_prefix_search_bug_reproduction() -> Result<()> {
+        let dir = tempdir()?;
+        let db_path = dir.path().join("test.db");
+        let db = Database::new(&db_path)?;
+
+        // Setup paths
+        let base_path = PathBuf::from("/home/user");
+        let path1 = base_path.join("proj/file1.txt");
+        let path2 = base_path.join("project/file2.txt");
+
+        // Insert files
+        let id1 = db.insert_file(&path1, "hash1")?;
+        let id2 = db.insert_file(&path2, "hash2")?;
+
+        // Insert dummy chunks
+        let embedding = vec![0.1; 128];
+        db.insert_chunk(id1, 0, "content1", 1, 1, &embedding)?;
+        db.insert_chunk(id2, 0, "content2", 1, 1, &embedding)?;
+
+        // Search with prefix "/home/user/proj"
+        // This should match ONLY "/home/user/proj/file1.txt"
+        // But due to the bug, it likely matches "/home/user/project/file2.txt" as well
+        let prefix = base_path.join("proj");
+        let results = db.search_similar(&embedding, &prefix, 10)?;
+
+        let found_paths: Vec<String> = results.iter()
+            .map(|r| r.path.to_string_lossy().to_string())
+            .collect();
+            
+        println!("Found paths: {:?}", found_paths);
+        
+        // Assert that we DO NOT match "project" when searching for "proj"
+        let matched_project = found_paths.iter().any(|p| p.contains("project"));
+        assert!(!matched_project, "Should not match sibling directory 'project' when searching for 'proj'");
+        
+        Ok(())
+    }
+}
diff --git a/src/core/repro_test.rs b/src/core/repro_test.rs
new file mode 100644
index 0000000..ff45588
--- /dev/null
+++ b/src/core/repro_test.rs
@@ -0,0 +1,48 @@
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use tempfile::tempdir;
+
+    #[test]
+    fn test_path_prefix_search_bug_reproduction() -> Result<()> {
+        let dir = tempdir()?;
+        let db_path = dir.path().join("test.db");
+        let db = Database::new(&db_path)?;
+
+        // Setup paths
+        let base_path = PathBuf::from("/home/user");
+        let path1 = base_path.join("proj/file1.txt");
+        let path2 = base_path.join("project/file2.txt");
+
+        // Insert files
+        let id1 = db.insert_file(&path1, "hash1")?;
+        let id2 = db.insert_file(&path2, "hash2")?;
+
+        // Insert dummy chunks
+        let embedding = vec![0.1; 128];
+        db.insert_chunk(id1, 0, "content1", 1, 1, &embedding)?;
+        db.insert_chunk(id2, 0, "content2", 1, 1, &embedding)?;
+
+        // Search with prefix "/home/user/proj"
+        // This should match ONLY "/home/user/proj/file1.txt"
+        // But due to the bug, it likely matches "/home/user/project/file2.txt" as well
+        let prefix = base_path.join("proj");
+        let results = db.search_similar(&embedding, &prefix, 10)?;
+
+        // If bug exists, we might get 2 results instead of 1
+        // Or specific result path matching expectations
+        
+        let found_paths: Vec<String> = results.iter()
+            .map(|r| r.path.to_string_lossy().to_string())
+            .collect();
+            
+        println!("Found paths: {:?}", found_paths);
+        
+        // Assert that we DO NOT match "project" when searching for "proj"
+        let matched_project = found_paths.iter().any(|p| p.contains("project"));
+        assert!(!matched_project, "Should not match sibling directory 'project' when searching for 'proj'");
+        
+        Ok(())
+    }
+}