Add back support for process.exit and exit code

Cargo.toml

@@ -84,7 +84,7 @@ tracing-subscriber = { version = "0.3", features = ["env-filter"] }
 tokio = { version = "1.24", features = ["full", "tracing"] }
 os_str_bytes = { version = "6.4", features = ["conversions"] }
 
-mlua-luau-runtime = { git = "https://github.com/lune-org/mlua-luau-runtime", rev = "a5ae251fa3d62bb023dfbfc8ef692b28ff7a9a16" }
+mlua-luau-runtime = { git = "https://github.com/lune-org/mlua-luau-runtime", rev = "4117cfba75c6df50bd5b12a776ad33e768393753" }
 mlua = { version = "0.9.5", features = [
     "async",
     "luau",

src/lune/builtins/process/mod.rs

@@ -1,7 +1,7 @@
 use std::{
     env::{self, consts},
     path,
-    process::Stdio,
+    process::{ExitCode, Stdio},
 };
 
 use mlua::prelude::*;
@@ -61,9 +61,9 @@ pub fn create(lua: &Lua) -> LuaResult<LuaTable> {
         .globals()
         .get::<_, LuaTable>("coroutine")?
         .get::<_, LuaFunction>("yield")?;
-    let set_scheduler_exit_code = lua.create_function(|_, _code: Option<u8>| {
-        // TODO: Implement this
-        // sched.set_exit_code(code.unwrap_or_default());
+    let set_scheduler_exit_code = lua.create_function(|lua, code: Option<u8>| {
+        let code = code.map(ExitCode::from).unwrap_or_default();
+        lua.set_exit_code(code);
         Ok(())
     })?;
     let process_exit = lua

src/lune/mod.rs

@@ -69,11 +69,15 @@ impl Runtime {
 
         rt.run().await;
 
-        // TODO: Grab exit code that user set, if available
-        let res = rt.get_thread_result(id).unwrap();
-        Ok(match res {
-            Ok(_) => ExitCode::SUCCESS,
-            Err(_) => ExitCode::FAILURE,
-        })
+        if let Some(code) = rt.get_exit_code() {
+            Ok(code)
+        } else if let Some(res) = rt.get_thread_result(id) {
+            Ok(match res {
+                Ok(_) => ExitCode::SUCCESS,
+                Err(_) => ExitCode::FAILURE,
+            })
+        } else {
+            Ok(ExitCode::SUCCESS)
+        }
     }
 }