blob: cefd6428f6b2928ed3cd6eb653d4285ea3be6ce7 [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);
Yegappan Lakshmanana13f3a42024-11-02 18:40:10 +010048 if (!abort)
49 abort = abort || set_ref_in_callback(&bp->b_ffu_cb, copyID);
Bram Moolenaar261f3462019-09-07 15:45:32 +020050 if (abort)
51 break;
52 }
53 return abort;
54}
55
56 buf_T *
57buflist_find_by_name(char_u *name, int curtab_only)
58{
59 int save_magic;
60 char_u *save_cpo;
61 buf_T *buf;
62
63 // Ignore 'magic' and 'cpoptions' here to make scripts portable
64 save_magic = p_magic;
65 p_magic = TRUE;
66 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +010067 p_cpo = empty_option;
Bram Moolenaar261f3462019-09-07 15:45:32 +020068
69 buf = buflist_findnr(buflist_findpat(name, name + STRLEN(name),
70 TRUE, FALSE, curtab_only));
71
72 p_magic = save_magic;
73 p_cpo = save_cpo;
74 return buf;
75}
76
77/*
78 * Find a buffer by number or exact name.
79 */
80 buf_T *
81find_buffer(typval_T *avar)
82{
83 buf_T *buf = NULL;
84
85 if (avar->v_type == VAR_NUMBER)
86 buf = buflist_findnr((int)avar->vval.v_number);
Bram Moolenaar7b45d462021-03-27 19:09:02 +010087 else if (in_vim9script() && check_for_string_arg(avar, 0) == FAIL)
88 return NULL;
Bram Moolenaar261f3462019-09-07 15:45:32 +020089 else if (avar->v_type == VAR_STRING && avar->vval.v_string != NULL)
90 {
91 buf = buflist_findname_exp(avar->vval.v_string);
92 if (buf == NULL)
93 {
94 // No full path name match, try a match with a URL or a "nofile"
95 // buffer, these don't use the full path.
96 FOR_ALL_BUFFERS(buf)
97 if (buf->b_fname != NULL
Bram Moolenaar6d4b2f52022-08-25 15:11:15 +010098 && (path_with_url(buf->b_fname) || bt_nofilename(buf))
Bram Moolenaar261f3462019-09-07 15:45:32 +020099 && STRCMP(buf->b_fname, avar->vval.v_string) == 0)
100 break;
101 }
102 }
103 return buf;
104}
105
106/*
107 * If there is a window for "curbuf", make it the current window.
108 */
109 static void
110find_win_for_curbuf(void)
111{
112 wininfo_T *wip;
113
Bram Moolenaar3da85972022-11-27 19:45:49 +0000114 // The b_wininfo list should have the windows that recently contained the
115 // buffer, going over this is faster than going over all the windows.
116 // Do check the buffer is still there.
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200117 FOR_ALL_BUF_WININFO(curbuf, wip)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200118 {
Bram Moolenaar3da85972022-11-27 19:45:49 +0000119 if (wip->wi_win != NULL && wip->wi_win->w_buffer == curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200120 {
121 curwin = wip->wi_win;
122 break;
123 }
124 }
125}
126
zeertzjq228e4222022-11-20 11:13:17 +0000127typedef struct {
128 win_T *cob_curwin_save;
129 aco_save_T cob_aco;
130 int cob_using_aco;
131 int cob_save_VIsual_active;
132} cob_T;
133
134/*
135 * Used before making a change in "buf", which is not the current one: Make
136 * "buf" the current buffer and find a window for this buffer, so that side
137 * effects are done correctly (e.g., adjusting marks).
138 *
139 * Information is saved in "cob" and MUST be restored by calling
140 * change_other_buffer_restore().
Bram Moolenaare76062c2022-11-28 18:51:43 +0000141 *
142 * If this fails then "curbuf" will not be equal to "buf".
zeertzjq228e4222022-11-20 11:13:17 +0000143 */
144 static void
145change_other_buffer_prepare(cob_T *cob, buf_T *buf)
146{
147 CLEAR_POINTER(cob);
148
149 // Set "curbuf" to the buffer being changed. Then make sure there is a
150 // window for it to handle any side effects.
151 cob->cob_save_VIsual_active = VIsual_active;
152 VIsual_active = FALSE;
153 cob->cob_curwin_save = curwin;
154 curbuf = buf;
155 find_win_for_curbuf(); // simplest: find existing window for "buf"
156
157 if (curwin->w_buffer != buf)
158 {
159 // No existing window for this buffer. It is dangerous to have
160 // curwin->w_buffer differ from "curbuf", use the autocmd window.
161 curbuf = curwin->w_buffer;
162 aucmd_prepbuf(&cob->cob_aco, buf);
Bram Moolenaare76062c2022-11-28 18:51:43 +0000163 if (curbuf == buf)
164 cob->cob_using_aco = TRUE;
zeertzjq228e4222022-11-20 11:13:17 +0000165 }
166}
167
168 static void
169change_other_buffer_restore(cob_T *cob)
170{
171 if (cob->cob_using_aco)
172 {
173 aucmd_restbuf(&cob->cob_aco);
174 }
175 else
176 {
177 curwin = cob->cob_curwin_save;
178 curbuf = curwin->w_buffer;
179 }
180 VIsual_active = cob->cob_save_VIsual_active;
181}
182
Bram Moolenaar261f3462019-09-07 15:45:32 +0200183/*
Bram Moolenaar34453202021-01-31 13:08:38 +0100184 * Set line or list of lines in buffer "buf" to "lines".
185 * Any type is allowed and converted to a string.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200186 */
187 static void
188set_buffer_lines(
189 buf_T *buf,
190 linenr_T lnum_arg,
191 int append,
192 typval_T *lines,
193 typval_T *rettv)
194{
195 linenr_T lnum = lnum_arg + (append ? 1 : 0);
196 char_u *line = NULL;
197 list_T *l = NULL;
198 listitem_T *li = NULL;
199 long added = 0;
200 linenr_T append_lnum;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200201
202 // When using the current buffer ml_mfp will be set if needed. Useful when
203 // setline() is used on startup. For other buffers the buffer must be
204 // loaded.
Bram Moolenaarc934bfa2022-11-19 13:59:43 +0000205 int is_curbuf = buf == curbuf;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200206 if (buf == NULL || (!is_curbuf && buf->b_ml.ml_mfp == NULL) || lnum < 1)
207 {
208 rettv->vval.v_number = 1; // FAIL
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +0000209 if (in_vim9script() && lnum < 1)
210 semsg(_(e_invalid_line_number_nr), lnum_arg);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200211 return;
212 }
213
zeertzjq228e4222022-11-20 11:13:17 +0000214 // After this don't use "return", goto "cleanup"!
215 cob_T cob;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200216 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000217 // set "curbuf" to "buf" and find a window for this buffer
218 change_other_buffer_prepare(&cob, buf);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200219
220 if (append)
221 // appendbufline() uses the line number below which we insert
222 append_lnum = lnum - 1;
223 else
224 // setbufline() uses the line number above which we insert, we only
225 // append if it's below the last line
226 append_lnum = curbuf->b_ml.ml_line_count;
227
228 if (lines->v_type == VAR_LIST)
229 {
230 l = lines->vval.v_list;
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200231 if (l == NULL || list_len(l) == 0)
232 {
Bram Moolenaarcd9c8d42022-11-05 23:46:43 +0000233 // not appending anything always succeeds
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200234 goto done;
235 }
Bram Moolenaar7e9f3512020-05-13 22:44:22 +0200236 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200237 li = l->lv_first;
238 }
239 else
Bram Moolenaar34453202021-01-31 13:08:38 +0100240 line = typval_tostring(lines, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200241
242 // default result is zero == OK
243 for (;;)
244 {
245 if (l != NULL)
246 {
247 // list argument, get next string
248 if (li == NULL)
249 break;
Bram Moolenaar34453202021-01-31 13:08:38 +0100250 vim_free(line);
251 line = typval_tostring(&li->li_tv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200252 li = li->li_next;
253 }
254
255 rettv->vval.v_number = 1; // FAIL
256 if (line == NULL || lnum > curbuf->b_ml.ml_line_count + 1)
257 break;
258
259 // When coming here from Insert mode, sync undo, so that this can be
260 // undone separately from what was previously inserted.
261 if (u_sync_once == 2)
262 {
263 u_sync_once = 1; // notify that u_sync() was called
264 u_sync(TRUE);
265 }
266
267 if (!append && lnum <= curbuf->b_ml.ml_line_count)
268 {
269 // Existing line, replace it.
270 // Removes any existing text properties.
271 if (u_savesub(lnum) == OK && ml_replace_len(
272 lnum, line, (colnr_T)STRLEN(line) + 1, TRUE, TRUE) == OK)
273 {
274 changed_bytes(lnum, 0);
275 if (is_curbuf && lnum == curwin->w_cursor.lnum)
276 check_cursor_col();
277 rettv->vval.v_number = 0; // OK
278 }
279 }
280 else if (added > 0 || u_save(lnum - 1, lnum) == OK)
281 {
282 // append the line
283 ++added;
284 if (ml_append(lnum - 1, line, (colnr_T)0, FALSE) == OK)
285 rettv->vval.v_number = 0; // OK
286 }
287
288 if (l == NULL) // only one string argument
289 break;
290 ++lnum;
291 }
Bram Moolenaar34453202021-01-31 13:08:38 +0100292 vim_free(line);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200293
294 if (added > 0)
295 {
296 win_T *wp;
297 tabpage_T *tp;
298
299 appended_lines_mark(append_lnum, added);
300
301 // Only adjust the cursor for buffers other than the current, unless it
302 // is the current window. For curbuf and other windows it has been
303 // done in mark_adjust_internal().
304 FOR_ALL_TAB_WINDOWS(tp, wp)
305 if (wp->w_buffer == buf
306 && (wp->w_buffer != curbuf || wp == curwin)
307 && wp->w_cursor.lnum > append_lnum)
308 wp->w_cursor.lnum += added;
309 check_cursor_col();
Bram Moolenaar58a3cae2022-09-08 13:43:10 +0100310
311 // Only update the window view if w_buffer matches curbuf, otherwise
312 // the computations will be wrong.
313 if (curwin->w_buffer == curbuf)
314 update_topline();
Bram Moolenaar261f3462019-09-07 15:45:32 +0200315 }
316
Bram Moolenaarad48e6c2020-04-21 22:19:45 +0200317done:
Bram Moolenaar261f3462019-09-07 15:45:32 +0200318 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000319 change_other_buffer_restore(&cob);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200320}
321
322/*
323 * "append(lnum, string/list)" function
324 */
325 void
326f_append(typval_T *argvars, typval_T *rettv)
327{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200328 linenr_T lnum;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000329 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200330
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200331 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
332 return;
333
334 lnum = tv_get_lnum(&argvars[0]);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000335 if (did_emsg == did_emsg_before)
336 set_buffer_lines(curbuf, lnum, TRUE, &argvars[1], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200337}
338
339/*
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200340 * Set or append lines to a buffer.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200341 */
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200342 static void
343buf_set_append_line(typval_T *argvars, typval_T *rettv, int append)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200344{
345 linenr_T lnum;
346 buf_T *buf;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000347 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200348
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200349 if (in_vim9script()
350 && (check_for_buffer_arg(argvars, 0) == FAIL
351 || check_for_lnum_arg(argvars, 1) == FAIL
352 || check_for_string_or_number_or_list_arg(argvars, 2) == FAIL))
353 return;
354
Bram Moolenaar261f3462019-09-07 15:45:32 +0200355 buf = tv_get_buf(&argvars[0], FALSE);
356 if (buf == NULL)
357 rettv->vval.v_number = 1; // FAIL
358 else
359 {
360 lnum = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000361 if (did_emsg == did_emsg_before)
362 set_buffer_lines(buf, lnum, append, &argvars[2], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200363 }
364}
365
366/*
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200367 * "appendbufline(buf, lnum, string/list)" function
368 */
369 void
370f_appendbufline(typval_T *argvars, typval_T *rettv)
371{
372 buf_set_append_line(argvars, rettv, TRUE);
373}
374
375/*
Bram Moolenaar261f3462019-09-07 15:45:32 +0200376 * "bufadd(expr)" function
377 */
378 void
379f_bufadd(typval_T *argvars, typval_T *rettv)
380{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200381 char_u *name;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200382
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200383 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
384 return;
385
386 name = tv_get_string(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200387 rettv->vval.v_number = buflist_add(*name == NUL ? NULL : name, 0);
388}
389
390/*
391 * "bufexists(expr)" function
392 */
393 void
394f_bufexists(typval_T *argvars, typval_T *rettv)
395{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200396 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
397 return;
398
Bram Moolenaar261f3462019-09-07 15:45:32 +0200399 rettv->vval.v_number = (find_buffer(&argvars[0]) != NULL);
400}
401
402/*
403 * "buflisted(expr)" function
404 */
405 void
406f_buflisted(typval_T *argvars, typval_T *rettv)
407{
408 buf_T *buf;
409
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200410 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
411 return;
412
Bram Moolenaar261f3462019-09-07 15:45:32 +0200413 buf = find_buffer(&argvars[0]);
414 rettv->vval.v_number = (buf != NULL && buf->b_p_bl);
415}
416
417/*
418 * "bufload(expr)" function
419 */
420 void
421f_bufload(typval_T *argvars, typval_T *rettv UNUSED)
422{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200423 buf_T *buf;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200424
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200425 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
426 return;
427
428 buf = get_buf_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200429 if (buf != NULL)
430 buffer_ensure_loaded(buf);
431}
432
433/*
434 * "bufloaded(expr)" function
435 */
436 void
437f_bufloaded(typval_T *argvars, typval_T *rettv)
438{
439 buf_T *buf;
440
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200441 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
442 return;
443
Bram Moolenaar261f3462019-09-07 15:45:32 +0200444 buf = find_buffer(&argvars[0]);
445 rettv->vval.v_number = (buf != NULL && buf->b_ml.ml_mfp != NULL);
446}
447
448/*
449 * "bufname(expr)" function
450 */
451 void
452f_bufname(typval_T *argvars, typval_T *rettv)
453{
454 buf_T *buf;
Bram Moolenaar02aaad92020-08-30 21:26:57 +0200455 typval_T *tv = &argvars[0];
Bram Moolenaar261f3462019-09-07 15:45:32 +0200456
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200457 if (in_vim9script() && check_for_opt_buffer_arg(argvars, 0) == FAIL)
458 return;
459
Bram Moolenaar02aaad92020-08-30 21:26:57 +0200460 if (tv->v_type == VAR_UNKNOWN)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200461 buf = curbuf;
462 else
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200463 buf = tv_get_buf_from_arg(tv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200464 rettv->v_type = VAR_STRING;
465 if (buf != NULL && buf->b_fname != NULL)
466 rettv->vval.v_string = vim_strsave(buf->b_fname);
467 else
468 rettv->vval.v_string = NULL;
469}
470
471/*
472 * "bufnr(expr)" function
473 */
474 void
475f_bufnr(typval_T *argvars, typval_T *rettv)
476{
477 buf_T *buf;
478 int error = FALSE;
479 char_u *name;
480
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200481 if (in_vim9script()
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200482 && (check_for_opt_buffer_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200483 || (argvars[0].v_type != VAR_UNKNOWN
484 && check_for_opt_bool_arg(argvars, 1) == FAIL)))
485 return;
486
Bram Moolenaar261f3462019-09-07 15:45:32 +0200487 if (argvars[0].v_type == VAR_UNKNOWN)
488 buf = curbuf;
489 else
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200490 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200491
492 // If the buffer isn't found and the second argument is not zero create a
493 // new buffer.
494 if (buf == NULL
495 && argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaarfe136c92020-09-04 18:35:26 +0200496 && tv_get_bool_chk(&argvars[1], &error) != 0
Bram Moolenaar261f3462019-09-07 15:45:32 +0200497 && !error
498 && (name = tv_get_string_chk(&argvars[0])) != NULL
499 && !error)
500 buf = buflist_new(name, NULL, (linenr_T)1, 0);
501
502 if (buf != NULL)
503 rettv->vval.v_number = buf->b_fnum;
504 else
505 rettv->vval.v_number = -1;
506}
507
508 static void
509buf_win_common(typval_T *argvars, typval_T *rettv, int get_nr)
510{
511 win_T *wp;
512 int winnr = 0;
513 buf_T *buf;
514
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200515 if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
516 return;
517
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200518 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200519 FOR_ALL_WINDOWS(wp)
520 {
521 ++winnr;
522 if (wp->w_buffer == buf)
523 break;
524 }
525 rettv->vval.v_number = (wp != NULL ? (get_nr ? winnr : wp->w_id) : -1);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200526}
527
528/*
529 * "bufwinid(nr)" function
530 */
531 void
532f_bufwinid(typval_T *argvars, typval_T *rettv)
533{
534 buf_win_common(argvars, rettv, FALSE);
535}
536
537/*
538 * "bufwinnr(nr)" function
539 */
540 void
541f_bufwinnr(typval_T *argvars, typval_T *rettv)
542{
543 buf_win_common(argvars, rettv, TRUE);
544}
545
546/*
547 * "deletebufline()" function
548 */
549 void
550f_deletebufline(typval_T *argvars, typval_T *rettv)
551{
552 buf_T *buf;
553 linenr_T first, last;
554 linenr_T lnum;
555 long count;
556 int is_curbuf;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200557 tabpage_T *tp;
558 win_T *wp;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000559 int did_emsg_before = did_emsg;
560
561 rettv->vval.v_number = 1; // FAIL by default
Bram Moolenaar261f3462019-09-07 15:45:32 +0200562
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200563 if (in_vim9script()
564 && (check_for_buffer_arg(argvars, 0) == FAIL
565 || check_for_lnum_arg(argvars, 1) == FAIL
566 || check_for_opt_lnum_arg(argvars, 2) == FAIL))
567 return;
568
Bram Moolenaar261f3462019-09-07 15:45:32 +0200569 buf = tv_get_buf(&argvars[0], FALSE);
570 if (buf == NULL)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200571 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200572
573 first = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000574 if (did_emsg > did_emsg_before)
575 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200576 if (argvars[2].v_type != VAR_UNKNOWN)
577 last = tv_get_lnum_buf(&argvars[2], buf);
578 else
579 last = first;
580
581 if (buf->b_ml.ml_mfp == NULL || first < 1
582 || first > buf->b_ml.ml_line_count || last < first)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200583 return;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200584
zeertzjq7af3ee22022-11-06 22:26:05 +0000585 // After this don't use "return", goto "cleanup"!
zeertzjq228e4222022-11-20 11:13:17 +0000586 is_curbuf = buf == curbuf;
587 cob_T cob;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200588 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000589 // set "curbuf" to "buf" and find a window for this buffer
590 change_other_buffer_prepare(&cob, buf);
591
Bram Moolenaar261f3462019-09-07 15:45:32 +0200592 if (last > curbuf->b_ml.ml_line_count)
593 last = curbuf->b_ml.ml_line_count;
594 count = last - first + 1;
595
596 // When coming here from Insert mode, sync undo, so that this can be
597 // undone separately from what was previously inserted.
598 if (u_sync_once == 2)
599 {
600 u_sync_once = 1; // notify that u_sync() was called
601 u_sync(TRUE);
602 }
603
604 if (u_save(first - 1, last + 1) == FAIL)
zeertzjq7af3ee22022-11-06 22:26:05 +0000605 goto cleanup;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200606
zeertzjq7af3ee22022-11-06 22:26:05 +0000607 for (lnum = first; lnum <= last; ++lnum)
608 ml_delete_flags(first, ML_DEL_MESSAGE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200609
zeertzjq7af3ee22022-11-06 22:26:05 +0000610 FOR_ALL_TAB_WINDOWS(tp, wp)
611 if (wp->w_buffer == buf)
612 {
613 if (wp->w_cursor.lnum > last)
614 wp->w_cursor.lnum -= count;
615 else if (wp->w_cursor.lnum > first)
616 wp->w_cursor.lnum = first;
617 if (wp->w_cursor.lnum > wp->w_buffer->b_ml.ml_line_count)
618 wp->w_cursor.lnum = wp->w_buffer->b_ml.ml_line_count;
619 wp->w_valid = 0;
620 if (wp->w_cursor.lnum <= wp->w_topline)
621 wp->w_topline = 1;
622 }
623 check_cursor_col();
624 deleted_lines_mark(first, count);
625 rettv->vval.v_number = 0; // OK
626
627cleanup:
Bram Moolenaar261f3462019-09-07 15:45:32 +0200628 if (!is_curbuf)
zeertzjq228e4222022-11-20 11:13:17 +0000629 change_other_buffer_restore(&cob);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200630}
631
632/*
633 * Returns buffer options, variables and other attributes in a dictionary.
634 */
635 static dict_T *
636get_buffer_info(buf_T *buf)
637{
638 dict_T *dict;
639 tabpage_T *tp;
640 win_T *wp;
641 list_T *windows;
642
643 dict = dict_alloc();
644 if (dict == NULL)
645 return NULL;
646
647 dict_add_number(dict, "bufnr", buf->b_fnum);
648 dict_add_string(dict, "name", buf->b_ffname);
649 dict_add_number(dict, "lnum", buf == curbuf ? curwin->w_cursor.lnum
650 : buflist_findlnum(buf));
Bram Moolenaara9e96792019-12-17 22:40:15 +0100651 dict_add_number(dict, "linecount", buf->b_ml.ml_line_count);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200652 dict_add_number(dict, "loaded", buf->b_ml.ml_mfp != NULL);
653 dict_add_number(dict, "listed", buf->b_p_bl);
654 dict_add_number(dict, "changed", bufIsChanged(buf));
655 dict_add_number(dict, "changedtick", CHANGEDTICK(buf));
656 dict_add_number(dict, "hidden",
657 buf->b_ml.ml_mfp != NULL && buf->b_nwindows == 0);
Sean Dewar1fb41032023-08-16 17:15:05 +0100658 dict_add_number(dict, "command", buf == cmdwin_buf);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200659
660 // Get a reference to buffer variables
661 dict_add_dict(dict, "variables", buf->b_vars);
662
663 // List of windows displaying this buffer
664 windows = list_alloc();
665 if (windows != NULL)
666 {
667 FOR_ALL_TAB_WINDOWS(tp, wp)
668 if (wp->w_buffer == buf)
669 list_append_number(windows, (varnumber_T)wp->w_id);
670 dict_add_list(dict, "windows", windows);
671 }
672
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +0100673#ifdef FEAT_PROP_POPUP
Bram Moolenaar261f3462019-09-07 15:45:32 +0200674 // List of popup windows displaying this buffer
675 windows = list_alloc();
676 if (windows != NULL)
677 {
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200678 FOR_ALL_POPUPWINS(wp)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200679 if (wp->w_buffer == buf)
680 list_append_number(windows, (varnumber_T)wp->w_id);
681 FOR_ALL_TABPAGES(tp)
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200682 FOR_ALL_POPUPWINS_IN_TAB(tp, wp)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200683 if (wp->w_buffer == buf)
684 list_append_number(windows, (varnumber_T)wp->w_id);
685
686 dict_add_list(dict, "popups", windows);
687 }
688#endif
689
690#ifdef FEAT_SIGNS
691 if (buf->b_signlist != NULL)
692 {
693 // List of signs placed in this buffer
694 list_T *signs = list_alloc();
695 if (signs != NULL)
696 {
697 get_buffer_signs(buf, signs);
698 dict_add_list(dict, "signs", signs);
699 }
700 }
701#endif
702
Bram Moolenaar52410572019-10-27 05:12:45 +0100703#ifdef FEAT_VIMINFO
704 dict_add_number(dict, "lastused", buf->b_last_used);
705#endif
706
Bram Moolenaar261f3462019-09-07 15:45:32 +0200707 return dict;
708}
709
710/*
711 * "getbufinfo()" function
712 */
713 void
714f_getbufinfo(typval_T *argvars, typval_T *rettv)
715{
716 buf_T *buf = NULL;
717 buf_T *argbuf = NULL;
718 dict_T *d;
719 int filtered = FALSE;
720 int sel_buflisted = FALSE;
721 int sel_bufloaded = FALSE;
722 int sel_bufmodified = FALSE;
723
Bram Moolenaar93a10962022-06-16 11:42:09 +0100724 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200725 return;
726
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200727 if (in_vim9script()
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200728 && check_for_opt_buffer_or_dict_arg(argvars, 0) == FAIL)
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200729 return;
730
Bram Moolenaar261f3462019-09-07 15:45:32 +0200731 // List of all the buffers or selected buffers
732 if (argvars[0].v_type == VAR_DICT)
733 {
734 dict_T *sel_d = argvars[0].vval.v_dict;
735
736 if (sel_d != NULL)
737 {
Bram Moolenaar261f3462019-09-07 15:45:32 +0200738 filtered = TRUE;
Bram Moolenaard61efa52022-07-23 09:52:04 +0100739 sel_buflisted = dict_get_bool(sel_d, "buflisted", FALSE);
740 sel_bufloaded = dict_get_bool(sel_d, "bufloaded", FALSE);
741 sel_bufmodified = dict_get_bool(sel_d, "bufmodified",
Bram Moolenaar036c2cf2020-09-05 17:37:07 +0200742 FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200743 }
744 }
745 else if (argvars[0].v_type != VAR_UNKNOWN)
746 {
747 // Information about one buffer. Argument specifies the buffer
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200748 argbuf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200749 if (argbuf == NULL)
750 return;
751 }
752
753 // Return information about all the buffers or a specified buffer
754 FOR_ALL_BUFFERS(buf)
755 {
756 if (argbuf != NULL && argbuf != buf)
757 continue;
758 if (filtered && ((sel_bufloaded && buf->b_ml.ml_mfp == NULL)
759 || (sel_buflisted && !buf->b_p_bl)
760 || (sel_bufmodified && !buf->b_changed)))
761 continue;
762
763 d = get_buffer_info(buf);
764 if (d != NULL)
765 list_append_dict(rettv->vval.v_list, d);
766 if (argbuf != NULL)
767 return;
768 }
769}
770
771/*
772 * Get line or list of lines from buffer "buf" into "rettv".
773 * Return a range (from start to end) of lines in rettv from the specified
774 * buffer.
775 * If 'retlist' is TRUE, then the lines are returned as a Vim List.
776 */
777 static void
778get_buffer_lines(
779 buf_T *buf,
780 linenr_T start,
781 linenr_T end,
782 int retlist,
783 typval_T *rettv)
784{
785 char_u *p;
786
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100787 if (retlist)
788 {
789 if (rettv_list_alloc(rettv) == FAIL)
790 return;
791 }
792 else
793 {
794 rettv->v_type = VAR_STRING;
795 rettv->vval.v_string = NULL;
796 }
Bram Moolenaar261f3462019-09-07 15:45:32 +0200797
798 if (buf == NULL || buf->b_ml.ml_mfp == NULL || start < 0)
799 return;
800
801 if (!retlist)
802 {
803 if (start >= 1 && start <= buf->b_ml.ml_line_count)
804 p = ml_get_buf(buf, start, FALSE);
805 else
806 p = (char_u *)"";
807 rettv->vval.v_string = vim_strsave(p);
808 }
809 else
810 {
811 if (end < start)
812 return;
813
814 if (start < 1)
815 start = 1;
816 if (end > buf->b_ml.ml_line_count)
817 end = buf->b_ml.ml_line_count;
818 while (start <= end)
819 if (list_append_string(rettv->vval.v_list,
820 ml_get_buf(buf, start++, FALSE), -1) == FAIL)
821 break;
822 }
823}
824
825/*
Bram Moolenaarce30ccc2022-11-21 19:57:04 +0000826 * "retlist" TRUE: "getbufline()" function
827 * "retlist" FALSE: "getbufoneline()" function
Bram Moolenaar261f3462019-09-07 15:45:32 +0200828 */
Bram Moolenaarce30ccc2022-11-21 19:57:04 +0000829 static void
830getbufline(typval_T *argvars, typval_T *rettv, int retlist)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200831{
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200832 linenr_T lnum = 1;
833 linenr_T end = 1;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200834 buf_T *buf;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000835 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200836
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200837 if (in_vim9script()
838 && (check_for_buffer_arg(argvars, 0) == FAIL
839 || check_for_lnum_arg(argvars, 1) == FAIL
840 || check_for_opt_lnum_arg(argvars, 2) == FAIL))
841 return;
842
Bram Moolenaar3767e3a2020-09-01 23:06:01 +0200843 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200844 if (buf != NULL)
845 {
846 lnum = tv_get_lnum_buf(&argvars[1], buf);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000847 if (did_emsg > did_emsg_before)
848 return;
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200849 if (argvars[2].v_type == VAR_UNKNOWN)
850 end = lnum;
851 else
852 end = tv_get_lnum_buf(&argvars[2], buf);
853 }
Bram Moolenaar261f3462019-09-07 15:45:32 +0200854
Bram Moolenaarce30ccc2022-11-21 19:57:04 +0000855 get_buffer_lines(buf, lnum, end, retlist, rettv);
856}
857
858/*
859 * "getbufline()" function
860 */
861 void
862f_getbufline(typval_T *argvars, typval_T *rettv)
863{
864 getbufline(argvars, rettv, TRUE);
865}
866
867/*
868 * "getbufoneline()" function
869 */
870 void
871f_getbufoneline(typval_T *argvars, typval_T *rettv)
872{
873 getbufline(argvars, rettv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200874}
875
876/*
877 * "getline(lnum, [end])" function
878 */
879 void
880f_getline(typval_T *argvars, typval_T *rettv)
881{
882 linenr_T lnum;
883 linenr_T end;
884 int retlist;
885
Yegappan Lakshmanancd917202021-07-21 19:09:09 +0200886 if (in_vim9script()
887 && (check_for_lnum_arg(argvars, 0) == FAIL
888 || check_for_opt_lnum_arg(argvars, 1) == FAIL))
889 return;
890
Bram Moolenaar261f3462019-09-07 15:45:32 +0200891 lnum = tv_get_lnum(argvars);
892 if (argvars[1].v_type == VAR_UNKNOWN)
893 {
894 end = 0;
895 retlist = FALSE;
896 }
897 else
898 {
899 end = tv_get_lnum(&argvars[1]);
900 retlist = TRUE;
901 }
902
903 get_buffer_lines(curbuf, lnum, end, retlist, rettv);
904}
905
906/*
907 * "setbufline()" function
908 */
909 void
910f_setbufline(typval_T *argvars, typval_T *rettv)
911{
Yegappan Lakshmanan4a155042021-07-30 21:32:45 +0200912 buf_set_append_line(argvars, rettv, FALSE);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200913}
914
915/*
916 * "setline()" function
917 */
918 void
919f_setline(typval_T *argvars, typval_T *rettv)
920{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200921 linenr_T lnum;
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000922 int did_emsg_before = did_emsg;
Bram Moolenaar261f3462019-09-07 15:45:32 +0200923
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200924 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
925 return;
926
927 lnum = tv_get_lnum(&argvars[0]);
Bram Moolenaar8b6256f2021-12-28 11:24:49 +0000928 if (did_emsg == did_emsg_before)
929 set_buffer_lines(curbuf, lnum, FALSE, &argvars[1], rettv);
Bram Moolenaar261f3462019-09-07 15:45:32 +0200930}
931#endif // FEAT_EVAL
932
Dominique Pelle748b3082022-01-08 12:41:16 +0000933#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3) || defined(PROTO)
Bram Moolenaar261f3462019-09-07 15:45:32 +0200934/*
935 * Make "buf" the current buffer. restore_buffer() MUST be called to undo.
936 * No autocommands will be executed. Use aucmd_prepbuf() if there are any.
937 */
938 void
939switch_buffer(bufref_T *save_curbuf, buf_T *buf)
940{
941 block_autocmds();
Bram Moolenaar3e0107e2021-01-02 13:53:59 +0100942#ifdef FEAT_FOLDING
943 ++disable_fold_update;
944#endif
Bram Moolenaar261f3462019-09-07 15:45:32 +0200945 set_bufref(save_curbuf, curbuf);
946 --curbuf->b_nwindows;
947 curbuf = buf;
948 curwin->w_buffer = buf;
949 ++curbuf->b_nwindows;
950}
951
952/*
953 * Restore the current buffer after using switch_buffer().
954 */
955 void
956restore_buffer(bufref_T *save_curbuf)
957{
958 unblock_autocmds();
Bram Moolenaar3e0107e2021-01-02 13:53:59 +0100959#ifdef FEAT_FOLDING
960 --disable_fold_update;
961#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100962 // Check for valid buffer, just in case.
Bram Moolenaar261f3462019-09-07 15:45:32 +0200963 if (bufref_valid(save_curbuf))
964 {
965 --curbuf->b_nwindows;
966 curwin->w_buffer = save_curbuf->br_buf;
967 curbuf = save_curbuf->br_buf;
968 ++curbuf->b_nwindows;
969 }
970}
971
972/*
973 * Find a window for buffer "buf".
974 * If found OK is returned and "wp" and "tp" are set to the window and tabpage.
975 * If not found FAIL is returned.
976 */
977 static int
978find_win_for_buf(
979 buf_T *buf,
980 win_T **wp,
981 tabpage_T **tp)
982{
983 FOR_ALL_TAB_WINDOWS(*tp, *wp)
984 if ((*wp)->w_buffer == buf)
985 return OK;
986 return FAIL;
987}
988
989/*
990 * Find a window that contains "buf" and switch to it.
991 * If there is no such window, use the current window and change "curbuf".
992 * Caller must initialize save_curbuf to NULL.
993 * restore_win_for_buf() MUST be called later!
994 */
995 void
996switch_to_win_for_buf(
Bram Moolenaar18f47402022-01-06 13:24:51 +0000997 buf_T *buf,
998 switchwin_T *switchwin,
999 bufref_T *save_curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +02001000{
1001 win_T *wp;
1002 tabpage_T *tp;
1003
1004 if (find_win_for_buf(buf, &wp, &tp) == FAIL)
1005 switch_buffer(save_curbuf, buf);
Bram Moolenaar18f47402022-01-06 13:24:51 +00001006 else if (switch_win(switchwin, wp, tp, TRUE) == FAIL)
Bram Moolenaar261f3462019-09-07 15:45:32 +02001007 {
Bram Moolenaar18f47402022-01-06 13:24:51 +00001008 restore_win(switchwin, TRUE);
Bram Moolenaar261f3462019-09-07 15:45:32 +02001009 switch_buffer(save_curbuf, buf);
1010 }
1011}
1012
1013 void
1014restore_win_for_buf(
Bram Moolenaar18f47402022-01-06 13:24:51 +00001015 switchwin_T *switchwin,
1016 bufref_T *save_curbuf)
Bram Moolenaar261f3462019-09-07 15:45:32 +02001017{
1018 if (save_curbuf->br_buf == NULL)
Bram Moolenaar18f47402022-01-06 13:24:51 +00001019 restore_win(switchwin, TRUE);
Bram Moolenaar261f3462019-09-07 15:45:32 +02001020 else
1021 restore_buffer(save_curbuf);
1022}
1023#endif