blob: 9852c618e894c1c4b6cdf45fe75162b3aadf8ead [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 Moolenaarbd2f3c32012-04-06 14:31:00 +0200123#define luaL_checkany dll_luaL_checkany
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200124#define luaL_checklstring dll_luaL_checklstring
125#define luaL_checkinteger dll_luaL_checkinteger
126#define luaL_optinteger dll_luaL_optinteger
127#define luaL_checktype dll_luaL_checktype
128#define luaL_error dll_luaL_error
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200129#define luaL_newstate dll_luaL_newstate
130#define luaL_buffinit dll_luaL_buffinit
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200131#define luaL_addlstring dll_luaL_addlstring
132#define luaL_pushresult dll_luaL_pushresult
Bram Moolenaardf1643a2020-05-17 18:53:27 +0200133#define luaL_loadstring dll_luaL_loadstring
Bram Moolenaar1e4c7d02020-06-25 20:56:42 +0200134#define luaL_ref dll_luaL_ref
135#define luaL_unref dll_luaL_unref
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100136// lua
Bram Moolenaar1dced572012-04-05 16:54:08 +0200137#if LUA_VERSION_NUM <= 501
138#define lua_tonumber dll_lua_tonumber
139#define lua_tointeger dll_lua_tointeger
140#define lua_call dll_lua_call
141#define lua_pcall dll_lua_pcall
142#else
143#define lua_tonumberx dll_lua_tonumberx
144#define lua_tointegerx dll_lua_tointegerx
145#define lua_callk dll_lua_callk
146#define lua_pcallk dll_lua_pcallk
147#define lua_getglobal dll_lua_getglobal
148#define lua_setglobal dll_lua_setglobal
Bram Moolenaar1dced572012-04-05 16:54:08 +0200149#endif
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200150#if LUA_VERSION_NUM <= 502
151#define lua_replace dll_lua_replace
152#define lua_remove dll_lua_remove
153#endif
154#if LUA_VERSION_NUM >= 503
155#define lua_rotate dll_lua_rotate
156#define lua_copy dll_lua_copy
157#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200158#define lua_typename dll_lua_typename
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200159#define lua_close dll_lua_close
160#define lua_gettop dll_lua_gettop
161#define lua_settop dll_lua_settop
162#define lua_pushvalue dll_lua_pushvalue
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200163#define lua_isnumber dll_lua_isnumber
164#define lua_isstring dll_lua_isstring
165#define lua_type dll_lua_type
166#define lua_rawequal dll_lua_rawequal
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200167#define lua_toboolean dll_lua_toboolean
168#define lua_tolstring dll_lua_tolstring
169#define lua_touserdata dll_lua_touserdata
170#define lua_pushnil dll_lua_pushnil
171#define lua_pushnumber dll_lua_pushnumber
172#define lua_pushinteger dll_lua_pushinteger
173#define lua_pushlstring dll_lua_pushlstring
174#define lua_pushstring dll_lua_pushstring
175#define lua_pushfstring dll_lua_pushfstring
176#define lua_pushcclosure dll_lua_pushcclosure
177#define lua_pushboolean dll_lua_pushboolean
178#define lua_pushlightuserdata dll_lua_pushlightuserdata
179#define lua_getfield dll_lua_getfield
180#define lua_rawget dll_lua_rawget
Bram Moolenaar1dced572012-04-05 16:54:08 +0200181#define lua_rawgeti dll_lua_rawgeti
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200182#define lua_createtable dll_lua_createtable
Bram Moolenaar830e3582018-08-21 14:23:35 +0200183#if LUA_VERSION_NUM >= 504
184 #define lua_newuserdatauv dll_lua_newuserdatauv
185#else
186 #define lua_newuserdata dll_lua_newuserdata
187#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200188#define lua_getmetatable dll_lua_getmetatable
189#define lua_setfield dll_lua_setfield
190#define lua_rawset dll_lua_rawset
191#define lua_rawseti dll_lua_rawseti
192#define lua_setmetatable dll_lua_setmetatable
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200193#define lua_next dll_lua_next
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100194// libs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200195#define luaopen_base dll_luaopen_base
196#define luaopen_table dll_luaopen_table
197#define luaopen_string dll_luaopen_string
198#define luaopen_math dll_luaopen_math
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200199#define luaopen_io dll_luaopen_io
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200200#define luaopen_os dll_luaopen_os
201#define luaopen_package dll_luaopen_package
202#define luaopen_debug dll_luaopen_debug
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200203#define luaL_openlibs dll_luaL_openlibs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200204
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100205// lauxlib
Bram Moolenaar1dced572012-04-05 16:54:08 +0200206#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200207void (*dll_luaL_register) (lua_State *L, const char *libname, const luaL_Reg *l);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200208char *(*dll_luaL_prepbuffer) (luaL_Buffer *B);
209void (*dll_luaL_openlib) (lua_State *L, const char *libname, const luaL_Reg *l, int nup);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200210int (*dll_luaL_typerror) (lua_State *L, int narg, const char *tname);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200211int (*dll_luaL_loadfile) (lua_State *L, const char *filename);
212int (*dll_luaL_loadbuffer) (lua_State *L, const char *buff, size_t sz, const char *name);
213#else
214char *(*dll_luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
215void (*dll_luaL_setfuncs) (lua_State *L, const luaL_Reg *l, int nup);
216int (*dll_luaL_loadfilex) (lua_State *L, const char *filename, const char *mode);
217int (*dll_luaL_loadbufferx) (lua_State *L, const char *buff, size_t sz, const char *name, const char *mode);
218int (*dll_luaL_argerror) (lua_State *L, int numarg, const char *extramsg);
219#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200220void (*dll_luaL_checkany) (lua_State *L, int narg);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200221const char *(*dll_luaL_checklstring) (lua_State *L, int numArg, size_t *l);
222lua_Integer (*dll_luaL_checkinteger) (lua_State *L, int numArg);
223lua_Integer (*dll_luaL_optinteger) (lua_State *L, int nArg, lua_Integer def);
224void (*dll_luaL_checktype) (lua_State *L, int narg, int t);
225int (*dll_luaL_error) (lua_State *L, const char *fmt, ...);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200226lua_State *(*dll_luaL_newstate) (void);
227void (*dll_luaL_buffinit) (lua_State *L, luaL_Buffer *B);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200228void (*dll_luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
229void (*dll_luaL_pushresult) (luaL_Buffer *B);
Bram Moolenaardf1643a2020-05-17 18:53:27 +0200230int (*dll_luaL_loadstring) (lua_State *L, const char *s);
Bram Moolenaar1e4c7d02020-06-25 20:56:42 +0200231int (*dll_luaL_ref) (lua_State *L, int idx);
232#if LUA_VERSION_NUM <= 502
233void (*dll_luaL_unref) (lua_State *L, int idx, int n);
234#else
235void (*dll_luaL_unref) (lua_State *L, int idx, lua_Integer n);
236#endif
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100237// lua
Bram Moolenaar1dced572012-04-05 16:54:08 +0200238#if LUA_VERSION_NUM <= 501
239lua_Number (*dll_lua_tonumber) (lua_State *L, int idx);
240lua_Integer (*dll_lua_tointeger) (lua_State *L, int idx);
241void (*dll_lua_call) (lua_State *L, int nargs, int nresults);
242int (*dll_lua_pcall) (lua_State *L, int nargs, int nresults, int errfunc);
243#else
244lua_Number (*dll_lua_tonumberx) (lua_State *L, int idx, int *isnum);
245lua_Integer (*dll_lua_tointegerx) (lua_State *L, int idx, int *isnum);
246void (*dll_lua_callk) (lua_State *L, int nargs, int nresults, int ctx,
Bram Moolenaardb913952012-06-29 12:54:53 +0200247 lua_CFunction k);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200248int (*dll_lua_pcallk) (lua_State *L, int nargs, int nresults, int errfunc,
Bram Moolenaardb913952012-06-29 12:54:53 +0200249 int ctx, lua_CFunction k);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200250void (*dll_lua_getglobal) (lua_State *L, const char *var);
251void (*dll_lua_setglobal) (lua_State *L, const char *var);
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200252#endif
253#if LUA_VERSION_NUM <= 502
254void (*dll_lua_replace) (lua_State *L, int idx);
255void (*dll_lua_remove) (lua_State *L, int idx);
256#endif
257#if LUA_VERSION_NUM >= 503
258void (*dll_lua_rotate) (lua_State *L, int idx, int n);
Bram Moolenaar9514b1f2015-06-25 18:27:32 +0200259void (*dll_lua_copy) (lua_State *L, int fromidx, int toidx);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200260#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200261const char *(*dll_lua_typename) (lua_State *L, int tp);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200262void (*dll_lua_close) (lua_State *L);
263int (*dll_lua_gettop) (lua_State *L);
264void (*dll_lua_settop) (lua_State *L, int idx);
265void (*dll_lua_pushvalue) (lua_State *L, int idx);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200266int (*dll_lua_isnumber) (lua_State *L, int idx);
267int (*dll_lua_isstring) (lua_State *L, int idx);
268int (*dll_lua_type) (lua_State *L, int idx);
269int (*dll_lua_rawequal) (lua_State *L, int idx1, int idx2);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200270int (*dll_lua_toboolean) (lua_State *L, int idx);
271const char *(*dll_lua_tolstring) (lua_State *L, int idx, size_t *len);
272void *(*dll_lua_touserdata) (lua_State *L, int idx);
273void (*dll_lua_pushnil) (lua_State *L);
274void (*dll_lua_pushnumber) (lua_State *L, lua_Number n);
275void (*dll_lua_pushinteger) (lua_State *L, lua_Integer n);
276void (*dll_lua_pushlstring) (lua_State *L, const char *s, size_t l);
277void (*dll_lua_pushstring) (lua_State *L, const char *s);
278const char *(*dll_lua_pushfstring) (lua_State *L, const char *fmt, ...);
279void (*dll_lua_pushcclosure) (lua_State *L, lua_CFunction fn, int n);
280void (*dll_lua_pushboolean) (lua_State *L, int b);
281void (*dll_lua_pushlightuserdata) (lua_State *L, void *p);
282void (*dll_lua_getfield) (lua_State *L, int idx, const char *k);
Bram Moolenaar17413672018-07-14 20:49:42 +0200283#if LUA_VERSION_NUM <= 502
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200284void (*dll_lua_rawget) (lua_State *L, int idx);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200285void (*dll_lua_rawgeti) (lua_State *L, int idx, int n);
Bram Moolenaar17413672018-07-14 20:49:42 +0200286#else
287int (*dll_lua_rawget) (lua_State *L, int idx);
288int (*dll_lua_rawgeti) (lua_State *L, int idx, lua_Integer n);
289#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200290void (*dll_lua_createtable) (lua_State *L, int narr, int nrec);
Bram Moolenaar830e3582018-08-21 14:23:35 +0200291#if LUA_VERSION_NUM >= 504
292void *(*dll_lua_newuserdatauv) (lua_State *L, size_t sz, int nuvalue);
293#else
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200294void *(*dll_lua_newuserdata) (lua_State *L, size_t sz);
Bram Moolenaar830e3582018-08-21 14:23:35 +0200295#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200296int (*dll_lua_getmetatable) (lua_State *L, int objindex);
297void (*dll_lua_setfield) (lua_State *L, int idx, const char *k);
298void (*dll_lua_rawset) (lua_State *L, int idx);
Bram Moolenaar17413672018-07-14 20:49:42 +0200299#if LUA_VERSION_NUM <= 502
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200300void (*dll_lua_rawseti) (lua_State *L, int idx, int n);
Bram Moolenaar17413672018-07-14 20:49:42 +0200301#else
302void (*dll_lua_rawseti) (lua_State *L, int idx, lua_Integer n);
303#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200304int (*dll_lua_setmetatable) (lua_State *L, int objindex);
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200305int (*dll_lua_next) (lua_State *L, int idx);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100306// libs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200307int (*dll_luaopen_base) (lua_State *L);
308int (*dll_luaopen_table) (lua_State *L);
309int (*dll_luaopen_string) (lua_State *L);
310int (*dll_luaopen_math) (lua_State *L);
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200311int (*dll_luaopen_io) (lua_State *L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200312int (*dll_luaopen_os) (lua_State *L);
313int (*dll_luaopen_package) (lua_State *L);
314int (*dll_luaopen_debug) (lua_State *L);
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200315void (*dll_luaL_openlibs) (lua_State *L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200316
317typedef void **luaV_function;
318typedef struct {
319 const char *name;
320 luaV_function func;
321} luaV_Reg;
322
323static const luaV_Reg luaV_dll[] = {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100324 // lauxlib
Bram Moolenaar1dced572012-04-05 16:54:08 +0200325#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200326 {"luaL_register", (luaV_function) &dll_luaL_register},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200327 {"luaL_prepbuffer", (luaV_function) &dll_luaL_prepbuffer},
328 {"luaL_openlib", (luaV_function) &dll_luaL_openlib},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200329 {"luaL_typerror", (luaV_function) &dll_luaL_typerror},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200330 {"luaL_loadfile", (luaV_function) &dll_luaL_loadfile},
331 {"luaL_loadbuffer", (luaV_function) &dll_luaL_loadbuffer},
332#else
333 {"luaL_prepbuffsize", (luaV_function) &dll_luaL_prepbuffsize},
334 {"luaL_setfuncs", (luaV_function) &dll_luaL_setfuncs},
335 {"luaL_loadfilex", (luaV_function) &dll_luaL_loadfilex},
336 {"luaL_loadbufferx", (luaV_function) &dll_luaL_loadbufferx},
337 {"luaL_argerror", (luaV_function) &dll_luaL_argerror},
338#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200339 {"luaL_checkany", (luaV_function) &dll_luaL_checkany},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200340 {"luaL_checklstring", (luaV_function) &dll_luaL_checklstring},
341 {"luaL_checkinteger", (luaV_function) &dll_luaL_checkinteger},
342 {"luaL_optinteger", (luaV_function) &dll_luaL_optinteger},
343 {"luaL_checktype", (luaV_function) &dll_luaL_checktype},
344 {"luaL_error", (luaV_function) &dll_luaL_error},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200345 {"luaL_newstate", (luaV_function) &dll_luaL_newstate},
346 {"luaL_buffinit", (luaV_function) &dll_luaL_buffinit},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200347 {"luaL_addlstring", (luaV_function) &dll_luaL_addlstring},
348 {"luaL_pushresult", (luaV_function) &dll_luaL_pushresult},
Bram Moolenaardf1643a2020-05-17 18:53:27 +0200349 {"luaL_loadstring", (luaV_function) &dll_luaL_loadstring},
Bram Moolenaar1e4c7d02020-06-25 20:56:42 +0200350 {"luaL_ref", (luaV_function) &dll_luaL_ref},
351 {"luaL_unref", (luaV_function) &dll_luaL_unref},
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100352 // lua
Bram Moolenaar1dced572012-04-05 16:54:08 +0200353#if LUA_VERSION_NUM <= 501
354 {"lua_tonumber", (luaV_function) &dll_lua_tonumber},
355 {"lua_tointeger", (luaV_function) &dll_lua_tointeger},
356 {"lua_call", (luaV_function) &dll_lua_call},
357 {"lua_pcall", (luaV_function) &dll_lua_pcall},
358#else
359 {"lua_tonumberx", (luaV_function) &dll_lua_tonumberx},
360 {"lua_tointegerx", (luaV_function) &dll_lua_tointegerx},
361 {"lua_callk", (luaV_function) &dll_lua_callk},
362 {"lua_pcallk", (luaV_function) &dll_lua_pcallk},
363 {"lua_getglobal", (luaV_function) &dll_lua_getglobal},
364 {"lua_setglobal", (luaV_function) &dll_lua_setglobal},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200365#endif
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200366#if LUA_VERSION_NUM <= 502
367 {"lua_replace", (luaV_function) &dll_lua_replace},
368 {"lua_remove", (luaV_function) &dll_lua_remove},
369#endif
370#if LUA_VERSION_NUM >= 503
371 {"lua_rotate", (luaV_function) &dll_lua_rotate},
372 {"lua_copy", (luaV_function) &dll_lua_copy},
373#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200374 {"lua_typename", (luaV_function) &dll_lua_typename},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200375 {"lua_close", (luaV_function) &dll_lua_close},
376 {"lua_gettop", (luaV_function) &dll_lua_gettop},
377 {"lua_settop", (luaV_function) &dll_lua_settop},
378 {"lua_pushvalue", (luaV_function) &dll_lua_pushvalue},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200379 {"lua_isnumber", (luaV_function) &dll_lua_isnumber},
380 {"lua_isstring", (luaV_function) &dll_lua_isstring},
381 {"lua_type", (luaV_function) &dll_lua_type},
382 {"lua_rawequal", (luaV_function) &dll_lua_rawequal},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200383 {"lua_toboolean", (luaV_function) &dll_lua_toboolean},
384 {"lua_tolstring", (luaV_function) &dll_lua_tolstring},
385 {"lua_touserdata", (luaV_function) &dll_lua_touserdata},
386 {"lua_pushnil", (luaV_function) &dll_lua_pushnil},
387 {"lua_pushnumber", (luaV_function) &dll_lua_pushnumber},
388 {"lua_pushinteger", (luaV_function) &dll_lua_pushinteger},
389 {"lua_pushlstring", (luaV_function) &dll_lua_pushlstring},
390 {"lua_pushstring", (luaV_function) &dll_lua_pushstring},
391 {"lua_pushfstring", (luaV_function) &dll_lua_pushfstring},
392 {"lua_pushcclosure", (luaV_function) &dll_lua_pushcclosure},
393 {"lua_pushboolean", (luaV_function) &dll_lua_pushboolean},
394 {"lua_pushlightuserdata", (luaV_function) &dll_lua_pushlightuserdata},
395 {"lua_getfield", (luaV_function) &dll_lua_getfield},
396 {"lua_rawget", (luaV_function) &dll_lua_rawget},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200397 {"lua_rawgeti", (luaV_function) &dll_lua_rawgeti},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200398 {"lua_createtable", (luaV_function) &dll_lua_createtable},
Bram Moolenaar830e3582018-08-21 14:23:35 +0200399#if LUA_VERSION_NUM >= 504
400 {"lua_newuserdatauv", (luaV_function) &dll_lua_newuserdatauv},
401#else
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200402 {"lua_newuserdata", (luaV_function) &dll_lua_newuserdata},
Bram Moolenaar830e3582018-08-21 14:23:35 +0200403#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200404 {"lua_getmetatable", (luaV_function) &dll_lua_getmetatable},
405 {"lua_setfield", (luaV_function) &dll_lua_setfield},
406 {"lua_rawset", (luaV_function) &dll_lua_rawset},
407 {"lua_rawseti", (luaV_function) &dll_lua_rawseti},
408 {"lua_setmetatable", (luaV_function) &dll_lua_setmetatable},
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200409 {"lua_next", (luaV_function) &dll_lua_next},
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100410 // libs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200411 {"luaopen_base", (luaV_function) &dll_luaopen_base},
412 {"luaopen_table", (luaV_function) &dll_luaopen_table},
413 {"luaopen_string", (luaV_function) &dll_luaopen_string},
414 {"luaopen_math", (luaV_function) &dll_luaopen_math},
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200415 {"luaopen_io", (luaV_function) &dll_luaopen_io},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200416 {"luaopen_os", (luaV_function) &dll_luaopen_os},
417 {"luaopen_package", (luaV_function) &dll_luaopen_package},
418 {"luaopen_debug", (luaV_function) &dll_luaopen_debug},
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200419 {"luaL_openlibs", (luaV_function) &dll_luaL_openlibs},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200420 {NULL, NULL}
421};
422
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200423static HANDLE hinstLua = NULL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200424
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200425 static int
426lua_link_init(char *libname, int verbose)
427{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200428 const luaV_Reg *reg;
429 if (hinstLua) return OK;
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200430 hinstLua = load_dll(libname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200431 if (!hinstLua)
432 {
433 if (verbose)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100434 semsg(_(e_loadlib), libname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200435 return FAIL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200436 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200437 for (reg = luaV_dll; reg->func; reg++)
438 {
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200439 if ((*reg->func = symbol_from_dll(hinstLua, reg->name)) == NULL)
440 {
441 close_dll(hinstLua);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200442 hinstLua = 0;
443 if (verbose)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100444 semsg(_(e_loadfunc), reg->name);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200445 return FAIL;
446 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200447 }
448 return OK;
449}
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100450#endif // DYNAMIC_LUA
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200451
Bram Moolenaard90b6c02016-08-28 18:10:45 +0200452#if defined(DYNAMIC_LUA) || defined(PROTO)
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200453 int
454lua_enabled(int verbose)
455{
Bram Moolenaar25e4fcd2016-01-09 14:57:47 +0100456 return lua_link_init((char *)p_luadll, verbose) == OK;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200457}
Bram Moolenaard90b6c02016-08-28 18:10:45 +0200458#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200459
Bram Moolenaar1dced572012-04-05 16:54:08 +0200460#if LUA_VERSION_NUM > 501
461 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100462luaL_typeerror(lua_State *L, int narg, const char *tname)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200463{
464 const char *msg = lua_pushfstring(L, "%s expected, got %s",
Bram Moolenaardb913952012-06-29 12:54:53 +0200465 tname, luaL_typename(L, narg));
Bram Moolenaar1dced572012-04-05 16:54:08 +0200466 return luaL_argerror(L, narg, msg);
467}
468#endif
469
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200470
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100471// ======= Internal =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200472
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200473 static void
474luaV_newmetatable(lua_State *L, const char *tname)
475{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200476 lua_newtable(L);
477 lua_pushlightuserdata(L, (void *) tname);
478 lua_pushvalue(L, -2);
479 lua_rawset(L, LUA_REGISTRYINDEX);
480}
481
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200482 static void *
483luaV_toudata(lua_State *L, int ud, const char *tname)
484{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200485 void *p = lua_touserdata(L, ud);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200486
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100487 if (p != NULL) // value is userdata?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200488 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100489 if (lua_getmetatable(L, ud)) // does it have a metatable?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200490 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100491 luaV_getfield(L, tname); // get metatable
492 if (lua_rawequal(L, -1, -2)) // MTs match?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200493 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100494 lua_pop(L, 2); // MTs
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200495 return p;
496 }
497 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200498 }
499 return NULL;
500}
501
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200502 static void *
Bram Moolenaar1dced572012-04-05 16:54:08 +0200503luaV_checkcache(lua_State *L, void *p)
504{
505 luaV_getudata(L, p);
506 if (lua_isnil(L, -1)) luaL_error(L, "invalid object");
507 lua_pop(L, 1);
508 return p;
509}
510
511#define luaV_unbox(L,luatyp,ud) (*((luatyp *) lua_touserdata((L),(ud))))
512
513#define luaV_checkvalid(L,luatyp,ud) \
514 luaV_checkcache((L), (void *) luaV_unbox((L),luatyp,(ud)))
515
516 static void *
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200517luaV_checkudata(lua_State *L, int ud, const char *tname)
518{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200519 void *p = luaV_toudata(L, ud, tname);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200520 if (p == NULL) luaL_typeerror(L, ud, tname);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200521 return p;
522}
523
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200524 static void
525luaV_pushtypval(lua_State *L, typval_T *tv)
526{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200527 if (tv == NULL)
528 {
529 lua_pushnil(L);
530 return;
531 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200532 switch (tv->v_type)
533 {
534 case VAR_STRING:
Bram Moolenaard04da7c2012-10-14 03:41:59 +0200535 lua_pushstring(L, tv->vval.v_string == NULL
536 ? "" : (char *)tv->vval.v_string);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200537 break;
538 case VAR_NUMBER:
539 lua_pushinteger(L, (int) tv->vval.v_number);
540 break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200541#ifdef FEAT_FLOAT
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200542 case VAR_FLOAT:
543 lua_pushnumber(L, (lua_Number) tv->vval.v_float);
544 break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200545#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +0200546 case VAR_LIST:
547 luaV_pushlist(L, tv->vval.v_list);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200548 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200549 case VAR_DICT:
550 luaV_pushdict(L, tv->vval.v_dict);
551 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +0100552 case VAR_BOOL:
Bram Moolenaar520e1e42016-01-23 19:46:28 +0100553 case VAR_SPECIAL:
554 if (tv->vval.v_number <= VVAL_TRUE)
555 lua_pushinteger(L, (int) tv->vval.v_number);
556 else
557 lua_pushnil(L);
558 break;
Bram Moolenaarca06da92018-07-01 15:12:05 +0200559 case VAR_FUNC:
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100560 luaV_pushfuncref(L, tv->vval.v_string);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200561 break;
Bram Moolenaarb7828692019-03-23 13:57:02 +0100562 case VAR_BLOB:
563 luaV_pushblob(L, tv->vval.v_blob);
564 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200565 default:
566 lua_pushnil(L);
567 }
568}
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200569
Bram Moolenaarca06da92018-07-01 15:12:05 +0200570/*
571 * Converts lua value at 'pos' to typval 'tv'.
572 * Returns OK or FAIL.
573 */
574 static int
575luaV_totypval(lua_State *L, int pos, typval_T *tv)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200576{
Bram Moolenaarca06da92018-07-01 15:12:05 +0200577 int status = OK;
578
Bram Moolenaara9a8e5f2020-07-02 21:17:57 +0200579 tv->v_lock = 0;
580
Bram Moolenaarca06da92018-07-01 15:12:05 +0200581 switch (lua_type(L, pos))
582 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200583 case LUA_TBOOLEAN:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +0100584 tv->v_type = VAR_BOOL;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200585 tv->vval.v_number = (varnumber_T) lua_toboolean(L, pos);
586 break;
Bram Moolenaar9067cd62019-01-01 00:41:54 +0100587 case LUA_TNIL:
588 tv->v_type = VAR_SPECIAL;
589 tv->vval.v_number = VVAL_NULL;
590 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200591 case LUA_TSTRING:
592 tv->v_type = VAR_STRING;
593 tv->vval.v_string = vim_strsave((char_u *) lua_tostring(L, pos));
594 break;
595 case LUA_TNUMBER:
596#ifdef FEAT_FLOAT
Bram Moolenaareb04f082020-05-17 14:32:35 +0200597 {
598 const lua_Number n = lua_tonumber(L, pos);
599
600 if (n > (lua_Number)INT64_MAX || n < (lua_Number)INT64_MIN
601 || ((lua_Number)((varnumber_T)n)) != n)
602 {
603 tv->v_type = VAR_FLOAT;
604 tv->vval.v_float = (float_T)n;
605 }
606 else
607 {
608 tv->v_type = VAR_NUMBER;
609 tv->vval.v_number = (varnumber_T)n;
610 }
611 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200612#else
613 tv->v_type = VAR_NUMBER;
614 tv->vval.v_number = (varnumber_T) lua_tointeger(L, pos);
615#endif
616 break;
Bram Moolenaar801ab062020-06-25 19:27:56 +0200617 case LUA_TFUNCTION:
618 {
619 char_u *name;
620 lua_pushvalue(L, pos);
621 luaV_CFuncState *state = ALLOC_CLEAR_ONE(luaV_CFuncState);
622 state->lua_funcref = luaL_ref(L, LUA_REGISTRYINDEX);
623 state->L = L;
624 state->lua_tableref = LUA_NOREF;
625 name = register_cfunc(&luaV_call_lua_func,
626 &luaV_call_lua_func_free, state);
627 tv->v_type = VAR_FUNC;
628 tv->vval.v_string = vim_strsave(name);
629 break;
630 }
631 case LUA_TTABLE:
632 {
633 lua_pushvalue(L, pos);
634 int lua_tableref = luaL_ref(L, LUA_REGISTRYINDEX);
635 if (lua_getmetatable(L, pos)) {
636 lua_getfield(L, -1, LUA___CALL);
637 if (lua_isfunction(L, -1)) {
638 char_u *name;
639 int lua_funcref = luaL_ref(L, LUA_REGISTRYINDEX);
640 luaV_CFuncState *state = ALLOC_CLEAR_ONE(luaV_CFuncState);
641 state->lua_funcref = lua_funcref;
642 state->L = L;
643 state->lua_tableref = lua_tableref;
644 name = register_cfunc(&luaV_call_lua_func,
645 &luaV_call_lua_func_free, state);
646 tv->v_type = VAR_FUNC;
647 tv->vval.v_string = vim_strsave(name);
648 break;
649 }
650 }
651 tv->v_type = VAR_NUMBER;
652 tv->vval.v_number = 0;
653 status = FAIL;
654 break;
655 }
Bram Moolenaarca06da92018-07-01 15:12:05 +0200656 case LUA_TUSERDATA:
657 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200658 void *p = lua_touserdata(L, pos);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200659
Bram Moolenaarb7828692019-03-23 13:57:02 +0100660 if (lua_getmetatable(L, pos)) // has metatable?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200661 {
Bram Moolenaarb7828692019-03-23 13:57:02 +0100662 // check list
Bram Moolenaar1dced572012-04-05 16:54:08 +0200663 luaV_getfield(L, LUAVIM_LIST);
664 if (lua_rawequal(L, -1, -2))
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200665 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200666 tv->v_type = VAR_LIST;
667 tv->vval.v_list = *((luaV_List *) p);
668 ++tv->vval.v_list->lv_refcount;
Bram Moolenaarb7828692019-03-23 13:57:02 +0100669 lua_pop(L, 2); // MTs
Bram Moolenaarca06da92018-07-01 15:12:05 +0200670 break;
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200671 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100672 // check dict
Bram Moolenaar1dced572012-04-05 16:54:08 +0200673 luaV_getfield(L, LUAVIM_DICT);
674 if (lua_rawequal(L, -1, -3))
675 {
676 tv->v_type = VAR_DICT;
677 tv->vval.v_dict = *((luaV_Dict *) p);
678 ++tv->vval.v_dict->dv_refcount;
Bram Moolenaarb7828692019-03-23 13:57:02 +0100679 lua_pop(L, 3); // MTs
Bram Moolenaarca06da92018-07-01 15:12:05 +0200680 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200681 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100682 // check blob
683 luaV_getfield(L, LUAVIM_BLOB);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200684 if (lua_rawequal(L, -1, -4))
685 {
Bram Moolenaarb7828692019-03-23 13:57:02 +0100686 tv->v_type = VAR_BLOB;
687 tv->vval.v_blob = *((luaV_Blob *) p);
688 ++tv->vval.v_blob->bv_refcount;
689 lua_pop(L, 4); // MTs
690 break;
691 }
692 // check funcref
693 luaV_getfield(L, LUAVIM_FUNCREF);
694 if (lua_rawequal(L, -1, -5))
695 {
Bram Moolenaarca06da92018-07-01 15:12:05 +0200696 luaV_Funcref *f = (luaV_Funcref *) p;
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100697 func_ref(f->name);
698 tv->v_type = VAR_FUNC;
699 tv->vval.v_string = vim_strsave(f->name);
Bram Moolenaarb7828692019-03-23 13:57:02 +0100700 lua_pop(L, 5); // MTs
Bram Moolenaarca06da92018-07-01 15:12:05 +0200701 break;
702 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100703 lua_pop(L, 4); // MTs
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200704 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200705 }
Bram Moolenaarb7828692019-03-23 13:57:02 +0100706 // FALLTHROUGH
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200707 default:
Bram Moolenaar1dced572012-04-05 16:54:08 +0200708 tv->v_type = VAR_NUMBER;
709 tv->vval.v_number = 0;
Bram Moolenaarca06da92018-07-01 15:12:05 +0200710 status = FAIL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200711 }
Bram Moolenaarca06da92018-07-01 15:12:05 +0200712 return status;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200713}
714
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100715/*
716 * similar to luaL_addlstring, but replaces \0 with \n if toline and
717 * \n with \0 otherwise
718 */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200719 static void
720luaV_addlstring(luaL_Buffer *b, const char *s, size_t l, int toline)
721{
722 while (l--)
723 {
724 if (*s == '\0' && toline)
725 luaL_addchar(b, '\n');
726 else if (*s == '\n' && !toline)
727 luaL_addchar(b, '\0');
728 else
729 luaL_addchar(b, *s);
730 s++;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200731 }
732}
733
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200734 static void
735luaV_pushline(lua_State *L, buf_T *buf, linenr_T n)
736{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200737 const char *s = (const char *) ml_get_buf(buf, n, FALSE);
738 luaL_Buffer b;
739 luaL_buffinit(L, &b);
740 luaV_addlstring(&b, s, strlen(s), 0);
741 luaL_pushresult(&b);
742}
743
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200744 static char_u *
745luaV_toline(lua_State *L, int pos)
746{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200747 size_t l;
748 const char *s = lua_tolstring(L, pos, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200749
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200750 luaL_Buffer b;
751 luaL_buffinit(L, &b);
752 luaV_addlstring(&b, s, l, 1);
753 luaL_pushresult(&b);
754 return (char_u *) lua_tostring(L, -1);
755}
756
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100757/*
758 * pops a string s from the top of the stack and calls mf(t) for pieces t of
759 * s separated by newlines
760 */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200761 static void
762luaV_msgfunc(lua_State *L, msgfunc_T mf)
763{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200764 luaL_Buffer b;
765 size_t l;
766 const char *p, *s = lua_tolstring(L, -1, &l);
767 luaL_buffinit(L, &b);
768 luaV_addlstring(&b, s, l, 0);
769 luaL_pushresult(&b);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100770 // break string
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200771 p = s = lua_tolstring(L, -1, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200772 while (l--)
773 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100774 if (*p++ == '\0') // break?
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200775 {
776 mf((char_u *) s);
777 s = p;
778 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200779 }
780 mf((char_u *) s);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100781 lua_pop(L, 2); // original and modified strings
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200782}
783
Bram Moolenaar1dced572012-04-05 16:54:08 +0200784#define luaV_newtype(typ,tname,luatyp,luatname) \
785 static luatyp * \
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100786 luaV_new##tname(lua_State *L, typ *obj) \
Bram Moolenaar1dced572012-04-05 16:54:08 +0200787 { \
788 luatyp *o = (luatyp *) lua_newuserdata(L, sizeof(luatyp)); \
789 *o = obj; \
790 luaV_setudata(L, obj); /* cache[obj] = udata */ \
791 luaV_getfield(L, luatname); \
792 lua_setmetatable(L, -2); \
793 return o; \
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200794 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200795
796#define luaV_pushtype(typ,tname,luatyp) \
797 static luatyp * \
Bram Moolenaarca06da92018-07-01 15:12:05 +0200798 luaV_push##tname(lua_State *L, typ *obj) \
Bram Moolenaar1dced572012-04-05 16:54:08 +0200799 { \
800 luatyp *o = NULL; \
801 if (obj == NULL) \
802 lua_pushnil(L); \
803 else { \
804 luaV_getudata(L, obj); \
805 if (lua_isnil(L, -1)) /* not interned? */ \
806 { \
807 lua_pop(L, 1); \
808 o = luaV_new##tname(L, obj); \
809 } \
810 else \
811 o = (luatyp *) lua_touserdata(L, -1); \
812 } \
813 return o; \
814 }
815
816#define luaV_type_tostring(tname,luatname) \
817 static int \
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100818 luaV_##tname##_tostring(lua_State *L) \
Bram Moolenaar1dced572012-04-05 16:54:08 +0200819 { \
820 lua_pushfstring(L, "%s: %p", luatname, lua_touserdata(L, 1)); \
821 return 1; \
822 }
823
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100824// ======= List type =======
Bram Moolenaar1dced572012-04-05 16:54:08 +0200825
826 static luaV_List *
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100827luaV_newlist(lua_State *L, list_T *lis)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200828{
829 luaV_List *l = (luaV_List *) lua_newuserdata(L, sizeof(luaV_List));
830 *l = lis;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100831 lis->lv_refcount++; // reference in Lua
832 luaV_setudata(L, lis); // cache[lis] = udata
Bram Moolenaar1dced572012-04-05 16:54:08 +0200833 luaV_getfield(L, LUAVIM_LIST);
834 lua_setmetatable(L, -2);
835 return l;
836}
837
838luaV_pushtype(list_T, list, luaV_List)
839luaV_type_tostring(list, LUAVIM_LIST)
840
841 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100842luaV_list_len(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200843{
844 list_T *l = luaV_unbox(L, luaV_List, 1);
Bram Moolenaarb7828692019-03-23 13:57:02 +0100845 lua_pushinteger(L, (int) list_len(l));
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200846 return 1;
847}
848
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200849 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100850luaV_list_iter(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200851{
852 listitem_T *li = (listitem_T *) lua_touserdata(L, lua_upvalueindex(2));
853 if (li == NULL) return 0;
854 luaV_pushtypval(L, &li->li_tv);
855 lua_pushlightuserdata(L, (void *) li->li_next);
856 lua_replace(L, lua_upvalueindex(2));
857 return 1;
858}
859
860 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100861luaV_list_call(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200862{
863 list_T *l = luaV_unbox(L, luaV_List, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100864 lua_pushvalue(L, lua_upvalueindex(1)); // pass cache table along
Bram Moolenaar1dced572012-04-05 16:54:08 +0200865 lua_pushlightuserdata(L, (void *) l->lv_first);
866 lua_pushcclosure(L, luaV_list_iter, 2);
867 return 1;
868}
869
870 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100871luaV_list_index(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200872{
873 list_T *l = luaV_unbox(L, luaV_List, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100874 if (lua_isnumber(L, 2)) // list item?
Bram Moolenaar1dced572012-04-05 16:54:08 +0200875 {
Bram Moolenaarbd846172020-06-27 12:32:57 +0200876 long n = (long) luaL_checkinteger(L, 2);
877 listitem_T *li;
878
879 // Lua array index starts with 1 while Vim uses 0, subtract 1 to
880 // normalize.
881 n -= 1;
882 li = list_find(l, n);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200883 if (li == NULL)
884 lua_pushnil(L);
885 else
886 luaV_pushtypval(L, &li->li_tv);
887 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +0100888 else if (lua_isstring(L, 2)) // method?
Bram Moolenaar1dced572012-04-05 16:54:08 +0200889 {
890 const char *s = lua_tostring(L, 2);
891 if (strncmp(s, "add", 3) == 0
Bram Moolenaarb3766472013-04-15 13:49:21 +0200892 || strncmp(s, "insert", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200893 {
894 lua_getmetatable(L, 1);
895 lua_getfield(L, -1, s);
896 }
897 else
898 lua_pushnil(L);
899 }
900 else
901 lua_pushnil(L);
902 return 1;
903}
904
905 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100906luaV_list_newindex(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200907{
908 list_T *l = luaV_unbox(L, luaV_List, 1);
909 long n = (long) luaL_checkinteger(L, 2);
910 listitem_T *li;
Bram Moolenaarbd846172020-06-27 12:32:57 +0200911
912 // Lua array index starts with 1 while Vim uses 0, subtract 1 to normalize.
913 n -= 1;
914
Bram Moolenaar1dced572012-04-05 16:54:08 +0200915 if (l->lv_lock)
916 luaL_error(L, "list is locked");
917 li = list_find(l, n);
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200918 if (li == NULL)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200919 {
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200920 if (!lua_isnil(L, 3))
921 {
922 typval_T v;
923 luaV_checktypval(L, 3, &v, "inserting list item");
924 if (list_insert_tv(l, &v, li) == FAIL)
925 luaL_error(L, "failed to add item to list");
926 clear_tv(&v);
927 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200928 }
929 else
930 {
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200931 if (lua_isnil(L, 3)) // remove?
932 {
933 vimlist_remove(l, li, li);
934 listitem_free(l, li);
935 }
936 else
937 {
938 typval_T v;
939 luaV_checktypval(L, 3, &v, "setting list item");
940 clear_tv(&li->li_tv);
Bram Moolenaare49b8e82020-07-01 13:52:55 +0200941 li->li_tv = v;
Bram Moolenaara1f9f862020-06-28 22:41:26 +0200942 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200943 }
944 return 0;
945}
946
947 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100948luaV_list_add(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200949{
950 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
951 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200952 typval_T v;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200953 if (l->lv_lock)
954 luaL_error(L, "list is locked");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200955 lua_settop(L, 2);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200956 luaV_checktypval(L, 2, &v, "adding list item");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200957 if (list_append_tv(l, &v) == FAIL)
Bram Moolenaarca06da92018-07-01 15:12:05 +0200958 luaL_error(L, "failed to add item to list");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200959 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200960 lua_settop(L, 1);
961 return 1;
962}
963
964 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +0100965luaV_list_insert(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200966{
967 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
968 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaar46538ee2015-02-17 16:28:55 +0100969 long pos = (long) luaL_optinteger(L, 3, 0);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200970 listitem_T *li = NULL;
971 typval_T v;
972 if (l->lv_lock)
973 luaL_error(L, "list is locked");
974 if (pos < l->lv_len)
975 {
976 li = list_find(l, pos);
977 if (li == NULL)
978 luaL_error(L, "invalid position");
979 }
980 lua_settop(L, 2);
Bram Moolenaarca06da92018-07-01 15:12:05 +0200981 luaV_checktypval(L, 2, &v, "inserting list item");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200982 if (list_insert_tv(l, &v, li) == FAIL)
Bram Moolenaarca06da92018-07-01 15:12:05 +0200983 luaL_error(L, "failed to add item to list");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200984 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200985 lua_settop(L, 1);
986 return 1;
987}
988
989static const luaL_Reg luaV_List_mt[] = {
990 {"__tostring", luaV_list_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200991 {"__len", luaV_list_len},
992 {"__call", luaV_list_call},
993 {"__index", luaV_list_index},
994 {"__newindex", luaV_list_newindex},
995 {"add", luaV_list_add},
996 {"insert", luaV_list_insert},
997 {NULL, NULL}
998};
999
1000
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001001// ======= Dict type =======
Bram Moolenaar1dced572012-04-05 16:54:08 +02001002
1003 static luaV_Dict *
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001004luaV_newdict(lua_State *L, dict_T *dic)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001005{
1006 luaV_Dict *d = (luaV_Dict *) lua_newuserdata(L, sizeof(luaV_Dict));
1007 *d = dic;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001008 dic->dv_refcount++; // reference in Lua
1009 luaV_setudata(L, dic); // cache[dic] = udata
Bram Moolenaar1dced572012-04-05 16:54:08 +02001010 luaV_getfield(L, LUAVIM_DICT);
1011 lua_setmetatable(L, -2);
1012 return d;
1013}
1014
1015luaV_pushtype(dict_T, dict, luaV_Dict)
1016luaV_type_tostring(dict, LUAVIM_DICT)
1017
1018 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001019luaV_dict_len(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001020{
1021 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
Bram Moolenaarb7828692019-03-23 13:57:02 +01001022 lua_pushinteger(L, (int) dict_len(d));
Bram Moolenaar1dced572012-04-05 16:54:08 +02001023 return 1;
1024}
1025
1026 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001027luaV_dict_iter(lua_State *L UNUSED)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001028{
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001029#ifdef FEAT_EVAL
Bram Moolenaar1dced572012-04-05 16:54:08 +02001030 hashitem_T *hi = (hashitem_T *) lua_touserdata(L, lua_upvalueindex(2));
1031 int n = lua_tointeger(L, lua_upvalueindex(3));
1032 dictitem_T *di;
1033 if (n <= 0) return 0;
1034 while (HASHITEM_EMPTY(hi)) hi++;
1035 di = dict_lookup(hi);
1036 lua_pushstring(L, (char *) hi->hi_key);
1037 luaV_pushtypval(L, &di->di_tv);
1038 lua_pushlightuserdata(L, (void *) (hi + 1));
1039 lua_replace(L, lua_upvalueindex(2));
1040 lua_pushinteger(L, n - 1);
1041 lua_replace(L, lua_upvalueindex(3));
1042 return 2;
Bram Moolenaarfeeaa682013-02-14 22:19:51 +01001043#else
1044 return 0;
1045#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001046}
1047
1048 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001049luaV_dict_call(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001050{
1051 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
1052 hashtab_T *ht = &d->dv_hashtab;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001053 lua_pushvalue(L, lua_upvalueindex(1)); // pass cache table along
Bram Moolenaar1dced572012-04-05 16:54:08 +02001054 lua_pushlightuserdata(L, (void *) ht->ht_array);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001055 lua_pushinteger(L, ht->ht_used); // # remaining items
Bram Moolenaar1dced572012-04-05 16:54:08 +02001056 lua_pushcclosure(L, luaV_dict_iter, 3);
1057 return 1;
1058}
1059
1060 static int
Bram Moolenaarca06da92018-07-01 15:12:05 +02001061luaV_dict_index(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001062{
1063 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
1064 char_u *key = (char_u *) luaL_checkstring(L, 2);
1065 dictitem_T *di = dict_find(d, key, -1);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001066
Bram Moolenaar1dced572012-04-05 16:54:08 +02001067 if (di == NULL)
1068 lua_pushnil(L);
1069 else
Bram Moolenaarca06da92018-07-01 15:12:05 +02001070 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001071 luaV_pushtypval(L, &di->di_tv);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001072 if (di->di_tv.v_type == VAR_FUNC) // funcref?
Bram Moolenaarca06da92018-07-01 15:12:05 +02001073 {
1074 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, -1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001075 f->self = d; // keep "self" reference
Bram Moolenaarca06da92018-07-01 15:12:05 +02001076 d->dv_refcount++;
1077 }
1078 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001079 return 1;
1080}
1081
1082 static int
Bram Moolenaarca06da92018-07-01 15:12:05 +02001083luaV_dict_newindex(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001084{
1085 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
1086 char_u *key = (char_u *) luaL_checkstring(L, 2);
1087 dictitem_T *di;
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001088 typval_T tv;
Bram Moolenaar713bf9e92019-03-16 16:38:41 +01001089
Bram Moolenaar1dced572012-04-05 16:54:08 +02001090 if (d->dv_lock)
1091 luaL_error(L, "dict is locked");
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001092 if (key == NULL)
1093 return 0;
1094 if (*key == NUL)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001095 luaL_error(L, "empty key");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001096 if (!lua_isnil(L, 3)) // read value?
Bram Moolenaar17413672018-07-14 20:49:42 +02001097 {
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001098 luaV_checktypval(L, 3, &tv, "setting dict item");
1099 if (d->dv_scope == VAR_DEF_SCOPE && tv.v_type == VAR_FUNC)
1100 {
1101 clear_tv(&tv);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001102 luaL_error(L, "cannot assign funcref to builtin scope");
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001103 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001104 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001105 di = dict_find(d, key, -1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001106 if (di == NULL) // non-existing key?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001107 {
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001108 if (lua_isnil(L, 3))
1109 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001110 di = dictitem_alloc(key);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001111 if (di == NULL)
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001112 {
1113 clear_tv(&tv);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001114 return 0;
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001115 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001116 if (dict_add(d, di) == FAIL)
1117 {
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001118 vim_free(di);
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001119 clear_tv(&tv);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001120 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001121 }
1122 }
1123 else
1124 clear_tv(&di->di_tv);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001125 if (lua_isnil(L, 3)) // remove?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001126 {
1127 hashitem_T *hi = hash_find(&d->dv_hashtab, di->di_key);
1128 hash_remove(&d->dv_hashtab, hi);
1129 dictitem_free(di);
1130 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001131 else
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001132 di->di_tv = tv;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001133 return 0;
1134}
1135
1136static const luaL_Reg luaV_Dict_mt[] = {
1137 {"__tostring", luaV_dict_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +02001138 {"__len", luaV_dict_len},
1139 {"__call", luaV_dict_call},
1140 {"__index", luaV_dict_index},
1141 {"__newindex", luaV_dict_newindex},
1142 {NULL, NULL}
1143};
1144
1145
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001146// ======= Blob type =======
Bram Moolenaarb7828692019-03-23 13:57:02 +01001147
1148 static luaV_Blob *
1149luaV_newblob(lua_State *L, blob_T *blo)
1150{
1151 luaV_Blob *b = (luaV_Blob *) lua_newuserdata(L, sizeof(luaV_Blob));
1152 *b = blo;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001153 blo->bv_refcount++; // reference in Lua
1154 luaV_setudata(L, blo); // cache[blo] = udata
Bram Moolenaarb7828692019-03-23 13:57:02 +01001155 luaV_getfield(L, LUAVIM_BLOB);
1156 lua_setmetatable(L, -2);
1157 return b;
1158}
1159
1160luaV_pushtype(blob_T, blob, luaV_Blob)
1161luaV_type_tostring(blob, LUAVIM_BLOB)
1162
1163 static int
1164luaV_blob_gc(lua_State *L)
1165{
1166 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1167 blob_unref(b);
1168 return 0;
1169}
1170
1171 static int
1172luaV_blob_len(lua_State *L)
1173{
1174 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1175 lua_pushinteger(L, (int) blob_len(b));
1176 return 1;
1177}
1178
1179 static int
1180luaV_blob_index(lua_State *L)
1181{
1182 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1183 if (lua_isnumber(L, 2))
1184 {
1185 int idx = luaL_checkinteger(L, 2);
1186 if (idx < blob_len(b))
1187 lua_pushnumber(L, (lua_Number) blob_get(b, idx));
1188 else
1189 lua_pushnil(L);
1190 }
1191 else if (lua_isstring(L, 2))
1192 {
1193 const char *s = lua_tostring(L, 2);
1194 if (strncmp(s, "add", 3) == 0)
1195 {
1196 lua_getmetatable(L, 1);
1197 lua_getfield(L, -1, s);
1198 }
1199 else
1200 lua_pushnil(L);
1201 }
1202 else
1203 lua_pushnil(L);
1204 return 1;
1205}
1206
1207 static int
1208luaV_blob_newindex(lua_State *L)
1209{
1210 blob_T *b = luaV_unbox(L, luaV_Blob, 1);
1211 if (b->bv_lock)
1212 luaL_error(L, "blob is locked");
1213 if (lua_isnumber(L, 2))
1214 {
1215 long len = blob_len(b);
1216 int idx = luaL_checkinteger(L, 2);
1217 int val = luaL_checkinteger(L, 3);
1218 if (idx < len || (idx == len && ga_grow(&b->bv_ga, 1) == OK))
1219 {
1220 blob_set(b, idx, (char_u) val);
1221 if (idx == len)
1222 ++b->bv_ga.ga_len;
1223 }
1224 else
1225 luaL_error(L, "index out of range");
1226 }
1227 return 0;
1228}
1229
1230 static int
1231luaV_blob_add(lua_State *L)
1232{
1233 luaV_Blob *blo = luaV_checkudata(L, 1, LUAVIM_BLOB);
1234 blob_T *b = (blob_T *) luaV_checkcache(L, (void *) *blo);
1235 if (b->bv_lock)
1236 luaL_error(L, "blob is locked");
1237 lua_settop(L, 2);
1238 if (!lua_isstring(L, 2))
1239 luaL_error(L, "string expected, got %s", luaL_typename(L, 2));
1240 else
1241 {
1242 size_t i, l = 0;
1243 const char *s = lua_tolstring(L, 2, &l);
1244
Bram Moolenaar5f1d3ae2020-02-11 22:37:35 +01001245 if (ga_grow(&b->bv_ga, (int)l) == OK)
Bram Moolenaar6fb5c972019-03-26 21:44:20 +01001246 for (i = 0; i < l; ++i)
1247 ga_append(&b->bv_ga, s[i]);
Bram Moolenaarb7828692019-03-23 13:57:02 +01001248 }
1249 lua_settop(L, 1);
1250 return 1;
1251}
1252
1253static const luaL_Reg luaV_Blob_mt[] = {
1254 {"__tostring", luaV_blob_tostring},
1255 {"__gc", luaV_blob_gc},
1256 {"__len", luaV_blob_len},
1257 {"__index", luaV_blob_index},
1258 {"__newindex", luaV_blob_newindex},
1259 {"add", luaV_blob_add},
1260 {NULL, NULL}
1261};
1262
1263
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001264// ======= Funcref type =======
Bram Moolenaarca06da92018-07-01 15:12:05 +02001265
1266 static luaV_Funcref *
1267luaV_newfuncref(lua_State *L, char_u *name)
1268{
1269 luaV_Funcref *f = (luaV_Funcref *)lua_newuserdata(L, sizeof(luaV_Funcref));
1270
1271 if (name != NULL)
1272 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001273 func_ref(name);
1274 f->name = vim_strsave(name);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001275 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001276 f->self = NULL;
1277 luaV_getfield(L, LUAVIM_FUNCREF);
1278 lua_setmetatable(L, -2);
1279 return f;
1280}
1281
1282 static luaV_Funcref *
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001283luaV_pushfuncref(lua_State *L, char_u *name)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001284{
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001285 return luaV_newfuncref(L, name);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001286}
1287
1288
1289luaV_type_tostring(funcref, LUAVIM_FUNCREF)
1290
1291 static int
1292luaV_funcref_gc(lua_State *L)
1293{
1294 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, 1);
1295
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001296 func_unref(f->name);
1297 vim_free(f->name);
1298 // NOTE: Don't call "dict_unref(f->self)", because the dict of "f->self"
1299 // will be (or has been already) freed by Vim's garbage collection.
Bram Moolenaarca06da92018-07-01 15:12:05 +02001300 return 0;
1301}
1302
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001303// equivalent to string(funcref)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001304 static int
1305luaV_funcref_len(lua_State *L)
1306{
1307 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, 1);
1308
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001309 lua_pushstring(L, (const char *) f->name);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001310 return 1;
1311}
1312
1313 static int
1314luaV_funcref_call(lua_State *L)
1315{
1316 luaV_Funcref *f = (luaV_Funcref *) lua_touserdata(L, 1);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001317 int i, n = lua_gettop(L) - 1; // #args
1318 int status = FAIL;
1319 typval_T args;
1320 typval_T rettv;
Bram Moolenaarca06da92018-07-01 15:12:05 +02001321
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001322 args.v_type = VAR_LIST;
1323 args.vval.v_list = list_alloc();
1324 rettv.v_type = VAR_UNKNOWN; // as in clear_tv
1325 if (args.vval.v_list != NULL)
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001326 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001327 typval_T v;
1328
Bram Moolenaar17413672018-07-14 20:49:42 +02001329 for (i = 0; i < n; i++)
1330 {
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001331 luaV_checktypval(L, i + 2, &v, "calling funcref");
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001332 list_append_tv(args.vval.v_list, &v);
Bram Moolenaar713bf9e92019-03-16 16:38:41 +01001333 clear_tv(&v);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001334 }
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001335 status = func_call(f->name, &args, NULL, f->self, &rettv);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001336 if (status == OK)
1337 luaV_pushtypval(L, &rettv);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001338 clear_tv(&args);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001339 clear_tv(&rettv);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001340 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001341 if (status != OK)
1342 luaL_error(L, "cannot call funcref");
1343 return 1;
1344}
1345
1346static const luaL_Reg luaV_Funcref_mt[] = {
1347 {"__tostring", luaV_funcref_tostring},
1348 {"__gc", luaV_funcref_gc},
1349 {"__len", luaV_funcref_len},
1350 {"__call", luaV_funcref_call},
1351 {NULL, NULL}
1352};
1353
1354
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001355// ======= Buffer type =======
Bram Moolenaar1dced572012-04-05 16:54:08 +02001356
1357luaV_newtype(buf_T, buffer, luaV_Buffer, LUAVIM_BUFFER)
1358luaV_pushtype(buf_T, buffer, luaV_Buffer)
1359luaV_type_tostring(buffer, LUAVIM_BUFFER)
1360
1361 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001362luaV_buffer_len(lua_State *L)
1363{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001364 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
1365 lua_pushinteger(L, b->b_ml.ml_line_count);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001366 return 1;
1367}
1368
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001369 static int
1370luaV_buffer_call(lua_State *L)
1371{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001372 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001373 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001374 set_curbuf(b, DOBUF_SPLIT);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001375 return 1;
1376}
1377
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001378 static int
1379luaV_buffer_index(lua_State *L)
1380{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001381 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001382 linenr_T n = (linenr_T) lua_tointeger(L, 2);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001383 if (n > 0 && n <= b->b_ml.ml_line_count)
1384 luaV_pushline(L, b, n);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001385 else if (lua_isstring(L, 2))
1386 {
1387 const char *s = lua_tostring(L, 2);
1388 if (strncmp(s, "name", 4) == 0)
Bram Moolenaarfe08df42018-07-07 23:07:41 +02001389 lua_pushstring(L, (b->b_sfname == NULL)
1390 ? "" : (char *) b->b_sfname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001391 else if (strncmp(s, "fname", 5) == 0)
Bram Moolenaarfe08df42018-07-07 23:07:41 +02001392 lua_pushstring(L, (b->b_ffname == NULL)
1393 ? "" : (char *) b->b_ffname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001394 else if (strncmp(s, "number", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001395 lua_pushinteger(L, b->b_fnum);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001396 // methods
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001397 else if (strncmp(s, "insert", 6) == 0
1398 || strncmp(s, "next", 4) == 0
1399 || strncmp(s, "previous", 8) == 0
1400 || strncmp(s, "isvalid", 7) == 0)
1401 {
1402 lua_getmetatable(L, 1);
1403 lua_getfield(L, -1, s);
1404 }
1405 else
1406 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001407 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001408 else
1409 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001410 return 1;
1411}
1412
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001413 static int
1414luaV_buffer_newindex(lua_State *L)
1415{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001416 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001417 linenr_T n = (linenr_T) luaL_checkinteger(L, 2);
1418#ifdef HAVE_SANDBOX
1419 luaV_checksandbox(L);
1420#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001421 if (n < 1 || n > b->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001422 luaL_error(L, "invalid line number");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001423 if (lua_isnil(L, 3)) // delete line
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001424 {
1425 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001426 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001427 if (u_savedel(n, 1L) == FAIL)
1428 {
1429 curbuf = buf;
1430 luaL_error(L, "cannot save undo information");
1431 }
Bram Moolenaarca70c072020-05-30 20:30:46 +02001432 else if (ml_delete(n) == FAIL)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001433 {
1434 curbuf = buf;
1435 luaL_error(L, "cannot delete line");
1436 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001437 else
1438 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001439 deleted_lines_mark(n, 1L);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001440 if (b == curwin->w_buffer) // fix cursor in current window?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001441 {
1442 if (curwin->w_cursor.lnum >= n)
1443 {
1444 if (curwin->w_cursor.lnum > n)
1445 {
1446 curwin->w_cursor.lnum -= 1;
1447 check_cursor_col();
1448 }
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001449 else
1450 check_cursor();
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001451 changed_cline_bef_curs();
1452 }
1453 invalidate_botline();
1454 }
1455 }
1456 curbuf = buf;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001457 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001458 else if (lua_isstring(L, 3)) // update line
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001459 {
1460 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001461 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001462 if (u_savesub(n) == FAIL)
1463 {
1464 curbuf = buf;
1465 luaL_error(L, "cannot save undo information");
1466 }
1467 else if (ml_replace(n, luaV_toline(L, 3), TRUE) == FAIL)
1468 {
1469 curbuf = buf;
1470 luaL_error(L, "cannot replace line");
1471 }
1472 else changed_bytes(n, 0);
1473 curbuf = buf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001474 if (b == curwin->w_buffer)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001475 check_cursor_col();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001476 }
1477 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001478 luaL_error(L, "wrong argument to change line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001479 return 0;
1480}
1481
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001482 static int
1483luaV_buffer_insert(lua_State *L)
1484{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001485 luaV_Buffer *lb = luaV_checkudata(L, 1, LUAVIM_BUFFER);
1486 buf_T *b = (buf_T *) luaV_checkcache(L, (void *) *lb);
1487 linenr_T last = b->b_ml.ml_line_count;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001488 linenr_T n = (linenr_T) luaL_optinteger(L, 3, last);
1489 buf_T *buf;
1490 luaL_checktype(L, 2, LUA_TSTRING);
1491#ifdef HAVE_SANDBOX
1492 luaV_checksandbox(L);
1493#endif
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001494 // fix insertion line
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001495 if (n < 0) n = 0;
1496 if (n > last) n = last;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001497 // insert
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001498 buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001499 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001500 if (u_save(n, n + 1) == FAIL)
1501 {
1502 curbuf = buf;
1503 luaL_error(L, "cannot save undo information");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001504 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001505 else if (ml_append(n, luaV_toline(L, 2), 0, FALSE) == FAIL)
1506 {
1507 curbuf = buf;
1508 luaL_error(L, "cannot insert line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001509 }
1510 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001511 appended_lines_mark(n, 1L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001512 curbuf = buf;
1513 update_screen(VALID);
1514 return 0;
1515}
1516
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001517 static int
1518luaV_buffer_next(lua_State *L)
1519{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001520 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001521 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1522 luaV_pushbuffer(L, buf->b_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001523 return 1;
1524}
1525
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001526 static int
1527luaV_buffer_previous(lua_State *L)
1528{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001529 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001530 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1531 luaV_pushbuffer(L, buf->b_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001532 return 1;
1533}
1534
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001535 static int
1536luaV_buffer_isvalid(lua_State *L)
1537{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001538 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001539 luaV_getudata(L, *b);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001540 lua_pushboolean(L, !lua_isnil(L, -1));
1541 return 1;
1542}
1543
1544static const luaL_Reg luaV_Buffer_mt[] = {
1545 {"__tostring", luaV_buffer_tostring},
1546 {"__len", luaV_buffer_len},
1547 {"__call", luaV_buffer_call},
1548 {"__index", luaV_buffer_index},
1549 {"__newindex", luaV_buffer_newindex},
1550 {"insert", luaV_buffer_insert},
1551 {"next", luaV_buffer_next},
1552 {"previous", luaV_buffer_previous},
1553 {"isvalid", luaV_buffer_isvalid},
1554 {NULL, NULL}
1555};
1556
1557
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001558// ======= Window type =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001559
Bram Moolenaar1dced572012-04-05 16:54:08 +02001560luaV_newtype(win_T, window, luaV_Window, LUAVIM_WINDOW)
1561luaV_pushtype(win_T, window, luaV_Window)
1562luaV_type_tostring(window, LUAVIM_WINDOW)
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001563
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001564 static int
1565luaV_window_call(lua_State *L)
1566{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001567 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001568 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001569 win_goto(w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001570 return 1;
1571}
1572
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001573 static int
1574luaV_window_index(lua_State *L)
1575{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001576 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001577 const char *s = luaL_checkstring(L, 2);
1578 if (strncmp(s, "buffer", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001579 luaV_pushbuffer(L, w->w_buffer);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001580 else if (strncmp(s, "line", 4) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001581 lua_pushinteger(L, w->w_cursor.lnum);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001582 else if (strncmp(s, "col", 3) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001583 lua_pushinteger(L, w->w_cursor.col + 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001584 else if (strncmp(s, "width", 5) == 0)
Bram Moolenaar02631462017-09-22 15:20:32 +02001585 lua_pushinteger(L, w->w_width);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001586 else if (strncmp(s, "height", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001587 lua_pushinteger(L, w->w_height);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001588 // methods
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001589 else if (strncmp(s, "next", 4) == 0
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001590 || strncmp(s, "previous", 8) == 0
1591 || strncmp(s, "isvalid", 7) == 0)
1592 {
1593 lua_getmetatable(L, 1);
1594 lua_getfield(L, -1, s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001595 }
1596 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001597 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001598 return 1;
1599}
1600
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001601 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01001602luaV_window_newindex(lua_State *L)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001603{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001604 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001605 const char *s = luaL_checkstring(L, 2);
1606 int v = luaL_checkinteger(L, 3);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001607 if (strncmp(s, "line", 4) == 0)
1608 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001609#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001610 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001611#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001612 if (v < 1 || v > w->w_buffer->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001613 luaL_error(L, "line out of range");
Bram Moolenaar1dced572012-04-05 16:54:08 +02001614 w->w_cursor.lnum = v;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001615 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001616 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001617 else if (strncmp(s, "col", 3) == 0)
1618 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001619#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001620 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001621#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001622 w->w_cursor.col = v - 1;
Bram Moolenaar53901442018-07-25 22:02:36 +02001623 w->w_set_curswant = TRUE;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001624 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001625 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001626 else if (strncmp(s, "width", 5) == 0)
1627 {
1628 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001629#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001630 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001631#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001632 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001633 win_setwidth(v);
1634 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001635 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001636 else if (strncmp(s, "height", 6) == 0)
1637 {
1638 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001639#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001640 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001641#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001642 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001643 win_setheight(v);
1644 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001645 }
1646 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001647 luaL_error(L, "invalid window property: `%s'", s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001648 return 0;
1649}
1650
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001651 static int
1652luaV_window_next(lua_State *L)
1653{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001654 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001655 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1656 luaV_pushwindow(L, win->w_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001657 return 1;
1658}
1659
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001660 static int
1661luaV_window_previous(lua_State *L)
1662{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001663 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001664 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1665 luaV_pushwindow(L, win->w_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001666 return 1;
1667}
1668
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001669 static int
1670luaV_window_isvalid(lua_State *L)
1671{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001672 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001673 luaV_getudata(L, *w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001674 lua_pushboolean(L, !lua_isnil(L, -1));
1675 return 1;
1676}
1677
1678static const luaL_Reg luaV_Window_mt[] = {
1679 {"__tostring", luaV_window_tostring},
1680 {"__call", luaV_window_call},
1681 {"__index", luaV_window_index},
1682 {"__newindex", luaV_window_newindex},
1683 {"next", luaV_window_next},
1684 {"previous", luaV_window_previous},
1685 {"isvalid", luaV_window_isvalid},
1686 {NULL, NULL}
1687};
1688
1689
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001690// ======= Vim module =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001691
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001692 static int
1693luaV_print(lua_State *L)
1694{
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001695 int i, n = lua_gettop(L); // nargs
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001696 const char *s;
1697 size_t l;
1698 luaL_Buffer b;
1699 luaL_buffinit(L, &b);
1700 lua_getglobal(L, "tostring");
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001701 for (i = 1; i <= n; i++)
1702 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001703 lua_pushvalue(L, -1); // tostring
1704 lua_pushvalue(L, i); // arg
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001705 lua_call(L, 1, 1);
1706 s = lua_tolstring(L, -1, &l);
1707 if (s == NULL)
1708 return luaL_error(L, "cannot convert to string");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001709 if (i > 1) luaL_addchar(&b, ' '); // use space instead of tab
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001710 luaV_addlstring(&b, s, l, 0);
1711 lua_pop(L, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001712 }
1713 luaL_pushresult(&b);
Bram Moolenaarb98678a2019-10-19 15:18:44 +02001714 if (!got_int)
1715 luaV_msg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001716 return 0;
1717}
1718
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001719 static int
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001720luaV_debug(lua_State *L)
1721{
1722 lua_settop(L, 0);
1723 lua_getglobal(L, "vim");
1724 lua_getfield(L, -1, "eval");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001725 lua_remove(L, -2); // vim.eval at position 1
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001726 for (;;)
1727 {
1728 const char *input;
1729 size_t l;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001730 lua_pushvalue(L, 1); // vim.eval
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001731 lua_pushliteral(L, "input('lua_debug> ')");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001732 lua_call(L, 1, 1); // return string
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001733 input = lua_tolstring(L, -1, &l);
1734 if (l == 0 || strcmp(input, "cont") == 0)
1735 return 0;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001736 msg_putchar('\n'); // avoid outputting on input line
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001737 if (luaL_loadbuffer(L, input, l, "=(debug command)")
1738 || lua_pcall(L, 0, 0, 0))
1739 luaV_emsg(L);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001740 lua_settop(L, 1); // remove eventual returns, but keep vim.eval
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001741 }
1742}
1743
1744 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001745luaV_command(lua_State *L)
1746{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001747 do_cmdline_cmd((char_u *) luaL_checkstring(L, 1));
1748 update_screen(VALID);
1749 return 0;
1750}
1751
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001752 static int
1753luaV_eval(lua_State *L)
1754{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001755 typval_T *tv = eval_expr((char_u *) luaL_checkstring(L, 1), NULL);
1756 if (tv == NULL) luaL_error(L, "invalid expression");
1757 luaV_pushtypval(L, tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +02001758 free_tv(tv);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001759 return 1;
1760}
1761
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001762 static int
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001763luaV_beep(lua_State *L UNUSED)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001764{
Bram Moolenaar165bc692015-07-21 17:53:25 +02001765 vim_beep(BO_LANG);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001766 return 0;
1767}
1768
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001769 static int
1770luaV_line(lua_State *L)
1771{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001772 luaV_pushline(L, curbuf, curwin->w_cursor.lnum);
1773 return 1;
1774}
1775
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001776 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001777luaV_list(lua_State *L)
1778{
Bram Moolenaarca06da92018-07-01 15:12:05 +02001779 list_T *l;
1780 int initarg = !lua_isnoneornil(L, 1);
1781
1782 if (initarg && lua_type(L, 1) != LUA_TTABLE)
1783 luaL_error(L, "table expected, got %s", luaL_typename(L, 1));
1784 l = list_alloc();
Bram Moolenaar1dced572012-04-05 16:54:08 +02001785 if (l == NULL)
1786 lua_pushnil(L);
1787 else
Bram Moolenaarca06da92018-07-01 15:12:05 +02001788 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001789 luaV_newlist(L, l);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001790 if (initarg) // traverse table to init list
Bram Moolenaar17413672018-07-14 20:49:42 +02001791 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001792 int notnil, i = 0;
1793 typval_T v;
Bram Moolenaar17413672018-07-14 20:49:42 +02001794 do
1795 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001796 lua_rawgeti(L, 1, ++i);
1797 notnil = !lua_isnil(L, -1);
Bram Moolenaar17413672018-07-14 20:49:42 +02001798 if (notnil)
1799 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001800 luaV_checktypval(L, -1, &v, "vim.list");
1801 list_append_tv(l, &v);
Bram Moolenaar713bf9e92019-03-16 16:38:41 +01001802 clear_tv(&v);
Bram Moolenaarca06da92018-07-01 15:12:05 +02001803 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001804 lua_pop(L, 1); // value
Bram Moolenaarca06da92018-07-01 15:12:05 +02001805 } while (notnil);
1806 }
1807 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001808 return 1;
1809}
1810
1811 static int
1812luaV_dict(lua_State *L)
1813{
Bram Moolenaarca06da92018-07-01 15:12:05 +02001814 dict_T *d;
1815 int initarg = !lua_isnoneornil(L, 1);
1816
1817 if (initarg && lua_type(L, 1) != LUA_TTABLE)
1818 luaL_error(L, "table expected, got %s", luaL_typename(L, 1));
1819 d = dict_alloc();
Bram Moolenaar1dced572012-04-05 16:54:08 +02001820 if (d == NULL)
1821 lua_pushnil(L);
1822 else
Bram Moolenaarca06da92018-07-01 15:12:05 +02001823 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001824 luaV_newdict(L, d);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001825 if (initarg) // traverse table to init dict
Bram Moolenaarca06da92018-07-01 15:12:05 +02001826 {
1827 lua_pushnil(L);
1828 while (lua_next(L, 1))
1829 {
1830 char_u *key;
1831 dictitem_T *di;
1832 typval_T v;
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001833
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001834 lua_pushvalue(L, -2); // dup key in case it's a number
Bram Moolenaarca06da92018-07-01 15:12:05 +02001835 key = (char_u *) lua_tostring(L, -1);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001836 if (key == NULL)
1837 {
1838 lua_pushnil(L);
1839 return 1;
1840 }
1841 if (*key == NUL)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001842 luaL_error(L, "table has empty key");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001843 luaV_checktypval(L, -2, &v, "vim.dict"); // value
Bram Moolenaarca06da92018-07-01 15:12:05 +02001844 di = dictitem_alloc(key);
Bram Moolenaard6ef5f92018-07-13 22:08:23 +02001845 if (di == NULL || dict_add(d, di) == FAIL)
1846 {
Bram Moolenaarca06da92018-07-01 15:12:05 +02001847 vim_free(di);
1848 lua_pushnil(L);
1849 return 1;
1850 }
Bram Moolenaare49b8e82020-07-01 13:52:55 +02001851 di->di_tv = v;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001852 lua_pop(L, 2); // key copy and value
Bram Moolenaarca06da92018-07-01 15:12:05 +02001853 }
1854 }
1855 }
1856 return 1;
1857}
1858
1859 static int
Bram Moolenaarb7828692019-03-23 13:57:02 +01001860luaV_blob(lua_State *L)
1861{
1862 blob_T *b;
1863 int initarg = !lua_isnoneornil(L, 1);
1864
1865 if (initarg && !lua_isstring(L, 1))
1866 luaL_error(L, "string expected, got %s", luaL_typename(L, 1));
1867 b = blob_alloc();
1868 if (b == NULL)
1869 lua_pushnil(L);
1870 else
1871 {
1872 luaV_newblob(L, b);
1873 if (initarg)
1874 {
1875 size_t i, l = 0;
1876 const char *s = lua_tolstring(L, 1, &l);
1877
Bram Moolenaar5f1d3ae2020-02-11 22:37:35 +01001878 if (ga_grow(&b->bv_ga, (int)l) == OK)
Bram Moolenaar6fb5c972019-03-26 21:44:20 +01001879 for (i = 0; i < l; ++i)
1880 ga_append(&b->bv_ga, s[i]);
Bram Moolenaarb7828692019-03-23 13:57:02 +01001881 }
1882 }
1883 return 1;
1884}
1885
1886 static int
Bram Moolenaarca06da92018-07-01 15:12:05 +02001887luaV_funcref(lua_State *L)
1888{
1889 const char *name = luaL_checkstring(L, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001890 // note: not checking if function exists (needs function_exists)
Bram Moolenaarca06da92018-07-01 15:12:05 +02001891 if (name == NULL || *name == NUL || VIM_ISDIGIT(*name))
1892 luaL_error(L, "invalid function name: %s", name);
1893 luaV_newfuncref(L, (char_u *) name);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001894 return 1;
1895}
1896
1897 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001898luaV_buffer(lua_State *L)
1899{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001900 buf_T *buf;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001901 if (lua_isstring(L, 1)) // get by number or name?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001902 {
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001903 if (lua_isnumber(L, 1)) // by number?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001904 {
1905 int n = lua_tointeger(L, 1);
Bram Moolenaar29323592016-07-24 22:04:11 +02001906 FOR_ALL_BUFFERS(buf)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001907 if (buf->b_fnum == n) break;
1908 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001909 else // by name
1910 {
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001911 size_t l;
1912 const char *s = lua_tolstring(L, 1, &l);
Bram Moolenaar29323592016-07-24 22:04:11 +02001913 FOR_ALL_BUFFERS(buf)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001914 {
1915 if (buf->b_ffname == NULL || buf->b_sfname == NULL)
1916 {
1917 if (l == 0) break;
1918 }
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001919 else if (strncmp(s, (char *)buf->b_ffname, l) == 0
1920 || strncmp(s, (char *)buf->b_sfname, l) == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001921 break;
1922 }
1923 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001924 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001925 else
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001926 buf = (lua_toboolean(L, 1)) ? firstbuf : curbuf; // first buffer?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001927 luaV_pushbuffer(L, buf);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001928 return 1;
1929}
1930
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001931 static int
1932luaV_window(lua_State *L)
1933{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001934 win_T *win;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001935 if (lua_isnumber(L, 1)) // get by number?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001936 {
1937 int n = lua_tointeger(L, 1);
1938 for (win = firstwin; win != NULL; win = win->w_next, n--)
1939 if (n == 1) break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001940 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001941 else
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001942 win = (lua_toboolean(L, 1)) ? firstwin : curwin; // first window?
Bram Moolenaar1dced572012-04-05 16:54:08 +02001943 luaV_pushwindow(L, win);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001944 return 1;
1945}
1946
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001947 static int
1948luaV_open(lua_State *L)
1949{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001950 char_u *s = NULL;
1951#ifdef HAVE_SANDBOX
1952 luaV_checksandbox(L);
1953#endif
1954 if (lua_isstring(L, 1)) s = (char_u *) lua_tostring(L, 1);
Bram Moolenaarfa263a52011-12-08 16:00:16 +01001955 luaV_pushbuffer(L, buflist_new(s, NULL, 1L, BLN_LISTED));
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001956 return 1;
1957}
1958
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001959 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001960luaV_type(lua_State *L)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001961{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001962 luaL_checkany(L, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01001963 if (lua_type(L, 1) == LUA_TUSERDATA) // check vim udata?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001964 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001965 lua_settop(L, 1);
1966 if (lua_getmetatable(L, 1))
1967 {
1968 luaV_getfield(L, LUAVIM_LIST);
1969 if (lua_rawequal(L, -1, 2))
1970 {
1971 lua_pushstring(L, "list");
1972 return 1;
1973 }
1974 luaV_getfield(L, LUAVIM_DICT);
1975 if (lua_rawequal(L, -1, 2))
1976 {
1977 lua_pushstring(L, "dict");
1978 return 1;
1979 }
Bram Moolenaarb7828692019-03-23 13:57:02 +01001980 luaV_getfield(L, LUAVIM_BLOB);
1981 if (lua_rawequal(L, -1, 2))
1982 {
1983 lua_pushstring(L, "blob");
1984 return 1;
1985 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02001986 luaV_getfield(L, LUAVIM_FUNCREF);
1987 if (lua_rawequal(L, -1, 2))
1988 {
1989 lua_pushstring(L, "funcref");
1990 return 1;
1991 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001992 luaV_getfield(L, LUAVIM_BUFFER);
1993 if (lua_rawequal(L, -1, 2))
1994 {
1995 lua_pushstring(L, "buffer");
1996 return 1;
1997 }
1998 luaV_getfield(L, LUAVIM_WINDOW);
1999 if (lua_rawequal(L, -1, 2))
2000 {
2001 lua_pushstring(L, "window");
2002 return 1;
2003 }
2004 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002005 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002006 lua_pushstring(L, luaL_typename(L, 1)); // fallback
Bram Moolenaar1dced572012-04-05 16:54:08 +02002007 return 1;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002008}
2009
Bram Moolenaareb04f082020-05-17 14:32:35 +02002010 static int
2011luaV_call(lua_State *L)
2012{
2013 int argc = lua_gettop(L) - 1;
2014 size_t funcname_len;
2015 char_u *funcname;
2016 char *error = NULL;
2017 typval_T rettv;
2018 typval_T argv[MAX_FUNC_ARGS + 1];
2019 int i = 0;
2020
2021 if (argc > MAX_FUNC_ARGS)
2022 return luaL_error(L, "Function called with too many arguments");
2023
2024 funcname = (char_u *)luaL_checklstring(L, 1, &funcname_len);
2025
2026 for (; i < argc; i++)
2027 {
2028 if (luaV_totypval(L, i + 2, &argv[i]) == FAIL)
2029 {
2030 error = "lua: cannot convert value";
2031 goto free_vim_args;
2032 }
2033 }
2034
2035 argv[argc].v_type = VAR_UNKNOWN;
2036
2037 if (call_vim_function(funcname, argc, argv, &rettv) == FAIL)
2038 {
2039 error = "lua: call_vim_function failed";
2040 goto free_vim_args;
2041 }
2042
2043 luaV_pushtypval(L, &rettv);
2044 clear_tv(&rettv);
2045
2046free_vim_args:
2047 while (i > 0)
2048 clear_tv(&argv[--i]);
2049
2050 if (error == NULL)
2051 return 1;
2052 else
2053 return luaL_error(L, error);
2054}
2055
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002056static const luaL_Reg luaV_module[] = {
2057 {"command", luaV_command},
2058 {"eval", luaV_eval},
2059 {"beep", luaV_beep},
2060 {"line", luaV_line},
Bram Moolenaar1dced572012-04-05 16:54:08 +02002061 {"list", luaV_list},
2062 {"dict", luaV_dict},
Bram Moolenaarb7828692019-03-23 13:57:02 +01002063 {"blob", luaV_blob},
Bram Moolenaarca06da92018-07-01 15:12:05 +02002064 {"funcref", luaV_funcref},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002065 {"buffer", luaV_buffer},
2066 {"window", luaV_window},
2067 {"open", luaV_open},
Bram Moolenaar1dced572012-04-05 16:54:08 +02002068 {"type", luaV_type},
Bram Moolenaareb04f082020-05-17 14:32:35 +02002069 {"call", luaV_call},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002070 {NULL, NULL}
2071};
2072
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002073/*
2074 * for freeing list, dict, buffer and window objects; lightuserdata as arg
2075 */
Bram Moolenaar1dced572012-04-05 16:54:08 +02002076 static int
2077luaV_free(lua_State *L)
2078{
2079 lua_pushnil(L);
2080 luaV_setudata(L, lua_touserdata(L, 1));
2081 return 0;
2082}
2083
2084 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002085luaV_luaeval(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02002086{
2087 luaL_Buffer b;
2088 size_t l;
2089 const char *str = lua_tolstring(L, 1, &l);
2090 typval_T *arg = (typval_T *) lua_touserdata(L, 2);
2091 typval_T *rettv = (typval_T *) lua_touserdata(L, 3);
2092 luaL_buffinit(L, &b);
2093 luaL_addlstring(&b, LUAVIM_EVALHEADER, sizeof(LUAVIM_EVALHEADER) - 1);
2094 luaL_addlstring(&b, str, l);
2095 luaL_pushresult(&b);
2096 str = lua_tolstring(L, -1, &l);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002097 if (luaL_loadbuffer(L, str, l, LUAVIM_EVALNAME)) // compile error?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002098 {
2099 luaV_emsg(L);
2100 return 0;
2101 }
2102 luaV_pushtypval(L, arg);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002103 if (lua_pcall(L, 1, 1, 0)) // running error?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002104 {
2105 luaV_emsg(L);
2106 return 0;
2107 }
Bram Moolenaarca06da92018-07-01 15:12:05 +02002108 if (luaV_totypval(L, -1, rettv) == FAIL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002109 emsg("luaeval: cannot convert value");
Bram Moolenaarf554a322015-02-04 23:08:01 +01002110 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02002111}
2112
2113 static int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002114luaV_setref(lua_State *L)
Bram Moolenaar1dced572012-04-05 16:54:08 +02002115{
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002116 int copyID = lua_tointeger(L, 1);
2117 int abort = FALSE;
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002118
Bram Moolenaar1dced572012-04-05 16:54:08 +02002119 luaV_getfield(L, LUAVIM_LIST);
2120 luaV_getfield(L, LUAVIM_DICT);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002121 luaV_getfield(L, LUAVIM_FUNCREF);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002122 lua_pushnil(L);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002123 // traverse cache table
Bram Moolenaarb84634d2015-02-04 22:02:37 +01002124 while (!abort && lua_next(L, lua_upvalueindex(1)) != 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02002125 {
2126 lua_getmetatable(L, -1);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002127 if (lua_rawequal(L, -1, 2)) // list?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002128 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002129 list_T *l = (list_T *)lua_touserdata(L, 5); // key
2130
Bram Moolenaar7be3ab22019-06-23 01:46:15 +02002131 abort = set_ref_in_list(l, copyID);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002132 }
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002133 else if (lua_rawequal(L, -1, 3)) // dict?
Bram Moolenaar1dced572012-04-05 16:54:08 +02002134 {
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002135 dict_T *d = (dict_T *)lua_touserdata(L, 5); // key
2136
Bram Moolenaar7be3ab22019-06-23 01:46:15 +02002137 abort = set_ref_in_dict(d, copyID);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002138 }
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002139 else if (lua_rawequal(L, -1, 4)) // funcref?
2140 {
2141 luaV_Funcref *f = (luaV_Funcref *)lua_touserdata(L, 5); // key
2142
Bram Moolenaar7be3ab22019-06-23 01:46:15 +02002143 abort = set_ref_in_dict(f->self, copyID);
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002144 }
2145 lua_pop(L, 2); // metatable and value
Bram Moolenaar1dced572012-04-05 16:54:08 +02002146 }
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002147 lua_pushinteger(L, abort);
Bram Moolenaarf554a322015-02-04 23:08:01 +01002148 return 1;
Bram Moolenaar1dced572012-04-05 16:54:08 +02002149}
2150
Bram Moolenaareb04f082020-05-17 14:32:35 +02002151#define LUA_VIM_FN_CODE \
Bram Moolenaar788fbb42020-05-31 14:08:12 +02002152 "vim.fn = setmetatable({}, {\n"\
2153 " __index = function (t, key)\n"\
2154 " local function _fn(...)\n"\
2155 " return vim.call(key, ...)\n"\
2156 " end\n"\
2157 " t[key] = _fn\n"\
2158 " return _fn\n"\
2159 " end\n"\
2160 " })"
2161
2162#define LUA_VIM_UPDATE_PACKAGE_PATHS \
2163 "local last_vim_paths = {}\n"\
2164 "vim._update_package_paths = function ()\n"\
2165 " local cur_vim_paths = {}\n"\
2166 " local function split(s, delimiter)\n"\
2167 " result = {}\n"\
2168 " for match in (s..delimiter):gmatch(\"(.-)\"..delimiter) do\n"\
2169 " table.insert(result, match)\n"\
2170 " end\n"\
2171 " return result\n"\
2172 " end\n"\
2173 " local rtps = split(vim.eval('&runtimepath'), ',')\n"\
2174 " local sep = package.config:sub(1, 1)\n"\
2175 " for _, key in ipairs({'path', 'cpath'}) do\n"\
2176 " local orig_str = package[key] .. ';'\n"\
2177 " local pathtrails_ordered = {}\n"\
2178 " -- Note: ignores trailing item without trailing `;`. Not using something\n"\
2179 " -- simpler in order to preserve empty items (stand for default path).\n"\
2180 " local orig = {}\n"\
2181 " for s in orig_str:gmatch('[^;]*;') do\n"\
2182 " s = s:sub(1, -2) -- Strip trailing semicolon\n"\
2183 " orig[#orig + 1] = s\n"\
2184 " end\n"\
2185 " if key == 'path' then\n"\
2186 " -- /?.lua and /?/init.lua\n"\
2187 " pathtrails_ordered = {sep .. '?.lua', sep .. '?' .. sep .. 'init.lua'}\n"\
2188 " else\n"\
2189 " local pathtrails = {}\n"\
2190 " for _, s in ipairs(orig) do\n"\
2191 " -- Find out path patterns. pathtrail should contain something like\n"\
2192 " -- /?.so, \?.dll. This allows not to bother determining what correct\n"\
2193 " -- suffixes are.\n"\
2194 " local pathtrail = s:match('[/\\\\][^/\\\\]*%?.*$')\n"\
2195 " if pathtrail and not pathtrails[pathtrail] then\n"\
2196 " pathtrails[pathtrail] = true\n"\
2197 " pathtrails_ordered[#pathtrails_ordered + 1] = pathtrail\n"\
2198 " end\n"\
2199 " end\n"\
2200 " end\n"\
2201 " local new = {}\n"\
2202 " for _, rtp in ipairs(rtps) do\n"\
2203 " if not rtp:match(';') then\n"\
2204 " for _, pathtrail in pairs(pathtrails_ordered) do\n"\
2205 " local new_path = rtp .. sep .. 'lua' .. pathtrail\n"\
2206 " -- Always keep paths from &runtimepath at the start:\n"\
2207 " -- append them here disregarding orig possibly containing one of them.\n"\
2208 " new[#new + 1] = new_path\n"\
2209 " cur_vim_paths[new_path] = true\n"\
2210 " end\n"\
2211 " end\n"\
2212 " end\n"\
2213 " for _, orig_path in ipairs(orig) do\n"\
2214 " -- Handle removing obsolete paths originating from &runtimepath: such\n"\
2215 " -- paths either belong to cur_nvim_paths and were already added above or\n"\
2216 " -- to last_nvim_paths and should not be added at all if corresponding\n"\
2217 " -- entry was removed from &runtimepath list.\n"\
2218 " if not (cur_vim_paths[orig_path] or last_vim_paths[orig_path]) then\n"\
2219 " new[#new + 1] = orig_path\n"\
2220 " end\n"\
2221 " end\n"\
2222 " package[key] = table.concat(new, ';')\n"\
2223 " end\n"\
2224 " last_vim_paths = cur_vim_paths\n"\
2225 "end"
Bram Moolenaareb04f082020-05-17 14:32:35 +02002226
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002227 static int
2228luaopen_vim(lua_State *L)
2229{
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002230 // set cache table
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002231 lua_newtable(L);
2232 lua_newtable(L);
Bram Moolenaar1dced572012-04-05 16:54:08 +02002233 lua_pushstring(L, "v");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002234 lua_setfield(L, -2, "__mode");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002235 lua_setmetatable(L, -2); // cache is weak-valued
2236 // print
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002237 lua_pushcfunction(L, luaV_print);
2238 lua_setglobal(L, "print");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002239 // debug.debug
Bram Moolenaar38e2b062011-09-21 17:15:39 +02002240 lua_getglobal(L, "debug");
2241 lua_pushcfunction(L, luaV_debug);
2242 lua_setfield(L, -2, "debug");
2243 lua_pop(L, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002244 // free
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002245 lua_pushlightuserdata(L, (void *) LUAVIM_FREE);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002246 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002247 lua_pushcclosure(L, luaV_free, 1);
2248 lua_rawset(L, LUA_REGISTRYINDEX);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002249 // luaeval
Bram Moolenaar1dced572012-04-05 16:54:08 +02002250 lua_pushlightuserdata(L, (void *) LUAVIM_LUAEVAL);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002251 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002252 lua_pushcclosure(L, luaV_luaeval, 1);
2253 lua_rawset(L, LUA_REGISTRYINDEX);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002254 // setref
Bram Moolenaar1dced572012-04-05 16:54:08 +02002255 lua_pushlightuserdata(L, (void *) LUAVIM_SETREF);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002256 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002257 lua_pushcclosure(L, luaV_setref, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002258 lua_rawset(L, LUA_REGISTRYINDEX);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002259 // register
Bram Moolenaar1dced572012-04-05 16:54:08 +02002260 luaV_newmetatable(L, LUAVIM_LIST);
2261 lua_pushvalue(L, 1);
2262 luaV_openlib(L, luaV_List_mt, 1);
2263 luaV_newmetatable(L, LUAVIM_DICT);
2264 lua_pushvalue(L, 1);
2265 luaV_openlib(L, luaV_Dict_mt, 1);
Bram Moolenaarb7828692019-03-23 13:57:02 +01002266 luaV_newmetatable(L, LUAVIM_BLOB);
2267 lua_pushvalue(L, 1);
2268 luaV_openlib(L, luaV_Blob_mt, 1);
Bram Moolenaarca06da92018-07-01 15:12:05 +02002269 luaV_newmetatable(L, LUAVIM_FUNCREF);
2270 lua_pushvalue(L, 1);
2271 luaV_openlib(L, luaV_Funcref_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002272 luaV_newmetatable(L, LUAVIM_BUFFER);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002273 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002274 luaV_openlib(L, luaV_Buffer_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002275 luaV_newmetatable(L, LUAVIM_WINDOW);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002276 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002277 luaV_openlib(L, luaV_Window_mt, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002278 lua_newtable(L); // vim table
2279 lua_pushvalue(L, 1); // cache table
Bram Moolenaar1dced572012-04-05 16:54:08 +02002280 luaV_openlib(L, luaV_module, 1);
2281 lua_setglobal(L, LUAVIM_NAME);
Bram Moolenaareb04f082020-05-17 14:32:35 +02002282 // custom code
Bram Moolenaar9309eb22020-05-17 16:53:56 +02002283 (void)luaL_dostring(L, LUA_VIM_FN_CODE);
Bram Moolenaar788fbb42020-05-31 14:08:12 +02002284 (void)luaL_dostring(L, LUA_VIM_UPDATE_PACKAGE_PATHS);
2285
2286 lua_getglobal(L, "vim");
2287 lua_getfield(L, -1, "_update_package_paths");
2288
2289 if (lua_pcall(L, 0, 0, 0))
2290 luaV_emsg(L);
2291
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002292 return 0;
2293}
2294
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002295 static lua_State *
2296luaV_newstate(void)
2297{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002298 lua_State *L = luaL_newstate();
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002299 luaL_openlibs(L); // core libs
2300 lua_pushcfunction(L, luaopen_vim); // vim
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002301 lua_call(L, 0, 0);
2302 return L;
2303}
2304
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002305 static void
2306luaV_setrange(lua_State *L, int line1, int line2)
2307{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002308 lua_getglobal(L, LUAVIM_NAME);
2309 lua_pushinteger(L, line1);
2310 lua_setfield(L, -2, "firstline");
2311 lua_pushinteger(L, line2);
2312 lua_setfield(L, -2, "lastline");
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002313 lua_pop(L, 1); // vim table
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002314}
2315
2316
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002317// ======= Interface =======
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002318
2319static lua_State *L = NULL;
2320
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002321 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02002322lua_isopen(void)
Bram Moolenaar2bd6a1b2010-08-12 22:14:01 +02002323{
2324 return L != NULL;
2325}
2326
2327 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002328lua_init(void)
2329{
Bram Moolenaar1dced572012-04-05 16:54:08 +02002330 if (!lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002331 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002332#ifdef DYNAMIC_LUA
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002333 if (!lua_enabled(TRUE))
2334 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002335 emsg(_("Lua library cannot be loaded."));
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002336 return FAIL;
2337 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002338#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002339 L = luaV_newstate();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002340 }
2341 return OK;
2342}
2343
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002344 void
2345lua_end(void)
2346{
Bram Moolenaar1dced572012-04-05 16:54:08 +02002347 if (lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002348 {
2349 lua_close(L);
2350 L = NULL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002351 }
2352}
2353
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002354/*
2355 * ex commands
2356 */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002357 void
2358ex_lua(exarg_T *eap)
2359{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002360 char *script;
2361 if (lua_init() == FAIL) return;
2362 script = (char *) script_get(eap, eap->arg);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002363 if (!eap->skip)
2364 {
2365 char *s = (script) ? script : (char *) eap->arg;
2366 luaV_setrange(L, eap->line1, eap->line2);
2367 if (luaL_loadbuffer(L, s, strlen(s), LUAVIM_CHUNKNAME)
2368 || lua_pcall(L, 0, 0, 0))
2369 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002370 }
2371 if (script != NULL) vim_free(script);
2372}
2373
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002374 void
2375ex_luado(exarg_T *eap)
2376{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002377 linenr_T l;
2378 const char *s = (const char *) eap->arg;
2379 luaL_Buffer b;
2380 size_t len;
Bram Moolenaard58f03b2017-01-29 22:48:45 +01002381 buf_T *was_curbuf = curbuf;
2382
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002383 if (lua_init() == FAIL) return;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002384 if (u_save(eap->line1 - 1, eap->line2 + 1) == FAIL)
2385 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002386 emsg(_("cannot save undo information"));
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002387 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002388 }
2389 luaV_setrange(L, eap->line1, eap->line2);
2390 luaL_buffinit(L, &b);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002391 luaL_addlstring(&b, "return function(line, linenr) ", 30); // header
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002392 luaL_addlstring(&b, s, strlen(s));
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002393 luaL_addlstring(&b, " end", 4); // footer
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002394 luaL_pushresult(&b);
2395 s = lua_tolstring(L, -1, &len);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002396 if (luaL_loadbuffer(L, s, len, LUAVIM_CHUNKNAME))
2397 {
2398 luaV_emsg(L);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002399 lua_pop(L, 1); // function body
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002400 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002401 }
2402 lua_call(L, 0, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002403 lua_replace(L, -2); // function -> body
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002404 for (l = eap->line1; l <= eap->line2; l++)
2405 {
Bram Moolenaare49b8e82020-07-01 13:52:55 +02002406 // Check the line number, the command may have deleted lines.
Bram Moolenaard58f03b2017-01-29 22:48:45 +01002407 if (l > curbuf->b_ml.ml_line_count)
2408 break;
2409
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002410 lua_pushvalue(L, -1); // function
2411 luaV_pushline(L, curbuf, l); // current line as arg
2412 lua_pushinteger(L, l); // current line number as arg
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +02002413 if (lua_pcall(L, 2, 1, 0))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002414 {
2415 luaV_emsg(L);
2416 break;
2417 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002418 // Catch the command switching to another buffer.
Bram Moolenaard58f03b2017-01-29 22:48:45 +01002419 if (curbuf != was_curbuf)
2420 break;
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002421 if (lua_isstring(L, -1)) // update line?
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002422 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002423#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002424 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002425#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002426 ml_replace(l, luaV_toline(L, -1), TRUE);
2427 changed_bytes(l, 0);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002428 lua_pop(L, 1); // result from luaV_toline
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002429 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002430 lua_pop(L, 1); // line
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002431 }
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002432 lua_pop(L, 1); // function
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002433 check_cursor();
2434 update_screen(NOT_VALID);
2435}
2436
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002437 void
2438ex_luafile(exarg_T *eap)
2439{
2440 if (lua_init() == FAIL)
2441 return;
2442 if (!eap->skip)
2443 {
2444 luaV_setrange(L, eap->line1, eap->line2);
2445 if (luaL_loadfile(L, (char *) eap->arg) || lua_pcall(L, 0, 0, 0))
2446 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002447 }
2448}
2449
Bram Moolenaar1dced572012-04-05 16:54:08 +02002450#define luaV_freetype(typ,tname) \
2451 void \
2452 lua_##tname##_free(typ *o) \
2453 { \
2454 if (!lua_isopen()) return; \
2455 luaV_getfield(L, LUAVIM_FREE); \
2456 lua_pushlightuserdata(L, (void *) o); \
2457 lua_call(L, 1, 0); \
2458 }
2459
2460luaV_freetype(buf_T, buffer)
2461luaV_freetype(win_T, window)
2462
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002463 void
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002464do_luaeval(char_u *str, typval_T *arg, typval_T *rettv)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002465{
Bram Moolenaar1dced572012-04-05 16:54:08 +02002466 lua_init();
2467 luaV_getfield(L, LUAVIM_LUAEVAL);
2468 lua_pushstring(L, (char *) str);
2469 lua_pushlightuserdata(L, (void *) arg);
2470 lua_pushlightuserdata(L, (void *) rettv);
2471 lua_call(L, 3, 0);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002472}
2473
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002474 int
Bram Moolenaar4eefe472019-03-19 21:59:19 +01002475set_ref_in_lua(int copyID)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02002476{
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002477 int aborted = 0;
2478
2479 if (lua_isopen())
2480 {
2481 luaV_getfield(L, LUAVIM_SETREF);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002482 // call the function with 1 arg, getting 1 result back
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002483 lua_pushinteger(L, copyID);
2484 lua_call(L, 1, 1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002485 // get the result
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002486 aborted = lua_tointeger(L, -1);
Bram Moolenaar2ab2e862019-12-04 21:24:53 +01002487 // pop result off the stack
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01002488 lua_pop(L, 1);
2489 }
2490 return aborted;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002491}
2492
Bram Moolenaar788fbb42020-05-31 14:08:12 +02002493 void
2494update_package_paths_in_lua()
2495{
2496 if (lua_isopen())
2497 {
2498 lua_getglobal(L, "vim");
2499 lua_getfield(L, -1, "_update_package_paths");
2500
2501 if (lua_pcall(L, 0, 0, 0))
2502 luaV_emsg(L);
2503 }
2504}
2505
Bram Moolenaar801ab062020-06-25 19:27:56 +02002506/*
2507 * Native C function callback
2508 */
2509 static int
2510luaV_call_lua_func(
2511 int argcount,
2512 typval_T *argvars,
2513 typval_T *rettv,
2514 void *state)
2515{
2516 int i;
2517 int luaargcount = argcount;
2518 luaV_CFuncState *funcstate = (luaV_CFuncState*)state;
2519 lua_rawgeti(funcstate->L, LUA_REGISTRYINDEX, funcstate->lua_funcref);
2520
2521 if (funcstate->lua_tableref != LUA_NOREF)
2522 {
2523 // First arg for metatable __call method is a table
2524 luaargcount += 1;
2525 lua_rawgeti(funcstate->L, LUA_REGISTRYINDEX, funcstate->lua_tableref);
2526 }
2527
2528 for (i = 0; i < argcount; ++i)
2529 luaV_pushtypval(funcstate->L, &argvars[i]);
2530
2531 if (lua_pcall(funcstate->L, luaargcount, 1, 0))
2532 {
2533 luaV_emsg(funcstate->L);
2534 return FCERR_OTHER;
2535 }
2536
2537 luaV_checktypval(funcstate->L, -1, rettv, "get return value");
2538 return FCERR_NONE;
2539}
2540
2541/*
2542 * Free up any lua references held by the func state.
2543 */
2544 static void
2545luaV_call_lua_func_free(void *state)
2546{
2547 luaV_CFuncState *funcstate = (luaV_CFuncState*)state;
2548 luaL_unref(L, LUA_REGISTRYINDEX, funcstate->lua_funcref);
2549 funcstate->L = NULL;
2550 if (funcstate->lua_tableref != LUA_NOREF)
2551 luaL_unref(L, LUA_REGISTRYINDEX, funcstate->lua_tableref);
2552 VIM_CLEAR(funcstate);
2553}
2554
Bram Moolenaar0ba04292010-07-14 23:23:17 +02002555#endif