Variable declaration

1 files changed, 12 insertions, 2 deletions

diff --git a/src/Parser.hs b/src/Parser.hs
index ea698c8..f341bc2 100644
--- a/src/Parser.hs
+++ b/src/Parser.hs
@@ -43,7 +43,7 @@ variable :: Parser Expr
 variable = do
   pos <- getPosition
   var <- identifier
-  return $ Var var pos
+  return $ Variable var pos
 
 typ :: Parser Type
 typ = do
@@ -116,6 +116,16 @@ call = do
   args <- parens $ commaSep expr
   return $ Call ident args pos
 
+var :: Parser Expr
+var = do
+  pos <- getPosition
+  reserved "var"
+  (ident, typ, _) <- arg
+  reservedOp "=" <?> "assignation"
+  value <- expr
+  reservedOp ";"
+  return $ Var ident typ value pos
+
 factor :: Parser Expr
 factor =
   number
@@ -150,7 +160,7 @@ program = do
   n <-
     many $ do
       function
-        -- TODO: variable decl
+        <|> var
         <|> statement <?> "statement"
   return $ [m] ++ n