aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/nvim/lua/executor.c25
-rw-r--r--src/nvim/os/env.h7
-rw-r--r--test/functional/lua/overrides_spec.lua16
3 files changed, 48 insertions, 0 deletions
diff --git a/src/nvim/lua/executor.c b/src/nvim/lua/executor.c
index 93069893cf..d62b5e2c21 100644
--- a/src/nvim/lua/executor.c
+++ b/src/nvim/lua/executor.c
@@ -24,6 +24,10 @@
#include "nvim/undo.h"
#include "nvim/ascii.h"
+#ifdef WIN32
+#include "nvim/os/env.h"
+#endif
+
#include "nvim/lua/executor.h"
#include "nvim/lua/converter.h"
@@ -118,6 +122,14 @@ static int nlua_state_init(lua_State *const lstate) FUNC_ATTR_NONNULL_ALL
lua_setfield(lstate, -2, "debug");
lua_pop(lstate, 1);
+#ifdef WIN32
+ // os.getenv
+ lua_getglobal(lstate, "os");
+ lua_pushcfunction(lstate, &nlua_getenv);
+ lua_setfield(lstate, -2, "getenv");
+ lua_pop(lstate, 1);
+#endif
+
// vim
if (luaL_dostring(lstate, (char *)&vim_module[0])) {
nlua_error(lstate, _("E5106: Error while creating vim module: %.*s"));
@@ -337,6 +349,19 @@ int nlua_debug(lua_State *lstate)
return 0;
}
+#ifdef WIN32
+/// os.getenv implementation: On Windows, uv_os_setenv does not update _environ,
+/// so we need to use os_getenv instead of getenv. Therefore we will apply a
+/// monkey patch here.
+///
+/// @param lstate Lua interpreter state.
+static int nlua_getenv(lua_State *lstate)
+{
+ lua_pushstring(lstate, os_getenv(luaL_checkstring(lstate, 1)));
+ return 1;
+}
+#endif
+
/// Evaluate lua string
///
/// Used for luaeval().
diff --git a/src/nvim/os/env.h b/src/nvim/os/env.h
new file mode 100644
index 0000000000..e0abd87220
--- /dev/null
+++ b/src/nvim/os/env.h
@@ -0,0 +1,7 @@
+#ifndef NVIM_OS_ENV_H
+#define NVIM_OS_ENV_H
+
+#ifdef INCLUDE_GENERATED_DECLARATIONS
+# include "os/env.h.generated.h"
+#endif
+#endif // NVIM_OS_ENV_H
diff --git a/test/functional/lua/overrides_spec.lua b/test/functional/lua/overrides_spec.lua
index 007d40874f..e9d7f3355a 100644
--- a/test/functional/lua/overrides_spec.lua
+++ b/test/functional/lua/overrides_spec.lua
@@ -300,3 +300,19 @@ describe('package.path/package.cpath', function()
eq(new_paths_str, eval_lua('package.path'):sub(1, #new_paths_str))
end)
end)
+
+describe('os.getenv', function()
+ it('returns nothing for not set env var', function()
+ eq(NIL, funcs.luaeval('os.getenv("XTEST_1")'))
+ end)
+ it('returns env var set by the parent process', function()
+ local value = 'foo'
+ clear({env = {['XTEST_1']=value}})
+ eq(value, funcs.luaeval('os.getenv("XTEST_1")'))
+ end)
+ it('returns env var set by let', function()
+ local value = 'foo'
+ meths.command('let $XTEST_1 = "'..value..'"')
+ eq(value, funcs.luaeval('os.getenv("XTEST_1")'))
+ end)
+end)