blob: 571dd622315309c5e0ea63aca909459906a9df29 [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 Moolenaar0ba04292010-07-14 23:23:17 +0200405 return lua_link_init(DYNAMIC_LUA_DLL, verbose) == OK;
406}
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;
502 default:
503 lua_pushnil(L);
504 }
505}
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200506
Bram Moolenaar1dced572012-04-05 16:54:08 +0200507/* converts lua value at 'pos' to typval 'tv' */
508 static void
509luaV_totypval (lua_State *L, int pos, typval_T *tv)
510{
511 switch(lua_type(L, pos)) {
512 case LUA_TBOOLEAN:
513 tv->v_type = VAR_NUMBER;
514 tv->vval.v_number = (varnumber_T) lua_toboolean(L, pos);
515 break;
516 case LUA_TSTRING:
517 tv->v_type = VAR_STRING;
518 tv->vval.v_string = vim_strsave((char_u *) lua_tostring(L, pos));
519 break;
520 case LUA_TNUMBER:
521#ifdef FEAT_FLOAT
522 tv->v_type = VAR_FLOAT;
523 tv->vval.v_float = (float_T) lua_tonumber(L, pos);
524#else
525 tv->v_type = VAR_NUMBER;
526 tv->vval.v_number = (varnumber_T) lua_tointeger(L, pos);
527#endif
528 break;
529 case LUA_TUSERDATA: {
530 void *p = lua_touserdata(L, pos);
531 if (lua_getmetatable(L, pos)) /* has metatable? */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200532 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200533 /* check list */
534 luaV_getfield(L, LUAVIM_LIST);
535 if (lua_rawequal(L, -1, -2))
Bram Moolenaar2334b6d2010-07-22 21:32:16 +0200536 {
Bram Moolenaar1dced572012-04-05 16:54:08 +0200537 tv->v_type = VAR_LIST;
538 tv->vval.v_list = *((luaV_List *) p);
539 ++tv->vval.v_list->lv_refcount;
540 lua_pop(L, 2); /* MTs */
541 return;
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200542 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200543 /* check dict */
544 luaV_getfield(L, LUAVIM_DICT);
545 if (lua_rawequal(L, -1, -3))
546 {
547 tv->v_type = VAR_DICT;
548 tv->vval.v_dict = *((luaV_Dict *) p);
549 ++tv->vval.v_dict->dv_refcount;
550 lua_pop(L, 3); /* MTs */
551 return;
552 }
553 lua_pop(L, 3); /* MTs */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200554 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200555 break;
556 }
557 default:
Bram Moolenaar1dced572012-04-05 16:54:08 +0200558 tv->v_type = VAR_NUMBER;
559 tv->vval.v_number = 0;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200560 }
561}
562
563/* similar to luaL_addlstring, but replaces \0 with \n if toline and
564 * \n with \0 otherwise */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200565 static void
566luaV_addlstring(luaL_Buffer *b, const char *s, size_t l, int toline)
567{
568 while (l--)
569 {
570 if (*s == '\0' && toline)
571 luaL_addchar(b, '\n');
572 else if (*s == '\n' && !toline)
573 luaL_addchar(b, '\0');
574 else
575 luaL_addchar(b, *s);
576 s++;
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200577 }
578}
579
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200580 static void
581luaV_pushline(lua_State *L, buf_T *buf, linenr_T n)
582{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200583 const char *s = (const char *) ml_get_buf(buf, n, FALSE);
584 luaL_Buffer b;
585 luaL_buffinit(L, &b);
586 luaV_addlstring(&b, s, strlen(s), 0);
587 luaL_pushresult(&b);
588}
589
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200590 static char_u *
591luaV_toline(lua_State *L, int pos)
592{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200593 size_t l;
594 const char *s = lua_tolstring(L, pos, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200595
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200596 luaL_Buffer b;
597 luaL_buffinit(L, &b);
598 luaV_addlstring(&b, s, l, 1);
599 luaL_pushresult(&b);
600 return (char_u *) lua_tostring(L, -1);
601}
602
603/* pops a string s from the top of the stack and calls mf(t) for pieces t of
604 * s separated by newlines */
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200605 static void
606luaV_msgfunc(lua_State *L, msgfunc_T mf)
607{
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200608 luaL_Buffer b;
609 size_t l;
610 const char *p, *s = lua_tolstring(L, -1, &l);
611 luaL_buffinit(L, &b);
612 luaV_addlstring(&b, s, l, 0);
613 luaL_pushresult(&b);
614 /* break string */
615 p = s = lua_tolstring(L, -1, &l);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200616 while (l--)
617 {
618 if (*p++ == '\0') /* break? */
619 {
620 mf((char_u *) s);
621 s = p;
622 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200623 }
624 mf((char_u *) s);
625 lua_pop(L, 2); /* original and modified strings */
626}
627
Bram Moolenaar1dced572012-04-05 16:54:08 +0200628#define luaV_newtype(typ,tname,luatyp,luatname) \
629 static luatyp * \
630 luaV_new##tname (lua_State *L, typ *obj) \
631 { \
632 luatyp *o = (luatyp *) lua_newuserdata(L, sizeof(luatyp)); \
633 *o = obj; \
634 luaV_setudata(L, obj); /* cache[obj] = udata */ \
635 luaV_getfield(L, luatname); \
636 lua_setmetatable(L, -2); \
637 return o; \
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200638 }
Bram Moolenaar1dced572012-04-05 16:54:08 +0200639
640#define luaV_pushtype(typ,tname,luatyp) \
641 static luatyp * \
642 luaV_push##tname (lua_State *L, typ *obj) \
643 { \
644 luatyp *o = NULL; \
645 if (obj == NULL) \
646 lua_pushnil(L); \
647 else { \
648 luaV_getudata(L, obj); \
649 if (lua_isnil(L, -1)) /* not interned? */ \
650 { \
651 lua_pop(L, 1); \
652 o = luaV_new##tname(L, obj); \
653 } \
654 else \
655 o = (luatyp *) lua_touserdata(L, -1); \
656 } \
657 return o; \
658 }
659
660#define luaV_type_tostring(tname,luatname) \
661 static int \
662 luaV_##tname##_tostring (lua_State *L) \
663 { \
664 lua_pushfstring(L, "%s: %p", luatname, lua_touserdata(L, 1)); \
665 return 1; \
666 }
667
Bram Moolenaar1dced572012-04-05 16:54:08 +0200668/* ======= List type ======= */
669
670 static luaV_List *
671luaV_newlist (lua_State *L, list_T *lis)
672{
673 luaV_List *l = (luaV_List *) lua_newuserdata(L, sizeof(luaV_List));
674 *l = lis;
675 lis->lv_refcount++; /* reference in Lua */
676 luaV_setudata(L, lis); /* cache[lis] = udata */
677 luaV_getfield(L, LUAVIM_LIST);
678 lua_setmetatable(L, -2);
679 return l;
680}
681
682luaV_pushtype(list_T, list, luaV_List)
683luaV_type_tostring(list, LUAVIM_LIST)
684
685 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +0200686luaV_list_len (lua_State *L)
687{
688 list_T *l = luaV_unbox(L, luaV_List, 1);
689 lua_pushinteger(L, (l == NULL) ? 0 : (int) l->lv_len);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200690 return 1;
691}
692
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200693 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +0200694luaV_list_iter (lua_State *L)
695{
696 listitem_T *li = (listitem_T *) lua_touserdata(L, lua_upvalueindex(2));
697 if (li == NULL) return 0;
698 luaV_pushtypval(L, &li->li_tv);
699 lua_pushlightuserdata(L, (void *) li->li_next);
700 lua_replace(L, lua_upvalueindex(2));
701 return 1;
702}
703
704 static int
705luaV_list_call (lua_State *L)
706{
707 list_T *l = luaV_unbox(L, luaV_List, 1);
708 lua_pushvalue(L, lua_upvalueindex(1)); /* pass cache table along */
709 lua_pushlightuserdata(L, (void *) l->lv_first);
710 lua_pushcclosure(L, luaV_list_iter, 2);
711 return 1;
712}
713
714 static int
715luaV_list_index (lua_State *L)
716{
717 list_T *l = luaV_unbox(L, luaV_List, 1);
718 if (lua_isnumber(L, 2)) /* list item? */
719 {
720 listitem_T *li = list_find(l, (long) luaL_checkinteger(L, 2));
721 if (li == NULL)
722 lua_pushnil(L);
723 else
724 luaV_pushtypval(L, &li->li_tv);
725 }
726 else if (lua_isstring(L, 2)) /* method? */
727 {
728 const char *s = lua_tostring(L, 2);
729 if (strncmp(s, "add", 3) == 0
Bram Moolenaarb3766472013-04-15 13:49:21 +0200730 || strncmp(s, "insert", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200731 {
732 lua_getmetatable(L, 1);
733 lua_getfield(L, -1, s);
734 }
735 else
736 lua_pushnil(L);
737 }
738 else
739 lua_pushnil(L);
740 return 1;
741}
742
743 static int
744luaV_list_newindex (lua_State *L)
745{
746 list_T *l = luaV_unbox(L, luaV_List, 1);
747 long n = (long) luaL_checkinteger(L, 2);
748 listitem_T *li;
749 if (l->lv_lock)
750 luaL_error(L, "list is locked");
751 li = list_find(l, n);
752 if (li == NULL) return 0;
753 if (lua_isnil(L, 3)) /* remove? */
754 {
Bram Moolenaar3ec7f4e2014-05-07 17:31:37 +0200755 vimlist_remove(l, li, li);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200756 clear_tv(&li->li_tv);
757 vim_free(li);
758 }
759 else
760 {
761 typval_T v;
762 luaV_totypval(L, 3, &v);
763 clear_tv(&li->li_tv);
764 copy_tv(&v, &li->li_tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200765 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200766 }
767 return 0;
768}
769
770 static int
771luaV_list_add (lua_State *L)
772{
773 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
774 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200775 typval_T v;
Bram Moolenaar1dced572012-04-05 16:54:08 +0200776 if (l->lv_lock)
777 luaL_error(L, "list is locked");
Bram Moolenaarb3766472013-04-15 13:49:21 +0200778 lua_settop(L, 2);
779 luaV_totypval(L, 2, &v);
780 if (list_append_tv(l, &v) == FAIL)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200781 {
Bram Moolenaarb3766472013-04-15 13:49:21 +0200782 clear_tv(&v);
783 luaL_error(L, "Failed to add item to list");
Bram Moolenaar1dced572012-04-05 16:54:08 +0200784 }
Bram Moolenaarb3766472013-04-15 13:49:21 +0200785 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200786 lua_settop(L, 1);
787 return 1;
788}
789
790 static int
791luaV_list_insert (lua_State *L)
792{
793 luaV_List *lis = luaV_checkudata(L, 1, LUAVIM_LIST);
794 list_T *l = (list_T *) luaV_checkcache(L, (void *) *lis);
Bram Moolenaar46538ee2015-02-17 16:28:55 +0100795 long pos = (long) luaL_optinteger(L, 3, 0);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200796 listitem_T *li = NULL;
797 typval_T v;
798 if (l->lv_lock)
799 luaL_error(L, "list is locked");
800 if (pos < l->lv_len)
801 {
802 li = list_find(l, pos);
803 if (li == NULL)
804 luaL_error(L, "invalid position");
805 }
806 lua_settop(L, 2);
807 luaV_totypval(L, 2, &v);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200808 if (list_insert_tv(l, &v, li) == FAIL)
809 {
810 clear_tv(&v);
811 luaL_error(L, "Failed to add item to list");
812 }
813 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200814 lua_settop(L, 1);
815 return 1;
816}
817
818static const luaL_Reg luaV_List_mt[] = {
819 {"__tostring", luaV_list_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200820 {"__len", luaV_list_len},
821 {"__call", luaV_list_call},
822 {"__index", luaV_list_index},
823 {"__newindex", luaV_list_newindex},
824 {"add", luaV_list_add},
825 {"insert", luaV_list_insert},
826 {NULL, NULL}
827};
828
829
830/* ======= Dict type ======= */
831
832 static luaV_Dict *
833luaV_newdict (lua_State *L, dict_T *dic)
834{
835 luaV_Dict *d = (luaV_Dict *) lua_newuserdata(L, sizeof(luaV_Dict));
836 *d = dic;
837 dic->dv_refcount++; /* reference in Lua */
838 luaV_setudata(L, dic); /* cache[dic] = udata */
839 luaV_getfield(L, LUAVIM_DICT);
840 lua_setmetatable(L, -2);
841 return d;
842}
843
844luaV_pushtype(dict_T, dict, luaV_Dict)
845luaV_type_tostring(dict, LUAVIM_DICT)
846
847 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +0200848luaV_dict_len (lua_State *L)
849{
850 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
851 lua_pushinteger(L, (d == NULL) ? 0 : (int) d->dv_hashtab.ht_used);
852 return 1;
853}
854
855 static int
Bram Moolenaarfeeaa682013-02-14 22:19:51 +0100856luaV_dict_iter (lua_State *L UNUSED)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200857{
Bram Moolenaarfeeaa682013-02-14 22:19:51 +0100858#ifdef FEAT_EVAL
Bram Moolenaar1dced572012-04-05 16:54:08 +0200859 hashitem_T *hi = (hashitem_T *) lua_touserdata(L, lua_upvalueindex(2));
860 int n = lua_tointeger(L, lua_upvalueindex(3));
861 dictitem_T *di;
862 if (n <= 0) return 0;
863 while (HASHITEM_EMPTY(hi)) hi++;
864 di = dict_lookup(hi);
865 lua_pushstring(L, (char *) hi->hi_key);
866 luaV_pushtypval(L, &di->di_tv);
867 lua_pushlightuserdata(L, (void *) (hi + 1));
868 lua_replace(L, lua_upvalueindex(2));
869 lua_pushinteger(L, n - 1);
870 lua_replace(L, lua_upvalueindex(3));
871 return 2;
Bram Moolenaarfeeaa682013-02-14 22:19:51 +0100872#else
873 return 0;
874#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +0200875}
876
877 static int
878luaV_dict_call (lua_State *L)
879{
880 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
881 hashtab_T *ht = &d->dv_hashtab;
882 lua_pushvalue(L, lua_upvalueindex(1)); /* pass cache table along */
883 lua_pushlightuserdata(L, (void *) ht->ht_array);
884 lua_pushinteger(L, ht->ht_used); /* # remaining items */
885 lua_pushcclosure(L, luaV_dict_iter, 3);
886 return 1;
887}
888
889 static int
890luaV_dict_index (lua_State *L)
891{
892 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
893 char_u *key = (char_u *) luaL_checkstring(L, 2);
894 dictitem_T *di = dict_find(d, key, -1);
895 if (di == NULL)
896 lua_pushnil(L);
897 else
898 luaV_pushtypval(L, &di->di_tv);
899 return 1;
900}
901
902 static int
903luaV_dict_newindex (lua_State *L)
904{
905 dict_T *d = luaV_unbox(L, luaV_Dict, 1);
906 char_u *key = (char_u *) luaL_checkstring(L, 2);
907 dictitem_T *di;
908 if (d->dv_lock)
909 luaL_error(L, "dict is locked");
910 di = dict_find(d, key, -1);
911 if (di == NULL) /* non-existing key? */
912 {
913 if (lua_isnil(L, 3)) return 0;
914 di = dictitem_alloc(key);
915 if (di == NULL) return 0;
916 if (dict_add(d, di) == FAIL)
917 {
918 vim_free(di);
919 return 0;
920 }
921 }
922 else
923 clear_tv(&di->di_tv);
924 if (lua_isnil(L, 3)) /* remove? */
925 {
926 hashitem_T *hi = hash_find(&d->dv_hashtab, di->di_key);
927 hash_remove(&d->dv_hashtab, hi);
928 dictitem_free(di);
929 }
930 else {
931 typval_T v;
932 luaV_totypval(L, 3, &v);
933 copy_tv(&v, &di->di_tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +0200934 clear_tv(&v);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200935 }
936 return 0;
937}
938
939static const luaL_Reg luaV_Dict_mt[] = {
940 {"__tostring", luaV_dict_tostring},
Bram Moolenaar1dced572012-04-05 16:54:08 +0200941 {"__len", luaV_dict_len},
942 {"__call", luaV_dict_call},
943 {"__index", luaV_dict_index},
944 {"__newindex", luaV_dict_newindex},
945 {NULL, NULL}
946};
947
948
949/* ======= Buffer type ======= */
950
951luaV_newtype(buf_T, buffer, luaV_Buffer, LUAVIM_BUFFER)
952luaV_pushtype(buf_T, buffer, luaV_Buffer)
953luaV_type_tostring(buffer, LUAVIM_BUFFER)
954
955 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200956luaV_buffer_len(lua_State *L)
957{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200958 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
959 lua_pushinteger(L, b->b_ml.ml_line_count);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200960 return 1;
961}
962
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200963 static int
964luaV_buffer_call(lua_State *L)
965{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200966 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200967 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200968 set_curbuf(b, DOBUF_SPLIT);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200969 return 1;
970}
971
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200972 static int
973luaV_buffer_index(lua_State *L)
974{
Bram Moolenaar1dced572012-04-05 16:54:08 +0200975 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200976 linenr_T n = (linenr_T) lua_tointeger(L, 2);
Bram Moolenaar1dced572012-04-05 16:54:08 +0200977 if (n > 0 && n <= b->b_ml.ml_line_count)
978 luaV_pushline(L, b, n);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200979 else if (lua_isstring(L, 2))
980 {
981 const char *s = lua_tostring(L, 2);
982 if (strncmp(s, "name", 4) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200983 lua_pushstring(L, (char *) b->b_sfname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200984 else if (strncmp(s, "fname", 5) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200985 lua_pushstring(L, (char *) b->b_ffname);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200986 else if (strncmp(s, "number", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +0200987 lua_pushinteger(L, b->b_fnum);
Bram Moolenaar55d5c032010-07-17 23:52:29 +0200988 /* methods */
989 else if (strncmp(s, "insert", 6) == 0
990 || strncmp(s, "next", 4) == 0
991 || strncmp(s, "previous", 8) == 0
992 || strncmp(s, "isvalid", 7) == 0)
993 {
994 lua_getmetatable(L, 1);
995 lua_getfield(L, -1, s);
996 }
997 else
998 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +0200999 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001000 else
1001 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001002 return 1;
1003}
1004
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001005 static int
1006luaV_buffer_newindex(lua_State *L)
1007{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001008 buf_T *b = (buf_T *) luaV_checkvalid(L, luaV_Buffer, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001009 linenr_T n = (linenr_T) luaL_checkinteger(L, 2);
1010#ifdef HAVE_SANDBOX
1011 luaV_checksandbox(L);
1012#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001013 if (n < 1 || n > b->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001014 luaL_error(L, "invalid line number");
1015 if (lua_isnil(L, 3)) /* delete line */
1016 {
1017 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001018 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001019 if (u_savedel(n, 1L) == FAIL)
1020 {
1021 curbuf = buf;
1022 luaL_error(L, "cannot save undo information");
1023 }
1024 else if (ml_delete(n, FALSE) == FAIL)
1025 {
1026 curbuf = buf;
1027 luaL_error(L, "cannot delete line");
1028 }
1029 else {
1030 deleted_lines_mark(n, 1L);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001031 if (b == curwin->w_buffer) /* fix cursor in current window? */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001032 {
1033 if (curwin->w_cursor.lnum >= n)
1034 {
1035 if (curwin->w_cursor.lnum > n)
1036 {
1037 curwin->w_cursor.lnum -= 1;
1038 check_cursor_col();
1039 }
1040 else check_cursor();
1041 changed_cline_bef_curs();
1042 }
1043 invalidate_botline();
1044 }
1045 }
1046 curbuf = buf;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001047 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001048 else if (lua_isstring(L, 3)) /* update line */
1049 {
1050 buf_T *buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001051 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001052 if (u_savesub(n) == FAIL)
1053 {
1054 curbuf = buf;
1055 luaL_error(L, "cannot save undo information");
1056 }
1057 else if (ml_replace(n, luaV_toline(L, 3), TRUE) == FAIL)
1058 {
1059 curbuf = buf;
1060 luaL_error(L, "cannot replace line");
1061 }
1062 else changed_bytes(n, 0);
1063 curbuf = buf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001064 if (b == curwin->w_buffer)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001065 check_cursor_col();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001066 }
1067 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001068 luaL_error(L, "wrong argument to change line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001069 return 0;
1070}
1071
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001072 static int
1073luaV_buffer_insert(lua_State *L)
1074{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001075 luaV_Buffer *lb = luaV_checkudata(L, 1, LUAVIM_BUFFER);
1076 buf_T *b = (buf_T *) luaV_checkcache(L, (void *) *lb);
1077 linenr_T last = b->b_ml.ml_line_count;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001078 linenr_T n = (linenr_T) luaL_optinteger(L, 3, last);
1079 buf_T *buf;
1080 luaL_checktype(L, 2, LUA_TSTRING);
1081#ifdef HAVE_SANDBOX
1082 luaV_checksandbox(L);
1083#endif
1084 /* fix insertion line */
1085 if (n < 0) n = 0;
1086 if (n > last) n = last;
1087 /* insert */
1088 buf = curbuf;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001089 curbuf = b;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001090 if (u_save(n, n + 1) == FAIL)
1091 {
1092 curbuf = buf;
1093 luaL_error(L, "cannot save undo information");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001094 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001095 else if (ml_append(n, luaV_toline(L, 2), 0, FALSE) == FAIL)
1096 {
1097 curbuf = buf;
1098 luaL_error(L, "cannot insert line");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001099 }
1100 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001101 appended_lines_mark(n, 1L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001102 curbuf = buf;
1103 update_screen(VALID);
1104 return 0;
1105}
1106
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001107 static int
1108luaV_buffer_next(lua_State *L)
1109{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001110 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001111 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1112 luaV_pushbuffer(L, buf->b_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001113 return 1;
1114}
1115
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001116 static int
1117luaV_buffer_previous(lua_State *L)
1118{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001119 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001120 buf_T *buf = (buf_T *) luaV_checkcache(L, (void *) *b);
1121 luaV_pushbuffer(L, buf->b_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001122 return 1;
1123}
1124
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001125 static int
1126luaV_buffer_isvalid(lua_State *L)
1127{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001128 luaV_Buffer *b = luaV_checkudata(L, 1, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001129 luaV_getudata(L, *b);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001130 lua_pushboolean(L, !lua_isnil(L, -1));
1131 return 1;
1132}
1133
1134static const luaL_Reg luaV_Buffer_mt[] = {
1135 {"__tostring", luaV_buffer_tostring},
1136 {"__len", luaV_buffer_len},
1137 {"__call", luaV_buffer_call},
1138 {"__index", luaV_buffer_index},
1139 {"__newindex", luaV_buffer_newindex},
1140 {"insert", luaV_buffer_insert},
1141 {"next", luaV_buffer_next},
1142 {"previous", luaV_buffer_previous},
1143 {"isvalid", luaV_buffer_isvalid},
1144 {NULL, NULL}
1145};
1146
1147
1148/* ======= Window type ======= */
1149
Bram Moolenaar1dced572012-04-05 16:54:08 +02001150luaV_newtype(win_T, window, luaV_Window, LUAVIM_WINDOW)
1151luaV_pushtype(win_T, window, luaV_Window)
1152luaV_type_tostring(window, LUAVIM_WINDOW)
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001153
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001154 static int
1155luaV_window_call(lua_State *L)
1156{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001157 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001158 lua_settop(L, 1);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001159 win_goto(w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001160 return 1;
1161}
1162
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001163 static int
1164luaV_window_index(lua_State *L)
1165{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001166 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001167 const char *s = luaL_checkstring(L, 2);
1168 if (strncmp(s, "buffer", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001169 luaV_pushbuffer(L, w->w_buffer);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001170 else if (strncmp(s, "line", 4) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001171 lua_pushinteger(L, w->w_cursor.lnum);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001172 else if (strncmp(s, "col", 3) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001173 lua_pushinteger(L, w->w_cursor.col + 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001174#ifdef FEAT_VERTSPLIT
1175 else if (strncmp(s, "width", 5) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001176 lua_pushinteger(L, W_WIDTH(w));
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001177#endif
1178 else if (strncmp(s, "height", 6) == 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001179 lua_pushinteger(L, w->w_height);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001180 /* methods */
1181 else if (strncmp(s, "next", 4) == 0
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001182 || strncmp(s, "previous", 8) == 0
1183 || strncmp(s, "isvalid", 7) == 0)
1184 {
1185 lua_getmetatable(L, 1);
1186 lua_getfield(L, -1, s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001187 }
1188 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001189 lua_pushnil(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001190 return 1;
1191}
1192
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001193 static int
1194luaV_window_newindex (lua_State *L)
1195{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001196 win_T *w = (win_T *) luaV_checkvalid(L, luaV_Window, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001197 const char *s = luaL_checkstring(L, 2);
1198 int v = luaL_checkinteger(L, 3);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001199 if (strncmp(s, "line", 4) == 0)
1200 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001201#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001202 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001203#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001204 if (v < 1 || v > w->w_buffer->b_ml.ml_line_count)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001205 luaL_error(L, "line out of range");
Bram Moolenaar1dced572012-04-05 16:54:08 +02001206 w->w_cursor.lnum = v;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001207 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001208 }
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001209 else if (strncmp(s, "col", 3) == 0)
1210 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001211#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001212 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001213#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001214 w->w_cursor.col = v - 1;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001215 update_screen(VALID);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001216 }
1217#ifdef FEAT_VERTSPLIT
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001218 else if (strncmp(s, "width", 5) == 0)
1219 {
1220 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001221#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001222 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001223#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001224 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001225 win_setwidth(v);
1226 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001227 }
1228#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001229 else if (strncmp(s, "height", 6) == 0)
1230 {
1231 win_T *win = curwin;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001232#ifdef FEAT_GUI
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001233 need_mouse_correct = TRUE;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001234#endif
Bram Moolenaar1dced572012-04-05 16:54:08 +02001235 curwin = w;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001236 win_setheight(v);
1237 curwin = win;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001238 }
1239 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001240 luaL_error(L, "invalid window property: `%s'", s);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001241 return 0;
1242}
1243
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001244 static int
1245luaV_window_next(lua_State *L)
1246{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001247 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001248 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1249 luaV_pushwindow(L, win->w_next);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001250 return 1;
1251}
1252
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001253 static int
1254luaV_window_previous(lua_State *L)
1255{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001256 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001257 win_T *win = (win_T *) luaV_checkcache(L, (void *) *w);
1258 luaV_pushwindow(L, win->w_prev);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001259 return 1;
1260}
1261
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001262 static int
1263luaV_window_isvalid(lua_State *L)
1264{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001265 luaV_Window *w = luaV_checkudata(L, 1, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001266 luaV_getudata(L, *w);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001267 lua_pushboolean(L, !lua_isnil(L, -1));
1268 return 1;
1269}
1270
1271static const luaL_Reg luaV_Window_mt[] = {
1272 {"__tostring", luaV_window_tostring},
1273 {"__call", luaV_window_call},
1274 {"__index", luaV_window_index},
1275 {"__newindex", luaV_window_newindex},
1276 {"next", luaV_window_next},
1277 {"previous", luaV_window_previous},
1278 {"isvalid", luaV_window_isvalid},
1279 {NULL, NULL}
1280};
1281
1282
1283/* ======= Vim module ======= */
1284
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001285 static int
1286luaV_print(lua_State *L)
1287{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001288 int i, n = lua_gettop(L); /* nargs */
1289 const char *s;
1290 size_t l;
1291 luaL_Buffer b;
1292 luaL_buffinit(L, &b);
1293 lua_getglobal(L, "tostring");
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001294 for (i = 1; i <= n; i++)
1295 {
1296 lua_pushvalue(L, -1); /* tostring */
1297 lua_pushvalue(L, i); /* arg */
1298 lua_call(L, 1, 1);
1299 s = lua_tolstring(L, -1, &l);
1300 if (s == NULL)
1301 return luaL_error(L, "cannot convert to string");
1302 if (i > 1) luaL_addchar(&b, ' '); /* use space instead of tab */
1303 luaV_addlstring(&b, s, l, 0);
1304 lua_pop(L, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001305 }
1306 luaL_pushresult(&b);
1307 luaV_msg(L);
1308 return 0;
1309}
1310
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001311 static int
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001312luaV_debug(lua_State *L)
1313{
1314 lua_settop(L, 0);
1315 lua_getglobal(L, "vim");
1316 lua_getfield(L, -1, "eval");
1317 lua_remove(L, -2); /* vim.eval at position 1 */
1318 for (;;)
1319 {
1320 const char *input;
1321 size_t l;
1322 lua_pushvalue(L, 1); /* vim.eval */
1323 lua_pushliteral(L, "input('lua_debug> ')");
1324 lua_call(L, 1, 1); /* return string */
1325 input = lua_tolstring(L, -1, &l);
1326 if (l == 0 || strcmp(input, "cont") == 0)
1327 return 0;
1328 msg_putchar('\n'); /* avoid outputting on input line */
1329 if (luaL_loadbuffer(L, input, l, "=(debug command)")
1330 || lua_pcall(L, 0, 0, 0))
1331 luaV_emsg(L);
1332 lua_settop(L, 1); /* remove eventual returns, but keep vim.eval */
1333 }
1334}
1335
1336 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001337luaV_command(lua_State *L)
1338{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001339 do_cmdline_cmd((char_u *) luaL_checkstring(L, 1));
1340 update_screen(VALID);
1341 return 0;
1342}
1343
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001344 static int
1345luaV_eval(lua_State *L)
1346{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001347 typval_T *tv = eval_expr((char_u *) luaL_checkstring(L, 1), NULL);
1348 if (tv == NULL) luaL_error(L, "invalid expression");
1349 luaV_pushtypval(L, tv);
Bram Moolenaarb3766472013-04-15 13:49:21 +02001350 free_tv(tv);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001351 return 1;
1352}
1353
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001354 static int
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001355luaV_beep(lua_State *L UNUSED)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001356{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001357 vim_beep();
1358 return 0;
1359}
1360
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001361 static int
1362luaV_line(lua_State *L)
1363{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001364 luaV_pushline(L, curbuf, curwin->w_cursor.lnum);
1365 return 1;
1366}
1367
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001368 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001369luaV_list(lua_State *L)
1370{
1371 list_T *l = list_alloc();
1372 if (l == NULL)
1373 lua_pushnil(L);
1374 else
1375 luaV_newlist(L, l);
1376 return 1;
1377}
1378
1379 static int
1380luaV_dict(lua_State *L)
1381{
1382 dict_T *d = dict_alloc();
1383 if (d == NULL)
1384 lua_pushnil(L);
1385 else
1386 luaV_newdict(L, d);
1387 return 1;
1388}
1389
1390 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001391luaV_buffer(lua_State *L)
1392{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001393 buf_T *buf;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001394 if (lua_isstring(L, 1)) /* get by number or name? */
1395 {
1396 if (lua_isnumber(L, 1)) /* by number? */
1397 {
1398 int n = lua_tointeger(L, 1);
1399 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1400 if (buf->b_fnum == n) break;
1401 }
1402 else { /* by name */
1403 size_t l;
1404 const char *s = lua_tolstring(L, 1, &l);
1405 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1406 {
1407 if (buf->b_ffname == NULL || buf->b_sfname == NULL)
1408 {
1409 if (l == 0) break;
1410 }
Bram Moolenaar0d2e4fc2010-07-18 12:35:47 +02001411 else if (strncmp(s, (char *)buf->b_ffname, l) == 0
1412 || strncmp(s, (char *)buf->b_sfname, l) == 0)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001413 break;
1414 }
1415 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001416 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001417 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001418 buf = (lua_toboolean(L, 1)) ? firstbuf : curbuf; /* first buffer? */
Bram Moolenaar1dced572012-04-05 16:54:08 +02001419 luaV_pushbuffer(L, buf);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001420 return 1;
1421}
1422
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001423 static int
1424luaV_window(lua_State *L)
1425{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001426 win_T *win;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001427 if (lua_isnumber(L, 1)) /* get by number? */
1428 {
1429 int n = lua_tointeger(L, 1);
1430 for (win = firstwin; win != NULL; win = win->w_next, n--)
1431 if (n == 1) break;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001432 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001433 else
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001434 win = (lua_toboolean(L, 1)) ? firstwin : curwin; /* first window? */
Bram Moolenaar1dced572012-04-05 16:54:08 +02001435 luaV_pushwindow(L, win);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001436 return 1;
1437}
1438
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001439 static int
1440luaV_open(lua_State *L)
1441{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001442 char_u *s = NULL;
1443#ifdef HAVE_SANDBOX
1444 luaV_checksandbox(L);
1445#endif
1446 if (lua_isstring(L, 1)) s = (char_u *) lua_tostring(L, 1);
Bram Moolenaarfa263a52011-12-08 16:00:16 +01001447 luaV_pushbuffer(L, buflist_new(s, NULL, 1L, BLN_LISTED));
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001448 return 1;
1449}
1450
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001451 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001452luaV_type(lua_State *L)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001453{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001454 luaL_checkany(L, 1);
1455 if (lua_type(L, 1) == LUA_TUSERDATA) /* check vim udata? */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001456 {
Bram Moolenaar1dced572012-04-05 16:54:08 +02001457 lua_settop(L, 1);
1458 if (lua_getmetatable(L, 1))
1459 {
1460 luaV_getfield(L, LUAVIM_LIST);
1461 if (lua_rawequal(L, -1, 2))
1462 {
1463 lua_pushstring(L, "list");
1464 return 1;
1465 }
1466 luaV_getfield(L, LUAVIM_DICT);
1467 if (lua_rawequal(L, -1, 2))
1468 {
1469 lua_pushstring(L, "dict");
1470 return 1;
1471 }
1472 luaV_getfield(L, LUAVIM_BUFFER);
1473 if (lua_rawequal(L, -1, 2))
1474 {
1475 lua_pushstring(L, "buffer");
1476 return 1;
1477 }
1478 luaV_getfield(L, LUAVIM_WINDOW);
1479 if (lua_rawequal(L, -1, 2))
1480 {
1481 lua_pushstring(L, "window");
1482 return 1;
1483 }
1484 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001485 }
Bram Moolenaar1dced572012-04-05 16:54:08 +02001486 lua_pushstring(L, luaL_typename(L, 1)); /* fallback */
1487 return 1;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001488}
1489
1490static const luaL_Reg luaV_module[] = {
1491 {"command", luaV_command},
1492 {"eval", luaV_eval},
1493 {"beep", luaV_beep},
1494 {"line", luaV_line},
Bram Moolenaar1dced572012-04-05 16:54:08 +02001495 {"list", luaV_list},
1496 {"dict", luaV_dict},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001497 {"buffer", luaV_buffer},
1498 {"window", luaV_window},
1499 {"open", luaV_open},
Bram Moolenaar1dced572012-04-05 16:54:08 +02001500 {"type", luaV_type},
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001501 {NULL, NULL}
1502};
1503
Bram Moolenaar1dced572012-04-05 16:54:08 +02001504/* for freeing list, dict, buffer and window objects; lightuserdata as arg */
1505 static int
1506luaV_free(lua_State *L)
1507{
1508 lua_pushnil(L);
1509 luaV_setudata(L, lua_touserdata(L, 1));
1510 return 0;
1511}
1512
1513 static int
1514luaV_luaeval (lua_State *L)
1515{
1516 luaL_Buffer b;
1517 size_t l;
1518 const char *str = lua_tolstring(L, 1, &l);
1519 typval_T *arg = (typval_T *) lua_touserdata(L, 2);
1520 typval_T *rettv = (typval_T *) lua_touserdata(L, 3);
1521 luaL_buffinit(L, &b);
1522 luaL_addlstring(&b, LUAVIM_EVALHEADER, sizeof(LUAVIM_EVALHEADER) - 1);
1523 luaL_addlstring(&b, str, l);
1524 luaL_pushresult(&b);
1525 str = lua_tolstring(L, -1, &l);
1526 if (luaL_loadbuffer(L, str, l, LUAVIM_EVALNAME)) /* compile error? */
1527 {
1528 luaV_emsg(L);
1529 return 0;
1530 }
1531 luaV_pushtypval(L, arg);
1532 if (lua_pcall(L, 1, 1, 0)) /* running error? */
1533 {
1534 luaV_emsg(L);
1535 return 0;
1536 }
1537 luaV_totypval(L, -1, rettv);
Bram Moolenaarf554a322015-02-04 23:08:01 +01001538 return 0;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001539}
1540
1541 static int
1542luaV_setref (lua_State *L)
1543{
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001544 int copyID = lua_tointeger(L, 1);
1545 int abort = FALSE;
1546 typval_T tv;
1547
Bram Moolenaar1dced572012-04-05 16:54:08 +02001548 luaV_getfield(L, LUAVIM_LIST);
1549 luaV_getfield(L, LUAVIM_DICT);
1550 lua_pushnil(L);
Bram Moolenaarb84634d2015-02-04 22:02:37 +01001551 /* traverse cache table */
1552 while (!abort && lua_next(L, lua_upvalueindex(1)) != 0)
Bram Moolenaar1dced572012-04-05 16:54:08 +02001553 {
1554 lua_getmetatable(L, -1);
1555 if (lua_rawequal(L, -1, 2)) /* list? */
1556 {
1557 tv.v_type = VAR_LIST;
1558 tv.vval.v_list = (list_T *) lua_touserdata(L, 4); /* key */
1559 }
1560 else if (lua_rawequal(L, -1, 3)) /* dict? */
1561 {
1562 tv.v_type = VAR_DICT;
1563 tv.vval.v_dict = (dict_T *) lua_touserdata(L, 4); /* key */
1564 }
1565 lua_pop(L, 2); /* metatable and value */
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001566 abort = set_ref_in_item(&tv, copyID, NULL, NULL);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001567 }
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001568 lua_pushinteger(L, abort);
Bram Moolenaarf554a322015-02-04 23:08:01 +01001569 return 1;
Bram Moolenaar1dced572012-04-05 16:54:08 +02001570}
1571
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001572 static int
1573luaopen_vim(lua_State *L)
1574{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001575 /* set cache table */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001576 lua_newtable(L);
1577 lua_newtable(L);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001578 lua_pushstring(L, "v");
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001579 lua_setfield(L, -2, "__mode");
Bram Moolenaar1dced572012-04-05 16:54:08 +02001580 lua_setmetatable(L, -2); /* cache is weak-valued */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001581 /* print */
1582 lua_pushcfunction(L, luaV_print);
1583 lua_setglobal(L, "print");
Bram Moolenaar38e2b062011-09-21 17:15:39 +02001584 /* debug.debug */
1585 lua_getglobal(L, "debug");
1586 lua_pushcfunction(L, luaV_debug);
1587 lua_setfield(L, -2, "debug");
1588 lua_pop(L, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001589 /* free */
1590 lua_pushlightuserdata(L, (void *) LUAVIM_FREE);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001591 lua_pushvalue(L, 1); /* cache table */
1592 lua_pushcclosure(L, luaV_free, 1);
1593 lua_rawset(L, LUA_REGISTRYINDEX);
1594 /* luaeval */
1595 lua_pushlightuserdata(L, (void *) LUAVIM_LUAEVAL);
1596 lua_pushvalue(L, 1); /* cache table */
1597 lua_pushcclosure(L, luaV_luaeval, 1);
1598 lua_rawset(L, LUA_REGISTRYINDEX);
1599 /* setref */
1600 lua_pushlightuserdata(L, (void *) LUAVIM_SETREF);
1601 lua_pushvalue(L, 1); /* cache table */
1602 lua_pushcclosure(L, luaV_setref, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001603 lua_rawset(L, LUA_REGISTRYINDEX);
1604 /* register */
Bram Moolenaar1dced572012-04-05 16:54:08 +02001605 luaV_newmetatable(L, LUAVIM_LIST);
1606 lua_pushvalue(L, 1);
1607 luaV_openlib(L, luaV_List_mt, 1);
1608 luaV_newmetatable(L, LUAVIM_DICT);
1609 lua_pushvalue(L, 1);
1610 luaV_openlib(L, luaV_Dict_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001611 luaV_newmetatable(L, LUAVIM_BUFFER);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001612 lua_pushvalue(L, 1); /* cache table */
1613 luaV_openlib(L, luaV_Buffer_mt, 1);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001614 luaV_newmetatable(L, LUAVIM_WINDOW);
Bram Moolenaar1dced572012-04-05 16:54:08 +02001615 lua_pushvalue(L, 1); /* cache table */
1616 luaV_openlib(L, luaV_Window_mt, 1);
1617 lua_newtable(L); /* vim table */
1618 lua_pushvalue(L, 1); /* cache table */
1619 luaV_openlib(L, luaV_module, 1);
1620 lua_setglobal(L, LUAVIM_NAME);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001621 return 0;
1622}
1623
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001624 static lua_State *
1625luaV_newstate(void)
1626{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001627 lua_State *L = luaL_newstate();
Bram Moolenaar16c98f92010-07-28 22:46:08 +02001628 luaL_openlibs(L); /* core libs */
1629 lua_pushcfunction(L, luaopen_vim); /* vim */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001630 lua_call(L, 0, 0);
1631 return L;
1632}
1633
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001634 static void
1635luaV_setrange(lua_State *L, int line1, int line2)
1636{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001637 lua_getglobal(L, LUAVIM_NAME);
1638 lua_pushinteger(L, line1);
1639 lua_setfield(L, -2, "firstline");
1640 lua_pushinteger(L, line2);
1641 lua_setfield(L, -2, "lastline");
1642 lua_pop(L, 1); /* vim table */
1643}
1644
1645
1646/* ======= Interface ======= */
1647
1648static lua_State *L = NULL;
1649
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001650 static int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001651lua_isopen(void)
Bram Moolenaar2bd6a1b2010-08-12 22:14:01 +02001652{
1653 return L != NULL;
1654}
1655
1656 static int
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001657lua_init(void)
1658{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001659 if (!lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001660 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001661#ifdef DYNAMIC_LUA
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001662 if (!lua_enabled(TRUE))
1663 {
1664 EMSG(_("Lua library cannot be loaded."));
1665 return FAIL;
1666 }
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001667#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001668 L = luaV_newstate();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001669 }
1670 return OK;
1671}
1672
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001673 void
1674lua_end(void)
1675{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001676 if (lua_isopen())
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001677 {
1678 lua_close(L);
1679 L = NULL;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001680#ifdef DYNAMIC_LUA
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001681 end_dynamic_lua();
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001682#endif
1683 }
1684}
1685
1686/* ex commands */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001687 void
1688ex_lua(exarg_T *eap)
1689{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001690 char *script;
1691 if (lua_init() == FAIL) return;
1692 script = (char *) script_get(eap, eap->arg);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001693 if (!eap->skip)
1694 {
1695 char *s = (script) ? script : (char *) eap->arg;
1696 luaV_setrange(L, eap->line1, eap->line2);
1697 if (luaL_loadbuffer(L, s, strlen(s), LUAVIM_CHUNKNAME)
1698 || lua_pcall(L, 0, 0, 0))
1699 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001700 }
1701 if (script != NULL) vim_free(script);
1702}
1703
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001704 void
1705ex_luado(exarg_T *eap)
1706{
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001707 linenr_T l;
1708 const char *s = (const char *) eap->arg;
1709 luaL_Buffer b;
1710 size_t len;
1711 if (lua_init() == FAIL) return;
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001712 if (u_save(eap->line1 - 1, eap->line2 + 1) == FAIL)
1713 {
1714 EMSG(_("cannot save undo information"));
1715 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001716 }
1717 luaV_setrange(L, eap->line1, eap->line2);
1718 luaL_buffinit(L, &b);
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +02001719 luaL_addlstring(&b, "return function(line, linenr) ", 30); /* header */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001720 luaL_addlstring(&b, s, strlen(s));
1721 luaL_addlstring(&b, " end", 4); /* footer */
1722 luaL_pushresult(&b);
1723 s = lua_tolstring(L, -1, &len);
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001724 if (luaL_loadbuffer(L, s, len, LUAVIM_CHUNKNAME))
1725 {
1726 luaV_emsg(L);
1727 lua_pop(L, 1); /* function body */
1728 return;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001729 }
1730 lua_call(L, 0, 1);
1731 lua_replace(L, -2); /* function -> body */
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001732 for (l = eap->line1; l <= eap->line2; l++)
1733 {
1734 lua_pushvalue(L, -1); /* function */
1735 luaV_pushline(L, curbuf, l); /* current line as arg */
Bram Moolenaarbd2f3c32012-04-06 14:31:00 +02001736 lua_pushinteger(L, l); /* current line number as arg */
1737 if (lua_pcall(L, 2, 1, 0))
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001738 {
1739 luaV_emsg(L);
1740 break;
1741 }
1742 if (lua_isstring(L, -1)) /* update line? */
1743 {
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001744#ifdef HAVE_SANDBOX
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001745 luaV_checksandbox(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001746#endif
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001747 ml_replace(l, luaV_toline(L, -1), TRUE);
1748 changed_bytes(l, 0);
1749 lua_pop(L, 1); /* result from luaV_toline */
1750 }
1751 lua_pop(L, 1); /* line */
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001752 }
1753 lua_pop(L, 1); /* function */
1754 check_cursor();
1755 update_screen(NOT_VALID);
1756}
1757
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001758 void
1759ex_luafile(exarg_T *eap)
1760{
1761 if (lua_init() == FAIL)
1762 return;
1763 if (!eap->skip)
1764 {
1765 luaV_setrange(L, eap->line1, eap->line2);
1766 if (luaL_loadfile(L, (char *) eap->arg) || lua_pcall(L, 0, 0, 0))
1767 luaV_emsg(L);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001768 }
1769}
1770
Bram Moolenaar1dced572012-04-05 16:54:08 +02001771#define luaV_freetype(typ,tname) \
1772 void \
1773 lua_##tname##_free(typ *o) \
1774 { \
1775 if (!lua_isopen()) return; \
1776 luaV_getfield(L, LUAVIM_FREE); \
1777 lua_pushlightuserdata(L, (void *) o); \
1778 lua_call(L, 1, 0); \
1779 }
1780
1781luaV_freetype(buf_T, buffer)
1782luaV_freetype(win_T, window)
1783
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001784 void
Bram Moolenaar1dced572012-04-05 16:54:08 +02001785do_luaeval (char_u *str, typval_T *arg, typval_T *rettv)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001786{
Bram Moolenaar1dced572012-04-05 16:54:08 +02001787 lua_init();
1788 luaV_getfield(L, LUAVIM_LUAEVAL);
1789 lua_pushstring(L, (char *) str);
1790 lua_pushlightuserdata(L, (void *) arg);
1791 lua_pushlightuserdata(L, (void *) rettv);
1792 lua_call(L, 3, 0);
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001793}
1794
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001795 int
Bram Moolenaar1dced572012-04-05 16:54:08 +02001796set_ref_in_lua (int copyID)
Bram Moolenaar55d5c032010-07-17 23:52:29 +02001797{
Bram Moolenaar2459a5e2015-02-03 12:55:18 +01001798 int aborted = 0;
1799
1800 if (lua_isopen())
1801 {
1802 luaV_getfield(L, LUAVIM_SETREF);
1803 /* call the function with 1 arg, getting 1 result back */
1804 lua_pushinteger(L, copyID);
1805 lua_call(L, 1, 1);
1806 /* get the result */
1807 aborted = lua_tointeger(L, -1);
1808 /* pop result off the stack */
1809 lua_pop(L, 1);
1810 }
1811 return aborted;
Bram Moolenaar0ba04292010-07-14 23:23:17 +02001812}
1813
1814#endif