diff options
author | Daniel Hahler <git@thequod.de> | 2019-07-30 08:00:36 +0200 |
---|---|---|
committer | Justin M. Keyes <justinkz@gmail.com> | 2019-08-06 01:17:32 +0200 |
commit | fd66ad226221a22f223dcfca337e893d0a9cca46 (patch) | |
tree | 7638ea511f479ee980e1cf8c43b41d7afc98f9e9 /src/nvim/eval.c | |
parent | b09e03c64d0f38a43b5ef068141bc86e365cd7fa (diff) | |
download | rneovim-fd66ad226221a22f223dcfca337e893d0a9cca46.tar.gz rneovim-fd66ad226221a22f223dcfca337e893d0a9cca46.tar.bz2 rneovim-fd66ad226221a22f223dcfca337e893d0a9cca46.zip |
vim-patch:8.1.1305: there is no easy way to manipulate environment variables
Problem: There is no easy way to manipulate environment variables.
Solution: Add environ(), getenv() and setenv(). (Yasuhiro Matsumoto,
closes vim/vim#2875)
https://github.com/vim/vim/commit/691ddeefb545d8488e5a495af61caba2e57b3de9
Diffstat (limited to 'src/nvim/eval.c')
-rw-r--r-- | src/nvim/eval.c | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/src/nvim/eval.c b/src/nvim/eval.c index d82a081c27..91c82a393e 100644 --- a/src/nvim/eval.c +++ b/src/nvim/eval.c @@ -8495,6 +8495,54 @@ static void f_empty(typval_T *argvars, typval_T *rettv, FunPtr fptr) rettv->vval.v_number = n; } +/// "environ()" function +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); + +# ifdef WIN32 + if (*_wenviron == NULL) { + return; + } +# else + 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); +# else + if ((entry = (char_u *)environ[i]) == NULL) { + return; + } + 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); + } +} + /* * "escape({string}, {chars})" function */ @@ -8508,6 +8556,21 @@ static void f_escape(typval_T *argvars, typval_T *rettv, FunPtr fptr) rettv->v_type = VAR_STRING; } +/// "getenv()" function +static void f_getenv(typval_T *argvars, typval_T *rettv, FunPtr fptr) +{ + char_u *p = (char_u *)vim_getenv(tv_get_string(&argvars[0])); + + if (p == NULL || *p == NUL) { + rettv->v_type = VAR_SPECIAL; + rettv->vval.v_number = kSpecialVarNull; + return; + } + p = vim_strsave(p); + rettv->vval.v_string = p; + rettv->v_type = VAR_STRING; +} + /* * "eval()" function */ @@ -15319,6 +15382,20 @@ static void f_setcmdpos(typval_T *argvars, typval_T *rettv, FunPtr fptr) } } +/// "setenv()" function +static void f_setenv(typval_T *argvars, typval_T *rettv, FunPtr fptr) +{ + char namebuf[NUMBUFLEN]; + char valbuf[NUMBUFLEN]; + const char *name = tv_get_string_buf(&argvars[0], namebuf); + + if (argvars[1].v_type == VAR_SPECIAL + && argvars[1].vval.v_number == kSpecialVarNull) { + os_unsetenv(name); + } else { + vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf)); + } +} /// "setfperm({fname}, {mode})" function static void f_setfperm(typval_T *argvars, typval_T *rettv, FunPtr fptr) |