Fix directories not working in user prompt.

Fixes #273

Author: ericdallo

CHANGELOG.md

@@ -3,6 +3,7 @@
 ## Unreleased
 
 - Improve error handling on chat messages, avoiding stuck and losing chat history. #272
+- Fix directories not working in user prompt. #273
 
 ## 0.92.2
 

src/eca/features/context.clj

@@ -126,11 +126,11 @@
   "Extract all contexts (@something) and refine them.
    Parse lines if present in contexts like @/path/to/file:L1-L4"
   [prompt db]
-  (let [;; Capture @<path> with optional :L<start>-L<end>
+  (let [ ;; Capture @<path> with optional :L<start>-L<end>
         context-pattern #"@([^\s:]+)(?::L(\d+)-L(\d+))?"
         matches (re-seq context-pattern prompt)
         raw-contexts (mapv (fn [[_ path s e]]
-                             (assoc-some {:type "file"
+                             (assoc-some {:type (if (fs/directory? path) "directory" "file")
                                           :path path}
                                          :lines-range (when (and s e)
                                                         {:start (Integer/parseInt s)

test/eca/features/context_test.clj

@@ -321,22 +321,22 @@
                 :path b-file
                 :content b-content}])
              (#'f.context/parse-agents-file a-file)))))))
-  (testing "Missing referenced file is ignored"
-    (let [a-file (h/file-path "/fake/AGENTS.md")
-          a-content (multi-str
-                     "- do foo"
-                     "- check @missing.md for missing things")
-          missing-file (h/file-path "/fake/missing.md")]
-      (with-redefs [llm-api/refine-file-context (fn [p _l]
-                                                  (condp = p
-                                                    a-file a-content
-                                                    missing-file nil))]
-        (is (match?
-             (m/in-any-order
-              [{:type :agents-file
-                :path a-file
-                :content a-content}])
-             (#'f.context/parse-agents-file a-file))))))
+(testing "Missing referenced file is ignored"
+  (let [a-file (h/file-path "/fake/AGENTS.md")
+        a-content (multi-str
+                   "- do foo"
+                   "- check @missing.md for missing things")
+        missing-file (h/file-path "/fake/missing.md")]
+    (with-redefs [llm-api/refine-file-context (fn [p _l]
+                                                (condp = p
+                                                  a-file a-content
+                                                  missing-file nil))]
+      (is (match?
+           (m/in-any-order
+            [{:type :agents-file
+              :path a-file
+              :content a-content}])
+           (#'f.context/parse-agents-file a-file))))))
 
 (deftest contexts-str-from-prompt-test
   (testing "not context mention"
@@ -359,4 +359,18 @@
              :path "/path/to/file"
              :lines-range {:start 1 :end 4}
              :content "Some content"}]
-           (f.context/contexts-str-from-prompt "check @/path/to/file:L1-L4" (h/db)))))))
+           (f.context/contexts-str-from-prompt "check @/path/to/file:L1-L4" (h/db))))))
+  (testing "directory"
+    (with-redefs [fs/readable? (constantly true)
+                  fs/directory? (fn [path]
+                                  (= "/path/to/folder" path))
+                  fs/glob (constantly ["/path/to/folder/foo.txt" "/path/to/folder/bar.s"])
+                  llm-api/refine-file-context (constantly "Some content")]
+      (is (match?
+           [{:type :file
+             :path "/path/to/folder/foo.txt"
+             :content "Some content"}
+            {:type :file
+             :path "/path/to/folder/bar.s"
+             :content "Some content"}]
+           (f.context/contexts-str-from-prompt "check @/path/to/folder" (h/db)))))))