blob: 65cb59f8d41fbaada08c2d42c89ae63a3a05fc5c [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 Moolenaar6f02b002021-01-10 20:22:54 +010050static void f_charcol(typval_T *argvars, typval_T *rettv);
Bram Moolenaar17793ef2020-12-28 12:56:58 +010051static void f_charidx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020052static void f_col(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020053static void f_confirm(typval_T *argvars, typval_T *rettv);
54static void f_copy(typval_T *argvars, typval_T *rettv);
55#ifdef FEAT_FLOAT
56static void f_cos(typval_T *argvars, typval_T *rettv);
57static void f_cosh(typval_T *argvars, typval_T *rettv);
58#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020059static void f_cursor(typval_T *argsvars, typval_T *rettv);
Bram Moolenaar4f974752019-02-17 17:44:42 +010060#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +020061static void f_debugbreak(typval_T *argvars, typval_T *rettv);
62#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020063static void f_deepcopy(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020064static void f_did_filetype(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4132eb52020-02-14 16:53:00 +010065static void f_echoraw(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020066static void f_empty(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020067static void f_environ(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020068static void f_escape(typval_T *argvars, typval_T *rettv);
69static void f_eval(typval_T *argvars, typval_T *rettv);
70static void f_eventhandler(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020071static void f_execute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020072static void f_exists(typval_T *argvars, typval_T *rettv);
73#ifdef FEAT_FLOAT
74static void f_exp(typval_T *argvars, typval_T *rettv);
75#endif
76static void f_expand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar80dad482019-06-09 17:22:31 +020077static void f_expandcmd(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020078static void f_feedkeys(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020079#ifdef FEAT_FLOAT
80static void f_float2nr(typval_T *argvars, typval_T *rettv);
81static void f_floor(typval_T *argvars, typval_T *rettv);
82static void f_fmod(typval_T *argvars, typval_T *rettv);
83#endif
84static void f_fnameescape(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020085static void f_foreground(typval_T *argvars, typval_T *rettv);
Bram Moolenaar437bafe2016-08-01 15:40:54 +020086static void f_funcref(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020087static void f_function(typval_T *argvars, typval_T *rettv);
88static void f_garbagecollect(typval_T *argvars, typval_T *rettv);
89static void f_get(typval_T *argvars, typval_T *rettv);
Bram Moolenaar07ad8162018-02-13 13:59:59 +010090static void f_getchangelist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +010091static void f_getcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020092static void f_getcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020093static void f_getenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020094static void f_getfontname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4f505882018-02-10 21:06:32 +010095static void f_getjumplist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020096static void f_getpid(typval_T *argvars, typval_T *rettv);
97static void f_getcurpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +010098static void f_getcursorcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020099static void f_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200100static void f_getreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarbb861e22020-06-07 18:16:36 +0200101static void f_getreginfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200102static void f_getregtype(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100103static void f_gettagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +0200104static void f_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200105static void f_haslocaldir(typval_T *argvars, typval_T *rettv);
106static void f_hasmapto(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200107static void f_hlID(typval_T *argvars, typval_T *rettv);
108static void f_hlexists(typval_T *argvars, typval_T *rettv);
109static void f_hostname(typval_T *argvars, typval_T *rettv);
110static void f_iconv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200111static void f_index(typval_T *argvars, typval_T *rettv);
112static void f_input(typval_T *argvars, typval_T *rettv);
113static void f_inputdialog(typval_T *argvars, typval_T *rettv);
114static void f_inputlist(typval_T *argvars, typval_T *rettv);
115static void f_inputrestore(typval_T *argvars, typval_T *rettv);
116static void f_inputsave(typval_T *argvars, typval_T *rettv);
117static void f_inputsecret(typval_T *argvars, typval_T *rettv);
Bram Moolenaar67a2deb2019-11-25 00:05:32 +0100118static void f_interrupt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200119static void f_invert(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200120static void f_islocked(typval_T *argvars, typval_T *rettv);
121#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
Bram Moolenaarfda1bff2019-04-04 13:44:37 +0200122static void f_isinf(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200123static void f_isnan(typval_T *argvars, typval_T *rettv);
124#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200125static void f_last_buffer_nr(typval_T *argvars, typval_T *rettv);
126static void f_len(typval_T *argvars, typval_T *rettv);
127static void f_libcall(typval_T *argvars, typval_T *rettv);
128static void f_libcallnr(typval_T *argvars, typval_T *rettv);
129static void f_line(typval_T *argvars, typval_T *rettv);
130static void f_line2byte(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200131#ifdef FEAT_FLOAT
132static void f_log(typval_T *argvars, typval_T *rettv);
133static void f_log10(typval_T *argvars, typval_T *rettv);
134#endif
135#ifdef FEAT_LUA
136static void f_luaeval(typval_T *argvars, typval_T *rettv);
137#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200138static void f_maparg(typval_T *argvars, typval_T *rettv);
139static void f_mapcheck(typval_T *argvars, typval_T *rettv);
140static void f_match(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200141static void f_matchend(typval_T *argvars, typval_T *rettv);
142static void f_matchlist(typval_T *argvars, typval_T *rettv);
143static void f_matchstr(typval_T *argvars, typval_T *rettv);
144static void f_matchstrpos(typval_T *argvars, typval_T *rettv);
145static void f_max(typval_T *argvars, typval_T *rettv);
146static void f_min(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200147#ifdef FEAT_MZSCHEME
148static void f_mzeval(typval_T *argvars, typval_T *rettv);
149#endif
150static void f_nextnonblank(typval_T *argvars, typval_T *rettv);
151static void f_nr2char(typval_T *argvars, typval_T *rettv);
152static void f_or(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200153#ifdef FEAT_PERL
154static void f_perleval(typval_T *argvars, typval_T *rettv);
155#endif
156#ifdef FEAT_FLOAT
157static void f_pow(typval_T *argvars, typval_T *rettv);
158#endif
159static void f_prevnonblank(typval_T *argvars, typval_T *rettv);
160static void f_printf(typval_T *argvars, typval_T *rettv);
Bram Moolenaare9bd5722019-08-17 19:36:06 +0200161static void f_pum_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200162static void f_pumvisible(typval_T *argvars, typval_T *rettv);
163#ifdef FEAT_PYTHON3
164static void f_py3eval(typval_T *argvars, typval_T *rettv);
165#endif
166#ifdef FEAT_PYTHON
167static void f_pyeval(typval_T *argvars, typval_T *rettv);
168#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100169#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
170static void f_pyxeval(typval_T *argvars, typval_T *rettv);
171#endif
Bram Moolenaar4f645c52020-02-08 16:40:39 +0100172static void f_test_srand_seed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100173static void f_rand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200174static void f_range(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b6d9112018-05-22 20:35:17 +0200175static void f_reg_executing(typval_T *argvars, typval_T *rettv);
176static void f_reg_recording(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200177static void f_rename(typval_T *argvars, typval_T *rettv);
178static void f_repeat(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200179#ifdef FEAT_FLOAT
180static void f_round(typval_T *argvars, typval_T *rettv);
181#endif
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100182#ifdef FEAT_RUBY
183static void f_rubyeval(typval_T *argvars, typval_T *rettv);
184#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200185static void f_screenattr(typval_T *argvars, typval_T *rettv);
186static void f_screenchar(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100187static void f_screenchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200188static void f_screencol(typval_T *argvars, typval_T *rettv);
189static void f_screenrow(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100190static void f_screenstring(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200191static void f_search(typval_T *argvars, typval_T *rettv);
192static void f_searchdecl(typval_T *argvars, typval_T *rettv);
193static void f_searchpair(typval_T *argvars, typval_T *rettv);
194static void f_searchpairpos(typval_T *argvars, typval_T *rettv);
195static void f_searchpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100196static void f_setcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200197static void f_setcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100198static void f_setcursorcharpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +0200199static void f_setenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200200static void f_setfperm(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200201static void f_setpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200202static void f_setreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100203static void f_settagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200204#ifdef FEAT_CRYPT
205static void f_sha256(typval_T *argvars, typval_T *rettv);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200206#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200207static void f_shellescape(typval_T *argvars, typval_T *rettv);
208static void f_shiftwidth(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200209#ifdef FEAT_FLOAT
210static void f_sin(typval_T *argvars, typval_T *rettv);
211static void f_sinh(typval_T *argvars, typval_T *rettv);
212#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200213static void f_soundfold(typval_T *argvars, typval_T *rettv);
214static void f_spellbadword(typval_T *argvars, typval_T *rettv);
215static void f_spellsuggest(typval_T *argvars, typval_T *rettv);
216static void f_split(typval_T *argvars, typval_T *rettv);
217#ifdef FEAT_FLOAT
218static void f_sqrt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100219#endif
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100220static void f_srand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100221#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200222static void f_str2float(typval_T *argvars, typval_T *rettv);
223#endif
Bram Moolenaar9d401282019-04-06 13:18:12 +0200224static void f_str2list(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200225static void f_str2nr(typval_T *argvars, typval_T *rettv);
226static void f_strchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200227static void f_strgetchar(typval_T *argvars, typval_T *rettv);
228static void f_stridx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200229static void f_strlen(typval_T *argvars, typval_T *rettv);
230static void f_strcharpart(typval_T *argvars, typval_T *rettv);
231static void f_strpart(typval_T *argvars, typval_T *rettv);
232static void f_strridx(typval_T *argvars, typval_T *rettv);
233static void f_strtrans(typval_T *argvars, typval_T *rettv);
234static void f_strdisplaywidth(typval_T *argvars, typval_T *rettv);
235static void f_strwidth(typval_T *argvars, typval_T *rettv);
236static void f_submatch(typval_T *argvars, typval_T *rettv);
237static void f_substitute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar00f123a2018-08-21 20:28:54 +0200238static void f_swapinfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar110bd602018-09-16 18:46:59 +0200239static void f_swapname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200240static void f_synID(typval_T *argvars, typval_T *rettv);
241static void f_synIDattr(typval_T *argvars, typval_T *rettv);
242static void f_synIDtrans(typval_T *argvars, typval_T *rettv);
243static void f_synstack(typval_T *argvars, typval_T *rettv);
244static void f_synconcealed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200245static void f_tabpagebuflist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200246static void f_taglist(typval_T *argvars, typval_T *rettv);
247static void f_tagfiles(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200248#ifdef FEAT_FLOAT
249static void f_tan(typval_T *argvars, typval_T *rettv);
250static void f_tanh(typval_T *argvars, typval_T *rettv);
251#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200252static void f_tolower(typval_T *argvars, typval_T *rettv);
253static void f_toupper(typval_T *argvars, typval_T *rettv);
254static void f_tr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +0100255static void f_trim(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200256#ifdef FEAT_FLOAT
257static void f_trunc(typval_T *argvars, typval_T *rettv);
258#endif
259static void f_type(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200260static void f_virtcol(typval_T *argvars, typval_T *rettv);
261static void f_visualmode(typval_T *argvars, typval_T *rettv);
262static void f_wildmenumode(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0c1e3742019-12-27 13:49:24 +0100263static void f_windowsversion(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200264static void f_wordcount(typval_T *argvars, typval_T *rettv);
265static void f_xor(typval_T *argvars, typval_T *rettv);
266
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100267
Bram Moolenaar94738d82020-10-21 14:25:07 +0200268/*
269 * Functions that check the argument type of a builtin function.
270 * Each function returns FAIL and gives an error message if the type is wrong.
271 */
272
273// Context passed to an arg_ function.
274typedef struct {
Bram Moolenaarca174532020-10-21 16:42:22 +0200275 int arg_count; // actual argument count
276 type_T **arg_types; // list of argument types
277 int arg_idx; // current argument index (first arg is zero)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200278} argcontext_T;
279
280// A function to check one argument type. The first argument is the type to
281// check. If needed, other argument types can be obtained with the context.
282// E.g. if "arg_idx" is 1, then (type - 1) is the first argument type.
283typedef int (*argcheck_T)(type_T *, argcontext_T *);
284
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100285/*
286 * Check "type" is a float or a number.
287 */
Bram Moolenaar94738d82020-10-21 14:25:07 +0200288 static int
289arg_float_or_nr(type_T *type, argcontext_T *context)
290{
Bram Moolenaarca174532020-10-21 16:42:22 +0200291 if (type->tt_type == VAR_ANY
292 || type->tt_type == VAR_FLOAT || type->tt_type == VAR_NUMBER)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200293 return OK;
294 arg_type_mismatch(&t_number, type, context->arg_idx + 1);
295 return FAIL;
296}
297
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100298/*
299 * Check "type" is a number.
300 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200301 static int
302arg_number(type_T *type, argcontext_T *context)
303{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100304 return check_arg_type(&t_number, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200305}
306
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100307/*
308 * Check "type" is a string.
309 */
310 static int
311arg_string(type_T *type, argcontext_T *context)
312{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100313 return check_arg_type(&t_string, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100314}
315
316/*
317 * Check "type" is a list or a blob.
318 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200319 static int
320arg_list_or_blob(type_T *type, argcontext_T *context)
321{
322 if (type->tt_type == VAR_ANY
323 || type->tt_type == VAR_LIST || type->tt_type == VAR_BLOB)
324 return OK;
325 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
326 return FAIL;
327}
328
329/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100330 * Check "type" is a list or a dict.
331 */
332 static int
333arg_list_or_dict(type_T *type, argcontext_T *context)
334{
335 if (type->tt_type == VAR_ANY
336 || type->tt_type == VAR_LIST || type->tt_type == VAR_DICT)
337 return OK;
338 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
339 return FAIL;
340}
341
342/*
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100343 * Check "type" is the same type as the previous argument.
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100344 * Must not be used for the first argcheck_T entry.
345 */
346 static int
347arg_same_as_prev(type_T *type, argcontext_T *context)
348{
349 type_T *prev_type = context->arg_types[context->arg_idx - 1];
350
Bram Moolenaar193f6202020-11-16 20:08:35 +0100351 return check_arg_type(prev_type, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100352}
353
354/*
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100355 * Check "type" is the same basic type as the previous argument, checks list or
356 * dict vs other type, but not member type.
357 * Must not be used for the first argcheck_T entry.
358 */
359 static int
360arg_same_struct_as_prev(type_T *type, argcontext_T *context)
361{
362 type_T *prev_type = context->arg_types[context->arg_idx - 1];
363
364 if (prev_type->tt_type != context->arg_types[context->arg_idx]->tt_type)
365 return check_arg_type(prev_type, type, context->arg_idx + 1);
366 return OK;
367}
368
369/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100370 * Check "type" is an item of the list or blob of the previous arg.
Bram Moolenaarca174532020-10-21 16:42:22 +0200371 * Must not be used for the first argcheck_T entry.
372 */
373 static int
374arg_item_of_prev(type_T *type, argcontext_T *context)
375{
376 type_T *prev_type = context->arg_types[context->arg_idx - 1];
377 type_T *expected;
378
379 if (prev_type->tt_type == VAR_LIST)
380 expected = prev_type->tt_member;
381 else if (prev_type->tt_type == VAR_BLOB)
382 expected = &t_number;
383 else
384 // probably VAR_ANY, can't check
385 return OK;
386
Bram Moolenaar193f6202020-11-16 20:08:35 +0100387 return check_arg_type(expected, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200388}
389
Bram Moolenaar94738d82020-10-21 14:25:07 +0200390/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100391 * Check "type" which is the third argument of extend().
392 */
393 static int
394arg_extend3(type_T *type, argcontext_T *context)
395{
396 type_T *first_type = context->arg_types[context->arg_idx - 2];
397
398 if (first_type->tt_type == VAR_LIST)
399 return arg_number(type, context);
400 if (first_type->tt_type == VAR_DICT)
401 return arg_string(type, context);
402 return OK;
403}
404
405
406/*
Bram Moolenaar94738d82020-10-21 14:25:07 +0200407 * Lists of functions that check the argument types of a builtin function.
408 */
409argcheck_T arg1_float_or_nr[] = {arg_float_or_nr};
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100410argcheck_T arg2_listblob_item[] = {arg_list_or_blob, arg_item_of_prev};
411argcheck_T arg23_extend[] = {arg_list_or_dict, arg_same_as_prev, arg_extend3};
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100412argcheck_T arg23_extendnew[] = {arg_list_or_dict, arg_same_struct_as_prev, arg_extend3};
Bram Moolenaarca174532020-10-21 16:42:22 +0200413argcheck_T arg3_insert[] = {arg_list_or_blob, arg_item_of_prev, arg_number};
Bram Moolenaar94738d82020-10-21 14:25:07 +0200414
415/*
416 * Functions that return the return type of a builtin function.
Bram Moolenaara1224cb2020-10-22 12:31:49 +0200417 * Note that "argtypes" is NULL if "argcount" is zero.
Bram Moolenaar94738d82020-10-21 14:25:07 +0200418 */
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100419 static type_T *
420ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
421{
422 return &t_void;
423}
424 static type_T *
425ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
426{
427 return &t_any;
428}
429 static type_T *
Bram Moolenaar403dc312020-10-17 19:29:51 +0200430ret_bool(int argcount UNUSED, type_T **argtypes UNUSED)
431{
432 return &t_bool;
433}
434 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100435ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
436{
437 return &t_number;
438}
439 static type_T *
440ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
441{
442 return &t_float;
443}
444 static type_T *
445ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
446{
447 return &t_string;
448}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200449 static type_T *
450ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100451{
452 return &t_list_any;
453}
454 static type_T *
455ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
456{
457 return &t_list_number;
458}
459 static type_T *
460ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
461{
462 return &t_list_string;
463}
464 static type_T *
465ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
466{
467 return &t_list_dict_any;
468}
469 static type_T *
470ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
471{
472 return &t_dict_any;
473}
474 static type_T *
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100475ret_job_info(int argcount, type_T **argtypes UNUSED)
476{
477 if (argcount == 0)
478 return &t_list_job;
479 return &t_dict_any;
480}
481 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100482ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
483{
484 return &t_dict_number;
485}
486 static type_T *
487ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
488{
489 return &t_dict_string;
490}
491 static type_T *
492ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
493{
494 return &t_blob;
495}
496 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200497ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100498{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200499 return &t_func_any;
500}
501 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100502ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
503{
504 return &t_channel;
505}
506 static type_T *
507ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
508{
509 return &t_job;
510}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200511 static type_T *
512ret_first_arg(int argcount, type_T **argtypes)
513{
514 if (argcount > 0)
515 return argtypes[0];
516 return &t_void;
517}
Bram Moolenaarea696852020-11-09 18:31:39 +0100518// for map(): returns first argument but item type may differ
519 static type_T *
520ret_first_cont(int argcount UNUSED, type_T **argtypes)
521{
522 if (argtypes[0]->tt_type == VAR_LIST)
523 return &t_list_any;
524 if (argtypes[0]->tt_type == VAR_DICT)
525 return &t_dict_any;
526 if (argtypes[0]->tt_type == VAR_BLOB)
527 return argtypes[0];
528 return &t_any;
529}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200530
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200531/*
532 * Used for getqflist(): returns list if there is no argument, dict if there is
533 * one.
534 */
535 static type_T *
536ret_list_or_dict_0(int argcount, type_T **argtypes UNUSED)
537{
538 if (argcount > 0)
539 return &t_dict_any;
540 return &t_list_dict_any;
541}
542
543/*
544 * Used for getloclist(): returns list if there is one argument, dict if there
545 * are two.
546 */
547 static type_T *
548ret_list_or_dict_1(int argcount, type_T **argtypes UNUSED)
549{
550 if (argcount > 1)
551 return &t_dict_any;
552 return &t_list_dict_any;
553}
554
Bram Moolenaar846178a2020-07-05 17:04:13 +0200555 static type_T *
556ret_argv(int argcount, type_T **argtypes UNUSED)
557{
558 // argv() returns list of strings
559 if (argcount == 0)
560 return &t_list_string;
561
562 // argv(0) returns a string, but argv(-1] returns a list
563 return &t_any;
564}
565
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200566 static type_T *
567ret_remove(int argcount UNUSED, type_T **argtypes)
568{
Bram Moolenaar5e654232020-09-16 15:22:00 +0200569 if (argtypes != NULL)
570 {
571 if (argtypes[0]->tt_type == VAR_LIST
572 || argtypes[0]->tt_type == VAR_DICT)
573 return argtypes[0]->tt_member;
574 if (argtypes[0]->tt_type == VAR_BLOB)
575 return &t_number;
576 }
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200577 return &t_any;
578}
579
Bram Moolenaar3d945cc2020-08-06 21:26:59 +0200580 static type_T *
581ret_getreg(int argcount, type_T **argtypes UNUSED)
582{
583 // Assume that if the third argument is passed it's non-zero
584 if (argcount == 3)
585 return &t_list_string;
586 return &t_string;
587}
588
Bram Moolenaar4a6d1b62020-08-08 17:55:49 +0200589 static type_T *
590ret_maparg(int argcount, type_T **argtypes UNUSED)
591{
592 // Assume that if the fourth argument is passed it's non-zero
593 if (argcount == 4)
594 return &t_dict_any;
595 return &t_string;
596}
597
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100598static type_T *ret_f_function(int argcount, type_T **argtypes);
599
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200600/*
601 * Array with names and number of arguments of all internal functions
602 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
603 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200604typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200605{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200606 char *f_name; // function name
607 char f_min_argc; // minimal number of arguments
608 char f_max_argc; // maximal number of arguments
609 char f_argtype; // for method: FEARG_ values
Bram Moolenaar94738d82020-10-21 14:25:07 +0200610 argcheck_T *f_argcheck; // list of functions to check argument types
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100611 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
612 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200613 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200614 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200615} funcentry_T;
616
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200617// values for f_argtype; zero means it cannot be used as a method
618#define FEARG_1 1 // base is the first argument
619#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200620#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200621#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200622#define FEARG_LAST 9 // base is the last argument
623
Bram Moolenaar15c47602020-03-26 22:16:48 +0100624#ifdef FEAT_FLOAT
625# define FLOAT_FUNC(name) name
626#else
627# define FLOAT_FUNC(name) NULL
628#endif
629#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
630# define MATH_FUNC(name) name
631#else
632# define MATH_FUNC(name) NULL
633#endif
634#ifdef FEAT_TIMERS
635# define TIMER_FUNC(name) name
636#else
637# define TIMER_FUNC(name) NULL
638#endif
639#ifdef FEAT_JOB_CHANNEL
640# define JOB_FUNC(name) name
641#else
642# define JOB_FUNC(name) NULL
643#endif
644#ifdef FEAT_PROP_POPUP
645# define PROP_FUNC(name) name
646#else
647# define PROP_FUNC(name) NULL
648#endif
649#ifdef FEAT_SIGNS
650# define SIGN_FUNC(name) name
651#else
652# define SIGN_FUNC(name) NULL
653#endif
654#ifdef FEAT_SOUND
655# define SOUND_FUNC(name) name
656#else
657# define SOUND_FUNC(name) NULL
658#endif
659#ifdef FEAT_TERMINAL
660# define TERM_FUNC(name) name
661#else
662# define TERM_FUNC(name) NULL
663#endif
664
Bram Moolenaarac92e252019-08-03 21:58:38 +0200665static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200666{
Bram Moolenaar94738d82020-10-21 14:25:07 +0200667 {"abs", 1, 1, FEARG_1, arg1_float_or_nr,
668 ret_any, FLOAT_FUNC(f_abs)},
669 {"acos", 1, 1, FEARG_1, NULL,
670 ret_float, FLOAT_FUNC(f_acos)},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100671 {"add", 2, 2, FEARG_1, NULL /* arg2_listblob_item */,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200672 ret_first_arg, f_add},
673 {"and", 2, 2, FEARG_1, NULL,
674 ret_number, f_and},
675 {"append", 2, 2, FEARG_2, NULL,
676 ret_number, f_append},
677 {"appendbufline", 3, 3, FEARG_3, NULL,
678 ret_number, f_appendbufline},
679 {"argc", 0, 1, 0, NULL,
680 ret_number, f_argc},
681 {"argidx", 0, 0, 0, NULL,
682 ret_number, f_argidx},
683 {"arglistid", 0, 2, 0, NULL,
684 ret_number, f_arglistid},
685 {"argv", 0, 2, 0, NULL,
686 ret_argv, f_argv},
687 {"asin", 1, 1, FEARG_1, NULL,
688 ret_float, FLOAT_FUNC(f_asin)},
689 {"assert_beeps", 1, 2, FEARG_1, NULL,
690 ret_number, f_assert_beeps},
691 {"assert_equal", 2, 3, FEARG_2, NULL,
692 ret_number, f_assert_equal},
693 {"assert_equalfile", 2, 3, FEARG_1, NULL,
694 ret_number, f_assert_equalfile},
695 {"assert_exception", 1, 2, 0, NULL,
696 ret_number, f_assert_exception},
697 {"assert_fails", 1, 5, FEARG_1, NULL,
698 ret_number, f_assert_fails},
699 {"assert_false", 1, 2, FEARG_1, NULL,
700 ret_number, f_assert_false},
701 {"assert_inrange", 3, 4, FEARG_3, NULL,
702 ret_number, f_assert_inrange},
703 {"assert_match", 2, 3, FEARG_2, NULL,
704 ret_number, f_assert_match},
705 {"assert_notequal", 2, 3, FEARG_2, NULL,
706 ret_number, f_assert_notequal},
707 {"assert_notmatch", 2, 3, FEARG_2, NULL,
708 ret_number, f_assert_notmatch},
709 {"assert_report", 1, 1, FEARG_1, NULL,
710 ret_number, f_assert_report},
711 {"assert_true", 1, 2, FEARG_1, NULL,
712 ret_number, f_assert_true},
713 {"atan", 1, 1, FEARG_1, NULL,
714 ret_float, FLOAT_FUNC(f_atan)},
715 {"atan2", 2, 2, FEARG_1, NULL,
716 ret_float, FLOAT_FUNC(f_atan2)},
717 {"balloon_gettext", 0, 0, 0, NULL,
718 ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100719#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100720 f_balloon_gettext
721#else
722 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100723#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100724 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200725 {"balloon_show", 1, 1, FEARG_1, NULL,
726 ret_void,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100727#ifdef FEAT_BEVAL
728 f_balloon_show
729#else
730 NULL
731#endif
732 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200733 {"balloon_split", 1, 1, FEARG_1, NULL,
734 ret_list_string,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100735#if defined(FEAT_BEVAL_TERM)
736 f_balloon_split
737#else
738 NULL
739#endif
740 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200741 {"browse", 4, 4, 0, NULL,
742 ret_string, f_browse},
743 {"browsedir", 2, 2, 0, NULL,
744 ret_string, f_browsedir},
745 {"bufadd", 1, 1, FEARG_1, NULL,
746 ret_number, f_bufadd},
747 {"bufexists", 1, 1, FEARG_1, NULL,
748 ret_number, f_bufexists},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200749 {"buffer_exists", 1, 1, FEARG_1, NULL, // obsolete
750 ret_number, f_bufexists},
751 {"buffer_name", 0, 1, FEARG_1, NULL, // obsolete
752 ret_string, f_bufname},
753 {"buffer_number", 0, 1, FEARG_1, NULL, // obsolete
754 ret_number, f_bufnr},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200755 {"buflisted", 1, 1, FEARG_1, NULL,
756 ret_number, f_buflisted},
757 {"bufload", 1, 1, FEARG_1, NULL,
758 ret_void, f_bufload},
759 {"bufloaded", 1, 1, FEARG_1, NULL,
760 ret_number, f_bufloaded},
761 {"bufname", 0, 1, FEARG_1, NULL,
762 ret_string, f_bufname},
763 {"bufnr", 0, 2, FEARG_1, NULL,
764 ret_number, f_bufnr},
765 {"bufwinid", 1, 1, FEARG_1, NULL,
766 ret_number, f_bufwinid},
767 {"bufwinnr", 1, 1, FEARG_1, NULL,
768 ret_number, f_bufwinnr},
769 {"byte2line", 1, 1, FEARG_1, NULL,
770 ret_number, f_byte2line},
771 {"byteidx", 2, 2, FEARG_1, NULL,
772 ret_number, f_byteidx},
773 {"byteidxcomp", 2, 2, FEARG_1, NULL,
774 ret_number, f_byteidxcomp},
775 {"call", 2, 3, FEARG_1, NULL,
776 ret_any, f_call},
777 {"ceil", 1, 1, FEARG_1, NULL,
778 ret_float, FLOAT_FUNC(f_ceil)},
779 {"ch_canread", 1, 1, FEARG_1, NULL,
780 ret_number, JOB_FUNC(f_ch_canread)},
781 {"ch_close", 1, 1, FEARG_1, NULL,
782 ret_void, JOB_FUNC(f_ch_close)},
783 {"ch_close_in", 1, 1, FEARG_1, NULL,
784 ret_void, JOB_FUNC(f_ch_close_in)},
785 {"ch_evalexpr", 2, 3, FEARG_1, NULL,
786 ret_any, JOB_FUNC(f_ch_evalexpr)},
787 {"ch_evalraw", 2, 3, FEARG_1, NULL,
788 ret_any, JOB_FUNC(f_ch_evalraw)},
789 {"ch_getbufnr", 2, 2, FEARG_1, NULL,
790 ret_number, JOB_FUNC(f_ch_getbufnr)},
791 {"ch_getjob", 1, 1, FEARG_1, NULL,
792 ret_job, JOB_FUNC(f_ch_getjob)},
793 {"ch_info", 1, 1, FEARG_1, NULL,
794 ret_dict_any, JOB_FUNC(f_ch_info)},
795 {"ch_log", 1, 2, FEARG_1, NULL,
796 ret_void, JOB_FUNC(f_ch_log)},
797 {"ch_logfile", 1, 2, FEARG_1, NULL,
798 ret_void, JOB_FUNC(f_ch_logfile)},
799 {"ch_open", 1, 2, FEARG_1, NULL,
800 ret_channel, JOB_FUNC(f_ch_open)},
801 {"ch_read", 1, 2, FEARG_1, NULL,
802 ret_string, JOB_FUNC(f_ch_read)},
803 {"ch_readblob", 1, 2, FEARG_1, NULL,
804 ret_blob, JOB_FUNC(f_ch_readblob)},
805 {"ch_readraw", 1, 2, FEARG_1, NULL,
806 ret_string, JOB_FUNC(f_ch_readraw)},
807 {"ch_sendexpr", 2, 3, FEARG_1, NULL,
808 ret_void, JOB_FUNC(f_ch_sendexpr)},
809 {"ch_sendraw", 2, 3, FEARG_1, NULL,
810 ret_void, JOB_FUNC(f_ch_sendraw)},
811 {"ch_setoptions", 2, 2, FEARG_1, NULL,
812 ret_void, JOB_FUNC(f_ch_setoptions)},
813 {"ch_status", 1, 2, FEARG_1, NULL,
814 ret_string, JOB_FUNC(f_ch_status)},
815 {"changenr", 0, 0, 0, NULL,
816 ret_number, f_changenr},
817 {"char2nr", 1, 2, FEARG_1, NULL,
818 ret_number, f_char2nr},
819 {"charclass", 1, 1, FEARG_1, NULL,
820 ret_number, f_charclass},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100821 {"charcol", 1, 1, FEARG_1, NULL,
822 ret_number, f_charcol},
Bram Moolenaar17793ef2020-12-28 12:56:58 +0100823 {"charidx", 2, 3, FEARG_1, NULL,
824 ret_number, f_charidx},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200825 {"chdir", 1, 1, FEARG_1, NULL,
826 ret_string, f_chdir},
827 {"cindent", 1, 1, FEARG_1, NULL,
828 ret_number, f_cindent},
829 {"clearmatches", 0, 1, FEARG_1, NULL,
830 ret_void, f_clearmatches},
831 {"col", 1, 1, FEARG_1, NULL,
832 ret_number, f_col},
833 {"complete", 2, 2, FEARG_2, NULL,
834 ret_void, f_complete},
835 {"complete_add", 1, 1, FEARG_1, NULL,
836 ret_number, f_complete_add},
837 {"complete_check", 0, 0, 0, NULL,
838 ret_number, f_complete_check},
839 {"complete_info", 0, 1, FEARG_1, NULL,
840 ret_dict_any, f_complete_info},
841 {"confirm", 1, 4, FEARG_1, NULL,
842 ret_number, f_confirm},
843 {"copy", 1, 1, FEARG_1, NULL,
844 ret_first_arg, f_copy},
845 {"cos", 1, 1, FEARG_1, NULL,
846 ret_float, FLOAT_FUNC(f_cos)},
847 {"cosh", 1, 1, FEARG_1, NULL,
848 ret_float, FLOAT_FUNC(f_cosh)},
849 {"count", 2, 4, FEARG_1, NULL,
850 ret_number, f_count},
851 {"cscope_connection",0,3, 0, NULL,
852 ret_number, f_cscope_connection},
853 {"cursor", 1, 3, FEARG_1, NULL,
854 ret_number, f_cursor},
855 {"debugbreak", 1, 1, FEARG_1, NULL,
856 ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100857#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100858 f_debugbreak
859#else
860 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200861#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100862 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200863 {"deepcopy", 1, 2, FEARG_1, NULL,
864 ret_first_arg, f_deepcopy},
865 {"delete", 1, 2, FEARG_1, NULL,
866 ret_number, f_delete},
867 {"deletebufline", 2, 3, FEARG_1, NULL,
868 ret_number, f_deletebufline},
869 {"did_filetype", 0, 0, 0, NULL,
870 ret_number, f_did_filetype},
871 {"diff_filler", 1, 1, FEARG_1, NULL,
872 ret_number, f_diff_filler},
873 {"diff_hlID", 2, 2, FEARG_1, NULL,
874 ret_number, f_diff_hlID},
875 {"echoraw", 1, 1, FEARG_1, NULL,
876 ret_number, f_echoraw},
877 {"empty", 1, 1, FEARG_1, NULL,
878 ret_number, f_empty},
879 {"environ", 0, 0, 0, NULL,
880 ret_dict_string, f_environ},
881 {"escape", 2, 2, FEARG_1, NULL,
882 ret_string, f_escape},
883 {"eval", 1, 1, FEARG_1, NULL,
884 ret_any, f_eval},
885 {"eventhandler", 0, 0, 0, NULL,
886 ret_number, f_eventhandler},
887 {"executable", 1, 1, FEARG_1, NULL,
888 ret_number, f_executable},
889 {"execute", 1, 2, FEARG_1, NULL,
890 ret_string, f_execute},
891 {"exepath", 1, 1, FEARG_1, NULL,
892 ret_string, f_exepath},
893 {"exists", 1, 1, FEARG_1, NULL,
894 ret_number, f_exists},
895 {"exp", 1, 1, FEARG_1, NULL,
896 ret_float, FLOAT_FUNC(f_exp)},
897 {"expand", 1, 3, FEARG_1, NULL,
898 ret_any, f_expand},
899 {"expandcmd", 1, 1, FEARG_1, NULL,
900 ret_string, f_expandcmd},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100901 {"extend", 2, 3, FEARG_1, arg23_extend,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200902 ret_first_arg, f_extend},
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100903 {"extendnew", 2, 3, FEARG_1, arg23_extendnew,
904 ret_first_cont, f_extendnew},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200905 {"feedkeys", 1, 2, FEARG_1, NULL,
906 ret_void, f_feedkeys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200907 {"file_readable", 1, 1, FEARG_1, NULL, // obsolete
908 ret_number, f_filereadable},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200909 {"filereadable", 1, 1, FEARG_1, NULL,
910 ret_number, f_filereadable},
911 {"filewritable", 1, 1, FEARG_1, NULL,
912 ret_number, f_filewritable},
913 {"filter", 2, 2, FEARG_1, NULL,
914 ret_first_arg, f_filter},
915 {"finddir", 1, 3, FEARG_1, NULL,
916 ret_string, f_finddir},
917 {"findfile", 1, 3, FEARG_1, NULL,
918 ret_string, f_findfile},
919 {"flatten", 1, 2, FEARG_1, NULL,
920 ret_list_any, f_flatten},
921 {"float2nr", 1, 1, FEARG_1, NULL,
922 ret_number, FLOAT_FUNC(f_float2nr)},
923 {"floor", 1, 1, FEARG_1, NULL,
924 ret_float, FLOAT_FUNC(f_floor)},
925 {"fmod", 2, 2, FEARG_1, NULL,
926 ret_float, FLOAT_FUNC(f_fmod)},
927 {"fnameescape", 1, 1, FEARG_1, NULL,
928 ret_string, f_fnameescape},
929 {"fnamemodify", 2, 2, FEARG_1, NULL,
930 ret_string, f_fnamemodify},
931 {"foldclosed", 1, 1, FEARG_1, NULL,
932 ret_number, f_foldclosed},
933 {"foldclosedend", 1, 1, FEARG_1, NULL,
934 ret_number, f_foldclosedend},
935 {"foldlevel", 1, 1, FEARG_1, NULL,
936 ret_number, f_foldlevel},
937 {"foldtext", 0, 0, 0, NULL,
938 ret_string, f_foldtext},
939 {"foldtextresult", 1, 1, FEARG_1, NULL,
940 ret_string, f_foldtextresult},
941 {"foreground", 0, 0, 0, NULL,
942 ret_void, f_foreground},
943 {"funcref", 1, 3, FEARG_1, NULL,
944 ret_func_any, f_funcref},
945 {"function", 1, 3, FEARG_1, NULL,
946 ret_f_function, f_function},
947 {"garbagecollect", 0, 1, 0, NULL,
948 ret_void, f_garbagecollect},
949 {"get", 2, 3, FEARG_1, NULL,
950 ret_any, f_get},
951 {"getbufinfo", 0, 1, FEARG_1, NULL,
952 ret_list_dict_any, f_getbufinfo},
953 {"getbufline", 2, 3, FEARG_1, NULL,
954 ret_list_string, f_getbufline},
955 {"getbufvar", 2, 3, FEARG_1, NULL,
956 ret_any, f_getbufvar},
957 {"getchangelist", 0, 1, FEARG_1, NULL,
958 ret_list_any, f_getchangelist},
959 {"getchar", 0, 1, 0, NULL,
960 ret_number, f_getchar},
961 {"getcharmod", 0, 0, 0, NULL,
962 ret_number, f_getcharmod},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100963 {"getcharpos", 1, 1, FEARG_1, NULL,
964 ret_list_number, f_getcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200965 {"getcharsearch", 0, 0, 0, NULL,
966 ret_dict_any, f_getcharsearch},
967 {"getcmdline", 0, 0, 0, NULL,
968 ret_string, f_getcmdline},
969 {"getcmdpos", 0, 0, 0, NULL,
970 ret_number, f_getcmdpos},
971 {"getcmdtype", 0, 0, 0, NULL,
972 ret_string, f_getcmdtype},
973 {"getcmdwintype", 0, 0, 0, NULL,
974 ret_string, f_getcmdwintype},
975 {"getcompletion", 2, 3, FEARG_1, NULL,
976 ret_list_string, f_getcompletion},
977 {"getcurpos", 0, 1, FEARG_1, NULL,
978 ret_list_number, f_getcurpos},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100979 {"getcursorcharpos", 0, 1, FEARG_1, NULL,
980 ret_list_number, f_getcursorcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200981 {"getcwd", 0, 2, FEARG_1, NULL,
982 ret_string, f_getcwd},
983 {"getenv", 1, 1, FEARG_1, NULL,
984 ret_string, f_getenv},
985 {"getfontname", 0, 1, 0, NULL,
986 ret_string, f_getfontname},
987 {"getfperm", 1, 1, FEARG_1, NULL,
988 ret_string, f_getfperm},
989 {"getfsize", 1, 1, FEARG_1, NULL,
990 ret_number, f_getfsize},
991 {"getftime", 1, 1, FEARG_1, NULL,
992 ret_number, f_getftime},
993 {"getftype", 1, 1, FEARG_1, NULL,
994 ret_string, f_getftype},
995 {"getimstatus", 0, 0, 0, NULL,
996 ret_number, f_getimstatus},
997 {"getjumplist", 0, 2, FEARG_1, NULL,
998 ret_list_any, f_getjumplist},
999 {"getline", 1, 2, FEARG_1, NULL,
1000 ret_f_getline, f_getline},
1001 {"getloclist", 1, 2, 0, NULL,
1002 ret_list_or_dict_1, f_getloclist},
1003 {"getmarklist", 0, 1, FEARG_1, NULL,
1004 ret_list_dict_any, f_getmarklist},
1005 {"getmatches", 0, 1, 0, NULL,
1006 ret_list_dict_any, f_getmatches},
1007 {"getmousepos", 0, 0, 0, NULL,
1008 ret_dict_number, f_getmousepos},
1009 {"getpid", 0, 0, 0, NULL,
1010 ret_number, f_getpid},
1011 {"getpos", 1, 1, FEARG_1, NULL,
1012 ret_list_number, f_getpos},
1013 {"getqflist", 0, 1, 0, NULL,
1014 ret_list_or_dict_0, f_getqflist},
1015 {"getreg", 0, 3, FEARG_1, NULL,
1016 ret_getreg, f_getreg},
1017 {"getreginfo", 0, 1, FEARG_1, NULL,
1018 ret_dict_any, f_getreginfo},
1019 {"getregtype", 0, 1, FEARG_1, NULL,
1020 ret_string, f_getregtype},
1021 {"gettabinfo", 0, 1, FEARG_1, NULL,
1022 ret_list_dict_any, f_gettabinfo},
1023 {"gettabvar", 2, 3, FEARG_1, NULL,
1024 ret_any, f_gettabvar},
1025 {"gettabwinvar", 3, 4, FEARG_1, NULL,
1026 ret_any, f_gettabwinvar},
1027 {"gettagstack", 0, 1, FEARG_1, NULL,
1028 ret_dict_any, f_gettagstack},
1029 {"gettext", 1, 1, FEARG_1, NULL,
1030 ret_string, f_gettext},
1031 {"getwininfo", 0, 1, FEARG_1, NULL,
1032 ret_list_dict_any, f_getwininfo},
1033 {"getwinpos", 0, 1, FEARG_1, NULL,
1034 ret_list_number, f_getwinpos},
1035 {"getwinposx", 0, 0, 0, NULL,
1036 ret_number, f_getwinposx},
1037 {"getwinposy", 0, 0, 0, NULL,
1038 ret_number, f_getwinposy},
1039 {"getwinvar", 2, 3, FEARG_1, NULL,
1040 ret_any, f_getwinvar},
1041 {"glob", 1, 4, FEARG_1, NULL,
1042 ret_any, f_glob},
1043 {"glob2regpat", 1, 1, FEARG_1, NULL,
1044 ret_string, f_glob2regpat},
1045 {"globpath", 2, 5, FEARG_2, NULL,
1046 ret_any, f_globpath},
1047 {"has", 1, 2, 0, NULL,
1048 ret_number, f_has},
1049 {"has_key", 2, 2, FEARG_1, NULL,
1050 ret_number, f_has_key},
1051 {"haslocaldir", 0, 2, FEARG_1, NULL,
1052 ret_number, f_haslocaldir},
1053 {"hasmapto", 1, 3, FEARG_1, NULL,
1054 ret_number, f_hasmapto},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001055 {"highlightID", 1, 1, FEARG_1, NULL, // obsolete
1056 ret_number, f_hlID},
1057 {"highlight_exists",1, 1, FEARG_1, NULL, // obsolete
1058 ret_number, f_hlexists},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001059 {"histadd", 2, 2, FEARG_2, NULL,
1060 ret_number, f_histadd},
1061 {"histdel", 1, 2, FEARG_1, NULL,
1062 ret_number, f_histdel},
1063 {"histget", 1, 2, FEARG_1, NULL,
1064 ret_string, f_histget},
1065 {"histnr", 1, 1, FEARG_1, NULL,
1066 ret_number, f_histnr},
1067 {"hlID", 1, 1, FEARG_1, NULL,
1068 ret_number, f_hlID},
1069 {"hlexists", 1, 1, FEARG_1, NULL,
1070 ret_number, f_hlexists},
1071 {"hostname", 0, 0, 0, NULL,
1072 ret_string, f_hostname},
1073 {"iconv", 3, 3, FEARG_1, NULL,
1074 ret_string, f_iconv},
1075 {"indent", 1, 1, FEARG_1, NULL,
1076 ret_number, f_indent},
1077 {"index", 2, 4, FEARG_1, NULL,
1078 ret_number, f_index},
1079 {"input", 1, 3, FEARG_1, NULL,
1080 ret_string, f_input},
1081 {"inputdialog", 1, 3, FEARG_1, NULL,
1082 ret_string, f_inputdialog},
1083 {"inputlist", 1, 1, FEARG_1, NULL,
1084 ret_number, f_inputlist},
1085 {"inputrestore", 0, 0, 0, NULL,
1086 ret_number, f_inputrestore},
1087 {"inputsave", 0, 0, 0, NULL,
1088 ret_number, f_inputsave},
1089 {"inputsecret", 1, 2, FEARG_1, NULL,
1090 ret_string, f_inputsecret},
Bram Moolenaarca174532020-10-21 16:42:22 +02001091 {"insert", 2, 3, FEARG_1, arg3_insert,
Bram Moolenaar94738d82020-10-21 14:25:07 +02001092 ret_first_arg, f_insert},
1093 {"interrupt", 0, 0, 0, NULL,
1094 ret_void, f_interrupt},
1095 {"invert", 1, 1, FEARG_1, NULL,
1096 ret_number, f_invert},
1097 {"isdirectory", 1, 1, FEARG_1, NULL,
1098 ret_number, f_isdirectory},
1099 {"isinf", 1, 1, FEARG_1, NULL,
1100 ret_number, MATH_FUNC(f_isinf)},
1101 {"islocked", 1, 1, FEARG_1, NULL,
1102 ret_number, f_islocked},
1103 {"isnan", 1, 1, FEARG_1, NULL,
1104 ret_number, MATH_FUNC(f_isnan)},
1105 {"items", 1, 1, FEARG_1, NULL,
1106 ret_list_any, f_items},
1107 {"job_getchannel", 1, 1, FEARG_1, NULL,
1108 ret_channel, JOB_FUNC(f_job_getchannel)},
1109 {"job_info", 0, 1, FEARG_1, NULL,
Bram Moolenaar64ed4d42021-01-12 21:22:31 +01001110 ret_job_info, JOB_FUNC(f_job_info)},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001111 {"job_setoptions", 2, 2, FEARG_1, NULL,
1112 ret_void, JOB_FUNC(f_job_setoptions)},
1113 {"job_start", 1, 2, FEARG_1, NULL,
1114 ret_job, JOB_FUNC(f_job_start)},
1115 {"job_status", 1, 1, FEARG_1, NULL,
1116 ret_string, JOB_FUNC(f_job_status)},
1117 {"job_stop", 1, 2, FEARG_1, NULL,
1118 ret_number, JOB_FUNC(f_job_stop)},
1119 {"join", 1, 2, FEARG_1, NULL,
1120 ret_string, f_join},
1121 {"js_decode", 1, 1, FEARG_1, NULL,
1122 ret_any, f_js_decode},
1123 {"js_encode", 1, 1, FEARG_1, NULL,
1124 ret_string, f_js_encode},
1125 {"json_decode", 1, 1, FEARG_1, NULL,
1126 ret_any, f_json_decode},
1127 {"json_encode", 1, 1, FEARG_1, NULL,
1128 ret_string, f_json_encode},
1129 {"keys", 1, 1, FEARG_1, NULL,
1130 ret_list_string, f_keys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001131 {"last_buffer_nr", 0, 0, 0, NULL, // obsolete
1132 ret_number, f_last_buffer_nr},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001133 {"len", 1, 1, FEARG_1, NULL,
1134 ret_number, f_len},
1135 {"libcall", 3, 3, FEARG_3, NULL,
1136 ret_string, f_libcall},
1137 {"libcallnr", 3, 3, FEARG_3, NULL,
1138 ret_number, f_libcallnr},
1139 {"line", 1, 2, FEARG_1, NULL,
1140 ret_number, f_line},
1141 {"line2byte", 1, 1, FEARG_1, NULL,
1142 ret_number, f_line2byte},
1143 {"lispindent", 1, 1, FEARG_1, NULL,
1144 ret_number, f_lispindent},
1145 {"list2str", 1, 2, FEARG_1, NULL,
1146 ret_string, f_list2str},
1147 {"listener_add", 1, 2, FEARG_2, NULL,
1148 ret_number, f_listener_add},
1149 {"listener_flush", 0, 1, FEARG_1, NULL,
1150 ret_void, f_listener_flush},
1151 {"listener_remove", 1, 1, FEARG_1, NULL,
1152 ret_number, f_listener_remove},
1153 {"localtime", 0, 0, 0, NULL,
1154 ret_number, f_localtime},
1155 {"log", 1, 1, FEARG_1, NULL,
1156 ret_float, FLOAT_FUNC(f_log)},
1157 {"log10", 1, 1, FEARG_1, NULL,
1158 ret_float, FLOAT_FUNC(f_log10)},
1159 {"luaeval", 1, 2, FEARG_1, NULL,
1160 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001161#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +01001162 f_luaeval
1163#else
1164 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001165#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001166 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001167 {"map", 2, 2, FEARG_1, NULL,
Bram Moolenaarea696852020-11-09 18:31:39 +01001168 ret_first_cont, f_map},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001169 {"maparg", 1, 4, FEARG_1, NULL,
1170 ret_maparg, f_maparg},
1171 {"mapcheck", 1, 3, FEARG_1, NULL,
1172 ret_string, f_mapcheck},
Bram Moolenaarea696852020-11-09 18:31:39 +01001173 {"mapnew", 2, 2, FEARG_1, NULL,
1174 ret_first_cont, f_mapnew},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001175 {"mapset", 3, 3, FEARG_1, NULL,
1176 ret_void, f_mapset},
1177 {"match", 2, 4, FEARG_1, NULL,
1178 ret_any, f_match},
1179 {"matchadd", 2, 5, FEARG_1, NULL,
1180 ret_number, f_matchadd},
1181 {"matchaddpos", 2, 5, FEARG_1, NULL,
1182 ret_number, f_matchaddpos},
1183 {"matcharg", 1, 1, FEARG_1, NULL,
1184 ret_list_string, f_matcharg},
1185 {"matchdelete", 1, 2, FEARG_1, NULL,
1186 ret_number, f_matchdelete},
1187 {"matchend", 2, 4, FEARG_1, NULL,
1188 ret_number, f_matchend},
1189 {"matchfuzzy", 2, 3, FEARG_1, NULL,
1190 ret_list_string, f_matchfuzzy},
1191 {"matchfuzzypos", 2, 3, FEARG_1, NULL,
1192 ret_list_any, f_matchfuzzypos},
1193 {"matchlist", 2, 4, FEARG_1, NULL,
1194 ret_list_string, f_matchlist},
1195 {"matchstr", 2, 4, FEARG_1, NULL,
1196 ret_string, f_matchstr},
1197 {"matchstrpos", 2, 4, FEARG_1, NULL,
1198 ret_list_any, f_matchstrpos},
1199 {"max", 1, 1, FEARG_1, NULL,
1200 ret_any, f_max},
1201 {"menu_info", 1, 2, FEARG_1, NULL,
1202 ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001203#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +01001204 f_menu_info
1205#else
1206 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001207#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001208 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001209 {"min", 1, 1, FEARG_1, NULL,
1210 ret_any, f_min},
1211 {"mkdir", 1, 3, FEARG_1, NULL,
1212 ret_number, f_mkdir},
1213 {"mode", 0, 1, FEARG_1, NULL,
1214 ret_string, f_mode},
1215 {"mzeval", 1, 1, FEARG_1, NULL,
1216 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001217#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001218 f_mzeval
1219#else
1220 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001221#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001222 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001223 {"nextnonblank", 1, 1, FEARG_1, NULL,
1224 ret_number, f_nextnonblank},
1225 {"nr2char", 1, 2, FEARG_1, NULL,
1226 ret_string, f_nr2char},
1227 {"or", 2, 2, FEARG_1, NULL,
1228 ret_number, f_or},
1229 {"pathshorten", 1, 2, FEARG_1, NULL,
1230 ret_string, f_pathshorten},
1231 {"perleval", 1, 1, FEARG_1, NULL,
1232 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001233#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +01001234 f_perleval
1235#else
1236 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001237#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001238 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001239 {"popup_atcursor", 2, 2, FEARG_1, NULL,
1240 ret_number, PROP_FUNC(f_popup_atcursor)},
1241 {"popup_beval", 2, 2, FEARG_1, NULL,
1242 ret_number, PROP_FUNC(f_popup_beval)},
1243 {"popup_clear", 0, 1, 0, NULL,
1244 ret_void, PROP_FUNC(f_popup_clear)},
1245 {"popup_close", 1, 2, FEARG_1, NULL,
1246 ret_void, PROP_FUNC(f_popup_close)},
1247 {"popup_create", 2, 2, FEARG_1, NULL,
1248 ret_number, PROP_FUNC(f_popup_create)},
1249 {"popup_dialog", 2, 2, FEARG_1, NULL,
1250 ret_number, PROP_FUNC(f_popup_dialog)},
1251 {"popup_filter_menu", 2, 2, 0, NULL,
1252 ret_bool, PROP_FUNC(f_popup_filter_menu)},
1253 {"popup_filter_yesno", 2, 2, 0, NULL,
1254 ret_bool, PROP_FUNC(f_popup_filter_yesno)},
1255 {"popup_findinfo", 0, 0, 0, NULL,
1256 ret_number, PROP_FUNC(f_popup_findinfo)},
1257 {"popup_findpreview", 0, 0, 0, NULL,
1258 ret_number, PROP_FUNC(f_popup_findpreview)},
1259 {"popup_getoptions", 1, 1, FEARG_1, NULL,
1260 ret_dict_any, PROP_FUNC(f_popup_getoptions)},
1261 {"popup_getpos", 1, 1, FEARG_1, NULL,
1262 ret_dict_any, PROP_FUNC(f_popup_getpos)},
1263 {"popup_hide", 1, 1, FEARG_1, NULL,
1264 ret_void, PROP_FUNC(f_popup_hide)},
1265 {"popup_list", 0, 0, 0, NULL,
1266 ret_list_number, PROP_FUNC(f_popup_list)},
1267 {"popup_locate", 2, 2, 0, NULL,
1268 ret_number, PROP_FUNC(f_popup_locate)},
1269 {"popup_menu", 2, 2, FEARG_1, NULL,
1270 ret_number, PROP_FUNC(f_popup_menu)},
1271 {"popup_move", 2, 2, FEARG_1, NULL,
1272 ret_void, PROP_FUNC(f_popup_move)},
1273 {"popup_notification", 2, 2, FEARG_1, NULL,
1274 ret_number, PROP_FUNC(f_popup_notification)},
1275 {"popup_setoptions", 2, 2, FEARG_1, NULL,
1276 ret_void, PROP_FUNC(f_popup_setoptions)},
1277 {"popup_settext", 2, 2, FEARG_1, NULL,
1278 ret_void, PROP_FUNC(f_popup_settext)},
1279 {"popup_show", 1, 1, FEARG_1, NULL,
1280 ret_void, PROP_FUNC(f_popup_show)},
1281 {"pow", 2, 2, FEARG_1, NULL,
1282 ret_float, FLOAT_FUNC(f_pow)},
1283 {"prevnonblank", 1, 1, FEARG_1, NULL,
1284 ret_number, f_prevnonblank},
1285 {"printf", 1, 19, FEARG_2, NULL,
1286 ret_string, f_printf},
1287 {"prompt_getprompt", 1, 1, FEARG_1, NULL,
1288 ret_string, JOB_FUNC(f_prompt_getprompt)},
1289 {"prompt_setcallback", 2, 2, FEARG_1, NULL,
1290 ret_void, JOB_FUNC(f_prompt_setcallback)},
1291 {"prompt_setinterrupt", 2, 2, FEARG_1, NULL,
1292 ret_void, JOB_FUNC(f_prompt_setinterrupt)},
1293 {"prompt_setprompt", 2, 2, FEARG_1, NULL,
1294 ret_void, JOB_FUNC(f_prompt_setprompt)},
1295 {"prop_add", 3, 3, FEARG_1, NULL,
1296 ret_void, PROP_FUNC(f_prop_add)},
1297 {"prop_clear", 1, 3, FEARG_1, NULL,
1298 ret_void, PROP_FUNC(f_prop_clear)},
1299 {"prop_find", 1, 2, FEARG_1, NULL,
1300 ret_dict_any, PROP_FUNC(f_prop_find)},
1301 {"prop_list", 1, 2, FEARG_1, NULL,
1302 ret_list_dict_any, PROP_FUNC(f_prop_list)},
1303 {"prop_remove", 1, 3, FEARG_1, NULL,
1304 ret_number, PROP_FUNC(f_prop_remove)},
1305 {"prop_type_add", 2, 2, FEARG_1, NULL,
1306 ret_void, PROP_FUNC(f_prop_type_add)},
1307 {"prop_type_change", 2, 2, FEARG_1, NULL,
1308 ret_void, PROP_FUNC(f_prop_type_change)},
1309 {"prop_type_delete", 1, 2, FEARG_1, NULL,
1310 ret_void, PROP_FUNC(f_prop_type_delete)},
1311 {"prop_type_get", 1, 2, FEARG_1, NULL,
1312 ret_dict_any, PROP_FUNC(f_prop_type_get)},
1313 {"prop_type_list", 0, 1, FEARG_1, NULL,
1314 ret_list_string, PROP_FUNC(f_prop_type_list)},
1315 {"pum_getpos", 0, 0, 0, NULL,
1316 ret_dict_number, f_pum_getpos},
1317 {"pumvisible", 0, 0, 0, NULL,
1318 ret_number, f_pumvisible},
1319 {"py3eval", 1, 1, FEARG_1, NULL,
1320 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001321#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +01001322 f_py3eval
1323#else
1324 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001325#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001326 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001327 {"pyeval", 1, 1, FEARG_1, NULL,
1328 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001329#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +01001330 f_pyeval
1331#else
1332 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001333#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001334 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001335 {"pyxeval", 1, 1, FEARG_1, NULL,
1336 ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001337#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +01001338 f_pyxeval
1339#else
1340 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001341#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001342 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001343 {"rand", 0, 1, FEARG_1, NULL,
1344 ret_number, f_rand},
1345 {"range", 1, 3, FEARG_1, NULL,
1346 ret_list_number, f_range},
1347 {"readdir", 1, 3, FEARG_1, NULL,
1348 ret_list_string, f_readdir},
1349 {"readdirex", 1, 3, FEARG_1, NULL,
1350 ret_list_dict_any, f_readdirex},
1351 {"readfile", 1, 3, FEARG_1, NULL,
1352 ret_any, f_readfile},
1353 {"reduce", 2, 3, FEARG_1, NULL,
1354 ret_any, f_reduce},
1355 {"reg_executing", 0, 0, 0, NULL,
1356 ret_string, f_reg_executing},
1357 {"reg_recording", 0, 0, 0, NULL,
1358 ret_string, f_reg_recording},
1359 {"reltime", 0, 2, FEARG_1, NULL,
1360 ret_list_any, f_reltime},
1361 {"reltimefloat", 1, 1, FEARG_1, NULL,
1362 ret_float, FLOAT_FUNC(f_reltimefloat)},
1363 {"reltimestr", 1, 1, FEARG_1, NULL,
1364 ret_string, f_reltimestr},
1365 {"remote_expr", 2, 4, FEARG_1, NULL,
1366 ret_string, f_remote_expr},
1367 {"remote_foreground", 1, 1, FEARG_1, NULL,
1368 ret_string, f_remote_foreground},
1369 {"remote_peek", 1, 2, FEARG_1, NULL,
1370 ret_number, f_remote_peek},
1371 {"remote_read", 1, 2, FEARG_1, NULL,
1372 ret_string, f_remote_read},
1373 {"remote_send", 2, 3, FEARG_1, NULL,
1374 ret_string, f_remote_send},
1375 {"remote_startserver", 1, 1, FEARG_1, NULL,
1376 ret_void, f_remote_startserver},
1377 {"remove", 2, 3, FEARG_1, NULL,
1378 ret_remove, f_remove},
1379 {"rename", 2, 2, FEARG_1, NULL,
1380 ret_number, f_rename},
1381 {"repeat", 2, 2, FEARG_1, NULL,
1382 ret_first_arg, f_repeat},
1383 {"resolve", 1, 1, FEARG_1, NULL,
1384 ret_string, f_resolve},
1385 {"reverse", 1, 1, FEARG_1, NULL,
1386 ret_first_arg, f_reverse},
1387 {"round", 1, 1, FEARG_1, NULL,
1388 ret_float, FLOAT_FUNC(f_round)},
1389 {"rubyeval", 1, 1, FEARG_1, NULL,
1390 ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001391#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +01001392 f_rubyeval
1393#else
1394 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001395#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001396 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001397 {"screenattr", 2, 2, FEARG_1, NULL,
1398 ret_number, f_screenattr},
1399 {"screenchar", 2, 2, FEARG_1, NULL,
1400 ret_number, f_screenchar},
1401 {"screenchars", 2, 2, FEARG_1, NULL,
1402 ret_list_number, f_screenchars},
1403 {"screencol", 0, 0, 0, NULL,
1404 ret_number, f_screencol},
1405 {"screenpos", 3, 3, FEARG_1, NULL,
1406 ret_dict_number, f_screenpos},
1407 {"screenrow", 0, 0, 0, NULL,
1408 ret_number, f_screenrow},
1409 {"screenstring", 2, 2, FEARG_1, NULL,
1410 ret_string, f_screenstring},
1411 {"search", 1, 5, FEARG_1, NULL,
1412 ret_number, f_search},
1413 {"searchcount", 0, 1, FEARG_1, NULL,
1414 ret_dict_any, f_searchcount},
1415 {"searchdecl", 1, 3, FEARG_1, NULL,
1416 ret_number, f_searchdecl},
1417 {"searchpair", 3, 7, 0, NULL,
1418 ret_number, f_searchpair},
1419 {"searchpairpos", 3, 7, 0, NULL,
1420 ret_list_number, f_searchpairpos},
1421 {"searchpos", 1, 5, FEARG_1, NULL,
1422 ret_list_number, f_searchpos},
1423 {"server2client", 2, 2, FEARG_1, NULL,
1424 ret_number, f_server2client},
1425 {"serverlist", 0, 0, 0, NULL,
1426 ret_string, f_serverlist},
1427 {"setbufline", 3, 3, FEARG_3, NULL,
1428 ret_number, f_setbufline},
1429 {"setbufvar", 3, 3, FEARG_3, NULL,
1430 ret_void, f_setbufvar},
1431 {"setcellwidths", 1, 1, FEARG_1, NULL,
1432 ret_void, f_setcellwidths},
Bram Moolenaar6f02b002021-01-10 20:22:54 +01001433 {"setcharpos", 2, 2, FEARG_2, NULL,
1434 ret_number, f_setcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001435 {"setcharsearch", 1, 1, FEARG_1, NULL,
1436 ret_void, f_setcharsearch},
1437 {"setcmdpos", 1, 1, FEARG_1, NULL,
1438 ret_number, f_setcmdpos},
Bram Moolenaar6f02b002021-01-10 20:22:54 +01001439 {"setcursorcharpos", 1, 3, FEARG_1, NULL,
1440 ret_number, f_setcursorcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001441 {"setenv", 2, 2, FEARG_2, NULL,
1442 ret_void, f_setenv},
1443 {"setfperm", 2, 2, FEARG_1, NULL,
1444 ret_number, f_setfperm},
1445 {"setline", 2, 2, FEARG_2, NULL,
1446 ret_number, f_setline},
1447 {"setloclist", 2, 4, FEARG_2, NULL,
1448 ret_number, f_setloclist},
1449 {"setmatches", 1, 2, FEARG_1, NULL,
1450 ret_number, f_setmatches},
1451 {"setpos", 2, 2, FEARG_2, NULL,
1452 ret_number, f_setpos},
1453 {"setqflist", 1, 3, FEARG_1, NULL,
1454 ret_number, f_setqflist},
1455 {"setreg", 2, 3, FEARG_2, NULL,
1456 ret_number, f_setreg},
1457 {"settabvar", 3, 3, FEARG_3, NULL,
1458 ret_void, f_settabvar},
1459 {"settabwinvar", 4, 4, FEARG_4, NULL,
1460 ret_void, f_settabwinvar},
1461 {"settagstack", 2, 3, FEARG_2, NULL,
1462 ret_number, f_settagstack},
1463 {"setwinvar", 3, 3, FEARG_3, NULL,
1464 ret_void, f_setwinvar},
1465 {"sha256", 1, 1, FEARG_1, NULL,
1466 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001467#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +01001468 f_sha256
1469#else
1470 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001471#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001472 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001473 {"shellescape", 1, 2, FEARG_1, NULL,
1474 ret_string, f_shellescape},
1475 {"shiftwidth", 0, 1, FEARG_1, NULL,
1476 ret_number, f_shiftwidth},
1477 {"sign_define", 1, 2, FEARG_1, NULL,
1478 ret_any, SIGN_FUNC(f_sign_define)},
1479 {"sign_getdefined", 0, 1, FEARG_1, NULL,
1480 ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
1481 {"sign_getplaced", 0, 2, FEARG_1, NULL,
1482 ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
1483 {"sign_jump", 3, 3, FEARG_1, NULL,
1484 ret_number, SIGN_FUNC(f_sign_jump)},
1485 {"sign_place", 4, 5, FEARG_1, NULL,
1486 ret_number, SIGN_FUNC(f_sign_place)},
1487 {"sign_placelist", 1, 1, FEARG_1, NULL,
1488 ret_list_number, SIGN_FUNC(f_sign_placelist)},
1489 {"sign_undefine", 0, 1, FEARG_1, NULL,
1490 ret_number, SIGN_FUNC(f_sign_undefine)},
1491 {"sign_unplace", 1, 2, FEARG_1, NULL,
1492 ret_number, SIGN_FUNC(f_sign_unplace)},
1493 {"sign_unplacelist", 1, 2, FEARG_1, NULL,
1494 ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
1495 {"simplify", 1, 1, FEARG_1, NULL,
1496 ret_string, f_simplify},
1497 {"sin", 1, 1, FEARG_1, NULL,
1498 ret_float, FLOAT_FUNC(f_sin)},
1499 {"sinh", 1, 1, FEARG_1, NULL,
1500 ret_float, FLOAT_FUNC(f_sinh)},
1501 {"sort", 1, 3, FEARG_1, NULL,
1502 ret_first_arg, f_sort},
1503 {"sound_clear", 0, 0, 0, NULL,
1504 ret_void, SOUND_FUNC(f_sound_clear)},
1505 {"sound_playevent", 1, 2, FEARG_1, NULL,
1506 ret_number, SOUND_FUNC(f_sound_playevent)},
1507 {"sound_playfile", 1, 2, FEARG_1, NULL,
1508 ret_number, SOUND_FUNC(f_sound_playfile)},
1509 {"sound_stop", 1, 1, FEARG_1, NULL,
1510 ret_void, SOUND_FUNC(f_sound_stop)},
1511 {"soundfold", 1, 1, FEARG_1, NULL,
1512 ret_string, f_soundfold},
1513 {"spellbadword", 0, 1, FEARG_1, NULL,
1514 ret_list_string, f_spellbadword},
1515 {"spellsuggest", 1, 3, FEARG_1, NULL,
1516 ret_list_string, f_spellsuggest},
1517 {"split", 1, 3, FEARG_1, NULL,
1518 ret_list_string, f_split},
1519 {"sqrt", 1, 1, FEARG_1, NULL,
1520 ret_float, FLOAT_FUNC(f_sqrt)},
1521 {"srand", 0, 1, FEARG_1, NULL,
1522 ret_list_number, f_srand},
1523 {"state", 0, 1, FEARG_1, NULL,
1524 ret_string, f_state},
1525 {"str2float", 1, 1, FEARG_1, NULL,
1526 ret_float, FLOAT_FUNC(f_str2float)},
1527 {"str2list", 1, 2, FEARG_1, NULL,
1528 ret_list_number, f_str2list},
1529 {"str2nr", 1, 3, FEARG_1, NULL,
1530 ret_number, f_str2nr},
1531 {"strcharpart", 2, 3, FEARG_1, NULL,
1532 ret_string, f_strcharpart},
1533 {"strchars", 1, 2, FEARG_1, NULL,
1534 ret_number, f_strchars},
1535 {"strdisplaywidth", 1, 2, FEARG_1, NULL,
1536 ret_number, f_strdisplaywidth},
1537 {"strftime", 1, 2, FEARG_1, NULL,
1538 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001539#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001540 f_strftime
1541#else
1542 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001543#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001544 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001545 {"strgetchar", 2, 2, FEARG_1, NULL,
1546 ret_number, f_strgetchar},
1547 {"stridx", 2, 3, FEARG_1, NULL,
1548 ret_number, f_stridx},
1549 {"string", 1, 1, FEARG_1, NULL,
1550 ret_string, f_string},
1551 {"strlen", 1, 1, FEARG_1, NULL,
1552 ret_number, f_strlen},
1553 {"strpart", 2, 4, FEARG_1, NULL,
1554 ret_string, f_strpart},
1555 {"strptime", 2, 2, FEARG_1, NULL,
1556 ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +01001557#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001558 f_strptime
1559#else
1560 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +01001561#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001562 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001563 {"strridx", 2, 3, FEARG_1, NULL,
1564 ret_number, f_strridx},
1565 {"strtrans", 1, 1, FEARG_1, NULL,
1566 ret_string, f_strtrans},
1567 {"strwidth", 1, 1, FEARG_1, NULL,
1568 ret_number, f_strwidth},
1569 {"submatch", 1, 2, FEARG_1, NULL,
1570 ret_string, f_submatch},
1571 {"substitute", 4, 4, FEARG_1, NULL,
1572 ret_string, f_substitute},
1573 {"swapinfo", 1, 1, FEARG_1, NULL,
1574 ret_dict_any, f_swapinfo},
1575 {"swapname", 1, 1, FEARG_1, NULL,
1576 ret_string, f_swapname},
1577 {"synID", 3, 3, 0, NULL,
1578 ret_number, f_synID},
1579 {"synIDattr", 2, 3, FEARG_1, NULL,
1580 ret_string, f_synIDattr},
1581 {"synIDtrans", 1, 1, FEARG_1, NULL,
1582 ret_number, f_synIDtrans},
1583 {"synconcealed", 2, 2, 0, NULL,
1584 ret_list_any, f_synconcealed},
1585 {"synstack", 2, 2, 0, NULL,
1586 ret_list_number, f_synstack},
1587 {"system", 1, 2, FEARG_1, NULL,
1588 ret_string, f_system},
1589 {"systemlist", 1, 2, FEARG_1, NULL,
1590 ret_list_string, f_systemlist},
1591 {"tabpagebuflist", 0, 1, FEARG_1, NULL,
1592 ret_list_number, f_tabpagebuflist},
1593 {"tabpagenr", 0, 1, 0, NULL,
1594 ret_number, f_tabpagenr},
1595 {"tabpagewinnr", 1, 2, FEARG_1, NULL,
1596 ret_number, f_tabpagewinnr},
1597 {"tagfiles", 0, 0, 0, NULL,
1598 ret_list_string, f_tagfiles},
1599 {"taglist", 1, 2, FEARG_1, NULL,
1600 ret_list_dict_any, f_taglist},
1601 {"tan", 1, 1, FEARG_1, NULL,
1602 ret_float, FLOAT_FUNC(f_tan)},
1603 {"tanh", 1, 1, FEARG_1, NULL,
1604 ret_float, FLOAT_FUNC(f_tanh)},
1605 {"tempname", 0, 0, 0, NULL,
1606 ret_string, f_tempname},
1607 {"term_dumpdiff", 2, 3, FEARG_1, NULL,
1608 ret_number, TERM_FUNC(f_term_dumpdiff)},
1609 {"term_dumpload", 1, 2, FEARG_1, NULL,
1610 ret_number, TERM_FUNC(f_term_dumpload)},
1611 {"term_dumpwrite", 2, 3, FEARG_2, NULL,
1612 ret_void, TERM_FUNC(f_term_dumpwrite)},
1613 {"term_getaltscreen", 1, 1, FEARG_1, NULL,
1614 ret_number, TERM_FUNC(f_term_getaltscreen)},
1615 {"term_getansicolors", 1, 1, FEARG_1, NULL,
1616 ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001617#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001618 f_term_getansicolors
1619#else
1620 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001621#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001622 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001623 {"term_getattr", 2, 2, FEARG_1, NULL,
1624 ret_number, TERM_FUNC(f_term_getattr)},
1625 {"term_getcursor", 1, 1, FEARG_1, NULL,
1626 ret_list_any, TERM_FUNC(f_term_getcursor)},
1627 {"term_getjob", 1, 1, FEARG_1, NULL,
1628 ret_job, TERM_FUNC(f_term_getjob)},
1629 {"term_getline", 2, 2, FEARG_1, NULL,
1630 ret_string, TERM_FUNC(f_term_getline)},
1631 {"term_getscrolled", 1, 1, FEARG_1, NULL,
1632 ret_number, TERM_FUNC(f_term_getscrolled)},
1633 {"term_getsize", 1, 1, FEARG_1, NULL,
1634 ret_list_number, TERM_FUNC(f_term_getsize)},
1635 {"term_getstatus", 1, 1, FEARG_1, NULL,
1636 ret_string, TERM_FUNC(f_term_getstatus)},
1637 {"term_gettitle", 1, 1, FEARG_1, NULL,
1638 ret_string, TERM_FUNC(f_term_gettitle)},
1639 {"term_gettty", 1, 2, FEARG_1, NULL,
1640 ret_string, TERM_FUNC(f_term_gettty)},
1641 {"term_list", 0, 0, 0, NULL,
1642 ret_list_number, TERM_FUNC(f_term_list)},
1643 {"term_scrape", 2, 2, FEARG_1, NULL,
1644 ret_list_dict_any, TERM_FUNC(f_term_scrape)},
1645 {"term_sendkeys", 2, 2, FEARG_1, NULL,
1646 ret_void, TERM_FUNC(f_term_sendkeys)},
1647 {"term_setansicolors", 2, 2, FEARG_1, NULL,
1648 ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001649#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001650 f_term_setansicolors
1651#else
1652 NULL
1653#endif
1654 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001655 {"term_setapi", 2, 2, FEARG_1, NULL,
1656 ret_void, TERM_FUNC(f_term_setapi)},
1657 {"term_setkill", 2, 2, FEARG_1, NULL,
1658 ret_void, TERM_FUNC(f_term_setkill)},
1659 {"term_setrestore", 2, 2, FEARG_1, NULL,
1660 ret_void, TERM_FUNC(f_term_setrestore)},
1661 {"term_setsize", 3, 3, FEARG_1, NULL,
1662 ret_void, TERM_FUNC(f_term_setsize)},
1663 {"term_start", 1, 2, FEARG_1, NULL,
1664 ret_number, TERM_FUNC(f_term_start)},
1665 {"term_wait", 1, 2, FEARG_1, NULL,
1666 ret_void, TERM_FUNC(f_term_wait)},
1667 {"terminalprops", 0, 0, 0, NULL,
1668 ret_dict_string, f_terminalprops},
1669 {"test_alloc_fail", 3, 3, FEARG_1, NULL,
1670 ret_void, f_test_alloc_fail},
1671 {"test_autochdir", 0, 0, 0, NULL,
1672 ret_void, f_test_autochdir},
1673 {"test_feedinput", 1, 1, FEARG_1, NULL,
1674 ret_void, f_test_feedinput},
1675 {"test_garbagecollect_now", 0, 0, 0, NULL,
1676 ret_void, f_test_garbagecollect_now},
1677 {"test_garbagecollect_soon", 0, 0, 0, NULL,
1678 ret_void, f_test_garbagecollect_soon},
1679 {"test_getvalue", 1, 1, FEARG_1, NULL,
1680 ret_number, f_test_getvalue},
1681 {"test_ignore_error", 1, 1, FEARG_1, NULL,
1682 ret_void, f_test_ignore_error},
1683 {"test_null_blob", 0, 0, 0, NULL,
1684 ret_blob, f_test_null_blob},
1685 {"test_null_channel", 0, 0, 0, NULL,
1686 ret_channel, JOB_FUNC(f_test_null_channel)},
1687 {"test_null_dict", 0, 0, 0, NULL,
1688 ret_dict_any, f_test_null_dict},
1689 {"test_null_function", 0, 0, 0, NULL,
1690 ret_func_any, f_test_null_function},
1691 {"test_null_job", 0, 0, 0, NULL,
1692 ret_job, JOB_FUNC(f_test_null_job)},
1693 {"test_null_list", 0, 0, 0, NULL,
1694 ret_list_any, f_test_null_list},
1695 {"test_null_partial", 0, 0, 0, NULL,
1696 ret_func_any, f_test_null_partial},
1697 {"test_null_string", 0, 0, 0, NULL,
1698 ret_string, f_test_null_string},
1699 {"test_option_not_set", 1, 1, FEARG_1, NULL,
1700 ret_void, f_test_option_not_set},
1701 {"test_override", 2, 2, FEARG_2, NULL,
1702 ret_void, f_test_override},
1703 {"test_refcount", 1, 1, FEARG_1, NULL,
1704 ret_number, f_test_refcount},
1705 {"test_scrollbar", 3, 3, FEARG_2, NULL,
1706 ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +02001707#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +01001708 f_test_scrollbar
1709#else
1710 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +02001711#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001712 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001713 {"test_setmouse", 2, 2, 0, NULL,
1714 ret_void, f_test_setmouse},
1715 {"test_settime", 1, 1, FEARG_1, NULL,
1716 ret_void, f_test_settime},
1717 {"test_srand_seed", 0, 1, FEARG_1, NULL,
1718 ret_void, f_test_srand_seed},
1719 {"test_unknown", 0, 0, 0, NULL,
1720 ret_any, f_test_unknown},
1721 {"test_void", 0, 0, 0, NULL,
1722 ret_void, f_test_void},
1723 {"timer_info", 0, 1, FEARG_1, NULL,
1724 ret_list_dict_any, TIMER_FUNC(f_timer_info)},
1725 {"timer_pause", 2, 2, FEARG_1, NULL,
1726 ret_void, TIMER_FUNC(f_timer_pause)},
1727 {"timer_start", 2, 3, FEARG_1, NULL,
1728 ret_number, TIMER_FUNC(f_timer_start)},
1729 {"timer_stop", 1, 1, FEARG_1, NULL,
1730 ret_void, TIMER_FUNC(f_timer_stop)},
1731 {"timer_stopall", 0, 0, 0, NULL,
1732 ret_void, TIMER_FUNC(f_timer_stopall)},
1733 {"tolower", 1, 1, FEARG_1, NULL,
1734 ret_string, f_tolower},
1735 {"toupper", 1, 1, FEARG_1, NULL,
1736 ret_string, f_toupper},
1737 {"tr", 3, 3, FEARG_1, NULL,
1738 ret_string, f_tr},
1739 {"trim", 1, 3, FEARG_1, NULL,
1740 ret_string, f_trim},
1741 {"trunc", 1, 1, FEARG_1, NULL,
1742 ret_float, FLOAT_FUNC(f_trunc)},
1743 {"type", 1, 1, FEARG_1, NULL,
1744 ret_number, f_type},
1745 {"undofile", 1, 1, FEARG_1, NULL,
1746 ret_string, f_undofile},
1747 {"undotree", 0, 0, 0, NULL,
1748 ret_dict_any, f_undotree},
1749 {"uniq", 1, 3, FEARG_1, NULL,
1750 ret_list_any, f_uniq},
1751 {"values", 1, 1, FEARG_1, NULL,
1752 ret_list_any, f_values},
1753 {"virtcol", 1, 1, FEARG_1, NULL,
1754 ret_number, f_virtcol},
1755 {"visualmode", 0, 1, 0, NULL,
1756 ret_string, f_visualmode},
1757 {"wildmenumode", 0, 0, 0, NULL,
1758 ret_number, f_wildmenumode},
1759 {"win_execute", 2, 3, FEARG_2, NULL,
1760 ret_string, f_win_execute},
1761 {"win_findbuf", 1, 1, FEARG_1, NULL,
1762 ret_list_number, f_win_findbuf},
1763 {"win_getid", 0, 2, FEARG_1, NULL,
1764 ret_number, f_win_getid},
1765 {"win_gettype", 0, 1, FEARG_1, NULL,
1766 ret_string, f_win_gettype},
1767 {"win_gotoid", 1, 1, FEARG_1, NULL,
1768 ret_number, f_win_gotoid},
1769 {"win_id2tabwin", 1, 1, FEARG_1, NULL,
1770 ret_list_number, f_win_id2tabwin},
1771 {"win_id2win", 1, 1, FEARG_1, NULL,
1772 ret_number, f_win_id2win},
1773 {"win_screenpos", 1, 1, FEARG_1, NULL,
1774 ret_list_number, f_win_screenpos},
1775 {"win_splitmove", 2, 3, FEARG_1, NULL,
1776 ret_number, f_win_splitmove},
1777 {"winbufnr", 1, 1, FEARG_1, NULL,
1778 ret_number, f_winbufnr},
1779 {"wincol", 0, 0, 0, NULL,
1780 ret_number, f_wincol},
1781 {"windowsversion", 0, 0, 0, NULL,
1782 ret_string, f_windowsversion},
1783 {"winheight", 1, 1, FEARG_1, NULL,
1784 ret_number, f_winheight},
1785 {"winlayout", 0, 1, FEARG_1, NULL,
1786 ret_list_any, f_winlayout},
1787 {"winline", 0, 0, 0, NULL,
1788 ret_number, f_winline},
1789 {"winnr", 0, 1, FEARG_1, NULL,
1790 ret_number, f_winnr},
1791 {"winrestcmd", 0, 0, 0, NULL,
1792 ret_string, f_winrestcmd},
1793 {"winrestview", 1, 1, FEARG_1, NULL,
1794 ret_void, f_winrestview},
1795 {"winsaveview", 0, 0, 0, NULL,
Bram Moolenaar43b69b32021-01-07 20:23:33 +01001796 ret_dict_number, f_winsaveview},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001797 {"winwidth", 1, 1, FEARG_1, NULL,
1798 ret_number, f_winwidth},
1799 {"wordcount", 0, 0, 0, NULL,
1800 ret_dict_number, f_wordcount},
1801 {"writefile", 2, 3, FEARG_1, NULL,
1802 ret_number, f_writefile},
1803 {"xor", 2, 2, FEARG_1, NULL,
1804 ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001805};
1806
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001807/*
1808 * Function given to ExpandGeneric() to obtain the list of internal
1809 * or user defined function names.
1810 */
1811 char_u *
1812get_function_name(expand_T *xp, int idx)
1813{
1814 static int intidx = -1;
1815 char_u *name;
1816
1817 if (idx == 0)
1818 intidx = -1;
1819 if (intidx < 0)
1820 {
1821 name = get_user_func_name(xp, idx);
1822 if (name != NULL)
1823 return name;
1824 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001825 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001826 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001827 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001828 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001829 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001830 STRCAT(IObuff, ")");
1831 return IObuff;
1832 }
1833
1834 return NULL;
1835}
1836
1837/*
1838 * Function given to ExpandGeneric() to obtain the list of internal or
1839 * user defined variable or function names.
1840 */
1841 char_u *
1842get_expr_name(expand_T *xp, int idx)
1843{
1844 static int intidx = -1;
1845 char_u *name;
1846
1847 if (idx == 0)
1848 intidx = -1;
1849 if (intidx < 0)
1850 {
1851 name = get_function_name(xp, idx);
1852 if (name != NULL)
1853 return name;
1854 }
1855 return get_user_var_name(xp, ++intidx);
1856}
1857
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001858/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001859 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001860 * Return index, or -1 if not found or "implemented" is TRUE and the function
1861 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001862 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001863 static int
1864find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001865{
1866 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001867 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001868 int cmp;
1869 int x;
1870
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001871 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001872
1873 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001874 while (first <= last)
1875 {
1876 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001877 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001878 if (cmp < 0)
1879 last = x - 1;
1880 else if (cmp > 0)
1881 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001882 else if (implemented && global_functions[x].f_func == NULL)
1883 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001884 else
1885 return x;
1886 }
1887 return -1;
1888}
1889
Bram Moolenaar15c47602020-03-26 22:16:48 +01001890/*
1891 * Find internal function "name" in table "global_functions".
1892 * Return index, or -1 if not found or the function is not implemented.
1893 */
1894 int
1895find_internal_func(char_u *name)
1896{
1897 return find_internal_func_opt(name, TRUE);
1898}
1899
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001900 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001901has_internal_func(char_u *name)
1902{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001903 return find_internal_func_opt(name, TRUE) >= 0;
1904}
1905
1906 static int
1907has_internal_func_name(char_u *name)
1908{
1909 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001910}
1911
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001912 char *
1913internal_func_name(int idx)
1914{
1915 return global_functions[idx].f_name;
1916}
1917
Bram Moolenaar94738d82020-10-21 14:25:07 +02001918/*
1919 * Check the argument types for builting function "idx".
1920 * Uses the list of types on the type stack: "types".
1921 * Return FAIL and gives an error message when a type is wrong.
1922 */
1923 int
Bram Moolenaarca174532020-10-21 16:42:22 +02001924internal_func_check_arg_types(type_T **types, int idx, int argcount)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001925{
1926 argcheck_T *argchecks = global_functions[idx].f_argcheck;
1927 int i;
1928
1929 if (argchecks != NULL)
1930 {
1931 argcontext_T context;
1932
1933 context.arg_count = argcount;
Bram Moolenaarca174532020-10-21 16:42:22 +02001934 context.arg_types = types;
Bram Moolenaar94738d82020-10-21 14:25:07 +02001935 for (i = 0; i < argcount; ++i)
1936 if (argchecks[i] != NULL)
1937 {
1938 context.arg_idx = i;
Bram Moolenaarca174532020-10-21 16:42:22 +02001939 if (argchecks[i](types[i], &context) == FAIL)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001940 return FAIL;
1941 }
1942 }
1943 return OK;
1944}
1945
Bram Moolenaara1224cb2020-10-22 12:31:49 +02001946/*
1947 * Call the "f_retfunc" function to obtain the return type of function "idx".
1948 * "argtypes" is the list of argument types or NULL when there are no
1949 * arguments.
1950 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001951 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001952internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001953{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001954 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001955}
1956
1957/*
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01001958 * Return TRUE if "idx" is for the map() function.
1959 */
1960 int
1961internal_func_is_map(int idx)
1962{
1963 return global_functions[idx].f_func == f_map;
1964}
1965
1966/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001967 * Check the argument count to use for internal function "idx".
Bram Moolenaar389df252020-07-09 21:20:47 +02001968 * Returns -1 for failure, 0 if no method base accepted, 1 if method base is
1969 * first argument, 2 if method base is second argument, etc. 9 if method base
1970 * is last argument.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001971 */
1972 int
1973check_internal_func(int idx, int argcount)
1974{
1975 int res;
1976 char *name;
1977
1978 if (argcount < global_functions[idx].f_min_argc)
1979 res = FCERR_TOOFEW;
1980 else if (argcount > global_functions[idx].f_max_argc)
1981 res = FCERR_TOOMANY;
1982 else
Bram Moolenaar389df252020-07-09 21:20:47 +02001983 return global_functions[idx].f_argtype;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001984
1985 name = internal_func_name(idx);
1986 if (res == FCERR_TOOMANY)
1987 semsg(_(e_toomanyarg), name);
1988 else
1989 semsg(_(e_toofewarg), name);
Bram Moolenaar389df252020-07-09 21:20:47 +02001990 return -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001991}
1992
Bram Moolenaarac92e252019-08-03 21:58:38 +02001993 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001994call_internal_func(
1995 char_u *name,
1996 int argcount,
1997 typval_T *argvars,
1998 typval_T *rettv)
1999{
2000 int i;
2001
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002002 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002003 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01002004 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002005 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002006 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002007 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002008 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002009 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002010 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01002011 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002012}
2013
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002014 void
2015call_internal_func_by_idx(
2016 int idx,
2017 typval_T *argvars,
2018 typval_T *rettv)
2019{
2020 global_functions[idx].f_func(argvars, rettv);
2021}
2022
Bram Moolenaarac92e252019-08-03 21:58:38 +02002023/*
2024 * Invoke a method for base->method().
2025 */
2026 int
2027call_internal_method(
2028 char_u *name,
2029 int argcount,
2030 typval_T *argvars,
2031 typval_T *rettv,
2032 typval_T *basetv)
2033{
2034 int i;
2035 int fi;
2036 typval_T argv[MAX_FUNC_ARGS + 1];
2037
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002038 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02002039 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01002040 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02002041 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01002042 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002043 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002044 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002045 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002046 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002047
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002048 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002049 {
2050 // base value goes last
2051 for (i = 0; i < argcount; ++i)
2052 argv[i] = argvars[i];
2053 argv[argcount] = *basetv;
2054 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002055 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002056 {
2057 // base value goes second
2058 argv[0] = argvars[0];
2059 argv[1] = *basetv;
2060 for (i = 1; i < argcount; ++i)
2061 argv[i + 1] = argvars[i];
2062 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02002063 else if (global_functions[fi].f_argtype == FEARG_3)
2064 {
2065 // base value goes third
2066 argv[0] = argvars[0];
2067 argv[1] = argvars[1];
2068 argv[2] = *basetv;
2069 for (i = 2; i < argcount; ++i)
2070 argv[i + 1] = argvars[i];
2071 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02002072 else if (global_functions[fi].f_argtype == FEARG_4)
2073 {
2074 // base value goes fourth
2075 argv[0] = argvars[0];
2076 argv[1] = argvars[1];
2077 argv[2] = argvars[2];
2078 argv[3] = *basetv;
2079 for (i = 3; i < argcount; ++i)
2080 argv[i + 1] = argvars[i];
2081 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02002082 else
2083 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002084 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02002085 argv[0] = *basetv;
2086 for (i = 0; i < argcount; ++i)
2087 argv[i + 1] = argvars[i];
2088 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02002089 argv[argcount + 1].v_type = VAR_UNKNOWN;
2090
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002091 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01002092 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002093}
2094
2095/*
2096 * Return TRUE for a non-zero Number and a non-empty String.
2097 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02002098 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002099non_zero_arg(typval_T *argvars)
2100{
2101 return ((argvars[0].v_type == VAR_NUMBER
2102 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002103 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002104 && argvars[0].vval.v_number == VVAL_TRUE)
2105 || (argvars[0].v_type == VAR_STRING
2106 && argvars[0].vval.v_string != NULL
2107 && *argvars[0].vval.v_string != NUL));
2108}
2109
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002110#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002111/*
2112 * Get the float value of "argvars[0]" into "f".
2113 * Returns FAIL when the argument is not a Number or Float.
2114 */
2115 static int
2116get_float_arg(typval_T *argvars, float_T *f)
2117{
2118 if (argvars[0].v_type == VAR_FLOAT)
2119 {
2120 *f = argvars[0].vval.v_float;
2121 return OK;
2122 }
2123 if (argvars[0].v_type == VAR_NUMBER)
2124 {
2125 *f = (float_T)argvars[0].vval.v_number;
2126 return OK;
2127 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002128 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002129 return FAIL;
2130}
2131
2132/*
2133 * "abs(expr)" function
2134 */
2135 static void
2136f_abs(typval_T *argvars, typval_T *rettv)
2137{
2138 if (argvars[0].v_type == VAR_FLOAT)
2139 {
2140 rettv->v_type = VAR_FLOAT;
2141 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
2142 }
2143 else
2144 {
2145 varnumber_T n;
2146 int error = FALSE;
2147
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002148 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002149 if (error)
2150 rettv->vval.v_number = -1;
2151 else if (n > 0)
2152 rettv->vval.v_number = n;
2153 else
2154 rettv->vval.v_number = -n;
2155 }
2156}
2157
2158/*
2159 * "acos()" function
2160 */
2161 static void
2162f_acos(typval_T *argvars, typval_T *rettv)
2163{
2164 float_T f = 0.0;
2165
2166 rettv->v_type = VAR_FLOAT;
2167 if (get_float_arg(argvars, &f) == OK)
2168 rettv->vval.v_float = acos(f);
2169 else
2170 rettv->vval.v_float = 0.0;
2171}
2172#endif
2173
2174/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002175 * "and(expr, expr)" function
2176 */
2177 static void
2178f_and(typval_T *argvars, typval_T *rettv)
2179{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002180 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
2181 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02002182}
2183
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002184#ifdef FEAT_FLOAT
2185/*
2186 * "asin()" function
2187 */
2188 static void
2189f_asin(typval_T *argvars, typval_T *rettv)
2190{
2191 float_T f = 0.0;
2192
2193 rettv->v_type = VAR_FLOAT;
2194 if (get_float_arg(argvars, &f) == OK)
2195 rettv->vval.v_float = asin(f);
2196 else
2197 rettv->vval.v_float = 0.0;
2198}
2199
2200/*
2201 * "atan()" function
2202 */
2203 static void
2204f_atan(typval_T *argvars, typval_T *rettv)
2205{
2206 float_T f = 0.0;
2207
2208 rettv->v_type = VAR_FLOAT;
2209 if (get_float_arg(argvars, &f) == OK)
2210 rettv->vval.v_float = atan(f);
2211 else
2212 rettv->vval.v_float = 0.0;
2213}
2214
2215/*
2216 * "atan2()" function
2217 */
2218 static void
2219f_atan2(typval_T *argvars, typval_T *rettv)
2220{
2221 float_T fx = 0.0, fy = 0.0;
2222
2223 rettv->v_type = VAR_FLOAT;
2224 if (get_float_arg(argvars, &fx) == OK
2225 && get_float_arg(&argvars[1], &fy) == OK)
2226 rettv->vval.v_float = atan2(fx, fy);
2227 else
2228 rettv->vval.v_float = 0.0;
2229}
2230#endif
2231
2232/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01002233 * "balloon_show()" function
2234 */
2235#ifdef FEAT_BEVAL
2236 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002237f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
2238{
2239 rettv->v_type = VAR_STRING;
2240 if (balloonEval != NULL)
2241 {
2242 if (balloonEval->msg == NULL)
2243 rettv->vval.v_string = NULL;
2244 else
2245 rettv->vval.v_string = vim_strsave(balloonEval->msg);
2246 }
2247}
2248
2249 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01002250f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
2251{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01002252 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002253 {
2254 if (argvars[0].v_type == VAR_LIST
2255# ifdef FEAT_GUI
2256 && !gui.in_use
2257# endif
2258 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002259 {
2260 list_T *l = argvars[0].vval.v_list;
2261
2262 // empty list removes the balloon
2263 post_balloon(balloonEval, NULL,
2264 l == NULL || l->lv_len == 0 ? NULL : l);
2265 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002266 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002267 {
2268 char_u *mesg = tv_get_string_chk(&argvars[0]);
2269
2270 if (mesg != NULL)
2271 // empty string removes the balloon
2272 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
2273 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002274 }
2275}
2276
Bram Moolenaar669a8282017-11-19 20:13:05 +01002277# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002278 static void
2279f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
2280{
2281 if (rettv_list_alloc(rettv) == OK)
2282 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002283 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002284
2285 if (msg != NULL)
2286 {
2287 pumitem_T *array;
2288 int size = split_message(msg, &array);
2289 int i;
2290
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002291 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01002292 for (i = 1; i < size - 1; ++i)
2293 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01002294 while (size > 0)
2295 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002296 vim_free(array);
2297 }
2298 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01002299}
Bram Moolenaar669a8282017-11-19 20:13:05 +01002300# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01002301#endif
2302
2303/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002304 * Get the buffer from "arg" and give an error and return NULL if it is not
2305 * valid.
2306 */
Bram Moolenaara3347722019-05-11 21:14:24 +02002307 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002308get_buf_arg(typval_T *arg)
2309{
2310 buf_T *buf;
2311
2312 ++emsg_off;
2313 buf = tv_get_buf(arg, FALSE);
2314 --emsg_off;
2315 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002316 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002317 return buf;
2318}
2319
2320/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002321 * "byte2line(byte)" function
2322 */
2323 static void
2324f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
2325{
2326#ifndef FEAT_BYTEOFF
2327 rettv->vval.v_number = -1;
2328#else
2329 long boff = 0;
2330
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002331 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002332 if (boff < 0)
2333 rettv->vval.v_number = -1;
2334 else
2335 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
2336 (linenr_T)0, &boff);
2337#endif
2338}
2339
2340 static void
2341byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
2342{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002343 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002344 char_u *str;
2345 varnumber_T idx;
2346
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002347 str = tv_get_string_chk(&argvars[0]);
2348 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002349 rettv->vval.v_number = -1;
2350 if (str == NULL || idx < 0)
2351 return;
2352
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002353 t = str;
2354 for ( ; idx > 0; idx--)
2355 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002356 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002357 return;
2358 if (enc_utf8 && comp)
2359 t += utf_ptr2len(t);
2360 else
2361 t += (*mb_ptr2len)(t);
2362 }
2363 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002364}
2365
2366/*
2367 * "byteidx()" function
2368 */
2369 static void
2370f_byteidx(typval_T *argvars, typval_T *rettv)
2371{
2372 byteidx(argvars, rettv, FALSE);
2373}
2374
2375/*
2376 * "byteidxcomp()" function
2377 */
2378 static void
2379f_byteidxcomp(typval_T *argvars, typval_T *rettv)
2380{
2381 byteidx(argvars, rettv, TRUE);
2382}
2383
2384/*
2385 * "call(func, arglist [, dict])" function
2386 */
2387 static void
2388f_call(typval_T *argvars, typval_T *rettv)
2389{
2390 char_u *func;
2391 partial_T *partial = NULL;
2392 dict_T *selfdict = NULL;
2393
2394 if (argvars[1].v_type != VAR_LIST)
2395 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002396 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002397 return;
2398 }
2399 if (argvars[1].vval.v_list == NULL)
2400 return;
2401
2402 if (argvars[0].v_type == VAR_FUNC)
2403 func = argvars[0].vval.v_string;
2404 else if (argvars[0].v_type == VAR_PARTIAL)
2405 {
2406 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002407 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002408 }
2409 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002410 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002411 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002412 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002413
2414 if (argvars[2].v_type != VAR_UNKNOWN)
2415 {
2416 if (argvars[2].v_type != VAR_DICT)
2417 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002418 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002419 return;
2420 }
2421 selfdict = argvars[2].vval.v_dict;
2422 }
2423
2424 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
2425}
2426
2427#ifdef FEAT_FLOAT
2428/*
2429 * "ceil({float})" function
2430 */
2431 static void
2432f_ceil(typval_T *argvars, typval_T *rettv)
2433{
2434 float_T f = 0.0;
2435
2436 rettv->v_type = VAR_FLOAT;
2437 if (get_float_arg(argvars, &f) == OK)
2438 rettv->vval.v_float = ceil(f);
2439 else
2440 rettv->vval.v_float = 0.0;
2441}
2442#endif
2443
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002444/*
2445 * "changenr()" function
2446 */
2447 static void
2448f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
2449{
2450 rettv->vval.v_number = curbuf->b_u_seq_cur;
2451}
2452
2453/*
2454 * "char2nr(string)" function
2455 */
2456 static void
2457f_char2nr(typval_T *argvars, typval_T *rettv)
2458{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002459 if (has_mbyte)
2460 {
2461 int utf8 = 0;
2462
2463 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar24f77502020-09-04 19:50:57 +02002464 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002465
2466 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01002467 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002468 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002469 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002470 }
2471 else
Bram Moolenaar13505972019-01-24 15:04:48 +01002472 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002473}
2474
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002475/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002476 * Get the current cursor column and store it in 'rettv'. If 'charcol' is TRUE,
2477 * returns the character index of the column. Otherwise, returns the byte index
2478 * of the column.
2479 */
2480 static void
2481get_col(typval_T *argvars, typval_T *rettv, int charcol)
2482{
2483 colnr_T col = 0;
2484 pos_T *fp;
2485 int fnum = curbuf->b_fnum;
2486
2487 fp = var2fpos(&argvars[0], FALSE, &fnum, charcol);
2488 if (fp != NULL && fnum == curbuf->b_fnum)
2489 {
2490 if (fp->col == MAXCOL)
2491 {
2492 // '> can be MAXCOL, get the length of the line then
2493 if (fp->lnum <= curbuf->b_ml.ml_line_count)
2494 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
2495 else
2496 col = MAXCOL;
2497 }
2498 else
2499 {
2500 col = fp->col + 1;
2501 // col(".") when the cursor is on the NUL at the end of the line
2502 // because of "coladd" can be seen as an extra column.
2503 if (virtual_active() && fp == &curwin->w_cursor)
2504 {
2505 char_u *p = ml_get_cursor();
2506
2507 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
2508 curwin->w_virtcol - curwin->w_cursor.coladd))
2509 {
2510 int l;
2511
2512 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
2513 col += l;
2514 }
2515 }
2516 }
2517 }
2518 rettv->vval.v_number = col;
2519}
2520
2521/*
2522 * "charcol()" function
2523 */
2524 static void
2525f_charcol(typval_T *argvars, typval_T *rettv)
2526{
2527 get_col(argvars, rettv, TRUE);
2528}
2529
2530/*
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002531 * "charidx()" function
2532 */
2533 static void
2534f_charidx(typval_T *argvars, typval_T *rettv)
2535{
2536 char_u *str;
2537 varnumber_T idx;
2538 int countcc = FALSE;
2539 char_u *p;
2540 int len;
2541 int (*ptr2len)(char_u *);
2542
2543 rettv->vval.v_number = -1;
2544
2545 if (argvars[0].v_type != VAR_STRING || argvars[1].v_type != VAR_NUMBER
2546 || (argvars[2].v_type != VAR_UNKNOWN
2547 && argvars[2].v_type != VAR_NUMBER))
2548 {
2549 emsg(_(e_invarg));
2550 return;
2551 }
2552
2553 str = tv_get_string_chk(&argvars[0]);
2554 idx = tv_get_number_chk(&argvars[1], NULL);
2555 if (str == NULL || idx < 0)
2556 return;
2557
2558 if (argvars[2].v_type != VAR_UNKNOWN)
2559 countcc = (int)tv_get_bool(&argvars[2]);
2560 if (countcc < 0 || countcc > 1)
2561 {
2562 semsg(_(e_using_number_as_bool_nr), countcc);
2563 return;
2564 }
2565
2566 if (enc_utf8 && countcc)
2567 ptr2len = utf_ptr2len;
2568 else
2569 ptr2len = mb_ptr2len;
2570
2571 for (p = str, len = 0; p <= str + idx; len++)
2572 {
2573 if (*p == NUL)
2574 return;
2575 p += ptr2len(p);
2576 }
2577
2578 rettv->vval.v_number = len > 0 ? len - 1 : 0;
2579}
2580
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02002581 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01002582get_optional_window(typval_T *argvars, int idx)
2583{
2584 win_T *win = curwin;
2585
2586 if (argvars[idx].v_type != VAR_UNKNOWN)
2587 {
2588 win = find_win_by_nr_or_id(&argvars[idx]);
2589 if (win == NULL)
2590 {
2591 emsg(_(e_invalwindow));
2592 return NULL;
2593 }
2594 }
2595 return win;
2596}
2597
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002598/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002599 * "col(string)" function
2600 */
2601 static void
2602f_col(typval_T *argvars, typval_T *rettv)
2603{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002604 get_col(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002605}
2606
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002607/*
2608 * "confirm(message, buttons[, default [, type]])" function
2609 */
2610 static void
2611f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2612{
2613#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2614 char_u *message;
2615 char_u *buttons = NULL;
2616 char_u buf[NUMBUFLEN];
2617 char_u buf2[NUMBUFLEN];
2618 int def = 1;
2619 int type = VIM_GENERIC;
2620 char_u *typestr;
2621 int error = FALSE;
2622
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002623 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002624 if (message == NULL)
2625 error = TRUE;
2626 if (argvars[1].v_type != VAR_UNKNOWN)
2627 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002628 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002629 if (buttons == NULL)
2630 error = TRUE;
2631 if (argvars[2].v_type != VAR_UNKNOWN)
2632 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002633 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002634 if (argvars[3].v_type != VAR_UNKNOWN)
2635 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002636 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002637 if (typestr == NULL)
2638 error = TRUE;
2639 else
2640 {
2641 switch (TOUPPER_ASC(*typestr))
2642 {
2643 case 'E': type = VIM_ERROR; break;
2644 case 'Q': type = VIM_QUESTION; break;
2645 case 'I': type = VIM_INFO; break;
2646 case 'W': type = VIM_WARNING; break;
2647 case 'G': type = VIM_GENERIC; break;
2648 }
2649 }
2650 }
2651 }
2652 }
2653
2654 if (buttons == NULL || *buttons == NUL)
2655 buttons = (char_u *)_("&Ok");
2656
2657 if (!error)
2658 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
2659 def, NULL, FALSE);
2660#endif
2661}
2662
2663/*
2664 * "copy()" function
2665 */
2666 static void
2667f_copy(typval_T *argvars, typval_T *rettv)
2668{
2669 item_copy(&argvars[0], rettv, FALSE, 0);
2670}
2671
2672#ifdef FEAT_FLOAT
2673/*
2674 * "cos()" function
2675 */
2676 static void
2677f_cos(typval_T *argvars, typval_T *rettv)
2678{
2679 float_T f = 0.0;
2680
2681 rettv->v_type = VAR_FLOAT;
2682 if (get_float_arg(argvars, &f) == OK)
2683 rettv->vval.v_float = cos(f);
2684 else
2685 rettv->vval.v_float = 0.0;
2686}
2687
2688/*
2689 * "cosh()" function
2690 */
2691 static void
2692f_cosh(typval_T *argvars, typval_T *rettv)
2693{
2694 float_T f = 0.0;
2695
2696 rettv->v_type = VAR_FLOAT;
2697 if (get_float_arg(argvars, &f) == OK)
2698 rettv->vval.v_float = cosh(f);
2699 else
2700 rettv->vval.v_float = 0.0;
2701}
2702#endif
2703
2704/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002705 * Set the cursor position.
2706 * If 'charcol' is TRUE, then use the column number as a character offet.
2707 * Otherwise use the column number as a byte offset.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002708 */
2709 static void
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002710set_cursorpos(typval_T *argvars, typval_T *rettv, int charcol)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002711{
2712 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002713 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002714 int set_curswant = TRUE;
2715
2716 rettv->vval.v_number = -1;
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002717 if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002718 {
2719 pos_T pos;
2720 colnr_T curswant = -1;
2721
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002722 if (list2fpos(argvars, &pos, NULL, &curswant, charcol) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002723 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002724 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002725 return;
2726 }
2727 line = pos.lnum;
2728 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002729 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002730 if (curswant >= 0)
2731 {
2732 curwin->w_curswant = curswant - 1;
2733 set_curswant = FALSE;
2734 }
2735 }
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002736 else if ((argvars[0].v_type == VAR_NUMBER ||
2737 argvars[0].v_type == VAR_STRING)
2738 && argvars[1].v_type == VAR_NUMBER)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002739 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002740 line = tv_get_lnum(argvars);
Bram Moolenaar9a963372020-12-21 21:58:46 +01002741 if (line < 0)
2742 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002743 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002744 if (charcol)
2745 col = buf_charidx_to_byteidx(curbuf, line, col);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002746 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002747 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002748 }
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002749 else
2750 {
2751 emsg(_(e_invarg));
2752 return;
2753 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002754 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002755 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002756 if (line > 0)
2757 curwin->w_cursor.lnum = line;
2758 if (col > 0)
2759 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002760 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002761
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002762 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002763 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002764 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002765 if (has_mbyte)
2766 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002767
2768 curwin->w_set_curswant = set_curswant;
2769 rettv->vval.v_number = 0;
2770}
2771
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002772/*
2773 * "cursor(lnum, col)" function, or
2774 * "cursor(list)"
2775 *
2776 * Moves the cursor to the specified line and column.
2777 * Returns 0 when the position could be set, -1 otherwise.
2778 */
2779 static void
2780f_cursor(typval_T *argvars, typval_T *rettv)
2781{
2782 set_cursorpos(argvars, rettv, FALSE);
2783}
2784
Bram Moolenaar4f974752019-02-17 17:44:42 +01002785#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002786/*
2787 * "debugbreak()" function
2788 */
2789 static void
2790f_debugbreak(typval_T *argvars, typval_T *rettv)
2791{
2792 int pid;
2793
2794 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002795 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002796 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002797 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002798 else
2799 {
2800 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
2801
2802 if (hProcess != NULL)
2803 {
2804 DebugBreakProcess(hProcess);
2805 CloseHandle(hProcess);
2806 rettv->vval.v_number = OK;
2807 }
2808 }
2809}
2810#endif
2811
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002812/*
2813 * "deepcopy()" function
2814 */
2815 static void
2816f_deepcopy(typval_T *argvars, typval_T *rettv)
2817{
2818 int noref = 0;
2819 int copyID;
2820
2821 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar44b4a242020-09-05 17:18:28 +02002822 noref = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002823 if (noref < 0 || noref > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02002824 semsg(_(e_using_number_as_bool_nr), noref);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002825 else
2826 {
2827 copyID = get_copyID();
2828 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
2829 }
2830}
2831
2832/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002833 * "did_filetype()" function
2834 */
2835 static void
2836f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2837{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002838 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002839}
2840
2841/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01002842 * "echoraw({expr})" function
2843 */
2844 static void
2845f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
2846{
2847 char_u *str = tv_get_string_chk(&argvars[0]);
2848
2849 if (str != NULL && *str != NUL)
2850 {
2851 out_str(str);
2852 out_flush();
2853 }
2854}
2855
2856/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002857 * "empty({expr})" function
2858 */
2859 static void
2860f_empty(typval_T *argvars, typval_T *rettv)
2861{
2862 int n = FALSE;
2863
2864 switch (argvars[0].v_type)
2865 {
2866 case VAR_STRING:
2867 case VAR_FUNC:
2868 n = argvars[0].vval.v_string == NULL
2869 || *argvars[0].vval.v_string == NUL;
2870 break;
2871 case VAR_PARTIAL:
2872 n = FALSE;
2873 break;
2874 case VAR_NUMBER:
2875 n = argvars[0].vval.v_number == 0;
2876 break;
2877 case VAR_FLOAT:
2878#ifdef FEAT_FLOAT
2879 n = argvars[0].vval.v_float == 0.0;
2880 break;
2881#endif
2882 case VAR_LIST:
2883 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002884 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002885 break;
2886 case VAR_DICT:
2887 n = argvars[0].vval.v_dict == NULL
2888 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2889 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002890 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002891 case VAR_SPECIAL:
2892 n = argvars[0].vval.v_number != VVAL_TRUE;
2893 break;
2894
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002895 case VAR_BLOB:
2896 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002897 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2898 break;
2899
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002900 case VAR_JOB:
2901#ifdef FEAT_JOB_CHANNEL
2902 n = argvars[0].vval.v_job == NULL
2903 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2904 break;
2905#endif
2906 case VAR_CHANNEL:
2907#ifdef FEAT_JOB_CHANNEL
2908 n = argvars[0].vval.v_channel == NULL
2909 || !channel_is_open(argvars[0].vval.v_channel);
2910 break;
2911#endif
2912 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02002913 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002914 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002915 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002916 n = TRUE;
2917 break;
2918 }
2919
2920 rettv->vval.v_number = n;
2921}
2922
2923/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002924 * "environ()" function
2925 */
2926 static void
2927f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2928{
2929#if !defined(AMIGA)
2930 int i = 0;
2931 char_u *entry, *value;
2932# ifdef MSWIN
2933 extern wchar_t **_wenviron;
2934# else
2935 extern char **environ;
2936# endif
2937
2938 if (rettv_dict_alloc(rettv) != OK)
2939 return;
2940
2941# ifdef MSWIN
2942 if (*_wenviron == NULL)
2943 return;
2944# else
2945 if (*environ == NULL)
2946 return;
2947# endif
2948
2949 for (i = 0; ; ++i)
2950 {
2951# ifdef MSWIN
2952 short_u *p;
2953
2954 if ((p = (short_u *)_wenviron[i]) == NULL)
2955 return;
2956 entry = utf16_to_enc(p, NULL);
2957# else
2958 if ((entry = (char_u *)environ[i]) == NULL)
2959 return;
2960 entry = vim_strsave(entry);
2961# endif
2962 if (entry == NULL) // out of memory
2963 return;
2964 if ((value = vim_strchr(entry, '=')) == NULL)
2965 {
2966 vim_free(entry);
2967 continue;
2968 }
2969 *value++ = NUL;
2970 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2971 vim_free(entry);
2972 }
2973#endif
2974}
2975
2976/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002977 * "escape({string}, {chars})" function
2978 */
2979 static void
2980f_escape(typval_T *argvars, typval_T *rettv)
2981{
2982 char_u buf[NUMBUFLEN];
2983
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002984 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2985 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002986 rettv->v_type = VAR_STRING;
2987}
2988
2989/*
2990 * "eval()" function
2991 */
2992 static void
2993f_eval(typval_T *argvars, typval_T *rettv)
2994{
2995 char_u *s, *p;
2996
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002997 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002998 if (s != NULL)
2999 s = skipwhite(s);
3000
3001 p = s;
Bram Moolenaar5409f5d2020-06-24 18:37:35 +02003002 if (s == NULL || eval1(&s, rettv, &EVALARG_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003003 {
3004 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003005 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003006 need_clr_eos = FALSE;
3007 rettv->v_type = VAR_NUMBER;
3008 rettv->vval.v_number = 0;
3009 }
3010 else if (*s != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02003011 semsg(_(e_trailing_arg), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003012}
3013
3014/*
3015 * "eventhandler()" function
3016 */
3017 static void
3018f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
3019{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003020 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003021}
3022
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003023static garray_T redir_execute_ga;
3024
3025/*
3026 * Append "value[value_len]" to the execute() output.
3027 */
3028 void
3029execute_redir_str(char_u *value, int value_len)
3030{
3031 int len;
3032
3033 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003034 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003035 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003036 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003037 if (ga_grow(&redir_execute_ga, len) == OK)
3038 {
3039 mch_memmove((char *)redir_execute_ga.ga_data
3040 + redir_execute_ga.ga_len, value, len);
3041 redir_execute_ga.ga_len += len;
3042 }
3043}
3044
3045/*
3046 * Get next line from a list.
3047 * Called by do_cmdline() to get the next line.
3048 * Returns allocated string, or NULL for end of function.
3049 */
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003050 static char_u *
3051get_list_line(
3052 int c UNUSED,
3053 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02003054 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +02003055 getline_opt_T options UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003056{
3057 listitem_T **p = (listitem_T **)cookie;
3058 listitem_T *item = *p;
3059 char_u buf[NUMBUFLEN];
3060 char_u *s;
3061
3062 if (item == NULL)
3063 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003064 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003065 *p = item->li_next;
3066 return s == NULL ? NULL : vim_strsave(s);
3067}
3068
3069/*
3070 * "execute()" function
3071 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02003072 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003073execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003074{
3075 char_u *cmd = NULL;
3076 list_T *list = NULL;
3077 int save_msg_silent = msg_silent;
3078 int save_emsg_silent = emsg_silent;
3079 int save_emsg_noredir = emsg_noredir;
3080 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01003081 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003082 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003083 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003084 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003085
3086 rettv->vval.v_string = NULL;
3087 rettv->v_type = VAR_STRING;
3088
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003089 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003090 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003091 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003092 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003093 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003094 return;
3095 ++list->lv_refcount;
3096 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01003097 else if (argvars[arg_off].v_type == VAR_JOB
3098 || argvars[arg_off].v_type == VAR_CHANNEL)
3099 {
3100 emsg(_(e_inval_string));
3101 return;
3102 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003103 else
3104 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003105 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003106 if (cmd == NULL)
3107 return;
3108 }
3109
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003110 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003111 {
3112 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003113 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003114
3115 if (s == NULL)
3116 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003117 if (*s == NUL)
3118 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003119 if (STRNCMP(s, "silent", 6) == 0)
3120 ++msg_silent;
3121 if (STRCMP(s, "silent!") == 0)
3122 {
3123 emsg_silent = TRUE;
3124 emsg_noredir = TRUE;
3125 }
3126 }
3127 else
3128 ++msg_silent;
3129
3130 if (redir_execute)
3131 save_ga = redir_execute_ga;
3132 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
3133 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01003134 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003135 if (!echo_output)
3136 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003137
3138 if (cmd != NULL)
3139 do_cmdline_cmd(cmd);
3140 else
3141 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003142 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003143
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003144 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003145 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003146 do_cmdline(NULL, get_list_line, (void *)&item,
3147 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
3148 --list->lv_refcount;
3149 }
3150
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003151 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01003152 if (ga_grow(&redir_execute_ga, 1) == OK)
3153 {
3154 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
3155 rettv->vval.v_string = redir_execute_ga.ga_data;
3156 }
3157 else
3158 {
3159 ga_clear(&redir_execute_ga);
3160 rettv->vval.v_string = NULL;
3161 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003162 msg_silent = save_msg_silent;
3163 emsg_silent = save_emsg_silent;
3164 emsg_noredir = save_emsg_noredir;
3165
3166 redir_execute = save_redir_execute;
3167 if (redir_execute)
3168 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01003169 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003170
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003171 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003172 if (echo_output)
3173 // When not working silently: put it in column zero. A following
3174 // "echon" will overwrite the message, unavoidably.
3175 msg_col = 0;
3176 else
3177 // When working silently: Put it back where it was, since nothing
3178 // should have been written.
3179 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003180}
3181
3182/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003183 * "execute()" function
3184 */
3185 static void
3186f_execute(typval_T *argvars, typval_T *rettv)
3187{
3188 execute_common(argvars, rettv, 0);
3189}
3190
3191/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003192 * "exists()" function
3193 */
3194 static void
3195f_exists(typval_T *argvars, typval_T *rettv)
3196{
3197 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003198 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003199
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003200 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003201 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003202 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003203 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003204 if (mch_getenv(p + 1) != NULL)
3205 n = TRUE;
3206 else
3207 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003208 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003209 p = expand_env_save(p);
3210 if (p != NULL && *p != '$')
3211 n = TRUE;
3212 vim_free(p);
3213 }
3214 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003215 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003216 {
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +02003217 n = (eval_option(&p, NULL, TRUE) == OK);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003218 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003219 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003220 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003221 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003222 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02003223 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003224 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01003225 else if (*p == '?') // internal function only
3226 {
3227 n = has_internal_func_name(p + 1);
3228 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003229 else if (*p == ':')
3230 {
3231 n = cmd_exists(p + 1);
3232 }
3233 else if (*p == '#')
3234 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003235 if (p[1] == '#')
3236 n = autocmd_supported(p + 2);
3237 else
3238 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003239 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003240 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003241 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01003242 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003243 }
3244
3245 rettv->vval.v_number = n;
3246}
3247
3248#ifdef FEAT_FLOAT
3249/*
3250 * "exp()" function
3251 */
3252 static void
3253f_exp(typval_T *argvars, typval_T *rettv)
3254{
3255 float_T f = 0.0;
3256
3257 rettv->v_type = VAR_FLOAT;
3258 if (get_float_arg(argvars, &f) == OK)
3259 rettv->vval.v_float = exp(f);
3260 else
3261 rettv->vval.v_float = 0.0;
3262}
3263#endif
3264
3265/*
3266 * "expand()" function
3267 */
3268 static void
3269f_expand(typval_T *argvars, typval_T *rettv)
3270{
3271 char_u *s;
3272 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003273 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003274 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
3275 expand_T xpc;
3276 int error = FALSE;
3277 char_u *result;
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003278#ifdef BACKSLASH_IN_FILENAME
3279 char_u *p_csl_save = p_csl;
3280
3281 // avoid using 'completeslash' here
3282 p_csl = empty_option;
3283#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003284
3285 rettv->v_type = VAR_STRING;
3286 if (argvars[1].v_type != VAR_UNKNOWN
3287 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003288 && tv_get_bool_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003289 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02003290 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003291
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003292 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003293 if (*s == '%' || *s == '#' || *s == '<')
3294 {
3295 ++emsg_off;
3296 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
3297 --emsg_off;
3298 if (rettv->v_type == VAR_LIST)
3299 {
3300 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
3301 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02003302 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003303 }
3304 else
3305 rettv->vval.v_string = result;
3306 }
3307 else
3308 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003309 // When the optional second argument is non-zero, don't remove matches
3310 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003311 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003312 && tv_get_bool_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003313 options |= WILD_KEEP_ALL;
3314 if (!error)
3315 {
3316 ExpandInit(&xpc);
3317 xpc.xp_context = EXPAND_FILES;
3318 if (p_wic)
3319 options += WILD_ICASE;
3320 if (rettv->v_type == VAR_STRING)
3321 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
3322 options, WILD_ALL);
3323 else if (rettv_list_alloc(rettv) != FAIL)
3324 {
3325 int i;
3326
3327 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
3328 for (i = 0; i < xpc.xp_numfiles; i++)
3329 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
3330 ExpandCleanup(&xpc);
3331 }
3332 }
3333 else
3334 rettv->vval.v_string = NULL;
3335 }
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003336#ifdef BACKSLASH_IN_FILENAME
3337 p_csl = p_csl_save;
3338#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003339}
3340
3341/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02003342 * "expandcmd()" function
3343 * Expand all the special characters in a command string.
3344 */
3345 static void
3346f_expandcmd(typval_T *argvars, typval_T *rettv)
3347{
3348 exarg_T eap;
3349 char_u *cmdstr;
3350 char *errormsg = NULL;
3351
3352 rettv->v_type = VAR_STRING;
3353 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
3354
3355 memset(&eap, 0, sizeof(eap));
3356 eap.cmd = cmdstr;
3357 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02003358 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02003359 eap.usefilter = FALSE;
3360 eap.nextcmd = NULL;
3361 eap.cmdidx = CMD_USER;
3362
3363 expand_filename(&eap, &cmdstr, &errormsg);
3364 if (errormsg != NULL && *errormsg != NUL)
3365 emsg(errormsg);
3366
3367 rettv->vval.v_string = cmdstr;
3368}
3369
3370/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003371 * "feedkeys()" function
3372 */
3373 static void
3374f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
3375{
3376 int remap = TRUE;
3377 int insert = FALSE;
3378 char_u *keys, *flags;
3379 char_u nbuf[NUMBUFLEN];
3380 int typed = FALSE;
3381 int execute = FALSE;
3382 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003383 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003384 char_u *keys_esc;
3385
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003386 // This is not allowed in the sandbox. If the commands would still be
3387 // executed in the sandbox it would be OK, but it probably happens later,
3388 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003389 if (check_secure())
3390 return;
3391
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003392 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003393
3394 if (argvars[1].v_type != VAR_UNKNOWN)
3395 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003396 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003397 for ( ; *flags != NUL; ++flags)
3398 {
3399 switch (*flags)
3400 {
3401 case 'n': remap = FALSE; break;
3402 case 'm': remap = TRUE; break;
3403 case 't': typed = TRUE; break;
3404 case 'i': insert = TRUE; break;
3405 case 'x': execute = TRUE; break;
3406 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003407 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003408 }
3409 }
3410 }
3411
3412 if (*keys != NUL || execute)
3413 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003414 // Need to escape K_SPECIAL and CSI before putting the string in the
3415 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003416 keys_esc = vim_strsave_escape_csi(keys);
3417 if (keys_esc != NULL)
3418 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003419 if (lowlevel)
3420 {
3421#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01003422 int idx;
3423 int len = (int)STRLEN(keys);
3424
3425 for (idx = 0; idx < len; ++idx)
3426 {
3427 // if a CTRL-C was typed, set got_int, similar to what
3428 // happens in fill_input_buf()
3429 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
3430 got_int = TRUE;
3431 add_to_input_buf(keys + idx, 1);
3432 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003433#else
3434 emsg(_("E980: lowlevel input not supported"));
3435#endif
3436 }
3437 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003438 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003439 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003440 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003441 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003442#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003443 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003444#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003445 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003446 typebuf_was_filled = TRUE;
3447 }
3448 vim_free(keys_esc);
3449
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003450 if (execute)
3451 {
3452 int save_msg_scroll = msg_scroll;
3453
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003454 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003455 msg_scroll = FALSE;
3456
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003457 if (!dangerous)
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003458 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003459 ++ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003460 ++in_feedkeys;
3461 }
Bram Moolenaar905dd902019-04-07 14:21:47 +02003462 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003463 if (!dangerous)
Bram Moolenaar189832b2020-09-23 12:29:11 +02003464 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003465 --ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003466 --in_feedkeys;
Bram Moolenaar189832b2020-09-23 12:29:11 +02003467 }
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003468
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003469 msg_scroll |= save_msg_scroll;
3470 }
3471 }
3472 }
3473}
3474
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003475#ifdef FEAT_FLOAT
3476/*
3477 * "float2nr({float})" function
3478 */
3479 static void
3480f_float2nr(typval_T *argvars, typval_T *rettv)
3481{
3482 float_T f = 0.0;
3483
3484 if (get_float_arg(argvars, &f) == OK)
3485 {
Bram Moolenaar37184272020-05-23 19:30:05 +02003486 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003487 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02003488 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003489 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003490 else
3491 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003492 }
3493}
3494
3495/*
3496 * "floor({float})" function
3497 */
3498 static void
3499f_floor(typval_T *argvars, typval_T *rettv)
3500{
3501 float_T f = 0.0;
3502
3503 rettv->v_type = VAR_FLOAT;
3504 if (get_float_arg(argvars, &f) == OK)
3505 rettv->vval.v_float = floor(f);
3506 else
3507 rettv->vval.v_float = 0.0;
3508}
3509
3510/*
3511 * "fmod()" function
3512 */
3513 static void
3514f_fmod(typval_T *argvars, typval_T *rettv)
3515{
3516 float_T fx = 0.0, fy = 0.0;
3517
3518 rettv->v_type = VAR_FLOAT;
3519 if (get_float_arg(argvars, &fx) == OK
3520 && get_float_arg(&argvars[1], &fy) == OK)
3521 rettv->vval.v_float = fmod(fx, fy);
3522 else
3523 rettv->vval.v_float = 0.0;
3524}
3525#endif
3526
3527/*
3528 * "fnameescape({string})" function
3529 */
3530 static void
3531f_fnameescape(typval_T *argvars, typval_T *rettv)
3532{
3533 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003534 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003535 rettv->v_type = VAR_STRING;
3536}
3537
3538/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003539 * "foreground()" function
3540 */
3541 static void
3542f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3543{
3544#ifdef FEAT_GUI
3545 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003546 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003547 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003548 return;
3549 }
3550#endif
3551#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003552 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003553#endif
3554}
3555
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003556 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003557common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003558{
3559 char_u *s;
3560 char_u *name;
3561 int use_string = FALSE;
3562 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003563 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003564 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003565
3566 if (argvars[0].v_type == VAR_FUNC)
3567 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003568 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003569 s = argvars[0].vval.v_string;
3570 }
3571 else if (argvars[0].v_type == VAR_PARTIAL
3572 && argvars[0].vval.v_partial != NULL)
3573 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003574 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003575 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003576 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003577 }
3578 else
3579 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003580 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003581 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003582 use_string = TRUE;
3583 }
3584
Bram Moolenaar843b8842016-08-21 14:36:15 +02003585 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003586 {
3587 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003588 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar32b3f822021-01-06 21:59:39 +01003589 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF,
3590 NULL, NULL, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003591 if (*name != NUL)
3592 s = NULL;
3593 }
3594
Bram Moolenaar843b8842016-08-21 14:36:15 +02003595 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
3596 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003597 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003598 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003599 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003600 ? find_func(trans_name, is_global, NULL) == NULL
3601 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003602 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003603 else
3604 {
3605 int dict_idx = 0;
3606 int arg_idx = 0;
3607 list_T *list = NULL;
3608
3609 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
3610 {
3611 char sid_buf[25];
3612 int off = *s == 's' ? 2 : 5;
3613
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003614 // Expand s: and <SID> into <SNR>nr_, so that the function can
3615 // also be called from another script. Using trans_function_name()
3616 // would also work, but some plugins depend on the name being
3617 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02003618 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02003619 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003620 if (name != NULL)
3621 {
3622 STRCPY(name, sid_buf);
3623 STRCAT(name, s + off);
3624 }
3625 }
3626 else
3627 name = vim_strsave(s);
3628
3629 if (argvars[1].v_type != VAR_UNKNOWN)
3630 {
3631 if (argvars[2].v_type != VAR_UNKNOWN)
3632 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003633 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003634 arg_idx = 1;
3635 dict_idx = 2;
3636 }
3637 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003638 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003639 dict_idx = 1;
3640 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003641 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003642 arg_idx = 1;
3643 if (dict_idx > 0)
3644 {
3645 if (argvars[dict_idx].v_type != VAR_DICT)
3646 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003647 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003648 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003649 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003650 }
3651 if (argvars[dict_idx].vval.v_dict == NULL)
3652 dict_idx = 0;
3653 }
3654 if (arg_idx > 0)
3655 {
3656 if (argvars[arg_idx].v_type != VAR_LIST)
3657 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003658 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003659 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003660 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003661 }
3662 list = argvars[arg_idx].vval.v_list;
3663 if (list == NULL || list->lv_len == 0)
3664 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003665 else if (list->lv_len > MAX_FUNC_ARGS)
3666 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01003667 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003668 vim_free(name);
3669 goto theend;
3670 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003671 }
3672 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003673 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003674 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003675 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003676
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003677 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003678 if (pt == NULL)
3679 vim_free(name);
3680 else
3681 {
3682 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
3683 {
3684 listitem_T *li;
3685 int i = 0;
3686 int arg_len = 0;
3687 int lv_len = 0;
3688
3689 if (arg_pt != NULL)
3690 arg_len = arg_pt->pt_argc;
3691 if (list != NULL)
3692 lv_len = list->lv_len;
3693 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003694 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003695 if (pt->pt_argv == NULL)
3696 {
3697 vim_free(pt);
3698 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003699 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003700 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003701 for (i = 0; i < arg_len; i++)
3702 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
3703 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003704 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003705 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003706 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003707 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003708 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003709 }
3710
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003711 // For "function(dict.func, [], dict)" and "func" is a partial
3712 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003713 if (dict_idx > 0)
3714 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003715 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003716 pt->pt_dict = argvars[dict_idx].vval.v_dict;
3717 ++pt->pt_dict->dv_refcount;
3718 }
3719 else if (arg_pt != NULL)
3720 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003721 // If the dict was bound automatically the result is also
3722 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003723 pt->pt_dict = arg_pt->pt_dict;
3724 pt->pt_auto = arg_pt->pt_auto;
3725 if (pt->pt_dict != NULL)
3726 ++pt->pt_dict->dv_refcount;
3727 }
3728
3729 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003730 if (arg_pt != NULL && arg_pt->pt_func != NULL)
3731 {
3732 pt->pt_func = arg_pt->pt_func;
3733 func_ptr_ref(pt->pt_func);
3734 vim_free(name);
3735 }
3736 else if (is_funcref)
3737 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003738 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003739 func_ptr_ref(pt->pt_func);
3740 vim_free(name);
3741 }
3742 else
3743 {
3744 pt->pt_name = name;
3745 func_ref(name);
3746 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003747 }
3748 rettv->v_type = VAR_PARTIAL;
3749 rettv->vval.v_partial = pt;
3750 }
3751 else
3752 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003753 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003754 rettv->v_type = VAR_FUNC;
3755 rettv->vval.v_string = name;
3756 func_ref(name);
3757 }
3758 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003759theend:
3760 vim_free(trans_name);
3761}
3762
3763/*
3764 * "funcref()" function
3765 */
3766 static void
3767f_funcref(typval_T *argvars, typval_T *rettv)
3768{
3769 common_function(argvars, rettv, TRUE);
3770}
3771
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003772 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003773ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003774{
3775 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
3776 return &t_func_any;
Bram Moolenaar5e654232020-09-16 15:22:00 +02003777 return &t_func_unknown;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003778}
3779
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003780/*
3781 * "function()" function
3782 */
3783 static void
3784f_function(typval_T *argvars, typval_T *rettv)
3785{
3786 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003787}
3788
3789/*
3790 * "garbagecollect()" function
3791 */
3792 static void
3793f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
3794{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003795 // This is postponed until we are back at the toplevel, because we may be
3796 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003797 want_garbage_collect = TRUE;
3798
Bram Moolenaar2df47312020-09-05 17:30:44 +02003799 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003800 garbage_collect_at_exit = TRUE;
3801}
3802
3803/*
3804 * "get()" function
3805 */
3806 static void
3807f_get(typval_T *argvars, typval_T *rettv)
3808{
3809 listitem_T *li;
3810 list_T *l;
3811 dictitem_T *di;
3812 dict_T *d;
3813 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003814 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003815
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003816 if (argvars[0].v_type == VAR_BLOB)
3817 {
3818 int error = FALSE;
3819 int idx = tv_get_number_chk(&argvars[1], &error);
3820
3821 if (!error)
3822 {
3823 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003824 if (idx < 0)
3825 idx = blob_len(argvars[0].vval.v_blob) + idx;
3826 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
3827 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003828 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003829 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003830 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003831 tv = rettv;
3832 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003833 }
3834 }
3835 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003836 {
3837 if ((l = argvars[0].vval.v_list) != NULL)
3838 {
3839 int error = FALSE;
3840
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003841 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003842 if (!error && li != NULL)
3843 tv = &li->li_tv;
3844 }
3845 }
3846 else if (argvars[0].v_type == VAR_DICT)
3847 {
3848 if ((d = argvars[0].vval.v_dict) != NULL)
3849 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003850 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003851 if (di != NULL)
3852 tv = &di->di_tv;
3853 }
3854 }
3855 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
3856 {
3857 partial_T *pt;
3858 partial_T fref_pt;
3859
3860 if (argvars[0].v_type == VAR_PARTIAL)
3861 pt = argvars[0].vval.v_partial;
3862 else
3863 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02003864 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003865 fref_pt.pt_name = argvars[0].vval.v_string;
3866 pt = &fref_pt;
3867 }
3868
3869 if (pt != NULL)
3870 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003871 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003872 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003873
3874 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
3875 {
3876 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003877 n = partial_name(pt);
3878 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003879 rettv->vval.v_string = NULL;
3880 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003881 {
3882 rettv->vval.v_string = vim_strsave(n);
3883 if (rettv->v_type == VAR_FUNC)
3884 func_ref(rettv->vval.v_string);
3885 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003886 }
3887 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003888 {
3889 what_is_dict = TRUE;
3890 if (pt->pt_dict != NULL)
3891 rettv_dict_set(rettv, pt->pt_dict);
3892 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003893 else if (STRCMP(what, "args") == 0)
3894 {
3895 rettv->v_type = VAR_LIST;
3896 if (rettv_list_alloc(rettv) == OK)
3897 {
3898 int i;
3899
3900 for (i = 0; i < pt->pt_argc; ++i)
3901 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3902 }
3903 }
3904 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003905 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003906
3907 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3908 // third argument
3909 if (!what_is_dict)
3910 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003911 }
3912 }
3913 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003914 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003915
3916 if (tv == NULL)
3917 {
3918 if (argvars[2].v_type != VAR_UNKNOWN)
3919 copy_tv(&argvars[2], rettv);
3920 }
3921 else
3922 copy_tv(tv, rettv);
3923}
3924
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003925/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003926 * "getchangelist()" function
3927 */
3928 static void
3929f_getchangelist(typval_T *argvars, typval_T *rettv)
3930{
3931#ifdef FEAT_JUMPLIST
3932 buf_T *buf;
3933 int i;
3934 list_T *l;
3935 dict_T *d;
3936#endif
3937
3938 if (rettv_list_alloc(rettv) != OK)
3939 return;
3940
3941#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003942 if (argvars[0].v_type == VAR_UNKNOWN)
3943 buf = curbuf;
3944 else
Bram Moolenaara5d38412020-09-02 21:02:35 +02003945 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003946 if (buf == NULL)
3947 return;
3948
3949 l = list_alloc();
3950 if (l == NULL)
3951 return;
3952
3953 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3954 return;
3955 /*
3956 * The current window change list index tracks only the position in the
3957 * current buffer change list. For other buffers, use the change list
3958 * length as the current index.
3959 */
3960 list_append_number(rettv->vval.v_list,
3961 (varnumber_T)((buf == curwin->w_buffer)
3962 ? curwin->w_changelistidx : buf->b_changelistlen));
3963
3964 for (i = 0; i < buf->b_changelistlen; ++i)
3965 {
3966 if (buf->b_changelist[i].lnum == 0)
3967 continue;
3968 if ((d = dict_alloc()) == NULL)
3969 return;
3970 if (list_append_dict(l, d) == FAIL)
3971 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003972 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3973 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003974 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003975 }
3976#endif
3977}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003978
Bram Moolenaar6f02b002021-01-10 20:22:54 +01003979 static void
3980getpos_both(
3981 typval_T *argvars,
3982 typval_T *rettv,
3983 int getcurpos,
3984 int charcol)
3985{
3986 pos_T *fp = NULL;
3987 pos_T pos;
3988 win_T *wp = curwin;
3989 list_T *l;
3990 int fnum = -1;
3991
3992 if (rettv_list_alloc(rettv) == OK)
3993 {
3994 l = rettv->vval.v_list;
3995 if (getcurpos)
3996 {
3997 if (argvars[0].v_type != VAR_UNKNOWN)
3998 {
3999 wp = find_win_by_nr_or_id(&argvars[0]);
4000 if (wp != NULL)
4001 fp = &wp->w_cursor;
4002 }
4003 else
4004 fp = &curwin->w_cursor;
4005 if (fp != NULL && charcol)
4006 {
4007 pos = *fp;
4008 pos.col = buf_byteidx_to_charidx(wp->w_buffer, pos.lnum,
4009 pos.col) - 1;
4010 fp = &pos;
4011 }
4012 }
4013 else
4014 fp = var2fpos(&argvars[0], TRUE, &fnum, charcol);
4015 if (fnum != -1)
4016 list_append_number(l, (varnumber_T)fnum);
4017 else
4018 list_append_number(l, (varnumber_T)0);
4019 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
4020 : (varnumber_T)0);
4021 list_append_number(l, (fp != NULL)
4022 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
4023 : (varnumber_T)0);
4024 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
4025 (varnumber_T)0);
4026 if (getcurpos)
4027 {
4028 int save_set_curswant = curwin->w_set_curswant;
4029 colnr_T save_curswant = curwin->w_curswant;
4030 colnr_T save_virtcol = curwin->w_virtcol;
4031
4032 if (wp == curwin)
4033 update_curswant();
4034 list_append_number(l, wp == NULL ? 0 : wp->w_curswant == MAXCOL
4035 ? (varnumber_T)MAXCOL : (varnumber_T)wp->w_curswant + 1);
4036
4037 // Do not change "curswant", as it is unexpected that a get
4038 // function has a side effect.
4039 if (wp == curwin && save_set_curswant)
4040 {
4041 curwin->w_set_curswant = save_set_curswant;
4042 curwin->w_curswant = save_curswant;
4043 curwin->w_virtcol = save_virtcol;
4044 curwin->w_valid &= ~VALID_VIRTCOL;
4045 }
4046 }
4047 }
4048 else
4049 rettv->vval.v_number = FALSE;
4050}
4051
4052/*
4053 * "getcharpos()" function
4054 */
4055 static void
4056f_getcharpos(typval_T *argvars UNUSED, typval_T *rettv)
4057{
4058 getpos_both(argvars, rettv, FALSE, TRUE);
4059}
4060
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004061/*
4062 * "getcharsearch()" function
4063 */
4064 static void
4065f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
4066{
4067 if (rettv_dict_alloc(rettv) != FAIL)
4068 {
4069 dict_T *dict = rettv->vval.v_dict;
4070
Bram Moolenaare0be1672018-07-08 16:50:37 +02004071 dict_add_string(dict, "char", last_csearch());
4072 dict_add_number(dict, "forward", last_csearch_forward());
4073 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004074 }
4075}
4076
4077/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02004078 * "getenv()" function
4079 */
4080 static void
4081f_getenv(typval_T *argvars, typval_T *rettv)
4082{
4083 int mustfree = FALSE;
4084 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
4085
4086 if (p == NULL)
4087 {
4088 rettv->v_type = VAR_SPECIAL;
4089 rettv->vval.v_number = VVAL_NULL;
4090 return;
4091 }
4092 if (!mustfree)
4093 p = vim_strsave(p);
4094 rettv->vval.v_string = p;
4095 rettv->v_type = VAR_STRING;
4096}
4097
4098/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004099 * "getfontname()" function
4100 */
4101 static void
4102f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
4103{
4104 rettv->v_type = VAR_STRING;
4105 rettv->vval.v_string = NULL;
4106#ifdef FEAT_GUI
4107 if (gui.in_use)
4108 {
4109 GuiFont font;
4110 char_u *name = NULL;
4111
4112 if (argvars[0].v_type == VAR_UNKNOWN)
4113 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004114 // Get the "Normal" font. Either the name saved by
4115 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004116 font = gui.norm_font;
4117 name = hl_get_font_name();
4118 }
4119 else
4120 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004121 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004122 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004123 return;
4124 font = gui_mch_get_font(name, FALSE);
4125 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004126 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004127 }
4128 rettv->vval.v_string = gui_mch_get_fontname(font, name);
4129 if (argvars[0].v_type != VAR_UNKNOWN)
4130 gui_mch_free_font(font);
4131 }
4132#endif
4133}
4134
4135/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01004136 * "getjumplist()" function
4137 */
4138 static void
4139f_getjumplist(typval_T *argvars, typval_T *rettv)
4140{
4141#ifdef FEAT_JUMPLIST
4142 win_T *wp;
4143 int i;
4144 list_T *l;
4145 dict_T *d;
4146#endif
4147
4148 if (rettv_list_alloc(rettv) != OK)
4149 return;
4150
4151#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004152 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004153 if (wp == NULL)
4154 return;
4155
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01004156 cleanup_jumplist(wp, TRUE);
4157
Bram Moolenaar4f505882018-02-10 21:06:32 +01004158 l = list_alloc();
4159 if (l == NULL)
4160 return;
4161
4162 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4163 return;
4164 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
4165
4166 for (i = 0; i < wp->w_jumplistlen; ++i)
4167 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004168 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
4169 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01004170 if ((d = dict_alloc()) == NULL)
4171 return;
4172 if (list_append_dict(l, d) == FAIL)
4173 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02004174 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
4175 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004176 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004177 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004178 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02004179 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004180 }
4181#endif
4182}
4183
4184/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004185 * "getpid()" function
4186 */
4187 static void
4188f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
4189{
4190 rettv->vval.v_number = mch_get_pid();
4191}
4192
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004193/*
4194 * "getcurpos()" function
4195 */
4196 static void
4197f_getcurpos(typval_T *argvars, typval_T *rettv)
4198{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01004199 getpos_both(argvars, rettv, TRUE, FALSE);
4200}
4201
4202 static void
4203f_getcursorcharpos(typval_T *argvars, typval_T *rettv)
4204{
4205 getpos_both(argvars, rettv, TRUE, TRUE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004206}
4207
4208/*
4209 * "getpos(string)" function
4210 */
4211 static void
4212f_getpos(typval_T *argvars, typval_T *rettv)
4213{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01004214 getpos_both(argvars, rettv, FALSE, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004215}
4216
4217/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004218 * "getreg()" function
4219 */
4220 static void
4221f_getreg(typval_T *argvars, typval_T *rettv)
4222{
4223 char_u *strregname;
4224 int regname;
4225 int arg2 = FALSE;
4226 int return_list = FALSE;
4227 int error = FALSE;
4228
4229 if (argvars[0].v_type != VAR_UNKNOWN)
4230 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004231 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004232 error = strregname == NULL;
4233 if (argvars[1].v_type != VAR_UNKNOWN)
4234 {
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004235 arg2 = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004236 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004237 return_list = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004238 }
4239 }
4240 else
4241 strregname = get_vim_var_str(VV_REG);
4242
4243 if (error)
4244 return;
4245
4246 regname = (strregname == NULL ? '"' : *strregname);
4247 if (regname == 0)
4248 regname = '"';
4249
4250 if (return_list)
4251 {
4252 rettv->v_type = VAR_LIST;
4253 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
4254 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
4255 if (rettv->vval.v_list == NULL)
4256 (void)rettv_list_alloc(rettv);
4257 else
4258 ++rettv->vval.v_list->lv_refcount;
4259 }
4260 else
4261 {
4262 rettv->v_type = VAR_STRING;
4263 rettv->vval.v_string = get_reg_contents(regname,
4264 arg2 ? GREG_EXPR_SRC : 0);
4265 }
4266}
4267
4268/*
4269 * "getregtype()" function
4270 */
4271 static void
4272f_getregtype(typval_T *argvars, typval_T *rettv)
4273{
4274 char_u *strregname;
4275 int regname;
4276 char_u buf[NUMBUFLEN + 2];
4277 long reglen = 0;
4278
4279 if (argvars[0].v_type != VAR_UNKNOWN)
4280 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004281 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004282 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004283 {
4284 rettv->v_type = VAR_STRING;
4285 rettv->vval.v_string = NULL;
4286 return;
4287 }
4288 }
4289 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004290 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004291 strregname = get_vim_var_str(VV_REG);
4292
4293 regname = (strregname == NULL ? '"' : *strregname);
4294 if (regname == 0)
4295 regname = '"';
4296
4297 buf[0] = NUL;
4298 buf[1] = NUL;
4299 switch (get_reg_type(regname, &reglen))
4300 {
4301 case MLINE: buf[0] = 'V'; break;
4302 case MCHAR: buf[0] = 'v'; break;
4303 case MBLOCK:
4304 buf[0] = Ctrl_V;
4305 sprintf((char *)buf + 1, "%ld", reglen + 1);
4306 break;
4307 }
4308 rettv->v_type = VAR_STRING;
4309 rettv->vval.v_string = vim_strsave(buf);
4310}
4311
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02004312/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01004313 * "gettagstack()" function
4314 */
4315 static void
4316f_gettagstack(typval_T *argvars, typval_T *rettv)
4317{
4318 win_T *wp = curwin; // default is current window
4319
4320 if (rettv_dict_alloc(rettv) != OK)
4321 return;
4322
4323 if (argvars[0].v_type != VAR_UNKNOWN)
4324 {
4325 wp = find_win_by_nr_or_id(&argvars[0]);
4326 if (wp == NULL)
4327 return;
4328 }
4329
4330 get_tagstack(wp, rettv->vval.v_dict);
4331}
4332
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +02004333/*
4334 * "gettext()" function
4335 */
4336 static void
4337f_gettext(typval_T *argvars, typval_T *rettv)
4338{
4339 if (argvars[0].v_type != VAR_STRING
4340 || argvars[0].vval.v_string == NULL
4341 || *argvars[0].vval.v_string == NUL)
4342 {
4343 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
4344 }
4345 else
4346 {
4347 rettv->v_type = VAR_STRING;
4348 rettv->vval.v_string = vim_strsave(
4349 (char_u *)_(argvars[0].vval.v_string));
4350 }
4351}
4352
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004353// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004354#include "version.h"
4355
4356/*
4357 * "has()" function
4358 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01004359 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004360f_has(typval_T *argvars, typval_T *rettv)
4361{
4362 int i;
4363 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01004364 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004365 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01004366 typedef struct {
4367 char *name;
4368 short present;
4369 } has_item_T;
4370 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004371 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004372 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004373#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01004374 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004375#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004376 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004377#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004378 },
4379 {"arp",
4380#if defined(AMIGA) && defined(FEAT_ARP)
4381 1
4382#else
4383 0
4384#endif
4385 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004386 {"haiku",
4387#ifdef __HAIKU__
4388 1
4389#else
4390 0
4391#endif
4392 },
4393 {"bsd",
4394#if defined(BSD) && !defined(MACOS_X)
4395 1
4396#else
4397 0
4398#endif
4399 },
4400 {"hpux",
4401#ifdef hpux
4402 1
4403#else
4404 0
4405#endif
4406 },
4407 {"linux",
4408#ifdef __linux__
4409 1
4410#else
4411 0
4412#endif
4413 },
4414 {"mac", // Mac OS X (and, once, Mac OS Classic)
4415#ifdef MACOS_X
4416 1
4417#else
4418 0
4419#endif
4420 },
4421 {"osx", // Mac OS X
4422#ifdef MACOS_X
4423 1
4424#else
4425 0
4426#endif
4427 },
4428 {"macunix", // Mac OS X, with the darwin feature
4429#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4430 1
4431#else
4432 0
4433#endif
4434 },
4435 {"osxdarwin", // synonym for macunix
4436#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4437 1
4438#else
4439 0
4440#endif
4441 },
4442 {"qnx",
4443#ifdef __QNX__
4444 1
4445#else
4446 0
4447#endif
4448 },
4449 {"sun",
4450#ifdef SUN_SYSTEM
4451 1
4452#else
4453 0
4454#endif
4455 },
4456 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004457#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01004458 1
4459#else
4460 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004461#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004462 },
4463 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004464#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01004465 1
4466#else
4467 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004468#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004469 },
4470 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01004471#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01004472 1
4473#else
4474 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004475#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004476 },
4477 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01004478#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01004479 1
4480#else
4481 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004482#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004483 },
4484 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01004485#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01004486 1
4487#else
4488 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004489#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004490 },
4491 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004492#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004493 1
4494#else
4495 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004496#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004497 },
4498 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004499#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004500 1
4501#else
4502 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004503#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004504 },
4505 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004506#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01004507 1
4508#else
4509 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004510#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004511 },
4512 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004513#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004514 1
4515#else
4516 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004517#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004518 },
4519 {"autocmd", 1},
4520 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004521#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01004522 1
4523#else
4524 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004525#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004526 },
4527 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004528#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004529 1
4530#else
4531 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004532#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004533 },
4534 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004535#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01004536 1
4537#else
4538 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004539#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004540 },
4541 {"balloon_multiline",
4542#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
4543 // MS-Windows requires runtime check, see below
4544 1
4545#else
4546 0
4547#endif
4548 },
4549 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004550#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01004551 1
4552#else
4553 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01004554#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004555 },
4556 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004557#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01004558 1
4559#else
4560 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004561#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004562 },
4563 {"all_builtin_terms",
4564#if defined(ALL_BUILTIN_TCAPS)
4565 1
4566#else
4567 0
4568#endif
4569 },
4570 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004571#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01004572 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004573 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01004574 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004575#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004576 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004577#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004578 },
4579 {"byte_offset",
4580#ifdef FEAT_BYTEOFF
4581 1
4582#else
4583 0
4584#endif
4585 },
4586 {"channel",
4587#ifdef FEAT_JOB_CHANNEL
4588 1
4589#else
4590 0
4591#endif
4592 },
4593 {"cindent",
4594#ifdef FEAT_CINDENT
4595 1
4596#else
4597 0
4598#endif
4599 },
4600 {"clientserver",
4601#ifdef FEAT_CLIENTSERVER
4602 1
4603#else
4604 0
4605#endif
4606 },
4607 {"clipboard",
4608#ifdef FEAT_CLIPBOARD
4609 1
4610#else
4611 0
4612#endif
4613 },
4614 {"cmdline_compl", 1},
4615 {"cmdline_hist", 1},
4616 {"comments", 1},
4617 {"conceal",
4618#ifdef FEAT_CONCEAL
4619 1
4620#else
4621 0
4622#endif
4623 },
4624 {"cryptv",
4625#ifdef FEAT_CRYPT
4626 1
4627#else
4628 0
4629#endif
4630 },
4631 {"crypt-blowfish",
4632#ifdef FEAT_CRYPT
4633 1
4634#else
4635 0
4636#endif
4637 },
4638 {"crypt-blowfish2",
4639#ifdef FEAT_CRYPT
4640 1
4641#else
4642 0
4643#endif
4644 },
4645 {"cscope",
4646#ifdef FEAT_CSCOPE
4647 1
4648#else
4649 0
4650#endif
4651 },
4652 {"cursorbind", 1},
4653 {"cursorshape",
4654#ifdef CURSOR_SHAPE
4655 1
4656#else
4657 0
4658#endif
4659 },
4660 {"debug",
4661#ifdef DEBUG
4662 1
4663#else
4664 0
4665#endif
4666 },
4667 {"dialog_con",
4668#ifdef FEAT_CON_DIALOG
4669 1
4670#else
4671 0
4672#endif
4673 },
4674 {"dialog_gui",
4675#ifdef FEAT_GUI_DIALOG
4676 1
4677#else
4678 0
4679#endif
4680 },
4681 {"diff",
4682#ifdef FEAT_DIFF
4683 1
4684#else
4685 0
4686#endif
4687 },
4688 {"digraphs",
4689#ifdef FEAT_DIGRAPHS
4690 1
4691#else
4692 0
4693#endif
4694 },
4695 {"directx",
4696#ifdef FEAT_DIRECTX
4697 1
4698#else
4699 0
4700#endif
4701 },
4702 {"dnd",
4703#ifdef FEAT_DND
4704 1
4705#else
4706 0
4707#endif
4708 },
4709 {"emacs_tags",
4710#ifdef FEAT_EMACS_TAGS
4711 1
4712#else
4713 0
4714#endif
4715 },
4716 {"eval", 1}, // always present, of course!
4717 {"ex_extra", 1}, // graduated feature
4718 {"extra_search",
4719#ifdef FEAT_SEARCH_EXTRA
4720 1
4721#else
4722 0
4723#endif
4724 },
4725 {"file_in_path",
4726#ifdef FEAT_SEARCHPATH
4727 1
4728#else
4729 0
4730#endif
4731 },
4732 {"filterpipe",
4733#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
4734 1
4735#else
4736 0
4737#endif
4738 },
4739 {"find_in_path",
4740#ifdef FEAT_FIND_ID
4741 1
4742#else
4743 0
4744#endif
4745 },
4746 {"float",
4747#ifdef FEAT_FLOAT
4748 1
4749#else
4750 0
4751#endif
4752 },
4753 {"folding",
4754#ifdef FEAT_FOLDING
4755 1
4756#else
4757 0
4758#endif
4759 },
4760 {"footer",
4761#ifdef FEAT_FOOTER
4762 1
4763#else
4764 0
4765#endif
4766 },
4767 {"fork",
4768#if !defined(USE_SYSTEM) && defined(UNIX)
4769 1
4770#else
4771 0
4772#endif
4773 },
4774 {"gettext",
4775#ifdef FEAT_GETTEXT
4776 1
4777#else
4778 0
4779#endif
4780 },
4781 {"gui",
4782#ifdef FEAT_GUI
4783 1
4784#else
4785 0
4786#endif
4787 },
4788 {"gui_neXtaw",
4789#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
4790 1
4791#else
4792 0
4793#endif
4794 },
4795 {"gui_athena",
4796#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
4797 1
4798#else
4799 0
4800#endif
4801 },
4802 {"gui_gtk",
4803#ifdef FEAT_GUI_GTK
4804 1
4805#else
4806 0
4807#endif
4808 },
4809 {"gui_gtk2",
4810#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
4811 1
4812#else
4813 0
4814#endif
4815 },
4816 {"gui_gtk3",
4817#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
4818 1
4819#else
4820 0
4821#endif
4822 },
4823 {"gui_gnome",
4824#ifdef FEAT_GUI_GNOME
4825 1
4826#else
4827 0
4828#endif
4829 },
4830 {"gui_haiku",
4831#ifdef FEAT_GUI_HAIKU
4832 1
4833#else
4834 0
4835#endif
4836 },
Bram Moolenaar097148e2020-08-11 21:58:20 +02004837 {"gui_mac", 0},
Bram Moolenaar79296512020-03-22 16:17:14 +01004838 {"gui_motif",
4839#ifdef FEAT_GUI_MOTIF
4840 1
4841#else
4842 0
4843#endif
4844 },
4845 {"gui_photon",
4846#ifdef FEAT_GUI_PHOTON
4847 1
4848#else
4849 0
4850#endif
4851 },
4852 {"gui_win32",
4853#ifdef FEAT_GUI_MSWIN
4854 1
4855#else
4856 0
4857#endif
4858 },
4859 {"iconv",
4860#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
4861 1
4862#else
4863 0
4864#endif
4865 },
4866 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02004867 {"ipv6",
4868#ifdef FEAT_IPV6
4869 1
4870#else
4871 0
4872#endif
4873 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004874 {"job",
4875#ifdef FEAT_JOB_CHANNEL
4876 1
4877#else
4878 0
4879#endif
4880 },
4881 {"jumplist",
4882#ifdef FEAT_JUMPLIST
4883 1
4884#else
4885 0
4886#endif
4887 },
4888 {"keymap",
4889#ifdef FEAT_KEYMAP
4890 1
4891#else
4892 0
4893#endif
4894 },
4895 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
4896 {"langmap",
4897#ifdef FEAT_LANGMAP
4898 1
4899#else
4900 0
4901#endif
4902 },
4903 {"libcall",
4904#ifdef FEAT_LIBCALL
4905 1
4906#else
4907 0
4908#endif
4909 },
4910 {"linebreak",
4911#ifdef FEAT_LINEBREAK
4912 1
4913#else
4914 0
4915#endif
4916 },
4917 {"lispindent",
4918#ifdef FEAT_LISP
4919 1
4920#else
4921 0
4922#endif
4923 },
4924 {"listcmds", 1},
4925 {"localmap", 1},
4926 {"lua",
4927#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4928 1
4929#else
4930 0
4931#endif
4932 },
4933 {"menu",
4934#ifdef FEAT_MENU
4935 1
4936#else
4937 0
4938#endif
4939 },
4940 {"mksession",
4941#ifdef FEAT_SESSION
4942 1
4943#else
4944 0
4945#endif
4946 },
4947 {"modify_fname", 1},
4948 {"mouse", 1},
4949 {"mouseshape",
4950#ifdef FEAT_MOUSESHAPE
4951 1
4952#else
4953 0
4954#endif
4955 },
4956 {"mouse_dec",
4957#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4958 1
4959#else
4960 0
4961#endif
4962 },
4963 {"mouse_gpm",
4964#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4965 1
4966#else
4967 0
4968#endif
4969 },
4970 {"mouse_jsbterm",
4971#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4972 1
4973#else
4974 0
4975#endif
4976 },
4977 {"mouse_netterm",
4978#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4979 1
4980#else
4981 0
4982#endif
4983 },
4984 {"mouse_pterm",
4985#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4986 1
4987#else
4988 0
4989#endif
4990 },
4991 {"mouse_sgr",
4992#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4993 1
4994#else
4995 0
4996#endif
4997 },
4998 {"mouse_sysmouse",
4999#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
5000 1
5001#else
5002 0
5003#endif
5004 },
5005 {"mouse_urxvt",
5006#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
5007 1
5008#else
5009 0
5010#endif
5011 },
5012 {"mouse_xterm",
5013#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
5014 1
5015#else
5016 0
5017#endif
5018 },
5019 {"multi_byte", 1},
5020 {"multi_byte_ime",
5021#ifdef FEAT_MBYTE_IME
5022 1
5023#else
5024 0
5025#endif
5026 },
5027 {"multi_lang",
5028#ifdef FEAT_MULTI_LANG
5029 1
5030#else
5031 0
5032#endif
5033 },
5034 {"mzscheme",
5035#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
5036 1
5037#else
5038 0
5039#endif
5040 },
5041 {"num64", 1},
5042 {"ole",
5043#ifdef FEAT_OLE
5044 1
5045#else
5046 0
5047#endif
5048 },
5049 {"packages",
5050#ifdef FEAT_EVAL
5051 1
5052#else
5053 0
5054#endif
5055 },
5056 {"path_extra",
5057#ifdef FEAT_PATH_EXTRA
5058 1
5059#else
5060 0
5061#endif
5062 },
5063 {"perl",
5064#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
5065 1
5066#else
5067 0
5068#endif
5069 },
5070 {"persistent_undo",
5071#ifdef FEAT_PERSISTENT_UNDO
5072 1
5073#else
5074 0
5075#endif
5076 },
5077 {"python_compiled",
5078#if defined(FEAT_PYTHON)
5079 1
5080#else
5081 0
5082#endif
5083 },
5084 {"python_dynamic",
5085#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
5086 1
5087#else
5088 0
5089#endif
5090 },
5091 {"python",
5092#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
5093 1
5094#else
5095 0
5096#endif
5097 },
5098 {"pythonx",
5099#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
5100 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
5101 1
5102#else
5103 0
5104#endif
5105 },
5106 {"python3_compiled",
5107#if defined(FEAT_PYTHON3)
5108 1
5109#else
5110 0
5111#endif
5112 },
5113 {"python3_dynamic",
5114#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
5115 1
5116#else
5117 0
5118#endif
5119 },
5120 {"python3",
5121#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
5122 1
5123#else
5124 0
5125#endif
5126 },
5127 {"popupwin",
5128#ifdef FEAT_PROP_POPUP
5129 1
5130#else
5131 0
5132#endif
5133 },
5134 {"postscript",
5135#ifdef FEAT_POSTSCRIPT
5136 1
5137#else
5138 0
5139#endif
5140 },
5141 {"printer",
5142#ifdef FEAT_PRINTER
5143 1
5144#else
5145 0
5146#endif
5147 },
5148 {"profile",
5149#ifdef FEAT_PROFILE
5150 1
5151#else
5152 0
5153#endif
5154 },
5155 {"reltime",
5156#ifdef FEAT_RELTIME
5157 1
5158#else
5159 0
5160#endif
5161 },
5162 {"quickfix",
5163#ifdef FEAT_QUICKFIX
5164 1
5165#else
5166 0
5167#endif
5168 },
5169 {"rightleft",
5170#ifdef FEAT_RIGHTLEFT
5171 1
5172#else
5173 0
5174#endif
5175 },
5176 {"ruby",
5177#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
5178 1
5179#else
5180 0
5181#endif
5182 },
5183 {"scrollbind", 1},
5184 {"showcmd",
5185#ifdef FEAT_CMDL_INFO
5186 1
5187#else
5188 0
5189#endif
5190 },
5191 {"cmdline_info",
5192#ifdef FEAT_CMDL_INFO
5193 1
5194#else
5195 0
5196#endif
5197 },
5198 {"signs",
5199#ifdef FEAT_SIGNS
5200 1
5201#else
5202 0
5203#endif
5204 },
5205 {"smartindent",
5206#ifdef FEAT_SMARTINDENT
5207 1
5208#else
5209 0
5210#endif
5211 },
5212 {"startuptime",
5213#ifdef STARTUPTIME
5214 1
5215#else
5216 0
5217#endif
5218 },
5219 {"statusline",
5220#ifdef FEAT_STL_OPT
5221 1
5222#else
5223 0
5224#endif
5225 },
5226 {"netbeans_intg",
5227#ifdef FEAT_NETBEANS_INTG
5228 1
5229#else
5230 0
5231#endif
5232 },
5233 {"sound",
5234#ifdef FEAT_SOUND
5235 1
5236#else
5237 0
5238#endif
5239 },
5240 {"spell",
5241#ifdef FEAT_SPELL
5242 1
5243#else
5244 0
5245#endif
5246 },
5247 {"syntax",
5248#ifdef FEAT_SYN_HL
5249 1
5250#else
5251 0
5252#endif
5253 },
5254 {"system",
5255#if defined(USE_SYSTEM) || !defined(UNIX)
5256 1
5257#else
5258 0
5259#endif
5260 },
5261 {"tag_binary",
5262#ifdef FEAT_TAG_BINS
5263 1
5264#else
5265 0
5266#endif
5267 },
5268 {"tcl",
5269#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
5270 1
5271#else
5272 0
5273#endif
5274 },
5275 {"termguicolors",
5276#ifdef FEAT_TERMGUICOLORS
5277 1
5278#else
5279 0
5280#endif
5281 },
5282 {"terminal",
5283#if defined(FEAT_TERMINAL) && !defined(MSWIN)
5284 1
5285#else
5286 0
5287#endif
5288 },
5289 {"terminfo",
5290#ifdef TERMINFO
5291 1
5292#else
5293 0
5294#endif
5295 },
5296 {"termresponse",
5297#ifdef FEAT_TERMRESPONSE
5298 1
5299#else
5300 0
5301#endif
5302 },
5303 {"textobjects",
5304#ifdef FEAT_TEXTOBJ
5305 1
5306#else
5307 0
5308#endif
5309 },
5310 {"textprop",
5311#ifdef FEAT_PROP_POPUP
5312 1
5313#else
5314 0
5315#endif
5316 },
5317 {"tgetent",
5318#ifdef HAVE_TGETENT
5319 1
5320#else
5321 0
5322#endif
5323 },
5324 {"timers",
5325#ifdef FEAT_TIMERS
5326 1
5327#else
5328 0
5329#endif
5330 },
5331 {"title",
5332#ifdef FEAT_TITLE
5333 1
5334#else
5335 0
5336#endif
5337 },
5338 {"toolbar",
5339#ifdef FEAT_TOOLBAR
5340 1
5341#else
5342 0
5343#endif
5344 },
5345 {"unnamedplus",
5346#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
5347 1
5348#else
5349 0
5350#endif
5351 },
5352 {"user-commands", 1}, // was accidentally included in 5.4
5353 {"user_commands", 1},
5354 {"vartabs",
5355#ifdef FEAT_VARTABS
5356 1
5357#else
5358 0
5359#endif
5360 },
5361 {"vertsplit", 1},
5362 {"viminfo",
5363#ifdef FEAT_VIMINFO
5364 1
5365#else
5366 0
5367#endif
5368 },
5369 {"vimscript-1", 1},
5370 {"vimscript-2", 1},
5371 {"vimscript-3", 1},
5372 {"vimscript-4", 1},
5373 {"virtualedit", 1},
5374 {"visual", 1},
5375 {"visualextra", 1},
5376 {"vreplace", 1},
5377 {"vtp",
5378#ifdef FEAT_VTP
5379 1
5380#else
5381 0
5382#endif
5383 },
5384 {"wildignore",
5385#ifdef FEAT_WILDIGN
5386 1
5387#else
5388 0
5389#endif
5390 },
5391 {"wildmenu",
5392#ifdef FEAT_WILDMENU
5393 1
5394#else
5395 0
5396#endif
5397 },
5398 {"windows", 1},
5399 {"winaltkeys",
5400#ifdef FEAT_WAK
5401 1
5402#else
5403 0
5404#endif
5405 },
5406 {"writebackup",
5407#ifdef FEAT_WRITEBACKUP
5408 1
5409#else
5410 0
5411#endif
5412 },
5413 {"xim",
5414#ifdef FEAT_XIM
5415 1
5416#else
5417 0
5418#endif
5419 },
5420 {"xfontset",
5421#ifdef FEAT_XFONTSET
5422 1
5423#else
5424 0
5425#endif
5426 },
5427 {"xpm",
5428#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
5429 1
5430#else
5431 0
5432#endif
5433 },
5434 {"xpm_w32", // for backward compatibility
5435#ifdef FEAT_XPM_W32
5436 1
5437#else
5438 0
5439#endif
5440 },
5441 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005442#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01005443 1
5444#else
5445 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005446#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005447 },
5448 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005449#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01005450 1
5451#else
5452 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005453#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005454 },
5455 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005456#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01005457 1
5458#else
5459 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005460#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005461 },
5462 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005463#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01005464 1
5465#else
5466 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005467#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005468 },
5469 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005470#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01005471 1
5472#else
5473 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005474#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005475 },
5476 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005477 };
5478
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005479 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01005480 for (i = 0; has_list[i].name != NULL; ++i)
5481 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005482 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005483 x = TRUE;
5484 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005485 break;
5486 }
5487
Bram Moolenaar79296512020-03-22 16:17:14 +01005488 // features also in has_list[] but sometimes enabled at runtime
5489 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005490 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005491 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02005492 {
5493 // intentionally empty
5494 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01005495#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005496 else if (STRICMP(name, "balloon_multiline") == 0)
5497 n = multiline_balloon_available();
5498#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005499#ifdef VIMDLL
5500 else if (STRICMP(name, "filterpipe") == 0)
5501 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005502#endif
5503#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5504 else if (STRICMP(name, "iconv") == 0)
5505 n = iconv_enabled(FALSE);
5506#endif
5507#ifdef DYNAMIC_LUA
5508 else if (STRICMP(name, "lua") == 0)
5509 n = lua_enabled(FALSE);
5510#endif
5511#ifdef DYNAMIC_MZSCHEME
5512 else if (STRICMP(name, "mzscheme") == 0)
5513 n = mzscheme_enabled(FALSE);
5514#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005515#ifdef DYNAMIC_PERL
5516 else if (STRICMP(name, "perl") == 0)
5517 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005518#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005519#ifdef DYNAMIC_PYTHON
5520 else if (STRICMP(name, "python") == 0)
5521 n = python_enabled(FALSE);
5522#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005523#ifdef DYNAMIC_PYTHON3
5524 else if (STRICMP(name, "python3") == 0)
5525 n = python3_enabled(FALSE);
5526#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005527#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5528 else if (STRICMP(name, "pythonx") == 0)
5529 {
5530# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
5531 if (p_pyx == 0)
5532 n = python3_enabled(FALSE) || python_enabled(FALSE);
5533 else if (p_pyx == 3)
5534 n = python3_enabled(FALSE);
5535 else if (p_pyx == 2)
5536 n = python_enabled(FALSE);
5537# elif defined(DYNAMIC_PYTHON)
5538 n = python_enabled(FALSE);
5539# elif defined(DYNAMIC_PYTHON3)
5540 n = python3_enabled(FALSE);
5541# endif
5542 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005543#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005544#ifdef DYNAMIC_RUBY
5545 else if (STRICMP(name, "ruby") == 0)
5546 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005547#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005548#ifdef DYNAMIC_TCL
5549 else if (STRICMP(name, "tcl") == 0)
5550 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02005551#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01005552#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02005553 else if (STRICMP(name, "terminal") == 0)
5554 n = terminal_enabled();
5555#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005556 }
5557
Bram Moolenaar79296512020-03-22 16:17:14 +01005558 // features not in has_list[]
5559 if (x == FALSE)
5560 {
5561 if (STRNICMP(name, "patch", 5) == 0)
5562 {
5563 x = TRUE;
5564 if (name[5] == '-'
5565 && STRLEN(name) >= 11
5566 && vim_isdigit(name[6])
5567 && vim_isdigit(name[8])
5568 && vim_isdigit(name[10]))
5569 {
5570 int major = atoi((char *)name + 6);
5571 int minor = atoi((char *)name + 8);
5572
5573 // Expect "patch-9.9.01234".
5574 n = (major < VIM_VERSION_MAJOR
5575 || (major == VIM_VERSION_MAJOR
5576 && (minor < VIM_VERSION_MINOR
5577 || (minor == VIM_VERSION_MINOR
5578 && has_patch(atoi((char *)name + 10))))));
5579 }
5580 else
5581 n = has_patch(atoi((char *)name + 5));
5582 }
5583 else if (STRICMP(name, "vim_starting") == 0)
5584 {
5585 x = TRUE;
5586 n = (starting != 0);
5587 }
5588 else if (STRICMP(name, "ttyin") == 0)
5589 {
5590 x = TRUE;
5591 n = mch_input_isatty();
5592 }
5593 else if (STRICMP(name, "ttyout") == 0)
5594 {
5595 x = TRUE;
5596 n = stdout_isatty;
5597 }
5598 else if (STRICMP(name, "multi_byte_encoding") == 0)
5599 {
5600 x = TRUE;
5601 n = has_mbyte;
5602 }
5603 else if (STRICMP(name, "gui_running") == 0)
5604 {
5605 x = TRUE;
5606#ifdef FEAT_GUI
5607 n = (gui.in_use || gui.starting);
5608#endif
5609 }
5610 else if (STRICMP(name, "browse") == 0)
5611 {
5612 x = TRUE;
5613#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5614 n = gui.in_use; // gui_mch_browse() works when GUI is running
5615#endif
5616 }
5617 else if (STRICMP(name, "syntax_items") == 0)
5618 {
5619 x = TRUE;
5620#ifdef FEAT_SYN_HL
5621 n = syntax_present(curwin);
5622#endif
5623 }
5624 else if (STRICMP(name, "vcon") == 0)
5625 {
5626 x = TRUE;
5627#ifdef FEAT_VTP
5628 n = is_term_win32() && has_vtp_working();
5629#endif
5630 }
5631 else if (STRICMP(name, "netbeans_enabled") == 0)
5632 {
5633 x = TRUE;
5634#ifdef FEAT_NETBEANS_INTG
5635 n = netbeans_active();
5636#endif
5637 }
5638 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
5639 {
5640 x = TRUE;
5641#ifdef FEAT_MOUSE_GPM
5642 n = gpm_enabled();
5643#endif
5644 }
5645 else if (STRICMP(name, "conpty") == 0)
5646 {
5647 x = TRUE;
5648#if defined(FEAT_TERMINAL) && defined(MSWIN)
5649 n = use_conpty();
5650#endif
5651 }
5652 else if (STRICMP(name, "clipboard_working") == 0)
5653 {
5654 x = TRUE;
5655#ifdef FEAT_CLIPBOARD
5656 n = clip_star.available;
5657#endif
5658 }
5659 }
5660
Bram Moolenaar04637e22020-09-05 18:45:29 +02005661 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[1]))
Bram Moolenaar79296512020-03-22 16:17:14 +01005662 // return whether feature could ever be enabled
5663 rettv->vval.v_number = x;
5664 else
5665 // return whether feature is enabled
5666 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005667}
5668
5669/*
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005670 * Return TRUE if "feature" can change later.
5671 * Also when checking for the feature has side effects, such as loading a DLL.
5672 */
5673 int
5674dynamic_feature(char_u *feature)
5675{
5676 return (feature == NULL
5677#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
5678 || STRICMP(feature, "balloon_multiline") == 0
5679#endif
5680#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5681 || (STRICMP(feature, "browse") == 0 && !gui.in_use)
5682#endif
5683#ifdef VIMDLL
5684 || STRICMP(feature, "filterpipe") == 0
5685#endif
Bram Moolenaar29b281b2020-11-10 20:58:00 +01005686#if defined(FEAT_GUI) && !defined(ALWAYS_USE_GUI) && !defined(VIMDLL)
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005687 // this can only change on Unix where the ":gui" command could be
5688 // used.
5689 || (STRICMP(feature, "gui_running") == 0 && !gui.in_use)
5690#endif
5691#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5692 || STRICMP(feature, "iconv") == 0
5693#endif
5694#ifdef DYNAMIC_LUA
5695 || STRICMP(feature, "lua") == 0
5696#endif
5697#ifdef FEAT_MOUSE_GPM
5698 || (STRICMP(feature, "mouse_gpm_enabled") == 0 && !gpm_enabled())
5699#endif
5700#ifdef DYNAMIC_MZSCHEME
5701 || STRICMP(feature, "mzscheme") == 0
5702#endif
5703#ifdef FEAT_NETBEANS_INTG
5704 || STRICMP(feature, "netbeans_enabled") == 0
5705#endif
5706#ifdef DYNAMIC_PERL
5707 || STRICMP(feature, "perl") == 0
5708#endif
5709#ifdef DYNAMIC_PYTHON
5710 || STRICMP(feature, "python") == 0
5711#endif
5712#ifdef DYNAMIC_PYTHON3
5713 || STRICMP(feature, "python3") == 0
5714#endif
5715#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5716 || STRICMP(feature, "pythonx") == 0
5717#endif
5718#ifdef DYNAMIC_RUBY
5719 || STRICMP(feature, "ruby") == 0
5720#endif
5721#ifdef FEAT_SYN_HL
5722 || STRICMP(feature, "syntax_items") == 0
5723#endif
5724#ifdef DYNAMIC_TCL
5725 || STRICMP(feature, "tcl") == 0
5726#endif
5727 // once "starting" is zero it will stay that way
5728 || (STRICMP(feature, "vim_starting") == 0 && starting != 0)
5729 || STRICMP(feature, "multi_byte_encoding") == 0
5730#if defined(FEAT_TERMINAL) && defined(MSWIN)
5731 || STRICMP(feature, "conpty") == 0
5732#endif
5733 );
5734}
5735
5736/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005737 * "haslocaldir()" function
5738 */
5739 static void
5740f_haslocaldir(typval_T *argvars, typval_T *rettv)
5741{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005742 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005743 win_T *wp = NULL;
5744
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005745 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
5746
5747 // Check for window-local and tab-local directories
5748 if (wp != NULL && wp->w_localdir != NULL)
5749 rettv->vval.v_number = 1;
5750 else if (tp != NULL && tp->tp_localdir != NULL)
5751 rettv->vval.v_number = 2;
5752 else
5753 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005754}
5755
5756/*
5757 * "hasmapto()" function
5758 */
5759 static void
5760f_hasmapto(typval_T *argvars, typval_T *rettv)
5761{
5762 char_u *name;
5763 char_u *mode;
5764 char_u buf[NUMBUFLEN];
5765 int abbr = FALSE;
5766
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005767 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005768 if (argvars[1].v_type == VAR_UNKNOWN)
5769 mode = (char_u *)"nvo";
5770 else
5771 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005772 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005773 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02005774 abbr = (int)tv_get_bool(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005775 }
5776
5777 if (map_to_exists(name, mode, abbr))
5778 rettv->vval.v_number = TRUE;
5779 else
5780 rettv->vval.v_number = FALSE;
5781}
5782
5783/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005784 * "highlightID(name)" function
5785 */
5786 static void
5787f_hlID(typval_T *argvars, typval_T *rettv)
5788{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005789 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005790}
5791
5792/*
5793 * "highlight_exists()" function
5794 */
5795 static void
5796f_hlexists(typval_T *argvars, typval_T *rettv)
5797{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005798 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005799}
5800
5801/*
5802 * "hostname()" function
5803 */
5804 static void
5805f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
5806{
5807 char_u hostname[256];
5808
5809 mch_get_host_name(hostname, 256);
5810 rettv->v_type = VAR_STRING;
5811 rettv->vval.v_string = vim_strsave(hostname);
5812}
5813
5814/*
5815 * iconv() function
5816 */
5817 static void
5818f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
5819{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005820 char_u buf1[NUMBUFLEN];
5821 char_u buf2[NUMBUFLEN];
5822 char_u *from, *to, *str;
5823 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005824
5825 rettv->v_type = VAR_STRING;
5826 rettv->vval.v_string = NULL;
5827
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005828 str = tv_get_string(&argvars[0]);
5829 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
5830 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005831 vimconv.vc_type = CONV_NONE;
5832 convert_setup(&vimconv, from, to);
5833
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005834 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005835 if (vimconv.vc_type == CONV_NONE)
5836 rettv->vval.v_string = vim_strsave(str);
5837 else
5838 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
5839
5840 convert_setup(&vimconv, NULL, NULL);
5841 vim_free(from);
5842 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005843}
5844
5845/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005846 * "index()" function
5847 */
5848 static void
5849f_index(typval_T *argvars, typval_T *rettv)
5850{
5851 list_T *l;
5852 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005853 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005854 long idx = 0;
5855 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005856 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005857
5858 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005859 if (argvars[0].v_type == VAR_BLOB)
5860 {
5861 typval_T tv;
5862 int start = 0;
5863
5864 if (argvars[2].v_type != VAR_UNKNOWN)
5865 {
5866 start = tv_get_number_chk(&argvars[2], &error);
5867 if (error)
5868 return;
5869 }
5870 b = argvars[0].vval.v_blob;
5871 if (b == NULL)
5872 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01005873 if (start < 0)
5874 {
5875 start = blob_len(b) + start;
5876 if (start < 0)
5877 start = 0;
5878 }
5879
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005880 for (idx = start; idx < blob_len(b); ++idx)
5881 {
5882 tv.v_type = VAR_NUMBER;
5883 tv.vval.v_number = blob_get(b, idx);
5884 if (tv_equal(&tv, &argvars[1], ic, FALSE))
5885 {
5886 rettv->vval.v_number = idx;
5887 return;
5888 }
5889 }
5890 return;
5891 }
5892 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005893 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01005894 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005895 return;
5896 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005897
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005898 l = argvars[0].vval.v_list;
5899 if (l != NULL)
5900 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005901 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005902 item = l->lv_first;
5903 if (argvars[2].v_type != VAR_UNKNOWN)
5904 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005905 // Start at specified item. Use the cached index that list_find()
5906 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005907 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005908 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005909 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar6c553f92020-09-02 22:10:34 +02005910 ic = (int)tv_get_bool_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005911 if (error)
5912 item = NULL;
5913 }
5914
5915 for ( ; item != NULL; item = item->li_next, ++idx)
5916 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
5917 {
5918 rettv->vval.v_number = idx;
5919 break;
5920 }
5921 }
5922}
5923
5924static int inputsecret_flag = 0;
5925
5926/*
5927 * "input()" function
5928 * Also handles inputsecret() when inputsecret is set.
5929 */
5930 static void
5931f_input(typval_T *argvars, typval_T *rettv)
5932{
5933 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
5934}
5935
5936/*
5937 * "inputdialog()" function
5938 */
5939 static void
5940f_inputdialog(typval_T *argvars, typval_T *rettv)
5941{
5942#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005943 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005944 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
5945 {
5946 char_u *message;
5947 char_u buf[NUMBUFLEN];
5948 char_u *defstr = (char_u *)"";
5949
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005950 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005951 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005952 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005953 vim_strncpy(IObuff, defstr, IOSIZE - 1);
5954 else
5955 IObuff[0] = NUL;
5956 if (message != NULL && defstr != NULL
5957 && do_dialog(VIM_QUESTION, NULL, message,
5958 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
5959 rettv->vval.v_string = vim_strsave(IObuff);
5960 else
5961 {
5962 if (message != NULL && defstr != NULL
5963 && argvars[1].v_type != VAR_UNKNOWN
5964 && argvars[2].v_type != VAR_UNKNOWN)
5965 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005966 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005967 else
5968 rettv->vval.v_string = NULL;
5969 }
5970 rettv->v_type = VAR_STRING;
5971 }
5972 else
5973#endif
5974 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
5975}
5976
5977/*
5978 * "inputlist()" function
5979 */
5980 static void
5981f_inputlist(typval_T *argvars, typval_T *rettv)
5982{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005983 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005984 listitem_T *li;
5985 int selected;
5986 int mouse_used;
5987
5988#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005989 // While starting up, there is no place to enter text. When running tests
5990 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005991 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005992 return;
5993#endif
5994 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5995 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005996 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005997 return;
5998 }
5999
6000 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006001 msg_row = Rows - 1; // for when 'cmdheight' > 1
6002 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006003 msg_scroll = TRUE;
6004 msg_clr_eos();
6005
Bram Moolenaar50985eb2020-01-27 22:09:39 +01006006 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006007 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02006008 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006009 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01006010 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006011 msg_putchar('\n');
6012 }
6013
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006014 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006015 selected = prompt_for_number(&mouse_used);
6016 if (mouse_used)
6017 selected -= lines_left;
6018
6019 rettv->vval.v_number = selected;
6020}
6021
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006022static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
6023
6024/*
6025 * "inputrestore()" function
6026 */
6027 static void
6028f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
6029{
6030 if (ga_userinput.ga_len > 0)
6031 {
6032 --ga_userinput.ga_len;
6033 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
6034 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006035 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006036 }
6037 else if (p_verbose > 1)
6038 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01006039 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006040 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006041 }
6042}
6043
6044/*
6045 * "inputsave()" function
6046 */
6047 static void
6048f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
6049{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006050 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006051 if (ga_grow(&ga_userinput, 1) == OK)
6052 {
6053 save_typeahead((tasave_T *)(ga_userinput.ga_data)
6054 + ga_userinput.ga_len);
6055 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006056 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006057 }
6058 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006059 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006060}
6061
6062/*
6063 * "inputsecret()" function
6064 */
6065 static void
6066f_inputsecret(typval_T *argvars, typval_T *rettv)
6067{
6068 ++cmdline_star;
6069 ++inputsecret_flag;
6070 f_input(argvars, rettv);
6071 --cmdline_star;
6072 --inputsecret_flag;
6073}
6074
6075/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01006076 * "interrupt()" function
6077 */
6078 static void
6079f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6080{
6081 got_int = TRUE;
6082}
6083
6084/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006085 * "invert(expr)" function
6086 */
6087 static void
6088f_invert(typval_T *argvars, typval_T *rettv)
6089{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006090 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006091}
6092
6093/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006094 * "islocked()" function
6095 */
6096 static void
6097f_islocked(typval_T *argvars, typval_T *rettv)
6098{
6099 lval_T lv;
6100 char_u *end;
6101 dictitem_T *di;
6102
6103 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006104 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01006105 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006106 if (end != NULL && lv.ll_name != NULL)
6107 {
6108 if (*end != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02006109 semsg(_(e_trailing_arg), end);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006110 else
6111 {
6112 if (lv.ll_tv == NULL)
6113 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01006114 di = find_var(lv.ll_name, NULL, TRUE);
6115 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006116 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006117 // Consider a variable locked when:
6118 // 1. the variable itself is locked
6119 // 2. the value of the variable is locked.
6120 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01006121 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
6122 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006123 }
6124 }
6125 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006126 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006127 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006128 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006129 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006130 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006131 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
6132 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006133 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006134 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
6135 }
6136 }
6137
6138 clear_lval(&lv);
6139}
6140
6141#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
6142/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02006143 * "isinf()" function
6144 */
6145 static void
6146f_isinf(typval_T *argvars, typval_T *rettv)
6147{
6148 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
6149 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
6150}
6151
6152/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006153 * "isnan()" function
6154 */
6155 static void
6156f_isnan(typval_T *argvars, typval_T *rettv)
6157{
6158 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
6159 && isnan(argvars[0].vval.v_float);
6160}
6161#endif
6162
6163/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006164 * "last_buffer_nr()" function.
6165 */
6166 static void
6167f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
6168{
6169 int n = 0;
6170 buf_T *buf;
6171
Bram Moolenaar29323592016-07-24 22:04:11 +02006172 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006173 if (n < buf->b_fnum)
6174 n = buf->b_fnum;
6175
6176 rettv->vval.v_number = n;
6177}
6178
6179/*
6180 * "len()" function
6181 */
6182 static void
6183f_len(typval_T *argvars, typval_T *rettv)
6184{
6185 switch (argvars[0].v_type)
6186 {
6187 case VAR_STRING:
6188 case VAR_NUMBER:
6189 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006190 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006191 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01006192 case VAR_BLOB:
6193 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
6194 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006195 case VAR_LIST:
6196 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
6197 break;
6198 case VAR_DICT:
6199 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
6200 break;
6201 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02006202 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006203 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01006204 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006205 case VAR_SPECIAL:
6206 case VAR_FLOAT:
6207 case VAR_FUNC:
6208 case VAR_PARTIAL:
6209 case VAR_JOB:
6210 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006211 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006212 break;
6213 }
6214}
6215
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006216 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01006217libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006218{
6219#ifdef FEAT_LIBCALL
6220 char_u *string_in;
6221 char_u **string_result;
6222 int nr_result;
6223#endif
6224
6225 rettv->v_type = type;
6226 if (type != VAR_NUMBER)
6227 rettv->vval.v_string = NULL;
6228
6229 if (check_restricted() || check_secure())
6230 return;
6231
6232#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006233 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006234 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
6235 {
6236 string_in = NULL;
6237 if (argvars[2].v_type == VAR_STRING)
6238 string_in = argvars[2].vval.v_string;
6239 if (type == VAR_NUMBER)
6240 string_result = NULL;
6241 else
6242 string_result = &rettv->vval.v_string;
6243 if (mch_libcall(argvars[0].vval.v_string,
6244 argvars[1].vval.v_string,
6245 string_in,
6246 argvars[2].vval.v_number,
6247 string_result,
6248 &nr_result) == OK
6249 && type == VAR_NUMBER)
6250 rettv->vval.v_number = nr_result;
6251 }
6252#endif
6253}
6254
6255/*
6256 * "libcall()" function
6257 */
6258 static void
6259f_libcall(typval_T *argvars, typval_T *rettv)
6260{
6261 libcall_common(argvars, rettv, VAR_STRING);
6262}
6263
6264/*
6265 * "libcallnr()" function
6266 */
6267 static void
6268f_libcallnr(typval_T *argvars, typval_T *rettv)
6269{
6270 libcall_common(argvars, rettv, VAR_NUMBER);
6271}
6272
6273/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006274 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006275 */
6276 static void
6277f_line(typval_T *argvars, typval_T *rettv)
6278{
6279 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006280 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006281 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006282 int id;
6283 tabpage_T *tp;
6284 win_T *wp;
6285 win_T *save_curwin;
6286 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006287
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006288 if (argvars[1].v_type != VAR_UNKNOWN)
6289 {
6290 // use window specified in the second argument
6291 id = (int)tv_get_number(&argvars[1]);
6292 wp = win_id2wp_tp(id, &tp);
6293 if (wp != NULL && tp != NULL)
6294 {
6295 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
6296 == OK)
6297 {
6298 check_cursor();
Bram Moolenaar6f02b002021-01-10 20:22:54 +01006299 fp = var2fpos(&argvars[0], TRUE, &fnum, FALSE);
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006300 }
6301 restore_win_noblock(save_curwin, save_curtab, TRUE);
6302 }
6303 }
6304 else
6305 // use current window
Bram Moolenaar6f02b002021-01-10 20:22:54 +01006306 fp = var2fpos(&argvars[0], TRUE, &fnum, FALSE);
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006307
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006308 if (fp != NULL)
6309 lnum = fp->lnum;
6310 rettv->vval.v_number = lnum;
6311}
6312
6313/*
6314 * "line2byte(lnum)" function
6315 */
6316 static void
6317f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
6318{
6319#ifndef FEAT_BYTEOFF
6320 rettv->vval.v_number = -1;
6321#else
6322 linenr_T lnum;
6323
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006324 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006325 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
6326 rettv->vval.v_number = -1;
6327 else
6328 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
6329 if (rettv->vval.v_number >= 0)
6330 ++rettv->vval.v_number;
6331#endif
6332}
6333
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006334#ifdef FEAT_FLOAT
6335/*
6336 * "log()" function
6337 */
6338 static void
6339f_log(typval_T *argvars, typval_T *rettv)
6340{
6341 float_T f = 0.0;
6342
6343 rettv->v_type = VAR_FLOAT;
6344 if (get_float_arg(argvars, &f) == OK)
6345 rettv->vval.v_float = log(f);
6346 else
6347 rettv->vval.v_float = 0.0;
6348}
6349
6350/*
6351 * "log10()" function
6352 */
6353 static void
6354f_log10(typval_T *argvars, typval_T *rettv)
6355{
6356 float_T f = 0.0;
6357
6358 rettv->v_type = VAR_FLOAT;
6359 if (get_float_arg(argvars, &f) == OK)
6360 rettv->vval.v_float = log10(f);
6361 else
6362 rettv->vval.v_float = 0.0;
6363}
6364#endif
6365
6366#ifdef FEAT_LUA
6367/*
6368 * "luaeval()" function
6369 */
6370 static void
6371f_luaeval(typval_T *argvars, typval_T *rettv)
6372{
6373 char_u *str;
6374 char_u buf[NUMBUFLEN];
6375
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006376 if (check_restricted() || check_secure())
6377 return;
6378
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006379 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006380 do_luaeval(str, argvars + 1, rettv);
6381}
6382#endif
6383
6384/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006385 * "maparg()" function
6386 */
6387 static void
6388f_maparg(typval_T *argvars, typval_T *rettv)
6389{
6390 get_maparg(argvars, rettv, TRUE);
6391}
6392
6393/*
6394 * "mapcheck()" function
6395 */
6396 static void
6397f_mapcheck(typval_T *argvars, typval_T *rettv)
6398{
6399 get_maparg(argvars, rettv, FALSE);
6400}
6401
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006402typedef enum
6403{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006404 MATCH_END, // matchend()
6405 MATCH_MATCH, // match()
6406 MATCH_STR, // matchstr()
6407 MATCH_LIST, // matchlist()
6408 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006409} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006410
6411 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006412find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006413{
6414 char_u *str = NULL;
6415 long len = 0;
6416 char_u *expr = NULL;
6417 char_u *pat;
6418 regmatch_T regmatch;
6419 char_u patbuf[NUMBUFLEN];
6420 char_u strbuf[NUMBUFLEN];
6421 char_u *save_cpo;
6422 long start = 0;
6423 long nth = 1;
6424 colnr_T startcol = 0;
6425 int match = 0;
6426 list_T *l = NULL;
6427 listitem_T *li = NULL;
6428 long idx = 0;
6429 char_u *tofree = NULL;
6430
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006431 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006432 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01006433 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006434
6435 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006436 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006437 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006438 // type MATCH_LIST: return empty list when there are no matches.
6439 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006440 if (rettv_list_alloc(rettv) == FAIL)
6441 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006442 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006443 && (list_append_string(rettv->vval.v_list,
6444 (char_u *)"", 0) == FAIL
6445 || list_append_number(rettv->vval.v_list,
6446 (varnumber_T)-1) == FAIL
6447 || list_append_number(rettv->vval.v_list,
6448 (varnumber_T)-1) == FAIL
6449 || list_append_number(rettv->vval.v_list,
6450 (varnumber_T)-1) == FAIL))
6451 {
6452 list_free(rettv->vval.v_list);
6453 rettv->vval.v_list = NULL;
6454 goto theend;
6455 }
6456 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006457 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006458 {
6459 rettv->v_type = VAR_STRING;
6460 rettv->vval.v_string = NULL;
6461 }
6462
6463 if (argvars[0].v_type == VAR_LIST)
6464 {
6465 if ((l = argvars[0].vval.v_list) == NULL)
6466 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006467 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006468 li = l->lv_first;
6469 }
6470 else
6471 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006472 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006473 len = (long)STRLEN(str);
6474 }
6475
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006476 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006477 if (pat == NULL)
6478 goto theend;
6479
6480 if (argvars[2].v_type != VAR_UNKNOWN)
6481 {
6482 int error = FALSE;
6483
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006484 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006485 if (error)
6486 goto theend;
6487 if (l != NULL)
6488 {
6489 li = list_find(l, start);
6490 if (li == NULL)
6491 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006492 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006493 }
6494 else
6495 {
6496 if (start < 0)
6497 start = 0;
6498 if (start > len)
6499 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006500 // When "count" argument is there ignore matches before "start",
6501 // otherwise skip part of the string. Differs when pattern is "^"
6502 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006503 if (argvars[3].v_type != VAR_UNKNOWN)
6504 startcol = start;
6505 else
6506 {
6507 str += start;
6508 len -= start;
6509 }
6510 }
6511
6512 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006513 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006514 if (error)
6515 goto theend;
6516 }
6517
6518 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
6519 if (regmatch.regprog != NULL)
6520 {
6521 regmatch.rm_ic = p_ic;
6522
6523 for (;;)
6524 {
6525 if (l != NULL)
6526 {
6527 if (li == NULL)
6528 {
6529 match = FALSE;
6530 break;
6531 }
6532 vim_free(tofree);
6533 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
6534 if (str == NULL)
6535 break;
6536 }
6537
6538 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
6539
6540 if (match && --nth <= 0)
6541 break;
6542 if (l == NULL && !match)
6543 break;
6544
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006545 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006546 if (l != NULL)
6547 {
6548 li = li->li_next;
6549 ++idx;
6550 }
6551 else
6552 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006553 startcol = (colnr_T)(regmatch.startp[0]
6554 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006555 if (startcol > (colnr_T)len
6556 || str + startcol <= regmatch.startp[0])
6557 {
6558 match = FALSE;
6559 break;
6560 }
6561 }
6562 }
6563
6564 if (match)
6565 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006566 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006567 {
6568 listitem_T *li1 = rettv->vval.v_list->lv_first;
6569 listitem_T *li2 = li1->li_next;
6570 listitem_T *li3 = li2->li_next;
6571 listitem_T *li4 = li3->li_next;
6572
6573 vim_free(li1->li_tv.vval.v_string);
6574 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006575 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006576 li3->li_tv.vval.v_number =
6577 (varnumber_T)(regmatch.startp[0] - expr);
6578 li4->li_tv.vval.v_number =
6579 (varnumber_T)(regmatch.endp[0] - expr);
6580 if (l != NULL)
6581 li2->li_tv.vval.v_number = (varnumber_T)idx;
6582 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006583 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006584 {
6585 int i;
6586
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006587 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006588 for (i = 0; i < NSUBEXP; ++i)
6589 {
6590 if (regmatch.endp[i] == NULL)
6591 {
6592 if (list_append_string(rettv->vval.v_list,
6593 (char_u *)"", 0) == FAIL)
6594 break;
6595 }
6596 else if (list_append_string(rettv->vval.v_list,
6597 regmatch.startp[i],
6598 (int)(regmatch.endp[i] - regmatch.startp[i]))
6599 == FAIL)
6600 break;
6601 }
6602 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006603 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006604 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006605 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006606 if (l != NULL)
6607 copy_tv(&li->li_tv, rettv);
6608 else
6609 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006610 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006611 }
6612 else if (l != NULL)
6613 rettv->vval.v_number = idx;
6614 else
6615 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006616 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006617 rettv->vval.v_number =
6618 (varnumber_T)(regmatch.startp[0] - str);
6619 else
6620 rettv->vval.v_number =
6621 (varnumber_T)(regmatch.endp[0] - str);
6622 rettv->vval.v_number += (varnumber_T)(str - expr);
6623 }
6624 }
6625 vim_regfree(regmatch.regprog);
6626 }
6627
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006628theend:
6629 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006630 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006631 listitem_remove(rettv->vval.v_list,
6632 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006633 vim_free(tofree);
6634 p_cpo = save_cpo;
6635}
6636
6637/*
6638 * "match()" function
6639 */
6640 static void
6641f_match(typval_T *argvars, typval_T *rettv)
6642{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006643 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006644}
6645
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006646/*
6647 * "matchend()" function
6648 */
6649 static void
6650f_matchend(typval_T *argvars, typval_T *rettv)
6651{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006652 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006653}
6654
6655/*
6656 * "matchlist()" function
6657 */
6658 static void
6659f_matchlist(typval_T *argvars, typval_T *rettv)
6660{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006661 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006662}
6663
6664/*
6665 * "matchstr()" function
6666 */
6667 static void
6668f_matchstr(typval_T *argvars, typval_T *rettv)
6669{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006670 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006671}
6672
6673/*
6674 * "matchstrpos()" function
6675 */
6676 static void
6677f_matchstrpos(typval_T *argvars, typval_T *rettv)
6678{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006679 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006680}
6681
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006682 static void
6683max_min(typval_T *argvars, typval_T *rettv, int domax)
6684{
6685 varnumber_T n = 0;
6686 varnumber_T i;
6687 int error = FALSE;
6688
6689 if (argvars[0].v_type == VAR_LIST)
6690 {
6691 list_T *l;
6692 listitem_T *li;
6693
6694 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006695 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006696 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006697 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006698 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006699 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
6700 n = l->lv_u.nonmat.lv_start;
6701 else
6702 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
6703 * l->lv_u.nonmat.lv_stride;
6704 }
6705 else
6706 {
6707 li = l->lv_first;
6708 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006709 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006710 n = tv_get_number_chk(&li->li_tv, &error);
6711 for (;;)
6712 {
6713 li = li->li_next;
6714 if (li == NULL)
6715 break;
6716 i = tv_get_number_chk(&li->li_tv, &error);
6717 if (domax ? i > n : i < n)
6718 n = i;
6719 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006720 }
6721 }
6722 }
6723 }
6724 else if (argvars[0].v_type == VAR_DICT)
6725 {
6726 dict_T *d;
6727 int first = TRUE;
6728 hashitem_T *hi;
6729 int todo;
6730
6731 d = argvars[0].vval.v_dict;
6732 if (d != NULL)
6733 {
6734 todo = (int)d->dv_hashtab.ht_used;
6735 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
6736 {
6737 if (!HASHITEM_EMPTY(hi))
6738 {
6739 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006740 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006741 if (first)
6742 {
6743 n = i;
6744 first = FALSE;
6745 }
6746 else if (domax ? i > n : i < n)
6747 n = i;
6748 }
6749 }
6750 }
6751 }
6752 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006753 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006754 rettv->vval.v_number = error ? 0 : n;
6755}
6756
6757/*
6758 * "max()" function
6759 */
6760 static void
6761f_max(typval_T *argvars, typval_T *rettv)
6762{
6763 max_min(argvars, rettv, TRUE);
6764}
6765
6766/*
6767 * "min()" function
6768 */
6769 static void
6770f_min(typval_T *argvars, typval_T *rettv)
6771{
6772 max_min(argvars, rettv, FALSE);
6773}
6774
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006775#if defined(FEAT_MZSCHEME) || defined(PROTO)
6776/*
6777 * "mzeval()" function
6778 */
6779 static void
6780f_mzeval(typval_T *argvars, typval_T *rettv)
6781{
6782 char_u *str;
6783 char_u buf[NUMBUFLEN];
6784
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006785 if (check_restricted() || check_secure())
6786 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006787 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006788 do_mzeval(str, rettv);
6789}
6790
6791 void
6792mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
6793{
6794 typval_T argvars[3];
6795
6796 argvars[0].v_type = VAR_STRING;
6797 argvars[0].vval.v_string = name;
6798 copy_tv(args, &argvars[1]);
6799 argvars[2].v_type = VAR_UNKNOWN;
6800 f_call(argvars, rettv);
6801 clear_tv(&argvars[1]);
6802}
6803#endif
6804
6805/*
6806 * "nextnonblank()" function
6807 */
6808 static void
6809f_nextnonblank(typval_T *argvars, typval_T *rettv)
6810{
6811 linenr_T lnum;
6812
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006813 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006814 {
6815 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
6816 {
6817 lnum = 0;
6818 break;
6819 }
6820 if (*skipwhite(ml_get(lnum)) != NUL)
6821 break;
6822 }
6823 rettv->vval.v_number = lnum;
6824}
6825
6826/*
6827 * "nr2char()" function
6828 */
6829 static void
6830f_nr2char(typval_T *argvars, typval_T *rettv)
6831{
6832 char_u buf[NUMBUFLEN];
6833
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006834 if (has_mbyte)
6835 {
6836 int utf8 = 0;
6837
6838 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaared6a4302020-09-05 20:29:41 +02006839 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006840 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01006841 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006842 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006843 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006844 }
6845 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006846 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006847 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006848 buf[1] = NUL;
6849 }
6850 rettv->v_type = VAR_STRING;
6851 rettv->vval.v_string = vim_strsave(buf);
6852}
6853
6854/*
6855 * "or(expr, expr)" function
6856 */
6857 static void
6858f_or(typval_T *argvars, typval_T *rettv)
6859{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006860 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
6861 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006862}
6863
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006864#ifdef FEAT_PERL
6865/*
6866 * "perleval()" function
6867 */
6868 static void
6869f_perleval(typval_T *argvars, typval_T *rettv)
6870{
6871 char_u *str;
6872 char_u buf[NUMBUFLEN];
6873
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006874 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006875 do_perleval(str, rettv);
6876}
6877#endif
6878
6879#ifdef FEAT_FLOAT
6880/*
6881 * "pow()" function
6882 */
6883 static void
6884f_pow(typval_T *argvars, typval_T *rettv)
6885{
6886 float_T fx = 0.0, fy = 0.0;
6887
6888 rettv->v_type = VAR_FLOAT;
6889 if (get_float_arg(argvars, &fx) == OK
6890 && get_float_arg(&argvars[1], &fy) == OK)
6891 rettv->vval.v_float = pow(fx, fy);
6892 else
6893 rettv->vval.v_float = 0.0;
6894}
6895#endif
6896
6897/*
6898 * "prevnonblank()" function
6899 */
6900 static void
6901f_prevnonblank(typval_T *argvars, typval_T *rettv)
6902{
6903 linenr_T lnum;
6904
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006905 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006906 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
6907 lnum = 0;
6908 else
6909 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
6910 --lnum;
6911 rettv->vval.v_number = lnum;
6912}
6913
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006914// This dummy va_list is here because:
6915// - passing a NULL pointer doesn't work when va_list isn't a pointer
6916// - locally in the function results in a "used before set" warning
6917// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006918static va_list ap;
6919
6920/*
6921 * "printf()" function
6922 */
6923 static void
6924f_printf(typval_T *argvars, typval_T *rettv)
6925{
6926 char_u buf[NUMBUFLEN];
6927 int len;
6928 char_u *s;
6929 int saved_did_emsg = did_emsg;
6930 char *fmt;
6931
6932 rettv->v_type = VAR_STRING;
6933 rettv->vval.v_string = NULL;
6934
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006935 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006936 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006937 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006938 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006939 if (!did_emsg)
6940 {
6941 s = alloc(len + 1);
6942 if (s != NULL)
6943 {
6944 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006945 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
6946 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006947 }
6948 }
6949 did_emsg |= saved_did_emsg;
6950}
6951
6952/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006953 * "pum_getpos()" function
6954 */
6955 static void
6956f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6957{
6958 if (rettv_dict_alloc(rettv) != OK)
6959 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006960 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006961}
6962
6963/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006964 * "pumvisible()" function
6965 */
6966 static void
6967f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6968{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006969 if (pum_visible())
6970 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006971}
6972
6973#ifdef FEAT_PYTHON3
6974/*
6975 * "py3eval()" function
6976 */
6977 static void
6978f_py3eval(typval_T *argvars, typval_T *rettv)
6979{
6980 char_u *str;
6981 char_u buf[NUMBUFLEN];
6982
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006983 if (check_restricted() || check_secure())
6984 return;
6985
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006986 if (p_pyx == 0)
6987 p_pyx = 3;
6988
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006989 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006990 do_py3eval(str, rettv);
6991}
6992#endif
6993
6994#ifdef FEAT_PYTHON
6995/*
6996 * "pyeval()" function
6997 */
6998 static void
6999f_pyeval(typval_T *argvars, typval_T *rettv)
7000{
7001 char_u *str;
7002 char_u buf[NUMBUFLEN];
7003
Bram Moolenaar8c62a082019-02-08 14:34:10 +01007004 if (check_restricted() || check_secure())
7005 return;
7006
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01007007 if (p_pyx == 0)
7008 p_pyx = 2;
7009
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007010 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007011 do_pyeval(str, rettv);
7012}
7013#endif
7014
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01007015#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
7016/*
7017 * "pyxeval()" function
7018 */
7019 static void
7020f_pyxeval(typval_T *argvars, typval_T *rettv)
7021{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01007022 if (check_restricted() || check_secure())
7023 return;
7024
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01007025# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
7026 init_pyxversion();
7027 if (p_pyx == 2)
7028 f_pyeval(argvars, rettv);
7029 else
7030 f_py3eval(argvars, rettv);
7031# elif defined(FEAT_PYTHON)
7032 f_pyeval(argvars, rettv);
7033# elif defined(FEAT_PYTHON3)
7034 f_py3eval(argvars, rettv);
7035# endif
7036}
7037#endif
7038
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007039static UINT32_T srand_seed_for_testing = 0;
7040static int srand_seed_for_testing_is_used = FALSE;
7041
7042 static void
7043f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
7044{
7045 if (argvars[0].v_type == VAR_UNKNOWN)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007046 srand_seed_for_testing_is_used = FALSE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007047 else
7048 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007049 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
7050 srand_seed_for_testing_is_used = TRUE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007051 }
7052}
7053
7054 static void
7055init_srand(UINT32_T *x)
7056{
7057#ifndef MSWIN
7058 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
7059#endif
7060
7061 if (srand_seed_for_testing_is_used)
7062 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007063 *x = srand_seed_for_testing;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007064 return;
7065 }
7066#ifndef MSWIN
7067 if (dev_urandom_state != FAIL)
7068 {
7069 int fd = open("/dev/urandom", O_RDONLY);
7070 struct {
7071 union {
7072 UINT32_T number;
7073 char bytes[sizeof(UINT32_T)];
7074 } contents;
7075 } buf;
7076
7077 // Attempt reading /dev/urandom.
7078 if (fd == -1)
7079 dev_urandom_state = FAIL;
7080 else
7081 {
7082 buf.contents.number = 0;
7083 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
7084 != sizeof(UINT32_T))
7085 dev_urandom_state = FAIL;
7086 else
7087 {
7088 dev_urandom_state = OK;
7089 *x = buf.contents.number;
7090 }
7091 close(fd);
7092 }
7093 }
7094 if (dev_urandom_state != OK)
7095 // Reading /dev/urandom doesn't work, fall back to time().
7096#endif
7097 *x = vim_time();
7098}
7099
7100#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
7101#define SPLITMIX32(x, z) ( \
7102 z = (x += 0x9e3779b9), \
7103 z = (z ^ (z >> 16)) * 0x85ebca6b, \
7104 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
7105 z ^ (z >> 16) \
7106 )
7107#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
7108 result = ROTL(y * 5, 7) * 9; \
7109 t = y << 9; \
7110 z ^= x; \
7111 w ^= y; \
7112 y ^= z, x ^= w; \
7113 z ^= t; \
7114 w = ROTL(w, 11);
7115
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007116/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007117 * "rand()" function
7118 */
7119 static void
7120f_rand(typval_T *argvars, typval_T *rettv)
7121{
7122 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007123 static UINT32_T gx, gy, gz, gw;
7124 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007125 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar0fd797e2020-11-05 20:46:32 +01007126 UINT32_T x = 0, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007127
7128 if (argvars[0].v_type == VAR_UNKNOWN)
7129 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007130 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007131 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007132 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007133 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007134 init_srand(&x);
7135
7136 gx = SPLITMIX32(x, z);
7137 gy = SPLITMIX32(x, z);
7138 gz = SPLITMIX32(x, z);
7139 gw = SPLITMIX32(x, z);
7140 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007141 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007142
7143 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007144 }
7145 else if (argvars[0].v_type == VAR_LIST)
7146 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007147 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007148 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007149 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007150
7151 lx = list_find(l, 0L);
7152 ly = list_find(l, 1L);
7153 lz = list_find(l, 2L);
7154 lw = list_find(l, 3L);
7155 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
7156 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
7157 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
7158 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
7159 x = (UINT32_T)lx->li_tv.vval.v_number;
7160 y = (UINT32_T)ly->li_tv.vval.v_number;
7161 z = (UINT32_T)lz->li_tv.vval.v_number;
7162 w = (UINT32_T)lw->li_tv.vval.v_number;
7163
7164 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
7165
7166 lx->li_tv.vval.v_number = (varnumber_T)x;
7167 ly->li_tv.vval.v_number = (varnumber_T)y;
7168 lz->li_tv.vval.v_number = (varnumber_T)z;
7169 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007170 }
7171 else
7172 goto theend;
7173
7174 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007175 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007176 return;
7177
7178theend:
7179 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007180 rettv->v_type = VAR_NUMBER;
7181 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007182}
7183
7184/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007185 * "srand()" function
7186 */
7187 static void
7188f_srand(typval_T *argvars, typval_T *rettv)
7189{
7190 UINT32_T x = 0, z;
7191
7192 if (rettv_list_alloc(rettv) == FAIL)
7193 return;
7194 if (argvars[0].v_type == VAR_UNKNOWN)
7195 {
7196 init_srand(&x);
7197 }
7198 else
7199 {
7200 int error = FALSE;
7201
7202 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
7203 if (error)
7204 return;
7205 }
7206
7207 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7208 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7209 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7210 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7211}
7212
7213#undef ROTL
7214#undef SPLITMIX32
7215#undef SHUFFLE_XOSHIRO128STARSTAR
7216
7217/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007218 * "range()" function
7219 */
7220 static void
7221f_range(typval_T *argvars, typval_T *rettv)
7222{
7223 varnumber_T start;
7224 varnumber_T end;
7225 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007226 int error = FALSE;
7227
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007228 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007229 if (argvars[1].v_type == VAR_UNKNOWN)
7230 {
7231 end = start - 1;
7232 start = 0;
7233 }
7234 else
7235 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007236 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007237 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007238 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007239 }
7240
7241 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007242 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007243 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007244 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007245 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007246 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007247 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007248 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007249 list_T *list = rettv->vval.v_list;
7250
7251 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007252 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007253 // be called.
7254 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01007255 list->lv_u.nonmat.lv_start = start;
7256 list->lv_u.nonmat.lv_end = end;
7257 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007258 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007259 }
7260}
7261
7262/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007263 * Materialize "list".
7264 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007265 */
7266 void
7267range_list_materialize(list_T *list)
7268{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007269 varnumber_T start = list->lv_u.nonmat.lv_start;
7270 varnumber_T end = list->lv_u.nonmat.lv_end;
7271 int stride = list->lv_u.nonmat.lv_stride;
7272 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007273
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007274 list->lv_first = NULL;
7275 list->lv_u.mat.lv_last = NULL;
7276 list->lv_len = 0;
7277 list->lv_u.mat.lv_idx_item = NULL;
7278 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
7279 if (list_append_number(list, (varnumber_T)i) == FAIL)
7280 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007281}
7282
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007283/*
7284 * "getreginfo()" function
7285 */
7286 static void
7287f_getreginfo(typval_T *argvars, typval_T *rettv)
7288{
7289 char_u *strregname;
7290 int regname;
7291 char_u buf[NUMBUFLEN + 2];
7292 long reglen = 0;
7293 dict_T *dict;
7294 list_T *list;
7295
7296 if (argvars[0].v_type != VAR_UNKNOWN)
7297 {
7298 strregname = tv_get_string_chk(&argvars[0]);
7299 if (strregname == NULL)
7300 return;
7301 }
7302 else
7303 strregname = get_vim_var_str(VV_REG);
7304
7305 regname = (strregname == NULL ? '"' : *strregname);
7306 if (regname == 0 || regname == '@')
7307 regname = '"';
7308
7309 if (rettv_dict_alloc(rettv) == FAIL)
7310 return;
7311 dict = rettv->vval.v_dict;
7312
7313 list = (list_T *)get_reg_contents(regname, GREG_EXPR_SRC | GREG_LIST);
7314 if (list == NULL)
7315 return;
Bram Moolenaar91639192020-06-29 19:55:58 +02007316 (void)dict_add_list(dict, "regcontents", list);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007317
7318 buf[0] = NUL;
7319 buf[1] = NUL;
7320 switch (get_reg_type(regname, &reglen))
7321 {
7322 case MLINE: buf[0] = 'V'; break;
7323 case MCHAR: buf[0] = 'v'; break;
7324 case MBLOCK:
7325 vim_snprintf((char *)buf, sizeof(buf), "%c%ld", Ctrl_V,
7326 reglen + 1);
7327 break;
7328 }
Bram Moolenaar91639192020-06-29 19:55:58 +02007329 (void)dict_add_string(dict, (char *)"regtype", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007330
7331 buf[0] = get_register_name(get_unname_register());
7332 buf[1] = NUL;
7333 if (regname == '"')
Bram Moolenaar91639192020-06-29 19:55:58 +02007334 (void)dict_add_string(dict, (char *)"points_to", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007335 else
7336 {
7337 dictitem_T *item = dictitem_alloc((char_u *)"isunnamed");
7338
7339 if (item != NULL)
7340 {
7341 item->di_tv.v_type = VAR_SPECIAL;
7342 item->di_tv.vval.v_number = regname == buf[0]
7343 ? VVAL_TRUE : VVAL_FALSE;
Bram Moolenaar91639192020-06-29 19:55:58 +02007344 (void)dict_add(dict, item);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007345 }
7346 }
7347}
7348
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02007349 static void
7350return_register(int regname, typval_T *rettv)
7351{
7352 char_u buf[2] = {0, 0};
7353
7354 buf[0] = (char_u)regname;
7355 rettv->v_type = VAR_STRING;
7356 rettv->vval.v_string = vim_strsave(buf);
7357}
7358
7359/*
7360 * "reg_executing()" function
7361 */
7362 static void
7363f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
7364{
7365 return_register(reg_executing, rettv);
7366}
7367
7368/*
7369 * "reg_recording()" function
7370 */
7371 static void
7372f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
7373{
7374 return_register(reg_recording, rettv);
7375}
7376
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01007377/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007378 * "rename({from}, {to})" function
7379 */
7380 static void
7381f_rename(typval_T *argvars, typval_T *rettv)
7382{
7383 char_u buf[NUMBUFLEN];
7384
7385 if (check_restricted() || check_secure())
7386 rettv->vval.v_number = -1;
7387 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007388 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
7389 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007390}
7391
7392/*
7393 * "repeat()" function
7394 */
7395 static void
7396f_repeat(typval_T *argvars, typval_T *rettv)
7397{
7398 char_u *p;
7399 int n;
7400 int slen;
7401 int len;
7402 char_u *r;
7403 int i;
7404
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007405 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007406 if (argvars[0].v_type == VAR_LIST)
7407 {
7408 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
7409 while (n-- > 0)
7410 if (list_extend(rettv->vval.v_list,
7411 argvars[0].vval.v_list, NULL) == FAIL)
7412 break;
7413 }
7414 else
7415 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007416 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007417 rettv->v_type = VAR_STRING;
7418 rettv->vval.v_string = NULL;
7419
7420 slen = (int)STRLEN(p);
7421 len = slen * n;
7422 if (len <= 0)
7423 return;
7424
7425 r = alloc(len + 1);
7426 if (r != NULL)
7427 {
7428 for (i = 0; i < n; i++)
7429 mch_memmove(r + i * slen, p, (size_t)slen);
7430 r[len] = NUL;
7431 }
7432
7433 rettv->vval.v_string = r;
7434 }
7435}
7436
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007437#define SP_NOMOVE 0x01 // don't move cursor
7438#define SP_REPEAT 0x02 // repeat to find outer pair
7439#define SP_RETCOUNT 0x04 // return matchcount
7440#define SP_SETPCMARK 0x08 // set previous context mark
7441#define SP_START 0x10 // accept match at start position
7442#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
7443#define SP_END 0x40 // leave cursor at end of match
7444#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007445
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007446/*
7447 * Get flags for a search function.
7448 * Possibly sets "p_ws".
7449 * Returns BACKWARD, FORWARD or zero (for an error).
7450 */
7451 static int
7452get_search_arg(typval_T *varp, int *flagsp)
7453{
7454 int dir = FORWARD;
7455 char_u *flags;
7456 char_u nbuf[NUMBUFLEN];
7457 int mask;
7458
7459 if (varp->v_type != VAR_UNKNOWN)
7460 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007461 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007462 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007463 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007464 while (*flags != NUL)
7465 {
7466 switch (*flags)
7467 {
7468 case 'b': dir = BACKWARD; break;
7469 case 'w': p_ws = TRUE; break;
7470 case 'W': p_ws = FALSE; break;
7471 default: mask = 0;
7472 if (flagsp != NULL)
7473 switch (*flags)
7474 {
7475 case 'c': mask = SP_START; break;
7476 case 'e': mask = SP_END; break;
7477 case 'm': mask = SP_RETCOUNT; break;
7478 case 'n': mask = SP_NOMOVE; break;
7479 case 'p': mask = SP_SUBPAT; break;
7480 case 'r': mask = SP_REPEAT; break;
7481 case 's': mask = SP_SETPCMARK; break;
7482 case 'z': mask = SP_COLUMN; break;
7483 }
7484 if (mask == 0)
7485 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007486 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007487 dir = 0;
7488 }
7489 else
7490 *flagsp |= mask;
7491 }
7492 if (dir == 0)
7493 break;
7494 ++flags;
7495 }
7496 }
7497 return dir;
7498}
7499
7500/*
7501 * Shared by search() and searchpos() functions.
7502 */
7503 static int
7504search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
7505{
7506 int flags;
7507 char_u *pat;
7508 pos_T pos;
7509 pos_T save_cursor;
7510 int save_p_ws = p_ws;
7511 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007512 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007513 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007514#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007515 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007516 long time_limit = 0;
7517#endif
7518 int options = SEARCH_KEEP;
7519 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007520 searchit_arg_T sia;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007521 int use_skip = FALSE;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007522 pos_T firstpos;
7523
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007524 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007525 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007526 if (dir == 0)
7527 goto theend;
7528 flags = *flagsp;
7529 if (flags & SP_START)
7530 options |= SEARCH_START;
7531 if (flags & SP_END)
7532 options |= SEARCH_END;
7533 if (flags & SP_COLUMN)
7534 options |= SEARCH_COL;
7535
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007536 // Optional arguments: line number to stop searching, timeout and skip.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007537 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
7538 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007539 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007540 if (lnum_stop < 0)
7541 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007542 if (argvars[3].v_type != VAR_UNKNOWN)
7543 {
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007544#ifdef FEAT_RELTIME
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007545 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007546 if (time_limit < 0)
7547 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007548#endif
Bram Moolenaara9c01042020-06-07 14:50:50 +02007549 use_skip = eval_expr_valid_arg(&argvars[4]);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007550 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007551 }
7552
7553#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007554 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007555 profile_setlimit(time_limit, &tm);
7556#endif
7557
7558 /*
7559 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
7560 * Check to make sure only those flags are set.
7561 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
7562 * flags cannot be set. Check for that condition also.
7563 */
7564 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
7565 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7566 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007567 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007568 goto theend;
7569 }
7570
7571 pos = save_cursor = curwin->w_cursor;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007572 CLEAR_FIELD(firstpos);
Bram Moolenaara80faa82020-04-12 19:37:17 +02007573 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007574 sia.sa_stop_lnum = (linenr_T)lnum_stop;
7575#ifdef FEAT_RELTIME
7576 sia.sa_tm = &tm;
7577#endif
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007578
7579 // Repeat until {skip} returns FALSE.
7580 for (;;)
7581 {
7582 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007583 options, RE_SEARCH, &sia);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007584 // finding the first match again means there is no match where {skip}
7585 // evaluates to zero.
7586 if (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos))
7587 subpatnum = FAIL;
7588
Bram Moolenaara9c01042020-06-07 14:50:50 +02007589 if (subpatnum == FAIL || !use_skip)
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007590 // didn't find it or no skip argument
7591 break;
7592 firstpos = pos;
7593
Bram Moolenaara9c01042020-06-07 14:50:50 +02007594 // If the skip expression matches, ignore this match.
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007595 {
7596 int do_skip;
7597 int err;
7598 pos_T save_pos = curwin->w_cursor;
7599
7600 curwin->w_cursor = pos;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007601 err = FALSE;
7602 do_skip = eval_expr_to_bool(&argvars[4], &err);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007603 curwin->w_cursor = save_pos;
7604 if (err)
7605 {
7606 // Evaluating {skip} caused an error, break here.
7607 subpatnum = FAIL;
7608 break;
7609 }
7610 if (!do_skip)
7611 break;
7612 }
7613 }
7614
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007615 if (subpatnum != FAIL)
7616 {
7617 if (flags & SP_SUBPAT)
7618 retval = subpatnum;
7619 else
7620 retval = pos.lnum;
7621 if (flags & SP_SETPCMARK)
7622 setpcmark();
7623 curwin->w_cursor = pos;
7624 if (match_pos != NULL)
7625 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007626 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007627 match_pos->lnum = pos.lnum;
7628 match_pos->col = pos.col + 1;
7629 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007630 // "/$" will put the cursor after the end of the line, may need to
7631 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007632 check_cursor();
7633 }
7634
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007635 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007636 if (flags & SP_NOMOVE)
7637 curwin->w_cursor = save_cursor;
7638 else
7639 curwin->w_set_curswant = TRUE;
7640theend:
7641 p_ws = save_p_ws;
7642
7643 return retval;
7644}
7645
7646#ifdef FEAT_FLOAT
7647
7648/*
7649 * round() is not in C90, use ceil() or floor() instead.
7650 */
7651 float_T
7652vim_round(float_T f)
7653{
7654 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
7655}
7656
7657/*
7658 * "round({float})" function
7659 */
7660 static void
7661f_round(typval_T *argvars, typval_T *rettv)
7662{
7663 float_T f = 0.0;
7664
7665 rettv->v_type = VAR_FLOAT;
7666 if (get_float_arg(argvars, &f) == OK)
7667 rettv->vval.v_float = vim_round(f);
7668 else
7669 rettv->vval.v_float = 0.0;
7670}
7671#endif
7672
Bram Moolenaare99be0e2019-03-26 22:51:09 +01007673#ifdef FEAT_RUBY
7674/*
7675 * "rubyeval()" function
7676 */
7677 static void
7678f_rubyeval(typval_T *argvars, typval_T *rettv)
7679{
7680 char_u *str;
7681 char_u buf[NUMBUFLEN];
7682
7683 str = tv_get_string_buf(&argvars[0], buf);
7684 do_rubyeval(str, rettv);
7685}
7686#endif
7687
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007688/*
7689 * "screenattr()" function
7690 */
7691 static void
7692f_screenattr(typval_T *argvars, typval_T *rettv)
7693{
7694 int row;
7695 int col;
7696 int c;
7697
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007698 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7699 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007700 if (row < 0 || row >= screen_Rows
7701 || col < 0 || col >= screen_Columns)
7702 c = -1;
7703 else
7704 c = ScreenAttrs[LineOffset[row] + col];
7705 rettv->vval.v_number = c;
7706}
7707
7708/*
7709 * "screenchar()" function
7710 */
7711 static void
7712f_screenchar(typval_T *argvars, typval_T *rettv)
7713{
7714 int row;
7715 int col;
7716 int off;
7717 int c;
7718
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007719 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7720 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007721 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007722 c = -1;
7723 else
7724 {
7725 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007726 if (enc_utf8 && ScreenLinesUC[off] != 0)
7727 c = ScreenLinesUC[off];
7728 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007729 c = ScreenLines[off];
7730 }
7731 rettv->vval.v_number = c;
7732}
7733
7734/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007735 * "screenchars()" function
7736 */
7737 static void
7738f_screenchars(typval_T *argvars, typval_T *rettv)
7739{
7740 int row;
7741 int col;
7742 int off;
7743 int c;
7744 int i;
7745
7746 if (rettv_list_alloc(rettv) == FAIL)
7747 return;
7748 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7749 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7750 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7751 return;
7752
7753 off = LineOffset[row] + col;
7754 if (enc_utf8 && ScreenLinesUC[off] != 0)
7755 c = ScreenLinesUC[off];
7756 else
7757 c = ScreenLines[off];
7758 list_append_number(rettv->vval.v_list, (varnumber_T)c);
7759
7760 if (enc_utf8)
7761
7762 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7763 list_append_number(rettv->vval.v_list,
7764 (varnumber_T)ScreenLinesC[i][off]);
7765}
7766
7767/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007768 * "screencol()" function
7769 *
7770 * First column is 1 to be consistent with virtcol().
7771 */
7772 static void
7773f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
7774{
7775 rettv->vval.v_number = screen_screencol() + 1;
7776}
7777
7778/*
7779 * "screenrow()" function
7780 */
7781 static void
7782f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
7783{
7784 rettv->vval.v_number = screen_screenrow() + 1;
7785}
7786
7787/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007788 * "screenstring()" function
7789 */
7790 static void
7791f_screenstring(typval_T *argvars, typval_T *rettv)
7792{
7793 int row;
7794 int col;
7795 int off;
7796 int c;
7797 int i;
7798 char_u buf[MB_MAXBYTES + 1];
7799 int buflen = 0;
7800
7801 rettv->vval.v_string = NULL;
7802 rettv->v_type = VAR_STRING;
7803
7804 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7805 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7806 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7807 return;
7808
7809 off = LineOffset[row] + col;
7810 if (enc_utf8 && ScreenLinesUC[off] != 0)
7811 c = ScreenLinesUC[off];
7812 else
7813 c = ScreenLines[off];
7814 buflen += mb_char2bytes(c, buf);
7815
7816 if (enc_utf8)
7817 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7818 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
7819
7820 buf[buflen] = NUL;
7821 rettv->vval.v_string = vim_strsave(buf);
7822}
7823
7824/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007825 * "search()" function
7826 */
7827 static void
7828f_search(typval_T *argvars, typval_T *rettv)
7829{
7830 int flags = 0;
7831
7832 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
7833}
7834
7835/*
7836 * "searchdecl()" function
7837 */
7838 static void
7839f_searchdecl(typval_T *argvars, typval_T *rettv)
7840{
Bram Moolenaar30788d32020-09-05 21:35:16 +02007841 int locally = TRUE;
7842 int thisblock = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007843 int error = FALSE;
7844 char_u *name;
7845
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007846 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007847
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007848 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007849 if (argvars[1].v_type != VAR_UNKNOWN)
7850 {
Bram Moolenaar30788d32020-09-05 21:35:16 +02007851 locally = !(int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007852 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar30788d32020-09-05 21:35:16 +02007853 thisblock = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007854 }
7855 if (!error && name != NULL)
7856 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
7857 locally, thisblock, SEARCH_KEEP) == FAIL;
7858}
7859
7860/*
7861 * Used by searchpair() and searchpairpos()
7862 */
7863 static int
7864searchpair_cmn(typval_T *argvars, pos_T *match_pos)
7865{
7866 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01007867 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007868 int save_p_ws = p_ws;
7869 int dir;
7870 int flags = 0;
7871 char_u nbuf1[NUMBUFLEN];
7872 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007873 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007874 long lnum_stop = 0;
7875 long time_limit = 0;
7876
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007877 // Get the three pattern arguments: start, middle, end. Will result in an
7878 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007879 spat = tv_get_string_chk(&argvars[0]);
7880 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
7881 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007882 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007883 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007884
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007885 // Handle the optional fourth argument: flags
7886 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007887 if (dir == 0)
7888 goto theend;
7889
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007890 // Don't accept SP_END or SP_SUBPAT.
7891 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007892 if ((flags & (SP_END | SP_SUBPAT)) != 0
7893 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7894 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007895 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007896 goto theend;
7897 }
7898
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007899 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007900 if (flags & SP_REPEAT)
7901 p_ws = FALSE;
7902
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007903 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007904 if (argvars[3].v_type == VAR_UNKNOWN
7905 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01007906 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007907 else
7908 {
Bram Moolenaara9c01042020-06-07 14:50:50 +02007909 // Type is checked later.
Bram Moolenaar48570482017-10-30 21:48:41 +01007910 skip = &argvars[4];
Bram Moolenaara9c01042020-06-07 14:50:50 +02007911
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007912 if (argvars[5].v_type != VAR_UNKNOWN)
7913 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007914 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007915 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007916 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007917 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007918 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007919 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007920#ifdef FEAT_RELTIME
7921 if (argvars[6].v_type != VAR_UNKNOWN)
7922 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007923 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007924 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007925 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007926 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007927 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007928 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007929 }
7930#endif
7931 }
7932 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007933
7934 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
7935 match_pos, lnum_stop, time_limit);
7936
7937theend:
7938 p_ws = save_p_ws;
7939
7940 return retval;
7941}
7942
7943/*
7944 * "searchpair()" function
7945 */
7946 static void
7947f_searchpair(typval_T *argvars, typval_T *rettv)
7948{
7949 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
7950}
7951
7952/*
7953 * "searchpairpos()" function
7954 */
7955 static void
7956f_searchpairpos(typval_T *argvars, typval_T *rettv)
7957{
7958 pos_T match_pos;
7959 int lnum = 0;
7960 int col = 0;
7961
7962 if (rettv_list_alloc(rettv) == FAIL)
7963 return;
7964
7965 if (searchpair_cmn(argvars, &match_pos) > 0)
7966 {
7967 lnum = match_pos.lnum;
7968 col = match_pos.col;
7969 }
7970
7971 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7972 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7973}
7974
7975/*
7976 * Search for a start/middle/end thing.
7977 * Used by searchpair(), see its documentation for the details.
7978 * Returns 0 or -1 for no match,
7979 */
7980 long
7981do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007982 char_u *spat, // start pattern
7983 char_u *mpat, // middle pattern
7984 char_u *epat, // end pattern
7985 int dir, // BACKWARD or FORWARD
7986 typval_T *skip, // skip expression
7987 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007988 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007989 linenr_T lnum_stop, // stop at this line if not zero
7990 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007991{
7992 char_u *save_cpo;
7993 char_u *pat, *pat2 = NULL, *pat3 = NULL;
7994 long retval = 0;
7995 pos_T pos;
7996 pos_T firstpos;
7997 pos_T foundpos;
7998 pos_T save_cursor;
7999 pos_T save_pos;
8000 int n;
8001 int r;
8002 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01008003 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008004 int err;
8005 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008006#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008007 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008008#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008009
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008010 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008011 save_cpo = p_cpo;
8012 p_cpo = empty_option;
8013
8014#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008015 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008016 profile_setlimit(time_limit, &tm);
8017#endif
8018
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008019 // Make two search patterns: start/end (pat2, for in nested pairs) and
8020 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02008021 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
8022 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008023 if (pat2 == NULL || pat3 == NULL)
8024 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01008025 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008026 if (*mpat == NUL)
8027 STRCPY(pat3, pat2);
8028 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01008029 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008030 spat, epat, mpat);
8031 if (flags & SP_START)
8032 options |= SEARCH_START;
8033
Bram Moolenaar48570482017-10-30 21:48:41 +01008034 if (skip != NULL)
Bram Moolenaara9c01042020-06-07 14:50:50 +02008035 use_skip = eval_expr_valid_arg(skip);
Bram Moolenaar48570482017-10-30 21:48:41 +01008036
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008037 save_cursor = curwin->w_cursor;
8038 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008039 CLEAR_POS(&firstpos);
8040 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008041 pat = pat3;
8042 for (;;)
8043 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008044 searchit_arg_T sia;
8045
Bram Moolenaara80faa82020-04-12 19:37:17 +02008046 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008047 sia.sa_stop_lnum = lnum_stop;
8048#ifdef FEAT_RELTIME
8049 sia.sa_tm = &tm;
8050#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01008051 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008052 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008053 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008054 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008055 break;
8056
8057 if (firstpos.lnum == 0)
8058 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008059 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008060 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008061 // Found the same position again. Can happen with a pattern that
8062 // has "\zs" at the end and searching backwards. Advance one
8063 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008064 if (dir == BACKWARD)
8065 decl(&pos);
8066 else
8067 incl(&pos);
8068 }
8069 foundpos = pos;
8070
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008071 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008072 options &= ~SEARCH_START;
8073
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008074 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01008075 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008076 {
8077 save_pos = curwin->w_cursor;
8078 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01008079 err = FALSE;
8080 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008081 curwin->w_cursor = save_pos;
8082 if (err)
8083 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008084 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008085 curwin->w_cursor = save_cursor;
8086 retval = -1;
8087 break;
8088 }
8089 if (r)
8090 continue;
8091 }
8092
8093 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
8094 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008095 // Found end when searching backwards or start when searching
8096 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008097 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008098 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008099 }
8100 else
8101 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008102 // Found end when searching forward or start when searching
8103 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008104 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008105 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008106 }
8107
8108 if (nest == 0)
8109 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008110 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008111 if (flags & SP_RETCOUNT)
8112 ++retval;
8113 else
8114 retval = pos.lnum;
8115 if (flags & SP_SETPCMARK)
8116 setpcmark();
8117 curwin->w_cursor = pos;
8118 if (!(flags & SP_REPEAT))
8119 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008120 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008121 }
8122 }
8123
8124 if (match_pos != NULL)
8125 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008126 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008127 match_pos->lnum = curwin->w_cursor.lnum;
8128 match_pos->col = curwin->w_cursor.col + 1;
8129 }
8130
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008131 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008132 if ((flags & SP_NOMOVE) || retval == 0)
8133 curwin->w_cursor = save_cursor;
8134
8135theend:
8136 vim_free(pat2);
8137 vim_free(pat3);
8138 if (p_cpo == empty_option)
8139 p_cpo = save_cpo;
8140 else
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008141 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008142 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008143 // If it's still empty it was changed and restored, need to restore in
8144 // the complicated way.
8145 if (*p_cpo == NUL)
8146 set_option_value((char_u *)"cpo", 0L, save_cpo, 0);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008147 free_string_option(save_cpo);
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008148 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008149
8150 return retval;
8151}
8152
8153/*
8154 * "searchpos()" function
8155 */
8156 static void
8157f_searchpos(typval_T *argvars, typval_T *rettv)
8158{
8159 pos_T match_pos;
8160 int lnum = 0;
8161 int col = 0;
8162 int n;
8163 int flags = 0;
8164
8165 if (rettv_list_alloc(rettv) == FAIL)
8166 return;
8167
8168 n = search_cmn(argvars, &match_pos, &flags);
8169 if (n > 0)
8170 {
8171 lnum = match_pos.lnum;
8172 col = match_pos.col;
8173 }
8174
8175 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
8176 list_append_number(rettv->vval.v_list, (varnumber_T)col);
8177 if (flags & SP_SUBPAT)
8178 list_append_number(rettv->vval.v_list, (varnumber_T)n);
8179}
8180
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008181/*
8182 * Set the cursor or mark position.
8183 * If 'charpos' is TRUE, then use the column number as a character offet.
8184 * Otherwise use the column number as a byte offset.
8185 */
8186 static void
8187set_position(typval_T *argvars, typval_T *rettv, int charpos)
8188{
8189 pos_T pos;
8190 int fnum;
8191 char_u *name;
8192 colnr_T curswant = -1;
8193
8194 rettv->vval.v_number = -1;
8195
8196 name = tv_get_string_chk(argvars);
8197 if (name != NULL)
8198 {
8199 if (list2fpos(&argvars[1], &pos, &fnum, &curswant, charpos) == OK)
8200 {
8201 if (pos.col != MAXCOL && --pos.col < 0)
8202 pos.col = 0;
8203 if ((name[0] == '.' && name[1] == NUL))
8204 {
8205 // set cursor; "fnum" is ignored
8206 curwin->w_cursor = pos;
8207 if (curswant >= 0)
8208 {
8209 curwin->w_curswant = curswant - 1;
8210 curwin->w_set_curswant = FALSE;
8211 }
8212 check_cursor();
8213 rettv->vval.v_number = 0;
8214 }
8215 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
8216 {
8217 // set mark
8218 if (setmark_pos(name[1], &pos, fnum) == OK)
8219 rettv->vval.v_number = 0;
8220 }
8221 else
8222 emsg(_(e_invarg));
8223 }
8224 }
8225}
8226/*
8227 * "setcharpos()" function
8228 */
8229 static void
8230f_setcharpos(typval_T *argvars, typval_T *rettv)
8231{
8232 set_position(argvars, rettv, TRUE);
8233}
8234
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008235 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008236f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
8237{
8238 dict_T *d;
8239 dictitem_T *di;
8240 char_u *csearch;
8241
8242 if (argvars[0].v_type != VAR_DICT)
8243 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008244 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008245 return;
8246 }
8247
8248 if ((d = argvars[0].vval.v_dict) != NULL)
8249 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01008250 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008251 if (csearch != NULL)
8252 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008253 if (enc_utf8)
8254 {
8255 int pcc[MAX_MCO];
8256 int c = utfc_ptr2char(csearch, pcc);
8257
8258 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
8259 }
8260 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008261 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02008262 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008263 }
8264
8265 di = dict_find(d, (char_u *)"forward", -1);
8266 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008267 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008268 ? FORWARD : BACKWARD);
8269
8270 di = dict_find(d, (char_u *)"until", -1);
8271 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008272 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008273 }
8274}
8275
8276/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008277 * "setcursorcharpos" function
8278 */
8279 static void
8280f_setcursorcharpos(typval_T *argvars, typval_T *rettv UNUSED)
8281{
8282 set_cursorpos(argvars, rettv, TRUE);
8283}
8284
8285/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02008286 * "setenv()" function
8287 */
8288 static void
8289f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
8290{
8291 char_u namebuf[NUMBUFLEN];
8292 char_u valbuf[NUMBUFLEN];
8293 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
8294
8295 if (argvars[1].v_type == VAR_SPECIAL
8296 && argvars[1].vval.v_number == VVAL_NULL)
8297 vim_unsetenv(name);
8298 else
8299 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
8300}
8301
8302/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008303 * "setfperm({fname}, {mode})" function
8304 */
8305 static void
8306f_setfperm(typval_T *argvars, typval_T *rettv)
8307{
8308 char_u *fname;
8309 char_u modebuf[NUMBUFLEN];
8310 char_u *mode_str;
8311 int i;
8312 int mask;
8313 int mode = 0;
8314
8315 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008316 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008317 if (fname == NULL)
8318 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008319 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008320 if (mode_str == NULL)
8321 return;
8322 if (STRLEN(mode_str) != 9)
8323 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008324 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008325 return;
8326 }
8327
8328 mask = 1;
8329 for (i = 8; i >= 0; --i)
8330 {
8331 if (mode_str[i] != '-')
8332 mode |= mask;
8333 mask = mask << 1;
8334 }
8335 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
8336}
8337
8338/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008339 * "setpos()" function
8340 */
8341 static void
8342f_setpos(typval_T *argvars, typval_T *rettv)
8343{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008344 set_position(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008345}
8346
8347/*
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008348 * Translate a register type string to the yank type and block length
8349 */
8350 static int
8351get_yank_type(char_u **pp, char_u *yank_type, long *block_len)
8352{
8353 char_u *stropt = *pp;
8354 switch (*stropt)
8355 {
8356 case 'v': case 'c': // character-wise selection
8357 *yank_type = MCHAR;
8358 break;
8359 case 'V': case 'l': // line-wise selection
8360 *yank_type = MLINE;
8361 break;
8362 case 'b': case Ctrl_V: // block-wise selection
8363 *yank_type = MBLOCK;
8364 if (VIM_ISDIGIT(stropt[1]))
8365 {
8366 ++stropt;
8367 *block_len = getdigits(&stropt) - 1;
8368 --stropt;
8369 }
8370 break;
8371 default:
8372 return FAIL;
8373 }
8374 *pp = stropt;
8375 return OK;
8376}
8377
8378/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008379 * "setreg()" function
8380 */
8381 static void
8382f_setreg(typval_T *argvars, typval_T *rettv)
8383{
8384 int regname;
8385 char_u *strregname;
8386 char_u *stropt;
8387 char_u *strval;
8388 int append;
8389 char_u yank_type;
8390 long block_len;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008391 typval_T *regcontents;
8392 int pointreg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008393
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008394 pointreg = 0;
8395 regcontents = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008396 block_len = -1;
8397 yank_type = MAUTO;
8398 append = FALSE;
8399
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008400 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008401 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008402
8403 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008404 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008405 regname = *strregname;
8406 if (regname == 0 || regname == '@')
8407 regname = '"';
8408
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008409 if (argvars[1].v_type == VAR_DICT)
8410 {
8411 dict_T *d = argvars[1].vval.v_dict;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008412 dictitem_T *di;
8413
8414 if (d == NULL || d->dv_hashtab.ht_used == 0)
8415 {
8416 // Empty dict, clear the register (like setreg(0, []))
8417 char_u *lstval[2] = {NULL, NULL};
8418 write_reg_contents_lst(regname, lstval, 0, FALSE, MAUTO, -1);
8419 return;
8420 }
8421
8422 di = dict_find(d, (char_u *)"regcontents", -1);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008423 if (di != NULL)
8424 regcontents = &di->di_tv;
8425
8426 stropt = dict_get_string(d, (char_u *)"regtype", FALSE);
8427 if (stropt != NULL)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008428 {
8429 int ret = get_yank_type(&stropt, &yank_type, &block_len);
8430
8431 if (ret == FAIL || *++stropt != NUL)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008432 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008433 semsg(_(e_invargval), "value");
8434 return;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008435 }
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008436 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008437
8438 if (regname == '"')
8439 {
8440 stropt = dict_get_string(d, (char_u *)"points_to", FALSE);
8441 if (stropt != NULL)
8442 {
8443 pointreg = *stropt;
8444 regname = pointreg;
8445 }
8446 }
Bram Moolenaar6a950582020-08-28 16:39:33 +02008447 else if (dict_get_bool(d, (char_u *)"isunnamed", -1) > 0)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008448 pointreg = regname;
8449 }
8450 else
8451 regcontents = &argvars[1];
8452
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008453 if (argvars[2].v_type != VAR_UNKNOWN)
8454 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008455 if (yank_type != MAUTO)
8456 {
8457 semsg(_(e_toomanyarg), "setreg");
8458 return;
8459 }
8460
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008461 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008462 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008463 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008464 for (; *stropt != NUL; ++stropt)
8465 switch (*stropt)
8466 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008467 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008468 append = TRUE;
8469 break;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008470 default:
8471 get_yank_type(&stropt, &yank_type, &block_len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008472 }
8473 }
8474
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008475 if (regcontents && regcontents->v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008476 {
8477 char_u **lstval;
8478 char_u **allocval;
8479 char_u buf[NUMBUFLEN];
8480 char_u **curval;
8481 char_u **curallocval;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008482 list_T *ll = regcontents->vval.v_list;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008483 listitem_T *li;
8484 int len;
8485
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008486 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008487 len = ll == NULL ? 0 : ll->lv_len;
8488
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008489 // First half: use for pointers to result lines; second half: use for
8490 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02008491 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008492 if (lstval == NULL)
8493 return;
8494 curval = lstval;
8495 allocval = lstval + len + 2;
8496 curallocval = allocval;
8497
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008498 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008499 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02008500 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02008501 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008502 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008503 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008504 if (strval == NULL)
8505 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008506 if (strval == buf)
8507 {
8508 // Need to make a copy, next tv_get_string_buf_chk() will
8509 // overwrite the string.
8510 strval = vim_strsave(buf);
8511 if (strval == NULL)
8512 goto free_lstval;
8513 *curallocval++ = strval;
8514 }
8515 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008516 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008517 }
8518 *curval++ = NULL;
8519
8520 write_reg_contents_lst(regname, lstval, -1,
8521 append, yank_type, block_len);
8522free_lstval:
8523 while (curallocval > allocval)
8524 vim_free(*--curallocval);
8525 vim_free(lstval);
8526 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008527 else if (regcontents)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008528 {
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008529 strval = tv_get_string_chk(regcontents);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008530 if (strval == NULL)
8531 return;
8532 write_reg_contents_ex(regname, strval, -1,
8533 append, yank_type, block_len);
8534 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008535 if (pointreg != 0)
8536 get_yank_register(pointreg, TRUE);
8537
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008538 rettv->vval.v_number = 0;
8539}
8540
8541/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008542 * "settagstack()" function
8543 */
8544 static void
8545f_settagstack(typval_T *argvars, typval_T *rettv)
8546{
8547 static char *e_invact2 = N_("E962: Invalid action: '%s'");
8548 win_T *wp;
8549 dict_T *d;
8550 int action = 'r';
8551
8552 rettv->vval.v_number = -1;
8553
8554 // first argument: window number or id
8555 wp = find_win_by_nr_or_id(&argvars[0]);
8556 if (wp == NULL)
8557 return;
8558
8559 // second argument: dict with items to set in the tag stack
8560 if (argvars[1].v_type != VAR_DICT)
8561 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008562 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008563 return;
8564 }
8565 d = argvars[1].vval.v_dict;
8566 if (d == NULL)
8567 return;
8568
8569 // third argument: action - 'a' for append and 'r' for replace.
8570 // default is to replace the stack.
8571 if (argvars[2].v_type == VAR_UNKNOWN)
8572 action = 'r';
8573 else if (argvars[2].v_type == VAR_STRING)
8574 {
8575 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008576 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008577 if (actstr == NULL)
8578 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01008579 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
8580 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008581 action = *actstr;
8582 else
8583 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008584 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008585 return;
8586 }
8587 }
8588 else
8589 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008590 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008591 return;
8592 }
8593
8594 if (set_tagstack(wp, d, action) == OK)
8595 rettv->vval.v_number = 0;
8596}
8597
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008598#ifdef FEAT_CRYPT
8599/*
8600 * "sha256({string})" function
8601 */
8602 static void
8603f_sha256(typval_T *argvars, typval_T *rettv)
8604{
8605 char_u *p;
8606
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008607 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008608 rettv->vval.v_string = vim_strsave(
8609 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
8610 rettv->v_type = VAR_STRING;
8611}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008612#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008613
8614/*
8615 * "shellescape({string})" function
8616 */
8617 static void
8618f_shellescape(typval_T *argvars, typval_T *rettv)
8619{
Bram Moolenaar20615522017-06-05 18:46:26 +02008620 int do_special = non_zero_arg(&argvars[1]);
8621
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008622 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008623 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008624 rettv->v_type = VAR_STRING;
8625}
8626
8627/*
8628 * shiftwidth() function
8629 */
8630 static void
8631f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
8632{
Bram Moolenaarf9514162018-11-22 03:08:29 +01008633 rettv->vval.v_number = 0;
8634
8635 if (argvars[0].v_type != VAR_UNKNOWN)
8636 {
8637 long col;
8638
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008639 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01008640 if (col < 0)
8641 return; // type error; errmsg already given
8642#ifdef FEAT_VARTABS
8643 rettv->vval.v_number = get_sw_value_col(curbuf, col);
8644 return;
8645#endif
8646 }
8647
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008648 rettv->vval.v_number = get_sw_value(curbuf);
8649}
8650
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008651#ifdef FEAT_FLOAT
8652/*
8653 * "sin()" function
8654 */
8655 static void
8656f_sin(typval_T *argvars, typval_T *rettv)
8657{
8658 float_T f = 0.0;
8659
8660 rettv->v_type = VAR_FLOAT;
8661 if (get_float_arg(argvars, &f) == OK)
8662 rettv->vval.v_float = sin(f);
8663 else
8664 rettv->vval.v_float = 0.0;
8665}
8666
8667/*
8668 * "sinh()" function
8669 */
8670 static void
8671f_sinh(typval_T *argvars, typval_T *rettv)
8672{
8673 float_T f = 0.0;
8674
8675 rettv->v_type = VAR_FLOAT;
8676 if (get_float_arg(argvars, &f) == OK)
8677 rettv->vval.v_float = sinh(f);
8678 else
8679 rettv->vval.v_float = 0.0;
8680}
8681#endif
8682
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008683/*
8684 * "soundfold({word})" function
8685 */
8686 static void
8687f_soundfold(typval_T *argvars, typval_T *rettv)
8688{
8689 char_u *s;
8690
8691 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008692 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008693#ifdef FEAT_SPELL
8694 rettv->vval.v_string = eval_soundfold(s);
8695#else
8696 rettv->vval.v_string = vim_strsave(s);
8697#endif
8698}
8699
8700/*
8701 * "spellbadword()" function
8702 */
8703 static void
8704f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
8705{
8706 char_u *word = (char_u *)"";
8707 hlf_T attr = HLF_COUNT;
8708 int len = 0;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008709#ifdef FEAT_SPELL
8710 int wo_spell_save = curwin->w_p_spell;
8711
8712 if (!curwin->w_p_spell)
8713 {
8714 did_set_spelllang(curwin);
8715 curwin->w_p_spell = TRUE;
8716 }
8717
8718 if (*curwin->w_s->b_p_spl == NUL)
8719 {
8720 emsg(_(e_no_spell));
8721 curwin->w_p_spell = wo_spell_save;
8722 return;
8723 }
8724#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008725
8726 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008727 {
8728#ifdef FEAT_SPELL
8729 curwin->w_p_spell = wo_spell_save;
8730#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008731 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008732 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008733
8734#ifdef FEAT_SPELL
8735 if (argvars[0].v_type == VAR_UNKNOWN)
8736 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008737 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008738 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
8739 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008740 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008741 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008742 curwin->w_set_curswant = TRUE;
8743 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008744 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008745 else if (*curbuf->b_s.b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008746 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008747 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008748 int capcol = -1;
8749
8750 if (str != NULL)
8751 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008752 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008753 while (*str != NUL)
8754 {
8755 len = spell_check(curwin, str, &attr, &capcol, FALSE);
8756 if (attr != HLF_COUNT)
8757 {
8758 word = str;
8759 break;
8760 }
8761 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02008762 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02008763 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008764 }
8765 }
8766 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008767 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008768#endif
8769
8770 list_append_string(rettv->vval.v_list, word, len);
8771 list_append_string(rettv->vval.v_list, (char_u *)(
8772 attr == HLF_SPB ? "bad" :
8773 attr == HLF_SPR ? "rare" :
8774 attr == HLF_SPL ? "local" :
8775 attr == HLF_SPC ? "caps" :
8776 ""), -1);
8777}
8778
8779/*
8780 * "spellsuggest()" function
8781 */
8782 static void
8783f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
8784{
8785#ifdef FEAT_SPELL
8786 char_u *str;
8787 int typeerr = FALSE;
8788 int maxcount;
8789 garray_T ga;
8790 int i;
8791 listitem_T *li;
8792 int need_capital = FALSE;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008793 int wo_spell_save = curwin->w_p_spell;
8794
8795 if (!curwin->w_p_spell)
8796 {
8797 did_set_spelllang(curwin);
8798 curwin->w_p_spell = TRUE;
8799 }
8800
8801 if (*curwin->w_s->b_p_spl == NUL)
8802 {
8803 emsg(_(e_no_spell));
8804 curwin->w_p_spell = wo_spell_save;
8805 return;
8806 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008807#endif
8808
8809 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008810 {
8811#ifdef FEAT_SPELL
8812 curwin->w_p_spell = wo_spell_save;
8813#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008814 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008815 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008816
8817#ifdef FEAT_SPELL
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008818 if (*curwin->w_s->b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008819 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008820 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008821 if (argvars[1].v_type != VAR_UNKNOWN)
8822 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008823 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008824 if (maxcount <= 0)
8825 return;
8826 if (argvars[2].v_type != VAR_UNKNOWN)
8827 {
Bram Moolenaar7c27f332020-09-05 22:45:55 +02008828 need_capital = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008829 if (typeerr)
8830 return;
8831 }
8832 }
8833 else
8834 maxcount = 25;
8835
8836 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
8837
8838 for (i = 0; i < ga.ga_len; ++i)
8839 {
8840 str = ((char_u **)ga.ga_data)[i];
8841
8842 li = listitem_alloc();
8843 if (li == NULL)
8844 vim_free(str);
8845 else
8846 {
8847 li->li_tv.v_type = VAR_STRING;
8848 li->li_tv.v_lock = 0;
8849 li->li_tv.vval.v_string = str;
8850 list_append(rettv->vval.v_list, li);
8851 }
8852 }
8853 ga_clear(&ga);
8854 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008855 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008856#endif
8857}
8858
8859 static void
8860f_split(typval_T *argvars, typval_T *rettv)
8861{
8862 char_u *str;
8863 char_u *end;
8864 char_u *pat = NULL;
8865 regmatch_T regmatch;
8866 char_u patbuf[NUMBUFLEN];
8867 char_u *save_cpo;
8868 int match;
8869 colnr_T col = 0;
8870 int keepempty = FALSE;
8871 int typeerr = FALSE;
8872
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008873 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008874 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008875 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008876
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008877 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008878 if (argvars[1].v_type != VAR_UNKNOWN)
8879 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008880 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008881 if (pat == NULL)
8882 typeerr = TRUE;
8883 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02008884 keepempty = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008885 }
8886 if (pat == NULL || *pat == NUL)
8887 pat = (char_u *)"[\\x01- ]\\+";
8888
8889 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008890 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008891 if (typeerr)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008892 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008893
8894 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
8895 if (regmatch.regprog != NULL)
8896 {
8897 regmatch.rm_ic = FALSE;
8898 while (*str != NUL || keepempty)
8899 {
8900 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008901 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008902 else
8903 match = vim_regexec_nl(&regmatch, str, col);
8904 if (match)
8905 end = regmatch.startp[0];
8906 else
8907 end = str + STRLEN(str);
8908 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
8909 && *str != NUL && match && end < regmatch.endp[0]))
8910 {
8911 if (list_append_string(rettv->vval.v_list, str,
8912 (int)(end - str)) == FAIL)
8913 break;
8914 }
8915 if (!match)
8916 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01008917 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008918 if (regmatch.endp[0] > str)
8919 col = 0;
8920 else
Bram Moolenaar13505972019-01-24 15:04:48 +01008921 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008922 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008923 str = regmatch.endp[0];
8924 }
8925
8926 vim_regfree(regmatch.regprog);
8927 }
8928
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008929theend:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008930 p_cpo = save_cpo;
8931}
8932
8933#ifdef FEAT_FLOAT
8934/*
8935 * "sqrt()" function
8936 */
8937 static void
8938f_sqrt(typval_T *argvars, typval_T *rettv)
8939{
8940 float_T f = 0.0;
8941
8942 rettv->v_type = VAR_FLOAT;
8943 if (get_float_arg(argvars, &f) == OK)
8944 rettv->vval.v_float = sqrt(f);
8945 else
8946 rettv->vval.v_float = 0.0;
8947}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008948#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008949
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008950#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01008951/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008952 * "str2float()" function
8953 */
8954 static void
8955f_str2float(typval_T *argvars, typval_T *rettv)
8956{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008957 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008958 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008959
Bram Moolenaar08243d22017-01-10 16:12:29 +01008960 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008961 p = skipwhite(p + 1);
8962 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01008963 if (isneg)
8964 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008965 rettv->v_type = VAR_FLOAT;
8966}
8967#endif
8968
8969/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02008970 * "str2list()" function
8971 */
8972 static void
8973f_str2list(typval_T *argvars, typval_T *rettv)
8974{
8975 char_u *p;
8976 int utf8 = FALSE;
8977
8978 if (rettv_list_alloc(rettv) == FAIL)
8979 return;
8980
8981 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaara48f7862020-09-05 20:16:57 +02008982 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar9d401282019-04-06 13:18:12 +02008983
8984 p = tv_get_string(&argvars[0]);
8985
8986 if (has_mbyte || utf8)
8987 {
8988 int (*ptr2len)(char_u *);
8989 int (*ptr2char)(char_u *);
8990
8991 if (utf8 || enc_utf8)
8992 {
8993 ptr2len = utf_ptr2len;
8994 ptr2char = utf_ptr2char;
8995 }
8996 else
8997 {
8998 ptr2len = mb_ptr2len;
8999 ptr2char = mb_ptr2char;
9000 }
9001
9002 for ( ; *p != NUL; p += (*ptr2len)(p))
9003 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
9004 }
9005 else
9006 for ( ; *p != NUL; ++p)
9007 list_append_number(rettv->vval.v_list, *p);
9008}
9009
9010/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009011 * "str2nr()" function
9012 */
9013 static void
9014f_str2nr(typval_T *argvars, typval_T *rettv)
9015{
9016 int base = 10;
9017 char_u *p;
9018 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009019 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01009020 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009021
9022 if (argvars[1].v_type != VAR_UNKNOWN)
9023 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009024 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009025 if (base != 2 && base != 8 && base != 10 && base != 16)
9026 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009027 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009028 return;
9029 }
Bram Moolenaar3986b942020-09-06 16:09:04 +02009030 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[2]))
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009031 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009032 }
9033
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009034 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01009035 isneg = (*p == '-');
9036 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009037 p = skipwhite(p + 1);
9038 switch (base)
9039 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009040 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02009041 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009042 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009043 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02009044 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
9045 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01009046 if (isneg)
9047 rettv->vval.v_number = -n;
9048 else
9049 rettv->vval.v_number = n;
9050
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009051}
9052
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009053/*
9054 * "strgetchar()" function
9055 */
9056 static void
9057f_strgetchar(typval_T *argvars, typval_T *rettv)
9058{
9059 char_u *str;
9060 int len;
9061 int error = FALSE;
9062 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01009063 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009064
9065 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009066 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009067 if (str == NULL)
9068 return;
9069 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009070 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009071 if (error)
9072 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009073
Bram Moolenaar13505972019-01-24 15:04:48 +01009074 while (charidx >= 0 && byteidx < len)
9075 {
9076 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009077 {
Bram Moolenaar13505972019-01-24 15:04:48 +01009078 rettv->vval.v_number = mb_ptr2char(str + byteidx);
9079 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009080 }
Bram Moolenaar13505972019-01-24 15:04:48 +01009081 --charidx;
9082 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009083 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009084}
9085
9086/*
9087 * "stridx()" function
9088 */
9089 static void
9090f_stridx(typval_T *argvars, typval_T *rettv)
9091{
9092 char_u buf[NUMBUFLEN];
9093 char_u *needle;
9094 char_u *haystack;
9095 char_u *save_haystack;
9096 char_u *pos;
9097 int start_idx;
9098
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009099 needle = tv_get_string_chk(&argvars[1]);
9100 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009101 rettv->vval.v_number = -1;
9102 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009103 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009104
9105 if (argvars[2].v_type != VAR_UNKNOWN)
9106 {
9107 int error = FALSE;
9108
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009109 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009110 if (error || start_idx >= (int)STRLEN(haystack))
9111 return;
9112 if (start_idx >= 0)
9113 haystack += start_idx;
9114 }
9115
9116 pos = (char_u *)strstr((char *)haystack, (char *)needle);
9117 if (pos != NULL)
9118 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
9119}
9120
9121/*
9122 * "string()" function
9123 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01009124 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009125f_string(typval_T *argvars, typval_T *rettv)
9126{
9127 char_u *tofree;
9128 char_u numbuf[NUMBUFLEN];
9129
9130 rettv->v_type = VAR_STRING;
9131 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
9132 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009133 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009134 if (rettv->vval.v_string != NULL && tofree == NULL)
9135 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
9136}
9137
9138/*
9139 * "strlen()" function
9140 */
9141 static void
9142f_strlen(typval_T *argvars, typval_T *rettv)
9143{
9144 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009145 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009146}
9147
9148/*
9149 * "strchars()" function
9150 */
9151 static void
9152f_strchars(typval_T *argvars, typval_T *rettv)
9153{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009154 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar3986b942020-09-06 16:09:04 +02009155 int skipcc = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009156 varnumber_T len = 0;
9157 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009158
9159 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02009160 skipcc = (int)tv_get_bool(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009161 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02009162 semsg(_(e_using_number_as_bool_nr), skipcc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009163 else
9164 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009165 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
9166 while (*s != NUL)
9167 {
9168 func_mb_ptr2char_adv(&s);
9169 ++len;
9170 }
9171 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009172 }
9173}
9174
9175/*
9176 * "strdisplaywidth()" function
9177 */
9178 static void
9179f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
9180{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009181 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009182 int col = 0;
9183
9184 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009185 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009186
9187 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
9188}
9189
9190/*
9191 * "strwidth()" function
9192 */
9193 static void
9194f_strwidth(typval_T *argvars, typval_T *rettv)
9195{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009196 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009197
Bram Moolenaar13505972019-01-24 15:04:48 +01009198 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009199}
9200
9201/*
9202 * "strcharpart()" function
9203 */
9204 static void
9205f_strcharpart(typval_T *argvars, typval_T *rettv)
9206{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009207 char_u *p;
9208 int nchar;
9209 int nbyte = 0;
9210 int charlen;
9211 int len = 0;
9212 int slen;
9213 int error = FALSE;
9214
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009215 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009216 slen = (int)STRLEN(p);
9217
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009218 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009219 if (!error)
9220 {
9221 if (nchar > 0)
9222 while (nchar > 0 && nbyte < slen)
9223 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009224 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009225 --nchar;
9226 }
9227 else
9228 nbyte = nchar;
9229 if (argvars[2].v_type != VAR_UNKNOWN)
9230 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009231 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009232 while (charlen > 0 && nbyte + len < slen)
9233 {
9234 int off = nbyte + len;
9235
9236 if (off < 0)
9237 len += 1;
9238 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009239 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009240 --charlen;
9241 }
9242 }
9243 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009244 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009245 }
9246
9247 /*
9248 * Only return the overlap between the specified part and the actual
9249 * string.
9250 */
9251 if (nbyte < 0)
9252 {
9253 len += nbyte;
9254 nbyte = 0;
9255 }
9256 else if (nbyte > slen)
9257 nbyte = slen;
9258 if (len < 0)
9259 len = 0;
9260 else if (nbyte + len > slen)
9261 len = slen - nbyte;
9262
9263 rettv->v_type = VAR_STRING;
9264 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009265}
9266
9267/*
9268 * "strpart()" function
9269 */
9270 static void
9271f_strpart(typval_T *argvars, typval_T *rettv)
9272{
9273 char_u *p;
9274 int n;
9275 int len;
9276 int slen;
9277 int error = FALSE;
9278
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009279 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009280 slen = (int)STRLEN(p);
9281
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009282 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009283 if (error)
9284 len = 0;
9285 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009286 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009287 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009288 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009289
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009290 // Only return the overlap between the specified part and the actual
9291 // string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009292 if (n < 0)
9293 {
9294 len += n;
9295 n = 0;
9296 }
9297 else if (n > slen)
9298 n = slen;
9299 if (len < 0)
9300 len = 0;
9301 else if (n + len > slen)
9302 len = slen - n;
9303
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009304 if (argvars[2].v_type != VAR_UNKNOWN && argvars[3].v_type != VAR_UNKNOWN)
9305 {
9306 int off;
9307
9308 // length in characters
9309 for (off = n; off < slen && len > 0; --len)
9310 off += mb_ptr2len(p + off);
9311 len = off - n;
9312 }
9313
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009314 rettv->v_type = VAR_STRING;
9315 rettv->vval.v_string = vim_strnsave(p + n, len);
9316}
9317
9318/*
9319 * "strridx()" function
9320 */
9321 static void
9322f_strridx(typval_T *argvars, typval_T *rettv)
9323{
9324 char_u buf[NUMBUFLEN];
9325 char_u *needle;
9326 char_u *haystack;
9327 char_u *rest;
9328 char_u *lastmatch = NULL;
9329 int haystack_len, end_idx;
9330
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009331 needle = tv_get_string_chk(&argvars[1]);
9332 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009333
9334 rettv->vval.v_number = -1;
9335 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009336 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009337
9338 haystack_len = (int)STRLEN(haystack);
9339 if (argvars[2].v_type != VAR_UNKNOWN)
9340 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009341 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009342 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009343 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009344 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009345 }
9346 else
9347 end_idx = haystack_len;
9348
9349 if (*needle == NUL)
9350 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009351 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009352 lastmatch = haystack + end_idx;
9353 }
9354 else
9355 {
9356 for (rest = haystack; *rest != '\0'; ++rest)
9357 {
9358 rest = (char_u *)strstr((char *)rest, (char *)needle);
9359 if (rest == NULL || rest > haystack + end_idx)
9360 break;
9361 lastmatch = rest;
9362 }
9363 }
9364
9365 if (lastmatch == NULL)
9366 rettv->vval.v_number = -1;
9367 else
9368 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
9369}
9370
9371/*
9372 * "strtrans()" function
9373 */
9374 static void
9375f_strtrans(typval_T *argvars, typval_T *rettv)
9376{
9377 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009378 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009379}
9380
9381/*
9382 * "submatch()" function
9383 */
9384 static void
9385f_submatch(typval_T *argvars, typval_T *rettv)
9386{
9387 int error = FALSE;
9388 int no;
9389 int retList = 0;
9390
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009391 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009392 if (error)
9393 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009394 if (no < 0 || no >= NSUBEXP)
9395 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009396 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01009397 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009398 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009399 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaarad304702020-09-06 18:22:53 +02009400 retList = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009401 if (error)
9402 return;
9403
9404 if (retList == 0)
9405 {
9406 rettv->v_type = VAR_STRING;
9407 rettv->vval.v_string = reg_submatch(no);
9408 }
9409 else
9410 {
9411 rettv->v_type = VAR_LIST;
9412 rettv->vval.v_list = reg_submatch_list(no);
9413 }
9414}
9415
9416/*
9417 * "substitute()" function
9418 */
9419 static void
9420f_substitute(typval_T *argvars, typval_T *rettv)
9421{
9422 char_u patbuf[NUMBUFLEN];
9423 char_u subbuf[NUMBUFLEN];
9424 char_u flagsbuf[NUMBUFLEN];
9425
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009426 char_u *str = tv_get_string_chk(&argvars[0]);
9427 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009428 char_u *sub = NULL;
9429 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009430 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009431
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009432 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
9433 expr = &argvars[2];
9434 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009435 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009436
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009437 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009438 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
9439 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009440 rettv->vval.v_string = NULL;
9441 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009442 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009443}
9444
9445/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009446 * "swapinfo(swap_filename)" function
9447 */
9448 static void
9449f_swapinfo(typval_T *argvars, typval_T *rettv)
9450{
9451 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009452 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009453}
9454
9455/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02009456 * "swapname(expr)" function
9457 */
9458 static void
9459f_swapname(typval_T *argvars, typval_T *rettv)
9460{
9461 buf_T *buf;
9462
9463 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01009464 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02009465 if (buf == NULL || buf->b_ml.ml_mfp == NULL
9466 || buf->b_ml.ml_mfp->mf_fname == NULL)
9467 rettv->vval.v_string = NULL;
9468 else
9469 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
9470}
9471
9472/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009473 * "synID(lnum, col, trans)" function
9474 */
9475 static void
9476f_synID(typval_T *argvars UNUSED, typval_T *rettv)
9477{
9478 int id = 0;
9479#ifdef FEAT_SYN_HL
9480 linenr_T lnum;
9481 colnr_T col;
9482 int trans;
9483 int transerr = FALSE;
9484
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009485 lnum = tv_get_lnum(argvars); // -1 on type error
9486 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaarfcb6d702020-09-05 21:41:56 +02009487 trans = (int)tv_get_bool_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009488
9489 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9490 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
9491 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
9492#endif
9493
9494 rettv->vval.v_number = id;
9495}
9496
9497/*
9498 * "synIDattr(id, what [, mode])" function
9499 */
9500 static void
9501f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
9502{
9503 char_u *p = NULL;
9504#ifdef FEAT_SYN_HL
9505 int id;
9506 char_u *what;
9507 char_u *mode;
9508 char_u modebuf[NUMBUFLEN];
9509 int modec;
9510
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009511 id = (int)tv_get_number(&argvars[0]);
9512 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009513 if (argvars[2].v_type != VAR_UNKNOWN)
9514 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009515 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009516 modec = TOLOWER_ASC(mode[0]);
9517 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009518 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009519 }
9520 else
9521 {
9522#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
9523 if (USE_24BIT)
9524 modec = 'g';
9525 else
9526#endif
9527 if (t_colors > 1)
9528 modec = 'c';
9529 else
9530 modec = 't';
9531 }
9532
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009533 switch (TOLOWER_ASC(what[0]))
9534 {
9535 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009536 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009537 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009538 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009539 p = highlight_has_attr(id, HL_BOLD, modec);
9540 break;
9541
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009542 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009543 p = highlight_color(id, what, modec);
9544 break;
9545
9546 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009547 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009548 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009549 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009550 p = highlight_has_attr(id, HL_ITALIC, modec);
9551 break;
9552
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009553 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02009554 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009555 break;
9556
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009557 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009558 p = highlight_has_attr(id, HL_INVERSE, modec);
9559 break;
9560
9561 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009562 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009563 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009564 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02009565 else if (TOLOWER_ASC(what[1]) == 't' &&
9566 TOLOWER_ASC(what[2]) == 'r')
9567 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009568 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009569 p = highlight_has_attr(id, HL_STANDOUT, modec);
9570 break;
9571
9572 case 'u':
Bram Moolenaar391c3622020-09-29 20:59:17 +02009573 if (TOLOWER_ASC(what[1]) == 'l') // ul
9574 p = highlight_color(id, what, modec);
9575 else if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009576 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009577 p = highlight_has_attr(id, HL_UNDERLINE, modec);
9578 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009579 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009580 p = highlight_has_attr(id, HL_UNDERCURL, modec);
9581 break;
9582 }
9583
9584 if (p != NULL)
9585 p = vim_strsave(p);
9586#endif
9587 rettv->v_type = VAR_STRING;
9588 rettv->vval.v_string = p;
9589}
9590
9591/*
9592 * "synIDtrans(id)" function
9593 */
9594 static void
9595f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
9596{
9597 int id;
9598
9599#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009600 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009601
9602 if (id > 0)
9603 id = syn_get_final_id(id);
9604 else
9605#endif
9606 id = 0;
9607
9608 rettv->vval.v_number = id;
9609}
9610
9611/*
9612 * "synconcealed(lnum, col)" function
9613 */
9614 static void
9615f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
9616{
9617#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
9618 linenr_T lnum;
9619 colnr_T col;
9620 int syntax_flags = 0;
9621 int cchar;
9622 int matchid = 0;
9623 char_u str[NUMBUFLEN];
9624#endif
9625
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009626 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009627
9628#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009629 lnum = tv_get_lnum(argvars); // -1 on type error
9630 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009631
Bram Moolenaara80faa82020-04-12 19:37:17 +02009632 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009633
9634 if (rettv_list_alloc(rettv) != FAIL)
9635 {
9636 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9637 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9638 && curwin->w_p_cole > 0)
9639 {
9640 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
9641 syntax_flags = get_syntax_info(&matchid);
9642
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009643 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009644 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
9645 {
9646 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02009647 if (cchar == NUL && curwin->w_p_cole == 1)
9648 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009649 if (cchar != NUL)
9650 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009651 if (has_mbyte)
9652 (*mb_char2bytes)(cchar, str);
9653 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009654 str[0] = cchar;
9655 }
9656 }
9657 }
9658
9659 list_append_number(rettv->vval.v_list,
9660 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009661 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009662 list_append_string(rettv->vval.v_list, str, -1);
9663 list_append_number(rettv->vval.v_list, matchid);
9664 }
9665#endif
9666}
9667
9668/*
9669 * "synstack(lnum, col)" function
9670 */
9671 static void
9672f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
9673{
9674#ifdef FEAT_SYN_HL
9675 linenr_T lnum;
9676 colnr_T col;
9677 int i;
9678 int id;
9679#endif
9680
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009681 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009682
9683#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009684 lnum = tv_get_lnum(argvars); // -1 on type error
9685 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009686
9687 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9688 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9689 && rettv_list_alloc(rettv) != FAIL)
9690 {
9691 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
9692 for (i = 0; ; ++i)
9693 {
9694 id = syn_get_stack_item(i);
9695 if (id < 0)
9696 break;
9697 if (list_append_number(rettv->vval.v_list, id) == FAIL)
9698 break;
9699 }
9700 }
9701#endif
9702}
9703
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009704/*
9705 * "tabpagebuflist()" function
9706 */
9707 static void
9708f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9709{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009710 tabpage_T *tp;
9711 win_T *wp = NULL;
9712
9713 if (argvars[0].v_type == VAR_UNKNOWN)
9714 wp = firstwin;
9715 else
9716 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009717 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009718 if (tp != NULL)
9719 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
9720 }
9721 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
9722 {
9723 for (; wp != NULL; wp = wp->w_next)
9724 if (list_append_number(rettv->vval.v_list,
9725 wp->w_buffer->b_fnum) == FAIL)
9726 break;
9727 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009728}
9729
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009730/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009731 * "tagfiles()" function
9732 */
9733 static void
9734f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
9735{
9736 char_u *fname;
9737 tagname_T tn;
9738 int first;
9739
9740 if (rettv_list_alloc(rettv) == FAIL)
9741 return;
9742 fname = alloc(MAXPATHL);
9743 if (fname == NULL)
9744 return;
9745
9746 for (first = TRUE; ; first = FALSE)
9747 if (get_tagfname(&tn, first, fname) == FAIL
9748 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
9749 break;
9750 tagname_free(&tn);
9751 vim_free(fname);
9752}
9753
9754/*
9755 * "taglist()" function
9756 */
9757 static void
9758f_taglist(typval_T *argvars, typval_T *rettv)
9759{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009760 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009761 char_u *tag_pattern;
9762
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009763 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009764
9765 rettv->vval.v_number = FALSE;
9766 if (*tag_pattern == NUL)
9767 return;
9768
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009769 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009770 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009771 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009772 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009773}
9774
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009775#ifdef FEAT_FLOAT
9776/*
9777 * "tan()" function
9778 */
9779 static void
9780f_tan(typval_T *argvars, typval_T *rettv)
9781{
9782 float_T f = 0.0;
9783
9784 rettv->v_type = VAR_FLOAT;
9785 if (get_float_arg(argvars, &f) == OK)
9786 rettv->vval.v_float = tan(f);
9787 else
9788 rettv->vval.v_float = 0.0;
9789}
9790
9791/*
9792 * "tanh()" function
9793 */
9794 static void
9795f_tanh(typval_T *argvars, typval_T *rettv)
9796{
9797 float_T f = 0.0;
9798
9799 rettv->v_type = VAR_FLOAT;
9800 if (get_float_arg(argvars, &f) == OK)
9801 rettv->vval.v_float = tanh(f);
9802 else
9803 rettv->vval.v_float = 0.0;
9804}
9805#endif
9806
9807/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009808 * "tolower(string)" function
9809 */
9810 static void
9811f_tolower(typval_T *argvars, typval_T *rettv)
9812{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009813 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009814 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009815}
9816
9817/*
9818 * "toupper(string)" function
9819 */
9820 static void
9821f_toupper(typval_T *argvars, typval_T *rettv)
9822{
9823 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009824 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009825}
9826
9827/*
9828 * "tr(string, fromstr, tostr)" function
9829 */
9830 static void
9831f_tr(typval_T *argvars, typval_T *rettv)
9832{
9833 char_u *in_str;
9834 char_u *fromstr;
9835 char_u *tostr;
9836 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009837 int inlen;
9838 int fromlen;
9839 int tolen;
9840 int idx;
9841 char_u *cpstr;
9842 int cplen;
9843 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009844 char_u buf[NUMBUFLEN];
9845 char_u buf2[NUMBUFLEN];
9846 garray_T ga;
9847
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009848 in_str = tv_get_string(&argvars[0]);
9849 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
9850 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009851
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009852 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009853 rettv->v_type = VAR_STRING;
9854 rettv->vval.v_string = NULL;
9855 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009856 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009857 ga_init2(&ga, (int)sizeof(char), 80);
9858
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009859 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009860 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009861 if (STRLEN(fromstr) != STRLEN(tostr))
9862 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009863error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009864 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009865 ga_clear(&ga);
9866 return;
9867 }
9868
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009869 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009870 while (*in_str != NUL)
9871 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009872 if (has_mbyte)
9873 {
9874 inlen = (*mb_ptr2len)(in_str);
9875 cpstr = in_str;
9876 cplen = inlen;
9877 idx = 0;
9878 for (p = fromstr; *p != NUL; p += fromlen)
9879 {
9880 fromlen = (*mb_ptr2len)(p);
9881 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
9882 {
9883 for (p = tostr; *p != NUL; p += tolen)
9884 {
9885 tolen = (*mb_ptr2len)(p);
9886 if (idx-- == 0)
9887 {
9888 cplen = tolen;
9889 cpstr = p;
9890 break;
9891 }
9892 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009893 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009894 goto error;
9895 break;
9896 }
9897 ++idx;
9898 }
9899
9900 if (first && cpstr == in_str)
9901 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009902 // Check that fromstr and tostr have the same number of
9903 // (multi-byte) characters. Done only once when a character
9904 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009905 first = FALSE;
9906 for (p = tostr; *p != NUL; p += tolen)
9907 {
9908 tolen = (*mb_ptr2len)(p);
9909 --idx;
9910 }
9911 if (idx != 0)
9912 goto error;
9913 }
9914
9915 (void)ga_grow(&ga, cplen);
9916 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
9917 ga.ga_len += cplen;
9918
9919 in_str += inlen;
9920 }
9921 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009922 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009923 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009924 p = vim_strchr(fromstr, *in_str);
9925 if (p != NULL)
9926 ga_append(&ga, tostr[p - fromstr]);
9927 else
9928 ga_append(&ga, *in_str);
9929 ++in_str;
9930 }
9931 }
9932
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009933 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009934 (void)ga_grow(&ga, 1);
9935 ga_append(&ga, NUL);
9936
9937 rettv->vval.v_string = ga.ga_data;
9938}
9939
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009940/*
9941 * "trim({expr})" function
9942 */
9943 static void
9944f_trim(typval_T *argvars, typval_T *rettv)
9945{
9946 char_u buf1[NUMBUFLEN];
9947 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009948 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009949 char_u *mask = NULL;
9950 char_u *tail;
9951 char_u *prev;
9952 char_u *p;
9953 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009954 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009955
9956 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009957 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009958 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009959 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009960
9961 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009962 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009963 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009964
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009965 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009966 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009967 int error = 0;
9968
9969 // leading or trailing characters to trim
9970 dir = (int)tv_get_number_chk(&argvars[2], &error);
9971 if (error)
9972 return;
9973 if (dir < 0 || dir > 2)
9974 {
9975 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
9976 return;
9977 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009978 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009979 }
9980
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009981 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009982 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009983 // Trim leading characters
9984 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009985 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009986 c1 = PTR2CHAR(head);
9987 if (mask == NULL)
9988 {
9989 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009990 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009991 }
9992 else
9993 {
9994 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9995 if (c1 == PTR2CHAR(p))
9996 break;
9997 if (*p == NUL)
9998 break;
9999 }
10000 MB_PTR_ADV(head);
10001 }
10002 }
10003
10004 tail = head + STRLEN(head);
10005 if (dir == 0 || dir == 2)
10006 {
10007 // Trim trailing characters
10008 for (; tail > head; tail = prev)
10009 {
10010 prev = tail;
10011 MB_PTR_BACK(head, prev);
10012 c1 = PTR2CHAR(prev);
10013 if (mask == NULL)
10014 {
10015 if (c1 > ' ' && c1 != 0xa0)
10016 break;
10017 }
10018 else
10019 {
10020 for (p = mask; *p != NUL; MB_PTR_ADV(p))
10021 if (c1 == PTR2CHAR(p))
10022 break;
10023 if (*p == NUL)
10024 break;
10025 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +010010026 }
10027 }
Bram Moolenaardf44a272020-06-07 20:49:05 +020010028 rettv->vval.v_string = vim_strnsave(head, tail - head);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +010010029}
10030
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010031#ifdef FEAT_FLOAT
10032/*
10033 * "trunc({float})" function
10034 */
10035 static void
10036f_trunc(typval_T *argvars, typval_T *rettv)
10037{
10038 float_T f = 0.0;
10039
10040 rettv->v_type = VAR_FLOAT;
10041 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010042 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010043 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
10044 else
10045 rettv->vval.v_float = 0.0;
10046}
10047#endif
10048
10049/*
10050 * "type(expr)" function
10051 */
10052 static void
10053f_type(typval_T *argvars, typval_T *rettv)
10054{
10055 int n = -1;
10056
10057 switch (argvars[0].v_type)
10058 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +010010059 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
10060 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010061 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +010010062 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
10063 case VAR_LIST: n = VAR_TYPE_LIST; break;
10064 case VAR_DICT: n = VAR_TYPE_DICT; break;
10065 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
10066 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
10067 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +020010068 case VAR_JOB: n = VAR_TYPE_JOB; break;
10069 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +010010070 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010071 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +020010072 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010010073 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +010010074 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010075 n = -1;
10076 break;
10077 }
10078 rettv->vval.v_number = n;
10079}
10080
10081/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010082 * "virtcol(string)" function
10083 */
10084 static void
10085f_virtcol(typval_T *argvars, typval_T *rettv)
10086{
10087 colnr_T vcol = 0;
10088 pos_T *fp;
10089 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +010010090 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010091
Bram Moolenaar6f02b002021-01-10 20:22:54 +010010092 fp = var2fpos(&argvars[0], FALSE, &fnum, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010093 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
10094 && fnum == curbuf->b_fnum)
10095 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +010010096 // Limit the column to a valid value, getvvcol() doesn't check.
10097 if (fp->col < 0)
10098 fp->col = 0;
10099 else
10100 {
10101 len = (int)STRLEN(ml_get(fp->lnum));
10102 if (fp->col > len)
10103 fp->col = len;
10104 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010105 getvvcol(curwin, fp, NULL, NULL, &vcol);
10106 ++vcol;
10107 }
10108
10109 rettv->vval.v_number = vcol;
10110}
10111
10112/*
10113 * "visualmode()" function
10114 */
10115 static void
10116f_visualmode(typval_T *argvars, typval_T *rettv)
10117{
10118 char_u str[2];
10119
10120 rettv->v_type = VAR_STRING;
10121 str[0] = curbuf->b_visual_mode_eval;
10122 str[1] = NUL;
10123 rettv->vval.v_string = vim_strsave(str);
10124
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010125 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010126 if (non_zero_arg(&argvars[0]))
10127 curbuf->b_visual_mode_eval = NUL;
10128}
10129
10130/*
10131 * "wildmenumode()" function
10132 */
10133 static void
10134f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
10135{
10136#ifdef FEAT_WILDMENU
10137 if (wild_menu_showing)
10138 rettv->vval.v_number = 1;
10139#endif
10140}
10141
10142/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +010010143 * "windowsversion()" function
10144 */
10145 static void
10146f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
10147{
10148 rettv->v_type = VAR_STRING;
10149 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
10150}
10151
10152/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010153 * "wordcount()" function
10154 */
10155 static void
10156f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
10157{
10158 if (rettv_dict_alloc(rettv) == FAIL)
10159 return;
10160 cursor_pos_info(rettv->vval.v_dict);
10161}
10162
10163/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010164 * "xor(expr, expr)" function
10165 */
10166 static void
10167f_xor(typval_T *argvars, typval_T *rettv)
10168{
Bram Moolenaard155d7a2018-12-21 16:04:21 +010010169 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
10170 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010171}
10172
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010173#endif // FEAT_EVAL