mirror of
https://github.com/neovim/neovim.git
synced 2025-02-25 18:55:25 -06:00
f_environ: cleanup/refactor
- use os_getenvname_at_index / os_getenv - f_getenv: empty (*p == NUL) is not null (undefined)
This commit is contained in:
parent
fd66ad2262
commit
d55b12ea50
@ -8498,48 +8498,19 @@ static void f_empty(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
|||||||
/// "environ()" function
|
/// "environ()" function
|
||||||
static void f_environ(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
static void f_environ(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
||||||
{
|
{
|
||||||
int i = 0;
|
|
||||||
char_u *entry, *value;
|
|
||||||
# ifdef WIN32
|
|
||||||
extern wchar_t **_wenviron;
|
|
||||||
# else
|
|
||||||
extern char **environ;
|
|
||||||
# endif
|
|
||||||
|
|
||||||
tv_dict_alloc_ret(rettv);
|
tv_dict_alloc_ret(rettv);
|
||||||
|
|
||||||
# ifdef WIN32
|
for (int i = 0; ; i++) {
|
||||||
if (*_wenviron == NULL) {
|
// TODO(justinmk): use os_copyfullenv from #7202 ?
|
||||||
return;
|
char *envname = os_getenvname_at_index((size_t)i);
|
||||||
}
|
if (envname == NULL) {
|
||||||
# else
|
break;
|
||||||
if (*environ == NULL) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
# endif
|
|
||||||
|
|
||||||
for (i = 0; ; i++) {
|
|
||||||
# ifdef WIN32
|
|
||||||
uint16_t *p;
|
|
||||||
|
|
||||||
if ((p = (uint16_t *)_wenviron[i]) == NULL) {
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
entry = utf16_to_enc(p, NULL);
|
const char *value = os_getenv(envname);
|
||||||
# else
|
tv_dict_add_str(rettv->vval.v_dict,
|
||||||
if ((entry = (char_u *)environ[i]) == NULL) {
|
(char *)envname, STRLEN((char *)envname),
|
||||||
return;
|
value == NULL ? "" : value);
|
||||||
}
|
xfree(envname);
|
||||||
entry = vim_strsave(entry);
|
|
||||||
# endif
|
|
||||||
if ((value = vim_strchr(entry, '=')) == NULL) {
|
|
||||||
xfree(entry);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
*value++ = NUL;
|
|
||||||
tv_dict_add_str(rettv->vval.v_dict, (char *)entry, STRLEN((char *)entry),
|
|
||||||
(const char *)value);
|
|
||||||
xfree(entry);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -8561,12 +8532,11 @@ static void f_getenv(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
|||||||
{
|
{
|
||||||
char_u *p = (char_u *)vim_getenv(tv_get_string(&argvars[0]));
|
char_u *p = (char_u *)vim_getenv(tv_get_string(&argvars[0]));
|
||||||
|
|
||||||
if (p == NULL || *p == NUL) {
|
if (p == NULL) {
|
||||||
rettv->v_type = VAR_SPECIAL;
|
rettv->v_type = VAR_SPECIAL;
|
||||||
rettv->vval.v_number = kSpecialVarNull;
|
rettv->vval.v_number = kSpecialVarNull;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
p = vim_strsave(p);
|
|
||||||
rettv->vval.v_string = p;
|
rettv->vval.v_string = p;
|
||||||
rettv->v_type = VAR_STRING;
|
rettv->v_type = VAR_STRING;
|
||||||
}
|
}
|
||||||
|
@ -45,6 +45,7 @@ void env_init(void)
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Like getenv(), but returns NULL if the variable is empty.
|
/// Like getenv(), but returns NULL if the variable is empty.
|
||||||
|
/// @see os_env_exists
|
||||||
const char *os_getenv(const char *name)
|
const char *os_getenv(const char *name)
|
||||||
FUNC_ATTR_NONNULL_ALL
|
FUNC_ATTR_NONNULL_ALL
|
||||||
{
|
{
|
||||||
|
12
test/functional/eval/environ_spec.lua
Normal file
12
test/functional/eval/environ_spec.lua
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
local helpers = require('test.functional.helpers')(after_each)
|
||||||
|
local clear = helpers.clear
|
||||||
|
local eq = helpers.eq
|
||||||
|
local environ = helpers.funcs.environ
|
||||||
|
|
||||||
|
describe('environ()', function()
|
||||||
|
it('handles empty env variable', function()
|
||||||
|
clear({env={EMPTY_VAR=""}})
|
||||||
|
eq("", environ()['EMPTY_VAR'])
|
||||||
|
eq(nil, environ()['DOES_NOT_EXIST'])
|
||||||
|
end)
|
||||||
|
end)
|
Loading…
Reference in New Issue
Block a user