Add parsing for def/set! expressions

1 files changed, 4 insertions, 0 deletions

diff --git a/src/lexer.c b/src/lexer.c
index cbcc175..704ee66 100755
--- a/src/lexer.c
+++ b/src/lexer.c
@@ -13,6 +13,8 @@ static const char* token_str[] = {
     [TOKEN_FALSE]   = "TOKEN_FALSE",
     [TOKEN_LAMBDA]  = "TOKEN_LAMBDA",
     [TOKEN_IF]      = "TOKEN_IF",
+    [TOKEN_DEF]     = "TOKEN_DEF",
+    [TOKEN_SET]     = "TOKEN_SET",
     [TOKEN_EOF]     = "TOKEN_EOF",
 };
 
@@ -128,6 +130,8 @@ find_primitive_type(const StringView value) {
     if (TOKEN_IS_KEYWORD(value, "false")) { return TOKEN_FALSE; }
     if (TOKEN_IS_KEYWORD(value, "lambda")) { return TOKEN_LAMBDA; }
     if (TOKEN_IS_KEYWORD(value, "if")) { return TOKEN_IF; }
+    if (TOKEN_IS_KEYWORD(value, "def")) { return TOKEN_DEF; }
+    if (TOKEN_IS_KEYWORD(value, "set!")) { return TOKEN_SET; }
 
     return TOKEN_SYMBOL;
 }