feat(lua): add <f-args> to user commands callback (#17522)

Works similar to ex <f-args>. It only splits the arguments if the
command has more than one posible argument. In cases were the command
can only have 1 argument opts.fargs = { opts.args }
This commit is contained in:
Javier Lopez
2022-02-27 14:35:06 -05:00
committed by GitHub
parent c65d93e60a
commit 1b5767aa34
5 changed files with 109 additions and 11 deletions

View File

@@ -1814,8 +1814,31 @@ void nlua_do_ucmd(ucmd_T *cmd, exarg_T *eap)
lua_pushinteger(lstate, eap->line2);
lua_setfield(lstate, -2, "line2");
lua_newtable(lstate); // f-args table
lua_pushstring(lstate, (const char *)eap->arg);
lua_setfield(lstate, -2, "args");
lua_pushvalue(lstate, -1); // Reference for potential use on f-args
lua_setfield(lstate, -4, "args");
// Split args by unescaped whitespace |<f-args>| (nargs dependent)
if (cmd->uc_argt & EX_NOSPC) {
// Commands where nargs = 1 or "?" fargs is the same as args
lua_rawseti(lstate, -2, 1);
} else {
// Commands with more than one possible argument we split
lua_pop(lstate, 1); // Pop the reference of opts.args
int length = (int)STRLEN(eap->arg);
int start = 0;
int end = 0;
int i = 1;
bool res = true;
while (res) {
res = uc_split_args_iter(eap->arg, i, &start, &end, length);
lua_pushlstring(lstate, (const char *)eap->arg + start, (size_t)(end - start + 1));
lua_rawseti(lstate, -2, i);
i++;
}
}
lua_setfield(lstate, -2, "fargs");
lua_pushstring(lstate, (const char *)&eap->regname);
lua_setfield(lstate, -2, "reg");