xref: /vim-8.2.3635/src/evalbuffer.c (revision a7c4e747)
1 /* 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
21 set_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;
29 	typval_T tv;
30 
31 	for (lnr = bp->b_listener; !abort && lnr != NULL; lnr = lnr->lr_next)
32 	{
33 	    if (lnr->lr_callback.cb_partial != NULL)
34 	    {
35 		tv.v_type = VAR_PARTIAL;
36 		tv.vval.v_partial = lnr->lr_callback.cb_partial;
37 		abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
38 	    }
39 	}
40 # ifdef FEAT_JOB_CHANNEL
41 	if (!abort && bp->b_prompt_callback.cb_partial != NULL)
42 	{
43 	    tv.v_type = VAR_PARTIAL;
44 	    tv.vval.v_partial = bp->b_prompt_callback.cb_partial;
45 	    abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
46 	}
47 	if (!abort && bp->b_prompt_interrupt.cb_partial != NULL)
48 	{
49 	    tv.v_type = VAR_PARTIAL;
50 	    tv.vval.v_partial = bp->b_prompt_interrupt.cb_partial;
51 	    abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
52 	}
53 # endif
54 	if (abort)
55 	    break;
56     }
57     return abort;
58 }
59 
60     buf_T *
61 buflist_find_by_name(char_u *name, int curtab_only)
62 {
63     int		save_magic;
64     char_u	*save_cpo;
65     buf_T	*buf;
66 
67     // Ignore 'magic' and 'cpoptions' here to make scripts portable
68     save_magic = p_magic;
69     p_magic = TRUE;
70     save_cpo = p_cpo;
71     p_cpo = (char_u *)"";
72 
73     buf = buflist_findnr(buflist_findpat(name, name + STRLEN(name),
74 						    TRUE, FALSE, curtab_only));
75 
76     p_magic = save_magic;
77     p_cpo = save_cpo;
78     return buf;
79 }
80 
81 /*
82  * Find a buffer by number or exact name.
83  */
84     buf_T *
85 find_buffer(typval_T *avar)
86 {
87     buf_T	*buf = NULL;
88 
89     if (avar->v_type == VAR_NUMBER)
90 	buf = buflist_findnr((int)avar->vval.v_number);
91     else if (avar->v_type == VAR_STRING && avar->vval.v_string != NULL)
92     {
93 	buf = buflist_findname_exp(avar->vval.v_string);
94 	if (buf == NULL)
95 	{
96 	    // No full path name match, try a match with a URL or a "nofile"
97 	    // buffer, these don't use the full path.
98 	    FOR_ALL_BUFFERS(buf)
99 		if (buf->b_fname != NULL
100 			&& (path_with_url(buf->b_fname)
101 #ifdef FEAT_QUICKFIX
102 			    || bt_nofilename(buf)
103 #endif
104 			   )
105 			&& STRCMP(buf->b_fname, avar->vval.v_string) == 0)
106 		    break;
107 	}
108     }
109     return buf;
110 }
111 
112 /*
113  * If there is a window for "curbuf", make it the current window.
114  */
115     static void
116 find_win_for_curbuf(void)
117 {
118     wininfo_T *wip;
119 
120     FOR_ALL_BUF_WININFO(curbuf, wip)
121     {
122 	if (wip->wi_win != NULL)
123 	{
124 	    curwin = wip->wi_win;
125 	    break;
126 	}
127     }
128 }
129 
130 /*
131  * Set line or list of lines in buffer "buf".
132  */
133     static void
134 set_buffer_lines(
135 	buf_T	    *buf,
136 	linenr_T    lnum_arg,
137 	int	    append,
138 	typval_T    *lines,
139 	typval_T    *rettv)
140 {
141     linenr_T    lnum = lnum_arg + (append ? 1 : 0);
142     char_u	*line = NULL;
143     list_T	*l = NULL;
144     listitem_T	*li = NULL;
145     long	added = 0;
146     linenr_T	append_lnum;
147     buf_T	*curbuf_save = NULL;
148     win_T	*curwin_save = NULL;
149     int		is_curbuf = buf == curbuf;
150 
151     // When using the current buffer ml_mfp will be set if needed.  Useful when
152     // setline() is used on startup.  For other buffers the buffer must be
153     // loaded.
154     if (buf == NULL || (!is_curbuf && buf->b_ml.ml_mfp == NULL) || lnum < 1)
155     {
156 	rettv->vval.v_number = 1;	// FAIL
157 	return;
158     }
159 
160     if (!is_curbuf)
161     {
162 	curbuf_save = curbuf;
163 	curwin_save = curwin;
164 	curbuf = buf;
165 	find_win_for_curbuf();
166     }
167 
168     if (append)
169 	// appendbufline() uses the line number below which we insert
170 	append_lnum = lnum - 1;
171     else
172 	// setbufline() uses the line number above which we insert, we only
173 	// append if it's below the last line
174 	append_lnum = curbuf->b_ml.ml_line_count;
175 
176     if (lines->v_type == VAR_LIST)
177     {
178 	l = lines->vval.v_list;
179 	if (l == NULL || list_len(l) == 0)
180 	{
181 	    // set proper return code
182 	    if (lnum > curbuf->b_ml.ml_line_count)
183 		rettv->vval.v_number = 1;	// FAIL
184 	    goto done;
185 	}
186 	CHECK_LIST_MATERIALIZE(l);
187 	li = l->lv_first;
188     }
189     else
190 	line = tv_get_string_chk(lines);
191 
192     // default result is zero == OK
193     for (;;)
194     {
195 	if (l != NULL)
196 	{
197 	    // list argument, get next string
198 	    if (li == NULL)
199 		break;
200 	    line = tv_get_string_chk(&li->li_tv);
201 	    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     }
241 
242     if (added > 0)
243     {
244 	win_T	    *wp;
245 	tabpage_T   *tp;
246 
247 	appended_lines_mark(append_lnum, added);
248 
249 	// Only adjust the cursor for buffers other than the current, unless it
250 	// is the current window.  For curbuf and other windows it has been
251 	// done in mark_adjust_internal().
252 	FOR_ALL_TAB_WINDOWS(tp, wp)
253 	    if (wp->w_buffer == buf
254 		    && (wp->w_buffer != curbuf || wp == curwin)
255 		    && wp->w_cursor.lnum > append_lnum)
256 		wp->w_cursor.lnum += added;
257 	check_cursor_col();
258 	update_topline();
259     }
260 
261 done:
262     if (!is_curbuf)
263     {
264 	curbuf = curbuf_save;
265 	curwin = curwin_save;
266     }
267 }
268 
269 /*
270  * "append(lnum, string/list)" function
271  */
272     void
273 f_append(typval_T *argvars, typval_T *rettv)
274 {
275     linenr_T	lnum = tv_get_lnum(&argvars[0]);
276 
277     set_buffer_lines(curbuf, lnum, TRUE, &argvars[1], rettv);
278 }
279 
280 /*
281  * "appendbufline(buf, lnum, string/list)" function
282  */
283     void
284 f_appendbufline(typval_T *argvars, typval_T *rettv)
285 {
286     linenr_T	lnum;
287     buf_T	*buf;
288 
289     buf = tv_get_buf(&argvars[0], FALSE);
290     if (buf == NULL)
291 	rettv->vval.v_number = 1; // FAIL
292     else
293     {
294 	lnum = tv_get_lnum_buf(&argvars[1], buf);
295 	set_buffer_lines(buf, lnum, TRUE, &argvars[2], rettv);
296     }
297 }
298 
299 /*
300  * "bufadd(expr)" function
301  */
302     void
303 f_bufadd(typval_T *argvars, typval_T *rettv)
304 {
305     char_u *name = tv_get_string(&argvars[0]);
306 
307     rettv->vval.v_number = buflist_add(*name == NUL ? NULL : name, 0);
308 }
309 
310 /*
311  * "bufexists(expr)" function
312  */
313     void
314 f_bufexists(typval_T *argvars, typval_T *rettv)
315 {
316     rettv->vval.v_number = (find_buffer(&argvars[0]) != NULL);
317 }
318 
319 /*
320  * "buflisted(expr)" function
321  */
322     void
323 f_buflisted(typval_T *argvars, typval_T *rettv)
324 {
325     buf_T	*buf;
326 
327     buf = find_buffer(&argvars[0]);
328     rettv->vval.v_number = (buf != NULL && buf->b_p_bl);
329 }
330 
331 /*
332  * "bufload(expr)" function
333  */
334     void
335 f_bufload(typval_T *argvars, typval_T *rettv UNUSED)
336 {
337     buf_T	*buf = get_buf_arg(&argvars[0]);
338 
339     if (buf != NULL)
340 	buffer_ensure_loaded(buf);
341 }
342 
343 /*
344  * "bufloaded(expr)" function
345  */
346     void
347 f_bufloaded(typval_T *argvars, typval_T *rettv)
348 {
349     buf_T	*buf;
350 
351     buf = find_buffer(&argvars[0]);
352     rettv->vval.v_number = (buf != NULL && buf->b_ml.ml_mfp != NULL);
353 }
354 
355 /*
356  * "bufname(expr)" function
357  */
358     void
359 f_bufname(typval_T *argvars, typval_T *rettv)
360 {
361     buf_T	*buf;
362 
363     if (argvars[0].v_type == VAR_UNKNOWN)
364 	buf = curbuf;
365     else
366     {
367 	(void)tv_get_number(&argvars[0]);	// issue errmsg if type error
368 	++emsg_off;
369 	buf = tv_get_buf(&argvars[0], FALSE);
370 	--emsg_off;
371     }
372     rettv->v_type = VAR_STRING;
373     if (buf != NULL && buf->b_fname != NULL)
374 	rettv->vval.v_string = vim_strsave(buf->b_fname);
375     else
376 	rettv->vval.v_string = NULL;
377 }
378 
379 /*
380  * "bufnr(expr)" function
381  */
382     void
383 f_bufnr(typval_T *argvars, typval_T *rettv)
384 {
385     buf_T	*buf;
386     int		error = FALSE;
387     char_u	*name;
388 
389     if (argvars[0].v_type == VAR_UNKNOWN)
390 	buf = curbuf;
391     else
392     {
393 	if (argvars[0].v_type != VAR_STRING)
394 	    (void)tv_get_number(&argvars[0]);    // issue errmsg if type error
395 	++emsg_off;
396 	buf = tv_get_buf(&argvars[0], FALSE);
397 	--emsg_off;
398     }
399 
400     // If the buffer isn't found and the second argument is not zero create a
401     // new buffer.
402     if (buf == NULL
403 	    && argvars[1].v_type != VAR_UNKNOWN
404 	    && tv_get_number_chk(&argvars[1], &error) != 0
405 	    && !error
406 	    && (name = tv_get_string_chk(&argvars[0])) != NULL
407 	    && !error)
408 	buf = buflist_new(name, NULL, (linenr_T)1, 0);
409 
410     if (buf != NULL)
411 	rettv->vval.v_number = buf->b_fnum;
412     else
413 	rettv->vval.v_number = -1;
414 }
415 
416     static void
417 buf_win_common(typval_T *argvars, typval_T *rettv, int get_nr)
418 {
419     win_T	*wp;
420     int		winnr = 0;
421     buf_T	*buf;
422 
423     (void)tv_get_number(&argvars[0]);	    // issue errmsg if type error
424     ++emsg_off;
425     buf = tv_get_buf(&argvars[0], TRUE);
426     FOR_ALL_WINDOWS(wp)
427     {
428 	++winnr;
429 	if (wp->w_buffer == buf)
430 	    break;
431     }
432     rettv->vval.v_number = (wp != NULL ? (get_nr ? winnr : wp->w_id) : -1);
433     --emsg_off;
434 }
435 
436 /*
437  * "bufwinid(nr)" function
438  */
439     void
440 f_bufwinid(typval_T *argvars, typval_T *rettv)
441 {
442     buf_win_common(argvars, rettv, FALSE);
443 }
444 
445 /*
446  * "bufwinnr(nr)" function
447  */
448     void
449 f_bufwinnr(typval_T *argvars, typval_T *rettv)
450 {
451     buf_win_common(argvars, rettv, TRUE);
452 }
453 
454 /*
455  * "deletebufline()" function
456  */
457     void
458 f_deletebufline(typval_T *argvars, typval_T *rettv)
459 {
460     buf_T	*buf;
461     linenr_T	first, last;
462     linenr_T	lnum;
463     long	count;
464     int		is_curbuf;
465     buf_T	*curbuf_save = NULL;
466     win_T	*curwin_save = NULL;
467     tabpage_T	*tp;
468     win_T	*wp;
469 
470     buf = tv_get_buf(&argvars[0], FALSE);
471     if (buf == NULL)
472     {
473 	rettv->vval.v_number = 1; // FAIL
474 	return;
475     }
476     is_curbuf = buf == curbuf;
477 
478     first = tv_get_lnum_buf(&argvars[1], buf);
479     if (argvars[2].v_type != VAR_UNKNOWN)
480 	last = tv_get_lnum_buf(&argvars[2], buf);
481     else
482 	last = first;
483 
484     if (buf->b_ml.ml_mfp == NULL || first < 1
485 			   || first > buf->b_ml.ml_line_count || last < first)
486     {
487 	rettv->vval.v_number = 1;	// FAIL
488 	return;
489     }
490 
491     if (!is_curbuf)
492     {
493 	curbuf_save = curbuf;
494 	curwin_save = curwin;
495 	curbuf = buf;
496 	find_win_for_curbuf();
497     }
498     if (last > curbuf->b_ml.ml_line_count)
499 	last = curbuf->b_ml.ml_line_count;
500     count = last - first + 1;
501 
502     // When coming here from Insert mode, sync undo, so that this can be
503     // undone separately from what was previously inserted.
504     if (u_sync_once == 2)
505     {
506 	u_sync_once = 1; // notify that u_sync() was called
507 	u_sync(TRUE);
508     }
509 
510     if (u_save(first - 1, last + 1) == FAIL)
511     {
512 	rettv->vval.v_number = 1;	// FAIL
513 	return;
514     }
515 
516     for (lnum = first; lnum <= last; ++lnum)
517 	ml_delete_flags(first, ML_DEL_MESSAGE);
518 
519     FOR_ALL_TAB_WINDOWS(tp, wp)
520 	if (wp->w_buffer == buf)
521 	{
522 	    if (wp->w_cursor.lnum > last)
523 		wp->w_cursor.lnum -= count;
524 	    else if (wp->w_cursor.lnum> first)
525 		wp->w_cursor.lnum = first;
526 	    if (wp->w_cursor.lnum > wp->w_buffer->b_ml.ml_line_count)
527 		wp->w_cursor.lnum = wp->w_buffer->b_ml.ml_line_count;
528 	}
529     check_cursor_col();
530     deleted_lines_mark(first, count);
531 
532     if (!is_curbuf)
533     {
534 	curbuf = curbuf_save;
535 	curwin = curwin_save;
536     }
537 }
538 
539 /*
540  * Returns buffer options, variables and other attributes in a dictionary.
541  */
542     static dict_T *
543 get_buffer_info(buf_T *buf)
544 {
545     dict_T	*dict;
546     tabpage_T	*tp;
547     win_T	*wp;
548     list_T	*windows;
549 
550     dict = dict_alloc();
551     if (dict == NULL)
552 	return NULL;
553 
554     dict_add_number(dict, "bufnr", buf->b_fnum);
555     dict_add_string(dict, "name", buf->b_ffname);
556     dict_add_number(dict, "lnum", buf == curbuf ? curwin->w_cursor.lnum
557 						     : buflist_findlnum(buf));
558     dict_add_number(dict, "linecount", buf->b_ml.ml_line_count);
559     dict_add_number(dict, "loaded", buf->b_ml.ml_mfp != NULL);
560     dict_add_number(dict, "listed", buf->b_p_bl);
561     dict_add_number(dict, "changed", bufIsChanged(buf));
562     dict_add_number(dict, "changedtick", CHANGEDTICK(buf));
563     dict_add_number(dict, "hidden",
564 			    buf->b_ml.ml_mfp != NULL && buf->b_nwindows == 0);
565 
566     // Get a reference to buffer variables
567     dict_add_dict(dict, "variables", buf->b_vars);
568 
569     // List of windows displaying this buffer
570     windows = list_alloc();
571     if (windows != NULL)
572     {
573 	FOR_ALL_TAB_WINDOWS(tp, wp)
574 	    if (wp->w_buffer == buf)
575 		list_append_number(windows, (varnumber_T)wp->w_id);
576 	dict_add_list(dict, "windows", windows);
577     }
578 
579 #ifdef FEAT_PROP_POPUP
580     // List of popup windows displaying this buffer
581     windows = list_alloc();
582     if (windows != NULL)
583     {
584 	FOR_ALL_POPUPWINS(wp)
585 	    if (wp->w_buffer == buf)
586 		list_append_number(windows, (varnumber_T)wp->w_id);
587 	FOR_ALL_TABPAGES(tp)
588 	    FOR_ALL_POPUPWINS_IN_TAB(tp, wp)
589 		if (wp->w_buffer == buf)
590 		    list_append_number(windows, (varnumber_T)wp->w_id);
591 
592 	dict_add_list(dict, "popups", windows);
593     }
594 #endif
595 
596 #ifdef FEAT_SIGNS
597     if (buf->b_signlist != NULL)
598     {
599 	// List of signs placed in this buffer
600 	list_T	*signs = list_alloc();
601 	if (signs != NULL)
602 	{
603 	    get_buffer_signs(buf, signs);
604 	    dict_add_list(dict, "signs", signs);
605 	}
606     }
607 #endif
608 
609 #ifdef FEAT_VIMINFO
610     dict_add_number(dict, "lastused", buf->b_last_used);
611 #endif
612 
613     return dict;
614 }
615 
616 /*
617  * "getbufinfo()" function
618  */
619     void
620 f_getbufinfo(typval_T *argvars, typval_T *rettv)
621 {
622     buf_T	*buf = NULL;
623     buf_T	*argbuf = NULL;
624     dict_T	*d;
625     int		filtered = FALSE;
626     int		sel_buflisted = FALSE;
627     int		sel_bufloaded = FALSE;
628     int		sel_bufmodified = FALSE;
629 
630     if (rettv_list_alloc(rettv) != OK)
631 	return;
632 
633     // List of all the buffers or selected buffers
634     if (argvars[0].v_type == VAR_DICT)
635     {
636 	dict_T	*sel_d = argvars[0].vval.v_dict;
637 
638 	if (sel_d != NULL)
639 	{
640 	    dictitem_T	*di;
641 
642 	    filtered = TRUE;
643 
644 	    di = dict_find(sel_d, (char_u *)"buflisted", -1);
645 	    if (di != NULL && tv_get_number(&di->di_tv))
646 		sel_buflisted = TRUE;
647 
648 	    di = dict_find(sel_d, (char_u *)"bufloaded", -1);
649 	    if (di != NULL && tv_get_number(&di->di_tv))
650 		sel_bufloaded = TRUE;
651 
652 	    di = dict_find(sel_d, (char_u *)"bufmodified", -1);
653 	    if (di != NULL && tv_get_number(&di->di_tv))
654 		sel_bufmodified = TRUE;
655 	}
656     }
657     else if (argvars[0].v_type != VAR_UNKNOWN)
658     {
659 	// Information about one buffer.  Argument specifies the buffer
660 	(void)tv_get_number(&argvars[0]);   // issue errmsg if type error
661 	++emsg_off;
662 	argbuf = tv_get_buf(&argvars[0], FALSE);
663 	--emsg_off;
664 	if (argbuf == NULL)
665 	    return;
666     }
667 
668     // Return information about all the buffers or a specified buffer
669     FOR_ALL_BUFFERS(buf)
670     {
671 	if (argbuf != NULL && argbuf != buf)
672 	    continue;
673 	if (filtered && ((sel_bufloaded && buf->b_ml.ml_mfp == NULL)
674 			|| (sel_buflisted && !buf->b_p_bl)
675 			|| (sel_bufmodified && !buf->b_changed)))
676 	    continue;
677 
678 	d = get_buffer_info(buf);
679 	if (d != NULL)
680 	    list_append_dict(rettv->vval.v_list, d);
681 	if (argbuf != NULL)
682 	    return;
683     }
684 }
685 
686 /*
687  * Get line or list of lines from buffer "buf" into "rettv".
688  * Return a range (from start to end) of lines in rettv from the specified
689  * buffer.
690  * If 'retlist' is TRUE, then the lines are returned as a Vim List.
691  */
692     static void
693 get_buffer_lines(
694     buf_T	*buf,
695     linenr_T	start,
696     linenr_T	end,
697     int		retlist,
698     typval_T	*rettv)
699 {
700     char_u	*p;
701 
702     if (retlist)
703     {
704 	if (rettv_list_alloc(rettv) == FAIL)
705 	    return;
706     }
707     else
708     {
709 	rettv->v_type = VAR_STRING;
710 	rettv->vval.v_string = NULL;
711     }
712 
713     if (buf == NULL || buf->b_ml.ml_mfp == NULL || start < 0)
714 	return;
715 
716     if (!retlist)
717     {
718 	if (start >= 1 && start <= buf->b_ml.ml_line_count)
719 	    p = ml_get_buf(buf, start, FALSE);
720 	else
721 	    p = (char_u *)"";
722 	rettv->vval.v_string = vim_strsave(p);
723     }
724     else
725     {
726 	if (end < start)
727 	    return;
728 
729 	if (start < 1)
730 	    start = 1;
731 	if (end > buf->b_ml.ml_line_count)
732 	    end = buf->b_ml.ml_line_count;
733 	while (start <= end)
734 	    if (list_append_string(rettv->vval.v_list,
735 				 ml_get_buf(buf, start++, FALSE), -1) == FAIL)
736 		break;
737     }
738 }
739 
740 /*
741  * "getbufline()" function
742  */
743     void
744 f_getbufline(typval_T *argvars, typval_T *rettv)
745 {
746     linenr_T	lnum;
747     linenr_T	end;
748     buf_T	*buf;
749 
750     (void)tv_get_number(&argvars[0]);	    // issue errmsg if type error
751     ++emsg_off;
752     buf = tv_get_buf(&argvars[0], FALSE);
753     --emsg_off;
754 
755     lnum = tv_get_lnum_buf(&argvars[1], buf);
756     if (argvars[2].v_type == VAR_UNKNOWN)
757 	end = lnum;
758     else
759 	end = tv_get_lnum_buf(&argvars[2], buf);
760 
761     get_buffer_lines(buf, lnum, end, TRUE, rettv);
762 }
763 
764     type_T *
765 ret_f_getline(int argcount, type_T **argtypes UNUSED)
766 {
767     return argcount == 1 ? &t_string : &t_list_string;
768 }
769 
770 /*
771  * "getline(lnum, [end])" function
772  */
773     void
774 f_getline(typval_T *argvars, typval_T *rettv)
775 {
776     linenr_T	lnum;
777     linenr_T	end;
778     int		retlist;
779 
780     lnum = tv_get_lnum(argvars);
781     if (argvars[1].v_type == VAR_UNKNOWN)
782     {
783 	end = 0;
784 	retlist = FALSE;
785     }
786     else
787     {
788 	end = tv_get_lnum(&argvars[1]);
789 	retlist = TRUE;
790     }
791 
792     get_buffer_lines(curbuf, lnum, end, retlist, rettv);
793 }
794 
795 /*
796  * "setbufline()" function
797  */
798     void
799 f_setbufline(typval_T *argvars, typval_T *rettv)
800 {
801     linenr_T	lnum;
802     buf_T	*buf;
803 
804     buf = tv_get_buf(&argvars[0], FALSE);
805     if (buf == NULL)
806 	rettv->vval.v_number = 1; // FAIL
807     else
808     {
809 	lnum = tv_get_lnum_buf(&argvars[1], buf);
810 	set_buffer_lines(buf, lnum, FALSE, &argvars[2], rettv);
811     }
812 }
813 
814 /*
815  * "setline()" function
816  */
817     void
818 f_setline(typval_T *argvars, typval_T *rettv)
819 {
820     linenr_T	lnum = tv_get_lnum(&argvars[0]);
821 
822     set_buffer_lines(curbuf, lnum, FALSE, &argvars[1], rettv);
823 }
824 #endif  // FEAT_EVAL
825 
826 #if defined(FEAT_JOB_CHANNEL) \
827 	|| defined(FEAT_PYTHON) || defined(FEAT_PYTHON3) \
828 	|| defined(PROTO)
829 /*
830  * Make "buf" the current buffer.  restore_buffer() MUST be called to undo.
831  * No autocommands will be executed.  Use aucmd_prepbuf() if there are any.
832  */
833     void
834 switch_buffer(bufref_T *save_curbuf, buf_T *buf)
835 {
836     block_autocmds();
837     set_bufref(save_curbuf, curbuf);
838     --curbuf->b_nwindows;
839     curbuf = buf;
840     curwin->w_buffer = buf;
841     ++curbuf->b_nwindows;
842 }
843 
844 /*
845  * Restore the current buffer after using switch_buffer().
846  */
847     void
848 restore_buffer(bufref_T *save_curbuf)
849 {
850     unblock_autocmds();
851     // Check for valid buffer, just in case.
852     if (bufref_valid(save_curbuf))
853     {
854 	--curbuf->b_nwindows;
855 	curwin->w_buffer = save_curbuf->br_buf;
856 	curbuf = save_curbuf->br_buf;
857 	++curbuf->b_nwindows;
858     }
859 }
860 
861 /*
862  * Find a window for buffer "buf".
863  * If found OK is returned and "wp" and "tp" are set to the window and tabpage.
864  * If not found FAIL is returned.
865  */
866     static int
867 find_win_for_buf(
868     buf_T     *buf,
869     win_T     **wp,
870     tabpage_T **tp)
871 {
872     FOR_ALL_TAB_WINDOWS(*tp, *wp)
873 	if ((*wp)->w_buffer == buf)
874 	    return OK;
875     return FAIL;
876 }
877 
878 /*
879  * Find a window that contains "buf" and switch to it.
880  * If there is no such window, use the current window and change "curbuf".
881  * Caller must initialize save_curbuf to NULL.
882  * restore_win_for_buf() MUST be called later!
883  */
884     void
885 switch_to_win_for_buf(
886     buf_T	*buf,
887     win_T	**save_curwinp,
888     tabpage_T	**save_curtabp,
889     bufref_T	*save_curbuf)
890 {
891     win_T	*wp;
892     tabpage_T	*tp;
893 
894     if (find_win_for_buf(buf, &wp, &tp) == FAIL)
895 	switch_buffer(save_curbuf, buf);
896     else if (switch_win(save_curwinp, save_curtabp, wp, tp, TRUE) == FAIL)
897     {
898 	restore_win(*save_curwinp, *save_curtabp, TRUE);
899 	switch_buffer(save_curbuf, buf);
900     }
901 }
902 
903     void
904 restore_win_for_buf(
905     win_T	*save_curwin,
906     tabpage_T	*save_curtab,
907     bufref_T	*save_curbuf)
908 {
909     if (save_curbuf->br_buf == NULL)
910 	restore_win(save_curwin, save_curtab, TRUE);
911     else
912 	restore_buffer(save_curbuf);
913 }
914 #endif
915