blob: 5803d07912b9f7b6e2eb2a24a5ae90193f04521f [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
18/* Only do the following when the feature is enabled. Needed for "make
19 * depend". */
20#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 Moolenaar0ba04292010-07-14 23:23:17 +020031typedef void (*msgfunc_T)(char_u *);
32
Bram Moolenaar1dced572012-04-05 16:54:08 +020033static const char LUAVIM_DICT[] = "dict";
34static const char LUAVIM_LIST[] = "list";
Bram Moolenaar0ba04292010-07-14 23:23:17 +020035static const char LUAVIM_BUFFER[] = "buffer";
36static const char LUAVIM_WINDOW[] = "window";
37static const char LUAVIM_FREE[] = "luaV_free";
Bram Moolenaar1dced572012-04-05 16:54:08 +020038static const char LUAVIM_LUAEVAL[] = "luaV_luaeval";
39static const char LUAVIM_SETREF[] = "luaV_setref";
Bram Moolenaar0ba04292010-07-14 23:23:17 +020040
Bram Moolenaar1dced572012-04-05 16:54:08 +020041/* most functions are closures with a cache table as first upvalue;
42 * get/setudata manage references to vim userdata in cache table through
43 * object pointers (light userdata) */
44#define luaV_getudata(L, v) \
45 lua_pushlightuserdata((L), (void *) (v)); \
46 lua_rawget((L), lua_upvalueindex(1))
47#define luaV_setudata(L, v) \
48 lua_pushlightuserdata((L), (void *) (v)); \
49 lua_pushvalue((L), -2); \
50 lua_rawset((L), lua_upvalueindex(1))
Bram Moolenaar0ba04292010-07-14 23:23:17 +020051#define luaV_getfield(L, s) \
52 lua_pushlightuserdata((L), (void *)(s)); \
53 lua_rawget((L), LUA_REGISTRYINDEX)
54#define luaV_checksandbox(L) \
55 if (sandbox) luaL_error((L), "not allowed in sandbox")
56#define luaV_msg(L) luaV_msgfunc((L), (msgfunc_T) msg)
57#define luaV_emsg(L) luaV_msgfunc((L), (msgfunc_T) emsg)
58
Bram Moolenaar1dced572012-04-05 16:54:08 +020059static luaV_List *luaV_pushlist (lua_State *L, list_T *lis);
60static luaV_Dict *luaV_pushdict (lua_State *L, dict_T *dic);
61
62#if LUA_VERSION_NUM <= 501
63#define luaV_openlib(L, l, n) luaL_openlib(L, NULL, l, n)
64#define luaL_typeerror luaL_typerror
65#else
66#define luaV_openlib luaL_setfuncs
67#endif
Bram Moolenaar0ba04292010-07-14 23:23:17 +020068
69#ifdef DYNAMIC_LUA
Bram Moolenaar2334b6d2010-07-22 21:32:16 +020070
71#ifndef WIN3264
72# include <dlfcn.h>
73# define HANDLE void*
74# define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
75# define symbol_from_dll dlsym
76# define close_dll dlclose
77#else
Bram Moolenaarebbcb822010-10-23 14:02:54 +020078# define load_dll vimLoadLib
Bram Moolenaar2334b6d2010-07-22 21:32:16 +020079# define symbol_from_dll GetProcAddress
80# define close_dll FreeLibrary
81#endif
82
Bram Moolenaar0ba04292010-07-14 23:23:17 +020083/* lauxlib */
Bram Moolenaar1dced572012-04-05 16:54:08 +020084#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +020085#define luaL_register dll_luaL_register
Bram Moolenaar1dced572012-04-05 16:54:08 +020086#define luaL_prepbuffer dll_luaL_prepbuffer
87#define luaL_openlib dll_luaL_openlib
Bram Moolenaar0ba04292010-07-14 23:23:17 +020088#define luaL_typerror dll_luaL_typerror
Bram Moolenaar1dced572012-04-05 16:54:08 +020089#define luaL_loadfile dll_luaL_loadfile
90#define luaL_loadbuffer dll_luaL_loadbuffer
91#else
92#define luaL_prepbuffsize dll_luaL_prepbuffsize
93#define luaL_setfuncs dll_luaL_setfuncs
94#define luaL_loadfilex dll_luaL_loadfilex
95#define luaL_loadbufferx dll_luaL_loadbufferx
96#define luaL_argerror dll_luaL_argerror
97#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +020098#define luaL_checkany dll_luaL_checkany
Bram Moolenaar0ba04292010-07-14 23:23:17 +020099#define luaL_checklstring dll_luaL_checklstring
100#define luaL_checkinteger dll_luaL_checkinteger
101#define luaL_optinteger dll_luaL_optinteger
102#define luaL_checktype dll_luaL_checktype
103#define luaL_error dll_luaL_error
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200104#define luaL_newstate dll_luaL_newstate
105#define luaL_buffinit dll_luaL_buffinit
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200106#define luaL_addlstring dll_luaL_addlstring
107#define luaL_pushresult dll_luaL_pushresult
108/* lua */
Bram Moolenaar1dced572012-04-05 16:54:08 +0200109#if LUA_VERSION_NUM <= 501
110#define lua_tonumber dll_lua_tonumber
111#define lua_tointeger dll_lua_tointeger
112#define lua_call dll_lua_call
113#define lua_pcall dll_lua_pcall
114#else
115#define lua_tonumberx dll_lua_tonumberx
116#define lua_tointegerx dll_lua_tointegerx
117#define lua_callk dll_lua_callk
118#define lua_pcallk dll_lua_pcallk
119#define lua_getglobal dll_lua_getglobal
120#define lua_setglobal dll_lua_setglobal
Bram Moolenaar1dced572012-04-05 16:54:08 +0200121#endif
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200122#if LUA_VERSION_NUM <= 502
123#define lua_replace dll_lua_replace
124#define lua_remove dll_lua_remove
125#endif
126#if LUA_VERSION_NUM >= 503
127#define lua_rotate dll_lua_rotate
128#define lua_copy dll_lua_copy
129#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200130#define lua_typename dll_lua_typename
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200131#define lua_close dll_lua_close
132#define lua_gettop dll_lua_gettop
133#define lua_settop dll_lua_settop
134#define lua_pushvalue dll_lua_pushvalue
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200135#define lua_isnumber dll_lua_isnumber
136#define lua_isstring dll_lua_isstring
137#define lua_type dll_lua_type
138#define lua_rawequal dll_lua_rawequal
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200139#define lua_toboolean dll_lua_toboolean
140#define lua_tolstring dll_lua_tolstring
141#define lua_touserdata dll_lua_touserdata
142#define lua_pushnil dll_lua_pushnil
143#define lua_pushnumber dll_lua_pushnumber
144#define lua_pushinteger dll_lua_pushinteger
145#define lua_pushlstring dll_lua_pushlstring
146#define lua_pushstring dll_lua_pushstring
147#define lua_pushfstring dll_lua_pushfstring
148#define lua_pushcclosure dll_lua_pushcclosure
149#define lua_pushboolean dll_lua_pushboolean
150#define lua_pushlightuserdata dll_lua_pushlightuserdata
151#define lua_getfield dll_lua_getfield
152#define lua_rawget dll_lua_rawget
Bram Moolenaar1dced572012-04-05 16:54:08 +0200153#define lua_rawgeti dll_lua_rawgeti
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200154#define lua_createtable dll_lua_createtable
155#define lua_newuserdata dll_lua_newuserdata
156#define lua_getmetatable dll_lua_getmetatable
157#define lua_setfield dll_lua_setfield
158#define lua_rawset dll_lua_rawset
159#define lua_rawseti dll_lua_rawseti
160#define lua_setmetatable dll_lua_setmetatable
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200161#define lua_next dll_lua_next
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200162/* libs */
163#define luaopen_base dll_luaopen_base
164#define luaopen_table dll_luaopen_table
165#define luaopen_string dll_luaopen_string
166#define luaopen_math dll_luaopen_math
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200167#define luaopen_io dll_luaopen_io
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200168#define luaopen_os dll_luaopen_os
169#define luaopen_package dll_luaopen_package
170#define luaopen_debug dll_luaopen_debug
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200171#define luaL_openlibs dll_luaL_openlibs
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200172
173/* lauxlib */
Bram Moolenaar1dced572012-04-05 16:54:08 +0200174#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200175void (*dll_luaL_register) (lua_State *L, const char *libname, const luaL_Reg *l);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200176char *(*dll_luaL_prepbuffer) (luaL_Buffer *B);
177void (*dll_luaL_openlib) (lua_State *L, const char *libname, const luaL_Reg *l, int nup);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200178int (*dll_luaL_typerror) (lua_State *L, int narg, const char *tname);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200179int (*dll_luaL_loadfile) (lua_State *L, const char *filename);
180int (*dll_luaL_loadbuffer) (lua_State *L, const char *buff, size_t sz, const char *name);
181#else
182char *(*dll_luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
183void (*dll_luaL_setfuncs) (lua_State *L, const luaL_Reg *l, int nup);
184int (*dll_luaL_loadfilex) (lua_State *L, const char *filename, const char *mode);
185int (*dll_luaL_loadbufferx) (lua_State *L, const char *buff, size_t sz, const char *name, const char *mode);
186int (*dll_luaL_argerror) (lua_State *L, int numarg, const char *extramsg);
187#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200188void (*dll_luaL_checkany) (lua_State *L, int narg);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200189const char *(*dll_luaL_checklstring) (lua_State *L, int numArg, size_t *l);
190lua_Integer (*dll_luaL_checkinteger) (lua_State *L, int numArg);
191lua_Integer (*dll_luaL_optinteger) (lua_State *L, int nArg, lua_Integer def);
192void (*dll_luaL_checktype) (lua_State *L, int narg, int t);
193int (*dll_luaL_error) (lua_State *L, const char *fmt, ...);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200194lua_State *(*dll_luaL_newstate) (void);
195void (*dll_luaL_buffinit) (lua_State *L, luaL_Buffer *B);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200196void (*dll_luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
197void (*dll_luaL_pushresult) (luaL_Buffer *B);
198/* lua */
Bram Moolenaar1dced572012-04-05 16:54:08 +0200199#if LUA_VERSION_NUM <= 501
200lua_Number (*dll_lua_tonumber) (lua_State *L, int idx);
201lua_Integer (*dll_lua_tointeger) (lua_State *L, int idx);
202void (*dll_lua_call) (lua_State *L, int nargs, int nresults);
203int (*dll_lua_pcall) (lua_State *L, int nargs, int nresults, int errfunc);
204#else
205lua_Number (*dll_lua_tonumberx) (lua_State *L, int idx, int *isnum);
206lua_Integer (*dll_lua_tointegerx) (lua_State *L, int idx, int *isnum);
207void (*dll_lua_callk) (lua_State *L, int nargs, int nresults, int ctx,
Bram Moolenaardb913952012-06-29 12:54:53 +0200208 lua_CFunction k);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200209int (*dll_lua_pcallk) (lua_State *L, int nargs, int nresults, int errfunc,
Bram Moolenaardb913952012-06-29 12:54:53 +0200210 int ctx, lua_CFunction k);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200211void (*dll_lua_getglobal) (lua_State *L, const char *var);
212void (*dll_lua_setglobal) (lua_State *L, const char *var);
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200213#endif
214#if LUA_VERSION_NUM <= 502
215void (*dll_lua_replace) (lua_State *L, int idx);
216void (*dll_lua_remove) (lua_State *L, int idx);
217#endif
218#if LUA_VERSION_NUM >= 503
219void (*dll_lua_rotate) (lua_State *L, int idx, int n);
Bram Moolenaar9514b1f2015-06-25 18:27:32 +0200220void (*dll_lua_copy) (lua_State *L, int fromidx, int toidx);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200221#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200222const char *(*dll_lua_typename) (lua_State *L, int tp);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200223void (*dll_lua_close) (lua_State *L);
224int (*dll_lua_gettop) (lua_State *L);
225void (*dll_lua_settop) (lua_State *L, int idx);
226void (*dll_lua_pushvalue) (lua_State *L, int idx);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200227int (*dll_lua_isnumber) (lua_State *L, int idx);
228int (*dll_lua_isstring) (lua_State *L, int idx);
229int (*dll_lua_type) (lua_State *L, int idx);
230int (*dll_lua_rawequal) (lua_State *L, int idx1, int idx2);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200231int (*dll_lua_toboolean) (lua_State *L, int idx);
232const char *(*dll_lua_tolstring) (lua_State *L, int idx, size_t *len);
233void *(*dll_lua_touserdata) (lua_State *L, int idx);
234void (*dll_lua_pushnil) (lua_State *L);
235void (*dll_lua_pushnumber) (lua_State *L, lua_Number n);
236void (*dll_lua_pushinteger) (lua_State *L, lua_Integer n);
237void (*dll_lua_pushlstring) (lua_State *L, const char *s, size_t l);
238void (*dll_lua_pushstring) (lua_State *L, const char *s);
239const char *(*dll_lua_pushfstring) (lua_State *L, const char *fmt, ...);
240void (*dll_lua_pushcclosure) (lua_State *L, lua_CFunction fn, int n);
241void (*dll_lua_pushboolean) (lua_State *L, int b);
242void (*dll_lua_pushlightuserdata) (lua_State *L, void *p);
243void (*dll_lua_getfield) (lua_State *L, int idx, const char *k);
244void (*dll_lua_rawget) (lua_State *L, int idx);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200245void (*dll_lua_rawgeti) (lua_State *L, int idx, int n);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200246void (*dll_lua_createtable) (lua_State *L, int narr, int nrec);
247void *(*dll_lua_newuserdata) (lua_State *L, size_t sz);
248int (*dll_lua_getmetatable) (lua_State *L, int objindex);
249void (*dll_lua_setfield) (lua_State *L, int idx, const char *k);
250void (*dll_lua_rawset) (lua_State *L, int idx);
251void (*dll_lua_rawseti) (lua_State *L, int idx, int n);
252int (*dll_lua_setmetatable) (lua_State *L, int objindex);
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200253int (*dll_lua_next) (lua_State *L, int idx);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200254/* libs */
255int (*dll_luaopen_base) (lua_State *L);
256int (*dll_luaopen_table) (lua_State *L);
257int (*dll_luaopen_string) (lua_State *L);
258int (*dll_luaopen_math) (lua_State *L);
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200259int (*dll_luaopen_io) (lua_State *L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200260int (*dll_luaopen_os) (lua_State *L);
261int (*dll_luaopen_package) (lua_State *L);
262int (*dll_luaopen_debug) (lua_State *L);
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200263void (*dll_luaL_openlibs) (lua_State *L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200264
265typedef void **luaV_function;
266typedef struct {
267 const char *name;
268 luaV_function func;
269} luaV_Reg;
270
271static const luaV_Reg luaV_dll[] = {
272 /* lauxlib */
Bram Moolenaar1dced572012-04-05 16:54:08 +0200273#if LUA_VERSION_NUM <= 501
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200274 {"luaL_register", (luaV_function) &dll_luaL_register},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200275 {"luaL_prepbuffer", (luaV_function) &dll_luaL_prepbuffer},
276 {"luaL_openlib", (luaV_function) &dll_luaL_openlib},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200277 {"luaL_typerror", (luaV_function) &dll_luaL_typerror},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200278 {"luaL_loadfile", (luaV_function) &dll_luaL_loadfile},
279 {"luaL_loadbuffer", (luaV_function) &dll_luaL_loadbuffer},
280#else
281 {"luaL_prepbuffsize", (luaV_function) &dll_luaL_prepbuffsize},
282 {"luaL_setfuncs", (luaV_function) &dll_luaL_setfuncs},
283 {"luaL_loadfilex", (luaV_function) &dll_luaL_loadfilex},
284 {"luaL_loadbufferx", (luaV_function) &dll_luaL_loadbufferx},
285 {"luaL_argerror", (luaV_function) &dll_luaL_argerror},
286#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200287 {"luaL_checkany", (luaV_function) &dll_luaL_checkany},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200288 {"luaL_checklstring", (luaV_function) &dll_luaL_checklstring},
289 {"luaL_checkinteger", (luaV_function) &dll_luaL_checkinteger},
290 {"luaL_optinteger", (luaV_function) &dll_luaL_optinteger},
291 {"luaL_checktype", (luaV_function) &dll_luaL_checktype},
292 {"luaL_error", (luaV_function) &dll_luaL_error},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200293 {"luaL_newstate", (luaV_function) &dll_luaL_newstate},
294 {"luaL_buffinit", (luaV_function) &dll_luaL_buffinit},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200295 {"luaL_addlstring", (luaV_function) &dll_luaL_addlstring},
296 {"luaL_pushresult", (luaV_function) &dll_luaL_pushresult},
297 /* lua */
Bram Moolenaar1dced572012-04-05 16:54:08 +0200298#if LUA_VERSION_NUM <= 501
299 {"lua_tonumber", (luaV_function) &dll_lua_tonumber},
300 {"lua_tointeger", (luaV_function) &dll_lua_tointeger},
301 {"lua_call", (luaV_function) &dll_lua_call},
302 {"lua_pcall", (luaV_function) &dll_lua_pcall},
303#else
304 {"lua_tonumberx", (luaV_function) &dll_lua_tonumberx},
305 {"lua_tointegerx", (luaV_function) &dll_lua_tointegerx},
306 {"lua_callk", (luaV_function) &dll_lua_callk},
307 {"lua_pcallk", (luaV_function) &dll_lua_pcallk},
308 {"lua_getglobal", (luaV_function) &dll_lua_getglobal},
309 {"lua_setglobal", (luaV_function) &dll_lua_setglobal},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200310#endif
Bram Moolenaar1f860d82015-06-27 18:36:16 +0200311#if LUA_VERSION_NUM <= 502
312 {"lua_replace", (luaV_function) &dll_lua_replace},
313 {"lua_remove", (luaV_function) &dll_lua_remove},
314#endif
315#if LUA_VERSION_NUM >= 503
316 {"lua_rotate", (luaV_function) &dll_lua_rotate},
317 {"lua_copy", (luaV_function) &dll_lua_copy},
318#endif
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200319 {"lua_typename", (luaV_function) &dll_lua_typename},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200320 {"lua_close", (luaV_function) &dll_lua_close},
321 {"lua_gettop", (luaV_function) &dll_lua_gettop},
322 {"lua_settop", (luaV_function) &dll_lua_settop},
323 {"lua_pushvalue", (luaV_function) &dll_lua_pushvalue},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200324 {"lua_isnumber", (luaV_function) &dll_lua_isnumber},
325 {"lua_isstring", (luaV_function) &dll_lua_isstring},
326 {"lua_type", (luaV_function) &dll_lua_type},
327 {"lua_rawequal", (luaV_function) &dll_lua_rawequal},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200328 {"lua_toboolean", (luaV_function) &dll_lua_toboolean},
329 {"lua_tolstring", (luaV_function) &dll_lua_tolstring},
330 {"lua_touserdata", (luaV_function) &dll_lua_touserdata},
331 {"lua_pushnil", (luaV_function) &dll_lua_pushnil},
332 {"lua_pushnumber", (luaV_function) &dll_lua_pushnumber},
333 {"lua_pushinteger", (luaV_function) &dll_lua_pushinteger},
334 {"lua_pushlstring", (luaV_function) &dll_lua_pushlstring},
335 {"lua_pushstring", (luaV_function) &dll_lua_pushstring},
336 {"lua_pushfstring", (luaV_function) &dll_lua_pushfstring},
337 {"lua_pushcclosure", (luaV_function) &dll_lua_pushcclosure},
338 {"lua_pushboolean", (luaV_function) &dll_lua_pushboolean},
339 {"lua_pushlightuserdata", (luaV_function) &dll_lua_pushlightuserdata},
340 {"lua_getfield", (luaV_function) &dll_lua_getfield},
341 {"lua_rawget", (luaV_function) &dll_lua_rawget},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200342 {"lua_rawgeti", (luaV_function) &dll_lua_rawgeti},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200343 {"lua_createtable", (luaV_function) &dll_lua_createtable},
344 {"lua_newuserdata", (luaV_function) &dll_lua_newuserdata},
345 {"lua_getmetatable", (luaV_function) &dll_lua_getmetatable},
346 {"lua_setfield", (luaV_function) &dll_lua_setfield},
347 {"lua_rawset", (luaV_function) &dll_lua_rawset},
348 {"lua_rawseti", (luaV_function) &dll_lua_rawseti},
349 {"lua_setmetatable", (luaV_function) &dll_lua_setmetatable},
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +0200350 {"lua_next", (luaV_function) &dll_lua_next},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200351 /* libs */
352 {"luaopen_base", (luaV_function) &dll_luaopen_base},
353 {"luaopen_table", (luaV_function) &dll_luaopen_table},
354 {"luaopen_string", (luaV_function) &dll_luaopen_string},
355 {"luaopen_math", (luaV_function) &dll_luaopen_math},
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200356 {"luaopen_io", (luaV_function) &dll_luaopen_io},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200357 {"luaopen_os", (luaV_function) &dll_luaopen_os},
358 {"luaopen_package", (luaV_function) &dll_luaopen_package},
359 {"luaopen_debug", (luaV_function) &dll_luaopen_debug},
Bram Moolenaar16c98f92010-07-28 22:46:08 +0200360 {"luaL_openlibs", (luaV_function) &dll_luaL_openlibs},
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200361 {NULL, NULL}
362};
363
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200364static HANDLE hinstLua = NULL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200365
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200366 static void
367end_dynamic_lua(void)
368{
369 if (hinstLua)
370 {
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200371 close_dll(hinstLua);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200372 hinstLua = 0;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200373 }
374}
375
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200376 static int
377lua_link_init(char *libname, int verbose)
378{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200379 const luaV_Reg *reg;
380 if (hinstLua) return OK;
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200381 hinstLua = load_dll(libname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200382 if (!hinstLua)
383 {
384 if (verbose)
385 EMSG2(_(e_loadlib), libname);
386 return FAIL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200387 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200388 for (reg = luaV_dll; reg->func; reg++)
389 {
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200390 if ((*reg->func = symbol_from_dll(hinstLua, reg->name)) == NULL)
391 {
392 close_dll(hinstLua);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200393 hinstLua = 0;
394 if (verbose)
395 EMSG2(_(e_loadfunc), reg->name);
396 return FAIL;
397 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200398 }
399 return OK;
400}
401
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200402 int
403lua_enabled(int verbose)
404{
Bram Moolenaar25e4fcd2016-01-09 14:57:47 +0100405 return lua_link_init((char *)p_luadll, verbose) == OK;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200406}
407
408#endif /* DYNAMIC_LUA */
409
Bram Moolenaar1dced572012-04-05 16:54:08 +0200410#if LUA_VERSION_NUM > 501
411 static int
412luaL_typeerror (lua_State *L, int narg, const char *tname)
413{
414 const char *msg = lua_pushfstring(L, "%s expected, got %s",
Bram Moolenaardb913952012-06-29 12:54:53 +0200415 tname, luaL_typename(L, narg));
Bram Moolenaar1dced572012-04-05 16:54:08 +0200416 return luaL_argerror(L, narg, msg);
417}
418#endif
419
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200420
421/* ======= Internal ======= */
422
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200423 static void
424luaV_newmetatable(lua_State *L, const char *tname)
425{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200426 lua_newtable(L);
427 lua_pushlightuserdata(L, (void *) tname);
428 lua_pushvalue(L, -2);
429 lua_rawset(L, LUA_REGISTRYINDEX);
430}
431
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200432 static void *
433luaV_toudata(lua_State *L, int ud, const char *tname)
434{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200435 void *p = lua_touserdata(L, ud);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200436
437 if (p != NULL) /* value is userdata? */
438 {
439 if (lua_getmetatable(L, ud)) /* does it have a metatable? */
440 {
441 luaV_getfield(L, tname); /* get metatable */
442 if (lua_rawequal(L, -1, -2)) /* MTs match? */
443 {
444 lua_pop(L, 2); /* MTs */
445 return p;
446 }
447 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200448 }
449 return NULL;
450}
451
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200452 static void *
Bram Moolenaar1dced572012-04-05 16:54:08 +0200453luaV_checkcache(lua_State *L, void *p)
454{
455 luaV_getudata(L, p);
456 if (lua_isnil(L, -1)) luaL_error(L, "invalid object");
457 lua_pop(L, 1);
458 return p;
459}
460
461#define luaV_unbox(L,luatyp,ud) (*((luatyp *) lua_touserdata((L),(ud))))
462
463#define luaV_checkvalid(L,luatyp,ud) \
464 luaV_checkcache((L), (void *) luaV_unbox((L),luatyp,(ud)))
465
466 static void *
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200467luaV_checkudata(lua_State *L, int ud, const char *tname)
468{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200469 void *p = luaV_toudata(L, ud, tname);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200470 if (p == NULL) luaL_typeerror(L, ud, tname);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200471 return p;
472}
473
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200474 static void
475luaV_pushtypval(lua_State *L, typval_T *tv)
476{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200477 if (tv == NULL)
478 {
479 lua_pushnil(L);
480 return;
481 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200482 switch (tv->v_type)
483 {
484 case VAR_STRING:
Bram Moolenaard04da7c2012-10-14 03:41:59 +0200485 lua_pushstring(L, tv->vval.v_string == NULL
486 ? "" : (char *)tv->vval.v_string);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200487 break;
488 case VAR_NUMBER:
489 lua_pushinteger(L, (int) tv->vval.v_number);
490 break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200491#ifdef FEAT_FLOAT
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200492 case VAR_FLOAT:
493 lua_pushnumber(L, (lua_Number) tv->vval.v_float);
494 break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200495#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +0200496 case VAR_LIST:
497 luaV_pushlist(L, tv->vval.v_list);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200498 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200499 case VAR_DICT:
500 luaV_pushdict(L, tv->vval.v_dict);
501 break;
Bram Moolenaar520e1e42016-01-23 19:46:28 +0100502 case VAR_SPECIAL:
503 if (tv->vval.v_number <= VVAL_TRUE)
504 lua_pushinteger(L, (int) tv->vval.v_number);
505 else
506 lua_pushnil(L);
507 break;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200508 default:
509 lua_pushnil(L);
510 }
511}
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200512
Bram Moolenaar1dced572012-04-05 16:54:08 +0200513/* converts lua value at 'pos' to typval 'tv' */
514 static void
515luaV_totypval (lua_State *L, int pos, typval_T *tv)
516{
517 switch(lua_type(L, pos)) {
518 case LUA_TBOOLEAN:
Bram Moolenaar520e1e42016-01-23 19:46:28 +0100519 tv->v_type = VAR_SPECIAL;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200520 tv->vval.v_number = (varnumber_T) lua_toboolean(L, pos);
521 break;
522 case LUA_TSTRING:
523 tv->v_type = VAR_STRING;
524 tv->vval.v_string = vim_strsave((char_u *) lua_tostring(L, pos));
525 break;
526 case LUA_TNUMBER:
527#ifdef FEAT_FLOAT
528 tv->v_type = VAR_FLOAT;
529 tv->vval.v_float = (float_T) lua_tonumber(L, pos);
530#else
531 tv->v_type = VAR_NUMBER;
532 tv->vval.v_number = (varnumber_T) lua_tointeger(L, pos);
533#endif
534 break;
535 case LUA_TUSERDATA: {
536 void *p = lua_touserdata(L, pos);
537 if (lua_getmetatable(L, pos)) /* has metatable? */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200538 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200539 /* check list */
540 luaV_getfield(L, LUAVIM_LIST);
541 if (lua_rawequal(L, -1, -2))
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200542 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200543 tv->v_type = VAR_LIST;
544 tv->vval.v_list = *((luaV_List *) p);
545 ++tv->vval.v_list->lv_refcount;
546 lua_pop(L, 2); /* MTs */
547 return;
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200548 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200549 /* check dict */
550 luaV_getfield(L, LUAVIM_DICT);
551 if (lua_rawequal(L, -1, -3))
552 {
553 tv->v_type = VAR_DICT;
554 tv->vval.v_dict = *((luaV_Dict *) p);
555 ++tv->vval.v_dict->dv_refcount;
556 lua_pop(L, 3); /* MTs */
557 return;
558 }
559 lua_pop(L, 3); /* MTs */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200560 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200561 break;
562 }
563 default:
Bram Moolenaar1dced572012-04-05 16:54:08 +0200564 tv->v_type = VAR_NUMBER;
565 tv->vval.v_number = 0;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200566 }
567}
568
569/* similar to luaL_addlstring, but replaces \0 with \n if toline and
570 * \n with \0 otherwise */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200571 static void
572luaV_addlstring(luaL_Buffer *b, const char *s, size_t l, int toline)
573{
574 while (l--)
575 {
576 if (*s == '\0' && toline)
577 luaL_addchar(b, '\n');
578 else if (*s == '\n' && !toline)
579 luaL_addchar(b, '\0');
580 else
581 luaL_addchar(b, *s);
582 s++;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200583 }
584}
585
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200586 static void
587luaV_pushline(lua_State *L, buf_T *buf, linenr_T n)
588{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200589 const char *s = (const char *) ml_get_buf(buf, n, FALSE);
590 luaL_Buffer b;
591 luaL_buffinit(L, &b);
592 luaV_addlstring(&b, s, strlen(s), 0);
593 luaL_pushresult(&b);
594}
595
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200596 static char_u *
597luaV_toline(lua_State *L, int pos)
598{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200599 size_t l;
600 const char *s = lua_tolstring(L, pos, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200601
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200602 luaL_Buffer b;
603 luaL_buffinit(L, &b);
604 luaV_addlstring(&b, s, l, 1);
605 luaL_pushresult(&b);
606 return (char_u *) lua_tostring(L, -1);
607}
608
609/* pops a string s from the top of the stack and calls mf(t) for pieces t of
610 * s separated by newlines */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200611 static void
612luaV_msgfunc(lua_State *L, msgfunc_T mf)
613{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200614 luaL_Buffer b;
615 size_t l;
616 const char *p, *s = lua_tolstring(L, -1, &l);
617 luaL_buffinit(L, &b);
618 luaV_addlstring(&b, s, l, 0);
619 luaL_pushresult(&b);
620 /* break string */
621 p = s = lua_tolstring(L, -1, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200622 while (l--)
623 {
624 if (*p++ == '\0') /* break? */
625 {
626 mf((char_u *) s);
627 s = p;
628 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200629 }
630 mf((char_u *) s);
631 lua_pop(L, 2); /* original and modified strings */
632}
633
Bram Moolenaar1dced572012-04-05 16:54:08 +0200634#define luaV_newtype(typ,tname,luatyp,luatname) \
635 static luatyp * \
636 luaV_new##tname (lua_State *L, typ *obj) \
637 { \
638 luatyp *o = (luatyp *) lua_newuserdata(L, sizeof(luatyp)); \
639 *o = obj; \
640 luaV_setudata(L, obj); /* cache[obj] = udata */ \
641 luaV_getfield(L, luatname); \
642 lua_setmetatable(L, -2); \
643 return o; \
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200644 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200645
646#define luaV_pushtype(typ,tname,luatyp) \
647 static luatyp * \
648 luaV_push##tname (lua_State *L, typ *obj) \
649 { \
650 luatyp *o = NULL; \
651 if (obj == NULL) \
652 lua_pushnil(L); \
653 else { \
654 luaV_getudata(L, obj); \
655 if (lua_isnil(L, -1)) /* not interned? */ \
656 { \
657 lua_pop(L, 1); \
658 o = luaV_new##tname(L, obj); \
659 } \
660 else \
661 o = (luatyp *) lua_touserdata(L, -1); \
662 } \
663 return o; \
664 }
665
666#define luaV_type_tostring(tname,luatname) \
667 static int \
668 luaV_##tname##_tostring (lua_State *L) \
669 { \
670 lua_pushfstring(L, "%s: %p", luatname, lua_touserdata(L, 1)); \
671 return 1; \
672 }
673
Bram Moolenaar1dced572012-04-05 16:54:08 +0200674/* ======= List type ======= */
675
676 static luaV_List *
677luaV_newlist (lua_State *L, list_T *lis)
678{
679 luaV_List *l = (luaV_List *) lua_newuserdata(L, sizeof(luaV_List));
680 *l = lis;
681 lis->lv_refcount++; /* reference in Lua */
682 luaV_setudata(L, lis); /* cache[lis] = udata */
683 luaV_getfield(L, LUAVIM_LIST);
684 lua_setmetatable(L, -2);
685 return l;
686}
687
688luaV_pushtype(list_T, list, luaV_List)
689luaV_type_tostring(list, LUAVIM_LIST)
690
691 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +0200692luaV_list_len (lua_State *L)
693{
694 list_T *l = luaV_unbox(L, luaV_List, 1);
695 lua_pushinteger(L, (l == NULL) ? 0 : (int) l->lv_len);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200696 return 1;
697}
698
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200699 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +0200700luaV_list_iter (lua_State *L)
701{
702 listitem_T *li = (listitem_T *) lua_touserdata(L, lua_upvalueindex(2));
703 if (li == NULL) return 0;
704 luaV_pushtypval(L, &li->li_tv);
705 lua_pushlightuserdata(L, (void *) li->li_next);
706 lua_replace(L, lua_upvalueindex(2));
707 return 1;
708}
709
710 static int
711luaV_list_call (lua_State *L)
712{
713 list_T *l = luaV_unbox(L, luaV_List, 1);
714 lua_pushvalue(L, lua_upvalueindex(1)); /* pass cache table along */
715 lua_pushlightuserdata(L, (void *) l->lv_first);
716 lua_pushcclosure(L, luaV_list_iter, 2);
717 return 1;
718}
719
720 static int
721luaV_list_index (lua_State *L)
722{
723 list_T *l = luaV_unbox(L, luaV_List, 1);
724 if (lua_isnumber(L, 2)) /* list item? */
725 {
726 listitem_T *li = list_find(l, (long) luaL_checkinteger(L, 2));
727 if (li == NULL)
728 lua_pushnil(L);
729 else
730 luaV_pushtypval(L, &li->li_tv);
731 }
732 else if (lua_isstring(L, 2)) /* method? */
733 {
734 const char *s = lua_tostring(L, 2);
735 if (strncmp(s, "add", 3) == 0
Bram Moolenaarb3766472013-04-15 13:49:21 +0200736 || strncmp(s, "insert", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200737 {
738 lua_getmetatable(L, 1);
739 lua_getfield(L, -1, s);
740 }
741 else
742 lua_pushnil(L);
743 }
744 else
745 lua_pushnil(L);
746 return 1;
747}
748
749 static int
750luaV_list_newindex (lua_State *L)
751{
752 list_T *l = luaV_unbox(L, luaV_List, 1);
753 long n = (long) luaL_checkinteger(L, 2);
754 listitem_T *li;
755 if (l->lv_lock)
756 luaL_error(L, "list is locked");
757 li = list_find(l, n);
758 if (li == NULL) return 0;
759 if (lua_isnil(L, 3)) /* remove? */
760 {
Bram Moolenaar3ec7f4e2014-05-07 17:31:37 +0200761 vimlist_remove(l, li, li);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200762 clear_tv(&li->li_tv);
763 vim_free(li);
764 }
765 else
766 {
767 typval_T v;
768 luaV_totypval(L, 3, &v);
769 clear_tv(&li->li_tv);
770 copy_tv(&v, &li->li_tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200771 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200772 }
773 return 0;
774}
775
776 static int
777luaV_list_add (lua_State *L)
778{
779 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
780 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200781 typval_T v;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200782 if (l->lv_lock)
783 luaL_error(L, "list is locked");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200784 lua_settop(L, 2);
785 luaV_totypval(L, 2, &v);
786 if (list_append_tv(l, &v) == FAIL)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200787 {
Bram Moolenaarb3766472013-04-15 13:49:21 +0200788 clear_tv(&v);
789 luaL_error(L, "Failed to add item to list");
Bram Moolenaar1dced572012-04-05 16:54:08 +0200790 }
Bram Moolenaarb3766472013-04-15 13:49:21 +0200791 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200792 lua_settop(L, 1);
793 return 1;
794}
795
796 static int
797luaV_list_insert (lua_State *L)
798{
799 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
800 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaar46538ee2015-02-17 16:28:55 +0100801 long pos = (long) luaL_optinteger(L, 3, 0);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200802 listitem_T *li = NULL;
803 typval_T v;
804 if (l->lv_lock)
805 luaL_error(L, "list is locked");
806 if (pos < l->lv_len)
807 {
808 li = list_find(l, pos);
809 if (li == NULL)
810 luaL_error(L, "invalid position");
811 }
812 lua_settop(L, 2);
813 luaV_totypval(L, 2, &v);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200814 if (list_insert_tv(l, &v, li) == FAIL)
815 {
816 clear_tv(&v);
817 luaL_error(L, "Failed to add item to list");
818 }
819 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200820 lua_settop(L, 1);
821 return 1;
822}
823
824static const luaL_Reg luaV_List_mt[] = {
825 {"__tostring", luaV_list_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200826 {"__len", luaV_list_len},
827 {"__call", luaV_list_call},
828 {"__index", luaV_list_index},
829 {"__newindex", luaV_list_newindex},
830 {"add", luaV_list_add},
831 {"insert", luaV_list_insert},
832 {NULL, NULL}
833};
834
835
836/* ======= Dict type ======= */
837
838 static luaV_Dict *
839luaV_newdict (lua_State *L, dict_T *dic)
840{
841 luaV_Dict *d = (luaV_Dict *) lua_newuserdata(L, sizeof(luaV_Dict));
842 *d = dic;
843 dic->dv_refcount++; /* reference in Lua */
844 luaV_setudata(L, dic); /* cache[dic] = udata */
845 luaV_getfield(L, LUAVIM_DICT);
846 lua_setmetatable(L, -2);
847 return d;
848}
849
850luaV_pushtype(dict_T, dict, luaV_Dict)
851luaV_type_tostring(dict, LUAVIM_DICT)
852
853 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +0200854luaV_dict_len (lua_State *L)
855{
856 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
857 lua_pushinteger(L, (d == NULL) ? 0 : (int) d->dv_hashtab.ht_used);
858 return 1;
859}
860
861 static int
Bram Moolenaarfeeaa682013-02-14 22:19:51 +0100862luaV_dict_iter (lua_State *L UNUSED)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200863{
Bram Moolenaarfeeaa682013-02-14 22:19:51 +0100864#ifdef FEAT_EVAL
Bram Moolenaar1dced572012-04-05 16:54:08 +0200865 hashitem_T *hi = (hashitem_T *) lua_touserdata(L, lua_upvalueindex(2));
866 int n = lua_tointeger(L, lua_upvalueindex(3));
867 dictitem_T *di;
868 if (n <= 0) return 0;
869 while (HASHITEM_EMPTY(hi)) hi++;
870 di = dict_lookup(hi);
871 lua_pushstring(L, (char *) hi->hi_key);
872 luaV_pushtypval(L, &di->di_tv);
873 lua_pushlightuserdata(L, (void *) (hi + 1));
874 lua_replace(L, lua_upvalueindex(2));
875 lua_pushinteger(L, n - 1);
876 lua_replace(L, lua_upvalueindex(3));
877 return 2;
Bram Moolenaarfeeaa682013-02-14 22:19:51 +0100878#else
879 return 0;
880#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +0200881}
882
883 static int
884luaV_dict_call (lua_State *L)
885{
886 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
887 hashtab_T *ht = &d->dv_hashtab;
888 lua_pushvalue(L, lua_upvalueindex(1)); /* pass cache table along */
889 lua_pushlightuserdata(L, (void *) ht->ht_array);
890 lua_pushinteger(L, ht->ht_used); /* # remaining items */
891 lua_pushcclosure(L, luaV_dict_iter, 3);
892 return 1;
893}
894
895 static int
896luaV_dict_index (lua_State *L)
897{
898 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
899 char_u *key = (char_u *) luaL_checkstring(L, 2);
900 dictitem_T *di = dict_find(d, key, -1);
901 if (di == NULL)
902 lua_pushnil(L);
903 else
904 luaV_pushtypval(L, &di->di_tv);
905 return 1;
906}
907
908 static int
909luaV_dict_newindex (lua_State *L)
910{
911 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
912 char_u *key = (char_u *) luaL_checkstring(L, 2);
913 dictitem_T *di;
914 if (d->dv_lock)
915 luaL_error(L, "dict is locked");
916 di = dict_find(d, key, -1);
917 if (di == NULL) /* non-existing key? */
918 {
919 if (lua_isnil(L, 3)) return 0;
920 di = dictitem_alloc(key);
921 if (di == NULL) return 0;
922 if (dict_add(d, di) == FAIL)
923 {
924 vim_free(di);
925 return 0;
926 }
927 }
928 else
929 clear_tv(&di->di_tv);
930 if (lua_isnil(L, 3)) /* remove? */
931 {
932 hashitem_T *hi = hash_find(&d->dv_hashtab, di->di_key);
933 hash_remove(&d->dv_hashtab, hi);
934 dictitem_free(di);
935 }
936 else {
937 typval_T v;
938 luaV_totypval(L, 3, &v);
939 copy_tv(&v, &di->di_tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200940 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200941 }
942 return 0;
943}
944
945static const luaL_Reg luaV_Dict_mt[] = {
946 {"__tostring", luaV_dict_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200947 {"__len", luaV_dict_len},
948 {"__call", luaV_dict_call},
949 {"__index", luaV_dict_index},
950 {"__newindex", luaV_dict_newindex},
951 {NULL, NULL}
952};
953
954
955/* ======= Buffer type ======= */
956
957luaV_newtype(buf_T, buffer, luaV_Buffer, LUAVIM_BUFFER)
958luaV_pushtype(buf_T, buffer, luaV_Buffer)
959luaV_type_tostring(buffer, LUAVIM_BUFFER)
960
961 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200962luaV_buffer_len(lua_State *L)
963{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200964 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
965 lua_pushinteger(L, b->b_ml.ml_line_count);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200966 return 1;
967}
968
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200969 static int
970luaV_buffer_call(lua_State *L)
971{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200972 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200973 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200974 set_curbuf(b, DOBUF_SPLIT);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200975 return 1;
976}
977
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200978 static int
979luaV_buffer_index(lua_State *L)
980{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200981 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200982 linenr_T n = (linenr_T) lua_tointeger(L, 2);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200983 if (n > 0 && n <= b->b_ml.ml_line_count)
984 luaV_pushline(L, b, n);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200985 else if (lua_isstring(L, 2))
986 {
987 const char *s = lua_tostring(L, 2);
988 if (strncmp(s, "name", 4) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200989 lua_pushstring(L, (char *) b->b_sfname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200990 else if (strncmp(s, "fname", 5) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200991 lua_pushstring(L, (char *) b->b_ffname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200992 else if (strncmp(s, "number", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200993 lua_pushinteger(L, b->b_fnum);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200994 /* methods */
995 else if (strncmp(s, "insert", 6) == 0
996 || strncmp(s, "next", 4) == 0
997 || strncmp(s, "previous", 8) == 0
998 || strncmp(s, "isvalid", 7) == 0)
999 {
1000 lua_getmetatable(L, 1);
1001 lua_getfield(L, -1, s);
1002 }
1003 else
1004 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001005 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001006 else
1007 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001008 return 1;
1009}
1010
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001011 static int
1012luaV_buffer_newindex(lua_State *L)
1013{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001014 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001015 linenr_T n = (linenr_T) luaL_checkinteger(L, 2);
1016#ifdef HAVE_SANDBOX
1017 luaV_checksandbox(L);
1018#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001019 if (n < 1 || n > b->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001020 luaL_error(L, "invalid line number");
1021 if (lua_isnil(L, 3)) /* delete line */
1022 {
1023 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001024 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001025 if (u_savedel(n, 1L) == FAIL)
1026 {
1027 curbuf = buf;
1028 luaL_error(L, "cannot save undo information");
1029 }
1030 else if (ml_delete(n, FALSE) == FAIL)
1031 {
1032 curbuf = buf;
1033 luaL_error(L, "cannot delete line");
1034 }
1035 else {
1036 deleted_lines_mark(n, 1L);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001037 if (b == curwin->w_buffer) /* fix cursor in current window? */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001038 {
1039 if (curwin->w_cursor.lnum >= n)
1040 {
1041 if (curwin->w_cursor.lnum > n)
1042 {
1043 curwin->w_cursor.lnum -= 1;
1044 check_cursor_col();
1045 }
1046 else check_cursor();
1047 changed_cline_bef_curs();
1048 }
1049 invalidate_botline();
1050 }
1051 }
1052 curbuf = buf;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001053 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001054 else if (lua_isstring(L, 3)) /* update line */
1055 {
1056 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001057 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001058 if (u_savesub(n) == FAIL)
1059 {
1060 curbuf = buf;
1061 luaL_error(L, "cannot save undo information");
1062 }
1063 else if (ml_replace(n, luaV_toline(L, 3), TRUE) == FAIL)
1064 {
1065 curbuf = buf;
1066 luaL_error(L, "cannot replace line");
1067 }
1068 else changed_bytes(n, 0);
1069 curbuf = buf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001070 if (b == curwin->w_buffer)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001071 check_cursor_col();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001072 }
1073 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001074 luaL_error(L, "wrong argument to change line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001075 return 0;
1076}
1077
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001078 static int
1079luaV_buffer_insert(lua_State *L)
1080{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001081 luaV_Buffer *lb = luaV_checkudata(L, 1, LUAVIM_BUFFER);
1082 buf_T *b = (buf_T *) luaV_checkcache(L, (void *) *lb);
1083 linenr_T last = b->b_ml.ml_line_count;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001084 linenr_T n = (linenr_T) luaL_optinteger(L, 3, last);
1085 buf_T *buf;
1086 luaL_checktype(L, 2, LUA_TSTRING);
1087#ifdef HAVE_SANDBOX
1088 luaV_checksandbox(L);
1089#endif
1090 /* fix insertion line */
1091 if (n < 0) n = 0;
1092 if (n > last) n = last;
1093 /* insert */
1094 buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001095 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001096 if (u_save(n, n + 1) == FAIL)
1097 {
1098 curbuf = buf;
1099 luaL_error(L, "cannot save undo information");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001100 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001101 else if (ml_append(n, luaV_toline(L, 2), 0, FALSE) == FAIL)
1102 {
1103 curbuf = buf;
1104 luaL_error(L, "cannot insert line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001105 }
1106 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001107 appended_lines_mark(n, 1L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001108 curbuf = buf;
1109 update_screen(VALID);
1110 return 0;
1111}
1112
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001113 static int
1114luaV_buffer_next(lua_State *L)
1115{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001116 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001117 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1118 luaV_pushbuffer(L, buf->b_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001119 return 1;
1120}
1121
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001122 static int
1123luaV_buffer_previous(lua_State *L)
1124{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001125 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001126 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1127 luaV_pushbuffer(L, buf->b_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001128 return 1;
1129}
1130
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001131 static int
1132luaV_buffer_isvalid(lua_State *L)
1133{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001134 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001135 luaV_getudata(L, *b);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001136 lua_pushboolean(L, !lua_isnil(L, -1));
1137 return 1;
1138}
1139
1140static const luaL_Reg luaV_Buffer_mt[] = {
1141 {"__tostring", luaV_buffer_tostring},
1142 {"__len", luaV_buffer_len},
1143 {"__call", luaV_buffer_call},
1144 {"__index", luaV_buffer_index},
1145 {"__newindex", luaV_buffer_newindex},
1146 {"insert", luaV_buffer_insert},
1147 {"next", luaV_buffer_next},
1148 {"previous", luaV_buffer_previous},
1149 {"isvalid", luaV_buffer_isvalid},
1150 {NULL, NULL}
1151};
1152
1153
1154/* ======= Window type ======= */
1155
Bram Moolenaar1dced572012-04-05 16:54:08 +02001156luaV_newtype(win_T, window, luaV_Window, LUAVIM_WINDOW)
1157luaV_pushtype(win_T, window, luaV_Window)
1158luaV_type_tostring(window, LUAVIM_WINDOW)
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001159
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001160 static int
1161luaV_window_call(lua_State *L)
1162{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001163 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001164 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001165 win_goto(w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001166 return 1;
1167}
1168
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001169 static int
1170luaV_window_index(lua_State *L)
1171{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001172 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001173 const char *s = luaL_checkstring(L, 2);
1174 if (strncmp(s, "buffer", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001175 luaV_pushbuffer(L, w->w_buffer);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001176 else if (strncmp(s, "line", 4) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001177 lua_pushinteger(L, w->w_cursor.lnum);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001178 else if (strncmp(s, "col", 3) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001179 lua_pushinteger(L, w->w_cursor.col + 1);
Bram Moolenaar44a2f922016-03-19 22:11:51 +01001180#ifdef FEAT_WINDOWS
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001181 else if (strncmp(s, "width", 5) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001182 lua_pushinteger(L, W_WIDTH(w));
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001183#endif
1184 else if (strncmp(s, "height", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001185 lua_pushinteger(L, w->w_height);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001186 /* methods */
1187 else if (strncmp(s, "next", 4) == 0
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001188 || strncmp(s, "previous", 8) == 0
1189 || strncmp(s, "isvalid", 7) == 0)
1190 {
1191 lua_getmetatable(L, 1);
1192 lua_getfield(L, -1, s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001193 }
1194 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001195 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001196 return 1;
1197}
1198
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001199 static int
1200luaV_window_newindex (lua_State *L)
1201{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001202 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001203 const char *s = luaL_checkstring(L, 2);
1204 int v = luaL_checkinteger(L, 3);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001205 if (strncmp(s, "line", 4) == 0)
1206 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001207#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001208 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001209#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001210 if (v < 1 || v > w->w_buffer->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001211 luaL_error(L, "line out of range");
Bram Moolenaar1dced572012-04-05 16:54:08 +02001212 w->w_cursor.lnum = v;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001213 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001214 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001215 else if (strncmp(s, "col", 3) == 0)
1216 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001217#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001218 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001219#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001220 w->w_cursor.col = v - 1;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001221 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001222 }
Bram Moolenaar44a2f922016-03-19 22:11:51 +01001223#ifdef FEAT_WINDOWS
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001224 else if (strncmp(s, "width", 5) == 0)
1225 {
1226 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001227#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001228 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001229#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001230 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001231 win_setwidth(v);
1232 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001233 }
1234#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001235 else if (strncmp(s, "height", 6) == 0)
1236 {
1237 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001238#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001239 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001240#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001241 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001242 win_setheight(v);
1243 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001244 }
1245 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001246 luaL_error(L, "invalid window property: `%s'", s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001247 return 0;
1248}
1249
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001250 static int
1251luaV_window_next(lua_State *L)
1252{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001253 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001254 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1255 luaV_pushwindow(L, win->w_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001256 return 1;
1257}
1258
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001259 static int
1260luaV_window_previous(lua_State *L)
1261{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001262 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001263 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1264 luaV_pushwindow(L, win->w_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001265 return 1;
1266}
1267
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001268 static int
1269luaV_window_isvalid(lua_State *L)
1270{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001271 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001272 luaV_getudata(L, *w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001273 lua_pushboolean(L, !lua_isnil(L, -1));
1274 return 1;
1275}
1276
1277static const luaL_Reg luaV_Window_mt[] = {
1278 {"__tostring", luaV_window_tostring},
1279 {"__call", luaV_window_call},
1280 {"__index", luaV_window_index},
1281 {"__newindex", luaV_window_newindex},
1282 {"next", luaV_window_next},
1283 {"previous", luaV_window_previous},
1284 {"isvalid", luaV_window_isvalid},
1285 {NULL, NULL}
1286};
1287
1288
1289/* ======= Vim module ======= */
1290
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001291 static int
1292luaV_print(lua_State *L)
1293{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001294 int i, n = lua_gettop(L); /* nargs */
1295 const char *s;
1296 size_t l;
1297 luaL_Buffer b;
1298 luaL_buffinit(L, &b);
1299 lua_getglobal(L, "tostring");
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001300 for (i = 1; i <= n; i++)
1301 {
1302 lua_pushvalue(L, -1); /* tostring */
1303 lua_pushvalue(L, i); /* arg */
1304 lua_call(L, 1, 1);
1305 s = lua_tolstring(L, -1, &l);
1306 if (s == NULL)
1307 return luaL_error(L, "cannot convert to string");
1308 if (i > 1) luaL_addchar(&b, ' '); /* use space instead of tab */
1309 luaV_addlstring(&b, s, l, 0);
1310 lua_pop(L, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001311 }
1312 luaL_pushresult(&b);
1313 luaV_msg(L);
1314 return 0;
1315}
1316
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001317 static int
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001318luaV_debug(lua_State *L)
1319{
1320 lua_settop(L, 0);
1321 lua_getglobal(L, "vim");
1322 lua_getfield(L, -1, "eval");
1323 lua_remove(L, -2); /* vim.eval at position 1 */
1324 for (;;)
1325 {
1326 const char *input;
1327 size_t l;
1328 lua_pushvalue(L, 1); /* vim.eval */
1329 lua_pushliteral(L, "input('lua_debug> ')");
1330 lua_call(L, 1, 1); /* return string */
1331 input = lua_tolstring(L, -1, &l);
1332 if (l == 0 || strcmp(input, "cont") == 0)
1333 return 0;
1334 msg_putchar('\n'); /* avoid outputting on input line */
1335 if (luaL_loadbuffer(L, input, l, "=(debug command)")
1336 || lua_pcall(L, 0, 0, 0))
1337 luaV_emsg(L);
1338 lua_settop(L, 1); /* remove eventual returns, but keep vim.eval */
1339 }
1340}
1341
1342 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001343luaV_command(lua_State *L)
1344{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001345 do_cmdline_cmd((char_u *) luaL_checkstring(L, 1));
1346 update_screen(VALID);
1347 return 0;
1348}
1349
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001350 static int
1351luaV_eval(lua_State *L)
1352{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001353 typval_T *tv = eval_expr((char_u *) luaL_checkstring(L, 1), NULL);
1354 if (tv == NULL) luaL_error(L, "invalid expression");
1355 luaV_pushtypval(L, tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +02001356 free_tv(tv);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001357 return 1;
1358}
1359
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001360 static int
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001361luaV_beep(lua_State *L UNUSED)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001362{
Bram Moolenaar165bc692015-07-21 17:53:25 +02001363 vim_beep(BO_LANG);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001364 return 0;
1365}
1366
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001367 static int
1368luaV_line(lua_State *L)
1369{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001370 luaV_pushline(L, curbuf, curwin->w_cursor.lnum);
1371 return 1;
1372}
1373
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001374 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001375luaV_list(lua_State *L)
1376{
1377 list_T *l = list_alloc();
1378 if (l == NULL)
1379 lua_pushnil(L);
1380 else
1381 luaV_newlist(L, l);
1382 return 1;
1383}
1384
1385 static int
1386luaV_dict(lua_State *L)
1387{
1388 dict_T *d = dict_alloc();
1389 if (d == NULL)
1390 lua_pushnil(L);
1391 else
1392 luaV_newdict(L, d);
1393 return 1;
1394}
1395
1396 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001397luaV_buffer(lua_State *L)
1398{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001399 buf_T *buf;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001400 if (lua_isstring(L, 1)) /* get by number or name? */
1401 {
1402 if (lua_isnumber(L, 1)) /* by number? */
1403 {
1404 int n = lua_tointeger(L, 1);
Bram Moolenaar29323592016-07-24 22:04:11 +02001405 FOR_ALL_BUFFERS(buf)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001406 if (buf->b_fnum == n) break;
1407 }
1408 else { /* by name */
1409 size_t l;
1410 const char *s = lua_tolstring(L, 1, &l);
Bram Moolenaar29323592016-07-24 22:04:11 +02001411 FOR_ALL_BUFFERS(buf)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001412 {
1413 if (buf->b_ffname == NULL || buf->b_sfname == NULL)
1414 {
1415 if (l == 0) break;
1416 }
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001417 else if (strncmp(s, (char *)buf->b_ffname, l) == 0
1418 || strncmp(s, (char *)buf->b_sfname, l) == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001419 break;
1420 }
1421 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001422 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001423 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001424 buf = (lua_toboolean(L, 1)) ? firstbuf : curbuf; /* first buffer? */
Bram Moolenaar1dced572012-04-05 16:54:08 +02001425 luaV_pushbuffer(L, buf);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001426 return 1;
1427}
1428
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001429 static int
1430luaV_window(lua_State *L)
1431{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001432 win_T *win;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001433 if (lua_isnumber(L, 1)) /* get by number? */
1434 {
1435 int n = lua_tointeger(L, 1);
1436 for (win = firstwin; win != NULL; win = win->w_next, n--)
1437 if (n == 1) break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001438 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001439 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001440 win = (lua_toboolean(L, 1)) ? firstwin : curwin; /* first window? */
Bram Moolenaar1dced572012-04-05 16:54:08 +02001441 luaV_pushwindow(L, win);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001442 return 1;
1443}
1444
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001445 static int
1446luaV_open(lua_State *L)
1447{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001448 char_u *s = NULL;
1449#ifdef HAVE_SANDBOX
1450 luaV_checksandbox(L);
1451#endif
1452 if (lua_isstring(L, 1)) s = (char_u *) lua_tostring(L, 1);
Bram Moolenaarfa263a52011-12-08 16:00:16 +01001453 luaV_pushbuffer(L, buflist_new(s, NULL, 1L, BLN_LISTED));
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001454 return 1;
1455}
1456
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001457 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001458luaV_type(lua_State *L)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001459{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001460 luaL_checkany(L, 1);
1461 if (lua_type(L, 1) == LUA_TUSERDATA) /* check vim udata? */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001462 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001463 lua_settop(L, 1);
1464 if (lua_getmetatable(L, 1))
1465 {
1466 luaV_getfield(L, LUAVIM_LIST);
1467 if (lua_rawequal(L, -1, 2))
1468 {
1469 lua_pushstring(L, "list");
1470 return 1;
1471 }
1472 luaV_getfield(L, LUAVIM_DICT);
1473 if (lua_rawequal(L, -1, 2))
1474 {
1475 lua_pushstring(L, "dict");
1476 return 1;
1477 }
1478 luaV_getfield(L, LUAVIM_BUFFER);
1479 if (lua_rawequal(L, -1, 2))
1480 {
1481 lua_pushstring(L, "buffer");
1482 return 1;
1483 }
1484 luaV_getfield(L, LUAVIM_WINDOW);
1485 if (lua_rawequal(L, -1, 2))
1486 {
1487 lua_pushstring(L, "window");
1488 return 1;
1489 }
1490 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001491 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001492 lua_pushstring(L, luaL_typename(L, 1)); /* fallback */
1493 return 1;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001494}
1495
1496static const luaL_Reg luaV_module[] = {
1497 {"command", luaV_command},
1498 {"eval", luaV_eval},
1499 {"beep", luaV_beep},
1500 {"line", luaV_line},
Bram Moolenaar1dced572012-04-05 16:54:08 +02001501 {"list", luaV_list},
1502 {"dict", luaV_dict},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001503 {"buffer", luaV_buffer},
1504 {"window", luaV_window},
1505 {"open", luaV_open},
Bram Moolenaar1dced572012-04-05 16:54:08 +02001506 {"type", luaV_type},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001507 {NULL, NULL}
1508};
1509
Bram Moolenaar1dced572012-04-05 16:54:08 +02001510/* for freeing list, dict, buffer and window objects; lightuserdata as arg */
1511 static int
1512luaV_free(lua_State *L)
1513{
1514 lua_pushnil(L);
1515 luaV_setudata(L, lua_touserdata(L, 1));
1516 return 0;
1517}
1518
1519 static int
1520luaV_luaeval (lua_State *L)
1521{
1522 luaL_Buffer b;
1523 size_t l;
1524 const char *str = lua_tolstring(L, 1, &l);
1525 typval_T *arg = (typval_T *) lua_touserdata(L, 2);
1526 typval_T *rettv = (typval_T *) lua_touserdata(L, 3);
1527 luaL_buffinit(L, &b);
1528 luaL_addlstring(&b, LUAVIM_EVALHEADER, sizeof(LUAVIM_EVALHEADER) - 1);
1529 luaL_addlstring(&b, str, l);
1530 luaL_pushresult(&b);
1531 str = lua_tolstring(L, -1, &l);
1532 if (luaL_loadbuffer(L, str, l, LUAVIM_EVALNAME)) /* compile error? */
1533 {
1534 luaV_emsg(L);
1535 return 0;
1536 }
1537 luaV_pushtypval(L, arg);
1538 if (lua_pcall(L, 1, 1, 0)) /* running error? */
1539 {
1540 luaV_emsg(L);
1541 return 0;
1542 }
1543 luaV_totypval(L, -1, rettv);
Bram Moolenaarf554a322015-02-04 23:08:01 +01001544 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001545}
1546
1547 static int
1548luaV_setref (lua_State *L)
1549{
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001550 int copyID = lua_tointeger(L, 1);
1551 int abort = FALSE;
1552 typval_T tv;
1553
Bram Moolenaar1dced572012-04-05 16:54:08 +02001554 luaV_getfield(L, LUAVIM_LIST);
1555 luaV_getfield(L, LUAVIM_DICT);
1556 lua_pushnil(L);
Bram Moolenaarb84634d2015-02-04 22:02:37 +01001557 /* traverse cache table */
1558 while (!abort && lua_next(L, lua_upvalueindex(1)) != 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001559 {
1560 lua_getmetatable(L, -1);
1561 if (lua_rawequal(L, -1, 2)) /* list? */
1562 {
1563 tv.v_type = VAR_LIST;
1564 tv.vval.v_list = (list_T *) lua_touserdata(L, 4); /* key */
Bram Moolenaarf609dcf2015-12-03 17:43:17 +01001565 abort = set_ref_in_item(&tv, copyID, NULL, NULL);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001566 }
1567 else if (lua_rawequal(L, -1, 3)) /* dict? */
1568 {
1569 tv.v_type = VAR_DICT;
1570 tv.vval.v_dict = (dict_T *) lua_touserdata(L, 4); /* key */
Bram Moolenaarf609dcf2015-12-03 17:43:17 +01001571 abort = set_ref_in_item(&tv, copyID, NULL, NULL);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001572 }
1573 lua_pop(L, 2); /* metatable and value */
Bram Moolenaar1dced572012-04-05 16:54:08 +02001574 }
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001575 lua_pushinteger(L, abort);
Bram Moolenaarf554a322015-02-04 23:08:01 +01001576 return 1;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001577}
1578
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001579 static int
1580luaopen_vim(lua_State *L)
1581{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001582 /* set cache table */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001583 lua_newtable(L);
1584 lua_newtable(L);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001585 lua_pushstring(L, "v");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001586 lua_setfield(L, -2, "__mode");
Bram Moolenaar1dced572012-04-05 16:54:08 +02001587 lua_setmetatable(L, -2); /* cache is weak-valued */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001588 /* print */
1589 lua_pushcfunction(L, luaV_print);
1590 lua_setglobal(L, "print");
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001591 /* debug.debug */
1592 lua_getglobal(L, "debug");
1593 lua_pushcfunction(L, luaV_debug);
1594 lua_setfield(L, -2, "debug");
1595 lua_pop(L, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001596 /* free */
1597 lua_pushlightuserdata(L, (void *) LUAVIM_FREE);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001598 lua_pushvalue(L, 1); /* cache table */
1599 lua_pushcclosure(L, luaV_free, 1);
1600 lua_rawset(L, LUA_REGISTRYINDEX);
1601 /* luaeval */
1602 lua_pushlightuserdata(L, (void *) LUAVIM_LUAEVAL);
1603 lua_pushvalue(L, 1); /* cache table */
1604 lua_pushcclosure(L, luaV_luaeval, 1);
1605 lua_rawset(L, LUA_REGISTRYINDEX);
1606 /* setref */
1607 lua_pushlightuserdata(L, (void *) LUAVIM_SETREF);
1608 lua_pushvalue(L, 1); /* cache table */
1609 lua_pushcclosure(L, luaV_setref, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001610 lua_rawset(L, LUA_REGISTRYINDEX);
1611 /* register */
Bram Moolenaar1dced572012-04-05 16:54:08 +02001612 luaV_newmetatable(L, LUAVIM_LIST);
1613 lua_pushvalue(L, 1);
1614 luaV_openlib(L, luaV_List_mt, 1);
1615 luaV_newmetatable(L, LUAVIM_DICT);
1616 lua_pushvalue(L, 1);
1617 luaV_openlib(L, luaV_Dict_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001618 luaV_newmetatable(L, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001619 lua_pushvalue(L, 1); /* cache table */
1620 luaV_openlib(L, luaV_Buffer_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001621 luaV_newmetatable(L, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001622 lua_pushvalue(L, 1); /* cache table */
1623 luaV_openlib(L, luaV_Window_mt, 1);
1624 lua_newtable(L); /* vim table */
1625 lua_pushvalue(L, 1); /* cache table */
1626 luaV_openlib(L, luaV_module, 1);
1627 lua_setglobal(L, LUAVIM_NAME);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001628 return 0;
1629}
1630
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001631 static lua_State *
1632luaV_newstate(void)
1633{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001634 lua_State *L = luaL_newstate();
Bram Moolenaar16c98f92010-07-28 22:46:08 +02001635 luaL_openlibs(L); /* core libs */
1636 lua_pushcfunction(L, luaopen_vim); /* vim */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001637 lua_call(L, 0, 0);
1638 return L;
1639}
1640
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001641 static void
1642luaV_setrange(lua_State *L, int line1, int line2)
1643{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001644 lua_getglobal(L, LUAVIM_NAME);
1645 lua_pushinteger(L, line1);
1646 lua_setfield(L, -2, "firstline");
1647 lua_pushinteger(L, line2);
1648 lua_setfield(L, -2, "lastline");
1649 lua_pop(L, 1); /* vim table */
1650}
1651
1652
1653/* ======= Interface ======= */
1654
1655static lua_State *L = NULL;
1656
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001657 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001658lua_isopen(void)
Bram Moolenaar2bd6a1b2010-08-12 22:14:01 +02001659{
1660 return L != NULL;
1661}
1662
1663 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001664lua_init(void)
1665{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001666 if (!lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001667 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001668#ifdef DYNAMIC_LUA
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001669 if (!lua_enabled(TRUE))
1670 {
1671 EMSG(_("Lua library cannot be loaded."));
1672 return FAIL;
1673 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001674#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001675 L = luaV_newstate();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001676 }
1677 return OK;
1678}
1679
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001680 void
1681lua_end(void)
1682{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001683 if (lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001684 {
1685 lua_close(L);
1686 L = NULL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001687#ifdef DYNAMIC_LUA
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001688 end_dynamic_lua();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001689#endif
1690 }
1691}
1692
1693/* ex commands */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001694 void
1695ex_lua(exarg_T *eap)
1696{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001697 char *script;
1698 if (lua_init() == FAIL) return;
1699 script = (char *) script_get(eap, eap->arg);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001700 if (!eap->skip)
1701 {
1702 char *s = (script) ? script : (char *) eap->arg;
1703 luaV_setrange(L, eap->line1, eap->line2);
1704 if (luaL_loadbuffer(L, s, strlen(s), LUAVIM_CHUNKNAME)
1705 || lua_pcall(L, 0, 0, 0))
1706 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001707 }
1708 if (script != NULL) vim_free(script);
1709}
1710
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001711 void
1712ex_luado(exarg_T *eap)
1713{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001714 linenr_T l;
1715 const char *s = (const char *) eap->arg;
1716 luaL_Buffer b;
1717 size_t len;
1718 if (lua_init() == FAIL) return;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001719 if (u_save(eap->line1 - 1, eap->line2 + 1) == FAIL)
1720 {
1721 EMSG(_("cannot save undo information"));
1722 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001723 }
1724 luaV_setrange(L, eap->line1, eap->line2);
1725 luaL_buffinit(L, &b);
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +02001726 luaL_addlstring(&b, "return function(line, linenr) ", 30); /* header */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001727 luaL_addlstring(&b, s, strlen(s));
1728 luaL_addlstring(&b, " end", 4); /* footer */
1729 luaL_pushresult(&b);
1730 s = lua_tolstring(L, -1, &len);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001731 if (luaL_loadbuffer(L, s, len, LUAVIM_CHUNKNAME))
1732 {
1733 luaV_emsg(L);
1734 lua_pop(L, 1); /* function body */
1735 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001736 }
1737 lua_call(L, 0, 1);
1738 lua_replace(L, -2); /* function -> body */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001739 for (l = eap->line1; l <= eap->line2; l++)
1740 {
1741 lua_pushvalue(L, -1); /* function */
1742 luaV_pushline(L, curbuf, l); /* current line as arg */
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +02001743 lua_pushinteger(L, l); /* current line number as arg */
1744 if (lua_pcall(L, 2, 1, 0))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001745 {
1746 luaV_emsg(L);
1747 break;
1748 }
1749 if (lua_isstring(L, -1)) /* update line? */
1750 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001751#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001752 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001753#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001754 ml_replace(l, luaV_toline(L, -1), TRUE);
1755 changed_bytes(l, 0);
1756 lua_pop(L, 1); /* result from luaV_toline */
1757 }
1758 lua_pop(L, 1); /* line */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001759 }
1760 lua_pop(L, 1); /* function */
1761 check_cursor();
1762 update_screen(NOT_VALID);
1763}
1764
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001765 void
1766ex_luafile(exarg_T *eap)
1767{
1768 if (lua_init() == FAIL)
1769 return;
1770 if (!eap->skip)
1771 {
1772 luaV_setrange(L, eap->line1, eap->line2);
1773 if (luaL_loadfile(L, (char *) eap->arg) || lua_pcall(L, 0, 0, 0))
1774 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001775 }
1776}
1777
Bram Moolenaar1dced572012-04-05 16:54:08 +02001778#define luaV_freetype(typ,tname) \
1779 void \
1780 lua_##tname##_free(typ *o) \
1781 { \
1782 if (!lua_isopen()) return; \
1783 luaV_getfield(L, LUAVIM_FREE); \
1784 lua_pushlightuserdata(L, (void *) o); \
1785 lua_call(L, 1, 0); \
1786 }
1787
1788luaV_freetype(buf_T, buffer)
1789luaV_freetype(win_T, window)
1790
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001791 void
Bram Moolenaar1dced572012-04-05 16:54:08 +02001792do_luaeval (char_u *str, typval_T *arg, typval_T *rettv)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001793{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001794 lua_init();
1795 luaV_getfield(L, LUAVIM_LUAEVAL);
1796 lua_pushstring(L, (char *) str);
1797 lua_pushlightuserdata(L, (void *) arg);
1798 lua_pushlightuserdata(L, (void *) rettv);
1799 lua_call(L, 3, 0);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001800}
1801
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001802 int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001803set_ref_in_lua (int copyID)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001804{
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001805 int aborted = 0;
1806
1807 if (lua_isopen())
1808 {
1809 luaV_getfield(L, LUAVIM_SETREF);
1810 /* call the function with 1 arg, getting 1 result back */
1811 lua_pushinteger(L, copyID);
1812 lua_call(L, 1, 1);
1813 /* get the result */
1814 aborted = lua_tointeger(L, -1);
1815 /* pop result off the stack */
1816 lua_pop(L, 1);
1817 }
1818 return aborted;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001819}
1820
1821#endif