From c7d578ec4d9b87c36f504e5a0691007439d2a025 Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Fri, 4 Mar 2022 21:00:32 +0100
Subject: Internal functions/types

---
 src/Fun/Syntax.hs | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(limited to 'src/Fun/Syntax.hs')

diff --git a/src/Fun/Syntax.hs b/src/Fun/Syntax.hs
index 86e8100..c426a03 100644
--- a/src/Fun/Syntax.hs
+++ b/src/Fun/Syntax.hs
@@ -10,6 +10,8 @@ mergeSyntax :: [Either [SyntaxError] a] -> ([a] -> b) -> Either [SyntaxError] b
 mergeSyntax d c | any isLeft d = Left $ concat $ lefts d
                 | otherwise    = Right $ c $ rights d
 
+-- TODO: Disallow same declarations in entire tree (w/o visibility)
+
 checkFunctionArity :: Block -> Either [SyntaxError] Block
 checkFunctionArity (FunctionBlock decl defns) =
   let declArity   = length $ dTypes decl
@@ -22,14 +24,14 @@ checkFunctionArity (FunctionBlock decl defns) =
         else Left [SyntaxError msg]
   where msg = "invalid arity in function " ++ (dName decl)
 
-checkMainExistence :: Tree -> Either [SyntaxError] Tree
-checkMainExistence tree =
-  let inProgram bs = filter (\b -> (dName . bDecl) b == "main") bs
+checkEntryExistence :: Tree -> Either [SyntaxError] Tree
+checkEntryExistence tree =
+  let inProgram bs = filter (\b -> (dName . bDecl) b == "_start") bs
       oneInProgram (Program bs) = length (inProgram bs) == 1
       inTree ps = map oneInProgram ps
       oneInTree (Tree ps) = length (filter (== True) (inTree ps)) == 1
   in  if oneInTree tree then Right tree else Left [SyntaxError msg]
-  where msg = "invalid amount of main functions"
+  where msg = "invalid amount of entry (_start) functions"
 
 checkFunctionPatternDistinguishability :: Block -> Either [SyntaxError] Block
 checkFunctionPatternDistinguishability f =
@@ -54,4 +56,4 @@ checkProgram (Program blocks) = mergeSyntax (map checkBlock blocks) Program
 
 checkTree :: Tree -> Either [SyntaxError] Tree
 checkTree (Tree programs) =
-  mergeSyntax (map checkProgram programs) Tree >>= checkMainExistence
+  mergeSyntax (map checkProgram programs) Tree >>= checkEntryExistence
-- 
cgit v1.2.3