fix: use lua stdlib in scripts (fix #437)

Author: boltlessengineer

lua/rest-nvim/script/lua.lua

@@ -32,8 +32,6 @@ local function create_prescript_env(ctx)
                 end,
             },
         },
-        ---Global variable `vim` you are familiar of
-        vim = vim,
     }
     return env
 end
@@ -77,8 +75,6 @@ local function create_handler_env(ctx, res)
         ---Raw response object
         ---@type rest.Response
         response = res,
-        ---Global variable `vim` you are familiar of
-        vim = vim,
     }
     return env
 end
@@ -88,6 +84,7 @@ end
 ---@param env table
 ---@return function
 local function load_lua(s, env)
+    env = setmetatable(env, { __index = _G })
     local f, error_msg = load(s, "script_variable", "bt", env)
     if error_msg then
         logger.error(error_msg)

spec/script_spec.lua

@@ -74,4 +74,15 @@ client.global.set("MYVAR", json.var)
         end)
         assert.same("boo", ctx:resolve("MYVAR"))
     end)
+    it("use lua stdlib from script", function ()
+        local ctx = Context:new()
+        local res = {}
+        local script = [[
+        local a = 123
+        vim.env["__TEST"] = tostring(a)
+        ]]
+        local h = require("rest-nvim.script.lua").load_post_req_hook(script, ctx)
+        h(res)
+        assert.same(vim.env["__TEST"], "123")
+    end)
 end)