luaur-web 0.1.3

//! `#[wasm_bindgen]` browser API for the Luau playground.
//!
//! These wrappers sit on top of the crate's existing `extern "C"` entry points
//! (`execute_script`, `check_script`) and the lower-level `run_code` /
//! `setup_state` logic. They are gated behind the `wasm` feature and present a
//! plain `&str` -> `String` interface that JavaScript can call directly.
//!
//! Two functions are exported:
//!
//! * [`run`]    — compile + execute the source on the VM, returning captured
//!                `print` output (and any error text).
//! * [`check`]  — type-check the source with the analyzer, returning the
//!                newline-joined diagnostics, or `"No errors."` when clean.
//!
//! ## Capturing `print`
//!
//! The VM's default `print` (`lua_b_print`) writes through `writestring` to
//! `std::io::stdout()`. On `wasm32-unknown-unknown` there is no real stdout, so
//! that output would be silently discarded. To make a browser playground
//! useful, [`run`] installs a capturing `print` global *before* the sandbox
//! freezes the global table; the capturing function appends to a thread-local
//! buffer that is drained and returned to JavaScript after the script finishes.
//! The capture mirrors `lua_b_print` exactly (tab-separated args, trailing
//! newline, `luaL_tolstring` coercion) so observable behavior is unchanged.

use alloc::string::String;
use alloc::string::ToString;
use core::cell::RefCell;
use core::ffi::c_char;
use core::ffi::c_int;

use wasm_bindgen::prelude::wasm_bindgen;

use crate::functions::check_script::check_script;
use crate::functions::run_code::run_code;

use luaur_common::set_luau_bool_flags;
use luaur_vm::functions::lua_close::lua_close;
use luaur_vm::functions::lua_gettop::lua_gettop;
use luaur_vm::functions::lua_l_newstate::lua_l_newstate;
use luaur_vm::functions::lua_l_openlibs::lua_l_openlibs;
use luaur_vm::functions::lua_l_sandbox::lua_l_sandbox;
use luaur_vm::functions::lua_l_sandboxthread::lua_l_sandboxthread;
use luaur_vm::functions::lua_l_tolstring::lua_l_tolstring;
use luaur_vm::functions::lua_pushcclosurek::lua_pushcclosurek;
use luaur_vm::macros::lua_pop::lua_pop;
use luaur_vm::macros::lua_setglobal::lua_setglobal;
use luaur_vm::type_aliases::lua_state::lua_State;

/// Module start hook: route Rust panics to `console.error` with a readable
/// message + location instead of an opaque `unreachable` wasm trap. Runs once
/// when the module is instantiated.
#[wasm_bindgen(start)]
pub fn wasm_start() {
    console_error_panic_hook::set_once();
}

thread_local! {
    /// Accumulates `print` output for the current `run` call. Drained when the
    /// VM finishes so subsequent runs start clean.
    static PRINT_BUFFER: RefCell<String> = const { RefCell::new(String::new()) };
}

/// Capturing replacement for the VM's `print`. Mirrors `lua_b_print`: each
/// argument is coerced with `luaL_tolstring`, arguments are tab-separated, and a
/// trailing newline is appended — but the bytes go to [`PRINT_BUFFER`] instead
/// of `stdout`.
unsafe fn capturing_print(l: *mut lua_State) -> c_int {
    let n = lua_gettop(l);
    let mut line = String::new();
    for i in 1..=n {
        let mut len: usize = 0;
        let s = lua_l_tolstring(l, i, &mut len);
        if i > 1 {
            line.push('\t');
        }
        if !s.is_null() {
            let bytes = core::slice::from_raw_parts(s as *const u8, len);
            line.push_str(&String::from_utf8_lossy(bytes));
        }
        lua_pop(l, 1);
    }
    line.push('\n');
    PRINT_BUFFER.with(|b| b.borrow_mut().push_str(&line));
    0
}

/// Compile and execute `source` on a fresh sandboxed Luau VM, returning the
/// program's captured `print` output followed by any runtime error text.
///
/// This is the browser counterpart of the crate's `extern "C"` `execute_script`
/// — it shares `setup_state` and `run_code`, but installs a capturing `print`
/// and returns the captured output as an owned `String`.
#[wasm_bindgen]
pub fn run(source: &str) -> String {
    // Enable the `Luau*` bool fast flags, matching `execute_script`.
    set_luau_bool_flags(true);

    unsafe {
        let l: *mut lua_State = lua_l_newstate();

        // Open the standard library + base `print`, then override `print` with
        // the capturing variant *before* the sandbox freezes the global table.
        lua_l_openlibs(l);
        lua_pushcclosurek(l, Some(capturing_print), c"print".as_ptr(), 0, None);
        lua_setglobal(l, c"print".as_ptr());

        // Freeze libraries / proxy the global table, exactly like
        // `setup_state` + `execute_script` do for the C++ web demo.
        lua_l_sandbox(l);
        lua_l_sandboxthread(l);

        // Reset the capture buffer for this run.
        PRINT_BUFFER.with(|b| b.borrow_mut().clear());

        let error = run_code(l, source);

        lua_close(l);

        let mut out = PRINT_BUFFER.with(|b| core::mem::take(&mut *b.borrow_mut()));
        if !error.is_empty() {
            if !out.is_empty() && !out.ends_with('\n') {
                out.push('\n');
            }
            out.push_str(&error);
        }
        out
    }
}

/// Type-check `source` with the analyzer (old solver) and return the
/// newline-joined `line: message` diagnostics, or `"No errors."` when clean.
///
/// This wraps the crate's `extern "C"` `check_script`, converting the returned
/// C string pointer back into an owned `String`.
#[wasm_bindgen]
pub fn check(source: &str) -> String {
    let c_source = match alloc::ffi::CString::new(source) {
        Ok(s) => s,
        Err(_) => return "error: source contains an interior NUL byte".to_string(),
    };

    let result_ptr = unsafe { check_script(c_source.as_ptr() as *const c_char, 0) };

    if result_ptr.is_null() {
        return "No errors.".to_string();
    }

    unsafe {
        core::ffi::CStr::from_ptr(result_ptr)
            .to_string_lossy()
            .into_owned()
    }
}