Make parser properly parse function definitions with 0 parameters (fix …

…#11)
rockofox · Dec 7, 2024 · e1fbd95 · e1fbd95
1 parent 54346cf
commit e1fbd95
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/lib/Parser.hs b/lib/Parser.hs
@@ -260,7 +260,7 @@ defArg = try structLit <|> var <|> parens listPattern <|> array <|> placeholder
 funcDef :: Parser Expr
 funcDef = do
     name <- identifier <|> gravis <?> "function name"
-    args <- some defArg <?> "function arguments"
+    args <- many defArg <?> "function arguments"
     symbol "="
     FuncDef name args <$> expr <?> "function body"
 

diff --git a/tests/ParserSpec.hs b/tests/ParserSpec.hs
@@ -21,6 +21,15 @@ parserCompilerFlags = initCompilerFlags{needsMain = False}
 
 spec :: Spec
 spec = do
+    describe "Function definitions" $ do
+        it "Should parse a simple function" $
+            parseProgram "add a b = a + b" parserCompilerFlags
+                `shouldBe` Right
+                    (Program [FuncDef{name = "add", args = [Var "a" anyPosition, Var "b" anyPosition], body = Add (Var "a" anyPosition) (Var "b" anyPosition)}])
+        it "Should prase a function with 0 parameters" $
+            parseProgram "thunk = 1" parserCompilerFlags
+                `shouldBe` Right
+                    (Program [FuncDef{name = "thunk", args = [], body = IntLit 1}])
     describe "compareTypes" $ do
         it "Should be true for exact matches" $
             property $