blob: 82a10d675c9279792d5bf98c8cb96d90c23a76af [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
Bram Moolenaar6d4b2f52022-08-25 15:11:15 +010096 && (path_with_url(buf->b_fname) || bt_nofilename(buf))
Bram Moolenaar261f3462019-09-07 15:45:32 +020097 && STRCMP(buf->b_fname, avar->vval.v_string) == 0)
98 break;
99 }
100 }
101 return buf;
102}
103
104/*
105 * If there is a window for "curbuf", make it the current window.
106 */
107 static void
108find_win_for_curbuf(void)
109{
110 wininfo_T *wip;
111
Bram Moolenaar3da85972022-11-27 19:45:49 +0000112 // The b_wininfo list should have the windows that recently contained the
113 // buffer, going over this is faster than going over all the windows.
114 // Do check the buffer is still there.
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200115 FOR_ALL_BUF_WININFO(curbuf, wip)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200116 {
Bram Moolenaar3da85972022-11-27 19:45:49 +0000117 if (wip->wi_win != NULL && wip->wi_win->w_buffer == curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200118 {
119 curwin = wip->wi_win;
120 break;
121 }
122 }
123}
124
zeertzjq228e4222022-11-20 11:13:17 +0000125typedef struct {
126 win_T *cob_curwin_save;
127 aco_save_T cob_aco;
128 int cob_using_aco;
129 int cob_save_VIsual_active;
130} cob_T;
131
132/*
133 * Used before making a change in "buf", which is not the current one: Make
134 * "buf" the current buffer and find a window for this buffer, so that side
135 * effects are done correctly (e.g., adjusting marks).
136 *
137 * Information is saved in "cob" and MUST be restored by calling
138 * change_other_buffer_restore().
139 */
140 static void
141change_other_buffer_prepare(cob_T *cob, buf_T *buf)
142{
143 CLEAR_POINTER(cob);
144
145 // Set "curbuf" to the buffer being changed. Then make sure there is a
146 // window for it to handle any side effects.
147 cob->cob_save_VIsual_active = VIsual_active;
148 VIsual_active = FALSE;
149 cob->cob_curwin_save = curwin;
150 curbuf = buf;
151 find_win_for_curbuf(); // simplest: find existing window for "buf"
152
153 if (curwin->w_buffer != buf)
154 {
155 // No existing window for this buffer. It is dangerous to have
156 // curwin->w_buffer differ from "curbuf", use the autocmd window.
157 curbuf = curwin->w_buffer;
158 aucmd_prepbuf(&cob->cob_aco, buf);
159 cob->cob_using_aco = TRUE;
160 }
161}
162
163 static void
164change_other_buffer_restore(cob_T *cob)
165{
166 if (cob->cob_using_aco)
167 {
168 aucmd_restbuf(&cob->cob_aco);
169 }
170 else
171 {
172 curwin = cob->cob_curwin_save;
173 curbuf = curwin->w_buffer;
174 }
175 VIsual_active = cob->cob_save_VIsual_active;
176}
177
Bram Moolenaar261f3462019-09-07 15:45:32 +0200178/*
Bram Moolenaar34453202021-01-31 13:08:38 +0100179 * Set line or list of lines in buffer "buf" to "lines".
180 * Any type is allowed and converted to a string.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200181 */
182 static void
183set_buffer_lines(
184 buf_T *buf,
185 linenr_T lnum_arg,
186 int append,
187 typval_T *lines,
188 typval_T *rettv)
189{
190 linenr_T lnum = lnum_arg + (append ? 1 : 0);
191 char_u *line = NULL;
192 list_T *l = NULL;
193 listitem_T *li = NULL;
194 long added = 0;
195 linenr_T append_lnum;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200196
197 // When using the current buffer ml_mfp will be set if needed. Useful when
198 // setline() is used on startup. For other buffers the buffer must be
199 // loaded.
Bram Moolenaarc934bfa2022-11-19 13:59:43 +0000200 int is_curbuf = buf == curbuf;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200201 if (buf == NULL || (!is_curbuf && buf->b_ml.ml_mfp == NULL) || lnum < 1)
202 {
203 rettv->vval.v_number = 1; // FAIL
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +0000204 if (in_vim9script() && lnum < 1)
205 semsg(_(e_invalid_line_number_nr), lnum_arg);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200206 return;
207 }
208
zeertzjq228e4222022-11-20 11:13:17 +0000209 // After this don't use "return", goto "cleanup"!
210 cob_T cob;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200211 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000212 // set "curbuf" to "buf" and find a window for this buffer
213 change_other_buffer_prepare(&cob, buf);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200214
215 if (append)
216 // appendbufline() uses the line number below which we insert
217 append_lnum = lnum - 1;
218 else
219 // setbufline() uses the line number above which we insert, we only
220 // append if it's below the last line
221 append_lnum = curbuf->b_ml.ml_line_count;
222
223 if (lines->v_type == VAR_LIST)
224 {
225 l = lines->vval.v_list;
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200226 if (l == NULL || list_len(l) == 0)
227 {
Bram Moolenaarcd9c8d42022-11-05 23:46:43 +0000228 // not appending anything always succeeds
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200229 goto done;
230 }
Bram Moolenaar7e9f3512020-05-13 22:44:22 +0200231 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200232 li = l->lv_first;
233 }
234 else
Bram Moolenaar34453202021-01-31 13:08:38 +0100235 line = typval_tostring(lines, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200236
237 // default result is zero == OK
238 for (;;)
239 {
240 if (l != NULL)
241 {
242 // list argument, get next string
243 if (li == NULL)
244 break;
Bram Moolenaar34453202021-01-31 13:08:38 +0100245 vim_free(line);
246 line = typval_tostring(&li->li_tv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200247 li = li->li_next;
248 }
249
250 rettv->vval.v_number = 1; // FAIL
251 if (line == NULL || lnum > curbuf->b_ml.ml_line_count + 1)
252 break;
253
254 // When coming here from Insert mode, sync undo, so that this can be
255 // undone separately from what was previously inserted.
256 if (u_sync_once == 2)
257 {
258 u_sync_once = 1; // notify that u_sync() was called
259 u_sync(TRUE);
260 }
261
262 if (!append && lnum <= curbuf->b_ml.ml_line_count)
263 {
264 // Existing line, replace it.
265 // Removes any existing text properties.
266 if (u_savesub(lnum) == OK && ml_replace_len(
267 lnum, line, (colnr_T)STRLEN(line) + 1, TRUE, TRUE) == OK)
268 {
269 changed_bytes(lnum, 0);
270 if (is_curbuf && lnum == curwin->w_cursor.lnum)
271 check_cursor_col();
272 rettv->vval.v_number = 0; // OK
273 }
274 }
275 else if (added > 0 || u_save(lnum - 1, lnum) == OK)
276 {
277 // append the line
278 ++added;
279 if (ml_append(lnum - 1, line, (colnr_T)0, FALSE) == OK)
280 rettv->vval.v_number = 0; // OK
281 }
282
283 if (l == NULL) // only one string argument
284 break;
285 ++lnum;
286 }
Bram Moolenaar34453202021-01-31 13:08:38 +0100287 vim_free(line);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200288
289 if (added > 0)
290 {
291 win_T *wp;
292 tabpage_T *tp;
293
294 appended_lines_mark(append_lnum, added);
295
296 // Only adjust the cursor for buffers other than the current, unless it
297 // is the current window. For curbuf and other windows it has been
298 // done in mark_adjust_internal().
299 FOR_ALL_TAB_WINDOWS(tp, wp)
300 if (wp->w_buffer == buf
301 && (wp->w_buffer != curbuf || wp == curwin)
302 && wp->w_cursor.lnum > append_lnum)
303 wp->w_cursor.lnum += added;
304 check_cursor_col();
Bram Moolenaar58a3cae2022-09-08 13:43:10 +0100305
306 // Only update the window view if w_buffer matches curbuf, otherwise
307 // the computations will be wrong.
308 if (curwin->w_buffer == curbuf)
309 update_topline();
Bram Moolenaar261f3462019-09-07 15:45:32 +0200310 }
311
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200312done:
Bram Moolenaar261f3462019-09-07 15:45:32 +0200313 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000314 change_other_buffer_restore(&cob);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200315}
316
317/*
318 * "append(lnum, string/list)" function
319 */
320 void
321f_append(typval_T *argvars, typval_T *rettv)
322{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200323 linenr_T lnum;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000324 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200325
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200326 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
327 return;
328
329 lnum = tv_get_lnum(&argvars[0]);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000330 if (did_emsg == did_emsg_before)
331 set_buffer_lines(curbuf, lnum, TRUE, &argvars[1], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200332}
333
334/*
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200335 * Set or append lines to a buffer.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200336 */
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200337 static void
338buf_set_append_line(typval_T *argvars, typval_T *rettv, int append)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200339{
340 linenr_T lnum;
341 buf_T *buf;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000342 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200343
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200344 if (in_vim9script()
345 && (check_for_buffer_arg(argvars, 0) == FAIL
346 || check_for_lnum_arg(argvars, 1) == FAIL
347 || check_for_string_or_number_or_list_arg(argvars, 2) == FAIL))
348 return;
349
Bram Moolenaar261f3462019-09-07 15:45:32 +0200350 buf = tv_get_buf(&argvars[0], FALSE);
351 if (buf == NULL)
352 rettv->vval.v_number = 1; // FAIL
353 else
354 {
355 lnum = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000356 if (did_emsg == did_emsg_before)
357 set_buffer_lines(buf, lnum, append, &argvars[2], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200358 }
359}
360
361/*
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200362 * "appendbufline(buf, lnum, string/list)" function
363 */
364 void
365f_appendbufline(typval_T *argvars, typval_T *rettv)
366{
367 buf_set_append_line(argvars, rettv, TRUE);
368}
369
370/*
Bram Moolenaar261f3462019-09-07 15:45:32 +0200371 * "bufadd(expr)" function
372 */
373 void
374f_bufadd(typval_T *argvars, typval_T *rettv)
375{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200376 char_u *name;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200377
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200378 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
379 return;
380
381 name = tv_get_string(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200382 rettv->vval.v_number = buflist_add(*name == NUL ? NULL : name, 0);
383}
384
385/*
386 * "bufexists(expr)" function
387 */
388 void
389f_bufexists(typval_T *argvars, typval_T *rettv)
390{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200391 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
392 return;
393
Bram Moolenaar261f3462019-09-07 15:45:32 +0200394 rettv->vval.v_number = (find_buffer(&argvars[0]) != NULL);
395}
396
397/*
398 * "buflisted(expr)" function
399 */
400 void
401f_buflisted(typval_T *argvars, typval_T *rettv)
402{
403 buf_T *buf;
404
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200405 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
406 return;
407
Bram Moolenaar261f3462019-09-07 15:45:32 +0200408 buf = find_buffer(&argvars[0]);
409 rettv->vval.v_number = (buf != NULL && buf->b_p_bl);
410}
411
412/*
413 * "bufload(expr)" function
414 */
415 void
416f_bufload(typval_T *argvars, typval_T *rettv UNUSED)
417{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200418 buf_T *buf;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200419
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200420 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
421 return;
422
423 buf = get_buf_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200424 if (buf != NULL)
425 buffer_ensure_loaded(buf);
426}
427
428/*
429 * "bufloaded(expr)" function
430 */
431 void
432f_bufloaded(typval_T *argvars, typval_T *rettv)
433{
434 buf_T *buf;
435
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200436 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
437 return;
438
Bram Moolenaar261f3462019-09-07 15:45:32 +0200439 buf = find_buffer(&argvars[0]);
440 rettv->vval.v_number = (buf != NULL && buf->b_ml.ml_mfp != NULL);
441}
442
443/*
444 * "bufname(expr)" function
445 */
446 void
447f_bufname(typval_T *argvars, typval_T *rettv)
448{
449 buf_T *buf;
Bram Moolenaar02aaad92020-08-30 21:26:57 +0200450 typval_T *tv = &argvars[0];
Bram Moolenaar261f3462019-09-07 15:45:32 +0200451
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200452 if (in_vim9script() && check_for_opt_buffer_arg(argvars, 0) == FAIL)
453 return;
454
Bram Moolenaar02aaad92020-08-30 21:26:57 +0200455 if (tv->v_type == VAR_UNKNOWN)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200456 buf = curbuf;
457 else
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200458 buf = tv_get_buf_from_arg(tv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200459 rettv->v_type = VAR_STRING;
460 if (buf != NULL && buf->b_fname != NULL)
461 rettv->vval.v_string = vim_strsave(buf->b_fname);
462 else
463 rettv->vval.v_string = NULL;
464}
465
466/*
467 * "bufnr(expr)" function
468 */
469 void
470f_bufnr(typval_T *argvars, typval_T *rettv)
471{
472 buf_T *buf;
473 int error = FALSE;
474 char_u *name;
475
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200476 if (in_vim9script()
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200477 && (check_for_opt_buffer_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200478 || (argvars[0].v_type != VAR_UNKNOWN
479 && check_for_opt_bool_arg(argvars, 1) == FAIL)))
480 return;
481
Bram Moolenaar261f3462019-09-07 15:45:32 +0200482 if (argvars[0].v_type == VAR_UNKNOWN)
483 buf = curbuf;
484 else
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200485 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200486
487 // If the buffer isn't found and the second argument is not zero create a
488 // new buffer.
489 if (buf == NULL
490 && argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaarfe136c92020-09-04 18:35:26 +0200491 && tv_get_bool_chk(&argvars[1], &error) != 0
Bram Moolenaar261f3462019-09-07 15:45:32 +0200492 && !error
493 && (name = tv_get_string_chk(&argvars[0])) != NULL
494 && !error)
495 buf = buflist_new(name, NULL, (linenr_T)1, 0);
496
497 if (buf != NULL)
498 rettv->vval.v_number = buf->b_fnum;
499 else
500 rettv->vval.v_number = -1;
501}
502
503 static void
504buf_win_common(typval_T *argvars, typval_T *rettv, int get_nr)
505{
506 win_T *wp;
507 int winnr = 0;
508 buf_T *buf;
509
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200510 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
511 return;
512
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200513 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200514 FOR_ALL_WINDOWS(wp)
515 {
516 ++winnr;
517 if (wp->w_buffer == buf)
518 break;
519 }
520 rettv->vval.v_number = (wp != NULL ? (get_nr ? winnr : wp->w_id) : -1);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200521}
522
523/*
524 * "bufwinid(nr)" function
525 */
526 void
527f_bufwinid(typval_T *argvars, typval_T *rettv)
528{
529 buf_win_common(argvars, rettv, FALSE);
530}
531
532/*
533 * "bufwinnr(nr)" function
534 */
535 void
536f_bufwinnr(typval_T *argvars, typval_T *rettv)
537{
538 buf_win_common(argvars, rettv, TRUE);
539}
540
541/*
542 * "deletebufline()" function
543 */
544 void
545f_deletebufline(typval_T *argvars, typval_T *rettv)
546{
547 buf_T *buf;
548 linenr_T first, last;
549 linenr_T lnum;
550 long count;
551 int is_curbuf;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200552 tabpage_T *tp;
553 win_T *wp;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000554 int did_emsg_before = did_emsg;
555
556 rettv->vval.v_number = 1; // FAIL by default
Bram Moolenaar261f3462019-09-07 15:45:32 +0200557
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200558 if (in_vim9script()
559 && (check_for_buffer_arg(argvars, 0) == FAIL
560 || check_for_lnum_arg(argvars, 1) == FAIL
561 || check_for_opt_lnum_arg(argvars, 2) == FAIL))
562 return;
563
Bram Moolenaar261f3462019-09-07 15:45:32 +0200564 buf = tv_get_buf(&argvars[0], FALSE);
565 if (buf == NULL)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200566 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200567
568 first = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000569 if (did_emsg > did_emsg_before)
570 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200571 if (argvars[2].v_type != VAR_UNKNOWN)
572 last = tv_get_lnum_buf(&argvars[2], buf);
573 else
574 last = first;
575
576 if (buf->b_ml.ml_mfp == NULL || first < 1
577 || first > buf->b_ml.ml_line_count || last < first)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200578 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200579
zeertzjq7af3ee22022-11-06 22:26:05 +0000580 // After this don't use "return", goto "cleanup"!
zeertzjq228e4222022-11-20 11:13:17 +0000581 is_curbuf = buf == curbuf;
582 cob_T cob;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200583 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000584 // set "curbuf" to "buf" and find a window for this buffer
585 change_other_buffer_prepare(&cob, buf);
586
Bram Moolenaar261f3462019-09-07 15:45:32 +0200587 if (last > curbuf->b_ml.ml_line_count)
588 last = curbuf->b_ml.ml_line_count;
589 count = last - first + 1;
590
591 // When coming here from Insert mode, sync undo, so that this can be
592 // undone separately from what was previously inserted.
593 if (u_sync_once == 2)
594 {
595 u_sync_once = 1; // notify that u_sync() was called
596 u_sync(TRUE);
597 }
598
599 if (u_save(first - 1, last + 1) == FAIL)
zeertzjq7af3ee22022-11-06 22:26:05 +0000600 goto cleanup;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200601
zeertzjq7af3ee22022-11-06 22:26:05 +0000602 for (lnum = first; lnum <= last; ++lnum)
603 ml_delete_flags(first, ML_DEL_MESSAGE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200604
zeertzjq7af3ee22022-11-06 22:26:05 +0000605 FOR_ALL_TAB_WINDOWS(tp, wp)
606 if (wp->w_buffer == buf)
607 {
608 if (wp->w_cursor.lnum > last)
609 wp->w_cursor.lnum -= count;
610 else if (wp->w_cursor.lnum > first)
611 wp->w_cursor.lnum = first;
612 if (wp->w_cursor.lnum > wp->w_buffer->b_ml.ml_line_count)
613 wp->w_cursor.lnum = wp->w_buffer->b_ml.ml_line_count;
614 wp->w_valid = 0;
615 if (wp->w_cursor.lnum <= wp->w_topline)
616 wp->w_topline = 1;
617 }
618 check_cursor_col();
619 deleted_lines_mark(first, count);
620 rettv->vval.v_number = 0; // OK
621
622cleanup:
Bram Moolenaar261f3462019-09-07 15:45:32 +0200623 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000624 change_other_buffer_restore(&cob);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200625}
626
627/*
628 * Returns buffer options, variables and other attributes in a dictionary.
629 */
630 static dict_T *
631get_buffer_info(buf_T *buf)
632{
633 dict_T *dict;
634 tabpage_T *tp;
635 win_T *wp;
636 list_T *windows;
637
638 dict = dict_alloc();
639 if (dict == NULL)
640 return NULL;
641
642 dict_add_number(dict, "bufnr", buf->b_fnum);
643 dict_add_string(dict, "name", buf->b_ffname);
644 dict_add_number(dict, "lnum", buf == curbuf ? curwin->w_cursor.lnum
645 : buflist_findlnum(buf));
Bram Moolenaara9e96792019-12-17 22:40:15 +0100646 dict_add_number(dict, "linecount", buf->b_ml.ml_line_count);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200647 dict_add_number(dict, "loaded", buf->b_ml.ml_mfp != NULL);
648 dict_add_number(dict, "listed", buf->b_p_bl);
649 dict_add_number(dict, "changed", bufIsChanged(buf));
650 dict_add_number(dict, "changedtick", CHANGEDTICK(buf));
651 dict_add_number(dict, "hidden",
652 buf->b_ml.ml_mfp != NULL && buf->b_nwindows == 0);
653
654 // Get a reference to buffer variables
655 dict_add_dict(dict, "variables", buf->b_vars);
656
657 // List of windows displaying this buffer
658 windows = list_alloc();
659 if (windows != NULL)
660 {
661 FOR_ALL_TAB_WINDOWS(tp, wp)
662 if (wp->w_buffer == buf)
663 list_append_number(windows, (varnumber_T)wp->w_id);
664 dict_add_list(dict, "windows", windows);
665 }
666
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +0100667#ifdef FEAT_PROP_POPUP
Bram Moolenaar261f3462019-09-07 15:45:32 +0200668 // List of popup windows displaying this buffer
669 windows = list_alloc();
670 if (windows != NULL)
671 {
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200672 FOR_ALL_POPUPWINS(wp)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200673 if (wp->w_buffer == buf)
674 list_append_number(windows, (varnumber_T)wp->w_id);
675 FOR_ALL_TABPAGES(tp)
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200676 FOR_ALL_POPUPWINS_IN_TAB(tp, wp)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200677 if (wp->w_buffer == buf)
678 list_append_number(windows, (varnumber_T)wp->w_id);
679
680 dict_add_list(dict, "popups", windows);
681 }
682#endif
683
684#ifdef FEAT_SIGNS
685 if (buf->b_signlist != NULL)
686 {
687 // List of signs placed in this buffer
688 list_T *signs = list_alloc();
689 if (signs != NULL)
690 {
691 get_buffer_signs(buf, signs);
692 dict_add_list(dict, "signs", signs);
693 }
694 }
695#endif
696
Bram Moolenaar52410572019-10-27 05:12:45 +0100697#ifdef FEAT_VIMINFO
698 dict_add_number(dict, "lastused", buf->b_last_used);
699#endif
700
Bram Moolenaar261f3462019-09-07 15:45:32 +0200701 return dict;
702}
703
704/*
705 * "getbufinfo()" function
706 */
707 void
708f_getbufinfo(typval_T *argvars, typval_T *rettv)
709{
710 buf_T *buf = NULL;
711 buf_T *argbuf = NULL;
712 dict_T *d;
713 int filtered = FALSE;
714 int sel_buflisted = FALSE;
715 int sel_bufloaded = FALSE;
716 int sel_bufmodified = FALSE;
717
Bram Moolenaar93a10962022-06-16 11:42:09 +0100718 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200719 return;
720
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200721 if (in_vim9script()
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200722 && check_for_opt_buffer_or_dict_arg(argvars, 0) == FAIL)
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200723 return;
724
Bram Moolenaar261f3462019-09-07 15:45:32 +0200725 // List of all the buffers or selected buffers
726 if (argvars[0].v_type == VAR_DICT)
727 {
728 dict_T *sel_d = argvars[0].vval.v_dict;
729
730 if (sel_d != NULL)
731 {
Bram Moolenaar261f3462019-09-07 15:45:32 +0200732 filtered = TRUE;
Bram Moolenaard61efa52022-07-23 09:52:04 +0100733 sel_buflisted = dict_get_bool(sel_d, "buflisted", FALSE);
734 sel_bufloaded = dict_get_bool(sel_d, "bufloaded", FALSE);
735 sel_bufmodified = dict_get_bool(sel_d, "bufmodified",
Bram Moolenaar036c2cf2020-09-05 17:37:07 +0200736 FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200737 }
738 }
739 else if (argvars[0].v_type != VAR_UNKNOWN)
740 {
741 // Information about one buffer. Argument specifies the buffer
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200742 argbuf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200743 if (argbuf == NULL)
744 return;
745 }
746
747 // Return information about all the buffers or a specified buffer
748 FOR_ALL_BUFFERS(buf)
749 {
750 if (argbuf != NULL && argbuf != buf)
751 continue;
752 if (filtered && ((sel_bufloaded && buf->b_ml.ml_mfp == NULL)
753 || (sel_buflisted && !buf->b_p_bl)
754 || (sel_bufmodified && !buf->b_changed)))
755 continue;
756
757 d = get_buffer_info(buf);
758 if (d != NULL)
759 list_append_dict(rettv->vval.v_list, d);
760 if (argbuf != NULL)
761 return;
762 }
763}
764
765/*
766 * Get line or list of lines from buffer "buf" into "rettv".
767 * Return a range (from start to end) of lines in rettv from the specified
768 * buffer.
769 * If 'retlist' is TRUE, then the lines are returned as a Vim List.
770 */
771 static void
772get_buffer_lines(
773 buf_T *buf,
774 linenr_T start,
775 linenr_T end,
776 int retlist,
777 typval_T *rettv)
778{
779 char_u *p;
780
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100781 if (retlist)
782 {
783 if (rettv_list_alloc(rettv) == FAIL)
784 return;
785 }
786 else
787 {
788 rettv->v_type = VAR_STRING;
789 rettv->vval.v_string = NULL;
790 }
Bram Moolenaar261f3462019-09-07 15:45:32 +0200791
792 if (buf == NULL || buf->b_ml.ml_mfp == NULL || start < 0)
793 return;
794
795 if (!retlist)
796 {
797 if (start >= 1 && start <= buf->b_ml.ml_line_count)
798 p = ml_get_buf(buf, start, FALSE);
799 else
800 p = (char_u *)"";
801 rettv->vval.v_string = vim_strsave(p);
802 }
803 else
804 {
805 if (end < start)
806 return;
807
808 if (start < 1)
809 start = 1;
810 if (end > buf->b_ml.ml_line_count)
811 end = buf->b_ml.ml_line_count;
812 while (start <= end)
813 if (list_append_string(rettv->vval.v_list,
814 ml_get_buf(buf, start++, FALSE), -1) == FAIL)
815 break;
816 }
817}
818
819/*
Bram Moolenaarce30ccc2022-11-21 19:57:04 +0000820 * "retlist" TRUE: "getbufline()" function
821 * "retlist" FALSE: "getbufoneline()" function
Bram Moolenaar261f3462019-09-07 15:45:32 +0200822 */
Bram Moolenaarce30ccc2022-11-21 19:57:04 +0000823 static void
824getbufline(typval_T *argvars, typval_T *rettv, int retlist)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200825{
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200826 linenr_T lnum = 1;
827 linenr_T end = 1;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200828 buf_T *buf;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000829 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200830
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200831 if (in_vim9script()
832 && (check_for_buffer_arg(argvars, 0) == FAIL
833 || check_for_lnum_arg(argvars, 1) == FAIL
834 || check_for_opt_lnum_arg(argvars, 2) == FAIL))
835 return;
836
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200837 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200838 if (buf != NULL)
839 {
840 lnum = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000841 if (did_emsg > did_emsg_before)
842 return;
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200843 if (argvars[2].v_type == VAR_UNKNOWN)
844 end = lnum;
845 else
846 end = tv_get_lnum_buf(&argvars[2], buf);
847 }
Bram Moolenaar261f3462019-09-07 15:45:32 +0200848
Bram Moolenaarce30ccc2022-11-21 19:57:04 +0000849 get_buffer_lines(buf, lnum, end, retlist, rettv);
850}
851
852/*
853 * "getbufline()" function
854 */
855 void
856f_getbufline(typval_T *argvars, typval_T *rettv)
857{
858 getbufline(argvars, rettv, TRUE);
859}
860
861/*
862 * "getbufoneline()" function
863 */
864 void
865f_getbufoneline(typval_T *argvars, typval_T *rettv)
866{
867 getbufline(argvars, rettv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200868}
869
870/*
871 * "getline(lnum, [end])" function
872 */
873 void
874f_getline(typval_T *argvars, typval_T *rettv)
875{
876 linenr_T lnum;
877 linenr_T end;
878 int retlist;
879
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200880 if (in_vim9script()
881 && (check_for_lnum_arg(argvars, 0) == FAIL
882 || check_for_opt_lnum_arg(argvars, 1) == FAIL))
883 return;
884
Bram Moolenaar261f3462019-09-07 15:45:32 +0200885 lnum = tv_get_lnum(argvars);
886 if (argvars[1].v_type == VAR_UNKNOWN)
887 {
888 end = 0;
889 retlist = FALSE;
890 }
891 else
892 {
893 end = tv_get_lnum(&argvars[1]);
894 retlist = TRUE;
895 }
896
897 get_buffer_lines(curbuf, lnum, end, retlist, rettv);
898}
899
900/*
901 * "setbufline()" function
902 */
903 void
904f_setbufline(typval_T *argvars, typval_T *rettv)
905{
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200906 buf_set_append_line(argvars, rettv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200907}
908
909/*
910 * "setline()" function
911 */
912 void
913f_setline(typval_T *argvars, typval_T *rettv)
914{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200915 linenr_T lnum;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000916 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200917
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200918 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
919 return;
920
921 lnum = tv_get_lnum(&argvars[0]);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000922 if (did_emsg == did_emsg_before)
923 set_buffer_lines(curbuf, lnum, FALSE, &argvars[1], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200924}
925#endif // FEAT_EVAL
926
Dominique Pelle748b3082022-01-08 12:41:16 +0000927#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3) || defined(PROTO)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200928/*
929 * Make "buf" the current buffer. restore_buffer() MUST be called to undo.
930 * No autocommands will be executed. Use aucmd_prepbuf() if there are any.
931 */
932 void
933switch_buffer(bufref_T *save_curbuf, buf_T *buf)
934{
935 block_autocmds();
Bram Moolenaar3e0107e2021-01-02 13:53:59 +0100936#ifdef FEAT_FOLDING
937 ++disable_fold_update;
938#endif
Bram Moolenaar261f3462019-09-07 15:45:32 +0200939 set_bufref(save_curbuf, curbuf);
940 --curbuf->b_nwindows;
941 curbuf = buf;
942 curwin->w_buffer = buf;
943 ++curbuf->b_nwindows;
944}
945
946/*
947 * Restore the current buffer after using switch_buffer().
948 */
949 void
950restore_buffer(bufref_T *save_curbuf)
951{
952 unblock_autocmds();
Bram Moolenaar3e0107e2021-01-02 13:53:59 +0100953#ifdef FEAT_FOLDING
954 --disable_fold_update;
955#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100956 // Check for valid buffer, just in case.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200957 if (bufref_valid(save_curbuf))
958 {
959 --curbuf->b_nwindows;
960 curwin->w_buffer = save_curbuf->br_buf;
961 curbuf = save_curbuf->br_buf;
962 ++curbuf->b_nwindows;
963 }
964}
965
966/*
967 * Find a window for buffer "buf".
968 * If found OK is returned and "wp" and "tp" are set to the window and tabpage.
969 * If not found FAIL is returned.
970 */
971 static int
972find_win_for_buf(
973 buf_T *buf,
974 win_T **wp,
975 tabpage_T **tp)
976{
977 FOR_ALL_TAB_WINDOWS(*tp, *wp)
978 if ((*wp)->w_buffer == buf)
979 return OK;
980 return FAIL;
981}
982
983/*
984 * Find a window that contains "buf" and switch to it.
985 * If there is no such window, use the current window and change "curbuf".
986 * Caller must initialize save_curbuf to NULL.
987 * restore_win_for_buf() MUST be called later!
988 */
989 void
990switch_to_win_for_buf(
Bram Moolenaar18f47402022-01-06 13:24:51 +0000991 buf_T *buf,
992 switchwin_T *switchwin,
993 bufref_T *save_curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200994{
995 win_T *wp;
996 tabpage_T *tp;
997
998 if (find_win_for_buf(buf, &wp, &tp) == FAIL)
999 switch_buffer(save_curbuf, buf);
Bram Moolenaar18f47402022-01-06 13:24:51 +00001000 else if (switch_win(switchwin, wp, tp, TRUE) == FAIL)
Bram Moolenaar261f3462019-09-07 15:45:32 +02001001 {
Bram Moolenaar18f47402022-01-06 13:24:51 +00001002 restore_win(switchwin, TRUE);
Bram Moolenaar261f3462019-09-07 15:45:32 +02001003 switch_buffer(save_curbuf, buf);
1004 }
1005}
1006
1007 void
1008restore_win_for_buf(
Bram Moolenaar18f47402022-01-06 13:24:51 +00001009 switchwin_T *switchwin,
1010 bufref_T *save_curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +02001011{
1012 if (save_curbuf->br_buf == NULL)
Bram Moolenaar18f47402022-01-06 13:24:51 +00001013 restore_win(switchwin, TRUE);
Bram Moolenaar261f3462019-09-07 15:45:32 +02001014 else
1015 restore_buffer(save_curbuf);
1016}
1017#endif