0
0
mirror of https://github.com/vim/vim.git synced 2025-10-09 06:14:17 -04:00

patch 8.2.1081: Lua: cannot use table.insert() and table.remove()

Problem:    Lua: cannot use table.insert() and table.remove().
Solution:   Add the list functions. (Prabir Shrestha, closes #6353)
This commit is contained in:
Bram Moolenaar
2020-06-28 22:41:26 +02:00
parent b7a78f7a67
commit a1f9f8666e
4 changed files with 64 additions and 11 deletions

View File

@@ -913,19 +913,32 @@ luaV_list_newindex(lua_State *L)
if (l->lv_lock)
luaL_error(L, "list is locked");
li = list_find(l, n);
if (li == NULL) return 0;
if (lua_isnil(L, 3)) // remove?
if (li == NULL)
{
vimlist_remove(l, li, li);
listitem_free(l, li);
if (!lua_isnil(L, 3))
{
typval_T v;
luaV_checktypval(L, 3, &v, "inserting list item");
if (list_insert_tv(l, &v, li) == FAIL)
luaL_error(L, "failed to add item to list");
clear_tv(&v);
}
}
else
{
typval_T v;
luaV_checktypval(L, 3, &v, "setting list item");
clear_tv(&li->li_tv);
copy_tv(&v, &li->li_tv);
clear_tv(&v);
if (lua_isnil(L, 3)) // remove?
{
vimlist_remove(l, li, li);
listitem_free(l, li);
}
else
{
typval_T v;
luaV_checktypval(L, 3, &v, "setting list item");
clear_tv(&li->li_tv);
copy_tv(&v, &li->li_tv);
clear_tv(&v);
}
}
return 0;
}

View File

@@ -353,6 +353,32 @@ func Test_lua_list_table()
call assert_fails('lua vim.list(true)', '[string "vim chunk"]:1: table expected, got boolean')
endfunc
func Test_lua_list_table_insert_remove()
let luaver = split(split(luaeval('_VERSION'), ' ')[1], '\.')
let major = str2nr(luaver[0])
let minor = str2nr(luaver[1])
if major < 5 || (major == 5 && minor < 3)
throw 'Skipped: Lua version < 5.3'
endif
let l = [1, 2]
lua t = vim.eval('l')
lua table.insert(t, 10)
lua t[#t + 1] = 20
lua table.insert(t, 2, 30)
call assert_equal(l, [1, 30, 2, 10, 20])
lua table.remove(t, 2)
call assert_equal(l, [1, 2, 10, 20])
lua t[3] = nil
call assert_equal(l, [1, 2, 20])
lua removed_value = table.remove(t, 3)
call assert_equal(luaeval('removed_value'), 20)
lua t = nil
lua removed_value = nil
unlet l
endfunc
" Test l() i.e. iterator on list
func Test_lua_list_iter()
lua l = vim.list():add('foo'):add('bar')

View File

@@ -754,6 +754,8 @@ static char *(features[]) =
static int included_patches[] =
{ /* Add new patch number below this line */
/**/
1081,
/**/
1080,
/**/