Internal functions/types

1 files changed, 15 insertions, 3 deletions

diff --git a/src/Fun/Grammar.hs b/src/Fun/Grammar.hs
index 7ec9e42..47191de 100644
--- a/src/Fun/Grammar.hs
+++ b/src/Fun/Grammar.hs
@@ -63,11 +63,16 @@ functionDeclarationDelimiter =
     <+-> space
     <?>  "function declaration delimiter"
 
+internalFunctions :: [String]
+internalFunctions = ["_start", "_asm"]
+
 functionName :: Parser String
 functionName =
   (special <|> letter)
     <+> iter (special <|> alphanum)
     >>> build
+    <|> internalWord
+    <=> (`elem` internalFunctions)
     <?> "function name"
   where build (first, rest) = first : rest
 
@@ -79,11 +84,17 @@ functionTypes =
     <?> "function types"
   where build (a, b) = a ++ [b]
 
+internalTypes :: [String]
+internalTypes = ["_void"]
+
 functionType :: Parser Type
 functionType =
   (letter <=> isUpper)
     <+>  oneOrMore alphanum
     >>>  build
+    <|>  internalWord
+    <=>  (`elem` internalTypes)
+    >>>  buildInternal
     <|>  literal '['
     <-+> functionType
     <+-> literal ']'
@@ -91,6 +102,7 @@ functionType =
     <?>  "function type"
  where
   build (first, rest) = resolveType $ first : rest
+  buildInternal t = resolveType t
   buildList t = ListType t
 
 functionTypeDelimiter :: Parser Char
@@ -105,10 +117,10 @@ functionFlagDelimiter :: Parser Char
 functionFlagDelimiter = literal '%' <?> "function flag delimiter"
 
 functionFlags :: [String]
-functionFlags = ["inline", "deprecated"]
+functionFlags = ["inline", "deprecated", "danger-void", "danger-asm"]
 
 functionFlag :: Parser String
-functionFlag = letters <=> (`elem` functionFlags) <?> "function flag"
+functionFlag = specialWord <=> (`elem` functionFlags) <?> "function flag"
 
 functionDefinition :: Parser FunctionDefinition
 functionDefinition =
@@ -155,4 +167,4 @@ functionBodyElement =
     <?> "function body element"
 
 statement :: Parser FunctionBodyElement --  TODO
-statement = accept "if" >>> Statement
+statement = acceptSpecial "if" >>> Statement