parse function arguments

1 files changed, 32 insertions, 51 deletions

diff --git a/source/mod_student/lexer.mll b/source/mod_student/lexer.mll
index a62b8ad..806cfb8 100644
--- a/source/mod_student/lexer.mll
+++ b/source/mod_student/lexer.mll
@@ -31,6 +31,7 @@
    *)
 
   let identifier    = ['a'-'z' '_' 'A' - 'Z']['_' 'A' - 'Z' 'a'-'z' '0'-'9']*
+  let integer       = ['0'-'9']*
   
   (* ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
      ----------------------------------------------------------------------------- *)
@@ -54,60 +55,40 @@
       (* vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv 
        * Miejsce na twoje reguły
        *)
-      | "("
-      { LPAREN }
-      | ")"
-      { RPAREN }
-      | "{"
-      { LBRACKET }
-      | "{"
-      { RBRACKET }
-      | ":"
-      { COLON }
-      | "+"
-      { OP_PLUS }
-      | "-"
-      { OP_MINUS }
-      | "*"
-      { OP_MULT }
-      | "/"
-      { OP_DIV }
-      | "%"
-      { OP_REM }
-      | "&"
-      { OP_AND }
-      | "|"
-      { OP_OR }
-      | "=="
-      { OP_EQ }
-      | "!="
-      { OP_NEQ }
-      | "<="
-      { OP_LE }
-      | ">="
-      { OP_GE }
-      | "<"
-      { OP_LT }
-      | ">"
-      { OP_GT }
-      | "="
-      { ASSIGN }
-      | "if"
-      { IF }
-      | "else"
-      { ELSE }
-      | "while"
-      { WHILE }
-      | "return"
-      { RETURN }
-      | "length"
-      { LENGTH }
-      | identifier as id
-      { IDENTIFIER id }
+      | "("       { LPAREN }
+      | ")"       { RPAREN }
+      | "{"       { LBRACKET }
+      | "{"       { RBRACKET }
+      | "["       { LSBRACKET }
+      | "]"       { RSBRACKET }
+      | ":"       { COLON }
+      | "+"       { OP_PLUS }
+      | "-"       { OP_MINUS }
+      | "*"       { OP_MULT }
+      | "/"       { OP_DIV }
+      | "%"       { OP_REM }
+      | "&"       { OP_AND }
+      | "|"       { OP_OR }
+      | "=="      { OP_EQ }
+      | "!="      { OP_NEQ }
+      | "<="      { OP_LE }
+      | ">="      { OP_GE }
+      | "<"       { OP_LT }
+      | ">"       { OP_GT }
+      | "="       { ASSIGN }
+      | "if"      { IF }
+      | "else"    { ELSE }
+      | "while"   { WHILE }
+      | "return"  { RETURN }
+      | "length"  { LENGTH }
+      | "int"     { T_INT }
+      | "bool"    { T_BOOL }
+      | identifier as id  { IDENTIFIER id }
+      | integer as i      { INT (int_of_string i) }
 
       (* ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
          ----------------------------------------------------------------------------- *)
-  
+      | " " {token lexbuf}
       | _
       { handleError (Lexing.lexeme_start_p lexbuf) (Lexing.lexeme lexbuf) }