From ef261a4c93ca1f60b593256ea6a87ec60537933d Mon Sep 17 00:00:00 2001
From: Carlos Zamora <uriel.zamora@encora.com>
Date: Sun, 22 Jun 2025 03:00:40 -0600
Subject: [PATCH] fix(ui): update directory-picker logic to compute full file
 paths

---
 src/server/templates/components/result.jinja | 38 ++++++++++++++++----
 1 file changed, 32 insertions(+), 6 deletions(-)

diff --git a/src/server/templates/components/result.jinja b/src/server/templates/components/result.jinja
index 151bc02f..55c1f533 100644
--- a/src/server/templates/components/result.jinja
+++ b/src/server/templates/components/result.jinja
@@ -1,22 +1,48 @@
 <script>
-    function getFileName(line) {
-        // Skips "|", "└", "├" found in file tree
-        const index = line.search(/[a-zA-Z0-9]/);
-        return line.substring(index).trim();
+    function getFileName(element) {
+        const indentSize = 4;
+        let path = "";
+        let prevIndentLevel = null;
+
+        while (element) {
+            const line = element.textContent;
+            const index = line.search(/[a-zA-Z0-9_.-]/);
+            const indentLevel = index / indentSize;
+
+            // Stop when we reach or go above the top-level directory
+            if (indentLevel <= 1) {
+                break;
+            }
+
+            // Only include directories that are one level above the previous
+            if (prevIndentLevel === null || indentLevel === prevIndentLevel - 1) {
+                const fileName = line.substring(index).trim();
+                path = fileName + path;
+                prevIndentLevel = indentLevel;
+            }
+
+            element = element.previousElementSibling;
+        }
+
+        return path;
     }
 
     function toggleFile(element) {
         const patternInput = document.getElementById("pattern");
         const patternFiles = patternInput.value ? patternInput.value.split(",").map(item => item.trim()) : [];
 
-        if (element.textContent.includes("Directory structure:")) {
+        const directoryContainer = document.getElementById("directory-structure-container");
+        const treeLineElements = Array.from(directoryContainer.children).filter(child => child.tagName === "PRE");
+
+        // Skip the first two tree lines (header and repository name)
+        if (treeLineElements[0] === element || treeLineElements[1] === element) {
             return;
         }
 
         element.classList.toggle('line-through');
         element.classList.toggle('text-gray-500');
 
-        const fileName = getFileName(element.textContent);
+        const fileName = getFileName(element);
         const fileIndex = patternFiles.indexOf(fileName);
 
         if (fileIndex !== -1) {