Extended lexer to support all types of tokens needed.

1 files changed, 17 insertions, 2 deletions

diff --git a/include/token_type.hpp b/include/token_type.hpp
index d123aaa..a62369d 100644
--- a/include/token_type.hpp
+++ b/include/token_type.hpp
@@ -3,20 +3,35 @@
 enum class TokenType {
     INT,
     STRING,
+    FUNCTION,
 
     NUMBER,
     IDENTIFIER,
+    RETURN,
+
+    IF,
+    WHILE,
+    FOR,
 
     PLUS,
     MINUS,
     MULTIPLY,
     DIVIDE,
+    EQUAL,
+    LESS_EQUAL,
+    GREATER_EQUAL,
+    LESS,
+    GREATER,
 
     ASSIGN,
+    ARROW,
 
     SEMICOLON,
+    LEFT_BRACKET,
+    RIGHT_BRACKET,
+    LEFT_BRACE,
+    RIGHT_BRACE,
 
     END_OF_FILE,
-    INVALID
-
+    INVALID,
 };