mumu 0.10.0

Lava Mumu is a language for those in the now and that know
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

// FILE: src/main.rs

use mumu::{Interpreter, Value};
use mumu::modules::compose::register_compose_and_pipe;
use mumu::modules::slog::register_slog;
use mumu::modules::sput::register_sput;
use mumu::modules::r#type::register_type;
use mumu::modules::ink::register_ink;
use mumu::modules::check::register_check;
use mumu::modules::include::register_include;

use mumu::parser::lexer::tokenize;
use mumu::parser::core::driver::parse_tokens;
use mumu::parser::ast::Stmt;

fn print_help() {
    println!("af [OPTIONS] <path-to-file>");
    println!("  -h, --help       Print this help message");
    println!("  --version        Print version & debug info");
    println!("Examples:");
    println!("  af script.af");
}

fn main() {
    if let Err(e) = try_main() {
        println!("Error: {}", e);
    }
}

fn try_main() -> Result<(), String> {
    let args: Vec<String> = std::env::args().collect();

    let mut debug_verbose = false;
    let mut file_path: Option<String> = None;

    let mut i = 1;
    while i < args.len() {
        match args[i].as_str() {
            "-h" | "--help" => {
                print_help();
                return Ok(());
            }
            "--version" => {
                println!("mumu version 0.9.1");
                println!("Environment: REPL + file-run example");
                return Ok(());
            }
            "-v" | "--verbose" => {
                debug_verbose = true;
                i += 1;
                continue;
            }
            other => {
                file_path = Some(other.to_string());
            }
        }
        i += 1;
    }

    // -------------------- REPL PATH --------------------
    if file_path.is_none() {
        // Spawn a REPL-mode interpreter, and set is_repl_mode=true
        return mumu::modules::repl::run_repl_mode(debug_verbose);
    }

    // -------------------- FILE EXECUTION PATH --------------------
    let file_path = file_path.unwrap();
    let code = std::fs::read_to_string(&file_path)
        .map_err(|e| format!("Failed to read '{}': {}", file_path, e))?;

    let tokens = tokenize(&code, debug_verbose)
        .map_err(|lex_err| format!("Lex error: {}", lex_err))?;

    let ast = parse_tokens(&tokens, debug_verbose)
        .map_err(|parse_err| format!("Parse error: {}", parse_err))?;

    let mut interp = Interpreter::new();
    interp.set_verbose(debug_verbose);
    interp.set_repl_mode(false); // This run is *not* REPL

    // Register all built-in modules
    register_compose_and_pipe(&mut interp);
    register_sput(&mut interp);
    register_slog(&mut interp);
    register_type(&mut interp);
    register_ink(&mut interp);
    register_check(&mut interp);
    register_include(&mut interp);

    let mut _last_val = Value::Int(0);

    // NEW: For correct plugin loading, execute each ImportSo immediately as you encounter it.
    let mut idx = 0;
    while idx < ast.len() {
        let stmt = &ast[idx];
        match stmt {
            Stmt::ImportSo { .. } => {
                // Execute extend/import statement immediately
                let val = interp
                    .exec_statement(stmt)
                    .map_err(|e| format!("Statement {} error: {}", idx + 1, e))?;
                _last_val = val;
                idx += 1;
                // continue to next statement
            }
            _ => {
                // For all subsequent non-ImportSo statements, execute as usual
                break;
            }
        }
    }
    // After all leading ImportSo/extend statements have been executed, execute the rest
    for (stmt_index, stmt) in ast[idx..].iter().enumerate() {
        let val = interp
            .exec_statement(stmt)
            .map_err(|e| format!("Statement {} error: {}", idx + stmt_index + 1, e))?;
        _last_val = val;
    }

    loop {
        if let Some(poll_fn) = interp.get_dynamic_function("poll_events") {
            let _ = poll_fn.lock().unwrap()(&mut interp, vec![]);
        }
        let total_tasks = interp.poll_all();
        if total_tasks == 0 {
            break;
        }
    }

    interp.remove_all_pollers();

    Ok(())
}