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

refactor(generators): style of generating and generated lua dispatch code

Browse Source
This commit is contained in:
bfredl
2024-02-01 12:30:25 +01:00
parent f9d81c43d2
commit ca2635adf9
1 changed files with 70 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

199
src/nvim/generators/gen_api_dispatch.lua
View File

@@ -666,11 +666,11 @@ local function include_headers(output_handle, headers_to_include)
end end
end end
local function write_shifted_output(_, str) local function write_shifted_output(str, ...)
str = str:gsub('\n ', '\n') str = str:gsub('\n ', '\n')
str = str:gsub('^ ', '') str = str:gsub('^ ', '')
str = str:gsub(' +$', '') str = str:gsub(' +$', '')
output:write(str) output:write(string.format(str, ...))
end end
-- start building lua output -- start building lua output
@@ -701,9 +701,7 @@ local lua_c_functions = {}
local function process_function(fn) local function process_function(fn)
local lua_c_function_name = ('nlua_api_%s'):format(fn.name) local lua_c_function_name = ('nlua_api_%s'):format(fn.name)
write_shifted_output( write_shifted_output(
output, [[
string.format(
[[
static int %s(lua_State *lstate) static int %s(lua_State *lstate)
{ {
@@ -714,11 +712,10 @@ local function process_function(fn)
goto exit_0; goto exit_0;
} }
]], ]],
lua_c_function_name, lua_c_function_name,
#fn.parameters, #fn.parameters,
#fn.parameters, #fn.parameters,
(#fn.parameters == 1) and '' or 's' (#fn.parameters == 1) and '' or 's'
)
) )
lua_c_functions[#lua_c_functions + 1] = { lua_c_functions[#lua_c_functions + 1] = {
binding = lua_c_function_name, binding = lua_c_function_name,
@@ -727,38 +724,29 @@ local function process_function(fn)
if not fn.fast then if not fn.fast then
write_shifted_output( write_shifted_output(
output, [[
string.format(
[[
if (!nlua_is_deferred_safe()) { if (!nlua_is_deferred_safe()) {
return luaL_error(lstate, e_luv_api_disabled, "%s"); return luaL_error(lstate, e_luv_api_disabled, "%s");
} }
]], ]],
fn.name fn.name
)
) )
end end
if fn.textlock then if fn.textlock then
write_shifted_output( write_shifted_output([[
output,
[[
if (text_locked()) { if (text_locked()) {
api_set_error(&err, kErrorTypeException, "%s", get_text_locked_msg()); api_set_error(&err, kErrorTypeException, "%%s", get_text_locked_msg());
goto exit_0; goto exit_0;
} }
]] ]])
)
elseif fn.textlock_allow_cmdwin then elseif fn.textlock_allow_cmdwin then
write_shifted_output( write_shifted_output([[
output,
[[
if (textlock != 0 || expr_map_locked()) { if (textlock != 0 || expr_map_locked()) {
api_set_error(&err, kErrorTypeException, "%s", e_textlock); api_set_error(&err, kErrorTypeException, "%%s", e_textlock);
goto exit_0; goto exit_0;
} }
]] ]])
)
end end
local cparams = '' local cparams = ''
@@ -776,44 +764,37 @@ local function process_function(fn)
local seterr = '' local seterr = ''
if string.match(param_type, '^KeyDict_') then if string.match(param_type, '^KeyDict_') then
write_shifted_output( write_shifted_output(
output, [[
string.format( %s %s = { 0 };
[[ nlua_pop_keydict(lstate, &%s, %s_get_field, &err_param, &err);
%s %s = { 0 }; nlua_pop_keydict(lstate, &%s, %s_get_field, &err_param, &err);]], ]],
param_type, param_type,
cparam, cparam,
cparam, cparam,
param_type param_type
)
) )
cparam = '&' .. cparam cparam = '&' .. cparam
errshift = 1 -- free incomplete dict on error errshift = 1 -- free incomplete dict on error
else else
write_shifted_output( write_shifted_output(
output, [[
string.format( const %s %s = nlua_pop_%s(lstate, %s&err);]],
[[ param[1],
const %s %s = nlua_pop_%s(lstate, %s&err);]], cparam,
param[1], param_type,
cparam, extra
param_type,
extra
)
) )
seterr = [[ seterr = '\n err_param = "' .. param[2] .. '";'
err_param = "]] .. param[2] .. [[";]]
end end
write_shifted_output( write_shifted_output([[
output,
string.format([[
if (ERROR_SET(&err)) {]] .. seterr .. [[ if (ERROR_SET(&err)) {]] .. seterr .. [[
goto exit_%u; goto exit_%u;
} }
]], #fn.parameters - j + errshift) ]], #fn.parameters - j + errshift)
)
free_code[#free_code + 1] = ('api_free_%s(%s);'):format(lc_param_type, cparam) free_code[#free_code + 1] = ('api_free_%s(%s);'):format(lc_param_type, cparam)
cparams = cparam .. ', ' .. cparams cparams = cparam .. ', ' .. cparams
end end
@@ -822,12 +803,7 @@ local function process_function(fn)
end end
if fn.arena_return then if fn.arena_return then
cparams = cparams .. '&arena, ' cparams = cparams .. '&arena, '
write_shifted_output( write_shifted_output(' Arena arena = ARENA_EMPTY;\n')
output,
[[
Arena arena = ARENA_EMPTY;
]]
)
end end
if fn.has_lua_imp then if fn.has_lua_imp then
@@ -844,27 +820,27 @@ local function process_function(fn)
local rev_i = #free_code - i + 1 local rev_i = #free_code - i + 1
local code = free_code[rev_i] local code = free_code[rev_i]
if i == 1 and not string.match(real_type(fn.parameters[1][1]), '^KeyDict_') then if i == 1 and not string.match(real_type(fn.parameters[1][1]), '^KeyDict_') then
free_at_exit_code = free_at_exit_code .. ('\n %s'):format(code) free_at_exit_code = free_at_exit_code .. ('\n %s'):format(code)
else else
free_at_exit_code = free_at_exit_code .. ('\n exit_%u:\n %s'):format(rev_i, code) free_at_exit_code = free_at_exit_code .. ('\nexit_%u:\n %s'):format(rev_i, code)
end end
end end
local err_throw_code = [[ local err_throw_code = [[
exit_0: exit_0:
if (ERROR_SET(&err)) { if (ERROR_SET(&err)) {
luaL_where(lstate, 1); luaL_where(lstate, 1);
if (err_param) { if (err_param) {
lua_pushstring(lstate, "Invalid '"); lua_pushstring(lstate, "Invalid '");
lua_pushstring(lstate, err_param); lua_pushstring(lstate, err_param);
lua_pushstring(lstate, "': "); lua_pushstring(lstate, "': ");
}
lua_pushstring(lstate, err.msg);
api_clear_error(&err);
lua_concat(lstate, err_param ? 5 : 2);
return lua_error(lstate);
} }
]] lua_pushstring(lstate, err.msg);
api_clear_error(&err);
lua_concat(lstate, err_param ? 5 : 2);
return lua_error(lstate);
}
]]
local return_type local return_type
if fn.return_type ~= 'void' then if fn.return_type ~= 'void' then
if fn.return_type:match('^ArrayOf') then if fn.return_type:match('^ArrayOf') then
@@ -874,97 +850,62 @@ local function process_function(fn)
end end
local free_retval local free_retval
if fn.arena_return then if fn.arena_return then
free_retval = 'arena_mem_free(arena_finish(&arena));' free_retval = ' arena_mem_free(arena_finish(&arena));'
else else
free_retval = 'api_free_' .. return_type:lower() .. '(ret);' free_retval = ' api_free_' .. return_type:lower() .. '(ret);'
end end
write_shifted_output( write_shifted_output(' %s ret = %s(%s);\n', fn.return_type, fn.name, cparams)
output,
string.format(
[[
%s ret = %s(%s);
]],
fn.return_type,
fn.name,
cparams
)
)
local ret_type = real_type(fn.return_type) local ret_type = real_type(fn.return_type)
if fn.has_lua_imp then if fn.has_lua_imp then
-- only push onto the Lua stack if we haven't already -- only push onto the Lua stack if we haven't already
write_shifted_output( write_shifted_output(string.format(
output, [[
string.format(
[[
if (lua_gettop(lstate) == 0) { if (lua_gettop(lstate) == 0) {
nlua_push_%s(lstate, ret, true); nlua_push_%s(lstate, ret, true);
} }
]], ]],
return_type return_type
) ))
)
elseif string.match(ret_type, '^KeyDict_') then elseif string.match(ret_type, '^KeyDict_') then
write_shifted_output( write_shifted_output(
output, ' nlua_push_keydict(lstate, &ret, %s_table);\n',
string.format( string.sub(ret_type, 9)
[[
nlua_push_keydict(lstate, &ret, %s_table);
]],
string.sub(ret_type, 9)
)
) )
else else
local special = (fn.since ~= nil and fn.since < 11) local special = (fn.since ~= nil and fn.since < 11)
write_shifted_output( write_shifted_output(' nlua_push_%s(lstate, ret, %s);\n', return_type, tostring(special))
output,
string.format(
[[
nlua_push_%s(lstate, ret, %s);
]],
return_type,
tostring(special)
)
)
end end
write_shifted_output( write_shifted_output(
output,
string.format( [[
[[
%s %s
%s %s
%s %s
return 1; return 1;
]], ]],
free_retval, free_retval,
free_at_exit_code, free_at_exit_code,
err_throw_code err_throw_code
)
) )
else else
write_shifted_output( write_shifted_output(
output, [[
string.format(
[[
%s(%s); %s(%s);
%s %s
%s %s
return 0; return 0;
]], ]],
fn.name, fn.name,
cparams, cparams,
free_at_exit_code, free_at_exit_code,
err_throw_code err_throw_code
)
) )
end end
write_shifted_output( write_shifted_output([[
output,
[[
} }
]] ]])
)
end end
for _, fn in ipairs(functions) do for _, fn in ipairs(functions) do