shithub: MicroHs

Download patch

ref: f9f47c644421659ae388a6b831ac706439529919
parent: ddc38da1edf82344f6a211e0a00e075549861ba8
author: Jan-Willem Maessen <jmaessen@fb.com>
date: Thu Jan 18 22:21:52 EST 2024

Tweak Desugar.lazier to use existing utility functions

--- a/src/MicroHs/Desugar.hs
+++ b/src/MicroHs/Desugar.hs
@@ -541,8 +541,7 @@
       (drops, keeps) = splitAt arity as
       -- reverse n-ary apply
       app :: [Ident] -> Exp -> Exp
-      app [] e = e
-      app (v:vs) e = app vs $ App e (Var v)
+      app vs e = apps e (map Var vs)
       -- Replace n recursive args with call to vfcn'
       repl :: [Ident] -> Exp -> Exp
       repl vs (Lam i e) = app vs $ Lam i $ repl [] e
@@ -552,11 +551,7 @@
       repl vs (Var v)
         | v == fcn && take arity vs == drops = app (drop arity vs) $ vfcn'
       repl vs e = app vs e
-      -- n-ary lambda
-      lam :: [Ident] -> Exp -> Exp
-      lam [] e = e
-      lam (v:vs) e = Lam v $ lam vs e
   in  if arity > 0
-      then (fcn, lam drops $ letRecE fcn' (lam keeps (repl [] body)) vfcn')
+      then (fcn, lams drops $ letRecE fcn' (lams keeps (repl [] body)) vfcn')
       else def
 lazier def = def
--