1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

//! The TokenKind enum maintains all of the different Tokens that could occur within the program.
//! Using an enum allows for easy extensibility.

#[derive(Debug)]
pub enum TokenKind {
    Void,
    Any,
    IntegerLiteral(i64),
    FloatLiteral(f64),
    BooleanLiteral(bool),
    StringLiteral(String),
    Identifier(String),
    Label(String),
    End,

    Push,
    Pop,
    Peek,
    Add,
    Sub,
    Mul,
    Div,
    LessThan,
    LessThanEqual,
    GreaterThan,
    GreaterThanEqual,
    Equal,
    NotEqual,
    Jump,
    RelativeJump,
    JumpIfTrue,
    JumpIfFalse,
    Print,
    PrintNewLine,
    Set,
    Call,
}

impl TokenKind {
    /// This function checks if the given name is an instrution. If it is, it returns the correct instruction token.
    /// This function is called by the lexer, specifically by the make_word function.
    ///
    /// # Arguments
    /// `name` - The name of the current word.
    pub fn is_instruction(name: &str) -> Option<TokenKind> {
        match name.to_ascii_lowercase().as_str() {
            "push" => Some(TokenKind::Push),
            "pop" => Some(TokenKind::Pop),
            "peek" => Some(TokenKind::Peek),
            "add" => Some(TokenKind::Add),
            "sub" => Some(TokenKind::Sub),
            "mul" => Some(TokenKind::Mul),
            "div" => Some(TokenKind::Div),
            "lt" => Some(TokenKind::LessThan),
            "lte" => Some(TokenKind::LessThanEqual),
            "gt" => Some(TokenKind::GreaterThan),
            "gte" => Some(TokenKind::GreaterThanEqual),
            "eq" => Some(TokenKind::Equal),
            "neq" => Some(TokenKind::NotEqual),
            "jmp" => Some(TokenKind::Jump),
            "rjmp" => Some(TokenKind::RelativeJump),
            "jmpt" => Some(TokenKind::JumpIfTrue),
            "jmpf" => Some(TokenKind::JumpIfFalse),
            "print" => Some(TokenKind::Print),
            "printn" => Some(TokenKind::PrintNewLine),
            "set" => Some(TokenKind::Set),
            "call" => Some(TokenKind::Call),

            _ => None,
        }
    }
}