vis: use marks instead of registers to store selections
[vis.git] / vis-lua.c
blob91ad6454e1d78d879bcb4e777d6a1bc53f624aa8
1 /***
2 * Lua Extension API for the [Vis Editor](https://github.com/martanne/vis).
4 * *WARNING:* there is no stability guarantee at this time, the API might
5 * change without notice!
7 * This document might be out of date, run `make luadoc` to regenerate it.
9 * @module vis
10 * @author Marc André Tanner
11 * @license ISC
12 * @release RELEASE
14 #include <stddef.h>
15 #include <stdarg.h>
16 #include <stdio.h>
17 #include <string.h>
18 #include <limits.h>
19 #include <unistd.h>
20 #include <libgen.h>
21 #include <sys/types.h>
22 #include <pwd.h>
24 #include "vis-lua.h"
25 #include "vis-core.h"
26 #include "text-motions.h"
27 #include "util.h"
29 #ifndef VIS_PATH
30 #define VIS_PATH "/usr/local/share/vis"
31 #endif
33 #define VIS_LUA_TYPE_VIS "vis"
34 #define VIS_LUA_TYPE_FILE "file"
35 #define VIS_LUA_TYPE_TEXT "text"
36 #define VIS_LUA_TYPE_MARK "mark"
37 #define VIS_LUA_TYPE_MARKS "marks"
38 #define VIS_LUA_TYPE_WINDOW "window"
39 #define VIS_LUA_TYPE_CURSOR "cursor"
40 #define VIS_LUA_TYPE_CURSORS "cursors"
41 #define VIS_LUA_TYPE_UI "ui"
42 #define VIS_LUA_TYPE_REGISTERS "registers"
43 #define VIS_LUA_TYPE_KEYACTION "keyaction"
45 #ifndef DEBUG_LUA
46 #define DEBUG_LUA 0
47 #endif
49 #if DEBUG_LUA
50 #define debug(...) do { printf(__VA_ARGS__); fflush(stdout); } while (0)
51 #else
52 #define debug(...) do { } while (0)
53 #endif
55 static void window_status_update(Vis *vis, Win *win) {
56 char left_parts[4][255] = { "", "", "", "" };
57 char right_parts[4][32] = { "", "", "", "" };
58 char left[sizeof(left_parts)+LENGTH(left_parts)*8];
59 char right[sizeof(right_parts)+LENGTH(right_parts)*8];
60 char status[sizeof(left)+sizeof(right)+1];
61 size_t left_count = 0;
62 size_t right_count = 0;
64 View *view = win->view;
65 File *file = win->file;
66 Text *txt = file->text;
67 int width = vis_window_width_get(win);
68 enum UiOption options = view_options_get(view);
69 bool focused = vis->win == win;
70 const char *filename = file_name_get(file);
71 const char *mode = vis->mode->status;
73 if (focused && mode)
74 strcpy(left_parts[left_count++], mode);
76 snprintf(left_parts[left_count], sizeof(left_parts[left_count])-1, "%s%s%s",
77 filename ? filename : "[No Name]",
78 text_modified(txt) ? " [+]" : "",
79 vis_macro_recording(vis) ? " @": "");
80 left_count++;
82 int cursor_count = view_selections_count(view);
83 if (cursor_count > 1) {
84 Selection *s = view_selections_primary_get(view);
85 int cursor_number = view_selections_number(s) + 1;
86 snprintf(right_parts[right_count], sizeof(right_parts[right_count])-1,
87 "%d/%d", cursor_number, cursor_count);
88 right_count++;
91 size_t size = text_size(txt);
92 size_t pos = view_cursor_get(view);
93 size_t percent = 0;
94 if (size > 0) {
95 double tmp = ((double)pos/(double)size)*100;
96 percent = (size_t)(tmp+1);
98 snprintf(right_parts[right_count], sizeof(right_parts[right_count])-1,
99 "%zu%%", percent);
100 right_count++;
102 if (!(options & UI_OPTION_LARGE_FILE)) {
103 Selection *sel = view_selections_primary_get(win->view);
104 size_t line = view_cursors_line(sel);
105 size_t col = view_cursors_col(sel);
106 if (col > UI_LARGE_FILE_LINE_SIZE) {
107 options |= UI_OPTION_LARGE_FILE;
108 view_options_set(win->view, options);
110 snprintf(right_parts[right_count], sizeof(right_parts[right_count])-1,
111 "%zu, %zu", line, col);
112 right_count++;
115 int left_len = snprintf(left, sizeof(left)-1, " %s%s%s%s%s%s%s",
116 left_parts[0],
117 left_parts[1][0] ? " » " : "",
118 left_parts[1],
119 left_parts[2][0] ? " » " : "",
120 left_parts[2],
121 left_parts[3][0] ? " » " : "",
122 left_parts[3]);
124 int right_len = snprintf(right, sizeof(right)-1, "%s%s%s%s%s%s%s ",
125 right_parts[0],
126 right_parts[1][0] ? " « " : "",
127 right_parts[1],
128 right_parts[2][0] ? " « " : "",
129 right_parts[2],
130 right_parts[3][0] ? " « " : "",
131 right_parts[3]);
133 if (left_len < 0 || right_len < 0)
134 return;
135 int left_width = text_string_width(left, left_len);
136 int right_width = text_string_width(right, right_len);
138 int spaces = width - left_width - right_width;
139 if (spaces < 1)
140 spaces = 1;
142 snprintf(status, sizeof(status)-1, "%s%*s%s", left, spaces, " ", right);
143 vis_window_status(win, status);
146 #if !CONFIG_LUA
148 bool vis_lua_path_add(Vis *vis, const char *path) { return true; }
149 bool vis_lua_paths_get(Vis *vis, char **lpath, char **cpath) { return false; }
150 void vis_lua_init(Vis *vis) { }
151 void vis_lua_start(Vis *vis) { }
152 void vis_lua_quit(Vis *vis) { }
153 void vis_lua_file_open(Vis *vis, File *file) { }
154 bool vis_lua_file_save_pre(Vis *vis, File *file, const char *path) { return true; }
155 void vis_lua_file_save_post(Vis *vis, File *file, const char *path) { }
156 void vis_lua_file_close(Vis *vis, File *file) { }
157 void vis_lua_win_open(Vis *vis, Win *win) { }
158 void vis_lua_win_close(Vis *vis, Win *win) { }
159 void vis_lua_win_highlight(Vis *vis, Win *win) { }
160 void vis_lua_win_status(Vis *vis, Win *win) { window_status_update(vis, win); }
162 #else
164 #if DEBUG_LUA
165 static void stack_dump_entry(lua_State *L, int i) {
166 int t = lua_type(L, i);
167 switch (t) {
168 case LUA_TNIL:
169 printf("nil");
170 break;
171 case LUA_TBOOLEAN:
172 printf(lua_toboolean(L, i) ? "true" : "false");
173 break;
174 case LUA_TLIGHTUSERDATA:
175 printf("lightuserdata(%p)", lua_touserdata(L, i));
176 break;
177 case LUA_TNUMBER:
178 printf("%g", lua_tonumber(L, i));
179 break;
180 case LUA_TSTRING:
181 printf("`%s'", lua_tostring(L, i));
182 break;
183 case LUA_TTABLE:
184 printf("table[");
185 lua_pushnil(L); /* first key */
186 while (lua_next(L, i > 0 ? i : i - 1)) {
187 stack_dump_entry(L, -2);
188 printf("=");
189 stack_dump_entry(L, -1);
190 printf(",");
191 lua_pop(L, 1); /* remove value, keep key */
193 printf("]");
194 break;
195 case LUA_TUSERDATA:
196 printf("userdata(%p)", lua_touserdata(L, i));
197 break;
198 default: /* other values */
199 printf("%s", lua_typename(L, t));
200 break;
204 static void stack_dump(lua_State *L, const char *format, ...) {
205 va_list ap;
206 va_start(ap, format);
207 vprintf(format, ap);
208 va_end(ap);
209 int top = lua_gettop(L);
210 for (int i = 1; i <= top; i++) {
211 printf("%d: ", i);
212 stack_dump_entry(L, i);
213 printf("\n");
215 printf("\n\n");
216 fflush(stdout);
219 #endif
221 static int panic_handler(lua_State *L) {
222 void *ud = NULL;
223 lua_getallocf(L, &ud);
224 if (ud) {
225 Vis *vis = ud;
226 vis->lua = NULL;
227 const char *msg = NULL;
228 if (lua_type(L, -1) == LUA_TSTRING)
229 msg = lua_tostring(L, -1);
230 vis_info_show(vis, "Fatal Lua error: %s", msg ? msg : "unknown reason");
231 lua_close(L);
232 if (vis->running)
233 siglongjmp(vis->sigbus_jmpbuf, 1);
235 return 0;
238 static int error_handler(lua_State *L) {
239 Vis *vis = lua_touserdata(L, lua_upvalueindex(1));
240 if (vis->errorhandler)
241 return 1;
242 vis->errorhandler = true;
243 size_t len;
244 const char *msg = lua_tostring(L, 1);
245 if (msg)
246 luaL_traceback(L, L, msg, 1);
247 msg = lua_tolstring(L, 1, &len);
248 vis_message_show(vis, msg);
249 vis->errorhandler = false;
250 return 1;
253 static int pcall(Vis *vis, lua_State *L, int nargs, int nresults) {
254 /* insert a custom error function below all arguments */
255 int msgh = lua_gettop(L) - nargs;
256 lua_pushlightuserdata(L, vis);
257 lua_pushcclosure(L, error_handler, 1);
258 lua_insert(L, msgh);
259 int ret = lua_pcall(L, nargs, nresults, msgh);
260 lua_remove(L, msgh);
261 return ret;
264 /* expects a lua function at stack position `narg` and stores a
265 * reference to it in the registry. The return value can be used
266 * to look it up.
268 * registry["vis.functions"][(void*)(function)] = function
270 static const void *func_ref_new(lua_State *L, int narg) {
271 const void *addr = lua_topointer(L, narg);
272 if (!lua_isfunction(L, narg) || !addr)
273 luaL_argerror(L, narg, "function expected");
274 lua_getfield(L, LUA_REGISTRYINDEX, "vis.functions");
275 lua_pushlightuserdata(L, (void*)addr);
276 lua_pushvalue(L, narg);
277 lua_settable(L, -3);
278 lua_pop(L, 1);
279 return addr;
282 /* retrieve function from registry and place it at the top of the stack */
283 static bool func_ref_get(lua_State *L, const void *addr) {
284 if (!addr)
285 return false;
286 lua_getfield(L, LUA_REGISTRYINDEX, "vis.functions");
287 lua_pushlightuserdata(L, (void*)addr);
288 lua_gettable(L, -2);
289 lua_remove(L, -2);
290 if (!lua_isfunction(L, -1)) {
291 lua_pop(L, 1);
292 return false;
294 return true;
297 /* creates a new metatable for a given type and stores a mapping:
299 * registry["vis.types"][metatable] = type
301 * leaves the metatable at the top of the stack.
303 static void obj_type_new(lua_State *L, const char *type) {
304 luaL_newmetatable(L, type);
305 lua_getglobal(L, "vis");
306 if (!lua_isnil(L, -1)) {
307 lua_getfield(L, -1, "types");
308 lua_pushvalue(L, -3);
309 lua_setfield(L, -2, type);
310 lua_pop(L, 1);
312 lua_pop(L, 1);
313 lua_getfield(L, LUA_REGISTRYINDEX, "vis.types");
314 lua_pushvalue(L, -2);
315 lua_pushstring(L, type);
316 lua_settable(L, -3);
317 lua_pop(L, 1);
320 /* get type of userdatum at the top of the stack:
322 * return registry["vis.types"][getmetatable(userdata)]
324 const char *obj_type_get(lua_State *L) {
325 if (lua_isnil(L, -1))
326 return "nil";
327 lua_getfield(L, LUA_REGISTRYINDEX, "vis.types");
328 lua_getmetatable(L, -2);
329 lua_gettable(L, -2);
330 // XXX: in theory string might become invalid when poped from stack
331 const char *type = lua_tostring(L, -1);
332 lua_pop(L, 2);
333 return type;
336 static void *obj_new(lua_State *L, size_t size, const char *type) {
337 void *obj = lua_newuserdata(L, size);
338 luaL_getmetatable(L, type);
339 lua_setmetatable(L, -2);
340 lua_newtable(L);
341 lua_setuservalue(L, -2);
342 return obj;
345 /* returns registry["vis.objects"][addr] if it is of correct type */
346 static void *obj_ref_get(lua_State *L, void *addr, const char *type) {
347 lua_getfield(L, LUA_REGISTRYINDEX, "vis.objects");
348 lua_pushlightuserdata(L, addr);
349 lua_gettable(L, -2);
350 lua_remove(L, -2);
351 if (lua_isnil(L, -1)) {
352 debug("get: vis.objects[%p] = nil\n", addr);
353 lua_pop(L, 1);
354 return NULL;
356 if (DEBUG_LUA) {
357 const char *actual_type = obj_type_get(L);
358 if (strcmp(type, actual_type) != 0)
359 debug("get: vis.objects[%p] = %s (BUG: expected %s)\n", addr, actual_type, type);
360 void **handle = luaL_checkudata(L, -1, type);
361 if (!handle)
362 debug("get: vis.objects[%p] = %s (BUG: invalid handle)\n", addr, type);
363 else if (*handle != addr)
364 debug("get: vis.objects[%p] = %s (BUG: handle mismatch %p)\n", addr, type, *handle);
366 return luaL_checkudata(L, -1, type);
369 /* expects a userdatum at the top of the stack and sets
371 * registry["vis.objects"][addr] = userdata
373 static void obj_ref_set(lua_State *L, void *addr) {
374 //debug("set: vis.objects[%p] = %s\n", addr, obj_type_get(L));
375 lua_getfield(L, LUA_REGISTRYINDEX, "vis.objects");
376 lua_pushlightuserdata(L, addr);
377 lua_pushvalue(L, -3);
378 lua_settable(L, -3);
379 lua_pop(L, 1);
382 /* invalidates an object reference
384 * registry["vis.objects"][addr] = nil
386 static void obj_ref_free(lua_State *L, void *addr) {
387 if (DEBUG_LUA) {
388 lua_getfield(L, LUA_REGISTRYINDEX, "vis.objects");
389 lua_pushlightuserdata(L, addr);
390 lua_gettable(L, -2);
391 lua_remove(L, -2);
392 if (lua_isnil(L, -1))
393 debug("free-unused: %p\n", addr);
394 else
395 debug("free: vis.objects[%p] = %s\n", addr, obj_type_get(L));
396 lua_pop(L, 1);
398 lua_pushnil(L);
399 obj_ref_set(L, addr);
402 /* creates a new object reference of given type if it does not already exist in the registry:
404 * if (registry["vis.types"][metatable(registry["vis.objects"][addr])] != type) {
405 * // XXX: should not happen
406 * registry["vis.objects"][addr] = new_obj(addr, type)
408 * return registry["vis.objects"][addr];
410 static void *obj_ref_new(lua_State *L, void *addr, const char *type) {
411 if (!addr) {
412 lua_pushnil(L);
413 return NULL;
415 lua_getfield(L, LUA_REGISTRYINDEX, "vis.objects");
416 lua_pushlightuserdata(L, addr);
417 lua_gettable(L, -2);
418 lua_remove(L, -2);
419 const char *old_type = obj_type_get(L);
420 if (strcmp(type, old_type) == 0) {
421 debug("new: vis.objects[%p] = %s (returning existing object)\n", addr, old_type);
422 void **handle = luaL_checkudata(L, -1, type);
423 if (!handle)
424 debug("new: vis.objects[%p] = %s (BUG: invalid handle)\n", addr, old_type);
425 else if (*handle != addr)
426 debug("new: vis.objects[%p] = %s (BUG: handle mismatch %p)\n", addr, old_type, *handle);
427 return addr;
429 if (!lua_isnil(L, -1))
430 debug("new: vis.objects[%p] = %s (WARNING: changing object type from %s)\n", addr, type, old_type);
431 else
432 debug("new: vis.objects[%p] = %s (creating new object)\n", addr, type);
433 lua_pop(L, 1);
434 void **handle = obj_new(L, sizeof(addr), type);
435 obj_ref_set(L, addr);
436 *handle = addr;
437 return addr;
440 /* retrieve object stored in reference at stack location `idx' */
441 static void *obj_ref_check_get(lua_State *L, int idx, const char *type) {
442 void **addr = luaL_checkudata(L, idx, type);
443 if (!obj_ref_get(L, *addr, type))
444 return NULL;
445 return *addr;
448 /* (type) check validity of object reference at stack location `idx' */
449 static void *obj_ref_check(lua_State *L, int idx, const char *type) {
450 void *obj = obj_ref_check_get(L, idx, type);
451 if (obj)
452 lua_pop(L, 1);
453 else
454 luaL_argerror(L, idx, "invalid object reference");
455 return obj;
458 static void *obj_ref_check_containerof(lua_State *L, int idx, const char *type, size_t offset) {
459 void *obj = obj_ref_check(L, idx, type);
460 return obj ? ((char*)obj-offset) : obj;
463 static void *obj_lightref_new(lua_State *L, void *addr, const char *type) {
464 if (!addr)
465 return NULL;
466 void **handle = obj_new(L, sizeof(addr), type);
467 *handle = addr;
468 return addr;
471 static void *obj_lightref_check(lua_State *L, int idx, const char *type) {
472 void **addr = luaL_checkudata(L, idx, type);
473 return *addr;
476 static int index_common(lua_State *L) {
477 lua_getmetatable(L, 1);
478 lua_pushvalue(L, 2);
479 lua_gettable(L, -2);
480 if (lua_isnil(L, -1)) {
481 lua_getuservalue(L, 1);
482 lua_pushvalue(L, 2);
483 lua_gettable(L, -2);
485 return 1;
488 static int newindex_common(lua_State *L) {
489 lua_getuservalue(L, 1);
490 lua_pushvalue(L, 2);
491 lua_pushvalue(L, 3);
492 lua_settable(L, -3);
493 return 0;
496 static size_t getpos(lua_State *L, int narg) {
497 return lua_tounsigned(L, narg);
500 static size_t checkpos(lua_State *L, int narg) {
501 lua_Number n = luaL_checknumber(L, narg);
502 if (n >= 0 && n <= SIZE_MAX && n == (size_t)n)
503 return n;
504 return luaL_argerror(L, narg, "expected position, got number");
507 static void pushpos(lua_State *L, size_t pos) {
508 if (pos == EPOS)
509 lua_pushnil(L);
510 else
511 lua_pushunsigned(L, pos);
514 static void pushrange(lua_State *L, Filerange *r) {
515 if (!r || !text_range_valid(r)) {
516 lua_pushnil(L);
517 return;
519 lua_createtable(L, 0, 2);
520 lua_pushstring(L, "start");
521 lua_pushunsigned(L, r->start);
522 lua_settable(L, -3);
523 lua_pushstring(L, "finish");
524 lua_pushunsigned(L, r->end);
525 lua_settable(L, -3);
528 static Filerange getrange(lua_State *L, int index) {
529 Filerange range = text_range_empty();
530 if (lua_istable(L, index)) {
531 lua_getfield(L, index, "start");
532 range.start = checkpos(L, -1);
533 lua_pop(L, 1);
534 lua_getfield(L, index, "finish");
535 range.end = checkpos(L, -1);
536 lua_pop(L, 1);
537 } else {
538 range.start = checkpos(L, index);
539 range.end = range.start + checkpos(L, index+1);
541 return range;
544 static const char *keymapping(Vis *vis, const char *keys, const Arg *arg) {
545 lua_State *L = vis->lua;
546 if (!func_ref_get(L, arg->v))
547 return keys;
548 lua_pushstring(L, keys);
549 if (pcall(vis, L, 1, 1) != 0)
550 return keys;
551 if (lua_type(L, -1) != LUA_TNUMBER)
552 return keys; /* invalid or no return value, assume zero */
553 lua_Number number = lua_tonumber(L, -1);
554 lua_Integer integer = lua_tointeger(L, -1);
555 if (number != integer)
556 return keys;
557 if (integer < 0)
558 return NULL; /* need more input */
559 size_t len = integer;
560 size_t max = strlen(keys);
561 return (len <= max) ? keys+len : keys;
564 /***
565 * The main editor object.
566 * @type Vis
569 /***
570 * Version information.
571 * @tfield string VERSION
572 * version information in `git describe` format, same as reported by `vis -v`.
574 /***
575 * Lua API object types
576 * @field types meta tables of userdata objects used for type checking
577 * @local
579 /***
580 * User interface.
581 * @tfield Ui ui the user interface being used
583 /***
584 * Mode constants.
585 * @tfield modes modes
587 /***
588 * Events.
589 * @tfield events events
591 /***
592 * Registers.
593 * @field registers array to access the register by single letter name
595 /***
596 * Scintillua lexer module.
597 * @field lexers might be `nil` if module is not found
599 /***
600 * LPeg lexer module.
601 * @field lpeg might be `nil` if module is not found
603 /***
604 * Current count.
605 * @tfield int count the specified count for the current command or `nil` if none was given
608 /***
609 * Create an iterator over all windows.
610 * @function windows
611 * @return the new iterator
612 * @see win
613 * @usage
614 * for win in vis:windows() do
615 * -- do something with win
616 * end
618 static int windows_iter(lua_State *L);
619 static int windows(lua_State *L) {
620 Vis *vis = obj_ref_check(L, 1, "vis");
621 Win **handle = lua_newuserdata(L, sizeof *handle);
622 *handle = vis->windows;
623 lua_pushcclosure(L, windows_iter, 1);
624 return 1;
627 static int windows_iter(lua_State *L) {
628 Win **handle = lua_touserdata(L, lua_upvalueindex(1));
629 if (!*handle)
630 return 0;
631 Win *win = obj_ref_new(L, *handle, VIS_LUA_TYPE_WINDOW);
632 if (win)
633 *handle = win->next;
634 return 1;
637 /***
638 * Create an iterator over all files.
639 * @function files
640 * @return the new iterator
641 * @usage
642 * for file in vis:files() do
643 * -- do something with file
644 * end
646 static int files_iter(lua_State *L);
647 static int files(lua_State *L) {
648 Vis *vis = obj_ref_check(L, 1, "vis");
649 File **handle = lua_newuserdata(L, sizeof *handle);
650 *handle = vis->files;
651 lua_pushcclosure(L, files_iter, 1);
652 return 1;
655 static int files_iter(lua_State *L) {
656 File **handle = lua_touserdata(L, lua_upvalueindex(1));
657 if (!*handle)
658 return 0;
659 File *file = obj_ref_new(L, *handle, VIS_LUA_TYPE_FILE);
660 if (file)
661 *handle = file->next;
662 return 1;
665 /***
666 * Create an iterator over all mark names.
667 * @function mark_names
668 * @return the new iterator
669 * @usage
670 * local marks = vis.win.file.marks;
671 * for mark in vis:mark_names() do
672 * local pos = marks[mark]
673 * if pos then
674 * -- do something with mark pos
675 * end
676 * end
678 static int mark_names_iter(lua_State *L);
679 static int mark_names(lua_State *L) {
680 enum VisMark *handle = lua_newuserdata(L, sizeof *handle);
681 *handle = 0;
682 lua_pushcclosure(L, mark_names_iter, 1);
683 return 1;
686 static int mark_names_iter(lua_State *L) {
687 char mark = '\0';
688 enum VisMark *handle = lua_touserdata(L, lua_upvalueindex(1));
689 if (*handle < LENGTH(vis_marks))
690 mark = vis_marks[*handle].name;
691 else if (VIS_MARK_a <= *handle && *handle <= VIS_MARK_z)
692 mark = 'a' + *handle - VIS_MARK_a;
693 if (mark) {
694 char name[2] = { mark, '\0' };
695 lua_pushstring(L, name);
696 (*handle)++;
697 return 1;
699 return 0;
702 /***
703 * Create an iterator over all register names.
704 * @function register_names
705 * @return the new iterator
706 * @usage
707 * for reg in vis:register_names() do
708 * local value = vis.registers[reg]
709 * if value then
710 * -- do something with register value
711 * end
712 * end
714 static int register_names_iter(lua_State *L);
715 static int register_names(lua_State *L) {
716 enum VisRegister *handle = lua_newuserdata(L, sizeof *handle);
717 *handle = 0;
718 lua_pushcclosure(L, register_names_iter, 1);
719 return 1;
722 static int register_names_iter(lua_State *L) {
723 char reg = '\0';
724 enum VisRegister *handle = lua_touserdata(L, lua_upvalueindex(1));
725 if (*handle < LENGTH(vis_registers))
726 reg = vis_registers[*handle].name;
727 else if (VIS_REG_a <= *handle && *handle <= VIS_REG_z)
728 reg = 'a' + *handle - VIS_REG_a;
729 if (reg) {
730 char name[2] = { reg, '\0' };
731 lua_pushstring(L, name);
732 (*handle)++;
733 return 1;
735 return 0;
738 /***
739 * Execute a `:`-command.
740 * @function command
741 * @tparam string command the command to execute
742 * @treturn bool whether the command succeeded
743 * @usage
744 * vis:command("set number")
746 static int command(lua_State *L) {
747 Vis *vis = obj_ref_check(L, 1, "vis");
748 const char *cmd = luaL_checkstring(L, 2);
749 bool ret = vis_cmd(vis, cmd);
750 lua_pushboolean(L, ret);
751 return 1;
754 /***
755 * Display a short message.
757 * The single line message will be displayed at the bottom of
758 * the scren and automatically hidden once a key is pressed.
760 * @function info
761 * @tparam string message the message to display
763 static int info(lua_State *L) {
764 Vis *vis = obj_ref_check(L, 1, "vis");
765 const char *msg = luaL_checkstring(L, 2);
766 vis_info_show(vis, "%s", msg);
767 return 0;
770 /***
771 * Display a multi line message.
773 * Opens a new window and displays an arbitrarily long message.
775 * @function message
776 * @tparam string message the message to display
778 static int message(lua_State *L) {
779 Vis *vis = obj_ref_check(L, 1, "vis");
780 const char *msg = luaL_checkstring(L, 2);
781 vis_message_show(vis, msg);
782 return 0;
785 /***
786 * Register a Lua function as key action.
787 * @function action_register
788 * @tparam string name the name of the action, can be referred to in key bindings as `<name>` pseudo key
789 * @tparam Function func the lua function implementing the key action (see @{keyhandler})
790 * @tparam[opt] string help the single line help text as displayed in `:help`
791 * @treturn KeyAction action the registered key action
792 * @see Vis:map
793 * @see Window:map
795 static int action_register(lua_State *L) {
796 Vis *vis = obj_ref_check(L, 1, "vis");
797 const char *name = luaL_checkstring(L, 2);
798 const void *func = func_ref_new(L, 3);
799 const char *help = luaL_optstring(L, 4, NULL);
800 KeyAction *action = vis_action_new(vis, name, help, keymapping, (Arg){ .v = func });
801 if (!action)
802 goto err;
803 if (!vis_action_register(vis, action))
804 goto err;
805 obj_ref_new(L, action, VIS_LUA_TYPE_KEYACTION);
806 return 1;
807 err:
808 vis_action_free(vis, action);
809 lua_pushnil(L);
810 return 1;
813 static int keymap(lua_State *L, Vis *vis, Win *win) {
814 int mode = luaL_checkint(L, 2);
815 const char *key = luaL_checkstring(L, 3);
816 const char *help = luaL_optstring(L, 5, NULL);
817 KeyBinding *binding = vis_binding_new(vis);
818 if (!binding)
819 goto err;
820 if (lua_isstring(L, 4)) {
821 const char *alias = luaL_checkstring(L, 4);
822 if (!(binding->alias = strdup(alias)))
823 goto err;
824 } else if (lua_isfunction(L, 4)) {
825 const void *func = func_ref_new(L, 4);
826 if (!(binding->action = vis_action_new(vis, NULL, help, keymapping, (Arg){ .v = func })))
827 goto err;
828 } else if (lua_isuserdata(L, 4)) {
829 binding->action = obj_ref_check(L, 4, VIS_LUA_TYPE_KEYACTION);
832 if (win) {
833 if (!vis_window_mode_map(win, mode, true, key, binding))
834 goto err;
835 } else {
836 if (!vis_mode_map(vis, mode, true, key, binding))
837 goto err;
840 lua_pushboolean(L, true);
841 return 1;
842 err:
843 vis_binding_free(vis, binding);
844 lua_pushboolean(L, false);
845 return 1;
848 /***
849 * Map a key to a Lua function.
851 * Creates a new key mapping in a given mode.
853 * @function map
854 * @tparam int mode the mode to which the mapping should be added
855 * @tparam string key the key to map
856 * @tparam function func the Lua function to handle the key mapping (see @{keyhandler})
857 * @tparam[opt] string help the single line help text as displayed in `:help`
858 * @treturn bool whether the mapping was successfully established
859 * @see Window:map
860 * @usage
861 * vis:map(vis.modes.INSERT, "<C-k>", function(keys)
862 * if #keys < 2 then
863 * return -1 -- need more input
864 * end
865 * local digraph = keys:sub(1, 2)
866 * if digraph == "l*" then
867 * vis:feedkeys('λ')
868 * return 2 -- consume 2 bytes of input
869 * end
870 * end, "Insert digraph")
872 /***
873 * Setup a key alias.
875 * This is equivalent to `vis:command('map! mode key alias')`.
877 * Mappings are always recursive!
878 * @function map
879 * @tparam int mode the mode to which the mapping should be added
880 * @tparam string key the key to map
881 * @tparam string alias the key to map to
882 * @treturn bool whether the mapping was successfully established
883 * @see Window:map
884 * @usage
885 * vis:map(vis.modes.NORMAL, "j", "k")
887 /***
888 * Map a key to a key action.
890 * @function map
891 * @tparam int mode the mode to which the mapping should be added
892 * @tparam string key the key to map
893 * @param action the action to map
894 * @treturn bool whether the mapping was successfully established
895 * @see Window:map
896 * @usage
897 * local action = vis:action_register("info", function()
898 * vis:info("Mapping works!")
899 * end, "Info message help text")
900 * vis:map(vis.modes.NORMAL, "gh", action)
901 * vis:map(vis.modes.NORMAL, "gl", action)
903 static int map(lua_State *L) {
904 Vis *vis = obj_ref_check(L, 1, "vis");
905 return keymap(L, vis, NULL);
908 /***
909 * Unmap a global key binding.
911 * @function unmap
912 * @tparam int mode the mode from which the mapping should be removed
913 * @tparam string key the mapping to remove
914 * @treturn bool whether the mapping was successfully removed
915 * @see Window:unmap
917 static int keyunmap(lua_State *L, Vis *vis, Win *win) {
918 enum VisMode mode = luaL_checkint(L, 2);
919 const char *key = luaL_checkstring(L, 3);
920 bool ret;
921 if (!win)
922 ret = vis_mode_unmap(vis, mode, key);
923 else
924 ret = vis_window_mode_unmap(win, mode, key);
925 lua_pushboolean(L, ret);
926 return 1;
929 static int unmap(lua_State *L) {
930 Vis *vis = obj_ref_check(L, 1, "vis");
931 return keyunmap(L, vis, NULL);
934 /***
935 * Get all currently active mappings of a mode.
937 * @function mappings
938 * @tparam int mode the mode to query
939 * @treturn table the active mappings and their associated help texts
940 * @usage
941 * local bindings = vis:mappings(vis.modes.NORMAL)
942 * for key, help in pairs(bindings) do
943 * -- do something
944 * end
945 * @see Vis:map
947 static bool binding_collect(const char *key, void *value, void *ctx) {
948 lua_State *L = ctx;
949 KeyBinding *binding = value;
950 lua_getfield(L, -1, key);
951 bool new = lua_isnil(L, -1);
952 lua_pop(L, 1);
953 if (new) {
954 const char *help = binding->alias ? binding->alias : VIS_HELP_USE(binding->action->help);
955 lua_pushstring(L, help ? help : "");
956 lua_setfield(L, -2, key);
958 return true;
961 static int mappings(lua_State *L) {
962 Vis *vis = obj_ref_check(L, 1, "vis");
963 lua_newtable(L);
964 for (Mode *mode = mode_get(vis, luaL_checkint(L, 2)); mode; mode = mode->parent) {
965 if (!mode->bindings)
966 continue;
967 map_iterate(mode->bindings, binding_collect, vis->lua);
969 return 1;
972 /***
973 * Execute a motion.
975 * @function motion
976 * @tparam int id the id of the motion to execute
977 * @treturn bool whether the id was valid
978 * @local
980 static int motion(lua_State *L) {
981 Vis *vis = obj_ref_check(L, 1, "vis");
982 enum VisMotion id = luaL_checkunsigned(L, 2);
983 // TODO handle var args?
984 lua_pushboolean(L, vis && vis_motion(vis, id));
985 return 1;
988 static size_t motion_lua(Vis *vis, Win *win, void *data, size_t pos) {
989 lua_State *L = vis->lua;
990 if (!L || !func_ref_get(L, data) || !obj_ref_new(L, win, VIS_LUA_TYPE_WINDOW))
991 return EPOS;
993 lua_pushunsigned(L, pos);
994 if (pcall(vis, L, 2, 1) != 0)
995 return EPOS;
996 return getpos(L, -1);
999 /***
1000 * Register a custom motion.
1002 * @function motion_register
1003 * @tparam function motion the Lua function implementing the motion
1004 * @treturn int the associated motion id, or `-1` on failure
1005 * @see motion, motion_new
1006 * @local
1007 * @usage
1008 * -- custom motion advancing to the next byte
1009 * local id = vis:motion_register(function(win, pos)
1010 * return pos+1
1011 * end)
1013 static int motion_register(lua_State *L) {
1014 Vis *vis = obj_ref_check(L, 1, "vis");
1015 const void *func = func_ref_new(L, 2);
1016 int id = vis_motion_register(vis, 0, (void*)func, motion_lua);
1017 lua_pushinteger(L, id);
1018 return 1;
1021 /***
1022 * Execute an operator.
1024 * @function operator
1025 * @tparam int id the id of the operator to execute
1026 * @treturn bool whether the id was valid
1027 * @local
1029 static int operator(lua_State *L) {
1030 Vis *vis = obj_ref_check(L, 1, "vis");
1031 enum VisOperator id = luaL_checkunsigned(L, 2);
1032 // TODO handle var args?
1033 lua_pushboolean(L, vis && vis_operator(vis, id));
1034 return 1;
1037 static size_t operator_lua(Vis *vis, Text *text, OperatorContext *c) {
1038 lua_State *L = vis->lua;
1039 if (!L || !func_ref_get(L, c->context))
1040 return EPOS;
1041 File *file = vis->files;
1042 while (file && (file->internal || file->text != text))
1043 file = file->next;
1044 if (!file || !obj_ref_new(L, file, VIS_LUA_TYPE_FILE))
1045 return EPOS;
1046 pushrange(L, &c->range);
1047 pushpos(L, c->pos);
1048 if (pcall(vis, L, 3, 1) != 0)
1049 return EPOS;
1050 return getpos(L, -1);
1053 /***
1054 * Register a custom operator.
1056 * @function operator_register
1057 * @tparam function operator the Lua function implementing the operator
1058 * @treturn int the associated operator id, or `-1` on failure
1059 * @see operator, operator_new
1060 * @local
1061 * @usage
1062 * -- custom operator replacing every 'a' with 'b'
1063 * local id = vis:operator_register(function(file, range, pos)
1064 * local data = file:content(range)
1065 * data = data:gsub("a", "b")
1066 * file:delete(range)
1067 * file:insert(range.start, data)
1068 * return range.start -- new cursor location
1069 * end)
1071 static int operator_register(lua_State *L) {
1072 Vis *vis = obj_ref_check(L, 1, "vis");
1073 const void *func = func_ref_new(L, 2);
1074 int id = vis_operator_register(vis, operator_lua, (void*)func);
1075 lua_pushinteger(L, id);
1076 return 1;
1079 /***
1080 * Execute a text object.
1082 * @function textobject
1083 * @tparam int id the id of the text object to execute
1084 * @treturn bool whether the id was valid
1085 * @see textobject_register, textobject_new
1086 * @local
1088 static int textobject(lua_State *L) {
1089 Vis *vis = obj_ref_check(L, 1, "vis");
1090 enum VisTextObject id = luaL_checkunsigned(L, 2);
1091 lua_pushboolean(L, vis_textobject(vis, id));
1092 return 1;
1095 static Filerange textobject_lua(Vis *vis, Win *win, void *data, size_t pos) {
1096 lua_State *L = vis->lua;
1097 if (!L || !func_ref_get(L, data) || !obj_ref_new(L, win, VIS_LUA_TYPE_WINDOW))
1098 return text_range_empty();
1099 lua_pushunsigned(L, pos);
1100 if (pcall(vis, L, 2, 2) != 0 || lua_isnil(L, -1))
1101 return text_range_empty();
1102 return text_range_new(getpos(L, -2), getpos(L, -1));
1105 /***
1106 * Register a custom text object.
1108 * @function textobject_register
1109 * @tparam function textobject the Lua function implementing the text object
1110 * @treturn int the associated text object id, or `-1` on failure
1111 * @see textobject, textobject_new
1112 * @local
1113 * @usage
1114 * -- custom text object covering the next byte
1115 * local id = vis:textobject_register(function(win, pos)
1116 * return pos, pos+1
1117 * end)
1119 static int textobject_register(lua_State *L) {
1120 Vis *vis = obj_ref_check(L, 1, "vis");
1121 const void *func = func_ref_new(L, 2);
1122 int id = vis_textobject_register(vis, 0, (void*)func, textobject_lua);
1123 lua_pushinteger(L, id);
1124 return 1;
1127 static bool option_lua(Vis *vis, Win *win, void *context, bool toggle,
1128 enum VisOption flags, const char *name, Arg *value) {
1129 lua_State *L = vis->lua;
1130 if (!L || !func_ref_get(L, context))
1131 return false;
1132 if (flags & VIS_OPTION_TYPE_BOOL)
1133 lua_pushboolean(L, value->b);
1134 else if (flags & VIS_OPTION_TYPE_STRING)
1135 lua_pushstring(L, value->s);
1136 else if (flags & VIS_OPTION_TYPE_NUMBER)
1137 lua_pushnumber(L, value->i);
1138 else
1139 return false;
1140 lua_pushboolean(L, toggle);
1141 return pcall(vis, L, 2, 2) == 0 && (!lua_isboolean(L, -1) || lua_toboolean(L, -1));
1144 /***
1145 * Register a custom `:set` option.
1147 * @function option_register
1148 * @tparam string name the option name
1149 * @tparam string type the option type (`bool`, `string` or `number`)
1150 * @tparam function handler the Lua function being called when the option is changed
1151 * @tparam[opt] string help the single line help text as displayed in `:help`
1152 * @treturn bool whether the option was successfully registered
1153 * @usage
1154 * vis:option_register("foo", "bool", function(value, toogle)
1155 * if not vis.win then return false end
1156 * vis.win.foo = toogle and not vis.win.foo or value
1157 * vis:info("Option foo = " .. tostring(vis.win.foo))
1158 * return true
1159 * end, "Foo enables superpowers")
1161 static int option_register(lua_State *L) {
1162 Vis *vis = obj_ref_check(L, 1, "vis");
1163 const char *name = luaL_checkstring(L, 2);
1164 const char *type = luaL_checkstring(L, 3);
1165 const void *func = func_ref_new(L, 4);
1166 const char *help = luaL_optstring(L, 5, NULL);
1167 const char *names[] = { name, NULL };
1168 enum VisOption flags = 0;
1169 if (strcmp(type, "string") == 0)
1170 flags |= VIS_OPTION_TYPE_STRING;
1171 else if (strcmp(type, "number") == 0)
1172 flags |= VIS_OPTION_TYPE_NUMBER;
1173 else
1174 flags |= VIS_OPTION_TYPE_BOOL;
1175 bool ret = vis_option_register(vis, names, flags, option_lua, (void*)func, help);
1176 lua_pushboolean(L, ret);
1177 return 1;
1180 /***
1181 * Unregister a `:set` option.
1183 * @function option_unregister
1184 * @tparam string name the option name
1185 * @treturn bool whether the option was successfully unregistered
1187 static int option_unregister(lua_State *L) {
1188 Vis *vis = obj_ref_check(L, 1, "vis");
1189 const char *name = luaL_checkstring(L, 2);
1190 bool ret = vis_option_unregister(vis, name);
1191 lua_pushboolean(L, ret);
1192 return 1;
1195 static bool command_lua(Vis *vis, Win *win, void *data, bool force, const char *argv[], Selection *sel, Filerange *range) {
1196 lua_State *L = vis->lua;
1197 if (!L || !func_ref_get(L, data))
1198 return false;
1199 lua_newtable(L);
1200 for (size_t i = 0; argv[i]; i++) {
1201 lua_pushunsigned(L, i);
1202 lua_pushstring(L, argv[i]);
1203 lua_settable(L, -3);
1205 lua_pushboolean(L, force);
1206 if (!obj_ref_new(L, win, VIS_LUA_TYPE_WINDOW))
1207 return false;
1208 if (!sel)
1209 sel = view_selections_primary_get(win->view);
1210 if (!obj_lightref_new(L, sel, VIS_LUA_TYPE_CURSOR))
1211 return false;
1212 pushrange(L, range);
1213 if (pcall(vis, L, 5, 1) != 0)
1214 return false;
1215 return lua_toboolean(L, -1);
1218 /***
1219 * Register a custom `:`-command.
1221 * @function command_register
1222 * @tparam string name the command name
1223 * @tparam function command the Lua function implementing the command
1224 * @tparam[opt] string help the single line help text as displayed in `:help`
1225 * @treturn bool whether the command has been successfully registered
1226 * @usage
1227 * vis:command_register("foo", function(argv, force, win, cursor, range)
1228 * for i,arg in ipairs(argv) do
1229 * print(i..": "..arg)
1230 * end
1231 * print("was command forced with ! "..(force and "yes" or "no"))
1232 * print(win.file.name)
1233 * print(cursor.pos)
1234 * print(range ~= nil and ('['..range.start..', '..range.finish..']') or "invalid range")
1235 * return true;
1236 * end)
1238 static int command_register(lua_State *L) {
1239 Vis *vis = obj_ref_check(L, 1, "vis");
1240 const char *name = luaL_checkstring(L, 2);
1241 const void *func = func_ref_new(L, 3);
1242 const char *help = luaL_optstring(L, 4, "");
1243 bool ret = vis_cmd_register(vis, name, help, (void*)func, command_lua);
1244 lua_pushboolean(L, ret);
1245 return 1;
1248 /***
1249 * Push keys to input queue and interpret them.
1251 * The keys are processed as if they were read from the keyboard.
1253 * @function feedkeys
1254 * @tparam string keys the keys to interpret
1256 static int feedkeys(lua_State *L) {
1257 Vis *vis = obj_ref_check(L, 1, "vis");
1258 const char *keys = luaL_checkstring(L, 2);
1259 vis_keys_feed(vis, keys);
1260 return 0;
1263 /***
1264 * Insert keys at all cursor positions of active window.
1266 * This function behaves as if the keys were entered in insert mode,
1267 * but in contrast to @{Vis:feedkeys} it bypasses the input queue,
1268 * meaning mappings do not apply and the keys will not be recorded in macros.
1270 * @function insert
1271 * @tparam string keys the keys to insert
1272 * @see Vis:feedkeys
1274 static int insert(lua_State *L) {
1275 Vis *vis = obj_ref_check(L, 1, "vis");
1276 size_t len;
1277 const char *keys = luaL_checklstring(L, 2, &len);
1278 vis_insert_key(vis, keys, len);
1279 return 0;
1282 /***
1283 * Replace keys at all cursor positions of active window.
1285 * This function behaves as if the keys were entered in replace mode,
1286 * but in contrast to @{Vis:feedkeys} it bypasses the input queue,
1287 * meaning mappings do not apply and the keys will not be recorded in macros.
1289 * @function replace
1290 * @tparam string keys the keys to insert
1291 * @see Vis:feedkeys
1293 static int replace(lua_State *L) {
1294 Vis *vis = obj_ref_check(L, 1, "vis");
1295 size_t len;
1296 const char *keys = luaL_checklstring(L, 2, &len);
1297 vis_replace_key(vis, keys, len);
1298 return 0;
1301 /***
1302 * Terminate editor process.
1304 * Termination happens upon the next iteration of the main event loop.
1305 * This means the calling Lua code will be executed further until it
1306 * eventually hands over control to the editor core. The exit status
1307 * of the most recent call is used.
1309 * All unsaved chanes will be lost!
1311 * @function exit
1312 * @tparam int code the exit status returned to the operating system
1314 static int exit_func(lua_State *L) {
1315 Vis *vis = obj_ref_check(L, 1, "vis");
1316 int code = luaL_checkint(L, 2);
1317 vis_exit(vis, code);
1318 return 0;
1321 /***
1322 * Pipe file range to external process and collect output.
1324 * The editor core will be blocked while the external process is running.
1326 * @function pipe
1327 * @tparam File file the file to which the range applies
1328 * @tparam Range range the range to pipe
1329 * @tparam string command the command to execute
1330 * @treturn int code the exit status of the executed command
1331 * @treturn string stdout the data written to stdout
1332 * @treturn string stderr the data written to stderr
1334 static int pipe_func(lua_State *L) {
1335 Vis *vis = obj_ref_check(L, 1, "vis");
1336 File *file = obj_ref_check(L, 2, VIS_LUA_TYPE_FILE);
1337 Filerange range = getrange(L, 3);
1338 const char *cmd = luaL_checkstring(L, 4);
1339 char *out = NULL, *err = NULL;
1340 int status = vis_pipe_collect(vis, file, &range, (const char*[]){ cmd, NULL }, &out, &err);
1341 lua_pushinteger(L, status);
1342 if (out)
1343 lua_pushstring(L, out);
1344 else
1345 lua_pushnil(L);
1346 free(out);
1347 if (err)
1348 lua_pushstring(L, err);
1349 else
1350 lua_pushnil(L);
1351 free(err);
1352 vis_draw(vis);
1353 return 3;
1355 /***
1356 * Currently active window.
1357 * @tfield Window win
1358 * @see windows
1360 /***
1361 * Currently active mode.
1362 * @tfield modes mode
1364 /***
1365 * Whether a macro is being recorded.
1366 * @tfield bool recording
1368 static int vis_index(lua_State *L) {
1369 Vis *vis = obj_ref_check(L, 1, "vis");
1371 if (lua_isstring(L, 2)) {
1372 const char *key = lua_tostring(L, 2);
1373 if (strcmp(key, "win") == 0) {
1374 if (vis->win)
1375 obj_ref_new(L, vis->win, VIS_LUA_TYPE_WINDOW);
1376 else
1377 lua_pushnil(L);
1378 return 1;
1381 if (strcmp(key, "mode") == 0) {
1382 lua_pushunsigned(L, vis->mode->id);
1383 return 1;
1386 if (strcmp(key, "recording") == 0) {
1387 lua_pushboolean(L, vis_macro_recording(vis));
1388 return 1;
1391 if (strcmp(key, "count") == 0) {
1392 int count = vis_count_get(vis);
1393 if (count == VIS_COUNT_UNKNOWN)
1394 lua_pushnil(L);
1395 else
1396 lua_pushunsigned(L, count);
1397 return 1;
1400 if (strcmp(key, "registers") == 0) {
1401 obj_ref_new(L, vis->ui, VIS_LUA_TYPE_REGISTERS);
1402 return 1;
1405 if (strcmp(key, "ui") == 0) {
1406 obj_ref_new(L, vis->ui, VIS_LUA_TYPE_UI);
1407 return 1;
1411 return index_common(L);
1414 static int vis_newindex(lua_State *L) {
1415 Vis *vis = obj_ref_check(L, 1, "vis");
1416 if (lua_isstring(L, 2)) {
1417 const char *key = lua_tostring(L, 2);
1418 if (strcmp(key, "mode") == 0) {
1419 enum VisMode mode = luaL_checkunsigned(L, 3);
1420 vis_mode_switch(vis, mode);
1421 return 0;
1424 if (strcmp(key, "count") == 0) {
1425 int count;
1426 if (lua_isnil(L, 3))
1427 count = VIS_COUNT_UNKNOWN;
1428 else
1429 count = luaL_checkunsigned(L, 3);
1430 vis_count_set(vis, count);
1431 return 0;
1434 if (strcmp(key, "win") == 0) {
1435 vis_window_focus(obj_ref_check(L, 3, VIS_LUA_TYPE_WINDOW));
1436 return 0;
1439 return newindex_common(L);
1442 static const struct luaL_Reg vis_lua[] = {
1443 { "files", files },
1444 { "windows", windows },
1445 { "mark_names", mark_names },
1446 { "register_names", register_names },
1447 { "command", command },
1448 { "info", info },
1449 { "message", message },
1450 { "map", map },
1451 { "unmap", unmap },
1452 { "mappings", mappings },
1453 { "operator", operator },
1454 { "operator_register", operator_register },
1455 { "motion", motion },
1456 { "motion_register", motion_register },
1457 { "textobject", textobject },
1458 { "textobject_register", textobject_register },
1459 { "option_register", option_register },
1460 { "option_unregister", option_unregister },
1461 { "command_register", command_register },
1462 { "feedkeys", feedkeys },
1463 { "insert", insert },
1464 { "replace", replace },
1465 { "action_register", action_register },
1466 { "exit", exit_func },
1467 { "pipe", pipe_func },
1468 { "__index", vis_index },
1469 { "__newindex", vis_newindex },
1470 { NULL, NULL },
1473 static const struct luaL_Reg ui_funcs[] = {
1474 { "__index", index_common },
1475 { NULL, NULL },
1478 static int registers_index(lua_State *L) {
1479 Vis *vis = lua_touserdata(L, lua_upvalueindex(1));
1480 const char *symbol = luaL_checkstring(L, 2);
1481 if (strlen(symbol) != 1)
1482 goto err;
1483 enum VisRegister reg = vis_register_from(vis, symbol[0]);
1484 if (reg >= VIS_REG_INVALID)
1485 goto err;
1486 size_t len;
1487 const char *value = vis_register_get(vis, reg, &len);
1488 lua_pushlstring(L, value ? value : "" , len);
1489 return 1;
1490 err:
1491 lua_pushnil(L);
1492 return 1;
1495 static int registers_newindex(lua_State *L) {
1496 Vis *vis = lua_touserdata(L, lua_upvalueindex(1));
1497 const char *symbol = luaL_checkstring(L, 2);
1498 if (strlen(symbol) != 1)
1499 return 0;
1500 enum VisRegister reg = vis_register_from(vis, symbol[0]);
1501 size_t len;
1502 const char *value = luaL_checklstring(L, 3, &len);
1503 vis_register_put(vis, reg, value, len+1);
1504 return 0;
1507 static int registers_len(lua_State *L) {
1508 Vis *vis = lua_touserdata(L, lua_upvalueindex(1));
1509 lua_pushunsigned(L, LENGTH(vis->registers));
1510 return 1;
1513 static const struct luaL_Reg registers_funcs[] = {
1514 { "__index", registers_index },
1515 { "__newindex", registers_newindex },
1516 { "__len", registers_len },
1517 { NULL, NULL },
1520 /***
1521 * A window object.
1522 * @type Window
1525 /***
1526 * Viewport currently being displayed.
1527 * @tfield Range viewport
1529 /***
1530 * The window width.
1531 * @tfield int width
1533 /***
1534 * The window height.
1535 * @tfield int height
1537 /***
1538 * The file being displayed in this window.
1539 * @tfield File file
1541 /***
1542 * The primary cursor of this window.
1543 * @tfield Cursor cursor
1545 /***
1546 * The cursors of this window.
1547 * @tfield Array(Cursor) cursors
1549 static int window_index(lua_State *L) {
1550 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1552 if (lua_isstring(L, 2)) {
1553 const char *key = lua_tostring(L, 2);
1555 if (strcmp(key, "viewport") == 0) {
1556 Filerange r = view_viewport_get(win->view);
1557 pushrange(L, &r);
1558 return 1;
1561 if (strcmp(key, "width") == 0) {
1562 lua_pushunsigned(L, vis_window_width_get(win));
1563 return 1;
1566 if (strcmp(key, "height") == 0) {
1567 lua_pushunsigned(L, vis_window_height_get(win));
1568 return 1;
1571 if (strcmp(key, "file") == 0) {
1572 obj_ref_new(L, win->file, VIS_LUA_TYPE_FILE);
1573 return 1;
1576 if (strcmp(key, "cursor") == 0) {
1577 Selection *sel = view_selections_primary_get(win->view);
1578 obj_lightref_new(L, sel, VIS_LUA_TYPE_CURSOR);
1579 return 1;
1582 if (strcmp(key, "cursors") == 0) {
1583 obj_ref_new(L, win->view, VIS_LUA_TYPE_CURSORS);
1584 return 1;
1588 return index_common(L);
1591 static int window_cursors_iterator_next(lua_State *L) {
1592 Selection **handle = lua_touserdata(L, lua_upvalueindex(1));
1593 if (!*handle)
1594 return 0;
1595 Selection *sel = obj_lightref_new(L, *handle, VIS_LUA_TYPE_CURSOR);
1596 if (!sel)
1597 return 0;
1598 *handle = view_selections_next(sel);
1599 return 1;
1602 /***
1603 * Create an iterator over all cursors of this window.
1604 * @function cursors_iterator
1605 * @return the new iterator
1607 static int window_cursors_iterator(lua_State *L) {
1608 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1609 Selection **handle = lua_newuserdata(L, sizeof *handle);
1610 *handle = view_selections(win->view);
1611 lua_pushcclosure(L, window_cursors_iterator_next, 1);
1612 return 1;
1615 /***
1616 * Set up a window local key mapping.
1617 * The function signatures are the same as for @{Vis:map}.
1618 * @function map
1619 * @param ...
1620 * @see Vis:map
1622 static int window_map(lua_State *L) {
1623 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1624 return keymap(L, win->vis, win);
1627 /***
1628 * Remove a window local key mapping.
1629 * The function signature is the same as for @{Vis:unmap}.
1630 * @function unmap
1631 * @param ...
1632 * @see Vis:unmap
1634 static int window_unmap(lua_State *L) {
1635 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1636 return keyunmap(L, win->vis, win);
1639 /***
1640 * Define a display style.
1641 * @function style_define
1642 * @tparam int id the style id to use
1643 * @tparam string style the style definition
1644 * @treturn bool whether the style definition has been successfully
1645 * associated with the given id
1646 * @see style
1647 * @usage
1648 * win:style_define(win.STYLE_DEFAULT, "fore:red")
1650 static int window_style_define(lua_State *L) {
1651 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1652 enum UiStyle id = luaL_checkunsigned(L, 2);
1653 const char *style = luaL_checkstring(L, 3);
1654 bool ret = view_style_define(win->view, id, style);
1655 lua_pushboolean(L, ret);
1656 return 1;
1659 /***
1660 * Style a window range.
1662 * The style will be cleared after every window redraw.
1663 * @function style
1664 * @tparam int id the display style as registered with @{style_define}
1665 * @tparam int start the absolute file position in bytes
1666 * @tparam int len the length in bytes to style
1667 * @see style_define
1668 * @usage
1669 * win:style(win.STYLE_DEFAULT, 0, 10)
1671 static int window_style(lua_State *L) {
1672 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1673 enum UiStyle style = luaL_checkunsigned(L, 2);
1674 size_t start = checkpos(L, 3);
1675 size_t end = checkpos(L, 4);
1676 view_style(win->view, style, start, end);
1677 return 0;
1680 /***
1681 * Set window status line.
1683 * @function status
1684 * @tparam string left the left aligned part of the status line
1685 * @tparam[opt] string right the right aligned part of the status line
1687 static int window_status(lua_State *L) {
1688 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1689 char status[1024] = "";
1690 int width = vis_window_width_get(win);
1691 const char *left = luaL_checkstring(L, 2);
1692 const char *right = luaL_optstring(L, 3, "");
1693 int left_width = text_string_width(left, strlen(left));
1694 int right_width = text_string_width(right, strlen(right));
1695 int spaces = width - left_width - right_width;
1696 if (spaces < 1)
1697 spaces = 1;
1698 snprintf(status, sizeof(status)-1, "%s%*s%s", left, spaces, " ", right);
1699 vis_window_status(win, status);
1700 return 0;
1703 /***
1704 * Redraw window content.
1706 * @function draw
1708 static int window_draw(lua_State *L) {
1709 Win *win = obj_ref_check(L, 1, VIS_LUA_TYPE_WINDOW);
1710 view_draw(win->view);
1711 return 0;
1714 static const struct luaL_Reg window_funcs[] = {
1715 { "__index", window_index },
1716 { "__newindex", newindex_common },
1717 { "cursors_iterator", window_cursors_iterator },
1718 { "map", window_map },
1719 { "unmap", window_unmap },
1720 { "style_define", window_style_define },
1721 { "style", window_style },
1722 { "status", window_status },
1723 { "draw", window_draw },
1724 { NULL, NULL },
1727 static int window_cursors_index(lua_State *L) {
1728 View *view = obj_ref_check(L, 1, VIS_LUA_TYPE_CURSORS);
1729 size_t index = luaL_checkunsigned(L, 2);
1730 size_t count = view_selections_count(view);
1731 if (index == 0 || index > count)
1732 goto err;
1733 for (Selection *s = view_selections(view); s; s = view_selections_next(s)) {
1734 if (!--index) {
1735 obj_lightref_new(L, s, VIS_LUA_TYPE_CURSOR);
1736 return 1;
1739 err:
1740 lua_pushnil(L);
1741 return 1;
1744 static int window_cursors_len(lua_State *L) {
1745 View *view = obj_ref_check(L, 1, VIS_LUA_TYPE_CURSORS);
1746 lua_pushunsigned(L, view_selections_count(view));
1747 return 1;
1750 static const struct luaL_Reg window_cursors_funcs[] = {
1751 { "__index", window_cursors_index },
1752 { "__len", window_cursors_len },
1753 { NULL, NULL },
1756 /***
1757 * A cursor object.
1759 * Cursors are represented as absolute byte offsets from the start of the file.
1760 * Valid cursor placements are within the closed interval `[0, file.size]`.
1761 * Cursors are currently implemented using character marks into the underlying
1762 * persistent [text management data structure](https://github.com/martanne/vis/wiki/Text-management-using-a-piece-chain).
1763 * This has a few consequences you should be aware of:
1765 * - A cursor becomes invalid when the underlying text range it is referencing
1766 * is deleted:
1768 * -- leaves cursor in an invalid state
1769 * win.file:delete(win.cursor.pos, 1)
1770 * assert(win.cursor.pos == nil)
1772 * Like a regular mark it will become valid again when the text is reverted
1773 * to the state before the deletion.
1775 * - Inserts after the cursor position (`> cursor.pos`) will not affect the
1776 * cursor postion.
1778 * local pos = win.cursor.pos
1779 * win.file:insert(pos+1, "-")
1780 * assert(win.cursor.pos == pos)
1782 * - Non-cached inserts before the cursor position (`<= cursor.pos`) will
1783 * affect the mark and adjust the cursor postion by the number of bytes
1784 * which were inserted.
1786 * local pos = win.cursor.pos
1787 * win.file:insert(pos, "-")
1788 * assert(win.cursor.pos == pos+1)
1790 * - Cached inserts before the cursor position (`<= cursor.pos`) will
1791 * not affect the cursor position because the underlying text is replaced
1792 * inplace.
1794 * For these reasons it is generally recommended to update the cursor position
1795 * after a modification. The general procedure amounts to:
1797 * 1. Read out the current cursor position
1798 * 2. Perform text modifications
1799 * 3. Update the cursor postion
1801 * This is what @{Vis:insert} and @{Vis:replace} do internally.
1803 * @type Cursor
1804 * @usage
1805 * local data = "new text"
1806 * local pos = win.cursor.pos
1807 * win.file:insert(pos, data)
1808 * win.cursor.pos = pos + #data
1811 /***
1812 * The zero based byte position in the file.
1814 * Might be `nil` if the cursor is in an invalid state.
1815 * Setting this field will move the cursor to the given position.
1816 * @tfield int pos
1818 /***
1819 * The 1-based line the cursor resides on.
1821 * @tfield int line
1822 * @see to
1824 /***
1825 * The 1-based column position the cursor resides on.
1826 * @tfield int col
1827 * @see to
1829 /***
1830 * The 1-based cursor index.
1831 * @tfield int number
1833 /***
1834 * The selection associated with this cursor.
1835 * @tfield Range selection the selection or `nil` if not in visual mode
1837 static int window_cursor_index(lua_State *L) {
1838 Selection *sel = obj_lightref_check(L, 1, VIS_LUA_TYPE_CURSOR);
1839 if (!sel) {
1840 lua_pushnil(L);
1841 return 1;
1844 if (lua_isstring(L, 2)) {
1845 const char *key = lua_tostring(L, 2);
1846 if (strcmp(key, "pos") == 0) {
1847 pushpos(L, view_cursors_pos(sel));
1848 return 1;
1851 if (strcmp(key, "line") == 0) {
1852 lua_pushunsigned(L, view_cursors_line(sel));
1853 return 1;
1856 if (strcmp(key, "col") == 0) {
1857 lua_pushunsigned(L, view_cursors_col(sel));
1858 return 1;
1861 if (strcmp(key, "number") == 0) {
1862 lua_pushunsigned(L, view_selections_number(sel)+1);
1863 return 1;
1866 if (strcmp(key, "selection") == 0) {
1867 Filerange range = view_selections_get(sel);
1868 pushrange(L, &range);
1869 return 1;
1873 return index_common(L);
1876 static int window_cursor_newindex(lua_State *L) {
1877 Selection *sel = obj_lightref_check(L, 1, VIS_LUA_TYPE_CURSOR);
1878 if (!sel)
1879 return 0;
1880 if (lua_isstring(L, 2)) {
1881 const char *key = lua_tostring(L, 2);
1882 if (strcmp(key, "pos") == 0) {
1883 size_t pos = checkpos(L, 3);
1884 view_cursors_to(sel, pos);
1885 return 0;
1888 if (strcmp(key, "selection") == 0) {
1889 Filerange range = getrange(L, 3);
1890 if (text_range_valid(&range)) {
1891 view_selections_set(sel, &range);
1892 view_selections_anchor(sel);
1893 } else {
1894 view_selection_clear(sel);
1896 return 0;
1899 return newindex_common(L);
1902 /***
1903 * Move cursor.
1904 * @function to
1905 * @tparam int line the 1-based line number
1906 * @tparam int col the 1-based column number
1908 static int window_cursor_to(lua_State *L) {
1909 Selection *sel = obj_lightref_check(L, 1, VIS_LUA_TYPE_CURSOR);
1910 if (sel) {
1911 size_t line = checkpos(L, 2);
1912 size_t col = checkpos(L, 3);
1913 view_cursors_place(sel, line, col);
1915 return 0;
1918 static const struct luaL_Reg window_cursor_funcs[] = {
1919 { "__index", window_cursor_index },
1920 { "__newindex", window_cursor_newindex },
1921 { "to", window_cursor_to },
1922 { NULL, NULL },
1925 /***
1926 * A file object.
1927 * @type File
1929 /***
1930 * File name.
1931 * @tfield string name the file name relative to current working directory or `nil` if not yet named
1933 /***
1934 * File path.
1935 * @tfield string path the absolute file path or `nil` if not yet named
1937 /***
1938 * File content by logical lines.
1940 * Assigning to array element `0` (`#lines+1`) will insert a new line at
1941 * the beginning (end) of the file.
1942 * @tfield Array(string) lines the file content accessible as 1-based array
1943 * @see content
1944 * @usage
1945 * local lines = vis.win.file.lines
1946 * for i=1, #lines do
1947 * lines[i] = i .. ": " .. lines[i]
1948 * end
1950 /***
1951 * File size in bytes.
1952 * @tfield int size the current file size in bytes
1954 /***
1955 * File state.
1956 * @tfield bool modified whether the file contains unsaved changes
1958 /***
1959 * File marks.
1960 * @field marks array to access the marks of this file by single letter name
1962 static int file_index(lua_State *L) {
1963 File *file = obj_ref_check(L, 1, VIS_LUA_TYPE_FILE);
1965 if (lua_isstring(L, 2)) {
1966 const char *key = lua_tostring(L, 2);
1967 if (strcmp(key, "name") == 0) {
1968 lua_pushstring(L, file_name_get(file));
1969 return 1;
1972 if (strcmp(key, "path") == 0) {
1973 lua_pushstring(L, file->name);
1974 return 1;
1977 if (strcmp(key, "lines") == 0) {
1978 obj_ref_new(L, file->text, VIS_LUA_TYPE_TEXT);
1979 return 1;
1982 if (strcmp(key, "size") == 0) {
1983 lua_pushunsigned(L, text_size(file->text));
1984 return 1;
1987 if (strcmp(key, "modified") == 0) {
1988 lua_pushboolean(L, text_modified(file->text));
1989 return 1;
1992 if (strcmp(key, "marks") == 0) {
1993 obj_ref_new(L, file->marks, VIS_LUA_TYPE_MARKS);
1994 return 1;
1998 return index_common(L);
2001 /***
2002 * Insert data at position.
2003 * @function insert
2004 * @tparam int pos the 0-based file position in bytes
2005 * @tparam string data the data to insert
2006 * @treturn bool whether the file content was successfully changed
2008 static int file_insert(lua_State *L) {
2009 File *file = obj_ref_check(L, 1, VIS_LUA_TYPE_FILE);
2010 size_t pos = checkpos(L, 2);
2011 size_t len;
2012 luaL_checkstring(L, 3);
2013 const char *data = lua_tolstring(L, 3, &len);
2014 lua_pushboolean(L, text_insert(file->text, pos, data, len));
2015 return 1;
2018 /***
2019 * Delete data at position.
2021 * @function delete
2022 * @tparam int pos the 0-based file position in bytes
2023 * @tparam int len the length in bytes to delete
2024 * @treturn bool whether the file content was successfully changed
2026 /***
2027 * Delete file range.
2029 * @function delete
2030 * @tparam Range range the range to delete
2031 * @treturn bool whether the file content was successfully changed
2033 static int file_delete(lua_State *L) {
2034 File *file = obj_ref_check(L, 1, VIS_LUA_TYPE_FILE);
2035 Filerange range = getrange(L, 2);
2036 lua_pushboolean(L, text_delete_range(file->text, &range));
2037 return 1;
2040 /***
2041 * Create an iterator over all lines of the file.
2043 * For large files this is probably faster than @{lines}.
2044 * @function lines_iterator
2045 * @return the new iterator
2046 * @see lines
2047 * @usage
2048 * for line in file:lines_iterator() do
2049 * -- do something with line
2050 * end
2052 static int file_lines_iterator_it(lua_State *L);
2053 static int file_lines_iterator(lua_State *L) {
2054 /* need to check second parameter first, because obj_ref_check_get
2055 * modifies the stack */
2056 size_t line = luaL_optunsigned(L, 2, 1);
2057 File *file = obj_ref_check_get(L, 1, VIS_LUA_TYPE_FILE);
2058 size_t *pos = lua_newuserdata(L, sizeof *pos);
2059 *pos = text_pos_by_lineno(file->text, line);
2060 lua_pushcclosure(L, file_lines_iterator_it, 2);
2061 return 1;
2064 static int file_lines_iterator_it(lua_State *L) {
2065 File *file = *(File**)lua_touserdata(L, lua_upvalueindex(1));
2066 size_t *start = lua_touserdata(L, lua_upvalueindex(2));
2067 if (*start == text_size(file->text))
2068 return 0;
2069 size_t end = text_line_end(file->text, *start);
2070 size_t len = end - *start;
2071 char *buf = malloc(len);
2072 if (!buf && len)
2073 return 0;
2074 len = text_bytes_get(file->text, *start, len, buf);
2075 lua_pushlstring(L, buf, len);
2076 free(buf);
2077 *start = text_line_next(file->text, end);
2078 return 1;
2081 /***
2082 * Get file content of position and length.
2084 * @function content
2085 * @tparam int pos the 0-based file position in bytes
2086 * @tparam int len the length in bytes to read
2087 * @treturn string the file content corresponding to the range
2088 * @see lines
2089 * @usage
2090 * local file = vis.win.file
2091 * local text = file:content(0, file.size)
2093 /***
2094 * Get file content of range.
2096 * @function content
2097 * @tparam Range range the range to read
2098 * @treturn string the file content corresponding to the range
2100 static int file_content(lua_State *L) {
2101 File *file = obj_ref_check(L, 1, VIS_LUA_TYPE_FILE);
2102 Filerange range = getrange(L, 2);
2103 if (!text_range_valid(&range))
2104 goto err;
2105 size_t len = text_range_size(&range);
2106 char *data = malloc(len);
2107 if (!data)
2108 goto err;
2109 len = text_bytes_get(file->text, range.start, len, data);
2110 lua_pushlstring(L, data, len);
2111 free(data);
2112 return 1;
2113 err:
2114 lua_pushnil(L);
2115 return 1;
2118 /***
2119 * Set mark.
2120 * @function mark_set
2121 * @tparam int pos the position to set the mark to, must be in [0, file.size]
2122 * @treturn Mark mark the mark which can be looked up later
2124 static int file_mark_set(lua_State *L) {
2125 File *file = obj_ref_check(L, 1, VIS_LUA_TYPE_FILE);
2126 size_t pos = checkpos(L, 2);
2127 Mark mark = text_mark_set(file->text, pos);
2128 if (mark)
2129 obj_lightref_new(L, (void*)mark, VIS_LUA_TYPE_MARK);
2130 else
2131 lua_pushnil(L);
2132 return 1;
2135 /***
2136 * Get position of mark.
2137 * @function mark_get
2138 * @tparam Mark mark the mark to look up
2139 * @treturn int pos the position of the mark, or `nil` if invalid
2141 static int file_mark_get(lua_State *L) {
2142 File *file = obj_ref_check(L, 1, VIS_LUA_TYPE_FILE);
2143 Mark mark = (Mark)obj_lightref_check(L, 2, VIS_LUA_TYPE_MARK);
2144 size_t pos = text_mark_get(file->text, mark);
2145 if (pos == EPOS)
2146 lua_pushnil(L);
2147 else
2148 lua_pushunsigned(L, pos);
2149 return 1;
2152 /***
2153 * Word text object.
2155 * @function text_object_word
2156 * @tparam int pos the position which must be part of the word
2157 * @treturn Range range the range
2160 static int file_text_object(lua_State *L) {
2161 Filerange range = text_range_empty();
2162 File *file = obj_ref_check(L, 1, VIS_LUA_TYPE_FILE);
2163 size_t pos = checkpos(L, 2);
2164 size_t idx = lua_tointeger(L, lua_upvalueindex(1));
2165 if (idx < LENGTH(vis_textobjects)) {
2166 const TextObject *txtobj = &vis_textobjects[idx];
2167 if (txtobj->txt)
2168 range = txtobj->txt(file->text, pos);
2170 pushrange(L, &range);
2171 return 1;
2174 static const struct luaL_Reg file_funcs[] = {
2175 { "__index", file_index },
2176 { "__newindex", newindex_common },
2177 { "insert", file_insert },
2178 { "delete", file_delete },
2179 { "lines_iterator", file_lines_iterator },
2180 { "content", file_content },
2181 { "mark_set", file_mark_set },
2182 { "mark_get", file_mark_get },
2183 { NULL, NULL },
2186 static int file_lines_index(lua_State *L) {
2187 Text *txt = obj_ref_check(L, 1, VIS_LUA_TYPE_TEXT);
2188 size_t line = luaL_checkunsigned(L, 2);
2189 size_t start = text_pos_by_lineno(txt, line);
2190 size_t end = text_line_end(txt, start);
2191 if (start != EPOS && end != EPOS) {
2192 size_t size = end - start;
2193 char *data = malloc(size);
2194 if (!data && size)
2195 goto err;
2196 size = text_bytes_get(txt, start, size, data);
2197 lua_pushlstring(L, data, size);
2198 free(data);
2199 return 1;
2201 err:
2202 lua_pushnil(L);
2203 return 1;
2206 static int file_lines_newindex(lua_State *L) {
2207 Text *txt = obj_ref_check(L, 1, VIS_LUA_TYPE_TEXT);
2208 size_t line = luaL_checkunsigned(L, 2);
2209 size_t size;
2210 const char *data = luaL_checklstring(L, 3, &size);
2211 if (line == 0) {
2212 text_insert(txt, 0, data, size);
2213 text_insert(txt, size, "\n", 1);
2214 return 0;
2216 size_t start = text_pos_by_lineno(txt, line);
2217 size_t end = text_line_end(txt, start);
2218 if (start != EPOS && end != EPOS) {
2219 text_delete(txt, start, end - start);
2220 text_insert(txt, start, data, size);
2221 if (text_size(txt) == start + size)
2222 text_insert(txt, text_size(txt), "\n", 1);
2224 return 0;
2227 static int file_lines_len(lua_State *L) {
2228 Text *txt = obj_ref_check(L, 1, VIS_LUA_TYPE_TEXT);
2229 size_t lines = 0;
2230 char lastchar;
2231 size_t size = text_size(txt);
2232 if (size > 0)
2233 lines = text_lineno_by_pos(txt, size);
2234 if (lines > 1 && text_byte_get(txt, size-1, &lastchar) && lastchar == '\n')
2235 lines--;
2236 lua_pushunsigned(L, lines);
2237 return 1;
2240 static const struct luaL_Reg file_lines_funcs[] = {
2241 { "__index", file_lines_index },
2242 { "__newindex", file_lines_newindex },
2243 { "__len", file_lines_len },
2244 { NULL, NULL },
2247 static int file_marks_index(lua_State *L) {
2248 Filerange *range = NULL;
2249 Vis *vis = lua_touserdata(L, lua_upvalueindex(1));
2250 File *file = obj_ref_check_containerof(L, 1, VIS_LUA_TYPE_MARKS, offsetof(File, marks));
2251 if (!file)
2252 goto err;
2253 const char *symbol = luaL_checkstring(L, 2);
2254 if (strlen(symbol) != 1)
2255 goto err;
2256 enum VisMark mark = vis_mark_from(vis, symbol[0]);
2257 if (mark == VIS_MARK_INVALID)
2258 goto err;
2259 Array arr = vis_register_selections_get(vis, mark);
2260 range = array_get(&arr, 0);
2261 array_release(&arr);
2262 err:
2263 pushrange(L, range);
2264 return 1;
2267 static int file_marks_newindex(lua_State *L) {
2268 Vis *vis = lua_touserdata(L, lua_upvalueindex(1));
2269 File *file = obj_ref_check_containerof(L, 1, VIS_LUA_TYPE_MARKS, offsetof(File, marks));
2270 if (!file)
2271 return 0;
2272 const char *symbol = luaL_checkstring(L, 2);
2273 if (strlen(symbol) != 1)
2274 return 0;
2275 enum VisMark mark = vis_mark_from(vis, symbol[0]);
2276 size_t pos = luaL_checkunsigned(L, 3);
2277 if (mark < LENGTH(file->marks)) {
2278 Array arr;
2279 array_init_sized(&arr, sizeof(Filerange));
2280 Filerange range = text_range_new(pos, pos);
2281 array_add(&arr, &range);
2282 vis_register_selections_set(vis, mark, &arr);
2283 array_release(&arr);
2285 return 0;
2288 static int file_marks_len(lua_State *L) {
2289 File *file = obj_ref_check_containerof(L, 1, VIS_LUA_TYPE_MARKS, offsetof(File, marks));
2290 lua_pushunsigned(L, file ? LENGTH(file->marks) : 0);
2291 return 1;
2294 static const struct luaL_Reg file_marks_funcs[] = {
2295 { "__index", file_marks_index },
2296 { "__newindex", file_marks_newindex },
2297 { "__len", file_marks_len },
2298 { NULL, NULL },
2301 /***
2302 * The user interface.
2304 * @type Ui
2306 /***
2307 * Number of available colors.
2308 * @tfield int colors
2311 /***
2312 * A file range.
2314 * For a valid range `start <= finish` holds.
2315 * An invalid range is represented as `nil`.
2316 * @type Range
2318 /***
2319 * The being of the range.
2320 * @tfield int start
2322 /***
2323 * The end of the range.
2324 * @tfield int finish
2327 /***
2328 * Modes.
2329 * @section Modes
2332 /***
2333 * Mode constants.
2334 * @table modes
2335 * @tfield int NORMAL
2336 * @tfield int OPERATOR_PENDING
2337 * @tfield int INSERT
2338 * @tfield int REPLACE
2339 * @tfield int VISUAL
2340 * @tfield int VISUAL_LINE
2341 * @see Vis:map
2342 * @see Window:map
2345 /***
2346 * Key Handling.
2348 * This section describes the contract between the editor core and Lua
2349 * key handling functions mapped to symbolic keys using either @{Vis:map}
2350 * or @{Window:map}.
2352 * @section Key_Handling
2355 /***
2356 * Example of a key handling function.
2358 * The keyhandler is invoked with the pending content of the input queue
2359 * given as argument. This might be the empty string if no further input
2360 * is available.
2362 * The function is expected to return the number of *bytes* it has
2363 * consumed from the passed input keys. A negative return value is
2364 * interpreted as an indication that not enough input was available. The
2365 * function will be called again once the user has provided more input. A
2366 * missing return value (i.e. `nil`) is interpreted as zero, meaning
2367 * no further input was consumed but the function completed successfully.
2369 * @function keyhandler
2370 * @tparam string keys the keys following the mapping
2371 * @treturn int the number of *bytes* being consumed by the function (see above)
2372 * @see Vis:action_register
2373 * @see Vis:map
2374 * @see Window:map
2375 * @usage
2376 * vis:map(vis.modes.INSERT, "<C-k>", function(keys)
2377 * if #keys < 2 then
2378 * return -1 -- need more input
2379 * end
2380 * local digraph = keys:sub(1, 2)
2381 * if digraph == "l*" then
2382 * vis:feedkeys('λ')
2383 * return 2 -- consume 2 bytes of input
2384 * end
2385 * end, "Insert digraph")
2388 /***
2389 * Core Events.
2391 * These events are invoked from the editor core.
2392 * The following functions are invoked if they are registered in the
2393 * `vis.events` table. Users scripts should generally use the [Events](#events)
2394 * mechanism instead which multiplexes these core events.
2396 * @section Core_Events
2399 static void vis_lua_event_get(lua_State *L, const char *name) {
2400 lua_getglobal(L, "vis");
2401 lua_getfield(L, -1, "events");
2402 if (lua_istable(L, -1)) {
2403 lua_getfield(L, -1, name);
2405 lua_remove(L, -2);
2408 static void vis_lua_event_call(Vis *vis, const char *name) {
2409 lua_State *L = vis->lua;
2410 vis_lua_event_get(L, name);
2411 if (lua_isfunction(L, -1))
2412 pcall(vis, L, 0, 0);
2413 lua_pop(L, 1);
2416 static bool vis_lua_path_strip(Vis *vis) {
2417 lua_State *L = vis->lua;
2418 lua_getglobal(L, "package");
2420 for (const char **var = (const char*[]){ "path", "cpath", NULL }; *var; var++) {
2422 lua_getfield(L, -1, *var);
2423 const char *path = lua_tostring(L, -1);
2424 lua_pop(L, 1);
2425 if (!path)
2426 return false;
2428 char *copy = strdup(path), *stripped = calloc(1, strlen(path)+2);
2429 if (!copy || !stripped) {
2430 free(copy);
2431 free(stripped);
2432 return false;
2435 for (char *elem = copy, *stripped_elem = stripped, *next; elem; elem = next) {
2436 if ((next = strstr(elem, ";")))
2437 *next++ = '\0';
2438 if (strstr(elem, "./"))
2439 continue; /* skip relative path entries */
2440 stripped_elem += sprintf(stripped_elem, "%s;", elem);
2443 lua_pushstring(L, stripped);
2444 lua_setfield(L, -2, *var);
2446 free(copy);
2447 free(stripped);
2450 lua_pop(L, 1); /* package */
2451 return true;
2454 bool vis_lua_path_add(Vis *vis, const char *path) {
2455 lua_State *L = vis->lua;
2456 if (!L || !path)
2457 return false;
2458 lua_getglobal(L, "package");
2459 lua_pushstring(L, path);
2460 lua_pushstring(L, "/?.lua;");
2461 lua_getfield(L, -3, "path");
2462 lua_concat(L, 3);
2463 lua_setfield(L, -2, "path");
2464 lua_pop(L, 1); /* package */
2465 return true;
2468 bool vis_lua_paths_get(Vis *vis, char **lpath, char **cpath) {
2469 lua_State *L = vis->lua;
2470 if (!L)
2471 return false;
2472 const char *s;
2473 lua_getglobal(L, "package");
2474 lua_getfield(L, -1, "path");
2475 s = lua_tostring(L, -1);
2476 *lpath = s ? strdup(s) : NULL;
2477 lua_getfield(L, -2, "cpath");
2478 s = lua_tostring(L, -1);
2479 *cpath = s ? strdup(s) : NULL;
2480 return true;
2483 static bool package_exist(Vis *vis, lua_State *L, const char *name) {
2484 const char lua[] =
2485 "local name = ...\n"
2486 "for _, searcher in ipairs(package.searchers or package.loaders) do\n"
2487 "local loader = searcher(name)\n"
2488 "if type(loader) == 'function' then\n"
2489 "return true\n"
2490 "end\n"
2491 "end\n"
2492 "return false\n";
2493 if (luaL_loadstring(L, lua) != LUA_OK)
2494 return false;
2495 lua_pushstring(L, name);
2496 /* an error indicates package exists */
2497 bool ret = lua_pcall(L, 1, 1, 0) != LUA_OK || lua_toboolean(L, -1);
2498 lua_pop(L, 1);
2499 return ret;
2502 static void *alloc_lua(void *ud, void *ptr, size_t osize, size_t nsize) {
2503 if (nsize == 0) {
2504 free(ptr);
2505 return NULL;
2506 } else {
2507 return realloc(ptr, nsize);
2511 /***
2512 * Editor initialization completed.
2513 * This event is emitted immediately after `visrc.lua` has been sourced, but
2514 * before any other events have occured, in particular the command line arguments
2515 * have not yet been processed.
2517 * Can be used to set *global* configuration options.
2518 * @function init
2520 void vis_lua_init(Vis *vis) {
2521 lua_State *L = lua_newstate(alloc_lua, vis);
2522 if (!L)
2523 return;
2524 vis->lua = L;
2525 lua_atpanic(L, &panic_handler);
2527 luaL_openlibs(L);
2529 #if CONFIG_LPEG
2530 extern int luaopen_lpeg(lua_State *L);
2531 lua_getglobal(L, "package");
2532 lua_getfield(L, -1, "preload");
2533 lua_pushcfunction(L, luaopen_lpeg);
2534 lua_setfield(L, -2, "lpeg");
2535 lua_pop(L, 2);
2536 #endif
2538 /* remove any relative paths from lua's default package.path */
2539 vis_lua_path_strip(vis);
2541 /* extends lua's package.path with:
2542 * - $VIS_PATH
2543 * - ./lua (relative path to the binary location)
2544 * - $XDG_CONFIG_HOME/vis (defaulting to $HOME/.config/vis)
2545 * - /etc/vis (for system-wide configuration provided by administrator)
2546 * - /usr/(local/)?share/vis (or whatever is specified during ./configure)
2547 * - package.path (standard lua search path)
2549 char path[PATH_MAX];
2551 vis_lua_path_add(vis, VIS_PATH);
2553 /* try to get users home directory */
2554 const char *home = getenv("HOME");
2555 if (!home || !*home) {
2556 struct passwd *pw = getpwuid(getuid());
2557 if (pw)
2558 home = pw->pw_dir;
2561 vis_lua_path_add(vis, "/etc/vis");
2563 const char *xdg_config = getenv("XDG_CONFIG_HOME");
2564 if (xdg_config) {
2565 snprintf(path, sizeof path, "%s/vis", xdg_config);
2566 vis_lua_path_add(vis, path);
2567 } else if (home && *home) {
2568 snprintf(path, sizeof path, "%s/.config/vis", home);
2569 vis_lua_path_add(vis, path);
2572 ssize_t len = readlink("/proc/self/exe", path, sizeof(path)-1);
2573 if (len > 0) {
2574 path[len] = '\0';
2575 /* some idotic dirname(3) implementations return pointers to statically
2576 * allocated memory, hence we use memmove to copy it back */
2577 char *dir = dirname(path);
2578 if (dir) {
2579 size_t len = strlen(dir)+1;
2580 if (len < sizeof(path) - sizeof("/lua")) {
2581 memmove(path, dir, len);
2582 strcat(path, "/lua");
2583 vis_lua_path_add(vis, path);
2588 vis_lua_path_add(vis, getenv("VIS_PATH"));
2590 /* table in registry to lookup object type, stores metatable -> type mapping */
2591 lua_newtable(L);
2592 lua_setfield(L, LUA_REGISTRYINDEX, "vis.types");
2593 /* table in registry to track lifetimes of C objects */
2594 lua_newtable(L);
2595 lua_setfield(L, LUA_REGISTRYINDEX, "vis.objects");
2596 /* table in registry to store references to Lua functions */
2597 lua_newtable(L);
2598 lua_setfield(L, LUA_REGISTRYINDEX, "vis.functions");
2599 /* metatable used to type check user data */
2600 obj_type_new(L, VIS_LUA_TYPE_VIS);
2601 luaL_setfuncs(L, vis_lua, 0);
2602 lua_newtable(L);
2603 lua_setfield(L, -2, "types");
2604 /* create reference to main vis object, such that the further
2605 * calls to obj_type_new can register the type meta tables in
2606 * vis.types[name] */
2607 obj_ref_new(L, vis, "vis");
2608 lua_setglobal(L, "vis");
2610 obj_type_new(L, VIS_LUA_TYPE_FILE);
2612 const struct {
2613 enum VisTextObject id;
2614 const char *name;
2615 } textobjects[] = {
2616 { VIS_TEXTOBJECT_INNER_WORD, "text_object_word" },
2619 for (size_t i = 0; i < LENGTH(textobjects); i++) {
2620 lua_pushunsigned(L, textobjects[i].id);
2621 lua_pushcclosure(L, file_text_object, 1);
2622 lua_setfield(L, -2, textobjects[i].name);
2625 luaL_setfuncs(L, file_funcs, 0);
2627 obj_type_new(L, VIS_LUA_TYPE_TEXT);
2628 luaL_setfuncs(L, file_lines_funcs, 0);
2629 obj_type_new(L, VIS_LUA_TYPE_WINDOW);
2630 luaL_setfuncs(L, window_funcs, 0);
2632 const struct {
2633 enum UiStyle id;
2634 const char *name;
2635 } styles[] = {
2636 { UI_STYLE_DEFAULT, "STYLE_DEFAULT" },
2637 { UI_STYLE_CURSOR, "STYLE_CURSOR" },
2638 { UI_STYLE_CURSOR_PRIMARY, "STYLE_CURSOR_PRIMARY" },
2639 { UI_STYLE_CURSOR_LINE, "STYLE_CURSOR_LINE" },
2640 { UI_STYLE_SELECTION, "STYLE_SELECTION" },
2641 { UI_STYLE_LINENUMBER, "STYLE_LINENUMBER" },
2642 { UI_STYLE_LINENUMBER_CURSOR, "STYLE_LINENUMBER_CURSOR" },
2643 { UI_STYLE_COLOR_COLUMN, "STYLE_COLOR_COLUMN" },
2644 { UI_STYLE_STATUS, "STYLE_STATUS" },
2645 { UI_STYLE_STATUS_FOCUSED, "STYLE_STATUS_FOCUSED" },
2646 { UI_STYLE_SEPARATOR, "STYLE_SEPARATOR" },
2647 { UI_STYLE_INFO, "STYLE_INFO" },
2648 { UI_STYLE_EOF, "STYLE_EOF" },
2651 for (size_t i = 0; i < LENGTH(styles); i++) {
2652 lua_pushunsigned(L, styles[i].id);
2653 lua_setfield(L, -2, styles[i].name);
2656 obj_type_new(L, VIS_LUA_TYPE_MARK);
2657 obj_type_new(L, VIS_LUA_TYPE_MARKS);
2658 lua_pushlightuserdata(L, vis);
2659 luaL_setfuncs(L, file_marks_funcs, 1);
2661 obj_type_new(L, VIS_LUA_TYPE_CURSOR);
2662 luaL_setfuncs(L, window_cursor_funcs, 0);
2663 obj_type_new(L, VIS_LUA_TYPE_CURSORS);
2664 luaL_setfuncs(L, window_cursors_funcs, 0);
2666 obj_type_new(L, VIS_LUA_TYPE_UI);
2667 luaL_setfuncs(L, ui_funcs, 0);
2668 lua_pushunsigned(L, vis->ui->colors(vis->ui));
2669 lua_setfield(L, -2, "colors");
2671 obj_type_new(L, VIS_LUA_TYPE_REGISTERS);
2672 lua_pushlightuserdata(L, vis);
2673 luaL_setfuncs(L, registers_funcs, 1);
2675 obj_type_new(L, VIS_LUA_TYPE_KEYACTION);
2677 lua_getglobal(L, "vis");
2678 lua_getmetatable(L, -1);
2680 lua_pushstring(L, VERSION);
2681 lua_setfield(L, -2, "VERSION");
2683 lua_newtable(L);
2685 static const struct {
2686 enum VisMode id;
2687 const char *name;
2688 } modes[] = {
2689 { VIS_MODE_NORMAL, "NORMAL" },
2690 { VIS_MODE_OPERATOR_PENDING, "OPERATOR_PENDING" },
2691 { VIS_MODE_VISUAL, "VISUAL" },
2692 { VIS_MODE_VISUAL_LINE, "VISUAL_LINE" },
2693 { VIS_MODE_INSERT, "INSERT" },
2694 { VIS_MODE_REPLACE, "REPLACE" },
2697 for (size_t i = 0; i < LENGTH(modes); i++) {
2698 lua_pushunsigned(L, modes[i].id);
2699 lua_setfield(L, -2, modes[i].name);
2702 lua_setfield(L, -2, "modes");
2704 if (!package_exist(vis, L, "visrc")) {
2705 vis_info_show(vis, "WARNING: failed to load visrc.lua");
2706 } else {
2707 lua_getglobal(L, "require");
2708 lua_pushstring(L, "visrc");
2709 pcall(vis, L, 1, 0);
2710 vis_lua_event_call(vis, "init");
2714 /***
2715 * Editor startup completed.
2716 * This event is emitted immediately before the main loop starts.
2717 * At this point all files are loaded and corresponding windows are created.
2718 * We are about to process interactive keyboard input.
2719 * @function start
2721 void vis_lua_start(Vis *vis) {
2722 vis_lua_event_call(vis, "start");
2726 * Editor is about to terminate.
2727 * @function quit
2729 void vis_lua_quit(Vis *vis) {
2730 if (!vis->lua)
2731 return;
2732 vis_lua_event_call(vis, "quit");
2733 lua_close(vis->lua);
2734 vis->lua = NULL;
2737 /***
2738 * Input key event in either input or replace mode.
2739 * @function input
2740 * @tparam string key
2741 * @treturn bool whether the key was cosumed or not
2743 static bool vis_lua_input(Vis *vis, const char *key, size_t len) {
2744 lua_State *L = vis->lua;
2745 if (!L || vis->win->file->internal)
2746 return false;
2747 bool ret = false;
2748 vis_lua_event_get(L, "input");
2749 if (lua_isfunction(L, -1)) {
2750 lua_pushlstring(L, key, len);
2751 if (pcall(vis, L, 1, 1) == 0) {
2752 ret = lua_isboolean(L, -1) && lua_toboolean(L, -1);
2753 lua_pop(L, 1);
2756 lua_pop(L, 1);
2757 return ret;
2760 void vis_lua_mode_insert_input(Vis *vis, const char *key, size_t len) {
2761 if (!vis_lua_input(vis, key, len))
2762 vis_insert_key(vis, key, len);
2765 void vis_lua_mode_replace_input(Vis *vis, const char *key, size_t len) {
2766 if (!vis_lua_input(vis, key, len))
2767 vis_replace_key(vis, key, len);
2770 /***
2771 * File open.
2772 * @function file_open
2773 * @tparam File file the file to be opened
2775 void vis_lua_file_open(Vis *vis, File *file) {
2776 debug("event: file-open: %s %p %p\n", file->name ? file->name : "unnamed", (void*)file, (void*)file->text);
2777 lua_State *L = vis->lua;
2778 if (!L)
2779 return;
2780 vis_lua_event_get(L, "file_open");
2781 if (lua_isfunction(L, -1)) {
2782 obj_ref_new(L, file, VIS_LUA_TYPE_FILE);
2783 pcall(vis, L, 1, 0);
2785 lua_pop(L, 1);
2788 /***
2789 * File pre save.
2790 * Triggered *before* the file is being written.
2791 * @function file_save_pre
2792 * @tparam File file the file being written
2793 * @tparam string path the absolute path to which the file will be written, `nil` if standard output
2794 * @treturn bool whether the write operation should be proceeded
2796 bool vis_lua_file_save_pre(Vis *vis, File *file, const char *path) {
2797 lua_State *L = vis->lua;
2798 if (!L)
2799 return true;
2800 vis_lua_event_get(L, "file_save_pre");
2801 if (lua_isfunction(L, -1)) {
2802 obj_ref_new(L, file, VIS_LUA_TYPE_FILE);
2803 lua_pushstring(L, path);
2804 if (pcall(vis, L, 2, 1) != 0)
2805 return false;
2806 return !lua_isboolean(L, -1) || lua_toboolean(L, -1);
2808 lua_pop(L, 1);
2809 return true;
2812 /***
2813 * File post save.
2814 * Triggered *after* a successfull write operation.
2815 * @function file_save_post
2816 * @tparam File file the file which was written
2817 * @tparam string path the absolute path to which it was written, `nil` if standard output
2819 void vis_lua_file_save_post(Vis *vis, File *file, const char *path) {
2820 lua_State *L = vis->lua;
2821 if (!L)
2822 return;
2823 vis_lua_event_get(L, "file_save_post");
2824 if (lua_isfunction(L, -1)) {
2825 obj_ref_new(L, file, VIS_LUA_TYPE_FILE);
2826 lua_pushstring(L, path);
2827 pcall(vis, L, 2, 0);
2829 lua_pop(L, 1);
2832 /***
2833 * File close.
2834 * The last window displaying the file has been closed.
2835 * @function file_close
2836 * @tparam File file the file being closed
2838 void vis_lua_file_close(Vis *vis, File *file) {
2839 debug("event: file-close: %s %p %p\n", file->name ? file->name : "unnamed", (void*)file, (void*)file->text);
2840 lua_State *L = vis->lua;
2841 if (!L)
2842 return;
2843 vis_lua_event_get(L, "file_close");
2844 if (lua_isfunction(L, -1)) {
2845 obj_ref_new(L, file, VIS_LUA_TYPE_FILE);
2846 pcall(vis, L, 1, 0);
2848 obj_ref_free(L, file->marks);
2849 obj_ref_free(L, file->text);
2850 obj_ref_free(L, file);
2851 lua_pop(L, 1);
2854 /***
2855 * Window open.
2856 * A new window has been created.
2857 * @function win_open
2858 * @tparam Window win the window being opened
2860 void vis_lua_win_open(Vis *vis, Win *win) {
2861 debug("event: win-open: %s %p %p\n", win->file->name ? win->file->name : "unnamed", (void*)win, (void*)win->view);
2862 lua_State *L = vis->lua;
2863 if (!L)
2864 return;
2865 vis_lua_event_get(L, "win_open");
2866 if (lua_isfunction(L, -1)) {
2867 obj_ref_new(L, win, VIS_LUA_TYPE_WINDOW);
2868 pcall(vis, L, 1, 0);
2870 lua_pop(L, 1);
2873 /***
2874 * Window close.
2875 * An window is being closed.
2876 * @function win_close
2877 * @tparam Window win the window being closed
2879 void vis_lua_win_close(Vis *vis, Win *win) {
2880 debug("event: win-close: %s %p %p\n", win->file->name ? win->file->name : "unnamed", (void*)win, (void*)win->view);
2881 lua_State *L = vis->lua;
2882 if (!L)
2883 return;
2884 vis_lua_event_get(L, "win_close");
2885 if (lua_isfunction(L, -1)) {
2886 obj_ref_new(L, win, VIS_LUA_TYPE_WINDOW);
2887 pcall(vis, L, 1, 0);
2889 obj_ref_free(L, win->view);
2890 obj_ref_free(L, win);
2891 lua_pop(L, 1);
2895 * Window highlight.
2896 * The window has been redrawn and the syntax highlighting needs to be performed.
2897 * @function win_highlight
2898 * @tparam Window win the window being redrawn
2899 * @see style
2901 void vis_lua_win_highlight(Vis *vis, Win *win) {
2902 lua_State *L = vis->lua;
2903 if (!L)
2904 return;
2905 vis_lua_event_get(L, "win_highlight");
2906 if (lua_isfunction(L, -1)) {
2907 obj_ref_new(L, win, VIS_LUA_TYPE_WINDOW);
2908 pcall(vis, L, 1, 0);
2910 lua_pop(L, 1);
2913 /***
2914 * Window status bar redraw.
2915 * @function win_status
2916 * @tparam Window win the affected window
2917 * @see status
2919 void vis_lua_win_status(Vis *vis, Win *win) {
2920 lua_State *L = vis->lua;
2921 if (!L || win->file->internal) {
2922 window_status_update(vis, win);
2923 return;
2925 vis_lua_event_get(L, "win_status");
2926 if (lua_isfunction(L, -1)) {
2927 obj_ref_new(L, win, VIS_LUA_TYPE_WINDOW);
2928 pcall(vis, L, 1, 0);
2929 } else {
2930 window_status_update(vis, win);
2932 lua_pop(L, 1);
2935 #endif