xref: /vim-8.2.3635/src/evalbuffer.c (revision 4a15504e)
1261f346fSBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet:
2261f346fSBram Moolenaar  *
3261f346fSBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
4261f346fSBram Moolenaar  *
5261f346fSBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
6261f346fSBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
7261f346fSBram Moolenaar  * See README.txt for an overview of the Vim source code.
8261f346fSBram Moolenaar  */
9261f346fSBram Moolenaar 
10261f346fSBram Moolenaar /*
11261f346fSBram Moolenaar  * evalbuffer.c: Buffer related builtin functions
12261f346fSBram Moolenaar  */
13261f346fSBram Moolenaar 
14261f346fSBram Moolenaar #include "vim.h"
15261f346fSBram Moolenaar 
16261f346fSBram Moolenaar #if defined(FEAT_EVAL) || defined(PROTO)
17261f346fSBram Moolenaar /*
18261f346fSBram Moolenaar  * Mark references in functions of buffers.
19261f346fSBram Moolenaar  */
20261f346fSBram Moolenaar     int
set_ref_in_buffers(int copyID)21261f346fSBram Moolenaar set_ref_in_buffers(int copyID)
22261f346fSBram Moolenaar {
23261f346fSBram Moolenaar     int		abort = FALSE;
24261f346fSBram Moolenaar     buf_T	*bp;
25261f346fSBram Moolenaar 
26261f346fSBram Moolenaar     FOR_ALL_BUFFERS(bp)
27261f346fSBram Moolenaar     {
28261f346fSBram Moolenaar 	listener_T *lnr;
29261f346fSBram Moolenaar 	typval_T tv;
30261f346fSBram Moolenaar 
31261f346fSBram Moolenaar 	for (lnr = bp->b_listener; !abort && lnr != NULL; lnr = lnr->lr_next)
32261f346fSBram Moolenaar 	{
33261f346fSBram Moolenaar 	    if (lnr->lr_callback.cb_partial != NULL)
34261f346fSBram Moolenaar 	    {
35261f346fSBram Moolenaar 		tv.v_type = VAR_PARTIAL;
36261f346fSBram Moolenaar 		tv.vval.v_partial = lnr->lr_callback.cb_partial;
37261f346fSBram Moolenaar 		abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
38261f346fSBram Moolenaar 	    }
39261f346fSBram Moolenaar 	}
40261f346fSBram Moolenaar # ifdef FEAT_JOB_CHANNEL
41261f346fSBram Moolenaar 	if (!abort && bp->b_prompt_callback.cb_partial != NULL)
42261f346fSBram Moolenaar 	{
43261f346fSBram Moolenaar 	    tv.v_type = VAR_PARTIAL;
44261f346fSBram Moolenaar 	    tv.vval.v_partial = bp->b_prompt_callback.cb_partial;
45261f346fSBram Moolenaar 	    abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
46261f346fSBram Moolenaar 	}
47261f346fSBram Moolenaar 	if (!abort && bp->b_prompt_interrupt.cb_partial != NULL)
48261f346fSBram Moolenaar 	{
49261f346fSBram Moolenaar 	    tv.v_type = VAR_PARTIAL;
50261f346fSBram Moolenaar 	    tv.vval.v_partial = bp->b_prompt_interrupt.cb_partial;
51261f346fSBram Moolenaar 	    abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
52261f346fSBram Moolenaar 	}
53261f346fSBram Moolenaar # endif
54261f346fSBram Moolenaar 	if (abort)
55261f346fSBram Moolenaar 	    break;
56261f346fSBram Moolenaar     }
57261f346fSBram Moolenaar     return abort;
58261f346fSBram Moolenaar }
59261f346fSBram Moolenaar 
60261f346fSBram Moolenaar     buf_T *
buflist_find_by_name(char_u * name,int curtab_only)61261f346fSBram Moolenaar buflist_find_by_name(char_u *name, int curtab_only)
62261f346fSBram Moolenaar {
63261f346fSBram Moolenaar     int		save_magic;
64261f346fSBram Moolenaar     char_u	*save_cpo;
65261f346fSBram Moolenaar     buf_T	*buf;
66261f346fSBram Moolenaar 
67261f346fSBram Moolenaar     // Ignore 'magic' and 'cpoptions' here to make scripts portable
68261f346fSBram Moolenaar     save_magic = p_magic;
69261f346fSBram Moolenaar     p_magic = TRUE;
70261f346fSBram Moolenaar     save_cpo = p_cpo;
71e5a2dc87SBram Moolenaar     p_cpo = empty_option;
72261f346fSBram Moolenaar 
73261f346fSBram Moolenaar     buf = buflist_findnr(buflist_findpat(name, name + STRLEN(name),
74261f346fSBram Moolenaar 						    TRUE, FALSE, curtab_only));
75261f346fSBram Moolenaar 
76261f346fSBram Moolenaar     p_magic = save_magic;
77261f346fSBram Moolenaar     p_cpo = save_cpo;
78261f346fSBram Moolenaar     return buf;
79261f346fSBram Moolenaar }
80261f346fSBram Moolenaar 
81261f346fSBram Moolenaar /*
82261f346fSBram Moolenaar  * Find a buffer by number or exact name.
83261f346fSBram Moolenaar  */
84261f346fSBram Moolenaar     buf_T *
find_buffer(typval_T * avar)85261f346fSBram Moolenaar find_buffer(typval_T *avar)
86261f346fSBram Moolenaar {
87261f346fSBram Moolenaar     buf_T	*buf = NULL;
88261f346fSBram Moolenaar 
89261f346fSBram Moolenaar     if (avar->v_type == VAR_NUMBER)
90261f346fSBram Moolenaar 	buf = buflist_findnr((int)avar->vval.v_number);
917b45d46cSBram Moolenaar     else if (in_vim9script() && check_for_string_arg(avar, 0) == FAIL)
927b45d46cSBram Moolenaar 	return NULL;
93261f346fSBram Moolenaar     else if (avar->v_type == VAR_STRING && avar->vval.v_string != NULL)
94261f346fSBram Moolenaar     {
95261f346fSBram Moolenaar 	buf = buflist_findname_exp(avar->vval.v_string);
96261f346fSBram Moolenaar 	if (buf == NULL)
97261f346fSBram Moolenaar 	{
98261f346fSBram Moolenaar 	    // No full path name match, try a match with a URL or a "nofile"
99261f346fSBram Moolenaar 	    // buffer, these don't use the full path.
100261f346fSBram Moolenaar 	    FOR_ALL_BUFFERS(buf)
101261f346fSBram Moolenaar 		if (buf->b_fname != NULL
102261f346fSBram Moolenaar 			&& (path_with_url(buf->b_fname)
103261f346fSBram Moolenaar #ifdef FEAT_QUICKFIX
104261f346fSBram Moolenaar 			    || bt_nofilename(buf)
105261f346fSBram Moolenaar #endif
106261f346fSBram Moolenaar 			   )
107261f346fSBram Moolenaar 			&& STRCMP(buf->b_fname, avar->vval.v_string) == 0)
108261f346fSBram Moolenaar 		    break;
109261f346fSBram Moolenaar 	}
110261f346fSBram Moolenaar     }
111261f346fSBram Moolenaar     return buf;
112261f346fSBram Moolenaar }
113261f346fSBram Moolenaar 
114261f346fSBram Moolenaar /*
115261f346fSBram Moolenaar  * If there is a window for "curbuf", make it the current window.
116261f346fSBram Moolenaar  */
117261f346fSBram Moolenaar     static void
find_win_for_curbuf(void)118261f346fSBram Moolenaar find_win_for_curbuf(void)
119261f346fSBram Moolenaar {
120261f346fSBram Moolenaar     wininfo_T *wip;
121261f346fSBram Moolenaar 
122aeea7215SBram Moolenaar     FOR_ALL_BUF_WININFO(curbuf, wip)
123261f346fSBram Moolenaar     {
124261f346fSBram Moolenaar 	if (wip->wi_win != NULL)
125261f346fSBram Moolenaar 	{
126261f346fSBram Moolenaar 	    curwin = wip->wi_win;
127261f346fSBram Moolenaar 	    break;
128261f346fSBram Moolenaar 	}
129261f346fSBram Moolenaar     }
130261f346fSBram Moolenaar }
131261f346fSBram Moolenaar 
132261f346fSBram Moolenaar /*
13334453208SBram Moolenaar  * Set line or list of lines in buffer "buf" to "lines".
13434453208SBram Moolenaar  * Any type is allowed and converted to a string.
135261f346fSBram Moolenaar  */
136261f346fSBram Moolenaar     static void
set_buffer_lines(buf_T * buf,linenr_T lnum_arg,int append,typval_T * lines,typval_T * rettv)137261f346fSBram Moolenaar set_buffer_lines(
138261f346fSBram Moolenaar 	buf_T	    *buf,
139261f346fSBram Moolenaar 	linenr_T    lnum_arg,
140261f346fSBram Moolenaar 	int	    append,
141261f346fSBram Moolenaar 	typval_T    *lines,
142261f346fSBram Moolenaar 	typval_T    *rettv)
143261f346fSBram Moolenaar {
144261f346fSBram Moolenaar     linenr_T    lnum = lnum_arg + (append ? 1 : 0);
145261f346fSBram Moolenaar     char_u	*line = NULL;
146261f346fSBram Moolenaar     list_T	*l = NULL;
147261f346fSBram Moolenaar     listitem_T	*li = NULL;
148261f346fSBram Moolenaar     long	added = 0;
149261f346fSBram Moolenaar     linenr_T	append_lnum;
150261f346fSBram Moolenaar     buf_T	*curbuf_save = NULL;
151261f346fSBram Moolenaar     win_T	*curwin_save = NULL;
152261f346fSBram Moolenaar     int		is_curbuf = buf == curbuf;
153261f346fSBram Moolenaar 
154261f346fSBram Moolenaar     // When using the current buffer ml_mfp will be set if needed.  Useful when
155261f346fSBram Moolenaar     // setline() is used on startup.  For other buffers the buffer must be
156261f346fSBram Moolenaar     // loaded.
157261f346fSBram Moolenaar     if (buf == NULL || (!is_curbuf && buf->b_ml.ml_mfp == NULL) || lnum < 1)
158261f346fSBram Moolenaar     {
159261f346fSBram Moolenaar 	rettv->vval.v_number = 1;	// FAIL
160261f346fSBram Moolenaar 	return;
161261f346fSBram Moolenaar     }
162261f346fSBram Moolenaar 
163261f346fSBram Moolenaar     if (!is_curbuf)
164261f346fSBram Moolenaar     {
165261f346fSBram Moolenaar 	curbuf_save = curbuf;
166261f346fSBram Moolenaar 	curwin_save = curwin;
167261f346fSBram Moolenaar 	curbuf = buf;
168261f346fSBram Moolenaar 	find_win_for_curbuf();
169261f346fSBram Moolenaar     }
170261f346fSBram Moolenaar 
171261f346fSBram Moolenaar     if (append)
172261f346fSBram Moolenaar 	// appendbufline() uses the line number below which we insert
173261f346fSBram Moolenaar 	append_lnum = lnum - 1;
174261f346fSBram Moolenaar     else
175261f346fSBram Moolenaar 	// setbufline() uses the line number above which we insert, we only
176261f346fSBram Moolenaar 	// append if it's below the last line
177261f346fSBram Moolenaar 	append_lnum = curbuf->b_ml.ml_line_count;
178261f346fSBram Moolenaar 
179261f346fSBram Moolenaar     if (lines->v_type == VAR_LIST)
180261f346fSBram Moolenaar     {
181261f346fSBram Moolenaar 	l = lines->vval.v_list;
182ad48e6c1SBram Moolenaar 	if (l == NULL || list_len(l) == 0)
183ad48e6c1SBram Moolenaar 	{
184ad48e6c1SBram Moolenaar 	    // set proper return code
185ad48e6c1SBram Moolenaar 	    if (lnum > curbuf->b_ml.ml_line_count)
186ad48e6c1SBram Moolenaar 		rettv->vval.v_number = 1;	// FAIL
187ad48e6c1SBram Moolenaar 	    goto done;
188ad48e6c1SBram Moolenaar 	}
1897e9f351bSBram Moolenaar 	CHECK_LIST_MATERIALIZE(l);
190261f346fSBram Moolenaar 	li = l->lv_first;
191261f346fSBram Moolenaar     }
192261f346fSBram Moolenaar     else
19334453208SBram Moolenaar 	line = typval_tostring(lines, FALSE);
194261f346fSBram Moolenaar 
195261f346fSBram Moolenaar     // default result is zero == OK
196261f346fSBram Moolenaar     for (;;)
197261f346fSBram Moolenaar     {
198261f346fSBram Moolenaar 	if (l != NULL)
199261f346fSBram Moolenaar 	{
200261f346fSBram Moolenaar 	    // list argument, get next string
201261f346fSBram Moolenaar 	    if (li == NULL)
202261f346fSBram Moolenaar 		break;
20334453208SBram Moolenaar 	    vim_free(line);
20434453208SBram Moolenaar 	    line = typval_tostring(&li->li_tv, FALSE);
205261f346fSBram Moolenaar 	    li = li->li_next;
206261f346fSBram Moolenaar 	}
207261f346fSBram Moolenaar 
208261f346fSBram Moolenaar 	rettv->vval.v_number = 1;	// FAIL
209261f346fSBram Moolenaar 	if (line == NULL || lnum > curbuf->b_ml.ml_line_count + 1)
210261f346fSBram Moolenaar 	    break;
211261f346fSBram Moolenaar 
212261f346fSBram Moolenaar 	// When coming here from Insert mode, sync undo, so that this can be
213261f346fSBram Moolenaar 	// undone separately from what was previously inserted.
214261f346fSBram Moolenaar 	if (u_sync_once == 2)
215261f346fSBram Moolenaar 	{
216261f346fSBram Moolenaar 	    u_sync_once = 1; // notify that u_sync() was called
217261f346fSBram Moolenaar 	    u_sync(TRUE);
218261f346fSBram Moolenaar 	}
219261f346fSBram Moolenaar 
220261f346fSBram Moolenaar 	if (!append && lnum <= curbuf->b_ml.ml_line_count)
221261f346fSBram Moolenaar 	{
222261f346fSBram Moolenaar 	    // Existing line, replace it.
223261f346fSBram Moolenaar 	    // Removes any existing text properties.
224261f346fSBram Moolenaar 	    if (u_savesub(lnum) == OK && ml_replace_len(
225261f346fSBram Moolenaar 		      lnum, line, (colnr_T)STRLEN(line) + 1, TRUE, TRUE) == OK)
226261f346fSBram Moolenaar 	    {
227261f346fSBram Moolenaar 		changed_bytes(lnum, 0);
228261f346fSBram Moolenaar 		if (is_curbuf && lnum == curwin->w_cursor.lnum)
229261f346fSBram Moolenaar 		    check_cursor_col();
230261f346fSBram Moolenaar 		rettv->vval.v_number = 0;	// OK
231261f346fSBram Moolenaar 	    }
232261f346fSBram Moolenaar 	}
233261f346fSBram Moolenaar 	else if (added > 0 || u_save(lnum - 1, lnum) == OK)
234261f346fSBram Moolenaar 	{
235261f346fSBram Moolenaar 	    // append the line
236261f346fSBram Moolenaar 	    ++added;
237261f346fSBram Moolenaar 	    if (ml_append(lnum - 1, line, (colnr_T)0, FALSE) == OK)
238261f346fSBram Moolenaar 		rettv->vval.v_number = 0;	// OK
239261f346fSBram Moolenaar 	}
240261f346fSBram Moolenaar 
241261f346fSBram Moolenaar 	if (l == NULL)			// only one string argument
242261f346fSBram Moolenaar 	    break;
243261f346fSBram Moolenaar 	++lnum;
244261f346fSBram Moolenaar     }
24534453208SBram Moolenaar     vim_free(line);
246261f346fSBram Moolenaar 
247261f346fSBram Moolenaar     if (added > 0)
248261f346fSBram Moolenaar     {
249261f346fSBram Moolenaar 	win_T	    *wp;
250261f346fSBram Moolenaar 	tabpage_T   *tp;
251261f346fSBram Moolenaar 
252261f346fSBram Moolenaar 	appended_lines_mark(append_lnum, added);
253261f346fSBram Moolenaar 
254261f346fSBram Moolenaar 	// Only adjust the cursor for buffers other than the current, unless it
255261f346fSBram Moolenaar 	// is the current window.  For curbuf and other windows it has been
256261f346fSBram Moolenaar 	// done in mark_adjust_internal().
257261f346fSBram Moolenaar 	FOR_ALL_TAB_WINDOWS(tp, wp)
258261f346fSBram Moolenaar 	    if (wp->w_buffer == buf
259261f346fSBram Moolenaar 		    && (wp->w_buffer != curbuf || wp == curwin)
260261f346fSBram Moolenaar 		    && wp->w_cursor.lnum > append_lnum)
261261f346fSBram Moolenaar 		wp->w_cursor.lnum += added;
262261f346fSBram Moolenaar 	check_cursor_col();
263261f346fSBram Moolenaar 	update_topline();
264261f346fSBram Moolenaar     }
265261f346fSBram Moolenaar 
266ad48e6c1SBram Moolenaar done:
267261f346fSBram Moolenaar     if (!is_curbuf)
268261f346fSBram Moolenaar     {
269261f346fSBram Moolenaar 	curbuf = curbuf_save;
270261f346fSBram Moolenaar 	curwin = curwin_save;
271261f346fSBram Moolenaar     }
272261f346fSBram Moolenaar }
273261f346fSBram Moolenaar 
274261f346fSBram Moolenaar /*
275261f346fSBram Moolenaar  * "append(lnum, string/list)" function
276261f346fSBram Moolenaar  */
277261f346fSBram Moolenaar     void
f_append(typval_T * argvars,typval_T * rettv)278261f346fSBram Moolenaar f_append(typval_T *argvars, typval_T *rettv)
279261f346fSBram Moolenaar {
2804490ec4eSYegappan Lakshmanan     linenr_T	lnum;
281261f346fSBram Moolenaar 
2824490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
2834490ec4eSYegappan Lakshmanan 	return;
2844490ec4eSYegappan Lakshmanan 
2854490ec4eSYegappan Lakshmanan     lnum = tv_get_lnum(&argvars[0]);
286261f346fSBram Moolenaar     set_buffer_lines(curbuf, lnum, TRUE, &argvars[1], rettv);
287261f346fSBram Moolenaar }
288261f346fSBram Moolenaar 
289261f346fSBram Moolenaar /*
290*4a15504eSYegappan Lakshmanan  * Set or append lines to a buffer.
291261f346fSBram Moolenaar  */
292*4a15504eSYegappan Lakshmanan     static void
buf_set_append_line(typval_T * argvars,typval_T * rettv,int append)293*4a15504eSYegappan Lakshmanan buf_set_append_line(typval_T *argvars, typval_T *rettv, int append)
294261f346fSBram Moolenaar {
295261f346fSBram Moolenaar     linenr_T	lnum;
296261f346fSBram Moolenaar     buf_T	*buf;
297261f346fSBram Moolenaar 
2984490ec4eSYegappan Lakshmanan     if (in_vim9script()
2994490ec4eSYegappan Lakshmanan 	    && (check_for_buffer_arg(argvars, 0) == FAIL
3004490ec4eSYegappan Lakshmanan 		|| check_for_lnum_arg(argvars, 1) == FAIL
3014490ec4eSYegappan Lakshmanan 		|| check_for_string_or_number_or_list_arg(argvars, 2) == FAIL))
3024490ec4eSYegappan Lakshmanan 	return;
3034490ec4eSYegappan Lakshmanan 
304261f346fSBram Moolenaar     buf = tv_get_buf(&argvars[0], FALSE);
305261f346fSBram Moolenaar     if (buf == NULL)
306261f346fSBram Moolenaar 	rettv->vval.v_number = 1; // FAIL
307261f346fSBram Moolenaar     else
308261f346fSBram Moolenaar     {
309261f346fSBram Moolenaar 	lnum = tv_get_lnum_buf(&argvars[1], buf);
310*4a15504eSYegappan Lakshmanan 	set_buffer_lines(buf, lnum, append, &argvars[2], rettv);
311261f346fSBram Moolenaar     }
312261f346fSBram Moolenaar }
313261f346fSBram Moolenaar 
314261f346fSBram Moolenaar /*
315*4a15504eSYegappan Lakshmanan  * "appendbufline(buf, lnum, string/list)" function
316*4a15504eSYegappan Lakshmanan  */
317*4a15504eSYegappan Lakshmanan     void
f_appendbufline(typval_T * argvars,typval_T * rettv)318*4a15504eSYegappan Lakshmanan f_appendbufline(typval_T *argvars, typval_T *rettv)
319*4a15504eSYegappan Lakshmanan {
320*4a15504eSYegappan Lakshmanan     buf_set_append_line(argvars, rettv, TRUE);
321*4a15504eSYegappan Lakshmanan }
322*4a15504eSYegappan Lakshmanan 
323*4a15504eSYegappan Lakshmanan /*
324261f346fSBram Moolenaar  * "bufadd(expr)" function
325261f346fSBram Moolenaar  */
326261f346fSBram Moolenaar     void
f_bufadd(typval_T * argvars,typval_T * rettv)327261f346fSBram Moolenaar f_bufadd(typval_T *argvars, typval_T *rettv)
328261f346fSBram Moolenaar {
3294490ec4eSYegappan Lakshmanan     char_u *name;
330261f346fSBram Moolenaar 
3314490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
3324490ec4eSYegappan Lakshmanan 	return;
3334490ec4eSYegappan Lakshmanan 
3344490ec4eSYegappan Lakshmanan     name = tv_get_string(&argvars[0]);
335261f346fSBram Moolenaar     rettv->vval.v_number = buflist_add(*name == NUL ? NULL : name, 0);
336261f346fSBram Moolenaar }
337261f346fSBram Moolenaar 
338261f346fSBram Moolenaar /*
339261f346fSBram Moolenaar  * "bufexists(expr)" function
340261f346fSBram Moolenaar  */
341261f346fSBram Moolenaar     void
f_bufexists(typval_T * argvars,typval_T * rettv)342261f346fSBram Moolenaar f_bufexists(typval_T *argvars, typval_T *rettv)
343261f346fSBram Moolenaar {
3444490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
3454490ec4eSYegappan Lakshmanan 	return;
3464490ec4eSYegappan Lakshmanan 
347261f346fSBram Moolenaar     rettv->vval.v_number = (find_buffer(&argvars[0]) != NULL);
348261f346fSBram Moolenaar }
349261f346fSBram Moolenaar 
350261f346fSBram Moolenaar /*
351261f346fSBram Moolenaar  * "buflisted(expr)" function
352261f346fSBram Moolenaar  */
353261f346fSBram Moolenaar     void
f_buflisted(typval_T * argvars,typval_T * rettv)354261f346fSBram Moolenaar f_buflisted(typval_T *argvars, typval_T *rettv)
355261f346fSBram Moolenaar {
356261f346fSBram Moolenaar     buf_T	*buf;
357261f346fSBram Moolenaar 
3584490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
3594490ec4eSYegappan Lakshmanan 	return;
3604490ec4eSYegappan Lakshmanan 
361261f346fSBram Moolenaar     buf = find_buffer(&argvars[0]);
362261f346fSBram Moolenaar     rettv->vval.v_number = (buf != NULL && buf->b_p_bl);
363261f346fSBram Moolenaar }
364261f346fSBram Moolenaar 
365261f346fSBram Moolenaar /*
366261f346fSBram Moolenaar  * "bufload(expr)" function
367261f346fSBram Moolenaar  */
368261f346fSBram Moolenaar     void
f_bufload(typval_T * argvars,typval_T * rettv UNUSED)369261f346fSBram Moolenaar f_bufload(typval_T *argvars, typval_T *rettv UNUSED)
370261f346fSBram Moolenaar {
3714490ec4eSYegappan Lakshmanan     buf_T	*buf;
372261f346fSBram Moolenaar 
3734490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
3744490ec4eSYegappan Lakshmanan 	return;
3754490ec4eSYegappan Lakshmanan 
3764490ec4eSYegappan Lakshmanan     buf = get_buf_arg(&argvars[0]);
377261f346fSBram Moolenaar     if (buf != NULL)
378261f346fSBram Moolenaar 	buffer_ensure_loaded(buf);
379261f346fSBram Moolenaar }
380261f346fSBram Moolenaar 
381261f346fSBram Moolenaar /*
382261f346fSBram Moolenaar  * "bufloaded(expr)" function
383261f346fSBram Moolenaar  */
384261f346fSBram Moolenaar     void
f_bufloaded(typval_T * argvars,typval_T * rettv)385261f346fSBram Moolenaar f_bufloaded(typval_T *argvars, typval_T *rettv)
386261f346fSBram Moolenaar {
387261f346fSBram Moolenaar     buf_T	*buf;
388261f346fSBram Moolenaar 
3894490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
3904490ec4eSYegappan Lakshmanan 	return;
3914490ec4eSYegappan Lakshmanan 
392261f346fSBram Moolenaar     buf = find_buffer(&argvars[0]);
393261f346fSBram Moolenaar     rettv->vval.v_number = (buf != NULL && buf->b_ml.ml_mfp != NULL);
394261f346fSBram Moolenaar }
395261f346fSBram Moolenaar 
396261f346fSBram Moolenaar /*
397261f346fSBram Moolenaar  * "bufname(expr)" function
398261f346fSBram Moolenaar  */
399261f346fSBram Moolenaar     void
f_bufname(typval_T * argvars,typval_T * rettv)400261f346fSBram Moolenaar f_bufname(typval_T *argvars, typval_T *rettv)
401261f346fSBram Moolenaar {
402261f346fSBram Moolenaar     buf_T	*buf;
40302aaad91SBram Moolenaar     typval_T	*tv = &argvars[0];
404261f346fSBram Moolenaar 
4054490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_opt_buffer_arg(argvars, 0) == FAIL)
4064490ec4eSYegappan Lakshmanan 	return;
4074490ec4eSYegappan Lakshmanan 
40802aaad91SBram Moolenaar     if (tv->v_type == VAR_UNKNOWN)
409261f346fSBram Moolenaar 	buf = curbuf;
410261f346fSBram Moolenaar     else
4113767e3a3SBram Moolenaar 	buf = tv_get_buf_from_arg(tv);
412261f346fSBram Moolenaar     rettv->v_type = VAR_STRING;
413261f346fSBram Moolenaar     if (buf != NULL && buf->b_fname != NULL)
414261f346fSBram Moolenaar 	rettv->vval.v_string = vim_strsave(buf->b_fname);
415261f346fSBram Moolenaar     else
416261f346fSBram Moolenaar 	rettv->vval.v_string = NULL;
417261f346fSBram Moolenaar }
418261f346fSBram Moolenaar 
419261f346fSBram Moolenaar /*
420261f346fSBram Moolenaar  * "bufnr(expr)" function
421261f346fSBram Moolenaar  */
422261f346fSBram Moolenaar     void
f_bufnr(typval_T * argvars,typval_T * rettv)423261f346fSBram Moolenaar f_bufnr(typval_T *argvars, typval_T *rettv)
424261f346fSBram Moolenaar {
425261f346fSBram Moolenaar     buf_T	*buf;
426261f346fSBram Moolenaar     int		error = FALSE;
427261f346fSBram Moolenaar     char_u	*name;
428261f346fSBram Moolenaar 
42983494b4aSYegappan Lakshmanan     if (in_vim9script()
430cd917207SYegappan Lakshmanan 	    && (check_for_opt_buffer_arg(argvars, 0) == FAIL
43183494b4aSYegappan Lakshmanan 		|| (argvars[0].v_type != VAR_UNKNOWN
43283494b4aSYegappan Lakshmanan 		    && check_for_opt_bool_arg(argvars, 1) == FAIL)))
43383494b4aSYegappan Lakshmanan 	return;
43483494b4aSYegappan Lakshmanan 
435261f346fSBram Moolenaar     if (argvars[0].v_type == VAR_UNKNOWN)
436261f346fSBram Moolenaar 	buf = curbuf;
437261f346fSBram Moolenaar     else
4383767e3a3SBram Moolenaar 	buf = tv_get_buf_from_arg(&argvars[0]);
439261f346fSBram Moolenaar 
440261f346fSBram Moolenaar     // If the buffer isn't found and the second argument is not zero create a
441261f346fSBram Moolenaar     // new buffer.
442261f346fSBram Moolenaar     if (buf == NULL
443261f346fSBram Moolenaar 	    && argvars[1].v_type != VAR_UNKNOWN
444fe136c9aSBram Moolenaar 	    && tv_get_bool_chk(&argvars[1], &error) != 0
445261f346fSBram Moolenaar 	    && !error
446261f346fSBram Moolenaar 	    && (name = tv_get_string_chk(&argvars[0])) != NULL
447261f346fSBram Moolenaar 	    && !error)
448261f346fSBram Moolenaar 	buf = buflist_new(name, NULL, (linenr_T)1, 0);
449261f346fSBram Moolenaar 
450261f346fSBram Moolenaar     if (buf != NULL)
451261f346fSBram Moolenaar 	rettv->vval.v_number = buf->b_fnum;
452261f346fSBram Moolenaar     else
453261f346fSBram Moolenaar 	rettv->vval.v_number = -1;
454261f346fSBram Moolenaar }
455261f346fSBram Moolenaar 
456261f346fSBram Moolenaar     static void
buf_win_common(typval_T * argvars,typval_T * rettv,int get_nr)457261f346fSBram Moolenaar buf_win_common(typval_T *argvars, typval_T *rettv, int get_nr)
458261f346fSBram Moolenaar {
459261f346fSBram Moolenaar     win_T	*wp;
460261f346fSBram Moolenaar     int		winnr = 0;
461261f346fSBram Moolenaar     buf_T	*buf;
462261f346fSBram Moolenaar 
4634490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_buffer_arg(argvars, 0) == FAIL)
4644490ec4eSYegappan Lakshmanan 	return;
4654490ec4eSYegappan Lakshmanan 
4663767e3a3SBram Moolenaar     buf = tv_get_buf_from_arg(&argvars[0]);
467261f346fSBram Moolenaar     FOR_ALL_WINDOWS(wp)
468261f346fSBram Moolenaar     {
469261f346fSBram Moolenaar 	++winnr;
470261f346fSBram Moolenaar 	if (wp->w_buffer == buf)
471261f346fSBram Moolenaar 	    break;
472261f346fSBram Moolenaar     }
473261f346fSBram Moolenaar     rettv->vval.v_number = (wp != NULL ? (get_nr ? winnr : wp->w_id) : -1);
474261f346fSBram Moolenaar }
475261f346fSBram Moolenaar 
476261f346fSBram Moolenaar /*
477261f346fSBram Moolenaar  * "bufwinid(nr)" function
478261f346fSBram Moolenaar  */
479261f346fSBram Moolenaar     void
f_bufwinid(typval_T * argvars,typval_T * rettv)480261f346fSBram Moolenaar f_bufwinid(typval_T *argvars, typval_T *rettv)
481261f346fSBram Moolenaar {
482261f346fSBram Moolenaar     buf_win_common(argvars, rettv, FALSE);
483261f346fSBram Moolenaar }
484261f346fSBram Moolenaar 
485261f346fSBram Moolenaar /*
486261f346fSBram Moolenaar  * "bufwinnr(nr)" function
487261f346fSBram Moolenaar  */
488261f346fSBram Moolenaar     void
f_bufwinnr(typval_T * argvars,typval_T * rettv)489261f346fSBram Moolenaar f_bufwinnr(typval_T *argvars, typval_T *rettv)
490261f346fSBram Moolenaar {
491261f346fSBram Moolenaar     buf_win_common(argvars, rettv, TRUE);
492261f346fSBram Moolenaar }
493261f346fSBram Moolenaar 
494261f346fSBram Moolenaar /*
495261f346fSBram Moolenaar  * "deletebufline()" function
496261f346fSBram Moolenaar  */
497261f346fSBram Moolenaar     void
f_deletebufline(typval_T * argvars,typval_T * rettv)498261f346fSBram Moolenaar f_deletebufline(typval_T *argvars, typval_T *rettv)
499261f346fSBram Moolenaar {
500261f346fSBram Moolenaar     buf_T	*buf;
501261f346fSBram Moolenaar     linenr_T	first, last;
502261f346fSBram Moolenaar     linenr_T	lnum;
503261f346fSBram Moolenaar     long	count;
504261f346fSBram Moolenaar     int		is_curbuf;
505261f346fSBram Moolenaar     buf_T	*curbuf_save = NULL;
506261f346fSBram Moolenaar     win_T	*curwin_save = NULL;
507261f346fSBram Moolenaar     tabpage_T	*tp;
508261f346fSBram Moolenaar     win_T	*wp;
509261f346fSBram Moolenaar 
51083494b4aSYegappan Lakshmanan     if (in_vim9script()
51183494b4aSYegappan Lakshmanan 	    && (check_for_buffer_arg(argvars, 0) == FAIL
51283494b4aSYegappan Lakshmanan 		|| check_for_lnum_arg(argvars, 1) == FAIL
51383494b4aSYegappan Lakshmanan 		|| check_for_opt_lnum_arg(argvars, 2) == FAIL))
51483494b4aSYegappan Lakshmanan 	return;
51583494b4aSYegappan Lakshmanan 
516261f346fSBram Moolenaar     buf = tv_get_buf(&argvars[0], FALSE);
517261f346fSBram Moolenaar     if (buf == NULL)
518261f346fSBram Moolenaar     {
519261f346fSBram Moolenaar 	rettv->vval.v_number = 1; // FAIL
520261f346fSBram Moolenaar 	return;
521261f346fSBram Moolenaar     }
522261f346fSBram Moolenaar     is_curbuf = buf == curbuf;
523261f346fSBram Moolenaar 
524261f346fSBram Moolenaar     first = tv_get_lnum_buf(&argvars[1], buf);
525261f346fSBram Moolenaar     if (argvars[2].v_type != VAR_UNKNOWN)
526261f346fSBram Moolenaar 	last = tv_get_lnum_buf(&argvars[2], buf);
527261f346fSBram Moolenaar     else
528261f346fSBram Moolenaar 	last = first;
529261f346fSBram Moolenaar 
530261f346fSBram Moolenaar     if (buf->b_ml.ml_mfp == NULL || first < 1
531261f346fSBram Moolenaar 			   || first > buf->b_ml.ml_line_count || last < first)
532261f346fSBram Moolenaar     {
533261f346fSBram Moolenaar 	rettv->vval.v_number = 1;	// FAIL
534261f346fSBram Moolenaar 	return;
535261f346fSBram Moolenaar     }
536261f346fSBram Moolenaar 
537261f346fSBram Moolenaar     if (!is_curbuf)
538261f346fSBram Moolenaar     {
539261f346fSBram Moolenaar 	curbuf_save = curbuf;
540261f346fSBram Moolenaar 	curwin_save = curwin;
541261f346fSBram Moolenaar 	curbuf = buf;
542261f346fSBram Moolenaar 	find_win_for_curbuf();
543261f346fSBram Moolenaar     }
544261f346fSBram Moolenaar     if (last > curbuf->b_ml.ml_line_count)
545261f346fSBram Moolenaar 	last = curbuf->b_ml.ml_line_count;
546261f346fSBram Moolenaar     count = last - first + 1;
547261f346fSBram Moolenaar 
548261f346fSBram Moolenaar     // When coming here from Insert mode, sync undo, so that this can be
549261f346fSBram Moolenaar     // undone separately from what was previously inserted.
550261f346fSBram Moolenaar     if (u_sync_once == 2)
551261f346fSBram Moolenaar     {
552261f346fSBram Moolenaar 	u_sync_once = 1; // notify that u_sync() was called
553261f346fSBram Moolenaar 	u_sync(TRUE);
554261f346fSBram Moolenaar     }
555261f346fSBram Moolenaar 
556261f346fSBram Moolenaar     if (u_save(first - 1, last + 1) == FAIL)
557261f346fSBram Moolenaar     {
558261f346fSBram Moolenaar 	rettv->vval.v_number = 1;	// FAIL
559261f346fSBram Moolenaar     }
560963ffa0aSBram Moolenaar     else
561963ffa0aSBram Moolenaar     {
562261f346fSBram Moolenaar 	for (lnum = first; lnum <= last; ++lnum)
563ca70c07bSBram Moolenaar 	    ml_delete_flags(first, ML_DEL_MESSAGE);
564261f346fSBram Moolenaar 
565261f346fSBram Moolenaar 	FOR_ALL_TAB_WINDOWS(tp, wp)
566261f346fSBram Moolenaar 	    if (wp->w_buffer == buf)
567261f346fSBram Moolenaar 	    {
568261f346fSBram Moolenaar 		if (wp->w_cursor.lnum > last)
569261f346fSBram Moolenaar 		    wp->w_cursor.lnum -= count;
570261f346fSBram Moolenaar 		else if (wp->w_cursor.lnum> first)
571261f346fSBram Moolenaar 		    wp->w_cursor.lnum = first;
572261f346fSBram Moolenaar 		if (wp->w_cursor.lnum > wp->w_buffer->b_ml.ml_line_count)
573261f346fSBram Moolenaar 		    wp->w_cursor.lnum = wp->w_buffer->b_ml.ml_line_count;
574261f346fSBram Moolenaar 	    }
575261f346fSBram Moolenaar 	check_cursor_col();
576261f346fSBram Moolenaar 	deleted_lines_mark(first, count);
577963ffa0aSBram Moolenaar     }
578261f346fSBram Moolenaar 
579261f346fSBram Moolenaar     if (!is_curbuf)
580261f346fSBram Moolenaar     {
581261f346fSBram Moolenaar 	curbuf = curbuf_save;
582261f346fSBram Moolenaar 	curwin = curwin_save;
583261f346fSBram Moolenaar     }
584261f346fSBram Moolenaar }
585261f346fSBram Moolenaar 
586261f346fSBram Moolenaar /*
587261f346fSBram Moolenaar  * Returns buffer options, variables and other attributes in a dictionary.
588261f346fSBram Moolenaar  */
589261f346fSBram Moolenaar     static dict_T *
get_buffer_info(buf_T * buf)590261f346fSBram Moolenaar get_buffer_info(buf_T *buf)
591261f346fSBram Moolenaar {
592261f346fSBram Moolenaar     dict_T	*dict;
593261f346fSBram Moolenaar     tabpage_T	*tp;
594261f346fSBram Moolenaar     win_T	*wp;
595261f346fSBram Moolenaar     list_T	*windows;
596261f346fSBram Moolenaar 
597261f346fSBram Moolenaar     dict = dict_alloc();
598261f346fSBram Moolenaar     if (dict == NULL)
599261f346fSBram Moolenaar 	return NULL;
600261f346fSBram Moolenaar 
601261f346fSBram Moolenaar     dict_add_number(dict, "bufnr", buf->b_fnum);
602261f346fSBram Moolenaar     dict_add_string(dict, "name", buf->b_ffname);
603261f346fSBram Moolenaar     dict_add_number(dict, "lnum", buf == curbuf ? curwin->w_cursor.lnum
604261f346fSBram Moolenaar 						     : buflist_findlnum(buf));
605a9e9679dSBram Moolenaar     dict_add_number(dict, "linecount", buf->b_ml.ml_line_count);
606261f346fSBram Moolenaar     dict_add_number(dict, "loaded", buf->b_ml.ml_mfp != NULL);
607261f346fSBram Moolenaar     dict_add_number(dict, "listed", buf->b_p_bl);
608261f346fSBram Moolenaar     dict_add_number(dict, "changed", bufIsChanged(buf));
609261f346fSBram Moolenaar     dict_add_number(dict, "changedtick", CHANGEDTICK(buf));
610261f346fSBram Moolenaar     dict_add_number(dict, "hidden",
611261f346fSBram Moolenaar 			    buf->b_ml.ml_mfp != NULL && buf->b_nwindows == 0);
612261f346fSBram Moolenaar 
613261f346fSBram Moolenaar     // Get a reference to buffer variables
614261f346fSBram Moolenaar     dict_add_dict(dict, "variables", buf->b_vars);
615261f346fSBram Moolenaar 
616261f346fSBram Moolenaar     // List of windows displaying this buffer
617261f346fSBram Moolenaar     windows = list_alloc();
618261f346fSBram Moolenaar     if (windows != NULL)
619261f346fSBram Moolenaar     {
620261f346fSBram Moolenaar 	FOR_ALL_TAB_WINDOWS(tp, wp)
621261f346fSBram Moolenaar 	    if (wp->w_buffer == buf)
622261f346fSBram Moolenaar 		list_append_number(windows, (varnumber_T)wp->w_id);
623261f346fSBram Moolenaar 	dict_add_list(dict, "windows", windows);
624261f346fSBram Moolenaar     }
625261f346fSBram Moolenaar 
62605ad5ff0SBram Moolenaar #ifdef FEAT_PROP_POPUP
627261f346fSBram Moolenaar     // List of popup windows displaying this buffer
628261f346fSBram Moolenaar     windows = list_alloc();
629261f346fSBram Moolenaar     if (windows != NULL)
630261f346fSBram Moolenaar     {
631aeea7215SBram Moolenaar 	FOR_ALL_POPUPWINS(wp)
632261f346fSBram Moolenaar 	    if (wp->w_buffer == buf)
633261f346fSBram Moolenaar 		list_append_number(windows, (varnumber_T)wp->w_id);
634261f346fSBram Moolenaar 	FOR_ALL_TABPAGES(tp)
635aeea7215SBram Moolenaar 	    FOR_ALL_POPUPWINS_IN_TAB(tp, wp)
636261f346fSBram Moolenaar 		if (wp->w_buffer == buf)
637261f346fSBram Moolenaar 		    list_append_number(windows, (varnumber_T)wp->w_id);
638261f346fSBram Moolenaar 
639261f346fSBram Moolenaar 	dict_add_list(dict, "popups", windows);
640261f346fSBram Moolenaar     }
641261f346fSBram Moolenaar #endif
642261f346fSBram Moolenaar 
643261f346fSBram Moolenaar #ifdef FEAT_SIGNS
644261f346fSBram Moolenaar     if (buf->b_signlist != NULL)
645261f346fSBram Moolenaar     {
646261f346fSBram Moolenaar 	// List of signs placed in this buffer
647261f346fSBram Moolenaar 	list_T	*signs = list_alloc();
648261f346fSBram Moolenaar 	if (signs != NULL)
649261f346fSBram Moolenaar 	{
650261f346fSBram Moolenaar 	    get_buffer_signs(buf, signs);
651261f346fSBram Moolenaar 	    dict_add_list(dict, "signs", signs);
652261f346fSBram Moolenaar 	}
653261f346fSBram Moolenaar     }
654261f346fSBram Moolenaar #endif
655261f346fSBram Moolenaar 
65652410575SBram Moolenaar #ifdef FEAT_VIMINFO
65752410575SBram Moolenaar     dict_add_number(dict, "lastused", buf->b_last_used);
65852410575SBram Moolenaar #endif
65952410575SBram Moolenaar 
660261f346fSBram Moolenaar     return dict;
661261f346fSBram Moolenaar }
662261f346fSBram Moolenaar 
663261f346fSBram Moolenaar /*
664261f346fSBram Moolenaar  * "getbufinfo()" function
665261f346fSBram Moolenaar  */
666261f346fSBram Moolenaar     void
f_getbufinfo(typval_T * argvars,typval_T * rettv)667261f346fSBram Moolenaar f_getbufinfo(typval_T *argvars, typval_T *rettv)
668261f346fSBram Moolenaar {
669261f346fSBram Moolenaar     buf_T	*buf = NULL;
670261f346fSBram Moolenaar     buf_T	*argbuf = NULL;
671261f346fSBram Moolenaar     dict_T	*d;
672261f346fSBram Moolenaar     int		filtered = FALSE;
673261f346fSBram Moolenaar     int		sel_buflisted = FALSE;
674261f346fSBram Moolenaar     int		sel_bufloaded = FALSE;
675261f346fSBram Moolenaar     int		sel_bufmodified = FALSE;
676261f346fSBram Moolenaar 
677261f346fSBram Moolenaar     if (rettv_list_alloc(rettv) != OK)
678261f346fSBram Moolenaar 	return;
679261f346fSBram Moolenaar 
680cd917207SYegappan Lakshmanan     if (in_vim9script()
6814490ec4eSYegappan Lakshmanan 	    && check_for_opt_buffer_or_dict_arg(argvars, 0) == FAIL)
682cd917207SYegappan Lakshmanan 	return;
683cd917207SYegappan Lakshmanan 
684261f346fSBram Moolenaar     // List of all the buffers or selected buffers
685261f346fSBram Moolenaar     if (argvars[0].v_type == VAR_DICT)
686261f346fSBram Moolenaar     {
687261f346fSBram Moolenaar 	dict_T	*sel_d = argvars[0].vval.v_dict;
688261f346fSBram Moolenaar 
689261f346fSBram Moolenaar 	if (sel_d != NULL)
690261f346fSBram Moolenaar 	{
691261f346fSBram Moolenaar 	    filtered = TRUE;
692036c2cf7SBram Moolenaar 	    sel_buflisted = dict_get_bool(sel_d, (char_u *)"buflisted", FALSE);
693036c2cf7SBram Moolenaar 	    sel_bufloaded = dict_get_bool(sel_d, (char_u *)"bufloaded", FALSE);
694036c2cf7SBram Moolenaar 	    sel_bufmodified = dict_get_bool(sel_d, (char_u *)"bufmodified",
695036c2cf7SBram Moolenaar 									FALSE);
696261f346fSBram Moolenaar 	}
697261f346fSBram Moolenaar     }
698261f346fSBram Moolenaar     else if (argvars[0].v_type != VAR_UNKNOWN)
699261f346fSBram Moolenaar     {
700261f346fSBram Moolenaar 	// Information about one buffer.  Argument specifies the buffer
7013767e3a3SBram Moolenaar 	argbuf = tv_get_buf_from_arg(&argvars[0]);
702261f346fSBram Moolenaar 	if (argbuf == NULL)
703261f346fSBram Moolenaar 	    return;
704261f346fSBram Moolenaar     }
705261f346fSBram Moolenaar 
706261f346fSBram Moolenaar     // Return information about all the buffers or a specified buffer
707261f346fSBram Moolenaar     FOR_ALL_BUFFERS(buf)
708261f346fSBram Moolenaar     {
709261f346fSBram Moolenaar 	if (argbuf != NULL && argbuf != buf)
710261f346fSBram Moolenaar 	    continue;
711261f346fSBram Moolenaar 	if (filtered && ((sel_bufloaded && buf->b_ml.ml_mfp == NULL)
712261f346fSBram Moolenaar 			|| (sel_buflisted && !buf->b_p_bl)
713261f346fSBram Moolenaar 			|| (sel_bufmodified && !buf->b_changed)))
714261f346fSBram Moolenaar 	    continue;
715261f346fSBram Moolenaar 
716261f346fSBram Moolenaar 	d = get_buffer_info(buf);
717261f346fSBram Moolenaar 	if (d != NULL)
718261f346fSBram Moolenaar 	    list_append_dict(rettv->vval.v_list, d);
719261f346fSBram Moolenaar 	if (argbuf != NULL)
720261f346fSBram Moolenaar 	    return;
721261f346fSBram Moolenaar     }
722261f346fSBram Moolenaar }
723261f346fSBram Moolenaar 
724261f346fSBram Moolenaar /*
725261f346fSBram Moolenaar  * Get line or list of lines from buffer "buf" into "rettv".
726261f346fSBram Moolenaar  * Return a range (from start to end) of lines in rettv from the specified
727261f346fSBram Moolenaar  * buffer.
728261f346fSBram Moolenaar  * If 'retlist' is TRUE, then the lines are returned as a Vim List.
729261f346fSBram Moolenaar  */
730261f346fSBram Moolenaar     static void
get_buffer_lines(buf_T * buf,linenr_T start,linenr_T end,int retlist,typval_T * rettv)731261f346fSBram Moolenaar get_buffer_lines(
732261f346fSBram Moolenaar     buf_T	*buf,
733261f346fSBram Moolenaar     linenr_T	start,
734261f346fSBram Moolenaar     linenr_T	end,
735261f346fSBram Moolenaar     int		retlist,
736261f346fSBram Moolenaar     typval_T	*rettv)
737261f346fSBram Moolenaar {
738261f346fSBram Moolenaar     char_u	*p;
739261f346fSBram Moolenaar 
7408a7d6542SBram Moolenaar     if (retlist)
7418a7d6542SBram Moolenaar     {
7428a7d6542SBram Moolenaar 	if (rettv_list_alloc(rettv) == FAIL)
7438a7d6542SBram Moolenaar 	    return;
7448a7d6542SBram Moolenaar     }
7458a7d6542SBram Moolenaar     else
7468a7d6542SBram Moolenaar     {
747261f346fSBram Moolenaar 	rettv->v_type = VAR_STRING;
748261f346fSBram Moolenaar 	rettv->vval.v_string = NULL;
7498a7d6542SBram Moolenaar     }
750261f346fSBram Moolenaar 
751261f346fSBram Moolenaar     if (buf == NULL || buf->b_ml.ml_mfp == NULL || start < 0)
752261f346fSBram Moolenaar 	return;
753261f346fSBram Moolenaar 
754261f346fSBram Moolenaar     if (!retlist)
755261f346fSBram Moolenaar     {
756261f346fSBram Moolenaar 	if (start >= 1 && start <= buf->b_ml.ml_line_count)
757261f346fSBram Moolenaar 	    p = ml_get_buf(buf, start, FALSE);
758261f346fSBram Moolenaar 	else
759261f346fSBram Moolenaar 	    p = (char_u *)"";
760261f346fSBram Moolenaar 	rettv->vval.v_string = vim_strsave(p);
761261f346fSBram Moolenaar     }
762261f346fSBram Moolenaar     else
763261f346fSBram Moolenaar     {
764261f346fSBram Moolenaar 	if (end < start)
765261f346fSBram Moolenaar 	    return;
766261f346fSBram Moolenaar 
767261f346fSBram Moolenaar 	if (start < 1)
768261f346fSBram Moolenaar 	    start = 1;
769261f346fSBram Moolenaar 	if (end > buf->b_ml.ml_line_count)
770261f346fSBram Moolenaar 	    end = buf->b_ml.ml_line_count;
771261f346fSBram Moolenaar 	while (start <= end)
772261f346fSBram Moolenaar 	    if (list_append_string(rettv->vval.v_list,
773261f346fSBram Moolenaar 				 ml_get_buf(buf, start++, FALSE), -1) == FAIL)
774261f346fSBram Moolenaar 		break;
775261f346fSBram Moolenaar     }
776261f346fSBram Moolenaar }
777261f346fSBram Moolenaar 
778261f346fSBram Moolenaar /*
779261f346fSBram Moolenaar  * "getbufline()" function
780261f346fSBram Moolenaar  */
781261f346fSBram Moolenaar     void
f_getbufline(typval_T * argvars,typval_T * rettv)782261f346fSBram Moolenaar f_getbufline(typval_T *argvars, typval_T *rettv)
783261f346fSBram Moolenaar {
784e6e70a10SBram Moolenaar     linenr_T	lnum = 1;
785e6e70a10SBram Moolenaar     linenr_T	end = 1;
786261f346fSBram Moolenaar     buf_T	*buf;
787261f346fSBram Moolenaar 
78883494b4aSYegappan Lakshmanan     if (in_vim9script()
78983494b4aSYegappan Lakshmanan 	    && (check_for_buffer_arg(argvars, 0) == FAIL
79083494b4aSYegappan Lakshmanan 		|| check_for_lnum_arg(argvars, 1) == FAIL
79183494b4aSYegappan Lakshmanan 		|| check_for_opt_lnum_arg(argvars, 2) == FAIL))
79283494b4aSYegappan Lakshmanan 	return;
79383494b4aSYegappan Lakshmanan 
7943767e3a3SBram Moolenaar     buf = tv_get_buf_from_arg(&argvars[0]);
795e6e70a10SBram Moolenaar     if (buf != NULL)
796e6e70a10SBram Moolenaar     {
797261f346fSBram Moolenaar 	lnum = tv_get_lnum_buf(&argvars[1], buf);
798261f346fSBram Moolenaar 	if (argvars[2].v_type == VAR_UNKNOWN)
799261f346fSBram Moolenaar 	    end = lnum;
800261f346fSBram Moolenaar 	else
801261f346fSBram Moolenaar 	    end = tv_get_lnum_buf(&argvars[2], buf);
802e6e70a10SBram Moolenaar     }
803261f346fSBram Moolenaar 
804261f346fSBram Moolenaar     get_buffer_lines(buf, lnum, end, TRUE, rettv);
805261f346fSBram Moolenaar }
806261f346fSBram Moolenaar 
807fbdd08edSBram Moolenaar     type_T *
ret_f_getline(int argcount,type_T ** argtypes UNUSED)808fbdd08edSBram Moolenaar ret_f_getline(int argcount, type_T **argtypes UNUSED)
809fbdd08edSBram Moolenaar {
810fbdd08edSBram Moolenaar     return argcount == 1 ? &t_string : &t_list_string;
811fbdd08edSBram Moolenaar }
812fbdd08edSBram Moolenaar 
813261f346fSBram Moolenaar /*
814261f346fSBram Moolenaar  * "getline(lnum, [end])" function
815261f346fSBram Moolenaar  */
816261f346fSBram Moolenaar     void
f_getline(typval_T * argvars,typval_T * rettv)817261f346fSBram Moolenaar f_getline(typval_T *argvars, typval_T *rettv)
818261f346fSBram Moolenaar {
819261f346fSBram Moolenaar     linenr_T	lnum;
820261f346fSBram Moolenaar     linenr_T	end;
821261f346fSBram Moolenaar     int		retlist;
822261f346fSBram Moolenaar 
823cd917207SYegappan Lakshmanan     if (in_vim9script()
824cd917207SYegappan Lakshmanan 	    && (check_for_lnum_arg(argvars, 0) == FAIL
825cd917207SYegappan Lakshmanan 		|| check_for_opt_lnum_arg(argvars, 1) == FAIL))
826cd917207SYegappan Lakshmanan 	return;
827cd917207SYegappan Lakshmanan 
828261f346fSBram Moolenaar     lnum = tv_get_lnum(argvars);
829261f346fSBram Moolenaar     if (argvars[1].v_type == VAR_UNKNOWN)
830261f346fSBram Moolenaar     {
831261f346fSBram Moolenaar 	end = 0;
832261f346fSBram Moolenaar 	retlist = FALSE;
833261f346fSBram Moolenaar     }
834261f346fSBram Moolenaar     else
835261f346fSBram Moolenaar     {
836261f346fSBram Moolenaar 	end = tv_get_lnum(&argvars[1]);
837261f346fSBram Moolenaar 	retlist = TRUE;
838261f346fSBram Moolenaar     }
839261f346fSBram Moolenaar 
840261f346fSBram Moolenaar     get_buffer_lines(curbuf, lnum, end, retlist, rettv);
841261f346fSBram Moolenaar }
842261f346fSBram Moolenaar 
843261f346fSBram Moolenaar /*
844261f346fSBram Moolenaar  * "setbufline()" function
845261f346fSBram Moolenaar  */
846261f346fSBram Moolenaar     void
f_setbufline(typval_T * argvars,typval_T * rettv)847261f346fSBram Moolenaar f_setbufline(typval_T *argvars, typval_T *rettv)
848261f346fSBram Moolenaar {
849*4a15504eSYegappan Lakshmanan     buf_set_append_line(argvars, rettv, FALSE);
850261f346fSBram Moolenaar }
851261f346fSBram Moolenaar 
852261f346fSBram Moolenaar /*
853261f346fSBram Moolenaar  * "setline()" function
854261f346fSBram Moolenaar  */
855261f346fSBram Moolenaar     void
f_setline(typval_T * argvars,typval_T * rettv)856261f346fSBram Moolenaar f_setline(typval_T *argvars, typval_T *rettv)
857261f346fSBram Moolenaar {
8584490ec4eSYegappan Lakshmanan     linenr_T	lnum;
859261f346fSBram Moolenaar 
8604490ec4eSYegappan Lakshmanan     if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
8614490ec4eSYegappan Lakshmanan 	return;
8624490ec4eSYegappan Lakshmanan 
8634490ec4eSYegappan Lakshmanan     lnum = tv_get_lnum(&argvars[0]);
864261f346fSBram Moolenaar     set_buffer_lines(curbuf, lnum, FALSE, &argvars[1], rettv);
865261f346fSBram Moolenaar }
866261f346fSBram Moolenaar #endif  // FEAT_EVAL
867261f346fSBram Moolenaar 
868261f346fSBram Moolenaar #if defined(FEAT_JOB_CHANNEL) \
869261f346fSBram Moolenaar 	|| defined(FEAT_PYTHON) || defined(FEAT_PYTHON3) \
870261f346fSBram Moolenaar 	|| defined(PROTO)
871261f346fSBram Moolenaar /*
872261f346fSBram Moolenaar  * Make "buf" the current buffer.  restore_buffer() MUST be called to undo.
873261f346fSBram Moolenaar  * No autocommands will be executed.  Use aucmd_prepbuf() if there are any.
874261f346fSBram Moolenaar  */
875261f346fSBram Moolenaar     void
switch_buffer(bufref_T * save_curbuf,buf_T * buf)876261f346fSBram Moolenaar switch_buffer(bufref_T *save_curbuf, buf_T *buf)
877261f346fSBram Moolenaar {
878261f346fSBram Moolenaar     block_autocmds();
8793e0107eaSBram Moolenaar #ifdef FEAT_FOLDING
8803e0107eaSBram Moolenaar     ++disable_fold_update;
8813e0107eaSBram Moolenaar #endif
882261f346fSBram Moolenaar     set_bufref(save_curbuf, curbuf);
883261f346fSBram Moolenaar     --curbuf->b_nwindows;
884261f346fSBram Moolenaar     curbuf = buf;
885261f346fSBram Moolenaar     curwin->w_buffer = buf;
886261f346fSBram Moolenaar     ++curbuf->b_nwindows;
887261f346fSBram Moolenaar }
888261f346fSBram Moolenaar 
889261f346fSBram Moolenaar /*
890261f346fSBram Moolenaar  * Restore the current buffer after using switch_buffer().
891261f346fSBram Moolenaar  */
892261f346fSBram Moolenaar     void
restore_buffer(bufref_T * save_curbuf)893261f346fSBram Moolenaar restore_buffer(bufref_T *save_curbuf)
894261f346fSBram Moolenaar {
895261f346fSBram Moolenaar     unblock_autocmds();
8963e0107eaSBram Moolenaar #ifdef FEAT_FOLDING
8973e0107eaSBram Moolenaar     --disable_fold_update;
8983e0107eaSBram Moolenaar #endif
8995d18efecSBram Moolenaar     // Check for valid buffer, just in case.
900261f346fSBram Moolenaar     if (bufref_valid(save_curbuf))
901261f346fSBram Moolenaar     {
902261f346fSBram Moolenaar 	--curbuf->b_nwindows;
903261f346fSBram Moolenaar 	curwin->w_buffer = save_curbuf->br_buf;
904261f346fSBram Moolenaar 	curbuf = save_curbuf->br_buf;
905261f346fSBram Moolenaar 	++curbuf->b_nwindows;
906261f346fSBram Moolenaar     }
907261f346fSBram Moolenaar }
908261f346fSBram Moolenaar 
909261f346fSBram Moolenaar /*
910261f346fSBram Moolenaar  * Find a window for buffer "buf".
911261f346fSBram Moolenaar  * If found OK is returned and "wp" and "tp" are set to the window and tabpage.
912261f346fSBram Moolenaar  * If not found FAIL is returned.
913261f346fSBram Moolenaar  */
914261f346fSBram Moolenaar     static int
find_win_for_buf(buf_T * buf,win_T ** wp,tabpage_T ** tp)915261f346fSBram Moolenaar find_win_for_buf(
916261f346fSBram Moolenaar     buf_T     *buf,
917261f346fSBram Moolenaar     win_T     **wp,
918261f346fSBram Moolenaar     tabpage_T **tp)
919261f346fSBram Moolenaar {
920261f346fSBram Moolenaar     FOR_ALL_TAB_WINDOWS(*tp, *wp)
921261f346fSBram Moolenaar 	if ((*wp)->w_buffer == buf)
922261f346fSBram Moolenaar 	    return OK;
923261f346fSBram Moolenaar     return FAIL;
924261f346fSBram Moolenaar }
925261f346fSBram Moolenaar 
926261f346fSBram Moolenaar /*
927261f346fSBram Moolenaar  * Find a window that contains "buf" and switch to it.
928261f346fSBram Moolenaar  * If there is no such window, use the current window and change "curbuf".
929261f346fSBram Moolenaar  * Caller must initialize save_curbuf to NULL.
930261f346fSBram Moolenaar  * restore_win_for_buf() MUST be called later!
931261f346fSBram Moolenaar  */
932261f346fSBram Moolenaar     void
switch_to_win_for_buf(buf_T * buf,win_T ** save_curwinp,tabpage_T ** save_curtabp,bufref_T * save_curbuf)933261f346fSBram Moolenaar switch_to_win_for_buf(
934261f346fSBram Moolenaar     buf_T	*buf,
935261f346fSBram Moolenaar     win_T	**save_curwinp,
936261f346fSBram Moolenaar     tabpage_T	**save_curtabp,
937261f346fSBram Moolenaar     bufref_T	*save_curbuf)
938261f346fSBram Moolenaar {
939261f346fSBram Moolenaar     win_T	*wp;
940261f346fSBram Moolenaar     tabpage_T	*tp;
941261f346fSBram Moolenaar 
942261f346fSBram Moolenaar     if (find_win_for_buf(buf, &wp, &tp) == FAIL)
943261f346fSBram Moolenaar 	switch_buffer(save_curbuf, buf);
944261f346fSBram Moolenaar     else if (switch_win(save_curwinp, save_curtabp, wp, tp, TRUE) == FAIL)
945261f346fSBram Moolenaar     {
946261f346fSBram Moolenaar 	restore_win(*save_curwinp, *save_curtabp, TRUE);
947261f346fSBram Moolenaar 	switch_buffer(save_curbuf, buf);
948261f346fSBram Moolenaar     }
949261f346fSBram Moolenaar }
950261f346fSBram Moolenaar 
951261f346fSBram Moolenaar     void
restore_win_for_buf(win_T * save_curwin,tabpage_T * save_curtab,bufref_T * save_curbuf)952261f346fSBram Moolenaar restore_win_for_buf(
953261f346fSBram Moolenaar     win_T	*save_curwin,
954261f346fSBram Moolenaar     tabpage_T	*save_curtab,
955261f346fSBram Moolenaar     bufref_T	*save_curbuf)
956261f346fSBram Moolenaar {
957261f346fSBram Moolenaar     if (save_curbuf->br_buf == NULL)
958261f346fSBram Moolenaar 	restore_win(save_curwin, save_curtab, TRUE);
959261f346fSBram Moolenaar     else
960261f346fSBram Moolenaar 	restore_buffer(save_curbuf);
961261f346fSBram Moolenaar }
962261f346fSBram Moolenaar #endif
963