luaur-repl-cli 0.1.0

Interactive Luau REPL (Rust).
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

use alloc::string::String;
use core::ffi::{c_char, c_void, CStr};

use luaur_ast::records::parse_options::ParseOptions;
use luaur_bytecode::records::bytecode_encoder::BytecodeEncoder;
use luaur_cli_lib::functions::read_file::read_file;
use luaur_code_gen::functions::luau_codegen_compile::luau_codegen_compile;
use luaur_compiler::functions::compile::compile;
use luaur_vm::enums::lua_status::lua_Status;
use luaur_vm::functions::lua_gettop::lua_gettop;
use luaur_vm::functions::lua_isstring::lua_isstring;
use luaur_vm::functions::lua_l_sandboxthread::lua_l_sandboxthread;
use luaur_vm::functions::lua_mainthread::lua_mainthread;
use luaur_vm::functions::lua_newthread::lua_newthread;
use luaur_vm::functions::lua_remove::lua_remove;
use luaur_vm::functions::lua_resume::lua_resume;
use luaur_vm::functions::lua_xmove::lua_xmove;
use luaur_vm::functions::luau_load::luau_load;
use luaur_vm::macros::lua_l_error::luaL_error;
use luaur_vm::macros::lua_tostring::lua_tostring;
use luaur_vm::type_aliases::lua_state::lua_State;

use crate::records::repl_requirer::ReplRequirer;

pub unsafe fn load(
    l: *mut lua_State,
    ctx: *mut c_void,
    _path: *const c_char,
    chunkname: *const c_char,
    loadname: *const c_char,
) -> i32 {
    let req = &*(ctx as *const ReplRequirer);

    // module needs to run in a new thread, isolated from the rest
    // note: we create ML on main thread so that it doesn't inherit environment of L
    let gl = lua_mainthread(l);
    let ml = lua_newthread(gl);
    lua_xmove(gl, l, 1);

    // new thread needs to have the globals sandboxed
    lua_l_sandboxthread(ml);

    let loadname_str = CStr::from_ptr(loadname).to_string_lossy();

    let contents = read_file(&loadname_str);
    let had_contents = contents.is_some();
    let mut status: i32 = lua_Status::LUA_OK as i32;

    if let Some(ref source) = contents {
        // now we can compile & run module on the new thread
        struct NoopEncoder;
        impl BytecodeEncoder for NoopEncoder {
            fn encode(&mut self, _data: &mut [u32]) {}
        }
        let options = (req.copts.unwrap())();
        let parse_options = ParseOptions::default();
        let mut encoder = NoopEncoder;
        let source_owned: String = source.clone();
        let bytecode = compile(
            &source_owned,
            &options,
            &parse_options,
            &mut encoder as *mut dyn BytecodeEncoder,
        );
        status = luau_load(
            ml,
            chunkname,
            bytecode.as_ptr() as *const c_char,
            bytecode.len(),
            0,
        );
    }

    if !had_contents {
        luaL_error!(l, "could not read file '{}'", loadname_str);
    }

    if status == 0 {
        if (req.codegenEnabled.unwrap())() {
            // The Rust codegen port exposes `luau_codegen_compile(L, idx)`; the
            // native CompilationOptions (CodeGen_ColdFunctions / recordCounters)
            // are not threaded through the public Rust API.
            luau_codegen_compile(ml, -1);
        }

        if (req.coverageActive.unwrap())() {
            (req.coverageTrack.unwrap())(ml as *mut c_void, -1);
        }

        if (req.countersActive.unwrap())() {
            (req.countersTrack.unwrap())(ml as *mut c_void, -1);
        }

        let status = lua_resume(ml, l, 0);

        if status == 0 {
            if lua_gettop(ml) != 1 {
                luaL_error!(l, "module must return a single value");
            }
        } else if status == lua_Status::LUA_YIELD as i32 {
            luaL_error!(l, "module can not yield");
        } else if lua_isstring(ml, -1) == 0 {
            luaL_error!(l, "unknown error while running module");
        } else {
            let msg = lua_tostring!(ml, -1);
            let msg = CStr::from_ptr(msg).to_string_lossy();
            luaL_error!(l, "error while running module: {}", msg);
        }
    }

    // add ML result to L stack
    lua_xmove(ml, l, 1);

    // remove ML thread from L stack
    lua_remove(l, -2);

    // added one value to L stack: module result
    1
}