blob: b6bd67bce5d335327ae05f3474ee4c4e725eda35 [file] [log] [blame]
Bram Moolenaar261f3462019-09-07 15:45:32 +02001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * evalbuffer.c: Buffer related builtin functions
12 */
13
14#include "vim.h"
15
16#if defined(FEAT_EVAL) || defined(PROTO)
17/*
18 * Mark references in functions of buffers.
19 */
20 int
21set_ref_in_buffers(int copyID)
22{
23 int abort = FALSE;
24 buf_T *bp;
25
26 FOR_ALL_BUFFERS(bp)
27 {
28 listener_T *lnr;
Bram Moolenaar261f3462019-09-07 15:45:32 +020029
30 for (lnr = bp->b_listener; !abort && lnr != NULL; lnr = lnr->lr_next)
Yegappan Lakshmanan6ae8fae2021-12-12 16:26:44 +000031 abort = abort || set_ref_in_callback(&lnr->lr_callback, copyID);
Bram Moolenaar261f3462019-09-07 15:45:32 +020032# ifdef FEAT_JOB_CHANNEL
Yegappan Lakshmanan6ae8fae2021-12-12 16:26:44 +000033 if (!abort)
34 abort = abort || set_ref_in_callback(&bp->b_prompt_callback, copyID);
35 if (!abort)
36 abort = abort || set_ref_in_callback(&bp->b_prompt_interrupt, copyID);
Bram Moolenaar261f3462019-09-07 15:45:32 +020037# endif
Yegappan Lakshmanan6ae8fae2021-12-12 16:26:44 +000038#ifdef FEAT_COMPL_FUNC
39 if (!abort)
40 abort = abort || set_ref_in_callback(&bp->b_cfu_cb, copyID);
41 if (!abort)
42 abort = abort || set_ref_in_callback(&bp->b_ofu_cb, copyID);
43 if (!abort)
44 abort = abort || set_ref_in_callback(&bp->b_tsrfu_cb, copyID);
45#endif
46 if (!abort)
47 abort = abort || set_ref_in_callback(&bp->b_tfu_cb, copyID);
Bram Moolenaar261f3462019-09-07 15:45:32 +020048 if (abort)
49 break;
50 }
51 return abort;
52}
53
54 buf_T *
55buflist_find_by_name(char_u *name, int curtab_only)
56{
57 int save_magic;
58 char_u *save_cpo;
59 buf_T *buf;
60
61 // Ignore 'magic' and 'cpoptions' here to make scripts portable
62 save_magic = p_magic;
63 p_magic = TRUE;
64 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +010065 p_cpo = empty_option;
Bram Moolenaar261f3462019-09-07 15:45:32 +020066
67 buf = buflist_findnr(buflist_findpat(name, name + STRLEN(name),
68 TRUE, FALSE, curtab_only));
69
70 p_magic = save_magic;
71 p_cpo = save_cpo;
72 return buf;
73}
74
75/*
76 * Find a buffer by number or exact name.
77 */
78 buf_T *
79find_buffer(typval_T *avar)
80{
81 buf_T *buf = NULL;
82
83 if (avar->v_type == VAR_NUMBER)
84 buf = buflist_findnr((int)avar->vval.v_number);
Bram Moolenaar7b45d462021-03-27 19:09:02 +010085 else if (in_vim9script() && check_for_string_arg(avar, 0) == FAIL)
86 return NULL;
Bram Moolenaar261f3462019-09-07 15:45:32 +020087 else if (avar->v_type == VAR_STRING && avar->vval.v_string != NULL)
88 {
89 buf = buflist_findname_exp(avar->vval.v_string);
90 if (buf == NULL)
91 {
92 // No full path name match, try a match with a URL or a "nofile"
93 // buffer, these don't use the full path.
94 FOR_ALL_BUFFERS(buf)
95 if (buf->b_fname != NULL
96 && (path_with_url(buf->b_fname)
97#ifdef FEAT_QUICKFIX
98 || bt_nofilename(buf)
99#endif
100 )
101 && STRCMP(buf->b_fname, avar->vval.v_string) == 0)
102 break;
103 }
104 }
105 return buf;
106}
107
108/*
109 * If there is a window for "curbuf", make it the current window.
110 */
111 static void
112find_win_for_curbuf(void)
113{
114 wininfo_T *wip;
115
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200116 FOR_ALL_BUF_WININFO(curbuf, wip)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200117 {
118 if (wip->wi_win != NULL)
119 {
120 curwin = wip->wi_win;
121 break;
122 }
123 }
124}
125
126/*
Bram Moolenaar34453202021-01-31 13:08:38 +0100127 * Set line or list of lines in buffer "buf" to "lines".
128 * Any type is allowed and converted to a string.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200129 */
130 static void
131set_buffer_lines(
132 buf_T *buf,
133 linenr_T lnum_arg,
134 int append,
135 typval_T *lines,
136 typval_T *rettv)
137{
138 linenr_T lnum = lnum_arg + (append ? 1 : 0);
139 char_u *line = NULL;
140 list_T *l = NULL;
141 listitem_T *li = NULL;
142 long added = 0;
143 linenr_T append_lnum;
144 buf_T *curbuf_save = NULL;
145 win_T *curwin_save = NULL;
146 int is_curbuf = buf == curbuf;
147
148 // When using the current buffer ml_mfp will be set if needed. Useful when
149 // setline() is used on startup. For other buffers the buffer must be
150 // loaded.
151 if (buf == NULL || (!is_curbuf && buf->b_ml.ml_mfp == NULL) || lnum < 1)
152 {
153 rettv->vval.v_number = 1; // FAIL
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +0000154 if (in_vim9script() && lnum < 1)
155 semsg(_(e_invalid_line_number_nr), lnum_arg);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200156 return;
157 }
158
159 if (!is_curbuf)
160 {
161 curbuf_save = curbuf;
162 curwin_save = curwin;
163 curbuf = buf;
164 find_win_for_curbuf();
165 }
166
167 if (append)
168 // appendbufline() uses the line number below which we insert
169 append_lnum = lnum - 1;
170 else
171 // setbufline() uses the line number above which we insert, we only
172 // append if it's below the last line
173 append_lnum = curbuf->b_ml.ml_line_count;
174
175 if (lines->v_type == VAR_LIST)
176 {
177 l = lines->vval.v_list;
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200178 if (l == NULL || list_len(l) == 0)
179 {
180 // set proper return code
181 if (lnum > curbuf->b_ml.ml_line_count)
182 rettv->vval.v_number = 1; // FAIL
183 goto done;
184 }
Bram Moolenaar7e9f3512020-05-13 22:44:22 +0200185 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200186 li = l->lv_first;
187 }
188 else
Bram Moolenaar34453202021-01-31 13:08:38 +0100189 line = typval_tostring(lines, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200190
191 // default result is zero == OK
192 for (;;)
193 {
194 if (l != NULL)
195 {
196 // list argument, get next string
197 if (li == NULL)
198 break;
Bram Moolenaar34453202021-01-31 13:08:38 +0100199 vim_free(line);
200 line = typval_tostring(&li->li_tv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200201 li = li->li_next;
202 }
203
204 rettv->vval.v_number = 1; // FAIL
205 if (line == NULL || lnum > curbuf->b_ml.ml_line_count + 1)
206 break;
207
208 // When coming here from Insert mode, sync undo, so that this can be
209 // undone separately from what was previously inserted.
210 if (u_sync_once == 2)
211 {
212 u_sync_once = 1; // notify that u_sync() was called
213 u_sync(TRUE);
214 }
215
216 if (!append && lnum <= curbuf->b_ml.ml_line_count)
217 {
218 // Existing line, replace it.
219 // Removes any existing text properties.
220 if (u_savesub(lnum) == OK && ml_replace_len(
221 lnum, line, (colnr_T)STRLEN(line) + 1, TRUE, TRUE) == OK)
222 {
223 changed_bytes(lnum, 0);
224 if (is_curbuf && lnum == curwin->w_cursor.lnum)
225 check_cursor_col();
226 rettv->vval.v_number = 0; // OK
227 }
228 }
229 else if (added > 0 || u_save(lnum - 1, lnum) == OK)
230 {
231 // append the line
232 ++added;
233 if (ml_append(lnum - 1, line, (colnr_T)0, FALSE) == OK)
234 rettv->vval.v_number = 0; // OK
235 }
236
237 if (l == NULL) // only one string argument
238 break;
239 ++lnum;
240 }
Bram Moolenaar34453202021-01-31 13:08:38 +0100241 vim_free(line);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200242
243 if (added > 0)
244 {
245 win_T *wp;
246 tabpage_T *tp;
247
248 appended_lines_mark(append_lnum, added);
249
250 // Only adjust the cursor for buffers other than the current, unless it
251 // is the current window. For curbuf and other windows it has been
252 // done in mark_adjust_internal().
253 FOR_ALL_TAB_WINDOWS(tp, wp)
254 if (wp->w_buffer == buf
255 && (wp->w_buffer != curbuf || wp == curwin)
256 && wp->w_cursor.lnum > append_lnum)
257 wp->w_cursor.lnum += added;
258 check_cursor_col();
259 update_topline();
260 }
261
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200262done:
Bram Moolenaar261f3462019-09-07 15:45:32 +0200263 if (!is_curbuf)
264 {
265 curbuf = curbuf_save;
266 curwin = curwin_save;
267 }
268}
269
270/*
271 * "append(lnum, string/list)" function
272 */
273 void
274f_append(typval_T *argvars, typval_T *rettv)
275{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200276 linenr_T lnum;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000277 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200278
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200279 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
280 return;
281
282 lnum = tv_get_lnum(&argvars[0]);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000283 if (did_emsg == did_emsg_before)
284 set_buffer_lines(curbuf, lnum, TRUE, &argvars[1], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200285}
286
287/*
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200288 * Set or append lines to a buffer.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200289 */
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200290 static void
291buf_set_append_line(typval_T *argvars, typval_T *rettv, int append)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200292{
293 linenr_T lnum;
294 buf_T *buf;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000295 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200296
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200297 if (in_vim9script()
298 && (check_for_buffer_arg(argvars, 0) == FAIL
299 || check_for_lnum_arg(argvars, 1) == FAIL
300 || check_for_string_or_number_or_list_arg(argvars, 2) == FAIL))
301 return;
302
Bram Moolenaar261f3462019-09-07 15:45:32 +0200303 buf = tv_get_buf(&argvars[0], FALSE);
304 if (buf == NULL)
305 rettv->vval.v_number = 1; // FAIL
306 else
307 {
308 lnum = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000309 if (did_emsg == did_emsg_before)
310 set_buffer_lines(buf, lnum, append, &argvars[2], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200311 }
312}
313
314/*
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200315 * "appendbufline(buf, lnum, string/list)" function
316 */
317 void
318f_appendbufline(typval_T *argvars, typval_T *rettv)
319{
320 buf_set_append_line(argvars, rettv, TRUE);
321}
322
323/*
Bram Moolenaar261f3462019-09-07 15:45:32 +0200324 * "bufadd(expr)" function
325 */
326 void
327f_bufadd(typval_T *argvars, typval_T *rettv)
328{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200329 char_u *name;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200330
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200331 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
332 return;
333
334 name = tv_get_string(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200335 rettv->vval.v_number = buflist_add(*name == NUL ? NULL : name, 0);
336}
337
338/*
339 * "bufexists(expr)" function
340 */
341 void
342f_bufexists(typval_T *argvars, typval_T *rettv)
343{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200344 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
345 return;
346
Bram Moolenaar261f3462019-09-07 15:45:32 +0200347 rettv->vval.v_number = (find_buffer(&argvars[0]) != NULL);
348}
349
350/*
351 * "buflisted(expr)" function
352 */
353 void
354f_buflisted(typval_T *argvars, typval_T *rettv)
355{
356 buf_T *buf;
357
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200358 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
359 return;
360
Bram Moolenaar261f3462019-09-07 15:45:32 +0200361 buf = find_buffer(&argvars[0]);
362 rettv->vval.v_number = (buf != NULL && buf->b_p_bl);
363}
364
365/*
366 * "bufload(expr)" function
367 */
368 void
369f_bufload(typval_T *argvars, typval_T *rettv UNUSED)
370{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200371 buf_T *buf;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200372
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200373 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
374 return;
375
376 buf = get_buf_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200377 if (buf != NULL)
378 buffer_ensure_loaded(buf);
379}
380
381/*
382 * "bufloaded(expr)" function
383 */
384 void
385f_bufloaded(typval_T *argvars, typval_T *rettv)
386{
387 buf_T *buf;
388
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200389 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
390 return;
391
Bram Moolenaar261f3462019-09-07 15:45:32 +0200392 buf = find_buffer(&argvars[0]);
393 rettv->vval.v_number = (buf != NULL && buf->b_ml.ml_mfp != NULL);
394}
395
396/*
397 * "bufname(expr)" function
398 */
399 void
400f_bufname(typval_T *argvars, typval_T *rettv)
401{
402 buf_T *buf;
Bram Moolenaar02aaad92020-08-30 21:26:57 +0200403 typval_T *tv = &argvars[0];
Bram Moolenaar261f3462019-09-07 15:45:32 +0200404
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200405 if (in_vim9script() && check_for_opt_buffer_arg(argvars, 0) == FAIL)
406 return;
407
Bram Moolenaar02aaad92020-08-30 21:26:57 +0200408 if (tv->v_type == VAR_UNKNOWN)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200409 buf = curbuf;
410 else
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200411 buf = tv_get_buf_from_arg(tv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200412 rettv->v_type = VAR_STRING;
413 if (buf != NULL && buf->b_fname != NULL)
414 rettv->vval.v_string = vim_strsave(buf->b_fname);
415 else
416 rettv->vval.v_string = NULL;
417}
418
419/*
420 * "bufnr(expr)" function
421 */
422 void
423f_bufnr(typval_T *argvars, typval_T *rettv)
424{
425 buf_T *buf;
426 int error = FALSE;
427 char_u *name;
428
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200429 if (in_vim9script()
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200430 && (check_for_opt_buffer_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200431 || (argvars[0].v_type != VAR_UNKNOWN
432 && check_for_opt_bool_arg(argvars, 1) == FAIL)))
433 return;
434
Bram Moolenaar261f3462019-09-07 15:45:32 +0200435 if (argvars[0].v_type == VAR_UNKNOWN)
436 buf = curbuf;
437 else
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200438 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200439
440 // If the buffer isn't found and the second argument is not zero create a
441 // new buffer.
442 if (buf == NULL
443 && argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaarfe136c92020-09-04 18:35:26 +0200444 && tv_get_bool_chk(&argvars[1], &error) != 0
Bram Moolenaar261f3462019-09-07 15:45:32 +0200445 && !error
446 && (name = tv_get_string_chk(&argvars[0])) != NULL
447 && !error)
448 buf = buflist_new(name, NULL, (linenr_T)1, 0);
449
450 if (buf != NULL)
451 rettv->vval.v_number = buf->b_fnum;
452 else
453 rettv->vval.v_number = -1;
454}
455
456 static void
457buf_win_common(typval_T *argvars, typval_T *rettv, int get_nr)
458{
459 win_T *wp;
460 int winnr = 0;
461 buf_T *buf;
462
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200463 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
464 return;
465
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200466 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200467 FOR_ALL_WINDOWS(wp)
468 {
469 ++winnr;
470 if (wp->w_buffer == buf)
471 break;
472 }
473 rettv->vval.v_number = (wp != NULL ? (get_nr ? winnr : wp->w_id) : -1);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200474}
475
476/*
477 * "bufwinid(nr)" function
478 */
479 void
480f_bufwinid(typval_T *argvars, typval_T *rettv)
481{
482 buf_win_common(argvars, rettv, FALSE);
483}
484
485/*
486 * "bufwinnr(nr)" function
487 */
488 void
489f_bufwinnr(typval_T *argvars, typval_T *rettv)
490{
491 buf_win_common(argvars, rettv, TRUE);
492}
493
494/*
495 * "deletebufline()" function
496 */
497 void
498f_deletebufline(typval_T *argvars, typval_T *rettv)
499{
500 buf_T *buf;
501 linenr_T first, last;
502 linenr_T lnum;
503 long count;
504 int is_curbuf;
505 buf_T *curbuf_save = NULL;
506 win_T *curwin_save = NULL;
507 tabpage_T *tp;
508 win_T *wp;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000509 int did_emsg_before = did_emsg;
510
511 rettv->vval.v_number = 1; // FAIL by default
Bram Moolenaar261f3462019-09-07 15:45:32 +0200512
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200513 if (in_vim9script()
514 && (check_for_buffer_arg(argvars, 0) == FAIL
515 || check_for_lnum_arg(argvars, 1) == FAIL
516 || check_for_opt_lnum_arg(argvars, 2) == FAIL))
517 return;
518
Bram Moolenaar261f3462019-09-07 15:45:32 +0200519 buf = tv_get_buf(&argvars[0], FALSE);
520 if (buf == NULL)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200521 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200522 is_curbuf = buf == curbuf;
523
524 first = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000525 if (did_emsg > did_emsg_before)
526 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200527 if (argvars[2].v_type != VAR_UNKNOWN)
528 last = tv_get_lnum_buf(&argvars[2], buf);
529 else
530 last = first;
531
532 if (buf->b_ml.ml_mfp == NULL || first < 1
533 || first > buf->b_ml.ml_line_count || last < first)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200534 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200535
536 if (!is_curbuf)
537 {
538 curbuf_save = curbuf;
539 curwin_save = curwin;
540 curbuf = buf;
541 find_win_for_curbuf();
542 }
543 if (last > curbuf->b_ml.ml_line_count)
544 last = curbuf->b_ml.ml_line_count;
545 count = last - first + 1;
546
547 // When coming here from Insert mode, sync undo, so that this can be
548 // undone separately from what was previously inserted.
549 if (u_sync_once == 2)
550 {
551 u_sync_once = 1; // notify that u_sync() was called
552 u_sync(TRUE);
553 }
554
555 if (u_save(first - 1, last + 1) == FAIL)
556 {
557 rettv->vval.v_number = 1; // FAIL
Bram Moolenaar261f3462019-09-07 15:45:32 +0200558 }
Bram Moolenaar963ffa02021-02-09 20:02:55 +0100559 else
560 {
561 for (lnum = first; lnum <= last; ++lnum)
562 ml_delete_flags(first, ML_DEL_MESSAGE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200563
Bram Moolenaar963ffa02021-02-09 20:02:55 +0100564 FOR_ALL_TAB_WINDOWS(tp, wp)
565 if (wp->w_buffer == buf)
566 {
567 if (wp->w_cursor.lnum > last)
568 wp->w_cursor.lnum -= count;
569 else if (wp->w_cursor.lnum> first)
570 wp->w_cursor.lnum = first;
571 if (wp->w_cursor.lnum > wp->w_buffer->b_ml.ml_line_count)
572 wp->w_cursor.lnum = wp->w_buffer->b_ml.ml_line_count;
573 }
574 check_cursor_col();
575 deleted_lines_mark(first, count);
576 }
Bram Moolenaar261f3462019-09-07 15:45:32 +0200577
578 if (!is_curbuf)
579 {
580 curbuf = curbuf_save;
581 curwin = curwin_save;
582 }
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000583 rettv->vval.v_number = 0; // OK
Bram Moolenaar261f3462019-09-07 15:45:32 +0200584}
585
586/*
587 * Returns buffer options, variables and other attributes in a dictionary.
588 */
589 static dict_T *
590get_buffer_info(buf_T *buf)
591{
592 dict_T *dict;
593 tabpage_T *tp;
594 win_T *wp;
595 list_T *windows;
596
597 dict = dict_alloc();
598 if (dict == NULL)
599 return NULL;
600
601 dict_add_number(dict, "bufnr", buf->b_fnum);
602 dict_add_string(dict, "name", buf->b_ffname);
603 dict_add_number(dict, "lnum", buf == curbuf ? curwin->w_cursor.lnum
604 : buflist_findlnum(buf));
Bram Moolenaara9e96792019-12-17 22:40:15 +0100605 dict_add_number(dict, "linecount", buf->b_ml.ml_line_count);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200606 dict_add_number(dict, "loaded", buf->b_ml.ml_mfp != NULL);
607 dict_add_number(dict, "listed", buf->b_p_bl);
608 dict_add_number(dict, "changed", bufIsChanged(buf));
609 dict_add_number(dict, "changedtick", CHANGEDTICK(buf));
610 dict_add_number(dict, "hidden",
611 buf->b_ml.ml_mfp != NULL && buf->b_nwindows == 0);
612
613 // Get a reference to buffer variables
614 dict_add_dict(dict, "variables", buf->b_vars);
615
616 // List of windows displaying this buffer
617 windows = list_alloc();
618 if (windows != NULL)
619 {
620 FOR_ALL_TAB_WINDOWS(tp, wp)
621 if (wp->w_buffer == buf)
622 list_append_number(windows, (varnumber_T)wp->w_id);
623 dict_add_list(dict, "windows", windows);
624 }
625
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +0100626#ifdef FEAT_PROP_POPUP
Bram Moolenaar261f3462019-09-07 15:45:32 +0200627 // List of popup windows displaying this buffer
628 windows = list_alloc();
629 if (windows != NULL)
630 {
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200631 FOR_ALL_POPUPWINS(wp)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200632 if (wp->w_buffer == buf)
633 list_append_number(windows, (varnumber_T)wp->w_id);
634 FOR_ALL_TABPAGES(tp)
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200635 FOR_ALL_POPUPWINS_IN_TAB(tp, wp)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200636 if (wp->w_buffer == buf)
637 list_append_number(windows, (varnumber_T)wp->w_id);
638
639 dict_add_list(dict, "popups", windows);
640 }
641#endif
642
643#ifdef FEAT_SIGNS
644 if (buf->b_signlist != NULL)
645 {
646 // List of signs placed in this buffer
647 list_T *signs = list_alloc();
648 if (signs != NULL)
649 {
650 get_buffer_signs(buf, signs);
651 dict_add_list(dict, "signs", signs);
652 }
653 }
654#endif
655
Bram Moolenaar52410572019-10-27 05:12:45 +0100656#ifdef FEAT_VIMINFO
657 dict_add_number(dict, "lastused", buf->b_last_used);
658#endif
659
Bram Moolenaar261f3462019-09-07 15:45:32 +0200660 return dict;
661}
662
663/*
664 * "getbufinfo()" function
665 */
666 void
667f_getbufinfo(typval_T *argvars, typval_T *rettv)
668{
669 buf_T *buf = NULL;
670 buf_T *argbuf = NULL;
671 dict_T *d;
672 int filtered = FALSE;
673 int sel_buflisted = FALSE;
674 int sel_bufloaded = FALSE;
675 int sel_bufmodified = FALSE;
676
677 if (rettv_list_alloc(rettv) != OK)
678 return;
679
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200680 if (in_vim9script()
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200681 && check_for_opt_buffer_or_dict_arg(argvars, 0) == FAIL)
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200682 return;
683
Bram Moolenaar261f3462019-09-07 15:45:32 +0200684 // List of all the buffers or selected buffers
685 if (argvars[0].v_type == VAR_DICT)
686 {
687 dict_T *sel_d = argvars[0].vval.v_dict;
688
689 if (sel_d != NULL)
690 {
Bram Moolenaar261f3462019-09-07 15:45:32 +0200691 filtered = TRUE;
Bram Moolenaar036c2cf2020-09-05 17:37:07 +0200692 sel_buflisted = dict_get_bool(sel_d, (char_u *)"buflisted", FALSE);
693 sel_bufloaded = dict_get_bool(sel_d, (char_u *)"bufloaded", FALSE);
694 sel_bufmodified = dict_get_bool(sel_d, (char_u *)"bufmodified",
695 FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200696 }
697 }
698 else if (argvars[0].v_type != VAR_UNKNOWN)
699 {
700 // Information about one buffer. Argument specifies the buffer
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200701 argbuf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200702 if (argbuf == NULL)
703 return;
704 }
705
706 // Return information about all the buffers or a specified buffer
707 FOR_ALL_BUFFERS(buf)
708 {
709 if (argbuf != NULL && argbuf != buf)
710 continue;
711 if (filtered && ((sel_bufloaded && buf->b_ml.ml_mfp == NULL)
712 || (sel_buflisted && !buf->b_p_bl)
713 || (sel_bufmodified && !buf->b_changed)))
714 continue;
715
716 d = get_buffer_info(buf);
717 if (d != NULL)
718 list_append_dict(rettv->vval.v_list, d);
719 if (argbuf != NULL)
720 return;
721 }
722}
723
724/*
725 * Get line or list of lines from buffer "buf" into "rettv".
726 * Return a range (from start to end) of lines in rettv from the specified
727 * buffer.
728 * If 'retlist' is TRUE, then the lines are returned as a Vim List.
729 */
730 static void
731get_buffer_lines(
732 buf_T *buf,
733 linenr_T start,
734 linenr_T end,
735 int retlist,
736 typval_T *rettv)
737{
738 char_u *p;
739
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100740 if (retlist)
741 {
742 if (rettv_list_alloc(rettv) == FAIL)
743 return;
744 }
745 else
746 {
747 rettv->v_type = VAR_STRING;
748 rettv->vval.v_string = NULL;
749 }
Bram Moolenaar261f3462019-09-07 15:45:32 +0200750
751 if (buf == NULL || buf->b_ml.ml_mfp == NULL || start < 0)
752 return;
753
754 if (!retlist)
755 {
756 if (start >= 1 && start <= buf->b_ml.ml_line_count)
757 p = ml_get_buf(buf, start, FALSE);
758 else
759 p = (char_u *)"";
760 rettv->vval.v_string = vim_strsave(p);
761 }
762 else
763 {
764 if (end < start)
765 return;
766
767 if (start < 1)
768 start = 1;
769 if (end > buf->b_ml.ml_line_count)
770 end = buf->b_ml.ml_line_count;
771 while (start <= end)
772 if (list_append_string(rettv->vval.v_list,
773 ml_get_buf(buf, start++, FALSE), -1) == FAIL)
774 break;
775 }
776}
777
778/*
779 * "getbufline()" function
780 */
781 void
782f_getbufline(typval_T *argvars, typval_T *rettv)
783{
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200784 linenr_T lnum = 1;
785 linenr_T end = 1;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200786 buf_T *buf;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000787 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200788
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200789 if (in_vim9script()
790 && (check_for_buffer_arg(argvars, 0) == FAIL
791 || check_for_lnum_arg(argvars, 1) == FAIL
792 || check_for_opt_lnum_arg(argvars, 2) == FAIL))
793 return;
794
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200795 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200796 if (buf != NULL)
797 {
798 lnum = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000799 if (did_emsg > did_emsg_before)
800 return;
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200801 if (argvars[2].v_type == VAR_UNKNOWN)
802 end = lnum;
803 else
804 end = tv_get_lnum_buf(&argvars[2], buf);
805 }
Bram Moolenaar261f3462019-09-07 15:45:32 +0200806
807 get_buffer_lines(buf, lnum, end, TRUE, rettv);
808}
809
810/*
811 * "getline(lnum, [end])" function
812 */
813 void
814f_getline(typval_T *argvars, typval_T *rettv)
815{
816 linenr_T lnum;
817 linenr_T end;
818 int retlist;
819
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200820 if (in_vim9script()
821 && (check_for_lnum_arg(argvars, 0) == FAIL
822 || check_for_opt_lnum_arg(argvars, 1) == FAIL))
823 return;
824
Bram Moolenaar261f3462019-09-07 15:45:32 +0200825 lnum = tv_get_lnum(argvars);
826 if (argvars[1].v_type == VAR_UNKNOWN)
827 {
828 end = 0;
829 retlist = FALSE;
830 }
831 else
832 {
833 end = tv_get_lnum(&argvars[1]);
834 retlist = TRUE;
835 }
836
837 get_buffer_lines(curbuf, lnum, end, retlist, rettv);
838}
839
840/*
841 * "setbufline()" function
842 */
843 void
844f_setbufline(typval_T *argvars, typval_T *rettv)
845{
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200846 buf_set_append_line(argvars, rettv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200847}
848
849/*
850 * "setline()" function
851 */
852 void
853f_setline(typval_T *argvars, typval_T *rettv)
854{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200855 linenr_T lnum;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000856 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200857
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200858 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
859 return;
860
861 lnum = tv_get_lnum(&argvars[0]);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000862 if (did_emsg == did_emsg_before)
863 set_buffer_lines(curbuf, lnum, FALSE, &argvars[1], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200864}
865#endif // FEAT_EVAL
866
867#if defined(FEAT_JOB_CHANNEL) \
868 || defined(FEAT_PYTHON) || defined(FEAT_PYTHON3) \
869 || defined(PROTO)
870/*
871 * Make "buf" the current buffer. restore_buffer() MUST be called to undo.
872 * No autocommands will be executed. Use aucmd_prepbuf() if there are any.
873 */
874 void
875switch_buffer(bufref_T *save_curbuf, buf_T *buf)
876{
877 block_autocmds();
Bram Moolenaar3e0107e2021-01-02 13:53:59 +0100878#ifdef FEAT_FOLDING
879 ++disable_fold_update;
880#endif
Bram Moolenaar261f3462019-09-07 15:45:32 +0200881 set_bufref(save_curbuf, curbuf);
882 --curbuf->b_nwindows;
883 curbuf = buf;
884 curwin->w_buffer = buf;
885 ++curbuf->b_nwindows;
886}
887
888/*
889 * Restore the current buffer after using switch_buffer().
890 */
891 void
892restore_buffer(bufref_T *save_curbuf)
893{
894 unblock_autocmds();
Bram Moolenaar3e0107e2021-01-02 13:53:59 +0100895#ifdef FEAT_FOLDING
896 --disable_fold_update;
897#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100898 // Check for valid buffer, just in case.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200899 if (bufref_valid(save_curbuf))
900 {
901 --curbuf->b_nwindows;
902 curwin->w_buffer = save_curbuf->br_buf;
903 curbuf = save_curbuf->br_buf;
904 ++curbuf->b_nwindows;
905 }
906}
907
908/*
909 * Find a window for buffer "buf".
910 * If found OK is returned and "wp" and "tp" are set to the window and tabpage.
911 * If not found FAIL is returned.
912 */
913 static int
914find_win_for_buf(
915 buf_T *buf,
916 win_T **wp,
917 tabpage_T **tp)
918{
919 FOR_ALL_TAB_WINDOWS(*tp, *wp)
920 if ((*wp)->w_buffer == buf)
921 return OK;
922 return FAIL;
923}
924
925/*
926 * Find a window that contains "buf" and switch to it.
927 * If there is no such window, use the current window and change "curbuf".
928 * Caller must initialize save_curbuf to NULL.
929 * restore_win_for_buf() MUST be called later!
930 */
931 void
932switch_to_win_for_buf(
Bram Moolenaar18f47402022-01-06 13:24:51 +0000933 buf_T *buf,
934 switchwin_T *switchwin,
935 bufref_T *save_curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200936{
937 win_T *wp;
938 tabpage_T *tp;
939
940 if (find_win_for_buf(buf, &wp, &tp) == FAIL)
941 switch_buffer(save_curbuf, buf);
Bram Moolenaar18f47402022-01-06 13:24:51 +0000942 else if (switch_win(switchwin, wp, tp, TRUE) == FAIL)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200943 {
Bram Moolenaar18f47402022-01-06 13:24:51 +0000944 restore_win(switchwin, TRUE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200945 switch_buffer(save_curbuf, buf);
946 }
947}
948
949 void
950restore_win_for_buf(
Bram Moolenaar18f47402022-01-06 13:24:51 +0000951 switchwin_T *switchwin,
952 bufref_T *save_curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200953{
954 if (save_curbuf->br_buf == NULL)
Bram Moolenaar18f47402022-01-06 13:24:51 +0000955 restore_win(switchwin, TRUE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200956 else
957 restore_buffer(save_curbuf);
958}
959#endif