doukutsu-rs/src/scripting/lua/mod.rs

use std::io::Read;
use std::ptr::null_mut;

use lua_ffi::lua_fn;
use lua_ffi::ffi::lua_State;
use lua_ffi::types::LuaValue;
use lua_ffi::{c_int, State, ThreadStatus};

use crate::common::Rect;
use crate::framework::context::Context;
use crate::framework::error::{GameError, GameResult};
use crate::framework::filesystem;
use crate::framework::filesystem::File;
use crate::scene::game_scene::GameScene;
use crate::scripting::lua::doukutsu::Doukutsu;
use crate::shared_game_state::SharedGameState;

mod doukutsu;
mod scene;

pub struct LuaScriptingState {
    state: Option<State>,
    state_ptr: *mut SharedGameState,
    ctx_ptr: *mut Context,
    game_scene: *mut GameScene,
}

pub(in crate::scripting) static DRS_API_GLOBAL: &str = "__doukutsu_rs";
pub(in crate::scripting) static DRS_RUNTIME_GLOBAL: &str = "__doukutsu_rs_runtime_dont_touch";

static BOOT_SCRIPT: &str = include_str!("boot.lua");

pub(in crate::scripting) fn check_status(status: ThreadStatus, state: &mut State) -> GameResult {
    match status {
        ThreadStatus::Ok | ThreadStatus::Yield => {
            return Ok(());
        }
        _ => {}
    }

    let error = state.to_str(-1).unwrap_or("???");
    match status {
        ThreadStatus::RuntimeError => Err(GameError::EventLoopError(format!("Lua Runtime Error: {}", error))),
        ThreadStatus::SyntaxError => Err(GameError::EventLoopError(format!("Lua Syntax Error: {}", error))),
        ThreadStatus::MemoryError => Err(GameError::EventLoopError(format!("Lua Memory Error: {}", error))),
        ThreadStatus::MsgHandlerError => {
            Err(GameError::EventLoopError(format!("Lua Message Handler Error: {}", error)))
        }
        ThreadStatus::FileError => Err(GameError::EventLoopError(format!("Lua File Error: {}", error))),
        ThreadStatus::Unknown => Err(GameError::EventLoopError(format!("Unknown Error: {}", error))),
        _ => Ok(()),
    }
}

fn print(state: &mut State) -> c_int {
    if let Some(msg) = state.to_str(1) {
        log::info!("[Lua] {}", msg);
    }

    0
}

impl LuaScriptingState {
    pub fn new() -> LuaScriptingState {
        LuaScriptingState { state: None, state_ptr: null_mut(), ctx_ptr: null_mut(), game_scene: null_mut() }
    }

    pub fn update_refs(&mut self, state: *mut SharedGameState, ctx: *mut Context) {
        self.state_ptr = state;
        self.ctx_ptr = ctx;
    }

    pub fn set_game_scene(&mut self, game_scene: *mut GameScene) {
        self.game_scene = game_scene;
    }

    fn load_script(mut state: &mut State, path: &str, mut script: File) -> bool {
        let mut buf = Vec::new();
        let res = script.read_to_end(&mut buf);

        if let Err(err) = res {
            log::warn!("Error reading script {}: {}", path, err);
            return false;
        }

        let name = format!("@{}", path);
        let res = state.load_buffer(&buf, &name);
        let res = check_status(res, &mut state);
        if let Err(err) = res {
            log::warn!("Error loading script {}: {}", path, err);
            return false;
        }

        state.get_global(DRS_RUNTIME_GLOBAL);
        state.get_field(-1, "_initializeScript");
        state.push_value(-3);

        let res = state.pcall(1, 0, 0);
        if let Err((_, err)) = res {
            log::warn!("Error evaluating script {}: {}", path, err);
            return false;
        }

        log::info!("Successfully loaded Lua script: {}", path);

        true
    }

    pub fn reload_scripts(&mut self, ctx: &mut Context) -> GameResult {
        let mut state = State::new();
        state.open_libs();

        state.push(lua_fn!(print));
        state.set_global("print");

        state.push(Doukutsu { ptr: self as *mut LuaScriptingState });
        state.set_global(DRS_API_GLOBAL);

        log::info!("Initializing Lua scripting engine...");
        let res = state.do_string(BOOT_SCRIPT);
        check_status(res, &mut state)?;

        if filesystem::exists(ctx, "/drs-scripts/") {
            let mut script_count = 0;
            let files = filesystem::read_dir(ctx, "/drs-scripts/")?
                .filter(|f| f.to_string_lossy().to_lowercase().ends_with(".lua"));

            for file in files {
                let path = file.clone();

                match filesystem::open(ctx, file) {
                    Ok(script) => {
                        if LuaScriptingState::load_script(&mut state, path.to_string_lossy().as_ref(), script) {
                            script_count += 1;
                        }
                    }
                    Err(err) => {
                        log::warn!("Error opening script {:?}: {}", path, err);
                    }
                }
            }

            if script_count > 0 {
                log::info!("{} Lua scripts have been loaded.", script_count);
            }
        }

        let modcs_path = "/Scripts/main.lua";
        if filesystem::exists(ctx, modcs_path) {
            log::info!("Loading ModCS main script...");

            match filesystem::open(ctx, modcs_path) {
                Ok(script) => {
                    if !LuaScriptingState::load_script(&mut state, modcs_path, script) {
                        log::warn!("Error loading ModCS main script.");
                    }
                }
                Err(err) => {
                    log::warn!("Error opening script {:?}: {}", modcs_path, err);
                }
            }
        }

        self.state = Some(state);

        Ok(())
    }
}

impl LuaValue for Rect<u16> {
    fn push_val(self, l: *mut lua_State) {
        unsafe {
            lua_ffi::ffi::lua_newtable(l);
            lua_ffi::ffi::lua_pushinteger(l, self.left as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 1);
            lua_ffi::ffi::lua_pushinteger(l, self.top as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 2);
            lua_ffi::ffi::lua_pushinteger(l, self.right as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 3);
            lua_ffi::ffi::lua_pushinteger(l, self.bottom as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 4);
        }
    }
}

impl LuaValue for Rect<i16> {
    fn push_val(self, l: *mut lua_State) {
        unsafe {
            lua_ffi::ffi::lua_newtable(l);
            lua_ffi::ffi::lua_pushinteger(l, self.left as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 1);
            lua_ffi::ffi::lua_pushinteger(l, self.top as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 2);
            lua_ffi::ffi::lua_pushinteger(l, self.right as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 3);
            lua_ffi::ffi::lua_pushinteger(l, self.bottom as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 4);
        }
    }
}

impl LuaValue for Rect<i32> {
    fn push_val(self, l: *mut lua_State) {
        unsafe {
            lua_ffi::ffi::lua_newtable(l);
            lua_ffi::ffi::lua_pushinteger(l, self.left as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 1);
            lua_ffi::ffi::lua_pushinteger(l, self.top as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 2);
            lua_ffi::ffi::lua_pushinteger(l, self.right as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 3);
            lua_ffi::ffi::lua_pushinteger(l, self.bottom as isize);
            lua_ffi::ffi::lua_rawseti(l, -2, 4);
        }
    }
}

impl LuaValue for Rect<f32> {
    fn push_val(self, l: *mut lua_State) {
        unsafe {
            lua_ffi::ffi::lua_newtable(l);
            lua_ffi::ffi::lua_pushnumber(l, self.left as f64);
            lua_ffi::ffi::lua_rawseti(l, -2, 1);
            lua_ffi::ffi::lua_pushnumber(l, self.top as f64);
            lua_ffi::ffi::lua_rawseti(l, -2, 2);
            lua_ffi::ffi::lua_pushnumber(l, self.right as f64);
            lua_ffi::ffi::lua_rawseti(l, -2, 3);
            lua_ffi::ffi::lua_pushnumber(l, self.bottom as f64);
            lua_ffi::ffi::lua_rawseti(l, -2, 4);
        }
    }
}