core: Allow :all on one line require

yaml · Jul 4, 2024 · 64af767 · 64af767
1 parent 1a40afa
commit 64af767
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/core/src/yamlscript/transformers.clj b/core/src/yamlscript/transformers.clj
@@ -126,9 +126,10 @@
                      [(conj args AS alias) (drop 2 rhs)]
                      [args rhs])
         args (if (seq rhs)
-               (if (every? :Sym rhs)
-                 (conj args REFER (Vec rhs))
-                 (die "Invalid 'require' arguments"))
+               (cond
+                 (every? :Sym rhs) (conj args REFER (Vec rhs))
+                 (= rhs [{:Key :all}]) (conj args REFER (first rhs))
+                 :else (die "Invalid 'require' arguments"))
                args)]
     args))
 

diff --git a/core/test/transformer.yaml b/core/test/transformer.yaml
@@ -94,6 +94,14 @@
       '[foo.fff :refer :all])
 
 
+- name: 'require ns: :all'
+  yamlscript: |
+    !yamlscript/v0
+    require foo::bar: :all
+  clojure: |
+    (require '[foo.bar :refer :all])
+
+
 - name: Anonymous function with name
   yamlscript: |
     !yamlscript/v0