frg 2.1.0

Compiled programming language with frogs!
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

use super::{TreeCursor, VarType, types};

pub fn parse(cursor: &mut TreeCursor, code: &str) -> VarType {
    cursor.goto_first_child();

    let return_type = Box::new(types::parse(cursor, code));

    cursor.goto_next_sibling();
    let param_types = parameter_list(cursor, code);

    cursor.goto_parent();
    VarType::Function {
        return_type,
        param_types,
    }
}

fn parameter_list(cursor: &mut TreeCursor, code: &str) -> Vec<VarType> {
    cursor.goto_first_child();
    // skip "("
    cursor.goto_next_sibling();

    let mut param_types = vec![];
    loop {
        match cursor.node().kind() {
            ")" => break,
            "," => {
                cursor.goto_next_sibling();
                continue;
            }
            "type" => param_types.push(types::parse(cursor, code)),
            _ => panic!("unexpected token in param types"),
        }

        if !cursor.goto_next_sibling() {
            break;
        }
    }

    cursor.goto_parent();
    param_types
}