From 7bfd32750dca1d9f1fa26cbe70356ae578fd2f29 Mon Sep 17 00:00:00 2001
From: bjcscat <bjcs.rbx@gmail.com>
Date: Sun, 25 Aug 2024 10:50:41 -0500
Subject: [PATCH] Change chunk env to use luau's load env parameter (#442)

---
 mlua-sys/src/luau/compat.rs |  9 +++++----
 src/state/raw.rs            | 11 ++++++++++-
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/mlua-sys/src/luau/compat.rs b/mlua-sys/src/luau/compat.rs
index 47ec5b6d..7ca9fbdf 100644
--- a/mlua-sys/src/luau/compat.rs
+++ b/mlua-sys/src/luau/compat.rs
@@ -326,6 +326,7 @@ pub unsafe fn luaL_loadbufferx(
     mut size: usize,
     name: *const c_char,
     mode: *const c_char,
+    env: c_int
 ) -> c_int {
     extern "C" {
         fn free(p: *mut c_void);
@@ -345,12 +346,12 @@ pub unsafe fn luaL_loadbufferx(
 
     if chunk_is_text {
         let data = luau_compile_(data, size, ptr::null_mut(), &mut size);
-        let ok = luau_load(L, name, data, size, 0) == 0;
+        let ok = luau_load(L, name, data, size, env) == 0;
         free(data as *mut c_void);
         if !ok {
             return LUA_ERRSYNTAX;
         }
-    } else if luau_load(L, name, data, size, 0) != 0 {
+    } else if luau_load(L, name, data, size, env) != 0 {
         return LUA_ERRSYNTAX;
     }
     LUA_OK
@@ -361,9 +362,9 @@ pub unsafe fn luaL_loadbuffer(
     L: *mut lua_State,
     data: *const c_char,
     size: usize,
-    name: *const c_char,
+    name: *const c_char
 ) -> c_int {
-    luaL_loadbufferx(L, data, size, name, ptr::null())
+    luaL_loadbufferx(L, data, size, name, ptr::null(), 0)
 }
 
 #[inline(always)]
diff --git a/src/state/raw.rs b/src/state/raw.rs
index 795dc2b0..845594f3 100644
--- a/src/state/raw.rs
+++ b/src/state/raw.rs
@@ -319,13 +319,22 @@ impl RawLua {
                 source.len(),
                 name.map(|n| n.as_ptr()).unwrap_or_else(ptr::null),
                 mode_str,
+                #[cfg(feature="luau")]
+                match &env {
+                    Some(env) => {
+                        self.push_ref(&env.0);
+                        -1
+                    }
+                    _ => 0
+                },
             ) {
                 ffi::LUA_OK => {
                     if let Some(env) = env {
+                        #[cfg(not(feature="luau"))]
                         self.push_ref(&env.0);
                         #[cfg(any(feature = "lua54", feature = "lua53", feature = "lua52"))]
                         ffi::lua_setupvalue(state, -2, 1);
-                        #[cfg(any(feature = "lua51", feature = "luajit", feature = "luau"))]
+                        #[cfg(any(feature = "lua51", feature = "luajit"))]
                         ffi::lua_setfenv(state, -2);
                     }