factorio-frontend 0.1.0

Rust frontend that lowers Factorio mod sources into factorio-ir
Documentation 
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

mod common;

use common::must_ok_parse;
use factorio_frontend::parse_module;
use factorio_ir::statement::Statement;

#[test]
fn parses_method_with_self() {
    let source = r"
pub fn reset(&mut self, player: ()) {
    return;
}
";

    let module = must_ok_parse(parse_module(source, "control.player_util"));
    let Statement::FunctionDecl(function) = &module.symbols[0].statement else {
        assert_eq!(1, 0, "expected function declaration");
        return;
    };

    assert_eq!(function.name, "reset");
    assert_eq!(function.params.len(), 2);
    assert_eq!(function.params[0].source_type.as_deref(), Some("&mut self"));
    assert_eq!(function.params[1].source_type.as_deref(), Some("()"));
    assert_eq!(function.body.statements, vec![Statement::Return(None)]);
}

#[test]
fn parses_implicit_return() {
    let source = r"
fn helper() -> i64 {
    1
}
";

    let module = must_ok_parse(parse_module(source, "control.example"));
    let Statement::FunctionDecl(function) = &module.body.statements[0] else {
        assert_eq!(1, 0, "expected helper function");
        return;
    };

    assert_eq!(function.name, "helper");
    assert_eq!(
        function
            .debug
            .as_ref()
            .and_then(|debug| debug.return_type.as_deref()),
        Some("i64")
    );
}