From 4f6dd49df3f19204694fcea55f38efd9c5118bf2 Mon Sep 17 00:00:00 2001
From: Johannes Stoelp <johannes.stoelp@gmail.com>
Date: Mon, 4 Oct 2021 22:51:42 +0200
Subject: ch5: added if/then/else

---
 src/lexer.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/lexer.rs')

diff --git a/src/lexer.rs b/src/lexer.rs
index a25f0ab..fdab5b4 100644
--- a/src/lexer.rs
+++ b/src/lexer.rs
@@ -6,6 +6,9 @@ pub enum Token {
     Identifier(String),
     Number(f64),
     Char(char),
+    If,
+    Then,
+    Else,
 }
 
 pub struct Lexer<I>
@@ -62,6 +65,9 @@ where
             match ident.as_ref() {
                 "def" => return Token::Def,
                 "extern" => return Token::Extern,
+                "if" => return Token::If,
+                "then" => return Token::Then,
+                "else" => return Token::Else,
                 _ => {}
             }
 
@@ -178,4 +184,12 @@ mod test {
         assert_eq!(Token::Identifier("c".into()), lex.gettok());
         assert_eq!(Token::Eof, lex.gettok());
     }
+
+    #[test]
+    fn test_ite() {
+        let mut lex = Lexer::new("if then else".chars());
+        assert_eq!(Token::If, lex.gettok());
+        assert_eq!(Token::Then, lex.gettok());
+        assert_eq!(Token::Else, lex.gettok());
+    }
 }
-- 
cgit v1.2.3