blob: 77f2afe54b92a5aad4b5aef7cf0d4aaa3fb76c13 [file] [log] [blame]
Bram Moolenaar1dced572012-04-05 16:54:08 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Lua interface by Luis Carvalho
6 *
7 * Do ":help uganda" in Vim to read copying and usage conditions.
8 * Do ":help credits" in Vim to see a list of people who contributed.
9 * See README.txt for an overview of the Vim source code.
10 */
11
Bram Moolenaare2793352011-01-17 19:53:27 +010012#include "vim.h"
13
Bram Moolenaar0ba04292010-07-14 23:23:17 +020014#include <lua.h>
15#include <lualib.h>
16#include <lauxlib.h>
Bram Moolenaar0ba04292010-07-14 23:23:17 +020017
Bram Moolenaar2ab2e862019-12-04 21:24:53 +010018// Only do the following when the feature is enabled. Needed for "make
19// depend".
Bram Moolenaar0ba04292010-07-14 23:23:17 +020020#if defined(FEAT_LUA) || defined(PROTO)
21
22#define LUAVIM_CHUNKNAME "vim chunk"
23#define LUAVIM_NAME "vim"
Bram Moolenaar1dced572012-04-05 16:54:08 +020024#define LUAVIM_EVALNAME "luaeval"
25#define LUAVIM_EVALHEADER "local _A=select(1,...) return "
Bram Moolenaar0ba04292010-07-14 23:23:17 +020026
27typedef buf_T *luaV_Buffer;
28typedef win_T *luaV_Window;
Bram Moolenaar1dced572012-04-05 16:54:08 +020029typedef dict_T *luaV_Dict;
30typedef list_T *luaV_List;
Bram Moolenaarb7828692019-03-23 13:57:02 +010031typedef blob_T *luaV_Blob;
Bram Moolenaarca06da92018-07-01 15:12:05 +020032typedef struct {
Bram Moolenaar4eefe472019-03-19 21:59:19 +010033 char_u *name; // funcref
Bram Moolenaarca06da92018-07-01 15:12:05 +020034 dict_T *self; // selfdict
35} luaV_Funcref;
Bram Moolenaar0ba04292010-07-14 23:23:17 +020036typedef void (*msgfunc_T)(char_u *);
37
Bram Moolenaar801ab062020-06-25 19:27:56 +020038typedef struct {
39 int lua_funcref; // ref to a lua func
40 int lua_tableref; // ref to a lua table if metatable else LUA_NOREF. used
41 // for __call
42 lua_State *L;
43} luaV_CFuncState;
44
Bram Moolenaar1dced572012-04-05 16:54:08 +020045static const char LUAVIM_DICT[] = "dict";
46static const char LUAVIM_LIST[] = "list";
Bram Moolenaarb7828692019-03-23 13:57:02 +010047static const char LUAVIM_BLOB[] = "blob";
Bram Moolenaarca06da92018-07-01 15:12:05 +020048static const char LUAVIM_FUNCREF[] = "funcref";
Bram Moolenaar0ba04292010-07-14 23:23:17 +020049static const char LUAVIM_BUFFER[] = "buffer";
50static const char LUAVIM_WINDOW[] = "window";
51static const char LUAVIM_FREE[] = "luaV_free";
Bram Moolenaar1dced572012-04-05 16:54:08 +020052static const char LUAVIM_LUAEVAL[] = "luaV_luaeval";
53static const char LUAVIM_SETREF[] = "luaV_setref";
Bram Moolenaar0ba04292010-07-14 23:23:17 +020054
Bram Moolenaar801ab062020-06-25 19:27:56 +020055static const char LUA___CALL[] = "__call";
56
Bram Moolenaar2ab2e862019-12-04 21:24:53 +010057// most functions are closures with a cache table as first upvalue;
58// get/setudata manage references to vim userdata in cache table through
59// object pointers (light userdata)
Bram Moolenaar1dced572012-04-05 16:54:08 +020060#define luaV_getudata(L, v) \
61 lua_pushlightuserdata((L), (void *) (v)); \
62 lua_rawget((L), lua_upvalueindex(1))
63#define luaV_setudata(L, v) \
64 lua_pushlightuserdata((L), (void *) (v)); \
65 lua_pushvalue((L), -2); \
66 lua_rawset((L), lua_upvalueindex(1))
Bram Moolenaar0ba04292010-07-14 23:23:17 +020067#define luaV_getfield(L, s) \
68 lua_pushlightuserdata((L), (void *)(s)); \
69 lua_rawget((L), LUA_REGISTRYINDEX)
70#define luaV_checksandbox(L) \
71 if (sandbox) luaL_error((L), "not allowed in sandbox")
72#define luaV_msg(L) luaV_msgfunc((L), (msgfunc_T) msg)
73#define luaV_emsg(L) luaV_msgfunc((L), (msgfunc_T) emsg)
Bram Moolenaarca06da92018-07-01 15:12:05 +020074#define luaV_checktypval(L, a, v, msg) \
75 do { \
Bram Moolenaar801ab062020-06-25 19:27:56 +020076 if (luaV_totypval(L, a, v) == FAIL) \
Bram Moolenaarca06da92018-07-01 15:12:05 +020077 luaL_error(L, msg ": cannot convert value"); \
78 } while (0)
Bram Moolenaar0ba04292010-07-14 23:23:17 +020079
Bram Moolenaarca06da92018-07-01 15:12:05 +020080static luaV_List *luaV_pushlist(lua_State *L, list_T *lis);
81static luaV_Dict *luaV_pushdict(lua_State *L, dict_T *dic);
Bram Moolenaarb7828692019-03-23 13:57:02 +010082static luaV_Blob *luaV_pushblob(lua_State *L, blob_T *blo);
Bram Moolenaar4eefe472019-03-19 21:59:19 +010083static luaV_Funcref *luaV_pushfuncref(lua_State *L, char_u *name);
Bram Moolenaar801ab062020-06-25 19:27:56 +020084static int luaV_call_lua_func(int argcount, typval_T *argvars, typval_T *rettv, void *state);
85static void luaV_call_lua_func_free(void *state);
Bram Moolenaar1dced572012-04-05 16:54:08 +020086
87#if LUA_VERSION_NUM <= 501
88#define luaV_openlib(L, l, n) luaL_openlib(L, NULL, l, n)
89#define luaL_typeerror luaL_typerror
90#else
91#define luaV_openlib luaL_setfuncs
92#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +020093
94#ifdef DYNAMIC_LUA
Bram Moolenaar2334b6d2010-07-22 21:32:16 +020095
Bram Moolenaar4f974752019-02-17 17:44:42 +010096#ifndef MSWIN
Bram Moolenaar2334b6d2010-07-22 21:32:16 +020097# include <dlfcn.h>
98# define HANDLE void*
99# define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
100# define symbol_from_dll dlsym
101# define close_dll dlclose
102#else
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200103# define load_dll vimLoadLib
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200104# define symbol_from_dll GetProcAddress
105# define close_dll FreeLibrary
106#endif
107
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100108// lauxlib
Bram Moolenaar1dced572012-04-05 16:54:08 +0200109#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200110#define luaL_register dll_luaL_register
Bram Moolenaar1dced572012-04-05 16:54:08 +0200111#define luaL_prepbuffer dll_luaL_prepbuffer
112#define luaL_openlib dll_luaL_openlib
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200113#define luaL_typerror dll_luaL_typerror
Bram Moolenaar1dced572012-04-05 16:54:08 +0200114#define luaL_loadfile dll_luaL_loadfile
115#define luaL_loadbuffer dll_luaL_loadbuffer
116#else
117#define luaL_prepbuffsize dll_luaL_prepbuffsize
118#define luaL_setfuncs dll_luaL_setfuncs
119#define luaL_loadfilex dll_luaL_loadfilex
120#define luaL_loadbufferx dll_luaL_loadbufferx
121#define luaL_argerror dll_luaL_argerror
122#endif
Bram Moolenaar5551b132020-07-14 21:54:28 +0200123#if LUA_VERSION_NUM >= 504
124#define luaL_typeerror dll_luaL_typeerror
125#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200126#define luaL_checkany dll_luaL_checkany
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200127#define luaL_checklstring dll_luaL_checklstring
128#define luaL_checkinteger dll_luaL_checkinteger
129#define luaL_optinteger dll_luaL_optinteger
130#define luaL_checktype dll_luaL_checktype
131#define luaL_error dll_luaL_error
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200132#define luaL_newstate dll_luaL_newstate
133#define luaL_buffinit dll_luaL_buffinit
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200134#define luaL_addlstring dll_luaL_addlstring
135#define luaL_pushresult dll_luaL_pushresult
Bram Moolenaardf1643a2020-05-17 18:53:27 +0200136#define luaL_loadstring dll_luaL_loadstring
Bram Moolenaar1e4c7d02020-06-25 20:56:42 +0200137#define luaL_ref dll_luaL_ref
138#define luaL_unref dll_luaL_unref
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100139// lua
Bram Moolenaar1dced572012-04-05 16:54:08 +0200140#if LUA_VERSION_NUM <= 501
141#define lua_tonumber dll_lua_tonumber
142#define lua_tointeger dll_lua_tointeger
143#define lua_call dll_lua_call
144#define lua_pcall dll_lua_pcall
145#else
146#define lua_tonumberx dll_lua_tonumberx
147#define lua_tointegerx dll_lua_tointegerx
148#define lua_callk dll_lua_callk
149#define lua_pcallk dll_lua_pcallk
150#define lua_getglobal dll_lua_getglobal
151#define lua_setglobal dll_lua_setglobal
Bram Moolenaar1dced572012-04-05 16:54:08 +0200152#endif
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200153#if LUA_VERSION_NUM <= 502
154#define lua_replace dll_lua_replace
155#define lua_remove dll_lua_remove
156#endif
157#if LUA_VERSION_NUM >= 503
158#define lua_rotate dll_lua_rotate
159#define lua_copy dll_lua_copy
160#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200161#define lua_typename dll_lua_typename
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200162#define lua_close dll_lua_close
163#define lua_gettop dll_lua_gettop
164#define lua_settop dll_lua_settop
165#define lua_pushvalue dll_lua_pushvalue
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200166#define lua_isnumber dll_lua_isnumber
167#define lua_isstring dll_lua_isstring
168#define lua_type dll_lua_type
169#define lua_rawequal dll_lua_rawequal
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200170#define lua_toboolean dll_lua_toboolean
171#define lua_tolstring dll_lua_tolstring
172#define lua_touserdata dll_lua_touserdata
173#define lua_pushnil dll_lua_pushnil
174#define lua_pushnumber dll_lua_pushnumber
175#define lua_pushinteger dll_lua_pushinteger
176#define lua_pushlstring dll_lua_pushlstring
177#define lua_pushstring dll_lua_pushstring
178#define lua_pushfstring dll_lua_pushfstring
179#define lua_pushcclosure dll_lua_pushcclosure
180#define lua_pushboolean dll_lua_pushboolean
181#define lua_pushlightuserdata dll_lua_pushlightuserdata
182#define lua_getfield dll_lua_getfield
183#define lua_rawget dll_lua_rawget
Bram Moolenaar1dced572012-04-05 16:54:08 +0200184#define lua_rawgeti dll_lua_rawgeti
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200185#define lua_createtable dll_lua_createtable
Bram Moolenaar830e3582018-08-21 14:23:35 +0200186#if LUA_VERSION_NUM >= 504
187 #define lua_newuserdatauv dll_lua_newuserdatauv
188#else
189 #define lua_newuserdata dll_lua_newuserdata
190#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200191#define lua_getmetatable dll_lua_getmetatable
192#define lua_setfield dll_lua_setfield
193#define lua_rawset dll_lua_rawset
194#define lua_rawseti dll_lua_rawseti
195#define lua_setmetatable dll_lua_setmetatable
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200196#define lua_next dll_lua_next
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100197// libs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200198#define luaopen_base dll_luaopen_base
199#define luaopen_table dll_luaopen_table
200#define luaopen_string dll_luaopen_string
201#define luaopen_math dll_luaopen_math
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200202#define luaopen_io dll_luaopen_io
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200203#define luaopen_os dll_luaopen_os
204#define luaopen_package dll_luaopen_package
205#define luaopen_debug dll_luaopen_debug
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200206#define luaL_openlibs dll_luaL_openlibs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200207
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100208// lauxlib
Bram Moolenaar1dced572012-04-05 16:54:08 +0200209#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200210void (*dll_luaL_register) (lua_State *L, const char *libname, const luaL_Reg *l);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200211char *(*dll_luaL_prepbuffer) (luaL_Buffer *B);
212void (*dll_luaL_openlib) (lua_State *L, const char *libname, const luaL_Reg *l, int nup);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200213int (*dll_luaL_typerror) (lua_State *L, int narg, const char *tname);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200214int (*dll_luaL_loadfile) (lua_State *L, const char *filename);
215int (*dll_luaL_loadbuffer) (lua_State *L, const char *buff, size_t sz, const char *name);
216#else
217char *(*dll_luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
218void (*dll_luaL_setfuncs) (lua_State *L, const luaL_Reg *l, int nup);
219int (*dll_luaL_loadfilex) (lua_State *L, const char *filename, const char *mode);
220int (*dll_luaL_loadbufferx) (lua_State *L, const char *buff, size_t sz, const char *name, const char *mode);
221int (*dll_luaL_argerror) (lua_State *L, int numarg, const char *extramsg);
222#endif
Bram Moolenaar5551b132020-07-14 21:54:28 +0200223#if LUA_VERSION_NUM >= 504
224int (*dll_luaL_typeerror) (lua_State *L, int narg, const char *tname);
225#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200226void (*dll_luaL_checkany) (lua_State *L, int narg);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200227const char *(*dll_luaL_checklstring) (lua_State *L, int numArg, size_t *l);
228lua_Integer (*dll_luaL_checkinteger) (lua_State *L, int numArg);
229lua_Integer (*dll_luaL_optinteger) (lua_State *L, int nArg, lua_Integer def);
230void (*dll_luaL_checktype) (lua_State *L, int narg, int t);
231int (*dll_luaL_error) (lua_State *L, const char *fmt, ...);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200232lua_State *(*dll_luaL_newstate) (void);
233void (*dll_luaL_buffinit) (lua_State *L, luaL_Buffer *B);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200234void (*dll_luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
235void (*dll_luaL_pushresult) (luaL_Buffer *B);
Bram Moolenaardf1643a2020-05-17 18:53:27 +0200236int (*dll_luaL_loadstring) (lua_State *L, const char *s);
Bram Moolenaar1e4c7d02020-06-25 20:56:42 +0200237int (*dll_luaL_ref) (lua_State *L, int idx);
238#if LUA_VERSION_NUM <= 502
239void (*dll_luaL_unref) (lua_State *L, int idx, int n);
240#else
241void (*dll_luaL_unref) (lua_State *L, int idx, lua_Integer n);
242#endif
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100243// lua
Bram Moolenaar1dced572012-04-05 16:54:08 +0200244#if LUA_VERSION_NUM <= 501
245lua_Number (*dll_lua_tonumber) (lua_State *L, int idx);
246lua_Integer (*dll_lua_tointeger) (lua_State *L, int idx);
247void (*dll_lua_call) (lua_State *L, int nargs, int nresults);
248int (*dll_lua_pcall) (lua_State *L, int nargs, int nresults, int errfunc);
249#else
250lua_Number (*dll_lua_tonumberx) (lua_State *L, int idx, int *isnum);
251lua_Integer (*dll_lua_tointegerx) (lua_State *L, int idx, int *isnum);
252void (*dll_lua_callk) (lua_State *L, int nargs, int nresults, int ctx,
Bram Moolenaardb913952012-06-29 12:54:53 +0200253 lua_CFunction k);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200254int (*dll_lua_pcallk) (lua_State *L, int nargs, int nresults, int errfunc,
Bram Moolenaardb913952012-06-29 12:54:53 +0200255 int ctx, lua_CFunction k);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200256void (*dll_lua_getglobal) (lua_State *L, const char *var);
257void (*dll_lua_setglobal) (lua_State *L, const char *var);
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200258#endif
259#if LUA_VERSION_NUM <= 502
260void (*dll_lua_replace) (lua_State *L, int idx);
261void (*dll_lua_remove) (lua_State *L, int idx);
262#endif
263#if LUA_VERSION_NUM >= 503
264void (*dll_lua_rotate) (lua_State *L, int idx, int n);
Bram Moolenaar9514b1f2015-06-25 18:27:32 +0200265void (*dll_lua_copy) (lua_State *L, int fromidx, int toidx);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200266#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200267const char *(*dll_lua_typename) (lua_State *L, int tp);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200268void (*dll_lua_close) (lua_State *L);
269int (*dll_lua_gettop) (lua_State *L);
270void (*dll_lua_settop) (lua_State *L, int idx);
271void (*dll_lua_pushvalue) (lua_State *L, int idx);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200272int (*dll_lua_isnumber) (lua_State *L, int idx);
273int (*dll_lua_isstring) (lua_State *L, int idx);
274int (*dll_lua_type) (lua_State *L, int idx);
275int (*dll_lua_rawequal) (lua_State *L, int idx1, int idx2);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200276int (*dll_lua_toboolean) (lua_State *L, int idx);
277const char *(*dll_lua_tolstring) (lua_State *L, int idx, size_t *len);
278void *(*dll_lua_touserdata) (lua_State *L, int idx);
279void (*dll_lua_pushnil) (lua_State *L);
280void (*dll_lua_pushnumber) (lua_State *L, lua_Number n);
281void (*dll_lua_pushinteger) (lua_State *L, lua_Integer n);
282void (*dll_lua_pushlstring) (lua_State *L, const char *s, size_t l);
283void (*dll_lua_pushstring) (lua_State *L, const char *s);
284const char *(*dll_lua_pushfstring) (lua_State *L, const char *fmt, ...);
285void (*dll_lua_pushcclosure) (lua_State *L, lua_CFunction fn, int n);
286void (*dll_lua_pushboolean) (lua_State *L, int b);
287void (*dll_lua_pushlightuserdata) (lua_State *L, void *p);
288void (*dll_lua_getfield) (lua_State *L, int idx, const char *k);
Bram Moolenaar17413672018-07-14 20:49:42 +0200289#if LUA_VERSION_NUM <= 502
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200290void (*dll_lua_rawget) (lua_State *L, int idx);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200291void (*dll_lua_rawgeti) (lua_State *L, int idx, int n);
Bram Moolenaar17413672018-07-14 20:49:42 +0200292#else
293int (*dll_lua_rawget) (lua_State *L, int idx);
294int (*dll_lua_rawgeti) (lua_State *L, int idx, lua_Integer n);
295#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200296void (*dll_lua_createtable) (lua_State *L, int narr, int nrec);
Bram Moolenaar830e3582018-08-21 14:23:35 +0200297#if LUA_VERSION_NUM >= 504
298void *(*dll_lua_newuserdatauv) (lua_State *L, size_t sz, int nuvalue);
299#else
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200300void *(*dll_lua_newuserdata) (lua_State *L, size_t sz);
Bram Moolenaar830e3582018-08-21 14:23:35 +0200301#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200302int (*dll_lua_getmetatable) (lua_State *L, int objindex);
303void (*dll_lua_setfield) (lua_State *L, int idx, const char *k);
304void (*dll_lua_rawset) (lua_State *L, int idx);
Bram Moolenaar17413672018-07-14 20:49:42 +0200305#if LUA_VERSION_NUM <= 502
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200306void (*dll_lua_rawseti) (lua_State *L, int idx, int n);
Bram Moolenaar17413672018-07-14 20:49:42 +0200307#else
308void (*dll_lua_rawseti) (lua_State *L, int idx, lua_Integer n);
309#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200310int (*dll_lua_setmetatable) (lua_State *L, int objindex);
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200311int (*dll_lua_next) (lua_State *L, int idx);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100312// libs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200313int (*dll_luaopen_base) (lua_State *L);
314int (*dll_luaopen_table) (lua_State *L);
315int (*dll_luaopen_string) (lua_State *L);
316int (*dll_luaopen_math) (lua_State *L);
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200317int (*dll_luaopen_io) (lua_State *L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200318int (*dll_luaopen_os) (lua_State *L);
319int (*dll_luaopen_package) (lua_State *L);
320int (*dll_luaopen_debug) (lua_State *L);
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200321void (*dll_luaL_openlibs) (lua_State *L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200322
323typedef void **luaV_function;
324typedef struct {
325 const char *name;
326 luaV_function func;
327} luaV_Reg;
328
329static const luaV_Reg luaV_dll[] = {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100330 // lauxlib
Bram Moolenaar1dced572012-04-05 16:54:08 +0200331#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200332 {"luaL_register", (luaV_function) &dll_luaL_register},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200333 {"luaL_prepbuffer", (luaV_function) &dll_luaL_prepbuffer},
334 {"luaL_openlib", (luaV_function) &dll_luaL_openlib},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200335 {"luaL_typerror", (luaV_function) &dll_luaL_typerror},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200336 {"luaL_loadfile", (luaV_function) &dll_luaL_loadfile},
337 {"luaL_loadbuffer", (luaV_function) &dll_luaL_loadbuffer},
338#else
339 {"luaL_prepbuffsize", (luaV_function) &dll_luaL_prepbuffsize},
340 {"luaL_setfuncs", (luaV_function) &dll_luaL_setfuncs},
341 {"luaL_loadfilex", (luaV_function) &dll_luaL_loadfilex},
342 {"luaL_loadbufferx", (luaV_function) &dll_luaL_loadbufferx},
343 {"luaL_argerror", (luaV_function) &dll_luaL_argerror},
344#endif
Bram Moolenaar5551b132020-07-14 21:54:28 +0200345#if LUA_VERSION_NUM >= 504
346 {"luaL_typeerror", (luaV_function) &dll_luaL_typeerror},
347#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200348 {"luaL_checkany", (luaV_function) &dll_luaL_checkany},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200349 {"luaL_checklstring", (luaV_function) &dll_luaL_checklstring},
350 {"luaL_checkinteger", (luaV_function) &dll_luaL_checkinteger},
351 {"luaL_optinteger", (luaV_function) &dll_luaL_optinteger},
352 {"luaL_checktype", (luaV_function) &dll_luaL_checktype},
353 {"luaL_error", (luaV_function) &dll_luaL_error},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200354 {"luaL_newstate", (luaV_function) &dll_luaL_newstate},
355 {"luaL_buffinit", (luaV_function) &dll_luaL_buffinit},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200356 {"luaL_addlstring", (luaV_function) &dll_luaL_addlstring},
357 {"luaL_pushresult", (luaV_function) &dll_luaL_pushresult},
Bram Moolenaardf1643a2020-05-17 18:53:27 +0200358 {"luaL_loadstring", (luaV_function) &dll_luaL_loadstring},
Bram Moolenaar1e4c7d02020-06-25 20:56:42 +0200359 {"luaL_ref", (luaV_function) &dll_luaL_ref},
360 {"luaL_unref", (luaV_function) &dll_luaL_unref},
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100361 // lua
Bram Moolenaar1dced572012-04-05 16:54:08 +0200362#if LUA_VERSION_NUM <= 501
363 {"lua_tonumber", (luaV_function) &dll_lua_tonumber},
364 {"lua_tointeger", (luaV_function) &dll_lua_tointeger},
365 {"lua_call", (luaV_function) &dll_lua_call},
366 {"lua_pcall", (luaV_function) &dll_lua_pcall},
367#else
368 {"lua_tonumberx", (luaV_function) &dll_lua_tonumberx},
369 {"lua_tointegerx", (luaV_function) &dll_lua_tointegerx},
370 {"lua_callk", (luaV_function) &dll_lua_callk},
371 {"lua_pcallk", (luaV_function) &dll_lua_pcallk},
372 {"lua_getglobal", (luaV_function) &dll_lua_getglobal},
373 {"lua_setglobal", (luaV_function) &dll_lua_setglobal},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200374#endif
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200375#if LUA_VERSION_NUM <= 502
376 {"lua_replace", (luaV_function) &dll_lua_replace},
377 {"lua_remove", (luaV_function) &dll_lua_remove},
378#endif
379#if LUA_VERSION_NUM >= 503
380 {"lua_rotate", (luaV_function) &dll_lua_rotate},
381 {"lua_copy", (luaV_function) &dll_lua_copy},
382#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200383 {"lua_typename", (luaV_function) &dll_lua_typename},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200384 {"lua_close", (luaV_function) &dll_lua_close},
385 {"lua_gettop", (luaV_function) &dll_lua_gettop},
386 {"lua_settop", (luaV_function) &dll_lua_settop},
387 {"lua_pushvalue", (luaV_function) &dll_lua_pushvalue},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200388 {"lua_isnumber", (luaV_function) &dll_lua_isnumber},
389 {"lua_isstring", (luaV_function) &dll_lua_isstring},
390 {"lua_type", (luaV_function) &dll_lua_type},
391 {"lua_rawequal", (luaV_function) &dll_lua_rawequal},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200392 {"lua_toboolean", (luaV_function) &dll_lua_toboolean},
393 {"lua_tolstring", (luaV_function) &dll_lua_tolstring},
394 {"lua_touserdata", (luaV_function) &dll_lua_touserdata},
395 {"lua_pushnil", (luaV_function) &dll_lua_pushnil},
396 {"lua_pushnumber", (luaV_function) &dll_lua_pushnumber},
397 {"lua_pushinteger", (luaV_function) &dll_lua_pushinteger},
398 {"lua_pushlstring", (luaV_function) &dll_lua_pushlstring},
399 {"lua_pushstring", (luaV_function) &dll_lua_pushstring},
400 {"lua_pushfstring", (luaV_function) &dll_lua_pushfstring},
401 {"lua_pushcclosure", (luaV_function) &dll_lua_pushcclosure},
402 {"lua_pushboolean", (luaV_function) &dll_lua_pushboolean},
403 {"lua_pushlightuserdata", (luaV_function) &dll_lua_pushlightuserdata},
404 {"lua_getfield", (luaV_function) &dll_lua_getfield},
405 {"lua_rawget", (luaV_function) &dll_lua_rawget},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200406 {"lua_rawgeti", (luaV_function) &dll_lua_rawgeti},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200407 {"lua_createtable", (luaV_function) &dll_lua_createtable},
Bram Moolenaar830e3582018-08-21 14:23:35 +0200408#if LUA_VERSION_NUM >= 504
409 {"lua_newuserdatauv", (luaV_function) &dll_lua_newuserdatauv},
410#else
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200411 {"lua_newuserdata", (luaV_function) &dll_lua_newuserdata},
Bram Moolenaar830e3582018-08-21 14:23:35 +0200412#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200413 {"lua_getmetatable", (luaV_function) &dll_lua_getmetatable},
414 {"lua_setfield", (luaV_function) &dll_lua_setfield},
415 {"lua_rawset", (luaV_function) &dll_lua_rawset},
416 {"lua_rawseti", (luaV_function) &dll_lua_rawseti},
417 {"lua_setmetatable", (luaV_function) &dll_lua_setmetatable},
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200418 {"lua_next", (luaV_function) &dll_lua_next},
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100419 // libs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200420 {"luaopen_base", (luaV_function) &dll_luaopen_base},
421 {"luaopen_table", (luaV_function) &dll_luaopen_table},
422 {"luaopen_string", (luaV_function) &dll_luaopen_string},
423 {"luaopen_math", (luaV_function) &dll_luaopen_math},
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200424 {"luaopen_io", (luaV_function) &dll_luaopen_io},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200425 {"luaopen_os", (luaV_function) &dll_luaopen_os},
426 {"luaopen_package", (luaV_function) &dll_luaopen_package},
427 {"luaopen_debug", (luaV_function) &dll_luaopen_debug},
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200428 {"luaL_openlibs", (luaV_function) &dll_luaL_openlibs},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200429 {NULL, NULL}
430};
431
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200432static HANDLE hinstLua = NULL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200433
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200434 static int
435lua_link_init(char *libname, int verbose)
436{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200437 const luaV_Reg *reg;
438 if (hinstLua) return OK;
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200439 hinstLua = load_dll(libname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200440 if (!hinstLua)
441 {
442 if (verbose)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100443 semsg(_(e_loadlib), libname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200444 return FAIL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200445 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200446 for (reg = luaV_dll; reg->func; reg++)
447 {
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200448 if ((*reg->func = symbol_from_dll(hinstLua, reg->name)) == NULL)
449 {
450 close_dll(hinstLua);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200451 hinstLua = 0;
452 if (verbose)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100453 semsg(_(e_loadfunc), reg->name);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200454 return FAIL;
455 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200456 }
457 return OK;
458}
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100459#endif // DYNAMIC_LUA
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200460
Bram Moolenaard90b6c02016-08-28 18:10:45 +0200461#if defined(DYNAMIC_LUA) || defined(PROTO)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200462 int
463lua_enabled(int verbose)
464{
Bram Moolenaar25e4fcd2016-01-09 14:57:47 +0100465 return lua_link_init((char *)p_luadll, verbose) == OK;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200466}
Bram Moolenaard90b6c02016-08-28 18:10:45 +0200467#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200468
Bram Moolenaar5551b132020-07-14 21:54:28 +0200469#if LUA_VERSION_NUM > 501 && LUA_VERSION_NUM < 504
Bram Moolenaar1dced572012-04-05 16:54:08 +0200470 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100471luaL_typeerror(lua_State *L, int narg, const char *tname)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200472{
473 const char *msg = lua_pushfstring(L, "%s expected, got %s",
Bram Moolenaardb913952012-06-29 12:54:53 +0200474 tname, luaL_typename(L, narg));
Bram Moolenaar1dced572012-04-05 16:54:08 +0200475 return luaL_argerror(L, narg, msg);
476}
477#endif
478
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200479
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100480// ======= Internal =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200481
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200482 static void
483luaV_newmetatable(lua_State *L, const char *tname)
484{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200485 lua_newtable(L);
486 lua_pushlightuserdata(L, (void *) tname);
487 lua_pushvalue(L, -2);
488 lua_rawset(L, LUA_REGISTRYINDEX);
489}
490
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200491 static void *
492luaV_toudata(lua_State *L, int ud, const char *tname)
493{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200494 void *p = lua_touserdata(L, ud);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200495
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100496 if (p != NULL) // value is userdata?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200497 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100498 if (lua_getmetatable(L, ud)) // does it have a metatable?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200499 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100500 luaV_getfield(L, tname); // get metatable
501 if (lua_rawequal(L, -1, -2)) // MTs match?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200502 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100503 lua_pop(L, 2); // MTs
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200504 return p;
505 }
506 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200507 }
508 return NULL;
509}
510
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200511 static void *
Bram Moolenaar1dced572012-04-05 16:54:08 +0200512luaV_checkcache(lua_State *L, void *p)
513{
514 luaV_getudata(L, p);
515 if (lua_isnil(L, -1)) luaL_error(L, "invalid object");
516 lua_pop(L, 1);
517 return p;
518}
519
520#define luaV_unbox(L,luatyp,ud) (*((luatyp *) lua_touserdata((L),(ud))))
521
522#define luaV_checkvalid(L,luatyp,ud) \
523 luaV_checkcache((L), (void *) luaV_unbox((L),luatyp,(ud)))
524
525 static void *
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200526luaV_checkudata(lua_State *L, int ud, const char *tname)
527{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200528 void *p = luaV_toudata(L, ud, tname);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200529 if (p == NULL) luaL_typeerror(L, ud, tname);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200530 return p;
531}
532
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200533 static void
534luaV_pushtypval(lua_State *L, typval_T *tv)
535{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200536 if (tv == NULL)
537 {
538 lua_pushnil(L);
539 return;
540 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200541 switch (tv->v_type)
542 {
543 case VAR_STRING:
Bram Moolenaard04da7c2012-10-14 03:41:59 +0200544 lua_pushstring(L, tv->vval.v_string == NULL
545 ? "" : (char *)tv->vval.v_string);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200546 break;
547 case VAR_NUMBER:
548 lua_pushinteger(L, (int) tv->vval.v_number);
549 break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200550#ifdef FEAT_FLOAT
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200551 case VAR_FLOAT:
552 lua_pushnumber(L, (lua_Number) tv->vval.v_float);
553 break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200554#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +0200555 case VAR_LIST:
556 luaV_pushlist(L, tv->vval.v_list);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200557 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200558 case VAR_DICT:
559 luaV_pushdict(L, tv->vval.v_dict);
560 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +0100561 case VAR_BOOL:
Bram Moolenaar520e1e42016-01-23 19:46:28 +0100562 case VAR_SPECIAL:
563 if (tv->vval.v_number <= VVAL_TRUE)
564 lua_pushinteger(L, (int) tv->vval.v_number);
565 else
566 lua_pushnil(L);
567 break;
Bram Moolenaarca06da92018-07-01 15:12:05 +0200568 case VAR_FUNC:
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100569 luaV_pushfuncref(L, tv->vval.v_string);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200570 break;
Bram Moolenaarb7828692019-03-23 13:57:02 +0100571 case VAR_BLOB:
572 luaV_pushblob(L, tv->vval.v_blob);
573 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200574 default:
575 lua_pushnil(L);
576 }
577}
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200578
Bram Moolenaarca06da92018-07-01 15:12:05 +0200579/*
580 * Converts lua value at 'pos' to typval 'tv'.
581 * Returns OK or FAIL.
582 */
583 static int
584luaV_totypval(lua_State *L, int pos, typval_T *tv)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200585{
Bram Moolenaarca06da92018-07-01 15:12:05 +0200586 int status = OK;
587
Bram Moolenaara9a8e5f2020-07-02 21:17:57 +0200588 tv->v_lock = 0;
589
Bram Moolenaarca06da92018-07-01 15:12:05 +0200590 switch (lua_type(L, pos))
591 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200592 case LUA_TBOOLEAN:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +0100593 tv->v_type = VAR_BOOL;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200594 tv->vval.v_number = (varnumber_T) lua_toboolean(L, pos);
595 break;
Bram Moolenaar9067cd62019-01-01 00:41:54 +0100596 case LUA_TNIL:
597 tv->v_type = VAR_SPECIAL;
598 tv->vval.v_number = VVAL_NULL;
599 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200600 case LUA_TSTRING:
601 tv->v_type = VAR_STRING;
602 tv->vval.v_string = vim_strsave((char_u *) lua_tostring(L, pos));
603 break;
604 case LUA_TNUMBER:
605#ifdef FEAT_FLOAT
Bram Moolenaareb04f082020-05-17 14:32:35 +0200606 {
607 const lua_Number n = lua_tonumber(L, pos);
608
609 if (n > (lua_Number)INT64_MAX || n < (lua_Number)INT64_MIN
610 || ((lua_Number)((varnumber_T)n)) != n)
611 {
612 tv->v_type = VAR_FLOAT;
613 tv->vval.v_float = (float_T)n;
614 }
615 else
616 {
617 tv->v_type = VAR_NUMBER;
618 tv->vval.v_number = (varnumber_T)n;
619 }
620 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200621#else
622 tv->v_type = VAR_NUMBER;
623 tv->vval.v_number = (varnumber_T) lua_tointeger(L, pos);
624#endif
625 break;
Bram Moolenaar801ab062020-06-25 19:27:56 +0200626 case LUA_TFUNCTION:
627 {
628 char_u *name;
Bram Moolenaar066e7da2020-07-18 12:50:35 +0200629 luaV_CFuncState *state;
630
Bram Moolenaar801ab062020-06-25 19:27:56 +0200631 lua_pushvalue(L, pos);
Bram Moolenaar066e7da2020-07-18 12:50:35 +0200632 state = ALLOC_CLEAR_ONE(luaV_CFuncState);
Bram Moolenaar801ab062020-06-25 19:27:56 +0200633 state->lua_funcref = luaL_ref(L, LUA_REGISTRYINDEX);
634 state->L = L;
635 state->lua_tableref = LUA_NOREF;
636 name = register_cfunc(&luaV_call_lua_func,
637 &luaV_call_lua_func_free, state);
638 tv->v_type = VAR_FUNC;
639 tv->vval.v_string = vim_strsave(name);
640 break;
641 }
642 case LUA_TTABLE:
643 {
Bram Moolenaar066e7da2020-07-18 12:50:35 +0200644 int lua_tableref;
645
Bram Moolenaar801ab062020-06-25 19:27:56 +0200646 lua_pushvalue(L, pos);
Bram Moolenaar066e7da2020-07-18 12:50:35 +0200647 lua_tableref = luaL_ref(L, LUA_REGISTRYINDEX);
Bram Moolenaar801ab062020-06-25 19:27:56 +0200648 if (lua_getmetatable(L, pos)) {
649 lua_getfield(L, -1, LUA___CALL);
650 if (lua_isfunction(L, -1)) {
651 char_u *name;
652 int lua_funcref = luaL_ref(L, LUA_REGISTRYINDEX);
653 luaV_CFuncState *state = ALLOC_CLEAR_ONE(luaV_CFuncState);
Bram Moolenaar066e7da2020-07-18 12:50:35 +0200654
Bram Moolenaar801ab062020-06-25 19:27:56 +0200655 state->lua_funcref = lua_funcref;
656 state->L = L;
657 state->lua_tableref = lua_tableref;
658 name = register_cfunc(&luaV_call_lua_func,
659 &luaV_call_lua_func_free, state);
660 tv->v_type = VAR_FUNC;
661 tv->vval.v_string = vim_strsave(name);
662 break;
663 }
664 }
665 tv->v_type = VAR_NUMBER;
666 tv->vval.v_number = 0;
667 status = FAIL;
668 break;
669 }
Bram Moolenaarca06da92018-07-01 15:12:05 +0200670 case LUA_TUSERDATA:
671 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200672 void *p = lua_touserdata(L, pos);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200673
Bram Moolenaarb7828692019-03-23 13:57:02 +0100674 if (lua_getmetatable(L, pos)) // has metatable?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200675 {
Bram Moolenaarb7828692019-03-23 13:57:02 +0100676 // check list
Bram Moolenaar1dced572012-04-05 16:54:08 +0200677 luaV_getfield(L, LUAVIM_LIST);
678 if (lua_rawequal(L, -1, -2))
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200679 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200680 tv->v_type = VAR_LIST;
681 tv->vval.v_list = *((luaV_List *) p);
682 ++tv->vval.v_list->lv_refcount;
Bram Moolenaarb7828692019-03-23 13:57:02 +0100683 lua_pop(L, 2); // MTs
Bram Moolenaarca06da92018-07-01 15:12:05 +0200684 break;
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200685 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100686 // check dict
Bram Moolenaar1dced572012-04-05 16:54:08 +0200687 luaV_getfield(L, LUAVIM_DICT);
688 if (lua_rawequal(L, -1, -3))
689 {
690 tv->v_type = VAR_DICT;
691 tv->vval.v_dict = *((luaV_Dict *) p);
692 ++tv->vval.v_dict->dv_refcount;
Bram Moolenaarb7828692019-03-23 13:57:02 +0100693 lua_pop(L, 3); // MTs
Bram Moolenaarca06da92018-07-01 15:12:05 +0200694 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200695 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100696 // check blob
697 luaV_getfield(L, LUAVIM_BLOB);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200698 if (lua_rawequal(L, -1, -4))
699 {
Bram Moolenaarb7828692019-03-23 13:57:02 +0100700 tv->v_type = VAR_BLOB;
701 tv->vval.v_blob = *((luaV_Blob *) p);
702 ++tv->vval.v_blob->bv_refcount;
703 lua_pop(L, 4); // MTs
704 break;
705 }
706 // check funcref
707 luaV_getfield(L, LUAVIM_FUNCREF);
708 if (lua_rawequal(L, -1, -5))
709 {
Bram Moolenaarca06da92018-07-01 15:12:05 +0200710 luaV_Funcref *f = (luaV_Funcref *) p;
Bram Moolenaar066e7da2020-07-18 12:50:35 +0200711
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100712 func_ref(f->name);
713 tv->v_type = VAR_FUNC;
714 tv->vval.v_string = vim_strsave(f->name);
Bram Moolenaarb7828692019-03-23 13:57:02 +0100715 lua_pop(L, 5); // MTs
Bram Moolenaarca06da92018-07-01 15:12:05 +0200716 break;
717 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100718 lua_pop(L, 4); // MTs
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200719 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200720 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100721 // FALLTHROUGH
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200722 default:
Bram Moolenaar1dced572012-04-05 16:54:08 +0200723 tv->v_type = VAR_NUMBER;
724 tv->vval.v_number = 0;
Bram Moolenaarca06da92018-07-01 15:12:05 +0200725 status = FAIL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200726 }
Bram Moolenaarca06da92018-07-01 15:12:05 +0200727 return status;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200728}
729
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100730/*
731 * similar to luaL_addlstring, but replaces \0 with \n if toline and
732 * \n with \0 otherwise
733 */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200734 static void
735luaV_addlstring(luaL_Buffer *b, const char *s, size_t l, int toline)
736{
737 while (l--)
738 {
739 if (*s == '\0' && toline)
740 luaL_addchar(b, '\n');
741 else if (*s == '\n' && !toline)
742 luaL_addchar(b, '\0');
743 else
744 luaL_addchar(b, *s);
745 s++;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200746 }
747}
748
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200749 static void
750luaV_pushline(lua_State *L, buf_T *buf, linenr_T n)
751{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200752 const char *s = (const char *) ml_get_buf(buf, n, FALSE);
753 luaL_Buffer b;
754 luaL_buffinit(L, &b);
755 luaV_addlstring(&b, s, strlen(s), 0);
756 luaL_pushresult(&b);
757}
758
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200759 static char_u *
760luaV_toline(lua_State *L, int pos)
761{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200762 size_t l;
763 const char *s = lua_tolstring(L, pos, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200764
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200765 luaL_Buffer b;
766 luaL_buffinit(L, &b);
767 luaV_addlstring(&b, s, l, 1);
768 luaL_pushresult(&b);
769 return (char_u *) lua_tostring(L, -1);
770}
771
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100772/*
773 * pops a string s from the top of the stack and calls mf(t) for pieces t of
774 * s separated by newlines
775 */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200776 static void
777luaV_msgfunc(lua_State *L, msgfunc_T mf)
778{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200779 luaL_Buffer b;
780 size_t l;
781 const char *p, *s = lua_tolstring(L, -1, &l);
782 luaL_buffinit(L, &b);
783 luaV_addlstring(&b, s, l, 0);
784 luaL_pushresult(&b);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100785 // break string
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200786 p = s = lua_tolstring(L, -1, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200787 while (l--)
788 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100789 if (*p++ == '\0') // break?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200790 {
791 mf((char_u *) s);
792 s = p;
793 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200794 }
795 mf((char_u *) s);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100796 lua_pop(L, 2); // original and modified strings
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200797}
798
Bram Moolenaar1dced572012-04-05 16:54:08 +0200799#define luaV_newtype(typ,tname,luatyp,luatname) \
800 static luatyp * \
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100801 luaV_new##tname(lua_State *L, typ *obj) \
Bram Moolenaar1dced572012-04-05 16:54:08 +0200802 { \
803 luatyp *o = (luatyp *) lua_newuserdata(L, sizeof(luatyp)); \
804 *o = obj; \
805 luaV_setudata(L, obj); /* cache[obj] = udata */ \
806 luaV_getfield(L, luatname); \
807 lua_setmetatable(L, -2); \
808 return o; \
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200809 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200810
811#define luaV_pushtype(typ,tname,luatyp) \
812 static luatyp * \
Bram Moolenaarca06da92018-07-01 15:12:05 +0200813 luaV_push##tname(lua_State *L, typ *obj) \
Bram Moolenaar1dced572012-04-05 16:54:08 +0200814 { \
815 luatyp *o = NULL; \
816 if (obj == NULL) \
817 lua_pushnil(L); \
818 else { \
819 luaV_getudata(L, obj); \
820 if (lua_isnil(L, -1)) /* not interned? */ \
821 { \
822 lua_pop(L, 1); \
823 o = luaV_new##tname(L, obj); \
824 } \
825 else \
826 o = (luatyp *) lua_touserdata(L, -1); \
827 } \
828 return o; \
829 }
830
831#define luaV_type_tostring(tname,luatname) \
832 static int \
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100833 luaV_##tname##_tostring(lua_State *L) \
Bram Moolenaar1dced572012-04-05 16:54:08 +0200834 { \
835 lua_pushfstring(L, "%s: %p", luatname, lua_touserdata(L, 1)); \
836 return 1; \
837 }
838
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100839// ======= List type =======
Bram Moolenaar1dced572012-04-05 16:54:08 +0200840
841 static luaV_List *
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100842luaV_newlist(lua_State *L, list_T *lis)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200843{
844 luaV_List *l = (luaV_List *) lua_newuserdata(L, sizeof(luaV_List));
845 *l = lis;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100846 lis->lv_refcount++; // reference in Lua
847 luaV_setudata(L, lis); // cache[lis] = udata
Bram Moolenaar1dced572012-04-05 16:54:08 +0200848 luaV_getfield(L, LUAVIM_LIST);
849 lua_setmetatable(L, -2);
850 return l;
851}
852
853luaV_pushtype(list_T, list, luaV_List)
854luaV_type_tostring(list, LUAVIM_LIST)
855
856 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100857luaV_list_len(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200858{
859 list_T *l = luaV_unbox(L, luaV_List, 1);
Bram Moolenaarb7828692019-03-23 13:57:02 +0100860 lua_pushinteger(L, (int) list_len(l));
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200861 return 1;
862}
863
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200864 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100865luaV_list_iter(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200866{
867 listitem_T *li = (listitem_T *) lua_touserdata(L, lua_upvalueindex(2));
868 if (li == NULL) return 0;
869 luaV_pushtypval(L, &li->li_tv);
870 lua_pushlightuserdata(L, (void *) li->li_next);
871 lua_replace(L, lua_upvalueindex(2));
872 return 1;
873}
874
875 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100876luaV_list_call(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200877{
878 list_T *l = luaV_unbox(L, luaV_List, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100879 lua_pushvalue(L, lua_upvalueindex(1)); // pass cache table along
Bram Moolenaar1dced572012-04-05 16:54:08 +0200880 lua_pushlightuserdata(L, (void *) l->lv_first);
881 lua_pushcclosure(L, luaV_list_iter, 2);
882 return 1;
883}
884
885 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100886luaV_list_index(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200887{
888 list_T *l = luaV_unbox(L, luaV_List, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100889 if (lua_isnumber(L, 2)) // list item?
Bram Moolenaar1dced572012-04-05 16:54:08 +0200890 {
Bram Moolenaarbd846172020-06-27 12:32:57 +0200891 long n = (long) luaL_checkinteger(L, 2);
892 listitem_T *li;
893
894 // Lua array index starts with 1 while Vim uses 0, subtract 1 to
895 // normalize.
896 n -= 1;
897 li = list_find(l, n);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200898 if (li == NULL)
899 lua_pushnil(L);
900 else
901 luaV_pushtypval(L, &li->li_tv);
902 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100903 else if (lua_isstring(L, 2)) // method?
Bram Moolenaar1dced572012-04-05 16:54:08 +0200904 {
905 const char *s = lua_tostring(L, 2);
906 if (strncmp(s, "add", 3) == 0
Bram Moolenaarb3766472013-04-15 13:49:21 +0200907 || strncmp(s, "insert", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200908 {
909 lua_getmetatable(L, 1);
910 lua_getfield(L, -1, s);
911 }
912 else
913 lua_pushnil(L);
914 }
915 else
916 lua_pushnil(L);
917 return 1;
918}
919
920 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100921luaV_list_newindex(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200922{
923 list_T *l = luaV_unbox(L, luaV_List, 1);
924 long n = (long) luaL_checkinteger(L, 2);
925 listitem_T *li;
Bram Moolenaarbd846172020-06-27 12:32:57 +0200926
927 // Lua array index starts with 1 while Vim uses 0, subtract 1 to normalize.
928 n -= 1;
929
Bram Moolenaar1dced572012-04-05 16:54:08 +0200930 if (l->lv_lock)
931 luaL_error(L, "list is locked");
932 li = list_find(l, n);
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200933 if (li == NULL)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200934 {
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200935 if (!lua_isnil(L, 3))
936 {
937 typval_T v;
938 luaV_checktypval(L, 3, &v, "inserting list item");
939 if (list_insert_tv(l, &v, li) == FAIL)
940 luaL_error(L, "failed to add item to list");
941 clear_tv(&v);
942 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200943 }
944 else
945 {
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200946 if (lua_isnil(L, 3)) // remove?
947 {
948 vimlist_remove(l, li, li);
949 listitem_free(l, li);
950 }
951 else
952 {
953 typval_T v;
954 luaV_checktypval(L, 3, &v, "setting list item");
955 clear_tv(&li->li_tv);
Bram Moolenaare49b8e82020-07-01 13:52:55 +0200956 li->li_tv = v;
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200957 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200958 }
959 return 0;
960}
961
962 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100963luaV_list_add(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200964{
965 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
966 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200967 typval_T v;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200968 if (l->lv_lock)
969 luaL_error(L, "list is locked");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200970 lua_settop(L, 2);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200971 luaV_checktypval(L, 2, &v, "adding list item");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200972 if (list_append_tv(l, &v) == FAIL)
Bram Moolenaarca06da92018-07-01 15:12:05 +0200973 luaL_error(L, "failed to add item to list");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200974 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200975 lua_settop(L, 1);
976 return 1;
977}
978
979 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100980luaV_list_insert(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200981{
982 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
983 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaar46538ee2015-02-17 16:28:55 +0100984 long pos = (long) luaL_optinteger(L, 3, 0);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200985 listitem_T *li = NULL;
986 typval_T v;
987 if (l->lv_lock)
988 luaL_error(L, "list is locked");
989 if (pos < l->lv_len)
990 {
991 li = list_find(l, pos);
992 if (li == NULL)
993 luaL_error(L, "invalid position");
994 }
995 lua_settop(L, 2);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200996 luaV_checktypval(L, 2, &v, "inserting list item");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200997 if (list_insert_tv(l, &v, li) == FAIL)
Bram Moolenaarca06da92018-07-01 15:12:05 +0200998 luaL_error(L, "failed to add item to list");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200999 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001000 lua_settop(L, 1);
1001 return 1;
1002}
1003
1004static const luaL_Reg luaV_List_mt[] = {
1005 {"__tostring", luaV_list_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +02001006 {"__len", luaV_list_len},
1007 {"__call", luaV_list_call},
1008 {"__index", luaV_list_index},
1009 {"__newindex", luaV_list_newindex},
1010 {"add", luaV_list_add},
1011 {"insert", luaV_list_insert},
1012 {NULL, NULL}
1013};
1014
1015
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001016// ======= Dict type =======
Bram Moolenaar1dced572012-04-05 16:54:08 +02001017
1018 static luaV_Dict *
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001019luaV_newdict(lua_State *L, dict_T *dic)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001020{
1021 luaV_Dict *d = (luaV_Dict *) lua_newuserdata(L, sizeof(luaV_Dict));
1022 *d = dic;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001023 dic->dv_refcount++; // reference in Lua
1024 luaV_setudata(L, dic); // cache[dic] = udata
Bram Moolenaar1dced572012-04-05 16:54:08 +02001025 luaV_getfield(L, LUAVIM_DICT);
1026 lua_setmetatable(L, -2);
1027 return d;
1028}
1029
1030luaV_pushtype(dict_T, dict, luaV_Dict)
1031luaV_type_tostring(dict, LUAVIM_DICT)
1032
1033 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001034luaV_dict_len(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001035{
1036 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
Bram Moolenaarb7828692019-03-23 13:57:02 +01001037 lua_pushinteger(L, (int) dict_len(d));
Bram Moolenaar1dced572012-04-05 16:54:08 +02001038 return 1;
1039}
1040
1041 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001042luaV_dict_iter(lua_State *L UNUSED)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001043{
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001044#ifdef FEAT_EVAL
Bram Moolenaar1dced572012-04-05 16:54:08 +02001045 hashitem_T *hi = (hashitem_T *) lua_touserdata(L, lua_upvalueindex(2));
1046 int n = lua_tointeger(L, lua_upvalueindex(3));
1047 dictitem_T *di;
1048 if (n <= 0) return 0;
1049 while (HASHITEM_EMPTY(hi)) hi++;
1050 di = dict_lookup(hi);
1051 lua_pushstring(L, (char *) hi->hi_key);
1052 luaV_pushtypval(L, &di->di_tv);
1053 lua_pushlightuserdata(L, (void *) (hi + 1));
1054 lua_replace(L, lua_upvalueindex(2));
1055 lua_pushinteger(L, n - 1);
1056 lua_replace(L, lua_upvalueindex(3));
1057 return 2;
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001058#else
1059 return 0;
1060#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001061}
1062
1063 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001064luaV_dict_call(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001065{
1066 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
1067 hashtab_T *ht = &d->dv_hashtab;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001068 lua_pushvalue(L, lua_upvalueindex(1)); // pass cache table along
Bram Moolenaar1dced572012-04-05 16:54:08 +02001069 lua_pushlightuserdata(L, (void *) ht->ht_array);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001070 lua_pushinteger(L, ht->ht_used); // # remaining items
Bram Moolenaar1dced572012-04-05 16:54:08 +02001071 lua_pushcclosure(L, luaV_dict_iter, 3);
1072 return 1;
1073}
1074
1075 static int
Bram Moolenaarca06da92018-07-01 15:12:05 +02001076luaV_dict_index(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001077{
1078 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
1079 char_u *key = (char_u *) luaL_checkstring(L, 2);
1080 dictitem_T *di = dict_find(d, key, -1);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001081
Bram Moolenaar1dced572012-04-05 16:54:08 +02001082 if (di == NULL)
1083 lua_pushnil(L);
1084 else
Bram Moolenaarca06da92018-07-01 15:12:05 +02001085 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001086 luaV_pushtypval(L, &di->di_tv);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001087 if (di->di_tv.v_type == VAR_FUNC) // funcref?
Bram Moolenaarca06da92018-07-01 15:12:05 +02001088 {
1089 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, -1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001090 f->self = d; // keep "self" reference
Bram Moolenaarca06da92018-07-01 15:12:05 +02001091 d->dv_refcount++;
1092 }
1093 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001094 return 1;
1095}
1096
1097 static int
Bram Moolenaarca06da92018-07-01 15:12:05 +02001098luaV_dict_newindex(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001099{
1100 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
1101 char_u *key = (char_u *) luaL_checkstring(L, 2);
1102 dictitem_T *di;
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001103 typval_T tv;
Bram Moolenaar713bf9e92019-03-16 16:38:41 +01001104
Bram Moolenaar1dced572012-04-05 16:54:08 +02001105 if (d->dv_lock)
1106 luaL_error(L, "dict is locked");
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001107 if (key == NULL)
1108 return 0;
1109 if (*key == NUL)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001110 luaL_error(L, "empty key");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001111 if (!lua_isnil(L, 3)) // read value?
Bram Moolenaar17413672018-07-14 20:49:42 +02001112 {
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001113 luaV_checktypval(L, 3, &tv, "setting dict item");
1114 if (d->dv_scope == VAR_DEF_SCOPE && tv.v_type == VAR_FUNC)
1115 {
1116 clear_tv(&tv);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001117 luaL_error(L, "cannot assign funcref to builtin scope");
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001118 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001119 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001120 di = dict_find(d, key, -1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001121 if (di == NULL) // non-existing key?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001122 {
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001123 if (lua_isnil(L, 3))
1124 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001125 di = dictitem_alloc(key);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001126 if (di == NULL)
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001127 {
1128 clear_tv(&tv);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001129 return 0;
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001130 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001131 if (dict_add(d, di) == FAIL)
1132 {
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001133 vim_free(di);
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001134 clear_tv(&tv);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001135 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001136 }
1137 }
1138 else
1139 clear_tv(&di->di_tv);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001140 if (lua_isnil(L, 3)) // remove?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001141 {
1142 hashitem_T *hi = hash_find(&d->dv_hashtab, di->di_key);
1143 hash_remove(&d->dv_hashtab, hi);
1144 dictitem_free(di);
1145 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001146 else
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001147 di->di_tv = tv;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001148 return 0;
1149}
1150
1151static const luaL_Reg luaV_Dict_mt[] = {
1152 {"__tostring", luaV_dict_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +02001153 {"__len", luaV_dict_len},
1154 {"__call", luaV_dict_call},
1155 {"__index", luaV_dict_index},
1156 {"__newindex", luaV_dict_newindex},
1157 {NULL, NULL}
1158};
1159
1160
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001161// ======= Blob type =======
Bram Moolenaarb7828692019-03-23 13:57:02 +01001162
1163 static luaV_Blob *
1164luaV_newblob(lua_State *L, blob_T *blo)
1165{
1166 luaV_Blob *b = (luaV_Blob *) lua_newuserdata(L, sizeof(luaV_Blob));
1167 *b = blo;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001168 blo->bv_refcount++; // reference in Lua
1169 luaV_setudata(L, blo); // cache[blo] = udata
Bram Moolenaarb7828692019-03-23 13:57:02 +01001170 luaV_getfield(L, LUAVIM_BLOB);
1171 lua_setmetatable(L, -2);
1172 return b;
1173}
1174
1175luaV_pushtype(blob_T, blob, luaV_Blob)
1176luaV_type_tostring(blob, LUAVIM_BLOB)
1177
1178 static int
1179luaV_blob_gc(lua_State *L)
1180{
1181 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1182 blob_unref(b);
1183 return 0;
1184}
1185
1186 static int
1187luaV_blob_len(lua_State *L)
1188{
1189 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1190 lua_pushinteger(L, (int) blob_len(b));
1191 return 1;
1192}
1193
1194 static int
1195luaV_blob_index(lua_State *L)
1196{
1197 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1198 if (lua_isnumber(L, 2))
1199 {
1200 int idx = luaL_checkinteger(L, 2);
1201 if (idx < blob_len(b))
1202 lua_pushnumber(L, (lua_Number) blob_get(b, idx));
1203 else
1204 lua_pushnil(L);
1205 }
1206 else if (lua_isstring(L, 2))
1207 {
1208 const char *s = lua_tostring(L, 2);
1209 if (strncmp(s, "add", 3) == 0)
1210 {
1211 lua_getmetatable(L, 1);
1212 lua_getfield(L, -1, s);
1213 }
1214 else
1215 lua_pushnil(L);
1216 }
1217 else
1218 lua_pushnil(L);
1219 return 1;
1220}
1221
1222 static int
1223luaV_blob_newindex(lua_State *L)
1224{
1225 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1226 if (b->bv_lock)
1227 luaL_error(L, "blob is locked");
1228 if (lua_isnumber(L, 2))
1229 {
1230 long len = blob_len(b);
1231 int idx = luaL_checkinteger(L, 2);
1232 int val = luaL_checkinteger(L, 3);
1233 if (idx < len || (idx == len && ga_grow(&b->bv_ga, 1) == OK))
1234 {
1235 blob_set(b, idx, (char_u) val);
1236 if (idx == len)
1237 ++b->bv_ga.ga_len;
1238 }
1239 else
1240 luaL_error(L, "index out of range");
1241 }
1242 return 0;
1243}
1244
1245 static int
1246luaV_blob_add(lua_State *L)
1247{
1248 luaV_Blob *blo = luaV_checkudata(L, 1, LUAVIM_BLOB);
1249 blob_T *b = (blob_T *) luaV_checkcache(L, (void *) *blo);
1250 if (b->bv_lock)
1251 luaL_error(L, "blob is locked");
1252 lua_settop(L, 2);
1253 if (!lua_isstring(L, 2))
1254 luaL_error(L, "string expected, got %s", luaL_typename(L, 2));
1255 else
1256 {
1257 size_t i, l = 0;
1258 const char *s = lua_tolstring(L, 2, &l);
1259
Bram Moolenaar5f1d3ae2020-02-11 22:37:35 +01001260 if (ga_grow(&b->bv_ga, (int)l) == OK)
Bram Moolenaar6fb5c972019-03-26 21:44:20 +01001261 for (i = 0; i < l; ++i)
1262 ga_append(&b->bv_ga, s[i]);
Bram Moolenaarb7828692019-03-23 13:57:02 +01001263 }
1264 lua_settop(L, 1);
1265 return 1;
1266}
1267
1268static const luaL_Reg luaV_Blob_mt[] = {
1269 {"__tostring", luaV_blob_tostring},
1270 {"__gc", luaV_blob_gc},
1271 {"__len", luaV_blob_len},
1272 {"__index", luaV_blob_index},
1273 {"__newindex", luaV_blob_newindex},
1274 {"add", luaV_blob_add},
1275 {NULL, NULL}
1276};
1277
1278
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001279// ======= Funcref type =======
Bram Moolenaarca06da92018-07-01 15:12:05 +02001280
1281 static luaV_Funcref *
1282luaV_newfuncref(lua_State *L, char_u *name)
1283{
1284 luaV_Funcref *f = (luaV_Funcref *)lua_newuserdata(L, sizeof(luaV_Funcref));
1285
1286 if (name != NULL)
1287 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001288 func_ref(name);
1289 f->name = vim_strsave(name);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001290 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001291 f->self = NULL;
1292 luaV_getfield(L, LUAVIM_FUNCREF);
1293 lua_setmetatable(L, -2);
1294 return f;
1295}
1296
1297 static luaV_Funcref *
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001298luaV_pushfuncref(lua_State *L, char_u *name)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001299{
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001300 return luaV_newfuncref(L, name);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001301}
1302
1303
1304luaV_type_tostring(funcref, LUAVIM_FUNCREF)
1305
1306 static int
1307luaV_funcref_gc(lua_State *L)
1308{
1309 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, 1);
1310
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001311 func_unref(f->name);
1312 vim_free(f->name);
1313 // NOTE: Don't call "dict_unref(f->self)", because the dict of "f->self"
1314 // will be (or has been already) freed by Vim's garbage collection.
Bram Moolenaarca06da92018-07-01 15:12:05 +02001315 return 0;
1316}
1317
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001318// equivalent to string(funcref)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001319 static int
1320luaV_funcref_len(lua_State *L)
1321{
1322 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, 1);
1323
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001324 lua_pushstring(L, (const char *) f->name);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001325 return 1;
1326}
1327
1328 static int
1329luaV_funcref_call(lua_State *L)
1330{
1331 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, 1);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001332 int i, n = lua_gettop(L) - 1; // #args
1333 int status = FAIL;
1334 typval_T args;
1335 typval_T rettv;
Bram Moolenaarca06da92018-07-01 15:12:05 +02001336
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001337 args.v_type = VAR_LIST;
1338 args.vval.v_list = list_alloc();
1339 rettv.v_type = VAR_UNKNOWN; // as in clear_tv
1340 if (args.vval.v_list != NULL)
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001341 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001342 typval_T v;
1343
Bram Moolenaar17413672018-07-14 20:49:42 +02001344 for (i = 0; i < n; i++)
1345 {
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001346 luaV_checktypval(L, i + 2, &v, "calling funcref");
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001347 list_append_tv(args.vval.v_list, &v);
Bram Moolenaar713bf9e92019-03-16 16:38:41 +01001348 clear_tv(&v);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001349 }
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001350 status = func_call(f->name, &args, NULL, f->self, &rettv);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001351 if (status == OK)
1352 luaV_pushtypval(L, &rettv);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001353 clear_tv(&args);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001354 clear_tv(&rettv);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001355 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001356 if (status != OK)
1357 luaL_error(L, "cannot call funcref");
1358 return 1;
1359}
1360
1361static const luaL_Reg luaV_Funcref_mt[] = {
1362 {"__tostring", luaV_funcref_tostring},
1363 {"__gc", luaV_funcref_gc},
1364 {"__len", luaV_funcref_len},
1365 {"__call", luaV_funcref_call},
1366 {NULL, NULL}
1367};
1368
1369
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001370// ======= Buffer type =======
Bram Moolenaar1dced572012-04-05 16:54:08 +02001371
1372luaV_newtype(buf_T, buffer, luaV_Buffer, LUAVIM_BUFFER)
1373luaV_pushtype(buf_T, buffer, luaV_Buffer)
1374luaV_type_tostring(buffer, LUAVIM_BUFFER)
1375
1376 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001377luaV_buffer_len(lua_State *L)
1378{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001379 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
1380 lua_pushinteger(L, b->b_ml.ml_line_count);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001381 return 1;
1382}
1383
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001384 static int
1385luaV_buffer_call(lua_State *L)
1386{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001387 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001388 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001389 set_curbuf(b, DOBUF_SPLIT);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001390 return 1;
1391}
1392
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001393 static int
1394luaV_buffer_index(lua_State *L)
1395{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001396 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001397 linenr_T n = (linenr_T) lua_tointeger(L, 2);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001398 if (n > 0 && n <= b->b_ml.ml_line_count)
1399 luaV_pushline(L, b, n);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001400 else if (lua_isstring(L, 2))
1401 {
1402 const char *s = lua_tostring(L, 2);
1403 if (strncmp(s, "name", 4) == 0)
Bram Moolenaarfe08df42018-07-07 23:07:41 +02001404 lua_pushstring(L, (b->b_sfname == NULL)
1405 ? "" : (char *) b->b_sfname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001406 else if (strncmp(s, "fname", 5) == 0)
Bram Moolenaarfe08df42018-07-07 23:07:41 +02001407 lua_pushstring(L, (b->b_ffname == NULL)
1408 ? "" : (char *) b->b_ffname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001409 else if (strncmp(s, "number", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001410 lua_pushinteger(L, b->b_fnum);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001411 // methods
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001412 else if (strncmp(s, "insert", 6) == 0
1413 || strncmp(s, "next", 4) == 0
1414 || strncmp(s, "previous", 8) == 0
1415 || strncmp(s, "isvalid", 7) == 0)
1416 {
1417 lua_getmetatable(L, 1);
1418 lua_getfield(L, -1, s);
1419 }
1420 else
1421 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001422 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001423 else
1424 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001425 return 1;
1426}
1427
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001428 static int
1429luaV_buffer_newindex(lua_State *L)
1430{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001431 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001432 linenr_T n = (linenr_T) luaL_checkinteger(L, 2);
1433#ifdef HAVE_SANDBOX
1434 luaV_checksandbox(L);
1435#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001436 if (n < 1 || n > b->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001437 luaL_error(L, "invalid line number");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001438 if (lua_isnil(L, 3)) // delete line
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001439 {
1440 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001441 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001442 if (u_savedel(n, 1L) == FAIL)
1443 {
1444 curbuf = buf;
1445 luaL_error(L, "cannot save undo information");
1446 }
Bram Moolenaarca70c072020-05-30 20:30:46 +02001447 else if (ml_delete(n) == FAIL)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001448 {
1449 curbuf = buf;
1450 luaL_error(L, "cannot delete line");
1451 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001452 else
1453 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001454 deleted_lines_mark(n, 1L);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001455 if (b == curwin->w_buffer) // fix cursor in current window?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001456 {
1457 if (curwin->w_cursor.lnum >= n)
1458 {
1459 if (curwin->w_cursor.lnum > n)
1460 {
1461 curwin->w_cursor.lnum -= 1;
1462 check_cursor_col();
1463 }
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001464 else
1465 check_cursor();
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001466 changed_cline_bef_curs();
1467 }
1468 invalidate_botline();
1469 }
1470 }
1471 curbuf = buf;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001472 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001473 else if (lua_isstring(L, 3)) // update line
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001474 {
1475 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001476 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001477 if (u_savesub(n) == FAIL)
1478 {
1479 curbuf = buf;
1480 luaL_error(L, "cannot save undo information");
1481 }
1482 else if (ml_replace(n, luaV_toline(L, 3), TRUE) == FAIL)
1483 {
1484 curbuf = buf;
1485 luaL_error(L, "cannot replace line");
1486 }
1487 else changed_bytes(n, 0);
1488 curbuf = buf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001489 if (b == curwin->w_buffer)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001490 check_cursor_col();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001491 }
1492 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001493 luaL_error(L, "wrong argument to change line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001494 return 0;
1495}
1496
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001497 static int
1498luaV_buffer_insert(lua_State *L)
1499{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001500 luaV_Buffer *lb = luaV_checkudata(L, 1, LUAVIM_BUFFER);
1501 buf_T *b = (buf_T *) luaV_checkcache(L, (void *) *lb);
1502 linenr_T last = b->b_ml.ml_line_count;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001503 linenr_T n = (linenr_T) luaL_optinteger(L, 3, last);
1504 buf_T *buf;
1505 luaL_checktype(L, 2, LUA_TSTRING);
1506#ifdef HAVE_SANDBOX
1507 luaV_checksandbox(L);
1508#endif
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001509 // fix insertion line
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001510 if (n < 0) n = 0;
1511 if (n > last) n = last;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001512 // insert
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001513 buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001514 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001515 if (u_save(n, n + 1) == FAIL)
1516 {
1517 curbuf = buf;
1518 luaL_error(L, "cannot save undo information");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001519 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001520 else if (ml_append(n, luaV_toline(L, 2), 0, FALSE) == FAIL)
1521 {
1522 curbuf = buf;
1523 luaL_error(L, "cannot insert line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001524 }
1525 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001526 appended_lines_mark(n, 1L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001527 curbuf = buf;
1528 update_screen(VALID);
1529 return 0;
1530}
1531
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001532 static int
1533luaV_buffer_next(lua_State *L)
1534{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001535 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001536 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1537 luaV_pushbuffer(L, buf->b_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001538 return 1;
1539}
1540
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001541 static int
1542luaV_buffer_previous(lua_State *L)
1543{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001544 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001545 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1546 luaV_pushbuffer(L, buf->b_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001547 return 1;
1548}
1549
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001550 static int
1551luaV_buffer_isvalid(lua_State *L)
1552{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001553 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001554 luaV_getudata(L, *b);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001555 lua_pushboolean(L, !lua_isnil(L, -1));
1556 return 1;
1557}
1558
1559static const luaL_Reg luaV_Buffer_mt[] = {
1560 {"__tostring", luaV_buffer_tostring},
1561 {"__len", luaV_buffer_len},
1562 {"__call", luaV_buffer_call},
1563 {"__index", luaV_buffer_index},
1564 {"__newindex", luaV_buffer_newindex},
1565 {"insert", luaV_buffer_insert},
1566 {"next", luaV_buffer_next},
1567 {"previous", luaV_buffer_previous},
1568 {"isvalid", luaV_buffer_isvalid},
1569 {NULL, NULL}
1570};
1571
1572
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001573// ======= Window type =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001574
Bram Moolenaar1dced572012-04-05 16:54:08 +02001575luaV_newtype(win_T, window, luaV_Window, LUAVIM_WINDOW)
1576luaV_pushtype(win_T, window, luaV_Window)
1577luaV_type_tostring(window, LUAVIM_WINDOW)
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001578
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001579 static int
1580luaV_window_call(lua_State *L)
1581{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001582 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001583 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001584 win_goto(w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001585 return 1;
1586}
1587
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001588 static int
1589luaV_window_index(lua_State *L)
1590{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001591 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001592 const char *s = luaL_checkstring(L, 2);
1593 if (strncmp(s, "buffer", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001594 luaV_pushbuffer(L, w->w_buffer);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001595 else if (strncmp(s, "line", 4) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001596 lua_pushinteger(L, w->w_cursor.lnum);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001597 else if (strncmp(s, "col", 3) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001598 lua_pushinteger(L, w->w_cursor.col + 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001599 else if (strncmp(s, "width", 5) == 0)
Bram Moolenaar02631462017-09-22 15:20:32 +02001600 lua_pushinteger(L, w->w_width);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001601 else if (strncmp(s, "height", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001602 lua_pushinteger(L, w->w_height);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001603 // methods
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001604 else if (strncmp(s, "next", 4) == 0
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001605 || strncmp(s, "previous", 8) == 0
1606 || strncmp(s, "isvalid", 7) == 0)
1607 {
1608 lua_getmetatable(L, 1);
1609 lua_getfield(L, -1, s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001610 }
1611 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001612 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001613 return 1;
1614}
1615
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001616 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001617luaV_window_newindex(lua_State *L)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001618{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001619 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001620 const char *s = luaL_checkstring(L, 2);
1621 int v = luaL_checkinteger(L, 3);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001622 if (strncmp(s, "line", 4) == 0)
1623 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001624#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001625 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001626#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001627 if (v < 1 || v > w->w_buffer->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001628 luaL_error(L, "line out of range");
Bram Moolenaar1dced572012-04-05 16:54:08 +02001629 w->w_cursor.lnum = v;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001630 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001631 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001632 else if (strncmp(s, "col", 3) == 0)
1633 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001634#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001635 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001636#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001637 w->w_cursor.col = v - 1;
Bram Moolenaar53901442018-07-25 22:02:36 +02001638 w->w_set_curswant = TRUE;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001639 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001640 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001641 else if (strncmp(s, "width", 5) == 0)
1642 {
1643 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001644#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001645 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001646#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001647 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001648 win_setwidth(v);
1649 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001650 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001651 else if (strncmp(s, "height", 6) == 0)
1652 {
1653 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001654#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001655 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001656#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001657 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001658 win_setheight(v);
1659 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001660 }
1661 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001662 luaL_error(L, "invalid window property: `%s'", s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001663 return 0;
1664}
1665
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001666 static int
1667luaV_window_next(lua_State *L)
1668{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001669 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001670 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1671 luaV_pushwindow(L, win->w_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001672 return 1;
1673}
1674
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001675 static int
1676luaV_window_previous(lua_State *L)
1677{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001678 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001679 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1680 luaV_pushwindow(L, win->w_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001681 return 1;
1682}
1683
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001684 static int
1685luaV_window_isvalid(lua_State *L)
1686{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001687 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001688 luaV_getudata(L, *w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001689 lua_pushboolean(L, !lua_isnil(L, -1));
1690 return 1;
1691}
1692
1693static const luaL_Reg luaV_Window_mt[] = {
1694 {"__tostring", luaV_window_tostring},
1695 {"__call", luaV_window_call},
1696 {"__index", luaV_window_index},
1697 {"__newindex", luaV_window_newindex},
1698 {"next", luaV_window_next},
1699 {"previous", luaV_window_previous},
1700 {"isvalid", luaV_window_isvalid},
1701 {NULL, NULL}
1702};
1703
1704
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001705// ======= Vim module =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001706
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001707 static int
1708luaV_print(lua_State *L)
1709{
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001710 int i, n = lua_gettop(L); // nargs
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001711 const char *s;
1712 size_t l;
1713 luaL_Buffer b;
1714 luaL_buffinit(L, &b);
1715 lua_getglobal(L, "tostring");
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001716 for (i = 1; i <= n; i++)
1717 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001718 lua_pushvalue(L, -1); // tostring
1719 lua_pushvalue(L, i); // arg
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001720 lua_call(L, 1, 1);
1721 s = lua_tolstring(L, -1, &l);
1722 if (s == NULL)
1723 return luaL_error(L, "cannot convert to string");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001724 if (i > 1) luaL_addchar(&b, ' '); // use space instead of tab
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001725 luaV_addlstring(&b, s, l, 0);
1726 lua_pop(L, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001727 }
1728 luaL_pushresult(&b);
Bram Moolenaarb98678a2019-10-19 15:18:44 +02001729 if (!got_int)
1730 luaV_msg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001731 return 0;
1732}
1733
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001734 static int
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001735luaV_debug(lua_State *L)
1736{
1737 lua_settop(L, 0);
1738 lua_getglobal(L, "vim");
1739 lua_getfield(L, -1, "eval");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001740 lua_remove(L, -2); // vim.eval at position 1
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001741 for (;;)
1742 {
1743 const char *input;
1744 size_t l;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001745 lua_pushvalue(L, 1); // vim.eval
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001746 lua_pushliteral(L, "input('lua_debug> ')");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001747 lua_call(L, 1, 1); // return string
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001748 input = lua_tolstring(L, -1, &l);
1749 if (l == 0 || strcmp(input, "cont") == 0)
1750 return 0;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001751 msg_putchar('\n'); // avoid outputting on input line
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001752 if (luaL_loadbuffer(L, input, l, "=(debug command)")
1753 || lua_pcall(L, 0, 0, 0))
1754 luaV_emsg(L);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001755 lua_settop(L, 1); // remove eventual returns, but keep vim.eval
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001756 }
1757}
1758
1759 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001760luaV_command(lua_State *L)
1761{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001762 do_cmdline_cmd((char_u *) luaL_checkstring(L, 1));
1763 update_screen(VALID);
1764 return 0;
1765}
1766
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001767 static int
1768luaV_eval(lua_State *L)
1769{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001770 typval_T *tv = eval_expr((char_u *) luaL_checkstring(L, 1), NULL);
1771 if (tv == NULL) luaL_error(L, "invalid expression");
1772 luaV_pushtypval(L, tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +02001773 free_tv(tv);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001774 return 1;
1775}
1776
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001777 static int
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001778luaV_beep(lua_State *L UNUSED)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001779{
Bram Moolenaar165bc692015-07-21 17:53:25 +02001780 vim_beep(BO_LANG);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001781 return 0;
1782}
1783
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001784 static int
1785luaV_line(lua_State *L)
1786{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001787 luaV_pushline(L, curbuf, curwin->w_cursor.lnum);
1788 return 1;
1789}
1790
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001791 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001792luaV_list(lua_State *L)
1793{
Bram Moolenaarca06da92018-07-01 15:12:05 +02001794 list_T *l;
1795 int initarg = !lua_isnoneornil(L, 1);
1796
1797 if (initarg && lua_type(L, 1) != LUA_TTABLE)
1798 luaL_error(L, "table expected, got %s", luaL_typename(L, 1));
1799 l = list_alloc();
Bram Moolenaar1dced572012-04-05 16:54:08 +02001800 if (l == NULL)
1801 lua_pushnil(L);
1802 else
Bram Moolenaarca06da92018-07-01 15:12:05 +02001803 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001804 luaV_newlist(L, l);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001805 if (initarg) // traverse table to init list
Bram Moolenaar17413672018-07-14 20:49:42 +02001806 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001807 int notnil, i = 0;
1808 typval_T v;
Bram Moolenaar17413672018-07-14 20:49:42 +02001809 do
1810 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001811 lua_rawgeti(L, 1, ++i);
1812 notnil = !lua_isnil(L, -1);
Bram Moolenaar17413672018-07-14 20:49:42 +02001813 if (notnil)
1814 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001815 luaV_checktypval(L, -1, &v, "vim.list");
1816 list_append_tv(l, &v);
Bram Moolenaar713bf9e92019-03-16 16:38:41 +01001817 clear_tv(&v);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001818 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001819 lua_pop(L, 1); // value
Bram Moolenaarca06da92018-07-01 15:12:05 +02001820 } while (notnil);
1821 }
1822 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001823 return 1;
1824}
1825
1826 static int
1827luaV_dict(lua_State *L)
1828{
Bram Moolenaarca06da92018-07-01 15:12:05 +02001829 dict_T *d;
1830 int initarg = !lua_isnoneornil(L, 1);
1831
1832 if (initarg && lua_type(L, 1) != LUA_TTABLE)
1833 luaL_error(L, "table expected, got %s", luaL_typename(L, 1));
1834 d = dict_alloc();
Bram Moolenaar1dced572012-04-05 16:54:08 +02001835 if (d == NULL)
1836 lua_pushnil(L);
1837 else
Bram Moolenaarca06da92018-07-01 15:12:05 +02001838 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001839 luaV_newdict(L, d);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001840 if (initarg) // traverse table to init dict
Bram Moolenaarca06da92018-07-01 15:12:05 +02001841 {
1842 lua_pushnil(L);
1843 while (lua_next(L, 1))
1844 {
1845 char_u *key;
1846 dictitem_T *di;
1847 typval_T v;
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001848
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001849 lua_pushvalue(L, -2); // dup key in case it's a number
Bram Moolenaarca06da92018-07-01 15:12:05 +02001850 key = (char_u *) lua_tostring(L, -1);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001851 if (key == NULL)
1852 {
1853 lua_pushnil(L);
1854 return 1;
1855 }
1856 if (*key == NUL)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001857 luaL_error(L, "table has empty key");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001858 luaV_checktypval(L, -2, &v, "vim.dict"); // value
Bram Moolenaarca06da92018-07-01 15:12:05 +02001859 di = dictitem_alloc(key);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001860 if (di == NULL || dict_add(d, di) == FAIL)
1861 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001862 vim_free(di);
1863 lua_pushnil(L);
1864 return 1;
1865 }
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001866 di->di_tv = v;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001867 lua_pop(L, 2); // key copy and value
Bram Moolenaarca06da92018-07-01 15:12:05 +02001868 }
1869 }
1870 }
1871 return 1;
1872}
1873
1874 static int
Bram Moolenaarb7828692019-03-23 13:57:02 +01001875luaV_blob(lua_State *L)
1876{
1877 blob_T *b;
1878 int initarg = !lua_isnoneornil(L, 1);
1879
1880 if (initarg && !lua_isstring(L, 1))
1881 luaL_error(L, "string expected, got %s", luaL_typename(L, 1));
1882 b = blob_alloc();
1883 if (b == NULL)
1884 lua_pushnil(L);
1885 else
1886 {
1887 luaV_newblob(L, b);
1888 if (initarg)
1889 {
1890 size_t i, l = 0;
1891 const char *s = lua_tolstring(L, 1, &l);
1892
Bram Moolenaar5f1d3ae2020-02-11 22:37:35 +01001893 if (ga_grow(&b->bv_ga, (int)l) == OK)
Bram Moolenaar6fb5c972019-03-26 21:44:20 +01001894 for (i = 0; i < l; ++i)
1895 ga_append(&b->bv_ga, s[i]);
Bram Moolenaarb7828692019-03-23 13:57:02 +01001896 }
1897 }
1898 return 1;
1899}
1900
1901 static int
Bram Moolenaarca06da92018-07-01 15:12:05 +02001902luaV_funcref(lua_State *L)
1903{
1904 const char *name = luaL_checkstring(L, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001905 // note: not checking if function exists (needs function_exists)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001906 if (name == NULL || *name == NUL || VIM_ISDIGIT(*name))
1907 luaL_error(L, "invalid function name: %s", name);
1908 luaV_newfuncref(L, (char_u *) name);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001909 return 1;
1910}
1911
1912 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001913luaV_buffer(lua_State *L)
1914{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001915 buf_T *buf;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001916 if (lua_isstring(L, 1)) // get by number or name?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001917 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001918 if (lua_isnumber(L, 1)) // by number?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001919 {
1920 int n = lua_tointeger(L, 1);
Bram Moolenaar29323592016-07-24 22:04:11 +02001921 FOR_ALL_BUFFERS(buf)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001922 if (buf->b_fnum == n) break;
1923 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001924 else // by name
1925 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001926 size_t l;
1927 const char *s = lua_tolstring(L, 1, &l);
Bram Moolenaar29323592016-07-24 22:04:11 +02001928 FOR_ALL_BUFFERS(buf)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001929 {
1930 if (buf->b_ffname == NULL || buf->b_sfname == NULL)
1931 {
1932 if (l == 0) break;
1933 }
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001934 else if (strncmp(s, (char *)buf->b_ffname, l) == 0
1935 || strncmp(s, (char *)buf->b_sfname, l) == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001936 break;
1937 }
1938 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001939 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001940 else
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001941 buf = (lua_toboolean(L, 1)) ? firstbuf : curbuf; // first buffer?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001942 luaV_pushbuffer(L, buf);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001943 return 1;
1944}
1945
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001946 static int
1947luaV_window(lua_State *L)
1948{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001949 win_T *win;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001950 if (lua_isnumber(L, 1)) // get by number?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001951 {
1952 int n = lua_tointeger(L, 1);
1953 for (win = firstwin; win != NULL; win = win->w_next, n--)
1954 if (n == 1) break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001955 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001956 else
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001957 win = (lua_toboolean(L, 1)) ? firstwin : curwin; // first window?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001958 luaV_pushwindow(L, win);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001959 return 1;
1960}
1961
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001962 static int
1963luaV_open(lua_State *L)
1964{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001965 char_u *s = NULL;
1966#ifdef HAVE_SANDBOX
1967 luaV_checksandbox(L);
1968#endif
1969 if (lua_isstring(L, 1)) s = (char_u *) lua_tostring(L, 1);
Bram Moolenaarfa263a52011-12-08 16:00:16 +01001970 luaV_pushbuffer(L, buflist_new(s, NULL, 1L, BLN_LISTED));
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001971 return 1;
1972}
1973
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001974 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001975luaV_type(lua_State *L)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001976{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001977 luaL_checkany(L, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001978 if (lua_type(L, 1) == LUA_TUSERDATA) // check vim udata?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001979 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001980 lua_settop(L, 1);
1981 if (lua_getmetatable(L, 1))
1982 {
1983 luaV_getfield(L, LUAVIM_LIST);
1984 if (lua_rawequal(L, -1, 2))
1985 {
1986 lua_pushstring(L, "list");
1987 return 1;
1988 }
1989 luaV_getfield(L, LUAVIM_DICT);
1990 if (lua_rawequal(L, -1, 2))
1991 {
1992 lua_pushstring(L, "dict");
1993 return 1;
1994 }
Bram Moolenaarb7828692019-03-23 13:57:02 +01001995 luaV_getfield(L, LUAVIM_BLOB);
1996 if (lua_rawequal(L, -1, 2))
1997 {
1998 lua_pushstring(L, "blob");
1999 return 1;
2000 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02002001 luaV_getfield(L, LUAVIM_FUNCREF);
2002 if (lua_rawequal(L, -1, 2))
2003 {
2004 lua_pushstring(L, "funcref");
2005 return 1;
2006 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02002007 luaV_getfield(L, LUAVIM_BUFFER);
2008 if (lua_rawequal(L, -1, 2))
2009 {
2010 lua_pushstring(L, "buffer");
2011 return 1;
2012 }
2013 luaV_getfield(L, LUAVIM_WINDOW);
2014 if (lua_rawequal(L, -1, 2))
2015 {
2016 lua_pushstring(L, "window");
2017 return 1;
2018 }
2019 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002020 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002021 lua_pushstring(L, luaL_typename(L, 1)); // fallback
Bram Moolenaar1dced572012-04-05 16:54:08 +02002022 return 1;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002023}
2024
Bram Moolenaareb04f082020-05-17 14:32:35 +02002025 static int
2026luaV_call(lua_State *L)
2027{
2028 int argc = lua_gettop(L) - 1;
2029 size_t funcname_len;
2030 char_u *funcname;
2031 char *error = NULL;
2032 typval_T rettv;
2033 typval_T argv[MAX_FUNC_ARGS + 1];
2034 int i = 0;
2035
2036 if (argc > MAX_FUNC_ARGS)
2037 return luaL_error(L, "Function called with too many arguments");
2038
2039 funcname = (char_u *)luaL_checklstring(L, 1, &funcname_len);
2040
2041 for (; i < argc; i++)
2042 {
2043 if (luaV_totypval(L, i + 2, &argv[i]) == FAIL)
2044 {
2045 error = "lua: cannot convert value";
2046 goto free_vim_args;
2047 }
2048 }
2049
2050 argv[argc].v_type = VAR_UNKNOWN;
2051
2052 if (call_vim_function(funcname, argc, argv, &rettv) == FAIL)
2053 {
2054 error = "lua: call_vim_function failed";
2055 goto free_vim_args;
2056 }
2057
2058 luaV_pushtypval(L, &rettv);
2059 clear_tv(&rettv);
2060
2061free_vim_args:
2062 while (i > 0)
2063 clear_tv(&argv[--i]);
2064
2065 if (error == NULL)
2066 return 1;
2067 else
2068 return luaL_error(L, error);
2069}
2070
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002071static const luaL_Reg luaV_module[] = {
2072 {"command", luaV_command},
2073 {"eval", luaV_eval},
2074 {"beep", luaV_beep},
2075 {"line", luaV_line},
Bram Moolenaar1dced572012-04-05 16:54:08 +02002076 {"list", luaV_list},
2077 {"dict", luaV_dict},
Bram Moolenaarb7828692019-03-23 13:57:02 +01002078 {"blob", luaV_blob},
Bram Moolenaarca06da92018-07-01 15:12:05 +02002079 {"funcref", luaV_funcref},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002080 {"buffer", luaV_buffer},
2081 {"window", luaV_window},
2082 {"open", luaV_open},
Bram Moolenaar1dced572012-04-05 16:54:08 +02002083 {"type", luaV_type},
Bram Moolenaareb04f082020-05-17 14:32:35 +02002084 {"call", luaV_call},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002085 {NULL, NULL}
2086};
2087
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002088/*
2089 * for freeing list, dict, buffer and window objects; lightuserdata as arg
2090 */
Bram Moolenaar1dced572012-04-05 16:54:08 +02002091 static int
2092luaV_free(lua_State *L)
2093{
2094 lua_pushnil(L);
2095 luaV_setudata(L, lua_touserdata(L, 1));
2096 return 0;
2097}
2098
2099 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002100luaV_luaeval(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02002101{
2102 luaL_Buffer b;
2103 size_t l;
2104 const char *str = lua_tolstring(L, 1, &l);
2105 typval_T *arg = (typval_T *) lua_touserdata(L, 2);
2106 typval_T *rettv = (typval_T *) lua_touserdata(L, 3);
2107 luaL_buffinit(L, &b);
2108 luaL_addlstring(&b, LUAVIM_EVALHEADER, sizeof(LUAVIM_EVALHEADER) - 1);
2109 luaL_addlstring(&b, str, l);
2110 luaL_pushresult(&b);
2111 str = lua_tolstring(L, -1, &l);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002112 if (luaL_loadbuffer(L, str, l, LUAVIM_EVALNAME)) // compile error?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002113 {
2114 luaV_emsg(L);
2115 return 0;
2116 }
2117 luaV_pushtypval(L, arg);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002118 if (lua_pcall(L, 1, 1, 0)) // running error?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002119 {
2120 luaV_emsg(L);
2121 return 0;
2122 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02002123 if (luaV_totypval(L, -1, rettv) == FAIL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002124 emsg("luaeval: cannot convert value");
Bram Moolenaarf554a322015-02-04 23:08:01 +01002125 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02002126}
2127
2128 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002129luaV_setref(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02002130{
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002131 int copyID = lua_tointeger(L, 1);
2132 int abort = FALSE;
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002133
Bram Moolenaar1dced572012-04-05 16:54:08 +02002134 luaV_getfield(L, LUAVIM_LIST);
2135 luaV_getfield(L, LUAVIM_DICT);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002136 luaV_getfield(L, LUAVIM_FUNCREF);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002137 lua_pushnil(L);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002138 // traverse cache table
Bram Moolenaarb84634d2015-02-04 22:02:37 +01002139 while (!abort && lua_next(L, lua_upvalueindex(1)) != 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02002140 {
2141 lua_getmetatable(L, -1);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002142 if (lua_rawequal(L, -1, 2)) // list?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002143 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002144 list_T *l = (list_T *)lua_touserdata(L, 5); // key
2145
Bram Moolenaar7be3ab22019-06-23 01:46:15 +02002146 abort = set_ref_in_list(l, copyID);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002147 }
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002148 else if (lua_rawequal(L, -1, 3)) // dict?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002149 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002150 dict_T *d = (dict_T *)lua_touserdata(L, 5); // key
2151
Bram Moolenaar7be3ab22019-06-23 01:46:15 +02002152 abort = set_ref_in_dict(d, copyID);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002153 }
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002154 else if (lua_rawequal(L, -1, 4)) // funcref?
2155 {
2156 luaV_Funcref *f = (luaV_Funcref *)lua_touserdata(L, 5); // key
2157
Bram Moolenaar7be3ab22019-06-23 01:46:15 +02002158 abort = set_ref_in_dict(f->self, copyID);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002159 }
2160 lua_pop(L, 2); // metatable and value
Bram Moolenaar1dced572012-04-05 16:54:08 +02002161 }
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002162 lua_pushinteger(L, abort);
Bram Moolenaarf554a322015-02-04 23:08:01 +01002163 return 1;
Bram Moolenaar1dced572012-04-05 16:54:08 +02002164}
2165
Bram Moolenaareb04f082020-05-17 14:32:35 +02002166#define LUA_VIM_FN_CODE \
Bram Moolenaar788fbb42020-05-31 14:08:12 +02002167 "vim.fn = setmetatable({}, {\n"\
2168 " __index = function (t, key)\n"\
2169 " local function _fn(...)\n"\
2170 " return vim.call(key, ...)\n"\
2171 " end\n"\
2172 " t[key] = _fn\n"\
2173 " return _fn\n"\
2174 " end\n"\
2175 " })"
2176
2177#define LUA_VIM_UPDATE_PACKAGE_PATHS \
2178 "local last_vim_paths = {}\n"\
2179 "vim._update_package_paths = function ()\n"\
2180 " local cur_vim_paths = {}\n"\
2181 " local function split(s, delimiter)\n"\
2182 " result = {}\n"\
2183 " for match in (s..delimiter):gmatch(\"(.-)\"..delimiter) do\n"\
2184 " table.insert(result, match)\n"\
2185 " end\n"\
2186 " return result\n"\
2187 " end\n"\
2188 " local rtps = split(vim.eval('&runtimepath'), ',')\n"\
2189 " local sep = package.config:sub(1, 1)\n"\
2190 " for _, key in ipairs({'path', 'cpath'}) do\n"\
2191 " local orig_str = package[key] .. ';'\n"\
2192 " local pathtrails_ordered = {}\n"\
2193 " -- Note: ignores trailing item without trailing `;`. Not using something\n"\
2194 " -- simpler in order to preserve empty items (stand for default path).\n"\
2195 " local orig = {}\n"\
2196 " for s in orig_str:gmatch('[^;]*;') do\n"\
2197 " s = s:sub(1, -2) -- Strip trailing semicolon\n"\
2198 " orig[#orig + 1] = s\n"\
2199 " end\n"\
2200 " if key == 'path' then\n"\
2201 " -- /?.lua and /?/init.lua\n"\
2202 " pathtrails_ordered = {sep .. '?.lua', sep .. '?' .. sep .. 'init.lua'}\n"\
2203 " else\n"\
2204 " local pathtrails = {}\n"\
2205 " for _, s in ipairs(orig) do\n"\
2206 " -- Find out path patterns. pathtrail should contain something like\n"\
2207 " -- /?.so, \?.dll. This allows not to bother determining what correct\n"\
2208 " -- suffixes are.\n"\
2209 " local pathtrail = s:match('[/\\\\][^/\\\\]*%?.*$')\n"\
2210 " if pathtrail and not pathtrails[pathtrail] then\n"\
2211 " pathtrails[pathtrail] = true\n"\
2212 " pathtrails_ordered[#pathtrails_ordered + 1] = pathtrail\n"\
2213 " end\n"\
2214 " end\n"\
2215 " end\n"\
2216 " local new = {}\n"\
2217 " for _, rtp in ipairs(rtps) do\n"\
2218 " if not rtp:match(';') then\n"\
2219 " for _, pathtrail in pairs(pathtrails_ordered) do\n"\
2220 " local new_path = rtp .. sep .. 'lua' .. pathtrail\n"\
2221 " -- Always keep paths from &runtimepath at the start:\n"\
2222 " -- append them here disregarding orig possibly containing one of them.\n"\
2223 " new[#new + 1] = new_path\n"\
2224 " cur_vim_paths[new_path] = true\n"\
2225 " end\n"\
2226 " end\n"\
2227 " end\n"\
2228 " for _, orig_path in ipairs(orig) do\n"\
2229 " -- Handle removing obsolete paths originating from &runtimepath: such\n"\
2230 " -- paths either belong to cur_nvim_paths and were already added above or\n"\
2231 " -- to last_nvim_paths and should not be added at all if corresponding\n"\
2232 " -- entry was removed from &runtimepath list.\n"\
2233 " if not (cur_vim_paths[orig_path] or last_vim_paths[orig_path]) then\n"\
2234 " new[#new + 1] = orig_path\n"\
2235 " end\n"\
2236 " end\n"\
2237 " package[key] = table.concat(new, ';')\n"\
2238 " end\n"\
2239 " last_vim_paths = cur_vim_paths\n"\
2240 "end"
Bram Moolenaareb04f082020-05-17 14:32:35 +02002241
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002242 static int
2243luaopen_vim(lua_State *L)
2244{
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002245 // set cache table
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002246 lua_newtable(L);
2247 lua_newtable(L);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002248 lua_pushstring(L, "v");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002249 lua_setfield(L, -2, "__mode");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002250 lua_setmetatable(L, -2); // cache is weak-valued
2251 // print
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002252 lua_pushcfunction(L, luaV_print);
2253 lua_setglobal(L, "print");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002254 // debug.debug
Bram Moolenaar38e2b062011-09-21 17:15:39 +02002255 lua_getglobal(L, "debug");
2256 lua_pushcfunction(L, luaV_debug);
2257 lua_setfield(L, -2, "debug");
2258 lua_pop(L, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002259 // free
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002260 lua_pushlightuserdata(L, (void *) LUAVIM_FREE);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002261 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002262 lua_pushcclosure(L, luaV_free, 1);
2263 lua_rawset(L, LUA_REGISTRYINDEX);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002264 // luaeval
Bram Moolenaar1dced572012-04-05 16:54:08 +02002265 lua_pushlightuserdata(L, (void *) LUAVIM_LUAEVAL);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002266 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002267 lua_pushcclosure(L, luaV_luaeval, 1);
2268 lua_rawset(L, LUA_REGISTRYINDEX);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002269 // setref
Bram Moolenaar1dced572012-04-05 16:54:08 +02002270 lua_pushlightuserdata(L, (void *) LUAVIM_SETREF);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002271 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002272 lua_pushcclosure(L, luaV_setref, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002273 lua_rawset(L, LUA_REGISTRYINDEX);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002274 // register
Bram Moolenaar1dced572012-04-05 16:54:08 +02002275 luaV_newmetatable(L, LUAVIM_LIST);
2276 lua_pushvalue(L, 1);
2277 luaV_openlib(L, luaV_List_mt, 1);
2278 luaV_newmetatable(L, LUAVIM_DICT);
2279 lua_pushvalue(L, 1);
2280 luaV_openlib(L, luaV_Dict_mt, 1);
Bram Moolenaarb7828692019-03-23 13:57:02 +01002281 luaV_newmetatable(L, LUAVIM_BLOB);
2282 lua_pushvalue(L, 1);
2283 luaV_openlib(L, luaV_Blob_mt, 1);
Bram Moolenaarca06da92018-07-01 15:12:05 +02002284 luaV_newmetatable(L, LUAVIM_FUNCREF);
2285 lua_pushvalue(L, 1);
2286 luaV_openlib(L, luaV_Funcref_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002287 luaV_newmetatable(L, LUAVIM_BUFFER);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002288 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002289 luaV_openlib(L, luaV_Buffer_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002290 luaV_newmetatable(L, LUAVIM_WINDOW);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002291 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002292 luaV_openlib(L, luaV_Window_mt, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002293 lua_newtable(L); // vim table
2294 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002295 luaV_openlib(L, luaV_module, 1);
2296 lua_setglobal(L, LUAVIM_NAME);
Bram Moolenaareb04f082020-05-17 14:32:35 +02002297 // custom code
Bram Moolenaar9309eb22020-05-17 16:53:56 +02002298 (void)luaL_dostring(L, LUA_VIM_FN_CODE);
Bram Moolenaar788fbb42020-05-31 14:08:12 +02002299 (void)luaL_dostring(L, LUA_VIM_UPDATE_PACKAGE_PATHS);
2300
2301 lua_getglobal(L, "vim");
2302 lua_getfield(L, -1, "_update_package_paths");
2303
2304 if (lua_pcall(L, 0, 0, 0))
2305 luaV_emsg(L);
2306
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002307 return 0;
2308}
2309
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002310 static lua_State *
2311luaV_newstate(void)
2312{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002313 lua_State *L = luaL_newstate();
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002314 luaL_openlibs(L); // core libs
2315 lua_pushcfunction(L, luaopen_vim); // vim
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002316 lua_call(L, 0, 0);
2317 return L;
2318}
2319
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002320 static void
2321luaV_setrange(lua_State *L, int line1, int line2)
2322{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002323 lua_getglobal(L, LUAVIM_NAME);
2324 lua_pushinteger(L, line1);
2325 lua_setfield(L, -2, "firstline");
2326 lua_pushinteger(L, line2);
2327 lua_setfield(L, -2, "lastline");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002328 lua_pop(L, 1); // vim table
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002329}
2330
2331
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002332// ======= Interface =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002333
2334static lua_State *L = NULL;
2335
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002336 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02002337lua_isopen(void)
Bram Moolenaar2bd6a1b2010-08-12 22:14:01 +02002338{
2339 return L != NULL;
2340}
2341
2342 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002343lua_init(void)
2344{
Bram Moolenaar1dced572012-04-05 16:54:08 +02002345 if (!lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002346 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002347#ifdef DYNAMIC_LUA
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002348 if (!lua_enabled(TRUE))
2349 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002350 emsg(_("Lua library cannot be loaded."));
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002351 return FAIL;
2352 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002353#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002354 L = luaV_newstate();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002355 }
2356 return OK;
2357}
2358
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002359 void
2360lua_end(void)
2361{
Bram Moolenaar1dced572012-04-05 16:54:08 +02002362 if (lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002363 {
2364 lua_close(L);
2365 L = NULL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002366 }
2367}
2368
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002369/*
2370 * ex commands
2371 */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002372 void
2373ex_lua(exarg_T *eap)
2374{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002375 char *script;
2376 if (lua_init() == FAIL) return;
2377 script = (char *) script_get(eap, eap->arg);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002378 if (!eap->skip)
2379 {
2380 char *s = (script) ? script : (char *) eap->arg;
2381 luaV_setrange(L, eap->line1, eap->line2);
2382 if (luaL_loadbuffer(L, s, strlen(s), LUAVIM_CHUNKNAME)
2383 || lua_pcall(L, 0, 0, 0))
2384 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002385 }
2386 if (script != NULL) vim_free(script);
2387}
2388
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002389 void
2390ex_luado(exarg_T *eap)
2391{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002392 linenr_T l;
2393 const char *s = (const char *) eap->arg;
2394 luaL_Buffer b;
2395 size_t len;
Bram Moolenaard58f03b2017-01-29 22:48:45 +01002396 buf_T *was_curbuf = curbuf;
2397
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002398 if (lua_init() == FAIL) return;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002399 if (u_save(eap->line1 - 1, eap->line2 + 1) == FAIL)
2400 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002401 emsg(_("cannot save undo information"));
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002402 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002403 }
2404 luaV_setrange(L, eap->line1, eap->line2);
2405 luaL_buffinit(L, &b);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002406 luaL_addlstring(&b, "return function(line, linenr) ", 30); // header
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002407 luaL_addlstring(&b, s, strlen(s));
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002408 luaL_addlstring(&b, " end", 4); // footer
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002409 luaL_pushresult(&b);
2410 s = lua_tolstring(L, -1, &len);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002411 if (luaL_loadbuffer(L, s, len, LUAVIM_CHUNKNAME))
2412 {
2413 luaV_emsg(L);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002414 lua_pop(L, 1); // function body
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002415 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002416 }
2417 lua_call(L, 0, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002418 lua_replace(L, -2); // function -> body
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002419 for (l = eap->line1; l <= eap->line2; l++)
2420 {
Bram Moolenaare49b8e82020-07-01 13:52:55 +02002421 // Check the line number, the command may have deleted lines.
Bram Moolenaard58f03b2017-01-29 22:48:45 +01002422 if (l > curbuf->b_ml.ml_line_count)
2423 break;
2424
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002425 lua_pushvalue(L, -1); // function
2426 luaV_pushline(L, curbuf, l); // current line as arg
2427 lua_pushinteger(L, l); // current line number as arg
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +02002428 if (lua_pcall(L, 2, 1, 0))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002429 {
2430 luaV_emsg(L);
2431 break;
2432 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002433 // Catch the command switching to another buffer.
Bram Moolenaard58f03b2017-01-29 22:48:45 +01002434 if (curbuf != was_curbuf)
2435 break;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002436 if (lua_isstring(L, -1)) // update line?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002437 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002438#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002439 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002440#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002441 ml_replace(l, luaV_toline(L, -1), TRUE);
2442 changed_bytes(l, 0);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002443 lua_pop(L, 1); // result from luaV_toline
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002444 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002445 lua_pop(L, 1); // line
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002446 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002447 lua_pop(L, 1); // function
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002448 check_cursor();
2449 update_screen(NOT_VALID);
2450}
2451
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002452 void
2453ex_luafile(exarg_T *eap)
2454{
2455 if (lua_init() == FAIL)
2456 return;
2457 if (!eap->skip)
2458 {
2459 luaV_setrange(L, eap->line1, eap->line2);
2460 if (luaL_loadfile(L, (char *) eap->arg) || lua_pcall(L, 0, 0, 0))
2461 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002462 }
2463}
2464
Bram Moolenaar1dced572012-04-05 16:54:08 +02002465#define luaV_freetype(typ,tname) \
2466 void \
2467 lua_##tname##_free(typ *o) \
2468 { \
2469 if (!lua_isopen()) return; \
2470 luaV_getfield(L, LUAVIM_FREE); \
2471 lua_pushlightuserdata(L, (void *) o); \
2472 lua_call(L, 1, 0); \
2473 }
2474
2475luaV_freetype(buf_T, buffer)
2476luaV_freetype(win_T, window)
2477
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002478 void
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002479do_luaeval(char_u *str, typval_T *arg, typval_T *rettv)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002480{
Bram Moolenaar1dced572012-04-05 16:54:08 +02002481 lua_init();
2482 luaV_getfield(L, LUAVIM_LUAEVAL);
2483 lua_pushstring(L, (char *) str);
2484 lua_pushlightuserdata(L, (void *) arg);
2485 lua_pushlightuserdata(L, (void *) rettv);
2486 lua_call(L, 3, 0);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002487}
2488
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002489 int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002490set_ref_in_lua(int copyID)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002491{
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002492 int aborted = 0;
2493
2494 if (lua_isopen())
2495 {
2496 luaV_getfield(L, LUAVIM_SETREF);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002497 // call the function with 1 arg, getting 1 result back
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002498 lua_pushinteger(L, copyID);
2499 lua_call(L, 1, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002500 // get the result
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002501 aborted = lua_tointeger(L, -1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002502 // pop result off the stack
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002503 lua_pop(L, 1);
2504 }
2505 return aborted;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002506}
2507
Bram Moolenaar788fbb42020-05-31 14:08:12 +02002508 void
2509update_package_paths_in_lua()
2510{
2511 if (lua_isopen())
2512 {
2513 lua_getglobal(L, "vim");
2514 lua_getfield(L, -1, "_update_package_paths");
2515
2516 if (lua_pcall(L, 0, 0, 0))
2517 luaV_emsg(L);
2518 }
2519}
2520
Bram Moolenaar801ab062020-06-25 19:27:56 +02002521/*
2522 * Native C function callback
2523 */
2524 static int
2525luaV_call_lua_func(
2526 int argcount,
2527 typval_T *argvars,
2528 typval_T *rettv,
2529 void *state)
2530{
2531 int i;
2532 int luaargcount = argcount;
2533 luaV_CFuncState *funcstate = (luaV_CFuncState*)state;
2534 lua_rawgeti(funcstate->L, LUA_REGISTRYINDEX, funcstate->lua_funcref);
2535
2536 if (funcstate->lua_tableref != LUA_NOREF)
2537 {
2538 // First arg for metatable __call method is a table
2539 luaargcount += 1;
2540 lua_rawgeti(funcstate->L, LUA_REGISTRYINDEX, funcstate->lua_tableref);
2541 }
2542
2543 for (i = 0; i < argcount; ++i)
2544 luaV_pushtypval(funcstate->L, &argvars[i]);
2545
2546 if (lua_pcall(funcstate->L, luaargcount, 1, 0))
2547 {
2548 luaV_emsg(funcstate->L);
2549 return FCERR_OTHER;
2550 }
2551
2552 luaV_checktypval(funcstate->L, -1, rettv, "get return value");
2553 return FCERR_NONE;
2554}
2555
2556/*
2557 * Free up any lua references held by the func state.
2558 */
2559 static void
2560luaV_call_lua_func_free(void *state)
2561{
2562 luaV_CFuncState *funcstate = (luaV_CFuncState*)state;
2563 luaL_unref(L, LUA_REGISTRYINDEX, funcstate->lua_funcref);
2564 funcstate->L = NULL;
2565 if (funcstate->lua_tableref != LUA_NOREF)
2566 luaL_unref(L, LUA_REGISTRYINDEX, funcstate->lua_tableref);
2567 VIM_CLEAR(funcstate);
2568}
2569
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002570#endif