mirrors/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2025-09-28 05:58:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

vim-patch:9.1.0547: No way to get the arity of a Vim function (#29638)

Browse Source 
Problem:  No way to get the arity of a Vim function
          (Austin Ziegler)
Solution: Enhance get() Vim script function to return the function
          argument info using get(func, "arity") (LemonBoy)

fixes: vim/vim#15097
closes: vim/vim#15109

48b7d05a4f

Co-authored-by: LemonBoy <thatlemon@gmail.com>
This commit is contained in:
zeertzjq
2024-07-10 08:07:16 +08:00
committed by GitHub
parent f3c7fb9db1
commit 545aafbeb8
8 changed files with 168 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/nvim/eval/funcs.c
View File

@@ -2372,6 +2372,33 @@ static void f_get(typval_T *argvars, typval_T *rettv, EvalFuncData fptr)
for (int i = 0; i < pt->pt_argc; i++) {
tv_list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
}
} else if (strcmp(what, "arity") == 0) {
int required = 0;
int optional = 0;
bool varargs = false;
const char *name = partial_name(pt);
get_func_arity(name, &required, &optional, &varargs);
rettv->v_type = VAR_DICT;
tv_dict_alloc_ret(rettv);
dict_T *dict = rettv->vval.v_dict;
// Take into account the arguments of the partial, if any.
// Note that it is possible to supply more arguments than the function
// accepts.
if (pt->pt_argc >= required + optional) {
required = optional = 0;
} else if (pt->pt_argc > required) {
optional -= pt->pt_argc - required;
required = 0;
} else {
required -= pt->pt_argc;
}
tv_dict_add_nr(dict, S_LEN("required"), required);
tv_dict_add_nr(dict, S_LEN("optional"), optional);
tv_dict_add_bool(dict, S_LEN("varargs"), varargs);
} else {
semsg(_(e_invarg2), what);
}