blob: 9b3b5beb664fb9018db36563ed67abb2885c99d8 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002 *
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 * evalfunc.c: Builtin functions
12 */
13#define USING_FLOAT_STUFF
14
15#include "vim.h"
16
17#if defined(FEAT_EVAL) || defined(PROTO)
18
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020019#ifdef VMS
20# include <float.h>
21#endif
22
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020023#ifdef FEAT_FLOAT
24static void f_abs(typval_T *argvars, typval_T *rettv);
25static void f_acos(typval_T *argvars, typval_T *rettv);
26#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020027static void f_and(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020028#ifdef FEAT_FLOAT
29static void f_asin(typval_T *argvars, typval_T *rettv);
30static void f_atan(typval_T *argvars, typval_T *rettv);
31static void f_atan2(typval_T *argvars, typval_T *rettv);
32#endif
Bram Moolenaar59716a22017-03-01 20:32:44 +010033#ifdef FEAT_BEVAL
Bram Moolenaarbe0a2592019-05-09 13:50:16 +020034static void f_balloon_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar59716a22017-03-01 20:32:44 +010035static void f_balloon_show(typval_T *argvars, typval_T *rettv);
Bram Moolenaar669a8282017-11-19 20:13:05 +010036# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +010037static void f_balloon_split(typval_T *argvars, typval_T *rettv);
Bram Moolenaar669a8282017-11-19 20:13:05 +010038# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +010039#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020040static void f_byte2line(typval_T *argvars, typval_T *rettv);
41static void byteidx(typval_T *argvars, typval_T *rettv, int comp);
42static void f_byteidx(typval_T *argvars, typval_T *rettv);
43static void f_byteidxcomp(typval_T *argvars, typval_T *rettv);
44static void f_call(typval_T *argvars, typval_T *rettv);
45#ifdef FEAT_FLOAT
46static void f_ceil(typval_T *argvars, typval_T *rettv);
47#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020048static void f_changenr(typval_T *argvars, typval_T *rettv);
49static void f_char2nr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar17793ef2020-12-28 12:56:58 +010050static void f_charidx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020051static void f_col(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020052static void f_confirm(typval_T *argvars, typval_T *rettv);
53static void f_copy(typval_T *argvars, typval_T *rettv);
54#ifdef FEAT_FLOAT
55static void f_cos(typval_T *argvars, typval_T *rettv);
56static void f_cosh(typval_T *argvars, typval_T *rettv);
57#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020058static void f_cursor(typval_T *argsvars, typval_T *rettv);
Bram Moolenaar4f974752019-02-17 17:44:42 +010059#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +020060static void f_debugbreak(typval_T *argvars, typval_T *rettv);
61#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020062static void f_deepcopy(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020063static void f_did_filetype(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4132eb52020-02-14 16:53:00 +010064static void f_echoraw(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020065static void f_empty(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020066static void f_environ(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020067static void f_escape(typval_T *argvars, typval_T *rettv);
68static void f_eval(typval_T *argvars, typval_T *rettv);
69static void f_eventhandler(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020070static void f_execute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020071static void f_exists(typval_T *argvars, typval_T *rettv);
72#ifdef FEAT_FLOAT
73static void f_exp(typval_T *argvars, typval_T *rettv);
74#endif
75static void f_expand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar80dad482019-06-09 17:22:31 +020076static void f_expandcmd(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020077static void f_feedkeys(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020078#ifdef FEAT_FLOAT
79static void f_float2nr(typval_T *argvars, typval_T *rettv);
80static void f_floor(typval_T *argvars, typval_T *rettv);
81static void f_fmod(typval_T *argvars, typval_T *rettv);
82#endif
83static void f_fnameescape(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020084static void f_foreground(typval_T *argvars, typval_T *rettv);
Bram Moolenaar437bafe2016-08-01 15:40:54 +020085static void f_funcref(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020086static void f_function(typval_T *argvars, typval_T *rettv);
87static void f_garbagecollect(typval_T *argvars, typval_T *rettv);
88static void f_get(typval_T *argvars, typval_T *rettv);
Bram Moolenaar07ad8162018-02-13 13:59:59 +010089static void f_getchangelist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020090static void f_getcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020091static void f_getenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020092static void f_getfontname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4f505882018-02-10 21:06:32 +010093static void f_getjumplist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020094static void f_getpid(typval_T *argvars, typval_T *rettv);
95static void f_getcurpos(typval_T *argvars, typval_T *rettv);
96static void f_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020097static void f_getreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarbb861e22020-06-07 18:16:36 +020098static void f_getreginfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020099static void f_getregtype(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100100static void f_gettagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +0200101static void f_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200102static void f_haslocaldir(typval_T *argvars, typval_T *rettv);
103static void f_hasmapto(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200104static void f_hlID(typval_T *argvars, typval_T *rettv);
105static void f_hlexists(typval_T *argvars, typval_T *rettv);
106static void f_hostname(typval_T *argvars, typval_T *rettv);
107static void f_iconv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200108static void f_index(typval_T *argvars, typval_T *rettv);
109static void f_input(typval_T *argvars, typval_T *rettv);
110static void f_inputdialog(typval_T *argvars, typval_T *rettv);
111static void f_inputlist(typval_T *argvars, typval_T *rettv);
112static void f_inputrestore(typval_T *argvars, typval_T *rettv);
113static void f_inputsave(typval_T *argvars, typval_T *rettv);
114static void f_inputsecret(typval_T *argvars, typval_T *rettv);
Bram Moolenaar67a2deb2019-11-25 00:05:32 +0100115static void f_interrupt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200116static void f_invert(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200117static void f_islocked(typval_T *argvars, typval_T *rettv);
118#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
Bram Moolenaarfda1bff2019-04-04 13:44:37 +0200119static void f_isinf(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200120static void f_isnan(typval_T *argvars, typval_T *rettv);
121#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200122static void f_last_buffer_nr(typval_T *argvars, typval_T *rettv);
123static void f_len(typval_T *argvars, typval_T *rettv);
124static void f_libcall(typval_T *argvars, typval_T *rettv);
125static void f_libcallnr(typval_T *argvars, typval_T *rettv);
126static void f_line(typval_T *argvars, typval_T *rettv);
127static void f_line2byte(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200128#ifdef FEAT_FLOAT
129static void f_log(typval_T *argvars, typval_T *rettv);
130static void f_log10(typval_T *argvars, typval_T *rettv);
131#endif
132#ifdef FEAT_LUA
133static void f_luaeval(typval_T *argvars, typval_T *rettv);
134#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200135static void f_maparg(typval_T *argvars, typval_T *rettv);
136static void f_mapcheck(typval_T *argvars, typval_T *rettv);
137static void f_match(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200138static void f_matchend(typval_T *argvars, typval_T *rettv);
139static void f_matchlist(typval_T *argvars, typval_T *rettv);
140static void f_matchstr(typval_T *argvars, typval_T *rettv);
141static void f_matchstrpos(typval_T *argvars, typval_T *rettv);
142static void f_max(typval_T *argvars, typval_T *rettv);
143static void f_min(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200144#ifdef FEAT_MZSCHEME
145static void f_mzeval(typval_T *argvars, typval_T *rettv);
146#endif
147static void f_nextnonblank(typval_T *argvars, typval_T *rettv);
148static void f_nr2char(typval_T *argvars, typval_T *rettv);
149static void f_or(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200150#ifdef FEAT_PERL
151static void f_perleval(typval_T *argvars, typval_T *rettv);
152#endif
153#ifdef FEAT_FLOAT
154static void f_pow(typval_T *argvars, typval_T *rettv);
155#endif
156static void f_prevnonblank(typval_T *argvars, typval_T *rettv);
157static void f_printf(typval_T *argvars, typval_T *rettv);
Bram Moolenaare9bd5722019-08-17 19:36:06 +0200158static void f_pum_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200159static void f_pumvisible(typval_T *argvars, typval_T *rettv);
160#ifdef FEAT_PYTHON3
161static void f_py3eval(typval_T *argvars, typval_T *rettv);
162#endif
163#ifdef FEAT_PYTHON
164static void f_pyeval(typval_T *argvars, typval_T *rettv);
165#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100166#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
167static void f_pyxeval(typval_T *argvars, typval_T *rettv);
168#endif
Bram Moolenaar4f645c52020-02-08 16:40:39 +0100169static void f_test_srand_seed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100170static void f_rand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200171static void f_range(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b6d9112018-05-22 20:35:17 +0200172static void f_reg_executing(typval_T *argvars, typval_T *rettv);
173static void f_reg_recording(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200174static void f_rename(typval_T *argvars, typval_T *rettv);
175static void f_repeat(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200176#ifdef FEAT_FLOAT
177static void f_round(typval_T *argvars, typval_T *rettv);
178#endif
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100179#ifdef FEAT_RUBY
180static void f_rubyeval(typval_T *argvars, typval_T *rettv);
181#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200182static void f_screenattr(typval_T *argvars, typval_T *rettv);
183static void f_screenchar(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100184static void f_screenchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200185static void f_screencol(typval_T *argvars, typval_T *rettv);
186static void f_screenrow(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100187static void f_screenstring(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200188static void f_search(typval_T *argvars, typval_T *rettv);
189static void f_searchdecl(typval_T *argvars, typval_T *rettv);
190static void f_searchpair(typval_T *argvars, typval_T *rettv);
191static void f_searchpairpos(typval_T *argvars, typval_T *rettv);
192static void f_searchpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200193static void f_setcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +0200194static void f_setenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200195static void f_setfperm(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200196static void f_setpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200197static void f_setreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100198static void f_settagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200199#ifdef FEAT_CRYPT
200static void f_sha256(typval_T *argvars, typval_T *rettv);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200201#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200202static void f_shellescape(typval_T *argvars, typval_T *rettv);
203static void f_shiftwidth(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200204#ifdef FEAT_FLOAT
205static void f_sin(typval_T *argvars, typval_T *rettv);
206static void f_sinh(typval_T *argvars, typval_T *rettv);
207#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200208static void f_soundfold(typval_T *argvars, typval_T *rettv);
209static void f_spellbadword(typval_T *argvars, typval_T *rettv);
210static void f_spellsuggest(typval_T *argvars, typval_T *rettv);
211static void f_split(typval_T *argvars, typval_T *rettv);
212#ifdef FEAT_FLOAT
213static void f_sqrt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100214#endif
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100215static void f_srand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100216#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200217static void f_str2float(typval_T *argvars, typval_T *rettv);
218#endif
Bram Moolenaar9d401282019-04-06 13:18:12 +0200219static void f_str2list(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200220static void f_str2nr(typval_T *argvars, typval_T *rettv);
221static void f_strchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200222static void f_strgetchar(typval_T *argvars, typval_T *rettv);
223static void f_stridx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200224static void f_strlen(typval_T *argvars, typval_T *rettv);
225static void f_strcharpart(typval_T *argvars, typval_T *rettv);
226static void f_strpart(typval_T *argvars, typval_T *rettv);
227static void f_strridx(typval_T *argvars, typval_T *rettv);
228static void f_strtrans(typval_T *argvars, typval_T *rettv);
229static void f_strdisplaywidth(typval_T *argvars, typval_T *rettv);
230static void f_strwidth(typval_T *argvars, typval_T *rettv);
231static void f_submatch(typval_T *argvars, typval_T *rettv);
232static void f_substitute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar00f123a2018-08-21 20:28:54 +0200233static void f_swapinfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar110bd602018-09-16 18:46:59 +0200234static void f_swapname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200235static void f_synID(typval_T *argvars, typval_T *rettv);
236static void f_synIDattr(typval_T *argvars, typval_T *rettv);
237static void f_synIDtrans(typval_T *argvars, typval_T *rettv);
238static void f_synstack(typval_T *argvars, typval_T *rettv);
239static void f_synconcealed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200240static void f_tabpagebuflist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200241static void f_taglist(typval_T *argvars, typval_T *rettv);
242static void f_tagfiles(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200243#ifdef FEAT_FLOAT
244static void f_tan(typval_T *argvars, typval_T *rettv);
245static void f_tanh(typval_T *argvars, typval_T *rettv);
246#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200247static void f_tolower(typval_T *argvars, typval_T *rettv);
248static void f_toupper(typval_T *argvars, typval_T *rettv);
249static void f_tr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +0100250static void f_trim(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200251#ifdef FEAT_FLOAT
252static void f_trunc(typval_T *argvars, typval_T *rettv);
253#endif
254static void f_type(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200255static void f_virtcol(typval_T *argvars, typval_T *rettv);
256static void f_visualmode(typval_T *argvars, typval_T *rettv);
257static void f_wildmenumode(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0c1e3742019-12-27 13:49:24 +0100258static void f_windowsversion(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200259static void f_wordcount(typval_T *argvars, typval_T *rettv);
260static void f_xor(typval_T *argvars, typval_T *rettv);
261
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100262
Bram Moolenaar94738d82020-10-21 14:25:07 +0200263/*
264 * Functions that check the argument type of a builtin function.
265 * Each function returns FAIL and gives an error message if the type is wrong.
266 */
267
268// Context passed to an arg_ function.
269typedef struct {
Bram Moolenaarca174532020-10-21 16:42:22 +0200270 int arg_count; // actual argument count
271 type_T **arg_types; // list of argument types
272 int arg_idx; // current argument index (first arg is zero)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200273} argcontext_T;
274
275// A function to check one argument type. The first argument is the type to
276// check. If needed, other argument types can be obtained with the context.
277// E.g. if "arg_idx" is 1, then (type - 1) is the first argument type.
278typedef int (*argcheck_T)(type_T *, argcontext_T *);
279
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100280/*
281 * Check "type" is a float or a number.
282 */
Bram Moolenaar94738d82020-10-21 14:25:07 +0200283 static int
284arg_float_or_nr(type_T *type, argcontext_T *context)
285{
Bram Moolenaarca174532020-10-21 16:42:22 +0200286 if (type->tt_type == VAR_ANY
287 || type->tt_type == VAR_FLOAT || type->tt_type == VAR_NUMBER)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200288 return OK;
289 arg_type_mismatch(&t_number, type, context->arg_idx + 1);
290 return FAIL;
291}
292
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100293/*
294 * Check "type" is a number.
295 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200296 static int
297arg_number(type_T *type, argcontext_T *context)
298{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100299 return check_arg_type(&t_number, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200300}
301
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100302/*
303 * Check "type" is a string.
304 */
305 static int
306arg_string(type_T *type, argcontext_T *context)
307{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100308 return check_arg_type(&t_string, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100309}
310
311/*
312 * Check "type" is a list or a blob.
313 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200314 static int
315arg_list_or_blob(type_T *type, argcontext_T *context)
316{
317 if (type->tt_type == VAR_ANY
318 || type->tt_type == VAR_LIST || type->tt_type == VAR_BLOB)
319 return OK;
320 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
321 return FAIL;
322}
323
324/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100325 * Check "type" is a list or a dict.
326 */
327 static int
328arg_list_or_dict(type_T *type, argcontext_T *context)
329{
330 if (type->tt_type == VAR_ANY
331 || type->tt_type == VAR_LIST || type->tt_type == VAR_DICT)
332 return OK;
333 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
334 return FAIL;
335}
336
337/*
338 * Check "type" is the same type as the previous argument
339 * Must not be used for the first argcheck_T entry.
340 */
341 static int
342arg_same_as_prev(type_T *type, argcontext_T *context)
343{
344 type_T *prev_type = context->arg_types[context->arg_idx - 1];
345
Bram Moolenaar193f6202020-11-16 20:08:35 +0100346 return check_arg_type(prev_type, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100347}
348
349/*
350 * Check "type" is an item of the list or blob of the previous arg.
Bram Moolenaarca174532020-10-21 16:42:22 +0200351 * Must not be used for the first argcheck_T entry.
352 */
353 static int
354arg_item_of_prev(type_T *type, argcontext_T *context)
355{
356 type_T *prev_type = context->arg_types[context->arg_idx - 1];
357 type_T *expected;
358
359 if (prev_type->tt_type == VAR_LIST)
360 expected = prev_type->tt_member;
361 else if (prev_type->tt_type == VAR_BLOB)
362 expected = &t_number;
363 else
364 // probably VAR_ANY, can't check
365 return OK;
366
Bram Moolenaar193f6202020-11-16 20:08:35 +0100367 return check_arg_type(expected, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200368}
369
Bram Moolenaar94738d82020-10-21 14:25:07 +0200370/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100371 * Check "type" which is the third argument of extend().
372 */
373 static int
374arg_extend3(type_T *type, argcontext_T *context)
375{
376 type_T *first_type = context->arg_types[context->arg_idx - 2];
377
378 if (first_type->tt_type == VAR_LIST)
379 return arg_number(type, context);
380 if (first_type->tt_type == VAR_DICT)
381 return arg_string(type, context);
382 return OK;
383}
384
385
386/*
Bram Moolenaar94738d82020-10-21 14:25:07 +0200387 * Lists of functions that check the argument types of a builtin function.
388 */
389argcheck_T arg1_float_or_nr[] = {arg_float_or_nr};
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100390argcheck_T arg2_listblob_item[] = {arg_list_or_blob, arg_item_of_prev};
391argcheck_T arg23_extend[] = {arg_list_or_dict, arg_same_as_prev, arg_extend3};
Bram Moolenaarca174532020-10-21 16:42:22 +0200392argcheck_T arg3_insert[] = {arg_list_or_blob, arg_item_of_prev, arg_number};
Bram Moolenaar94738d82020-10-21 14:25:07 +0200393
394/*
395 * Functions that return the return type of a builtin function.
Bram Moolenaara1224cb2020-10-22 12:31:49 +0200396 * Note that "argtypes" is NULL if "argcount" is zero.
Bram Moolenaar94738d82020-10-21 14:25:07 +0200397 */
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100398 static type_T *
399ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
400{
401 return &t_void;
402}
403 static type_T *
404ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
405{
406 return &t_any;
407}
408 static type_T *
Bram Moolenaar403dc312020-10-17 19:29:51 +0200409ret_bool(int argcount UNUSED, type_T **argtypes UNUSED)
410{
411 return &t_bool;
412}
413 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100414ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
415{
416 return &t_number;
417}
418 static type_T *
419ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
420{
421 return &t_float;
422}
423 static type_T *
424ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
425{
426 return &t_string;
427}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200428 static type_T *
429ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100430{
431 return &t_list_any;
432}
433 static type_T *
434ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
435{
436 return &t_list_number;
437}
438 static type_T *
439ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
440{
441 return &t_list_string;
442}
443 static type_T *
444ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
445{
446 return &t_list_dict_any;
447}
448 static type_T *
449ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
450{
451 return &t_dict_any;
452}
453 static type_T *
454ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
455{
456 return &t_dict_number;
457}
458 static type_T *
459ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
460{
461 return &t_dict_string;
462}
463 static type_T *
464ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
465{
466 return &t_blob;
467}
468 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200469ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100470{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200471 return &t_func_any;
472}
473 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100474ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
475{
476 return &t_channel;
477}
478 static type_T *
479ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
480{
481 return &t_job;
482}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200483 static type_T *
484ret_first_arg(int argcount, type_T **argtypes)
485{
486 if (argcount > 0)
487 return argtypes[0];
488 return &t_void;
489}
Bram Moolenaarea696852020-11-09 18:31:39 +0100490// for map(): returns first argument but item type may differ
491 static type_T *
492ret_first_cont(int argcount UNUSED, type_T **argtypes)
493{
494 if (argtypes[0]->tt_type == VAR_LIST)
495 return &t_list_any;
496 if (argtypes[0]->tt_type == VAR_DICT)
497 return &t_dict_any;
498 if (argtypes[0]->tt_type == VAR_BLOB)
499 return argtypes[0];
500 return &t_any;
501}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200502
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200503/*
504 * Used for getqflist(): returns list if there is no argument, dict if there is
505 * one.
506 */
507 static type_T *
508ret_list_or_dict_0(int argcount, type_T **argtypes UNUSED)
509{
510 if (argcount > 0)
511 return &t_dict_any;
512 return &t_list_dict_any;
513}
514
515/*
516 * Used for getloclist(): returns list if there is one argument, dict if there
517 * are two.
518 */
519 static type_T *
520ret_list_or_dict_1(int argcount, type_T **argtypes UNUSED)
521{
522 if (argcount > 1)
523 return &t_dict_any;
524 return &t_list_dict_any;
525}
526
Bram Moolenaar846178a2020-07-05 17:04:13 +0200527 static type_T *
528ret_argv(int argcount, type_T **argtypes UNUSED)
529{
530 // argv() returns list of strings
531 if (argcount == 0)
532 return &t_list_string;
533
534 // argv(0) returns a string, but argv(-1] returns a list
535 return &t_any;
536}
537
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200538 static type_T *
539ret_remove(int argcount UNUSED, type_T **argtypes)
540{
Bram Moolenaar5e654232020-09-16 15:22:00 +0200541 if (argtypes != NULL)
542 {
543 if (argtypes[0]->tt_type == VAR_LIST
544 || argtypes[0]->tt_type == VAR_DICT)
545 return argtypes[0]->tt_member;
546 if (argtypes[0]->tt_type == VAR_BLOB)
547 return &t_number;
548 }
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200549 return &t_any;
550}
551
Bram Moolenaar3d945cc2020-08-06 21:26:59 +0200552 static type_T *
553ret_getreg(int argcount, type_T **argtypes UNUSED)
554{
555 // Assume that if the third argument is passed it's non-zero
556 if (argcount == 3)
557 return &t_list_string;
558 return &t_string;
559}
560
Bram Moolenaar4a6d1b62020-08-08 17:55:49 +0200561 static type_T *
562ret_maparg(int argcount, type_T **argtypes UNUSED)
563{
564 // Assume that if the fourth argument is passed it's non-zero
565 if (argcount == 4)
566 return &t_dict_any;
567 return &t_string;
568}
569
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100570static type_T *ret_f_function(int argcount, type_T **argtypes);
571
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200572/*
573 * Array with names and number of arguments of all internal functions
574 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
575 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200576typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200577{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200578 char *f_name; // function name
579 char f_min_argc; // minimal number of arguments
580 char f_max_argc; // maximal number of arguments
581 char f_argtype; // for method: FEARG_ values
Bram Moolenaar94738d82020-10-21 14:25:07 +0200582 argcheck_T *f_argcheck; // list of functions to check argument types
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100583 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
584 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200585 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200586 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200587} funcentry_T;
588
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200589// values for f_argtype; zero means it cannot be used as a method
590#define FEARG_1 1 // base is the first argument
591#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200592#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200593#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200594#define FEARG_LAST 9 // base is the last argument
595
Bram Moolenaar15c47602020-03-26 22:16:48 +0100596#ifdef FEAT_FLOAT
597# define FLOAT_FUNC(name) name
598#else
599# define FLOAT_FUNC(name) NULL
600#endif
601#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
602# define MATH_FUNC(name) name
603#else
604# define MATH_FUNC(name) NULL
605#endif
606#ifdef FEAT_TIMERS
607# define TIMER_FUNC(name) name
608#else
609# define TIMER_FUNC(name) NULL
610#endif
611#ifdef FEAT_JOB_CHANNEL
612# define JOB_FUNC(name) name
613#else
614# define JOB_FUNC(name) NULL
615#endif
616#ifdef FEAT_PROP_POPUP
617# define PROP_FUNC(name) name
618#else
619# define PROP_FUNC(name) NULL
620#endif
621#ifdef FEAT_SIGNS
622# define SIGN_FUNC(name) name
623#else
624# define SIGN_FUNC(name) NULL
625#endif
626#ifdef FEAT_SOUND
627# define SOUND_FUNC(name) name
628#else
629# define SOUND_FUNC(name) NULL
630#endif
631#ifdef FEAT_TERMINAL
632# define TERM_FUNC(name) name
633#else
634# define TERM_FUNC(name) NULL
635#endif
636
Bram Moolenaarac92e252019-08-03 21:58:38 +0200637static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200638{
Bram Moolenaar94738d82020-10-21 14:25:07 +0200639 {"abs", 1, 1, FEARG_1, arg1_float_or_nr,
640 ret_any, FLOAT_FUNC(f_abs)},
641 {"acos", 1, 1, FEARG_1, NULL,
642 ret_float, FLOAT_FUNC(f_acos)},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100643 {"add", 2, 2, FEARG_1, NULL /* arg2_listblob_item */,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200644 ret_first_arg, f_add},
645 {"and", 2, 2, FEARG_1, NULL,
646 ret_number, f_and},
647 {"append", 2, 2, FEARG_2, NULL,
648 ret_number, f_append},
649 {"appendbufline", 3, 3, FEARG_3, NULL,
650 ret_number, f_appendbufline},
651 {"argc", 0, 1, 0, NULL,
652 ret_number, f_argc},
653 {"argidx", 0, 0, 0, NULL,
654 ret_number, f_argidx},
655 {"arglistid", 0, 2, 0, NULL,
656 ret_number, f_arglistid},
657 {"argv", 0, 2, 0, NULL,
658 ret_argv, f_argv},
659 {"asin", 1, 1, FEARG_1, NULL,
660 ret_float, FLOAT_FUNC(f_asin)},
661 {"assert_beeps", 1, 2, FEARG_1, NULL,
662 ret_number, f_assert_beeps},
663 {"assert_equal", 2, 3, FEARG_2, NULL,
664 ret_number, f_assert_equal},
665 {"assert_equalfile", 2, 3, FEARG_1, NULL,
666 ret_number, f_assert_equalfile},
667 {"assert_exception", 1, 2, 0, NULL,
668 ret_number, f_assert_exception},
669 {"assert_fails", 1, 5, FEARG_1, NULL,
670 ret_number, f_assert_fails},
671 {"assert_false", 1, 2, FEARG_1, NULL,
672 ret_number, f_assert_false},
673 {"assert_inrange", 3, 4, FEARG_3, NULL,
674 ret_number, f_assert_inrange},
675 {"assert_match", 2, 3, FEARG_2, NULL,
676 ret_number, f_assert_match},
677 {"assert_notequal", 2, 3, FEARG_2, NULL,
678 ret_number, f_assert_notequal},
679 {"assert_notmatch", 2, 3, FEARG_2, NULL,
680 ret_number, f_assert_notmatch},
681 {"assert_report", 1, 1, FEARG_1, NULL,
682 ret_number, f_assert_report},
683 {"assert_true", 1, 2, FEARG_1, NULL,
684 ret_number, f_assert_true},
685 {"atan", 1, 1, FEARG_1, NULL,
686 ret_float, FLOAT_FUNC(f_atan)},
687 {"atan2", 2, 2, FEARG_1, NULL,
688 ret_float, FLOAT_FUNC(f_atan2)},
689 {"balloon_gettext", 0, 0, 0, NULL,
690 ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100691#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100692 f_balloon_gettext
693#else
694 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100695#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100696 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200697 {"balloon_show", 1, 1, FEARG_1, NULL,
698 ret_void,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100699#ifdef FEAT_BEVAL
700 f_balloon_show
701#else
702 NULL
703#endif
704 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200705 {"balloon_split", 1, 1, FEARG_1, NULL,
706 ret_list_string,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100707#if defined(FEAT_BEVAL_TERM)
708 f_balloon_split
709#else
710 NULL
711#endif
712 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200713 {"browse", 4, 4, 0, NULL,
714 ret_string, f_browse},
715 {"browsedir", 2, 2, 0, NULL,
716 ret_string, f_browsedir},
717 {"bufadd", 1, 1, FEARG_1, NULL,
718 ret_number, f_bufadd},
719 {"bufexists", 1, 1, FEARG_1, NULL,
720 ret_number, f_bufexists},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200721 {"buffer_exists", 1, 1, FEARG_1, NULL, // obsolete
722 ret_number, f_bufexists},
723 {"buffer_name", 0, 1, FEARG_1, NULL, // obsolete
724 ret_string, f_bufname},
725 {"buffer_number", 0, 1, FEARG_1, NULL, // obsolete
726 ret_number, f_bufnr},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200727 {"buflisted", 1, 1, FEARG_1, NULL,
728 ret_number, f_buflisted},
729 {"bufload", 1, 1, FEARG_1, NULL,
730 ret_void, f_bufload},
731 {"bufloaded", 1, 1, FEARG_1, NULL,
732 ret_number, f_bufloaded},
733 {"bufname", 0, 1, FEARG_1, NULL,
734 ret_string, f_bufname},
735 {"bufnr", 0, 2, FEARG_1, NULL,
736 ret_number, f_bufnr},
737 {"bufwinid", 1, 1, FEARG_1, NULL,
738 ret_number, f_bufwinid},
739 {"bufwinnr", 1, 1, FEARG_1, NULL,
740 ret_number, f_bufwinnr},
741 {"byte2line", 1, 1, FEARG_1, NULL,
742 ret_number, f_byte2line},
743 {"byteidx", 2, 2, FEARG_1, NULL,
744 ret_number, f_byteidx},
745 {"byteidxcomp", 2, 2, FEARG_1, NULL,
746 ret_number, f_byteidxcomp},
747 {"call", 2, 3, FEARG_1, NULL,
748 ret_any, f_call},
749 {"ceil", 1, 1, FEARG_1, NULL,
750 ret_float, FLOAT_FUNC(f_ceil)},
751 {"ch_canread", 1, 1, FEARG_1, NULL,
752 ret_number, JOB_FUNC(f_ch_canread)},
753 {"ch_close", 1, 1, FEARG_1, NULL,
754 ret_void, JOB_FUNC(f_ch_close)},
755 {"ch_close_in", 1, 1, FEARG_1, NULL,
756 ret_void, JOB_FUNC(f_ch_close_in)},
757 {"ch_evalexpr", 2, 3, FEARG_1, NULL,
758 ret_any, JOB_FUNC(f_ch_evalexpr)},
759 {"ch_evalraw", 2, 3, FEARG_1, NULL,
760 ret_any, JOB_FUNC(f_ch_evalraw)},
761 {"ch_getbufnr", 2, 2, FEARG_1, NULL,
762 ret_number, JOB_FUNC(f_ch_getbufnr)},
763 {"ch_getjob", 1, 1, FEARG_1, NULL,
764 ret_job, JOB_FUNC(f_ch_getjob)},
765 {"ch_info", 1, 1, FEARG_1, NULL,
766 ret_dict_any, JOB_FUNC(f_ch_info)},
767 {"ch_log", 1, 2, FEARG_1, NULL,
768 ret_void, JOB_FUNC(f_ch_log)},
769 {"ch_logfile", 1, 2, FEARG_1, NULL,
770 ret_void, JOB_FUNC(f_ch_logfile)},
771 {"ch_open", 1, 2, FEARG_1, NULL,
772 ret_channel, JOB_FUNC(f_ch_open)},
773 {"ch_read", 1, 2, FEARG_1, NULL,
774 ret_string, JOB_FUNC(f_ch_read)},
775 {"ch_readblob", 1, 2, FEARG_1, NULL,
776 ret_blob, JOB_FUNC(f_ch_readblob)},
777 {"ch_readraw", 1, 2, FEARG_1, NULL,
778 ret_string, JOB_FUNC(f_ch_readraw)},
779 {"ch_sendexpr", 2, 3, FEARG_1, NULL,
780 ret_void, JOB_FUNC(f_ch_sendexpr)},
781 {"ch_sendraw", 2, 3, FEARG_1, NULL,
782 ret_void, JOB_FUNC(f_ch_sendraw)},
783 {"ch_setoptions", 2, 2, FEARG_1, NULL,
784 ret_void, JOB_FUNC(f_ch_setoptions)},
785 {"ch_status", 1, 2, FEARG_1, NULL,
786 ret_string, JOB_FUNC(f_ch_status)},
787 {"changenr", 0, 0, 0, NULL,
788 ret_number, f_changenr},
789 {"char2nr", 1, 2, FEARG_1, NULL,
790 ret_number, f_char2nr},
791 {"charclass", 1, 1, FEARG_1, NULL,
792 ret_number, f_charclass},
Bram Moolenaar17793ef2020-12-28 12:56:58 +0100793 {"charidx", 2, 3, FEARG_1, NULL,
794 ret_number, f_charidx},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200795 {"chdir", 1, 1, FEARG_1, NULL,
796 ret_string, f_chdir},
797 {"cindent", 1, 1, FEARG_1, NULL,
798 ret_number, f_cindent},
799 {"clearmatches", 0, 1, FEARG_1, NULL,
800 ret_void, f_clearmatches},
801 {"col", 1, 1, FEARG_1, NULL,
802 ret_number, f_col},
803 {"complete", 2, 2, FEARG_2, NULL,
804 ret_void, f_complete},
805 {"complete_add", 1, 1, FEARG_1, NULL,
806 ret_number, f_complete_add},
807 {"complete_check", 0, 0, 0, NULL,
808 ret_number, f_complete_check},
809 {"complete_info", 0, 1, FEARG_1, NULL,
810 ret_dict_any, f_complete_info},
811 {"confirm", 1, 4, FEARG_1, NULL,
812 ret_number, f_confirm},
813 {"copy", 1, 1, FEARG_1, NULL,
814 ret_first_arg, f_copy},
815 {"cos", 1, 1, FEARG_1, NULL,
816 ret_float, FLOAT_FUNC(f_cos)},
817 {"cosh", 1, 1, FEARG_1, NULL,
818 ret_float, FLOAT_FUNC(f_cosh)},
819 {"count", 2, 4, FEARG_1, NULL,
820 ret_number, f_count},
821 {"cscope_connection",0,3, 0, NULL,
822 ret_number, f_cscope_connection},
823 {"cursor", 1, 3, FEARG_1, NULL,
824 ret_number, f_cursor},
825 {"debugbreak", 1, 1, FEARG_1, NULL,
826 ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100827#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100828 f_debugbreak
829#else
830 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200831#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100832 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200833 {"deepcopy", 1, 2, FEARG_1, NULL,
834 ret_first_arg, f_deepcopy},
835 {"delete", 1, 2, FEARG_1, NULL,
836 ret_number, f_delete},
837 {"deletebufline", 2, 3, FEARG_1, NULL,
838 ret_number, f_deletebufline},
839 {"did_filetype", 0, 0, 0, NULL,
840 ret_number, f_did_filetype},
841 {"diff_filler", 1, 1, FEARG_1, NULL,
842 ret_number, f_diff_filler},
843 {"diff_hlID", 2, 2, FEARG_1, NULL,
844 ret_number, f_diff_hlID},
845 {"echoraw", 1, 1, FEARG_1, NULL,
846 ret_number, f_echoraw},
847 {"empty", 1, 1, FEARG_1, NULL,
848 ret_number, f_empty},
849 {"environ", 0, 0, 0, NULL,
850 ret_dict_string, f_environ},
851 {"escape", 2, 2, FEARG_1, NULL,
852 ret_string, f_escape},
853 {"eval", 1, 1, FEARG_1, NULL,
854 ret_any, f_eval},
855 {"eventhandler", 0, 0, 0, NULL,
856 ret_number, f_eventhandler},
857 {"executable", 1, 1, FEARG_1, NULL,
858 ret_number, f_executable},
859 {"execute", 1, 2, FEARG_1, NULL,
860 ret_string, f_execute},
861 {"exepath", 1, 1, FEARG_1, NULL,
862 ret_string, f_exepath},
863 {"exists", 1, 1, FEARG_1, NULL,
864 ret_number, f_exists},
865 {"exp", 1, 1, FEARG_1, NULL,
866 ret_float, FLOAT_FUNC(f_exp)},
867 {"expand", 1, 3, FEARG_1, NULL,
868 ret_any, f_expand},
869 {"expandcmd", 1, 1, FEARG_1, NULL,
870 ret_string, f_expandcmd},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100871 {"extend", 2, 3, FEARG_1, arg23_extend,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200872 ret_first_arg, f_extend},
873 {"feedkeys", 1, 2, FEARG_1, NULL,
874 ret_void, f_feedkeys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200875 {"file_readable", 1, 1, FEARG_1, NULL, // obsolete
876 ret_number, f_filereadable},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200877 {"filereadable", 1, 1, FEARG_1, NULL,
878 ret_number, f_filereadable},
879 {"filewritable", 1, 1, FEARG_1, NULL,
880 ret_number, f_filewritable},
881 {"filter", 2, 2, FEARG_1, NULL,
882 ret_first_arg, f_filter},
883 {"finddir", 1, 3, FEARG_1, NULL,
884 ret_string, f_finddir},
885 {"findfile", 1, 3, FEARG_1, NULL,
886 ret_string, f_findfile},
887 {"flatten", 1, 2, FEARG_1, NULL,
888 ret_list_any, f_flatten},
889 {"float2nr", 1, 1, FEARG_1, NULL,
890 ret_number, FLOAT_FUNC(f_float2nr)},
891 {"floor", 1, 1, FEARG_1, NULL,
892 ret_float, FLOAT_FUNC(f_floor)},
893 {"fmod", 2, 2, FEARG_1, NULL,
894 ret_float, FLOAT_FUNC(f_fmod)},
895 {"fnameescape", 1, 1, FEARG_1, NULL,
896 ret_string, f_fnameescape},
897 {"fnamemodify", 2, 2, FEARG_1, NULL,
898 ret_string, f_fnamemodify},
899 {"foldclosed", 1, 1, FEARG_1, NULL,
900 ret_number, f_foldclosed},
901 {"foldclosedend", 1, 1, FEARG_1, NULL,
902 ret_number, f_foldclosedend},
903 {"foldlevel", 1, 1, FEARG_1, NULL,
904 ret_number, f_foldlevel},
905 {"foldtext", 0, 0, 0, NULL,
906 ret_string, f_foldtext},
907 {"foldtextresult", 1, 1, FEARG_1, NULL,
908 ret_string, f_foldtextresult},
909 {"foreground", 0, 0, 0, NULL,
910 ret_void, f_foreground},
911 {"funcref", 1, 3, FEARG_1, NULL,
912 ret_func_any, f_funcref},
913 {"function", 1, 3, FEARG_1, NULL,
914 ret_f_function, f_function},
915 {"garbagecollect", 0, 1, 0, NULL,
916 ret_void, f_garbagecollect},
917 {"get", 2, 3, FEARG_1, NULL,
918 ret_any, f_get},
919 {"getbufinfo", 0, 1, FEARG_1, NULL,
920 ret_list_dict_any, f_getbufinfo},
921 {"getbufline", 2, 3, FEARG_1, NULL,
922 ret_list_string, f_getbufline},
923 {"getbufvar", 2, 3, FEARG_1, NULL,
924 ret_any, f_getbufvar},
925 {"getchangelist", 0, 1, FEARG_1, NULL,
926 ret_list_any, f_getchangelist},
927 {"getchar", 0, 1, 0, NULL,
928 ret_number, f_getchar},
929 {"getcharmod", 0, 0, 0, NULL,
930 ret_number, f_getcharmod},
931 {"getcharsearch", 0, 0, 0, NULL,
932 ret_dict_any, f_getcharsearch},
933 {"getcmdline", 0, 0, 0, NULL,
934 ret_string, f_getcmdline},
935 {"getcmdpos", 0, 0, 0, NULL,
936 ret_number, f_getcmdpos},
937 {"getcmdtype", 0, 0, 0, NULL,
938 ret_string, f_getcmdtype},
939 {"getcmdwintype", 0, 0, 0, NULL,
940 ret_string, f_getcmdwintype},
941 {"getcompletion", 2, 3, FEARG_1, NULL,
942 ret_list_string, f_getcompletion},
943 {"getcurpos", 0, 1, FEARG_1, NULL,
944 ret_list_number, f_getcurpos},
945 {"getcwd", 0, 2, FEARG_1, NULL,
946 ret_string, f_getcwd},
947 {"getenv", 1, 1, FEARG_1, NULL,
948 ret_string, f_getenv},
949 {"getfontname", 0, 1, 0, NULL,
950 ret_string, f_getfontname},
951 {"getfperm", 1, 1, FEARG_1, NULL,
952 ret_string, f_getfperm},
953 {"getfsize", 1, 1, FEARG_1, NULL,
954 ret_number, f_getfsize},
955 {"getftime", 1, 1, FEARG_1, NULL,
956 ret_number, f_getftime},
957 {"getftype", 1, 1, FEARG_1, NULL,
958 ret_string, f_getftype},
959 {"getimstatus", 0, 0, 0, NULL,
960 ret_number, f_getimstatus},
961 {"getjumplist", 0, 2, FEARG_1, NULL,
962 ret_list_any, f_getjumplist},
963 {"getline", 1, 2, FEARG_1, NULL,
964 ret_f_getline, f_getline},
965 {"getloclist", 1, 2, 0, NULL,
966 ret_list_or_dict_1, f_getloclist},
967 {"getmarklist", 0, 1, FEARG_1, NULL,
968 ret_list_dict_any, f_getmarklist},
969 {"getmatches", 0, 1, 0, NULL,
970 ret_list_dict_any, f_getmatches},
971 {"getmousepos", 0, 0, 0, NULL,
972 ret_dict_number, f_getmousepos},
973 {"getpid", 0, 0, 0, NULL,
974 ret_number, f_getpid},
975 {"getpos", 1, 1, FEARG_1, NULL,
976 ret_list_number, f_getpos},
977 {"getqflist", 0, 1, 0, NULL,
978 ret_list_or_dict_0, f_getqflist},
979 {"getreg", 0, 3, FEARG_1, NULL,
980 ret_getreg, f_getreg},
981 {"getreginfo", 0, 1, FEARG_1, NULL,
982 ret_dict_any, f_getreginfo},
983 {"getregtype", 0, 1, FEARG_1, NULL,
984 ret_string, f_getregtype},
985 {"gettabinfo", 0, 1, FEARG_1, NULL,
986 ret_list_dict_any, f_gettabinfo},
987 {"gettabvar", 2, 3, FEARG_1, NULL,
988 ret_any, f_gettabvar},
989 {"gettabwinvar", 3, 4, FEARG_1, NULL,
990 ret_any, f_gettabwinvar},
991 {"gettagstack", 0, 1, FEARG_1, NULL,
992 ret_dict_any, f_gettagstack},
993 {"gettext", 1, 1, FEARG_1, NULL,
994 ret_string, f_gettext},
995 {"getwininfo", 0, 1, FEARG_1, NULL,
996 ret_list_dict_any, f_getwininfo},
997 {"getwinpos", 0, 1, FEARG_1, NULL,
998 ret_list_number, f_getwinpos},
999 {"getwinposx", 0, 0, 0, NULL,
1000 ret_number, f_getwinposx},
1001 {"getwinposy", 0, 0, 0, NULL,
1002 ret_number, f_getwinposy},
1003 {"getwinvar", 2, 3, FEARG_1, NULL,
1004 ret_any, f_getwinvar},
1005 {"glob", 1, 4, FEARG_1, NULL,
1006 ret_any, f_glob},
1007 {"glob2regpat", 1, 1, FEARG_1, NULL,
1008 ret_string, f_glob2regpat},
1009 {"globpath", 2, 5, FEARG_2, NULL,
1010 ret_any, f_globpath},
1011 {"has", 1, 2, 0, NULL,
1012 ret_number, f_has},
1013 {"has_key", 2, 2, FEARG_1, NULL,
1014 ret_number, f_has_key},
1015 {"haslocaldir", 0, 2, FEARG_1, NULL,
1016 ret_number, f_haslocaldir},
1017 {"hasmapto", 1, 3, FEARG_1, NULL,
1018 ret_number, f_hasmapto},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001019 {"highlightID", 1, 1, FEARG_1, NULL, // obsolete
1020 ret_number, f_hlID},
1021 {"highlight_exists",1, 1, FEARG_1, NULL, // obsolete
1022 ret_number, f_hlexists},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001023 {"histadd", 2, 2, FEARG_2, NULL,
1024 ret_number, f_histadd},
1025 {"histdel", 1, 2, FEARG_1, NULL,
1026 ret_number, f_histdel},
1027 {"histget", 1, 2, FEARG_1, NULL,
1028 ret_string, f_histget},
1029 {"histnr", 1, 1, FEARG_1, NULL,
1030 ret_number, f_histnr},
1031 {"hlID", 1, 1, FEARG_1, NULL,
1032 ret_number, f_hlID},
1033 {"hlexists", 1, 1, FEARG_1, NULL,
1034 ret_number, f_hlexists},
1035 {"hostname", 0, 0, 0, NULL,
1036 ret_string, f_hostname},
1037 {"iconv", 3, 3, FEARG_1, NULL,
1038 ret_string, f_iconv},
1039 {"indent", 1, 1, FEARG_1, NULL,
1040 ret_number, f_indent},
1041 {"index", 2, 4, FEARG_1, NULL,
1042 ret_number, f_index},
1043 {"input", 1, 3, FEARG_1, NULL,
1044 ret_string, f_input},
1045 {"inputdialog", 1, 3, FEARG_1, NULL,
1046 ret_string, f_inputdialog},
1047 {"inputlist", 1, 1, FEARG_1, NULL,
1048 ret_number, f_inputlist},
1049 {"inputrestore", 0, 0, 0, NULL,
1050 ret_number, f_inputrestore},
1051 {"inputsave", 0, 0, 0, NULL,
1052 ret_number, f_inputsave},
1053 {"inputsecret", 1, 2, FEARG_1, NULL,
1054 ret_string, f_inputsecret},
Bram Moolenaarca174532020-10-21 16:42:22 +02001055 {"insert", 2, 3, FEARG_1, arg3_insert,
Bram Moolenaar94738d82020-10-21 14:25:07 +02001056 ret_first_arg, f_insert},
1057 {"interrupt", 0, 0, 0, NULL,
1058 ret_void, f_interrupt},
1059 {"invert", 1, 1, FEARG_1, NULL,
1060 ret_number, f_invert},
1061 {"isdirectory", 1, 1, FEARG_1, NULL,
1062 ret_number, f_isdirectory},
1063 {"isinf", 1, 1, FEARG_1, NULL,
1064 ret_number, MATH_FUNC(f_isinf)},
1065 {"islocked", 1, 1, FEARG_1, NULL,
1066 ret_number, f_islocked},
1067 {"isnan", 1, 1, FEARG_1, NULL,
1068 ret_number, MATH_FUNC(f_isnan)},
1069 {"items", 1, 1, FEARG_1, NULL,
1070 ret_list_any, f_items},
1071 {"job_getchannel", 1, 1, FEARG_1, NULL,
1072 ret_channel, JOB_FUNC(f_job_getchannel)},
1073 {"job_info", 0, 1, FEARG_1, NULL,
1074 ret_dict_any, JOB_FUNC(f_job_info)},
1075 {"job_setoptions", 2, 2, FEARG_1, NULL,
1076 ret_void, JOB_FUNC(f_job_setoptions)},
1077 {"job_start", 1, 2, FEARG_1, NULL,
1078 ret_job, JOB_FUNC(f_job_start)},
1079 {"job_status", 1, 1, FEARG_1, NULL,
1080 ret_string, JOB_FUNC(f_job_status)},
1081 {"job_stop", 1, 2, FEARG_1, NULL,
1082 ret_number, JOB_FUNC(f_job_stop)},
1083 {"join", 1, 2, FEARG_1, NULL,
1084 ret_string, f_join},
1085 {"js_decode", 1, 1, FEARG_1, NULL,
1086 ret_any, f_js_decode},
1087 {"js_encode", 1, 1, FEARG_1, NULL,
1088 ret_string, f_js_encode},
1089 {"json_decode", 1, 1, FEARG_1, NULL,
1090 ret_any, f_json_decode},
1091 {"json_encode", 1, 1, FEARG_1, NULL,
1092 ret_string, f_json_encode},
1093 {"keys", 1, 1, FEARG_1, NULL,
1094 ret_list_string, f_keys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001095 {"last_buffer_nr", 0, 0, 0, NULL, // obsolete
1096 ret_number, f_last_buffer_nr},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001097 {"len", 1, 1, FEARG_1, NULL,
1098 ret_number, f_len},
1099 {"libcall", 3, 3, FEARG_3, NULL,
1100 ret_string, f_libcall},
1101 {"libcallnr", 3, 3, FEARG_3, NULL,
1102 ret_number, f_libcallnr},
1103 {"line", 1, 2, FEARG_1, NULL,
1104 ret_number, f_line},
1105 {"line2byte", 1, 1, FEARG_1, NULL,
1106 ret_number, f_line2byte},
1107 {"lispindent", 1, 1, FEARG_1, NULL,
1108 ret_number, f_lispindent},
1109 {"list2str", 1, 2, FEARG_1, NULL,
1110 ret_string, f_list2str},
1111 {"listener_add", 1, 2, FEARG_2, NULL,
1112 ret_number, f_listener_add},
1113 {"listener_flush", 0, 1, FEARG_1, NULL,
1114 ret_void, f_listener_flush},
1115 {"listener_remove", 1, 1, FEARG_1, NULL,
1116 ret_number, f_listener_remove},
1117 {"localtime", 0, 0, 0, NULL,
1118 ret_number, f_localtime},
1119 {"log", 1, 1, FEARG_1, NULL,
1120 ret_float, FLOAT_FUNC(f_log)},
1121 {"log10", 1, 1, FEARG_1, NULL,
1122 ret_float, FLOAT_FUNC(f_log10)},
1123 {"luaeval", 1, 2, FEARG_1, NULL,
1124 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001125#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +01001126 f_luaeval
1127#else
1128 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001129#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001130 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001131 {"map", 2, 2, FEARG_1, NULL,
Bram Moolenaarea696852020-11-09 18:31:39 +01001132 ret_first_cont, f_map},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001133 {"maparg", 1, 4, FEARG_1, NULL,
1134 ret_maparg, f_maparg},
1135 {"mapcheck", 1, 3, FEARG_1, NULL,
1136 ret_string, f_mapcheck},
Bram Moolenaarea696852020-11-09 18:31:39 +01001137 {"mapnew", 2, 2, FEARG_1, NULL,
1138 ret_first_cont, f_mapnew},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001139 {"mapset", 3, 3, FEARG_1, NULL,
1140 ret_void, f_mapset},
1141 {"match", 2, 4, FEARG_1, NULL,
1142 ret_any, f_match},
1143 {"matchadd", 2, 5, FEARG_1, NULL,
1144 ret_number, f_matchadd},
1145 {"matchaddpos", 2, 5, FEARG_1, NULL,
1146 ret_number, f_matchaddpos},
1147 {"matcharg", 1, 1, FEARG_1, NULL,
1148 ret_list_string, f_matcharg},
1149 {"matchdelete", 1, 2, FEARG_1, NULL,
1150 ret_number, f_matchdelete},
1151 {"matchend", 2, 4, FEARG_1, NULL,
1152 ret_number, f_matchend},
1153 {"matchfuzzy", 2, 3, FEARG_1, NULL,
1154 ret_list_string, f_matchfuzzy},
1155 {"matchfuzzypos", 2, 3, FEARG_1, NULL,
1156 ret_list_any, f_matchfuzzypos},
1157 {"matchlist", 2, 4, FEARG_1, NULL,
1158 ret_list_string, f_matchlist},
1159 {"matchstr", 2, 4, FEARG_1, NULL,
1160 ret_string, f_matchstr},
1161 {"matchstrpos", 2, 4, FEARG_1, NULL,
1162 ret_list_any, f_matchstrpos},
1163 {"max", 1, 1, FEARG_1, NULL,
1164 ret_any, f_max},
1165 {"menu_info", 1, 2, FEARG_1, NULL,
1166 ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001167#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +01001168 f_menu_info
1169#else
1170 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001171#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001172 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001173 {"min", 1, 1, FEARG_1, NULL,
1174 ret_any, f_min},
1175 {"mkdir", 1, 3, FEARG_1, NULL,
1176 ret_number, f_mkdir},
1177 {"mode", 0, 1, FEARG_1, NULL,
1178 ret_string, f_mode},
1179 {"mzeval", 1, 1, FEARG_1, NULL,
1180 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001181#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001182 f_mzeval
1183#else
1184 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001185#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001186 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001187 {"nextnonblank", 1, 1, FEARG_1, NULL,
1188 ret_number, f_nextnonblank},
1189 {"nr2char", 1, 2, FEARG_1, NULL,
1190 ret_string, f_nr2char},
1191 {"or", 2, 2, FEARG_1, NULL,
1192 ret_number, f_or},
1193 {"pathshorten", 1, 2, FEARG_1, NULL,
1194 ret_string, f_pathshorten},
1195 {"perleval", 1, 1, FEARG_1, NULL,
1196 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001197#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +01001198 f_perleval
1199#else
1200 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001201#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001202 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001203 {"popup_atcursor", 2, 2, FEARG_1, NULL,
1204 ret_number, PROP_FUNC(f_popup_atcursor)},
1205 {"popup_beval", 2, 2, FEARG_1, NULL,
1206 ret_number, PROP_FUNC(f_popup_beval)},
1207 {"popup_clear", 0, 1, 0, NULL,
1208 ret_void, PROP_FUNC(f_popup_clear)},
1209 {"popup_close", 1, 2, FEARG_1, NULL,
1210 ret_void, PROP_FUNC(f_popup_close)},
1211 {"popup_create", 2, 2, FEARG_1, NULL,
1212 ret_number, PROP_FUNC(f_popup_create)},
1213 {"popup_dialog", 2, 2, FEARG_1, NULL,
1214 ret_number, PROP_FUNC(f_popup_dialog)},
1215 {"popup_filter_menu", 2, 2, 0, NULL,
1216 ret_bool, PROP_FUNC(f_popup_filter_menu)},
1217 {"popup_filter_yesno", 2, 2, 0, NULL,
1218 ret_bool, PROP_FUNC(f_popup_filter_yesno)},
1219 {"popup_findinfo", 0, 0, 0, NULL,
1220 ret_number, PROP_FUNC(f_popup_findinfo)},
1221 {"popup_findpreview", 0, 0, 0, NULL,
1222 ret_number, PROP_FUNC(f_popup_findpreview)},
1223 {"popup_getoptions", 1, 1, FEARG_1, NULL,
1224 ret_dict_any, PROP_FUNC(f_popup_getoptions)},
1225 {"popup_getpos", 1, 1, FEARG_1, NULL,
1226 ret_dict_any, PROP_FUNC(f_popup_getpos)},
1227 {"popup_hide", 1, 1, FEARG_1, NULL,
1228 ret_void, PROP_FUNC(f_popup_hide)},
1229 {"popup_list", 0, 0, 0, NULL,
1230 ret_list_number, PROP_FUNC(f_popup_list)},
1231 {"popup_locate", 2, 2, 0, NULL,
1232 ret_number, PROP_FUNC(f_popup_locate)},
1233 {"popup_menu", 2, 2, FEARG_1, NULL,
1234 ret_number, PROP_FUNC(f_popup_menu)},
1235 {"popup_move", 2, 2, FEARG_1, NULL,
1236 ret_void, PROP_FUNC(f_popup_move)},
1237 {"popup_notification", 2, 2, FEARG_1, NULL,
1238 ret_number, PROP_FUNC(f_popup_notification)},
1239 {"popup_setoptions", 2, 2, FEARG_1, NULL,
1240 ret_void, PROP_FUNC(f_popup_setoptions)},
1241 {"popup_settext", 2, 2, FEARG_1, NULL,
1242 ret_void, PROP_FUNC(f_popup_settext)},
1243 {"popup_show", 1, 1, FEARG_1, NULL,
1244 ret_void, PROP_FUNC(f_popup_show)},
1245 {"pow", 2, 2, FEARG_1, NULL,
1246 ret_float, FLOAT_FUNC(f_pow)},
1247 {"prevnonblank", 1, 1, FEARG_1, NULL,
1248 ret_number, f_prevnonblank},
1249 {"printf", 1, 19, FEARG_2, NULL,
1250 ret_string, f_printf},
1251 {"prompt_getprompt", 1, 1, FEARG_1, NULL,
1252 ret_string, JOB_FUNC(f_prompt_getprompt)},
1253 {"prompt_setcallback", 2, 2, FEARG_1, NULL,
1254 ret_void, JOB_FUNC(f_prompt_setcallback)},
1255 {"prompt_setinterrupt", 2, 2, FEARG_1, NULL,
1256 ret_void, JOB_FUNC(f_prompt_setinterrupt)},
1257 {"prompt_setprompt", 2, 2, FEARG_1, NULL,
1258 ret_void, JOB_FUNC(f_prompt_setprompt)},
1259 {"prop_add", 3, 3, FEARG_1, NULL,
1260 ret_void, PROP_FUNC(f_prop_add)},
1261 {"prop_clear", 1, 3, FEARG_1, NULL,
1262 ret_void, PROP_FUNC(f_prop_clear)},
1263 {"prop_find", 1, 2, FEARG_1, NULL,
1264 ret_dict_any, PROP_FUNC(f_prop_find)},
1265 {"prop_list", 1, 2, FEARG_1, NULL,
1266 ret_list_dict_any, PROP_FUNC(f_prop_list)},
1267 {"prop_remove", 1, 3, FEARG_1, NULL,
1268 ret_number, PROP_FUNC(f_prop_remove)},
1269 {"prop_type_add", 2, 2, FEARG_1, NULL,
1270 ret_void, PROP_FUNC(f_prop_type_add)},
1271 {"prop_type_change", 2, 2, FEARG_1, NULL,
1272 ret_void, PROP_FUNC(f_prop_type_change)},
1273 {"prop_type_delete", 1, 2, FEARG_1, NULL,
1274 ret_void, PROP_FUNC(f_prop_type_delete)},
1275 {"prop_type_get", 1, 2, FEARG_1, NULL,
1276 ret_dict_any, PROP_FUNC(f_prop_type_get)},
1277 {"prop_type_list", 0, 1, FEARG_1, NULL,
1278 ret_list_string, PROP_FUNC(f_prop_type_list)},
1279 {"pum_getpos", 0, 0, 0, NULL,
1280 ret_dict_number, f_pum_getpos},
1281 {"pumvisible", 0, 0, 0, NULL,
1282 ret_number, f_pumvisible},
1283 {"py3eval", 1, 1, FEARG_1, NULL,
1284 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001285#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +01001286 f_py3eval
1287#else
1288 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001289#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001290 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001291 {"pyeval", 1, 1, FEARG_1, NULL,
1292 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001293#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +01001294 f_pyeval
1295#else
1296 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001297#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001298 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001299 {"pyxeval", 1, 1, FEARG_1, NULL,
1300 ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001301#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +01001302 f_pyxeval
1303#else
1304 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001305#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001306 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001307 {"rand", 0, 1, FEARG_1, NULL,
1308 ret_number, f_rand},
1309 {"range", 1, 3, FEARG_1, NULL,
1310 ret_list_number, f_range},
1311 {"readdir", 1, 3, FEARG_1, NULL,
1312 ret_list_string, f_readdir},
1313 {"readdirex", 1, 3, FEARG_1, NULL,
1314 ret_list_dict_any, f_readdirex},
1315 {"readfile", 1, 3, FEARG_1, NULL,
1316 ret_any, f_readfile},
1317 {"reduce", 2, 3, FEARG_1, NULL,
1318 ret_any, f_reduce},
1319 {"reg_executing", 0, 0, 0, NULL,
1320 ret_string, f_reg_executing},
1321 {"reg_recording", 0, 0, 0, NULL,
1322 ret_string, f_reg_recording},
1323 {"reltime", 0, 2, FEARG_1, NULL,
1324 ret_list_any, f_reltime},
1325 {"reltimefloat", 1, 1, FEARG_1, NULL,
1326 ret_float, FLOAT_FUNC(f_reltimefloat)},
1327 {"reltimestr", 1, 1, FEARG_1, NULL,
1328 ret_string, f_reltimestr},
1329 {"remote_expr", 2, 4, FEARG_1, NULL,
1330 ret_string, f_remote_expr},
1331 {"remote_foreground", 1, 1, FEARG_1, NULL,
1332 ret_string, f_remote_foreground},
1333 {"remote_peek", 1, 2, FEARG_1, NULL,
1334 ret_number, f_remote_peek},
1335 {"remote_read", 1, 2, FEARG_1, NULL,
1336 ret_string, f_remote_read},
1337 {"remote_send", 2, 3, FEARG_1, NULL,
1338 ret_string, f_remote_send},
1339 {"remote_startserver", 1, 1, FEARG_1, NULL,
1340 ret_void, f_remote_startserver},
1341 {"remove", 2, 3, FEARG_1, NULL,
1342 ret_remove, f_remove},
1343 {"rename", 2, 2, FEARG_1, NULL,
1344 ret_number, f_rename},
1345 {"repeat", 2, 2, FEARG_1, NULL,
1346 ret_first_arg, f_repeat},
1347 {"resolve", 1, 1, FEARG_1, NULL,
1348 ret_string, f_resolve},
1349 {"reverse", 1, 1, FEARG_1, NULL,
1350 ret_first_arg, f_reverse},
1351 {"round", 1, 1, FEARG_1, NULL,
1352 ret_float, FLOAT_FUNC(f_round)},
1353 {"rubyeval", 1, 1, FEARG_1, NULL,
1354 ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001355#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +01001356 f_rubyeval
1357#else
1358 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001359#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001360 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001361 {"screenattr", 2, 2, FEARG_1, NULL,
1362 ret_number, f_screenattr},
1363 {"screenchar", 2, 2, FEARG_1, NULL,
1364 ret_number, f_screenchar},
1365 {"screenchars", 2, 2, FEARG_1, NULL,
1366 ret_list_number, f_screenchars},
1367 {"screencol", 0, 0, 0, NULL,
1368 ret_number, f_screencol},
1369 {"screenpos", 3, 3, FEARG_1, NULL,
1370 ret_dict_number, f_screenpos},
1371 {"screenrow", 0, 0, 0, NULL,
1372 ret_number, f_screenrow},
1373 {"screenstring", 2, 2, FEARG_1, NULL,
1374 ret_string, f_screenstring},
1375 {"search", 1, 5, FEARG_1, NULL,
1376 ret_number, f_search},
1377 {"searchcount", 0, 1, FEARG_1, NULL,
1378 ret_dict_any, f_searchcount},
1379 {"searchdecl", 1, 3, FEARG_1, NULL,
1380 ret_number, f_searchdecl},
1381 {"searchpair", 3, 7, 0, NULL,
1382 ret_number, f_searchpair},
1383 {"searchpairpos", 3, 7, 0, NULL,
1384 ret_list_number, f_searchpairpos},
1385 {"searchpos", 1, 5, FEARG_1, NULL,
1386 ret_list_number, f_searchpos},
1387 {"server2client", 2, 2, FEARG_1, NULL,
1388 ret_number, f_server2client},
1389 {"serverlist", 0, 0, 0, NULL,
1390 ret_string, f_serverlist},
1391 {"setbufline", 3, 3, FEARG_3, NULL,
1392 ret_number, f_setbufline},
1393 {"setbufvar", 3, 3, FEARG_3, NULL,
1394 ret_void, f_setbufvar},
1395 {"setcellwidths", 1, 1, FEARG_1, NULL,
1396 ret_void, f_setcellwidths},
1397 {"setcharsearch", 1, 1, FEARG_1, NULL,
1398 ret_void, f_setcharsearch},
1399 {"setcmdpos", 1, 1, FEARG_1, NULL,
1400 ret_number, f_setcmdpos},
1401 {"setenv", 2, 2, FEARG_2, NULL,
1402 ret_void, f_setenv},
1403 {"setfperm", 2, 2, FEARG_1, NULL,
1404 ret_number, f_setfperm},
1405 {"setline", 2, 2, FEARG_2, NULL,
1406 ret_number, f_setline},
1407 {"setloclist", 2, 4, FEARG_2, NULL,
1408 ret_number, f_setloclist},
1409 {"setmatches", 1, 2, FEARG_1, NULL,
1410 ret_number, f_setmatches},
1411 {"setpos", 2, 2, FEARG_2, NULL,
1412 ret_number, f_setpos},
1413 {"setqflist", 1, 3, FEARG_1, NULL,
1414 ret_number, f_setqflist},
1415 {"setreg", 2, 3, FEARG_2, NULL,
1416 ret_number, f_setreg},
1417 {"settabvar", 3, 3, FEARG_3, NULL,
1418 ret_void, f_settabvar},
1419 {"settabwinvar", 4, 4, FEARG_4, NULL,
1420 ret_void, f_settabwinvar},
1421 {"settagstack", 2, 3, FEARG_2, NULL,
1422 ret_number, f_settagstack},
1423 {"setwinvar", 3, 3, FEARG_3, NULL,
1424 ret_void, f_setwinvar},
1425 {"sha256", 1, 1, FEARG_1, NULL,
1426 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001427#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +01001428 f_sha256
1429#else
1430 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001431#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001432 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001433 {"shellescape", 1, 2, FEARG_1, NULL,
1434 ret_string, f_shellescape},
1435 {"shiftwidth", 0, 1, FEARG_1, NULL,
1436 ret_number, f_shiftwidth},
1437 {"sign_define", 1, 2, FEARG_1, NULL,
1438 ret_any, SIGN_FUNC(f_sign_define)},
1439 {"sign_getdefined", 0, 1, FEARG_1, NULL,
1440 ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
1441 {"sign_getplaced", 0, 2, FEARG_1, NULL,
1442 ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
1443 {"sign_jump", 3, 3, FEARG_1, NULL,
1444 ret_number, SIGN_FUNC(f_sign_jump)},
1445 {"sign_place", 4, 5, FEARG_1, NULL,
1446 ret_number, SIGN_FUNC(f_sign_place)},
1447 {"sign_placelist", 1, 1, FEARG_1, NULL,
1448 ret_list_number, SIGN_FUNC(f_sign_placelist)},
1449 {"sign_undefine", 0, 1, FEARG_1, NULL,
1450 ret_number, SIGN_FUNC(f_sign_undefine)},
1451 {"sign_unplace", 1, 2, FEARG_1, NULL,
1452 ret_number, SIGN_FUNC(f_sign_unplace)},
1453 {"sign_unplacelist", 1, 2, FEARG_1, NULL,
1454 ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
1455 {"simplify", 1, 1, FEARG_1, NULL,
1456 ret_string, f_simplify},
1457 {"sin", 1, 1, FEARG_1, NULL,
1458 ret_float, FLOAT_FUNC(f_sin)},
1459 {"sinh", 1, 1, FEARG_1, NULL,
1460 ret_float, FLOAT_FUNC(f_sinh)},
1461 {"sort", 1, 3, FEARG_1, NULL,
1462 ret_first_arg, f_sort},
1463 {"sound_clear", 0, 0, 0, NULL,
1464 ret_void, SOUND_FUNC(f_sound_clear)},
1465 {"sound_playevent", 1, 2, FEARG_1, NULL,
1466 ret_number, SOUND_FUNC(f_sound_playevent)},
1467 {"sound_playfile", 1, 2, FEARG_1, NULL,
1468 ret_number, SOUND_FUNC(f_sound_playfile)},
1469 {"sound_stop", 1, 1, FEARG_1, NULL,
1470 ret_void, SOUND_FUNC(f_sound_stop)},
1471 {"soundfold", 1, 1, FEARG_1, NULL,
1472 ret_string, f_soundfold},
1473 {"spellbadword", 0, 1, FEARG_1, NULL,
1474 ret_list_string, f_spellbadword},
1475 {"spellsuggest", 1, 3, FEARG_1, NULL,
1476 ret_list_string, f_spellsuggest},
1477 {"split", 1, 3, FEARG_1, NULL,
1478 ret_list_string, f_split},
1479 {"sqrt", 1, 1, FEARG_1, NULL,
1480 ret_float, FLOAT_FUNC(f_sqrt)},
1481 {"srand", 0, 1, FEARG_1, NULL,
1482 ret_list_number, f_srand},
1483 {"state", 0, 1, FEARG_1, NULL,
1484 ret_string, f_state},
1485 {"str2float", 1, 1, FEARG_1, NULL,
1486 ret_float, FLOAT_FUNC(f_str2float)},
1487 {"str2list", 1, 2, FEARG_1, NULL,
1488 ret_list_number, f_str2list},
1489 {"str2nr", 1, 3, FEARG_1, NULL,
1490 ret_number, f_str2nr},
1491 {"strcharpart", 2, 3, FEARG_1, NULL,
1492 ret_string, f_strcharpart},
1493 {"strchars", 1, 2, FEARG_1, NULL,
1494 ret_number, f_strchars},
1495 {"strdisplaywidth", 1, 2, FEARG_1, NULL,
1496 ret_number, f_strdisplaywidth},
1497 {"strftime", 1, 2, FEARG_1, NULL,
1498 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001499#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001500 f_strftime
1501#else
1502 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001503#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001504 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001505 {"strgetchar", 2, 2, FEARG_1, NULL,
1506 ret_number, f_strgetchar},
1507 {"stridx", 2, 3, FEARG_1, NULL,
1508 ret_number, f_stridx},
1509 {"string", 1, 1, FEARG_1, NULL,
1510 ret_string, f_string},
1511 {"strlen", 1, 1, FEARG_1, NULL,
1512 ret_number, f_strlen},
1513 {"strpart", 2, 4, FEARG_1, NULL,
1514 ret_string, f_strpart},
1515 {"strptime", 2, 2, FEARG_1, NULL,
1516 ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +01001517#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001518 f_strptime
1519#else
1520 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +01001521#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001522 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001523 {"strridx", 2, 3, FEARG_1, NULL,
1524 ret_number, f_strridx},
1525 {"strtrans", 1, 1, FEARG_1, NULL,
1526 ret_string, f_strtrans},
1527 {"strwidth", 1, 1, FEARG_1, NULL,
1528 ret_number, f_strwidth},
1529 {"submatch", 1, 2, FEARG_1, NULL,
1530 ret_string, f_submatch},
1531 {"substitute", 4, 4, FEARG_1, NULL,
1532 ret_string, f_substitute},
1533 {"swapinfo", 1, 1, FEARG_1, NULL,
1534 ret_dict_any, f_swapinfo},
1535 {"swapname", 1, 1, FEARG_1, NULL,
1536 ret_string, f_swapname},
1537 {"synID", 3, 3, 0, NULL,
1538 ret_number, f_synID},
1539 {"synIDattr", 2, 3, FEARG_1, NULL,
1540 ret_string, f_synIDattr},
1541 {"synIDtrans", 1, 1, FEARG_1, NULL,
1542 ret_number, f_synIDtrans},
1543 {"synconcealed", 2, 2, 0, NULL,
1544 ret_list_any, f_synconcealed},
1545 {"synstack", 2, 2, 0, NULL,
1546 ret_list_number, f_synstack},
1547 {"system", 1, 2, FEARG_1, NULL,
1548 ret_string, f_system},
1549 {"systemlist", 1, 2, FEARG_1, NULL,
1550 ret_list_string, f_systemlist},
1551 {"tabpagebuflist", 0, 1, FEARG_1, NULL,
1552 ret_list_number, f_tabpagebuflist},
1553 {"tabpagenr", 0, 1, 0, NULL,
1554 ret_number, f_tabpagenr},
1555 {"tabpagewinnr", 1, 2, FEARG_1, NULL,
1556 ret_number, f_tabpagewinnr},
1557 {"tagfiles", 0, 0, 0, NULL,
1558 ret_list_string, f_tagfiles},
1559 {"taglist", 1, 2, FEARG_1, NULL,
1560 ret_list_dict_any, f_taglist},
1561 {"tan", 1, 1, FEARG_1, NULL,
1562 ret_float, FLOAT_FUNC(f_tan)},
1563 {"tanh", 1, 1, FEARG_1, NULL,
1564 ret_float, FLOAT_FUNC(f_tanh)},
1565 {"tempname", 0, 0, 0, NULL,
1566 ret_string, f_tempname},
1567 {"term_dumpdiff", 2, 3, FEARG_1, NULL,
1568 ret_number, TERM_FUNC(f_term_dumpdiff)},
1569 {"term_dumpload", 1, 2, FEARG_1, NULL,
1570 ret_number, TERM_FUNC(f_term_dumpload)},
1571 {"term_dumpwrite", 2, 3, FEARG_2, NULL,
1572 ret_void, TERM_FUNC(f_term_dumpwrite)},
1573 {"term_getaltscreen", 1, 1, FEARG_1, NULL,
1574 ret_number, TERM_FUNC(f_term_getaltscreen)},
1575 {"term_getansicolors", 1, 1, FEARG_1, NULL,
1576 ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001577#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001578 f_term_getansicolors
1579#else
1580 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001581#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001582 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001583 {"term_getattr", 2, 2, FEARG_1, NULL,
1584 ret_number, TERM_FUNC(f_term_getattr)},
1585 {"term_getcursor", 1, 1, FEARG_1, NULL,
1586 ret_list_any, TERM_FUNC(f_term_getcursor)},
1587 {"term_getjob", 1, 1, FEARG_1, NULL,
1588 ret_job, TERM_FUNC(f_term_getjob)},
1589 {"term_getline", 2, 2, FEARG_1, NULL,
1590 ret_string, TERM_FUNC(f_term_getline)},
1591 {"term_getscrolled", 1, 1, FEARG_1, NULL,
1592 ret_number, TERM_FUNC(f_term_getscrolled)},
1593 {"term_getsize", 1, 1, FEARG_1, NULL,
1594 ret_list_number, TERM_FUNC(f_term_getsize)},
1595 {"term_getstatus", 1, 1, FEARG_1, NULL,
1596 ret_string, TERM_FUNC(f_term_getstatus)},
1597 {"term_gettitle", 1, 1, FEARG_1, NULL,
1598 ret_string, TERM_FUNC(f_term_gettitle)},
1599 {"term_gettty", 1, 2, FEARG_1, NULL,
1600 ret_string, TERM_FUNC(f_term_gettty)},
1601 {"term_list", 0, 0, 0, NULL,
1602 ret_list_number, TERM_FUNC(f_term_list)},
1603 {"term_scrape", 2, 2, FEARG_1, NULL,
1604 ret_list_dict_any, TERM_FUNC(f_term_scrape)},
1605 {"term_sendkeys", 2, 2, FEARG_1, NULL,
1606 ret_void, TERM_FUNC(f_term_sendkeys)},
1607 {"term_setansicolors", 2, 2, FEARG_1, NULL,
1608 ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001609#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001610 f_term_setansicolors
1611#else
1612 NULL
1613#endif
1614 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001615 {"term_setapi", 2, 2, FEARG_1, NULL,
1616 ret_void, TERM_FUNC(f_term_setapi)},
1617 {"term_setkill", 2, 2, FEARG_1, NULL,
1618 ret_void, TERM_FUNC(f_term_setkill)},
1619 {"term_setrestore", 2, 2, FEARG_1, NULL,
1620 ret_void, TERM_FUNC(f_term_setrestore)},
1621 {"term_setsize", 3, 3, FEARG_1, NULL,
1622 ret_void, TERM_FUNC(f_term_setsize)},
1623 {"term_start", 1, 2, FEARG_1, NULL,
1624 ret_number, TERM_FUNC(f_term_start)},
1625 {"term_wait", 1, 2, FEARG_1, NULL,
1626 ret_void, TERM_FUNC(f_term_wait)},
1627 {"terminalprops", 0, 0, 0, NULL,
1628 ret_dict_string, f_terminalprops},
1629 {"test_alloc_fail", 3, 3, FEARG_1, NULL,
1630 ret_void, f_test_alloc_fail},
1631 {"test_autochdir", 0, 0, 0, NULL,
1632 ret_void, f_test_autochdir},
1633 {"test_feedinput", 1, 1, FEARG_1, NULL,
1634 ret_void, f_test_feedinput},
1635 {"test_garbagecollect_now", 0, 0, 0, NULL,
1636 ret_void, f_test_garbagecollect_now},
1637 {"test_garbagecollect_soon", 0, 0, 0, NULL,
1638 ret_void, f_test_garbagecollect_soon},
1639 {"test_getvalue", 1, 1, FEARG_1, NULL,
1640 ret_number, f_test_getvalue},
1641 {"test_ignore_error", 1, 1, FEARG_1, NULL,
1642 ret_void, f_test_ignore_error},
1643 {"test_null_blob", 0, 0, 0, NULL,
1644 ret_blob, f_test_null_blob},
1645 {"test_null_channel", 0, 0, 0, NULL,
1646 ret_channel, JOB_FUNC(f_test_null_channel)},
1647 {"test_null_dict", 0, 0, 0, NULL,
1648 ret_dict_any, f_test_null_dict},
1649 {"test_null_function", 0, 0, 0, NULL,
1650 ret_func_any, f_test_null_function},
1651 {"test_null_job", 0, 0, 0, NULL,
1652 ret_job, JOB_FUNC(f_test_null_job)},
1653 {"test_null_list", 0, 0, 0, NULL,
1654 ret_list_any, f_test_null_list},
1655 {"test_null_partial", 0, 0, 0, NULL,
1656 ret_func_any, f_test_null_partial},
1657 {"test_null_string", 0, 0, 0, NULL,
1658 ret_string, f_test_null_string},
1659 {"test_option_not_set", 1, 1, FEARG_1, NULL,
1660 ret_void, f_test_option_not_set},
1661 {"test_override", 2, 2, FEARG_2, NULL,
1662 ret_void, f_test_override},
1663 {"test_refcount", 1, 1, FEARG_1, NULL,
1664 ret_number, f_test_refcount},
1665 {"test_scrollbar", 3, 3, FEARG_2, NULL,
1666 ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +02001667#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +01001668 f_test_scrollbar
1669#else
1670 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +02001671#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001672 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001673 {"test_setmouse", 2, 2, 0, NULL,
1674 ret_void, f_test_setmouse},
1675 {"test_settime", 1, 1, FEARG_1, NULL,
1676 ret_void, f_test_settime},
1677 {"test_srand_seed", 0, 1, FEARG_1, NULL,
1678 ret_void, f_test_srand_seed},
1679 {"test_unknown", 0, 0, 0, NULL,
1680 ret_any, f_test_unknown},
1681 {"test_void", 0, 0, 0, NULL,
1682 ret_void, f_test_void},
1683 {"timer_info", 0, 1, FEARG_1, NULL,
1684 ret_list_dict_any, TIMER_FUNC(f_timer_info)},
1685 {"timer_pause", 2, 2, FEARG_1, NULL,
1686 ret_void, TIMER_FUNC(f_timer_pause)},
1687 {"timer_start", 2, 3, FEARG_1, NULL,
1688 ret_number, TIMER_FUNC(f_timer_start)},
1689 {"timer_stop", 1, 1, FEARG_1, NULL,
1690 ret_void, TIMER_FUNC(f_timer_stop)},
1691 {"timer_stopall", 0, 0, 0, NULL,
1692 ret_void, TIMER_FUNC(f_timer_stopall)},
1693 {"tolower", 1, 1, FEARG_1, NULL,
1694 ret_string, f_tolower},
1695 {"toupper", 1, 1, FEARG_1, NULL,
1696 ret_string, f_toupper},
1697 {"tr", 3, 3, FEARG_1, NULL,
1698 ret_string, f_tr},
1699 {"trim", 1, 3, FEARG_1, NULL,
1700 ret_string, f_trim},
1701 {"trunc", 1, 1, FEARG_1, NULL,
1702 ret_float, FLOAT_FUNC(f_trunc)},
1703 {"type", 1, 1, FEARG_1, NULL,
1704 ret_number, f_type},
1705 {"undofile", 1, 1, FEARG_1, NULL,
1706 ret_string, f_undofile},
1707 {"undotree", 0, 0, 0, NULL,
1708 ret_dict_any, f_undotree},
1709 {"uniq", 1, 3, FEARG_1, NULL,
1710 ret_list_any, f_uniq},
1711 {"values", 1, 1, FEARG_1, NULL,
1712 ret_list_any, f_values},
1713 {"virtcol", 1, 1, FEARG_1, NULL,
1714 ret_number, f_virtcol},
1715 {"visualmode", 0, 1, 0, NULL,
1716 ret_string, f_visualmode},
1717 {"wildmenumode", 0, 0, 0, NULL,
1718 ret_number, f_wildmenumode},
1719 {"win_execute", 2, 3, FEARG_2, NULL,
1720 ret_string, f_win_execute},
1721 {"win_findbuf", 1, 1, FEARG_1, NULL,
1722 ret_list_number, f_win_findbuf},
1723 {"win_getid", 0, 2, FEARG_1, NULL,
1724 ret_number, f_win_getid},
1725 {"win_gettype", 0, 1, FEARG_1, NULL,
1726 ret_string, f_win_gettype},
1727 {"win_gotoid", 1, 1, FEARG_1, NULL,
1728 ret_number, f_win_gotoid},
1729 {"win_id2tabwin", 1, 1, FEARG_1, NULL,
1730 ret_list_number, f_win_id2tabwin},
1731 {"win_id2win", 1, 1, FEARG_1, NULL,
1732 ret_number, f_win_id2win},
1733 {"win_screenpos", 1, 1, FEARG_1, NULL,
1734 ret_list_number, f_win_screenpos},
1735 {"win_splitmove", 2, 3, FEARG_1, NULL,
1736 ret_number, f_win_splitmove},
1737 {"winbufnr", 1, 1, FEARG_1, NULL,
1738 ret_number, f_winbufnr},
1739 {"wincol", 0, 0, 0, NULL,
1740 ret_number, f_wincol},
1741 {"windowsversion", 0, 0, 0, NULL,
1742 ret_string, f_windowsversion},
1743 {"winheight", 1, 1, FEARG_1, NULL,
1744 ret_number, f_winheight},
1745 {"winlayout", 0, 1, FEARG_1, NULL,
1746 ret_list_any, f_winlayout},
1747 {"winline", 0, 0, 0, NULL,
1748 ret_number, f_winline},
1749 {"winnr", 0, 1, FEARG_1, NULL,
1750 ret_number, f_winnr},
1751 {"winrestcmd", 0, 0, 0, NULL,
1752 ret_string, f_winrestcmd},
1753 {"winrestview", 1, 1, FEARG_1, NULL,
1754 ret_void, f_winrestview},
1755 {"winsaveview", 0, 0, 0, NULL,
1756 ret_dict_any, f_winsaveview},
1757 {"winwidth", 1, 1, FEARG_1, NULL,
1758 ret_number, f_winwidth},
1759 {"wordcount", 0, 0, 0, NULL,
1760 ret_dict_number, f_wordcount},
1761 {"writefile", 2, 3, FEARG_1, NULL,
1762 ret_number, f_writefile},
1763 {"xor", 2, 2, FEARG_1, NULL,
1764 ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001765};
1766
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001767/*
1768 * Function given to ExpandGeneric() to obtain the list of internal
1769 * or user defined function names.
1770 */
1771 char_u *
1772get_function_name(expand_T *xp, int idx)
1773{
1774 static int intidx = -1;
1775 char_u *name;
1776
1777 if (idx == 0)
1778 intidx = -1;
1779 if (intidx < 0)
1780 {
1781 name = get_user_func_name(xp, idx);
1782 if (name != NULL)
1783 return name;
1784 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001785 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001786 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001787 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001788 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001789 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001790 STRCAT(IObuff, ")");
1791 return IObuff;
1792 }
1793
1794 return NULL;
1795}
1796
1797/*
1798 * Function given to ExpandGeneric() to obtain the list of internal or
1799 * user defined variable or function names.
1800 */
1801 char_u *
1802get_expr_name(expand_T *xp, int idx)
1803{
1804 static int intidx = -1;
1805 char_u *name;
1806
1807 if (idx == 0)
1808 intidx = -1;
1809 if (intidx < 0)
1810 {
1811 name = get_function_name(xp, idx);
1812 if (name != NULL)
1813 return name;
1814 }
1815 return get_user_var_name(xp, ++intidx);
1816}
1817
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001818/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001819 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001820 * Return index, or -1 if not found or "implemented" is TRUE and the function
1821 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001822 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001823 static int
1824find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001825{
1826 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001827 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001828 int cmp;
1829 int x;
1830
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001831 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001832
1833 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001834 while (first <= last)
1835 {
1836 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001837 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001838 if (cmp < 0)
1839 last = x - 1;
1840 else if (cmp > 0)
1841 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001842 else if (implemented && global_functions[x].f_func == NULL)
1843 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001844 else
1845 return x;
1846 }
1847 return -1;
1848}
1849
Bram Moolenaar15c47602020-03-26 22:16:48 +01001850/*
1851 * Find internal function "name" in table "global_functions".
1852 * Return index, or -1 if not found or the function is not implemented.
1853 */
1854 int
1855find_internal_func(char_u *name)
1856{
1857 return find_internal_func_opt(name, TRUE);
1858}
1859
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001860 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001861has_internal_func(char_u *name)
1862{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001863 return find_internal_func_opt(name, TRUE) >= 0;
1864}
1865
1866 static int
1867has_internal_func_name(char_u *name)
1868{
1869 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001870}
1871
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001872 char *
1873internal_func_name(int idx)
1874{
1875 return global_functions[idx].f_name;
1876}
1877
Bram Moolenaar94738d82020-10-21 14:25:07 +02001878/*
1879 * Check the argument types for builting function "idx".
1880 * Uses the list of types on the type stack: "types".
1881 * Return FAIL and gives an error message when a type is wrong.
1882 */
1883 int
Bram Moolenaarca174532020-10-21 16:42:22 +02001884internal_func_check_arg_types(type_T **types, int idx, int argcount)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001885{
1886 argcheck_T *argchecks = global_functions[idx].f_argcheck;
1887 int i;
1888
1889 if (argchecks != NULL)
1890 {
1891 argcontext_T context;
1892
1893 context.arg_count = argcount;
Bram Moolenaarca174532020-10-21 16:42:22 +02001894 context.arg_types = types;
Bram Moolenaar94738d82020-10-21 14:25:07 +02001895 for (i = 0; i < argcount; ++i)
1896 if (argchecks[i] != NULL)
1897 {
1898 context.arg_idx = i;
Bram Moolenaarca174532020-10-21 16:42:22 +02001899 if (argchecks[i](types[i], &context) == FAIL)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001900 return FAIL;
1901 }
1902 }
1903 return OK;
1904}
1905
Bram Moolenaara1224cb2020-10-22 12:31:49 +02001906/*
1907 * Call the "f_retfunc" function to obtain the return type of function "idx".
1908 * "argtypes" is the list of argument types or NULL when there are no
1909 * arguments.
1910 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001911 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001912internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001913{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001914 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001915}
1916
1917/*
1918 * Check the argument count to use for internal function "idx".
Bram Moolenaar389df252020-07-09 21:20:47 +02001919 * Returns -1 for failure, 0 if no method base accepted, 1 if method base is
1920 * first argument, 2 if method base is second argument, etc. 9 if method base
1921 * is last argument.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001922 */
1923 int
1924check_internal_func(int idx, int argcount)
1925{
1926 int res;
1927 char *name;
1928
1929 if (argcount < global_functions[idx].f_min_argc)
1930 res = FCERR_TOOFEW;
1931 else if (argcount > global_functions[idx].f_max_argc)
1932 res = FCERR_TOOMANY;
1933 else
Bram Moolenaar389df252020-07-09 21:20:47 +02001934 return global_functions[idx].f_argtype;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001935
1936 name = internal_func_name(idx);
1937 if (res == FCERR_TOOMANY)
1938 semsg(_(e_toomanyarg), name);
1939 else
1940 semsg(_(e_toofewarg), name);
Bram Moolenaar389df252020-07-09 21:20:47 +02001941 return -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001942}
1943
Bram Moolenaarac92e252019-08-03 21:58:38 +02001944 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001945call_internal_func(
1946 char_u *name,
1947 int argcount,
1948 typval_T *argvars,
1949 typval_T *rettv)
1950{
1951 int i;
1952
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001953 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001954 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001955 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001956 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001957 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001958 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001959 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001960 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001961 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001962 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001963}
1964
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001965 void
1966call_internal_func_by_idx(
1967 int idx,
1968 typval_T *argvars,
1969 typval_T *rettv)
1970{
1971 global_functions[idx].f_func(argvars, rettv);
1972}
1973
Bram Moolenaarac92e252019-08-03 21:58:38 +02001974/*
1975 * Invoke a method for base->method().
1976 */
1977 int
1978call_internal_method(
1979 char_u *name,
1980 int argcount,
1981 typval_T *argvars,
1982 typval_T *rettv,
1983 typval_T *basetv)
1984{
1985 int i;
1986 int fi;
1987 typval_T argv[MAX_FUNC_ARGS + 1];
1988
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001989 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02001990 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001991 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02001992 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001993 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001994 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001995 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001996 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001997 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001998
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001999 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002000 {
2001 // base value goes last
2002 for (i = 0; i < argcount; ++i)
2003 argv[i] = argvars[i];
2004 argv[argcount] = *basetv;
2005 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002006 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002007 {
2008 // base value goes second
2009 argv[0] = argvars[0];
2010 argv[1] = *basetv;
2011 for (i = 1; i < argcount; ++i)
2012 argv[i + 1] = argvars[i];
2013 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02002014 else if (global_functions[fi].f_argtype == FEARG_3)
2015 {
2016 // base value goes third
2017 argv[0] = argvars[0];
2018 argv[1] = argvars[1];
2019 argv[2] = *basetv;
2020 for (i = 2; i < argcount; ++i)
2021 argv[i + 1] = argvars[i];
2022 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02002023 else if (global_functions[fi].f_argtype == FEARG_4)
2024 {
2025 // base value goes fourth
2026 argv[0] = argvars[0];
2027 argv[1] = argvars[1];
2028 argv[2] = argvars[2];
2029 argv[3] = *basetv;
2030 for (i = 3; i < argcount; ++i)
2031 argv[i + 1] = argvars[i];
2032 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02002033 else
2034 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002035 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02002036 argv[0] = *basetv;
2037 for (i = 0; i < argcount; ++i)
2038 argv[i + 1] = argvars[i];
2039 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02002040 argv[argcount + 1].v_type = VAR_UNKNOWN;
2041
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002042 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01002043 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002044}
2045
2046/*
2047 * Return TRUE for a non-zero Number and a non-empty String.
2048 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02002049 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002050non_zero_arg(typval_T *argvars)
2051{
2052 return ((argvars[0].v_type == VAR_NUMBER
2053 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002054 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002055 && argvars[0].vval.v_number == VVAL_TRUE)
2056 || (argvars[0].v_type == VAR_STRING
2057 && argvars[0].vval.v_string != NULL
2058 && *argvars[0].vval.v_string != NUL));
2059}
2060
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002061#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002062/*
2063 * Get the float value of "argvars[0]" into "f".
2064 * Returns FAIL when the argument is not a Number or Float.
2065 */
2066 static int
2067get_float_arg(typval_T *argvars, float_T *f)
2068{
2069 if (argvars[0].v_type == VAR_FLOAT)
2070 {
2071 *f = argvars[0].vval.v_float;
2072 return OK;
2073 }
2074 if (argvars[0].v_type == VAR_NUMBER)
2075 {
2076 *f = (float_T)argvars[0].vval.v_number;
2077 return OK;
2078 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002079 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002080 return FAIL;
2081}
2082
2083/*
2084 * "abs(expr)" function
2085 */
2086 static void
2087f_abs(typval_T *argvars, typval_T *rettv)
2088{
2089 if (argvars[0].v_type == VAR_FLOAT)
2090 {
2091 rettv->v_type = VAR_FLOAT;
2092 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
2093 }
2094 else
2095 {
2096 varnumber_T n;
2097 int error = FALSE;
2098
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002099 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002100 if (error)
2101 rettv->vval.v_number = -1;
2102 else if (n > 0)
2103 rettv->vval.v_number = n;
2104 else
2105 rettv->vval.v_number = -n;
2106 }
2107}
2108
2109/*
2110 * "acos()" function
2111 */
2112 static void
2113f_acos(typval_T *argvars, typval_T *rettv)
2114{
2115 float_T f = 0.0;
2116
2117 rettv->v_type = VAR_FLOAT;
2118 if (get_float_arg(argvars, &f) == OK)
2119 rettv->vval.v_float = acos(f);
2120 else
2121 rettv->vval.v_float = 0.0;
2122}
2123#endif
2124
2125/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002126 * "and(expr, expr)" function
2127 */
2128 static void
2129f_and(typval_T *argvars, typval_T *rettv)
2130{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002131 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
2132 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02002133}
2134
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002135#ifdef FEAT_FLOAT
2136/*
2137 * "asin()" function
2138 */
2139 static void
2140f_asin(typval_T *argvars, typval_T *rettv)
2141{
2142 float_T f = 0.0;
2143
2144 rettv->v_type = VAR_FLOAT;
2145 if (get_float_arg(argvars, &f) == OK)
2146 rettv->vval.v_float = asin(f);
2147 else
2148 rettv->vval.v_float = 0.0;
2149}
2150
2151/*
2152 * "atan()" function
2153 */
2154 static void
2155f_atan(typval_T *argvars, typval_T *rettv)
2156{
2157 float_T f = 0.0;
2158
2159 rettv->v_type = VAR_FLOAT;
2160 if (get_float_arg(argvars, &f) == OK)
2161 rettv->vval.v_float = atan(f);
2162 else
2163 rettv->vval.v_float = 0.0;
2164}
2165
2166/*
2167 * "atan2()" function
2168 */
2169 static void
2170f_atan2(typval_T *argvars, typval_T *rettv)
2171{
2172 float_T fx = 0.0, fy = 0.0;
2173
2174 rettv->v_type = VAR_FLOAT;
2175 if (get_float_arg(argvars, &fx) == OK
2176 && get_float_arg(&argvars[1], &fy) == OK)
2177 rettv->vval.v_float = atan2(fx, fy);
2178 else
2179 rettv->vval.v_float = 0.0;
2180}
2181#endif
2182
2183/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01002184 * "balloon_show()" function
2185 */
2186#ifdef FEAT_BEVAL
2187 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002188f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
2189{
2190 rettv->v_type = VAR_STRING;
2191 if (balloonEval != NULL)
2192 {
2193 if (balloonEval->msg == NULL)
2194 rettv->vval.v_string = NULL;
2195 else
2196 rettv->vval.v_string = vim_strsave(balloonEval->msg);
2197 }
2198}
2199
2200 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01002201f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
2202{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01002203 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002204 {
2205 if (argvars[0].v_type == VAR_LIST
2206# ifdef FEAT_GUI
2207 && !gui.in_use
2208# endif
2209 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002210 {
2211 list_T *l = argvars[0].vval.v_list;
2212
2213 // empty list removes the balloon
2214 post_balloon(balloonEval, NULL,
2215 l == NULL || l->lv_len == 0 ? NULL : l);
2216 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002217 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002218 {
2219 char_u *mesg = tv_get_string_chk(&argvars[0]);
2220
2221 if (mesg != NULL)
2222 // empty string removes the balloon
2223 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
2224 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002225 }
2226}
2227
Bram Moolenaar669a8282017-11-19 20:13:05 +01002228# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002229 static void
2230f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
2231{
2232 if (rettv_list_alloc(rettv) == OK)
2233 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002234 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002235
2236 if (msg != NULL)
2237 {
2238 pumitem_T *array;
2239 int size = split_message(msg, &array);
2240 int i;
2241
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002242 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01002243 for (i = 1; i < size - 1; ++i)
2244 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01002245 while (size > 0)
2246 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002247 vim_free(array);
2248 }
2249 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01002250}
Bram Moolenaar669a8282017-11-19 20:13:05 +01002251# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01002252#endif
2253
2254/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002255 * Get the buffer from "arg" and give an error and return NULL if it is not
2256 * valid.
2257 */
Bram Moolenaara3347722019-05-11 21:14:24 +02002258 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002259get_buf_arg(typval_T *arg)
2260{
2261 buf_T *buf;
2262
2263 ++emsg_off;
2264 buf = tv_get_buf(arg, FALSE);
2265 --emsg_off;
2266 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002267 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002268 return buf;
2269}
2270
2271/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002272 * "byte2line(byte)" function
2273 */
2274 static void
2275f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
2276{
2277#ifndef FEAT_BYTEOFF
2278 rettv->vval.v_number = -1;
2279#else
2280 long boff = 0;
2281
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002282 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002283 if (boff < 0)
2284 rettv->vval.v_number = -1;
2285 else
2286 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
2287 (linenr_T)0, &boff);
2288#endif
2289}
2290
2291 static void
2292byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
2293{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002294 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002295 char_u *str;
2296 varnumber_T idx;
2297
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002298 str = tv_get_string_chk(&argvars[0]);
2299 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002300 rettv->vval.v_number = -1;
2301 if (str == NULL || idx < 0)
2302 return;
2303
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002304 t = str;
2305 for ( ; idx > 0; idx--)
2306 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002307 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002308 return;
2309 if (enc_utf8 && comp)
2310 t += utf_ptr2len(t);
2311 else
2312 t += (*mb_ptr2len)(t);
2313 }
2314 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002315}
2316
2317/*
2318 * "byteidx()" function
2319 */
2320 static void
2321f_byteidx(typval_T *argvars, typval_T *rettv)
2322{
2323 byteidx(argvars, rettv, FALSE);
2324}
2325
2326/*
2327 * "byteidxcomp()" function
2328 */
2329 static void
2330f_byteidxcomp(typval_T *argvars, typval_T *rettv)
2331{
2332 byteidx(argvars, rettv, TRUE);
2333}
2334
2335/*
2336 * "call(func, arglist [, dict])" function
2337 */
2338 static void
2339f_call(typval_T *argvars, typval_T *rettv)
2340{
2341 char_u *func;
2342 partial_T *partial = NULL;
2343 dict_T *selfdict = NULL;
2344
2345 if (argvars[1].v_type != VAR_LIST)
2346 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002347 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002348 return;
2349 }
2350 if (argvars[1].vval.v_list == NULL)
2351 return;
2352
2353 if (argvars[0].v_type == VAR_FUNC)
2354 func = argvars[0].vval.v_string;
2355 else if (argvars[0].v_type == VAR_PARTIAL)
2356 {
2357 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002358 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002359 }
2360 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002361 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002362 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002363 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002364
2365 if (argvars[2].v_type != VAR_UNKNOWN)
2366 {
2367 if (argvars[2].v_type != VAR_DICT)
2368 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002369 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002370 return;
2371 }
2372 selfdict = argvars[2].vval.v_dict;
2373 }
2374
2375 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
2376}
2377
2378#ifdef FEAT_FLOAT
2379/*
2380 * "ceil({float})" function
2381 */
2382 static void
2383f_ceil(typval_T *argvars, typval_T *rettv)
2384{
2385 float_T f = 0.0;
2386
2387 rettv->v_type = VAR_FLOAT;
2388 if (get_float_arg(argvars, &f) == OK)
2389 rettv->vval.v_float = ceil(f);
2390 else
2391 rettv->vval.v_float = 0.0;
2392}
2393#endif
2394
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002395/*
2396 * "changenr()" function
2397 */
2398 static void
2399f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
2400{
2401 rettv->vval.v_number = curbuf->b_u_seq_cur;
2402}
2403
2404/*
2405 * "char2nr(string)" function
2406 */
2407 static void
2408f_char2nr(typval_T *argvars, typval_T *rettv)
2409{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002410 if (has_mbyte)
2411 {
2412 int utf8 = 0;
2413
2414 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar24f77502020-09-04 19:50:57 +02002415 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002416
2417 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01002418 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002419 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002420 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002421 }
2422 else
Bram Moolenaar13505972019-01-24 15:04:48 +01002423 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002424}
2425
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002426/*
2427 * "charidx()" function
2428 */
2429 static void
2430f_charidx(typval_T *argvars, typval_T *rettv)
2431{
2432 char_u *str;
2433 varnumber_T idx;
2434 int countcc = FALSE;
2435 char_u *p;
2436 int len;
2437 int (*ptr2len)(char_u *);
2438
2439 rettv->vval.v_number = -1;
2440
2441 if (argvars[0].v_type != VAR_STRING || argvars[1].v_type != VAR_NUMBER
2442 || (argvars[2].v_type != VAR_UNKNOWN
2443 && argvars[2].v_type != VAR_NUMBER))
2444 {
2445 emsg(_(e_invarg));
2446 return;
2447 }
2448
2449 str = tv_get_string_chk(&argvars[0]);
2450 idx = tv_get_number_chk(&argvars[1], NULL);
2451 if (str == NULL || idx < 0)
2452 return;
2453
2454 if (argvars[2].v_type != VAR_UNKNOWN)
2455 countcc = (int)tv_get_bool(&argvars[2]);
2456 if (countcc < 0 || countcc > 1)
2457 {
2458 semsg(_(e_using_number_as_bool_nr), countcc);
2459 return;
2460 }
2461
2462 if (enc_utf8 && countcc)
2463 ptr2len = utf_ptr2len;
2464 else
2465 ptr2len = mb_ptr2len;
2466
2467 for (p = str, len = 0; p <= str + idx; len++)
2468 {
2469 if (*p == NUL)
2470 return;
2471 p += ptr2len(p);
2472 }
2473
2474 rettv->vval.v_number = len > 0 ? len - 1 : 0;
2475}
2476
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02002477 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01002478get_optional_window(typval_T *argvars, int idx)
2479{
2480 win_T *win = curwin;
2481
2482 if (argvars[idx].v_type != VAR_UNKNOWN)
2483 {
2484 win = find_win_by_nr_or_id(&argvars[idx]);
2485 if (win == NULL)
2486 {
2487 emsg(_(e_invalwindow));
2488 return NULL;
2489 }
2490 }
2491 return win;
2492}
2493
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002494/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002495 * "col(string)" function
2496 */
2497 static void
2498f_col(typval_T *argvars, typval_T *rettv)
2499{
2500 colnr_T col = 0;
2501 pos_T *fp;
2502 int fnum = curbuf->b_fnum;
2503
2504 fp = var2fpos(&argvars[0], FALSE, &fnum);
2505 if (fp != NULL && fnum == curbuf->b_fnum)
2506 {
2507 if (fp->col == MAXCOL)
2508 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002509 // '> can be MAXCOL, get the length of the line then
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002510 if (fp->lnum <= curbuf->b_ml.ml_line_count)
2511 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
2512 else
2513 col = MAXCOL;
2514 }
2515 else
2516 {
2517 col = fp->col + 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002518 // col(".") when the cursor is on the NUL at the end of the line
2519 // because of "coladd" can be seen as an extra column.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002520 if (virtual_active() && fp == &curwin->w_cursor)
2521 {
2522 char_u *p = ml_get_cursor();
2523
2524 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
2525 curwin->w_virtcol - curwin->w_cursor.coladd))
2526 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002527 int l;
2528
2529 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
2530 col += l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002531 }
2532 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002533 }
2534 }
2535 rettv->vval.v_number = col;
2536}
2537
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002538/*
2539 * "confirm(message, buttons[, default [, type]])" function
2540 */
2541 static void
2542f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2543{
2544#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2545 char_u *message;
2546 char_u *buttons = NULL;
2547 char_u buf[NUMBUFLEN];
2548 char_u buf2[NUMBUFLEN];
2549 int def = 1;
2550 int type = VIM_GENERIC;
2551 char_u *typestr;
2552 int error = FALSE;
2553
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002554 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002555 if (message == NULL)
2556 error = TRUE;
2557 if (argvars[1].v_type != VAR_UNKNOWN)
2558 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002559 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002560 if (buttons == NULL)
2561 error = TRUE;
2562 if (argvars[2].v_type != VAR_UNKNOWN)
2563 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002564 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002565 if (argvars[3].v_type != VAR_UNKNOWN)
2566 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002567 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002568 if (typestr == NULL)
2569 error = TRUE;
2570 else
2571 {
2572 switch (TOUPPER_ASC(*typestr))
2573 {
2574 case 'E': type = VIM_ERROR; break;
2575 case 'Q': type = VIM_QUESTION; break;
2576 case 'I': type = VIM_INFO; break;
2577 case 'W': type = VIM_WARNING; break;
2578 case 'G': type = VIM_GENERIC; break;
2579 }
2580 }
2581 }
2582 }
2583 }
2584
2585 if (buttons == NULL || *buttons == NUL)
2586 buttons = (char_u *)_("&Ok");
2587
2588 if (!error)
2589 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
2590 def, NULL, FALSE);
2591#endif
2592}
2593
2594/*
2595 * "copy()" function
2596 */
2597 static void
2598f_copy(typval_T *argvars, typval_T *rettv)
2599{
2600 item_copy(&argvars[0], rettv, FALSE, 0);
2601}
2602
2603#ifdef FEAT_FLOAT
2604/*
2605 * "cos()" function
2606 */
2607 static void
2608f_cos(typval_T *argvars, typval_T *rettv)
2609{
2610 float_T f = 0.0;
2611
2612 rettv->v_type = VAR_FLOAT;
2613 if (get_float_arg(argvars, &f) == OK)
2614 rettv->vval.v_float = cos(f);
2615 else
2616 rettv->vval.v_float = 0.0;
2617}
2618
2619/*
2620 * "cosh()" function
2621 */
2622 static void
2623f_cosh(typval_T *argvars, typval_T *rettv)
2624{
2625 float_T f = 0.0;
2626
2627 rettv->v_type = VAR_FLOAT;
2628 if (get_float_arg(argvars, &f) == OK)
2629 rettv->vval.v_float = cosh(f);
2630 else
2631 rettv->vval.v_float = 0.0;
2632}
2633#endif
2634
2635/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002636 * "cursor(lnum, col)" function, or
2637 * "cursor(list)"
2638 *
2639 * Moves the cursor to the specified line and column.
2640 * Returns 0 when the position could be set, -1 otherwise.
2641 */
2642 static void
2643f_cursor(typval_T *argvars, typval_T *rettv)
2644{
2645 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002646 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002647 int set_curswant = TRUE;
2648
2649 rettv->vval.v_number = -1;
2650 if (argvars[1].v_type == VAR_UNKNOWN)
2651 {
2652 pos_T pos;
2653 colnr_T curswant = -1;
2654
2655 if (list2fpos(argvars, &pos, NULL, &curswant) == FAIL)
2656 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002657 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002658 return;
2659 }
2660 line = pos.lnum;
2661 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002662 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002663 if (curswant >= 0)
2664 {
2665 curwin->w_curswant = curswant - 1;
2666 set_curswant = FALSE;
2667 }
2668 }
2669 else
2670 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002671 line = tv_get_lnum(argvars);
Bram Moolenaar9a963372020-12-21 21:58:46 +01002672 if (line < 0)
2673 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002674 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002675 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002676 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002677 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002678 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002679 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002680 if (line > 0)
2681 curwin->w_cursor.lnum = line;
2682 if (col > 0)
2683 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002684 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002685
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002686 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002687 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002688 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002689 if (has_mbyte)
2690 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002691
2692 curwin->w_set_curswant = set_curswant;
2693 rettv->vval.v_number = 0;
2694}
2695
Bram Moolenaar4f974752019-02-17 17:44:42 +01002696#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002697/*
2698 * "debugbreak()" function
2699 */
2700 static void
2701f_debugbreak(typval_T *argvars, typval_T *rettv)
2702{
2703 int pid;
2704
2705 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002706 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002707 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002708 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002709 else
2710 {
2711 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
2712
2713 if (hProcess != NULL)
2714 {
2715 DebugBreakProcess(hProcess);
2716 CloseHandle(hProcess);
2717 rettv->vval.v_number = OK;
2718 }
2719 }
2720}
2721#endif
2722
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002723/*
2724 * "deepcopy()" function
2725 */
2726 static void
2727f_deepcopy(typval_T *argvars, typval_T *rettv)
2728{
2729 int noref = 0;
2730 int copyID;
2731
2732 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar44b4a242020-09-05 17:18:28 +02002733 noref = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002734 if (noref < 0 || noref > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02002735 semsg(_(e_using_number_as_bool_nr), noref);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002736 else
2737 {
2738 copyID = get_copyID();
2739 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
2740 }
2741}
2742
2743/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002744 * "did_filetype()" function
2745 */
2746 static void
2747f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2748{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002749 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002750}
2751
2752/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01002753 * "echoraw({expr})" function
2754 */
2755 static void
2756f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
2757{
2758 char_u *str = tv_get_string_chk(&argvars[0]);
2759
2760 if (str != NULL && *str != NUL)
2761 {
2762 out_str(str);
2763 out_flush();
2764 }
2765}
2766
2767/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002768 * "empty({expr})" function
2769 */
2770 static void
2771f_empty(typval_T *argvars, typval_T *rettv)
2772{
2773 int n = FALSE;
2774
2775 switch (argvars[0].v_type)
2776 {
2777 case VAR_STRING:
2778 case VAR_FUNC:
2779 n = argvars[0].vval.v_string == NULL
2780 || *argvars[0].vval.v_string == NUL;
2781 break;
2782 case VAR_PARTIAL:
2783 n = FALSE;
2784 break;
2785 case VAR_NUMBER:
2786 n = argvars[0].vval.v_number == 0;
2787 break;
2788 case VAR_FLOAT:
2789#ifdef FEAT_FLOAT
2790 n = argvars[0].vval.v_float == 0.0;
2791 break;
2792#endif
2793 case VAR_LIST:
2794 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002795 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002796 break;
2797 case VAR_DICT:
2798 n = argvars[0].vval.v_dict == NULL
2799 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2800 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002801 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002802 case VAR_SPECIAL:
2803 n = argvars[0].vval.v_number != VVAL_TRUE;
2804 break;
2805
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002806 case VAR_BLOB:
2807 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002808 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2809 break;
2810
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002811 case VAR_JOB:
2812#ifdef FEAT_JOB_CHANNEL
2813 n = argvars[0].vval.v_job == NULL
2814 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2815 break;
2816#endif
2817 case VAR_CHANNEL:
2818#ifdef FEAT_JOB_CHANNEL
2819 n = argvars[0].vval.v_channel == NULL
2820 || !channel_is_open(argvars[0].vval.v_channel);
2821 break;
2822#endif
2823 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02002824 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002825 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002826 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002827 n = TRUE;
2828 break;
2829 }
2830
2831 rettv->vval.v_number = n;
2832}
2833
2834/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002835 * "environ()" function
2836 */
2837 static void
2838f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2839{
2840#if !defined(AMIGA)
2841 int i = 0;
2842 char_u *entry, *value;
2843# ifdef MSWIN
2844 extern wchar_t **_wenviron;
2845# else
2846 extern char **environ;
2847# endif
2848
2849 if (rettv_dict_alloc(rettv) != OK)
2850 return;
2851
2852# ifdef MSWIN
2853 if (*_wenviron == NULL)
2854 return;
2855# else
2856 if (*environ == NULL)
2857 return;
2858# endif
2859
2860 for (i = 0; ; ++i)
2861 {
2862# ifdef MSWIN
2863 short_u *p;
2864
2865 if ((p = (short_u *)_wenviron[i]) == NULL)
2866 return;
2867 entry = utf16_to_enc(p, NULL);
2868# else
2869 if ((entry = (char_u *)environ[i]) == NULL)
2870 return;
2871 entry = vim_strsave(entry);
2872# endif
2873 if (entry == NULL) // out of memory
2874 return;
2875 if ((value = vim_strchr(entry, '=')) == NULL)
2876 {
2877 vim_free(entry);
2878 continue;
2879 }
2880 *value++ = NUL;
2881 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2882 vim_free(entry);
2883 }
2884#endif
2885}
2886
2887/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002888 * "escape({string}, {chars})" function
2889 */
2890 static void
2891f_escape(typval_T *argvars, typval_T *rettv)
2892{
2893 char_u buf[NUMBUFLEN];
2894
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002895 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2896 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002897 rettv->v_type = VAR_STRING;
2898}
2899
2900/*
2901 * "eval()" function
2902 */
2903 static void
2904f_eval(typval_T *argvars, typval_T *rettv)
2905{
2906 char_u *s, *p;
2907
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002908 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002909 if (s != NULL)
2910 s = skipwhite(s);
2911
2912 p = s;
Bram Moolenaar5409f5d2020-06-24 18:37:35 +02002913 if (s == NULL || eval1(&s, rettv, &EVALARG_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002914 {
2915 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002916 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002917 need_clr_eos = FALSE;
2918 rettv->v_type = VAR_NUMBER;
2919 rettv->vval.v_number = 0;
2920 }
2921 else if (*s != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02002922 semsg(_(e_trailing_arg), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002923}
2924
2925/*
2926 * "eventhandler()" function
2927 */
2928 static void
2929f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2930{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002931 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002932}
2933
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002934static garray_T redir_execute_ga;
2935
2936/*
2937 * Append "value[value_len]" to the execute() output.
2938 */
2939 void
2940execute_redir_str(char_u *value, int value_len)
2941{
2942 int len;
2943
2944 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002945 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002946 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002947 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002948 if (ga_grow(&redir_execute_ga, len) == OK)
2949 {
2950 mch_memmove((char *)redir_execute_ga.ga_data
2951 + redir_execute_ga.ga_len, value, len);
2952 redir_execute_ga.ga_len += len;
2953 }
2954}
2955
2956/*
2957 * Get next line from a list.
2958 * Called by do_cmdline() to get the next line.
2959 * Returns allocated string, or NULL for end of function.
2960 */
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002961 static char_u *
2962get_list_line(
2963 int c UNUSED,
2964 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002965 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +02002966 getline_opt_T options UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002967{
2968 listitem_T **p = (listitem_T **)cookie;
2969 listitem_T *item = *p;
2970 char_u buf[NUMBUFLEN];
2971 char_u *s;
2972
2973 if (item == NULL)
2974 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002975 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002976 *p = item->li_next;
2977 return s == NULL ? NULL : vim_strsave(s);
2978}
2979
2980/*
2981 * "execute()" function
2982 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002983 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002984execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002985{
2986 char_u *cmd = NULL;
2987 list_T *list = NULL;
2988 int save_msg_silent = msg_silent;
2989 int save_emsg_silent = emsg_silent;
2990 int save_emsg_noredir = emsg_noredir;
2991 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002992 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002993 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002994 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002995 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002996
2997 rettv->vval.v_string = NULL;
2998 rettv->v_type = VAR_STRING;
2999
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003000 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003001 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003002 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003003 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003004 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003005 return;
3006 ++list->lv_refcount;
3007 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01003008 else if (argvars[arg_off].v_type == VAR_JOB
3009 || argvars[arg_off].v_type == VAR_CHANNEL)
3010 {
3011 emsg(_(e_inval_string));
3012 return;
3013 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003014 else
3015 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003016 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003017 if (cmd == NULL)
3018 return;
3019 }
3020
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003021 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003022 {
3023 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003024 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003025
3026 if (s == NULL)
3027 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003028 if (*s == NUL)
3029 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003030 if (STRNCMP(s, "silent", 6) == 0)
3031 ++msg_silent;
3032 if (STRCMP(s, "silent!") == 0)
3033 {
3034 emsg_silent = TRUE;
3035 emsg_noredir = TRUE;
3036 }
3037 }
3038 else
3039 ++msg_silent;
3040
3041 if (redir_execute)
3042 save_ga = redir_execute_ga;
3043 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
3044 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01003045 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003046 if (!echo_output)
3047 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003048
3049 if (cmd != NULL)
3050 do_cmdline_cmd(cmd);
3051 else
3052 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003053 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003054
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003055 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003056 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003057 do_cmdline(NULL, get_list_line, (void *)&item,
3058 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
3059 --list->lv_refcount;
3060 }
3061
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003062 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01003063 if (ga_grow(&redir_execute_ga, 1) == OK)
3064 {
3065 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
3066 rettv->vval.v_string = redir_execute_ga.ga_data;
3067 }
3068 else
3069 {
3070 ga_clear(&redir_execute_ga);
3071 rettv->vval.v_string = NULL;
3072 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003073 msg_silent = save_msg_silent;
3074 emsg_silent = save_emsg_silent;
3075 emsg_noredir = save_emsg_noredir;
3076
3077 redir_execute = save_redir_execute;
3078 if (redir_execute)
3079 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01003080 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003081
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003082 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003083 if (echo_output)
3084 // When not working silently: put it in column zero. A following
3085 // "echon" will overwrite the message, unavoidably.
3086 msg_col = 0;
3087 else
3088 // When working silently: Put it back where it was, since nothing
3089 // should have been written.
3090 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003091}
3092
3093/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003094 * "execute()" function
3095 */
3096 static void
3097f_execute(typval_T *argvars, typval_T *rettv)
3098{
3099 execute_common(argvars, rettv, 0);
3100}
3101
3102/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003103 * "exists()" function
3104 */
3105 static void
3106f_exists(typval_T *argvars, typval_T *rettv)
3107{
3108 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003109 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003110
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003111 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003112 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003113 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003114 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003115 if (mch_getenv(p + 1) != NULL)
3116 n = TRUE;
3117 else
3118 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003119 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003120 p = expand_env_save(p);
3121 if (p != NULL && *p != '$')
3122 n = TRUE;
3123 vim_free(p);
3124 }
3125 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003126 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003127 {
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +02003128 n = (eval_option(&p, NULL, TRUE) == OK);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003129 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003130 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003131 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003132 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003133 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02003134 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003135 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01003136 else if (*p == '?') // internal function only
3137 {
3138 n = has_internal_func_name(p + 1);
3139 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003140 else if (*p == ':')
3141 {
3142 n = cmd_exists(p + 1);
3143 }
3144 else if (*p == '#')
3145 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003146 if (p[1] == '#')
3147 n = autocmd_supported(p + 2);
3148 else
3149 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003150 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003151 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003152 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01003153 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003154 }
3155
3156 rettv->vval.v_number = n;
3157}
3158
3159#ifdef FEAT_FLOAT
3160/*
3161 * "exp()" function
3162 */
3163 static void
3164f_exp(typval_T *argvars, typval_T *rettv)
3165{
3166 float_T f = 0.0;
3167
3168 rettv->v_type = VAR_FLOAT;
3169 if (get_float_arg(argvars, &f) == OK)
3170 rettv->vval.v_float = exp(f);
3171 else
3172 rettv->vval.v_float = 0.0;
3173}
3174#endif
3175
3176/*
3177 * "expand()" function
3178 */
3179 static void
3180f_expand(typval_T *argvars, typval_T *rettv)
3181{
3182 char_u *s;
3183 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003184 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003185 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
3186 expand_T xpc;
3187 int error = FALSE;
3188 char_u *result;
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003189#ifdef BACKSLASH_IN_FILENAME
3190 char_u *p_csl_save = p_csl;
3191
3192 // avoid using 'completeslash' here
3193 p_csl = empty_option;
3194#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003195
3196 rettv->v_type = VAR_STRING;
3197 if (argvars[1].v_type != VAR_UNKNOWN
3198 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003199 && tv_get_bool_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003200 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02003201 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003202
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003203 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003204 if (*s == '%' || *s == '#' || *s == '<')
3205 {
3206 ++emsg_off;
3207 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
3208 --emsg_off;
3209 if (rettv->v_type == VAR_LIST)
3210 {
3211 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
3212 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02003213 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003214 }
3215 else
3216 rettv->vval.v_string = result;
3217 }
3218 else
3219 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003220 // When the optional second argument is non-zero, don't remove matches
3221 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003222 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003223 && tv_get_bool_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003224 options |= WILD_KEEP_ALL;
3225 if (!error)
3226 {
3227 ExpandInit(&xpc);
3228 xpc.xp_context = EXPAND_FILES;
3229 if (p_wic)
3230 options += WILD_ICASE;
3231 if (rettv->v_type == VAR_STRING)
3232 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
3233 options, WILD_ALL);
3234 else if (rettv_list_alloc(rettv) != FAIL)
3235 {
3236 int i;
3237
3238 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
3239 for (i = 0; i < xpc.xp_numfiles; i++)
3240 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
3241 ExpandCleanup(&xpc);
3242 }
3243 }
3244 else
3245 rettv->vval.v_string = NULL;
3246 }
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003247#ifdef BACKSLASH_IN_FILENAME
3248 p_csl = p_csl_save;
3249#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003250}
3251
3252/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02003253 * "expandcmd()" function
3254 * Expand all the special characters in a command string.
3255 */
3256 static void
3257f_expandcmd(typval_T *argvars, typval_T *rettv)
3258{
3259 exarg_T eap;
3260 char_u *cmdstr;
3261 char *errormsg = NULL;
3262
3263 rettv->v_type = VAR_STRING;
3264 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
3265
3266 memset(&eap, 0, sizeof(eap));
3267 eap.cmd = cmdstr;
3268 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02003269 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02003270 eap.usefilter = FALSE;
3271 eap.nextcmd = NULL;
3272 eap.cmdidx = CMD_USER;
3273
3274 expand_filename(&eap, &cmdstr, &errormsg);
3275 if (errormsg != NULL && *errormsg != NUL)
3276 emsg(errormsg);
3277
3278 rettv->vval.v_string = cmdstr;
3279}
3280
3281/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003282 * "feedkeys()" function
3283 */
3284 static void
3285f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
3286{
3287 int remap = TRUE;
3288 int insert = FALSE;
3289 char_u *keys, *flags;
3290 char_u nbuf[NUMBUFLEN];
3291 int typed = FALSE;
3292 int execute = FALSE;
3293 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003294 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003295 char_u *keys_esc;
3296
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003297 // This is not allowed in the sandbox. If the commands would still be
3298 // executed in the sandbox it would be OK, but it probably happens later,
3299 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003300 if (check_secure())
3301 return;
3302
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003303 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003304
3305 if (argvars[1].v_type != VAR_UNKNOWN)
3306 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003307 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003308 for ( ; *flags != NUL; ++flags)
3309 {
3310 switch (*flags)
3311 {
3312 case 'n': remap = FALSE; break;
3313 case 'm': remap = TRUE; break;
3314 case 't': typed = TRUE; break;
3315 case 'i': insert = TRUE; break;
3316 case 'x': execute = TRUE; break;
3317 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003318 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003319 }
3320 }
3321 }
3322
3323 if (*keys != NUL || execute)
3324 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003325 // Need to escape K_SPECIAL and CSI before putting the string in the
3326 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003327 keys_esc = vim_strsave_escape_csi(keys);
3328 if (keys_esc != NULL)
3329 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003330 if (lowlevel)
3331 {
3332#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01003333 int idx;
3334 int len = (int)STRLEN(keys);
3335
3336 for (idx = 0; idx < len; ++idx)
3337 {
3338 // if a CTRL-C was typed, set got_int, similar to what
3339 // happens in fill_input_buf()
3340 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
3341 got_int = TRUE;
3342 add_to_input_buf(keys + idx, 1);
3343 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003344#else
3345 emsg(_("E980: lowlevel input not supported"));
3346#endif
3347 }
3348 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003349 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003350 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003351 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003352 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003353#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003354 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003355#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003356 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003357 typebuf_was_filled = TRUE;
3358 }
3359 vim_free(keys_esc);
3360
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003361 if (execute)
3362 {
3363 int save_msg_scroll = msg_scroll;
3364
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003365 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003366 msg_scroll = FALSE;
3367
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003368 if (!dangerous)
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003369 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003370 ++ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003371 ++in_feedkeys;
3372 }
Bram Moolenaar905dd902019-04-07 14:21:47 +02003373 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003374 if (!dangerous)
Bram Moolenaar189832b2020-09-23 12:29:11 +02003375 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003376 --ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003377 --in_feedkeys;
Bram Moolenaar189832b2020-09-23 12:29:11 +02003378 }
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003379
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003380 msg_scroll |= save_msg_scroll;
3381 }
3382 }
3383 }
3384}
3385
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003386#ifdef FEAT_FLOAT
3387/*
3388 * "float2nr({float})" function
3389 */
3390 static void
3391f_float2nr(typval_T *argvars, typval_T *rettv)
3392{
3393 float_T f = 0.0;
3394
3395 if (get_float_arg(argvars, &f) == OK)
3396 {
Bram Moolenaar37184272020-05-23 19:30:05 +02003397 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003398 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02003399 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003400 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003401 else
3402 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003403 }
3404}
3405
3406/*
3407 * "floor({float})" function
3408 */
3409 static void
3410f_floor(typval_T *argvars, typval_T *rettv)
3411{
3412 float_T f = 0.0;
3413
3414 rettv->v_type = VAR_FLOAT;
3415 if (get_float_arg(argvars, &f) == OK)
3416 rettv->vval.v_float = floor(f);
3417 else
3418 rettv->vval.v_float = 0.0;
3419}
3420
3421/*
3422 * "fmod()" function
3423 */
3424 static void
3425f_fmod(typval_T *argvars, typval_T *rettv)
3426{
3427 float_T fx = 0.0, fy = 0.0;
3428
3429 rettv->v_type = VAR_FLOAT;
3430 if (get_float_arg(argvars, &fx) == OK
3431 && get_float_arg(&argvars[1], &fy) == OK)
3432 rettv->vval.v_float = fmod(fx, fy);
3433 else
3434 rettv->vval.v_float = 0.0;
3435}
3436#endif
3437
3438/*
3439 * "fnameescape({string})" function
3440 */
3441 static void
3442f_fnameescape(typval_T *argvars, typval_T *rettv)
3443{
3444 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003445 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003446 rettv->v_type = VAR_STRING;
3447}
3448
3449/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003450 * "foreground()" function
3451 */
3452 static void
3453f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3454{
3455#ifdef FEAT_GUI
3456 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003457 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003458 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003459 return;
3460 }
3461#endif
3462#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003463 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003464#endif
3465}
3466
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003467 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003468common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003469{
3470 char_u *s;
3471 char_u *name;
3472 int use_string = FALSE;
3473 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003474 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003475 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003476
3477 if (argvars[0].v_type == VAR_FUNC)
3478 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003479 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003480 s = argvars[0].vval.v_string;
3481 }
3482 else if (argvars[0].v_type == VAR_PARTIAL
3483 && argvars[0].vval.v_partial != NULL)
3484 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003485 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003486 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003487 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003488 }
3489 else
3490 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003491 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003492 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003493 use_string = TRUE;
3494 }
3495
Bram Moolenaar843b8842016-08-21 14:36:15 +02003496 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003497 {
3498 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003499 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003500 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
3501 if (*name != NUL)
3502 s = NULL;
3503 }
3504
Bram Moolenaar843b8842016-08-21 14:36:15 +02003505 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
3506 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003507 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003508 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003509 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003510 ? find_func(trans_name, is_global, NULL) == NULL
3511 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003512 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003513 else
3514 {
3515 int dict_idx = 0;
3516 int arg_idx = 0;
3517 list_T *list = NULL;
3518
3519 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
3520 {
3521 char sid_buf[25];
3522 int off = *s == 's' ? 2 : 5;
3523
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003524 // Expand s: and <SID> into <SNR>nr_, so that the function can
3525 // also be called from another script. Using trans_function_name()
3526 // would also work, but some plugins depend on the name being
3527 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02003528 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02003529 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003530 if (name != NULL)
3531 {
3532 STRCPY(name, sid_buf);
3533 STRCAT(name, s + off);
3534 }
3535 }
3536 else
3537 name = vim_strsave(s);
3538
3539 if (argvars[1].v_type != VAR_UNKNOWN)
3540 {
3541 if (argvars[2].v_type != VAR_UNKNOWN)
3542 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003543 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003544 arg_idx = 1;
3545 dict_idx = 2;
3546 }
3547 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003548 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003549 dict_idx = 1;
3550 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003551 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003552 arg_idx = 1;
3553 if (dict_idx > 0)
3554 {
3555 if (argvars[dict_idx].v_type != VAR_DICT)
3556 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003557 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003558 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003559 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003560 }
3561 if (argvars[dict_idx].vval.v_dict == NULL)
3562 dict_idx = 0;
3563 }
3564 if (arg_idx > 0)
3565 {
3566 if (argvars[arg_idx].v_type != VAR_LIST)
3567 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003568 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003569 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003570 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003571 }
3572 list = argvars[arg_idx].vval.v_list;
3573 if (list == NULL || list->lv_len == 0)
3574 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003575 else if (list->lv_len > MAX_FUNC_ARGS)
3576 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01003577 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003578 vim_free(name);
3579 goto theend;
3580 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003581 }
3582 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003583 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003584 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003585 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003586
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003587 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003588 if (pt == NULL)
3589 vim_free(name);
3590 else
3591 {
3592 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
3593 {
3594 listitem_T *li;
3595 int i = 0;
3596 int arg_len = 0;
3597 int lv_len = 0;
3598
3599 if (arg_pt != NULL)
3600 arg_len = arg_pt->pt_argc;
3601 if (list != NULL)
3602 lv_len = list->lv_len;
3603 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003604 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003605 if (pt->pt_argv == NULL)
3606 {
3607 vim_free(pt);
3608 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003609 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003610 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003611 for (i = 0; i < arg_len; i++)
3612 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
3613 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003614 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003615 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003616 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003617 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003618 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003619 }
3620
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003621 // For "function(dict.func, [], dict)" and "func" is a partial
3622 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003623 if (dict_idx > 0)
3624 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003625 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003626 pt->pt_dict = argvars[dict_idx].vval.v_dict;
3627 ++pt->pt_dict->dv_refcount;
3628 }
3629 else if (arg_pt != NULL)
3630 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003631 // If the dict was bound automatically the result is also
3632 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003633 pt->pt_dict = arg_pt->pt_dict;
3634 pt->pt_auto = arg_pt->pt_auto;
3635 if (pt->pt_dict != NULL)
3636 ++pt->pt_dict->dv_refcount;
3637 }
3638
3639 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003640 if (arg_pt != NULL && arg_pt->pt_func != NULL)
3641 {
3642 pt->pt_func = arg_pt->pt_func;
3643 func_ptr_ref(pt->pt_func);
3644 vim_free(name);
3645 }
3646 else if (is_funcref)
3647 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003648 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003649 func_ptr_ref(pt->pt_func);
3650 vim_free(name);
3651 }
3652 else
3653 {
3654 pt->pt_name = name;
3655 func_ref(name);
3656 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003657 }
3658 rettv->v_type = VAR_PARTIAL;
3659 rettv->vval.v_partial = pt;
3660 }
3661 else
3662 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003663 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003664 rettv->v_type = VAR_FUNC;
3665 rettv->vval.v_string = name;
3666 func_ref(name);
3667 }
3668 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003669theend:
3670 vim_free(trans_name);
3671}
3672
3673/*
3674 * "funcref()" function
3675 */
3676 static void
3677f_funcref(typval_T *argvars, typval_T *rettv)
3678{
3679 common_function(argvars, rettv, TRUE);
3680}
3681
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003682 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003683ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003684{
3685 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
3686 return &t_func_any;
Bram Moolenaar5e654232020-09-16 15:22:00 +02003687 return &t_func_unknown;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003688}
3689
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003690/*
3691 * "function()" function
3692 */
3693 static void
3694f_function(typval_T *argvars, typval_T *rettv)
3695{
3696 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003697}
3698
3699/*
3700 * "garbagecollect()" function
3701 */
3702 static void
3703f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
3704{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003705 // This is postponed until we are back at the toplevel, because we may be
3706 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003707 want_garbage_collect = TRUE;
3708
Bram Moolenaar2df47312020-09-05 17:30:44 +02003709 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003710 garbage_collect_at_exit = TRUE;
3711}
3712
3713/*
3714 * "get()" function
3715 */
3716 static void
3717f_get(typval_T *argvars, typval_T *rettv)
3718{
3719 listitem_T *li;
3720 list_T *l;
3721 dictitem_T *di;
3722 dict_T *d;
3723 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003724 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003725
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003726 if (argvars[0].v_type == VAR_BLOB)
3727 {
3728 int error = FALSE;
3729 int idx = tv_get_number_chk(&argvars[1], &error);
3730
3731 if (!error)
3732 {
3733 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003734 if (idx < 0)
3735 idx = blob_len(argvars[0].vval.v_blob) + idx;
3736 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
3737 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003738 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003739 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003740 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003741 tv = rettv;
3742 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003743 }
3744 }
3745 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003746 {
3747 if ((l = argvars[0].vval.v_list) != NULL)
3748 {
3749 int error = FALSE;
3750
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003751 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003752 if (!error && li != NULL)
3753 tv = &li->li_tv;
3754 }
3755 }
3756 else if (argvars[0].v_type == VAR_DICT)
3757 {
3758 if ((d = argvars[0].vval.v_dict) != NULL)
3759 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003760 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003761 if (di != NULL)
3762 tv = &di->di_tv;
3763 }
3764 }
3765 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
3766 {
3767 partial_T *pt;
3768 partial_T fref_pt;
3769
3770 if (argvars[0].v_type == VAR_PARTIAL)
3771 pt = argvars[0].vval.v_partial;
3772 else
3773 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02003774 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003775 fref_pt.pt_name = argvars[0].vval.v_string;
3776 pt = &fref_pt;
3777 }
3778
3779 if (pt != NULL)
3780 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003781 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003782 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003783
3784 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
3785 {
3786 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003787 n = partial_name(pt);
3788 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003789 rettv->vval.v_string = NULL;
3790 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003791 {
3792 rettv->vval.v_string = vim_strsave(n);
3793 if (rettv->v_type == VAR_FUNC)
3794 func_ref(rettv->vval.v_string);
3795 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003796 }
3797 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003798 {
3799 what_is_dict = TRUE;
3800 if (pt->pt_dict != NULL)
3801 rettv_dict_set(rettv, pt->pt_dict);
3802 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003803 else if (STRCMP(what, "args") == 0)
3804 {
3805 rettv->v_type = VAR_LIST;
3806 if (rettv_list_alloc(rettv) == OK)
3807 {
3808 int i;
3809
3810 for (i = 0; i < pt->pt_argc; ++i)
3811 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3812 }
3813 }
3814 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003815 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003816
3817 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3818 // third argument
3819 if (!what_is_dict)
3820 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003821 }
3822 }
3823 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003824 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003825
3826 if (tv == NULL)
3827 {
3828 if (argvars[2].v_type != VAR_UNKNOWN)
3829 copy_tv(&argvars[2], rettv);
3830 }
3831 else
3832 copy_tv(tv, rettv);
3833}
3834
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003835/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003836 * "getchangelist()" function
3837 */
3838 static void
3839f_getchangelist(typval_T *argvars, typval_T *rettv)
3840{
3841#ifdef FEAT_JUMPLIST
3842 buf_T *buf;
3843 int i;
3844 list_T *l;
3845 dict_T *d;
3846#endif
3847
3848 if (rettv_list_alloc(rettv) != OK)
3849 return;
3850
3851#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003852 if (argvars[0].v_type == VAR_UNKNOWN)
3853 buf = curbuf;
3854 else
Bram Moolenaara5d38412020-09-02 21:02:35 +02003855 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003856 if (buf == NULL)
3857 return;
3858
3859 l = list_alloc();
3860 if (l == NULL)
3861 return;
3862
3863 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3864 return;
3865 /*
3866 * The current window change list index tracks only the position in the
3867 * current buffer change list. For other buffers, use the change list
3868 * length as the current index.
3869 */
3870 list_append_number(rettv->vval.v_list,
3871 (varnumber_T)((buf == curwin->w_buffer)
3872 ? curwin->w_changelistidx : buf->b_changelistlen));
3873
3874 for (i = 0; i < buf->b_changelistlen; ++i)
3875 {
3876 if (buf->b_changelist[i].lnum == 0)
3877 continue;
3878 if ((d = dict_alloc()) == NULL)
3879 return;
3880 if (list_append_dict(l, d) == FAIL)
3881 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003882 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3883 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003884 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003885 }
3886#endif
3887}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003888
3889/*
3890 * "getcharsearch()" function
3891 */
3892 static void
3893f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3894{
3895 if (rettv_dict_alloc(rettv) != FAIL)
3896 {
3897 dict_T *dict = rettv->vval.v_dict;
3898
Bram Moolenaare0be1672018-07-08 16:50:37 +02003899 dict_add_string(dict, "char", last_csearch());
3900 dict_add_number(dict, "forward", last_csearch_forward());
3901 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003902 }
3903}
3904
3905/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003906 * "getenv()" function
3907 */
3908 static void
3909f_getenv(typval_T *argvars, typval_T *rettv)
3910{
3911 int mustfree = FALSE;
3912 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3913
3914 if (p == NULL)
3915 {
3916 rettv->v_type = VAR_SPECIAL;
3917 rettv->vval.v_number = VVAL_NULL;
3918 return;
3919 }
3920 if (!mustfree)
3921 p = vim_strsave(p);
3922 rettv->vval.v_string = p;
3923 rettv->v_type = VAR_STRING;
3924}
3925
3926/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003927 * "getfontname()" function
3928 */
3929 static void
3930f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3931{
3932 rettv->v_type = VAR_STRING;
3933 rettv->vval.v_string = NULL;
3934#ifdef FEAT_GUI
3935 if (gui.in_use)
3936 {
3937 GuiFont font;
3938 char_u *name = NULL;
3939
3940 if (argvars[0].v_type == VAR_UNKNOWN)
3941 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003942 // Get the "Normal" font. Either the name saved by
3943 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003944 font = gui.norm_font;
3945 name = hl_get_font_name();
3946 }
3947 else
3948 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003949 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003950 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003951 return;
3952 font = gui_mch_get_font(name, FALSE);
3953 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003954 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003955 }
3956 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3957 if (argvars[0].v_type != VAR_UNKNOWN)
3958 gui_mch_free_font(font);
3959 }
3960#endif
3961}
3962
3963/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003964 * "getjumplist()" function
3965 */
3966 static void
3967f_getjumplist(typval_T *argvars, typval_T *rettv)
3968{
3969#ifdef FEAT_JUMPLIST
3970 win_T *wp;
3971 int i;
3972 list_T *l;
3973 dict_T *d;
3974#endif
3975
3976 if (rettv_list_alloc(rettv) != OK)
3977 return;
3978
3979#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003980 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003981 if (wp == NULL)
3982 return;
3983
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003984 cleanup_jumplist(wp, TRUE);
3985
Bram Moolenaar4f505882018-02-10 21:06:32 +01003986 l = list_alloc();
3987 if (l == NULL)
3988 return;
3989
3990 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3991 return;
3992 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3993
3994 for (i = 0; i < wp->w_jumplistlen; ++i)
3995 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003996 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3997 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003998 if ((d = dict_alloc()) == NULL)
3999 return;
4000 if (list_append_dict(l, d) == FAIL)
4001 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02004002 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
4003 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004004 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004005 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004006 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02004007 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004008 }
4009#endif
4010}
4011
4012/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004013 * "getpid()" function
4014 */
4015 static void
4016f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
4017{
4018 rettv->vval.v_number = mch_get_pid();
4019}
4020
4021 static void
4022getpos_both(
4023 typval_T *argvars,
4024 typval_T *rettv,
4025 int getcurpos)
4026{
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004027 pos_T *fp = NULL;
4028 win_T *wp = curwin;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004029 list_T *l;
4030 int fnum = -1;
4031
4032 if (rettv_list_alloc(rettv) == OK)
4033 {
4034 l = rettv->vval.v_list;
4035 if (getcurpos)
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004036 {
4037 if (argvars[0].v_type != VAR_UNKNOWN)
4038 {
4039 wp = find_win_by_nr_or_id(&argvars[0]);
4040 if (wp != NULL)
4041 fp = &wp->w_cursor;
4042 }
4043 else
4044 fp = &curwin->w_cursor;
4045 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004046 else
4047 fp = var2fpos(&argvars[0], TRUE, &fnum);
4048 if (fnum != -1)
4049 list_append_number(l, (varnumber_T)fnum);
4050 else
4051 list_append_number(l, (varnumber_T)0);
4052 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
4053 : (varnumber_T)0);
4054 list_append_number(l, (fp != NULL)
4055 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
4056 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01004057 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004058 (varnumber_T)0);
4059 if (getcurpos)
4060 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01004061 int save_set_curswant = curwin->w_set_curswant;
4062 colnr_T save_curswant = curwin->w_curswant;
4063 colnr_T save_virtcol = curwin->w_virtcol;
4064
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004065 if (wp == curwin)
4066 update_curswant();
4067 list_append_number(l, wp == NULL ? 0 : wp->w_curswant == MAXCOL
4068 ? (varnumber_T)MAXCOL : (varnumber_T)wp->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01004069
4070 // Do not change "curswant", as it is unexpected that a get
4071 // function has a side effect.
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004072 if (wp == curwin && save_set_curswant)
Bram Moolenaar19a66852019-03-07 11:25:32 +01004073 {
4074 curwin->w_set_curswant = save_set_curswant;
4075 curwin->w_curswant = save_curswant;
4076 curwin->w_virtcol = save_virtcol;
4077 curwin->w_valid &= ~VALID_VIRTCOL;
4078 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004079 }
4080 }
4081 else
4082 rettv->vval.v_number = FALSE;
4083}
4084
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004085/*
4086 * "getcurpos()" function
4087 */
4088 static void
4089f_getcurpos(typval_T *argvars, typval_T *rettv)
4090{
4091 getpos_both(argvars, rettv, TRUE);
4092}
4093
4094/*
4095 * "getpos(string)" function
4096 */
4097 static void
4098f_getpos(typval_T *argvars, typval_T *rettv)
4099{
4100 getpos_both(argvars, rettv, FALSE);
4101}
4102
4103/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004104 * "getreg()" function
4105 */
4106 static void
4107f_getreg(typval_T *argvars, typval_T *rettv)
4108{
4109 char_u *strregname;
4110 int regname;
4111 int arg2 = FALSE;
4112 int return_list = FALSE;
4113 int error = FALSE;
4114
4115 if (argvars[0].v_type != VAR_UNKNOWN)
4116 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004117 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004118 error = strregname == NULL;
4119 if (argvars[1].v_type != VAR_UNKNOWN)
4120 {
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004121 arg2 = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004122 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004123 return_list = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004124 }
4125 }
4126 else
4127 strregname = get_vim_var_str(VV_REG);
4128
4129 if (error)
4130 return;
4131
4132 regname = (strregname == NULL ? '"' : *strregname);
4133 if (regname == 0)
4134 regname = '"';
4135
4136 if (return_list)
4137 {
4138 rettv->v_type = VAR_LIST;
4139 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
4140 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
4141 if (rettv->vval.v_list == NULL)
4142 (void)rettv_list_alloc(rettv);
4143 else
4144 ++rettv->vval.v_list->lv_refcount;
4145 }
4146 else
4147 {
4148 rettv->v_type = VAR_STRING;
4149 rettv->vval.v_string = get_reg_contents(regname,
4150 arg2 ? GREG_EXPR_SRC : 0);
4151 }
4152}
4153
4154/*
4155 * "getregtype()" function
4156 */
4157 static void
4158f_getregtype(typval_T *argvars, typval_T *rettv)
4159{
4160 char_u *strregname;
4161 int regname;
4162 char_u buf[NUMBUFLEN + 2];
4163 long reglen = 0;
4164
4165 if (argvars[0].v_type != VAR_UNKNOWN)
4166 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004167 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004168 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004169 {
4170 rettv->v_type = VAR_STRING;
4171 rettv->vval.v_string = NULL;
4172 return;
4173 }
4174 }
4175 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004176 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004177 strregname = get_vim_var_str(VV_REG);
4178
4179 regname = (strregname == NULL ? '"' : *strregname);
4180 if (regname == 0)
4181 regname = '"';
4182
4183 buf[0] = NUL;
4184 buf[1] = NUL;
4185 switch (get_reg_type(regname, &reglen))
4186 {
4187 case MLINE: buf[0] = 'V'; break;
4188 case MCHAR: buf[0] = 'v'; break;
4189 case MBLOCK:
4190 buf[0] = Ctrl_V;
4191 sprintf((char *)buf + 1, "%ld", reglen + 1);
4192 break;
4193 }
4194 rettv->v_type = VAR_STRING;
4195 rettv->vval.v_string = vim_strsave(buf);
4196}
4197
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02004198/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01004199 * "gettagstack()" function
4200 */
4201 static void
4202f_gettagstack(typval_T *argvars, typval_T *rettv)
4203{
4204 win_T *wp = curwin; // default is current window
4205
4206 if (rettv_dict_alloc(rettv) != OK)
4207 return;
4208
4209 if (argvars[0].v_type != VAR_UNKNOWN)
4210 {
4211 wp = find_win_by_nr_or_id(&argvars[0]);
4212 if (wp == NULL)
4213 return;
4214 }
4215
4216 get_tagstack(wp, rettv->vval.v_dict);
4217}
4218
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +02004219/*
4220 * "gettext()" function
4221 */
4222 static void
4223f_gettext(typval_T *argvars, typval_T *rettv)
4224{
4225 if (argvars[0].v_type != VAR_STRING
4226 || argvars[0].vval.v_string == NULL
4227 || *argvars[0].vval.v_string == NUL)
4228 {
4229 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
4230 }
4231 else
4232 {
4233 rettv->v_type = VAR_STRING;
4234 rettv->vval.v_string = vim_strsave(
4235 (char_u *)_(argvars[0].vval.v_string));
4236 }
4237}
4238
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004239// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004240#include "version.h"
4241
4242/*
4243 * "has()" function
4244 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01004245 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004246f_has(typval_T *argvars, typval_T *rettv)
4247{
4248 int i;
4249 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01004250 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004251 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01004252 typedef struct {
4253 char *name;
4254 short present;
4255 } has_item_T;
4256 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004257 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004258 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004259#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01004260 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004261#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004262 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004263#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004264 },
4265 {"arp",
4266#if defined(AMIGA) && defined(FEAT_ARP)
4267 1
4268#else
4269 0
4270#endif
4271 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004272 {"haiku",
4273#ifdef __HAIKU__
4274 1
4275#else
4276 0
4277#endif
4278 },
4279 {"bsd",
4280#if defined(BSD) && !defined(MACOS_X)
4281 1
4282#else
4283 0
4284#endif
4285 },
4286 {"hpux",
4287#ifdef hpux
4288 1
4289#else
4290 0
4291#endif
4292 },
4293 {"linux",
4294#ifdef __linux__
4295 1
4296#else
4297 0
4298#endif
4299 },
4300 {"mac", // Mac OS X (and, once, Mac OS Classic)
4301#ifdef MACOS_X
4302 1
4303#else
4304 0
4305#endif
4306 },
4307 {"osx", // Mac OS X
4308#ifdef MACOS_X
4309 1
4310#else
4311 0
4312#endif
4313 },
4314 {"macunix", // Mac OS X, with the darwin feature
4315#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4316 1
4317#else
4318 0
4319#endif
4320 },
4321 {"osxdarwin", // synonym for macunix
4322#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4323 1
4324#else
4325 0
4326#endif
4327 },
4328 {"qnx",
4329#ifdef __QNX__
4330 1
4331#else
4332 0
4333#endif
4334 },
4335 {"sun",
4336#ifdef SUN_SYSTEM
4337 1
4338#else
4339 0
4340#endif
4341 },
4342 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004343#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01004344 1
4345#else
4346 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004347#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004348 },
4349 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004350#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01004351 1
4352#else
4353 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004354#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004355 },
4356 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01004357#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01004358 1
4359#else
4360 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004361#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004362 },
4363 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01004364#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01004365 1
4366#else
4367 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004368#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004369 },
4370 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01004371#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01004372 1
4373#else
4374 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004375#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004376 },
4377 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004378#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004379 1
4380#else
4381 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004382#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004383 },
4384 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004385#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004386 1
4387#else
4388 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004389#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004390 },
4391 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004392#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01004393 1
4394#else
4395 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004396#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004397 },
4398 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004399#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004400 1
4401#else
4402 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004403#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004404 },
4405 {"autocmd", 1},
4406 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004407#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01004408 1
4409#else
4410 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004411#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004412 },
4413 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004414#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004415 1
4416#else
4417 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004418#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004419 },
4420 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004421#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01004422 1
4423#else
4424 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004425#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004426 },
4427 {"balloon_multiline",
4428#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
4429 // MS-Windows requires runtime check, see below
4430 1
4431#else
4432 0
4433#endif
4434 },
4435 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004436#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01004437 1
4438#else
4439 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01004440#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004441 },
4442 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004443#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01004444 1
4445#else
4446 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004447#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004448 },
4449 {"all_builtin_terms",
4450#if defined(ALL_BUILTIN_TCAPS)
4451 1
4452#else
4453 0
4454#endif
4455 },
4456 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004457#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01004458 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004459 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01004460 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004461#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004462 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004463#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004464 },
4465 {"byte_offset",
4466#ifdef FEAT_BYTEOFF
4467 1
4468#else
4469 0
4470#endif
4471 },
4472 {"channel",
4473#ifdef FEAT_JOB_CHANNEL
4474 1
4475#else
4476 0
4477#endif
4478 },
4479 {"cindent",
4480#ifdef FEAT_CINDENT
4481 1
4482#else
4483 0
4484#endif
4485 },
4486 {"clientserver",
4487#ifdef FEAT_CLIENTSERVER
4488 1
4489#else
4490 0
4491#endif
4492 },
4493 {"clipboard",
4494#ifdef FEAT_CLIPBOARD
4495 1
4496#else
4497 0
4498#endif
4499 },
4500 {"cmdline_compl", 1},
4501 {"cmdline_hist", 1},
4502 {"comments", 1},
4503 {"conceal",
4504#ifdef FEAT_CONCEAL
4505 1
4506#else
4507 0
4508#endif
4509 },
4510 {"cryptv",
4511#ifdef FEAT_CRYPT
4512 1
4513#else
4514 0
4515#endif
4516 },
4517 {"crypt-blowfish",
4518#ifdef FEAT_CRYPT
4519 1
4520#else
4521 0
4522#endif
4523 },
4524 {"crypt-blowfish2",
4525#ifdef FEAT_CRYPT
4526 1
4527#else
4528 0
4529#endif
4530 },
4531 {"cscope",
4532#ifdef FEAT_CSCOPE
4533 1
4534#else
4535 0
4536#endif
4537 },
4538 {"cursorbind", 1},
4539 {"cursorshape",
4540#ifdef CURSOR_SHAPE
4541 1
4542#else
4543 0
4544#endif
4545 },
4546 {"debug",
4547#ifdef DEBUG
4548 1
4549#else
4550 0
4551#endif
4552 },
4553 {"dialog_con",
4554#ifdef FEAT_CON_DIALOG
4555 1
4556#else
4557 0
4558#endif
4559 },
4560 {"dialog_gui",
4561#ifdef FEAT_GUI_DIALOG
4562 1
4563#else
4564 0
4565#endif
4566 },
4567 {"diff",
4568#ifdef FEAT_DIFF
4569 1
4570#else
4571 0
4572#endif
4573 },
4574 {"digraphs",
4575#ifdef FEAT_DIGRAPHS
4576 1
4577#else
4578 0
4579#endif
4580 },
4581 {"directx",
4582#ifdef FEAT_DIRECTX
4583 1
4584#else
4585 0
4586#endif
4587 },
4588 {"dnd",
4589#ifdef FEAT_DND
4590 1
4591#else
4592 0
4593#endif
4594 },
4595 {"emacs_tags",
4596#ifdef FEAT_EMACS_TAGS
4597 1
4598#else
4599 0
4600#endif
4601 },
4602 {"eval", 1}, // always present, of course!
4603 {"ex_extra", 1}, // graduated feature
4604 {"extra_search",
4605#ifdef FEAT_SEARCH_EXTRA
4606 1
4607#else
4608 0
4609#endif
4610 },
4611 {"file_in_path",
4612#ifdef FEAT_SEARCHPATH
4613 1
4614#else
4615 0
4616#endif
4617 },
4618 {"filterpipe",
4619#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
4620 1
4621#else
4622 0
4623#endif
4624 },
4625 {"find_in_path",
4626#ifdef FEAT_FIND_ID
4627 1
4628#else
4629 0
4630#endif
4631 },
4632 {"float",
4633#ifdef FEAT_FLOAT
4634 1
4635#else
4636 0
4637#endif
4638 },
4639 {"folding",
4640#ifdef FEAT_FOLDING
4641 1
4642#else
4643 0
4644#endif
4645 },
4646 {"footer",
4647#ifdef FEAT_FOOTER
4648 1
4649#else
4650 0
4651#endif
4652 },
4653 {"fork",
4654#if !defined(USE_SYSTEM) && defined(UNIX)
4655 1
4656#else
4657 0
4658#endif
4659 },
4660 {"gettext",
4661#ifdef FEAT_GETTEXT
4662 1
4663#else
4664 0
4665#endif
4666 },
4667 {"gui",
4668#ifdef FEAT_GUI
4669 1
4670#else
4671 0
4672#endif
4673 },
4674 {"gui_neXtaw",
4675#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
4676 1
4677#else
4678 0
4679#endif
4680 },
4681 {"gui_athena",
4682#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
4683 1
4684#else
4685 0
4686#endif
4687 },
4688 {"gui_gtk",
4689#ifdef FEAT_GUI_GTK
4690 1
4691#else
4692 0
4693#endif
4694 },
4695 {"gui_gtk2",
4696#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
4697 1
4698#else
4699 0
4700#endif
4701 },
4702 {"gui_gtk3",
4703#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
4704 1
4705#else
4706 0
4707#endif
4708 },
4709 {"gui_gnome",
4710#ifdef FEAT_GUI_GNOME
4711 1
4712#else
4713 0
4714#endif
4715 },
4716 {"gui_haiku",
4717#ifdef FEAT_GUI_HAIKU
4718 1
4719#else
4720 0
4721#endif
4722 },
Bram Moolenaar097148e2020-08-11 21:58:20 +02004723 {"gui_mac", 0},
Bram Moolenaar79296512020-03-22 16:17:14 +01004724 {"gui_motif",
4725#ifdef FEAT_GUI_MOTIF
4726 1
4727#else
4728 0
4729#endif
4730 },
4731 {"gui_photon",
4732#ifdef FEAT_GUI_PHOTON
4733 1
4734#else
4735 0
4736#endif
4737 },
4738 {"gui_win32",
4739#ifdef FEAT_GUI_MSWIN
4740 1
4741#else
4742 0
4743#endif
4744 },
4745 {"iconv",
4746#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
4747 1
4748#else
4749 0
4750#endif
4751 },
4752 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02004753 {"ipv6",
4754#ifdef FEAT_IPV6
4755 1
4756#else
4757 0
4758#endif
4759 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004760 {"job",
4761#ifdef FEAT_JOB_CHANNEL
4762 1
4763#else
4764 0
4765#endif
4766 },
4767 {"jumplist",
4768#ifdef FEAT_JUMPLIST
4769 1
4770#else
4771 0
4772#endif
4773 },
4774 {"keymap",
4775#ifdef FEAT_KEYMAP
4776 1
4777#else
4778 0
4779#endif
4780 },
4781 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
4782 {"langmap",
4783#ifdef FEAT_LANGMAP
4784 1
4785#else
4786 0
4787#endif
4788 },
4789 {"libcall",
4790#ifdef FEAT_LIBCALL
4791 1
4792#else
4793 0
4794#endif
4795 },
4796 {"linebreak",
4797#ifdef FEAT_LINEBREAK
4798 1
4799#else
4800 0
4801#endif
4802 },
4803 {"lispindent",
4804#ifdef FEAT_LISP
4805 1
4806#else
4807 0
4808#endif
4809 },
4810 {"listcmds", 1},
4811 {"localmap", 1},
4812 {"lua",
4813#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4814 1
4815#else
4816 0
4817#endif
4818 },
4819 {"menu",
4820#ifdef FEAT_MENU
4821 1
4822#else
4823 0
4824#endif
4825 },
4826 {"mksession",
4827#ifdef FEAT_SESSION
4828 1
4829#else
4830 0
4831#endif
4832 },
4833 {"modify_fname", 1},
4834 {"mouse", 1},
4835 {"mouseshape",
4836#ifdef FEAT_MOUSESHAPE
4837 1
4838#else
4839 0
4840#endif
4841 },
4842 {"mouse_dec",
4843#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4844 1
4845#else
4846 0
4847#endif
4848 },
4849 {"mouse_gpm",
4850#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4851 1
4852#else
4853 0
4854#endif
4855 },
4856 {"mouse_jsbterm",
4857#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4858 1
4859#else
4860 0
4861#endif
4862 },
4863 {"mouse_netterm",
4864#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4865 1
4866#else
4867 0
4868#endif
4869 },
4870 {"mouse_pterm",
4871#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4872 1
4873#else
4874 0
4875#endif
4876 },
4877 {"mouse_sgr",
4878#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4879 1
4880#else
4881 0
4882#endif
4883 },
4884 {"mouse_sysmouse",
4885#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4886 1
4887#else
4888 0
4889#endif
4890 },
4891 {"mouse_urxvt",
4892#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4893 1
4894#else
4895 0
4896#endif
4897 },
4898 {"mouse_xterm",
4899#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4900 1
4901#else
4902 0
4903#endif
4904 },
4905 {"multi_byte", 1},
4906 {"multi_byte_ime",
4907#ifdef FEAT_MBYTE_IME
4908 1
4909#else
4910 0
4911#endif
4912 },
4913 {"multi_lang",
4914#ifdef FEAT_MULTI_LANG
4915 1
4916#else
4917 0
4918#endif
4919 },
4920 {"mzscheme",
4921#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4922 1
4923#else
4924 0
4925#endif
4926 },
4927 {"num64", 1},
4928 {"ole",
4929#ifdef FEAT_OLE
4930 1
4931#else
4932 0
4933#endif
4934 },
4935 {"packages",
4936#ifdef FEAT_EVAL
4937 1
4938#else
4939 0
4940#endif
4941 },
4942 {"path_extra",
4943#ifdef FEAT_PATH_EXTRA
4944 1
4945#else
4946 0
4947#endif
4948 },
4949 {"perl",
4950#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4951 1
4952#else
4953 0
4954#endif
4955 },
4956 {"persistent_undo",
4957#ifdef FEAT_PERSISTENT_UNDO
4958 1
4959#else
4960 0
4961#endif
4962 },
4963 {"python_compiled",
4964#if defined(FEAT_PYTHON)
4965 1
4966#else
4967 0
4968#endif
4969 },
4970 {"python_dynamic",
4971#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4972 1
4973#else
4974 0
4975#endif
4976 },
4977 {"python",
4978#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4979 1
4980#else
4981 0
4982#endif
4983 },
4984 {"pythonx",
4985#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4986 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4987 1
4988#else
4989 0
4990#endif
4991 },
4992 {"python3_compiled",
4993#if defined(FEAT_PYTHON3)
4994 1
4995#else
4996 0
4997#endif
4998 },
4999 {"python3_dynamic",
5000#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
5001 1
5002#else
5003 0
5004#endif
5005 },
5006 {"python3",
5007#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
5008 1
5009#else
5010 0
5011#endif
5012 },
5013 {"popupwin",
5014#ifdef FEAT_PROP_POPUP
5015 1
5016#else
5017 0
5018#endif
5019 },
5020 {"postscript",
5021#ifdef FEAT_POSTSCRIPT
5022 1
5023#else
5024 0
5025#endif
5026 },
5027 {"printer",
5028#ifdef FEAT_PRINTER
5029 1
5030#else
5031 0
5032#endif
5033 },
5034 {"profile",
5035#ifdef FEAT_PROFILE
5036 1
5037#else
5038 0
5039#endif
5040 },
5041 {"reltime",
5042#ifdef FEAT_RELTIME
5043 1
5044#else
5045 0
5046#endif
5047 },
5048 {"quickfix",
5049#ifdef FEAT_QUICKFIX
5050 1
5051#else
5052 0
5053#endif
5054 },
5055 {"rightleft",
5056#ifdef FEAT_RIGHTLEFT
5057 1
5058#else
5059 0
5060#endif
5061 },
5062 {"ruby",
5063#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
5064 1
5065#else
5066 0
5067#endif
5068 },
5069 {"scrollbind", 1},
5070 {"showcmd",
5071#ifdef FEAT_CMDL_INFO
5072 1
5073#else
5074 0
5075#endif
5076 },
5077 {"cmdline_info",
5078#ifdef FEAT_CMDL_INFO
5079 1
5080#else
5081 0
5082#endif
5083 },
5084 {"signs",
5085#ifdef FEAT_SIGNS
5086 1
5087#else
5088 0
5089#endif
5090 },
5091 {"smartindent",
5092#ifdef FEAT_SMARTINDENT
5093 1
5094#else
5095 0
5096#endif
5097 },
5098 {"startuptime",
5099#ifdef STARTUPTIME
5100 1
5101#else
5102 0
5103#endif
5104 },
5105 {"statusline",
5106#ifdef FEAT_STL_OPT
5107 1
5108#else
5109 0
5110#endif
5111 },
5112 {"netbeans_intg",
5113#ifdef FEAT_NETBEANS_INTG
5114 1
5115#else
5116 0
5117#endif
5118 },
5119 {"sound",
5120#ifdef FEAT_SOUND
5121 1
5122#else
5123 0
5124#endif
5125 },
5126 {"spell",
5127#ifdef FEAT_SPELL
5128 1
5129#else
5130 0
5131#endif
5132 },
5133 {"syntax",
5134#ifdef FEAT_SYN_HL
5135 1
5136#else
5137 0
5138#endif
5139 },
5140 {"system",
5141#if defined(USE_SYSTEM) || !defined(UNIX)
5142 1
5143#else
5144 0
5145#endif
5146 },
5147 {"tag_binary",
5148#ifdef FEAT_TAG_BINS
5149 1
5150#else
5151 0
5152#endif
5153 },
5154 {"tcl",
5155#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
5156 1
5157#else
5158 0
5159#endif
5160 },
5161 {"termguicolors",
5162#ifdef FEAT_TERMGUICOLORS
5163 1
5164#else
5165 0
5166#endif
5167 },
5168 {"terminal",
5169#if defined(FEAT_TERMINAL) && !defined(MSWIN)
5170 1
5171#else
5172 0
5173#endif
5174 },
5175 {"terminfo",
5176#ifdef TERMINFO
5177 1
5178#else
5179 0
5180#endif
5181 },
5182 {"termresponse",
5183#ifdef FEAT_TERMRESPONSE
5184 1
5185#else
5186 0
5187#endif
5188 },
5189 {"textobjects",
5190#ifdef FEAT_TEXTOBJ
5191 1
5192#else
5193 0
5194#endif
5195 },
5196 {"textprop",
5197#ifdef FEAT_PROP_POPUP
5198 1
5199#else
5200 0
5201#endif
5202 },
5203 {"tgetent",
5204#ifdef HAVE_TGETENT
5205 1
5206#else
5207 0
5208#endif
5209 },
5210 {"timers",
5211#ifdef FEAT_TIMERS
5212 1
5213#else
5214 0
5215#endif
5216 },
5217 {"title",
5218#ifdef FEAT_TITLE
5219 1
5220#else
5221 0
5222#endif
5223 },
5224 {"toolbar",
5225#ifdef FEAT_TOOLBAR
5226 1
5227#else
5228 0
5229#endif
5230 },
5231 {"unnamedplus",
5232#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
5233 1
5234#else
5235 0
5236#endif
5237 },
5238 {"user-commands", 1}, // was accidentally included in 5.4
5239 {"user_commands", 1},
5240 {"vartabs",
5241#ifdef FEAT_VARTABS
5242 1
5243#else
5244 0
5245#endif
5246 },
5247 {"vertsplit", 1},
5248 {"viminfo",
5249#ifdef FEAT_VIMINFO
5250 1
5251#else
5252 0
5253#endif
5254 },
5255 {"vimscript-1", 1},
5256 {"vimscript-2", 1},
5257 {"vimscript-3", 1},
5258 {"vimscript-4", 1},
5259 {"virtualedit", 1},
5260 {"visual", 1},
5261 {"visualextra", 1},
5262 {"vreplace", 1},
5263 {"vtp",
5264#ifdef FEAT_VTP
5265 1
5266#else
5267 0
5268#endif
5269 },
5270 {"wildignore",
5271#ifdef FEAT_WILDIGN
5272 1
5273#else
5274 0
5275#endif
5276 },
5277 {"wildmenu",
5278#ifdef FEAT_WILDMENU
5279 1
5280#else
5281 0
5282#endif
5283 },
5284 {"windows", 1},
5285 {"winaltkeys",
5286#ifdef FEAT_WAK
5287 1
5288#else
5289 0
5290#endif
5291 },
5292 {"writebackup",
5293#ifdef FEAT_WRITEBACKUP
5294 1
5295#else
5296 0
5297#endif
5298 },
5299 {"xim",
5300#ifdef FEAT_XIM
5301 1
5302#else
5303 0
5304#endif
5305 },
5306 {"xfontset",
5307#ifdef FEAT_XFONTSET
5308 1
5309#else
5310 0
5311#endif
5312 },
5313 {"xpm",
5314#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
5315 1
5316#else
5317 0
5318#endif
5319 },
5320 {"xpm_w32", // for backward compatibility
5321#ifdef FEAT_XPM_W32
5322 1
5323#else
5324 0
5325#endif
5326 },
5327 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005328#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01005329 1
5330#else
5331 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005332#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005333 },
5334 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005335#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01005336 1
5337#else
5338 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005339#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005340 },
5341 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005342#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01005343 1
5344#else
5345 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005346#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005347 },
5348 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005349#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01005350 1
5351#else
5352 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005353#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005354 },
5355 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005356#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01005357 1
5358#else
5359 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005360#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005361 },
5362 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005363 };
5364
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005365 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01005366 for (i = 0; has_list[i].name != NULL; ++i)
5367 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005368 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005369 x = TRUE;
5370 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005371 break;
5372 }
5373
Bram Moolenaar79296512020-03-22 16:17:14 +01005374 // features also in has_list[] but sometimes enabled at runtime
5375 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005376 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005377 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02005378 {
5379 // intentionally empty
5380 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01005381#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005382 else if (STRICMP(name, "balloon_multiline") == 0)
5383 n = multiline_balloon_available();
5384#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005385#ifdef VIMDLL
5386 else if (STRICMP(name, "filterpipe") == 0)
5387 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005388#endif
5389#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5390 else if (STRICMP(name, "iconv") == 0)
5391 n = iconv_enabled(FALSE);
5392#endif
5393#ifdef DYNAMIC_LUA
5394 else if (STRICMP(name, "lua") == 0)
5395 n = lua_enabled(FALSE);
5396#endif
5397#ifdef DYNAMIC_MZSCHEME
5398 else if (STRICMP(name, "mzscheme") == 0)
5399 n = mzscheme_enabled(FALSE);
5400#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005401#ifdef DYNAMIC_PERL
5402 else if (STRICMP(name, "perl") == 0)
5403 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005404#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005405#ifdef DYNAMIC_PYTHON
5406 else if (STRICMP(name, "python") == 0)
5407 n = python_enabled(FALSE);
5408#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005409#ifdef DYNAMIC_PYTHON3
5410 else if (STRICMP(name, "python3") == 0)
5411 n = python3_enabled(FALSE);
5412#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005413#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5414 else if (STRICMP(name, "pythonx") == 0)
5415 {
5416# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
5417 if (p_pyx == 0)
5418 n = python3_enabled(FALSE) || python_enabled(FALSE);
5419 else if (p_pyx == 3)
5420 n = python3_enabled(FALSE);
5421 else if (p_pyx == 2)
5422 n = python_enabled(FALSE);
5423# elif defined(DYNAMIC_PYTHON)
5424 n = python_enabled(FALSE);
5425# elif defined(DYNAMIC_PYTHON3)
5426 n = python3_enabled(FALSE);
5427# endif
5428 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005429#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005430#ifdef DYNAMIC_RUBY
5431 else if (STRICMP(name, "ruby") == 0)
5432 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005433#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005434#ifdef DYNAMIC_TCL
5435 else if (STRICMP(name, "tcl") == 0)
5436 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02005437#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01005438#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02005439 else if (STRICMP(name, "terminal") == 0)
5440 n = terminal_enabled();
5441#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005442 }
5443
Bram Moolenaar79296512020-03-22 16:17:14 +01005444 // features not in has_list[]
5445 if (x == FALSE)
5446 {
5447 if (STRNICMP(name, "patch", 5) == 0)
5448 {
5449 x = TRUE;
5450 if (name[5] == '-'
5451 && STRLEN(name) >= 11
5452 && vim_isdigit(name[6])
5453 && vim_isdigit(name[8])
5454 && vim_isdigit(name[10]))
5455 {
5456 int major = atoi((char *)name + 6);
5457 int minor = atoi((char *)name + 8);
5458
5459 // Expect "patch-9.9.01234".
5460 n = (major < VIM_VERSION_MAJOR
5461 || (major == VIM_VERSION_MAJOR
5462 && (minor < VIM_VERSION_MINOR
5463 || (minor == VIM_VERSION_MINOR
5464 && has_patch(atoi((char *)name + 10))))));
5465 }
5466 else
5467 n = has_patch(atoi((char *)name + 5));
5468 }
5469 else if (STRICMP(name, "vim_starting") == 0)
5470 {
5471 x = TRUE;
5472 n = (starting != 0);
5473 }
5474 else if (STRICMP(name, "ttyin") == 0)
5475 {
5476 x = TRUE;
5477 n = mch_input_isatty();
5478 }
5479 else if (STRICMP(name, "ttyout") == 0)
5480 {
5481 x = TRUE;
5482 n = stdout_isatty;
5483 }
5484 else if (STRICMP(name, "multi_byte_encoding") == 0)
5485 {
5486 x = TRUE;
5487 n = has_mbyte;
5488 }
5489 else if (STRICMP(name, "gui_running") == 0)
5490 {
5491 x = TRUE;
5492#ifdef FEAT_GUI
5493 n = (gui.in_use || gui.starting);
5494#endif
5495 }
5496 else if (STRICMP(name, "browse") == 0)
5497 {
5498 x = TRUE;
5499#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5500 n = gui.in_use; // gui_mch_browse() works when GUI is running
5501#endif
5502 }
5503 else if (STRICMP(name, "syntax_items") == 0)
5504 {
5505 x = TRUE;
5506#ifdef FEAT_SYN_HL
5507 n = syntax_present(curwin);
5508#endif
5509 }
5510 else if (STRICMP(name, "vcon") == 0)
5511 {
5512 x = TRUE;
5513#ifdef FEAT_VTP
5514 n = is_term_win32() && has_vtp_working();
5515#endif
5516 }
5517 else if (STRICMP(name, "netbeans_enabled") == 0)
5518 {
5519 x = TRUE;
5520#ifdef FEAT_NETBEANS_INTG
5521 n = netbeans_active();
5522#endif
5523 }
5524 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
5525 {
5526 x = TRUE;
5527#ifdef FEAT_MOUSE_GPM
5528 n = gpm_enabled();
5529#endif
5530 }
5531 else if (STRICMP(name, "conpty") == 0)
5532 {
5533 x = TRUE;
5534#if defined(FEAT_TERMINAL) && defined(MSWIN)
5535 n = use_conpty();
5536#endif
5537 }
5538 else if (STRICMP(name, "clipboard_working") == 0)
5539 {
5540 x = TRUE;
5541#ifdef FEAT_CLIPBOARD
5542 n = clip_star.available;
5543#endif
5544 }
5545 }
5546
Bram Moolenaar04637e22020-09-05 18:45:29 +02005547 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[1]))
Bram Moolenaar79296512020-03-22 16:17:14 +01005548 // return whether feature could ever be enabled
5549 rettv->vval.v_number = x;
5550 else
5551 // return whether feature is enabled
5552 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005553}
5554
5555/*
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005556 * Return TRUE if "feature" can change later.
5557 * Also when checking for the feature has side effects, such as loading a DLL.
5558 */
5559 int
5560dynamic_feature(char_u *feature)
5561{
5562 return (feature == NULL
5563#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
5564 || STRICMP(feature, "balloon_multiline") == 0
5565#endif
5566#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5567 || (STRICMP(feature, "browse") == 0 && !gui.in_use)
5568#endif
5569#ifdef VIMDLL
5570 || STRICMP(feature, "filterpipe") == 0
5571#endif
Bram Moolenaar29b281b2020-11-10 20:58:00 +01005572#if defined(FEAT_GUI) && !defined(ALWAYS_USE_GUI) && !defined(VIMDLL)
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005573 // this can only change on Unix where the ":gui" command could be
5574 // used.
5575 || (STRICMP(feature, "gui_running") == 0 && !gui.in_use)
5576#endif
5577#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5578 || STRICMP(feature, "iconv") == 0
5579#endif
5580#ifdef DYNAMIC_LUA
5581 || STRICMP(feature, "lua") == 0
5582#endif
5583#ifdef FEAT_MOUSE_GPM
5584 || (STRICMP(feature, "mouse_gpm_enabled") == 0 && !gpm_enabled())
5585#endif
5586#ifdef DYNAMIC_MZSCHEME
5587 || STRICMP(feature, "mzscheme") == 0
5588#endif
5589#ifdef FEAT_NETBEANS_INTG
5590 || STRICMP(feature, "netbeans_enabled") == 0
5591#endif
5592#ifdef DYNAMIC_PERL
5593 || STRICMP(feature, "perl") == 0
5594#endif
5595#ifdef DYNAMIC_PYTHON
5596 || STRICMP(feature, "python") == 0
5597#endif
5598#ifdef DYNAMIC_PYTHON3
5599 || STRICMP(feature, "python3") == 0
5600#endif
5601#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5602 || STRICMP(feature, "pythonx") == 0
5603#endif
5604#ifdef DYNAMIC_RUBY
5605 || STRICMP(feature, "ruby") == 0
5606#endif
5607#ifdef FEAT_SYN_HL
5608 || STRICMP(feature, "syntax_items") == 0
5609#endif
5610#ifdef DYNAMIC_TCL
5611 || STRICMP(feature, "tcl") == 0
5612#endif
5613 // once "starting" is zero it will stay that way
5614 || (STRICMP(feature, "vim_starting") == 0 && starting != 0)
5615 || STRICMP(feature, "multi_byte_encoding") == 0
5616#if defined(FEAT_TERMINAL) && defined(MSWIN)
5617 || STRICMP(feature, "conpty") == 0
5618#endif
5619 );
5620}
5621
5622/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005623 * "haslocaldir()" function
5624 */
5625 static void
5626f_haslocaldir(typval_T *argvars, typval_T *rettv)
5627{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005628 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005629 win_T *wp = NULL;
5630
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005631 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
5632
5633 // Check for window-local and tab-local directories
5634 if (wp != NULL && wp->w_localdir != NULL)
5635 rettv->vval.v_number = 1;
5636 else if (tp != NULL && tp->tp_localdir != NULL)
5637 rettv->vval.v_number = 2;
5638 else
5639 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005640}
5641
5642/*
5643 * "hasmapto()" function
5644 */
5645 static void
5646f_hasmapto(typval_T *argvars, typval_T *rettv)
5647{
5648 char_u *name;
5649 char_u *mode;
5650 char_u buf[NUMBUFLEN];
5651 int abbr = FALSE;
5652
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005653 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005654 if (argvars[1].v_type == VAR_UNKNOWN)
5655 mode = (char_u *)"nvo";
5656 else
5657 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005658 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005659 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02005660 abbr = (int)tv_get_bool(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005661 }
5662
5663 if (map_to_exists(name, mode, abbr))
5664 rettv->vval.v_number = TRUE;
5665 else
5666 rettv->vval.v_number = FALSE;
5667}
5668
5669/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005670 * "highlightID(name)" function
5671 */
5672 static void
5673f_hlID(typval_T *argvars, typval_T *rettv)
5674{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005675 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005676}
5677
5678/*
5679 * "highlight_exists()" function
5680 */
5681 static void
5682f_hlexists(typval_T *argvars, typval_T *rettv)
5683{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005684 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005685}
5686
5687/*
5688 * "hostname()" function
5689 */
5690 static void
5691f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
5692{
5693 char_u hostname[256];
5694
5695 mch_get_host_name(hostname, 256);
5696 rettv->v_type = VAR_STRING;
5697 rettv->vval.v_string = vim_strsave(hostname);
5698}
5699
5700/*
5701 * iconv() function
5702 */
5703 static void
5704f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
5705{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005706 char_u buf1[NUMBUFLEN];
5707 char_u buf2[NUMBUFLEN];
5708 char_u *from, *to, *str;
5709 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005710
5711 rettv->v_type = VAR_STRING;
5712 rettv->vval.v_string = NULL;
5713
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005714 str = tv_get_string(&argvars[0]);
5715 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
5716 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005717 vimconv.vc_type = CONV_NONE;
5718 convert_setup(&vimconv, from, to);
5719
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005720 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005721 if (vimconv.vc_type == CONV_NONE)
5722 rettv->vval.v_string = vim_strsave(str);
5723 else
5724 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
5725
5726 convert_setup(&vimconv, NULL, NULL);
5727 vim_free(from);
5728 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005729}
5730
5731/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005732 * "index()" function
5733 */
5734 static void
5735f_index(typval_T *argvars, typval_T *rettv)
5736{
5737 list_T *l;
5738 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005739 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005740 long idx = 0;
5741 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005742 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005743
5744 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005745 if (argvars[0].v_type == VAR_BLOB)
5746 {
5747 typval_T tv;
5748 int start = 0;
5749
5750 if (argvars[2].v_type != VAR_UNKNOWN)
5751 {
5752 start = tv_get_number_chk(&argvars[2], &error);
5753 if (error)
5754 return;
5755 }
5756 b = argvars[0].vval.v_blob;
5757 if (b == NULL)
5758 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01005759 if (start < 0)
5760 {
5761 start = blob_len(b) + start;
5762 if (start < 0)
5763 start = 0;
5764 }
5765
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005766 for (idx = start; idx < blob_len(b); ++idx)
5767 {
5768 tv.v_type = VAR_NUMBER;
5769 tv.vval.v_number = blob_get(b, idx);
5770 if (tv_equal(&tv, &argvars[1], ic, FALSE))
5771 {
5772 rettv->vval.v_number = idx;
5773 return;
5774 }
5775 }
5776 return;
5777 }
5778 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005779 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01005780 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005781 return;
5782 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005783
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005784 l = argvars[0].vval.v_list;
5785 if (l != NULL)
5786 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005787 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005788 item = l->lv_first;
5789 if (argvars[2].v_type != VAR_UNKNOWN)
5790 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005791 // Start at specified item. Use the cached index that list_find()
5792 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005793 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005794 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005795 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar6c553f92020-09-02 22:10:34 +02005796 ic = (int)tv_get_bool_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005797 if (error)
5798 item = NULL;
5799 }
5800
5801 for ( ; item != NULL; item = item->li_next, ++idx)
5802 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
5803 {
5804 rettv->vval.v_number = idx;
5805 break;
5806 }
5807 }
5808}
5809
5810static int inputsecret_flag = 0;
5811
5812/*
5813 * "input()" function
5814 * Also handles inputsecret() when inputsecret is set.
5815 */
5816 static void
5817f_input(typval_T *argvars, typval_T *rettv)
5818{
5819 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
5820}
5821
5822/*
5823 * "inputdialog()" function
5824 */
5825 static void
5826f_inputdialog(typval_T *argvars, typval_T *rettv)
5827{
5828#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005829 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005830 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
5831 {
5832 char_u *message;
5833 char_u buf[NUMBUFLEN];
5834 char_u *defstr = (char_u *)"";
5835
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005836 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005837 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005838 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005839 vim_strncpy(IObuff, defstr, IOSIZE - 1);
5840 else
5841 IObuff[0] = NUL;
5842 if (message != NULL && defstr != NULL
5843 && do_dialog(VIM_QUESTION, NULL, message,
5844 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
5845 rettv->vval.v_string = vim_strsave(IObuff);
5846 else
5847 {
5848 if (message != NULL && defstr != NULL
5849 && argvars[1].v_type != VAR_UNKNOWN
5850 && argvars[2].v_type != VAR_UNKNOWN)
5851 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005852 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005853 else
5854 rettv->vval.v_string = NULL;
5855 }
5856 rettv->v_type = VAR_STRING;
5857 }
5858 else
5859#endif
5860 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
5861}
5862
5863/*
5864 * "inputlist()" function
5865 */
5866 static void
5867f_inputlist(typval_T *argvars, typval_T *rettv)
5868{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005869 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005870 listitem_T *li;
5871 int selected;
5872 int mouse_used;
5873
5874#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005875 // While starting up, there is no place to enter text. When running tests
5876 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005877 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005878 return;
5879#endif
5880 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5881 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005882 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005883 return;
5884 }
5885
5886 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005887 msg_row = Rows - 1; // for when 'cmdheight' > 1
5888 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005889 msg_scroll = TRUE;
5890 msg_clr_eos();
5891
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005892 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005893 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02005894 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005895 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005896 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005897 msg_putchar('\n');
5898 }
5899
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005900 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005901 selected = prompt_for_number(&mouse_used);
5902 if (mouse_used)
5903 selected -= lines_left;
5904
5905 rettv->vval.v_number = selected;
5906}
5907
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005908static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
5909
5910/*
5911 * "inputrestore()" function
5912 */
5913 static void
5914f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
5915{
5916 if (ga_userinput.ga_len > 0)
5917 {
5918 --ga_userinput.ga_len;
5919 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
5920 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005921 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005922 }
5923 else if (p_verbose > 1)
5924 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005925 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005926 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005927 }
5928}
5929
5930/*
5931 * "inputsave()" function
5932 */
5933 static void
5934f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
5935{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005936 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005937 if (ga_grow(&ga_userinput, 1) == OK)
5938 {
5939 save_typeahead((tasave_T *)(ga_userinput.ga_data)
5940 + ga_userinput.ga_len);
5941 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005942 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005943 }
5944 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005945 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005946}
5947
5948/*
5949 * "inputsecret()" function
5950 */
5951 static void
5952f_inputsecret(typval_T *argvars, typval_T *rettv)
5953{
5954 ++cmdline_star;
5955 ++inputsecret_flag;
5956 f_input(argvars, rettv);
5957 --cmdline_star;
5958 --inputsecret_flag;
5959}
5960
5961/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005962 * "interrupt()" function
5963 */
5964 static void
5965f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5966{
5967 got_int = TRUE;
5968}
5969
5970/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005971 * "invert(expr)" function
5972 */
5973 static void
5974f_invert(typval_T *argvars, typval_T *rettv)
5975{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005976 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005977}
5978
5979/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005980 * "islocked()" function
5981 */
5982 static void
5983f_islocked(typval_T *argvars, typval_T *rettv)
5984{
5985 lval_T lv;
5986 char_u *end;
5987 dictitem_T *di;
5988
5989 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005990 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005991 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005992 if (end != NULL && lv.ll_name != NULL)
5993 {
5994 if (*end != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02005995 semsg(_(e_trailing_arg), end);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005996 else
5997 {
5998 if (lv.ll_tv == NULL)
5999 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01006000 di = find_var(lv.ll_name, NULL, TRUE);
6001 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006002 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006003 // Consider a variable locked when:
6004 // 1. the variable itself is locked
6005 // 2. the value of the variable is locked.
6006 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01006007 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
6008 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006009 }
6010 }
6011 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006012 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006013 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006014 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006015 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006016 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006017 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
6018 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006019 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006020 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
6021 }
6022 }
6023
6024 clear_lval(&lv);
6025}
6026
6027#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
6028/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02006029 * "isinf()" function
6030 */
6031 static void
6032f_isinf(typval_T *argvars, typval_T *rettv)
6033{
6034 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
6035 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
6036}
6037
6038/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006039 * "isnan()" function
6040 */
6041 static void
6042f_isnan(typval_T *argvars, typval_T *rettv)
6043{
6044 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
6045 && isnan(argvars[0].vval.v_float);
6046}
6047#endif
6048
6049/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006050 * "last_buffer_nr()" function.
6051 */
6052 static void
6053f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
6054{
6055 int n = 0;
6056 buf_T *buf;
6057
Bram Moolenaar29323592016-07-24 22:04:11 +02006058 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006059 if (n < buf->b_fnum)
6060 n = buf->b_fnum;
6061
6062 rettv->vval.v_number = n;
6063}
6064
6065/*
6066 * "len()" function
6067 */
6068 static void
6069f_len(typval_T *argvars, typval_T *rettv)
6070{
6071 switch (argvars[0].v_type)
6072 {
6073 case VAR_STRING:
6074 case VAR_NUMBER:
6075 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006076 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006077 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01006078 case VAR_BLOB:
6079 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
6080 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006081 case VAR_LIST:
6082 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
6083 break;
6084 case VAR_DICT:
6085 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
6086 break;
6087 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02006088 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006089 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01006090 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006091 case VAR_SPECIAL:
6092 case VAR_FLOAT:
6093 case VAR_FUNC:
6094 case VAR_PARTIAL:
6095 case VAR_JOB:
6096 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006097 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006098 break;
6099 }
6100}
6101
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006102 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01006103libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006104{
6105#ifdef FEAT_LIBCALL
6106 char_u *string_in;
6107 char_u **string_result;
6108 int nr_result;
6109#endif
6110
6111 rettv->v_type = type;
6112 if (type != VAR_NUMBER)
6113 rettv->vval.v_string = NULL;
6114
6115 if (check_restricted() || check_secure())
6116 return;
6117
6118#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006119 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006120 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
6121 {
6122 string_in = NULL;
6123 if (argvars[2].v_type == VAR_STRING)
6124 string_in = argvars[2].vval.v_string;
6125 if (type == VAR_NUMBER)
6126 string_result = NULL;
6127 else
6128 string_result = &rettv->vval.v_string;
6129 if (mch_libcall(argvars[0].vval.v_string,
6130 argvars[1].vval.v_string,
6131 string_in,
6132 argvars[2].vval.v_number,
6133 string_result,
6134 &nr_result) == OK
6135 && type == VAR_NUMBER)
6136 rettv->vval.v_number = nr_result;
6137 }
6138#endif
6139}
6140
6141/*
6142 * "libcall()" function
6143 */
6144 static void
6145f_libcall(typval_T *argvars, typval_T *rettv)
6146{
6147 libcall_common(argvars, rettv, VAR_STRING);
6148}
6149
6150/*
6151 * "libcallnr()" function
6152 */
6153 static void
6154f_libcallnr(typval_T *argvars, typval_T *rettv)
6155{
6156 libcall_common(argvars, rettv, VAR_NUMBER);
6157}
6158
6159/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006160 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006161 */
6162 static void
6163f_line(typval_T *argvars, typval_T *rettv)
6164{
6165 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006166 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006167 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006168 int id;
6169 tabpage_T *tp;
6170 win_T *wp;
6171 win_T *save_curwin;
6172 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006173
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006174 if (argvars[1].v_type != VAR_UNKNOWN)
6175 {
6176 // use window specified in the second argument
6177 id = (int)tv_get_number(&argvars[1]);
6178 wp = win_id2wp_tp(id, &tp);
6179 if (wp != NULL && tp != NULL)
6180 {
6181 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
6182 == OK)
6183 {
6184 check_cursor();
6185 fp = var2fpos(&argvars[0], TRUE, &fnum);
6186 }
6187 restore_win_noblock(save_curwin, save_curtab, TRUE);
6188 }
6189 }
6190 else
6191 // use current window
6192 fp = var2fpos(&argvars[0], TRUE, &fnum);
6193
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006194 if (fp != NULL)
6195 lnum = fp->lnum;
6196 rettv->vval.v_number = lnum;
6197}
6198
6199/*
6200 * "line2byte(lnum)" function
6201 */
6202 static void
6203f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
6204{
6205#ifndef FEAT_BYTEOFF
6206 rettv->vval.v_number = -1;
6207#else
6208 linenr_T lnum;
6209
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006210 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006211 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
6212 rettv->vval.v_number = -1;
6213 else
6214 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
6215 if (rettv->vval.v_number >= 0)
6216 ++rettv->vval.v_number;
6217#endif
6218}
6219
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006220#ifdef FEAT_FLOAT
6221/*
6222 * "log()" function
6223 */
6224 static void
6225f_log(typval_T *argvars, typval_T *rettv)
6226{
6227 float_T f = 0.0;
6228
6229 rettv->v_type = VAR_FLOAT;
6230 if (get_float_arg(argvars, &f) == OK)
6231 rettv->vval.v_float = log(f);
6232 else
6233 rettv->vval.v_float = 0.0;
6234}
6235
6236/*
6237 * "log10()" function
6238 */
6239 static void
6240f_log10(typval_T *argvars, typval_T *rettv)
6241{
6242 float_T f = 0.0;
6243
6244 rettv->v_type = VAR_FLOAT;
6245 if (get_float_arg(argvars, &f) == OK)
6246 rettv->vval.v_float = log10(f);
6247 else
6248 rettv->vval.v_float = 0.0;
6249}
6250#endif
6251
6252#ifdef FEAT_LUA
6253/*
6254 * "luaeval()" function
6255 */
6256 static void
6257f_luaeval(typval_T *argvars, typval_T *rettv)
6258{
6259 char_u *str;
6260 char_u buf[NUMBUFLEN];
6261
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006262 if (check_restricted() || check_secure())
6263 return;
6264
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006265 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006266 do_luaeval(str, argvars + 1, rettv);
6267}
6268#endif
6269
6270/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006271 * "maparg()" function
6272 */
6273 static void
6274f_maparg(typval_T *argvars, typval_T *rettv)
6275{
6276 get_maparg(argvars, rettv, TRUE);
6277}
6278
6279/*
6280 * "mapcheck()" function
6281 */
6282 static void
6283f_mapcheck(typval_T *argvars, typval_T *rettv)
6284{
6285 get_maparg(argvars, rettv, FALSE);
6286}
6287
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006288typedef enum
6289{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006290 MATCH_END, // matchend()
6291 MATCH_MATCH, // match()
6292 MATCH_STR, // matchstr()
6293 MATCH_LIST, // matchlist()
6294 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006295} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006296
6297 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006298find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006299{
6300 char_u *str = NULL;
6301 long len = 0;
6302 char_u *expr = NULL;
6303 char_u *pat;
6304 regmatch_T regmatch;
6305 char_u patbuf[NUMBUFLEN];
6306 char_u strbuf[NUMBUFLEN];
6307 char_u *save_cpo;
6308 long start = 0;
6309 long nth = 1;
6310 colnr_T startcol = 0;
6311 int match = 0;
6312 list_T *l = NULL;
6313 listitem_T *li = NULL;
6314 long idx = 0;
6315 char_u *tofree = NULL;
6316
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006317 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006318 save_cpo = p_cpo;
6319 p_cpo = (char_u *)"";
6320
6321 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006322 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006323 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006324 // type MATCH_LIST: return empty list when there are no matches.
6325 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006326 if (rettv_list_alloc(rettv) == FAIL)
6327 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006328 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006329 && (list_append_string(rettv->vval.v_list,
6330 (char_u *)"", 0) == FAIL
6331 || list_append_number(rettv->vval.v_list,
6332 (varnumber_T)-1) == FAIL
6333 || list_append_number(rettv->vval.v_list,
6334 (varnumber_T)-1) == FAIL
6335 || list_append_number(rettv->vval.v_list,
6336 (varnumber_T)-1) == FAIL))
6337 {
6338 list_free(rettv->vval.v_list);
6339 rettv->vval.v_list = NULL;
6340 goto theend;
6341 }
6342 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006343 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006344 {
6345 rettv->v_type = VAR_STRING;
6346 rettv->vval.v_string = NULL;
6347 }
6348
6349 if (argvars[0].v_type == VAR_LIST)
6350 {
6351 if ((l = argvars[0].vval.v_list) == NULL)
6352 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006353 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006354 li = l->lv_first;
6355 }
6356 else
6357 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006358 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006359 len = (long)STRLEN(str);
6360 }
6361
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006362 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006363 if (pat == NULL)
6364 goto theend;
6365
6366 if (argvars[2].v_type != VAR_UNKNOWN)
6367 {
6368 int error = FALSE;
6369
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006370 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006371 if (error)
6372 goto theend;
6373 if (l != NULL)
6374 {
6375 li = list_find(l, start);
6376 if (li == NULL)
6377 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006378 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006379 }
6380 else
6381 {
6382 if (start < 0)
6383 start = 0;
6384 if (start > len)
6385 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006386 // When "count" argument is there ignore matches before "start",
6387 // otherwise skip part of the string. Differs when pattern is "^"
6388 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006389 if (argvars[3].v_type != VAR_UNKNOWN)
6390 startcol = start;
6391 else
6392 {
6393 str += start;
6394 len -= start;
6395 }
6396 }
6397
6398 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006399 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006400 if (error)
6401 goto theend;
6402 }
6403
6404 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
6405 if (regmatch.regprog != NULL)
6406 {
6407 regmatch.rm_ic = p_ic;
6408
6409 for (;;)
6410 {
6411 if (l != NULL)
6412 {
6413 if (li == NULL)
6414 {
6415 match = FALSE;
6416 break;
6417 }
6418 vim_free(tofree);
6419 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
6420 if (str == NULL)
6421 break;
6422 }
6423
6424 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
6425
6426 if (match && --nth <= 0)
6427 break;
6428 if (l == NULL && !match)
6429 break;
6430
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006431 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006432 if (l != NULL)
6433 {
6434 li = li->li_next;
6435 ++idx;
6436 }
6437 else
6438 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006439 startcol = (colnr_T)(regmatch.startp[0]
6440 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006441 if (startcol > (colnr_T)len
6442 || str + startcol <= regmatch.startp[0])
6443 {
6444 match = FALSE;
6445 break;
6446 }
6447 }
6448 }
6449
6450 if (match)
6451 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006452 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006453 {
6454 listitem_T *li1 = rettv->vval.v_list->lv_first;
6455 listitem_T *li2 = li1->li_next;
6456 listitem_T *li3 = li2->li_next;
6457 listitem_T *li4 = li3->li_next;
6458
6459 vim_free(li1->li_tv.vval.v_string);
6460 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006461 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006462 li3->li_tv.vval.v_number =
6463 (varnumber_T)(regmatch.startp[0] - expr);
6464 li4->li_tv.vval.v_number =
6465 (varnumber_T)(regmatch.endp[0] - expr);
6466 if (l != NULL)
6467 li2->li_tv.vval.v_number = (varnumber_T)idx;
6468 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006469 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006470 {
6471 int i;
6472
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006473 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006474 for (i = 0; i < NSUBEXP; ++i)
6475 {
6476 if (regmatch.endp[i] == NULL)
6477 {
6478 if (list_append_string(rettv->vval.v_list,
6479 (char_u *)"", 0) == FAIL)
6480 break;
6481 }
6482 else if (list_append_string(rettv->vval.v_list,
6483 regmatch.startp[i],
6484 (int)(regmatch.endp[i] - regmatch.startp[i]))
6485 == FAIL)
6486 break;
6487 }
6488 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006489 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006490 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006491 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006492 if (l != NULL)
6493 copy_tv(&li->li_tv, rettv);
6494 else
6495 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006496 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006497 }
6498 else if (l != NULL)
6499 rettv->vval.v_number = idx;
6500 else
6501 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006502 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006503 rettv->vval.v_number =
6504 (varnumber_T)(regmatch.startp[0] - str);
6505 else
6506 rettv->vval.v_number =
6507 (varnumber_T)(regmatch.endp[0] - str);
6508 rettv->vval.v_number += (varnumber_T)(str - expr);
6509 }
6510 }
6511 vim_regfree(regmatch.regprog);
6512 }
6513
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006514theend:
6515 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006516 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006517 listitem_remove(rettv->vval.v_list,
6518 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006519 vim_free(tofree);
6520 p_cpo = save_cpo;
6521}
6522
6523/*
6524 * "match()" function
6525 */
6526 static void
6527f_match(typval_T *argvars, typval_T *rettv)
6528{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006529 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006530}
6531
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006532/*
6533 * "matchend()" function
6534 */
6535 static void
6536f_matchend(typval_T *argvars, typval_T *rettv)
6537{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006538 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006539}
6540
6541/*
6542 * "matchlist()" function
6543 */
6544 static void
6545f_matchlist(typval_T *argvars, typval_T *rettv)
6546{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006547 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006548}
6549
6550/*
6551 * "matchstr()" function
6552 */
6553 static void
6554f_matchstr(typval_T *argvars, typval_T *rettv)
6555{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006556 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006557}
6558
6559/*
6560 * "matchstrpos()" function
6561 */
6562 static void
6563f_matchstrpos(typval_T *argvars, typval_T *rettv)
6564{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006565 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006566}
6567
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006568 static void
6569max_min(typval_T *argvars, typval_T *rettv, int domax)
6570{
6571 varnumber_T n = 0;
6572 varnumber_T i;
6573 int error = FALSE;
6574
6575 if (argvars[0].v_type == VAR_LIST)
6576 {
6577 list_T *l;
6578 listitem_T *li;
6579
6580 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006581 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006582 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006583 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006584 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006585 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
6586 n = l->lv_u.nonmat.lv_start;
6587 else
6588 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
6589 * l->lv_u.nonmat.lv_stride;
6590 }
6591 else
6592 {
6593 li = l->lv_first;
6594 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006595 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006596 n = tv_get_number_chk(&li->li_tv, &error);
6597 for (;;)
6598 {
6599 li = li->li_next;
6600 if (li == NULL)
6601 break;
6602 i = tv_get_number_chk(&li->li_tv, &error);
6603 if (domax ? i > n : i < n)
6604 n = i;
6605 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006606 }
6607 }
6608 }
6609 }
6610 else if (argvars[0].v_type == VAR_DICT)
6611 {
6612 dict_T *d;
6613 int first = TRUE;
6614 hashitem_T *hi;
6615 int todo;
6616
6617 d = argvars[0].vval.v_dict;
6618 if (d != NULL)
6619 {
6620 todo = (int)d->dv_hashtab.ht_used;
6621 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
6622 {
6623 if (!HASHITEM_EMPTY(hi))
6624 {
6625 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006626 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006627 if (first)
6628 {
6629 n = i;
6630 first = FALSE;
6631 }
6632 else if (domax ? i > n : i < n)
6633 n = i;
6634 }
6635 }
6636 }
6637 }
6638 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006639 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006640 rettv->vval.v_number = error ? 0 : n;
6641}
6642
6643/*
6644 * "max()" function
6645 */
6646 static void
6647f_max(typval_T *argvars, typval_T *rettv)
6648{
6649 max_min(argvars, rettv, TRUE);
6650}
6651
6652/*
6653 * "min()" function
6654 */
6655 static void
6656f_min(typval_T *argvars, typval_T *rettv)
6657{
6658 max_min(argvars, rettv, FALSE);
6659}
6660
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006661#if defined(FEAT_MZSCHEME) || defined(PROTO)
6662/*
6663 * "mzeval()" function
6664 */
6665 static void
6666f_mzeval(typval_T *argvars, typval_T *rettv)
6667{
6668 char_u *str;
6669 char_u buf[NUMBUFLEN];
6670
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006671 if (check_restricted() || check_secure())
6672 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006673 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006674 do_mzeval(str, rettv);
6675}
6676
6677 void
6678mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
6679{
6680 typval_T argvars[3];
6681
6682 argvars[0].v_type = VAR_STRING;
6683 argvars[0].vval.v_string = name;
6684 copy_tv(args, &argvars[1]);
6685 argvars[2].v_type = VAR_UNKNOWN;
6686 f_call(argvars, rettv);
6687 clear_tv(&argvars[1]);
6688}
6689#endif
6690
6691/*
6692 * "nextnonblank()" function
6693 */
6694 static void
6695f_nextnonblank(typval_T *argvars, typval_T *rettv)
6696{
6697 linenr_T lnum;
6698
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006699 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006700 {
6701 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
6702 {
6703 lnum = 0;
6704 break;
6705 }
6706 if (*skipwhite(ml_get(lnum)) != NUL)
6707 break;
6708 }
6709 rettv->vval.v_number = lnum;
6710}
6711
6712/*
6713 * "nr2char()" function
6714 */
6715 static void
6716f_nr2char(typval_T *argvars, typval_T *rettv)
6717{
6718 char_u buf[NUMBUFLEN];
6719
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006720 if (has_mbyte)
6721 {
6722 int utf8 = 0;
6723
6724 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaared6a4302020-09-05 20:29:41 +02006725 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006726 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01006727 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006728 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006729 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006730 }
6731 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006732 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006733 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006734 buf[1] = NUL;
6735 }
6736 rettv->v_type = VAR_STRING;
6737 rettv->vval.v_string = vim_strsave(buf);
6738}
6739
6740/*
6741 * "or(expr, expr)" function
6742 */
6743 static void
6744f_or(typval_T *argvars, typval_T *rettv)
6745{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006746 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
6747 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006748}
6749
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006750#ifdef FEAT_PERL
6751/*
6752 * "perleval()" function
6753 */
6754 static void
6755f_perleval(typval_T *argvars, typval_T *rettv)
6756{
6757 char_u *str;
6758 char_u buf[NUMBUFLEN];
6759
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006760 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006761 do_perleval(str, rettv);
6762}
6763#endif
6764
6765#ifdef FEAT_FLOAT
6766/*
6767 * "pow()" function
6768 */
6769 static void
6770f_pow(typval_T *argvars, typval_T *rettv)
6771{
6772 float_T fx = 0.0, fy = 0.0;
6773
6774 rettv->v_type = VAR_FLOAT;
6775 if (get_float_arg(argvars, &fx) == OK
6776 && get_float_arg(&argvars[1], &fy) == OK)
6777 rettv->vval.v_float = pow(fx, fy);
6778 else
6779 rettv->vval.v_float = 0.0;
6780}
6781#endif
6782
6783/*
6784 * "prevnonblank()" function
6785 */
6786 static void
6787f_prevnonblank(typval_T *argvars, typval_T *rettv)
6788{
6789 linenr_T lnum;
6790
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006791 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006792 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
6793 lnum = 0;
6794 else
6795 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
6796 --lnum;
6797 rettv->vval.v_number = lnum;
6798}
6799
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006800// This dummy va_list is here because:
6801// - passing a NULL pointer doesn't work when va_list isn't a pointer
6802// - locally in the function results in a "used before set" warning
6803// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006804static va_list ap;
6805
6806/*
6807 * "printf()" function
6808 */
6809 static void
6810f_printf(typval_T *argvars, typval_T *rettv)
6811{
6812 char_u buf[NUMBUFLEN];
6813 int len;
6814 char_u *s;
6815 int saved_did_emsg = did_emsg;
6816 char *fmt;
6817
6818 rettv->v_type = VAR_STRING;
6819 rettv->vval.v_string = NULL;
6820
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006821 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006822 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006823 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006824 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006825 if (!did_emsg)
6826 {
6827 s = alloc(len + 1);
6828 if (s != NULL)
6829 {
6830 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006831 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
6832 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006833 }
6834 }
6835 did_emsg |= saved_did_emsg;
6836}
6837
6838/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006839 * "pum_getpos()" function
6840 */
6841 static void
6842f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6843{
6844 if (rettv_dict_alloc(rettv) != OK)
6845 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006846 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006847}
6848
6849/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006850 * "pumvisible()" function
6851 */
6852 static void
6853f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6854{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006855 if (pum_visible())
6856 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006857}
6858
6859#ifdef FEAT_PYTHON3
6860/*
6861 * "py3eval()" function
6862 */
6863 static void
6864f_py3eval(typval_T *argvars, typval_T *rettv)
6865{
6866 char_u *str;
6867 char_u buf[NUMBUFLEN];
6868
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006869 if (check_restricted() || check_secure())
6870 return;
6871
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006872 if (p_pyx == 0)
6873 p_pyx = 3;
6874
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006875 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006876 do_py3eval(str, rettv);
6877}
6878#endif
6879
6880#ifdef FEAT_PYTHON
6881/*
6882 * "pyeval()" function
6883 */
6884 static void
6885f_pyeval(typval_T *argvars, typval_T *rettv)
6886{
6887 char_u *str;
6888 char_u buf[NUMBUFLEN];
6889
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006890 if (check_restricted() || check_secure())
6891 return;
6892
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006893 if (p_pyx == 0)
6894 p_pyx = 2;
6895
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006896 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006897 do_pyeval(str, rettv);
6898}
6899#endif
6900
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006901#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
6902/*
6903 * "pyxeval()" function
6904 */
6905 static void
6906f_pyxeval(typval_T *argvars, typval_T *rettv)
6907{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006908 if (check_restricted() || check_secure())
6909 return;
6910
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006911# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
6912 init_pyxversion();
6913 if (p_pyx == 2)
6914 f_pyeval(argvars, rettv);
6915 else
6916 f_py3eval(argvars, rettv);
6917# elif defined(FEAT_PYTHON)
6918 f_pyeval(argvars, rettv);
6919# elif defined(FEAT_PYTHON3)
6920 f_py3eval(argvars, rettv);
6921# endif
6922}
6923#endif
6924
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006925static UINT32_T srand_seed_for_testing = 0;
6926static int srand_seed_for_testing_is_used = FALSE;
6927
6928 static void
6929f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
6930{
6931 if (argvars[0].v_type == VAR_UNKNOWN)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006932 srand_seed_for_testing_is_used = FALSE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006933 else
6934 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006935 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
6936 srand_seed_for_testing_is_used = TRUE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006937 }
6938}
6939
6940 static void
6941init_srand(UINT32_T *x)
6942{
6943#ifndef MSWIN
6944 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6945#endif
6946
6947 if (srand_seed_for_testing_is_used)
6948 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006949 *x = srand_seed_for_testing;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006950 return;
6951 }
6952#ifndef MSWIN
6953 if (dev_urandom_state != FAIL)
6954 {
6955 int fd = open("/dev/urandom", O_RDONLY);
6956 struct {
6957 union {
6958 UINT32_T number;
6959 char bytes[sizeof(UINT32_T)];
6960 } contents;
6961 } buf;
6962
6963 // Attempt reading /dev/urandom.
6964 if (fd == -1)
6965 dev_urandom_state = FAIL;
6966 else
6967 {
6968 buf.contents.number = 0;
6969 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6970 != sizeof(UINT32_T))
6971 dev_urandom_state = FAIL;
6972 else
6973 {
6974 dev_urandom_state = OK;
6975 *x = buf.contents.number;
6976 }
6977 close(fd);
6978 }
6979 }
6980 if (dev_urandom_state != OK)
6981 // Reading /dev/urandom doesn't work, fall back to time().
6982#endif
6983 *x = vim_time();
6984}
6985
6986#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6987#define SPLITMIX32(x, z) ( \
6988 z = (x += 0x9e3779b9), \
6989 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6990 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6991 z ^ (z >> 16) \
6992 )
6993#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6994 result = ROTL(y * 5, 7) * 9; \
6995 t = y << 9; \
6996 z ^= x; \
6997 w ^= y; \
6998 y ^= z, x ^= w; \
6999 z ^= t; \
7000 w = ROTL(w, 11);
7001
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007002/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007003 * "rand()" function
7004 */
7005 static void
7006f_rand(typval_T *argvars, typval_T *rettv)
7007{
7008 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007009 static UINT32_T gx, gy, gz, gw;
7010 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007011 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar0fd797e2020-11-05 20:46:32 +01007012 UINT32_T x = 0, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007013
7014 if (argvars[0].v_type == VAR_UNKNOWN)
7015 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007016 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007017 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007018 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007019 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007020 init_srand(&x);
7021
7022 gx = SPLITMIX32(x, z);
7023 gy = SPLITMIX32(x, z);
7024 gz = SPLITMIX32(x, z);
7025 gw = SPLITMIX32(x, z);
7026 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007027 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007028
7029 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007030 }
7031 else if (argvars[0].v_type == VAR_LIST)
7032 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007033 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007034 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007035 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007036
7037 lx = list_find(l, 0L);
7038 ly = list_find(l, 1L);
7039 lz = list_find(l, 2L);
7040 lw = list_find(l, 3L);
7041 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
7042 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
7043 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
7044 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
7045 x = (UINT32_T)lx->li_tv.vval.v_number;
7046 y = (UINT32_T)ly->li_tv.vval.v_number;
7047 z = (UINT32_T)lz->li_tv.vval.v_number;
7048 w = (UINT32_T)lw->li_tv.vval.v_number;
7049
7050 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
7051
7052 lx->li_tv.vval.v_number = (varnumber_T)x;
7053 ly->li_tv.vval.v_number = (varnumber_T)y;
7054 lz->li_tv.vval.v_number = (varnumber_T)z;
7055 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007056 }
7057 else
7058 goto theend;
7059
7060 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007061 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007062 return;
7063
7064theend:
7065 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007066 rettv->v_type = VAR_NUMBER;
7067 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007068}
7069
7070/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007071 * "srand()" function
7072 */
7073 static void
7074f_srand(typval_T *argvars, typval_T *rettv)
7075{
7076 UINT32_T x = 0, z;
7077
7078 if (rettv_list_alloc(rettv) == FAIL)
7079 return;
7080 if (argvars[0].v_type == VAR_UNKNOWN)
7081 {
7082 init_srand(&x);
7083 }
7084 else
7085 {
7086 int error = FALSE;
7087
7088 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
7089 if (error)
7090 return;
7091 }
7092
7093 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7094 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7095 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7096 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7097}
7098
7099#undef ROTL
7100#undef SPLITMIX32
7101#undef SHUFFLE_XOSHIRO128STARSTAR
7102
7103/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007104 * "range()" function
7105 */
7106 static void
7107f_range(typval_T *argvars, typval_T *rettv)
7108{
7109 varnumber_T start;
7110 varnumber_T end;
7111 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007112 int error = FALSE;
7113
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007114 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007115 if (argvars[1].v_type == VAR_UNKNOWN)
7116 {
7117 end = start - 1;
7118 start = 0;
7119 }
7120 else
7121 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007122 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007123 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007124 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007125 }
7126
7127 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007128 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007129 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007130 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007131 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007132 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007133 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007134 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007135 list_T *list = rettv->vval.v_list;
7136
7137 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007138 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007139 // be called.
7140 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01007141 list->lv_u.nonmat.lv_start = start;
7142 list->lv_u.nonmat.lv_end = end;
7143 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007144 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007145 }
7146}
7147
7148/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007149 * Materialize "list".
7150 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007151 */
7152 void
7153range_list_materialize(list_T *list)
7154{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007155 varnumber_T start = list->lv_u.nonmat.lv_start;
7156 varnumber_T end = list->lv_u.nonmat.lv_end;
7157 int stride = list->lv_u.nonmat.lv_stride;
7158 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007159
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007160 list->lv_first = NULL;
7161 list->lv_u.mat.lv_last = NULL;
7162 list->lv_len = 0;
7163 list->lv_u.mat.lv_idx_item = NULL;
7164 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
7165 if (list_append_number(list, (varnumber_T)i) == FAIL)
7166 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007167}
7168
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007169/*
7170 * "getreginfo()" function
7171 */
7172 static void
7173f_getreginfo(typval_T *argvars, typval_T *rettv)
7174{
7175 char_u *strregname;
7176 int regname;
7177 char_u buf[NUMBUFLEN + 2];
7178 long reglen = 0;
7179 dict_T *dict;
7180 list_T *list;
7181
7182 if (argvars[0].v_type != VAR_UNKNOWN)
7183 {
7184 strregname = tv_get_string_chk(&argvars[0]);
7185 if (strregname == NULL)
7186 return;
7187 }
7188 else
7189 strregname = get_vim_var_str(VV_REG);
7190
7191 regname = (strregname == NULL ? '"' : *strregname);
7192 if (regname == 0 || regname == '@')
7193 regname = '"';
7194
7195 if (rettv_dict_alloc(rettv) == FAIL)
7196 return;
7197 dict = rettv->vval.v_dict;
7198
7199 list = (list_T *)get_reg_contents(regname, GREG_EXPR_SRC | GREG_LIST);
7200 if (list == NULL)
7201 return;
Bram Moolenaar91639192020-06-29 19:55:58 +02007202 (void)dict_add_list(dict, "regcontents", list);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007203
7204 buf[0] = NUL;
7205 buf[1] = NUL;
7206 switch (get_reg_type(regname, &reglen))
7207 {
7208 case MLINE: buf[0] = 'V'; break;
7209 case MCHAR: buf[0] = 'v'; break;
7210 case MBLOCK:
7211 vim_snprintf((char *)buf, sizeof(buf), "%c%ld", Ctrl_V,
7212 reglen + 1);
7213 break;
7214 }
Bram Moolenaar91639192020-06-29 19:55:58 +02007215 (void)dict_add_string(dict, (char *)"regtype", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007216
7217 buf[0] = get_register_name(get_unname_register());
7218 buf[1] = NUL;
7219 if (regname == '"')
Bram Moolenaar91639192020-06-29 19:55:58 +02007220 (void)dict_add_string(dict, (char *)"points_to", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007221 else
7222 {
7223 dictitem_T *item = dictitem_alloc((char_u *)"isunnamed");
7224
7225 if (item != NULL)
7226 {
7227 item->di_tv.v_type = VAR_SPECIAL;
7228 item->di_tv.vval.v_number = regname == buf[0]
7229 ? VVAL_TRUE : VVAL_FALSE;
Bram Moolenaar91639192020-06-29 19:55:58 +02007230 (void)dict_add(dict, item);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007231 }
7232 }
7233}
7234
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02007235 static void
7236return_register(int regname, typval_T *rettv)
7237{
7238 char_u buf[2] = {0, 0};
7239
7240 buf[0] = (char_u)regname;
7241 rettv->v_type = VAR_STRING;
7242 rettv->vval.v_string = vim_strsave(buf);
7243}
7244
7245/*
7246 * "reg_executing()" function
7247 */
7248 static void
7249f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
7250{
7251 return_register(reg_executing, rettv);
7252}
7253
7254/*
7255 * "reg_recording()" function
7256 */
7257 static void
7258f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
7259{
7260 return_register(reg_recording, rettv);
7261}
7262
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01007263/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007264 * "rename({from}, {to})" function
7265 */
7266 static void
7267f_rename(typval_T *argvars, typval_T *rettv)
7268{
7269 char_u buf[NUMBUFLEN];
7270
7271 if (check_restricted() || check_secure())
7272 rettv->vval.v_number = -1;
7273 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007274 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
7275 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007276}
7277
7278/*
7279 * "repeat()" function
7280 */
7281 static void
7282f_repeat(typval_T *argvars, typval_T *rettv)
7283{
7284 char_u *p;
7285 int n;
7286 int slen;
7287 int len;
7288 char_u *r;
7289 int i;
7290
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007291 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007292 if (argvars[0].v_type == VAR_LIST)
7293 {
7294 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
7295 while (n-- > 0)
7296 if (list_extend(rettv->vval.v_list,
7297 argvars[0].vval.v_list, NULL) == FAIL)
7298 break;
7299 }
7300 else
7301 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007302 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007303 rettv->v_type = VAR_STRING;
7304 rettv->vval.v_string = NULL;
7305
7306 slen = (int)STRLEN(p);
7307 len = slen * n;
7308 if (len <= 0)
7309 return;
7310
7311 r = alloc(len + 1);
7312 if (r != NULL)
7313 {
7314 for (i = 0; i < n; i++)
7315 mch_memmove(r + i * slen, p, (size_t)slen);
7316 r[len] = NUL;
7317 }
7318
7319 rettv->vval.v_string = r;
7320 }
7321}
7322
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007323#define SP_NOMOVE 0x01 // don't move cursor
7324#define SP_REPEAT 0x02 // repeat to find outer pair
7325#define SP_RETCOUNT 0x04 // return matchcount
7326#define SP_SETPCMARK 0x08 // set previous context mark
7327#define SP_START 0x10 // accept match at start position
7328#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
7329#define SP_END 0x40 // leave cursor at end of match
7330#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007331
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007332/*
7333 * Get flags for a search function.
7334 * Possibly sets "p_ws".
7335 * Returns BACKWARD, FORWARD or zero (for an error).
7336 */
7337 static int
7338get_search_arg(typval_T *varp, int *flagsp)
7339{
7340 int dir = FORWARD;
7341 char_u *flags;
7342 char_u nbuf[NUMBUFLEN];
7343 int mask;
7344
7345 if (varp->v_type != VAR_UNKNOWN)
7346 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007347 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007348 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007349 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007350 while (*flags != NUL)
7351 {
7352 switch (*flags)
7353 {
7354 case 'b': dir = BACKWARD; break;
7355 case 'w': p_ws = TRUE; break;
7356 case 'W': p_ws = FALSE; break;
7357 default: mask = 0;
7358 if (flagsp != NULL)
7359 switch (*flags)
7360 {
7361 case 'c': mask = SP_START; break;
7362 case 'e': mask = SP_END; break;
7363 case 'm': mask = SP_RETCOUNT; break;
7364 case 'n': mask = SP_NOMOVE; break;
7365 case 'p': mask = SP_SUBPAT; break;
7366 case 'r': mask = SP_REPEAT; break;
7367 case 's': mask = SP_SETPCMARK; break;
7368 case 'z': mask = SP_COLUMN; break;
7369 }
7370 if (mask == 0)
7371 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007372 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007373 dir = 0;
7374 }
7375 else
7376 *flagsp |= mask;
7377 }
7378 if (dir == 0)
7379 break;
7380 ++flags;
7381 }
7382 }
7383 return dir;
7384}
7385
7386/*
7387 * Shared by search() and searchpos() functions.
7388 */
7389 static int
7390search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
7391{
7392 int flags;
7393 char_u *pat;
7394 pos_T pos;
7395 pos_T save_cursor;
7396 int save_p_ws = p_ws;
7397 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007398 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007399 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007400#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007401 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007402 long time_limit = 0;
7403#endif
7404 int options = SEARCH_KEEP;
7405 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007406 searchit_arg_T sia;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007407 int use_skip = FALSE;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007408 pos_T firstpos;
7409
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007410 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007411 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007412 if (dir == 0)
7413 goto theend;
7414 flags = *flagsp;
7415 if (flags & SP_START)
7416 options |= SEARCH_START;
7417 if (flags & SP_END)
7418 options |= SEARCH_END;
7419 if (flags & SP_COLUMN)
7420 options |= SEARCH_COL;
7421
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007422 // Optional arguments: line number to stop searching, timeout and skip.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007423 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
7424 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007425 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007426 if (lnum_stop < 0)
7427 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007428 if (argvars[3].v_type != VAR_UNKNOWN)
7429 {
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007430#ifdef FEAT_RELTIME
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007431 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007432 if (time_limit < 0)
7433 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007434#endif
Bram Moolenaara9c01042020-06-07 14:50:50 +02007435 use_skip = eval_expr_valid_arg(&argvars[4]);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007436 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007437 }
7438
7439#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007440 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007441 profile_setlimit(time_limit, &tm);
7442#endif
7443
7444 /*
7445 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
7446 * Check to make sure only those flags are set.
7447 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
7448 * flags cannot be set. Check for that condition also.
7449 */
7450 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
7451 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7452 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007453 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007454 goto theend;
7455 }
7456
7457 pos = save_cursor = curwin->w_cursor;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007458 CLEAR_FIELD(firstpos);
Bram Moolenaara80faa82020-04-12 19:37:17 +02007459 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007460 sia.sa_stop_lnum = (linenr_T)lnum_stop;
7461#ifdef FEAT_RELTIME
7462 sia.sa_tm = &tm;
7463#endif
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007464
7465 // Repeat until {skip} returns FALSE.
7466 for (;;)
7467 {
7468 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007469 options, RE_SEARCH, &sia);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007470 // finding the first match again means there is no match where {skip}
7471 // evaluates to zero.
7472 if (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos))
7473 subpatnum = FAIL;
7474
Bram Moolenaara9c01042020-06-07 14:50:50 +02007475 if (subpatnum == FAIL || !use_skip)
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007476 // didn't find it or no skip argument
7477 break;
7478 firstpos = pos;
7479
Bram Moolenaara9c01042020-06-07 14:50:50 +02007480 // If the skip expression matches, ignore this match.
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007481 {
7482 int do_skip;
7483 int err;
7484 pos_T save_pos = curwin->w_cursor;
7485
7486 curwin->w_cursor = pos;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007487 err = FALSE;
7488 do_skip = eval_expr_to_bool(&argvars[4], &err);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007489 curwin->w_cursor = save_pos;
7490 if (err)
7491 {
7492 // Evaluating {skip} caused an error, break here.
7493 subpatnum = FAIL;
7494 break;
7495 }
7496 if (!do_skip)
7497 break;
7498 }
7499 }
7500
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007501 if (subpatnum != FAIL)
7502 {
7503 if (flags & SP_SUBPAT)
7504 retval = subpatnum;
7505 else
7506 retval = pos.lnum;
7507 if (flags & SP_SETPCMARK)
7508 setpcmark();
7509 curwin->w_cursor = pos;
7510 if (match_pos != NULL)
7511 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007512 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007513 match_pos->lnum = pos.lnum;
7514 match_pos->col = pos.col + 1;
7515 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007516 // "/$" will put the cursor after the end of the line, may need to
7517 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007518 check_cursor();
7519 }
7520
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007521 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007522 if (flags & SP_NOMOVE)
7523 curwin->w_cursor = save_cursor;
7524 else
7525 curwin->w_set_curswant = TRUE;
7526theend:
7527 p_ws = save_p_ws;
7528
7529 return retval;
7530}
7531
7532#ifdef FEAT_FLOAT
7533
7534/*
7535 * round() is not in C90, use ceil() or floor() instead.
7536 */
7537 float_T
7538vim_round(float_T f)
7539{
7540 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
7541}
7542
7543/*
7544 * "round({float})" function
7545 */
7546 static void
7547f_round(typval_T *argvars, typval_T *rettv)
7548{
7549 float_T f = 0.0;
7550
7551 rettv->v_type = VAR_FLOAT;
7552 if (get_float_arg(argvars, &f) == OK)
7553 rettv->vval.v_float = vim_round(f);
7554 else
7555 rettv->vval.v_float = 0.0;
7556}
7557#endif
7558
Bram Moolenaare99be0e2019-03-26 22:51:09 +01007559#ifdef FEAT_RUBY
7560/*
7561 * "rubyeval()" function
7562 */
7563 static void
7564f_rubyeval(typval_T *argvars, typval_T *rettv)
7565{
7566 char_u *str;
7567 char_u buf[NUMBUFLEN];
7568
7569 str = tv_get_string_buf(&argvars[0], buf);
7570 do_rubyeval(str, rettv);
7571}
7572#endif
7573
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007574/*
7575 * "screenattr()" function
7576 */
7577 static void
7578f_screenattr(typval_T *argvars, typval_T *rettv)
7579{
7580 int row;
7581 int col;
7582 int c;
7583
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007584 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7585 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007586 if (row < 0 || row >= screen_Rows
7587 || col < 0 || col >= screen_Columns)
7588 c = -1;
7589 else
7590 c = ScreenAttrs[LineOffset[row] + col];
7591 rettv->vval.v_number = c;
7592}
7593
7594/*
7595 * "screenchar()" function
7596 */
7597 static void
7598f_screenchar(typval_T *argvars, typval_T *rettv)
7599{
7600 int row;
7601 int col;
7602 int off;
7603 int c;
7604
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007605 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7606 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007607 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007608 c = -1;
7609 else
7610 {
7611 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007612 if (enc_utf8 && ScreenLinesUC[off] != 0)
7613 c = ScreenLinesUC[off];
7614 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007615 c = ScreenLines[off];
7616 }
7617 rettv->vval.v_number = c;
7618}
7619
7620/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007621 * "screenchars()" function
7622 */
7623 static void
7624f_screenchars(typval_T *argvars, typval_T *rettv)
7625{
7626 int row;
7627 int col;
7628 int off;
7629 int c;
7630 int i;
7631
7632 if (rettv_list_alloc(rettv) == FAIL)
7633 return;
7634 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7635 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7636 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7637 return;
7638
7639 off = LineOffset[row] + col;
7640 if (enc_utf8 && ScreenLinesUC[off] != 0)
7641 c = ScreenLinesUC[off];
7642 else
7643 c = ScreenLines[off];
7644 list_append_number(rettv->vval.v_list, (varnumber_T)c);
7645
7646 if (enc_utf8)
7647
7648 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7649 list_append_number(rettv->vval.v_list,
7650 (varnumber_T)ScreenLinesC[i][off]);
7651}
7652
7653/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007654 * "screencol()" function
7655 *
7656 * First column is 1 to be consistent with virtcol().
7657 */
7658 static void
7659f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
7660{
7661 rettv->vval.v_number = screen_screencol() + 1;
7662}
7663
7664/*
7665 * "screenrow()" function
7666 */
7667 static void
7668f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
7669{
7670 rettv->vval.v_number = screen_screenrow() + 1;
7671}
7672
7673/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007674 * "screenstring()" function
7675 */
7676 static void
7677f_screenstring(typval_T *argvars, typval_T *rettv)
7678{
7679 int row;
7680 int col;
7681 int off;
7682 int c;
7683 int i;
7684 char_u buf[MB_MAXBYTES + 1];
7685 int buflen = 0;
7686
7687 rettv->vval.v_string = NULL;
7688 rettv->v_type = VAR_STRING;
7689
7690 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7691 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7692 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7693 return;
7694
7695 off = LineOffset[row] + col;
7696 if (enc_utf8 && ScreenLinesUC[off] != 0)
7697 c = ScreenLinesUC[off];
7698 else
7699 c = ScreenLines[off];
7700 buflen += mb_char2bytes(c, buf);
7701
7702 if (enc_utf8)
7703 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7704 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
7705
7706 buf[buflen] = NUL;
7707 rettv->vval.v_string = vim_strsave(buf);
7708}
7709
7710/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007711 * "search()" function
7712 */
7713 static void
7714f_search(typval_T *argvars, typval_T *rettv)
7715{
7716 int flags = 0;
7717
7718 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
7719}
7720
7721/*
7722 * "searchdecl()" function
7723 */
7724 static void
7725f_searchdecl(typval_T *argvars, typval_T *rettv)
7726{
Bram Moolenaar30788d32020-09-05 21:35:16 +02007727 int locally = TRUE;
7728 int thisblock = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007729 int error = FALSE;
7730 char_u *name;
7731
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007732 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007733
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007734 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007735 if (argvars[1].v_type != VAR_UNKNOWN)
7736 {
Bram Moolenaar30788d32020-09-05 21:35:16 +02007737 locally = !(int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007738 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar30788d32020-09-05 21:35:16 +02007739 thisblock = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007740 }
7741 if (!error && name != NULL)
7742 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
7743 locally, thisblock, SEARCH_KEEP) == FAIL;
7744}
7745
7746/*
7747 * Used by searchpair() and searchpairpos()
7748 */
7749 static int
7750searchpair_cmn(typval_T *argvars, pos_T *match_pos)
7751{
7752 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01007753 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007754 int save_p_ws = p_ws;
7755 int dir;
7756 int flags = 0;
7757 char_u nbuf1[NUMBUFLEN];
7758 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007759 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007760 long lnum_stop = 0;
7761 long time_limit = 0;
7762
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007763 // Get the three pattern arguments: start, middle, end. Will result in an
7764 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007765 spat = tv_get_string_chk(&argvars[0]);
7766 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
7767 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007768 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007769 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007770
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007771 // Handle the optional fourth argument: flags
7772 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007773 if (dir == 0)
7774 goto theend;
7775
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007776 // Don't accept SP_END or SP_SUBPAT.
7777 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007778 if ((flags & (SP_END | SP_SUBPAT)) != 0
7779 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7780 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007781 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007782 goto theend;
7783 }
7784
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007785 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007786 if (flags & SP_REPEAT)
7787 p_ws = FALSE;
7788
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007789 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007790 if (argvars[3].v_type == VAR_UNKNOWN
7791 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01007792 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007793 else
7794 {
Bram Moolenaara9c01042020-06-07 14:50:50 +02007795 // Type is checked later.
Bram Moolenaar48570482017-10-30 21:48:41 +01007796 skip = &argvars[4];
Bram Moolenaara9c01042020-06-07 14:50:50 +02007797
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007798 if (argvars[5].v_type != VAR_UNKNOWN)
7799 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007800 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007801 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007802 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007803 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007804 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007805 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007806#ifdef FEAT_RELTIME
7807 if (argvars[6].v_type != VAR_UNKNOWN)
7808 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007809 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007810 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007811 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007812 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007813 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007814 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007815 }
7816#endif
7817 }
7818 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007819
7820 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
7821 match_pos, lnum_stop, time_limit);
7822
7823theend:
7824 p_ws = save_p_ws;
7825
7826 return retval;
7827}
7828
7829/*
7830 * "searchpair()" function
7831 */
7832 static void
7833f_searchpair(typval_T *argvars, typval_T *rettv)
7834{
7835 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
7836}
7837
7838/*
7839 * "searchpairpos()" function
7840 */
7841 static void
7842f_searchpairpos(typval_T *argvars, typval_T *rettv)
7843{
7844 pos_T match_pos;
7845 int lnum = 0;
7846 int col = 0;
7847
7848 if (rettv_list_alloc(rettv) == FAIL)
7849 return;
7850
7851 if (searchpair_cmn(argvars, &match_pos) > 0)
7852 {
7853 lnum = match_pos.lnum;
7854 col = match_pos.col;
7855 }
7856
7857 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7858 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7859}
7860
7861/*
7862 * Search for a start/middle/end thing.
7863 * Used by searchpair(), see its documentation for the details.
7864 * Returns 0 or -1 for no match,
7865 */
7866 long
7867do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007868 char_u *spat, // start pattern
7869 char_u *mpat, // middle pattern
7870 char_u *epat, // end pattern
7871 int dir, // BACKWARD or FORWARD
7872 typval_T *skip, // skip expression
7873 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007874 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007875 linenr_T lnum_stop, // stop at this line if not zero
7876 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007877{
7878 char_u *save_cpo;
7879 char_u *pat, *pat2 = NULL, *pat3 = NULL;
7880 long retval = 0;
7881 pos_T pos;
7882 pos_T firstpos;
7883 pos_T foundpos;
7884 pos_T save_cursor;
7885 pos_T save_pos;
7886 int n;
7887 int r;
7888 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01007889 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007890 int err;
7891 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007892#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007893 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007894#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007895
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007896 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007897 save_cpo = p_cpo;
7898 p_cpo = empty_option;
7899
7900#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007901 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007902 profile_setlimit(time_limit, &tm);
7903#endif
7904
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007905 // Make two search patterns: start/end (pat2, for in nested pairs) and
7906 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02007907 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
7908 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007909 if (pat2 == NULL || pat3 == NULL)
7910 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01007911 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007912 if (*mpat == NUL)
7913 STRCPY(pat3, pat2);
7914 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01007915 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007916 spat, epat, mpat);
7917 if (flags & SP_START)
7918 options |= SEARCH_START;
7919
Bram Moolenaar48570482017-10-30 21:48:41 +01007920 if (skip != NULL)
Bram Moolenaara9c01042020-06-07 14:50:50 +02007921 use_skip = eval_expr_valid_arg(skip);
Bram Moolenaar48570482017-10-30 21:48:41 +01007922
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007923 save_cursor = curwin->w_cursor;
7924 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007925 CLEAR_POS(&firstpos);
7926 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007927 pat = pat3;
7928 for (;;)
7929 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007930 searchit_arg_T sia;
7931
Bram Moolenaara80faa82020-04-12 19:37:17 +02007932 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007933 sia.sa_stop_lnum = lnum_stop;
7934#ifdef FEAT_RELTIME
7935 sia.sa_tm = &tm;
7936#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01007937 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007938 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007939 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007940 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007941 break;
7942
7943 if (firstpos.lnum == 0)
7944 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007945 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007946 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007947 // Found the same position again. Can happen with a pattern that
7948 // has "\zs" at the end and searching backwards. Advance one
7949 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007950 if (dir == BACKWARD)
7951 decl(&pos);
7952 else
7953 incl(&pos);
7954 }
7955 foundpos = pos;
7956
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007957 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007958 options &= ~SEARCH_START;
7959
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007960 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01007961 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007962 {
7963 save_pos = curwin->w_cursor;
7964 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01007965 err = FALSE;
7966 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007967 curwin->w_cursor = save_pos;
7968 if (err)
7969 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007970 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007971 curwin->w_cursor = save_cursor;
7972 retval = -1;
7973 break;
7974 }
7975 if (r)
7976 continue;
7977 }
7978
7979 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
7980 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007981 // Found end when searching backwards or start when searching
7982 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007983 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007984 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007985 }
7986 else
7987 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007988 // Found end when searching forward or start when searching
7989 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007990 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007991 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007992 }
7993
7994 if (nest == 0)
7995 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007996 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007997 if (flags & SP_RETCOUNT)
7998 ++retval;
7999 else
8000 retval = pos.lnum;
8001 if (flags & SP_SETPCMARK)
8002 setpcmark();
8003 curwin->w_cursor = pos;
8004 if (!(flags & SP_REPEAT))
8005 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008006 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008007 }
8008 }
8009
8010 if (match_pos != NULL)
8011 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008012 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008013 match_pos->lnum = curwin->w_cursor.lnum;
8014 match_pos->col = curwin->w_cursor.col + 1;
8015 }
8016
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008017 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008018 if ((flags & SP_NOMOVE) || retval == 0)
8019 curwin->w_cursor = save_cursor;
8020
8021theend:
8022 vim_free(pat2);
8023 vim_free(pat3);
8024 if (p_cpo == empty_option)
8025 p_cpo = save_cpo;
8026 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008027 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008028 free_string_option(save_cpo);
8029
8030 return retval;
8031}
8032
8033/*
8034 * "searchpos()" function
8035 */
8036 static void
8037f_searchpos(typval_T *argvars, typval_T *rettv)
8038{
8039 pos_T match_pos;
8040 int lnum = 0;
8041 int col = 0;
8042 int n;
8043 int flags = 0;
8044
8045 if (rettv_list_alloc(rettv) == FAIL)
8046 return;
8047
8048 n = search_cmn(argvars, &match_pos, &flags);
8049 if (n > 0)
8050 {
8051 lnum = match_pos.lnum;
8052 col = match_pos.col;
8053 }
8054
8055 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
8056 list_append_number(rettv->vval.v_list, (varnumber_T)col);
8057 if (flags & SP_SUBPAT)
8058 list_append_number(rettv->vval.v_list, (varnumber_T)n);
8059}
8060
8061 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008062f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
8063{
8064 dict_T *d;
8065 dictitem_T *di;
8066 char_u *csearch;
8067
8068 if (argvars[0].v_type != VAR_DICT)
8069 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008070 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008071 return;
8072 }
8073
8074 if ((d = argvars[0].vval.v_dict) != NULL)
8075 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01008076 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008077 if (csearch != NULL)
8078 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008079 if (enc_utf8)
8080 {
8081 int pcc[MAX_MCO];
8082 int c = utfc_ptr2char(csearch, pcc);
8083
8084 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
8085 }
8086 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008087 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02008088 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008089 }
8090
8091 di = dict_find(d, (char_u *)"forward", -1);
8092 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008093 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008094 ? FORWARD : BACKWARD);
8095
8096 di = dict_find(d, (char_u *)"until", -1);
8097 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008098 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008099 }
8100}
8101
8102/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02008103 * "setenv()" function
8104 */
8105 static void
8106f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
8107{
8108 char_u namebuf[NUMBUFLEN];
8109 char_u valbuf[NUMBUFLEN];
8110 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
8111
8112 if (argvars[1].v_type == VAR_SPECIAL
8113 && argvars[1].vval.v_number == VVAL_NULL)
8114 vim_unsetenv(name);
8115 else
8116 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
8117}
8118
8119/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008120 * "setfperm({fname}, {mode})" function
8121 */
8122 static void
8123f_setfperm(typval_T *argvars, typval_T *rettv)
8124{
8125 char_u *fname;
8126 char_u modebuf[NUMBUFLEN];
8127 char_u *mode_str;
8128 int i;
8129 int mask;
8130 int mode = 0;
8131
8132 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008133 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008134 if (fname == NULL)
8135 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008136 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008137 if (mode_str == NULL)
8138 return;
8139 if (STRLEN(mode_str) != 9)
8140 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008141 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008142 return;
8143 }
8144
8145 mask = 1;
8146 for (i = 8; i >= 0; --i)
8147 {
8148 if (mode_str[i] != '-')
8149 mode |= mask;
8150 mask = mask << 1;
8151 }
8152 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
8153}
8154
8155/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008156 * "setpos()" function
8157 */
8158 static void
8159f_setpos(typval_T *argvars, typval_T *rettv)
8160{
8161 pos_T pos;
8162 int fnum;
8163 char_u *name;
8164 colnr_T curswant = -1;
8165
8166 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008167 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008168 if (name != NULL)
8169 {
8170 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
8171 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008172 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008173 pos.col = 0;
8174 if (name[0] == '.' && name[1] == NUL)
8175 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008176 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01008177 curwin->w_cursor = pos;
8178 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008179 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01008180 curwin->w_curswant = curswant - 1;
8181 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008182 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01008183 check_cursor();
8184 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008185 }
8186 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
8187 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008188 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008189 if (setmark_pos(name[1], &pos, fnum) == OK)
8190 rettv->vval.v_number = 0;
8191 }
8192 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008193 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008194 }
8195 }
8196}
8197
8198/*
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008199 * Translate a register type string to the yank type and block length
8200 */
8201 static int
8202get_yank_type(char_u **pp, char_u *yank_type, long *block_len)
8203{
8204 char_u *stropt = *pp;
8205 switch (*stropt)
8206 {
8207 case 'v': case 'c': // character-wise selection
8208 *yank_type = MCHAR;
8209 break;
8210 case 'V': case 'l': // line-wise selection
8211 *yank_type = MLINE;
8212 break;
8213 case 'b': case Ctrl_V: // block-wise selection
8214 *yank_type = MBLOCK;
8215 if (VIM_ISDIGIT(stropt[1]))
8216 {
8217 ++stropt;
8218 *block_len = getdigits(&stropt) - 1;
8219 --stropt;
8220 }
8221 break;
8222 default:
8223 return FAIL;
8224 }
8225 *pp = stropt;
8226 return OK;
8227}
8228
8229/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008230 * "setreg()" function
8231 */
8232 static void
8233f_setreg(typval_T *argvars, typval_T *rettv)
8234{
8235 int regname;
8236 char_u *strregname;
8237 char_u *stropt;
8238 char_u *strval;
8239 int append;
8240 char_u yank_type;
8241 long block_len;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008242 typval_T *regcontents;
8243 int pointreg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008244
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008245 pointreg = 0;
8246 regcontents = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008247 block_len = -1;
8248 yank_type = MAUTO;
8249 append = FALSE;
8250
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008251 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008252 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008253
8254 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008255 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008256 regname = *strregname;
8257 if (regname == 0 || regname == '@')
8258 regname = '"';
8259
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008260 if (argvars[1].v_type == VAR_DICT)
8261 {
8262 dict_T *d = argvars[1].vval.v_dict;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008263 dictitem_T *di;
8264
8265 if (d == NULL || d->dv_hashtab.ht_used == 0)
8266 {
8267 // Empty dict, clear the register (like setreg(0, []))
8268 char_u *lstval[2] = {NULL, NULL};
8269 write_reg_contents_lst(regname, lstval, 0, FALSE, MAUTO, -1);
8270 return;
8271 }
8272
8273 di = dict_find(d, (char_u *)"regcontents", -1);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008274 if (di != NULL)
8275 regcontents = &di->di_tv;
8276
8277 stropt = dict_get_string(d, (char_u *)"regtype", FALSE);
8278 if (stropt != NULL)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008279 {
8280 int ret = get_yank_type(&stropt, &yank_type, &block_len);
8281
8282 if (ret == FAIL || *++stropt != NUL)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008283 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008284 semsg(_(e_invargval), "value");
8285 return;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008286 }
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008287 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008288
8289 if (regname == '"')
8290 {
8291 stropt = dict_get_string(d, (char_u *)"points_to", FALSE);
8292 if (stropt != NULL)
8293 {
8294 pointreg = *stropt;
8295 regname = pointreg;
8296 }
8297 }
Bram Moolenaar6a950582020-08-28 16:39:33 +02008298 else if (dict_get_bool(d, (char_u *)"isunnamed", -1) > 0)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008299 pointreg = regname;
8300 }
8301 else
8302 regcontents = &argvars[1];
8303
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008304 if (argvars[2].v_type != VAR_UNKNOWN)
8305 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008306 if (yank_type != MAUTO)
8307 {
8308 semsg(_(e_toomanyarg), "setreg");
8309 return;
8310 }
8311
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008312 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008313 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008314 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008315 for (; *stropt != NUL; ++stropt)
8316 switch (*stropt)
8317 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008318 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008319 append = TRUE;
8320 break;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008321 default:
8322 get_yank_type(&stropt, &yank_type, &block_len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008323 }
8324 }
8325
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008326 if (regcontents && regcontents->v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008327 {
8328 char_u **lstval;
8329 char_u **allocval;
8330 char_u buf[NUMBUFLEN];
8331 char_u **curval;
8332 char_u **curallocval;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008333 list_T *ll = regcontents->vval.v_list;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008334 listitem_T *li;
8335 int len;
8336
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008337 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008338 len = ll == NULL ? 0 : ll->lv_len;
8339
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008340 // First half: use for pointers to result lines; second half: use for
8341 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02008342 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008343 if (lstval == NULL)
8344 return;
8345 curval = lstval;
8346 allocval = lstval + len + 2;
8347 curallocval = allocval;
8348
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008349 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008350 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02008351 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02008352 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008353 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008354 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008355 if (strval == NULL)
8356 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008357 if (strval == buf)
8358 {
8359 // Need to make a copy, next tv_get_string_buf_chk() will
8360 // overwrite the string.
8361 strval = vim_strsave(buf);
8362 if (strval == NULL)
8363 goto free_lstval;
8364 *curallocval++ = strval;
8365 }
8366 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008367 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008368 }
8369 *curval++ = NULL;
8370
8371 write_reg_contents_lst(regname, lstval, -1,
8372 append, yank_type, block_len);
8373free_lstval:
8374 while (curallocval > allocval)
8375 vim_free(*--curallocval);
8376 vim_free(lstval);
8377 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008378 else if (regcontents)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008379 {
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008380 strval = tv_get_string_chk(regcontents);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008381 if (strval == NULL)
8382 return;
8383 write_reg_contents_ex(regname, strval, -1,
8384 append, yank_type, block_len);
8385 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008386 if (pointreg != 0)
8387 get_yank_register(pointreg, TRUE);
8388
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008389 rettv->vval.v_number = 0;
8390}
8391
8392/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008393 * "settagstack()" function
8394 */
8395 static void
8396f_settagstack(typval_T *argvars, typval_T *rettv)
8397{
8398 static char *e_invact2 = N_("E962: Invalid action: '%s'");
8399 win_T *wp;
8400 dict_T *d;
8401 int action = 'r';
8402
8403 rettv->vval.v_number = -1;
8404
8405 // first argument: window number or id
8406 wp = find_win_by_nr_or_id(&argvars[0]);
8407 if (wp == NULL)
8408 return;
8409
8410 // second argument: dict with items to set in the tag stack
8411 if (argvars[1].v_type != VAR_DICT)
8412 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008413 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008414 return;
8415 }
8416 d = argvars[1].vval.v_dict;
8417 if (d == NULL)
8418 return;
8419
8420 // third argument: action - 'a' for append and 'r' for replace.
8421 // default is to replace the stack.
8422 if (argvars[2].v_type == VAR_UNKNOWN)
8423 action = 'r';
8424 else if (argvars[2].v_type == VAR_STRING)
8425 {
8426 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008427 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008428 if (actstr == NULL)
8429 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01008430 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
8431 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008432 action = *actstr;
8433 else
8434 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008435 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008436 return;
8437 }
8438 }
8439 else
8440 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008441 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008442 return;
8443 }
8444
8445 if (set_tagstack(wp, d, action) == OK)
8446 rettv->vval.v_number = 0;
8447}
8448
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008449#ifdef FEAT_CRYPT
8450/*
8451 * "sha256({string})" function
8452 */
8453 static void
8454f_sha256(typval_T *argvars, typval_T *rettv)
8455{
8456 char_u *p;
8457
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008458 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008459 rettv->vval.v_string = vim_strsave(
8460 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
8461 rettv->v_type = VAR_STRING;
8462}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008463#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008464
8465/*
8466 * "shellescape({string})" function
8467 */
8468 static void
8469f_shellescape(typval_T *argvars, typval_T *rettv)
8470{
Bram Moolenaar20615522017-06-05 18:46:26 +02008471 int do_special = non_zero_arg(&argvars[1]);
8472
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008473 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008474 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008475 rettv->v_type = VAR_STRING;
8476}
8477
8478/*
8479 * shiftwidth() function
8480 */
8481 static void
8482f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
8483{
Bram Moolenaarf9514162018-11-22 03:08:29 +01008484 rettv->vval.v_number = 0;
8485
8486 if (argvars[0].v_type != VAR_UNKNOWN)
8487 {
8488 long col;
8489
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008490 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01008491 if (col < 0)
8492 return; // type error; errmsg already given
8493#ifdef FEAT_VARTABS
8494 rettv->vval.v_number = get_sw_value_col(curbuf, col);
8495 return;
8496#endif
8497 }
8498
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008499 rettv->vval.v_number = get_sw_value(curbuf);
8500}
8501
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008502#ifdef FEAT_FLOAT
8503/*
8504 * "sin()" function
8505 */
8506 static void
8507f_sin(typval_T *argvars, typval_T *rettv)
8508{
8509 float_T f = 0.0;
8510
8511 rettv->v_type = VAR_FLOAT;
8512 if (get_float_arg(argvars, &f) == OK)
8513 rettv->vval.v_float = sin(f);
8514 else
8515 rettv->vval.v_float = 0.0;
8516}
8517
8518/*
8519 * "sinh()" function
8520 */
8521 static void
8522f_sinh(typval_T *argvars, typval_T *rettv)
8523{
8524 float_T f = 0.0;
8525
8526 rettv->v_type = VAR_FLOAT;
8527 if (get_float_arg(argvars, &f) == OK)
8528 rettv->vval.v_float = sinh(f);
8529 else
8530 rettv->vval.v_float = 0.0;
8531}
8532#endif
8533
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008534/*
8535 * "soundfold({word})" function
8536 */
8537 static void
8538f_soundfold(typval_T *argvars, typval_T *rettv)
8539{
8540 char_u *s;
8541
8542 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008543 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008544#ifdef FEAT_SPELL
8545 rettv->vval.v_string = eval_soundfold(s);
8546#else
8547 rettv->vval.v_string = vim_strsave(s);
8548#endif
8549}
8550
8551/*
8552 * "spellbadword()" function
8553 */
8554 static void
8555f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
8556{
8557 char_u *word = (char_u *)"";
8558 hlf_T attr = HLF_COUNT;
8559 int len = 0;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008560#ifdef FEAT_SPELL
8561 int wo_spell_save = curwin->w_p_spell;
8562
8563 if (!curwin->w_p_spell)
8564 {
8565 did_set_spelllang(curwin);
8566 curwin->w_p_spell = TRUE;
8567 }
8568
8569 if (*curwin->w_s->b_p_spl == NUL)
8570 {
8571 emsg(_(e_no_spell));
8572 curwin->w_p_spell = wo_spell_save;
8573 return;
8574 }
8575#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008576
8577 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008578 {
8579#ifdef FEAT_SPELL
8580 curwin->w_p_spell = wo_spell_save;
8581#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008582 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008583 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008584
8585#ifdef FEAT_SPELL
8586 if (argvars[0].v_type == VAR_UNKNOWN)
8587 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008588 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008589 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
8590 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008591 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008592 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008593 curwin->w_set_curswant = TRUE;
8594 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008595 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008596 else if (*curbuf->b_s.b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008597 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008598 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008599 int capcol = -1;
8600
8601 if (str != NULL)
8602 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008603 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008604 while (*str != NUL)
8605 {
8606 len = spell_check(curwin, str, &attr, &capcol, FALSE);
8607 if (attr != HLF_COUNT)
8608 {
8609 word = str;
8610 break;
8611 }
8612 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02008613 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02008614 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008615 }
8616 }
8617 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008618 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008619#endif
8620
8621 list_append_string(rettv->vval.v_list, word, len);
8622 list_append_string(rettv->vval.v_list, (char_u *)(
8623 attr == HLF_SPB ? "bad" :
8624 attr == HLF_SPR ? "rare" :
8625 attr == HLF_SPL ? "local" :
8626 attr == HLF_SPC ? "caps" :
8627 ""), -1);
8628}
8629
8630/*
8631 * "spellsuggest()" function
8632 */
8633 static void
8634f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
8635{
8636#ifdef FEAT_SPELL
8637 char_u *str;
8638 int typeerr = FALSE;
8639 int maxcount;
8640 garray_T ga;
8641 int i;
8642 listitem_T *li;
8643 int need_capital = FALSE;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008644 int wo_spell_save = curwin->w_p_spell;
8645
8646 if (!curwin->w_p_spell)
8647 {
8648 did_set_spelllang(curwin);
8649 curwin->w_p_spell = TRUE;
8650 }
8651
8652 if (*curwin->w_s->b_p_spl == NUL)
8653 {
8654 emsg(_(e_no_spell));
8655 curwin->w_p_spell = wo_spell_save;
8656 return;
8657 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008658#endif
8659
8660 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008661 {
8662#ifdef FEAT_SPELL
8663 curwin->w_p_spell = wo_spell_save;
8664#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008665 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008666 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008667
8668#ifdef FEAT_SPELL
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008669 if (*curwin->w_s->b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008670 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008671 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008672 if (argvars[1].v_type != VAR_UNKNOWN)
8673 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008674 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008675 if (maxcount <= 0)
8676 return;
8677 if (argvars[2].v_type != VAR_UNKNOWN)
8678 {
Bram Moolenaar7c27f332020-09-05 22:45:55 +02008679 need_capital = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008680 if (typeerr)
8681 return;
8682 }
8683 }
8684 else
8685 maxcount = 25;
8686
8687 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
8688
8689 for (i = 0; i < ga.ga_len; ++i)
8690 {
8691 str = ((char_u **)ga.ga_data)[i];
8692
8693 li = listitem_alloc();
8694 if (li == NULL)
8695 vim_free(str);
8696 else
8697 {
8698 li->li_tv.v_type = VAR_STRING;
8699 li->li_tv.v_lock = 0;
8700 li->li_tv.vval.v_string = str;
8701 list_append(rettv->vval.v_list, li);
8702 }
8703 }
8704 ga_clear(&ga);
8705 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008706 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008707#endif
8708}
8709
8710 static void
8711f_split(typval_T *argvars, typval_T *rettv)
8712{
8713 char_u *str;
8714 char_u *end;
8715 char_u *pat = NULL;
8716 regmatch_T regmatch;
8717 char_u patbuf[NUMBUFLEN];
8718 char_u *save_cpo;
8719 int match;
8720 colnr_T col = 0;
8721 int keepempty = FALSE;
8722 int typeerr = FALSE;
8723
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008724 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008725 save_cpo = p_cpo;
8726 p_cpo = (char_u *)"";
8727
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008728 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008729 if (argvars[1].v_type != VAR_UNKNOWN)
8730 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008731 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008732 if (pat == NULL)
8733 typeerr = TRUE;
8734 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02008735 keepempty = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008736 }
8737 if (pat == NULL || *pat == NUL)
8738 pat = (char_u *)"[\\x01- ]\\+";
8739
8740 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008741 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008742 if (typeerr)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008743 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008744
8745 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
8746 if (regmatch.regprog != NULL)
8747 {
8748 regmatch.rm_ic = FALSE;
8749 while (*str != NUL || keepempty)
8750 {
8751 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008752 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008753 else
8754 match = vim_regexec_nl(&regmatch, str, col);
8755 if (match)
8756 end = regmatch.startp[0];
8757 else
8758 end = str + STRLEN(str);
8759 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
8760 && *str != NUL && match && end < regmatch.endp[0]))
8761 {
8762 if (list_append_string(rettv->vval.v_list, str,
8763 (int)(end - str)) == FAIL)
8764 break;
8765 }
8766 if (!match)
8767 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01008768 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008769 if (regmatch.endp[0] > str)
8770 col = 0;
8771 else
Bram Moolenaar13505972019-01-24 15:04:48 +01008772 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008773 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008774 str = regmatch.endp[0];
8775 }
8776
8777 vim_regfree(regmatch.regprog);
8778 }
8779
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008780theend:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008781 p_cpo = save_cpo;
8782}
8783
8784#ifdef FEAT_FLOAT
8785/*
8786 * "sqrt()" function
8787 */
8788 static void
8789f_sqrt(typval_T *argvars, typval_T *rettv)
8790{
8791 float_T f = 0.0;
8792
8793 rettv->v_type = VAR_FLOAT;
8794 if (get_float_arg(argvars, &f) == OK)
8795 rettv->vval.v_float = sqrt(f);
8796 else
8797 rettv->vval.v_float = 0.0;
8798}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008799#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008800
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008801#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01008802/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008803 * "str2float()" function
8804 */
8805 static void
8806f_str2float(typval_T *argvars, typval_T *rettv)
8807{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008808 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008809 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008810
Bram Moolenaar08243d22017-01-10 16:12:29 +01008811 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008812 p = skipwhite(p + 1);
8813 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01008814 if (isneg)
8815 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008816 rettv->v_type = VAR_FLOAT;
8817}
8818#endif
8819
8820/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02008821 * "str2list()" function
8822 */
8823 static void
8824f_str2list(typval_T *argvars, typval_T *rettv)
8825{
8826 char_u *p;
8827 int utf8 = FALSE;
8828
8829 if (rettv_list_alloc(rettv) == FAIL)
8830 return;
8831
8832 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaara48f7862020-09-05 20:16:57 +02008833 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar9d401282019-04-06 13:18:12 +02008834
8835 p = tv_get_string(&argvars[0]);
8836
8837 if (has_mbyte || utf8)
8838 {
8839 int (*ptr2len)(char_u *);
8840 int (*ptr2char)(char_u *);
8841
8842 if (utf8 || enc_utf8)
8843 {
8844 ptr2len = utf_ptr2len;
8845 ptr2char = utf_ptr2char;
8846 }
8847 else
8848 {
8849 ptr2len = mb_ptr2len;
8850 ptr2char = mb_ptr2char;
8851 }
8852
8853 for ( ; *p != NUL; p += (*ptr2len)(p))
8854 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
8855 }
8856 else
8857 for ( ; *p != NUL; ++p)
8858 list_append_number(rettv->vval.v_list, *p);
8859}
8860
8861/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008862 * "str2nr()" function
8863 */
8864 static void
8865f_str2nr(typval_T *argvars, typval_T *rettv)
8866{
8867 int base = 10;
8868 char_u *p;
8869 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008870 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01008871 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008872
8873 if (argvars[1].v_type != VAR_UNKNOWN)
8874 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008875 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008876 if (base != 2 && base != 8 && base != 10 && base != 16)
8877 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008878 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008879 return;
8880 }
Bram Moolenaar3986b942020-09-06 16:09:04 +02008881 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[2]))
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008882 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008883 }
8884
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008885 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008886 isneg = (*p == '-');
8887 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008888 p = skipwhite(p + 1);
8889 switch (base)
8890 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008891 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02008892 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008893 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008894 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02008895 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
8896 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01008897 if (isneg)
8898 rettv->vval.v_number = -n;
8899 else
8900 rettv->vval.v_number = n;
8901
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008902}
8903
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008904/*
8905 * "strgetchar()" function
8906 */
8907 static void
8908f_strgetchar(typval_T *argvars, typval_T *rettv)
8909{
8910 char_u *str;
8911 int len;
8912 int error = FALSE;
8913 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01008914 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008915
8916 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008917 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008918 if (str == NULL)
8919 return;
8920 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008921 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008922 if (error)
8923 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008924
Bram Moolenaar13505972019-01-24 15:04:48 +01008925 while (charidx >= 0 && byteidx < len)
8926 {
8927 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008928 {
Bram Moolenaar13505972019-01-24 15:04:48 +01008929 rettv->vval.v_number = mb_ptr2char(str + byteidx);
8930 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008931 }
Bram Moolenaar13505972019-01-24 15:04:48 +01008932 --charidx;
8933 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008934 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008935}
8936
8937/*
8938 * "stridx()" function
8939 */
8940 static void
8941f_stridx(typval_T *argvars, typval_T *rettv)
8942{
8943 char_u buf[NUMBUFLEN];
8944 char_u *needle;
8945 char_u *haystack;
8946 char_u *save_haystack;
8947 char_u *pos;
8948 int start_idx;
8949
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008950 needle = tv_get_string_chk(&argvars[1]);
8951 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008952 rettv->vval.v_number = -1;
8953 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008954 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008955
8956 if (argvars[2].v_type != VAR_UNKNOWN)
8957 {
8958 int error = FALSE;
8959
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008960 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008961 if (error || start_idx >= (int)STRLEN(haystack))
8962 return;
8963 if (start_idx >= 0)
8964 haystack += start_idx;
8965 }
8966
8967 pos = (char_u *)strstr((char *)haystack, (char *)needle);
8968 if (pos != NULL)
8969 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
8970}
8971
8972/*
8973 * "string()" function
8974 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01008975 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008976f_string(typval_T *argvars, typval_T *rettv)
8977{
8978 char_u *tofree;
8979 char_u numbuf[NUMBUFLEN];
8980
8981 rettv->v_type = VAR_STRING;
8982 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
8983 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008984 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008985 if (rettv->vval.v_string != NULL && tofree == NULL)
8986 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
8987}
8988
8989/*
8990 * "strlen()" function
8991 */
8992 static void
8993f_strlen(typval_T *argvars, typval_T *rettv)
8994{
8995 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008996 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008997}
8998
8999/*
9000 * "strchars()" function
9001 */
9002 static void
9003f_strchars(typval_T *argvars, typval_T *rettv)
9004{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009005 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar3986b942020-09-06 16:09:04 +02009006 int skipcc = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009007 varnumber_T len = 0;
9008 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009009
9010 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02009011 skipcc = (int)tv_get_bool(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009012 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02009013 semsg(_(e_using_number_as_bool_nr), skipcc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009014 else
9015 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009016 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
9017 while (*s != NUL)
9018 {
9019 func_mb_ptr2char_adv(&s);
9020 ++len;
9021 }
9022 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009023 }
9024}
9025
9026/*
9027 * "strdisplaywidth()" function
9028 */
9029 static void
9030f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
9031{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009032 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009033 int col = 0;
9034
9035 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009036 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009037
9038 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
9039}
9040
9041/*
9042 * "strwidth()" function
9043 */
9044 static void
9045f_strwidth(typval_T *argvars, typval_T *rettv)
9046{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009047 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009048
Bram Moolenaar13505972019-01-24 15:04:48 +01009049 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009050}
9051
9052/*
9053 * "strcharpart()" function
9054 */
9055 static void
9056f_strcharpart(typval_T *argvars, typval_T *rettv)
9057{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009058 char_u *p;
9059 int nchar;
9060 int nbyte = 0;
9061 int charlen;
9062 int len = 0;
9063 int slen;
9064 int error = FALSE;
9065
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009066 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009067 slen = (int)STRLEN(p);
9068
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009069 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009070 if (!error)
9071 {
9072 if (nchar > 0)
9073 while (nchar > 0 && nbyte < slen)
9074 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009075 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009076 --nchar;
9077 }
9078 else
9079 nbyte = nchar;
9080 if (argvars[2].v_type != VAR_UNKNOWN)
9081 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009082 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009083 while (charlen > 0 && nbyte + len < slen)
9084 {
9085 int off = nbyte + len;
9086
9087 if (off < 0)
9088 len += 1;
9089 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009090 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009091 --charlen;
9092 }
9093 }
9094 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009095 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009096 }
9097
9098 /*
9099 * Only return the overlap between the specified part and the actual
9100 * string.
9101 */
9102 if (nbyte < 0)
9103 {
9104 len += nbyte;
9105 nbyte = 0;
9106 }
9107 else if (nbyte > slen)
9108 nbyte = slen;
9109 if (len < 0)
9110 len = 0;
9111 else if (nbyte + len > slen)
9112 len = slen - nbyte;
9113
9114 rettv->v_type = VAR_STRING;
9115 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009116}
9117
9118/*
9119 * "strpart()" function
9120 */
9121 static void
9122f_strpart(typval_T *argvars, typval_T *rettv)
9123{
9124 char_u *p;
9125 int n;
9126 int len;
9127 int slen;
9128 int error = FALSE;
9129
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009130 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009131 slen = (int)STRLEN(p);
9132
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009133 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009134 if (error)
9135 len = 0;
9136 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009137 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009138 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009139 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009140
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009141 // Only return the overlap between the specified part and the actual
9142 // string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009143 if (n < 0)
9144 {
9145 len += n;
9146 n = 0;
9147 }
9148 else if (n > slen)
9149 n = slen;
9150 if (len < 0)
9151 len = 0;
9152 else if (n + len > slen)
9153 len = slen - n;
9154
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009155 if (argvars[2].v_type != VAR_UNKNOWN && argvars[3].v_type != VAR_UNKNOWN)
9156 {
9157 int off;
9158
9159 // length in characters
9160 for (off = n; off < slen && len > 0; --len)
9161 off += mb_ptr2len(p + off);
9162 len = off - n;
9163 }
9164
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009165 rettv->v_type = VAR_STRING;
9166 rettv->vval.v_string = vim_strnsave(p + n, len);
9167}
9168
9169/*
9170 * "strridx()" function
9171 */
9172 static void
9173f_strridx(typval_T *argvars, typval_T *rettv)
9174{
9175 char_u buf[NUMBUFLEN];
9176 char_u *needle;
9177 char_u *haystack;
9178 char_u *rest;
9179 char_u *lastmatch = NULL;
9180 int haystack_len, end_idx;
9181
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009182 needle = tv_get_string_chk(&argvars[1]);
9183 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009184
9185 rettv->vval.v_number = -1;
9186 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009187 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009188
9189 haystack_len = (int)STRLEN(haystack);
9190 if (argvars[2].v_type != VAR_UNKNOWN)
9191 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009192 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009193 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009194 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009195 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009196 }
9197 else
9198 end_idx = haystack_len;
9199
9200 if (*needle == NUL)
9201 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009202 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009203 lastmatch = haystack + end_idx;
9204 }
9205 else
9206 {
9207 for (rest = haystack; *rest != '\0'; ++rest)
9208 {
9209 rest = (char_u *)strstr((char *)rest, (char *)needle);
9210 if (rest == NULL || rest > haystack + end_idx)
9211 break;
9212 lastmatch = rest;
9213 }
9214 }
9215
9216 if (lastmatch == NULL)
9217 rettv->vval.v_number = -1;
9218 else
9219 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
9220}
9221
9222/*
9223 * "strtrans()" function
9224 */
9225 static void
9226f_strtrans(typval_T *argvars, typval_T *rettv)
9227{
9228 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009229 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009230}
9231
9232/*
9233 * "submatch()" function
9234 */
9235 static void
9236f_submatch(typval_T *argvars, typval_T *rettv)
9237{
9238 int error = FALSE;
9239 int no;
9240 int retList = 0;
9241
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009242 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009243 if (error)
9244 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009245 if (no < 0 || no >= NSUBEXP)
9246 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009247 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01009248 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009249 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009250 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaarad304702020-09-06 18:22:53 +02009251 retList = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009252 if (error)
9253 return;
9254
9255 if (retList == 0)
9256 {
9257 rettv->v_type = VAR_STRING;
9258 rettv->vval.v_string = reg_submatch(no);
9259 }
9260 else
9261 {
9262 rettv->v_type = VAR_LIST;
9263 rettv->vval.v_list = reg_submatch_list(no);
9264 }
9265}
9266
9267/*
9268 * "substitute()" function
9269 */
9270 static void
9271f_substitute(typval_T *argvars, typval_T *rettv)
9272{
9273 char_u patbuf[NUMBUFLEN];
9274 char_u subbuf[NUMBUFLEN];
9275 char_u flagsbuf[NUMBUFLEN];
9276
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009277 char_u *str = tv_get_string_chk(&argvars[0]);
9278 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009279 char_u *sub = NULL;
9280 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009281 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009282
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009283 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
9284 expr = &argvars[2];
9285 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009286 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009287
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009288 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009289 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
9290 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009291 rettv->vval.v_string = NULL;
9292 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009293 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009294}
9295
9296/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009297 * "swapinfo(swap_filename)" function
9298 */
9299 static void
9300f_swapinfo(typval_T *argvars, typval_T *rettv)
9301{
9302 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009303 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009304}
9305
9306/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02009307 * "swapname(expr)" function
9308 */
9309 static void
9310f_swapname(typval_T *argvars, typval_T *rettv)
9311{
9312 buf_T *buf;
9313
9314 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01009315 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02009316 if (buf == NULL || buf->b_ml.ml_mfp == NULL
9317 || buf->b_ml.ml_mfp->mf_fname == NULL)
9318 rettv->vval.v_string = NULL;
9319 else
9320 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
9321}
9322
9323/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009324 * "synID(lnum, col, trans)" function
9325 */
9326 static void
9327f_synID(typval_T *argvars UNUSED, typval_T *rettv)
9328{
9329 int id = 0;
9330#ifdef FEAT_SYN_HL
9331 linenr_T lnum;
9332 colnr_T col;
9333 int trans;
9334 int transerr = FALSE;
9335
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009336 lnum = tv_get_lnum(argvars); // -1 on type error
9337 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaarfcb6d702020-09-05 21:41:56 +02009338 trans = (int)tv_get_bool_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009339
9340 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9341 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
9342 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
9343#endif
9344
9345 rettv->vval.v_number = id;
9346}
9347
9348/*
9349 * "synIDattr(id, what [, mode])" function
9350 */
9351 static void
9352f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
9353{
9354 char_u *p = NULL;
9355#ifdef FEAT_SYN_HL
9356 int id;
9357 char_u *what;
9358 char_u *mode;
9359 char_u modebuf[NUMBUFLEN];
9360 int modec;
9361
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009362 id = (int)tv_get_number(&argvars[0]);
9363 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009364 if (argvars[2].v_type != VAR_UNKNOWN)
9365 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009366 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009367 modec = TOLOWER_ASC(mode[0]);
9368 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009369 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009370 }
9371 else
9372 {
9373#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
9374 if (USE_24BIT)
9375 modec = 'g';
9376 else
9377#endif
9378 if (t_colors > 1)
9379 modec = 'c';
9380 else
9381 modec = 't';
9382 }
9383
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009384 switch (TOLOWER_ASC(what[0]))
9385 {
9386 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009387 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009388 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009389 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009390 p = highlight_has_attr(id, HL_BOLD, modec);
9391 break;
9392
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009393 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009394 p = highlight_color(id, what, modec);
9395 break;
9396
9397 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009398 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009399 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009400 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009401 p = highlight_has_attr(id, HL_ITALIC, modec);
9402 break;
9403
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009404 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02009405 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009406 break;
9407
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009408 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009409 p = highlight_has_attr(id, HL_INVERSE, modec);
9410 break;
9411
9412 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009413 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009414 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009415 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02009416 else if (TOLOWER_ASC(what[1]) == 't' &&
9417 TOLOWER_ASC(what[2]) == 'r')
9418 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009419 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009420 p = highlight_has_attr(id, HL_STANDOUT, modec);
9421 break;
9422
9423 case 'u':
Bram Moolenaar391c3622020-09-29 20:59:17 +02009424 if (TOLOWER_ASC(what[1]) == 'l') // ul
9425 p = highlight_color(id, what, modec);
9426 else if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009427 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009428 p = highlight_has_attr(id, HL_UNDERLINE, modec);
9429 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009430 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009431 p = highlight_has_attr(id, HL_UNDERCURL, modec);
9432 break;
9433 }
9434
9435 if (p != NULL)
9436 p = vim_strsave(p);
9437#endif
9438 rettv->v_type = VAR_STRING;
9439 rettv->vval.v_string = p;
9440}
9441
9442/*
9443 * "synIDtrans(id)" function
9444 */
9445 static void
9446f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
9447{
9448 int id;
9449
9450#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009451 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009452
9453 if (id > 0)
9454 id = syn_get_final_id(id);
9455 else
9456#endif
9457 id = 0;
9458
9459 rettv->vval.v_number = id;
9460}
9461
9462/*
9463 * "synconcealed(lnum, col)" function
9464 */
9465 static void
9466f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
9467{
9468#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
9469 linenr_T lnum;
9470 colnr_T col;
9471 int syntax_flags = 0;
9472 int cchar;
9473 int matchid = 0;
9474 char_u str[NUMBUFLEN];
9475#endif
9476
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009477 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009478
9479#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009480 lnum = tv_get_lnum(argvars); // -1 on type error
9481 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009482
Bram Moolenaara80faa82020-04-12 19:37:17 +02009483 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009484
9485 if (rettv_list_alloc(rettv) != FAIL)
9486 {
9487 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9488 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9489 && curwin->w_p_cole > 0)
9490 {
9491 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
9492 syntax_flags = get_syntax_info(&matchid);
9493
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009494 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009495 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
9496 {
9497 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02009498 if (cchar == NUL && curwin->w_p_cole == 1)
9499 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009500 if (cchar != NUL)
9501 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009502 if (has_mbyte)
9503 (*mb_char2bytes)(cchar, str);
9504 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009505 str[0] = cchar;
9506 }
9507 }
9508 }
9509
9510 list_append_number(rettv->vval.v_list,
9511 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009512 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009513 list_append_string(rettv->vval.v_list, str, -1);
9514 list_append_number(rettv->vval.v_list, matchid);
9515 }
9516#endif
9517}
9518
9519/*
9520 * "synstack(lnum, col)" function
9521 */
9522 static void
9523f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
9524{
9525#ifdef FEAT_SYN_HL
9526 linenr_T lnum;
9527 colnr_T col;
9528 int i;
9529 int id;
9530#endif
9531
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009532 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009533
9534#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009535 lnum = tv_get_lnum(argvars); // -1 on type error
9536 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009537
9538 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9539 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9540 && rettv_list_alloc(rettv) != FAIL)
9541 {
9542 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
9543 for (i = 0; ; ++i)
9544 {
9545 id = syn_get_stack_item(i);
9546 if (id < 0)
9547 break;
9548 if (list_append_number(rettv->vval.v_list, id) == FAIL)
9549 break;
9550 }
9551 }
9552#endif
9553}
9554
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009555/*
9556 * "tabpagebuflist()" function
9557 */
9558 static void
9559f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9560{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009561 tabpage_T *tp;
9562 win_T *wp = NULL;
9563
9564 if (argvars[0].v_type == VAR_UNKNOWN)
9565 wp = firstwin;
9566 else
9567 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009568 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009569 if (tp != NULL)
9570 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
9571 }
9572 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
9573 {
9574 for (; wp != NULL; wp = wp->w_next)
9575 if (list_append_number(rettv->vval.v_list,
9576 wp->w_buffer->b_fnum) == FAIL)
9577 break;
9578 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009579}
9580
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009581/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009582 * "tagfiles()" function
9583 */
9584 static void
9585f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
9586{
9587 char_u *fname;
9588 tagname_T tn;
9589 int first;
9590
9591 if (rettv_list_alloc(rettv) == FAIL)
9592 return;
9593 fname = alloc(MAXPATHL);
9594 if (fname == NULL)
9595 return;
9596
9597 for (first = TRUE; ; first = FALSE)
9598 if (get_tagfname(&tn, first, fname) == FAIL
9599 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
9600 break;
9601 tagname_free(&tn);
9602 vim_free(fname);
9603}
9604
9605/*
9606 * "taglist()" function
9607 */
9608 static void
9609f_taglist(typval_T *argvars, typval_T *rettv)
9610{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009611 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009612 char_u *tag_pattern;
9613
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009614 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009615
9616 rettv->vval.v_number = FALSE;
9617 if (*tag_pattern == NUL)
9618 return;
9619
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009620 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009621 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009622 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009623 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009624}
9625
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009626#ifdef FEAT_FLOAT
9627/*
9628 * "tan()" function
9629 */
9630 static void
9631f_tan(typval_T *argvars, typval_T *rettv)
9632{
9633 float_T f = 0.0;
9634
9635 rettv->v_type = VAR_FLOAT;
9636 if (get_float_arg(argvars, &f) == OK)
9637 rettv->vval.v_float = tan(f);
9638 else
9639 rettv->vval.v_float = 0.0;
9640}
9641
9642/*
9643 * "tanh()" function
9644 */
9645 static void
9646f_tanh(typval_T *argvars, typval_T *rettv)
9647{
9648 float_T f = 0.0;
9649
9650 rettv->v_type = VAR_FLOAT;
9651 if (get_float_arg(argvars, &f) == OK)
9652 rettv->vval.v_float = tanh(f);
9653 else
9654 rettv->vval.v_float = 0.0;
9655}
9656#endif
9657
9658/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009659 * "tolower(string)" function
9660 */
9661 static void
9662f_tolower(typval_T *argvars, typval_T *rettv)
9663{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009664 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009665 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009666}
9667
9668/*
9669 * "toupper(string)" function
9670 */
9671 static void
9672f_toupper(typval_T *argvars, typval_T *rettv)
9673{
9674 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009675 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009676}
9677
9678/*
9679 * "tr(string, fromstr, tostr)" function
9680 */
9681 static void
9682f_tr(typval_T *argvars, typval_T *rettv)
9683{
9684 char_u *in_str;
9685 char_u *fromstr;
9686 char_u *tostr;
9687 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009688 int inlen;
9689 int fromlen;
9690 int tolen;
9691 int idx;
9692 char_u *cpstr;
9693 int cplen;
9694 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009695 char_u buf[NUMBUFLEN];
9696 char_u buf2[NUMBUFLEN];
9697 garray_T ga;
9698
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009699 in_str = tv_get_string(&argvars[0]);
9700 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
9701 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009702
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009703 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009704 rettv->v_type = VAR_STRING;
9705 rettv->vval.v_string = NULL;
9706 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009707 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009708 ga_init2(&ga, (int)sizeof(char), 80);
9709
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009710 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009711 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009712 if (STRLEN(fromstr) != STRLEN(tostr))
9713 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009714error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009715 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009716 ga_clear(&ga);
9717 return;
9718 }
9719
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009720 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009721 while (*in_str != NUL)
9722 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009723 if (has_mbyte)
9724 {
9725 inlen = (*mb_ptr2len)(in_str);
9726 cpstr = in_str;
9727 cplen = inlen;
9728 idx = 0;
9729 for (p = fromstr; *p != NUL; p += fromlen)
9730 {
9731 fromlen = (*mb_ptr2len)(p);
9732 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
9733 {
9734 for (p = tostr; *p != NUL; p += tolen)
9735 {
9736 tolen = (*mb_ptr2len)(p);
9737 if (idx-- == 0)
9738 {
9739 cplen = tolen;
9740 cpstr = p;
9741 break;
9742 }
9743 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009744 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009745 goto error;
9746 break;
9747 }
9748 ++idx;
9749 }
9750
9751 if (first && cpstr == in_str)
9752 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009753 // Check that fromstr and tostr have the same number of
9754 // (multi-byte) characters. Done only once when a character
9755 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009756 first = FALSE;
9757 for (p = tostr; *p != NUL; p += tolen)
9758 {
9759 tolen = (*mb_ptr2len)(p);
9760 --idx;
9761 }
9762 if (idx != 0)
9763 goto error;
9764 }
9765
9766 (void)ga_grow(&ga, cplen);
9767 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
9768 ga.ga_len += cplen;
9769
9770 in_str += inlen;
9771 }
9772 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009773 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009774 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009775 p = vim_strchr(fromstr, *in_str);
9776 if (p != NULL)
9777 ga_append(&ga, tostr[p - fromstr]);
9778 else
9779 ga_append(&ga, *in_str);
9780 ++in_str;
9781 }
9782 }
9783
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009784 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009785 (void)ga_grow(&ga, 1);
9786 ga_append(&ga, NUL);
9787
9788 rettv->vval.v_string = ga.ga_data;
9789}
9790
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009791/*
9792 * "trim({expr})" function
9793 */
9794 static void
9795f_trim(typval_T *argvars, typval_T *rettv)
9796{
9797 char_u buf1[NUMBUFLEN];
9798 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009799 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009800 char_u *mask = NULL;
9801 char_u *tail;
9802 char_u *prev;
9803 char_u *p;
9804 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009805 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009806
9807 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009808 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009809 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009810 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009811
9812 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009813 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009814 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009815
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009816 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009817 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009818 int error = 0;
9819
9820 // leading or trailing characters to trim
9821 dir = (int)tv_get_number_chk(&argvars[2], &error);
9822 if (error)
9823 return;
9824 if (dir < 0 || dir > 2)
9825 {
9826 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
9827 return;
9828 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009829 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009830 }
9831
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009832 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009833 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009834 // Trim leading characters
9835 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009836 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009837 c1 = PTR2CHAR(head);
9838 if (mask == NULL)
9839 {
9840 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009841 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009842 }
9843 else
9844 {
9845 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9846 if (c1 == PTR2CHAR(p))
9847 break;
9848 if (*p == NUL)
9849 break;
9850 }
9851 MB_PTR_ADV(head);
9852 }
9853 }
9854
9855 tail = head + STRLEN(head);
9856 if (dir == 0 || dir == 2)
9857 {
9858 // Trim trailing characters
9859 for (; tail > head; tail = prev)
9860 {
9861 prev = tail;
9862 MB_PTR_BACK(head, prev);
9863 c1 = PTR2CHAR(prev);
9864 if (mask == NULL)
9865 {
9866 if (c1 > ' ' && c1 != 0xa0)
9867 break;
9868 }
9869 else
9870 {
9871 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9872 if (c1 == PTR2CHAR(p))
9873 break;
9874 if (*p == NUL)
9875 break;
9876 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009877 }
9878 }
Bram Moolenaardf44a272020-06-07 20:49:05 +02009879 rettv->vval.v_string = vim_strnsave(head, tail - head);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009880}
9881
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009882#ifdef FEAT_FLOAT
9883/*
9884 * "trunc({float})" function
9885 */
9886 static void
9887f_trunc(typval_T *argvars, typval_T *rettv)
9888{
9889 float_T f = 0.0;
9890
9891 rettv->v_type = VAR_FLOAT;
9892 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009893 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009894 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
9895 else
9896 rettv->vval.v_float = 0.0;
9897}
9898#endif
9899
9900/*
9901 * "type(expr)" function
9902 */
9903 static void
9904f_type(typval_T *argvars, typval_T *rettv)
9905{
9906 int n = -1;
9907
9908 switch (argvars[0].v_type)
9909 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01009910 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
9911 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009912 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01009913 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
9914 case VAR_LIST: n = VAR_TYPE_LIST; break;
9915 case VAR_DICT: n = VAR_TYPE_DICT; break;
9916 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
9917 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
9918 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02009919 case VAR_JOB: n = VAR_TYPE_JOB; break;
9920 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01009921 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009922 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02009923 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01009924 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01009925 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009926 n = -1;
9927 break;
9928 }
9929 rettv->vval.v_number = n;
9930}
9931
9932/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009933 * "virtcol(string)" function
9934 */
9935 static void
9936f_virtcol(typval_T *argvars, typval_T *rettv)
9937{
9938 colnr_T vcol = 0;
9939 pos_T *fp;
9940 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01009941 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009942
9943 fp = var2fpos(&argvars[0], FALSE, &fnum);
9944 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
9945 && fnum == curbuf->b_fnum)
9946 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01009947 // Limit the column to a valid value, getvvcol() doesn't check.
9948 if (fp->col < 0)
9949 fp->col = 0;
9950 else
9951 {
9952 len = (int)STRLEN(ml_get(fp->lnum));
9953 if (fp->col > len)
9954 fp->col = len;
9955 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009956 getvvcol(curwin, fp, NULL, NULL, &vcol);
9957 ++vcol;
9958 }
9959
9960 rettv->vval.v_number = vcol;
9961}
9962
9963/*
9964 * "visualmode()" function
9965 */
9966 static void
9967f_visualmode(typval_T *argvars, typval_T *rettv)
9968{
9969 char_u str[2];
9970
9971 rettv->v_type = VAR_STRING;
9972 str[0] = curbuf->b_visual_mode_eval;
9973 str[1] = NUL;
9974 rettv->vval.v_string = vim_strsave(str);
9975
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009976 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009977 if (non_zero_arg(&argvars[0]))
9978 curbuf->b_visual_mode_eval = NUL;
9979}
9980
9981/*
9982 * "wildmenumode()" function
9983 */
9984 static void
9985f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9986{
9987#ifdef FEAT_WILDMENU
9988 if (wild_menu_showing)
9989 rettv->vval.v_number = 1;
9990#endif
9991}
9992
9993/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +01009994 * "windowsversion()" function
9995 */
9996 static void
9997f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9998{
9999 rettv->v_type = VAR_STRING;
10000 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
10001}
10002
10003/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010004 * "wordcount()" function
10005 */
10006 static void
10007f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
10008{
10009 if (rettv_dict_alloc(rettv) == FAIL)
10010 return;
10011 cursor_pos_info(rettv->vval.v_dict);
10012}
10013
10014/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010015 * "xor(expr, expr)" function
10016 */
10017 static void
10018f_xor(typval_T *argvars, typval_T *rettv)
10019{
Bram Moolenaard155d7a2018-12-21 16:04:21 +010010020 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
10021 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010022}
10023
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010024#endif // FEAT_EVAL