blob: 1abcd5e11ccecba7673b1b2c166834878306af69 [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/*
343 * Check "type" is the same type as the previous argument
344 * 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/*
355 * Check "type" is an item of the list or blob of the previous arg.
Bram Moolenaarca174532020-10-21 16:42:22 +0200356 * Must not be used for the first argcheck_T entry.
357 */
358 static int
359arg_item_of_prev(type_T *type, argcontext_T *context)
360{
361 type_T *prev_type = context->arg_types[context->arg_idx - 1];
362 type_T *expected;
363
364 if (prev_type->tt_type == VAR_LIST)
365 expected = prev_type->tt_member;
366 else if (prev_type->tt_type == VAR_BLOB)
367 expected = &t_number;
368 else
369 // probably VAR_ANY, can't check
370 return OK;
371
Bram Moolenaar193f6202020-11-16 20:08:35 +0100372 return check_arg_type(expected, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200373}
374
Bram Moolenaar94738d82020-10-21 14:25:07 +0200375/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100376 * Check "type" which is the third argument of extend().
377 */
378 static int
379arg_extend3(type_T *type, argcontext_T *context)
380{
381 type_T *first_type = context->arg_types[context->arg_idx - 2];
382
383 if (first_type->tt_type == VAR_LIST)
384 return arg_number(type, context);
385 if (first_type->tt_type == VAR_DICT)
386 return arg_string(type, context);
387 return OK;
388}
389
390
391/*
Bram Moolenaar94738d82020-10-21 14:25:07 +0200392 * Lists of functions that check the argument types of a builtin function.
393 */
394argcheck_T arg1_float_or_nr[] = {arg_float_or_nr};
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100395argcheck_T arg2_listblob_item[] = {arg_list_or_blob, arg_item_of_prev};
396argcheck_T arg23_extend[] = {arg_list_or_dict, arg_same_as_prev, arg_extend3};
Bram Moolenaarca174532020-10-21 16:42:22 +0200397argcheck_T arg3_insert[] = {arg_list_or_blob, arg_item_of_prev, arg_number};
Bram Moolenaar94738d82020-10-21 14:25:07 +0200398
399/*
400 * Functions that return the return type of a builtin function.
Bram Moolenaara1224cb2020-10-22 12:31:49 +0200401 * Note that "argtypes" is NULL if "argcount" is zero.
Bram Moolenaar94738d82020-10-21 14:25:07 +0200402 */
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100403 static type_T *
404ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
405{
406 return &t_void;
407}
408 static type_T *
409ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
410{
411 return &t_any;
412}
413 static type_T *
Bram Moolenaar403dc312020-10-17 19:29:51 +0200414ret_bool(int argcount UNUSED, type_T **argtypes UNUSED)
415{
416 return &t_bool;
417}
418 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100419ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
420{
421 return &t_number;
422}
423 static type_T *
424ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
425{
426 return &t_float;
427}
428 static type_T *
429ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
430{
431 return &t_string;
432}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200433 static type_T *
434ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100435{
436 return &t_list_any;
437}
438 static type_T *
439ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
440{
441 return &t_list_number;
442}
443 static type_T *
444ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
445{
446 return &t_list_string;
447}
448 static type_T *
449ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
450{
451 return &t_list_dict_any;
452}
453 static type_T *
454ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
455{
456 return &t_dict_any;
457}
458 static type_T *
459ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
460{
461 return &t_dict_number;
462}
463 static type_T *
464ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
465{
466 return &t_dict_string;
467}
468 static type_T *
469ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
470{
471 return &t_blob;
472}
473 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200474ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100475{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200476 return &t_func_any;
477}
478 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100479ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
480{
481 return &t_channel;
482}
483 static type_T *
484ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
485{
486 return &t_job;
487}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200488 static type_T *
489ret_first_arg(int argcount, type_T **argtypes)
490{
491 if (argcount > 0)
492 return argtypes[0];
493 return &t_void;
494}
Bram Moolenaarea696852020-11-09 18:31:39 +0100495// for map(): returns first argument but item type may differ
496 static type_T *
497ret_first_cont(int argcount UNUSED, type_T **argtypes)
498{
499 if (argtypes[0]->tt_type == VAR_LIST)
500 return &t_list_any;
501 if (argtypes[0]->tt_type == VAR_DICT)
502 return &t_dict_any;
503 if (argtypes[0]->tt_type == VAR_BLOB)
504 return argtypes[0];
505 return &t_any;
506}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200507
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200508/*
509 * Used for getqflist(): returns list if there is no argument, dict if there is
510 * one.
511 */
512 static type_T *
513ret_list_or_dict_0(int argcount, type_T **argtypes UNUSED)
514{
515 if (argcount > 0)
516 return &t_dict_any;
517 return &t_list_dict_any;
518}
519
520/*
521 * Used for getloclist(): returns list if there is one argument, dict if there
522 * are two.
523 */
524 static type_T *
525ret_list_or_dict_1(int argcount, type_T **argtypes UNUSED)
526{
527 if (argcount > 1)
528 return &t_dict_any;
529 return &t_list_dict_any;
530}
531
Bram Moolenaar846178a2020-07-05 17:04:13 +0200532 static type_T *
533ret_argv(int argcount, type_T **argtypes UNUSED)
534{
535 // argv() returns list of strings
536 if (argcount == 0)
537 return &t_list_string;
538
539 // argv(0) returns a string, but argv(-1] returns a list
540 return &t_any;
541}
542
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200543 static type_T *
544ret_remove(int argcount UNUSED, type_T **argtypes)
545{
Bram Moolenaar5e654232020-09-16 15:22:00 +0200546 if (argtypes != NULL)
547 {
548 if (argtypes[0]->tt_type == VAR_LIST
549 || argtypes[0]->tt_type == VAR_DICT)
550 return argtypes[0]->tt_member;
551 if (argtypes[0]->tt_type == VAR_BLOB)
552 return &t_number;
553 }
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200554 return &t_any;
555}
556
Bram Moolenaar3d945cc2020-08-06 21:26:59 +0200557 static type_T *
558ret_getreg(int argcount, type_T **argtypes UNUSED)
559{
560 // Assume that if the third argument is passed it's non-zero
561 if (argcount == 3)
562 return &t_list_string;
563 return &t_string;
564}
565
Bram Moolenaar4a6d1b62020-08-08 17:55:49 +0200566 static type_T *
567ret_maparg(int argcount, type_T **argtypes UNUSED)
568{
569 // Assume that if the fourth argument is passed it's non-zero
570 if (argcount == 4)
571 return &t_dict_any;
572 return &t_string;
573}
574
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100575static type_T *ret_f_function(int argcount, type_T **argtypes);
576
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200577/*
578 * Array with names and number of arguments of all internal functions
579 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
580 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200581typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200582{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200583 char *f_name; // function name
584 char f_min_argc; // minimal number of arguments
585 char f_max_argc; // maximal number of arguments
586 char f_argtype; // for method: FEARG_ values
Bram Moolenaar94738d82020-10-21 14:25:07 +0200587 argcheck_T *f_argcheck; // list of functions to check argument types
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100588 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
589 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200590 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200591 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200592} funcentry_T;
593
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200594// values for f_argtype; zero means it cannot be used as a method
595#define FEARG_1 1 // base is the first argument
596#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200597#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200598#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200599#define FEARG_LAST 9 // base is the last argument
600
Bram Moolenaar15c47602020-03-26 22:16:48 +0100601#ifdef FEAT_FLOAT
602# define FLOAT_FUNC(name) name
603#else
604# define FLOAT_FUNC(name) NULL
605#endif
606#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
607# define MATH_FUNC(name) name
608#else
609# define MATH_FUNC(name) NULL
610#endif
611#ifdef FEAT_TIMERS
612# define TIMER_FUNC(name) name
613#else
614# define TIMER_FUNC(name) NULL
615#endif
616#ifdef FEAT_JOB_CHANNEL
617# define JOB_FUNC(name) name
618#else
619# define JOB_FUNC(name) NULL
620#endif
621#ifdef FEAT_PROP_POPUP
622# define PROP_FUNC(name) name
623#else
624# define PROP_FUNC(name) NULL
625#endif
626#ifdef FEAT_SIGNS
627# define SIGN_FUNC(name) name
628#else
629# define SIGN_FUNC(name) NULL
630#endif
631#ifdef FEAT_SOUND
632# define SOUND_FUNC(name) name
633#else
634# define SOUND_FUNC(name) NULL
635#endif
636#ifdef FEAT_TERMINAL
637# define TERM_FUNC(name) name
638#else
639# define TERM_FUNC(name) NULL
640#endif
641
Bram Moolenaarac92e252019-08-03 21:58:38 +0200642static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200643{
Bram Moolenaar94738d82020-10-21 14:25:07 +0200644 {"abs", 1, 1, FEARG_1, arg1_float_or_nr,
645 ret_any, FLOAT_FUNC(f_abs)},
646 {"acos", 1, 1, FEARG_1, NULL,
647 ret_float, FLOAT_FUNC(f_acos)},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100648 {"add", 2, 2, FEARG_1, NULL /* arg2_listblob_item */,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200649 ret_first_arg, f_add},
650 {"and", 2, 2, FEARG_1, NULL,
651 ret_number, f_and},
652 {"append", 2, 2, FEARG_2, NULL,
653 ret_number, f_append},
654 {"appendbufline", 3, 3, FEARG_3, NULL,
655 ret_number, f_appendbufline},
656 {"argc", 0, 1, 0, NULL,
657 ret_number, f_argc},
658 {"argidx", 0, 0, 0, NULL,
659 ret_number, f_argidx},
660 {"arglistid", 0, 2, 0, NULL,
661 ret_number, f_arglistid},
662 {"argv", 0, 2, 0, NULL,
663 ret_argv, f_argv},
664 {"asin", 1, 1, FEARG_1, NULL,
665 ret_float, FLOAT_FUNC(f_asin)},
666 {"assert_beeps", 1, 2, FEARG_1, NULL,
667 ret_number, f_assert_beeps},
668 {"assert_equal", 2, 3, FEARG_2, NULL,
669 ret_number, f_assert_equal},
670 {"assert_equalfile", 2, 3, FEARG_1, NULL,
671 ret_number, f_assert_equalfile},
672 {"assert_exception", 1, 2, 0, NULL,
673 ret_number, f_assert_exception},
674 {"assert_fails", 1, 5, FEARG_1, NULL,
675 ret_number, f_assert_fails},
676 {"assert_false", 1, 2, FEARG_1, NULL,
677 ret_number, f_assert_false},
678 {"assert_inrange", 3, 4, FEARG_3, NULL,
679 ret_number, f_assert_inrange},
680 {"assert_match", 2, 3, FEARG_2, NULL,
681 ret_number, f_assert_match},
682 {"assert_notequal", 2, 3, FEARG_2, NULL,
683 ret_number, f_assert_notequal},
684 {"assert_notmatch", 2, 3, FEARG_2, NULL,
685 ret_number, f_assert_notmatch},
686 {"assert_report", 1, 1, FEARG_1, NULL,
687 ret_number, f_assert_report},
688 {"assert_true", 1, 2, FEARG_1, NULL,
689 ret_number, f_assert_true},
690 {"atan", 1, 1, FEARG_1, NULL,
691 ret_float, FLOAT_FUNC(f_atan)},
692 {"atan2", 2, 2, FEARG_1, NULL,
693 ret_float, FLOAT_FUNC(f_atan2)},
694 {"balloon_gettext", 0, 0, 0, NULL,
695 ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100696#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100697 f_balloon_gettext
698#else
699 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100700#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100701 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200702 {"balloon_show", 1, 1, FEARG_1, NULL,
703 ret_void,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100704#ifdef FEAT_BEVAL
705 f_balloon_show
706#else
707 NULL
708#endif
709 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200710 {"balloon_split", 1, 1, FEARG_1, NULL,
711 ret_list_string,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100712#if defined(FEAT_BEVAL_TERM)
713 f_balloon_split
714#else
715 NULL
716#endif
717 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200718 {"browse", 4, 4, 0, NULL,
719 ret_string, f_browse},
720 {"browsedir", 2, 2, 0, NULL,
721 ret_string, f_browsedir},
722 {"bufadd", 1, 1, FEARG_1, NULL,
723 ret_number, f_bufadd},
724 {"bufexists", 1, 1, FEARG_1, NULL,
725 ret_number, f_bufexists},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200726 {"buffer_exists", 1, 1, FEARG_1, NULL, // obsolete
727 ret_number, f_bufexists},
728 {"buffer_name", 0, 1, FEARG_1, NULL, // obsolete
729 ret_string, f_bufname},
730 {"buffer_number", 0, 1, FEARG_1, NULL, // obsolete
731 ret_number, f_bufnr},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200732 {"buflisted", 1, 1, FEARG_1, NULL,
733 ret_number, f_buflisted},
734 {"bufload", 1, 1, FEARG_1, NULL,
735 ret_void, f_bufload},
736 {"bufloaded", 1, 1, FEARG_1, NULL,
737 ret_number, f_bufloaded},
738 {"bufname", 0, 1, FEARG_1, NULL,
739 ret_string, f_bufname},
740 {"bufnr", 0, 2, FEARG_1, NULL,
741 ret_number, f_bufnr},
742 {"bufwinid", 1, 1, FEARG_1, NULL,
743 ret_number, f_bufwinid},
744 {"bufwinnr", 1, 1, FEARG_1, NULL,
745 ret_number, f_bufwinnr},
746 {"byte2line", 1, 1, FEARG_1, NULL,
747 ret_number, f_byte2line},
748 {"byteidx", 2, 2, FEARG_1, NULL,
749 ret_number, f_byteidx},
750 {"byteidxcomp", 2, 2, FEARG_1, NULL,
751 ret_number, f_byteidxcomp},
752 {"call", 2, 3, FEARG_1, NULL,
753 ret_any, f_call},
754 {"ceil", 1, 1, FEARG_1, NULL,
755 ret_float, FLOAT_FUNC(f_ceil)},
756 {"ch_canread", 1, 1, FEARG_1, NULL,
757 ret_number, JOB_FUNC(f_ch_canread)},
758 {"ch_close", 1, 1, FEARG_1, NULL,
759 ret_void, JOB_FUNC(f_ch_close)},
760 {"ch_close_in", 1, 1, FEARG_1, NULL,
761 ret_void, JOB_FUNC(f_ch_close_in)},
762 {"ch_evalexpr", 2, 3, FEARG_1, NULL,
763 ret_any, JOB_FUNC(f_ch_evalexpr)},
764 {"ch_evalraw", 2, 3, FEARG_1, NULL,
765 ret_any, JOB_FUNC(f_ch_evalraw)},
766 {"ch_getbufnr", 2, 2, FEARG_1, NULL,
767 ret_number, JOB_FUNC(f_ch_getbufnr)},
768 {"ch_getjob", 1, 1, FEARG_1, NULL,
769 ret_job, JOB_FUNC(f_ch_getjob)},
770 {"ch_info", 1, 1, FEARG_1, NULL,
771 ret_dict_any, JOB_FUNC(f_ch_info)},
772 {"ch_log", 1, 2, FEARG_1, NULL,
773 ret_void, JOB_FUNC(f_ch_log)},
774 {"ch_logfile", 1, 2, FEARG_1, NULL,
775 ret_void, JOB_FUNC(f_ch_logfile)},
776 {"ch_open", 1, 2, FEARG_1, NULL,
777 ret_channel, JOB_FUNC(f_ch_open)},
778 {"ch_read", 1, 2, FEARG_1, NULL,
779 ret_string, JOB_FUNC(f_ch_read)},
780 {"ch_readblob", 1, 2, FEARG_1, NULL,
781 ret_blob, JOB_FUNC(f_ch_readblob)},
782 {"ch_readraw", 1, 2, FEARG_1, NULL,
783 ret_string, JOB_FUNC(f_ch_readraw)},
784 {"ch_sendexpr", 2, 3, FEARG_1, NULL,
785 ret_void, JOB_FUNC(f_ch_sendexpr)},
786 {"ch_sendraw", 2, 3, FEARG_1, NULL,
787 ret_void, JOB_FUNC(f_ch_sendraw)},
788 {"ch_setoptions", 2, 2, FEARG_1, NULL,
789 ret_void, JOB_FUNC(f_ch_setoptions)},
790 {"ch_status", 1, 2, FEARG_1, NULL,
791 ret_string, JOB_FUNC(f_ch_status)},
792 {"changenr", 0, 0, 0, NULL,
793 ret_number, f_changenr},
794 {"char2nr", 1, 2, FEARG_1, NULL,
795 ret_number, f_char2nr},
796 {"charclass", 1, 1, FEARG_1, NULL,
797 ret_number, f_charclass},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100798 {"charcol", 1, 1, FEARG_1, NULL,
799 ret_number, f_charcol},
Bram Moolenaar17793ef2020-12-28 12:56:58 +0100800 {"charidx", 2, 3, FEARG_1, NULL,
801 ret_number, f_charidx},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200802 {"chdir", 1, 1, FEARG_1, NULL,
803 ret_string, f_chdir},
804 {"cindent", 1, 1, FEARG_1, NULL,
805 ret_number, f_cindent},
806 {"clearmatches", 0, 1, FEARG_1, NULL,
807 ret_void, f_clearmatches},
808 {"col", 1, 1, FEARG_1, NULL,
809 ret_number, f_col},
810 {"complete", 2, 2, FEARG_2, NULL,
811 ret_void, f_complete},
812 {"complete_add", 1, 1, FEARG_1, NULL,
813 ret_number, f_complete_add},
814 {"complete_check", 0, 0, 0, NULL,
815 ret_number, f_complete_check},
816 {"complete_info", 0, 1, FEARG_1, NULL,
817 ret_dict_any, f_complete_info},
818 {"confirm", 1, 4, FEARG_1, NULL,
819 ret_number, f_confirm},
820 {"copy", 1, 1, FEARG_1, NULL,
821 ret_first_arg, f_copy},
822 {"cos", 1, 1, FEARG_1, NULL,
823 ret_float, FLOAT_FUNC(f_cos)},
824 {"cosh", 1, 1, FEARG_1, NULL,
825 ret_float, FLOAT_FUNC(f_cosh)},
826 {"count", 2, 4, FEARG_1, NULL,
827 ret_number, f_count},
828 {"cscope_connection",0,3, 0, NULL,
829 ret_number, f_cscope_connection},
830 {"cursor", 1, 3, FEARG_1, NULL,
831 ret_number, f_cursor},
832 {"debugbreak", 1, 1, FEARG_1, NULL,
833 ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100834#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100835 f_debugbreak
836#else
837 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200838#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100839 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200840 {"deepcopy", 1, 2, FEARG_1, NULL,
841 ret_first_arg, f_deepcopy},
842 {"delete", 1, 2, FEARG_1, NULL,
843 ret_number, f_delete},
844 {"deletebufline", 2, 3, FEARG_1, NULL,
845 ret_number, f_deletebufline},
846 {"did_filetype", 0, 0, 0, NULL,
847 ret_number, f_did_filetype},
848 {"diff_filler", 1, 1, FEARG_1, NULL,
849 ret_number, f_diff_filler},
850 {"diff_hlID", 2, 2, FEARG_1, NULL,
851 ret_number, f_diff_hlID},
852 {"echoraw", 1, 1, FEARG_1, NULL,
853 ret_number, f_echoraw},
854 {"empty", 1, 1, FEARG_1, NULL,
855 ret_number, f_empty},
856 {"environ", 0, 0, 0, NULL,
857 ret_dict_string, f_environ},
858 {"escape", 2, 2, FEARG_1, NULL,
859 ret_string, f_escape},
860 {"eval", 1, 1, FEARG_1, NULL,
861 ret_any, f_eval},
862 {"eventhandler", 0, 0, 0, NULL,
863 ret_number, f_eventhandler},
864 {"executable", 1, 1, FEARG_1, NULL,
865 ret_number, f_executable},
866 {"execute", 1, 2, FEARG_1, NULL,
867 ret_string, f_execute},
868 {"exepath", 1, 1, FEARG_1, NULL,
869 ret_string, f_exepath},
870 {"exists", 1, 1, FEARG_1, NULL,
871 ret_number, f_exists},
872 {"exp", 1, 1, FEARG_1, NULL,
873 ret_float, FLOAT_FUNC(f_exp)},
874 {"expand", 1, 3, FEARG_1, NULL,
875 ret_any, f_expand},
876 {"expandcmd", 1, 1, FEARG_1, NULL,
877 ret_string, f_expandcmd},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100878 {"extend", 2, 3, FEARG_1, arg23_extend,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200879 ret_first_arg, f_extend},
880 {"feedkeys", 1, 2, FEARG_1, NULL,
881 ret_void, f_feedkeys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200882 {"file_readable", 1, 1, FEARG_1, NULL, // obsolete
883 ret_number, f_filereadable},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200884 {"filereadable", 1, 1, FEARG_1, NULL,
885 ret_number, f_filereadable},
886 {"filewritable", 1, 1, FEARG_1, NULL,
887 ret_number, f_filewritable},
888 {"filter", 2, 2, FEARG_1, NULL,
889 ret_first_arg, f_filter},
890 {"finddir", 1, 3, FEARG_1, NULL,
891 ret_string, f_finddir},
892 {"findfile", 1, 3, FEARG_1, NULL,
893 ret_string, f_findfile},
894 {"flatten", 1, 2, FEARG_1, NULL,
895 ret_list_any, f_flatten},
896 {"float2nr", 1, 1, FEARG_1, NULL,
897 ret_number, FLOAT_FUNC(f_float2nr)},
898 {"floor", 1, 1, FEARG_1, NULL,
899 ret_float, FLOAT_FUNC(f_floor)},
900 {"fmod", 2, 2, FEARG_1, NULL,
901 ret_float, FLOAT_FUNC(f_fmod)},
902 {"fnameescape", 1, 1, FEARG_1, NULL,
903 ret_string, f_fnameescape},
904 {"fnamemodify", 2, 2, FEARG_1, NULL,
905 ret_string, f_fnamemodify},
906 {"foldclosed", 1, 1, FEARG_1, NULL,
907 ret_number, f_foldclosed},
908 {"foldclosedend", 1, 1, FEARG_1, NULL,
909 ret_number, f_foldclosedend},
910 {"foldlevel", 1, 1, FEARG_1, NULL,
911 ret_number, f_foldlevel},
912 {"foldtext", 0, 0, 0, NULL,
913 ret_string, f_foldtext},
914 {"foldtextresult", 1, 1, FEARG_1, NULL,
915 ret_string, f_foldtextresult},
916 {"foreground", 0, 0, 0, NULL,
917 ret_void, f_foreground},
918 {"funcref", 1, 3, FEARG_1, NULL,
919 ret_func_any, f_funcref},
920 {"function", 1, 3, FEARG_1, NULL,
921 ret_f_function, f_function},
922 {"garbagecollect", 0, 1, 0, NULL,
923 ret_void, f_garbagecollect},
924 {"get", 2, 3, FEARG_1, NULL,
925 ret_any, f_get},
926 {"getbufinfo", 0, 1, FEARG_1, NULL,
927 ret_list_dict_any, f_getbufinfo},
928 {"getbufline", 2, 3, FEARG_1, NULL,
929 ret_list_string, f_getbufline},
930 {"getbufvar", 2, 3, FEARG_1, NULL,
931 ret_any, f_getbufvar},
932 {"getchangelist", 0, 1, FEARG_1, NULL,
933 ret_list_any, f_getchangelist},
934 {"getchar", 0, 1, 0, NULL,
935 ret_number, f_getchar},
936 {"getcharmod", 0, 0, 0, NULL,
937 ret_number, f_getcharmod},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100938 {"getcharpos", 1, 1, FEARG_1, NULL,
939 ret_list_number, f_getcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200940 {"getcharsearch", 0, 0, 0, NULL,
941 ret_dict_any, f_getcharsearch},
942 {"getcmdline", 0, 0, 0, NULL,
943 ret_string, f_getcmdline},
944 {"getcmdpos", 0, 0, 0, NULL,
945 ret_number, f_getcmdpos},
946 {"getcmdtype", 0, 0, 0, NULL,
947 ret_string, f_getcmdtype},
948 {"getcmdwintype", 0, 0, 0, NULL,
949 ret_string, f_getcmdwintype},
950 {"getcompletion", 2, 3, FEARG_1, NULL,
951 ret_list_string, f_getcompletion},
952 {"getcurpos", 0, 1, FEARG_1, NULL,
953 ret_list_number, f_getcurpos},
Bram Moolenaar6f02b002021-01-10 20:22:54 +0100954 {"getcursorcharpos", 0, 1, FEARG_1, NULL,
955 ret_list_number, f_getcursorcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200956 {"getcwd", 0, 2, FEARG_1, NULL,
957 ret_string, f_getcwd},
958 {"getenv", 1, 1, FEARG_1, NULL,
959 ret_string, f_getenv},
960 {"getfontname", 0, 1, 0, NULL,
961 ret_string, f_getfontname},
962 {"getfperm", 1, 1, FEARG_1, NULL,
963 ret_string, f_getfperm},
964 {"getfsize", 1, 1, FEARG_1, NULL,
965 ret_number, f_getfsize},
966 {"getftime", 1, 1, FEARG_1, NULL,
967 ret_number, f_getftime},
968 {"getftype", 1, 1, FEARG_1, NULL,
969 ret_string, f_getftype},
970 {"getimstatus", 0, 0, 0, NULL,
971 ret_number, f_getimstatus},
972 {"getjumplist", 0, 2, FEARG_1, NULL,
973 ret_list_any, f_getjumplist},
974 {"getline", 1, 2, FEARG_1, NULL,
975 ret_f_getline, f_getline},
976 {"getloclist", 1, 2, 0, NULL,
977 ret_list_or_dict_1, f_getloclist},
978 {"getmarklist", 0, 1, FEARG_1, NULL,
979 ret_list_dict_any, f_getmarklist},
980 {"getmatches", 0, 1, 0, NULL,
981 ret_list_dict_any, f_getmatches},
982 {"getmousepos", 0, 0, 0, NULL,
983 ret_dict_number, f_getmousepos},
984 {"getpid", 0, 0, 0, NULL,
985 ret_number, f_getpid},
986 {"getpos", 1, 1, FEARG_1, NULL,
987 ret_list_number, f_getpos},
988 {"getqflist", 0, 1, 0, NULL,
989 ret_list_or_dict_0, f_getqflist},
990 {"getreg", 0, 3, FEARG_1, NULL,
991 ret_getreg, f_getreg},
992 {"getreginfo", 0, 1, FEARG_1, NULL,
993 ret_dict_any, f_getreginfo},
994 {"getregtype", 0, 1, FEARG_1, NULL,
995 ret_string, f_getregtype},
996 {"gettabinfo", 0, 1, FEARG_1, NULL,
997 ret_list_dict_any, f_gettabinfo},
998 {"gettabvar", 2, 3, FEARG_1, NULL,
999 ret_any, f_gettabvar},
1000 {"gettabwinvar", 3, 4, FEARG_1, NULL,
1001 ret_any, f_gettabwinvar},
1002 {"gettagstack", 0, 1, FEARG_1, NULL,
1003 ret_dict_any, f_gettagstack},
1004 {"gettext", 1, 1, FEARG_1, NULL,
1005 ret_string, f_gettext},
1006 {"getwininfo", 0, 1, FEARG_1, NULL,
1007 ret_list_dict_any, f_getwininfo},
1008 {"getwinpos", 0, 1, FEARG_1, NULL,
1009 ret_list_number, f_getwinpos},
1010 {"getwinposx", 0, 0, 0, NULL,
1011 ret_number, f_getwinposx},
1012 {"getwinposy", 0, 0, 0, NULL,
1013 ret_number, f_getwinposy},
1014 {"getwinvar", 2, 3, FEARG_1, NULL,
1015 ret_any, f_getwinvar},
1016 {"glob", 1, 4, FEARG_1, NULL,
1017 ret_any, f_glob},
1018 {"glob2regpat", 1, 1, FEARG_1, NULL,
1019 ret_string, f_glob2regpat},
1020 {"globpath", 2, 5, FEARG_2, NULL,
1021 ret_any, f_globpath},
1022 {"has", 1, 2, 0, NULL,
1023 ret_number, f_has},
1024 {"has_key", 2, 2, FEARG_1, NULL,
1025 ret_number, f_has_key},
1026 {"haslocaldir", 0, 2, FEARG_1, NULL,
1027 ret_number, f_haslocaldir},
1028 {"hasmapto", 1, 3, FEARG_1, NULL,
1029 ret_number, f_hasmapto},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001030 {"highlightID", 1, 1, FEARG_1, NULL, // obsolete
1031 ret_number, f_hlID},
1032 {"highlight_exists",1, 1, FEARG_1, NULL, // obsolete
1033 ret_number, f_hlexists},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001034 {"histadd", 2, 2, FEARG_2, NULL,
1035 ret_number, f_histadd},
1036 {"histdel", 1, 2, FEARG_1, NULL,
1037 ret_number, f_histdel},
1038 {"histget", 1, 2, FEARG_1, NULL,
1039 ret_string, f_histget},
1040 {"histnr", 1, 1, FEARG_1, NULL,
1041 ret_number, f_histnr},
1042 {"hlID", 1, 1, FEARG_1, NULL,
1043 ret_number, f_hlID},
1044 {"hlexists", 1, 1, FEARG_1, NULL,
1045 ret_number, f_hlexists},
1046 {"hostname", 0, 0, 0, NULL,
1047 ret_string, f_hostname},
1048 {"iconv", 3, 3, FEARG_1, NULL,
1049 ret_string, f_iconv},
1050 {"indent", 1, 1, FEARG_1, NULL,
1051 ret_number, f_indent},
1052 {"index", 2, 4, FEARG_1, NULL,
1053 ret_number, f_index},
1054 {"input", 1, 3, FEARG_1, NULL,
1055 ret_string, f_input},
1056 {"inputdialog", 1, 3, FEARG_1, NULL,
1057 ret_string, f_inputdialog},
1058 {"inputlist", 1, 1, FEARG_1, NULL,
1059 ret_number, f_inputlist},
1060 {"inputrestore", 0, 0, 0, NULL,
1061 ret_number, f_inputrestore},
1062 {"inputsave", 0, 0, 0, NULL,
1063 ret_number, f_inputsave},
1064 {"inputsecret", 1, 2, FEARG_1, NULL,
1065 ret_string, f_inputsecret},
Bram Moolenaarca174532020-10-21 16:42:22 +02001066 {"insert", 2, 3, FEARG_1, arg3_insert,
Bram Moolenaar94738d82020-10-21 14:25:07 +02001067 ret_first_arg, f_insert},
1068 {"interrupt", 0, 0, 0, NULL,
1069 ret_void, f_interrupt},
1070 {"invert", 1, 1, FEARG_1, NULL,
1071 ret_number, f_invert},
1072 {"isdirectory", 1, 1, FEARG_1, NULL,
1073 ret_number, f_isdirectory},
1074 {"isinf", 1, 1, FEARG_1, NULL,
1075 ret_number, MATH_FUNC(f_isinf)},
1076 {"islocked", 1, 1, FEARG_1, NULL,
1077 ret_number, f_islocked},
1078 {"isnan", 1, 1, FEARG_1, NULL,
1079 ret_number, MATH_FUNC(f_isnan)},
1080 {"items", 1, 1, FEARG_1, NULL,
1081 ret_list_any, f_items},
1082 {"job_getchannel", 1, 1, FEARG_1, NULL,
1083 ret_channel, JOB_FUNC(f_job_getchannel)},
1084 {"job_info", 0, 1, FEARG_1, NULL,
1085 ret_dict_any, JOB_FUNC(f_job_info)},
1086 {"job_setoptions", 2, 2, FEARG_1, NULL,
1087 ret_void, JOB_FUNC(f_job_setoptions)},
1088 {"job_start", 1, 2, FEARG_1, NULL,
1089 ret_job, JOB_FUNC(f_job_start)},
1090 {"job_status", 1, 1, FEARG_1, NULL,
1091 ret_string, JOB_FUNC(f_job_status)},
1092 {"job_stop", 1, 2, FEARG_1, NULL,
1093 ret_number, JOB_FUNC(f_job_stop)},
1094 {"join", 1, 2, FEARG_1, NULL,
1095 ret_string, f_join},
1096 {"js_decode", 1, 1, FEARG_1, NULL,
1097 ret_any, f_js_decode},
1098 {"js_encode", 1, 1, FEARG_1, NULL,
1099 ret_string, f_js_encode},
1100 {"json_decode", 1, 1, FEARG_1, NULL,
1101 ret_any, f_json_decode},
1102 {"json_encode", 1, 1, FEARG_1, NULL,
1103 ret_string, f_json_encode},
1104 {"keys", 1, 1, FEARG_1, NULL,
1105 ret_list_string, f_keys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001106 {"last_buffer_nr", 0, 0, 0, NULL, // obsolete
1107 ret_number, f_last_buffer_nr},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001108 {"len", 1, 1, FEARG_1, NULL,
1109 ret_number, f_len},
1110 {"libcall", 3, 3, FEARG_3, NULL,
1111 ret_string, f_libcall},
1112 {"libcallnr", 3, 3, FEARG_3, NULL,
1113 ret_number, f_libcallnr},
1114 {"line", 1, 2, FEARG_1, NULL,
1115 ret_number, f_line},
1116 {"line2byte", 1, 1, FEARG_1, NULL,
1117 ret_number, f_line2byte},
1118 {"lispindent", 1, 1, FEARG_1, NULL,
1119 ret_number, f_lispindent},
1120 {"list2str", 1, 2, FEARG_1, NULL,
1121 ret_string, f_list2str},
1122 {"listener_add", 1, 2, FEARG_2, NULL,
1123 ret_number, f_listener_add},
1124 {"listener_flush", 0, 1, FEARG_1, NULL,
1125 ret_void, f_listener_flush},
1126 {"listener_remove", 1, 1, FEARG_1, NULL,
1127 ret_number, f_listener_remove},
1128 {"localtime", 0, 0, 0, NULL,
1129 ret_number, f_localtime},
1130 {"log", 1, 1, FEARG_1, NULL,
1131 ret_float, FLOAT_FUNC(f_log)},
1132 {"log10", 1, 1, FEARG_1, NULL,
1133 ret_float, FLOAT_FUNC(f_log10)},
1134 {"luaeval", 1, 2, FEARG_1, NULL,
1135 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001136#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +01001137 f_luaeval
1138#else
1139 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001140#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001141 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001142 {"map", 2, 2, FEARG_1, NULL,
Bram Moolenaarea696852020-11-09 18:31:39 +01001143 ret_first_cont, f_map},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001144 {"maparg", 1, 4, FEARG_1, NULL,
1145 ret_maparg, f_maparg},
1146 {"mapcheck", 1, 3, FEARG_1, NULL,
1147 ret_string, f_mapcheck},
Bram Moolenaarea696852020-11-09 18:31:39 +01001148 {"mapnew", 2, 2, FEARG_1, NULL,
1149 ret_first_cont, f_mapnew},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001150 {"mapset", 3, 3, FEARG_1, NULL,
1151 ret_void, f_mapset},
1152 {"match", 2, 4, FEARG_1, NULL,
1153 ret_any, f_match},
1154 {"matchadd", 2, 5, FEARG_1, NULL,
1155 ret_number, f_matchadd},
1156 {"matchaddpos", 2, 5, FEARG_1, NULL,
1157 ret_number, f_matchaddpos},
1158 {"matcharg", 1, 1, FEARG_1, NULL,
1159 ret_list_string, f_matcharg},
1160 {"matchdelete", 1, 2, FEARG_1, NULL,
1161 ret_number, f_matchdelete},
1162 {"matchend", 2, 4, FEARG_1, NULL,
1163 ret_number, f_matchend},
1164 {"matchfuzzy", 2, 3, FEARG_1, NULL,
1165 ret_list_string, f_matchfuzzy},
1166 {"matchfuzzypos", 2, 3, FEARG_1, NULL,
1167 ret_list_any, f_matchfuzzypos},
1168 {"matchlist", 2, 4, FEARG_1, NULL,
1169 ret_list_string, f_matchlist},
1170 {"matchstr", 2, 4, FEARG_1, NULL,
1171 ret_string, f_matchstr},
1172 {"matchstrpos", 2, 4, FEARG_1, NULL,
1173 ret_list_any, f_matchstrpos},
1174 {"max", 1, 1, FEARG_1, NULL,
1175 ret_any, f_max},
1176 {"menu_info", 1, 2, FEARG_1, NULL,
1177 ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001178#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +01001179 f_menu_info
1180#else
1181 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001182#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001183 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001184 {"min", 1, 1, FEARG_1, NULL,
1185 ret_any, f_min},
1186 {"mkdir", 1, 3, FEARG_1, NULL,
1187 ret_number, f_mkdir},
1188 {"mode", 0, 1, FEARG_1, NULL,
1189 ret_string, f_mode},
1190 {"mzeval", 1, 1, FEARG_1, NULL,
1191 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001192#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001193 f_mzeval
1194#else
1195 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001196#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001197 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001198 {"nextnonblank", 1, 1, FEARG_1, NULL,
1199 ret_number, f_nextnonblank},
1200 {"nr2char", 1, 2, FEARG_1, NULL,
1201 ret_string, f_nr2char},
1202 {"or", 2, 2, FEARG_1, NULL,
1203 ret_number, f_or},
1204 {"pathshorten", 1, 2, FEARG_1, NULL,
1205 ret_string, f_pathshorten},
1206 {"perleval", 1, 1, FEARG_1, NULL,
1207 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001208#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +01001209 f_perleval
1210#else
1211 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001212#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001213 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001214 {"popup_atcursor", 2, 2, FEARG_1, NULL,
1215 ret_number, PROP_FUNC(f_popup_atcursor)},
1216 {"popup_beval", 2, 2, FEARG_1, NULL,
1217 ret_number, PROP_FUNC(f_popup_beval)},
1218 {"popup_clear", 0, 1, 0, NULL,
1219 ret_void, PROP_FUNC(f_popup_clear)},
1220 {"popup_close", 1, 2, FEARG_1, NULL,
1221 ret_void, PROP_FUNC(f_popup_close)},
1222 {"popup_create", 2, 2, FEARG_1, NULL,
1223 ret_number, PROP_FUNC(f_popup_create)},
1224 {"popup_dialog", 2, 2, FEARG_1, NULL,
1225 ret_number, PROP_FUNC(f_popup_dialog)},
1226 {"popup_filter_menu", 2, 2, 0, NULL,
1227 ret_bool, PROP_FUNC(f_popup_filter_menu)},
1228 {"popup_filter_yesno", 2, 2, 0, NULL,
1229 ret_bool, PROP_FUNC(f_popup_filter_yesno)},
1230 {"popup_findinfo", 0, 0, 0, NULL,
1231 ret_number, PROP_FUNC(f_popup_findinfo)},
1232 {"popup_findpreview", 0, 0, 0, NULL,
1233 ret_number, PROP_FUNC(f_popup_findpreview)},
1234 {"popup_getoptions", 1, 1, FEARG_1, NULL,
1235 ret_dict_any, PROP_FUNC(f_popup_getoptions)},
1236 {"popup_getpos", 1, 1, FEARG_1, NULL,
1237 ret_dict_any, PROP_FUNC(f_popup_getpos)},
1238 {"popup_hide", 1, 1, FEARG_1, NULL,
1239 ret_void, PROP_FUNC(f_popup_hide)},
1240 {"popup_list", 0, 0, 0, NULL,
1241 ret_list_number, PROP_FUNC(f_popup_list)},
1242 {"popup_locate", 2, 2, 0, NULL,
1243 ret_number, PROP_FUNC(f_popup_locate)},
1244 {"popup_menu", 2, 2, FEARG_1, NULL,
1245 ret_number, PROP_FUNC(f_popup_menu)},
1246 {"popup_move", 2, 2, FEARG_1, NULL,
1247 ret_void, PROP_FUNC(f_popup_move)},
1248 {"popup_notification", 2, 2, FEARG_1, NULL,
1249 ret_number, PROP_FUNC(f_popup_notification)},
1250 {"popup_setoptions", 2, 2, FEARG_1, NULL,
1251 ret_void, PROP_FUNC(f_popup_setoptions)},
1252 {"popup_settext", 2, 2, FEARG_1, NULL,
1253 ret_void, PROP_FUNC(f_popup_settext)},
1254 {"popup_show", 1, 1, FEARG_1, NULL,
1255 ret_void, PROP_FUNC(f_popup_show)},
1256 {"pow", 2, 2, FEARG_1, NULL,
1257 ret_float, FLOAT_FUNC(f_pow)},
1258 {"prevnonblank", 1, 1, FEARG_1, NULL,
1259 ret_number, f_prevnonblank},
1260 {"printf", 1, 19, FEARG_2, NULL,
1261 ret_string, f_printf},
1262 {"prompt_getprompt", 1, 1, FEARG_1, NULL,
1263 ret_string, JOB_FUNC(f_prompt_getprompt)},
1264 {"prompt_setcallback", 2, 2, FEARG_1, NULL,
1265 ret_void, JOB_FUNC(f_prompt_setcallback)},
1266 {"prompt_setinterrupt", 2, 2, FEARG_1, NULL,
1267 ret_void, JOB_FUNC(f_prompt_setinterrupt)},
1268 {"prompt_setprompt", 2, 2, FEARG_1, NULL,
1269 ret_void, JOB_FUNC(f_prompt_setprompt)},
1270 {"prop_add", 3, 3, FEARG_1, NULL,
1271 ret_void, PROP_FUNC(f_prop_add)},
1272 {"prop_clear", 1, 3, FEARG_1, NULL,
1273 ret_void, PROP_FUNC(f_prop_clear)},
1274 {"prop_find", 1, 2, FEARG_1, NULL,
1275 ret_dict_any, PROP_FUNC(f_prop_find)},
1276 {"prop_list", 1, 2, FEARG_1, NULL,
1277 ret_list_dict_any, PROP_FUNC(f_prop_list)},
1278 {"prop_remove", 1, 3, FEARG_1, NULL,
1279 ret_number, PROP_FUNC(f_prop_remove)},
1280 {"prop_type_add", 2, 2, FEARG_1, NULL,
1281 ret_void, PROP_FUNC(f_prop_type_add)},
1282 {"prop_type_change", 2, 2, FEARG_1, NULL,
1283 ret_void, PROP_FUNC(f_prop_type_change)},
1284 {"prop_type_delete", 1, 2, FEARG_1, NULL,
1285 ret_void, PROP_FUNC(f_prop_type_delete)},
1286 {"prop_type_get", 1, 2, FEARG_1, NULL,
1287 ret_dict_any, PROP_FUNC(f_prop_type_get)},
1288 {"prop_type_list", 0, 1, FEARG_1, NULL,
1289 ret_list_string, PROP_FUNC(f_prop_type_list)},
1290 {"pum_getpos", 0, 0, 0, NULL,
1291 ret_dict_number, f_pum_getpos},
1292 {"pumvisible", 0, 0, 0, NULL,
1293 ret_number, f_pumvisible},
1294 {"py3eval", 1, 1, FEARG_1, NULL,
1295 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001296#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +01001297 f_py3eval
1298#else
1299 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001300#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001301 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001302 {"pyeval", 1, 1, FEARG_1, NULL,
1303 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001304#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +01001305 f_pyeval
1306#else
1307 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001308#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001309 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001310 {"pyxeval", 1, 1, FEARG_1, NULL,
1311 ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001312#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +01001313 f_pyxeval
1314#else
1315 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001316#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001317 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001318 {"rand", 0, 1, FEARG_1, NULL,
1319 ret_number, f_rand},
1320 {"range", 1, 3, FEARG_1, NULL,
1321 ret_list_number, f_range},
1322 {"readdir", 1, 3, FEARG_1, NULL,
1323 ret_list_string, f_readdir},
1324 {"readdirex", 1, 3, FEARG_1, NULL,
1325 ret_list_dict_any, f_readdirex},
1326 {"readfile", 1, 3, FEARG_1, NULL,
1327 ret_any, f_readfile},
1328 {"reduce", 2, 3, FEARG_1, NULL,
1329 ret_any, f_reduce},
1330 {"reg_executing", 0, 0, 0, NULL,
1331 ret_string, f_reg_executing},
1332 {"reg_recording", 0, 0, 0, NULL,
1333 ret_string, f_reg_recording},
1334 {"reltime", 0, 2, FEARG_1, NULL,
1335 ret_list_any, f_reltime},
1336 {"reltimefloat", 1, 1, FEARG_1, NULL,
1337 ret_float, FLOAT_FUNC(f_reltimefloat)},
1338 {"reltimestr", 1, 1, FEARG_1, NULL,
1339 ret_string, f_reltimestr},
1340 {"remote_expr", 2, 4, FEARG_1, NULL,
1341 ret_string, f_remote_expr},
1342 {"remote_foreground", 1, 1, FEARG_1, NULL,
1343 ret_string, f_remote_foreground},
1344 {"remote_peek", 1, 2, FEARG_1, NULL,
1345 ret_number, f_remote_peek},
1346 {"remote_read", 1, 2, FEARG_1, NULL,
1347 ret_string, f_remote_read},
1348 {"remote_send", 2, 3, FEARG_1, NULL,
1349 ret_string, f_remote_send},
1350 {"remote_startserver", 1, 1, FEARG_1, NULL,
1351 ret_void, f_remote_startserver},
1352 {"remove", 2, 3, FEARG_1, NULL,
1353 ret_remove, f_remove},
1354 {"rename", 2, 2, FEARG_1, NULL,
1355 ret_number, f_rename},
1356 {"repeat", 2, 2, FEARG_1, NULL,
1357 ret_first_arg, f_repeat},
1358 {"resolve", 1, 1, FEARG_1, NULL,
1359 ret_string, f_resolve},
1360 {"reverse", 1, 1, FEARG_1, NULL,
1361 ret_first_arg, f_reverse},
1362 {"round", 1, 1, FEARG_1, NULL,
1363 ret_float, FLOAT_FUNC(f_round)},
1364 {"rubyeval", 1, 1, FEARG_1, NULL,
1365 ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001366#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +01001367 f_rubyeval
1368#else
1369 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001370#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001371 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001372 {"screenattr", 2, 2, FEARG_1, NULL,
1373 ret_number, f_screenattr},
1374 {"screenchar", 2, 2, FEARG_1, NULL,
1375 ret_number, f_screenchar},
1376 {"screenchars", 2, 2, FEARG_1, NULL,
1377 ret_list_number, f_screenchars},
1378 {"screencol", 0, 0, 0, NULL,
1379 ret_number, f_screencol},
1380 {"screenpos", 3, 3, FEARG_1, NULL,
1381 ret_dict_number, f_screenpos},
1382 {"screenrow", 0, 0, 0, NULL,
1383 ret_number, f_screenrow},
1384 {"screenstring", 2, 2, FEARG_1, NULL,
1385 ret_string, f_screenstring},
1386 {"search", 1, 5, FEARG_1, NULL,
1387 ret_number, f_search},
1388 {"searchcount", 0, 1, FEARG_1, NULL,
1389 ret_dict_any, f_searchcount},
1390 {"searchdecl", 1, 3, FEARG_1, NULL,
1391 ret_number, f_searchdecl},
1392 {"searchpair", 3, 7, 0, NULL,
1393 ret_number, f_searchpair},
1394 {"searchpairpos", 3, 7, 0, NULL,
1395 ret_list_number, f_searchpairpos},
1396 {"searchpos", 1, 5, FEARG_1, NULL,
1397 ret_list_number, f_searchpos},
1398 {"server2client", 2, 2, FEARG_1, NULL,
1399 ret_number, f_server2client},
1400 {"serverlist", 0, 0, 0, NULL,
1401 ret_string, f_serverlist},
1402 {"setbufline", 3, 3, FEARG_3, NULL,
1403 ret_number, f_setbufline},
1404 {"setbufvar", 3, 3, FEARG_3, NULL,
1405 ret_void, f_setbufvar},
1406 {"setcellwidths", 1, 1, FEARG_1, NULL,
1407 ret_void, f_setcellwidths},
Bram Moolenaar6f02b002021-01-10 20:22:54 +01001408 {"setcharpos", 2, 2, FEARG_2, NULL,
1409 ret_number, f_setcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001410 {"setcharsearch", 1, 1, FEARG_1, NULL,
1411 ret_void, f_setcharsearch},
1412 {"setcmdpos", 1, 1, FEARG_1, NULL,
1413 ret_number, f_setcmdpos},
Bram Moolenaar6f02b002021-01-10 20:22:54 +01001414 {"setcursorcharpos", 1, 3, FEARG_1, NULL,
1415 ret_number, f_setcursorcharpos},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001416 {"setenv", 2, 2, FEARG_2, NULL,
1417 ret_void, f_setenv},
1418 {"setfperm", 2, 2, FEARG_1, NULL,
1419 ret_number, f_setfperm},
1420 {"setline", 2, 2, FEARG_2, NULL,
1421 ret_number, f_setline},
1422 {"setloclist", 2, 4, FEARG_2, NULL,
1423 ret_number, f_setloclist},
1424 {"setmatches", 1, 2, FEARG_1, NULL,
1425 ret_number, f_setmatches},
1426 {"setpos", 2, 2, FEARG_2, NULL,
1427 ret_number, f_setpos},
1428 {"setqflist", 1, 3, FEARG_1, NULL,
1429 ret_number, f_setqflist},
1430 {"setreg", 2, 3, FEARG_2, NULL,
1431 ret_number, f_setreg},
1432 {"settabvar", 3, 3, FEARG_3, NULL,
1433 ret_void, f_settabvar},
1434 {"settabwinvar", 4, 4, FEARG_4, NULL,
1435 ret_void, f_settabwinvar},
1436 {"settagstack", 2, 3, FEARG_2, NULL,
1437 ret_number, f_settagstack},
1438 {"setwinvar", 3, 3, FEARG_3, NULL,
1439 ret_void, f_setwinvar},
1440 {"sha256", 1, 1, FEARG_1, NULL,
1441 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001442#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +01001443 f_sha256
1444#else
1445 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001446#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001447 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001448 {"shellescape", 1, 2, FEARG_1, NULL,
1449 ret_string, f_shellescape},
1450 {"shiftwidth", 0, 1, FEARG_1, NULL,
1451 ret_number, f_shiftwidth},
1452 {"sign_define", 1, 2, FEARG_1, NULL,
1453 ret_any, SIGN_FUNC(f_sign_define)},
1454 {"sign_getdefined", 0, 1, FEARG_1, NULL,
1455 ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
1456 {"sign_getplaced", 0, 2, FEARG_1, NULL,
1457 ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
1458 {"sign_jump", 3, 3, FEARG_1, NULL,
1459 ret_number, SIGN_FUNC(f_sign_jump)},
1460 {"sign_place", 4, 5, FEARG_1, NULL,
1461 ret_number, SIGN_FUNC(f_sign_place)},
1462 {"sign_placelist", 1, 1, FEARG_1, NULL,
1463 ret_list_number, SIGN_FUNC(f_sign_placelist)},
1464 {"sign_undefine", 0, 1, FEARG_1, NULL,
1465 ret_number, SIGN_FUNC(f_sign_undefine)},
1466 {"sign_unplace", 1, 2, FEARG_1, NULL,
1467 ret_number, SIGN_FUNC(f_sign_unplace)},
1468 {"sign_unplacelist", 1, 2, FEARG_1, NULL,
1469 ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
1470 {"simplify", 1, 1, FEARG_1, NULL,
1471 ret_string, f_simplify},
1472 {"sin", 1, 1, FEARG_1, NULL,
1473 ret_float, FLOAT_FUNC(f_sin)},
1474 {"sinh", 1, 1, FEARG_1, NULL,
1475 ret_float, FLOAT_FUNC(f_sinh)},
1476 {"sort", 1, 3, FEARG_1, NULL,
1477 ret_first_arg, f_sort},
1478 {"sound_clear", 0, 0, 0, NULL,
1479 ret_void, SOUND_FUNC(f_sound_clear)},
1480 {"sound_playevent", 1, 2, FEARG_1, NULL,
1481 ret_number, SOUND_FUNC(f_sound_playevent)},
1482 {"sound_playfile", 1, 2, FEARG_1, NULL,
1483 ret_number, SOUND_FUNC(f_sound_playfile)},
1484 {"sound_stop", 1, 1, FEARG_1, NULL,
1485 ret_void, SOUND_FUNC(f_sound_stop)},
1486 {"soundfold", 1, 1, FEARG_1, NULL,
1487 ret_string, f_soundfold},
1488 {"spellbadword", 0, 1, FEARG_1, NULL,
1489 ret_list_string, f_spellbadword},
1490 {"spellsuggest", 1, 3, FEARG_1, NULL,
1491 ret_list_string, f_spellsuggest},
1492 {"split", 1, 3, FEARG_1, NULL,
1493 ret_list_string, f_split},
1494 {"sqrt", 1, 1, FEARG_1, NULL,
1495 ret_float, FLOAT_FUNC(f_sqrt)},
1496 {"srand", 0, 1, FEARG_1, NULL,
1497 ret_list_number, f_srand},
1498 {"state", 0, 1, FEARG_1, NULL,
1499 ret_string, f_state},
1500 {"str2float", 1, 1, FEARG_1, NULL,
1501 ret_float, FLOAT_FUNC(f_str2float)},
1502 {"str2list", 1, 2, FEARG_1, NULL,
1503 ret_list_number, f_str2list},
1504 {"str2nr", 1, 3, FEARG_1, NULL,
1505 ret_number, f_str2nr},
1506 {"strcharpart", 2, 3, FEARG_1, NULL,
1507 ret_string, f_strcharpart},
1508 {"strchars", 1, 2, FEARG_1, NULL,
1509 ret_number, f_strchars},
1510 {"strdisplaywidth", 1, 2, FEARG_1, NULL,
1511 ret_number, f_strdisplaywidth},
1512 {"strftime", 1, 2, FEARG_1, NULL,
1513 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001514#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001515 f_strftime
1516#else
1517 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001518#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001519 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001520 {"strgetchar", 2, 2, FEARG_1, NULL,
1521 ret_number, f_strgetchar},
1522 {"stridx", 2, 3, FEARG_1, NULL,
1523 ret_number, f_stridx},
1524 {"string", 1, 1, FEARG_1, NULL,
1525 ret_string, f_string},
1526 {"strlen", 1, 1, FEARG_1, NULL,
1527 ret_number, f_strlen},
1528 {"strpart", 2, 4, FEARG_1, NULL,
1529 ret_string, f_strpart},
1530 {"strptime", 2, 2, FEARG_1, NULL,
1531 ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +01001532#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001533 f_strptime
1534#else
1535 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +01001536#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001537 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001538 {"strridx", 2, 3, FEARG_1, NULL,
1539 ret_number, f_strridx},
1540 {"strtrans", 1, 1, FEARG_1, NULL,
1541 ret_string, f_strtrans},
1542 {"strwidth", 1, 1, FEARG_1, NULL,
1543 ret_number, f_strwidth},
1544 {"submatch", 1, 2, FEARG_1, NULL,
1545 ret_string, f_submatch},
1546 {"substitute", 4, 4, FEARG_1, NULL,
1547 ret_string, f_substitute},
1548 {"swapinfo", 1, 1, FEARG_1, NULL,
1549 ret_dict_any, f_swapinfo},
1550 {"swapname", 1, 1, FEARG_1, NULL,
1551 ret_string, f_swapname},
1552 {"synID", 3, 3, 0, NULL,
1553 ret_number, f_synID},
1554 {"synIDattr", 2, 3, FEARG_1, NULL,
1555 ret_string, f_synIDattr},
1556 {"synIDtrans", 1, 1, FEARG_1, NULL,
1557 ret_number, f_synIDtrans},
1558 {"synconcealed", 2, 2, 0, NULL,
1559 ret_list_any, f_synconcealed},
1560 {"synstack", 2, 2, 0, NULL,
1561 ret_list_number, f_synstack},
1562 {"system", 1, 2, FEARG_1, NULL,
1563 ret_string, f_system},
1564 {"systemlist", 1, 2, FEARG_1, NULL,
1565 ret_list_string, f_systemlist},
1566 {"tabpagebuflist", 0, 1, FEARG_1, NULL,
1567 ret_list_number, f_tabpagebuflist},
1568 {"tabpagenr", 0, 1, 0, NULL,
1569 ret_number, f_tabpagenr},
1570 {"tabpagewinnr", 1, 2, FEARG_1, NULL,
1571 ret_number, f_tabpagewinnr},
1572 {"tagfiles", 0, 0, 0, NULL,
1573 ret_list_string, f_tagfiles},
1574 {"taglist", 1, 2, FEARG_1, NULL,
1575 ret_list_dict_any, f_taglist},
1576 {"tan", 1, 1, FEARG_1, NULL,
1577 ret_float, FLOAT_FUNC(f_tan)},
1578 {"tanh", 1, 1, FEARG_1, NULL,
1579 ret_float, FLOAT_FUNC(f_tanh)},
1580 {"tempname", 0, 0, 0, NULL,
1581 ret_string, f_tempname},
1582 {"term_dumpdiff", 2, 3, FEARG_1, NULL,
1583 ret_number, TERM_FUNC(f_term_dumpdiff)},
1584 {"term_dumpload", 1, 2, FEARG_1, NULL,
1585 ret_number, TERM_FUNC(f_term_dumpload)},
1586 {"term_dumpwrite", 2, 3, FEARG_2, NULL,
1587 ret_void, TERM_FUNC(f_term_dumpwrite)},
1588 {"term_getaltscreen", 1, 1, FEARG_1, NULL,
1589 ret_number, TERM_FUNC(f_term_getaltscreen)},
1590 {"term_getansicolors", 1, 1, FEARG_1, NULL,
1591 ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001592#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001593 f_term_getansicolors
1594#else
1595 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001596#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001597 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001598 {"term_getattr", 2, 2, FEARG_1, NULL,
1599 ret_number, TERM_FUNC(f_term_getattr)},
1600 {"term_getcursor", 1, 1, FEARG_1, NULL,
1601 ret_list_any, TERM_FUNC(f_term_getcursor)},
1602 {"term_getjob", 1, 1, FEARG_1, NULL,
1603 ret_job, TERM_FUNC(f_term_getjob)},
1604 {"term_getline", 2, 2, FEARG_1, NULL,
1605 ret_string, TERM_FUNC(f_term_getline)},
1606 {"term_getscrolled", 1, 1, FEARG_1, NULL,
1607 ret_number, TERM_FUNC(f_term_getscrolled)},
1608 {"term_getsize", 1, 1, FEARG_1, NULL,
1609 ret_list_number, TERM_FUNC(f_term_getsize)},
1610 {"term_getstatus", 1, 1, FEARG_1, NULL,
1611 ret_string, TERM_FUNC(f_term_getstatus)},
1612 {"term_gettitle", 1, 1, FEARG_1, NULL,
1613 ret_string, TERM_FUNC(f_term_gettitle)},
1614 {"term_gettty", 1, 2, FEARG_1, NULL,
1615 ret_string, TERM_FUNC(f_term_gettty)},
1616 {"term_list", 0, 0, 0, NULL,
1617 ret_list_number, TERM_FUNC(f_term_list)},
1618 {"term_scrape", 2, 2, FEARG_1, NULL,
1619 ret_list_dict_any, TERM_FUNC(f_term_scrape)},
1620 {"term_sendkeys", 2, 2, FEARG_1, NULL,
1621 ret_void, TERM_FUNC(f_term_sendkeys)},
1622 {"term_setansicolors", 2, 2, FEARG_1, NULL,
1623 ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001624#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001625 f_term_setansicolors
1626#else
1627 NULL
1628#endif
1629 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001630 {"term_setapi", 2, 2, FEARG_1, NULL,
1631 ret_void, TERM_FUNC(f_term_setapi)},
1632 {"term_setkill", 2, 2, FEARG_1, NULL,
1633 ret_void, TERM_FUNC(f_term_setkill)},
1634 {"term_setrestore", 2, 2, FEARG_1, NULL,
1635 ret_void, TERM_FUNC(f_term_setrestore)},
1636 {"term_setsize", 3, 3, FEARG_1, NULL,
1637 ret_void, TERM_FUNC(f_term_setsize)},
1638 {"term_start", 1, 2, FEARG_1, NULL,
1639 ret_number, TERM_FUNC(f_term_start)},
1640 {"term_wait", 1, 2, FEARG_1, NULL,
1641 ret_void, TERM_FUNC(f_term_wait)},
1642 {"terminalprops", 0, 0, 0, NULL,
1643 ret_dict_string, f_terminalprops},
1644 {"test_alloc_fail", 3, 3, FEARG_1, NULL,
1645 ret_void, f_test_alloc_fail},
1646 {"test_autochdir", 0, 0, 0, NULL,
1647 ret_void, f_test_autochdir},
1648 {"test_feedinput", 1, 1, FEARG_1, NULL,
1649 ret_void, f_test_feedinput},
1650 {"test_garbagecollect_now", 0, 0, 0, NULL,
1651 ret_void, f_test_garbagecollect_now},
1652 {"test_garbagecollect_soon", 0, 0, 0, NULL,
1653 ret_void, f_test_garbagecollect_soon},
1654 {"test_getvalue", 1, 1, FEARG_1, NULL,
1655 ret_number, f_test_getvalue},
1656 {"test_ignore_error", 1, 1, FEARG_1, NULL,
1657 ret_void, f_test_ignore_error},
1658 {"test_null_blob", 0, 0, 0, NULL,
1659 ret_blob, f_test_null_blob},
1660 {"test_null_channel", 0, 0, 0, NULL,
1661 ret_channel, JOB_FUNC(f_test_null_channel)},
1662 {"test_null_dict", 0, 0, 0, NULL,
1663 ret_dict_any, f_test_null_dict},
1664 {"test_null_function", 0, 0, 0, NULL,
1665 ret_func_any, f_test_null_function},
1666 {"test_null_job", 0, 0, 0, NULL,
1667 ret_job, JOB_FUNC(f_test_null_job)},
1668 {"test_null_list", 0, 0, 0, NULL,
1669 ret_list_any, f_test_null_list},
1670 {"test_null_partial", 0, 0, 0, NULL,
1671 ret_func_any, f_test_null_partial},
1672 {"test_null_string", 0, 0, 0, NULL,
1673 ret_string, f_test_null_string},
1674 {"test_option_not_set", 1, 1, FEARG_1, NULL,
1675 ret_void, f_test_option_not_set},
1676 {"test_override", 2, 2, FEARG_2, NULL,
1677 ret_void, f_test_override},
1678 {"test_refcount", 1, 1, FEARG_1, NULL,
1679 ret_number, f_test_refcount},
1680 {"test_scrollbar", 3, 3, FEARG_2, NULL,
1681 ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +02001682#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +01001683 f_test_scrollbar
1684#else
1685 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +02001686#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001687 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001688 {"test_setmouse", 2, 2, 0, NULL,
1689 ret_void, f_test_setmouse},
1690 {"test_settime", 1, 1, FEARG_1, NULL,
1691 ret_void, f_test_settime},
1692 {"test_srand_seed", 0, 1, FEARG_1, NULL,
1693 ret_void, f_test_srand_seed},
1694 {"test_unknown", 0, 0, 0, NULL,
1695 ret_any, f_test_unknown},
1696 {"test_void", 0, 0, 0, NULL,
1697 ret_void, f_test_void},
1698 {"timer_info", 0, 1, FEARG_1, NULL,
1699 ret_list_dict_any, TIMER_FUNC(f_timer_info)},
1700 {"timer_pause", 2, 2, FEARG_1, NULL,
1701 ret_void, TIMER_FUNC(f_timer_pause)},
1702 {"timer_start", 2, 3, FEARG_1, NULL,
1703 ret_number, TIMER_FUNC(f_timer_start)},
1704 {"timer_stop", 1, 1, FEARG_1, NULL,
1705 ret_void, TIMER_FUNC(f_timer_stop)},
1706 {"timer_stopall", 0, 0, 0, NULL,
1707 ret_void, TIMER_FUNC(f_timer_stopall)},
1708 {"tolower", 1, 1, FEARG_1, NULL,
1709 ret_string, f_tolower},
1710 {"toupper", 1, 1, FEARG_1, NULL,
1711 ret_string, f_toupper},
1712 {"tr", 3, 3, FEARG_1, NULL,
1713 ret_string, f_tr},
1714 {"trim", 1, 3, FEARG_1, NULL,
1715 ret_string, f_trim},
1716 {"trunc", 1, 1, FEARG_1, NULL,
1717 ret_float, FLOAT_FUNC(f_trunc)},
1718 {"type", 1, 1, FEARG_1, NULL,
1719 ret_number, f_type},
1720 {"undofile", 1, 1, FEARG_1, NULL,
1721 ret_string, f_undofile},
1722 {"undotree", 0, 0, 0, NULL,
1723 ret_dict_any, f_undotree},
1724 {"uniq", 1, 3, FEARG_1, NULL,
1725 ret_list_any, f_uniq},
1726 {"values", 1, 1, FEARG_1, NULL,
1727 ret_list_any, f_values},
1728 {"virtcol", 1, 1, FEARG_1, NULL,
1729 ret_number, f_virtcol},
1730 {"visualmode", 0, 1, 0, NULL,
1731 ret_string, f_visualmode},
1732 {"wildmenumode", 0, 0, 0, NULL,
1733 ret_number, f_wildmenumode},
1734 {"win_execute", 2, 3, FEARG_2, NULL,
1735 ret_string, f_win_execute},
1736 {"win_findbuf", 1, 1, FEARG_1, NULL,
1737 ret_list_number, f_win_findbuf},
1738 {"win_getid", 0, 2, FEARG_1, NULL,
1739 ret_number, f_win_getid},
1740 {"win_gettype", 0, 1, FEARG_1, NULL,
1741 ret_string, f_win_gettype},
1742 {"win_gotoid", 1, 1, FEARG_1, NULL,
1743 ret_number, f_win_gotoid},
1744 {"win_id2tabwin", 1, 1, FEARG_1, NULL,
1745 ret_list_number, f_win_id2tabwin},
1746 {"win_id2win", 1, 1, FEARG_1, NULL,
1747 ret_number, f_win_id2win},
1748 {"win_screenpos", 1, 1, FEARG_1, NULL,
1749 ret_list_number, f_win_screenpos},
1750 {"win_splitmove", 2, 3, FEARG_1, NULL,
1751 ret_number, f_win_splitmove},
1752 {"winbufnr", 1, 1, FEARG_1, NULL,
1753 ret_number, f_winbufnr},
1754 {"wincol", 0, 0, 0, NULL,
1755 ret_number, f_wincol},
1756 {"windowsversion", 0, 0, 0, NULL,
1757 ret_string, f_windowsversion},
1758 {"winheight", 1, 1, FEARG_1, NULL,
1759 ret_number, f_winheight},
1760 {"winlayout", 0, 1, FEARG_1, NULL,
1761 ret_list_any, f_winlayout},
1762 {"winline", 0, 0, 0, NULL,
1763 ret_number, f_winline},
1764 {"winnr", 0, 1, FEARG_1, NULL,
1765 ret_number, f_winnr},
1766 {"winrestcmd", 0, 0, 0, NULL,
1767 ret_string, f_winrestcmd},
1768 {"winrestview", 1, 1, FEARG_1, NULL,
1769 ret_void, f_winrestview},
1770 {"winsaveview", 0, 0, 0, NULL,
Bram Moolenaar43b69b32021-01-07 20:23:33 +01001771 ret_dict_number, f_winsaveview},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001772 {"winwidth", 1, 1, FEARG_1, NULL,
1773 ret_number, f_winwidth},
1774 {"wordcount", 0, 0, 0, NULL,
1775 ret_dict_number, f_wordcount},
1776 {"writefile", 2, 3, FEARG_1, NULL,
1777 ret_number, f_writefile},
1778 {"xor", 2, 2, FEARG_1, NULL,
1779 ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001780};
1781
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001782/*
1783 * Function given to ExpandGeneric() to obtain the list of internal
1784 * or user defined function names.
1785 */
1786 char_u *
1787get_function_name(expand_T *xp, int idx)
1788{
1789 static int intidx = -1;
1790 char_u *name;
1791
1792 if (idx == 0)
1793 intidx = -1;
1794 if (intidx < 0)
1795 {
1796 name = get_user_func_name(xp, idx);
1797 if (name != NULL)
1798 return name;
1799 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001800 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001801 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001802 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001803 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001804 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001805 STRCAT(IObuff, ")");
1806 return IObuff;
1807 }
1808
1809 return NULL;
1810}
1811
1812/*
1813 * Function given to ExpandGeneric() to obtain the list of internal or
1814 * user defined variable or function names.
1815 */
1816 char_u *
1817get_expr_name(expand_T *xp, int idx)
1818{
1819 static int intidx = -1;
1820 char_u *name;
1821
1822 if (idx == 0)
1823 intidx = -1;
1824 if (intidx < 0)
1825 {
1826 name = get_function_name(xp, idx);
1827 if (name != NULL)
1828 return name;
1829 }
1830 return get_user_var_name(xp, ++intidx);
1831}
1832
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001833/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001834 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001835 * Return index, or -1 if not found or "implemented" is TRUE and the function
1836 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001837 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001838 static int
1839find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001840{
1841 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001842 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001843 int cmp;
1844 int x;
1845
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001846 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001847
1848 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001849 while (first <= last)
1850 {
1851 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001852 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001853 if (cmp < 0)
1854 last = x - 1;
1855 else if (cmp > 0)
1856 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001857 else if (implemented && global_functions[x].f_func == NULL)
1858 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001859 else
1860 return x;
1861 }
1862 return -1;
1863}
1864
Bram Moolenaar15c47602020-03-26 22:16:48 +01001865/*
1866 * Find internal function "name" in table "global_functions".
1867 * Return index, or -1 if not found or the function is not implemented.
1868 */
1869 int
1870find_internal_func(char_u *name)
1871{
1872 return find_internal_func_opt(name, TRUE);
1873}
1874
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001875 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001876has_internal_func(char_u *name)
1877{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001878 return find_internal_func_opt(name, TRUE) >= 0;
1879}
1880
1881 static int
1882has_internal_func_name(char_u *name)
1883{
1884 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001885}
1886
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001887 char *
1888internal_func_name(int idx)
1889{
1890 return global_functions[idx].f_name;
1891}
1892
Bram Moolenaar94738d82020-10-21 14:25:07 +02001893/*
1894 * Check the argument types for builting function "idx".
1895 * Uses the list of types on the type stack: "types".
1896 * Return FAIL and gives an error message when a type is wrong.
1897 */
1898 int
Bram Moolenaarca174532020-10-21 16:42:22 +02001899internal_func_check_arg_types(type_T **types, int idx, int argcount)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001900{
1901 argcheck_T *argchecks = global_functions[idx].f_argcheck;
1902 int i;
1903
1904 if (argchecks != NULL)
1905 {
1906 argcontext_T context;
1907
1908 context.arg_count = argcount;
Bram Moolenaarca174532020-10-21 16:42:22 +02001909 context.arg_types = types;
Bram Moolenaar94738d82020-10-21 14:25:07 +02001910 for (i = 0; i < argcount; ++i)
1911 if (argchecks[i] != NULL)
1912 {
1913 context.arg_idx = i;
Bram Moolenaarca174532020-10-21 16:42:22 +02001914 if (argchecks[i](types[i], &context) == FAIL)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001915 return FAIL;
1916 }
1917 }
1918 return OK;
1919}
1920
Bram Moolenaara1224cb2020-10-22 12:31:49 +02001921/*
1922 * Call the "f_retfunc" function to obtain the return type of function "idx".
1923 * "argtypes" is the list of argument types or NULL when there are no
1924 * arguments.
1925 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001926 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001927internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001928{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001929 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001930}
1931
1932/*
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01001933 * Return TRUE if "idx" is for the map() function.
1934 */
1935 int
1936internal_func_is_map(int idx)
1937{
1938 return global_functions[idx].f_func == f_map;
1939}
1940
1941/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001942 * Check the argument count to use for internal function "idx".
Bram Moolenaar389df252020-07-09 21:20:47 +02001943 * Returns -1 for failure, 0 if no method base accepted, 1 if method base is
1944 * first argument, 2 if method base is second argument, etc. 9 if method base
1945 * is last argument.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001946 */
1947 int
1948check_internal_func(int idx, int argcount)
1949{
1950 int res;
1951 char *name;
1952
1953 if (argcount < global_functions[idx].f_min_argc)
1954 res = FCERR_TOOFEW;
1955 else if (argcount > global_functions[idx].f_max_argc)
1956 res = FCERR_TOOMANY;
1957 else
Bram Moolenaar389df252020-07-09 21:20:47 +02001958 return global_functions[idx].f_argtype;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001959
1960 name = internal_func_name(idx);
1961 if (res == FCERR_TOOMANY)
1962 semsg(_(e_toomanyarg), name);
1963 else
1964 semsg(_(e_toofewarg), name);
Bram Moolenaar389df252020-07-09 21:20:47 +02001965 return -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001966}
1967
Bram Moolenaarac92e252019-08-03 21:58:38 +02001968 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001969call_internal_func(
1970 char_u *name,
1971 int argcount,
1972 typval_T *argvars,
1973 typval_T *rettv)
1974{
1975 int i;
1976
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001977 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001978 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001979 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001980 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001981 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001982 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001983 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001984 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001985 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001986 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001987}
1988
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001989 void
1990call_internal_func_by_idx(
1991 int idx,
1992 typval_T *argvars,
1993 typval_T *rettv)
1994{
1995 global_functions[idx].f_func(argvars, rettv);
1996}
1997
Bram Moolenaarac92e252019-08-03 21:58:38 +02001998/*
1999 * Invoke a method for base->method().
2000 */
2001 int
2002call_internal_method(
2003 char_u *name,
2004 int argcount,
2005 typval_T *argvars,
2006 typval_T *rettv,
2007 typval_T *basetv)
2008{
2009 int i;
2010 int fi;
2011 typval_T argv[MAX_FUNC_ARGS + 1];
2012
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002013 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02002014 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01002015 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02002016 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01002017 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002018 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002019 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002020 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01002021 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02002022
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002023 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002024 {
2025 // base value goes last
2026 for (i = 0; i < argcount; ++i)
2027 argv[i] = argvars[i];
2028 argv[argcount] = *basetv;
2029 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002030 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002031 {
2032 // base value goes second
2033 argv[0] = argvars[0];
2034 argv[1] = *basetv;
2035 for (i = 1; i < argcount; ++i)
2036 argv[i + 1] = argvars[i];
2037 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02002038 else if (global_functions[fi].f_argtype == FEARG_3)
2039 {
2040 // base value goes third
2041 argv[0] = argvars[0];
2042 argv[1] = argvars[1];
2043 argv[2] = *basetv;
2044 for (i = 2; i < argcount; ++i)
2045 argv[i + 1] = argvars[i];
2046 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02002047 else if (global_functions[fi].f_argtype == FEARG_4)
2048 {
2049 // base value goes fourth
2050 argv[0] = argvars[0];
2051 argv[1] = argvars[1];
2052 argv[2] = argvars[2];
2053 argv[3] = *basetv;
2054 for (i = 3; i < argcount; ++i)
2055 argv[i + 1] = argvars[i];
2056 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02002057 else
2058 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002059 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02002060 argv[0] = *basetv;
2061 for (i = 0; i < argcount; ++i)
2062 argv[i + 1] = argvars[i];
2063 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02002064 argv[argcount + 1].v_type = VAR_UNKNOWN;
2065
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002066 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01002067 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002068}
2069
2070/*
2071 * Return TRUE for a non-zero Number and a non-empty String.
2072 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02002073 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002074non_zero_arg(typval_T *argvars)
2075{
2076 return ((argvars[0].v_type == VAR_NUMBER
2077 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002078 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002079 && argvars[0].vval.v_number == VVAL_TRUE)
2080 || (argvars[0].v_type == VAR_STRING
2081 && argvars[0].vval.v_string != NULL
2082 && *argvars[0].vval.v_string != NUL));
2083}
2084
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002085#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002086/*
2087 * Get the float value of "argvars[0]" into "f".
2088 * Returns FAIL when the argument is not a Number or Float.
2089 */
2090 static int
2091get_float_arg(typval_T *argvars, float_T *f)
2092{
2093 if (argvars[0].v_type == VAR_FLOAT)
2094 {
2095 *f = argvars[0].vval.v_float;
2096 return OK;
2097 }
2098 if (argvars[0].v_type == VAR_NUMBER)
2099 {
2100 *f = (float_T)argvars[0].vval.v_number;
2101 return OK;
2102 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002103 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002104 return FAIL;
2105}
2106
2107/*
2108 * "abs(expr)" function
2109 */
2110 static void
2111f_abs(typval_T *argvars, typval_T *rettv)
2112{
2113 if (argvars[0].v_type == VAR_FLOAT)
2114 {
2115 rettv->v_type = VAR_FLOAT;
2116 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
2117 }
2118 else
2119 {
2120 varnumber_T n;
2121 int error = FALSE;
2122
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002123 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002124 if (error)
2125 rettv->vval.v_number = -1;
2126 else if (n > 0)
2127 rettv->vval.v_number = n;
2128 else
2129 rettv->vval.v_number = -n;
2130 }
2131}
2132
2133/*
2134 * "acos()" function
2135 */
2136 static void
2137f_acos(typval_T *argvars, typval_T *rettv)
2138{
2139 float_T f = 0.0;
2140
2141 rettv->v_type = VAR_FLOAT;
2142 if (get_float_arg(argvars, &f) == OK)
2143 rettv->vval.v_float = acos(f);
2144 else
2145 rettv->vval.v_float = 0.0;
2146}
2147#endif
2148
2149/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002150 * "and(expr, expr)" function
2151 */
2152 static void
2153f_and(typval_T *argvars, typval_T *rettv)
2154{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002155 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
2156 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02002157}
2158
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002159#ifdef FEAT_FLOAT
2160/*
2161 * "asin()" function
2162 */
2163 static void
2164f_asin(typval_T *argvars, typval_T *rettv)
2165{
2166 float_T f = 0.0;
2167
2168 rettv->v_type = VAR_FLOAT;
2169 if (get_float_arg(argvars, &f) == OK)
2170 rettv->vval.v_float = asin(f);
2171 else
2172 rettv->vval.v_float = 0.0;
2173}
2174
2175/*
2176 * "atan()" function
2177 */
2178 static void
2179f_atan(typval_T *argvars, typval_T *rettv)
2180{
2181 float_T f = 0.0;
2182
2183 rettv->v_type = VAR_FLOAT;
2184 if (get_float_arg(argvars, &f) == OK)
2185 rettv->vval.v_float = atan(f);
2186 else
2187 rettv->vval.v_float = 0.0;
2188}
2189
2190/*
2191 * "atan2()" function
2192 */
2193 static void
2194f_atan2(typval_T *argvars, typval_T *rettv)
2195{
2196 float_T fx = 0.0, fy = 0.0;
2197
2198 rettv->v_type = VAR_FLOAT;
2199 if (get_float_arg(argvars, &fx) == OK
2200 && get_float_arg(&argvars[1], &fy) == OK)
2201 rettv->vval.v_float = atan2(fx, fy);
2202 else
2203 rettv->vval.v_float = 0.0;
2204}
2205#endif
2206
2207/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01002208 * "balloon_show()" function
2209 */
2210#ifdef FEAT_BEVAL
2211 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002212f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
2213{
2214 rettv->v_type = VAR_STRING;
2215 if (balloonEval != NULL)
2216 {
2217 if (balloonEval->msg == NULL)
2218 rettv->vval.v_string = NULL;
2219 else
2220 rettv->vval.v_string = vim_strsave(balloonEval->msg);
2221 }
2222}
2223
2224 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01002225f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
2226{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01002227 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002228 {
2229 if (argvars[0].v_type == VAR_LIST
2230# ifdef FEAT_GUI
2231 && !gui.in_use
2232# endif
2233 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002234 {
2235 list_T *l = argvars[0].vval.v_list;
2236
2237 // empty list removes the balloon
2238 post_balloon(balloonEval, NULL,
2239 l == NULL || l->lv_len == 0 ? NULL : l);
2240 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002241 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002242 {
2243 char_u *mesg = tv_get_string_chk(&argvars[0]);
2244
2245 if (mesg != NULL)
2246 // empty string removes the balloon
2247 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
2248 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002249 }
2250}
2251
Bram Moolenaar669a8282017-11-19 20:13:05 +01002252# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002253 static void
2254f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
2255{
2256 if (rettv_list_alloc(rettv) == OK)
2257 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002258 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002259
2260 if (msg != NULL)
2261 {
2262 pumitem_T *array;
2263 int size = split_message(msg, &array);
2264 int i;
2265
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002266 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01002267 for (i = 1; i < size - 1; ++i)
2268 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01002269 while (size > 0)
2270 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002271 vim_free(array);
2272 }
2273 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01002274}
Bram Moolenaar669a8282017-11-19 20:13:05 +01002275# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01002276#endif
2277
2278/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002279 * Get the buffer from "arg" and give an error and return NULL if it is not
2280 * valid.
2281 */
Bram Moolenaara3347722019-05-11 21:14:24 +02002282 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002283get_buf_arg(typval_T *arg)
2284{
2285 buf_T *buf;
2286
2287 ++emsg_off;
2288 buf = tv_get_buf(arg, FALSE);
2289 --emsg_off;
2290 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002291 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002292 return buf;
2293}
2294
2295/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002296 * "byte2line(byte)" function
2297 */
2298 static void
2299f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
2300{
2301#ifndef FEAT_BYTEOFF
2302 rettv->vval.v_number = -1;
2303#else
2304 long boff = 0;
2305
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002306 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002307 if (boff < 0)
2308 rettv->vval.v_number = -1;
2309 else
2310 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
2311 (linenr_T)0, &boff);
2312#endif
2313}
2314
2315 static void
2316byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
2317{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002318 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002319 char_u *str;
2320 varnumber_T idx;
2321
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002322 str = tv_get_string_chk(&argvars[0]);
2323 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002324 rettv->vval.v_number = -1;
2325 if (str == NULL || idx < 0)
2326 return;
2327
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002328 t = str;
2329 for ( ; idx > 0; idx--)
2330 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002331 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002332 return;
2333 if (enc_utf8 && comp)
2334 t += utf_ptr2len(t);
2335 else
2336 t += (*mb_ptr2len)(t);
2337 }
2338 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002339}
2340
2341/*
2342 * "byteidx()" function
2343 */
2344 static void
2345f_byteidx(typval_T *argvars, typval_T *rettv)
2346{
2347 byteidx(argvars, rettv, FALSE);
2348}
2349
2350/*
2351 * "byteidxcomp()" function
2352 */
2353 static void
2354f_byteidxcomp(typval_T *argvars, typval_T *rettv)
2355{
2356 byteidx(argvars, rettv, TRUE);
2357}
2358
2359/*
2360 * "call(func, arglist [, dict])" function
2361 */
2362 static void
2363f_call(typval_T *argvars, typval_T *rettv)
2364{
2365 char_u *func;
2366 partial_T *partial = NULL;
2367 dict_T *selfdict = NULL;
2368
2369 if (argvars[1].v_type != VAR_LIST)
2370 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002371 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002372 return;
2373 }
2374 if (argvars[1].vval.v_list == NULL)
2375 return;
2376
2377 if (argvars[0].v_type == VAR_FUNC)
2378 func = argvars[0].vval.v_string;
2379 else if (argvars[0].v_type == VAR_PARTIAL)
2380 {
2381 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002382 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002383 }
2384 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002385 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002386 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002387 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002388
2389 if (argvars[2].v_type != VAR_UNKNOWN)
2390 {
2391 if (argvars[2].v_type != VAR_DICT)
2392 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002393 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002394 return;
2395 }
2396 selfdict = argvars[2].vval.v_dict;
2397 }
2398
2399 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
2400}
2401
2402#ifdef FEAT_FLOAT
2403/*
2404 * "ceil({float})" function
2405 */
2406 static void
2407f_ceil(typval_T *argvars, typval_T *rettv)
2408{
2409 float_T f = 0.0;
2410
2411 rettv->v_type = VAR_FLOAT;
2412 if (get_float_arg(argvars, &f) == OK)
2413 rettv->vval.v_float = ceil(f);
2414 else
2415 rettv->vval.v_float = 0.0;
2416}
2417#endif
2418
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002419/*
2420 * "changenr()" function
2421 */
2422 static void
2423f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
2424{
2425 rettv->vval.v_number = curbuf->b_u_seq_cur;
2426}
2427
2428/*
2429 * "char2nr(string)" function
2430 */
2431 static void
2432f_char2nr(typval_T *argvars, typval_T *rettv)
2433{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002434 if (has_mbyte)
2435 {
2436 int utf8 = 0;
2437
2438 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar24f77502020-09-04 19:50:57 +02002439 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002440
2441 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01002442 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002443 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002444 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002445 }
2446 else
Bram Moolenaar13505972019-01-24 15:04:48 +01002447 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002448}
2449
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002450/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002451 * Get the current cursor column and store it in 'rettv'. If 'charcol' is TRUE,
2452 * returns the character index of the column. Otherwise, returns the byte index
2453 * of the column.
2454 */
2455 static void
2456get_col(typval_T *argvars, typval_T *rettv, int charcol)
2457{
2458 colnr_T col = 0;
2459 pos_T *fp;
2460 int fnum = curbuf->b_fnum;
2461
2462 fp = var2fpos(&argvars[0], FALSE, &fnum, charcol);
2463 if (fp != NULL && fnum == curbuf->b_fnum)
2464 {
2465 if (fp->col == MAXCOL)
2466 {
2467 // '> can be MAXCOL, get the length of the line then
2468 if (fp->lnum <= curbuf->b_ml.ml_line_count)
2469 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
2470 else
2471 col = MAXCOL;
2472 }
2473 else
2474 {
2475 col = fp->col + 1;
2476 // col(".") when the cursor is on the NUL at the end of the line
2477 // because of "coladd" can be seen as an extra column.
2478 if (virtual_active() && fp == &curwin->w_cursor)
2479 {
2480 char_u *p = ml_get_cursor();
2481
2482 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
2483 curwin->w_virtcol - curwin->w_cursor.coladd))
2484 {
2485 int l;
2486
2487 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
2488 col += l;
2489 }
2490 }
2491 }
2492 }
2493 rettv->vval.v_number = col;
2494}
2495
2496/*
2497 * "charcol()" function
2498 */
2499 static void
2500f_charcol(typval_T *argvars, typval_T *rettv)
2501{
2502 get_col(argvars, rettv, TRUE);
2503}
2504
2505/*
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002506 * "charidx()" function
2507 */
2508 static void
2509f_charidx(typval_T *argvars, typval_T *rettv)
2510{
2511 char_u *str;
2512 varnumber_T idx;
2513 int countcc = FALSE;
2514 char_u *p;
2515 int len;
2516 int (*ptr2len)(char_u *);
2517
2518 rettv->vval.v_number = -1;
2519
2520 if (argvars[0].v_type != VAR_STRING || argvars[1].v_type != VAR_NUMBER
2521 || (argvars[2].v_type != VAR_UNKNOWN
2522 && argvars[2].v_type != VAR_NUMBER))
2523 {
2524 emsg(_(e_invarg));
2525 return;
2526 }
2527
2528 str = tv_get_string_chk(&argvars[0]);
2529 idx = tv_get_number_chk(&argvars[1], NULL);
2530 if (str == NULL || idx < 0)
2531 return;
2532
2533 if (argvars[2].v_type != VAR_UNKNOWN)
2534 countcc = (int)tv_get_bool(&argvars[2]);
2535 if (countcc < 0 || countcc > 1)
2536 {
2537 semsg(_(e_using_number_as_bool_nr), countcc);
2538 return;
2539 }
2540
2541 if (enc_utf8 && countcc)
2542 ptr2len = utf_ptr2len;
2543 else
2544 ptr2len = mb_ptr2len;
2545
2546 for (p = str, len = 0; p <= str + idx; len++)
2547 {
2548 if (*p == NUL)
2549 return;
2550 p += ptr2len(p);
2551 }
2552
2553 rettv->vval.v_number = len > 0 ? len - 1 : 0;
2554}
2555
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02002556 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01002557get_optional_window(typval_T *argvars, int idx)
2558{
2559 win_T *win = curwin;
2560
2561 if (argvars[idx].v_type != VAR_UNKNOWN)
2562 {
2563 win = find_win_by_nr_or_id(&argvars[idx]);
2564 if (win == NULL)
2565 {
2566 emsg(_(e_invalwindow));
2567 return NULL;
2568 }
2569 }
2570 return win;
2571}
2572
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002573/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002574 * "col(string)" function
2575 */
2576 static void
2577f_col(typval_T *argvars, typval_T *rettv)
2578{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002579 get_col(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002580}
2581
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002582/*
2583 * "confirm(message, buttons[, default [, type]])" function
2584 */
2585 static void
2586f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2587{
2588#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2589 char_u *message;
2590 char_u *buttons = NULL;
2591 char_u buf[NUMBUFLEN];
2592 char_u buf2[NUMBUFLEN];
2593 int def = 1;
2594 int type = VIM_GENERIC;
2595 char_u *typestr;
2596 int error = FALSE;
2597
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002598 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002599 if (message == NULL)
2600 error = TRUE;
2601 if (argvars[1].v_type != VAR_UNKNOWN)
2602 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002603 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002604 if (buttons == NULL)
2605 error = TRUE;
2606 if (argvars[2].v_type != VAR_UNKNOWN)
2607 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002608 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002609 if (argvars[3].v_type != VAR_UNKNOWN)
2610 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002611 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002612 if (typestr == NULL)
2613 error = TRUE;
2614 else
2615 {
2616 switch (TOUPPER_ASC(*typestr))
2617 {
2618 case 'E': type = VIM_ERROR; break;
2619 case 'Q': type = VIM_QUESTION; break;
2620 case 'I': type = VIM_INFO; break;
2621 case 'W': type = VIM_WARNING; break;
2622 case 'G': type = VIM_GENERIC; break;
2623 }
2624 }
2625 }
2626 }
2627 }
2628
2629 if (buttons == NULL || *buttons == NUL)
2630 buttons = (char_u *)_("&Ok");
2631
2632 if (!error)
2633 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
2634 def, NULL, FALSE);
2635#endif
2636}
2637
2638/*
2639 * "copy()" function
2640 */
2641 static void
2642f_copy(typval_T *argvars, typval_T *rettv)
2643{
2644 item_copy(&argvars[0], rettv, FALSE, 0);
2645}
2646
2647#ifdef FEAT_FLOAT
2648/*
2649 * "cos()" function
2650 */
2651 static void
2652f_cos(typval_T *argvars, typval_T *rettv)
2653{
2654 float_T f = 0.0;
2655
2656 rettv->v_type = VAR_FLOAT;
2657 if (get_float_arg(argvars, &f) == OK)
2658 rettv->vval.v_float = cos(f);
2659 else
2660 rettv->vval.v_float = 0.0;
2661}
2662
2663/*
2664 * "cosh()" function
2665 */
2666 static void
2667f_cosh(typval_T *argvars, typval_T *rettv)
2668{
2669 float_T f = 0.0;
2670
2671 rettv->v_type = VAR_FLOAT;
2672 if (get_float_arg(argvars, &f) == OK)
2673 rettv->vval.v_float = cosh(f);
2674 else
2675 rettv->vval.v_float = 0.0;
2676}
2677#endif
2678
2679/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002680 * Set the cursor position.
2681 * If 'charcol' is TRUE, then use the column number as a character offet.
2682 * Otherwise use the column number as a byte offset.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002683 */
2684 static void
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002685set_cursorpos(typval_T *argvars, typval_T *rettv, int charcol)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002686{
2687 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002688 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002689 int set_curswant = TRUE;
2690
2691 rettv->vval.v_number = -1;
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002692 if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002693 {
2694 pos_T pos;
2695 colnr_T curswant = -1;
2696
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002697 if (list2fpos(argvars, &pos, NULL, &curswant, charcol) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002698 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002699 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002700 return;
2701 }
2702 line = pos.lnum;
2703 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002704 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002705 if (curswant >= 0)
2706 {
2707 curwin->w_curswant = curswant - 1;
2708 set_curswant = FALSE;
2709 }
2710 }
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002711 else if ((argvars[0].v_type == VAR_NUMBER ||
2712 argvars[0].v_type == VAR_STRING)
2713 && argvars[1].v_type == VAR_NUMBER)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002714 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002715 line = tv_get_lnum(argvars);
Bram Moolenaar9a963372020-12-21 21:58:46 +01002716 if (line < 0)
2717 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002718 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002719 if (charcol)
2720 col = buf_charidx_to_byteidx(curbuf, line, col);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002721 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002722 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002723 }
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002724 else
2725 {
2726 emsg(_(e_invarg));
2727 return;
2728 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002729 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002730 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002731 if (line > 0)
2732 curwin->w_cursor.lnum = line;
2733 if (col > 0)
2734 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002735 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002736
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002737 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002738 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002739 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002740 if (has_mbyte)
2741 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002742
2743 curwin->w_set_curswant = set_curswant;
2744 rettv->vval.v_number = 0;
2745}
2746
Bram Moolenaar6f02b002021-01-10 20:22:54 +01002747/*
2748 * "cursor(lnum, col)" function, or
2749 * "cursor(list)"
2750 *
2751 * Moves the cursor to the specified line and column.
2752 * Returns 0 when the position could be set, -1 otherwise.
2753 */
2754 static void
2755f_cursor(typval_T *argvars, typval_T *rettv)
2756{
2757 set_cursorpos(argvars, rettv, FALSE);
2758}
2759
Bram Moolenaar4f974752019-02-17 17:44:42 +01002760#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002761/*
2762 * "debugbreak()" function
2763 */
2764 static void
2765f_debugbreak(typval_T *argvars, typval_T *rettv)
2766{
2767 int pid;
2768
2769 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002770 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002771 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002772 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002773 else
2774 {
2775 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
2776
2777 if (hProcess != NULL)
2778 {
2779 DebugBreakProcess(hProcess);
2780 CloseHandle(hProcess);
2781 rettv->vval.v_number = OK;
2782 }
2783 }
2784}
2785#endif
2786
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002787/*
2788 * "deepcopy()" function
2789 */
2790 static void
2791f_deepcopy(typval_T *argvars, typval_T *rettv)
2792{
2793 int noref = 0;
2794 int copyID;
2795
2796 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar44b4a242020-09-05 17:18:28 +02002797 noref = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002798 if (noref < 0 || noref > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02002799 semsg(_(e_using_number_as_bool_nr), noref);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002800 else
2801 {
2802 copyID = get_copyID();
2803 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
2804 }
2805}
2806
2807/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002808 * "did_filetype()" function
2809 */
2810 static void
2811f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2812{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002813 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002814}
2815
2816/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01002817 * "echoraw({expr})" function
2818 */
2819 static void
2820f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
2821{
2822 char_u *str = tv_get_string_chk(&argvars[0]);
2823
2824 if (str != NULL && *str != NUL)
2825 {
2826 out_str(str);
2827 out_flush();
2828 }
2829}
2830
2831/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002832 * "empty({expr})" function
2833 */
2834 static void
2835f_empty(typval_T *argvars, typval_T *rettv)
2836{
2837 int n = FALSE;
2838
2839 switch (argvars[0].v_type)
2840 {
2841 case VAR_STRING:
2842 case VAR_FUNC:
2843 n = argvars[0].vval.v_string == NULL
2844 || *argvars[0].vval.v_string == NUL;
2845 break;
2846 case VAR_PARTIAL:
2847 n = FALSE;
2848 break;
2849 case VAR_NUMBER:
2850 n = argvars[0].vval.v_number == 0;
2851 break;
2852 case VAR_FLOAT:
2853#ifdef FEAT_FLOAT
2854 n = argvars[0].vval.v_float == 0.0;
2855 break;
2856#endif
2857 case VAR_LIST:
2858 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002859 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002860 break;
2861 case VAR_DICT:
2862 n = argvars[0].vval.v_dict == NULL
2863 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2864 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002865 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002866 case VAR_SPECIAL:
2867 n = argvars[0].vval.v_number != VVAL_TRUE;
2868 break;
2869
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002870 case VAR_BLOB:
2871 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002872 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2873 break;
2874
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002875 case VAR_JOB:
2876#ifdef FEAT_JOB_CHANNEL
2877 n = argvars[0].vval.v_job == NULL
2878 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2879 break;
2880#endif
2881 case VAR_CHANNEL:
2882#ifdef FEAT_JOB_CHANNEL
2883 n = argvars[0].vval.v_channel == NULL
2884 || !channel_is_open(argvars[0].vval.v_channel);
2885 break;
2886#endif
2887 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02002888 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002889 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002890 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002891 n = TRUE;
2892 break;
2893 }
2894
2895 rettv->vval.v_number = n;
2896}
2897
2898/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002899 * "environ()" function
2900 */
2901 static void
2902f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2903{
2904#if !defined(AMIGA)
2905 int i = 0;
2906 char_u *entry, *value;
2907# ifdef MSWIN
2908 extern wchar_t **_wenviron;
2909# else
2910 extern char **environ;
2911# endif
2912
2913 if (rettv_dict_alloc(rettv) != OK)
2914 return;
2915
2916# ifdef MSWIN
2917 if (*_wenviron == NULL)
2918 return;
2919# else
2920 if (*environ == NULL)
2921 return;
2922# endif
2923
2924 for (i = 0; ; ++i)
2925 {
2926# ifdef MSWIN
2927 short_u *p;
2928
2929 if ((p = (short_u *)_wenviron[i]) == NULL)
2930 return;
2931 entry = utf16_to_enc(p, NULL);
2932# else
2933 if ((entry = (char_u *)environ[i]) == NULL)
2934 return;
2935 entry = vim_strsave(entry);
2936# endif
2937 if (entry == NULL) // out of memory
2938 return;
2939 if ((value = vim_strchr(entry, '=')) == NULL)
2940 {
2941 vim_free(entry);
2942 continue;
2943 }
2944 *value++ = NUL;
2945 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2946 vim_free(entry);
2947 }
2948#endif
2949}
2950
2951/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002952 * "escape({string}, {chars})" function
2953 */
2954 static void
2955f_escape(typval_T *argvars, typval_T *rettv)
2956{
2957 char_u buf[NUMBUFLEN];
2958
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002959 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2960 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002961 rettv->v_type = VAR_STRING;
2962}
2963
2964/*
2965 * "eval()" function
2966 */
2967 static void
2968f_eval(typval_T *argvars, typval_T *rettv)
2969{
2970 char_u *s, *p;
2971
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002972 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002973 if (s != NULL)
2974 s = skipwhite(s);
2975
2976 p = s;
Bram Moolenaar5409f5d2020-06-24 18:37:35 +02002977 if (s == NULL || eval1(&s, rettv, &EVALARG_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002978 {
2979 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002980 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002981 need_clr_eos = FALSE;
2982 rettv->v_type = VAR_NUMBER;
2983 rettv->vval.v_number = 0;
2984 }
2985 else if (*s != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02002986 semsg(_(e_trailing_arg), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002987}
2988
2989/*
2990 * "eventhandler()" function
2991 */
2992 static void
2993f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2994{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002995 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002996}
2997
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002998static garray_T redir_execute_ga;
2999
3000/*
3001 * Append "value[value_len]" to the execute() output.
3002 */
3003 void
3004execute_redir_str(char_u *value, int value_len)
3005{
3006 int len;
3007
3008 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003009 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003010 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003011 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003012 if (ga_grow(&redir_execute_ga, len) == OK)
3013 {
3014 mch_memmove((char *)redir_execute_ga.ga_data
3015 + redir_execute_ga.ga_len, value, len);
3016 redir_execute_ga.ga_len += len;
3017 }
3018}
3019
3020/*
3021 * Get next line from a list.
3022 * Called by do_cmdline() to get the next line.
3023 * Returns allocated string, or NULL for end of function.
3024 */
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003025 static char_u *
3026get_list_line(
3027 int c UNUSED,
3028 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02003029 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +02003030 getline_opt_T options UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003031{
3032 listitem_T **p = (listitem_T **)cookie;
3033 listitem_T *item = *p;
3034 char_u buf[NUMBUFLEN];
3035 char_u *s;
3036
3037 if (item == NULL)
3038 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003039 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003040 *p = item->li_next;
3041 return s == NULL ? NULL : vim_strsave(s);
3042}
3043
3044/*
3045 * "execute()" function
3046 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02003047 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003048execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003049{
3050 char_u *cmd = NULL;
3051 list_T *list = NULL;
3052 int save_msg_silent = msg_silent;
3053 int save_emsg_silent = emsg_silent;
3054 int save_emsg_noredir = emsg_noredir;
3055 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01003056 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003057 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003058 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003059 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003060
3061 rettv->vval.v_string = NULL;
3062 rettv->v_type = VAR_STRING;
3063
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003064 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003065 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003066 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003067 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003068 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003069 return;
3070 ++list->lv_refcount;
3071 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01003072 else if (argvars[arg_off].v_type == VAR_JOB
3073 || argvars[arg_off].v_type == VAR_CHANNEL)
3074 {
3075 emsg(_(e_inval_string));
3076 return;
3077 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003078 else
3079 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003080 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003081 if (cmd == NULL)
3082 return;
3083 }
3084
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003085 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003086 {
3087 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003088 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003089
3090 if (s == NULL)
3091 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003092 if (*s == NUL)
3093 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003094 if (STRNCMP(s, "silent", 6) == 0)
3095 ++msg_silent;
3096 if (STRCMP(s, "silent!") == 0)
3097 {
3098 emsg_silent = TRUE;
3099 emsg_noredir = TRUE;
3100 }
3101 }
3102 else
3103 ++msg_silent;
3104
3105 if (redir_execute)
3106 save_ga = redir_execute_ga;
3107 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
3108 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01003109 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003110 if (!echo_output)
3111 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003112
3113 if (cmd != NULL)
3114 do_cmdline_cmd(cmd);
3115 else
3116 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003117 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003118
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003119 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003120 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003121 do_cmdline(NULL, get_list_line, (void *)&item,
3122 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
3123 --list->lv_refcount;
3124 }
3125
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003126 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01003127 if (ga_grow(&redir_execute_ga, 1) == OK)
3128 {
3129 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
3130 rettv->vval.v_string = redir_execute_ga.ga_data;
3131 }
3132 else
3133 {
3134 ga_clear(&redir_execute_ga);
3135 rettv->vval.v_string = NULL;
3136 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003137 msg_silent = save_msg_silent;
3138 emsg_silent = save_emsg_silent;
3139 emsg_noredir = save_emsg_noredir;
3140
3141 redir_execute = save_redir_execute;
3142 if (redir_execute)
3143 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01003144 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003145
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003146 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003147 if (echo_output)
3148 // When not working silently: put it in column zero. A following
3149 // "echon" will overwrite the message, unavoidably.
3150 msg_col = 0;
3151 else
3152 // When working silently: Put it back where it was, since nothing
3153 // should have been written.
3154 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003155}
3156
3157/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003158 * "execute()" function
3159 */
3160 static void
3161f_execute(typval_T *argvars, typval_T *rettv)
3162{
3163 execute_common(argvars, rettv, 0);
3164}
3165
3166/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003167 * "exists()" function
3168 */
3169 static void
3170f_exists(typval_T *argvars, typval_T *rettv)
3171{
3172 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003173 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003174
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003175 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003176 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003177 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003178 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003179 if (mch_getenv(p + 1) != NULL)
3180 n = TRUE;
3181 else
3182 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003183 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003184 p = expand_env_save(p);
3185 if (p != NULL && *p != '$')
3186 n = TRUE;
3187 vim_free(p);
3188 }
3189 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003190 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003191 {
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +02003192 n = (eval_option(&p, NULL, TRUE) == OK);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003193 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003194 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003195 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003196 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003197 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02003198 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003199 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01003200 else if (*p == '?') // internal function only
3201 {
3202 n = has_internal_func_name(p + 1);
3203 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003204 else if (*p == ':')
3205 {
3206 n = cmd_exists(p + 1);
3207 }
3208 else if (*p == '#')
3209 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003210 if (p[1] == '#')
3211 n = autocmd_supported(p + 2);
3212 else
3213 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003214 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003215 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003216 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01003217 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003218 }
3219
3220 rettv->vval.v_number = n;
3221}
3222
3223#ifdef FEAT_FLOAT
3224/*
3225 * "exp()" function
3226 */
3227 static void
3228f_exp(typval_T *argvars, typval_T *rettv)
3229{
3230 float_T f = 0.0;
3231
3232 rettv->v_type = VAR_FLOAT;
3233 if (get_float_arg(argvars, &f) == OK)
3234 rettv->vval.v_float = exp(f);
3235 else
3236 rettv->vval.v_float = 0.0;
3237}
3238#endif
3239
3240/*
3241 * "expand()" function
3242 */
3243 static void
3244f_expand(typval_T *argvars, typval_T *rettv)
3245{
3246 char_u *s;
3247 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003248 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003249 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
3250 expand_T xpc;
3251 int error = FALSE;
3252 char_u *result;
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003253#ifdef BACKSLASH_IN_FILENAME
3254 char_u *p_csl_save = p_csl;
3255
3256 // avoid using 'completeslash' here
3257 p_csl = empty_option;
3258#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003259
3260 rettv->v_type = VAR_STRING;
3261 if (argvars[1].v_type != VAR_UNKNOWN
3262 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003263 && tv_get_bool_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003264 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02003265 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003266
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003267 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003268 if (*s == '%' || *s == '#' || *s == '<')
3269 {
3270 ++emsg_off;
3271 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
3272 --emsg_off;
3273 if (rettv->v_type == VAR_LIST)
3274 {
3275 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
3276 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02003277 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003278 }
3279 else
3280 rettv->vval.v_string = result;
3281 }
3282 else
3283 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003284 // When the optional second argument is non-zero, don't remove matches
3285 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003286 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003287 && tv_get_bool_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003288 options |= WILD_KEEP_ALL;
3289 if (!error)
3290 {
3291 ExpandInit(&xpc);
3292 xpc.xp_context = EXPAND_FILES;
3293 if (p_wic)
3294 options += WILD_ICASE;
3295 if (rettv->v_type == VAR_STRING)
3296 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
3297 options, WILD_ALL);
3298 else if (rettv_list_alloc(rettv) != FAIL)
3299 {
3300 int i;
3301
3302 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
3303 for (i = 0; i < xpc.xp_numfiles; i++)
3304 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
3305 ExpandCleanup(&xpc);
3306 }
3307 }
3308 else
3309 rettv->vval.v_string = NULL;
3310 }
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003311#ifdef BACKSLASH_IN_FILENAME
3312 p_csl = p_csl_save;
3313#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003314}
3315
3316/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02003317 * "expandcmd()" function
3318 * Expand all the special characters in a command string.
3319 */
3320 static void
3321f_expandcmd(typval_T *argvars, typval_T *rettv)
3322{
3323 exarg_T eap;
3324 char_u *cmdstr;
3325 char *errormsg = NULL;
3326
3327 rettv->v_type = VAR_STRING;
3328 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
3329
3330 memset(&eap, 0, sizeof(eap));
3331 eap.cmd = cmdstr;
3332 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02003333 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02003334 eap.usefilter = FALSE;
3335 eap.nextcmd = NULL;
3336 eap.cmdidx = CMD_USER;
3337
3338 expand_filename(&eap, &cmdstr, &errormsg);
3339 if (errormsg != NULL && *errormsg != NUL)
3340 emsg(errormsg);
3341
3342 rettv->vval.v_string = cmdstr;
3343}
3344
3345/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003346 * "feedkeys()" function
3347 */
3348 static void
3349f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
3350{
3351 int remap = TRUE;
3352 int insert = FALSE;
3353 char_u *keys, *flags;
3354 char_u nbuf[NUMBUFLEN];
3355 int typed = FALSE;
3356 int execute = FALSE;
3357 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003358 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003359 char_u *keys_esc;
3360
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003361 // This is not allowed in the sandbox. If the commands would still be
3362 // executed in the sandbox it would be OK, but it probably happens later,
3363 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003364 if (check_secure())
3365 return;
3366
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003367 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003368
3369 if (argvars[1].v_type != VAR_UNKNOWN)
3370 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003371 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003372 for ( ; *flags != NUL; ++flags)
3373 {
3374 switch (*flags)
3375 {
3376 case 'n': remap = FALSE; break;
3377 case 'm': remap = TRUE; break;
3378 case 't': typed = TRUE; break;
3379 case 'i': insert = TRUE; break;
3380 case 'x': execute = TRUE; break;
3381 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003382 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003383 }
3384 }
3385 }
3386
3387 if (*keys != NUL || execute)
3388 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003389 // Need to escape K_SPECIAL and CSI before putting the string in the
3390 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003391 keys_esc = vim_strsave_escape_csi(keys);
3392 if (keys_esc != NULL)
3393 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003394 if (lowlevel)
3395 {
3396#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01003397 int idx;
3398 int len = (int)STRLEN(keys);
3399
3400 for (idx = 0; idx < len; ++idx)
3401 {
3402 // if a CTRL-C was typed, set got_int, similar to what
3403 // happens in fill_input_buf()
3404 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
3405 got_int = TRUE;
3406 add_to_input_buf(keys + idx, 1);
3407 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003408#else
3409 emsg(_("E980: lowlevel input not supported"));
3410#endif
3411 }
3412 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003413 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003414 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003415 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003416 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003417#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003418 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003419#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003420 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003421 typebuf_was_filled = TRUE;
3422 }
3423 vim_free(keys_esc);
3424
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003425 if (execute)
3426 {
3427 int save_msg_scroll = msg_scroll;
3428
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003429 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003430 msg_scroll = FALSE;
3431
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003432 if (!dangerous)
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003433 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003434 ++ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003435 ++in_feedkeys;
3436 }
Bram Moolenaar905dd902019-04-07 14:21:47 +02003437 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003438 if (!dangerous)
Bram Moolenaar189832b2020-09-23 12:29:11 +02003439 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003440 --ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003441 --in_feedkeys;
Bram Moolenaar189832b2020-09-23 12:29:11 +02003442 }
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003443
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003444 msg_scroll |= save_msg_scroll;
3445 }
3446 }
3447 }
3448}
3449
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003450#ifdef FEAT_FLOAT
3451/*
3452 * "float2nr({float})" function
3453 */
3454 static void
3455f_float2nr(typval_T *argvars, typval_T *rettv)
3456{
3457 float_T f = 0.0;
3458
3459 if (get_float_arg(argvars, &f) == OK)
3460 {
Bram Moolenaar37184272020-05-23 19:30:05 +02003461 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003462 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02003463 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003464 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003465 else
3466 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003467 }
3468}
3469
3470/*
3471 * "floor({float})" function
3472 */
3473 static void
3474f_floor(typval_T *argvars, typval_T *rettv)
3475{
3476 float_T f = 0.0;
3477
3478 rettv->v_type = VAR_FLOAT;
3479 if (get_float_arg(argvars, &f) == OK)
3480 rettv->vval.v_float = floor(f);
3481 else
3482 rettv->vval.v_float = 0.0;
3483}
3484
3485/*
3486 * "fmod()" function
3487 */
3488 static void
3489f_fmod(typval_T *argvars, typval_T *rettv)
3490{
3491 float_T fx = 0.0, fy = 0.0;
3492
3493 rettv->v_type = VAR_FLOAT;
3494 if (get_float_arg(argvars, &fx) == OK
3495 && get_float_arg(&argvars[1], &fy) == OK)
3496 rettv->vval.v_float = fmod(fx, fy);
3497 else
3498 rettv->vval.v_float = 0.0;
3499}
3500#endif
3501
3502/*
3503 * "fnameescape({string})" function
3504 */
3505 static void
3506f_fnameescape(typval_T *argvars, typval_T *rettv)
3507{
3508 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003509 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003510 rettv->v_type = VAR_STRING;
3511}
3512
3513/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003514 * "foreground()" function
3515 */
3516 static void
3517f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3518{
3519#ifdef FEAT_GUI
3520 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003521 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003522 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003523 return;
3524 }
3525#endif
3526#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003527 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003528#endif
3529}
3530
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003531 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003532common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003533{
3534 char_u *s;
3535 char_u *name;
3536 int use_string = FALSE;
3537 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003538 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003539 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003540
3541 if (argvars[0].v_type == VAR_FUNC)
3542 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003543 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003544 s = argvars[0].vval.v_string;
3545 }
3546 else if (argvars[0].v_type == VAR_PARTIAL
3547 && argvars[0].vval.v_partial != NULL)
3548 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003549 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003550 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003551 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003552 }
3553 else
3554 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003555 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003556 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003557 use_string = TRUE;
3558 }
3559
Bram Moolenaar843b8842016-08-21 14:36:15 +02003560 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003561 {
3562 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003563 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar32b3f822021-01-06 21:59:39 +01003564 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF,
3565 NULL, NULL, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003566 if (*name != NUL)
3567 s = NULL;
3568 }
3569
Bram Moolenaar843b8842016-08-21 14:36:15 +02003570 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
3571 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003572 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003573 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003574 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003575 ? find_func(trans_name, is_global, NULL) == NULL
3576 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003577 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003578 else
3579 {
3580 int dict_idx = 0;
3581 int arg_idx = 0;
3582 list_T *list = NULL;
3583
3584 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
3585 {
3586 char sid_buf[25];
3587 int off = *s == 's' ? 2 : 5;
3588
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003589 // Expand s: and <SID> into <SNR>nr_, so that the function can
3590 // also be called from another script. Using trans_function_name()
3591 // would also work, but some plugins depend on the name being
3592 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02003593 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02003594 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003595 if (name != NULL)
3596 {
3597 STRCPY(name, sid_buf);
3598 STRCAT(name, s + off);
3599 }
3600 }
3601 else
3602 name = vim_strsave(s);
3603
3604 if (argvars[1].v_type != VAR_UNKNOWN)
3605 {
3606 if (argvars[2].v_type != VAR_UNKNOWN)
3607 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003608 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003609 arg_idx = 1;
3610 dict_idx = 2;
3611 }
3612 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003613 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003614 dict_idx = 1;
3615 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003616 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003617 arg_idx = 1;
3618 if (dict_idx > 0)
3619 {
3620 if (argvars[dict_idx].v_type != VAR_DICT)
3621 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003622 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003623 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003624 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003625 }
3626 if (argvars[dict_idx].vval.v_dict == NULL)
3627 dict_idx = 0;
3628 }
3629 if (arg_idx > 0)
3630 {
3631 if (argvars[arg_idx].v_type != VAR_LIST)
3632 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003633 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003634 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003635 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003636 }
3637 list = argvars[arg_idx].vval.v_list;
3638 if (list == NULL || list->lv_len == 0)
3639 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003640 else if (list->lv_len > MAX_FUNC_ARGS)
3641 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01003642 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003643 vim_free(name);
3644 goto theend;
3645 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003646 }
3647 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003648 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003649 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003650 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003651
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003652 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003653 if (pt == NULL)
3654 vim_free(name);
3655 else
3656 {
3657 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
3658 {
3659 listitem_T *li;
3660 int i = 0;
3661 int arg_len = 0;
3662 int lv_len = 0;
3663
3664 if (arg_pt != NULL)
3665 arg_len = arg_pt->pt_argc;
3666 if (list != NULL)
3667 lv_len = list->lv_len;
3668 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003669 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003670 if (pt->pt_argv == NULL)
3671 {
3672 vim_free(pt);
3673 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003674 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003675 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003676 for (i = 0; i < arg_len; i++)
3677 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
3678 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003679 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003680 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003681 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003682 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003683 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003684 }
3685
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003686 // For "function(dict.func, [], dict)" and "func" is a partial
3687 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003688 if (dict_idx > 0)
3689 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003690 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003691 pt->pt_dict = argvars[dict_idx].vval.v_dict;
3692 ++pt->pt_dict->dv_refcount;
3693 }
3694 else if (arg_pt != NULL)
3695 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003696 // If the dict was bound automatically the result is also
3697 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003698 pt->pt_dict = arg_pt->pt_dict;
3699 pt->pt_auto = arg_pt->pt_auto;
3700 if (pt->pt_dict != NULL)
3701 ++pt->pt_dict->dv_refcount;
3702 }
3703
3704 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003705 if (arg_pt != NULL && arg_pt->pt_func != NULL)
3706 {
3707 pt->pt_func = arg_pt->pt_func;
3708 func_ptr_ref(pt->pt_func);
3709 vim_free(name);
3710 }
3711 else if (is_funcref)
3712 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003713 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003714 func_ptr_ref(pt->pt_func);
3715 vim_free(name);
3716 }
3717 else
3718 {
3719 pt->pt_name = name;
3720 func_ref(name);
3721 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003722 }
3723 rettv->v_type = VAR_PARTIAL;
3724 rettv->vval.v_partial = pt;
3725 }
3726 else
3727 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003728 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003729 rettv->v_type = VAR_FUNC;
3730 rettv->vval.v_string = name;
3731 func_ref(name);
3732 }
3733 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003734theend:
3735 vim_free(trans_name);
3736}
3737
3738/*
3739 * "funcref()" function
3740 */
3741 static void
3742f_funcref(typval_T *argvars, typval_T *rettv)
3743{
3744 common_function(argvars, rettv, TRUE);
3745}
3746
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003747 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003748ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003749{
3750 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
3751 return &t_func_any;
Bram Moolenaar5e654232020-09-16 15:22:00 +02003752 return &t_func_unknown;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003753}
3754
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003755/*
3756 * "function()" function
3757 */
3758 static void
3759f_function(typval_T *argvars, typval_T *rettv)
3760{
3761 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003762}
3763
3764/*
3765 * "garbagecollect()" function
3766 */
3767 static void
3768f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
3769{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003770 // This is postponed until we are back at the toplevel, because we may be
3771 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003772 want_garbage_collect = TRUE;
3773
Bram Moolenaar2df47312020-09-05 17:30:44 +02003774 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003775 garbage_collect_at_exit = TRUE;
3776}
3777
3778/*
3779 * "get()" function
3780 */
3781 static void
3782f_get(typval_T *argvars, typval_T *rettv)
3783{
3784 listitem_T *li;
3785 list_T *l;
3786 dictitem_T *di;
3787 dict_T *d;
3788 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003789 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003790
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003791 if (argvars[0].v_type == VAR_BLOB)
3792 {
3793 int error = FALSE;
3794 int idx = tv_get_number_chk(&argvars[1], &error);
3795
3796 if (!error)
3797 {
3798 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003799 if (idx < 0)
3800 idx = blob_len(argvars[0].vval.v_blob) + idx;
3801 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
3802 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003803 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003804 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003805 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003806 tv = rettv;
3807 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003808 }
3809 }
3810 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003811 {
3812 if ((l = argvars[0].vval.v_list) != NULL)
3813 {
3814 int error = FALSE;
3815
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003816 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003817 if (!error && li != NULL)
3818 tv = &li->li_tv;
3819 }
3820 }
3821 else if (argvars[0].v_type == VAR_DICT)
3822 {
3823 if ((d = argvars[0].vval.v_dict) != NULL)
3824 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003825 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003826 if (di != NULL)
3827 tv = &di->di_tv;
3828 }
3829 }
3830 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
3831 {
3832 partial_T *pt;
3833 partial_T fref_pt;
3834
3835 if (argvars[0].v_type == VAR_PARTIAL)
3836 pt = argvars[0].vval.v_partial;
3837 else
3838 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02003839 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003840 fref_pt.pt_name = argvars[0].vval.v_string;
3841 pt = &fref_pt;
3842 }
3843
3844 if (pt != NULL)
3845 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003846 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003847 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003848
3849 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
3850 {
3851 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003852 n = partial_name(pt);
3853 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003854 rettv->vval.v_string = NULL;
3855 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003856 {
3857 rettv->vval.v_string = vim_strsave(n);
3858 if (rettv->v_type == VAR_FUNC)
3859 func_ref(rettv->vval.v_string);
3860 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003861 }
3862 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003863 {
3864 what_is_dict = TRUE;
3865 if (pt->pt_dict != NULL)
3866 rettv_dict_set(rettv, pt->pt_dict);
3867 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003868 else if (STRCMP(what, "args") == 0)
3869 {
3870 rettv->v_type = VAR_LIST;
3871 if (rettv_list_alloc(rettv) == OK)
3872 {
3873 int i;
3874
3875 for (i = 0; i < pt->pt_argc; ++i)
3876 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3877 }
3878 }
3879 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003880 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003881
3882 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3883 // third argument
3884 if (!what_is_dict)
3885 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003886 }
3887 }
3888 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003889 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003890
3891 if (tv == NULL)
3892 {
3893 if (argvars[2].v_type != VAR_UNKNOWN)
3894 copy_tv(&argvars[2], rettv);
3895 }
3896 else
3897 copy_tv(tv, rettv);
3898}
3899
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003900/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003901 * "getchangelist()" function
3902 */
3903 static void
3904f_getchangelist(typval_T *argvars, typval_T *rettv)
3905{
3906#ifdef FEAT_JUMPLIST
3907 buf_T *buf;
3908 int i;
3909 list_T *l;
3910 dict_T *d;
3911#endif
3912
3913 if (rettv_list_alloc(rettv) != OK)
3914 return;
3915
3916#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003917 if (argvars[0].v_type == VAR_UNKNOWN)
3918 buf = curbuf;
3919 else
Bram Moolenaara5d38412020-09-02 21:02:35 +02003920 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003921 if (buf == NULL)
3922 return;
3923
3924 l = list_alloc();
3925 if (l == NULL)
3926 return;
3927
3928 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3929 return;
3930 /*
3931 * The current window change list index tracks only the position in the
3932 * current buffer change list. For other buffers, use the change list
3933 * length as the current index.
3934 */
3935 list_append_number(rettv->vval.v_list,
3936 (varnumber_T)((buf == curwin->w_buffer)
3937 ? curwin->w_changelistidx : buf->b_changelistlen));
3938
3939 for (i = 0; i < buf->b_changelistlen; ++i)
3940 {
3941 if (buf->b_changelist[i].lnum == 0)
3942 continue;
3943 if ((d = dict_alloc()) == NULL)
3944 return;
3945 if (list_append_dict(l, d) == FAIL)
3946 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003947 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3948 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003949 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003950 }
3951#endif
3952}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003953
Bram Moolenaar6f02b002021-01-10 20:22:54 +01003954 static void
3955getpos_both(
3956 typval_T *argvars,
3957 typval_T *rettv,
3958 int getcurpos,
3959 int charcol)
3960{
3961 pos_T *fp = NULL;
3962 pos_T pos;
3963 win_T *wp = curwin;
3964 list_T *l;
3965 int fnum = -1;
3966
3967 if (rettv_list_alloc(rettv) == OK)
3968 {
3969 l = rettv->vval.v_list;
3970 if (getcurpos)
3971 {
3972 if (argvars[0].v_type != VAR_UNKNOWN)
3973 {
3974 wp = find_win_by_nr_or_id(&argvars[0]);
3975 if (wp != NULL)
3976 fp = &wp->w_cursor;
3977 }
3978 else
3979 fp = &curwin->w_cursor;
3980 if (fp != NULL && charcol)
3981 {
3982 pos = *fp;
3983 pos.col = buf_byteidx_to_charidx(wp->w_buffer, pos.lnum,
3984 pos.col) - 1;
3985 fp = &pos;
3986 }
3987 }
3988 else
3989 fp = var2fpos(&argvars[0], TRUE, &fnum, charcol);
3990 if (fnum != -1)
3991 list_append_number(l, (varnumber_T)fnum);
3992 else
3993 list_append_number(l, (varnumber_T)0);
3994 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
3995 : (varnumber_T)0);
3996 list_append_number(l, (fp != NULL)
3997 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
3998 : (varnumber_T)0);
3999 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
4000 (varnumber_T)0);
4001 if (getcurpos)
4002 {
4003 int save_set_curswant = curwin->w_set_curswant;
4004 colnr_T save_curswant = curwin->w_curswant;
4005 colnr_T save_virtcol = curwin->w_virtcol;
4006
4007 if (wp == curwin)
4008 update_curswant();
4009 list_append_number(l, wp == NULL ? 0 : wp->w_curswant == MAXCOL
4010 ? (varnumber_T)MAXCOL : (varnumber_T)wp->w_curswant + 1);
4011
4012 // Do not change "curswant", as it is unexpected that a get
4013 // function has a side effect.
4014 if (wp == curwin && save_set_curswant)
4015 {
4016 curwin->w_set_curswant = save_set_curswant;
4017 curwin->w_curswant = save_curswant;
4018 curwin->w_virtcol = save_virtcol;
4019 curwin->w_valid &= ~VALID_VIRTCOL;
4020 }
4021 }
4022 }
4023 else
4024 rettv->vval.v_number = FALSE;
4025}
4026
4027/*
4028 * "getcharpos()" function
4029 */
4030 static void
4031f_getcharpos(typval_T *argvars UNUSED, typval_T *rettv)
4032{
4033 getpos_both(argvars, rettv, FALSE, TRUE);
4034}
4035
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004036/*
4037 * "getcharsearch()" function
4038 */
4039 static void
4040f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
4041{
4042 if (rettv_dict_alloc(rettv) != FAIL)
4043 {
4044 dict_T *dict = rettv->vval.v_dict;
4045
Bram Moolenaare0be1672018-07-08 16:50:37 +02004046 dict_add_string(dict, "char", last_csearch());
4047 dict_add_number(dict, "forward", last_csearch_forward());
4048 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004049 }
4050}
4051
4052/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02004053 * "getenv()" function
4054 */
4055 static void
4056f_getenv(typval_T *argvars, typval_T *rettv)
4057{
4058 int mustfree = FALSE;
4059 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
4060
4061 if (p == NULL)
4062 {
4063 rettv->v_type = VAR_SPECIAL;
4064 rettv->vval.v_number = VVAL_NULL;
4065 return;
4066 }
4067 if (!mustfree)
4068 p = vim_strsave(p);
4069 rettv->vval.v_string = p;
4070 rettv->v_type = VAR_STRING;
4071}
4072
4073/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004074 * "getfontname()" function
4075 */
4076 static void
4077f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
4078{
4079 rettv->v_type = VAR_STRING;
4080 rettv->vval.v_string = NULL;
4081#ifdef FEAT_GUI
4082 if (gui.in_use)
4083 {
4084 GuiFont font;
4085 char_u *name = NULL;
4086
4087 if (argvars[0].v_type == VAR_UNKNOWN)
4088 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004089 // Get the "Normal" font. Either the name saved by
4090 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004091 font = gui.norm_font;
4092 name = hl_get_font_name();
4093 }
4094 else
4095 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004096 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004097 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004098 return;
4099 font = gui_mch_get_font(name, FALSE);
4100 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004101 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004102 }
4103 rettv->vval.v_string = gui_mch_get_fontname(font, name);
4104 if (argvars[0].v_type != VAR_UNKNOWN)
4105 gui_mch_free_font(font);
4106 }
4107#endif
4108}
4109
4110/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01004111 * "getjumplist()" function
4112 */
4113 static void
4114f_getjumplist(typval_T *argvars, typval_T *rettv)
4115{
4116#ifdef FEAT_JUMPLIST
4117 win_T *wp;
4118 int i;
4119 list_T *l;
4120 dict_T *d;
4121#endif
4122
4123 if (rettv_list_alloc(rettv) != OK)
4124 return;
4125
4126#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004127 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004128 if (wp == NULL)
4129 return;
4130
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01004131 cleanup_jumplist(wp, TRUE);
4132
Bram Moolenaar4f505882018-02-10 21:06:32 +01004133 l = list_alloc();
4134 if (l == NULL)
4135 return;
4136
4137 if (list_append_list(rettv->vval.v_list, l) == FAIL)
4138 return;
4139 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
4140
4141 for (i = 0; i < wp->w_jumplistlen; ++i)
4142 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004143 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
4144 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01004145 if ((d = dict_alloc()) == NULL)
4146 return;
4147 if (list_append_dict(l, d) == FAIL)
4148 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02004149 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
4150 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004151 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004152 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004153 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02004154 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004155 }
4156#endif
4157}
4158
4159/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004160 * "getpid()" function
4161 */
4162 static void
4163f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
4164{
4165 rettv->vval.v_number = mch_get_pid();
4166}
4167
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004168/*
4169 * "getcurpos()" function
4170 */
4171 static void
4172f_getcurpos(typval_T *argvars, typval_T *rettv)
4173{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01004174 getpos_both(argvars, rettv, TRUE, FALSE);
4175}
4176
4177 static void
4178f_getcursorcharpos(typval_T *argvars, typval_T *rettv)
4179{
4180 getpos_both(argvars, rettv, TRUE, TRUE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004181}
4182
4183/*
4184 * "getpos(string)" function
4185 */
4186 static void
4187f_getpos(typval_T *argvars, typval_T *rettv)
4188{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01004189 getpos_both(argvars, rettv, FALSE, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004190}
4191
4192/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004193 * "getreg()" function
4194 */
4195 static void
4196f_getreg(typval_T *argvars, typval_T *rettv)
4197{
4198 char_u *strregname;
4199 int regname;
4200 int arg2 = FALSE;
4201 int return_list = FALSE;
4202 int error = FALSE;
4203
4204 if (argvars[0].v_type != VAR_UNKNOWN)
4205 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004206 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004207 error = strregname == NULL;
4208 if (argvars[1].v_type != VAR_UNKNOWN)
4209 {
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004210 arg2 = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004211 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004212 return_list = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004213 }
4214 }
4215 else
4216 strregname = get_vim_var_str(VV_REG);
4217
4218 if (error)
4219 return;
4220
4221 regname = (strregname == NULL ? '"' : *strregname);
4222 if (regname == 0)
4223 regname = '"';
4224
4225 if (return_list)
4226 {
4227 rettv->v_type = VAR_LIST;
4228 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
4229 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
4230 if (rettv->vval.v_list == NULL)
4231 (void)rettv_list_alloc(rettv);
4232 else
4233 ++rettv->vval.v_list->lv_refcount;
4234 }
4235 else
4236 {
4237 rettv->v_type = VAR_STRING;
4238 rettv->vval.v_string = get_reg_contents(regname,
4239 arg2 ? GREG_EXPR_SRC : 0);
4240 }
4241}
4242
4243/*
4244 * "getregtype()" function
4245 */
4246 static void
4247f_getregtype(typval_T *argvars, typval_T *rettv)
4248{
4249 char_u *strregname;
4250 int regname;
4251 char_u buf[NUMBUFLEN + 2];
4252 long reglen = 0;
4253
4254 if (argvars[0].v_type != VAR_UNKNOWN)
4255 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004256 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004257 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004258 {
4259 rettv->v_type = VAR_STRING;
4260 rettv->vval.v_string = NULL;
4261 return;
4262 }
4263 }
4264 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004265 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004266 strregname = get_vim_var_str(VV_REG);
4267
4268 regname = (strregname == NULL ? '"' : *strregname);
4269 if (regname == 0)
4270 regname = '"';
4271
4272 buf[0] = NUL;
4273 buf[1] = NUL;
4274 switch (get_reg_type(regname, &reglen))
4275 {
4276 case MLINE: buf[0] = 'V'; break;
4277 case MCHAR: buf[0] = 'v'; break;
4278 case MBLOCK:
4279 buf[0] = Ctrl_V;
4280 sprintf((char *)buf + 1, "%ld", reglen + 1);
4281 break;
4282 }
4283 rettv->v_type = VAR_STRING;
4284 rettv->vval.v_string = vim_strsave(buf);
4285}
4286
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02004287/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01004288 * "gettagstack()" function
4289 */
4290 static void
4291f_gettagstack(typval_T *argvars, typval_T *rettv)
4292{
4293 win_T *wp = curwin; // default is current window
4294
4295 if (rettv_dict_alloc(rettv) != OK)
4296 return;
4297
4298 if (argvars[0].v_type != VAR_UNKNOWN)
4299 {
4300 wp = find_win_by_nr_or_id(&argvars[0]);
4301 if (wp == NULL)
4302 return;
4303 }
4304
4305 get_tagstack(wp, rettv->vval.v_dict);
4306}
4307
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +02004308/*
4309 * "gettext()" function
4310 */
4311 static void
4312f_gettext(typval_T *argvars, typval_T *rettv)
4313{
4314 if (argvars[0].v_type != VAR_STRING
4315 || argvars[0].vval.v_string == NULL
4316 || *argvars[0].vval.v_string == NUL)
4317 {
4318 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
4319 }
4320 else
4321 {
4322 rettv->v_type = VAR_STRING;
4323 rettv->vval.v_string = vim_strsave(
4324 (char_u *)_(argvars[0].vval.v_string));
4325 }
4326}
4327
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004328// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004329#include "version.h"
4330
4331/*
4332 * "has()" function
4333 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01004334 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004335f_has(typval_T *argvars, typval_T *rettv)
4336{
4337 int i;
4338 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01004339 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004340 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01004341 typedef struct {
4342 char *name;
4343 short present;
4344 } has_item_T;
4345 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004346 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004347 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004348#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01004349 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004350#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004351 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004352#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004353 },
4354 {"arp",
4355#if defined(AMIGA) && defined(FEAT_ARP)
4356 1
4357#else
4358 0
4359#endif
4360 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004361 {"haiku",
4362#ifdef __HAIKU__
4363 1
4364#else
4365 0
4366#endif
4367 },
4368 {"bsd",
4369#if defined(BSD) && !defined(MACOS_X)
4370 1
4371#else
4372 0
4373#endif
4374 },
4375 {"hpux",
4376#ifdef hpux
4377 1
4378#else
4379 0
4380#endif
4381 },
4382 {"linux",
4383#ifdef __linux__
4384 1
4385#else
4386 0
4387#endif
4388 },
4389 {"mac", // Mac OS X (and, once, Mac OS Classic)
4390#ifdef MACOS_X
4391 1
4392#else
4393 0
4394#endif
4395 },
4396 {"osx", // Mac OS X
4397#ifdef MACOS_X
4398 1
4399#else
4400 0
4401#endif
4402 },
4403 {"macunix", // Mac OS X, with the darwin feature
4404#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4405 1
4406#else
4407 0
4408#endif
4409 },
4410 {"osxdarwin", // synonym for macunix
4411#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4412 1
4413#else
4414 0
4415#endif
4416 },
4417 {"qnx",
4418#ifdef __QNX__
4419 1
4420#else
4421 0
4422#endif
4423 },
4424 {"sun",
4425#ifdef SUN_SYSTEM
4426 1
4427#else
4428 0
4429#endif
4430 },
4431 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004432#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01004433 1
4434#else
4435 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004436#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004437 },
4438 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004439#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01004440 1
4441#else
4442 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004443#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004444 },
4445 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01004446#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01004447 1
4448#else
4449 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004450#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004451 },
4452 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01004453#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01004454 1
4455#else
4456 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004457#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004458 },
4459 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01004460#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01004461 1
4462#else
4463 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004464#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004465 },
4466 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004467#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004468 1
4469#else
4470 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004471#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004472 },
4473 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004474#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004475 1
4476#else
4477 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004478#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004479 },
4480 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004481#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01004482 1
4483#else
4484 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004485#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004486 },
4487 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004488#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004489 1
4490#else
4491 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004492#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004493 },
4494 {"autocmd", 1},
4495 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004496#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01004497 1
4498#else
4499 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004500#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004501 },
4502 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004503#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004504 1
4505#else
4506 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004507#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004508 },
4509 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004510#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01004511 1
4512#else
4513 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004514#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004515 },
4516 {"balloon_multiline",
4517#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
4518 // MS-Windows requires runtime check, see below
4519 1
4520#else
4521 0
4522#endif
4523 },
4524 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004525#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01004526 1
4527#else
4528 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01004529#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004530 },
4531 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004532#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01004533 1
4534#else
4535 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004536#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004537 },
4538 {"all_builtin_terms",
4539#if defined(ALL_BUILTIN_TCAPS)
4540 1
4541#else
4542 0
4543#endif
4544 },
4545 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004546#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01004547 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004548 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01004549 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004550#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004551 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004552#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004553 },
4554 {"byte_offset",
4555#ifdef FEAT_BYTEOFF
4556 1
4557#else
4558 0
4559#endif
4560 },
4561 {"channel",
4562#ifdef FEAT_JOB_CHANNEL
4563 1
4564#else
4565 0
4566#endif
4567 },
4568 {"cindent",
4569#ifdef FEAT_CINDENT
4570 1
4571#else
4572 0
4573#endif
4574 },
4575 {"clientserver",
4576#ifdef FEAT_CLIENTSERVER
4577 1
4578#else
4579 0
4580#endif
4581 },
4582 {"clipboard",
4583#ifdef FEAT_CLIPBOARD
4584 1
4585#else
4586 0
4587#endif
4588 },
4589 {"cmdline_compl", 1},
4590 {"cmdline_hist", 1},
4591 {"comments", 1},
4592 {"conceal",
4593#ifdef FEAT_CONCEAL
4594 1
4595#else
4596 0
4597#endif
4598 },
4599 {"cryptv",
4600#ifdef FEAT_CRYPT
4601 1
4602#else
4603 0
4604#endif
4605 },
4606 {"crypt-blowfish",
4607#ifdef FEAT_CRYPT
4608 1
4609#else
4610 0
4611#endif
4612 },
4613 {"crypt-blowfish2",
4614#ifdef FEAT_CRYPT
4615 1
4616#else
4617 0
4618#endif
4619 },
4620 {"cscope",
4621#ifdef FEAT_CSCOPE
4622 1
4623#else
4624 0
4625#endif
4626 },
4627 {"cursorbind", 1},
4628 {"cursorshape",
4629#ifdef CURSOR_SHAPE
4630 1
4631#else
4632 0
4633#endif
4634 },
4635 {"debug",
4636#ifdef DEBUG
4637 1
4638#else
4639 0
4640#endif
4641 },
4642 {"dialog_con",
4643#ifdef FEAT_CON_DIALOG
4644 1
4645#else
4646 0
4647#endif
4648 },
4649 {"dialog_gui",
4650#ifdef FEAT_GUI_DIALOG
4651 1
4652#else
4653 0
4654#endif
4655 },
4656 {"diff",
4657#ifdef FEAT_DIFF
4658 1
4659#else
4660 0
4661#endif
4662 },
4663 {"digraphs",
4664#ifdef FEAT_DIGRAPHS
4665 1
4666#else
4667 0
4668#endif
4669 },
4670 {"directx",
4671#ifdef FEAT_DIRECTX
4672 1
4673#else
4674 0
4675#endif
4676 },
4677 {"dnd",
4678#ifdef FEAT_DND
4679 1
4680#else
4681 0
4682#endif
4683 },
4684 {"emacs_tags",
4685#ifdef FEAT_EMACS_TAGS
4686 1
4687#else
4688 0
4689#endif
4690 },
4691 {"eval", 1}, // always present, of course!
4692 {"ex_extra", 1}, // graduated feature
4693 {"extra_search",
4694#ifdef FEAT_SEARCH_EXTRA
4695 1
4696#else
4697 0
4698#endif
4699 },
4700 {"file_in_path",
4701#ifdef FEAT_SEARCHPATH
4702 1
4703#else
4704 0
4705#endif
4706 },
4707 {"filterpipe",
4708#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
4709 1
4710#else
4711 0
4712#endif
4713 },
4714 {"find_in_path",
4715#ifdef FEAT_FIND_ID
4716 1
4717#else
4718 0
4719#endif
4720 },
4721 {"float",
4722#ifdef FEAT_FLOAT
4723 1
4724#else
4725 0
4726#endif
4727 },
4728 {"folding",
4729#ifdef FEAT_FOLDING
4730 1
4731#else
4732 0
4733#endif
4734 },
4735 {"footer",
4736#ifdef FEAT_FOOTER
4737 1
4738#else
4739 0
4740#endif
4741 },
4742 {"fork",
4743#if !defined(USE_SYSTEM) && defined(UNIX)
4744 1
4745#else
4746 0
4747#endif
4748 },
4749 {"gettext",
4750#ifdef FEAT_GETTEXT
4751 1
4752#else
4753 0
4754#endif
4755 },
4756 {"gui",
4757#ifdef FEAT_GUI
4758 1
4759#else
4760 0
4761#endif
4762 },
4763 {"gui_neXtaw",
4764#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
4765 1
4766#else
4767 0
4768#endif
4769 },
4770 {"gui_athena",
4771#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
4772 1
4773#else
4774 0
4775#endif
4776 },
4777 {"gui_gtk",
4778#ifdef FEAT_GUI_GTK
4779 1
4780#else
4781 0
4782#endif
4783 },
4784 {"gui_gtk2",
4785#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
4786 1
4787#else
4788 0
4789#endif
4790 },
4791 {"gui_gtk3",
4792#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
4793 1
4794#else
4795 0
4796#endif
4797 },
4798 {"gui_gnome",
4799#ifdef FEAT_GUI_GNOME
4800 1
4801#else
4802 0
4803#endif
4804 },
4805 {"gui_haiku",
4806#ifdef FEAT_GUI_HAIKU
4807 1
4808#else
4809 0
4810#endif
4811 },
Bram Moolenaar097148e2020-08-11 21:58:20 +02004812 {"gui_mac", 0},
Bram Moolenaar79296512020-03-22 16:17:14 +01004813 {"gui_motif",
4814#ifdef FEAT_GUI_MOTIF
4815 1
4816#else
4817 0
4818#endif
4819 },
4820 {"gui_photon",
4821#ifdef FEAT_GUI_PHOTON
4822 1
4823#else
4824 0
4825#endif
4826 },
4827 {"gui_win32",
4828#ifdef FEAT_GUI_MSWIN
4829 1
4830#else
4831 0
4832#endif
4833 },
4834 {"iconv",
4835#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
4836 1
4837#else
4838 0
4839#endif
4840 },
4841 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02004842 {"ipv6",
4843#ifdef FEAT_IPV6
4844 1
4845#else
4846 0
4847#endif
4848 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004849 {"job",
4850#ifdef FEAT_JOB_CHANNEL
4851 1
4852#else
4853 0
4854#endif
4855 },
4856 {"jumplist",
4857#ifdef FEAT_JUMPLIST
4858 1
4859#else
4860 0
4861#endif
4862 },
4863 {"keymap",
4864#ifdef FEAT_KEYMAP
4865 1
4866#else
4867 0
4868#endif
4869 },
4870 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
4871 {"langmap",
4872#ifdef FEAT_LANGMAP
4873 1
4874#else
4875 0
4876#endif
4877 },
4878 {"libcall",
4879#ifdef FEAT_LIBCALL
4880 1
4881#else
4882 0
4883#endif
4884 },
4885 {"linebreak",
4886#ifdef FEAT_LINEBREAK
4887 1
4888#else
4889 0
4890#endif
4891 },
4892 {"lispindent",
4893#ifdef FEAT_LISP
4894 1
4895#else
4896 0
4897#endif
4898 },
4899 {"listcmds", 1},
4900 {"localmap", 1},
4901 {"lua",
4902#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4903 1
4904#else
4905 0
4906#endif
4907 },
4908 {"menu",
4909#ifdef FEAT_MENU
4910 1
4911#else
4912 0
4913#endif
4914 },
4915 {"mksession",
4916#ifdef FEAT_SESSION
4917 1
4918#else
4919 0
4920#endif
4921 },
4922 {"modify_fname", 1},
4923 {"mouse", 1},
4924 {"mouseshape",
4925#ifdef FEAT_MOUSESHAPE
4926 1
4927#else
4928 0
4929#endif
4930 },
4931 {"mouse_dec",
4932#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4933 1
4934#else
4935 0
4936#endif
4937 },
4938 {"mouse_gpm",
4939#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4940 1
4941#else
4942 0
4943#endif
4944 },
4945 {"mouse_jsbterm",
4946#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4947 1
4948#else
4949 0
4950#endif
4951 },
4952 {"mouse_netterm",
4953#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4954 1
4955#else
4956 0
4957#endif
4958 },
4959 {"mouse_pterm",
4960#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4961 1
4962#else
4963 0
4964#endif
4965 },
4966 {"mouse_sgr",
4967#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4968 1
4969#else
4970 0
4971#endif
4972 },
4973 {"mouse_sysmouse",
4974#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4975 1
4976#else
4977 0
4978#endif
4979 },
4980 {"mouse_urxvt",
4981#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4982 1
4983#else
4984 0
4985#endif
4986 },
4987 {"mouse_xterm",
4988#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4989 1
4990#else
4991 0
4992#endif
4993 },
4994 {"multi_byte", 1},
4995 {"multi_byte_ime",
4996#ifdef FEAT_MBYTE_IME
4997 1
4998#else
4999 0
5000#endif
5001 },
5002 {"multi_lang",
5003#ifdef FEAT_MULTI_LANG
5004 1
5005#else
5006 0
5007#endif
5008 },
5009 {"mzscheme",
5010#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
5011 1
5012#else
5013 0
5014#endif
5015 },
5016 {"num64", 1},
5017 {"ole",
5018#ifdef FEAT_OLE
5019 1
5020#else
5021 0
5022#endif
5023 },
5024 {"packages",
5025#ifdef FEAT_EVAL
5026 1
5027#else
5028 0
5029#endif
5030 },
5031 {"path_extra",
5032#ifdef FEAT_PATH_EXTRA
5033 1
5034#else
5035 0
5036#endif
5037 },
5038 {"perl",
5039#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
5040 1
5041#else
5042 0
5043#endif
5044 },
5045 {"persistent_undo",
5046#ifdef FEAT_PERSISTENT_UNDO
5047 1
5048#else
5049 0
5050#endif
5051 },
5052 {"python_compiled",
5053#if defined(FEAT_PYTHON)
5054 1
5055#else
5056 0
5057#endif
5058 },
5059 {"python_dynamic",
5060#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
5061 1
5062#else
5063 0
5064#endif
5065 },
5066 {"python",
5067#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
5068 1
5069#else
5070 0
5071#endif
5072 },
5073 {"pythonx",
5074#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
5075 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
5076 1
5077#else
5078 0
5079#endif
5080 },
5081 {"python3_compiled",
5082#if defined(FEAT_PYTHON3)
5083 1
5084#else
5085 0
5086#endif
5087 },
5088 {"python3_dynamic",
5089#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
5090 1
5091#else
5092 0
5093#endif
5094 },
5095 {"python3",
5096#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
5097 1
5098#else
5099 0
5100#endif
5101 },
5102 {"popupwin",
5103#ifdef FEAT_PROP_POPUP
5104 1
5105#else
5106 0
5107#endif
5108 },
5109 {"postscript",
5110#ifdef FEAT_POSTSCRIPT
5111 1
5112#else
5113 0
5114#endif
5115 },
5116 {"printer",
5117#ifdef FEAT_PRINTER
5118 1
5119#else
5120 0
5121#endif
5122 },
5123 {"profile",
5124#ifdef FEAT_PROFILE
5125 1
5126#else
5127 0
5128#endif
5129 },
5130 {"reltime",
5131#ifdef FEAT_RELTIME
5132 1
5133#else
5134 0
5135#endif
5136 },
5137 {"quickfix",
5138#ifdef FEAT_QUICKFIX
5139 1
5140#else
5141 0
5142#endif
5143 },
5144 {"rightleft",
5145#ifdef FEAT_RIGHTLEFT
5146 1
5147#else
5148 0
5149#endif
5150 },
5151 {"ruby",
5152#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
5153 1
5154#else
5155 0
5156#endif
5157 },
5158 {"scrollbind", 1},
5159 {"showcmd",
5160#ifdef FEAT_CMDL_INFO
5161 1
5162#else
5163 0
5164#endif
5165 },
5166 {"cmdline_info",
5167#ifdef FEAT_CMDL_INFO
5168 1
5169#else
5170 0
5171#endif
5172 },
5173 {"signs",
5174#ifdef FEAT_SIGNS
5175 1
5176#else
5177 0
5178#endif
5179 },
5180 {"smartindent",
5181#ifdef FEAT_SMARTINDENT
5182 1
5183#else
5184 0
5185#endif
5186 },
5187 {"startuptime",
5188#ifdef STARTUPTIME
5189 1
5190#else
5191 0
5192#endif
5193 },
5194 {"statusline",
5195#ifdef FEAT_STL_OPT
5196 1
5197#else
5198 0
5199#endif
5200 },
5201 {"netbeans_intg",
5202#ifdef FEAT_NETBEANS_INTG
5203 1
5204#else
5205 0
5206#endif
5207 },
5208 {"sound",
5209#ifdef FEAT_SOUND
5210 1
5211#else
5212 0
5213#endif
5214 },
5215 {"spell",
5216#ifdef FEAT_SPELL
5217 1
5218#else
5219 0
5220#endif
5221 },
5222 {"syntax",
5223#ifdef FEAT_SYN_HL
5224 1
5225#else
5226 0
5227#endif
5228 },
5229 {"system",
5230#if defined(USE_SYSTEM) || !defined(UNIX)
5231 1
5232#else
5233 0
5234#endif
5235 },
5236 {"tag_binary",
5237#ifdef FEAT_TAG_BINS
5238 1
5239#else
5240 0
5241#endif
5242 },
5243 {"tcl",
5244#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
5245 1
5246#else
5247 0
5248#endif
5249 },
5250 {"termguicolors",
5251#ifdef FEAT_TERMGUICOLORS
5252 1
5253#else
5254 0
5255#endif
5256 },
5257 {"terminal",
5258#if defined(FEAT_TERMINAL) && !defined(MSWIN)
5259 1
5260#else
5261 0
5262#endif
5263 },
5264 {"terminfo",
5265#ifdef TERMINFO
5266 1
5267#else
5268 0
5269#endif
5270 },
5271 {"termresponse",
5272#ifdef FEAT_TERMRESPONSE
5273 1
5274#else
5275 0
5276#endif
5277 },
5278 {"textobjects",
5279#ifdef FEAT_TEXTOBJ
5280 1
5281#else
5282 0
5283#endif
5284 },
5285 {"textprop",
5286#ifdef FEAT_PROP_POPUP
5287 1
5288#else
5289 0
5290#endif
5291 },
5292 {"tgetent",
5293#ifdef HAVE_TGETENT
5294 1
5295#else
5296 0
5297#endif
5298 },
5299 {"timers",
5300#ifdef FEAT_TIMERS
5301 1
5302#else
5303 0
5304#endif
5305 },
5306 {"title",
5307#ifdef FEAT_TITLE
5308 1
5309#else
5310 0
5311#endif
5312 },
5313 {"toolbar",
5314#ifdef FEAT_TOOLBAR
5315 1
5316#else
5317 0
5318#endif
5319 },
5320 {"unnamedplus",
5321#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
5322 1
5323#else
5324 0
5325#endif
5326 },
5327 {"user-commands", 1}, // was accidentally included in 5.4
5328 {"user_commands", 1},
5329 {"vartabs",
5330#ifdef FEAT_VARTABS
5331 1
5332#else
5333 0
5334#endif
5335 },
5336 {"vertsplit", 1},
5337 {"viminfo",
5338#ifdef FEAT_VIMINFO
5339 1
5340#else
5341 0
5342#endif
5343 },
5344 {"vimscript-1", 1},
5345 {"vimscript-2", 1},
5346 {"vimscript-3", 1},
5347 {"vimscript-4", 1},
5348 {"virtualedit", 1},
5349 {"visual", 1},
5350 {"visualextra", 1},
5351 {"vreplace", 1},
5352 {"vtp",
5353#ifdef FEAT_VTP
5354 1
5355#else
5356 0
5357#endif
5358 },
5359 {"wildignore",
5360#ifdef FEAT_WILDIGN
5361 1
5362#else
5363 0
5364#endif
5365 },
5366 {"wildmenu",
5367#ifdef FEAT_WILDMENU
5368 1
5369#else
5370 0
5371#endif
5372 },
5373 {"windows", 1},
5374 {"winaltkeys",
5375#ifdef FEAT_WAK
5376 1
5377#else
5378 0
5379#endif
5380 },
5381 {"writebackup",
5382#ifdef FEAT_WRITEBACKUP
5383 1
5384#else
5385 0
5386#endif
5387 },
5388 {"xim",
5389#ifdef FEAT_XIM
5390 1
5391#else
5392 0
5393#endif
5394 },
5395 {"xfontset",
5396#ifdef FEAT_XFONTSET
5397 1
5398#else
5399 0
5400#endif
5401 },
5402 {"xpm",
5403#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
5404 1
5405#else
5406 0
5407#endif
5408 },
5409 {"xpm_w32", // for backward compatibility
5410#ifdef FEAT_XPM_W32
5411 1
5412#else
5413 0
5414#endif
5415 },
5416 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005417#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01005418 1
5419#else
5420 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005421#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005422 },
5423 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005424#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01005425 1
5426#else
5427 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005428#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005429 },
5430 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005431#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01005432 1
5433#else
5434 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005435#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005436 },
5437 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005438#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01005439 1
5440#else
5441 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005442#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005443 },
5444 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005445#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01005446 1
5447#else
5448 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005449#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005450 },
5451 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005452 };
5453
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005454 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01005455 for (i = 0; has_list[i].name != NULL; ++i)
5456 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005457 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005458 x = TRUE;
5459 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005460 break;
5461 }
5462
Bram Moolenaar79296512020-03-22 16:17:14 +01005463 // features also in has_list[] but sometimes enabled at runtime
5464 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005465 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005466 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02005467 {
5468 // intentionally empty
5469 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01005470#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005471 else if (STRICMP(name, "balloon_multiline") == 0)
5472 n = multiline_balloon_available();
5473#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005474#ifdef VIMDLL
5475 else if (STRICMP(name, "filterpipe") == 0)
5476 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005477#endif
5478#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5479 else if (STRICMP(name, "iconv") == 0)
5480 n = iconv_enabled(FALSE);
5481#endif
5482#ifdef DYNAMIC_LUA
5483 else if (STRICMP(name, "lua") == 0)
5484 n = lua_enabled(FALSE);
5485#endif
5486#ifdef DYNAMIC_MZSCHEME
5487 else if (STRICMP(name, "mzscheme") == 0)
5488 n = mzscheme_enabled(FALSE);
5489#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005490#ifdef DYNAMIC_PERL
5491 else if (STRICMP(name, "perl") == 0)
5492 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005493#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005494#ifdef DYNAMIC_PYTHON
5495 else if (STRICMP(name, "python") == 0)
5496 n = python_enabled(FALSE);
5497#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005498#ifdef DYNAMIC_PYTHON3
5499 else if (STRICMP(name, "python3") == 0)
5500 n = python3_enabled(FALSE);
5501#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005502#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5503 else if (STRICMP(name, "pythonx") == 0)
5504 {
5505# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
5506 if (p_pyx == 0)
5507 n = python3_enabled(FALSE) || python_enabled(FALSE);
5508 else if (p_pyx == 3)
5509 n = python3_enabled(FALSE);
5510 else if (p_pyx == 2)
5511 n = python_enabled(FALSE);
5512# elif defined(DYNAMIC_PYTHON)
5513 n = python_enabled(FALSE);
5514# elif defined(DYNAMIC_PYTHON3)
5515 n = python3_enabled(FALSE);
5516# endif
5517 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005518#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005519#ifdef DYNAMIC_RUBY
5520 else if (STRICMP(name, "ruby") == 0)
5521 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005522#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005523#ifdef DYNAMIC_TCL
5524 else if (STRICMP(name, "tcl") == 0)
5525 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02005526#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01005527#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02005528 else if (STRICMP(name, "terminal") == 0)
5529 n = terminal_enabled();
5530#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005531 }
5532
Bram Moolenaar79296512020-03-22 16:17:14 +01005533 // features not in has_list[]
5534 if (x == FALSE)
5535 {
5536 if (STRNICMP(name, "patch", 5) == 0)
5537 {
5538 x = TRUE;
5539 if (name[5] == '-'
5540 && STRLEN(name) >= 11
5541 && vim_isdigit(name[6])
5542 && vim_isdigit(name[8])
5543 && vim_isdigit(name[10]))
5544 {
5545 int major = atoi((char *)name + 6);
5546 int minor = atoi((char *)name + 8);
5547
5548 // Expect "patch-9.9.01234".
5549 n = (major < VIM_VERSION_MAJOR
5550 || (major == VIM_VERSION_MAJOR
5551 && (minor < VIM_VERSION_MINOR
5552 || (minor == VIM_VERSION_MINOR
5553 && has_patch(atoi((char *)name + 10))))));
5554 }
5555 else
5556 n = has_patch(atoi((char *)name + 5));
5557 }
5558 else if (STRICMP(name, "vim_starting") == 0)
5559 {
5560 x = TRUE;
5561 n = (starting != 0);
5562 }
5563 else if (STRICMP(name, "ttyin") == 0)
5564 {
5565 x = TRUE;
5566 n = mch_input_isatty();
5567 }
5568 else if (STRICMP(name, "ttyout") == 0)
5569 {
5570 x = TRUE;
5571 n = stdout_isatty;
5572 }
5573 else if (STRICMP(name, "multi_byte_encoding") == 0)
5574 {
5575 x = TRUE;
5576 n = has_mbyte;
5577 }
5578 else if (STRICMP(name, "gui_running") == 0)
5579 {
5580 x = TRUE;
5581#ifdef FEAT_GUI
5582 n = (gui.in_use || gui.starting);
5583#endif
5584 }
5585 else if (STRICMP(name, "browse") == 0)
5586 {
5587 x = TRUE;
5588#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5589 n = gui.in_use; // gui_mch_browse() works when GUI is running
5590#endif
5591 }
5592 else if (STRICMP(name, "syntax_items") == 0)
5593 {
5594 x = TRUE;
5595#ifdef FEAT_SYN_HL
5596 n = syntax_present(curwin);
5597#endif
5598 }
5599 else if (STRICMP(name, "vcon") == 0)
5600 {
5601 x = TRUE;
5602#ifdef FEAT_VTP
5603 n = is_term_win32() && has_vtp_working();
5604#endif
5605 }
5606 else if (STRICMP(name, "netbeans_enabled") == 0)
5607 {
5608 x = TRUE;
5609#ifdef FEAT_NETBEANS_INTG
5610 n = netbeans_active();
5611#endif
5612 }
5613 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
5614 {
5615 x = TRUE;
5616#ifdef FEAT_MOUSE_GPM
5617 n = gpm_enabled();
5618#endif
5619 }
5620 else if (STRICMP(name, "conpty") == 0)
5621 {
5622 x = TRUE;
5623#if defined(FEAT_TERMINAL) && defined(MSWIN)
5624 n = use_conpty();
5625#endif
5626 }
5627 else if (STRICMP(name, "clipboard_working") == 0)
5628 {
5629 x = TRUE;
5630#ifdef FEAT_CLIPBOARD
5631 n = clip_star.available;
5632#endif
5633 }
5634 }
5635
Bram Moolenaar04637e22020-09-05 18:45:29 +02005636 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[1]))
Bram Moolenaar79296512020-03-22 16:17:14 +01005637 // return whether feature could ever be enabled
5638 rettv->vval.v_number = x;
5639 else
5640 // return whether feature is enabled
5641 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005642}
5643
5644/*
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005645 * Return TRUE if "feature" can change later.
5646 * Also when checking for the feature has side effects, such as loading a DLL.
5647 */
5648 int
5649dynamic_feature(char_u *feature)
5650{
5651 return (feature == NULL
5652#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
5653 || STRICMP(feature, "balloon_multiline") == 0
5654#endif
5655#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5656 || (STRICMP(feature, "browse") == 0 && !gui.in_use)
5657#endif
5658#ifdef VIMDLL
5659 || STRICMP(feature, "filterpipe") == 0
5660#endif
Bram Moolenaar29b281b2020-11-10 20:58:00 +01005661#if defined(FEAT_GUI) && !defined(ALWAYS_USE_GUI) && !defined(VIMDLL)
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005662 // this can only change on Unix where the ":gui" command could be
5663 // used.
5664 || (STRICMP(feature, "gui_running") == 0 && !gui.in_use)
5665#endif
5666#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5667 || STRICMP(feature, "iconv") == 0
5668#endif
5669#ifdef DYNAMIC_LUA
5670 || STRICMP(feature, "lua") == 0
5671#endif
5672#ifdef FEAT_MOUSE_GPM
5673 || (STRICMP(feature, "mouse_gpm_enabled") == 0 && !gpm_enabled())
5674#endif
5675#ifdef DYNAMIC_MZSCHEME
5676 || STRICMP(feature, "mzscheme") == 0
5677#endif
5678#ifdef FEAT_NETBEANS_INTG
5679 || STRICMP(feature, "netbeans_enabled") == 0
5680#endif
5681#ifdef DYNAMIC_PERL
5682 || STRICMP(feature, "perl") == 0
5683#endif
5684#ifdef DYNAMIC_PYTHON
5685 || STRICMP(feature, "python") == 0
5686#endif
5687#ifdef DYNAMIC_PYTHON3
5688 || STRICMP(feature, "python3") == 0
5689#endif
5690#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5691 || STRICMP(feature, "pythonx") == 0
5692#endif
5693#ifdef DYNAMIC_RUBY
5694 || STRICMP(feature, "ruby") == 0
5695#endif
5696#ifdef FEAT_SYN_HL
5697 || STRICMP(feature, "syntax_items") == 0
5698#endif
5699#ifdef DYNAMIC_TCL
5700 || STRICMP(feature, "tcl") == 0
5701#endif
5702 // once "starting" is zero it will stay that way
5703 || (STRICMP(feature, "vim_starting") == 0 && starting != 0)
5704 || STRICMP(feature, "multi_byte_encoding") == 0
5705#if defined(FEAT_TERMINAL) && defined(MSWIN)
5706 || STRICMP(feature, "conpty") == 0
5707#endif
5708 );
5709}
5710
5711/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005712 * "haslocaldir()" function
5713 */
5714 static void
5715f_haslocaldir(typval_T *argvars, typval_T *rettv)
5716{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005717 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005718 win_T *wp = NULL;
5719
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005720 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
5721
5722 // Check for window-local and tab-local directories
5723 if (wp != NULL && wp->w_localdir != NULL)
5724 rettv->vval.v_number = 1;
5725 else if (tp != NULL && tp->tp_localdir != NULL)
5726 rettv->vval.v_number = 2;
5727 else
5728 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005729}
5730
5731/*
5732 * "hasmapto()" function
5733 */
5734 static void
5735f_hasmapto(typval_T *argvars, typval_T *rettv)
5736{
5737 char_u *name;
5738 char_u *mode;
5739 char_u buf[NUMBUFLEN];
5740 int abbr = FALSE;
5741
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005742 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005743 if (argvars[1].v_type == VAR_UNKNOWN)
5744 mode = (char_u *)"nvo";
5745 else
5746 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005747 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005748 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02005749 abbr = (int)tv_get_bool(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005750 }
5751
5752 if (map_to_exists(name, mode, abbr))
5753 rettv->vval.v_number = TRUE;
5754 else
5755 rettv->vval.v_number = FALSE;
5756}
5757
5758/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005759 * "highlightID(name)" function
5760 */
5761 static void
5762f_hlID(typval_T *argvars, typval_T *rettv)
5763{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005764 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005765}
5766
5767/*
5768 * "highlight_exists()" function
5769 */
5770 static void
5771f_hlexists(typval_T *argvars, typval_T *rettv)
5772{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005773 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005774}
5775
5776/*
5777 * "hostname()" function
5778 */
5779 static void
5780f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
5781{
5782 char_u hostname[256];
5783
5784 mch_get_host_name(hostname, 256);
5785 rettv->v_type = VAR_STRING;
5786 rettv->vval.v_string = vim_strsave(hostname);
5787}
5788
5789/*
5790 * iconv() function
5791 */
5792 static void
5793f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
5794{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005795 char_u buf1[NUMBUFLEN];
5796 char_u buf2[NUMBUFLEN];
5797 char_u *from, *to, *str;
5798 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005799
5800 rettv->v_type = VAR_STRING;
5801 rettv->vval.v_string = NULL;
5802
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005803 str = tv_get_string(&argvars[0]);
5804 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
5805 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005806 vimconv.vc_type = CONV_NONE;
5807 convert_setup(&vimconv, from, to);
5808
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005809 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005810 if (vimconv.vc_type == CONV_NONE)
5811 rettv->vval.v_string = vim_strsave(str);
5812 else
5813 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
5814
5815 convert_setup(&vimconv, NULL, NULL);
5816 vim_free(from);
5817 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005818}
5819
5820/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005821 * "index()" function
5822 */
5823 static void
5824f_index(typval_T *argvars, typval_T *rettv)
5825{
5826 list_T *l;
5827 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005828 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005829 long idx = 0;
5830 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005831 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005832
5833 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005834 if (argvars[0].v_type == VAR_BLOB)
5835 {
5836 typval_T tv;
5837 int start = 0;
5838
5839 if (argvars[2].v_type != VAR_UNKNOWN)
5840 {
5841 start = tv_get_number_chk(&argvars[2], &error);
5842 if (error)
5843 return;
5844 }
5845 b = argvars[0].vval.v_blob;
5846 if (b == NULL)
5847 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01005848 if (start < 0)
5849 {
5850 start = blob_len(b) + start;
5851 if (start < 0)
5852 start = 0;
5853 }
5854
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005855 for (idx = start; idx < blob_len(b); ++idx)
5856 {
5857 tv.v_type = VAR_NUMBER;
5858 tv.vval.v_number = blob_get(b, idx);
5859 if (tv_equal(&tv, &argvars[1], ic, FALSE))
5860 {
5861 rettv->vval.v_number = idx;
5862 return;
5863 }
5864 }
5865 return;
5866 }
5867 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005868 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01005869 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005870 return;
5871 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005872
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005873 l = argvars[0].vval.v_list;
5874 if (l != NULL)
5875 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005876 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005877 item = l->lv_first;
5878 if (argvars[2].v_type != VAR_UNKNOWN)
5879 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005880 // Start at specified item. Use the cached index that list_find()
5881 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005882 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005883 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005884 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar6c553f92020-09-02 22:10:34 +02005885 ic = (int)tv_get_bool_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005886 if (error)
5887 item = NULL;
5888 }
5889
5890 for ( ; item != NULL; item = item->li_next, ++idx)
5891 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
5892 {
5893 rettv->vval.v_number = idx;
5894 break;
5895 }
5896 }
5897}
5898
5899static int inputsecret_flag = 0;
5900
5901/*
5902 * "input()" function
5903 * Also handles inputsecret() when inputsecret is set.
5904 */
5905 static void
5906f_input(typval_T *argvars, typval_T *rettv)
5907{
5908 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
5909}
5910
5911/*
5912 * "inputdialog()" function
5913 */
5914 static void
5915f_inputdialog(typval_T *argvars, typval_T *rettv)
5916{
5917#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005918 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005919 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
5920 {
5921 char_u *message;
5922 char_u buf[NUMBUFLEN];
5923 char_u *defstr = (char_u *)"";
5924
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005925 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005926 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005927 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005928 vim_strncpy(IObuff, defstr, IOSIZE - 1);
5929 else
5930 IObuff[0] = NUL;
5931 if (message != NULL && defstr != NULL
5932 && do_dialog(VIM_QUESTION, NULL, message,
5933 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
5934 rettv->vval.v_string = vim_strsave(IObuff);
5935 else
5936 {
5937 if (message != NULL && defstr != NULL
5938 && argvars[1].v_type != VAR_UNKNOWN
5939 && argvars[2].v_type != VAR_UNKNOWN)
5940 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005941 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005942 else
5943 rettv->vval.v_string = NULL;
5944 }
5945 rettv->v_type = VAR_STRING;
5946 }
5947 else
5948#endif
5949 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
5950}
5951
5952/*
5953 * "inputlist()" function
5954 */
5955 static void
5956f_inputlist(typval_T *argvars, typval_T *rettv)
5957{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005958 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005959 listitem_T *li;
5960 int selected;
5961 int mouse_used;
5962
5963#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005964 // While starting up, there is no place to enter text. When running tests
5965 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005966 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005967 return;
5968#endif
5969 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5970 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005971 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005972 return;
5973 }
5974
5975 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005976 msg_row = Rows - 1; // for when 'cmdheight' > 1
5977 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005978 msg_scroll = TRUE;
5979 msg_clr_eos();
5980
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005981 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005982 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02005983 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005984 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005985 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005986 msg_putchar('\n');
5987 }
5988
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005989 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005990 selected = prompt_for_number(&mouse_used);
5991 if (mouse_used)
5992 selected -= lines_left;
5993
5994 rettv->vval.v_number = selected;
5995}
5996
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005997static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
5998
5999/*
6000 * "inputrestore()" function
6001 */
6002 static void
6003f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
6004{
6005 if (ga_userinput.ga_len > 0)
6006 {
6007 --ga_userinput.ga_len;
6008 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
6009 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006010 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006011 }
6012 else if (p_verbose > 1)
6013 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01006014 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006015 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006016 }
6017}
6018
6019/*
6020 * "inputsave()" function
6021 */
6022 static void
6023f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
6024{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006025 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006026 if (ga_grow(&ga_userinput, 1) == OK)
6027 {
6028 save_typeahead((tasave_T *)(ga_userinput.ga_data)
6029 + ga_userinput.ga_len);
6030 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006031 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006032 }
6033 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006034 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006035}
6036
6037/*
6038 * "inputsecret()" function
6039 */
6040 static void
6041f_inputsecret(typval_T *argvars, typval_T *rettv)
6042{
6043 ++cmdline_star;
6044 ++inputsecret_flag;
6045 f_input(argvars, rettv);
6046 --cmdline_star;
6047 --inputsecret_flag;
6048}
6049
6050/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01006051 * "interrupt()" function
6052 */
6053 static void
6054f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6055{
6056 got_int = TRUE;
6057}
6058
6059/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006060 * "invert(expr)" function
6061 */
6062 static void
6063f_invert(typval_T *argvars, typval_T *rettv)
6064{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006065 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006066}
6067
6068/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006069 * "islocked()" function
6070 */
6071 static void
6072f_islocked(typval_T *argvars, typval_T *rettv)
6073{
6074 lval_T lv;
6075 char_u *end;
6076 dictitem_T *di;
6077
6078 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006079 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01006080 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006081 if (end != NULL && lv.ll_name != NULL)
6082 {
6083 if (*end != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02006084 semsg(_(e_trailing_arg), end);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006085 else
6086 {
6087 if (lv.ll_tv == NULL)
6088 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01006089 di = find_var(lv.ll_name, NULL, TRUE);
6090 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006091 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006092 // Consider a variable locked when:
6093 // 1. the variable itself is locked
6094 // 2. the value of the variable is locked.
6095 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01006096 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
6097 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006098 }
6099 }
6100 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006101 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006102 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006103 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006104 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006105 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006106 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
6107 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006108 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006109 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
6110 }
6111 }
6112
6113 clear_lval(&lv);
6114}
6115
6116#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
6117/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02006118 * "isinf()" function
6119 */
6120 static void
6121f_isinf(typval_T *argvars, typval_T *rettv)
6122{
6123 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
6124 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
6125}
6126
6127/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006128 * "isnan()" function
6129 */
6130 static void
6131f_isnan(typval_T *argvars, typval_T *rettv)
6132{
6133 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
6134 && isnan(argvars[0].vval.v_float);
6135}
6136#endif
6137
6138/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006139 * "last_buffer_nr()" function.
6140 */
6141 static void
6142f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
6143{
6144 int n = 0;
6145 buf_T *buf;
6146
Bram Moolenaar29323592016-07-24 22:04:11 +02006147 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006148 if (n < buf->b_fnum)
6149 n = buf->b_fnum;
6150
6151 rettv->vval.v_number = n;
6152}
6153
6154/*
6155 * "len()" function
6156 */
6157 static void
6158f_len(typval_T *argvars, typval_T *rettv)
6159{
6160 switch (argvars[0].v_type)
6161 {
6162 case VAR_STRING:
6163 case VAR_NUMBER:
6164 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006165 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006166 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01006167 case VAR_BLOB:
6168 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
6169 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006170 case VAR_LIST:
6171 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
6172 break;
6173 case VAR_DICT:
6174 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
6175 break;
6176 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02006177 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006178 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01006179 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006180 case VAR_SPECIAL:
6181 case VAR_FLOAT:
6182 case VAR_FUNC:
6183 case VAR_PARTIAL:
6184 case VAR_JOB:
6185 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006186 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006187 break;
6188 }
6189}
6190
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006191 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01006192libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006193{
6194#ifdef FEAT_LIBCALL
6195 char_u *string_in;
6196 char_u **string_result;
6197 int nr_result;
6198#endif
6199
6200 rettv->v_type = type;
6201 if (type != VAR_NUMBER)
6202 rettv->vval.v_string = NULL;
6203
6204 if (check_restricted() || check_secure())
6205 return;
6206
6207#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006208 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006209 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
6210 {
6211 string_in = NULL;
6212 if (argvars[2].v_type == VAR_STRING)
6213 string_in = argvars[2].vval.v_string;
6214 if (type == VAR_NUMBER)
6215 string_result = NULL;
6216 else
6217 string_result = &rettv->vval.v_string;
6218 if (mch_libcall(argvars[0].vval.v_string,
6219 argvars[1].vval.v_string,
6220 string_in,
6221 argvars[2].vval.v_number,
6222 string_result,
6223 &nr_result) == OK
6224 && type == VAR_NUMBER)
6225 rettv->vval.v_number = nr_result;
6226 }
6227#endif
6228}
6229
6230/*
6231 * "libcall()" function
6232 */
6233 static void
6234f_libcall(typval_T *argvars, typval_T *rettv)
6235{
6236 libcall_common(argvars, rettv, VAR_STRING);
6237}
6238
6239/*
6240 * "libcallnr()" function
6241 */
6242 static void
6243f_libcallnr(typval_T *argvars, typval_T *rettv)
6244{
6245 libcall_common(argvars, rettv, VAR_NUMBER);
6246}
6247
6248/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006249 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006250 */
6251 static void
6252f_line(typval_T *argvars, typval_T *rettv)
6253{
6254 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006255 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006256 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006257 int id;
6258 tabpage_T *tp;
6259 win_T *wp;
6260 win_T *save_curwin;
6261 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006262
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006263 if (argvars[1].v_type != VAR_UNKNOWN)
6264 {
6265 // use window specified in the second argument
6266 id = (int)tv_get_number(&argvars[1]);
6267 wp = win_id2wp_tp(id, &tp);
6268 if (wp != NULL && tp != NULL)
6269 {
6270 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
6271 == OK)
6272 {
6273 check_cursor();
Bram Moolenaar6f02b002021-01-10 20:22:54 +01006274 fp = var2fpos(&argvars[0], TRUE, &fnum, FALSE);
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006275 }
6276 restore_win_noblock(save_curwin, save_curtab, TRUE);
6277 }
6278 }
6279 else
6280 // use current window
Bram Moolenaar6f02b002021-01-10 20:22:54 +01006281 fp = var2fpos(&argvars[0], TRUE, &fnum, FALSE);
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006282
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006283 if (fp != NULL)
6284 lnum = fp->lnum;
6285 rettv->vval.v_number = lnum;
6286}
6287
6288/*
6289 * "line2byte(lnum)" function
6290 */
6291 static void
6292f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
6293{
6294#ifndef FEAT_BYTEOFF
6295 rettv->vval.v_number = -1;
6296#else
6297 linenr_T lnum;
6298
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006299 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006300 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
6301 rettv->vval.v_number = -1;
6302 else
6303 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
6304 if (rettv->vval.v_number >= 0)
6305 ++rettv->vval.v_number;
6306#endif
6307}
6308
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006309#ifdef FEAT_FLOAT
6310/*
6311 * "log()" function
6312 */
6313 static void
6314f_log(typval_T *argvars, typval_T *rettv)
6315{
6316 float_T f = 0.0;
6317
6318 rettv->v_type = VAR_FLOAT;
6319 if (get_float_arg(argvars, &f) == OK)
6320 rettv->vval.v_float = log(f);
6321 else
6322 rettv->vval.v_float = 0.0;
6323}
6324
6325/*
6326 * "log10()" function
6327 */
6328 static void
6329f_log10(typval_T *argvars, typval_T *rettv)
6330{
6331 float_T f = 0.0;
6332
6333 rettv->v_type = VAR_FLOAT;
6334 if (get_float_arg(argvars, &f) == OK)
6335 rettv->vval.v_float = log10(f);
6336 else
6337 rettv->vval.v_float = 0.0;
6338}
6339#endif
6340
6341#ifdef FEAT_LUA
6342/*
6343 * "luaeval()" function
6344 */
6345 static void
6346f_luaeval(typval_T *argvars, typval_T *rettv)
6347{
6348 char_u *str;
6349 char_u buf[NUMBUFLEN];
6350
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006351 if (check_restricted() || check_secure())
6352 return;
6353
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006354 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006355 do_luaeval(str, argvars + 1, rettv);
6356}
6357#endif
6358
6359/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006360 * "maparg()" function
6361 */
6362 static void
6363f_maparg(typval_T *argvars, typval_T *rettv)
6364{
6365 get_maparg(argvars, rettv, TRUE);
6366}
6367
6368/*
6369 * "mapcheck()" function
6370 */
6371 static void
6372f_mapcheck(typval_T *argvars, typval_T *rettv)
6373{
6374 get_maparg(argvars, rettv, FALSE);
6375}
6376
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006377typedef enum
6378{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006379 MATCH_END, // matchend()
6380 MATCH_MATCH, // match()
6381 MATCH_STR, // matchstr()
6382 MATCH_LIST, // matchlist()
6383 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006384} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006385
6386 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006387find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006388{
6389 char_u *str = NULL;
6390 long len = 0;
6391 char_u *expr = NULL;
6392 char_u *pat;
6393 regmatch_T regmatch;
6394 char_u patbuf[NUMBUFLEN];
6395 char_u strbuf[NUMBUFLEN];
6396 char_u *save_cpo;
6397 long start = 0;
6398 long nth = 1;
6399 colnr_T startcol = 0;
6400 int match = 0;
6401 list_T *l = NULL;
6402 listitem_T *li = NULL;
6403 long idx = 0;
6404 char_u *tofree = NULL;
6405
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006406 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006407 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01006408 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006409
6410 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006411 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006412 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006413 // type MATCH_LIST: return empty list when there are no matches.
6414 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006415 if (rettv_list_alloc(rettv) == FAIL)
6416 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006417 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006418 && (list_append_string(rettv->vval.v_list,
6419 (char_u *)"", 0) == FAIL
6420 || list_append_number(rettv->vval.v_list,
6421 (varnumber_T)-1) == FAIL
6422 || list_append_number(rettv->vval.v_list,
6423 (varnumber_T)-1) == FAIL
6424 || list_append_number(rettv->vval.v_list,
6425 (varnumber_T)-1) == FAIL))
6426 {
6427 list_free(rettv->vval.v_list);
6428 rettv->vval.v_list = NULL;
6429 goto theend;
6430 }
6431 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006432 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006433 {
6434 rettv->v_type = VAR_STRING;
6435 rettv->vval.v_string = NULL;
6436 }
6437
6438 if (argvars[0].v_type == VAR_LIST)
6439 {
6440 if ((l = argvars[0].vval.v_list) == NULL)
6441 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006442 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006443 li = l->lv_first;
6444 }
6445 else
6446 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006447 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006448 len = (long)STRLEN(str);
6449 }
6450
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006451 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006452 if (pat == NULL)
6453 goto theend;
6454
6455 if (argvars[2].v_type != VAR_UNKNOWN)
6456 {
6457 int error = FALSE;
6458
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006459 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006460 if (error)
6461 goto theend;
6462 if (l != NULL)
6463 {
6464 li = list_find(l, start);
6465 if (li == NULL)
6466 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006467 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006468 }
6469 else
6470 {
6471 if (start < 0)
6472 start = 0;
6473 if (start > len)
6474 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006475 // When "count" argument is there ignore matches before "start",
6476 // otherwise skip part of the string. Differs when pattern is "^"
6477 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006478 if (argvars[3].v_type != VAR_UNKNOWN)
6479 startcol = start;
6480 else
6481 {
6482 str += start;
6483 len -= start;
6484 }
6485 }
6486
6487 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006488 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006489 if (error)
6490 goto theend;
6491 }
6492
6493 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
6494 if (regmatch.regprog != NULL)
6495 {
6496 regmatch.rm_ic = p_ic;
6497
6498 for (;;)
6499 {
6500 if (l != NULL)
6501 {
6502 if (li == NULL)
6503 {
6504 match = FALSE;
6505 break;
6506 }
6507 vim_free(tofree);
6508 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
6509 if (str == NULL)
6510 break;
6511 }
6512
6513 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
6514
6515 if (match && --nth <= 0)
6516 break;
6517 if (l == NULL && !match)
6518 break;
6519
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006520 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006521 if (l != NULL)
6522 {
6523 li = li->li_next;
6524 ++idx;
6525 }
6526 else
6527 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006528 startcol = (colnr_T)(regmatch.startp[0]
6529 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006530 if (startcol > (colnr_T)len
6531 || str + startcol <= regmatch.startp[0])
6532 {
6533 match = FALSE;
6534 break;
6535 }
6536 }
6537 }
6538
6539 if (match)
6540 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006541 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006542 {
6543 listitem_T *li1 = rettv->vval.v_list->lv_first;
6544 listitem_T *li2 = li1->li_next;
6545 listitem_T *li3 = li2->li_next;
6546 listitem_T *li4 = li3->li_next;
6547
6548 vim_free(li1->li_tv.vval.v_string);
6549 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006550 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006551 li3->li_tv.vval.v_number =
6552 (varnumber_T)(regmatch.startp[0] - expr);
6553 li4->li_tv.vval.v_number =
6554 (varnumber_T)(regmatch.endp[0] - expr);
6555 if (l != NULL)
6556 li2->li_tv.vval.v_number = (varnumber_T)idx;
6557 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006558 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006559 {
6560 int i;
6561
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006562 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006563 for (i = 0; i < NSUBEXP; ++i)
6564 {
6565 if (regmatch.endp[i] == NULL)
6566 {
6567 if (list_append_string(rettv->vval.v_list,
6568 (char_u *)"", 0) == FAIL)
6569 break;
6570 }
6571 else if (list_append_string(rettv->vval.v_list,
6572 regmatch.startp[i],
6573 (int)(regmatch.endp[i] - regmatch.startp[i]))
6574 == FAIL)
6575 break;
6576 }
6577 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006578 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006579 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006580 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006581 if (l != NULL)
6582 copy_tv(&li->li_tv, rettv);
6583 else
6584 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006585 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006586 }
6587 else if (l != NULL)
6588 rettv->vval.v_number = idx;
6589 else
6590 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006591 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006592 rettv->vval.v_number =
6593 (varnumber_T)(regmatch.startp[0] - str);
6594 else
6595 rettv->vval.v_number =
6596 (varnumber_T)(regmatch.endp[0] - str);
6597 rettv->vval.v_number += (varnumber_T)(str - expr);
6598 }
6599 }
6600 vim_regfree(regmatch.regprog);
6601 }
6602
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006603theend:
6604 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006605 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006606 listitem_remove(rettv->vval.v_list,
6607 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006608 vim_free(tofree);
6609 p_cpo = save_cpo;
6610}
6611
6612/*
6613 * "match()" function
6614 */
6615 static void
6616f_match(typval_T *argvars, typval_T *rettv)
6617{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006618 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006619}
6620
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006621/*
6622 * "matchend()" function
6623 */
6624 static void
6625f_matchend(typval_T *argvars, typval_T *rettv)
6626{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006627 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006628}
6629
6630/*
6631 * "matchlist()" function
6632 */
6633 static void
6634f_matchlist(typval_T *argvars, typval_T *rettv)
6635{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006636 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006637}
6638
6639/*
6640 * "matchstr()" function
6641 */
6642 static void
6643f_matchstr(typval_T *argvars, typval_T *rettv)
6644{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006645 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006646}
6647
6648/*
6649 * "matchstrpos()" function
6650 */
6651 static void
6652f_matchstrpos(typval_T *argvars, typval_T *rettv)
6653{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006654 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006655}
6656
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006657 static void
6658max_min(typval_T *argvars, typval_T *rettv, int domax)
6659{
6660 varnumber_T n = 0;
6661 varnumber_T i;
6662 int error = FALSE;
6663
6664 if (argvars[0].v_type == VAR_LIST)
6665 {
6666 list_T *l;
6667 listitem_T *li;
6668
6669 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006670 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006671 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006672 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006673 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006674 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
6675 n = l->lv_u.nonmat.lv_start;
6676 else
6677 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
6678 * l->lv_u.nonmat.lv_stride;
6679 }
6680 else
6681 {
6682 li = l->lv_first;
6683 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006684 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006685 n = tv_get_number_chk(&li->li_tv, &error);
6686 for (;;)
6687 {
6688 li = li->li_next;
6689 if (li == NULL)
6690 break;
6691 i = tv_get_number_chk(&li->li_tv, &error);
6692 if (domax ? i > n : i < n)
6693 n = i;
6694 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006695 }
6696 }
6697 }
6698 }
6699 else if (argvars[0].v_type == VAR_DICT)
6700 {
6701 dict_T *d;
6702 int first = TRUE;
6703 hashitem_T *hi;
6704 int todo;
6705
6706 d = argvars[0].vval.v_dict;
6707 if (d != NULL)
6708 {
6709 todo = (int)d->dv_hashtab.ht_used;
6710 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
6711 {
6712 if (!HASHITEM_EMPTY(hi))
6713 {
6714 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006715 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006716 if (first)
6717 {
6718 n = i;
6719 first = FALSE;
6720 }
6721 else if (domax ? i > n : i < n)
6722 n = i;
6723 }
6724 }
6725 }
6726 }
6727 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006728 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006729 rettv->vval.v_number = error ? 0 : n;
6730}
6731
6732/*
6733 * "max()" function
6734 */
6735 static void
6736f_max(typval_T *argvars, typval_T *rettv)
6737{
6738 max_min(argvars, rettv, TRUE);
6739}
6740
6741/*
6742 * "min()" function
6743 */
6744 static void
6745f_min(typval_T *argvars, typval_T *rettv)
6746{
6747 max_min(argvars, rettv, FALSE);
6748}
6749
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006750#if defined(FEAT_MZSCHEME) || defined(PROTO)
6751/*
6752 * "mzeval()" function
6753 */
6754 static void
6755f_mzeval(typval_T *argvars, typval_T *rettv)
6756{
6757 char_u *str;
6758 char_u buf[NUMBUFLEN];
6759
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006760 if (check_restricted() || check_secure())
6761 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006762 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006763 do_mzeval(str, rettv);
6764}
6765
6766 void
6767mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
6768{
6769 typval_T argvars[3];
6770
6771 argvars[0].v_type = VAR_STRING;
6772 argvars[0].vval.v_string = name;
6773 copy_tv(args, &argvars[1]);
6774 argvars[2].v_type = VAR_UNKNOWN;
6775 f_call(argvars, rettv);
6776 clear_tv(&argvars[1]);
6777}
6778#endif
6779
6780/*
6781 * "nextnonblank()" function
6782 */
6783 static void
6784f_nextnonblank(typval_T *argvars, typval_T *rettv)
6785{
6786 linenr_T lnum;
6787
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006788 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006789 {
6790 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
6791 {
6792 lnum = 0;
6793 break;
6794 }
6795 if (*skipwhite(ml_get(lnum)) != NUL)
6796 break;
6797 }
6798 rettv->vval.v_number = lnum;
6799}
6800
6801/*
6802 * "nr2char()" function
6803 */
6804 static void
6805f_nr2char(typval_T *argvars, typval_T *rettv)
6806{
6807 char_u buf[NUMBUFLEN];
6808
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006809 if (has_mbyte)
6810 {
6811 int utf8 = 0;
6812
6813 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaared6a4302020-09-05 20:29:41 +02006814 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006815 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01006816 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006817 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006818 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006819 }
6820 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006821 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006822 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006823 buf[1] = NUL;
6824 }
6825 rettv->v_type = VAR_STRING;
6826 rettv->vval.v_string = vim_strsave(buf);
6827}
6828
6829/*
6830 * "or(expr, expr)" function
6831 */
6832 static void
6833f_or(typval_T *argvars, typval_T *rettv)
6834{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006835 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
6836 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006837}
6838
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006839#ifdef FEAT_PERL
6840/*
6841 * "perleval()" function
6842 */
6843 static void
6844f_perleval(typval_T *argvars, typval_T *rettv)
6845{
6846 char_u *str;
6847 char_u buf[NUMBUFLEN];
6848
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006849 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006850 do_perleval(str, rettv);
6851}
6852#endif
6853
6854#ifdef FEAT_FLOAT
6855/*
6856 * "pow()" function
6857 */
6858 static void
6859f_pow(typval_T *argvars, typval_T *rettv)
6860{
6861 float_T fx = 0.0, fy = 0.0;
6862
6863 rettv->v_type = VAR_FLOAT;
6864 if (get_float_arg(argvars, &fx) == OK
6865 && get_float_arg(&argvars[1], &fy) == OK)
6866 rettv->vval.v_float = pow(fx, fy);
6867 else
6868 rettv->vval.v_float = 0.0;
6869}
6870#endif
6871
6872/*
6873 * "prevnonblank()" function
6874 */
6875 static void
6876f_prevnonblank(typval_T *argvars, typval_T *rettv)
6877{
6878 linenr_T lnum;
6879
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006880 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006881 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
6882 lnum = 0;
6883 else
6884 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
6885 --lnum;
6886 rettv->vval.v_number = lnum;
6887}
6888
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006889// This dummy va_list is here because:
6890// - passing a NULL pointer doesn't work when va_list isn't a pointer
6891// - locally in the function results in a "used before set" warning
6892// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006893static va_list ap;
6894
6895/*
6896 * "printf()" function
6897 */
6898 static void
6899f_printf(typval_T *argvars, typval_T *rettv)
6900{
6901 char_u buf[NUMBUFLEN];
6902 int len;
6903 char_u *s;
6904 int saved_did_emsg = did_emsg;
6905 char *fmt;
6906
6907 rettv->v_type = VAR_STRING;
6908 rettv->vval.v_string = NULL;
6909
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006910 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006911 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006912 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006913 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006914 if (!did_emsg)
6915 {
6916 s = alloc(len + 1);
6917 if (s != NULL)
6918 {
6919 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006920 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
6921 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006922 }
6923 }
6924 did_emsg |= saved_did_emsg;
6925}
6926
6927/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006928 * "pum_getpos()" function
6929 */
6930 static void
6931f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6932{
6933 if (rettv_dict_alloc(rettv) != OK)
6934 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006935 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006936}
6937
6938/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006939 * "pumvisible()" function
6940 */
6941 static void
6942f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6943{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006944 if (pum_visible())
6945 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006946}
6947
6948#ifdef FEAT_PYTHON3
6949/*
6950 * "py3eval()" function
6951 */
6952 static void
6953f_py3eval(typval_T *argvars, typval_T *rettv)
6954{
6955 char_u *str;
6956 char_u buf[NUMBUFLEN];
6957
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006958 if (check_restricted() || check_secure())
6959 return;
6960
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006961 if (p_pyx == 0)
6962 p_pyx = 3;
6963
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006964 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006965 do_py3eval(str, rettv);
6966}
6967#endif
6968
6969#ifdef FEAT_PYTHON
6970/*
6971 * "pyeval()" function
6972 */
6973 static void
6974f_pyeval(typval_T *argvars, typval_T *rettv)
6975{
6976 char_u *str;
6977 char_u buf[NUMBUFLEN];
6978
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006979 if (check_restricted() || check_secure())
6980 return;
6981
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006982 if (p_pyx == 0)
6983 p_pyx = 2;
6984
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006985 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006986 do_pyeval(str, rettv);
6987}
6988#endif
6989
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006990#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
6991/*
6992 * "pyxeval()" function
6993 */
6994 static void
6995f_pyxeval(typval_T *argvars, typval_T *rettv)
6996{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006997 if (check_restricted() || check_secure())
6998 return;
6999
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01007000# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
7001 init_pyxversion();
7002 if (p_pyx == 2)
7003 f_pyeval(argvars, rettv);
7004 else
7005 f_py3eval(argvars, rettv);
7006# elif defined(FEAT_PYTHON)
7007 f_pyeval(argvars, rettv);
7008# elif defined(FEAT_PYTHON3)
7009 f_py3eval(argvars, rettv);
7010# endif
7011}
7012#endif
7013
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007014static UINT32_T srand_seed_for_testing = 0;
7015static int srand_seed_for_testing_is_used = FALSE;
7016
7017 static void
7018f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
7019{
7020 if (argvars[0].v_type == VAR_UNKNOWN)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007021 srand_seed_for_testing_is_used = FALSE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007022 else
7023 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007024 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
7025 srand_seed_for_testing_is_used = TRUE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007026 }
7027}
7028
7029 static void
7030init_srand(UINT32_T *x)
7031{
7032#ifndef MSWIN
7033 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
7034#endif
7035
7036 if (srand_seed_for_testing_is_used)
7037 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007038 *x = srand_seed_for_testing;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007039 return;
7040 }
7041#ifndef MSWIN
7042 if (dev_urandom_state != FAIL)
7043 {
7044 int fd = open("/dev/urandom", O_RDONLY);
7045 struct {
7046 union {
7047 UINT32_T number;
7048 char bytes[sizeof(UINT32_T)];
7049 } contents;
7050 } buf;
7051
7052 // Attempt reading /dev/urandom.
7053 if (fd == -1)
7054 dev_urandom_state = FAIL;
7055 else
7056 {
7057 buf.contents.number = 0;
7058 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
7059 != sizeof(UINT32_T))
7060 dev_urandom_state = FAIL;
7061 else
7062 {
7063 dev_urandom_state = OK;
7064 *x = buf.contents.number;
7065 }
7066 close(fd);
7067 }
7068 }
7069 if (dev_urandom_state != OK)
7070 // Reading /dev/urandom doesn't work, fall back to time().
7071#endif
7072 *x = vim_time();
7073}
7074
7075#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
7076#define SPLITMIX32(x, z) ( \
7077 z = (x += 0x9e3779b9), \
7078 z = (z ^ (z >> 16)) * 0x85ebca6b, \
7079 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
7080 z ^ (z >> 16) \
7081 )
7082#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
7083 result = ROTL(y * 5, 7) * 9; \
7084 t = y << 9; \
7085 z ^= x; \
7086 w ^= y; \
7087 y ^= z, x ^= w; \
7088 z ^= t; \
7089 w = ROTL(w, 11);
7090
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007091/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007092 * "rand()" function
7093 */
7094 static void
7095f_rand(typval_T *argvars, typval_T *rettv)
7096{
7097 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007098 static UINT32_T gx, gy, gz, gw;
7099 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007100 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar0fd797e2020-11-05 20:46:32 +01007101 UINT32_T x = 0, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007102
7103 if (argvars[0].v_type == VAR_UNKNOWN)
7104 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007105 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007106 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007107 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007108 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007109 init_srand(&x);
7110
7111 gx = SPLITMIX32(x, z);
7112 gy = SPLITMIX32(x, z);
7113 gz = SPLITMIX32(x, z);
7114 gw = SPLITMIX32(x, z);
7115 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007116 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007117
7118 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007119 }
7120 else if (argvars[0].v_type == VAR_LIST)
7121 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007122 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007123 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007124 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007125
7126 lx = list_find(l, 0L);
7127 ly = list_find(l, 1L);
7128 lz = list_find(l, 2L);
7129 lw = list_find(l, 3L);
7130 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
7131 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
7132 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
7133 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
7134 x = (UINT32_T)lx->li_tv.vval.v_number;
7135 y = (UINT32_T)ly->li_tv.vval.v_number;
7136 z = (UINT32_T)lz->li_tv.vval.v_number;
7137 w = (UINT32_T)lw->li_tv.vval.v_number;
7138
7139 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
7140
7141 lx->li_tv.vval.v_number = (varnumber_T)x;
7142 ly->li_tv.vval.v_number = (varnumber_T)y;
7143 lz->li_tv.vval.v_number = (varnumber_T)z;
7144 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007145 }
7146 else
7147 goto theend;
7148
7149 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007150 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007151 return;
7152
7153theend:
7154 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007155 rettv->v_type = VAR_NUMBER;
7156 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007157}
7158
7159/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007160 * "srand()" function
7161 */
7162 static void
7163f_srand(typval_T *argvars, typval_T *rettv)
7164{
7165 UINT32_T x = 0, z;
7166
7167 if (rettv_list_alloc(rettv) == FAIL)
7168 return;
7169 if (argvars[0].v_type == VAR_UNKNOWN)
7170 {
7171 init_srand(&x);
7172 }
7173 else
7174 {
7175 int error = FALSE;
7176
7177 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
7178 if (error)
7179 return;
7180 }
7181
7182 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7183 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7184 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7185 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7186}
7187
7188#undef ROTL
7189#undef SPLITMIX32
7190#undef SHUFFLE_XOSHIRO128STARSTAR
7191
7192/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007193 * "range()" function
7194 */
7195 static void
7196f_range(typval_T *argvars, typval_T *rettv)
7197{
7198 varnumber_T start;
7199 varnumber_T end;
7200 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007201 int error = FALSE;
7202
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007203 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007204 if (argvars[1].v_type == VAR_UNKNOWN)
7205 {
7206 end = start - 1;
7207 start = 0;
7208 }
7209 else
7210 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007211 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007212 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007213 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007214 }
7215
7216 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007217 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007218 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007219 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007220 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007221 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007222 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007223 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007224 list_T *list = rettv->vval.v_list;
7225
7226 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007227 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007228 // be called.
7229 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01007230 list->lv_u.nonmat.lv_start = start;
7231 list->lv_u.nonmat.lv_end = end;
7232 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007233 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007234 }
7235}
7236
7237/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007238 * Materialize "list".
7239 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007240 */
7241 void
7242range_list_materialize(list_T *list)
7243{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007244 varnumber_T start = list->lv_u.nonmat.lv_start;
7245 varnumber_T end = list->lv_u.nonmat.lv_end;
7246 int stride = list->lv_u.nonmat.lv_stride;
7247 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007248
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007249 list->lv_first = NULL;
7250 list->lv_u.mat.lv_last = NULL;
7251 list->lv_len = 0;
7252 list->lv_u.mat.lv_idx_item = NULL;
7253 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
7254 if (list_append_number(list, (varnumber_T)i) == FAIL)
7255 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007256}
7257
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007258/*
7259 * "getreginfo()" function
7260 */
7261 static void
7262f_getreginfo(typval_T *argvars, typval_T *rettv)
7263{
7264 char_u *strregname;
7265 int regname;
7266 char_u buf[NUMBUFLEN + 2];
7267 long reglen = 0;
7268 dict_T *dict;
7269 list_T *list;
7270
7271 if (argvars[0].v_type != VAR_UNKNOWN)
7272 {
7273 strregname = tv_get_string_chk(&argvars[0]);
7274 if (strregname == NULL)
7275 return;
7276 }
7277 else
7278 strregname = get_vim_var_str(VV_REG);
7279
7280 regname = (strregname == NULL ? '"' : *strregname);
7281 if (regname == 0 || regname == '@')
7282 regname = '"';
7283
7284 if (rettv_dict_alloc(rettv) == FAIL)
7285 return;
7286 dict = rettv->vval.v_dict;
7287
7288 list = (list_T *)get_reg_contents(regname, GREG_EXPR_SRC | GREG_LIST);
7289 if (list == NULL)
7290 return;
Bram Moolenaar91639192020-06-29 19:55:58 +02007291 (void)dict_add_list(dict, "regcontents", list);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007292
7293 buf[0] = NUL;
7294 buf[1] = NUL;
7295 switch (get_reg_type(regname, &reglen))
7296 {
7297 case MLINE: buf[0] = 'V'; break;
7298 case MCHAR: buf[0] = 'v'; break;
7299 case MBLOCK:
7300 vim_snprintf((char *)buf, sizeof(buf), "%c%ld", Ctrl_V,
7301 reglen + 1);
7302 break;
7303 }
Bram Moolenaar91639192020-06-29 19:55:58 +02007304 (void)dict_add_string(dict, (char *)"regtype", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007305
7306 buf[0] = get_register_name(get_unname_register());
7307 buf[1] = NUL;
7308 if (regname == '"')
Bram Moolenaar91639192020-06-29 19:55:58 +02007309 (void)dict_add_string(dict, (char *)"points_to", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007310 else
7311 {
7312 dictitem_T *item = dictitem_alloc((char_u *)"isunnamed");
7313
7314 if (item != NULL)
7315 {
7316 item->di_tv.v_type = VAR_SPECIAL;
7317 item->di_tv.vval.v_number = regname == buf[0]
7318 ? VVAL_TRUE : VVAL_FALSE;
Bram Moolenaar91639192020-06-29 19:55:58 +02007319 (void)dict_add(dict, item);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007320 }
7321 }
7322}
7323
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02007324 static void
7325return_register(int regname, typval_T *rettv)
7326{
7327 char_u buf[2] = {0, 0};
7328
7329 buf[0] = (char_u)regname;
7330 rettv->v_type = VAR_STRING;
7331 rettv->vval.v_string = vim_strsave(buf);
7332}
7333
7334/*
7335 * "reg_executing()" function
7336 */
7337 static void
7338f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
7339{
7340 return_register(reg_executing, rettv);
7341}
7342
7343/*
7344 * "reg_recording()" function
7345 */
7346 static void
7347f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
7348{
7349 return_register(reg_recording, rettv);
7350}
7351
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01007352/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007353 * "rename({from}, {to})" function
7354 */
7355 static void
7356f_rename(typval_T *argvars, typval_T *rettv)
7357{
7358 char_u buf[NUMBUFLEN];
7359
7360 if (check_restricted() || check_secure())
7361 rettv->vval.v_number = -1;
7362 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007363 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
7364 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007365}
7366
7367/*
7368 * "repeat()" function
7369 */
7370 static void
7371f_repeat(typval_T *argvars, typval_T *rettv)
7372{
7373 char_u *p;
7374 int n;
7375 int slen;
7376 int len;
7377 char_u *r;
7378 int i;
7379
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007380 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007381 if (argvars[0].v_type == VAR_LIST)
7382 {
7383 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
7384 while (n-- > 0)
7385 if (list_extend(rettv->vval.v_list,
7386 argvars[0].vval.v_list, NULL) == FAIL)
7387 break;
7388 }
7389 else
7390 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007391 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007392 rettv->v_type = VAR_STRING;
7393 rettv->vval.v_string = NULL;
7394
7395 slen = (int)STRLEN(p);
7396 len = slen * n;
7397 if (len <= 0)
7398 return;
7399
7400 r = alloc(len + 1);
7401 if (r != NULL)
7402 {
7403 for (i = 0; i < n; i++)
7404 mch_memmove(r + i * slen, p, (size_t)slen);
7405 r[len] = NUL;
7406 }
7407
7408 rettv->vval.v_string = r;
7409 }
7410}
7411
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007412#define SP_NOMOVE 0x01 // don't move cursor
7413#define SP_REPEAT 0x02 // repeat to find outer pair
7414#define SP_RETCOUNT 0x04 // return matchcount
7415#define SP_SETPCMARK 0x08 // set previous context mark
7416#define SP_START 0x10 // accept match at start position
7417#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
7418#define SP_END 0x40 // leave cursor at end of match
7419#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007420
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007421/*
7422 * Get flags for a search function.
7423 * Possibly sets "p_ws".
7424 * Returns BACKWARD, FORWARD or zero (for an error).
7425 */
7426 static int
7427get_search_arg(typval_T *varp, int *flagsp)
7428{
7429 int dir = FORWARD;
7430 char_u *flags;
7431 char_u nbuf[NUMBUFLEN];
7432 int mask;
7433
7434 if (varp->v_type != VAR_UNKNOWN)
7435 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007436 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007437 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007438 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007439 while (*flags != NUL)
7440 {
7441 switch (*flags)
7442 {
7443 case 'b': dir = BACKWARD; break;
7444 case 'w': p_ws = TRUE; break;
7445 case 'W': p_ws = FALSE; break;
7446 default: mask = 0;
7447 if (flagsp != NULL)
7448 switch (*flags)
7449 {
7450 case 'c': mask = SP_START; break;
7451 case 'e': mask = SP_END; break;
7452 case 'm': mask = SP_RETCOUNT; break;
7453 case 'n': mask = SP_NOMOVE; break;
7454 case 'p': mask = SP_SUBPAT; break;
7455 case 'r': mask = SP_REPEAT; break;
7456 case 's': mask = SP_SETPCMARK; break;
7457 case 'z': mask = SP_COLUMN; break;
7458 }
7459 if (mask == 0)
7460 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007461 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007462 dir = 0;
7463 }
7464 else
7465 *flagsp |= mask;
7466 }
7467 if (dir == 0)
7468 break;
7469 ++flags;
7470 }
7471 }
7472 return dir;
7473}
7474
7475/*
7476 * Shared by search() and searchpos() functions.
7477 */
7478 static int
7479search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
7480{
7481 int flags;
7482 char_u *pat;
7483 pos_T pos;
7484 pos_T save_cursor;
7485 int save_p_ws = p_ws;
7486 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007487 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007488 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007489#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007490 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007491 long time_limit = 0;
7492#endif
7493 int options = SEARCH_KEEP;
7494 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007495 searchit_arg_T sia;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007496 int use_skip = FALSE;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007497 pos_T firstpos;
7498
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007499 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007500 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007501 if (dir == 0)
7502 goto theend;
7503 flags = *flagsp;
7504 if (flags & SP_START)
7505 options |= SEARCH_START;
7506 if (flags & SP_END)
7507 options |= SEARCH_END;
7508 if (flags & SP_COLUMN)
7509 options |= SEARCH_COL;
7510
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007511 // Optional arguments: line number to stop searching, timeout and skip.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007512 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
7513 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007514 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007515 if (lnum_stop < 0)
7516 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007517 if (argvars[3].v_type != VAR_UNKNOWN)
7518 {
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007519#ifdef FEAT_RELTIME
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007520 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007521 if (time_limit < 0)
7522 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007523#endif
Bram Moolenaara9c01042020-06-07 14:50:50 +02007524 use_skip = eval_expr_valid_arg(&argvars[4]);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007525 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007526 }
7527
7528#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007529 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007530 profile_setlimit(time_limit, &tm);
7531#endif
7532
7533 /*
7534 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
7535 * Check to make sure only those flags are set.
7536 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
7537 * flags cannot be set. Check for that condition also.
7538 */
7539 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
7540 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7541 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007542 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007543 goto theend;
7544 }
7545
7546 pos = save_cursor = curwin->w_cursor;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007547 CLEAR_FIELD(firstpos);
Bram Moolenaara80faa82020-04-12 19:37:17 +02007548 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007549 sia.sa_stop_lnum = (linenr_T)lnum_stop;
7550#ifdef FEAT_RELTIME
7551 sia.sa_tm = &tm;
7552#endif
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007553
7554 // Repeat until {skip} returns FALSE.
7555 for (;;)
7556 {
7557 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007558 options, RE_SEARCH, &sia);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007559 // finding the first match again means there is no match where {skip}
7560 // evaluates to zero.
7561 if (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos))
7562 subpatnum = FAIL;
7563
Bram Moolenaara9c01042020-06-07 14:50:50 +02007564 if (subpatnum == FAIL || !use_skip)
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007565 // didn't find it or no skip argument
7566 break;
7567 firstpos = pos;
7568
Bram Moolenaara9c01042020-06-07 14:50:50 +02007569 // If the skip expression matches, ignore this match.
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007570 {
7571 int do_skip;
7572 int err;
7573 pos_T save_pos = curwin->w_cursor;
7574
7575 curwin->w_cursor = pos;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007576 err = FALSE;
7577 do_skip = eval_expr_to_bool(&argvars[4], &err);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007578 curwin->w_cursor = save_pos;
7579 if (err)
7580 {
7581 // Evaluating {skip} caused an error, break here.
7582 subpatnum = FAIL;
7583 break;
7584 }
7585 if (!do_skip)
7586 break;
7587 }
7588 }
7589
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007590 if (subpatnum != FAIL)
7591 {
7592 if (flags & SP_SUBPAT)
7593 retval = subpatnum;
7594 else
7595 retval = pos.lnum;
7596 if (flags & SP_SETPCMARK)
7597 setpcmark();
7598 curwin->w_cursor = pos;
7599 if (match_pos != NULL)
7600 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007601 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007602 match_pos->lnum = pos.lnum;
7603 match_pos->col = pos.col + 1;
7604 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007605 // "/$" will put the cursor after the end of the line, may need to
7606 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007607 check_cursor();
7608 }
7609
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007610 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007611 if (flags & SP_NOMOVE)
7612 curwin->w_cursor = save_cursor;
7613 else
7614 curwin->w_set_curswant = TRUE;
7615theend:
7616 p_ws = save_p_ws;
7617
7618 return retval;
7619}
7620
7621#ifdef FEAT_FLOAT
7622
7623/*
7624 * round() is not in C90, use ceil() or floor() instead.
7625 */
7626 float_T
7627vim_round(float_T f)
7628{
7629 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
7630}
7631
7632/*
7633 * "round({float})" function
7634 */
7635 static void
7636f_round(typval_T *argvars, typval_T *rettv)
7637{
7638 float_T f = 0.0;
7639
7640 rettv->v_type = VAR_FLOAT;
7641 if (get_float_arg(argvars, &f) == OK)
7642 rettv->vval.v_float = vim_round(f);
7643 else
7644 rettv->vval.v_float = 0.0;
7645}
7646#endif
7647
Bram Moolenaare99be0e2019-03-26 22:51:09 +01007648#ifdef FEAT_RUBY
7649/*
7650 * "rubyeval()" function
7651 */
7652 static void
7653f_rubyeval(typval_T *argvars, typval_T *rettv)
7654{
7655 char_u *str;
7656 char_u buf[NUMBUFLEN];
7657
7658 str = tv_get_string_buf(&argvars[0], buf);
7659 do_rubyeval(str, rettv);
7660}
7661#endif
7662
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007663/*
7664 * "screenattr()" function
7665 */
7666 static void
7667f_screenattr(typval_T *argvars, typval_T *rettv)
7668{
7669 int row;
7670 int col;
7671 int c;
7672
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007673 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7674 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007675 if (row < 0 || row >= screen_Rows
7676 || col < 0 || col >= screen_Columns)
7677 c = -1;
7678 else
7679 c = ScreenAttrs[LineOffset[row] + col];
7680 rettv->vval.v_number = c;
7681}
7682
7683/*
7684 * "screenchar()" function
7685 */
7686 static void
7687f_screenchar(typval_T *argvars, typval_T *rettv)
7688{
7689 int row;
7690 int col;
7691 int off;
7692 int c;
7693
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007694 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7695 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007696 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007697 c = -1;
7698 else
7699 {
7700 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007701 if (enc_utf8 && ScreenLinesUC[off] != 0)
7702 c = ScreenLinesUC[off];
7703 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007704 c = ScreenLines[off];
7705 }
7706 rettv->vval.v_number = c;
7707}
7708
7709/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007710 * "screenchars()" function
7711 */
7712 static void
7713f_screenchars(typval_T *argvars, typval_T *rettv)
7714{
7715 int row;
7716 int col;
7717 int off;
7718 int c;
7719 int i;
7720
7721 if (rettv_list_alloc(rettv) == FAIL)
7722 return;
7723 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7724 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7725 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7726 return;
7727
7728 off = LineOffset[row] + col;
7729 if (enc_utf8 && ScreenLinesUC[off] != 0)
7730 c = ScreenLinesUC[off];
7731 else
7732 c = ScreenLines[off];
7733 list_append_number(rettv->vval.v_list, (varnumber_T)c);
7734
7735 if (enc_utf8)
7736
7737 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7738 list_append_number(rettv->vval.v_list,
7739 (varnumber_T)ScreenLinesC[i][off]);
7740}
7741
7742/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007743 * "screencol()" function
7744 *
7745 * First column is 1 to be consistent with virtcol().
7746 */
7747 static void
7748f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
7749{
7750 rettv->vval.v_number = screen_screencol() + 1;
7751}
7752
7753/*
7754 * "screenrow()" function
7755 */
7756 static void
7757f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
7758{
7759 rettv->vval.v_number = screen_screenrow() + 1;
7760}
7761
7762/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007763 * "screenstring()" function
7764 */
7765 static void
7766f_screenstring(typval_T *argvars, typval_T *rettv)
7767{
7768 int row;
7769 int col;
7770 int off;
7771 int c;
7772 int i;
7773 char_u buf[MB_MAXBYTES + 1];
7774 int buflen = 0;
7775
7776 rettv->vval.v_string = NULL;
7777 rettv->v_type = VAR_STRING;
7778
7779 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7780 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7781 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7782 return;
7783
7784 off = LineOffset[row] + col;
7785 if (enc_utf8 && ScreenLinesUC[off] != 0)
7786 c = ScreenLinesUC[off];
7787 else
7788 c = ScreenLines[off];
7789 buflen += mb_char2bytes(c, buf);
7790
7791 if (enc_utf8)
7792 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7793 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
7794
7795 buf[buflen] = NUL;
7796 rettv->vval.v_string = vim_strsave(buf);
7797}
7798
7799/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007800 * "search()" function
7801 */
7802 static void
7803f_search(typval_T *argvars, typval_T *rettv)
7804{
7805 int flags = 0;
7806
7807 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
7808}
7809
7810/*
7811 * "searchdecl()" function
7812 */
7813 static void
7814f_searchdecl(typval_T *argvars, typval_T *rettv)
7815{
Bram Moolenaar30788d32020-09-05 21:35:16 +02007816 int locally = TRUE;
7817 int thisblock = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007818 int error = FALSE;
7819 char_u *name;
7820
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007821 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007822
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007823 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007824 if (argvars[1].v_type != VAR_UNKNOWN)
7825 {
Bram Moolenaar30788d32020-09-05 21:35:16 +02007826 locally = !(int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007827 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar30788d32020-09-05 21:35:16 +02007828 thisblock = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007829 }
7830 if (!error && name != NULL)
7831 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
7832 locally, thisblock, SEARCH_KEEP) == FAIL;
7833}
7834
7835/*
7836 * Used by searchpair() and searchpairpos()
7837 */
7838 static int
7839searchpair_cmn(typval_T *argvars, pos_T *match_pos)
7840{
7841 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01007842 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007843 int save_p_ws = p_ws;
7844 int dir;
7845 int flags = 0;
7846 char_u nbuf1[NUMBUFLEN];
7847 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007848 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007849 long lnum_stop = 0;
7850 long time_limit = 0;
7851
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007852 // Get the three pattern arguments: start, middle, end. Will result in an
7853 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007854 spat = tv_get_string_chk(&argvars[0]);
7855 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
7856 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007857 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007858 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007859
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007860 // Handle the optional fourth argument: flags
7861 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007862 if (dir == 0)
7863 goto theend;
7864
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007865 // Don't accept SP_END or SP_SUBPAT.
7866 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007867 if ((flags & (SP_END | SP_SUBPAT)) != 0
7868 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7869 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007870 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007871 goto theend;
7872 }
7873
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007874 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007875 if (flags & SP_REPEAT)
7876 p_ws = FALSE;
7877
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007878 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007879 if (argvars[3].v_type == VAR_UNKNOWN
7880 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01007881 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007882 else
7883 {
Bram Moolenaara9c01042020-06-07 14:50:50 +02007884 // Type is checked later.
Bram Moolenaar48570482017-10-30 21:48:41 +01007885 skip = &argvars[4];
Bram Moolenaara9c01042020-06-07 14:50:50 +02007886
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007887 if (argvars[5].v_type != VAR_UNKNOWN)
7888 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007889 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007890 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007891 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007892 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007893 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007894 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007895#ifdef FEAT_RELTIME
7896 if (argvars[6].v_type != VAR_UNKNOWN)
7897 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007898 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007899 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007900 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007901 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007902 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007903 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007904 }
7905#endif
7906 }
7907 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007908
7909 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
7910 match_pos, lnum_stop, time_limit);
7911
7912theend:
7913 p_ws = save_p_ws;
7914
7915 return retval;
7916}
7917
7918/*
7919 * "searchpair()" function
7920 */
7921 static void
7922f_searchpair(typval_T *argvars, typval_T *rettv)
7923{
7924 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
7925}
7926
7927/*
7928 * "searchpairpos()" function
7929 */
7930 static void
7931f_searchpairpos(typval_T *argvars, typval_T *rettv)
7932{
7933 pos_T match_pos;
7934 int lnum = 0;
7935 int col = 0;
7936
7937 if (rettv_list_alloc(rettv) == FAIL)
7938 return;
7939
7940 if (searchpair_cmn(argvars, &match_pos) > 0)
7941 {
7942 lnum = match_pos.lnum;
7943 col = match_pos.col;
7944 }
7945
7946 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7947 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7948}
7949
7950/*
7951 * Search for a start/middle/end thing.
7952 * Used by searchpair(), see its documentation for the details.
7953 * Returns 0 or -1 for no match,
7954 */
7955 long
7956do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007957 char_u *spat, // start pattern
7958 char_u *mpat, // middle pattern
7959 char_u *epat, // end pattern
7960 int dir, // BACKWARD or FORWARD
7961 typval_T *skip, // skip expression
7962 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007963 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007964 linenr_T lnum_stop, // stop at this line if not zero
7965 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007966{
7967 char_u *save_cpo;
7968 char_u *pat, *pat2 = NULL, *pat3 = NULL;
7969 long retval = 0;
7970 pos_T pos;
7971 pos_T firstpos;
7972 pos_T foundpos;
7973 pos_T save_cursor;
7974 pos_T save_pos;
7975 int n;
7976 int r;
7977 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01007978 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007979 int err;
7980 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007981#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007982 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007983#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007984
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007985 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007986 save_cpo = p_cpo;
7987 p_cpo = empty_option;
7988
7989#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007990 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007991 profile_setlimit(time_limit, &tm);
7992#endif
7993
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007994 // Make two search patterns: start/end (pat2, for in nested pairs) and
7995 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02007996 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
7997 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007998 if (pat2 == NULL || pat3 == NULL)
7999 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01008000 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008001 if (*mpat == NUL)
8002 STRCPY(pat3, pat2);
8003 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01008004 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008005 spat, epat, mpat);
8006 if (flags & SP_START)
8007 options |= SEARCH_START;
8008
Bram Moolenaar48570482017-10-30 21:48:41 +01008009 if (skip != NULL)
Bram Moolenaara9c01042020-06-07 14:50:50 +02008010 use_skip = eval_expr_valid_arg(skip);
Bram Moolenaar48570482017-10-30 21:48:41 +01008011
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008012 save_cursor = curwin->w_cursor;
8013 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008014 CLEAR_POS(&firstpos);
8015 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008016 pat = pat3;
8017 for (;;)
8018 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008019 searchit_arg_T sia;
8020
Bram Moolenaara80faa82020-04-12 19:37:17 +02008021 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008022 sia.sa_stop_lnum = lnum_stop;
8023#ifdef FEAT_RELTIME
8024 sia.sa_tm = &tm;
8025#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01008026 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02008027 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008028 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008029 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008030 break;
8031
8032 if (firstpos.lnum == 0)
8033 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01008034 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008035 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008036 // Found the same position again. Can happen with a pattern that
8037 // has "\zs" at the end and searching backwards. Advance one
8038 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008039 if (dir == BACKWARD)
8040 decl(&pos);
8041 else
8042 incl(&pos);
8043 }
8044 foundpos = pos;
8045
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008046 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008047 options &= ~SEARCH_START;
8048
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008049 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01008050 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008051 {
8052 save_pos = curwin->w_cursor;
8053 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01008054 err = FALSE;
8055 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008056 curwin->w_cursor = save_pos;
8057 if (err)
8058 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008059 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008060 curwin->w_cursor = save_cursor;
8061 retval = -1;
8062 break;
8063 }
8064 if (r)
8065 continue;
8066 }
8067
8068 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
8069 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008070 // Found end when searching backwards or start when searching
8071 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008072 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008073 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008074 }
8075 else
8076 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008077 // Found end when searching forward or start when searching
8078 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008079 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008080 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008081 }
8082
8083 if (nest == 0)
8084 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008085 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008086 if (flags & SP_RETCOUNT)
8087 ++retval;
8088 else
8089 retval = pos.lnum;
8090 if (flags & SP_SETPCMARK)
8091 setpcmark();
8092 curwin->w_cursor = pos;
8093 if (!(flags & SP_REPEAT))
8094 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008095 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008096 }
8097 }
8098
8099 if (match_pos != NULL)
8100 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008101 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008102 match_pos->lnum = curwin->w_cursor.lnum;
8103 match_pos->col = curwin->w_cursor.col + 1;
8104 }
8105
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008106 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008107 if ((flags & SP_NOMOVE) || retval == 0)
8108 curwin->w_cursor = save_cursor;
8109
8110theend:
8111 vim_free(pat2);
8112 vim_free(pat3);
8113 if (p_cpo == empty_option)
8114 p_cpo = save_cpo;
8115 else
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008116 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008117 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008118 // If it's still empty it was changed and restored, need to restore in
8119 // the complicated way.
8120 if (*p_cpo == NUL)
8121 set_option_value((char_u *)"cpo", 0L, save_cpo, 0);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008122 free_string_option(save_cpo);
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008123 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008124
8125 return retval;
8126}
8127
8128/*
8129 * "searchpos()" function
8130 */
8131 static void
8132f_searchpos(typval_T *argvars, typval_T *rettv)
8133{
8134 pos_T match_pos;
8135 int lnum = 0;
8136 int col = 0;
8137 int n;
8138 int flags = 0;
8139
8140 if (rettv_list_alloc(rettv) == FAIL)
8141 return;
8142
8143 n = search_cmn(argvars, &match_pos, &flags);
8144 if (n > 0)
8145 {
8146 lnum = match_pos.lnum;
8147 col = match_pos.col;
8148 }
8149
8150 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
8151 list_append_number(rettv->vval.v_list, (varnumber_T)col);
8152 if (flags & SP_SUBPAT)
8153 list_append_number(rettv->vval.v_list, (varnumber_T)n);
8154}
8155
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008156/*
8157 * Set the cursor or mark position.
8158 * If 'charpos' is TRUE, then use the column number as a character offet.
8159 * Otherwise use the column number as a byte offset.
8160 */
8161 static void
8162set_position(typval_T *argvars, typval_T *rettv, int charpos)
8163{
8164 pos_T pos;
8165 int fnum;
8166 char_u *name;
8167 colnr_T curswant = -1;
8168
8169 rettv->vval.v_number = -1;
8170
8171 name = tv_get_string_chk(argvars);
8172 if (name != NULL)
8173 {
8174 if (list2fpos(&argvars[1], &pos, &fnum, &curswant, charpos) == OK)
8175 {
8176 if (pos.col != MAXCOL && --pos.col < 0)
8177 pos.col = 0;
8178 if ((name[0] == '.' && name[1] == NUL))
8179 {
8180 // set cursor; "fnum" is ignored
8181 curwin->w_cursor = pos;
8182 if (curswant >= 0)
8183 {
8184 curwin->w_curswant = curswant - 1;
8185 curwin->w_set_curswant = FALSE;
8186 }
8187 check_cursor();
8188 rettv->vval.v_number = 0;
8189 }
8190 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
8191 {
8192 // set mark
8193 if (setmark_pos(name[1], &pos, fnum) == OK)
8194 rettv->vval.v_number = 0;
8195 }
8196 else
8197 emsg(_(e_invarg));
8198 }
8199 }
8200}
8201/*
8202 * "setcharpos()" function
8203 */
8204 static void
8205f_setcharpos(typval_T *argvars, typval_T *rettv)
8206{
8207 set_position(argvars, rettv, TRUE);
8208}
8209
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008210 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008211f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
8212{
8213 dict_T *d;
8214 dictitem_T *di;
8215 char_u *csearch;
8216
8217 if (argvars[0].v_type != VAR_DICT)
8218 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008219 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008220 return;
8221 }
8222
8223 if ((d = argvars[0].vval.v_dict) != NULL)
8224 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01008225 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008226 if (csearch != NULL)
8227 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008228 if (enc_utf8)
8229 {
8230 int pcc[MAX_MCO];
8231 int c = utfc_ptr2char(csearch, pcc);
8232
8233 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
8234 }
8235 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008236 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02008237 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008238 }
8239
8240 di = dict_find(d, (char_u *)"forward", -1);
8241 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008242 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008243 ? FORWARD : BACKWARD);
8244
8245 di = dict_find(d, (char_u *)"until", -1);
8246 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008247 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008248 }
8249}
8250
8251/*
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008252 * "setcursorcharpos" function
8253 */
8254 static void
8255f_setcursorcharpos(typval_T *argvars, typval_T *rettv UNUSED)
8256{
8257 set_cursorpos(argvars, rettv, TRUE);
8258}
8259
8260/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02008261 * "setenv()" function
8262 */
8263 static void
8264f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
8265{
8266 char_u namebuf[NUMBUFLEN];
8267 char_u valbuf[NUMBUFLEN];
8268 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
8269
8270 if (argvars[1].v_type == VAR_SPECIAL
8271 && argvars[1].vval.v_number == VVAL_NULL)
8272 vim_unsetenv(name);
8273 else
8274 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
8275}
8276
8277/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008278 * "setfperm({fname}, {mode})" function
8279 */
8280 static void
8281f_setfperm(typval_T *argvars, typval_T *rettv)
8282{
8283 char_u *fname;
8284 char_u modebuf[NUMBUFLEN];
8285 char_u *mode_str;
8286 int i;
8287 int mask;
8288 int mode = 0;
8289
8290 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008291 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008292 if (fname == NULL)
8293 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008294 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008295 if (mode_str == NULL)
8296 return;
8297 if (STRLEN(mode_str) != 9)
8298 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008299 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008300 return;
8301 }
8302
8303 mask = 1;
8304 for (i = 8; i >= 0; --i)
8305 {
8306 if (mode_str[i] != '-')
8307 mode |= mask;
8308 mask = mask << 1;
8309 }
8310 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
8311}
8312
8313/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008314 * "setpos()" function
8315 */
8316 static void
8317f_setpos(typval_T *argvars, typval_T *rettv)
8318{
Bram Moolenaar6f02b002021-01-10 20:22:54 +01008319 set_position(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008320}
8321
8322/*
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008323 * Translate a register type string to the yank type and block length
8324 */
8325 static int
8326get_yank_type(char_u **pp, char_u *yank_type, long *block_len)
8327{
8328 char_u *stropt = *pp;
8329 switch (*stropt)
8330 {
8331 case 'v': case 'c': // character-wise selection
8332 *yank_type = MCHAR;
8333 break;
8334 case 'V': case 'l': // line-wise selection
8335 *yank_type = MLINE;
8336 break;
8337 case 'b': case Ctrl_V: // block-wise selection
8338 *yank_type = MBLOCK;
8339 if (VIM_ISDIGIT(stropt[1]))
8340 {
8341 ++stropt;
8342 *block_len = getdigits(&stropt) - 1;
8343 --stropt;
8344 }
8345 break;
8346 default:
8347 return FAIL;
8348 }
8349 *pp = stropt;
8350 return OK;
8351}
8352
8353/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008354 * "setreg()" function
8355 */
8356 static void
8357f_setreg(typval_T *argvars, typval_T *rettv)
8358{
8359 int regname;
8360 char_u *strregname;
8361 char_u *stropt;
8362 char_u *strval;
8363 int append;
8364 char_u yank_type;
8365 long block_len;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008366 typval_T *regcontents;
8367 int pointreg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008368
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008369 pointreg = 0;
8370 regcontents = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008371 block_len = -1;
8372 yank_type = MAUTO;
8373 append = FALSE;
8374
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008375 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008376 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008377
8378 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008379 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008380 regname = *strregname;
8381 if (regname == 0 || regname == '@')
8382 regname = '"';
8383
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008384 if (argvars[1].v_type == VAR_DICT)
8385 {
8386 dict_T *d = argvars[1].vval.v_dict;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008387 dictitem_T *di;
8388
8389 if (d == NULL || d->dv_hashtab.ht_used == 0)
8390 {
8391 // Empty dict, clear the register (like setreg(0, []))
8392 char_u *lstval[2] = {NULL, NULL};
8393 write_reg_contents_lst(regname, lstval, 0, FALSE, MAUTO, -1);
8394 return;
8395 }
8396
8397 di = dict_find(d, (char_u *)"regcontents", -1);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008398 if (di != NULL)
8399 regcontents = &di->di_tv;
8400
8401 stropt = dict_get_string(d, (char_u *)"regtype", FALSE);
8402 if (stropt != NULL)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008403 {
8404 int ret = get_yank_type(&stropt, &yank_type, &block_len);
8405
8406 if (ret == FAIL || *++stropt != NUL)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008407 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008408 semsg(_(e_invargval), "value");
8409 return;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008410 }
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008411 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008412
8413 if (regname == '"')
8414 {
8415 stropt = dict_get_string(d, (char_u *)"points_to", FALSE);
8416 if (stropt != NULL)
8417 {
8418 pointreg = *stropt;
8419 regname = pointreg;
8420 }
8421 }
Bram Moolenaar6a950582020-08-28 16:39:33 +02008422 else if (dict_get_bool(d, (char_u *)"isunnamed", -1) > 0)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008423 pointreg = regname;
8424 }
8425 else
8426 regcontents = &argvars[1];
8427
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008428 if (argvars[2].v_type != VAR_UNKNOWN)
8429 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008430 if (yank_type != MAUTO)
8431 {
8432 semsg(_(e_toomanyarg), "setreg");
8433 return;
8434 }
8435
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008436 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008437 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008438 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008439 for (; *stropt != NUL; ++stropt)
8440 switch (*stropt)
8441 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008442 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008443 append = TRUE;
8444 break;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008445 default:
8446 get_yank_type(&stropt, &yank_type, &block_len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008447 }
8448 }
8449
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008450 if (regcontents && regcontents->v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008451 {
8452 char_u **lstval;
8453 char_u **allocval;
8454 char_u buf[NUMBUFLEN];
8455 char_u **curval;
8456 char_u **curallocval;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008457 list_T *ll = regcontents->vval.v_list;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008458 listitem_T *li;
8459 int len;
8460
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008461 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008462 len = ll == NULL ? 0 : ll->lv_len;
8463
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008464 // First half: use for pointers to result lines; second half: use for
8465 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02008466 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008467 if (lstval == NULL)
8468 return;
8469 curval = lstval;
8470 allocval = lstval + len + 2;
8471 curallocval = allocval;
8472
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008473 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008474 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02008475 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02008476 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008477 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008478 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008479 if (strval == NULL)
8480 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008481 if (strval == buf)
8482 {
8483 // Need to make a copy, next tv_get_string_buf_chk() will
8484 // overwrite the string.
8485 strval = vim_strsave(buf);
8486 if (strval == NULL)
8487 goto free_lstval;
8488 *curallocval++ = strval;
8489 }
8490 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008491 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008492 }
8493 *curval++ = NULL;
8494
8495 write_reg_contents_lst(regname, lstval, -1,
8496 append, yank_type, block_len);
8497free_lstval:
8498 while (curallocval > allocval)
8499 vim_free(*--curallocval);
8500 vim_free(lstval);
8501 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008502 else if (regcontents)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008503 {
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008504 strval = tv_get_string_chk(regcontents);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008505 if (strval == NULL)
8506 return;
8507 write_reg_contents_ex(regname, strval, -1,
8508 append, yank_type, block_len);
8509 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008510 if (pointreg != 0)
8511 get_yank_register(pointreg, TRUE);
8512
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008513 rettv->vval.v_number = 0;
8514}
8515
8516/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008517 * "settagstack()" function
8518 */
8519 static void
8520f_settagstack(typval_T *argvars, typval_T *rettv)
8521{
8522 static char *e_invact2 = N_("E962: Invalid action: '%s'");
8523 win_T *wp;
8524 dict_T *d;
8525 int action = 'r';
8526
8527 rettv->vval.v_number = -1;
8528
8529 // first argument: window number or id
8530 wp = find_win_by_nr_or_id(&argvars[0]);
8531 if (wp == NULL)
8532 return;
8533
8534 // second argument: dict with items to set in the tag stack
8535 if (argvars[1].v_type != VAR_DICT)
8536 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008537 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008538 return;
8539 }
8540 d = argvars[1].vval.v_dict;
8541 if (d == NULL)
8542 return;
8543
8544 // third argument: action - 'a' for append and 'r' for replace.
8545 // default is to replace the stack.
8546 if (argvars[2].v_type == VAR_UNKNOWN)
8547 action = 'r';
8548 else if (argvars[2].v_type == VAR_STRING)
8549 {
8550 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008551 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008552 if (actstr == NULL)
8553 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01008554 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
8555 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008556 action = *actstr;
8557 else
8558 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008559 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008560 return;
8561 }
8562 }
8563 else
8564 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008565 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008566 return;
8567 }
8568
8569 if (set_tagstack(wp, d, action) == OK)
8570 rettv->vval.v_number = 0;
8571}
8572
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008573#ifdef FEAT_CRYPT
8574/*
8575 * "sha256({string})" function
8576 */
8577 static void
8578f_sha256(typval_T *argvars, typval_T *rettv)
8579{
8580 char_u *p;
8581
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008582 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008583 rettv->vval.v_string = vim_strsave(
8584 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
8585 rettv->v_type = VAR_STRING;
8586}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008587#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008588
8589/*
8590 * "shellescape({string})" function
8591 */
8592 static void
8593f_shellescape(typval_T *argvars, typval_T *rettv)
8594{
Bram Moolenaar20615522017-06-05 18:46:26 +02008595 int do_special = non_zero_arg(&argvars[1]);
8596
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008597 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008598 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008599 rettv->v_type = VAR_STRING;
8600}
8601
8602/*
8603 * shiftwidth() function
8604 */
8605 static void
8606f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
8607{
Bram Moolenaarf9514162018-11-22 03:08:29 +01008608 rettv->vval.v_number = 0;
8609
8610 if (argvars[0].v_type != VAR_UNKNOWN)
8611 {
8612 long col;
8613
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008614 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01008615 if (col < 0)
8616 return; // type error; errmsg already given
8617#ifdef FEAT_VARTABS
8618 rettv->vval.v_number = get_sw_value_col(curbuf, col);
8619 return;
8620#endif
8621 }
8622
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008623 rettv->vval.v_number = get_sw_value(curbuf);
8624}
8625
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008626#ifdef FEAT_FLOAT
8627/*
8628 * "sin()" function
8629 */
8630 static void
8631f_sin(typval_T *argvars, typval_T *rettv)
8632{
8633 float_T f = 0.0;
8634
8635 rettv->v_type = VAR_FLOAT;
8636 if (get_float_arg(argvars, &f) == OK)
8637 rettv->vval.v_float = sin(f);
8638 else
8639 rettv->vval.v_float = 0.0;
8640}
8641
8642/*
8643 * "sinh()" function
8644 */
8645 static void
8646f_sinh(typval_T *argvars, typval_T *rettv)
8647{
8648 float_T f = 0.0;
8649
8650 rettv->v_type = VAR_FLOAT;
8651 if (get_float_arg(argvars, &f) == OK)
8652 rettv->vval.v_float = sinh(f);
8653 else
8654 rettv->vval.v_float = 0.0;
8655}
8656#endif
8657
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008658/*
8659 * "soundfold({word})" function
8660 */
8661 static void
8662f_soundfold(typval_T *argvars, typval_T *rettv)
8663{
8664 char_u *s;
8665
8666 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008667 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008668#ifdef FEAT_SPELL
8669 rettv->vval.v_string = eval_soundfold(s);
8670#else
8671 rettv->vval.v_string = vim_strsave(s);
8672#endif
8673}
8674
8675/*
8676 * "spellbadword()" function
8677 */
8678 static void
8679f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
8680{
8681 char_u *word = (char_u *)"";
8682 hlf_T attr = HLF_COUNT;
8683 int len = 0;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008684#ifdef FEAT_SPELL
8685 int wo_spell_save = curwin->w_p_spell;
8686
8687 if (!curwin->w_p_spell)
8688 {
8689 did_set_spelllang(curwin);
8690 curwin->w_p_spell = TRUE;
8691 }
8692
8693 if (*curwin->w_s->b_p_spl == NUL)
8694 {
8695 emsg(_(e_no_spell));
8696 curwin->w_p_spell = wo_spell_save;
8697 return;
8698 }
8699#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008700
8701 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008702 {
8703#ifdef FEAT_SPELL
8704 curwin->w_p_spell = wo_spell_save;
8705#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008706 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008707 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008708
8709#ifdef FEAT_SPELL
8710 if (argvars[0].v_type == VAR_UNKNOWN)
8711 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008712 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008713 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
8714 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008715 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008716 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008717 curwin->w_set_curswant = TRUE;
8718 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008719 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008720 else if (*curbuf->b_s.b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008721 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008722 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008723 int capcol = -1;
8724
8725 if (str != NULL)
8726 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008727 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008728 while (*str != NUL)
8729 {
8730 len = spell_check(curwin, str, &attr, &capcol, FALSE);
8731 if (attr != HLF_COUNT)
8732 {
8733 word = str;
8734 break;
8735 }
8736 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02008737 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02008738 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008739 }
8740 }
8741 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008742 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008743#endif
8744
8745 list_append_string(rettv->vval.v_list, word, len);
8746 list_append_string(rettv->vval.v_list, (char_u *)(
8747 attr == HLF_SPB ? "bad" :
8748 attr == HLF_SPR ? "rare" :
8749 attr == HLF_SPL ? "local" :
8750 attr == HLF_SPC ? "caps" :
8751 ""), -1);
8752}
8753
8754/*
8755 * "spellsuggest()" function
8756 */
8757 static void
8758f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
8759{
8760#ifdef FEAT_SPELL
8761 char_u *str;
8762 int typeerr = FALSE;
8763 int maxcount;
8764 garray_T ga;
8765 int i;
8766 listitem_T *li;
8767 int need_capital = FALSE;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008768 int wo_spell_save = curwin->w_p_spell;
8769
8770 if (!curwin->w_p_spell)
8771 {
8772 did_set_spelllang(curwin);
8773 curwin->w_p_spell = TRUE;
8774 }
8775
8776 if (*curwin->w_s->b_p_spl == NUL)
8777 {
8778 emsg(_(e_no_spell));
8779 curwin->w_p_spell = wo_spell_save;
8780 return;
8781 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008782#endif
8783
8784 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008785 {
8786#ifdef FEAT_SPELL
8787 curwin->w_p_spell = wo_spell_save;
8788#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008789 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008790 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008791
8792#ifdef FEAT_SPELL
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008793 if (*curwin->w_s->b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008794 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008795 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008796 if (argvars[1].v_type != VAR_UNKNOWN)
8797 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008798 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008799 if (maxcount <= 0)
8800 return;
8801 if (argvars[2].v_type != VAR_UNKNOWN)
8802 {
Bram Moolenaar7c27f332020-09-05 22:45:55 +02008803 need_capital = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008804 if (typeerr)
8805 return;
8806 }
8807 }
8808 else
8809 maxcount = 25;
8810
8811 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
8812
8813 for (i = 0; i < ga.ga_len; ++i)
8814 {
8815 str = ((char_u **)ga.ga_data)[i];
8816
8817 li = listitem_alloc();
8818 if (li == NULL)
8819 vim_free(str);
8820 else
8821 {
8822 li->li_tv.v_type = VAR_STRING;
8823 li->li_tv.v_lock = 0;
8824 li->li_tv.vval.v_string = str;
8825 list_append(rettv->vval.v_list, li);
8826 }
8827 }
8828 ga_clear(&ga);
8829 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008830 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008831#endif
8832}
8833
8834 static void
8835f_split(typval_T *argvars, typval_T *rettv)
8836{
8837 char_u *str;
8838 char_u *end;
8839 char_u *pat = NULL;
8840 regmatch_T regmatch;
8841 char_u patbuf[NUMBUFLEN];
8842 char_u *save_cpo;
8843 int match;
8844 colnr_T col = 0;
8845 int keepempty = FALSE;
8846 int typeerr = FALSE;
8847
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008848 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008849 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008850 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008851
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008852 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008853 if (argvars[1].v_type != VAR_UNKNOWN)
8854 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008855 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008856 if (pat == NULL)
8857 typeerr = TRUE;
8858 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02008859 keepempty = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008860 }
8861 if (pat == NULL || *pat == NUL)
8862 pat = (char_u *)"[\\x01- ]\\+";
8863
8864 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008865 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008866 if (typeerr)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008867 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008868
8869 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
8870 if (regmatch.regprog != NULL)
8871 {
8872 regmatch.rm_ic = FALSE;
8873 while (*str != NUL || keepempty)
8874 {
8875 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008876 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008877 else
8878 match = vim_regexec_nl(&regmatch, str, col);
8879 if (match)
8880 end = regmatch.startp[0];
8881 else
8882 end = str + STRLEN(str);
8883 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
8884 && *str != NUL && match && end < regmatch.endp[0]))
8885 {
8886 if (list_append_string(rettv->vval.v_list, str,
8887 (int)(end - str)) == FAIL)
8888 break;
8889 }
8890 if (!match)
8891 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01008892 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008893 if (regmatch.endp[0] > str)
8894 col = 0;
8895 else
Bram Moolenaar13505972019-01-24 15:04:48 +01008896 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008897 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008898 str = regmatch.endp[0];
8899 }
8900
8901 vim_regfree(regmatch.regprog);
8902 }
8903
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008904theend:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008905 p_cpo = save_cpo;
8906}
8907
8908#ifdef FEAT_FLOAT
8909/*
8910 * "sqrt()" function
8911 */
8912 static void
8913f_sqrt(typval_T *argvars, typval_T *rettv)
8914{
8915 float_T f = 0.0;
8916
8917 rettv->v_type = VAR_FLOAT;
8918 if (get_float_arg(argvars, &f) == OK)
8919 rettv->vval.v_float = sqrt(f);
8920 else
8921 rettv->vval.v_float = 0.0;
8922}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008923#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008924
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008925#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01008926/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008927 * "str2float()" function
8928 */
8929 static void
8930f_str2float(typval_T *argvars, typval_T *rettv)
8931{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008932 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008933 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008934
Bram Moolenaar08243d22017-01-10 16:12:29 +01008935 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008936 p = skipwhite(p + 1);
8937 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01008938 if (isneg)
8939 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008940 rettv->v_type = VAR_FLOAT;
8941}
8942#endif
8943
8944/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02008945 * "str2list()" function
8946 */
8947 static void
8948f_str2list(typval_T *argvars, typval_T *rettv)
8949{
8950 char_u *p;
8951 int utf8 = FALSE;
8952
8953 if (rettv_list_alloc(rettv) == FAIL)
8954 return;
8955
8956 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaara48f7862020-09-05 20:16:57 +02008957 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar9d401282019-04-06 13:18:12 +02008958
8959 p = tv_get_string(&argvars[0]);
8960
8961 if (has_mbyte || utf8)
8962 {
8963 int (*ptr2len)(char_u *);
8964 int (*ptr2char)(char_u *);
8965
8966 if (utf8 || enc_utf8)
8967 {
8968 ptr2len = utf_ptr2len;
8969 ptr2char = utf_ptr2char;
8970 }
8971 else
8972 {
8973 ptr2len = mb_ptr2len;
8974 ptr2char = mb_ptr2char;
8975 }
8976
8977 for ( ; *p != NUL; p += (*ptr2len)(p))
8978 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
8979 }
8980 else
8981 for ( ; *p != NUL; ++p)
8982 list_append_number(rettv->vval.v_list, *p);
8983}
8984
8985/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008986 * "str2nr()" function
8987 */
8988 static void
8989f_str2nr(typval_T *argvars, typval_T *rettv)
8990{
8991 int base = 10;
8992 char_u *p;
8993 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008994 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01008995 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008996
8997 if (argvars[1].v_type != VAR_UNKNOWN)
8998 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008999 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009000 if (base != 2 && base != 8 && base != 10 && base != 16)
9001 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009002 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009003 return;
9004 }
Bram Moolenaar3986b942020-09-06 16:09:04 +02009005 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[2]))
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009006 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009007 }
9008
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009009 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01009010 isneg = (*p == '-');
9011 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009012 p = skipwhite(p + 1);
9013 switch (base)
9014 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009015 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02009016 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02009017 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009018 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02009019 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
9020 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01009021 if (isneg)
9022 rettv->vval.v_number = -n;
9023 else
9024 rettv->vval.v_number = n;
9025
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009026}
9027
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009028/*
9029 * "strgetchar()" function
9030 */
9031 static void
9032f_strgetchar(typval_T *argvars, typval_T *rettv)
9033{
9034 char_u *str;
9035 int len;
9036 int error = FALSE;
9037 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01009038 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009039
9040 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009041 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009042 if (str == NULL)
9043 return;
9044 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009045 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009046 if (error)
9047 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009048
Bram Moolenaar13505972019-01-24 15:04:48 +01009049 while (charidx >= 0 && byteidx < len)
9050 {
9051 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009052 {
Bram Moolenaar13505972019-01-24 15:04:48 +01009053 rettv->vval.v_number = mb_ptr2char(str + byteidx);
9054 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009055 }
Bram Moolenaar13505972019-01-24 15:04:48 +01009056 --charidx;
9057 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009058 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009059}
9060
9061/*
9062 * "stridx()" function
9063 */
9064 static void
9065f_stridx(typval_T *argvars, typval_T *rettv)
9066{
9067 char_u buf[NUMBUFLEN];
9068 char_u *needle;
9069 char_u *haystack;
9070 char_u *save_haystack;
9071 char_u *pos;
9072 int start_idx;
9073
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009074 needle = tv_get_string_chk(&argvars[1]);
9075 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009076 rettv->vval.v_number = -1;
9077 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009078 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009079
9080 if (argvars[2].v_type != VAR_UNKNOWN)
9081 {
9082 int error = FALSE;
9083
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009084 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009085 if (error || start_idx >= (int)STRLEN(haystack))
9086 return;
9087 if (start_idx >= 0)
9088 haystack += start_idx;
9089 }
9090
9091 pos = (char_u *)strstr((char *)haystack, (char *)needle);
9092 if (pos != NULL)
9093 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
9094}
9095
9096/*
9097 * "string()" function
9098 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01009099 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009100f_string(typval_T *argvars, typval_T *rettv)
9101{
9102 char_u *tofree;
9103 char_u numbuf[NUMBUFLEN];
9104
9105 rettv->v_type = VAR_STRING;
9106 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
9107 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009108 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009109 if (rettv->vval.v_string != NULL && tofree == NULL)
9110 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
9111}
9112
9113/*
9114 * "strlen()" function
9115 */
9116 static void
9117f_strlen(typval_T *argvars, typval_T *rettv)
9118{
9119 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009120 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009121}
9122
9123/*
9124 * "strchars()" function
9125 */
9126 static void
9127f_strchars(typval_T *argvars, typval_T *rettv)
9128{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009129 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar3986b942020-09-06 16:09:04 +02009130 int skipcc = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009131 varnumber_T len = 0;
9132 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009133
9134 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02009135 skipcc = (int)tv_get_bool(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009136 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02009137 semsg(_(e_using_number_as_bool_nr), skipcc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009138 else
9139 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009140 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
9141 while (*s != NUL)
9142 {
9143 func_mb_ptr2char_adv(&s);
9144 ++len;
9145 }
9146 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009147 }
9148}
9149
9150/*
9151 * "strdisplaywidth()" function
9152 */
9153 static void
9154f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
9155{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009156 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009157 int col = 0;
9158
9159 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009160 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009161
9162 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
9163}
9164
9165/*
9166 * "strwidth()" function
9167 */
9168 static void
9169f_strwidth(typval_T *argvars, typval_T *rettv)
9170{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009171 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009172
Bram Moolenaar13505972019-01-24 15:04:48 +01009173 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009174}
9175
9176/*
9177 * "strcharpart()" function
9178 */
9179 static void
9180f_strcharpart(typval_T *argvars, typval_T *rettv)
9181{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009182 char_u *p;
9183 int nchar;
9184 int nbyte = 0;
9185 int charlen;
9186 int len = 0;
9187 int slen;
9188 int error = FALSE;
9189
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009190 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009191 slen = (int)STRLEN(p);
9192
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009193 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009194 if (!error)
9195 {
9196 if (nchar > 0)
9197 while (nchar > 0 && nbyte < slen)
9198 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009199 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009200 --nchar;
9201 }
9202 else
9203 nbyte = nchar;
9204 if (argvars[2].v_type != VAR_UNKNOWN)
9205 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009206 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009207 while (charlen > 0 && nbyte + len < slen)
9208 {
9209 int off = nbyte + len;
9210
9211 if (off < 0)
9212 len += 1;
9213 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009214 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009215 --charlen;
9216 }
9217 }
9218 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009219 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009220 }
9221
9222 /*
9223 * Only return the overlap between the specified part and the actual
9224 * string.
9225 */
9226 if (nbyte < 0)
9227 {
9228 len += nbyte;
9229 nbyte = 0;
9230 }
9231 else if (nbyte > slen)
9232 nbyte = slen;
9233 if (len < 0)
9234 len = 0;
9235 else if (nbyte + len > slen)
9236 len = slen - nbyte;
9237
9238 rettv->v_type = VAR_STRING;
9239 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009240}
9241
9242/*
9243 * "strpart()" function
9244 */
9245 static void
9246f_strpart(typval_T *argvars, typval_T *rettv)
9247{
9248 char_u *p;
9249 int n;
9250 int len;
9251 int slen;
9252 int error = FALSE;
9253
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009254 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009255 slen = (int)STRLEN(p);
9256
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009257 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009258 if (error)
9259 len = 0;
9260 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009261 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009262 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009263 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009264
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009265 // Only return the overlap between the specified part and the actual
9266 // string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009267 if (n < 0)
9268 {
9269 len += n;
9270 n = 0;
9271 }
9272 else if (n > slen)
9273 n = slen;
9274 if (len < 0)
9275 len = 0;
9276 else if (n + len > slen)
9277 len = slen - n;
9278
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009279 if (argvars[2].v_type != VAR_UNKNOWN && argvars[3].v_type != VAR_UNKNOWN)
9280 {
9281 int off;
9282
9283 // length in characters
9284 for (off = n; off < slen && len > 0; --len)
9285 off += mb_ptr2len(p + off);
9286 len = off - n;
9287 }
9288
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009289 rettv->v_type = VAR_STRING;
9290 rettv->vval.v_string = vim_strnsave(p + n, len);
9291}
9292
9293/*
9294 * "strridx()" function
9295 */
9296 static void
9297f_strridx(typval_T *argvars, typval_T *rettv)
9298{
9299 char_u buf[NUMBUFLEN];
9300 char_u *needle;
9301 char_u *haystack;
9302 char_u *rest;
9303 char_u *lastmatch = NULL;
9304 int haystack_len, end_idx;
9305
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009306 needle = tv_get_string_chk(&argvars[1]);
9307 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009308
9309 rettv->vval.v_number = -1;
9310 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009311 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009312
9313 haystack_len = (int)STRLEN(haystack);
9314 if (argvars[2].v_type != VAR_UNKNOWN)
9315 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009316 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009317 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009318 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009319 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009320 }
9321 else
9322 end_idx = haystack_len;
9323
9324 if (*needle == NUL)
9325 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009326 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009327 lastmatch = haystack + end_idx;
9328 }
9329 else
9330 {
9331 for (rest = haystack; *rest != '\0'; ++rest)
9332 {
9333 rest = (char_u *)strstr((char *)rest, (char *)needle);
9334 if (rest == NULL || rest > haystack + end_idx)
9335 break;
9336 lastmatch = rest;
9337 }
9338 }
9339
9340 if (lastmatch == NULL)
9341 rettv->vval.v_number = -1;
9342 else
9343 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
9344}
9345
9346/*
9347 * "strtrans()" function
9348 */
9349 static void
9350f_strtrans(typval_T *argvars, typval_T *rettv)
9351{
9352 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009353 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009354}
9355
9356/*
9357 * "submatch()" function
9358 */
9359 static void
9360f_submatch(typval_T *argvars, typval_T *rettv)
9361{
9362 int error = FALSE;
9363 int no;
9364 int retList = 0;
9365
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009366 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009367 if (error)
9368 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009369 if (no < 0 || no >= NSUBEXP)
9370 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009371 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01009372 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009373 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009374 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaarad304702020-09-06 18:22:53 +02009375 retList = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009376 if (error)
9377 return;
9378
9379 if (retList == 0)
9380 {
9381 rettv->v_type = VAR_STRING;
9382 rettv->vval.v_string = reg_submatch(no);
9383 }
9384 else
9385 {
9386 rettv->v_type = VAR_LIST;
9387 rettv->vval.v_list = reg_submatch_list(no);
9388 }
9389}
9390
9391/*
9392 * "substitute()" function
9393 */
9394 static void
9395f_substitute(typval_T *argvars, typval_T *rettv)
9396{
9397 char_u patbuf[NUMBUFLEN];
9398 char_u subbuf[NUMBUFLEN];
9399 char_u flagsbuf[NUMBUFLEN];
9400
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009401 char_u *str = tv_get_string_chk(&argvars[0]);
9402 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009403 char_u *sub = NULL;
9404 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009405 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009406
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009407 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
9408 expr = &argvars[2];
9409 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009410 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009411
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009412 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009413 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
9414 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009415 rettv->vval.v_string = NULL;
9416 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009417 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009418}
9419
9420/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009421 * "swapinfo(swap_filename)" function
9422 */
9423 static void
9424f_swapinfo(typval_T *argvars, typval_T *rettv)
9425{
9426 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009427 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009428}
9429
9430/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02009431 * "swapname(expr)" function
9432 */
9433 static void
9434f_swapname(typval_T *argvars, typval_T *rettv)
9435{
9436 buf_T *buf;
9437
9438 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01009439 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02009440 if (buf == NULL || buf->b_ml.ml_mfp == NULL
9441 || buf->b_ml.ml_mfp->mf_fname == NULL)
9442 rettv->vval.v_string = NULL;
9443 else
9444 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
9445}
9446
9447/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009448 * "synID(lnum, col, trans)" function
9449 */
9450 static void
9451f_synID(typval_T *argvars UNUSED, typval_T *rettv)
9452{
9453 int id = 0;
9454#ifdef FEAT_SYN_HL
9455 linenr_T lnum;
9456 colnr_T col;
9457 int trans;
9458 int transerr = FALSE;
9459
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009460 lnum = tv_get_lnum(argvars); // -1 on type error
9461 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaarfcb6d702020-09-05 21:41:56 +02009462 trans = (int)tv_get_bool_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009463
9464 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9465 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
9466 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
9467#endif
9468
9469 rettv->vval.v_number = id;
9470}
9471
9472/*
9473 * "synIDattr(id, what [, mode])" function
9474 */
9475 static void
9476f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
9477{
9478 char_u *p = NULL;
9479#ifdef FEAT_SYN_HL
9480 int id;
9481 char_u *what;
9482 char_u *mode;
9483 char_u modebuf[NUMBUFLEN];
9484 int modec;
9485
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009486 id = (int)tv_get_number(&argvars[0]);
9487 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009488 if (argvars[2].v_type != VAR_UNKNOWN)
9489 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009490 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009491 modec = TOLOWER_ASC(mode[0]);
9492 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009493 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009494 }
9495 else
9496 {
9497#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
9498 if (USE_24BIT)
9499 modec = 'g';
9500 else
9501#endif
9502 if (t_colors > 1)
9503 modec = 'c';
9504 else
9505 modec = 't';
9506 }
9507
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009508 switch (TOLOWER_ASC(what[0]))
9509 {
9510 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009511 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009512 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009513 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009514 p = highlight_has_attr(id, HL_BOLD, modec);
9515 break;
9516
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009517 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009518 p = highlight_color(id, what, modec);
9519 break;
9520
9521 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009522 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009523 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009524 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009525 p = highlight_has_attr(id, HL_ITALIC, modec);
9526 break;
9527
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009528 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02009529 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009530 break;
9531
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009532 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009533 p = highlight_has_attr(id, HL_INVERSE, modec);
9534 break;
9535
9536 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009537 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009538 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009539 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02009540 else if (TOLOWER_ASC(what[1]) == 't' &&
9541 TOLOWER_ASC(what[2]) == 'r')
9542 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009543 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009544 p = highlight_has_attr(id, HL_STANDOUT, modec);
9545 break;
9546
9547 case 'u':
Bram Moolenaar391c3622020-09-29 20:59:17 +02009548 if (TOLOWER_ASC(what[1]) == 'l') // ul
9549 p = highlight_color(id, what, modec);
9550 else if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009551 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009552 p = highlight_has_attr(id, HL_UNDERLINE, modec);
9553 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009554 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009555 p = highlight_has_attr(id, HL_UNDERCURL, modec);
9556 break;
9557 }
9558
9559 if (p != NULL)
9560 p = vim_strsave(p);
9561#endif
9562 rettv->v_type = VAR_STRING;
9563 rettv->vval.v_string = p;
9564}
9565
9566/*
9567 * "synIDtrans(id)" function
9568 */
9569 static void
9570f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
9571{
9572 int id;
9573
9574#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009575 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009576
9577 if (id > 0)
9578 id = syn_get_final_id(id);
9579 else
9580#endif
9581 id = 0;
9582
9583 rettv->vval.v_number = id;
9584}
9585
9586/*
9587 * "synconcealed(lnum, col)" function
9588 */
9589 static void
9590f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
9591{
9592#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
9593 linenr_T lnum;
9594 colnr_T col;
9595 int syntax_flags = 0;
9596 int cchar;
9597 int matchid = 0;
9598 char_u str[NUMBUFLEN];
9599#endif
9600
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009601 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009602
9603#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009604 lnum = tv_get_lnum(argvars); // -1 on type error
9605 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009606
Bram Moolenaara80faa82020-04-12 19:37:17 +02009607 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009608
9609 if (rettv_list_alloc(rettv) != FAIL)
9610 {
9611 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9612 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9613 && curwin->w_p_cole > 0)
9614 {
9615 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
9616 syntax_flags = get_syntax_info(&matchid);
9617
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009618 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009619 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
9620 {
9621 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02009622 if (cchar == NUL && curwin->w_p_cole == 1)
9623 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009624 if (cchar != NUL)
9625 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009626 if (has_mbyte)
9627 (*mb_char2bytes)(cchar, str);
9628 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009629 str[0] = cchar;
9630 }
9631 }
9632 }
9633
9634 list_append_number(rettv->vval.v_list,
9635 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009636 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009637 list_append_string(rettv->vval.v_list, str, -1);
9638 list_append_number(rettv->vval.v_list, matchid);
9639 }
9640#endif
9641}
9642
9643/*
9644 * "synstack(lnum, col)" function
9645 */
9646 static void
9647f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
9648{
9649#ifdef FEAT_SYN_HL
9650 linenr_T lnum;
9651 colnr_T col;
9652 int i;
9653 int id;
9654#endif
9655
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009656 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009657
9658#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009659 lnum = tv_get_lnum(argvars); // -1 on type error
9660 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009661
9662 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9663 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9664 && rettv_list_alloc(rettv) != FAIL)
9665 {
9666 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
9667 for (i = 0; ; ++i)
9668 {
9669 id = syn_get_stack_item(i);
9670 if (id < 0)
9671 break;
9672 if (list_append_number(rettv->vval.v_list, id) == FAIL)
9673 break;
9674 }
9675 }
9676#endif
9677}
9678
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009679/*
9680 * "tabpagebuflist()" function
9681 */
9682 static void
9683f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9684{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009685 tabpage_T *tp;
9686 win_T *wp = NULL;
9687
9688 if (argvars[0].v_type == VAR_UNKNOWN)
9689 wp = firstwin;
9690 else
9691 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009692 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009693 if (tp != NULL)
9694 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
9695 }
9696 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
9697 {
9698 for (; wp != NULL; wp = wp->w_next)
9699 if (list_append_number(rettv->vval.v_list,
9700 wp->w_buffer->b_fnum) == FAIL)
9701 break;
9702 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009703}
9704
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009705/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009706 * "tagfiles()" function
9707 */
9708 static void
9709f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
9710{
9711 char_u *fname;
9712 tagname_T tn;
9713 int first;
9714
9715 if (rettv_list_alloc(rettv) == FAIL)
9716 return;
9717 fname = alloc(MAXPATHL);
9718 if (fname == NULL)
9719 return;
9720
9721 for (first = TRUE; ; first = FALSE)
9722 if (get_tagfname(&tn, first, fname) == FAIL
9723 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
9724 break;
9725 tagname_free(&tn);
9726 vim_free(fname);
9727}
9728
9729/*
9730 * "taglist()" function
9731 */
9732 static void
9733f_taglist(typval_T *argvars, typval_T *rettv)
9734{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009735 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009736 char_u *tag_pattern;
9737
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009738 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009739
9740 rettv->vval.v_number = FALSE;
9741 if (*tag_pattern == NUL)
9742 return;
9743
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009744 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009745 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009746 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009747 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009748}
9749
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009750#ifdef FEAT_FLOAT
9751/*
9752 * "tan()" function
9753 */
9754 static void
9755f_tan(typval_T *argvars, typval_T *rettv)
9756{
9757 float_T f = 0.0;
9758
9759 rettv->v_type = VAR_FLOAT;
9760 if (get_float_arg(argvars, &f) == OK)
9761 rettv->vval.v_float = tan(f);
9762 else
9763 rettv->vval.v_float = 0.0;
9764}
9765
9766/*
9767 * "tanh()" function
9768 */
9769 static void
9770f_tanh(typval_T *argvars, typval_T *rettv)
9771{
9772 float_T f = 0.0;
9773
9774 rettv->v_type = VAR_FLOAT;
9775 if (get_float_arg(argvars, &f) == OK)
9776 rettv->vval.v_float = tanh(f);
9777 else
9778 rettv->vval.v_float = 0.0;
9779}
9780#endif
9781
9782/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009783 * "tolower(string)" function
9784 */
9785 static void
9786f_tolower(typval_T *argvars, typval_T *rettv)
9787{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009788 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009789 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009790}
9791
9792/*
9793 * "toupper(string)" function
9794 */
9795 static void
9796f_toupper(typval_T *argvars, typval_T *rettv)
9797{
9798 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009799 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009800}
9801
9802/*
9803 * "tr(string, fromstr, tostr)" function
9804 */
9805 static void
9806f_tr(typval_T *argvars, typval_T *rettv)
9807{
9808 char_u *in_str;
9809 char_u *fromstr;
9810 char_u *tostr;
9811 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009812 int inlen;
9813 int fromlen;
9814 int tolen;
9815 int idx;
9816 char_u *cpstr;
9817 int cplen;
9818 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009819 char_u buf[NUMBUFLEN];
9820 char_u buf2[NUMBUFLEN];
9821 garray_T ga;
9822
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009823 in_str = tv_get_string(&argvars[0]);
9824 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
9825 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009826
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009827 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009828 rettv->v_type = VAR_STRING;
9829 rettv->vval.v_string = NULL;
9830 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009831 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009832 ga_init2(&ga, (int)sizeof(char), 80);
9833
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009834 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009835 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009836 if (STRLEN(fromstr) != STRLEN(tostr))
9837 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009838error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009839 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009840 ga_clear(&ga);
9841 return;
9842 }
9843
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009844 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009845 while (*in_str != NUL)
9846 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009847 if (has_mbyte)
9848 {
9849 inlen = (*mb_ptr2len)(in_str);
9850 cpstr = in_str;
9851 cplen = inlen;
9852 idx = 0;
9853 for (p = fromstr; *p != NUL; p += fromlen)
9854 {
9855 fromlen = (*mb_ptr2len)(p);
9856 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
9857 {
9858 for (p = tostr; *p != NUL; p += tolen)
9859 {
9860 tolen = (*mb_ptr2len)(p);
9861 if (idx-- == 0)
9862 {
9863 cplen = tolen;
9864 cpstr = p;
9865 break;
9866 }
9867 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009868 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009869 goto error;
9870 break;
9871 }
9872 ++idx;
9873 }
9874
9875 if (first && cpstr == in_str)
9876 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009877 // Check that fromstr and tostr have the same number of
9878 // (multi-byte) characters. Done only once when a character
9879 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009880 first = FALSE;
9881 for (p = tostr; *p != NUL; p += tolen)
9882 {
9883 tolen = (*mb_ptr2len)(p);
9884 --idx;
9885 }
9886 if (idx != 0)
9887 goto error;
9888 }
9889
9890 (void)ga_grow(&ga, cplen);
9891 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
9892 ga.ga_len += cplen;
9893
9894 in_str += inlen;
9895 }
9896 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009897 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009898 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009899 p = vim_strchr(fromstr, *in_str);
9900 if (p != NULL)
9901 ga_append(&ga, tostr[p - fromstr]);
9902 else
9903 ga_append(&ga, *in_str);
9904 ++in_str;
9905 }
9906 }
9907
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009908 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009909 (void)ga_grow(&ga, 1);
9910 ga_append(&ga, NUL);
9911
9912 rettv->vval.v_string = ga.ga_data;
9913}
9914
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009915/*
9916 * "trim({expr})" function
9917 */
9918 static void
9919f_trim(typval_T *argvars, typval_T *rettv)
9920{
9921 char_u buf1[NUMBUFLEN];
9922 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009923 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009924 char_u *mask = NULL;
9925 char_u *tail;
9926 char_u *prev;
9927 char_u *p;
9928 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009929 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009930
9931 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009932 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009933 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009934 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009935
9936 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009937 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009938 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009939
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009940 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009941 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009942 int error = 0;
9943
9944 // leading or trailing characters to trim
9945 dir = (int)tv_get_number_chk(&argvars[2], &error);
9946 if (error)
9947 return;
9948 if (dir < 0 || dir > 2)
9949 {
9950 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
9951 return;
9952 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009953 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009954 }
9955
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009956 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009957 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009958 // Trim leading characters
9959 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009960 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009961 c1 = PTR2CHAR(head);
9962 if (mask == NULL)
9963 {
9964 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009965 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009966 }
9967 else
9968 {
9969 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9970 if (c1 == PTR2CHAR(p))
9971 break;
9972 if (*p == NUL)
9973 break;
9974 }
9975 MB_PTR_ADV(head);
9976 }
9977 }
9978
9979 tail = head + STRLEN(head);
9980 if (dir == 0 || dir == 2)
9981 {
9982 // Trim trailing characters
9983 for (; tail > head; tail = prev)
9984 {
9985 prev = tail;
9986 MB_PTR_BACK(head, prev);
9987 c1 = PTR2CHAR(prev);
9988 if (mask == NULL)
9989 {
9990 if (c1 > ' ' && c1 != 0xa0)
9991 break;
9992 }
9993 else
9994 {
9995 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9996 if (c1 == PTR2CHAR(p))
9997 break;
9998 if (*p == NUL)
9999 break;
10000 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +010010001 }
10002 }
Bram Moolenaardf44a272020-06-07 20:49:05 +020010003 rettv->vval.v_string = vim_strnsave(head, tail - head);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +010010004}
10005
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010006#ifdef FEAT_FLOAT
10007/*
10008 * "trunc({float})" function
10009 */
10010 static void
10011f_trunc(typval_T *argvars, typval_T *rettv)
10012{
10013 float_T f = 0.0;
10014
10015 rettv->v_type = VAR_FLOAT;
10016 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010017 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010018 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
10019 else
10020 rettv->vval.v_float = 0.0;
10021}
10022#endif
10023
10024/*
10025 * "type(expr)" function
10026 */
10027 static void
10028f_type(typval_T *argvars, typval_T *rettv)
10029{
10030 int n = -1;
10031
10032 switch (argvars[0].v_type)
10033 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +010010034 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
10035 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010036 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +010010037 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
10038 case VAR_LIST: n = VAR_TYPE_LIST; break;
10039 case VAR_DICT: n = VAR_TYPE_DICT; break;
10040 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
10041 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
10042 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +020010043 case VAR_JOB: n = VAR_TYPE_JOB; break;
10044 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +010010045 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010046 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +020010047 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010010048 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +010010049 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010050 n = -1;
10051 break;
10052 }
10053 rettv->vval.v_number = n;
10054}
10055
10056/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010057 * "virtcol(string)" function
10058 */
10059 static void
10060f_virtcol(typval_T *argvars, typval_T *rettv)
10061{
10062 colnr_T vcol = 0;
10063 pos_T *fp;
10064 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +010010065 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010066
Bram Moolenaar6f02b002021-01-10 20:22:54 +010010067 fp = var2fpos(&argvars[0], FALSE, &fnum, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010068 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
10069 && fnum == curbuf->b_fnum)
10070 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +010010071 // Limit the column to a valid value, getvvcol() doesn't check.
10072 if (fp->col < 0)
10073 fp->col = 0;
10074 else
10075 {
10076 len = (int)STRLEN(ml_get(fp->lnum));
10077 if (fp->col > len)
10078 fp->col = len;
10079 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010080 getvvcol(curwin, fp, NULL, NULL, &vcol);
10081 ++vcol;
10082 }
10083
10084 rettv->vval.v_number = vcol;
10085}
10086
10087/*
10088 * "visualmode()" function
10089 */
10090 static void
10091f_visualmode(typval_T *argvars, typval_T *rettv)
10092{
10093 char_u str[2];
10094
10095 rettv->v_type = VAR_STRING;
10096 str[0] = curbuf->b_visual_mode_eval;
10097 str[1] = NUL;
10098 rettv->vval.v_string = vim_strsave(str);
10099
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010100 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010101 if (non_zero_arg(&argvars[0]))
10102 curbuf->b_visual_mode_eval = NUL;
10103}
10104
10105/*
10106 * "wildmenumode()" function
10107 */
10108 static void
10109f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
10110{
10111#ifdef FEAT_WILDMENU
10112 if (wild_menu_showing)
10113 rettv->vval.v_number = 1;
10114#endif
10115}
10116
10117/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +010010118 * "windowsversion()" function
10119 */
10120 static void
10121f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
10122{
10123 rettv->v_type = VAR_STRING;
10124 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
10125}
10126
10127/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010128 * "wordcount()" function
10129 */
10130 static void
10131f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
10132{
10133 if (rettv_dict_alloc(rettv) == FAIL)
10134 return;
10135 cursor_pos_info(rettv->vval.v_dict);
10136}
10137
10138/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010139 * "xor(expr, expr)" function
10140 */
10141 static void
10142f_xor(typval_T *argvars, typval_T *rettv)
10143{
Bram Moolenaard155d7a2018-12-21 16:04:21 +010010144 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
10145 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010146}
10147
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010148#endif // FEAT_EVAL