Adding in a Threading macro (->)

paultag · Mar 13, 2013 · daa6443 · daa6443
1 parent d68c114
commit daa6443
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/hy/core/bootstrap.py b/hy/core/bootstrap.py
@@ -45,3 +45,15 @@ def cond_macro(tree):
         ret = n
 
     return root
+
+
+@macro("_>")
+def threading_macro(tree):
+    tree.pop(0)  # ->
+    tree.reverse()
+    ret = tree.pop(0)
+    root = ret
+    for node in tree:
+        ret.insert(1, node)
+        ret = node
+    return root
diff --git a/tests/native_tests/language.hy b/tests/native_tests/language.hy
@@ -131,3 +131,8 @@
   (def foo "3")
   (assert (= *foo* FOO))
   (assert (!= *foo* foo)))
+
+(defn test-threading []
+  "NATIVE: test threading macro"
+  (assert (= (-> (.upper "a b c d") (.replace "A" "X") (.split))
+             ["X" "B" "C" "D"])))