Mercurial > vim
diff src/evalfunc.c @ 22844:36fc73078bce v8.2.1969
patch 8.2.1969: Vim9: map() may change the list or dict item type
Commit: https://github.com/vim/vim/commit/ea696852e7abcdebaf7f17a7f23dc90df1f5e2ed
Author: Bram Moolenaar <Bram@vim.org>
Date: Mon Nov 9 18:31:39 2020 +0100
patch 8.2.1969: Vim9: map() may change the list or dict item type
Problem: Vim9: map() may change the list or dict item type.
Solution: Add mapnew().
| author | Bram Moolenaar <Bram@vim.org> |
|---|---|
| date | Mon, 09 Nov 2020 18:45:04 +0100 |
| parents | f2fbbb72ff28 |
| children | eea4e7b67f24 |
line wrap: on
line diff
--- a/src/evalfunc.c +++ b/src/evalfunc.c @@ -479,7 +479,6 @@ ret_job(int argcount UNUSED, type_T **ar { return &t_job; } - static type_T * ret_first_arg(int argcount, type_T **argtypes) { @@ -487,6 +486,18 @@ ret_first_arg(int argcount, type_T **arg return argtypes[0]; return &t_void; } +// for map(): returns first argument but item type may differ + static type_T * +ret_first_cont(int argcount UNUSED, type_T **argtypes) +{ + if (argtypes[0]->tt_type == VAR_LIST) + return &t_list_any; + if (argtypes[0]->tt_type == VAR_DICT) + return &t_dict_any; + if (argtypes[0]->tt_type == VAR_BLOB) + return argtypes[0]; + return &t_any; +} /* * Used for getqflist(): returns list if there is no argument, dict if there is @@ -1115,11 +1126,13 @@ static funcentry_T global_functions[] = #endif }, {"map", 2, 2, FEARG_1, NULL, - ret_any, f_map}, + ret_first_cont, f_map}, {"maparg", 1, 4, FEARG_1, NULL, ret_maparg, f_maparg}, {"mapcheck", 1, 3, FEARG_1, NULL, ret_string, f_mapcheck}, + {"mapnew", 2, 2, FEARG_1, NULL, + ret_first_cont, f_mapnew}, {"mapset", 3, 3, FEARG_1, NULL, ret_void, f_mapset}, {"match", 2, 4, FEARG_1, NULL,