Merge pull request #54 from EpitechPromo2027/53-parse-recursive-lambd…

…a-calls add: parse recursive `lambda` calls
Jabolol · Dec 9, 2024 · d2be80f · d2be80f
2 parents 8baad29 + 7ebb3b1
commit d2be80f
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/lib/Ast/Parser.hs b/lib/Ast/Parser.hs
@@ -95,6 +95,7 @@ parseFunctionDefine = do
     args <- M.many $ parseVarName <* M.optional sc
     return (name, args)
   mapM_ (\arg -> S.modify $ E.insertVar arg $ Var arg) args
+  S.modify $ E.insertFn name $ Lambda args $ Seq []
   value <- parseExpr
   let lambda = Lambda args value
   S.modify $ E.insertFn name lambda

diff --git a/test/Ast/ParserSpec.hs b/test/Ast/ParserSpec.hs
@@ -119,6 +119,34 @@ spec = do
                   (Op Add (Var "x") (Var "y"))
               ]
           )
+    it "parses a recursive lambda" $ do
+      parse "" "(define (pow x n) ( if (< n 2) (x) (* x (pow x (- n 1)))))"
+        `shouldBe` Right
+          ( AST
+              [ Define
+                  "pow"
+                  ( Lambda
+                      ["x", "n"]
+                      ( If
+                          ( Op Lt (Var "n") (Lit (LInt 2))
+                          )
+                          (Seq [Var "x"])
+                          ( Op
+                              Mult
+                              (Var "x")
+                              ( Call
+                                  (Var "pow")
+                                  ( Seq
+                                      [ Var "x",
+                                        Op Sub (Var "n") (Lit (LInt 1))
+                                      ]
+                                  )
+                              )
+                          )
+                      )
+                  )
+              ]
+          )
 
     it "fails to parse a lambda with invalid parameter list" $ do
       parse "" "(lambda 42 (* 2 2))" `shouldSatisfy` isLeft