about summary refs log tree commit diff
path: root/src/lex/lexer.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/lex/lexer.rs')
-rw-r--r--src/lex/lexer.rs7
1 files changed, 6 insertions, 1 deletions
diff --git a/src/lex/lexer.rs b/src/lex/lexer.rs
index 6e188a0..b6fee0d 100644
--- a/src/lex/lexer.rs
+++ b/src/lex/lexer.rs
@@ -181,10 +181,15 @@ impl<'s> Lexer<'s> {
 
         let variant = match buffer.as_str() {
             "fn" => TokenVariant::KeywordFn,
+            "if" => TokenVariant::KeywordIf,
+            "elif" => TokenVariant::KeywordElif,
+            "else" => TokenVariant::KeywordElse,
             "type" => TokenVariant::KeywordType,
             "form" => TokenVariant::KeywordForm,
             "self" => TokenVariant::KeywordSelf,
-            _ => TokenVariant::Identifier(buffer),
+            "return" => TokenVariant::KeywordReturn,
+            "print" => TokenVariant::KeywordPrint,
+            _ => TokenVariant::Ident(buffer),
         };
 
         Token { location, variant }