blob: ed9f06fed436e93b5cb79cf29c5669f1bb30decb [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * evalfunc.c: Builtin functions
12 */
13#define USING_FLOAT_STUFF
14
15#include "vim.h"
16
17#if defined(FEAT_EVAL) || defined(PROTO)
18
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020019#ifdef VMS
20# include <float.h>
21#endif
22
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020023#ifdef FEAT_FLOAT
24static void f_abs(typval_T *argvars, typval_T *rettv);
25static void f_acos(typval_T *argvars, typval_T *rettv);
26#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020027static void f_and(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020028#ifdef FEAT_FLOAT
29static void f_asin(typval_T *argvars, typval_T *rettv);
30static void f_atan(typval_T *argvars, typval_T *rettv);
31static void f_atan2(typval_T *argvars, typval_T *rettv);
32#endif
Bram Moolenaar59716a22017-03-01 20:32:44 +010033#ifdef FEAT_BEVAL
Bram Moolenaarbe0a2592019-05-09 13:50:16 +020034static void f_balloon_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar59716a22017-03-01 20:32:44 +010035static void f_balloon_show(typval_T *argvars, typval_T *rettv);
Bram Moolenaar669a8282017-11-19 20:13:05 +010036# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +010037static void f_balloon_split(typval_T *argvars, typval_T *rettv);
Bram Moolenaar669a8282017-11-19 20:13:05 +010038# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +010039#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020040static void f_byte2line(typval_T *argvars, typval_T *rettv);
41static void byteidx(typval_T *argvars, typval_T *rettv, int comp);
42static void f_byteidx(typval_T *argvars, typval_T *rettv);
43static void f_byteidxcomp(typval_T *argvars, typval_T *rettv);
44static void f_call(typval_T *argvars, typval_T *rettv);
45#ifdef FEAT_FLOAT
46static void f_ceil(typval_T *argvars, typval_T *rettv);
47#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020048static void f_changenr(typval_T *argvars, typval_T *rettv);
49static void f_char2nr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar17793ef2020-12-28 12:56:58 +010050static void f_charidx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020051static void f_col(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020052static void f_confirm(typval_T *argvars, typval_T *rettv);
53static void f_copy(typval_T *argvars, typval_T *rettv);
54#ifdef FEAT_FLOAT
55static void f_cos(typval_T *argvars, typval_T *rettv);
56static void f_cosh(typval_T *argvars, typval_T *rettv);
57#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020058static void f_cursor(typval_T *argsvars, typval_T *rettv);
Bram Moolenaar4f974752019-02-17 17:44:42 +010059#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +020060static void f_debugbreak(typval_T *argvars, typval_T *rettv);
61#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020062static void f_deepcopy(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020063static void f_did_filetype(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4132eb52020-02-14 16:53:00 +010064static void f_echoraw(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020065static void f_empty(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020066static void f_environ(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020067static void f_escape(typval_T *argvars, typval_T *rettv);
68static void f_eval(typval_T *argvars, typval_T *rettv);
69static void f_eventhandler(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020070static void f_execute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020071static void f_exists(typval_T *argvars, typval_T *rettv);
72#ifdef FEAT_FLOAT
73static void f_exp(typval_T *argvars, typval_T *rettv);
74#endif
75static void f_expand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar80dad482019-06-09 17:22:31 +020076static void f_expandcmd(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020077static void f_feedkeys(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020078#ifdef FEAT_FLOAT
79static void f_float2nr(typval_T *argvars, typval_T *rettv);
80static void f_floor(typval_T *argvars, typval_T *rettv);
81static void f_fmod(typval_T *argvars, typval_T *rettv);
82#endif
83static void f_fnameescape(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020084static void f_foreground(typval_T *argvars, typval_T *rettv);
Bram Moolenaar437bafe2016-08-01 15:40:54 +020085static void f_funcref(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020086static void f_function(typval_T *argvars, typval_T *rettv);
87static void f_garbagecollect(typval_T *argvars, typval_T *rettv);
88static void f_get(typval_T *argvars, typval_T *rettv);
Bram Moolenaar07ad8162018-02-13 13:59:59 +010089static void f_getchangelist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020090static void f_getcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020091static void f_getenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020092static void f_getfontname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4f505882018-02-10 21:06:32 +010093static void f_getjumplist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020094static void f_getpid(typval_T *argvars, typval_T *rettv);
95static void f_getcurpos(typval_T *argvars, typval_T *rettv);
96static void f_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020097static void f_getreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarbb861e22020-06-07 18:16:36 +020098static void f_getreginfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020099static void f_getregtype(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100100static void f_gettagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +0200101static void f_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200102static void f_haslocaldir(typval_T *argvars, typval_T *rettv);
103static void f_hasmapto(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200104static void f_hlID(typval_T *argvars, typval_T *rettv);
105static void f_hlexists(typval_T *argvars, typval_T *rettv);
106static void f_hostname(typval_T *argvars, typval_T *rettv);
107static void f_iconv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200108static void f_index(typval_T *argvars, typval_T *rettv);
109static void f_input(typval_T *argvars, typval_T *rettv);
110static void f_inputdialog(typval_T *argvars, typval_T *rettv);
111static void f_inputlist(typval_T *argvars, typval_T *rettv);
112static void f_inputrestore(typval_T *argvars, typval_T *rettv);
113static void f_inputsave(typval_T *argvars, typval_T *rettv);
114static void f_inputsecret(typval_T *argvars, typval_T *rettv);
Bram Moolenaar67a2deb2019-11-25 00:05:32 +0100115static void f_interrupt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200116static void f_invert(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200117static void f_islocked(typval_T *argvars, typval_T *rettv);
118#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
Bram Moolenaarfda1bff2019-04-04 13:44:37 +0200119static void f_isinf(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200120static void f_isnan(typval_T *argvars, typval_T *rettv);
121#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200122static void f_last_buffer_nr(typval_T *argvars, typval_T *rettv);
123static void f_len(typval_T *argvars, typval_T *rettv);
124static void f_libcall(typval_T *argvars, typval_T *rettv);
125static void f_libcallnr(typval_T *argvars, typval_T *rettv);
126static void f_line(typval_T *argvars, typval_T *rettv);
127static void f_line2byte(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200128#ifdef FEAT_FLOAT
129static void f_log(typval_T *argvars, typval_T *rettv);
130static void f_log10(typval_T *argvars, typval_T *rettv);
131#endif
132#ifdef FEAT_LUA
133static void f_luaeval(typval_T *argvars, typval_T *rettv);
134#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200135static void f_maparg(typval_T *argvars, typval_T *rettv);
136static void f_mapcheck(typval_T *argvars, typval_T *rettv);
137static void f_match(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200138static void f_matchend(typval_T *argvars, typval_T *rettv);
139static void f_matchlist(typval_T *argvars, typval_T *rettv);
140static void f_matchstr(typval_T *argvars, typval_T *rettv);
141static void f_matchstrpos(typval_T *argvars, typval_T *rettv);
142static void f_max(typval_T *argvars, typval_T *rettv);
143static void f_min(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200144#ifdef FEAT_MZSCHEME
145static void f_mzeval(typval_T *argvars, typval_T *rettv);
146#endif
147static void f_nextnonblank(typval_T *argvars, typval_T *rettv);
148static void f_nr2char(typval_T *argvars, typval_T *rettv);
149static void f_or(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200150#ifdef FEAT_PERL
151static void f_perleval(typval_T *argvars, typval_T *rettv);
152#endif
153#ifdef FEAT_FLOAT
154static void f_pow(typval_T *argvars, typval_T *rettv);
155#endif
156static void f_prevnonblank(typval_T *argvars, typval_T *rettv);
157static void f_printf(typval_T *argvars, typval_T *rettv);
Bram Moolenaare9bd5722019-08-17 19:36:06 +0200158static void f_pum_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200159static void f_pumvisible(typval_T *argvars, typval_T *rettv);
160#ifdef FEAT_PYTHON3
161static void f_py3eval(typval_T *argvars, typval_T *rettv);
162#endif
163#ifdef FEAT_PYTHON
164static void f_pyeval(typval_T *argvars, typval_T *rettv);
165#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100166#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
167static void f_pyxeval(typval_T *argvars, typval_T *rettv);
168#endif
Bram Moolenaar4f645c52020-02-08 16:40:39 +0100169static void f_test_srand_seed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100170static void f_rand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200171static void f_range(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b6d9112018-05-22 20:35:17 +0200172static void f_reg_executing(typval_T *argvars, typval_T *rettv);
173static void f_reg_recording(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200174static void f_rename(typval_T *argvars, typval_T *rettv);
175static void f_repeat(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200176#ifdef FEAT_FLOAT
177static void f_round(typval_T *argvars, typval_T *rettv);
178#endif
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100179#ifdef FEAT_RUBY
180static void f_rubyeval(typval_T *argvars, typval_T *rettv);
181#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200182static void f_screenattr(typval_T *argvars, typval_T *rettv);
183static void f_screenchar(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100184static void f_screenchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200185static void f_screencol(typval_T *argvars, typval_T *rettv);
186static void f_screenrow(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100187static void f_screenstring(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200188static void f_search(typval_T *argvars, typval_T *rettv);
189static void f_searchdecl(typval_T *argvars, typval_T *rettv);
190static void f_searchpair(typval_T *argvars, typval_T *rettv);
191static void f_searchpairpos(typval_T *argvars, typval_T *rettv);
192static void f_searchpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200193static void f_setcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +0200194static void f_setenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200195static void f_setfperm(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200196static void f_setpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200197static void f_setreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100198static void f_settagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200199#ifdef FEAT_CRYPT
200static void f_sha256(typval_T *argvars, typval_T *rettv);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200201#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200202static void f_shellescape(typval_T *argvars, typval_T *rettv);
203static void f_shiftwidth(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200204#ifdef FEAT_FLOAT
205static void f_sin(typval_T *argvars, typval_T *rettv);
206static void f_sinh(typval_T *argvars, typval_T *rettv);
207#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200208static void f_soundfold(typval_T *argvars, typval_T *rettv);
209static void f_spellbadword(typval_T *argvars, typval_T *rettv);
210static void f_spellsuggest(typval_T *argvars, typval_T *rettv);
211static void f_split(typval_T *argvars, typval_T *rettv);
212#ifdef FEAT_FLOAT
213static void f_sqrt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100214#endif
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100215static void f_srand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100216#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200217static void f_str2float(typval_T *argvars, typval_T *rettv);
218#endif
Bram Moolenaar9d401282019-04-06 13:18:12 +0200219static void f_str2list(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200220static void f_str2nr(typval_T *argvars, typval_T *rettv);
221static void f_strchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200222static void f_strgetchar(typval_T *argvars, typval_T *rettv);
223static void f_stridx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200224static void f_strlen(typval_T *argvars, typval_T *rettv);
225static void f_strcharpart(typval_T *argvars, typval_T *rettv);
226static void f_strpart(typval_T *argvars, typval_T *rettv);
227static void f_strridx(typval_T *argvars, typval_T *rettv);
228static void f_strtrans(typval_T *argvars, typval_T *rettv);
229static void f_strdisplaywidth(typval_T *argvars, typval_T *rettv);
230static void f_strwidth(typval_T *argvars, typval_T *rettv);
231static void f_submatch(typval_T *argvars, typval_T *rettv);
232static void f_substitute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar00f123a2018-08-21 20:28:54 +0200233static void f_swapinfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar110bd602018-09-16 18:46:59 +0200234static void f_swapname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200235static void f_synID(typval_T *argvars, typval_T *rettv);
236static void f_synIDattr(typval_T *argvars, typval_T *rettv);
237static void f_synIDtrans(typval_T *argvars, typval_T *rettv);
238static void f_synstack(typval_T *argvars, typval_T *rettv);
239static void f_synconcealed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200240static void f_tabpagebuflist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200241static void f_taglist(typval_T *argvars, typval_T *rettv);
242static void f_tagfiles(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200243#ifdef FEAT_FLOAT
244static void f_tan(typval_T *argvars, typval_T *rettv);
245static void f_tanh(typval_T *argvars, typval_T *rettv);
246#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200247static void f_tolower(typval_T *argvars, typval_T *rettv);
248static void f_toupper(typval_T *argvars, typval_T *rettv);
249static void f_tr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +0100250static void f_trim(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200251#ifdef FEAT_FLOAT
252static void f_trunc(typval_T *argvars, typval_T *rettv);
253#endif
254static void f_type(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200255static void f_virtcol(typval_T *argvars, typval_T *rettv);
256static void f_visualmode(typval_T *argvars, typval_T *rettv);
257static void f_wildmenumode(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0c1e3742019-12-27 13:49:24 +0100258static void f_windowsversion(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200259static void f_wordcount(typval_T *argvars, typval_T *rettv);
260static void f_xor(typval_T *argvars, typval_T *rettv);
261
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100262
Bram Moolenaar94738d82020-10-21 14:25:07 +0200263/*
264 * Functions that check the argument type of a builtin function.
265 * Each function returns FAIL and gives an error message if the type is wrong.
266 */
267
268// Context passed to an arg_ function.
269typedef struct {
Bram Moolenaarca174532020-10-21 16:42:22 +0200270 int arg_count; // actual argument count
271 type_T **arg_types; // list of argument types
272 int arg_idx; // current argument index (first arg is zero)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200273} argcontext_T;
274
275// A function to check one argument type. The first argument is the type to
276// check. If needed, other argument types can be obtained with the context.
277// E.g. if "arg_idx" is 1, then (type - 1) is the first argument type.
278typedef int (*argcheck_T)(type_T *, argcontext_T *);
279
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100280/*
281 * Check "type" is a float or a number.
282 */
Bram Moolenaar94738d82020-10-21 14:25:07 +0200283 static int
284arg_float_or_nr(type_T *type, argcontext_T *context)
285{
Bram Moolenaarca174532020-10-21 16:42:22 +0200286 if (type->tt_type == VAR_ANY
287 || type->tt_type == VAR_FLOAT || type->tt_type == VAR_NUMBER)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200288 return OK;
289 arg_type_mismatch(&t_number, type, context->arg_idx + 1);
290 return FAIL;
291}
292
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100293/*
294 * Check "type" is a number.
295 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200296 static int
297arg_number(type_T *type, argcontext_T *context)
298{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100299 return check_arg_type(&t_number, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200300}
301
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100302/*
303 * Check "type" is a string.
304 */
305 static int
306arg_string(type_T *type, argcontext_T *context)
307{
Bram Moolenaar193f6202020-11-16 20:08:35 +0100308 return check_arg_type(&t_string, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100309}
310
311/*
312 * Check "type" is a list or a blob.
313 */
Bram Moolenaarca174532020-10-21 16:42:22 +0200314 static int
315arg_list_or_blob(type_T *type, argcontext_T *context)
316{
317 if (type->tt_type == VAR_ANY
318 || type->tt_type == VAR_LIST || type->tt_type == VAR_BLOB)
319 return OK;
320 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
321 return FAIL;
322}
323
324/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100325 * Check "type" is a list or a dict.
326 */
327 static int
328arg_list_or_dict(type_T *type, argcontext_T *context)
329{
330 if (type->tt_type == VAR_ANY
331 || type->tt_type == VAR_LIST || type->tt_type == VAR_DICT)
332 return OK;
333 arg_type_mismatch(&t_list_any, type, context->arg_idx + 1);
334 return FAIL;
335}
336
337/*
338 * Check "type" is the same type as the previous argument
339 * Must not be used for the first argcheck_T entry.
340 */
341 static int
342arg_same_as_prev(type_T *type, argcontext_T *context)
343{
344 type_T *prev_type = context->arg_types[context->arg_idx - 1];
345
Bram Moolenaar193f6202020-11-16 20:08:35 +0100346 return check_arg_type(prev_type, type, context->arg_idx + 1);
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100347}
348
349/*
350 * Check "type" is an item of the list or blob of the previous arg.
Bram Moolenaarca174532020-10-21 16:42:22 +0200351 * Must not be used for the first argcheck_T entry.
352 */
353 static int
354arg_item_of_prev(type_T *type, argcontext_T *context)
355{
356 type_T *prev_type = context->arg_types[context->arg_idx - 1];
357 type_T *expected;
358
359 if (prev_type->tt_type == VAR_LIST)
360 expected = prev_type->tt_member;
361 else if (prev_type->tt_type == VAR_BLOB)
362 expected = &t_number;
363 else
364 // probably VAR_ANY, can't check
365 return OK;
366
Bram Moolenaar193f6202020-11-16 20:08:35 +0100367 return check_arg_type(expected, type, context->arg_idx + 1);
Bram Moolenaarca174532020-10-21 16:42:22 +0200368}
369
Bram Moolenaar94738d82020-10-21 14:25:07 +0200370/*
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100371 * Check "type" which is the third argument of extend().
372 */
373 static int
374arg_extend3(type_T *type, argcontext_T *context)
375{
376 type_T *first_type = context->arg_types[context->arg_idx - 2];
377
378 if (first_type->tt_type == VAR_LIST)
379 return arg_number(type, context);
380 if (first_type->tt_type == VAR_DICT)
381 return arg_string(type, context);
382 return OK;
383}
384
385
386/*
Bram Moolenaar94738d82020-10-21 14:25:07 +0200387 * Lists of functions that check the argument types of a builtin function.
388 */
389argcheck_T arg1_float_or_nr[] = {arg_float_or_nr};
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100390argcheck_T arg2_listblob_item[] = {arg_list_or_blob, arg_item_of_prev};
391argcheck_T arg23_extend[] = {arg_list_or_dict, arg_same_as_prev, arg_extend3};
Bram Moolenaarca174532020-10-21 16:42:22 +0200392argcheck_T arg3_insert[] = {arg_list_or_blob, arg_item_of_prev, arg_number};
Bram Moolenaar94738d82020-10-21 14:25:07 +0200393
394/*
395 * Functions that return the return type of a builtin function.
Bram Moolenaara1224cb2020-10-22 12:31:49 +0200396 * Note that "argtypes" is NULL if "argcount" is zero.
Bram Moolenaar94738d82020-10-21 14:25:07 +0200397 */
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100398 static type_T *
399ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
400{
401 return &t_void;
402}
403 static type_T *
404ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
405{
406 return &t_any;
407}
408 static type_T *
Bram Moolenaar403dc312020-10-17 19:29:51 +0200409ret_bool(int argcount UNUSED, type_T **argtypes UNUSED)
410{
411 return &t_bool;
412}
413 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100414ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
415{
416 return &t_number;
417}
418 static type_T *
419ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
420{
421 return &t_float;
422}
423 static type_T *
424ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
425{
426 return &t_string;
427}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200428 static type_T *
429ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100430{
431 return &t_list_any;
432}
433 static type_T *
434ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
435{
436 return &t_list_number;
437}
438 static type_T *
439ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
440{
441 return &t_list_string;
442}
443 static type_T *
444ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
445{
446 return &t_list_dict_any;
447}
448 static type_T *
449ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
450{
451 return &t_dict_any;
452}
453 static type_T *
454ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
455{
456 return &t_dict_number;
457}
458 static type_T *
459ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
460{
461 return &t_dict_string;
462}
463 static type_T *
464ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
465{
466 return &t_blob;
467}
468 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200469ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100470{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200471 return &t_func_any;
472}
473 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100474ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
475{
476 return &t_channel;
477}
478 static type_T *
479ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
480{
481 return &t_job;
482}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200483 static type_T *
484ret_first_arg(int argcount, type_T **argtypes)
485{
486 if (argcount > 0)
487 return argtypes[0];
488 return &t_void;
489}
Bram Moolenaarea696852020-11-09 18:31:39 +0100490// for map(): returns first argument but item type may differ
491 static type_T *
492ret_first_cont(int argcount UNUSED, type_T **argtypes)
493{
494 if (argtypes[0]->tt_type == VAR_LIST)
495 return &t_list_any;
496 if (argtypes[0]->tt_type == VAR_DICT)
497 return &t_dict_any;
498 if (argtypes[0]->tt_type == VAR_BLOB)
499 return argtypes[0];
500 return &t_any;
501}
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200502
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200503/*
504 * Used for getqflist(): returns list if there is no argument, dict if there is
505 * one.
506 */
507 static type_T *
508ret_list_or_dict_0(int argcount, type_T **argtypes UNUSED)
509{
510 if (argcount > 0)
511 return &t_dict_any;
512 return &t_list_dict_any;
513}
514
515/*
516 * Used for getloclist(): returns list if there is one argument, dict if there
517 * are two.
518 */
519 static type_T *
520ret_list_or_dict_1(int argcount, type_T **argtypes UNUSED)
521{
522 if (argcount > 1)
523 return &t_dict_any;
524 return &t_list_dict_any;
525}
526
Bram Moolenaar846178a2020-07-05 17:04:13 +0200527 static type_T *
528ret_argv(int argcount, type_T **argtypes UNUSED)
529{
530 // argv() returns list of strings
531 if (argcount == 0)
532 return &t_list_string;
533
534 // argv(0) returns a string, but argv(-1] returns a list
535 return &t_any;
536}
537
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200538 static type_T *
539ret_remove(int argcount UNUSED, type_T **argtypes)
540{
Bram Moolenaar5e654232020-09-16 15:22:00 +0200541 if (argtypes != NULL)
542 {
543 if (argtypes[0]->tt_type == VAR_LIST
544 || argtypes[0]->tt_type == VAR_DICT)
545 return argtypes[0]->tt_member;
546 if (argtypes[0]->tt_type == VAR_BLOB)
547 return &t_number;
548 }
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200549 return &t_any;
550}
551
Bram Moolenaar3d945cc2020-08-06 21:26:59 +0200552 static type_T *
553ret_getreg(int argcount, type_T **argtypes UNUSED)
554{
555 // Assume that if the third argument is passed it's non-zero
556 if (argcount == 3)
557 return &t_list_string;
558 return &t_string;
559}
560
Bram Moolenaar4a6d1b62020-08-08 17:55:49 +0200561 static type_T *
562ret_maparg(int argcount, type_T **argtypes UNUSED)
563{
564 // Assume that if the fourth argument is passed it's non-zero
565 if (argcount == 4)
566 return &t_dict_any;
567 return &t_string;
568}
569
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100570static type_T *ret_f_function(int argcount, type_T **argtypes);
571
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200572/*
573 * Array with names and number of arguments of all internal functions
574 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
575 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200576typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200577{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200578 char *f_name; // function name
579 char f_min_argc; // minimal number of arguments
580 char f_max_argc; // maximal number of arguments
581 char f_argtype; // for method: FEARG_ values
Bram Moolenaar94738d82020-10-21 14:25:07 +0200582 argcheck_T *f_argcheck; // list of functions to check argument types
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100583 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
584 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200585 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200586 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200587} funcentry_T;
588
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200589// values for f_argtype; zero means it cannot be used as a method
590#define FEARG_1 1 // base is the first argument
591#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200592#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200593#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200594#define FEARG_LAST 9 // base is the last argument
595
Bram Moolenaar15c47602020-03-26 22:16:48 +0100596#ifdef FEAT_FLOAT
597# define FLOAT_FUNC(name) name
598#else
599# define FLOAT_FUNC(name) NULL
600#endif
601#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
602# define MATH_FUNC(name) name
603#else
604# define MATH_FUNC(name) NULL
605#endif
606#ifdef FEAT_TIMERS
607# define TIMER_FUNC(name) name
608#else
609# define TIMER_FUNC(name) NULL
610#endif
611#ifdef FEAT_JOB_CHANNEL
612# define JOB_FUNC(name) name
613#else
614# define JOB_FUNC(name) NULL
615#endif
616#ifdef FEAT_PROP_POPUP
617# define PROP_FUNC(name) name
618#else
619# define PROP_FUNC(name) NULL
620#endif
621#ifdef FEAT_SIGNS
622# define SIGN_FUNC(name) name
623#else
624# define SIGN_FUNC(name) NULL
625#endif
626#ifdef FEAT_SOUND
627# define SOUND_FUNC(name) name
628#else
629# define SOUND_FUNC(name) NULL
630#endif
631#ifdef FEAT_TERMINAL
632# define TERM_FUNC(name) name
633#else
634# define TERM_FUNC(name) NULL
635#endif
636
Bram Moolenaarac92e252019-08-03 21:58:38 +0200637static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200638{
Bram Moolenaar94738d82020-10-21 14:25:07 +0200639 {"abs", 1, 1, FEARG_1, arg1_float_or_nr,
640 ret_any, FLOAT_FUNC(f_abs)},
641 {"acos", 1, 1, FEARG_1, NULL,
642 ret_float, FLOAT_FUNC(f_acos)},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100643 {"add", 2, 2, FEARG_1, NULL /* arg2_listblob_item */,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200644 ret_first_arg, f_add},
645 {"and", 2, 2, FEARG_1, NULL,
646 ret_number, f_and},
647 {"append", 2, 2, FEARG_2, NULL,
648 ret_number, f_append},
649 {"appendbufline", 3, 3, FEARG_3, NULL,
650 ret_number, f_appendbufline},
651 {"argc", 0, 1, 0, NULL,
652 ret_number, f_argc},
653 {"argidx", 0, 0, 0, NULL,
654 ret_number, f_argidx},
655 {"arglistid", 0, 2, 0, NULL,
656 ret_number, f_arglistid},
657 {"argv", 0, 2, 0, NULL,
658 ret_argv, f_argv},
659 {"asin", 1, 1, FEARG_1, NULL,
660 ret_float, FLOAT_FUNC(f_asin)},
661 {"assert_beeps", 1, 2, FEARG_1, NULL,
662 ret_number, f_assert_beeps},
663 {"assert_equal", 2, 3, FEARG_2, NULL,
664 ret_number, f_assert_equal},
665 {"assert_equalfile", 2, 3, FEARG_1, NULL,
666 ret_number, f_assert_equalfile},
667 {"assert_exception", 1, 2, 0, NULL,
668 ret_number, f_assert_exception},
669 {"assert_fails", 1, 5, FEARG_1, NULL,
670 ret_number, f_assert_fails},
671 {"assert_false", 1, 2, FEARG_1, NULL,
672 ret_number, f_assert_false},
673 {"assert_inrange", 3, 4, FEARG_3, NULL,
674 ret_number, f_assert_inrange},
675 {"assert_match", 2, 3, FEARG_2, NULL,
676 ret_number, f_assert_match},
677 {"assert_notequal", 2, 3, FEARG_2, NULL,
678 ret_number, f_assert_notequal},
679 {"assert_notmatch", 2, 3, FEARG_2, NULL,
680 ret_number, f_assert_notmatch},
681 {"assert_report", 1, 1, FEARG_1, NULL,
682 ret_number, f_assert_report},
683 {"assert_true", 1, 2, FEARG_1, NULL,
684 ret_number, f_assert_true},
685 {"atan", 1, 1, FEARG_1, NULL,
686 ret_float, FLOAT_FUNC(f_atan)},
687 {"atan2", 2, 2, FEARG_1, NULL,
688 ret_float, FLOAT_FUNC(f_atan2)},
689 {"balloon_gettext", 0, 0, 0, NULL,
690 ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100691#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100692 f_balloon_gettext
693#else
694 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100695#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100696 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200697 {"balloon_show", 1, 1, FEARG_1, NULL,
698 ret_void,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100699#ifdef FEAT_BEVAL
700 f_balloon_show
701#else
702 NULL
703#endif
704 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200705 {"balloon_split", 1, 1, FEARG_1, NULL,
706 ret_list_string,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100707#if defined(FEAT_BEVAL_TERM)
708 f_balloon_split
709#else
710 NULL
711#endif
712 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200713 {"browse", 4, 4, 0, NULL,
714 ret_string, f_browse},
715 {"browsedir", 2, 2, 0, NULL,
716 ret_string, f_browsedir},
717 {"bufadd", 1, 1, FEARG_1, NULL,
718 ret_number, f_bufadd},
719 {"bufexists", 1, 1, FEARG_1, NULL,
720 ret_number, f_bufexists},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200721 {"buffer_exists", 1, 1, FEARG_1, NULL, // obsolete
722 ret_number, f_bufexists},
723 {"buffer_name", 0, 1, FEARG_1, NULL, // obsolete
724 ret_string, f_bufname},
725 {"buffer_number", 0, 1, FEARG_1, NULL, // obsolete
726 ret_number, f_bufnr},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200727 {"buflisted", 1, 1, FEARG_1, NULL,
728 ret_number, f_buflisted},
729 {"bufload", 1, 1, FEARG_1, NULL,
730 ret_void, f_bufload},
731 {"bufloaded", 1, 1, FEARG_1, NULL,
732 ret_number, f_bufloaded},
733 {"bufname", 0, 1, FEARG_1, NULL,
734 ret_string, f_bufname},
735 {"bufnr", 0, 2, FEARG_1, NULL,
736 ret_number, f_bufnr},
737 {"bufwinid", 1, 1, FEARG_1, NULL,
738 ret_number, f_bufwinid},
739 {"bufwinnr", 1, 1, FEARG_1, NULL,
740 ret_number, f_bufwinnr},
741 {"byte2line", 1, 1, FEARG_1, NULL,
742 ret_number, f_byte2line},
743 {"byteidx", 2, 2, FEARG_1, NULL,
744 ret_number, f_byteidx},
745 {"byteidxcomp", 2, 2, FEARG_1, NULL,
746 ret_number, f_byteidxcomp},
747 {"call", 2, 3, FEARG_1, NULL,
748 ret_any, f_call},
749 {"ceil", 1, 1, FEARG_1, NULL,
750 ret_float, FLOAT_FUNC(f_ceil)},
751 {"ch_canread", 1, 1, FEARG_1, NULL,
752 ret_number, JOB_FUNC(f_ch_canread)},
753 {"ch_close", 1, 1, FEARG_1, NULL,
754 ret_void, JOB_FUNC(f_ch_close)},
755 {"ch_close_in", 1, 1, FEARG_1, NULL,
756 ret_void, JOB_FUNC(f_ch_close_in)},
757 {"ch_evalexpr", 2, 3, FEARG_1, NULL,
758 ret_any, JOB_FUNC(f_ch_evalexpr)},
759 {"ch_evalraw", 2, 3, FEARG_1, NULL,
760 ret_any, JOB_FUNC(f_ch_evalraw)},
761 {"ch_getbufnr", 2, 2, FEARG_1, NULL,
762 ret_number, JOB_FUNC(f_ch_getbufnr)},
763 {"ch_getjob", 1, 1, FEARG_1, NULL,
764 ret_job, JOB_FUNC(f_ch_getjob)},
765 {"ch_info", 1, 1, FEARG_1, NULL,
766 ret_dict_any, JOB_FUNC(f_ch_info)},
767 {"ch_log", 1, 2, FEARG_1, NULL,
768 ret_void, JOB_FUNC(f_ch_log)},
769 {"ch_logfile", 1, 2, FEARG_1, NULL,
770 ret_void, JOB_FUNC(f_ch_logfile)},
771 {"ch_open", 1, 2, FEARG_1, NULL,
772 ret_channel, JOB_FUNC(f_ch_open)},
773 {"ch_read", 1, 2, FEARG_1, NULL,
774 ret_string, JOB_FUNC(f_ch_read)},
775 {"ch_readblob", 1, 2, FEARG_1, NULL,
776 ret_blob, JOB_FUNC(f_ch_readblob)},
777 {"ch_readraw", 1, 2, FEARG_1, NULL,
778 ret_string, JOB_FUNC(f_ch_readraw)},
779 {"ch_sendexpr", 2, 3, FEARG_1, NULL,
780 ret_void, JOB_FUNC(f_ch_sendexpr)},
781 {"ch_sendraw", 2, 3, FEARG_1, NULL,
782 ret_void, JOB_FUNC(f_ch_sendraw)},
783 {"ch_setoptions", 2, 2, FEARG_1, NULL,
784 ret_void, JOB_FUNC(f_ch_setoptions)},
785 {"ch_status", 1, 2, FEARG_1, NULL,
786 ret_string, JOB_FUNC(f_ch_status)},
787 {"changenr", 0, 0, 0, NULL,
788 ret_number, f_changenr},
789 {"char2nr", 1, 2, FEARG_1, NULL,
790 ret_number, f_char2nr},
791 {"charclass", 1, 1, FEARG_1, NULL,
792 ret_number, f_charclass},
Bram Moolenaar17793ef2020-12-28 12:56:58 +0100793 {"charidx", 2, 3, FEARG_1, NULL,
794 ret_number, f_charidx},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200795 {"chdir", 1, 1, FEARG_1, NULL,
796 ret_string, f_chdir},
797 {"cindent", 1, 1, FEARG_1, NULL,
798 ret_number, f_cindent},
799 {"clearmatches", 0, 1, FEARG_1, NULL,
800 ret_void, f_clearmatches},
801 {"col", 1, 1, FEARG_1, NULL,
802 ret_number, f_col},
803 {"complete", 2, 2, FEARG_2, NULL,
804 ret_void, f_complete},
805 {"complete_add", 1, 1, FEARG_1, NULL,
806 ret_number, f_complete_add},
807 {"complete_check", 0, 0, 0, NULL,
808 ret_number, f_complete_check},
809 {"complete_info", 0, 1, FEARG_1, NULL,
810 ret_dict_any, f_complete_info},
811 {"confirm", 1, 4, FEARG_1, NULL,
812 ret_number, f_confirm},
813 {"copy", 1, 1, FEARG_1, NULL,
814 ret_first_arg, f_copy},
815 {"cos", 1, 1, FEARG_1, NULL,
816 ret_float, FLOAT_FUNC(f_cos)},
817 {"cosh", 1, 1, FEARG_1, NULL,
818 ret_float, FLOAT_FUNC(f_cosh)},
819 {"count", 2, 4, FEARG_1, NULL,
820 ret_number, f_count},
821 {"cscope_connection",0,3, 0, NULL,
822 ret_number, f_cscope_connection},
823 {"cursor", 1, 3, FEARG_1, NULL,
824 ret_number, f_cursor},
825 {"debugbreak", 1, 1, FEARG_1, NULL,
826 ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100827#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100828 f_debugbreak
829#else
830 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200831#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100832 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200833 {"deepcopy", 1, 2, FEARG_1, NULL,
834 ret_first_arg, f_deepcopy},
835 {"delete", 1, 2, FEARG_1, NULL,
836 ret_number, f_delete},
837 {"deletebufline", 2, 3, FEARG_1, NULL,
838 ret_number, f_deletebufline},
839 {"did_filetype", 0, 0, 0, NULL,
840 ret_number, f_did_filetype},
841 {"diff_filler", 1, 1, FEARG_1, NULL,
842 ret_number, f_diff_filler},
843 {"diff_hlID", 2, 2, FEARG_1, NULL,
844 ret_number, f_diff_hlID},
845 {"echoraw", 1, 1, FEARG_1, NULL,
846 ret_number, f_echoraw},
847 {"empty", 1, 1, FEARG_1, NULL,
848 ret_number, f_empty},
849 {"environ", 0, 0, 0, NULL,
850 ret_dict_string, f_environ},
851 {"escape", 2, 2, FEARG_1, NULL,
852 ret_string, f_escape},
853 {"eval", 1, 1, FEARG_1, NULL,
854 ret_any, f_eval},
855 {"eventhandler", 0, 0, 0, NULL,
856 ret_number, f_eventhandler},
857 {"executable", 1, 1, FEARG_1, NULL,
858 ret_number, f_executable},
859 {"execute", 1, 2, FEARG_1, NULL,
860 ret_string, f_execute},
861 {"exepath", 1, 1, FEARG_1, NULL,
862 ret_string, f_exepath},
863 {"exists", 1, 1, FEARG_1, NULL,
864 ret_number, f_exists},
865 {"exp", 1, 1, FEARG_1, NULL,
866 ret_float, FLOAT_FUNC(f_exp)},
867 {"expand", 1, 3, FEARG_1, NULL,
868 ret_any, f_expand},
869 {"expandcmd", 1, 1, FEARG_1, NULL,
870 ret_string, f_expandcmd},
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100871 {"extend", 2, 3, FEARG_1, arg23_extend,
Bram Moolenaar94738d82020-10-21 14:25:07 +0200872 ret_first_arg, f_extend},
873 {"feedkeys", 1, 2, FEARG_1, NULL,
874 ret_void, f_feedkeys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +0200875 {"file_readable", 1, 1, FEARG_1, NULL, // obsolete
876 ret_number, f_filereadable},
Bram Moolenaar94738d82020-10-21 14:25:07 +0200877 {"filereadable", 1, 1, FEARG_1, NULL,
878 ret_number, f_filereadable},
879 {"filewritable", 1, 1, FEARG_1, NULL,
880 ret_number, f_filewritable},
881 {"filter", 2, 2, FEARG_1, NULL,
882 ret_first_arg, f_filter},
883 {"finddir", 1, 3, FEARG_1, NULL,
884 ret_string, f_finddir},
885 {"findfile", 1, 3, FEARG_1, NULL,
886 ret_string, f_findfile},
887 {"flatten", 1, 2, FEARG_1, NULL,
888 ret_list_any, f_flatten},
889 {"float2nr", 1, 1, FEARG_1, NULL,
890 ret_number, FLOAT_FUNC(f_float2nr)},
891 {"floor", 1, 1, FEARG_1, NULL,
892 ret_float, FLOAT_FUNC(f_floor)},
893 {"fmod", 2, 2, FEARG_1, NULL,
894 ret_float, FLOAT_FUNC(f_fmod)},
895 {"fnameescape", 1, 1, FEARG_1, NULL,
896 ret_string, f_fnameescape},
897 {"fnamemodify", 2, 2, FEARG_1, NULL,
898 ret_string, f_fnamemodify},
899 {"foldclosed", 1, 1, FEARG_1, NULL,
900 ret_number, f_foldclosed},
901 {"foldclosedend", 1, 1, FEARG_1, NULL,
902 ret_number, f_foldclosedend},
903 {"foldlevel", 1, 1, FEARG_1, NULL,
904 ret_number, f_foldlevel},
905 {"foldtext", 0, 0, 0, NULL,
906 ret_string, f_foldtext},
907 {"foldtextresult", 1, 1, FEARG_1, NULL,
908 ret_string, f_foldtextresult},
909 {"foreground", 0, 0, 0, NULL,
910 ret_void, f_foreground},
911 {"funcref", 1, 3, FEARG_1, NULL,
912 ret_func_any, f_funcref},
913 {"function", 1, 3, FEARG_1, NULL,
914 ret_f_function, f_function},
915 {"garbagecollect", 0, 1, 0, NULL,
916 ret_void, f_garbagecollect},
917 {"get", 2, 3, FEARG_1, NULL,
918 ret_any, f_get},
919 {"getbufinfo", 0, 1, FEARG_1, NULL,
920 ret_list_dict_any, f_getbufinfo},
921 {"getbufline", 2, 3, FEARG_1, NULL,
922 ret_list_string, f_getbufline},
923 {"getbufvar", 2, 3, FEARG_1, NULL,
924 ret_any, f_getbufvar},
925 {"getchangelist", 0, 1, FEARG_1, NULL,
926 ret_list_any, f_getchangelist},
927 {"getchar", 0, 1, 0, NULL,
928 ret_number, f_getchar},
929 {"getcharmod", 0, 0, 0, NULL,
930 ret_number, f_getcharmod},
931 {"getcharsearch", 0, 0, 0, NULL,
932 ret_dict_any, f_getcharsearch},
933 {"getcmdline", 0, 0, 0, NULL,
934 ret_string, f_getcmdline},
935 {"getcmdpos", 0, 0, 0, NULL,
936 ret_number, f_getcmdpos},
937 {"getcmdtype", 0, 0, 0, NULL,
938 ret_string, f_getcmdtype},
939 {"getcmdwintype", 0, 0, 0, NULL,
940 ret_string, f_getcmdwintype},
941 {"getcompletion", 2, 3, FEARG_1, NULL,
942 ret_list_string, f_getcompletion},
943 {"getcurpos", 0, 1, FEARG_1, NULL,
944 ret_list_number, f_getcurpos},
945 {"getcwd", 0, 2, FEARG_1, NULL,
946 ret_string, f_getcwd},
947 {"getenv", 1, 1, FEARG_1, NULL,
948 ret_string, f_getenv},
949 {"getfontname", 0, 1, 0, NULL,
950 ret_string, f_getfontname},
951 {"getfperm", 1, 1, FEARG_1, NULL,
952 ret_string, f_getfperm},
953 {"getfsize", 1, 1, FEARG_1, NULL,
954 ret_number, f_getfsize},
955 {"getftime", 1, 1, FEARG_1, NULL,
956 ret_number, f_getftime},
957 {"getftype", 1, 1, FEARG_1, NULL,
958 ret_string, f_getftype},
959 {"getimstatus", 0, 0, 0, NULL,
960 ret_number, f_getimstatus},
961 {"getjumplist", 0, 2, FEARG_1, NULL,
962 ret_list_any, f_getjumplist},
963 {"getline", 1, 2, FEARG_1, NULL,
964 ret_f_getline, f_getline},
965 {"getloclist", 1, 2, 0, NULL,
966 ret_list_or_dict_1, f_getloclist},
967 {"getmarklist", 0, 1, FEARG_1, NULL,
968 ret_list_dict_any, f_getmarklist},
969 {"getmatches", 0, 1, 0, NULL,
970 ret_list_dict_any, f_getmatches},
971 {"getmousepos", 0, 0, 0, NULL,
972 ret_dict_number, f_getmousepos},
973 {"getpid", 0, 0, 0, NULL,
974 ret_number, f_getpid},
975 {"getpos", 1, 1, FEARG_1, NULL,
976 ret_list_number, f_getpos},
977 {"getqflist", 0, 1, 0, NULL,
978 ret_list_or_dict_0, f_getqflist},
979 {"getreg", 0, 3, FEARG_1, NULL,
980 ret_getreg, f_getreg},
981 {"getreginfo", 0, 1, FEARG_1, NULL,
982 ret_dict_any, f_getreginfo},
983 {"getregtype", 0, 1, FEARG_1, NULL,
984 ret_string, f_getregtype},
985 {"gettabinfo", 0, 1, FEARG_1, NULL,
986 ret_list_dict_any, f_gettabinfo},
987 {"gettabvar", 2, 3, FEARG_1, NULL,
988 ret_any, f_gettabvar},
989 {"gettabwinvar", 3, 4, FEARG_1, NULL,
990 ret_any, f_gettabwinvar},
991 {"gettagstack", 0, 1, FEARG_1, NULL,
992 ret_dict_any, f_gettagstack},
993 {"gettext", 1, 1, FEARG_1, NULL,
994 ret_string, f_gettext},
995 {"getwininfo", 0, 1, FEARG_1, NULL,
996 ret_list_dict_any, f_getwininfo},
997 {"getwinpos", 0, 1, FEARG_1, NULL,
998 ret_list_number, f_getwinpos},
999 {"getwinposx", 0, 0, 0, NULL,
1000 ret_number, f_getwinposx},
1001 {"getwinposy", 0, 0, 0, NULL,
1002 ret_number, f_getwinposy},
1003 {"getwinvar", 2, 3, FEARG_1, NULL,
1004 ret_any, f_getwinvar},
1005 {"glob", 1, 4, FEARG_1, NULL,
1006 ret_any, f_glob},
1007 {"glob2regpat", 1, 1, FEARG_1, NULL,
1008 ret_string, f_glob2regpat},
1009 {"globpath", 2, 5, FEARG_2, NULL,
1010 ret_any, f_globpath},
1011 {"has", 1, 2, 0, NULL,
1012 ret_number, f_has},
1013 {"has_key", 2, 2, FEARG_1, NULL,
1014 ret_number, f_has_key},
1015 {"haslocaldir", 0, 2, FEARG_1, NULL,
1016 ret_number, f_haslocaldir},
1017 {"hasmapto", 1, 3, FEARG_1, NULL,
1018 ret_number, f_hasmapto},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001019 {"highlightID", 1, 1, FEARG_1, NULL, // obsolete
1020 ret_number, f_hlID},
1021 {"highlight_exists",1, 1, FEARG_1, NULL, // obsolete
1022 ret_number, f_hlexists},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001023 {"histadd", 2, 2, FEARG_2, NULL,
1024 ret_number, f_histadd},
1025 {"histdel", 1, 2, FEARG_1, NULL,
1026 ret_number, f_histdel},
1027 {"histget", 1, 2, FEARG_1, NULL,
1028 ret_string, f_histget},
1029 {"histnr", 1, 1, FEARG_1, NULL,
1030 ret_number, f_histnr},
1031 {"hlID", 1, 1, FEARG_1, NULL,
1032 ret_number, f_hlID},
1033 {"hlexists", 1, 1, FEARG_1, NULL,
1034 ret_number, f_hlexists},
1035 {"hostname", 0, 0, 0, NULL,
1036 ret_string, f_hostname},
1037 {"iconv", 3, 3, FEARG_1, NULL,
1038 ret_string, f_iconv},
1039 {"indent", 1, 1, FEARG_1, NULL,
1040 ret_number, f_indent},
1041 {"index", 2, 4, FEARG_1, NULL,
1042 ret_number, f_index},
1043 {"input", 1, 3, FEARG_1, NULL,
1044 ret_string, f_input},
1045 {"inputdialog", 1, 3, FEARG_1, NULL,
1046 ret_string, f_inputdialog},
1047 {"inputlist", 1, 1, FEARG_1, NULL,
1048 ret_number, f_inputlist},
1049 {"inputrestore", 0, 0, 0, NULL,
1050 ret_number, f_inputrestore},
1051 {"inputsave", 0, 0, 0, NULL,
1052 ret_number, f_inputsave},
1053 {"inputsecret", 1, 2, FEARG_1, NULL,
1054 ret_string, f_inputsecret},
Bram Moolenaarca174532020-10-21 16:42:22 +02001055 {"insert", 2, 3, FEARG_1, arg3_insert,
Bram Moolenaar94738d82020-10-21 14:25:07 +02001056 ret_first_arg, f_insert},
1057 {"interrupt", 0, 0, 0, NULL,
1058 ret_void, f_interrupt},
1059 {"invert", 1, 1, FEARG_1, NULL,
1060 ret_number, f_invert},
1061 {"isdirectory", 1, 1, FEARG_1, NULL,
1062 ret_number, f_isdirectory},
1063 {"isinf", 1, 1, FEARG_1, NULL,
1064 ret_number, MATH_FUNC(f_isinf)},
1065 {"islocked", 1, 1, FEARG_1, NULL,
1066 ret_number, f_islocked},
1067 {"isnan", 1, 1, FEARG_1, NULL,
1068 ret_number, MATH_FUNC(f_isnan)},
1069 {"items", 1, 1, FEARG_1, NULL,
1070 ret_list_any, f_items},
1071 {"job_getchannel", 1, 1, FEARG_1, NULL,
1072 ret_channel, JOB_FUNC(f_job_getchannel)},
1073 {"job_info", 0, 1, FEARG_1, NULL,
1074 ret_dict_any, JOB_FUNC(f_job_info)},
1075 {"job_setoptions", 2, 2, FEARG_1, NULL,
1076 ret_void, JOB_FUNC(f_job_setoptions)},
1077 {"job_start", 1, 2, FEARG_1, NULL,
1078 ret_job, JOB_FUNC(f_job_start)},
1079 {"job_status", 1, 1, FEARG_1, NULL,
1080 ret_string, JOB_FUNC(f_job_status)},
1081 {"job_stop", 1, 2, FEARG_1, NULL,
1082 ret_number, JOB_FUNC(f_job_stop)},
1083 {"join", 1, 2, FEARG_1, NULL,
1084 ret_string, f_join},
1085 {"js_decode", 1, 1, FEARG_1, NULL,
1086 ret_any, f_js_decode},
1087 {"js_encode", 1, 1, FEARG_1, NULL,
1088 ret_string, f_js_encode},
1089 {"json_decode", 1, 1, FEARG_1, NULL,
1090 ret_any, f_json_decode},
1091 {"json_encode", 1, 1, FEARG_1, NULL,
1092 ret_string, f_json_encode},
1093 {"keys", 1, 1, FEARG_1, NULL,
1094 ret_list_string, f_keys},
Bram Moolenaarb8f519e2020-10-21 14:49:08 +02001095 {"last_buffer_nr", 0, 0, 0, NULL, // obsolete
1096 ret_number, f_last_buffer_nr},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001097 {"len", 1, 1, FEARG_1, NULL,
1098 ret_number, f_len},
1099 {"libcall", 3, 3, FEARG_3, NULL,
1100 ret_string, f_libcall},
1101 {"libcallnr", 3, 3, FEARG_3, NULL,
1102 ret_number, f_libcallnr},
1103 {"line", 1, 2, FEARG_1, NULL,
1104 ret_number, f_line},
1105 {"line2byte", 1, 1, FEARG_1, NULL,
1106 ret_number, f_line2byte},
1107 {"lispindent", 1, 1, FEARG_1, NULL,
1108 ret_number, f_lispindent},
1109 {"list2str", 1, 2, FEARG_1, NULL,
1110 ret_string, f_list2str},
1111 {"listener_add", 1, 2, FEARG_2, NULL,
1112 ret_number, f_listener_add},
1113 {"listener_flush", 0, 1, FEARG_1, NULL,
1114 ret_void, f_listener_flush},
1115 {"listener_remove", 1, 1, FEARG_1, NULL,
1116 ret_number, f_listener_remove},
1117 {"localtime", 0, 0, 0, NULL,
1118 ret_number, f_localtime},
1119 {"log", 1, 1, FEARG_1, NULL,
1120 ret_float, FLOAT_FUNC(f_log)},
1121 {"log10", 1, 1, FEARG_1, NULL,
1122 ret_float, FLOAT_FUNC(f_log10)},
1123 {"luaeval", 1, 2, FEARG_1, NULL,
1124 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001125#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +01001126 f_luaeval
1127#else
1128 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001129#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001130 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001131 {"map", 2, 2, FEARG_1, NULL,
Bram Moolenaarea696852020-11-09 18:31:39 +01001132 ret_first_cont, f_map},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001133 {"maparg", 1, 4, FEARG_1, NULL,
1134 ret_maparg, f_maparg},
1135 {"mapcheck", 1, 3, FEARG_1, NULL,
1136 ret_string, f_mapcheck},
Bram Moolenaarea696852020-11-09 18:31:39 +01001137 {"mapnew", 2, 2, FEARG_1, NULL,
1138 ret_first_cont, f_mapnew},
Bram Moolenaar94738d82020-10-21 14:25:07 +02001139 {"mapset", 3, 3, FEARG_1, NULL,
1140 ret_void, f_mapset},
1141 {"match", 2, 4, FEARG_1, NULL,
1142 ret_any, f_match},
1143 {"matchadd", 2, 5, FEARG_1, NULL,
1144 ret_number, f_matchadd},
1145 {"matchaddpos", 2, 5, FEARG_1, NULL,
1146 ret_number, f_matchaddpos},
1147 {"matcharg", 1, 1, FEARG_1, NULL,
1148 ret_list_string, f_matcharg},
1149 {"matchdelete", 1, 2, FEARG_1, NULL,
1150 ret_number, f_matchdelete},
1151 {"matchend", 2, 4, FEARG_1, NULL,
1152 ret_number, f_matchend},
1153 {"matchfuzzy", 2, 3, FEARG_1, NULL,
1154 ret_list_string, f_matchfuzzy},
1155 {"matchfuzzypos", 2, 3, FEARG_1, NULL,
1156 ret_list_any, f_matchfuzzypos},
1157 {"matchlist", 2, 4, FEARG_1, NULL,
1158 ret_list_string, f_matchlist},
1159 {"matchstr", 2, 4, FEARG_1, NULL,
1160 ret_string, f_matchstr},
1161 {"matchstrpos", 2, 4, FEARG_1, NULL,
1162 ret_list_any, f_matchstrpos},
1163 {"max", 1, 1, FEARG_1, NULL,
1164 ret_any, f_max},
1165 {"menu_info", 1, 2, FEARG_1, NULL,
1166 ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001167#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +01001168 f_menu_info
1169#else
1170 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001171#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001172 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001173 {"min", 1, 1, FEARG_1, NULL,
1174 ret_any, f_min},
1175 {"mkdir", 1, 3, FEARG_1, NULL,
1176 ret_number, f_mkdir},
1177 {"mode", 0, 1, FEARG_1, NULL,
1178 ret_string, f_mode},
1179 {"mzeval", 1, 1, FEARG_1, NULL,
1180 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001181#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001182 f_mzeval
1183#else
1184 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001185#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001186 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001187 {"nextnonblank", 1, 1, FEARG_1, NULL,
1188 ret_number, f_nextnonblank},
1189 {"nr2char", 1, 2, FEARG_1, NULL,
1190 ret_string, f_nr2char},
1191 {"or", 2, 2, FEARG_1, NULL,
1192 ret_number, f_or},
1193 {"pathshorten", 1, 2, FEARG_1, NULL,
1194 ret_string, f_pathshorten},
1195 {"perleval", 1, 1, FEARG_1, NULL,
1196 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001197#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +01001198 f_perleval
1199#else
1200 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001201#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001202 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001203 {"popup_atcursor", 2, 2, FEARG_1, NULL,
1204 ret_number, PROP_FUNC(f_popup_atcursor)},
1205 {"popup_beval", 2, 2, FEARG_1, NULL,
1206 ret_number, PROP_FUNC(f_popup_beval)},
1207 {"popup_clear", 0, 1, 0, NULL,
1208 ret_void, PROP_FUNC(f_popup_clear)},
1209 {"popup_close", 1, 2, FEARG_1, NULL,
1210 ret_void, PROP_FUNC(f_popup_close)},
1211 {"popup_create", 2, 2, FEARG_1, NULL,
1212 ret_number, PROP_FUNC(f_popup_create)},
1213 {"popup_dialog", 2, 2, FEARG_1, NULL,
1214 ret_number, PROP_FUNC(f_popup_dialog)},
1215 {"popup_filter_menu", 2, 2, 0, NULL,
1216 ret_bool, PROP_FUNC(f_popup_filter_menu)},
1217 {"popup_filter_yesno", 2, 2, 0, NULL,
1218 ret_bool, PROP_FUNC(f_popup_filter_yesno)},
1219 {"popup_findinfo", 0, 0, 0, NULL,
1220 ret_number, PROP_FUNC(f_popup_findinfo)},
1221 {"popup_findpreview", 0, 0, 0, NULL,
1222 ret_number, PROP_FUNC(f_popup_findpreview)},
1223 {"popup_getoptions", 1, 1, FEARG_1, NULL,
1224 ret_dict_any, PROP_FUNC(f_popup_getoptions)},
1225 {"popup_getpos", 1, 1, FEARG_1, NULL,
1226 ret_dict_any, PROP_FUNC(f_popup_getpos)},
1227 {"popup_hide", 1, 1, FEARG_1, NULL,
1228 ret_void, PROP_FUNC(f_popup_hide)},
1229 {"popup_list", 0, 0, 0, NULL,
1230 ret_list_number, PROP_FUNC(f_popup_list)},
1231 {"popup_locate", 2, 2, 0, NULL,
1232 ret_number, PROP_FUNC(f_popup_locate)},
1233 {"popup_menu", 2, 2, FEARG_1, NULL,
1234 ret_number, PROP_FUNC(f_popup_menu)},
1235 {"popup_move", 2, 2, FEARG_1, NULL,
1236 ret_void, PROP_FUNC(f_popup_move)},
1237 {"popup_notification", 2, 2, FEARG_1, NULL,
1238 ret_number, PROP_FUNC(f_popup_notification)},
1239 {"popup_setoptions", 2, 2, FEARG_1, NULL,
1240 ret_void, PROP_FUNC(f_popup_setoptions)},
1241 {"popup_settext", 2, 2, FEARG_1, NULL,
1242 ret_void, PROP_FUNC(f_popup_settext)},
1243 {"popup_show", 1, 1, FEARG_1, NULL,
1244 ret_void, PROP_FUNC(f_popup_show)},
1245 {"pow", 2, 2, FEARG_1, NULL,
1246 ret_float, FLOAT_FUNC(f_pow)},
1247 {"prevnonblank", 1, 1, FEARG_1, NULL,
1248 ret_number, f_prevnonblank},
1249 {"printf", 1, 19, FEARG_2, NULL,
1250 ret_string, f_printf},
1251 {"prompt_getprompt", 1, 1, FEARG_1, NULL,
1252 ret_string, JOB_FUNC(f_prompt_getprompt)},
1253 {"prompt_setcallback", 2, 2, FEARG_1, NULL,
1254 ret_void, JOB_FUNC(f_prompt_setcallback)},
1255 {"prompt_setinterrupt", 2, 2, FEARG_1, NULL,
1256 ret_void, JOB_FUNC(f_prompt_setinterrupt)},
1257 {"prompt_setprompt", 2, 2, FEARG_1, NULL,
1258 ret_void, JOB_FUNC(f_prompt_setprompt)},
1259 {"prop_add", 3, 3, FEARG_1, NULL,
1260 ret_void, PROP_FUNC(f_prop_add)},
1261 {"prop_clear", 1, 3, FEARG_1, NULL,
1262 ret_void, PROP_FUNC(f_prop_clear)},
1263 {"prop_find", 1, 2, FEARG_1, NULL,
1264 ret_dict_any, PROP_FUNC(f_prop_find)},
1265 {"prop_list", 1, 2, FEARG_1, NULL,
1266 ret_list_dict_any, PROP_FUNC(f_prop_list)},
1267 {"prop_remove", 1, 3, FEARG_1, NULL,
1268 ret_number, PROP_FUNC(f_prop_remove)},
1269 {"prop_type_add", 2, 2, FEARG_1, NULL,
1270 ret_void, PROP_FUNC(f_prop_type_add)},
1271 {"prop_type_change", 2, 2, FEARG_1, NULL,
1272 ret_void, PROP_FUNC(f_prop_type_change)},
1273 {"prop_type_delete", 1, 2, FEARG_1, NULL,
1274 ret_void, PROP_FUNC(f_prop_type_delete)},
1275 {"prop_type_get", 1, 2, FEARG_1, NULL,
1276 ret_dict_any, PROP_FUNC(f_prop_type_get)},
1277 {"prop_type_list", 0, 1, FEARG_1, NULL,
1278 ret_list_string, PROP_FUNC(f_prop_type_list)},
1279 {"pum_getpos", 0, 0, 0, NULL,
1280 ret_dict_number, f_pum_getpos},
1281 {"pumvisible", 0, 0, 0, NULL,
1282 ret_number, f_pumvisible},
1283 {"py3eval", 1, 1, FEARG_1, NULL,
1284 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001285#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +01001286 f_py3eval
1287#else
1288 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001289#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001290 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001291 {"pyeval", 1, 1, FEARG_1, NULL,
1292 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001293#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +01001294 f_pyeval
1295#else
1296 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001297#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001298 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001299 {"pyxeval", 1, 1, FEARG_1, NULL,
1300 ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001301#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +01001302 f_pyxeval
1303#else
1304 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001305#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001306 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001307 {"rand", 0, 1, FEARG_1, NULL,
1308 ret_number, f_rand},
1309 {"range", 1, 3, FEARG_1, NULL,
1310 ret_list_number, f_range},
1311 {"readdir", 1, 3, FEARG_1, NULL,
1312 ret_list_string, f_readdir},
1313 {"readdirex", 1, 3, FEARG_1, NULL,
1314 ret_list_dict_any, f_readdirex},
1315 {"readfile", 1, 3, FEARG_1, NULL,
1316 ret_any, f_readfile},
1317 {"reduce", 2, 3, FEARG_1, NULL,
1318 ret_any, f_reduce},
1319 {"reg_executing", 0, 0, 0, NULL,
1320 ret_string, f_reg_executing},
1321 {"reg_recording", 0, 0, 0, NULL,
1322 ret_string, f_reg_recording},
1323 {"reltime", 0, 2, FEARG_1, NULL,
1324 ret_list_any, f_reltime},
1325 {"reltimefloat", 1, 1, FEARG_1, NULL,
1326 ret_float, FLOAT_FUNC(f_reltimefloat)},
1327 {"reltimestr", 1, 1, FEARG_1, NULL,
1328 ret_string, f_reltimestr},
1329 {"remote_expr", 2, 4, FEARG_1, NULL,
1330 ret_string, f_remote_expr},
1331 {"remote_foreground", 1, 1, FEARG_1, NULL,
1332 ret_string, f_remote_foreground},
1333 {"remote_peek", 1, 2, FEARG_1, NULL,
1334 ret_number, f_remote_peek},
1335 {"remote_read", 1, 2, FEARG_1, NULL,
1336 ret_string, f_remote_read},
1337 {"remote_send", 2, 3, FEARG_1, NULL,
1338 ret_string, f_remote_send},
1339 {"remote_startserver", 1, 1, FEARG_1, NULL,
1340 ret_void, f_remote_startserver},
1341 {"remove", 2, 3, FEARG_1, NULL,
1342 ret_remove, f_remove},
1343 {"rename", 2, 2, FEARG_1, NULL,
1344 ret_number, f_rename},
1345 {"repeat", 2, 2, FEARG_1, NULL,
1346 ret_first_arg, f_repeat},
1347 {"resolve", 1, 1, FEARG_1, NULL,
1348 ret_string, f_resolve},
1349 {"reverse", 1, 1, FEARG_1, NULL,
1350 ret_first_arg, f_reverse},
1351 {"round", 1, 1, FEARG_1, NULL,
1352 ret_float, FLOAT_FUNC(f_round)},
1353 {"rubyeval", 1, 1, FEARG_1, NULL,
1354 ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001355#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +01001356 f_rubyeval
1357#else
1358 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001359#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001360 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001361 {"screenattr", 2, 2, FEARG_1, NULL,
1362 ret_number, f_screenattr},
1363 {"screenchar", 2, 2, FEARG_1, NULL,
1364 ret_number, f_screenchar},
1365 {"screenchars", 2, 2, FEARG_1, NULL,
1366 ret_list_number, f_screenchars},
1367 {"screencol", 0, 0, 0, NULL,
1368 ret_number, f_screencol},
1369 {"screenpos", 3, 3, FEARG_1, NULL,
1370 ret_dict_number, f_screenpos},
1371 {"screenrow", 0, 0, 0, NULL,
1372 ret_number, f_screenrow},
1373 {"screenstring", 2, 2, FEARG_1, NULL,
1374 ret_string, f_screenstring},
1375 {"search", 1, 5, FEARG_1, NULL,
1376 ret_number, f_search},
1377 {"searchcount", 0, 1, FEARG_1, NULL,
1378 ret_dict_any, f_searchcount},
1379 {"searchdecl", 1, 3, FEARG_1, NULL,
1380 ret_number, f_searchdecl},
1381 {"searchpair", 3, 7, 0, NULL,
1382 ret_number, f_searchpair},
1383 {"searchpairpos", 3, 7, 0, NULL,
1384 ret_list_number, f_searchpairpos},
1385 {"searchpos", 1, 5, FEARG_1, NULL,
1386 ret_list_number, f_searchpos},
1387 {"server2client", 2, 2, FEARG_1, NULL,
1388 ret_number, f_server2client},
1389 {"serverlist", 0, 0, 0, NULL,
1390 ret_string, f_serverlist},
1391 {"setbufline", 3, 3, FEARG_3, NULL,
1392 ret_number, f_setbufline},
1393 {"setbufvar", 3, 3, FEARG_3, NULL,
1394 ret_void, f_setbufvar},
1395 {"setcellwidths", 1, 1, FEARG_1, NULL,
1396 ret_void, f_setcellwidths},
1397 {"setcharsearch", 1, 1, FEARG_1, NULL,
1398 ret_void, f_setcharsearch},
1399 {"setcmdpos", 1, 1, FEARG_1, NULL,
1400 ret_number, f_setcmdpos},
1401 {"setenv", 2, 2, FEARG_2, NULL,
1402 ret_void, f_setenv},
1403 {"setfperm", 2, 2, FEARG_1, NULL,
1404 ret_number, f_setfperm},
1405 {"setline", 2, 2, FEARG_2, NULL,
1406 ret_number, f_setline},
1407 {"setloclist", 2, 4, FEARG_2, NULL,
1408 ret_number, f_setloclist},
1409 {"setmatches", 1, 2, FEARG_1, NULL,
1410 ret_number, f_setmatches},
1411 {"setpos", 2, 2, FEARG_2, NULL,
1412 ret_number, f_setpos},
1413 {"setqflist", 1, 3, FEARG_1, NULL,
1414 ret_number, f_setqflist},
1415 {"setreg", 2, 3, FEARG_2, NULL,
1416 ret_number, f_setreg},
1417 {"settabvar", 3, 3, FEARG_3, NULL,
1418 ret_void, f_settabvar},
1419 {"settabwinvar", 4, 4, FEARG_4, NULL,
1420 ret_void, f_settabwinvar},
1421 {"settagstack", 2, 3, FEARG_2, NULL,
1422 ret_number, f_settagstack},
1423 {"setwinvar", 3, 3, FEARG_3, NULL,
1424 ret_void, f_setwinvar},
1425 {"sha256", 1, 1, FEARG_1, NULL,
1426 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001427#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +01001428 f_sha256
1429#else
1430 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001431#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001432 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001433 {"shellescape", 1, 2, FEARG_1, NULL,
1434 ret_string, f_shellescape},
1435 {"shiftwidth", 0, 1, FEARG_1, NULL,
1436 ret_number, f_shiftwidth},
1437 {"sign_define", 1, 2, FEARG_1, NULL,
1438 ret_any, SIGN_FUNC(f_sign_define)},
1439 {"sign_getdefined", 0, 1, FEARG_1, NULL,
1440 ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
1441 {"sign_getplaced", 0, 2, FEARG_1, NULL,
1442 ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
1443 {"sign_jump", 3, 3, FEARG_1, NULL,
1444 ret_number, SIGN_FUNC(f_sign_jump)},
1445 {"sign_place", 4, 5, FEARG_1, NULL,
1446 ret_number, SIGN_FUNC(f_sign_place)},
1447 {"sign_placelist", 1, 1, FEARG_1, NULL,
1448 ret_list_number, SIGN_FUNC(f_sign_placelist)},
1449 {"sign_undefine", 0, 1, FEARG_1, NULL,
1450 ret_number, SIGN_FUNC(f_sign_undefine)},
1451 {"sign_unplace", 1, 2, FEARG_1, NULL,
1452 ret_number, SIGN_FUNC(f_sign_unplace)},
1453 {"sign_unplacelist", 1, 2, FEARG_1, NULL,
1454 ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
1455 {"simplify", 1, 1, FEARG_1, NULL,
1456 ret_string, f_simplify},
1457 {"sin", 1, 1, FEARG_1, NULL,
1458 ret_float, FLOAT_FUNC(f_sin)},
1459 {"sinh", 1, 1, FEARG_1, NULL,
1460 ret_float, FLOAT_FUNC(f_sinh)},
1461 {"sort", 1, 3, FEARG_1, NULL,
1462 ret_first_arg, f_sort},
1463 {"sound_clear", 0, 0, 0, NULL,
1464 ret_void, SOUND_FUNC(f_sound_clear)},
1465 {"sound_playevent", 1, 2, FEARG_1, NULL,
1466 ret_number, SOUND_FUNC(f_sound_playevent)},
1467 {"sound_playfile", 1, 2, FEARG_1, NULL,
1468 ret_number, SOUND_FUNC(f_sound_playfile)},
1469 {"sound_stop", 1, 1, FEARG_1, NULL,
1470 ret_void, SOUND_FUNC(f_sound_stop)},
1471 {"soundfold", 1, 1, FEARG_1, NULL,
1472 ret_string, f_soundfold},
1473 {"spellbadword", 0, 1, FEARG_1, NULL,
1474 ret_list_string, f_spellbadword},
1475 {"spellsuggest", 1, 3, FEARG_1, NULL,
1476 ret_list_string, f_spellsuggest},
1477 {"split", 1, 3, FEARG_1, NULL,
1478 ret_list_string, f_split},
1479 {"sqrt", 1, 1, FEARG_1, NULL,
1480 ret_float, FLOAT_FUNC(f_sqrt)},
1481 {"srand", 0, 1, FEARG_1, NULL,
1482 ret_list_number, f_srand},
1483 {"state", 0, 1, FEARG_1, NULL,
1484 ret_string, f_state},
1485 {"str2float", 1, 1, FEARG_1, NULL,
1486 ret_float, FLOAT_FUNC(f_str2float)},
1487 {"str2list", 1, 2, FEARG_1, NULL,
1488 ret_list_number, f_str2list},
1489 {"str2nr", 1, 3, FEARG_1, NULL,
1490 ret_number, f_str2nr},
1491 {"strcharpart", 2, 3, FEARG_1, NULL,
1492 ret_string, f_strcharpart},
1493 {"strchars", 1, 2, FEARG_1, NULL,
1494 ret_number, f_strchars},
1495 {"strdisplaywidth", 1, 2, FEARG_1, NULL,
1496 ret_number, f_strdisplaywidth},
1497 {"strftime", 1, 2, FEARG_1, NULL,
1498 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001499#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001500 f_strftime
1501#else
1502 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001503#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001504 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001505 {"strgetchar", 2, 2, FEARG_1, NULL,
1506 ret_number, f_strgetchar},
1507 {"stridx", 2, 3, FEARG_1, NULL,
1508 ret_number, f_stridx},
1509 {"string", 1, 1, FEARG_1, NULL,
1510 ret_string, f_string},
1511 {"strlen", 1, 1, FEARG_1, NULL,
1512 ret_number, f_strlen},
1513 {"strpart", 2, 4, FEARG_1, NULL,
1514 ret_string, f_strpart},
1515 {"strptime", 2, 2, FEARG_1, NULL,
1516 ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +01001517#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001518 f_strptime
1519#else
1520 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +01001521#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001522 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001523 {"strridx", 2, 3, FEARG_1, NULL,
1524 ret_number, f_strridx},
1525 {"strtrans", 1, 1, FEARG_1, NULL,
1526 ret_string, f_strtrans},
1527 {"strwidth", 1, 1, FEARG_1, NULL,
1528 ret_number, f_strwidth},
1529 {"submatch", 1, 2, FEARG_1, NULL,
1530 ret_string, f_submatch},
1531 {"substitute", 4, 4, FEARG_1, NULL,
1532 ret_string, f_substitute},
1533 {"swapinfo", 1, 1, FEARG_1, NULL,
1534 ret_dict_any, f_swapinfo},
1535 {"swapname", 1, 1, FEARG_1, NULL,
1536 ret_string, f_swapname},
1537 {"synID", 3, 3, 0, NULL,
1538 ret_number, f_synID},
1539 {"synIDattr", 2, 3, FEARG_1, NULL,
1540 ret_string, f_synIDattr},
1541 {"synIDtrans", 1, 1, FEARG_1, NULL,
1542 ret_number, f_synIDtrans},
1543 {"synconcealed", 2, 2, 0, NULL,
1544 ret_list_any, f_synconcealed},
1545 {"synstack", 2, 2, 0, NULL,
1546 ret_list_number, f_synstack},
1547 {"system", 1, 2, FEARG_1, NULL,
1548 ret_string, f_system},
1549 {"systemlist", 1, 2, FEARG_1, NULL,
1550 ret_list_string, f_systemlist},
1551 {"tabpagebuflist", 0, 1, FEARG_1, NULL,
1552 ret_list_number, f_tabpagebuflist},
1553 {"tabpagenr", 0, 1, 0, NULL,
1554 ret_number, f_tabpagenr},
1555 {"tabpagewinnr", 1, 2, FEARG_1, NULL,
1556 ret_number, f_tabpagewinnr},
1557 {"tagfiles", 0, 0, 0, NULL,
1558 ret_list_string, f_tagfiles},
1559 {"taglist", 1, 2, FEARG_1, NULL,
1560 ret_list_dict_any, f_taglist},
1561 {"tan", 1, 1, FEARG_1, NULL,
1562 ret_float, FLOAT_FUNC(f_tan)},
1563 {"tanh", 1, 1, FEARG_1, NULL,
1564 ret_float, FLOAT_FUNC(f_tanh)},
1565 {"tempname", 0, 0, 0, NULL,
1566 ret_string, f_tempname},
1567 {"term_dumpdiff", 2, 3, FEARG_1, NULL,
1568 ret_number, TERM_FUNC(f_term_dumpdiff)},
1569 {"term_dumpload", 1, 2, FEARG_1, NULL,
1570 ret_number, TERM_FUNC(f_term_dumpload)},
1571 {"term_dumpwrite", 2, 3, FEARG_2, NULL,
1572 ret_void, TERM_FUNC(f_term_dumpwrite)},
1573 {"term_getaltscreen", 1, 1, FEARG_1, NULL,
1574 ret_number, TERM_FUNC(f_term_getaltscreen)},
1575 {"term_getansicolors", 1, 1, FEARG_1, NULL,
1576 ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001577#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001578 f_term_getansicolors
1579#else
1580 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001581#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001582 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001583 {"term_getattr", 2, 2, FEARG_1, NULL,
1584 ret_number, TERM_FUNC(f_term_getattr)},
1585 {"term_getcursor", 1, 1, FEARG_1, NULL,
1586 ret_list_any, TERM_FUNC(f_term_getcursor)},
1587 {"term_getjob", 1, 1, FEARG_1, NULL,
1588 ret_job, TERM_FUNC(f_term_getjob)},
1589 {"term_getline", 2, 2, FEARG_1, NULL,
1590 ret_string, TERM_FUNC(f_term_getline)},
1591 {"term_getscrolled", 1, 1, FEARG_1, NULL,
1592 ret_number, TERM_FUNC(f_term_getscrolled)},
1593 {"term_getsize", 1, 1, FEARG_1, NULL,
1594 ret_list_number, TERM_FUNC(f_term_getsize)},
1595 {"term_getstatus", 1, 1, FEARG_1, NULL,
1596 ret_string, TERM_FUNC(f_term_getstatus)},
1597 {"term_gettitle", 1, 1, FEARG_1, NULL,
1598 ret_string, TERM_FUNC(f_term_gettitle)},
1599 {"term_gettty", 1, 2, FEARG_1, NULL,
1600 ret_string, TERM_FUNC(f_term_gettty)},
1601 {"term_list", 0, 0, 0, NULL,
1602 ret_list_number, TERM_FUNC(f_term_list)},
1603 {"term_scrape", 2, 2, FEARG_1, NULL,
1604 ret_list_dict_any, TERM_FUNC(f_term_scrape)},
1605 {"term_sendkeys", 2, 2, FEARG_1, NULL,
1606 ret_void, TERM_FUNC(f_term_sendkeys)},
1607 {"term_setansicolors", 2, 2, FEARG_1, NULL,
1608 ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001609#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001610 f_term_setansicolors
1611#else
1612 NULL
1613#endif
1614 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001615 {"term_setapi", 2, 2, FEARG_1, NULL,
1616 ret_void, TERM_FUNC(f_term_setapi)},
1617 {"term_setkill", 2, 2, FEARG_1, NULL,
1618 ret_void, TERM_FUNC(f_term_setkill)},
1619 {"term_setrestore", 2, 2, FEARG_1, NULL,
1620 ret_void, TERM_FUNC(f_term_setrestore)},
1621 {"term_setsize", 3, 3, FEARG_1, NULL,
1622 ret_void, TERM_FUNC(f_term_setsize)},
1623 {"term_start", 1, 2, FEARG_1, NULL,
1624 ret_number, TERM_FUNC(f_term_start)},
1625 {"term_wait", 1, 2, FEARG_1, NULL,
1626 ret_void, TERM_FUNC(f_term_wait)},
1627 {"terminalprops", 0, 0, 0, NULL,
1628 ret_dict_string, f_terminalprops},
1629 {"test_alloc_fail", 3, 3, FEARG_1, NULL,
1630 ret_void, f_test_alloc_fail},
1631 {"test_autochdir", 0, 0, 0, NULL,
1632 ret_void, f_test_autochdir},
1633 {"test_feedinput", 1, 1, FEARG_1, NULL,
1634 ret_void, f_test_feedinput},
1635 {"test_garbagecollect_now", 0, 0, 0, NULL,
1636 ret_void, f_test_garbagecollect_now},
1637 {"test_garbagecollect_soon", 0, 0, 0, NULL,
1638 ret_void, f_test_garbagecollect_soon},
1639 {"test_getvalue", 1, 1, FEARG_1, NULL,
1640 ret_number, f_test_getvalue},
1641 {"test_ignore_error", 1, 1, FEARG_1, NULL,
1642 ret_void, f_test_ignore_error},
1643 {"test_null_blob", 0, 0, 0, NULL,
1644 ret_blob, f_test_null_blob},
1645 {"test_null_channel", 0, 0, 0, NULL,
1646 ret_channel, JOB_FUNC(f_test_null_channel)},
1647 {"test_null_dict", 0, 0, 0, NULL,
1648 ret_dict_any, f_test_null_dict},
1649 {"test_null_function", 0, 0, 0, NULL,
1650 ret_func_any, f_test_null_function},
1651 {"test_null_job", 0, 0, 0, NULL,
1652 ret_job, JOB_FUNC(f_test_null_job)},
1653 {"test_null_list", 0, 0, 0, NULL,
1654 ret_list_any, f_test_null_list},
1655 {"test_null_partial", 0, 0, 0, NULL,
1656 ret_func_any, f_test_null_partial},
1657 {"test_null_string", 0, 0, 0, NULL,
1658 ret_string, f_test_null_string},
1659 {"test_option_not_set", 1, 1, FEARG_1, NULL,
1660 ret_void, f_test_option_not_set},
1661 {"test_override", 2, 2, FEARG_2, NULL,
1662 ret_void, f_test_override},
1663 {"test_refcount", 1, 1, FEARG_1, NULL,
1664 ret_number, f_test_refcount},
1665 {"test_scrollbar", 3, 3, FEARG_2, NULL,
1666 ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +02001667#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +01001668 f_test_scrollbar
1669#else
1670 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +02001671#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001672 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001673 {"test_setmouse", 2, 2, 0, NULL,
1674 ret_void, f_test_setmouse},
1675 {"test_settime", 1, 1, FEARG_1, NULL,
1676 ret_void, f_test_settime},
1677 {"test_srand_seed", 0, 1, FEARG_1, NULL,
1678 ret_void, f_test_srand_seed},
1679 {"test_unknown", 0, 0, 0, NULL,
1680 ret_any, f_test_unknown},
1681 {"test_void", 0, 0, 0, NULL,
1682 ret_void, f_test_void},
1683 {"timer_info", 0, 1, FEARG_1, NULL,
1684 ret_list_dict_any, TIMER_FUNC(f_timer_info)},
1685 {"timer_pause", 2, 2, FEARG_1, NULL,
1686 ret_void, TIMER_FUNC(f_timer_pause)},
1687 {"timer_start", 2, 3, FEARG_1, NULL,
1688 ret_number, TIMER_FUNC(f_timer_start)},
1689 {"timer_stop", 1, 1, FEARG_1, NULL,
1690 ret_void, TIMER_FUNC(f_timer_stop)},
1691 {"timer_stopall", 0, 0, 0, NULL,
1692 ret_void, TIMER_FUNC(f_timer_stopall)},
1693 {"tolower", 1, 1, FEARG_1, NULL,
1694 ret_string, f_tolower},
1695 {"toupper", 1, 1, FEARG_1, NULL,
1696 ret_string, f_toupper},
1697 {"tr", 3, 3, FEARG_1, NULL,
1698 ret_string, f_tr},
1699 {"trim", 1, 3, FEARG_1, NULL,
1700 ret_string, f_trim},
1701 {"trunc", 1, 1, FEARG_1, NULL,
1702 ret_float, FLOAT_FUNC(f_trunc)},
1703 {"type", 1, 1, FEARG_1, NULL,
1704 ret_number, f_type},
1705 {"undofile", 1, 1, FEARG_1, NULL,
1706 ret_string, f_undofile},
1707 {"undotree", 0, 0, 0, NULL,
1708 ret_dict_any, f_undotree},
1709 {"uniq", 1, 3, FEARG_1, NULL,
1710 ret_list_any, f_uniq},
1711 {"values", 1, 1, FEARG_1, NULL,
1712 ret_list_any, f_values},
1713 {"virtcol", 1, 1, FEARG_1, NULL,
1714 ret_number, f_virtcol},
1715 {"visualmode", 0, 1, 0, NULL,
1716 ret_string, f_visualmode},
1717 {"wildmenumode", 0, 0, 0, NULL,
1718 ret_number, f_wildmenumode},
1719 {"win_execute", 2, 3, FEARG_2, NULL,
1720 ret_string, f_win_execute},
1721 {"win_findbuf", 1, 1, FEARG_1, NULL,
1722 ret_list_number, f_win_findbuf},
1723 {"win_getid", 0, 2, FEARG_1, NULL,
1724 ret_number, f_win_getid},
1725 {"win_gettype", 0, 1, FEARG_1, NULL,
1726 ret_string, f_win_gettype},
1727 {"win_gotoid", 1, 1, FEARG_1, NULL,
1728 ret_number, f_win_gotoid},
1729 {"win_id2tabwin", 1, 1, FEARG_1, NULL,
1730 ret_list_number, f_win_id2tabwin},
1731 {"win_id2win", 1, 1, FEARG_1, NULL,
1732 ret_number, f_win_id2win},
1733 {"win_screenpos", 1, 1, FEARG_1, NULL,
1734 ret_list_number, f_win_screenpos},
1735 {"win_splitmove", 2, 3, FEARG_1, NULL,
1736 ret_number, f_win_splitmove},
1737 {"winbufnr", 1, 1, FEARG_1, NULL,
1738 ret_number, f_winbufnr},
1739 {"wincol", 0, 0, 0, NULL,
1740 ret_number, f_wincol},
1741 {"windowsversion", 0, 0, 0, NULL,
1742 ret_string, f_windowsversion},
1743 {"winheight", 1, 1, FEARG_1, NULL,
1744 ret_number, f_winheight},
1745 {"winlayout", 0, 1, FEARG_1, NULL,
1746 ret_list_any, f_winlayout},
1747 {"winline", 0, 0, 0, NULL,
1748 ret_number, f_winline},
1749 {"winnr", 0, 1, FEARG_1, NULL,
1750 ret_number, f_winnr},
1751 {"winrestcmd", 0, 0, 0, NULL,
1752 ret_string, f_winrestcmd},
1753 {"winrestview", 1, 1, FEARG_1, NULL,
1754 ret_void, f_winrestview},
1755 {"winsaveview", 0, 0, 0, NULL,
1756 ret_dict_any, f_winsaveview},
1757 {"winwidth", 1, 1, FEARG_1, NULL,
1758 ret_number, f_winwidth},
1759 {"wordcount", 0, 0, 0, NULL,
1760 ret_dict_number, f_wordcount},
1761 {"writefile", 2, 3, FEARG_1, NULL,
1762 ret_number, f_writefile},
1763 {"xor", 2, 2, FEARG_1, NULL,
1764 ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001765};
1766
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001767/*
1768 * Function given to ExpandGeneric() to obtain the list of internal
1769 * or user defined function names.
1770 */
1771 char_u *
1772get_function_name(expand_T *xp, int idx)
1773{
1774 static int intidx = -1;
1775 char_u *name;
1776
1777 if (idx == 0)
1778 intidx = -1;
1779 if (intidx < 0)
1780 {
1781 name = get_user_func_name(xp, idx);
1782 if (name != NULL)
1783 return name;
1784 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001785 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001786 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001787 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001788 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001789 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001790 STRCAT(IObuff, ")");
1791 return IObuff;
1792 }
1793
1794 return NULL;
1795}
1796
1797/*
1798 * Function given to ExpandGeneric() to obtain the list of internal or
1799 * user defined variable or function names.
1800 */
1801 char_u *
1802get_expr_name(expand_T *xp, int idx)
1803{
1804 static int intidx = -1;
1805 char_u *name;
1806
1807 if (idx == 0)
1808 intidx = -1;
1809 if (intidx < 0)
1810 {
1811 name = get_function_name(xp, idx);
1812 if (name != NULL)
1813 return name;
1814 }
1815 return get_user_var_name(xp, ++intidx);
1816}
1817
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001818/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001819 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001820 * Return index, or -1 if not found or "implemented" is TRUE and the function
1821 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001822 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001823 static int
1824find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001825{
1826 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001827 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001828 int cmp;
1829 int x;
1830
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001831 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001832
1833 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001834 while (first <= last)
1835 {
1836 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001837 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001838 if (cmp < 0)
1839 last = x - 1;
1840 else if (cmp > 0)
1841 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001842 else if (implemented && global_functions[x].f_func == NULL)
1843 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001844 else
1845 return x;
1846 }
1847 return -1;
1848}
1849
Bram Moolenaar15c47602020-03-26 22:16:48 +01001850/*
1851 * Find internal function "name" in table "global_functions".
1852 * Return index, or -1 if not found or the function is not implemented.
1853 */
1854 int
1855find_internal_func(char_u *name)
1856{
1857 return find_internal_func_opt(name, TRUE);
1858}
1859
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001860 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001861has_internal_func(char_u *name)
1862{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001863 return find_internal_func_opt(name, TRUE) >= 0;
1864}
1865
1866 static int
1867has_internal_func_name(char_u *name)
1868{
1869 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001870}
1871
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001872 char *
1873internal_func_name(int idx)
1874{
1875 return global_functions[idx].f_name;
1876}
1877
Bram Moolenaar94738d82020-10-21 14:25:07 +02001878/*
1879 * Check the argument types for builting function "idx".
1880 * Uses the list of types on the type stack: "types".
1881 * Return FAIL and gives an error message when a type is wrong.
1882 */
1883 int
Bram Moolenaarca174532020-10-21 16:42:22 +02001884internal_func_check_arg_types(type_T **types, int idx, int argcount)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001885{
1886 argcheck_T *argchecks = global_functions[idx].f_argcheck;
1887 int i;
1888
1889 if (argchecks != NULL)
1890 {
1891 argcontext_T context;
1892
1893 context.arg_count = argcount;
Bram Moolenaarca174532020-10-21 16:42:22 +02001894 context.arg_types = types;
Bram Moolenaar94738d82020-10-21 14:25:07 +02001895 for (i = 0; i < argcount; ++i)
1896 if (argchecks[i] != NULL)
1897 {
1898 context.arg_idx = i;
Bram Moolenaarca174532020-10-21 16:42:22 +02001899 if (argchecks[i](types[i], &context) == FAIL)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001900 return FAIL;
1901 }
1902 }
1903 return OK;
1904}
1905
Bram Moolenaara1224cb2020-10-22 12:31:49 +02001906/*
1907 * Call the "f_retfunc" function to obtain the return type of function "idx".
1908 * "argtypes" is the list of argument types or NULL when there are no
1909 * arguments.
1910 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001911 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001912internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001913{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001914 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001915}
1916
1917/*
1918 * Check the argument count to use for internal function "idx".
Bram Moolenaar389df252020-07-09 21:20:47 +02001919 * Returns -1 for failure, 0 if no method base accepted, 1 if method base is
1920 * first argument, 2 if method base is second argument, etc. 9 if method base
1921 * is last argument.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001922 */
1923 int
1924check_internal_func(int idx, int argcount)
1925{
1926 int res;
1927 char *name;
1928
1929 if (argcount < global_functions[idx].f_min_argc)
1930 res = FCERR_TOOFEW;
1931 else if (argcount > global_functions[idx].f_max_argc)
1932 res = FCERR_TOOMANY;
1933 else
Bram Moolenaar389df252020-07-09 21:20:47 +02001934 return global_functions[idx].f_argtype;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001935
1936 name = internal_func_name(idx);
1937 if (res == FCERR_TOOMANY)
1938 semsg(_(e_toomanyarg), name);
1939 else
1940 semsg(_(e_toofewarg), name);
Bram Moolenaar389df252020-07-09 21:20:47 +02001941 return -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001942}
1943
Bram Moolenaarac92e252019-08-03 21:58:38 +02001944 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001945call_internal_func(
1946 char_u *name,
1947 int argcount,
1948 typval_T *argvars,
1949 typval_T *rettv)
1950{
1951 int i;
1952
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001953 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001954 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001955 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001956 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001957 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001958 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001959 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001960 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001961 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001962 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001963}
1964
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001965 void
1966call_internal_func_by_idx(
1967 int idx,
1968 typval_T *argvars,
1969 typval_T *rettv)
1970{
1971 global_functions[idx].f_func(argvars, rettv);
1972}
1973
Bram Moolenaarac92e252019-08-03 21:58:38 +02001974/*
1975 * Invoke a method for base->method().
1976 */
1977 int
1978call_internal_method(
1979 char_u *name,
1980 int argcount,
1981 typval_T *argvars,
1982 typval_T *rettv,
1983 typval_T *basetv)
1984{
1985 int i;
1986 int fi;
1987 typval_T argv[MAX_FUNC_ARGS + 1];
1988
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001989 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02001990 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001991 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02001992 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001993 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001994 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001995 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001996 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001997 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001998
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001999 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002000 {
2001 // base value goes last
2002 for (i = 0; i < argcount; ++i)
2003 argv[i] = argvars[i];
2004 argv[argcount] = *basetv;
2005 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002006 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02002007 {
2008 // base value goes second
2009 argv[0] = argvars[0];
2010 argv[1] = *basetv;
2011 for (i = 1; i < argcount; ++i)
2012 argv[i + 1] = argvars[i];
2013 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02002014 else if (global_functions[fi].f_argtype == FEARG_3)
2015 {
2016 // base value goes third
2017 argv[0] = argvars[0];
2018 argv[1] = argvars[1];
2019 argv[2] = *basetv;
2020 for (i = 2; i < argcount; ++i)
2021 argv[i + 1] = argvars[i];
2022 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02002023 else if (global_functions[fi].f_argtype == FEARG_4)
2024 {
2025 // base value goes fourth
2026 argv[0] = argvars[0];
2027 argv[1] = argvars[1];
2028 argv[2] = argvars[2];
2029 argv[3] = *basetv;
2030 for (i = 3; i < argcount; ++i)
2031 argv[i + 1] = argvars[i];
2032 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02002033 else
2034 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002035 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02002036 argv[0] = *basetv;
2037 for (i = 0; i < argcount; ++i)
2038 argv[i + 1] = argvars[i];
2039 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02002040 argv[argcount + 1].v_type = VAR_UNKNOWN;
2041
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02002042 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01002043 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002044}
2045
2046/*
2047 * Return TRUE for a non-zero Number and a non-empty String.
2048 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02002049 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002050non_zero_arg(typval_T *argvars)
2051{
2052 return ((argvars[0].v_type == VAR_NUMBER
2053 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002054 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002055 && argvars[0].vval.v_number == VVAL_TRUE)
2056 || (argvars[0].v_type == VAR_STRING
2057 && argvars[0].vval.v_string != NULL
2058 && *argvars[0].vval.v_string != NUL));
2059}
2060
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002061#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002062/*
2063 * Get the float value of "argvars[0]" into "f".
2064 * Returns FAIL when the argument is not a Number or Float.
2065 */
2066 static int
2067get_float_arg(typval_T *argvars, float_T *f)
2068{
2069 if (argvars[0].v_type == VAR_FLOAT)
2070 {
2071 *f = argvars[0].vval.v_float;
2072 return OK;
2073 }
2074 if (argvars[0].v_type == VAR_NUMBER)
2075 {
2076 *f = (float_T)argvars[0].vval.v_number;
2077 return OK;
2078 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002079 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002080 return FAIL;
2081}
2082
2083/*
2084 * "abs(expr)" function
2085 */
2086 static void
2087f_abs(typval_T *argvars, typval_T *rettv)
2088{
2089 if (argvars[0].v_type == VAR_FLOAT)
2090 {
2091 rettv->v_type = VAR_FLOAT;
2092 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
2093 }
2094 else
2095 {
2096 varnumber_T n;
2097 int error = FALSE;
2098
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002099 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002100 if (error)
2101 rettv->vval.v_number = -1;
2102 else if (n > 0)
2103 rettv->vval.v_number = n;
2104 else
2105 rettv->vval.v_number = -n;
2106 }
2107}
2108
2109/*
2110 * "acos()" function
2111 */
2112 static void
2113f_acos(typval_T *argvars, typval_T *rettv)
2114{
2115 float_T f = 0.0;
2116
2117 rettv->v_type = VAR_FLOAT;
2118 if (get_float_arg(argvars, &f) == OK)
2119 rettv->vval.v_float = acos(f);
2120 else
2121 rettv->vval.v_float = 0.0;
2122}
2123#endif
2124
2125/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002126 * "and(expr, expr)" function
2127 */
2128 static void
2129f_and(typval_T *argvars, typval_T *rettv)
2130{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002131 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
2132 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02002133}
2134
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002135#ifdef FEAT_FLOAT
2136/*
2137 * "asin()" function
2138 */
2139 static void
2140f_asin(typval_T *argvars, typval_T *rettv)
2141{
2142 float_T f = 0.0;
2143
2144 rettv->v_type = VAR_FLOAT;
2145 if (get_float_arg(argvars, &f) == OK)
2146 rettv->vval.v_float = asin(f);
2147 else
2148 rettv->vval.v_float = 0.0;
2149}
2150
2151/*
2152 * "atan()" function
2153 */
2154 static void
2155f_atan(typval_T *argvars, typval_T *rettv)
2156{
2157 float_T f = 0.0;
2158
2159 rettv->v_type = VAR_FLOAT;
2160 if (get_float_arg(argvars, &f) == OK)
2161 rettv->vval.v_float = atan(f);
2162 else
2163 rettv->vval.v_float = 0.0;
2164}
2165
2166/*
2167 * "atan2()" function
2168 */
2169 static void
2170f_atan2(typval_T *argvars, typval_T *rettv)
2171{
2172 float_T fx = 0.0, fy = 0.0;
2173
2174 rettv->v_type = VAR_FLOAT;
2175 if (get_float_arg(argvars, &fx) == OK
2176 && get_float_arg(&argvars[1], &fy) == OK)
2177 rettv->vval.v_float = atan2(fx, fy);
2178 else
2179 rettv->vval.v_float = 0.0;
2180}
2181#endif
2182
2183/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01002184 * "balloon_show()" function
2185 */
2186#ifdef FEAT_BEVAL
2187 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002188f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
2189{
2190 rettv->v_type = VAR_STRING;
2191 if (balloonEval != NULL)
2192 {
2193 if (balloonEval->msg == NULL)
2194 rettv->vval.v_string = NULL;
2195 else
2196 rettv->vval.v_string = vim_strsave(balloonEval->msg);
2197 }
2198}
2199
2200 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01002201f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
2202{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01002203 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002204 {
2205 if (argvars[0].v_type == VAR_LIST
2206# ifdef FEAT_GUI
2207 && !gui.in_use
2208# endif
2209 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002210 {
2211 list_T *l = argvars[0].vval.v_list;
2212
2213 // empty list removes the balloon
2214 post_balloon(balloonEval, NULL,
2215 l == NULL || l->lv_len == 0 ? NULL : l);
2216 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002217 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002218 {
2219 char_u *mesg = tv_get_string_chk(&argvars[0]);
2220
2221 if (mesg != NULL)
2222 // empty string removes the balloon
2223 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
2224 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002225 }
2226}
2227
Bram Moolenaar669a8282017-11-19 20:13:05 +01002228# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002229 static void
2230f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
2231{
2232 if (rettv_list_alloc(rettv) == OK)
2233 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002234 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002235
2236 if (msg != NULL)
2237 {
2238 pumitem_T *array;
2239 int size = split_message(msg, &array);
2240 int i;
2241
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002242 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01002243 for (i = 1; i < size - 1; ++i)
2244 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01002245 while (size > 0)
2246 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002247 vim_free(array);
2248 }
2249 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01002250}
Bram Moolenaar669a8282017-11-19 20:13:05 +01002251# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01002252#endif
2253
2254/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002255 * Get the buffer from "arg" and give an error and return NULL if it is not
2256 * valid.
2257 */
Bram Moolenaara3347722019-05-11 21:14:24 +02002258 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002259get_buf_arg(typval_T *arg)
2260{
2261 buf_T *buf;
2262
2263 ++emsg_off;
2264 buf = tv_get_buf(arg, FALSE);
2265 --emsg_off;
2266 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002267 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002268 return buf;
2269}
2270
2271/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002272 * "byte2line(byte)" function
2273 */
2274 static void
2275f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
2276{
2277#ifndef FEAT_BYTEOFF
2278 rettv->vval.v_number = -1;
2279#else
2280 long boff = 0;
2281
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002282 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002283 if (boff < 0)
2284 rettv->vval.v_number = -1;
2285 else
2286 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
2287 (linenr_T)0, &boff);
2288#endif
2289}
2290
2291 static void
2292byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
2293{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002294 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002295 char_u *str;
2296 varnumber_T idx;
2297
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002298 str = tv_get_string_chk(&argvars[0]);
2299 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002300 rettv->vval.v_number = -1;
2301 if (str == NULL || idx < 0)
2302 return;
2303
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002304 t = str;
2305 for ( ; idx > 0; idx--)
2306 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002307 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002308 return;
2309 if (enc_utf8 && comp)
2310 t += utf_ptr2len(t);
2311 else
2312 t += (*mb_ptr2len)(t);
2313 }
2314 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002315}
2316
2317/*
2318 * "byteidx()" function
2319 */
2320 static void
2321f_byteidx(typval_T *argvars, typval_T *rettv)
2322{
2323 byteidx(argvars, rettv, FALSE);
2324}
2325
2326/*
2327 * "byteidxcomp()" function
2328 */
2329 static void
2330f_byteidxcomp(typval_T *argvars, typval_T *rettv)
2331{
2332 byteidx(argvars, rettv, TRUE);
2333}
2334
2335/*
2336 * "call(func, arglist [, dict])" function
2337 */
2338 static void
2339f_call(typval_T *argvars, typval_T *rettv)
2340{
2341 char_u *func;
2342 partial_T *partial = NULL;
2343 dict_T *selfdict = NULL;
2344
2345 if (argvars[1].v_type != VAR_LIST)
2346 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002347 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002348 return;
2349 }
2350 if (argvars[1].vval.v_list == NULL)
2351 return;
2352
2353 if (argvars[0].v_type == VAR_FUNC)
2354 func = argvars[0].vval.v_string;
2355 else if (argvars[0].v_type == VAR_PARTIAL)
2356 {
2357 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002358 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002359 }
2360 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002361 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002362 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002363 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002364
2365 if (argvars[2].v_type != VAR_UNKNOWN)
2366 {
2367 if (argvars[2].v_type != VAR_DICT)
2368 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002369 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002370 return;
2371 }
2372 selfdict = argvars[2].vval.v_dict;
2373 }
2374
2375 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
2376}
2377
2378#ifdef FEAT_FLOAT
2379/*
2380 * "ceil({float})" function
2381 */
2382 static void
2383f_ceil(typval_T *argvars, typval_T *rettv)
2384{
2385 float_T f = 0.0;
2386
2387 rettv->v_type = VAR_FLOAT;
2388 if (get_float_arg(argvars, &f) == OK)
2389 rettv->vval.v_float = ceil(f);
2390 else
2391 rettv->vval.v_float = 0.0;
2392}
2393#endif
2394
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002395/*
2396 * "changenr()" function
2397 */
2398 static void
2399f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
2400{
2401 rettv->vval.v_number = curbuf->b_u_seq_cur;
2402}
2403
2404/*
2405 * "char2nr(string)" function
2406 */
2407 static void
2408f_char2nr(typval_T *argvars, typval_T *rettv)
2409{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002410 if (has_mbyte)
2411 {
2412 int utf8 = 0;
2413
2414 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar24f77502020-09-04 19:50:57 +02002415 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002416
2417 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01002418 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002419 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002420 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002421 }
2422 else
Bram Moolenaar13505972019-01-24 15:04:48 +01002423 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002424}
2425
Bram Moolenaar17793ef2020-12-28 12:56:58 +01002426/*
2427 * "charidx()" function
2428 */
2429 static void
2430f_charidx(typval_T *argvars, typval_T *rettv)
2431{
2432 char_u *str;
2433 varnumber_T idx;
2434 int countcc = FALSE;
2435 char_u *p;
2436 int len;
2437 int (*ptr2len)(char_u *);
2438
2439 rettv->vval.v_number = -1;
2440
2441 if (argvars[0].v_type != VAR_STRING || argvars[1].v_type != VAR_NUMBER
2442 || (argvars[2].v_type != VAR_UNKNOWN
2443 && argvars[2].v_type != VAR_NUMBER))
2444 {
2445 emsg(_(e_invarg));
2446 return;
2447 }
2448
2449 str = tv_get_string_chk(&argvars[0]);
2450 idx = tv_get_number_chk(&argvars[1], NULL);
2451 if (str == NULL || idx < 0)
2452 return;
2453
2454 if (argvars[2].v_type != VAR_UNKNOWN)
2455 countcc = (int)tv_get_bool(&argvars[2]);
2456 if (countcc < 0 || countcc > 1)
2457 {
2458 semsg(_(e_using_number_as_bool_nr), countcc);
2459 return;
2460 }
2461
2462 if (enc_utf8 && countcc)
2463 ptr2len = utf_ptr2len;
2464 else
2465 ptr2len = mb_ptr2len;
2466
2467 for (p = str, len = 0; p <= str + idx; len++)
2468 {
2469 if (*p == NUL)
2470 return;
2471 p += ptr2len(p);
2472 }
2473
2474 rettv->vval.v_number = len > 0 ? len - 1 : 0;
2475}
2476
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02002477 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01002478get_optional_window(typval_T *argvars, int idx)
2479{
2480 win_T *win = curwin;
2481
2482 if (argvars[idx].v_type != VAR_UNKNOWN)
2483 {
2484 win = find_win_by_nr_or_id(&argvars[idx]);
2485 if (win == NULL)
2486 {
2487 emsg(_(e_invalwindow));
2488 return NULL;
2489 }
2490 }
2491 return win;
2492}
2493
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002494/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002495 * "col(string)" function
2496 */
2497 static void
2498f_col(typval_T *argvars, typval_T *rettv)
2499{
2500 colnr_T col = 0;
2501 pos_T *fp;
2502 int fnum = curbuf->b_fnum;
2503
2504 fp = var2fpos(&argvars[0], FALSE, &fnum);
2505 if (fp != NULL && fnum == curbuf->b_fnum)
2506 {
2507 if (fp->col == MAXCOL)
2508 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002509 // '> can be MAXCOL, get the length of the line then
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002510 if (fp->lnum <= curbuf->b_ml.ml_line_count)
2511 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
2512 else
2513 col = MAXCOL;
2514 }
2515 else
2516 {
2517 col = fp->col + 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002518 // col(".") when the cursor is on the NUL at the end of the line
2519 // because of "coladd" can be seen as an extra column.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002520 if (virtual_active() && fp == &curwin->w_cursor)
2521 {
2522 char_u *p = ml_get_cursor();
2523
2524 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
2525 curwin->w_virtcol - curwin->w_cursor.coladd))
2526 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002527 int l;
2528
2529 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
2530 col += l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002531 }
2532 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002533 }
2534 }
2535 rettv->vval.v_number = col;
2536}
2537
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002538/*
2539 * "confirm(message, buttons[, default [, type]])" function
2540 */
2541 static void
2542f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2543{
2544#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2545 char_u *message;
2546 char_u *buttons = NULL;
2547 char_u buf[NUMBUFLEN];
2548 char_u buf2[NUMBUFLEN];
2549 int def = 1;
2550 int type = VIM_GENERIC;
2551 char_u *typestr;
2552 int error = FALSE;
2553
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002554 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002555 if (message == NULL)
2556 error = TRUE;
2557 if (argvars[1].v_type != VAR_UNKNOWN)
2558 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002559 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002560 if (buttons == NULL)
2561 error = TRUE;
2562 if (argvars[2].v_type != VAR_UNKNOWN)
2563 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002564 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002565 if (argvars[3].v_type != VAR_UNKNOWN)
2566 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002567 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002568 if (typestr == NULL)
2569 error = TRUE;
2570 else
2571 {
2572 switch (TOUPPER_ASC(*typestr))
2573 {
2574 case 'E': type = VIM_ERROR; break;
2575 case 'Q': type = VIM_QUESTION; break;
2576 case 'I': type = VIM_INFO; break;
2577 case 'W': type = VIM_WARNING; break;
2578 case 'G': type = VIM_GENERIC; break;
2579 }
2580 }
2581 }
2582 }
2583 }
2584
2585 if (buttons == NULL || *buttons == NUL)
2586 buttons = (char_u *)_("&Ok");
2587
2588 if (!error)
2589 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
2590 def, NULL, FALSE);
2591#endif
2592}
2593
2594/*
2595 * "copy()" function
2596 */
2597 static void
2598f_copy(typval_T *argvars, typval_T *rettv)
2599{
2600 item_copy(&argvars[0], rettv, FALSE, 0);
2601}
2602
2603#ifdef FEAT_FLOAT
2604/*
2605 * "cos()" function
2606 */
2607 static void
2608f_cos(typval_T *argvars, typval_T *rettv)
2609{
2610 float_T f = 0.0;
2611
2612 rettv->v_type = VAR_FLOAT;
2613 if (get_float_arg(argvars, &f) == OK)
2614 rettv->vval.v_float = cos(f);
2615 else
2616 rettv->vval.v_float = 0.0;
2617}
2618
2619/*
2620 * "cosh()" function
2621 */
2622 static void
2623f_cosh(typval_T *argvars, typval_T *rettv)
2624{
2625 float_T f = 0.0;
2626
2627 rettv->v_type = VAR_FLOAT;
2628 if (get_float_arg(argvars, &f) == OK)
2629 rettv->vval.v_float = cosh(f);
2630 else
2631 rettv->vval.v_float = 0.0;
2632}
2633#endif
2634
2635/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002636 * "cursor(lnum, col)" function, or
2637 * "cursor(list)"
2638 *
2639 * Moves the cursor to the specified line and column.
2640 * Returns 0 when the position could be set, -1 otherwise.
2641 */
2642 static void
2643f_cursor(typval_T *argvars, typval_T *rettv)
2644{
2645 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002646 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002647 int set_curswant = TRUE;
2648
2649 rettv->vval.v_number = -1;
2650 if (argvars[1].v_type == VAR_UNKNOWN)
2651 {
2652 pos_T pos;
2653 colnr_T curswant = -1;
2654
2655 if (list2fpos(argvars, &pos, NULL, &curswant) == FAIL)
2656 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002657 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002658 return;
2659 }
2660 line = pos.lnum;
2661 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002662 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002663 if (curswant >= 0)
2664 {
2665 curwin->w_curswant = curswant - 1;
2666 set_curswant = FALSE;
2667 }
2668 }
2669 else
2670 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002671 line = tv_get_lnum(argvars);
Bram Moolenaar9a963372020-12-21 21:58:46 +01002672 if (line < 0)
2673 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002674 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002675 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002676 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002677 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002678 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002679 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002680 if (line > 0)
2681 curwin->w_cursor.lnum = line;
2682 if (col > 0)
2683 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002684 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002685
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002686 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002687 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002688 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002689 if (has_mbyte)
2690 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002691
2692 curwin->w_set_curswant = set_curswant;
2693 rettv->vval.v_number = 0;
2694}
2695
Bram Moolenaar4f974752019-02-17 17:44:42 +01002696#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002697/*
2698 * "debugbreak()" function
2699 */
2700 static void
2701f_debugbreak(typval_T *argvars, typval_T *rettv)
2702{
2703 int pid;
2704
2705 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002706 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002707 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002708 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002709 else
2710 {
2711 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
2712
2713 if (hProcess != NULL)
2714 {
2715 DebugBreakProcess(hProcess);
2716 CloseHandle(hProcess);
2717 rettv->vval.v_number = OK;
2718 }
2719 }
2720}
2721#endif
2722
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002723/*
2724 * "deepcopy()" function
2725 */
2726 static void
2727f_deepcopy(typval_T *argvars, typval_T *rettv)
2728{
2729 int noref = 0;
2730 int copyID;
2731
2732 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar44b4a242020-09-05 17:18:28 +02002733 noref = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002734 if (noref < 0 || noref > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02002735 semsg(_(e_using_number_as_bool_nr), noref);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002736 else
2737 {
2738 copyID = get_copyID();
2739 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
2740 }
2741}
2742
2743/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002744 * "did_filetype()" function
2745 */
2746 static void
2747f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2748{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002749 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002750}
2751
2752/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01002753 * "echoraw({expr})" function
2754 */
2755 static void
2756f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
2757{
2758 char_u *str = tv_get_string_chk(&argvars[0]);
2759
2760 if (str != NULL && *str != NUL)
2761 {
2762 out_str(str);
2763 out_flush();
2764 }
2765}
2766
2767/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002768 * "empty({expr})" function
2769 */
2770 static void
2771f_empty(typval_T *argvars, typval_T *rettv)
2772{
2773 int n = FALSE;
2774
2775 switch (argvars[0].v_type)
2776 {
2777 case VAR_STRING:
2778 case VAR_FUNC:
2779 n = argvars[0].vval.v_string == NULL
2780 || *argvars[0].vval.v_string == NUL;
2781 break;
2782 case VAR_PARTIAL:
2783 n = FALSE;
2784 break;
2785 case VAR_NUMBER:
2786 n = argvars[0].vval.v_number == 0;
2787 break;
2788 case VAR_FLOAT:
2789#ifdef FEAT_FLOAT
2790 n = argvars[0].vval.v_float == 0.0;
2791 break;
2792#endif
2793 case VAR_LIST:
2794 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002795 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002796 break;
2797 case VAR_DICT:
2798 n = argvars[0].vval.v_dict == NULL
2799 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2800 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002801 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002802 case VAR_SPECIAL:
2803 n = argvars[0].vval.v_number != VVAL_TRUE;
2804 break;
2805
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002806 case VAR_BLOB:
2807 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002808 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2809 break;
2810
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002811 case VAR_JOB:
2812#ifdef FEAT_JOB_CHANNEL
2813 n = argvars[0].vval.v_job == NULL
2814 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2815 break;
2816#endif
2817 case VAR_CHANNEL:
2818#ifdef FEAT_JOB_CHANNEL
2819 n = argvars[0].vval.v_channel == NULL
2820 || !channel_is_open(argvars[0].vval.v_channel);
2821 break;
2822#endif
2823 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02002824 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002825 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002826 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002827 n = TRUE;
2828 break;
2829 }
2830
2831 rettv->vval.v_number = n;
2832}
2833
2834/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002835 * "environ()" function
2836 */
2837 static void
2838f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2839{
2840#if !defined(AMIGA)
2841 int i = 0;
2842 char_u *entry, *value;
2843# ifdef MSWIN
2844 extern wchar_t **_wenviron;
2845# else
2846 extern char **environ;
2847# endif
2848
2849 if (rettv_dict_alloc(rettv) != OK)
2850 return;
2851
2852# ifdef MSWIN
2853 if (*_wenviron == NULL)
2854 return;
2855# else
2856 if (*environ == NULL)
2857 return;
2858# endif
2859
2860 for (i = 0; ; ++i)
2861 {
2862# ifdef MSWIN
2863 short_u *p;
2864
2865 if ((p = (short_u *)_wenviron[i]) == NULL)
2866 return;
2867 entry = utf16_to_enc(p, NULL);
2868# else
2869 if ((entry = (char_u *)environ[i]) == NULL)
2870 return;
2871 entry = vim_strsave(entry);
2872# endif
2873 if (entry == NULL) // out of memory
2874 return;
2875 if ((value = vim_strchr(entry, '=')) == NULL)
2876 {
2877 vim_free(entry);
2878 continue;
2879 }
2880 *value++ = NUL;
2881 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2882 vim_free(entry);
2883 }
2884#endif
2885}
2886
2887/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002888 * "escape({string}, {chars})" function
2889 */
2890 static void
2891f_escape(typval_T *argvars, typval_T *rettv)
2892{
2893 char_u buf[NUMBUFLEN];
2894
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002895 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2896 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002897 rettv->v_type = VAR_STRING;
2898}
2899
2900/*
2901 * "eval()" function
2902 */
2903 static void
2904f_eval(typval_T *argvars, typval_T *rettv)
2905{
2906 char_u *s, *p;
2907
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002908 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002909 if (s != NULL)
2910 s = skipwhite(s);
2911
2912 p = s;
Bram Moolenaar5409f5d2020-06-24 18:37:35 +02002913 if (s == NULL || eval1(&s, rettv, &EVALARG_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002914 {
2915 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002916 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002917 need_clr_eos = FALSE;
2918 rettv->v_type = VAR_NUMBER;
2919 rettv->vval.v_number = 0;
2920 }
2921 else if (*s != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02002922 semsg(_(e_trailing_arg), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002923}
2924
2925/*
2926 * "eventhandler()" function
2927 */
2928 static void
2929f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2930{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002931 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002932}
2933
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002934static garray_T redir_execute_ga;
2935
2936/*
2937 * Append "value[value_len]" to the execute() output.
2938 */
2939 void
2940execute_redir_str(char_u *value, int value_len)
2941{
2942 int len;
2943
2944 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002945 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002946 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002947 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002948 if (ga_grow(&redir_execute_ga, len) == OK)
2949 {
2950 mch_memmove((char *)redir_execute_ga.ga_data
2951 + redir_execute_ga.ga_len, value, len);
2952 redir_execute_ga.ga_len += len;
2953 }
2954}
2955
2956/*
2957 * Get next line from a list.
2958 * Called by do_cmdline() to get the next line.
2959 * Returns allocated string, or NULL for end of function.
2960 */
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002961 static char_u *
2962get_list_line(
2963 int c UNUSED,
2964 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002965 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +02002966 getline_opt_T options UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002967{
2968 listitem_T **p = (listitem_T **)cookie;
2969 listitem_T *item = *p;
2970 char_u buf[NUMBUFLEN];
2971 char_u *s;
2972
2973 if (item == NULL)
2974 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002975 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002976 *p = item->li_next;
2977 return s == NULL ? NULL : vim_strsave(s);
2978}
2979
2980/*
2981 * "execute()" function
2982 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002983 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002984execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002985{
2986 char_u *cmd = NULL;
2987 list_T *list = NULL;
2988 int save_msg_silent = msg_silent;
2989 int save_emsg_silent = emsg_silent;
2990 int save_emsg_noredir = emsg_noredir;
2991 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002992 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002993 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002994 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002995 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002996
2997 rettv->vval.v_string = NULL;
2998 rettv->v_type = VAR_STRING;
2999
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003000 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003001 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003002 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003003 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003004 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003005 return;
3006 ++list->lv_refcount;
3007 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01003008 else if (argvars[arg_off].v_type == VAR_JOB
3009 || argvars[arg_off].v_type == VAR_CHANNEL)
3010 {
3011 emsg(_(e_inval_string));
3012 return;
3013 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003014 else
3015 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003016 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003017 if (cmd == NULL)
3018 return;
3019 }
3020
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003021 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003022 {
3023 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003024 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003025
3026 if (s == NULL)
3027 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003028 if (*s == NUL)
3029 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003030 if (STRNCMP(s, "silent", 6) == 0)
3031 ++msg_silent;
3032 if (STRCMP(s, "silent!") == 0)
3033 {
3034 emsg_silent = TRUE;
3035 emsg_noredir = TRUE;
3036 }
3037 }
3038 else
3039 ++msg_silent;
3040
3041 if (redir_execute)
3042 save_ga = redir_execute_ga;
3043 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
3044 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01003045 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003046 if (!echo_output)
3047 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003048
3049 if (cmd != NULL)
3050 do_cmdline_cmd(cmd);
3051 else
3052 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003053 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003054
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003055 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003056 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003057 do_cmdline(NULL, get_list_line, (void *)&item,
3058 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
3059 --list->lv_refcount;
3060 }
3061
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003062 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01003063 if (ga_grow(&redir_execute_ga, 1) == OK)
3064 {
3065 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
3066 rettv->vval.v_string = redir_execute_ga.ga_data;
3067 }
3068 else
3069 {
3070 ga_clear(&redir_execute_ga);
3071 rettv->vval.v_string = NULL;
3072 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003073 msg_silent = save_msg_silent;
3074 emsg_silent = save_emsg_silent;
3075 emsg_noredir = save_emsg_noredir;
3076
3077 redir_execute = save_redir_execute;
3078 if (redir_execute)
3079 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01003080 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003081
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01003082 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01003083 if (echo_output)
3084 // When not working silently: put it in column zero. A following
3085 // "echon" will overwrite the message, unavoidably.
3086 msg_col = 0;
3087 else
3088 // When working silently: Put it back where it was, since nothing
3089 // should have been written.
3090 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003091}
3092
3093/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02003094 * "execute()" function
3095 */
3096 static void
3097f_execute(typval_T *argvars, typval_T *rettv)
3098{
3099 execute_common(argvars, rettv, 0);
3100}
3101
3102/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003103 * "exists()" function
3104 */
3105 static void
3106f_exists(typval_T *argvars, typval_T *rettv)
3107{
3108 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003109 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003110
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003111 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003112 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003113 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003114 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003115 if (mch_getenv(p + 1) != NULL)
3116 n = TRUE;
3117 else
3118 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003119 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003120 p = expand_env_save(p);
3121 if (p != NULL && *p != '$')
3122 n = TRUE;
3123 vim_free(p);
3124 }
3125 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003126 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003127 {
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +02003128 n = (eval_option(&p, NULL, TRUE) == OK);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003129 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003130 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003131 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003132 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003133 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02003134 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003135 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01003136 else if (*p == '?') // internal function only
3137 {
3138 n = has_internal_func_name(p + 1);
3139 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003140 else if (*p == ':')
3141 {
3142 n = cmd_exists(p + 1);
3143 }
3144 else if (*p == '#')
3145 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003146 if (p[1] == '#')
3147 n = autocmd_supported(p + 2);
3148 else
3149 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003150 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003151 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003152 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01003153 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003154 }
3155
3156 rettv->vval.v_number = n;
3157}
3158
3159#ifdef FEAT_FLOAT
3160/*
3161 * "exp()" function
3162 */
3163 static void
3164f_exp(typval_T *argvars, typval_T *rettv)
3165{
3166 float_T f = 0.0;
3167
3168 rettv->v_type = VAR_FLOAT;
3169 if (get_float_arg(argvars, &f) == OK)
3170 rettv->vval.v_float = exp(f);
3171 else
3172 rettv->vval.v_float = 0.0;
3173}
3174#endif
3175
3176/*
3177 * "expand()" function
3178 */
3179 static void
3180f_expand(typval_T *argvars, typval_T *rettv)
3181{
3182 char_u *s;
3183 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003184 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003185 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
3186 expand_T xpc;
3187 int error = FALSE;
3188 char_u *result;
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003189#ifdef BACKSLASH_IN_FILENAME
3190 char_u *p_csl_save = p_csl;
3191
3192 // avoid using 'completeslash' here
3193 p_csl = empty_option;
3194#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003195
3196 rettv->v_type = VAR_STRING;
3197 if (argvars[1].v_type != VAR_UNKNOWN
3198 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003199 && tv_get_bool_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003200 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02003201 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003202
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003203 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003204 if (*s == '%' || *s == '#' || *s == '<')
3205 {
3206 ++emsg_off;
3207 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
3208 --emsg_off;
3209 if (rettv->v_type == VAR_LIST)
3210 {
3211 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
3212 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02003213 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003214 }
3215 else
3216 rettv->vval.v_string = result;
3217 }
3218 else
3219 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003220 // When the optional second argument is non-zero, don't remove matches
3221 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003222 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003223 && tv_get_bool_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003224 options |= WILD_KEEP_ALL;
3225 if (!error)
3226 {
3227 ExpandInit(&xpc);
3228 xpc.xp_context = EXPAND_FILES;
3229 if (p_wic)
3230 options += WILD_ICASE;
3231 if (rettv->v_type == VAR_STRING)
3232 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
3233 options, WILD_ALL);
3234 else if (rettv_list_alloc(rettv) != FAIL)
3235 {
3236 int i;
3237
3238 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
3239 for (i = 0; i < xpc.xp_numfiles; i++)
3240 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
3241 ExpandCleanup(&xpc);
3242 }
3243 }
3244 else
3245 rettv->vval.v_string = NULL;
3246 }
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003247#ifdef BACKSLASH_IN_FILENAME
3248 p_csl = p_csl_save;
3249#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003250}
3251
3252/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02003253 * "expandcmd()" function
3254 * Expand all the special characters in a command string.
3255 */
3256 static void
3257f_expandcmd(typval_T *argvars, typval_T *rettv)
3258{
3259 exarg_T eap;
3260 char_u *cmdstr;
3261 char *errormsg = NULL;
3262
3263 rettv->v_type = VAR_STRING;
3264 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
3265
3266 memset(&eap, 0, sizeof(eap));
3267 eap.cmd = cmdstr;
3268 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02003269 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02003270 eap.usefilter = FALSE;
3271 eap.nextcmd = NULL;
3272 eap.cmdidx = CMD_USER;
3273
3274 expand_filename(&eap, &cmdstr, &errormsg);
3275 if (errormsg != NULL && *errormsg != NUL)
3276 emsg(errormsg);
3277
3278 rettv->vval.v_string = cmdstr;
3279}
3280
3281/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003282 * "feedkeys()" function
3283 */
3284 static void
3285f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
3286{
3287 int remap = TRUE;
3288 int insert = FALSE;
3289 char_u *keys, *flags;
3290 char_u nbuf[NUMBUFLEN];
3291 int typed = FALSE;
3292 int execute = FALSE;
3293 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003294 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003295 char_u *keys_esc;
3296
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003297 // This is not allowed in the sandbox. If the commands would still be
3298 // executed in the sandbox it would be OK, but it probably happens later,
3299 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003300 if (check_secure())
3301 return;
3302
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003303 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003304
3305 if (argvars[1].v_type != VAR_UNKNOWN)
3306 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003307 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003308 for ( ; *flags != NUL; ++flags)
3309 {
3310 switch (*flags)
3311 {
3312 case 'n': remap = FALSE; break;
3313 case 'm': remap = TRUE; break;
3314 case 't': typed = TRUE; break;
3315 case 'i': insert = TRUE; break;
3316 case 'x': execute = TRUE; break;
3317 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003318 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003319 }
3320 }
3321 }
3322
3323 if (*keys != NUL || execute)
3324 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003325 // Need to escape K_SPECIAL and CSI before putting the string in the
3326 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003327 keys_esc = vim_strsave_escape_csi(keys);
3328 if (keys_esc != NULL)
3329 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003330 if (lowlevel)
3331 {
3332#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01003333 int idx;
3334 int len = (int)STRLEN(keys);
3335
3336 for (idx = 0; idx < len; ++idx)
3337 {
3338 // if a CTRL-C was typed, set got_int, similar to what
3339 // happens in fill_input_buf()
3340 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
3341 got_int = TRUE;
3342 add_to_input_buf(keys + idx, 1);
3343 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003344#else
3345 emsg(_("E980: lowlevel input not supported"));
3346#endif
3347 }
3348 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003349 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003350 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003351 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003352 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003353#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003354 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003355#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003356 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003357 typebuf_was_filled = TRUE;
3358 }
3359 vim_free(keys_esc);
3360
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003361 if (execute)
3362 {
3363 int save_msg_scroll = msg_scroll;
3364
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003365 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003366 msg_scroll = FALSE;
3367
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003368 if (!dangerous)
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003369 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003370 ++ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003371 ++in_feedkeys;
3372 }
Bram Moolenaar905dd902019-04-07 14:21:47 +02003373 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003374 if (!dangerous)
Bram Moolenaar189832b2020-09-23 12:29:11 +02003375 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003376 --ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003377 --in_feedkeys;
Bram Moolenaar189832b2020-09-23 12:29:11 +02003378 }
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003379
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003380 msg_scroll |= save_msg_scroll;
3381 }
3382 }
3383 }
3384}
3385
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003386#ifdef FEAT_FLOAT
3387/*
3388 * "float2nr({float})" function
3389 */
3390 static void
3391f_float2nr(typval_T *argvars, typval_T *rettv)
3392{
3393 float_T f = 0.0;
3394
3395 if (get_float_arg(argvars, &f) == OK)
3396 {
Bram Moolenaar37184272020-05-23 19:30:05 +02003397 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003398 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02003399 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003400 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003401 else
3402 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003403 }
3404}
3405
3406/*
3407 * "floor({float})" function
3408 */
3409 static void
3410f_floor(typval_T *argvars, typval_T *rettv)
3411{
3412 float_T f = 0.0;
3413
3414 rettv->v_type = VAR_FLOAT;
3415 if (get_float_arg(argvars, &f) == OK)
3416 rettv->vval.v_float = floor(f);
3417 else
3418 rettv->vval.v_float = 0.0;
3419}
3420
3421/*
3422 * "fmod()" function
3423 */
3424 static void
3425f_fmod(typval_T *argvars, typval_T *rettv)
3426{
3427 float_T fx = 0.0, fy = 0.0;
3428
3429 rettv->v_type = VAR_FLOAT;
3430 if (get_float_arg(argvars, &fx) == OK
3431 && get_float_arg(&argvars[1], &fy) == OK)
3432 rettv->vval.v_float = fmod(fx, fy);
3433 else
3434 rettv->vval.v_float = 0.0;
3435}
3436#endif
3437
3438/*
3439 * "fnameescape({string})" function
3440 */
3441 static void
3442f_fnameescape(typval_T *argvars, typval_T *rettv)
3443{
3444 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003445 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003446 rettv->v_type = VAR_STRING;
3447}
3448
3449/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003450 * "foreground()" function
3451 */
3452 static void
3453f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3454{
3455#ifdef FEAT_GUI
3456 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003457 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003458 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003459 return;
3460 }
3461#endif
3462#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003463 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003464#endif
3465}
3466
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003467 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003468common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003469{
3470 char_u *s;
3471 char_u *name;
3472 int use_string = FALSE;
3473 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003474 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003475 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003476
3477 if (argvars[0].v_type == VAR_FUNC)
3478 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003479 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003480 s = argvars[0].vval.v_string;
3481 }
3482 else if (argvars[0].v_type == VAR_PARTIAL
3483 && argvars[0].vval.v_partial != NULL)
3484 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003485 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003486 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003487 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003488 }
3489 else
3490 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003491 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003492 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003493 use_string = TRUE;
3494 }
3495
Bram Moolenaar843b8842016-08-21 14:36:15 +02003496 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003497 {
3498 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003499 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar32b3f822021-01-06 21:59:39 +01003500 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF,
3501 NULL, NULL, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003502 if (*name != NUL)
3503 s = NULL;
3504 }
3505
Bram Moolenaar843b8842016-08-21 14:36:15 +02003506 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
3507 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003508 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003509 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003510 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003511 ? find_func(trans_name, is_global, NULL) == NULL
3512 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003513 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003514 else
3515 {
3516 int dict_idx = 0;
3517 int arg_idx = 0;
3518 list_T *list = NULL;
3519
3520 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
3521 {
3522 char sid_buf[25];
3523 int off = *s == 's' ? 2 : 5;
3524
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003525 // Expand s: and <SID> into <SNR>nr_, so that the function can
3526 // also be called from another script. Using trans_function_name()
3527 // would also work, but some plugins depend on the name being
3528 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02003529 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02003530 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003531 if (name != NULL)
3532 {
3533 STRCPY(name, sid_buf);
3534 STRCAT(name, s + off);
3535 }
3536 }
3537 else
3538 name = vim_strsave(s);
3539
3540 if (argvars[1].v_type != VAR_UNKNOWN)
3541 {
3542 if (argvars[2].v_type != VAR_UNKNOWN)
3543 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003544 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003545 arg_idx = 1;
3546 dict_idx = 2;
3547 }
3548 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003549 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003550 dict_idx = 1;
3551 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003552 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003553 arg_idx = 1;
3554 if (dict_idx > 0)
3555 {
3556 if (argvars[dict_idx].v_type != VAR_DICT)
3557 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003558 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003559 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003560 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003561 }
3562 if (argvars[dict_idx].vval.v_dict == NULL)
3563 dict_idx = 0;
3564 }
3565 if (arg_idx > 0)
3566 {
3567 if (argvars[arg_idx].v_type != VAR_LIST)
3568 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003569 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003570 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003571 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003572 }
3573 list = argvars[arg_idx].vval.v_list;
3574 if (list == NULL || list->lv_len == 0)
3575 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003576 else if (list->lv_len > MAX_FUNC_ARGS)
3577 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01003578 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003579 vim_free(name);
3580 goto theend;
3581 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003582 }
3583 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003584 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003585 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003586 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003587
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003588 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003589 if (pt == NULL)
3590 vim_free(name);
3591 else
3592 {
3593 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
3594 {
3595 listitem_T *li;
3596 int i = 0;
3597 int arg_len = 0;
3598 int lv_len = 0;
3599
3600 if (arg_pt != NULL)
3601 arg_len = arg_pt->pt_argc;
3602 if (list != NULL)
3603 lv_len = list->lv_len;
3604 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003605 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003606 if (pt->pt_argv == NULL)
3607 {
3608 vim_free(pt);
3609 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003610 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003611 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003612 for (i = 0; i < arg_len; i++)
3613 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
3614 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003615 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003616 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003617 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003618 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003619 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003620 }
3621
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003622 // For "function(dict.func, [], dict)" and "func" is a partial
3623 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003624 if (dict_idx > 0)
3625 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003626 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003627 pt->pt_dict = argvars[dict_idx].vval.v_dict;
3628 ++pt->pt_dict->dv_refcount;
3629 }
3630 else if (arg_pt != NULL)
3631 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003632 // If the dict was bound automatically the result is also
3633 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003634 pt->pt_dict = arg_pt->pt_dict;
3635 pt->pt_auto = arg_pt->pt_auto;
3636 if (pt->pt_dict != NULL)
3637 ++pt->pt_dict->dv_refcount;
3638 }
3639
3640 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003641 if (arg_pt != NULL && arg_pt->pt_func != NULL)
3642 {
3643 pt->pt_func = arg_pt->pt_func;
3644 func_ptr_ref(pt->pt_func);
3645 vim_free(name);
3646 }
3647 else if (is_funcref)
3648 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003649 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003650 func_ptr_ref(pt->pt_func);
3651 vim_free(name);
3652 }
3653 else
3654 {
3655 pt->pt_name = name;
3656 func_ref(name);
3657 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003658 }
3659 rettv->v_type = VAR_PARTIAL;
3660 rettv->vval.v_partial = pt;
3661 }
3662 else
3663 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003664 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003665 rettv->v_type = VAR_FUNC;
3666 rettv->vval.v_string = name;
3667 func_ref(name);
3668 }
3669 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003670theend:
3671 vim_free(trans_name);
3672}
3673
3674/*
3675 * "funcref()" function
3676 */
3677 static void
3678f_funcref(typval_T *argvars, typval_T *rettv)
3679{
3680 common_function(argvars, rettv, TRUE);
3681}
3682
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003683 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003684ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003685{
3686 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
3687 return &t_func_any;
Bram Moolenaar5e654232020-09-16 15:22:00 +02003688 return &t_func_unknown;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003689}
3690
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003691/*
3692 * "function()" function
3693 */
3694 static void
3695f_function(typval_T *argvars, typval_T *rettv)
3696{
3697 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003698}
3699
3700/*
3701 * "garbagecollect()" function
3702 */
3703 static void
3704f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
3705{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003706 // This is postponed until we are back at the toplevel, because we may be
3707 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003708 want_garbage_collect = TRUE;
3709
Bram Moolenaar2df47312020-09-05 17:30:44 +02003710 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003711 garbage_collect_at_exit = TRUE;
3712}
3713
3714/*
3715 * "get()" function
3716 */
3717 static void
3718f_get(typval_T *argvars, typval_T *rettv)
3719{
3720 listitem_T *li;
3721 list_T *l;
3722 dictitem_T *di;
3723 dict_T *d;
3724 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003725 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003726
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003727 if (argvars[0].v_type == VAR_BLOB)
3728 {
3729 int error = FALSE;
3730 int idx = tv_get_number_chk(&argvars[1], &error);
3731
3732 if (!error)
3733 {
3734 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003735 if (idx < 0)
3736 idx = blob_len(argvars[0].vval.v_blob) + idx;
3737 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
3738 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003739 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003740 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003741 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003742 tv = rettv;
3743 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003744 }
3745 }
3746 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003747 {
3748 if ((l = argvars[0].vval.v_list) != NULL)
3749 {
3750 int error = FALSE;
3751
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003752 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003753 if (!error && li != NULL)
3754 tv = &li->li_tv;
3755 }
3756 }
3757 else if (argvars[0].v_type == VAR_DICT)
3758 {
3759 if ((d = argvars[0].vval.v_dict) != NULL)
3760 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003761 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003762 if (di != NULL)
3763 tv = &di->di_tv;
3764 }
3765 }
3766 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
3767 {
3768 partial_T *pt;
3769 partial_T fref_pt;
3770
3771 if (argvars[0].v_type == VAR_PARTIAL)
3772 pt = argvars[0].vval.v_partial;
3773 else
3774 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02003775 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003776 fref_pt.pt_name = argvars[0].vval.v_string;
3777 pt = &fref_pt;
3778 }
3779
3780 if (pt != NULL)
3781 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003782 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003783 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003784
3785 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
3786 {
3787 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003788 n = partial_name(pt);
3789 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003790 rettv->vval.v_string = NULL;
3791 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003792 {
3793 rettv->vval.v_string = vim_strsave(n);
3794 if (rettv->v_type == VAR_FUNC)
3795 func_ref(rettv->vval.v_string);
3796 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003797 }
3798 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003799 {
3800 what_is_dict = TRUE;
3801 if (pt->pt_dict != NULL)
3802 rettv_dict_set(rettv, pt->pt_dict);
3803 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003804 else if (STRCMP(what, "args") == 0)
3805 {
3806 rettv->v_type = VAR_LIST;
3807 if (rettv_list_alloc(rettv) == OK)
3808 {
3809 int i;
3810
3811 for (i = 0; i < pt->pt_argc; ++i)
3812 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3813 }
3814 }
3815 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003816 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003817
3818 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3819 // third argument
3820 if (!what_is_dict)
3821 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003822 }
3823 }
3824 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003825 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003826
3827 if (tv == NULL)
3828 {
3829 if (argvars[2].v_type != VAR_UNKNOWN)
3830 copy_tv(&argvars[2], rettv);
3831 }
3832 else
3833 copy_tv(tv, rettv);
3834}
3835
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003836/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003837 * "getchangelist()" function
3838 */
3839 static void
3840f_getchangelist(typval_T *argvars, typval_T *rettv)
3841{
3842#ifdef FEAT_JUMPLIST
3843 buf_T *buf;
3844 int i;
3845 list_T *l;
3846 dict_T *d;
3847#endif
3848
3849 if (rettv_list_alloc(rettv) != OK)
3850 return;
3851
3852#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003853 if (argvars[0].v_type == VAR_UNKNOWN)
3854 buf = curbuf;
3855 else
Bram Moolenaara5d38412020-09-02 21:02:35 +02003856 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003857 if (buf == NULL)
3858 return;
3859
3860 l = list_alloc();
3861 if (l == NULL)
3862 return;
3863
3864 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3865 return;
3866 /*
3867 * The current window change list index tracks only the position in the
3868 * current buffer change list. For other buffers, use the change list
3869 * length as the current index.
3870 */
3871 list_append_number(rettv->vval.v_list,
3872 (varnumber_T)((buf == curwin->w_buffer)
3873 ? curwin->w_changelistidx : buf->b_changelistlen));
3874
3875 for (i = 0; i < buf->b_changelistlen; ++i)
3876 {
3877 if (buf->b_changelist[i].lnum == 0)
3878 continue;
3879 if ((d = dict_alloc()) == NULL)
3880 return;
3881 if (list_append_dict(l, d) == FAIL)
3882 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003883 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3884 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003885 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003886 }
3887#endif
3888}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003889
3890/*
3891 * "getcharsearch()" function
3892 */
3893 static void
3894f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3895{
3896 if (rettv_dict_alloc(rettv) != FAIL)
3897 {
3898 dict_T *dict = rettv->vval.v_dict;
3899
Bram Moolenaare0be1672018-07-08 16:50:37 +02003900 dict_add_string(dict, "char", last_csearch());
3901 dict_add_number(dict, "forward", last_csearch_forward());
3902 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003903 }
3904}
3905
3906/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003907 * "getenv()" function
3908 */
3909 static void
3910f_getenv(typval_T *argvars, typval_T *rettv)
3911{
3912 int mustfree = FALSE;
3913 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3914
3915 if (p == NULL)
3916 {
3917 rettv->v_type = VAR_SPECIAL;
3918 rettv->vval.v_number = VVAL_NULL;
3919 return;
3920 }
3921 if (!mustfree)
3922 p = vim_strsave(p);
3923 rettv->vval.v_string = p;
3924 rettv->v_type = VAR_STRING;
3925}
3926
3927/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003928 * "getfontname()" function
3929 */
3930 static void
3931f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3932{
3933 rettv->v_type = VAR_STRING;
3934 rettv->vval.v_string = NULL;
3935#ifdef FEAT_GUI
3936 if (gui.in_use)
3937 {
3938 GuiFont font;
3939 char_u *name = NULL;
3940
3941 if (argvars[0].v_type == VAR_UNKNOWN)
3942 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003943 // Get the "Normal" font. Either the name saved by
3944 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003945 font = gui.norm_font;
3946 name = hl_get_font_name();
3947 }
3948 else
3949 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003950 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003951 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003952 return;
3953 font = gui_mch_get_font(name, FALSE);
3954 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003955 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003956 }
3957 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3958 if (argvars[0].v_type != VAR_UNKNOWN)
3959 gui_mch_free_font(font);
3960 }
3961#endif
3962}
3963
3964/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003965 * "getjumplist()" function
3966 */
3967 static void
3968f_getjumplist(typval_T *argvars, typval_T *rettv)
3969{
3970#ifdef FEAT_JUMPLIST
3971 win_T *wp;
3972 int i;
3973 list_T *l;
3974 dict_T *d;
3975#endif
3976
3977 if (rettv_list_alloc(rettv) != OK)
3978 return;
3979
3980#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003981 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003982 if (wp == NULL)
3983 return;
3984
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003985 cleanup_jumplist(wp, TRUE);
3986
Bram Moolenaar4f505882018-02-10 21:06:32 +01003987 l = list_alloc();
3988 if (l == NULL)
3989 return;
3990
3991 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3992 return;
3993 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3994
3995 for (i = 0; i < wp->w_jumplistlen; ++i)
3996 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003997 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3998 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003999 if ((d = dict_alloc()) == NULL)
4000 return;
4001 if (list_append_dict(l, d) == FAIL)
4002 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02004003 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
4004 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004005 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02004006 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01004007 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02004008 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01004009 }
4010#endif
4011}
4012
4013/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004014 * "getpid()" function
4015 */
4016 static void
4017f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
4018{
4019 rettv->vval.v_number = mch_get_pid();
4020}
4021
4022 static void
4023getpos_both(
4024 typval_T *argvars,
4025 typval_T *rettv,
4026 int getcurpos)
4027{
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004028 pos_T *fp = NULL;
4029 win_T *wp = curwin;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004030 list_T *l;
4031 int fnum = -1;
4032
4033 if (rettv_list_alloc(rettv) == OK)
4034 {
4035 l = rettv->vval.v_list;
4036 if (getcurpos)
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004037 {
4038 if (argvars[0].v_type != VAR_UNKNOWN)
4039 {
4040 wp = find_win_by_nr_or_id(&argvars[0]);
4041 if (wp != NULL)
4042 fp = &wp->w_cursor;
4043 }
4044 else
4045 fp = &curwin->w_cursor;
4046 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004047 else
4048 fp = var2fpos(&argvars[0], TRUE, &fnum);
4049 if (fnum != -1)
4050 list_append_number(l, (varnumber_T)fnum);
4051 else
4052 list_append_number(l, (varnumber_T)0);
4053 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
4054 : (varnumber_T)0);
4055 list_append_number(l, (fp != NULL)
4056 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
4057 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01004058 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004059 (varnumber_T)0);
4060 if (getcurpos)
4061 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01004062 int save_set_curswant = curwin->w_set_curswant;
4063 colnr_T save_curswant = curwin->w_curswant;
4064 colnr_T save_virtcol = curwin->w_virtcol;
4065
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004066 if (wp == curwin)
4067 update_curswant();
4068 list_append_number(l, wp == NULL ? 0 : wp->w_curswant == MAXCOL
4069 ? (varnumber_T)MAXCOL : (varnumber_T)wp->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01004070
4071 // Do not change "curswant", as it is unexpected that a get
4072 // function has a side effect.
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02004073 if (wp == curwin && save_set_curswant)
Bram Moolenaar19a66852019-03-07 11:25:32 +01004074 {
4075 curwin->w_set_curswant = save_set_curswant;
4076 curwin->w_curswant = save_curswant;
4077 curwin->w_virtcol = save_virtcol;
4078 curwin->w_valid &= ~VALID_VIRTCOL;
4079 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004080 }
4081 }
4082 else
4083 rettv->vval.v_number = FALSE;
4084}
4085
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004086/*
4087 * "getcurpos()" function
4088 */
4089 static void
4090f_getcurpos(typval_T *argvars, typval_T *rettv)
4091{
4092 getpos_both(argvars, rettv, TRUE);
4093}
4094
4095/*
4096 * "getpos(string)" function
4097 */
4098 static void
4099f_getpos(typval_T *argvars, typval_T *rettv)
4100{
4101 getpos_both(argvars, rettv, FALSE);
4102}
4103
4104/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004105 * "getreg()" function
4106 */
4107 static void
4108f_getreg(typval_T *argvars, typval_T *rettv)
4109{
4110 char_u *strregname;
4111 int regname;
4112 int arg2 = FALSE;
4113 int return_list = FALSE;
4114 int error = FALSE;
4115
4116 if (argvars[0].v_type != VAR_UNKNOWN)
4117 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004118 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004119 error = strregname == NULL;
4120 if (argvars[1].v_type != VAR_UNKNOWN)
4121 {
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004122 arg2 = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004123 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02004124 return_list = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004125 }
4126 }
4127 else
4128 strregname = get_vim_var_str(VV_REG);
4129
4130 if (error)
4131 return;
4132
4133 regname = (strregname == NULL ? '"' : *strregname);
4134 if (regname == 0)
4135 regname = '"';
4136
4137 if (return_list)
4138 {
4139 rettv->v_type = VAR_LIST;
4140 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
4141 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
4142 if (rettv->vval.v_list == NULL)
4143 (void)rettv_list_alloc(rettv);
4144 else
4145 ++rettv->vval.v_list->lv_refcount;
4146 }
4147 else
4148 {
4149 rettv->v_type = VAR_STRING;
4150 rettv->vval.v_string = get_reg_contents(regname,
4151 arg2 ? GREG_EXPR_SRC : 0);
4152 }
4153}
4154
4155/*
4156 * "getregtype()" function
4157 */
4158 static void
4159f_getregtype(typval_T *argvars, typval_T *rettv)
4160{
4161 char_u *strregname;
4162 int regname;
4163 char_u buf[NUMBUFLEN + 2];
4164 long reglen = 0;
4165
4166 if (argvars[0].v_type != VAR_UNKNOWN)
4167 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004168 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004169 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004170 {
4171 rettv->v_type = VAR_STRING;
4172 rettv->vval.v_string = NULL;
4173 return;
4174 }
4175 }
4176 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004177 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004178 strregname = get_vim_var_str(VV_REG);
4179
4180 regname = (strregname == NULL ? '"' : *strregname);
4181 if (regname == 0)
4182 regname = '"';
4183
4184 buf[0] = NUL;
4185 buf[1] = NUL;
4186 switch (get_reg_type(regname, &reglen))
4187 {
4188 case MLINE: buf[0] = 'V'; break;
4189 case MCHAR: buf[0] = 'v'; break;
4190 case MBLOCK:
4191 buf[0] = Ctrl_V;
4192 sprintf((char *)buf + 1, "%ld", reglen + 1);
4193 break;
4194 }
4195 rettv->v_type = VAR_STRING;
4196 rettv->vval.v_string = vim_strsave(buf);
4197}
4198
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02004199/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01004200 * "gettagstack()" function
4201 */
4202 static void
4203f_gettagstack(typval_T *argvars, typval_T *rettv)
4204{
4205 win_T *wp = curwin; // default is current window
4206
4207 if (rettv_dict_alloc(rettv) != OK)
4208 return;
4209
4210 if (argvars[0].v_type != VAR_UNKNOWN)
4211 {
4212 wp = find_win_by_nr_or_id(&argvars[0]);
4213 if (wp == NULL)
4214 return;
4215 }
4216
4217 get_tagstack(wp, rettv->vval.v_dict);
4218}
4219
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +02004220/*
4221 * "gettext()" function
4222 */
4223 static void
4224f_gettext(typval_T *argvars, typval_T *rettv)
4225{
4226 if (argvars[0].v_type != VAR_STRING
4227 || argvars[0].vval.v_string == NULL
4228 || *argvars[0].vval.v_string == NUL)
4229 {
4230 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
4231 }
4232 else
4233 {
4234 rettv->v_type = VAR_STRING;
4235 rettv->vval.v_string = vim_strsave(
4236 (char_u *)_(argvars[0].vval.v_string));
4237 }
4238}
4239
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004240// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004241#include "version.h"
4242
4243/*
4244 * "has()" function
4245 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01004246 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004247f_has(typval_T *argvars, typval_T *rettv)
4248{
4249 int i;
4250 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01004251 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004252 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01004253 typedef struct {
4254 char *name;
4255 short present;
4256 } has_item_T;
4257 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004258 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004259 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004260#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01004261 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004262#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004263 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004264#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004265 },
4266 {"arp",
4267#if defined(AMIGA) && defined(FEAT_ARP)
4268 1
4269#else
4270 0
4271#endif
4272 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004273 {"haiku",
4274#ifdef __HAIKU__
4275 1
4276#else
4277 0
4278#endif
4279 },
4280 {"bsd",
4281#if defined(BSD) && !defined(MACOS_X)
4282 1
4283#else
4284 0
4285#endif
4286 },
4287 {"hpux",
4288#ifdef hpux
4289 1
4290#else
4291 0
4292#endif
4293 },
4294 {"linux",
4295#ifdef __linux__
4296 1
4297#else
4298 0
4299#endif
4300 },
4301 {"mac", // Mac OS X (and, once, Mac OS Classic)
4302#ifdef MACOS_X
4303 1
4304#else
4305 0
4306#endif
4307 },
4308 {"osx", // Mac OS X
4309#ifdef MACOS_X
4310 1
4311#else
4312 0
4313#endif
4314 },
4315 {"macunix", // Mac OS X, with the darwin feature
4316#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4317 1
4318#else
4319 0
4320#endif
4321 },
4322 {"osxdarwin", // synonym for macunix
4323#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4324 1
4325#else
4326 0
4327#endif
4328 },
4329 {"qnx",
4330#ifdef __QNX__
4331 1
4332#else
4333 0
4334#endif
4335 },
4336 {"sun",
4337#ifdef SUN_SYSTEM
4338 1
4339#else
4340 0
4341#endif
4342 },
4343 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004344#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01004345 1
4346#else
4347 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004348#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004349 },
4350 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004351#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01004352 1
4353#else
4354 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004355#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004356 },
4357 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01004358#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01004359 1
4360#else
4361 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004362#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004363 },
4364 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01004365#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01004366 1
4367#else
4368 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004369#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004370 },
4371 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01004372#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01004373 1
4374#else
4375 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004376#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004377 },
4378 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004379#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004380 1
4381#else
4382 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004383#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004384 },
4385 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004386#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004387 1
4388#else
4389 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004390#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004391 },
4392 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004393#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01004394 1
4395#else
4396 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004397#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004398 },
4399 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004400#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004401 1
4402#else
4403 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004404#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004405 },
4406 {"autocmd", 1},
4407 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004408#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01004409 1
4410#else
4411 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004412#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004413 },
4414 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004415#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004416 1
4417#else
4418 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004419#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004420 },
4421 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004422#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01004423 1
4424#else
4425 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004426#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004427 },
4428 {"balloon_multiline",
4429#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
4430 // MS-Windows requires runtime check, see below
4431 1
4432#else
4433 0
4434#endif
4435 },
4436 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004437#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01004438 1
4439#else
4440 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01004441#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004442 },
4443 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004444#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01004445 1
4446#else
4447 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004448#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004449 },
4450 {"all_builtin_terms",
4451#if defined(ALL_BUILTIN_TCAPS)
4452 1
4453#else
4454 0
4455#endif
4456 },
4457 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004458#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01004459 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004460 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01004461 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004462#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004463 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004464#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004465 },
4466 {"byte_offset",
4467#ifdef FEAT_BYTEOFF
4468 1
4469#else
4470 0
4471#endif
4472 },
4473 {"channel",
4474#ifdef FEAT_JOB_CHANNEL
4475 1
4476#else
4477 0
4478#endif
4479 },
4480 {"cindent",
4481#ifdef FEAT_CINDENT
4482 1
4483#else
4484 0
4485#endif
4486 },
4487 {"clientserver",
4488#ifdef FEAT_CLIENTSERVER
4489 1
4490#else
4491 0
4492#endif
4493 },
4494 {"clipboard",
4495#ifdef FEAT_CLIPBOARD
4496 1
4497#else
4498 0
4499#endif
4500 },
4501 {"cmdline_compl", 1},
4502 {"cmdline_hist", 1},
4503 {"comments", 1},
4504 {"conceal",
4505#ifdef FEAT_CONCEAL
4506 1
4507#else
4508 0
4509#endif
4510 },
4511 {"cryptv",
4512#ifdef FEAT_CRYPT
4513 1
4514#else
4515 0
4516#endif
4517 },
4518 {"crypt-blowfish",
4519#ifdef FEAT_CRYPT
4520 1
4521#else
4522 0
4523#endif
4524 },
4525 {"crypt-blowfish2",
4526#ifdef FEAT_CRYPT
4527 1
4528#else
4529 0
4530#endif
4531 },
4532 {"cscope",
4533#ifdef FEAT_CSCOPE
4534 1
4535#else
4536 0
4537#endif
4538 },
4539 {"cursorbind", 1},
4540 {"cursorshape",
4541#ifdef CURSOR_SHAPE
4542 1
4543#else
4544 0
4545#endif
4546 },
4547 {"debug",
4548#ifdef DEBUG
4549 1
4550#else
4551 0
4552#endif
4553 },
4554 {"dialog_con",
4555#ifdef FEAT_CON_DIALOG
4556 1
4557#else
4558 0
4559#endif
4560 },
4561 {"dialog_gui",
4562#ifdef FEAT_GUI_DIALOG
4563 1
4564#else
4565 0
4566#endif
4567 },
4568 {"diff",
4569#ifdef FEAT_DIFF
4570 1
4571#else
4572 0
4573#endif
4574 },
4575 {"digraphs",
4576#ifdef FEAT_DIGRAPHS
4577 1
4578#else
4579 0
4580#endif
4581 },
4582 {"directx",
4583#ifdef FEAT_DIRECTX
4584 1
4585#else
4586 0
4587#endif
4588 },
4589 {"dnd",
4590#ifdef FEAT_DND
4591 1
4592#else
4593 0
4594#endif
4595 },
4596 {"emacs_tags",
4597#ifdef FEAT_EMACS_TAGS
4598 1
4599#else
4600 0
4601#endif
4602 },
4603 {"eval", 1}, // always present, of course!
4604 {"ex_extra", 1}, // graduated feature
4605 {"extra_search",
4606#ifdef FEAT_SEARCH_EXTRA
4607 1
4608#else
4609 0
4610#endif
4611 },
4612 {"file_in_path",
4613#ifdef FEAT_SEARCHPATH
4614 1
4615#else
4616 0
4617#endif
4618 },
4619 {"filterpipe",
4620#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
4621 1
4622#else
4623 0
4624#endif
4625 },
4626 {"find_in_path",
4627#ifdef FEAT_FIND_ID
4628 1
4629#else
4630 0
4631#endif
4632 },
4633 {"float",
4634#ifdef FEAT_FLOAT
4635 1
4636#else
4637 0
4638#endif
4639 },
4640 {"folding",
4641#ifdef FEAT_FOLDING
4642 1
4643#else
4644 0
4645#endif
4646 },
4647 {"footer",
4648#ifdef FEAT_FOOTER
4649 1
4650#else
4651 0
4652#endif
4653 },
4654 {"fork",
4655#if !defined(USE_SYSTEM) && defined(UNIX)
4656 1
4657#else
4658 0
4659#endif
4660 },
4661 {"gettext",
4662#ifdef FEAT_GETTEXT
4663 1
4664#else
4665 0
4666#endif
4667 },
4668 {"gui",
4669#ifdef FEAT_GUI
4670 1
4671#else
4672 0
4673#endif
4674 },
4675 {"gui_neXtaw",
4676#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
4677 1
4678#else
4679 0
4680#endif
4681 },
4682 {"gui_athena",
4683#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
4684 1
4685#else
4686 0
4687#endif
4688 },
4689 {"gui_gtk",
4690#ifdef FEAT_GUI_GTK
4691 1
4692#else
4693 0
4694#endif
4695 },
4696 {"gui_gtk2",
4697#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
4698 1
4699#else
4700 0
4701#endif
4702 },
4703 {"gui_gtk3",
4704#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
4705 1
4706#else
4707 0
4708#endif
4709 },
4710 {"gui_gnome",
4711#ifdef FEAT_GUI_GNOME
4712 1
4713#else
4714 0
4715#endif
4716 },
4717 {"gui_haiku",
4718#ifdef FEAT_GUI_HAIKU
4719 1
4720#else
4721 0
4722#endif
4723 },
Bram Moolenaar097148e2020-08-11 21:58:20 +02004724 {"gui_mac", 0},
Bram Moolenaar79296512020-03-22 16:17:14 +01004725 {"gui_motif",
4726#ifdef FEAT_GUI_MOTIF
4727 1
4728#else
4729 0
4730#endif
4731 },
4732 {"gui_photon",
4733#ifdef FEAT_GUI_PHOTON
4734 1
4735#else
4736 0
4737#endif
4738 },
4739 {"gui_win32",
4740#ifdef FEAT_GUI_MSWIN
4741 1
4742#else
4743 0
4744#endif
4745 },
4746 {"iconv",
4747#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
4748 1
4749#else
4750 0
4751#endif
4752 },
4753 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02004754 {"ipv6",
4755#ifdef FEAT_IPV6
4756 1
4757#else
4758 0
4759#endif
4760 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004761 {"job",
4762#ifdef FEAT_JOB_CHANNEL
4763 1
4764#else
4765 0
4766#endif
4767 },
4768 {"jumplist",
4769#ifdef FEAT_JUMPLIST
4770 1
4771#else
4772 0
4773#endif
4774 },
4775 {"keymap",
4776#ifdef FEAT_KEYMAP
4777 1
4778#else
4779 0
4780#endif
4781 },
4782 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
4783 {"langmap",
4784#ifdef FEAT_LANGMAP
4785 1
4786#else
4787 0
4788#endif
4789 },
4790 {"libcall",
4791#ifdef FEAT_LIBCALL
4792 1
4793#else
4794 0
4795#endif
4796 },
4797 {"linebreak",
4798#ifdef FEAT_LINEBREAK
4799 1
4800#else
4801 0
4802#endif
4803 },
4804 {"lispindent",
4805#ifdef FEAT_LISP
4806 1
4807#else
4808 0
4809#endif
4810 },
4811 {"listcmds", 1},
4812 {"localmap", 1},
4813 {"lua",
4814#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4815 1
4816#else
4817 0
4818#endif
4819 },
4820 {"menu",
4821#ifdef FEAT_MENU
4822 1
4823#else
4824 0
4825#endif
4826 },
4827 {"mksession",
4828#ifdef FEAT_SESSION
4829 1
4830#else
4831 0
4832#endif
4833 },
4834 {"modify_fname", 1},
4835 {"mouse", 1},
4836 {"mouseshape",
4837#ifdef FEAT_MOUSESHAPE
4838 1
4839#else
4840 0
4841#endif
4842 },
4843 {"mouse_dec",
4844#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4845 1
4846#else
4847 0
4848#endif
4849 },
4850 {"mouse_gpm",
4851#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4852 1
4853#else
4854 0
4855#endif
4856 },
4857 {"mouse_jsbterm",
4858#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4859 1
4860#else
4861 0
4862#endif
4863 },
4864 {"mouse_netterm",
4865#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4866 1
4867#else
4868 0
4869#endif
4870 },
4871 {"mouse_pterm",
4872#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4873 1
4874#else
4875 0
4876#endif
4877 },
4878 {"mouse_sgr",
4879#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4880 1
4881#else
4882 0
4883#endif
4884 },
4885 {"mouse_sysmouse",
4886#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4887 1
4888#else
4889 0
4890#endif
4891 },
4892 {"mouse_urxvt",
4893#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4894 1
4895#else
4896 0
4897#endif
4898 },
4899 {"mouse_xterm",
4900#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4901 1
4902#else
4903 0
4904#endif
4905 },
4906 {"multi_byte", 1},
4907 {"multi_byte_ime",
4908#ifdef FEAT_MBYTE_IME
4909 1
4910#else
4911 0
4912#endif
4913 },
4914 {"multi_lang",
4915#ifdef FEAT_MULTI_LANG
4916 1
4917#else
4918 0
4919#endif
4920 },
4921 {"mzscheme",
4922#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4923 1
4924#else
4925 0
4926#endif
4927 },
4928 {"num64", 1},
4929 {"ole",
4930#ifdef FEAT_OLE
4931 1
4932#else
4933 0
4934#endif
4935 },
4936 {"packages",
4937#ifdef FEAT_EVAL
4938 1
4939#else
4940 0
4941#endif
4942 },
4943 {"path_extra",
4944#ifdef FEAT_PATH_EXTRA
4945 1
4946#else
4947 0
4948#endif
4949 },
4950 {"perl",
4951#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4952 1
4953#else
4954 0
4955#endif
4956 },
4957 {"persistent_undo",
4958#ifdef FEAT_PERSISTENT_UNDO
4959 1
4960#else
4961 0
4962#endif
4963 },
4964 {"python_compiled",
4965#if defined(FEAT_PYTHON)
4966 1
4967#else
4968 0
4969#endif
4970 },
4971 {"python_dynamic",
4972#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4973 1
4974#else
4975 0
4976#endif
4977 },
4978 {"python",
4979#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4980 1
4981#else
4982 0
4983#endif
4984 },
4985 {"pythonx",
4986#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4987 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4988 1
4989#else
4990 0
4991#endif
4992 },
4993 {"python3_compiled",
4994#if defined(FEAT_PYTHON3)
4995 1
4996#else
4997 0
4998#endif
4999 },
5000 {"python3_dynamic",
5001#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
5002 1
5003#else
5004 0
5005#endif
5006 },
5007 {"python3",
5008#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
5009 1
5010#else
5011 0
5012#endif
5013 },
5014 {"popupwin",
5015#ifdef FEAT_PROP_POPUP
5016 1
5017#else
5018 0
5019#endif
5020 },
5021 {"postscript",
5022#ifdef FEAT_POSTSCRIPT
5023 1
5024#else
5025 0
5026#endif
5027 },
5028 {"printer",
5029#ifdef FEAT_PRINTER
5030 1
5031#else
5032 0
5033#endif
5034 },
5035 {"profile",
5036#ifdef FEAT_PROFILE
5037 1
5038#else
5039 0
5040#endif
5041 },
5042 {"reltime",
5043#ifdef FEAT_RELTIME
5044 1
5045#else
5046 0
5047#endif
5048 },
5049 {"quickfix",
5050#ifdef FEAT_QUICKFIX
5051 1
5052#else
5053 0
5054#endif
5055 },
5056 {"rightleft",
5057#ifdef FEAT_RIGHTLEFT
5058 1
5059#else
5060 0
5061#endif
5062 },
5063 {"ruby",
5064#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
5065 1
5066#else
5067 0
5068#endif
5069 },
5070 {"scrollbind", 1},
5071 {"showcmd",
5072#ifdef FEAT_CMDL_INFO
5073 1
5074#else
5075 0
5076#endif
5077 },
5078 {"cmdline_info",
5079#ifdef FEAT_CMDL_INFO
5080 1
5081#else
5082 0
5083#endif
5084 },
5085 {"signs",
5086#ifdef FEAT_SIGNS
5087 1
5088#else
5089 0
5090#endif
5091 },
5092 {"smartindent",
5093#ifdef FEAT_SMARTINDENT
5094 1
5095#else
5096 0
5097#endif
5098 },
5099 {"startuptime",
5100#ifdef STARTUPTIME
5101 1
5102#else
5103 0
5104#endif
5105 },
5106 {"statusline",
5107#ifdef FEAT_STL_OPT
5108 1
5109#else
5110 0
5111#endif
5112 },
5113 {"netbeans_intg",
5114#ifdef FEAT_NETBEANS_INTG
5115 1
5116#else
5117 0
5118#endif
5119 },
5120 {"sound",
5121#ifdef FEAT_SOUND
5122 1
5123#else
5124 0
5125#endif
5126 },
5127 {"spell",
5128#ifdef FEAT_SPELL
5129 1
5130#else
5131 0
5132#endif
5133 },
5134 {"syntax",
5135#ifdef FEAT_SYN_HL
5136 1
5137#else
5138 0
5139#endif
5140 },
5141 {"system",
5142#if defined(USE_SYSTEM) || !defined(UNIX)
5143 1
5144#else
5145 0
5146#endif
5147 },
5148 {"tag_binary",
5149#ifdef FEAT_TAG_BINS
5150 1
5151#else
5152 0
5153#endif
5154 },
5155 {"tcl",
5156#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
5157 1
5158#else
5159 0
5160#endif
5161 },
5162 {"termguicolors",
5163#ifdef FEAT_TERMGUICOLORS
5164 1
5165#else
5166 0
5167#endif
5168 },
5169 {"terminal",
5170#if defined(FEAT_TERMINAL) && !defined(MSWIN)
5171 1
5172#else
5173 0
5174#endif
5175 },
5176 {"terminfo",
5177#ifdef TERMINFO
5178 1
5179#else
5180 0
5181#endif
5182 },
5183 {"termresponse",
5184#ifdef FEAT_TERMRESPONSE
5185 1
5186#else
5187 0
5188#endif
5189 },
5190 {"textobjects",
5191#ifdef FEAT_TEXTOBJ
5192 1
5193#else
5194 0
5195#endif
5196 },
5197 {"textprop",
5198#ifdef FEAT_PROP_POPUP
5199 1
5200#else
5201 0
5202#endif
5203 },
5204 {"tgetent",
5205#ifdef HAVE_TGETENT
5206 1
5207#else
5208 0
5209#endif
5210 },
5211 {"timers",
5212#ifdef FEAT_TIMERS
5213 1
5214#else
5215 0
5216#endif
5217 },
5218 {"title",
5219#ifdef FEAT_TITLE
5220 1
5221#else
5222 0
5223#endif
5224 },
5225 {"toolbar",
5226#ifdef FEAT_TOOLBAR
5227 1
5228#else
5229 0
5230#endif
5231 },
5232 {"unnamedplus",
5233#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
5234 1
5235#else
5236 0
5237#endif
5238 },
5239 {"user-commands", 1}, // was accidentally included in 5.4
5240 {"user_commands", 1},
5241 {"vartabs",
5242#ifdef FEAT_VARTABS
5243 1
5244#else
5245 0
5246#endif
5247 },
5248 {"vertsplit", 1},
5249 {"viminfo",
5250#ifdef FEAT_VIMINFO
5251 1
5252#else
5253 0
5254#endif
5255 },
5256 {"vimscript-1", 1},
5257 {"vimscript-2", 1},
5258 {"vimscript-3", 1},
5259 {"vimscript-4", 1},
5260 {"virtualedit", 1},
5261 {"visual", 1},
5262 {"visualextra", 1},
5263 {"vreplace", 1},
5264 {"vtp",
5265#ifdef FEAT_VTP
5266 1
5267#else
5268 0
5269#endif
5270 },
5271 {"wildignore",
5272#ifdef FEAT_WILDIGN
5273 1
5274#else
5275 0
5276#endif
5277 },
5278 {"wildmenu",
5279#ifdef FEAT_WILDMENU
5280 1
5281#else
5282 0
5283#endif
5284 },
5285 {"windows", 1},
5286 {"winaltkeys",
5287#ifdef FEAT_WAK
5288 1
5289#else
5290 0
5291#endif
5292 },
5293 {"writebackup",
5294#ifdef FEAT_WRITEBACKUP
5295 1
5296#else
5297 0
5298#endif
5299 },
5300 {"xim",
5301#ifdef FEAT_XIM
5302 1
5303#else
5304 0
5305#endif
5306 },
5307 {"xfontset",
5308#ifdef FEAT_XFONTSET
5309 1
5310#else
5311 0
5312#endif
5313 },
5314 {"xpm",
5315#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
5316 1
5317#else
5318 0
5319#endif
5320 },
5321 {"xpm_w32", // for backward compatibility
5322#ifdef FEAT_XPM_W32
5323 1
5324#else
5325 0
5326#endif
5327 },
5328 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005329#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01005330 1
5331#else
5332 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005333#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005334 },
5335 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005336#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01005337 1
5338#else
5339 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005340#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005341 },
5342 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005343#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01005344 1
5345#else
5346 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005347#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005348 },
5349 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005350#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01005351 1
5352#else
5353 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005354#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005355 },
5356 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005357#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01005358 1
5359#else
5360 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005361#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005362 },
5363 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005364 };
5365
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005366 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01005367 for (i = 0; has_list[i].name != NULL; ++i)
5368 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005369 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005370 x = TRUE;
5371 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005372 break;
5373 }
5374
Bram Moolenaar79296512020-03-22 16:17:14 +01005375 // features also in has_list[] but sometimes enabled at runtime
5376 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005377 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005378 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02005379 {
5380 // intentionally empty
5381 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01005382#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005383 else if (STRICMP(name, "balloon_multiline") == 0)
5384 n = multiline_balloon_available();
5385#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005386#ifdef VIMDLL
5387 else if (STRICMP(name, "filterpipe") == 0)
5388 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005389#endif
5390#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5391 else if (STRICMP(name, "iconv") == 0)
5392 n = iconv_enabled(FALSE);
5393#endif
5394#ifdef DYNAMIC_LUA
5395 else if (STRICMP(name, "lua") == 0)
5396 n = lua_enabled(FALSE);
5397#endif
5398#ifdef DYNAMIC_MZSCHEME
5399 else if (STRICMP(name, "mzscheme") == 0)
5400 n = mzscheme_enabled(FALSE);
5401#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005402#ifdef DYNAMIC_PERL
5403 else if (STRICMP(name, "perl") == 0)
5404 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005405#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005406#ifdef DYNAMIC_PYTHON
5407 else if (STRICMP(name, "python") == 0)
5408 n = python_enabled(FALSE);
5409#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005410#ifdef DYNAMIC_PYTHON3
5411 else if (STRICMP(name, "python3") == 0)
5412 n = python3_enabled(FALSE);
5413#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005414#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5415 else if (STRICMP(name, "pythonx") == 0)
5416 {
5417# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
5418 if (p_pyx == 0)
5419 n = python3_enabled(FALSE) || python_enabled(FALSE);
5420 else if (p_pyx == 3)
5421 n = python3_enabled(FALSE);
5422 else if (p_pyx == 2)
5423 n = python_enabled(FALSE);
5424# elif defined(DYNAMIC_PYTHON)
5425 n = python_enabled(FALSE);
5426# elif defined(DYNAMIC_PYTHON3)
5427 n = python3_enabled(FALSE);
5428# endif
5429 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005430#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005431#ifdef DYNAMIC_RUBY
5432 else if (STRICMP(name, "ruby") == 0)
5433 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005434#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005435#ifdef DYNAMIC_TCL
5436 else if (STRICMP(name, "tcl") == 0)
5437 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02005438#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01005439#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02005440 else if (STRICMP(name, "terminal") == 0)
5441 n = terminal_enabled();
5442#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005443 }
5444
Bram Moolenaar79296512020-03-22 16:17:14 +01005445 // features not in has_list[]
5446 if (x == FALSE)
5447 {
5448 if (STRNICMP(name, "patch", 5) == 0)
5449 {
5450 x = TRUE;
5451 if (name[5] == '-'
5452 && STRLEN(name) >= 11
5453 && vim_isdigit(name[6])
5454 && vim_isdigit(name[8])
5455 && vim_isdigit(name[10]))
5456 {
5457 int major = atoi((char *)name + 6);
5458 int minor = atoi((char *)name + 8);
5459
5460 // Expect "patch-9.9.01234".
5461 n = (major < VIM_VERSION_MAJOR
5462 || (major == VIM_VERSION_MAJOR
5463 && (minor < VIM_VERSION_MINOR
5464 || (minor == VIM_VERSION_MINOR
5465 && has_patch(atoi((char *)name + 10))))));
5466 }
5467 else
5468 n = has_patch(atoi((char *)name + 5));
5469 }
5470 else if (STRICMP(name, "vim_starting") == 0)
5471 {
5472 x = TRUE;
5473 n = (starting != 0);
5474 }
5475 else if (STRICMP(name, "ttyin") == 0)
5476 {
5477 x = TRUE;
5478 n = mch_input_isatty();
5479 }
5480 else if (STRICMP(name, "ttyout") == 0)
5481 {
5482 x = TRUE;
5483 n = stdout_isatty;
5484 }
5485 else if (STRICMP(name, "multi_byte_encoding") == 0)
5486 {
5487 x = TRUE;
5488 n = has_mbyte;
5489 }
5490 else if (STRICMP(name, "gui_running") == 0)
5491 {
5492 x = TRUE;
5493#ifdef FEAT_GUI
5494 n = (gui.in_use || gui.starting);
5495#endif
5496 }
5497 else if (STRICMP(name, "browse") == 0)
5498 {
5499 x = TRUE;
5500#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5501 n = gui.in_use; // gui_mch_browse() works when GUI is running
5502#endif
5503 }
5504 else if (STRICMP(name, "syntax_items") == 0)
5505 {
5506 x = TRUE;
5507#ifdef FEAT_SYN_HL
5508 n = syntax_present(curwin);
5509#endif
5510 }
5511 else if (STRICMP(name, "vcon") == 0)
5512 {
5513 x = TRUE;
5514#ifdef FEAT_VTP
5515 n = is_term_win32() && has_vtp_working();
5516#endif
5517 }
5518 else if (STRICMP(name, "netbeans_enabled") == 0)
5519 {
5520 x = TRUE;
5521#ifdef FEAT_NETBEANS_INTG
5522 n = netbeans_active();
5523#endif
5524 }
5525 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
5526 {
5527 x = TRUE;
5528#ifdef FEAT_MOUSE_GPM
5529 n = gpm_enabled();
5530#endif
5531 }
5532 else if (STRICMP(name, "conpty") == 0)
5533 {
5534 x = TRUE;
5535#if defined(FEAT_TERMINAL) && defined(MSWIN)
5536 n = use_conpty();
5537#endif
5538 }
5539 else if (STRICMP(name, "clipboard_working") == 0)
5540 {
5541 x = TRUE;
5542#ifdef FEAT_CLIPBOARD
5543 n = clip_star.available;
5544#endif
5545 }
5546 }
5547
Bram Moolenaar04637e22020-09-05 18:45:29 +02005548 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[1]))
Bram Moolenaar79296512020-03-22 16:17:14 +01005549 // return whether feature could ever be enabled
5550 rettv->vval.v_number = x;
5551 else
5552 // return whether feature is enabled
5553 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005554}
5555
5556/*
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005557 * Return TRUE if "feature" can change later.
5558 * Also when checking for the feature has side effects, such as loading a DLL.
5559 */
5560 int
5561dynamic_feature(char_u *feature)
5562{
5563 return (feature == NULL
5564#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
5565 || STRICMP(feature, "balloon_multiline") == 0
5566#endif
5567#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5568 || (STRICMP(feature, "browse") == 0 && !gui.in_use)
5569#endif
5570#ifdef VIMDLL
5571 || STRICMP(feature, "filterpipe") == 0
5572#endif
Bram Moolenaar29b281b2020-11-10 20:58:00 +01005573#if defined(FEAT_GUI) && !defined(ALWAYS_USE_GUI) && !defined(VIMDLL)
Bram Moolenaar8cebd432020-11-08 12:49:47 +01005574 // this can only change on Unix where the ":gui" command could be
5575 // used.
5576 || (STRICMP(feature, "gui_running") == 0 && !gui.in_use)
5577#endif
5578#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5579 || STRICMP(feature, "iconv") == 0
5580#endif
5581#ifdef DYNAMIC_LUA
5582 || STRICMP(feature, "lua") == 0
5583#endif
5584#ifdef FEAT_MOUSE_GPM
5585 || (STRICMP(feature, "mouse_gpm_enabled") == 0 && !gpm_enabled())
5586#endif
5587#ifdef DYNAMIC_MZSCHEME
5588 || STRICMP(feature, "mzscheme") == 0
5589#endif
5590#ifdef FEAT_NETBEANS_INTG
5591 || STRICMP(feature, "netbeans_enabled") == 0
5592#endif
5593#ifdef DYNAMIC_PERL
5594 || STRICMP(feature, "perl") == 0
5595#endif
5596#ifdef DYNAMIC_PYTHON
5597 || STRICMP(feature, "python") == 0
5598#endif
5599#ifdef DYNAMIC_PYTHON3
5600 || STRICMP(feature, "python3") == 0
5601#endif
5602#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5603 || STRICMP(feature, "pythonx") == 0
5604#endif
5605#ifdef DYNAMIC_RUBY
5606 || STRICMP(feature, "ruby") == 0
5607#endif
5608#ifdef FEAT_SYN_HL
5609 || STRICMP(feature, "syntax_items") == 0
5610#endif
5611#ifdef DYNAMIC_TCL
5612 || STRICMP(feature, "tcl") == 0
5613#endif
5614 // once "starting" is zero it will stay that way
5615 || (STRICMP(feature, "vim_starting") == 0 && starting != 0)
5616 || STRICMP(feature, "multi_byte_encoding") == 0
5617#if defined(FEAT_TERMINAL) && defined(MSWIN)
5618 || STRICMP(feature, "conpty") == 0
5619#endif
5620 );
5621}
5622
5623/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005624 * "haslocaldir()" function
5625 */
5626 static void
5627f_haslocaldir(typval_T *argvars, typval_T *rettv)
5628{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005629 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005630 win_T *wp = NULL;
5631
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005632 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
5633
5634 // Check for window-local and tab-local directories
5635 if (wp != NULL && wp->w_localdir != NULL)
5636 rettv->vval.v_number = 1;
5637 else if (tp != NULL && tp->tp_localdir != NULL)
5638 rettv->vval.v_number = 2;
5639 else
5640 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005641}
5642
5643/*
5644 * "hasmapto()" function
5645 */
5646 static void
5647f_hasmapto(typval_T *argvars, typval_T *rettv)
5648{
5649 char_u *name;
5650 char_u *mode;
5651 char_u buf[NUMBUFLEN];
5652 int abbr = FALSE;
5653
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005654 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005655 if (argvars[1].v_type == VAR_UNKNOWN)
5656 mode = (char_u *)"nvo";
5657 else
5658 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005659 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005660 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02005661 abbr = (int)tv_get_bool(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005662 }
5663
5664 if (map_to_exists(name, mode, abbr))
5665 rettv->vval.v_number = TRUE;
5666 else
5667 rettv->vval.v_number = FALSE;
5668}
5669
5670/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005671 * "highlightID(name)" function
5672 */
5673 static void
5674f_hlID(typval_T *argvars, typval_T *rettv)
5675{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005676 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005677}
5678
5679/*
5680 * "highlight_exists()" function
5681 */
5682 static void
5683f_hlexists(typval_T *argvars, typval_T *rettv)
5684{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005685 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005686}
5687
5688/*
5689 * "hostname()" function
5690 */
5691 static void
5692f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
5693{
5694 char_u hostname[256];
5695
5696 mch_get_host_name(hostname, 256);
5697 rettv->v_type = VAR_STRING;
5698 rettv->vval.v_string = vim_strsave(hostname);
5699}
5700
5701/*
5702 * iconv() function
5703 */
5704 static void
5705f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
5706{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005707 char_u buf1[NUMBUFLEN];
5708 char_u buf2[NUMBUFLEN];
5709 char_u *from, *to, *str;
5710 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005711
5712 rettv->v_type = VAR_STRING;
5713 rettv->vval.v_string = NULL;
5714
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005715 str = tv_get_string(&argvars[0]);
5716 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
5717 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005718 vimconv.vc_type = CONV_NONE;
5719 convert_setup(&vimconv, from, to);
5720
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005721 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005722 if (vimconv.vc_type == CONV_NONE)
5723 rettv->vval.v_string = vim_strsave(str);
5724 else
5725 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
5726
5727 convert_setup(&vimconv, NULL, NULL);
5728 vim_free(from);
5729 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005730}
5731
5732/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005733 * "index()" function
5734 */
5735 static void
5736f_index(typval_T *argvars, typval_T *rettv)
5737{
5738 list_T *l;
5739 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005740 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005741 long idx = 0;
5742 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005743 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005744
5745 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005746 if (argvars[0].v_type == VAR_BLOB)
5747 {
5748 typval_T tv;
5749 int start = 0;
5750
5751 if (argvars[2].v_type != VAR_UNKNOWN)
5752 {
5753 start = tv_get_number_chk(&argvars[2], &error);
5754 if (error)
5755 return;
5756 }
5757 b = argvars[0].vval.v_blob;
5758 if (b == NULL)
5759 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01005760 if (start < 0)
5761 {
5762 start = blob_len(b) + start;
5763 if (start < 0)
5764 start = 0;
5765 }
5766
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005767 for (idx = start; idx < blob_len(b); ++idx)
5768 {
5769 tv.v_type = VAR_NUMBER;
5770 tv.vval.v_number = blob_get(b, idx);
5771 if (tv_equal(&tv, &argvars[1], ic, FALSE))
5772 {
5773 rettv->vval.v_number = idx;
5774 return;
5775 }
5776 }
5777 return;
5778 }
5779 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005780 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01005781 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005782 return;
5783 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005784
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005785 l = argvars[0].vval.v_list;
5786 if (l != NULL)
5787 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005788 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005789 item = l->lv_first;
5790 if (argvars[2].v_type != VAR_UNKNOWN)
5791 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005792 // Start at specified item. Use the cached index that list_find()
5793 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005794 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005795 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005796 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar6c553f92020-09-02 22:10:34 +02005797 ic = (int)tv_get_bool_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005798 if (error)
5799 item = NULL;
5800 }
5801
5802 for ( ; item != NULL; item = item->li_next, ++idx)
5803 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
5804 {
5805 rettv->vval.v_number = idx;
5806 break;
5807 }
5808 }
5809}
5810
5811static int inputsecret_flag = 0;
5812
5813/*
5814 * "input()" function
5815 * Also handles inputsecret() when inputsecret is set.
5816 */
5817 static void
5818f_input(typval_T *argvars, typval_T *rettv)
5819{
5820 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
5821}
5822
5823/*
5824 * "inputdialog()" function
5825 */
5826 static void
5827f_inputdialog(typval_T *argvars, typval_T *rettv)
5828{
5829#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005830 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005831 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
5832 {
5833 char_u *message;
5834 char_u buf[NUMBUFLEN];
5835 char_u *defstr = (char_u *)"";
5836
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005837 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005838 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005839 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005840 vim_strncpy(IObuff, defstr, IOSIZE - 1);
5841 else
5842 IObuff[0] = NUL;
5843 if (message != NULL && defstr != NULL
5844 && do_dialog(VIM_QUESTION, NULL, message,
5845 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
5846 rettv->vval.v_string = vim_strsave(IObuff);
5847 else
5848 {
5849 if (message != NULL && defstr != NULL
5850 && argvars[1].v_type != VAR_UNKNOWN
5851 && argvars[2].v_type != VAR_UNKNOWN)
5852 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005853 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005854 else
5855 rettv->vval.v_string = NULL;
5856 }
5857 rettv->v_type = VAR_STRING;
5858 }
5859 else
5860#endif
5861 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
5862}
5863
5864/*
5865 * "inputlist()" function
5866 */
5867 static void
5868f_inputlist(typval_T *argvars, typval_T *rettv)
5869{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005870 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005871 listitem_T *li;
5872 int selected;
5873 int mouse_used;
5874
5875#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005876 // While starting up, there is no place to enter text. When running tests
5877 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005878 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005879 return;
5880#endif
5881 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5882 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005883 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005884 return;
5885 }
5886
5887 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005888 msg_row = Rows - 1; // for when 'cmdheight' > 1
5889 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005890 msg_scroll = TRUE;
5891 msg_clr_eos();
5892
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005893 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005894 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02005895 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005896 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005897 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005898 msg_putchar('\n');
5899 }
5900
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005901 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005902 selected = prompt_for_number(&mouse_used);
5903 if (mouse_used)
5904 selected -= lines_left;
5905
5906 rettv->vval.v_number = selected;
5907}
5908
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005909static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
5910
5911/*
5912 * "inputrestore()" function
5913 */
5914 static void
5915f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
5916{
5917 if (ga_userinput.ga_len > 0)
5918 {
5919 --ga_userinput.ga_len;
5920 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
5921 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005922 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005923 }
5924 else if (p_verbose > 1)
5925 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005926 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005927 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005928 }
5929}
5930
5931/*
5932 * "inputsave()" function
5933 */
5934 static void
5935f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
5936{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005937 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005938 if (ga_grow(&ga_userinput, 1) == OK)
5939 {
5940 save_typeahead((tasave_T *)(ga_userinput.ga_data)
5941 + ga_userinput.ga_len);
5942 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005943 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005944 }
5945 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005946 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005947}
5948
5949/*
5950 * "inputsecret()" function
5951 */
5952 static void
5953f_inputsecret(typval_T *argvars, typval_T *rettv)
5954{
5955 ++cmdline_star;
5956 ++inputsecret_flag;
5957 f_input(argvars, rettv);
5958 --cmdline_star;
5959 --inputsecret_flag;
5960}
5961
5962/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005963 * "interrupt()" function
5964 */
5965 static void
5966f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5967{
5968 got_int = TRUE;
5969}
5970
5971/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005972 * "invert(expr)" function
5973 */
5974 static void
5975f_invert(typval_T *argvars, typval_T *rettv)
5976{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005977 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005978}
5979
5980/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005981 * "islocked()" function
5982 */
5983 static void
5984f_islocked(typval_T *argvars, typval_T *rettv)
5985{
5986 lval_T lv;
5987 char_u *end;
5988 dictitem_T *di;
5989
5990 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005991 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005992 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005993 if (end != NULL && lv.ll_name != NULL)
5994 {
5995 if (*end != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02005996 semsg(_(e_trailing_arg), end);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005997 else
5998 {
5999 if (lv.ll_tv == NULL)
6000 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01006001 di = find_var(lv.ll_name, NULL, TRUE);
6002 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006003 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006004 // Consider a variable locked when:
6005 // 1. the variable itself is locked
6006 // 2. the value of the variable is locked.
6007 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01006008 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
6009 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006010 }
6011 }
6012 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006013 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006014 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006015 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006016 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006017 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006018 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
6019 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006020 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006021 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
6022 }
6023 }
6024
6025 clear_lval(&lv);
6026}
6027
6028#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
6029/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02006030 * "isinf()" function
6031 */
6032 static void
6033f_isinf(typval_T *argvars, typval_T *rettv)
6034{
6035 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
6036 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
6037}
6038
6039/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006040 * "isnan()" function
6041 */
6042 static void
6043f_isnan(typval_T *argvars, typval_T *rettv)
6044{
6045 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
6046 && isnan(argvars[0].vval.v_float);
6047}
6048#endif
6049
6050/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006051 * "last_buffer_nr()" function.
6052 */
6053 static void
6054f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
6055{
6056 int n = 0;
6057 buf_T *buf;
6058
Bram Moolenaar29323592016-07-24 22:04:11 +02006059 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006060 if (n < buf->b_fnum)
6061 n = buf->b_fnum;
6062
6063 rettv->vval.v_number = n;
6064}
6065
6066/*
6067 * "len()" function
6068 */
6069 static void
6070f_len(typval_T *argvars, typval_T *rettv)
6071{
6072 switch (argvars[0].v_type)
6073 {
6074 case VAR_STRING:
6075 case VAR_NUMBER:
6076 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006077 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006078 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01006079 case VAR_BLOB:
6080 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
6081 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006082 case VAR_LIST:
6083 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
6084 break;
6085 case VAR_DICT:
6086 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
6087 break;
6088 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02006089 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006090 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01006091 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006092 case VAR_SPECIAL:
6093 case VAR_FLOAT:
6094 case VAR_FUNC:
6095 case VAR_PARTIAL:
6096 case VAR_JOB:
6097 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006098 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006099 break;
6100 }
6101}
6102
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006103 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01006104libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006105{
6106#ifdef FEAT_LIBCALL
6107 char_u *string_in;
6108 char_u **string_result;
6109 int nr_result;
6110#endif
6111
6112 rettv->v_type = type;
6113 if (type != VAR_NUMBER)
6114 rettv->vval.v_string = NULL;
6115
6116 if (check_restricted() || check_secure())
6117 return;
6118
6119#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006120 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006121 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
6122 {
6123 string_in = NULL;
6124 if (argvars[2].v_type == VAR_STRING)
6125 string_in = argvars[2].vval.v_string;
6126 if (type == VAR_NUMBER)
6127 string_result = NULL;
6128 else
6129 string_result = &rettv->vval.v_string;
6130 if (mch_libcall(argvars[0].vval.v_string,
6131 argvars[1].vval.v_string,
6132 string_in,
6133 argvars[2].vval.v_number,
6134 string_result,
6135 &nr_result) == OK
6136 && type == VAR_NUMBER)
6137 rettv->vval.v_number = nr_result;
6138 }
6139#endif
6140}
6141
6142/*
6143 * "libcall()" function
6144 */
6145 static void
6146f_libcall(typval_T *argvars, typval_T *rettv)
6147{
6148 libcall_common(argvars, rettv, VAR_STRING);
6149}
6150
6151/*
6152 * "libcallnr()" function
6153 */
6154 static void
6155f_libcallnr(typval_T *argvars, typval_T *rettv)
6156{
6157 libcall_common(argvars, rettv, VAR_NUMBER);
6158}
6159
6160/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006161 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006162 */
6163 static void
6164f_line(typval_T *argvars, typval_T *rettv)
6165{
6166 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006167 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006168 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006169 int id;
6170 tabpage_T *tp;
6171 win_T *wp;
6172 win_T *save_curwin;
6173 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006174
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02006175 if (argvars[1].v_type != VAR_UNKNOWN)
6176 {
6177 // use window specified in the second argument
6178 id = (int)tv_get_number(&argvars[1]);
6179 wp = win_id2wp_tp(id, &tp);
6180 if (wp != NULL && tp != NULL)
6181 {
6182 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
6183 == OK)
6184 {
6185 check_cursor();
6186 fp = var2fpos(&argvars[0], TRUE, &fnum);
6187 }
6188 restore_win_noblock(save_curwin, save_curtab, TRUE);
6189 }
6190 }
6191 else
6192 // use current window
6193 fp = var2fpos(&argvars[0], TRUE, &fnum);
6194
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006195 if (fp != NULL)
6196 lnum = fp->lnum;
6197 rettv->vval.v_number = lnum;
6198}
6199
6200/*
6201 * "line2byte(lnum)" function
6202 */
6203 static void
6204f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
6205{
6206#ifndef FEAT_BYTEOFF
6207 rettv->vval.v_number = -1;
6208#else
6209 linenr_T lnum;
6210
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006211 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006212 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
6213 rettv->vval.v_number = -1;
6214 else
6215 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
6216 if (rettv->vval.v_number >= 0)
6217 ++rettv->vval.v_number;
6218#endif
6219}
6220
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006221#ifdef FEAT_FLOAT
6222/*
6223 * "log()" function
6224 */
6225 static void
6226f_log(typval_T *argvars, typval_T *rettv)
6227{
6228 float_T f = 0.0;
6229
6230 rettv->v_type = VAR_FLOAT;
6231 if (get_float_arg(argvars, &f) == OK)
6232 rettv->vval.v_float = log(f);
6233 else
6234 rettv->vval.v_float = 0.0;
6235}
6236
6237/*
6238 * "log10()" function
6239 */
6240 static void
6241f_log10(typval_T *argvars, typval_T *rettv)
6242{
6243 float_T f = 0.0;
6244
6245 rettv->v_type = VAR_FLOAT;
6246 if (get_float_arg(argvars, &f) == OK)
6247 rettv->vval.v_float = log10(f);
6248 else
6249 rettv->vval.v_float = 0.0;
6250}
6251#endif
6252
6253#ifdef FEAT_LUA
6254/*
6255 * "luaeval()" function
6256 */
6257 static void
6258f_luaeval(typval_T *argvars, typval_T *rettv)
6259{
6260 char_u *str;
6261 char_u buf[NUMBUFLEN];
6262
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006263 if (check_restricted() || check_secure())
6264 return;
6265
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006266 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006267 do_luaeval(str, argvars + 1, rettv);
6268}
6269#endif
6270
6271/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006272 * "maparg()" function
6273 */
6274 static void
6275f_maparg(typval_T *argvars, typval_T *rettv)
6276{
6277 get_maparg(argvars, rettv, TRUE);
6278}
6279
6280/*
6281 * "mapcheck()" function
6282 */
6283 static void
6284f_mapcheck(typval_T *argvars, typval_T *rettv)
6285{
6286 get_maparg(argvars, rettv, FALSE);
6287}
6288
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006289typedef enum
6290{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006291 MATCH_END, // matchend()
6292 MATCH_MATCH, // match()
6293 MATCH_STR, // matchstr()
6294 MATCH_LIST, // matchlist()
6295 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006296} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006297
6298 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006299find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006300{
6301 char_u *str = NULL;
6302 long len = 0;
6303 char_u *expr = NULL;
6304 char_u *pat;
6305 regmatch_T regmatch;
6306 char_u patbuf[NUMBUFLEN];
6307 char_u strbuf[NUMBUFLEN];
6308 char_u *save_cpo;
6309 long start = 0;
6310 long nth = 1;
6311 colnr_T startcol = 0;
6312 int match = 0;
6313 list_T *l = NULL;
6314 listitem_T *li = NULL;
6315 long idx = 0;
6316 char_u *tofree = NULL;
6317
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006318 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006319 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01006320 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006321
6322 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006323 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006324 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006325 // type MATCH_LIST: return empty list when there are no matches.
6326 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006327 if (rettv_list_alloc(rettv) == FAIL)
6328 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006329 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006330 && (list_append_string(rettv->vval.v_list,
6331 (char_u *)"", 0) == FAIL
6332 || list_append_number(rettv->vval.v_list,
6333 (varnumber_T)-1) == FAIL
6334 || list_append_number(rettv->vval.v_list,
6335 (varnumber_T)-1) == FAIL
6336 || list_append_number(rettv->vval.v_list,
6337 (varnumber_T)-1) == FAIL))
6338 {
6339 list_free(rettv->vval.v_list);
6340 rettv->vval.v_list = NULL;
6341 goto theend;
6342 }
6343 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006344 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006345 {
6346 rettv->v_type = VAR_STRING;
6347 rettv->vval.v_string = NULL;
6348 }
6349
6350 if (argvars[0].v_type == VAR_LIST)
6351 {
6352 if ((l = argvars[0].vval.v_list) == NULL)
6353 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006354 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006355 li = l->lv_first;
6356 }
6357 else
6358 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006359 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006360 len = (long)STRLEN(str);
6361 }
6362
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006363 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006364 if (pat == NULL)
6365 goto theend;
6366
6367 if (argvars[2].v_type != VAR_UNKNOWN)
6368 {
6369 int error = FALSE;
6370
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006371 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006372 if (error)
6373 goto theend;
6374 if (l != NULL)
6375 {
6376 li = list_find(l, start);
6377 if (li == NULL)
6378 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006379 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006380 }
6381 else
6382 {
6383 if (start < 0)
6384 start = 0;
6385 if (start > len)
6386 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006387 // When "count" argument is there ignore matches before "start",
6388 // otherwise skip part of the string. Differs when pattern is "^"
6389 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006390 if (argvars[3].v_type != VAR_UNKNOWN)
6391 startcol = start;
6392 else
6393 {
6394 str += start;
6395 len -= start;
6396 }
6397 }
6398
6399 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006400 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006401 if (error)
6402 goto theend;
6403 }
6404
6405 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
6406 if (regmatch.regprog != NULL)
6407 {
6408 regmatch.rm_ic = p_ic;
6409
6410 for (;;)
6411 {
6412 if (l != NULL)
6413 {
6414 if (li == NULL)
6415 {
6416 match = FALSE;
6417 break;
6418 }
6419 vim_free(tofree);
6420 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
6421 if (str == NULL)
6422 break;
6423 }
6424
6425 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
6426
6427 if (match && --nth <= 0)
6428 break;
6429 if (l == NULL && !match)
6430 break;
6431
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006432 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006433 if (l != NULL)
6434 {
6435 li = li->li_next;
6436 ++idx;
6437 }
6438 else
6439 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006440 startcol = (colnr_T)(regmatch.startp[0]
6441 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006442 if (startcol > (colnr_T)len
6443 || str + startcol <= regmatch.startp[0])
6444 {
6445 match = FALSE;
6446 break;
6447 }
6448 }
6449 }
6450
6451 if (match)
6452 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006453 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006454 {
6455 listitem_T *li1 = rettv->vval.v_list->lv_first;
6456 listitem_T *li2 = li1->li_next;
6457 listitem_T *li3 = li2->li_next;
6458 listitem_T *li4 = li3->li_next;
6459
6460 vim_free(li1->li_tv.vval.v_string);
6461 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006462 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006463 li3->li_tv.vval.v_number =
6464 (varnumber_T)(regmatch.startp[0] - expr);
6465 li4->li_tv.vval.v_number =
6466 (varnumber_T)(regmatch.endp[0] - expr);
6467 if (l != NULL)
6468 li2->li_tv.vval.v_number = (varnumber_T)idx;
6469 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006470 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006471 {
6472 int i;
6473
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006474 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006475 for (i = 0; i < NSUBEXP; ++i)
6476 {
6477 if (regmatch.endp[i] == NULL)
6478 {
6479 if (list_append_string(rettv->vval.v_list,
6480 (char_u *)"", 0) == FAIL)
6481 break;
6482 }
6483 else if (list_append_string(rettv->vval.v_list,
6484 regmatch.startp[i],
6485 (int)(regmatch.endp[i] - regmatch.startp[i]))
6486 == FAIL)
6487 break;
6488 }
6489 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006490 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006491 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006492 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006493 if (l != NULL)
6494 copy_tv(&li->li_tv, rettv);
6495 else
6496 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006497 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006498 }
6499 else if (l != NULL)
6500 rettv->vval.v_number = idx;
6501 else
6502 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006503 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006504 rettv->vval.v_number =
6505 (varnumber_T)(regmatch.startp[0] - str);
6506 else
6507 rettv->vval.v_number =
6508 (varnumber_T)(regmatch.endp[0] - str);
6509 rettv->vval.v_number += (varnumber_T)(str - expr);
6510 }
6511 }
6512 vim_regfree(regmatch.regprog);
6513 }
6514
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006515theend:
6516 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006517 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006518 listitem_remove(rettv->vval.v_list,
6519 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006520 vim_free(tofree);
6521 p_cpo = save_cpo;
6522}
6523
6524/*
6525 * "match()" function
6526 */
6527 static void
6528f_match(typval_T *argvars, typval_T *rettv)
6529{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006530 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006531}
6532
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006533/*
6534 * "matchend()" function
6535 */
6536 static void
6537f_matchend(typval_T *argvars, typval_T *rettv)
6538{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006539 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006540}
6541
6542/*
6543 * "matchlist()" function
6544 */
6545 static void
6546f_matchlist(typval_T *argvars, typval_T *rettv)
6547{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006548 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006549}
6550
6551/*
6552 * "matchstr()" function
6553 */
6554 static void
6555f_matchstr(typval_T *argvars, typval_T *rettv)
6556{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006557 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006558}
6559
6560/*
6561 * "matchstrpos()" function
6562 */
6563 static void
6564f_matchstrpos(typval_T *argvars, typval_T *rettv)
6565{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006566 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006567}
6568
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006569 static void
6570max_min(typval_T *argvars, typval_T *rettv, int domax)
6571{
6572 varnumber_T n = 0;
6573 varnumber_T i;
6574 int error = FALSE;
6575
6576 if (argvars[0].v_type == VAR_LIST)
6577 {
6578 list_T *l;
6579 listitem_T *li;
6580
6581 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006582 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006583 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006584 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006585 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006586 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
6587 n = l->lv_u.nonmat.lv_start;
6588 else
6589 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
6590 * l->lv_u.nonmat.lv_stride;
6591 }
6592 else
6593 {
6594 li = l->lv_first;
6595 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006596 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006597 n = tv_get_number_chk(&li->li_tv, &error);
6598 for (;;)
6599 {
6600 li = li->li_next;
6601 if (li == NULL)
6602 break;
6603 i = tv_get_number_chk(&li->li_tv, &error);
6604 if (domax ? i > n : i < n)
6605 n = i;
6606 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006607 }
6608 }
6609 }
6610 }
6611 else if (argvars[0].v_type == VAR_DICT)
6612 {
6613 dict_T *d;
6614 int first = TRUE;
6615 hashitem_T *hi;
6616 int todo;
6617
6618 d = argvars[0].vval.v_dict;
6619 if (d != NULL)
6620 {
6621 todo = (int)d->dv_hashtab.ht_used;
6622 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
6623 {
6624 if (!HASHITEM_EMPTY(hi))
6625 {
6626 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006627 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006628 if (first)
6629 {
6630 n = i;
6631 first = FALSE;
6632 }
6633 else if (domax ? i > n : i < n)
6634 n = i;
6635 }
6636 }
6637 }
6638 }
6639 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006640 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006641 rettv->vval.v_number = error ? 0 : n;
6642}
6643
6644/*
6645 * "max()" function
6646 */
6647 static void
6648f_max(typval_T *argvars, typval_T *rettv)
6649{
6650 max_min(argvars, rettv, TRUE);
6651}
6652
6653/*
6654 * "min()" function
6655 */
6656 static void
6657f_min(typval_T *argvars, typval_T *rettv)
6658{
6659 max_min(argvars, rettv, FALSE);
6660}
6661
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006662#if defined(FEAT_MZSCHEME) || defined(PROTO)
6663/*
6664 * "mzeval()" function
6665 */
6666 static void
6667f_mzeval(typval_T *argvars, typval_T *rettv)
6668{
6669 char_u *str;
6670 char_u buf[NUMBUFLEN];
6671
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006672 if (check_restricted() || check_secure())
6673 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006674 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006675 do_mzeval(str, rettv);
6676}
6677
6678 void
6679mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
6680{
6681 typval_T argvars[3];
6682
6683 argvars[0].v_type = VAR_STRING;
6684 argvars[0].vval.v_string = name;
6685 copy_tv(args, &argvars[1]);
6686 argvars[2].v_type = VAR_UNKNOWN;
6687 f_call(argvars, rettv);
6688 clear_tv(&argvars[1]);
6689}
6690#endif
6691
6692/*
6693 * "nextnonblank()" function
6694 */
6695 static void
6696f_nextnonblank(typval_T *argvars, typval_T *rettv)
6697{
6698 linenr_T lnum;
6699
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006700 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006701 {
6702 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
6703 {
6704 lnum = 0;
6705 break;
6706 }
6707 if (*skipwhite(ml_get(lnum)) != NUL)
6708 break;
6709 }
6710 rettv->vval.v_number = lnum;
6711}
6712
6713/*
6714 * "nr2char()" function
6715 */
6716 static void
6717f_nr2char(typval_T *argvars, typval_T *rettv)
6718{
6719 char_u buf[NUMBUFLEN];
6720
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006721 if (has_mbyte)
6722 {
6723 int utf8 = 0;
6724
6725 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaared6a4302020-09-05 20:29:41 +02006726 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006727 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01006728 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006729 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006730 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006731 }
6732 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006733 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006734 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006735 buf[1] = NUL;
6736 }
6737 rettv->v_type = VAR_STRING;
6738 rettv->vval.v_string = vim_strsave(buf);
6739}
6740
6741/*
6742 * "or(expr, expr)" function
6743 */
6744 static void
6745f_or(typval_T *argvars, typval_T *rettv)
6746{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006747 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
6748 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006749}
6750
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006751#ifdef FEAT_PERL
6752/*
6753 * "perleval()" function
6754 */
6755 static void
6756f_perleval(typval_T *argvars, typval_T *rettv)
6757{
6758 char_u *str;
6759 char_u buf[NUMBUFLEN];
6760
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006761 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006762 do_perleval(str, rettv);
6763}
6764#endif
6765
6766#ifdef FEAT_FLOAT
6767/*
6768 * "pow()" function
6769 */
6770 static void
6771f_pow(typval_T *argvars, typval_T *rettv)
6772{
6773 float_T fx = 0.0, fy = 0.0;
6774
6775 rettv->v_type = VAR_FLOAT;
6776 if (get_float_arg(argvars, &fx) == OK
6777 && get_float_arg(&argvars[1], &fy) == OK)
6778 rettv->vval.v_float = pow(fx, fy);
6779 else
6780 rettv->vval.v_float = 0.0;
6781}
6782#endif
6783
6784/*
6785 * "prevnonblank()" function
6786 */
6787 static void
6788f_prevnonblank(typval_T *argvars, typval_T *rettv)
6789{
6790 linenr_T lnum;
6791
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006792 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006793 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
6794 lnum = 0;
6795 else
6796 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
6797 --lnum;
6798 rettv->vval.v_number = lnum;
6799}
6800
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006801// This dummy va_list is here because:
6802// - passing a NULL pointer doesn't work when va_list isn't a pointer
6803// - locally in the function results in a "used before set" warning
6804// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006805static va_list ap;
6806
6807/*
6808 * "printf()" function
6809 */
6810 static void
6811f_printf(typval_T *argvars, typval_T *rettv)
6812{
6813 char_u buf[NUMBUFLEN];
6814 int len;
6815 char_u *s;
6816 int saved_did_emsg = did_emsg;
6817 char *fmt;
6818
6819 rettv->v_type = VAR_STRING;
6820 rettv->vval.v_string = NULL;
6821
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006822 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006823 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006824 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006825 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006826 if (!did_emsg)
6827 {
6828 s = alloc(len + 1);
6829 if (s != NULL)
6830 {
6831 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006832 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
6833 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006834 }
6835 }
6836 did_emsg |= saved_did_emsg;
6837}
6838
6839/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006840 * "pum_getpos()" function
6841 */
6842 static void
6843f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6844{
6845 if (rettv_dict_alloc(rettv) != OK)
6846 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006847 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006848}
6849
6850/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006851 * "pumvisible()" function
6852 */
6853 static void
6854f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6855{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006856 if (pum_visible())
6857 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006858}
6859
6860#ifdef FEAT_PYTHON3
6861/*
6862 * "py3eval()" function
6863 */
6864 static void
6865f_py3eval(typval_T *argvars, typval_T *rettv)
6866{
6867 char_u *str;
6868 char_u buf[NUMBUFLEN];
6869
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006870 if (check_restricted() || check_secure())
6871 return;
6872
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006873 if (p_pyx == 0)
6874 p_pyx = 3;
6875
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006876 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006877 do_py3eval(str, rettv);
6878}
6879#endif
6880
6881#ifdef FEAT_PYTHON
6882/*
6883 * "pyeval()" function
6884 */
6885 static void
6886f_pyeval(typval_T *argvars, typval_T *rettv)
6887{
6888 char_u *str;
6889 char_u buf[NUMBUFLEN];
6890
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006891 if (check_restricted() || check_secure())
6892 return;
6893
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006894 if (p_pyx == 0)
6895 p_pyx = 2;
6896
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006897 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006898 do_pyeval(str, rettv);
6899}
6900#endif
6901
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006902#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
6903/*
6904 * "pyxeval()" function
6905 */
6906 static void
6907f_pyxeval(typval_T *argvars, typval_T *rettv)
6908{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006909 if (check_restricted() || check_secure())
6910 return;
6911
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006912# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
6913 init_pyxversion();
6914 if (p_pyx == 2)
6915 f_pyeval(argvars, rettv);
6916 else
6917 f_py3eval(argvars, rettv);
6918# elif defined(FEAT_PYTHON)
6919 f_pyeval(argvars, rettv);
6920# elif defined(FEAT_PYTHON3)
6921 f_py3eval(argvars, rettv);
6922# endif
6923}
6924#endif
6925
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006926static UINT32_T srand_seed_for_testing = 0;
6927static int srand_seed_for_testing_is_used = FALSE;
6928
6929 static void
6930f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
6931{
6932 if (argvars[0].v_type == VAR_UNKNOWN)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006933 srand_seed_for_testing_is_used = FALSE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006934 else
6935 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006936 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
6937 srand_seed_for_testing_is_used = TRUE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006938 }
6939}
6940
6941 static void
6942init_srand(UINT32_T *x)
6943{
6944#ifndef MSWIN
6945 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6946#endif
6947
6948 if (srand_seed_for_testing_is_used)
6949 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006950 *x = srand_seed_for_testing;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006951 return;
6952 }
6953#ifndef MSWIN
6954 if (dev_urandom_state != FAIL)
6955 {
6956 int fd = open("/dev/urandom", O_RDONLY);
6957 struct {
6958 union {
6959 UINT32_T number;
6960 char bytes[sizeof(UINT32_T)];
6961 } contents;
6962 } buf;
6963
6964 // Attempt reading /dev/urandom.
6965 if (fd == -1)
6966 dev_urandom_state = FAIL;
6967 else
6968 {
6969 buf.contents.number = 0;
6970 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6971 != sizeof(UINT32_T))
6972 dev_urandom_state = FAIL;
6973 else
6974 {
6975 dev_urandom_state = OK;
6976 *x = buf.contents.number;
6977 }
6978 close(fd);
6979 }
6980 }
6981 if (dev_urandom_state != OK)
6982 // Reading /dev/urandom doesn't work, fall back to time().
6983#endif
6984 *x = vim_time();
6985}
6986
6987#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6988#define SPLITMIX32(x, z) ( \
6989 z = (x += 0x9e3779b9), \
6990 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6991 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6992 z ^ (z >> 16) \
6993 )
6994#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6995 result = ROTL(y * 5, 7) * 9; \
6996 t = y << 9; \
6997 z ^= x; \
6998 w ^= y; \
6999 y ^= z, x ^= w; \
7000 z ^= t; \
7001 w = ROTL(w, 11);
7002
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007003/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007004 * "rand()" function
7005 */
7006 static void
7007f_rand(typval_T *argvars, typval_T *rettv)
7008{
7009 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007010 static UINT32_T gx, gy, gz, gw;
7011 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007012 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar0fd797e2020-11-05 20:46:32 +01007013 UINT32_T x = 0, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007014
7015 if (argvars[0].v_type == VAR_UNKNOWN)
7016 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007017 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007018 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007019 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007020 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007021 init_srand(&x);
7022
7023 gx = SPLITMIX32(x, z);
7024 gy = SPLITMIX32(x, z);
7025 gz = SPLITMIX32(x, z);
7026 gw = SPLITMIX32(x, z);
7027 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007028 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007029
7030 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007031 }
7032 else if (argvars[0].v_type == VAR_LIST)
7033 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007034 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007035 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007036 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007037
7038 lx = list_find(l, 0L);
7039 ly = list_find(l, 1L);
7040 lz = list_find(l, 2L);
7041 lw = list_find(l, 3L);
7042 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
7043 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
7044 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
7045 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
7046 x = (UINT32_T)lx->li_tv.vval.v_number;
7047 y = (UINT32_T)ly->li_tv.vval.v_number;
7048 z = (UINT32_T)lz->li_tv.vval.v_number;
7049 w = (UINT32_T)lw->li_tv.vval.v_number;
7050
7051 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
7052
7053 lx->li_tv.vval.v_number = (varnumber_T)x;
7054 ly->li_tv.vval.v_number = (varnumber_T)y;
7055 lz->li_tv.vval.v_number = (varnumber_T)z;
7056 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007057 }
7058 else
7059 goto theend;
7060
7061 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007062 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007063 return;
7064
7065theend:
7066 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01007067 rettv->v_type = VAR_NUMBER;
7068 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007069}
7070
7071/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01007072 * "srand()" function
7073 */
7074 static void
7075f_srand(typval_T *argvars, typval_T *rettv)
7076{
7077 UINT32_T x = 0, z;
7078
7079 if (rettv_list_alloc(rettv) == FAIL)
7080 return;
7081 if (argvars[0].v_type == VAR_UNKNOWN)
7082 {
7083 init_srand(&x);
7084 }
7085 else
7086 {
7087 int error = FALSE;
7088
7089 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
7090 if (error)
7091 return;
7092 }
7093
7094 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7095 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7096 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7097 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
7098}
7099
7100#undef ROTL
7101#undef SPLITMIX32
7102#undef SHUFFLE_XOSHIRO128STARSTAR
7103
7104/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007105 * "range()" function
7106 */
7107 static void
7108f_range(typval_T *argvars, typval_T *rettv)
7109{
7110 varnumber_T start;
7111 varnumber_T end;
7112 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007113 int error = FALSE;
7114
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007115 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007116 if (argvars[1].v_type == VAR_UNKNOWN)
7117 {
7118 end = start - 1;
7119 start = 0;
7120 }
7121 else
7122 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007123 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007124 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007125 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007126 }
7127
7128 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007129 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007130 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007131 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007132 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007133 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007134 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007135 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007136 list_T *list = rettv->vval.v_list;
7137
7138 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007139 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007140 // be called.
7141 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01007142 list->lv_u.nonmat.lv_start = start;
7143 list->lv_u.nonmat.lv_end = end;
7144 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007145 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007146 }
7147}
7148
7149/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007150 * Materialize "list".
7151 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007152 */
7153 void
7154range_list_materialize(list_T *list)
7155{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007156 varnumber_T start = list->lv_u.nonmat.lv_start;
7157 varnumber_T end = list->lv_u.nonmat.lv_end;
7158 int stride = list->lv_u.nonmat.lv_stride;
7159 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01007160
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007161 list->lv_first = NULL;
7162 list->lv_u.mat.lv_last = NULL;
7163 list->lv_len = 0;
7164 list->lv_u.mat.lv_idx_item = NULL;
7165 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
7166 if (list_append_number(list, (varnumber_T)i) == FAIL)
7167 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007168}
7169
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007170/*
7171 * "getreginfo()" function
7172 */
7173 static void
7174f_getreginfo(typval_T *argvars, typval_T *rettv)
7175{
7176 char_u *strregname;
7177 int regname;
7178 char_u buf[NUMBUFLEN + 2];
7179 long reglen = 0;
7180 dict_T *dict;
7181 list_T *list;
7182
7183 if (argvars[0].v_type != VAR_UNKNOWN)
7184 {
7185 strregname = tv_get_string_chk(&argvars[0]);
7186 if (strregname == NULL)
7187 return;
7188 }
7189 else
7190 strregname = get_vim_var_str(VV_REG);
7191
7192 regname = (strregname == NULL ? '"' : *strregname);
7193 if (regname == 0 || regname == '@')
7194 regname = '"';
7195
7196 if (rettv_dict_alloc(rettv) == FAIL)
7197 return;
7198 dict = rettv->vval.v_dict;
7199
7200 list = (list_T *)get_reg_contents(regname, GREG_EXPR_SRC | GREG_LIST);
7201 if (list == NULL)
7202 return;
Bram Moolenaar91639192020-06-29 19:55:58 +02007203 (void)dict_add_list(dict, "regcontents", list);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007204
7205 buf[0] = NUL;
7206 buf[1] = NUL;
7207 switch (get_reg_type(regname, &reglen))
7208 {
7209 case MLINE: buf[0] = 'V'; break;
7210 case MCHAR: buf[0] = 'v'; break;
7211 case MBLOCK:
7212 vim_snprintf((char *)buf, sizeof(buf), "%c%ld", Ctrl_V,
7213 reglen + 1);
7214 break;
7215 }
Bram Moolenaar91639192020-06-29 19:55:58 +02007216 (void)dict_add_string(dict, (char *)"regtype", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007217
7218 buf[0] = get_register_name(get_unname_register());
7219 buf[1] = NUL;
7220 if (regname == '"')
Bram Moolenaar91639192020-06-29 19:55:58 +02007221 (void)dict_add_string(dict, (char *)"points_to", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007222 else
7223 {
7224 dictitem_T *item = dictitem_alloc((char_u *)"isunnamed");
7225
7226 if (item != NULL)
7227 {
7228 item->di_tv.v_type = VAR_SPECIAL;
7229 item->di_tv.vval.v_number = regname == buf[0]
7230 ? VVAL_TRUE : VVAL_FALSE;
Bram Moolenaar91639192020-06-29 19:55:58 +02007231 (void)dict_add(dict, item);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007232 }
7233 }
7234}
7235
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02007236 static void
7237return_register(int regname, typval_T *rettv)
7238{
7239 char_u buf[2] = {0, 0};
7240
7241 buf[0] = (char_u)regname;
7242 rettv->v_type = VAR_STRING;
7243 rettv->vval.v_string = vim_strsave(buf);
7244}
7245
7246/*
7247 * "reg_executing()" function
7248 */
7249 static void
7250f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
7251{
7252 return_register(reg_executing, rettv);
7253}
7254
7255/*
7256 * "reg_recording()" function
7257 */
7258 static void
7259f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
7260{
7261 return_register(reg_recording, rettv);
7262}
7263
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01007264/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007265 * "rename({from}, {to})" function
7266 */
7267 static void
7268f_rename(typval_T *argvars, typval_T *rettv)
7269{
7270 char_u buf[NUMBUFLEN];
7271
7272 if (check_restricted() || check_secure())
7273 rettv->vval.v_number = -1;
7274 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007275 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
7276 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007277}
7278
7279/*
7280 * "repeat()" function
7281 */
7282 static void
7283f_repeat(typval_T *argvars, typval_T *rettv)
7284{
7285 char_u *p;
7286 int n;
7287 int slen;
7288 int len;
7289 char_u *r;
7290 int i;
7291
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007292 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007293 if (argvars[0].v_type == VAR_LIST)
7294 {
7295 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
7296 while (n-- > 0)
7297 if (list_extend(rettv->vval.v_list,
7298 argvars[0].vval.v_list, NULL) == FAIL)
7299 break;
7300 }
7301 else
7302 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007303 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007304 rettv->v_type = VAR_STRING;
7305 rettv->vval.v_string = NULL;
7306
7307 slen = (int)STRLEN(p);
7308 len = slen * n;
7309 if (len <= 0)
7310 return;
7311
7312 r = alloc(len + 1);
7313 if (r != NULL)
7314 {
7315 for (i = 0; i < n; i++)
7316 mch_memmove(r + i * slen, p, (size_t)slen);
7317 r[len] = NUL;
7318 }
7319
7320 rettv->vval.v_string = r;
7321 }
7322}
7323
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007324#define SP_NOMOVE 0x01 // don't move cursor
7325#define SP_REPEAT 0x02 // repeat to find outer pair
7326#define SP_RETCOUNT 0x04 // return matchcount
7327#define SP_SETPCMARK 0x08 // set previous context mark
7328#define SP_START 0x10 // accept match at start position
7329#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
7330#define SP_END 0x40 // leave cursor at end of match
7331#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007332
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007333/*
7334 * Get flags for a search function.
7335 * Possibly sets "p_ws".
7336 * Returns BACKWARD, FORWARD or zero (for an error).
7337 */
7338 static int
7339get_search_arg(typval_T *varp, int *flagsp)
7340{
7341 int dir = FORWARD;
7342 char_u *flags;
7343 char_u nbuf[NUMBUFLEN];
7344 int mask;
7345
7346 if (varp->v_type != VAR_UNKNOWN)
7347 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007348 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007349 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007350 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007351 while (*flags != NUL)
7352 {
7353 switch (*flags)
7354 {
7355 case 'b': dir = BACKWARD; break;
7356 case 'w': p_ws = TRUE; break;
7357 case 'W': p_ws = FALSE; break;
7358 default: mask = 0;
7359 if (flagsp != NULL)
7360 switch (*flags)
7361 {
7362 case 'c': mask = SP_START; break;
7363 case 'e': mask = SP_END; break;
7364 case 'm': mask = SP_RETCOUNT; break;
7365 case 'n': mask = SP_NOMOVE; break;
7366 case 'p': mask = SP_SUBPAT; break;
7367 case 'r': mask = SP_REPEAT; break;
7368 case 's': mask = SP_SETPCMARK; break;
7369 case 'z': mask = SP_COLUMN; break;
7370 }
7371 if (mask == 0)
7372 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007373 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007374 dir = 0;
7375 }
7376 else
7377 *flagsp |= mask;
7378 }
7379 if (dir == 0)
7380 break;
7381 ++flags;
7382 }
7383 }
7384 return dir;
7385}
7386
7387/*
7388 * Shared by search() and searchpos() functions.
7389 */
7390 static int
7391search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
7392{
7393 int flags;
7394 char_u *pat;
7395 pos_T pos;
7396 pos_T save_cursor;
7397 int save_p_ws = p_ws;
7398 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007399 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007400 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007401#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007402 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007403 long time_limit = 0;
7404#endif
7405 int options = SEARCH_KEEP;
7406 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007407 searchit_arg_T sia;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007408 int use_skip = FALSE;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007409 pos_T firstpos;
7410
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007411 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007412 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007413 if (dir == 0)
7414 goto theend;
7415 flags = *flagsp;
7416 if (flags & SP_START)
7417 options |= SEARCH_START;
7418 if (flags & SP_END)
7419 options |= SEARCH_END;
7420 if (flags & SP_COLUMN)
7421 options |= SEARCH_COL;
7422
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007423 // Optional arguments: line number to stop searching, timeout and skip.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007424 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
7425 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007426 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007427 if (lnum_stop < 0)
7428 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007429 if (argvars[3].v_type != VAR_UNKNOWN)
7430 {
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007431#ifdef FEAT_RELTIME
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007432 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007433 if (time_limit < 0)
7434 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007435#endif
Bram Moolenaara9c01042020-06-07 14:50:50 +02007436 use_skip = eval_expr_valid_arg(&argvars[4]);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007437 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007438 }
7439
7440#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007441 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007442 profile_setlimit(time_limit, &tm);
7443#endif
7444
7445 /*
7446 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
7447 * Check to make sure only those flags are set.
7448 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
7449 * flags cannot be set. Check for that condition also.
7450 */
7451 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
7452 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7453 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007454 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007455 goto theend;
7456 }
7457
7458 pos = save_cursor = curwin->w_cursor;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007459 CLEAR_FIELD(firstpos);
Bram Moolenaara80faa82020-04-12 19:37:17 +02007460 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007461 sia.sa_stop_lnum = (linenr_T)lnum_stop;
7462#ifdef FEAT_RELTIME
7463 sia.sa_tm = &tm;
7464#endif
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007465
7466 // Repeat until {skip} returns FALSE.
7467 for (;;)
7468 {
7469 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007470 options, RE_SEARCH, &sia);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007471 // finding the first match again means there is no match where {skip}
7472 // evaluates to zero.
7473 if (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos))
7474 subpatnum = FAIL;
7475
Bram Moolenaara9c01042020-06-07 14:50:50 +02007476 if (subpatnum == FAIL || !use_skip)
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007477 // didn't find it or no skip argument
7478 break;
7479 firstpos = pos;
7480
Bram Moolenaara9c01042020-06-07 14:50:50 +02007481 // If the skip expression matches, ignore this match.
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007482 {
7483 int do_skip;
7484 int err;
7485 pos_T save_pos = curwin->w_cursor;
7486
7487 curwin->w_cursor = pos;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007488 err = FALSE;
7489 do_skip = eval_expr_to_bool(&argvars[4], &err);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007490 curwin->w_cursor = save_pos;
7491 if (err)
7492 {
7493 // Evaluating {skip} caused an error, break here.
7494 subpatnum = FAIL;
7495 break;
7496 }
7497 if (!do_skip)
7498 break;
7499 }
7500 }
7501
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007502 if (subpatnum != FAIL)
7503 {
7504 if (flags & SP_SUBPAT)
7505 retval = subpatnum;
7506 else
7507 retval = pos.lnum;
7508 if (flags & SP_SETPCMARK)
7509 setpcmark();
7510 curwin->w_cursor = pos;
7511 if (match_pos != NULL)
7512 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007513 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007514 match_pos->lnum = pos.lnum;
7515 match_pos->col = pos.col + 1;
7516 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007517 // "/$" will put the cursor after the end of the line, may need to
7518 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007519 check_cursor();
7520 }
7521
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007522 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007523 if (flags & SP_NOMOVE)
7524 curwin->w_cursor = save_cursor;
7525 else
7526 curwin->w_set_curswant = TRUE;
7527theend:
7528 p_ws = save_p_ws;
7529
7530 return retval;
7531}
7532
7533#ifdef FEAT_FLOAT
7534
7535/*
7536 * round() is not in C90, use ceil() or floor() instead.
7537 */
7538 float_T
7539vim_round(float_T f)
7540{
7541 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
7542}
7543
7544/*
7545 * "round({float})" function
7546 */
7547 static void
7548f_round(typval_T *argvars, typval_T *rettv)
7549{
7550 float_T f = 0.0;
7551
7552 rettv->v_type = VAR_FLOAT;
7553 if (get_float_arg(argvars, &f) == OK)
7554 rettv->vval.v_float = vim_round(f);
7555 else
7556 rettv->vval.v_float = 0.0;
7557}
7558#endif
7559
Bram Moolenaare99be0e2019-03-26 22:51:09 +01007560#ifdef FEAT_RUBY
7561/*
7562 * "rubyeval()" function
7563 */
7564 static void
7565f_rubyeval(typval_T *argvars, typval_T *rettv)
7566{
7567 char_u *str;
7568 char_u buf[NUMBUFLEN];
7569
7570 str = tv_get_string_buf(&argvars[0], buf);
7571 do_rubyeval(str, rettv);
7572}
7573#endif
7574
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007575/*
7576 * "screenattr()" function
7577 */
7578 static void
7579f_screenattr(typval_T *argvars, typval_T *rettv)
7580{
7581 int row;
7582 int col;
7583 int c;
7584
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007585 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7586 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007587 if (row < 0 || row >= screen_Rows
7588 || col < 0 || col >= screen_Columns)
7589 c = -1;
7590 else
7591 c = ScreenAttrs[LineOffset[row] + col];
7592 rettv->vval.v_number = c;
7593}
7594
7595/*
7596 * "screenchar()" function
7597 */
7598 static void
7599f_screenchar(typval_T *argvars, typval_T *rettv)
7600{
7601 int row;
7602 int col;
7603 int off;
7604 int c;
7605
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007606 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7607 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007608 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007609 c = -1;
7610 else
7611 {
7612 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007613 if (enc_utf8 && ScreenLinesUC[off] != 0)
7614 c = ScreenLinesUC[off];
7615 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007616 c = ScreenLines[off];
7617 }
7618 rettv->vval.v_number = c;
7619}
7620
7621/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007622 * "screenchars()" function
7623 */
7624 static void
7625f_screenchars(typval_T *argvars, typval_T *rettv)
7626{
7627 int row;
7628 int col;
7629 int off;
7630 int c;
7631 int i;
7632
7633 if (rettv_list_alloc(rettv) == FAIL)
7634 return;
7635 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7636 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7637 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7638 return;
7639
7640 off = LineOffset[row] + col;
7641 if (enc_utf8 && ScreenLinesUC[off] != 0)
7642 c = ScreenLinesUC[off];
7643 else
7644 c = ScreenLines[off];
7645 list_append_number(rettv->vval.v_list, (varnumber_T)c);
7646
7647 if (enc_utf8)
7648
7649 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7650 list_append_number(rettv->vval.v_list,
7651 (varnumber_T)ScreenLinesC[i][off]);
7652}
7653
7654/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007655 * "screencol()" function
7656 *
7657 * First column is 1 to be consistent with virtcol().
7658 */
7659 static void
7660f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
7661{
7662 rettv->vval.v_number = screen_screencol() + 1;
7663}
7664
7665/*
7666 * "screenrow()" function
7667 */
7668 static void
7669f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
7670{
7671 rettv->vval.v_number = screen_screenrow() + 1;
7672}
7673
7674/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007675 * "screenstring()" function
7676 */
7677 static void
7678f_screenstring(typval_T *argvars, typval_T *rettv)
7679{
7680 int row;
7681 int col;
7682 int off;
7683 int c;
7684 int i;
7685 char_u buf[MB_MAXBYTES + 1];
7686 int buflen = 0;
7687
7688 rettv->vval.v_string = NULL;
7689 rettv->v_type = VAR_STRING;
7690
7691 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7692 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7693 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7694 return;
7695
7696 off = LineOffset[row] + col;
7697 if (enc_utf8 && ScreenLinesUC[off] != 0)
7698 c = ScreenLinesUC[off];
7699 else
7700 c = ScreenLines[off];
7701 buflen += mb_char2bytes(c, buf);
7702
7703 if (enc_utf8)
7704 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7705 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
7706
7707 buf[buflen] = NUL;
7708 rettv->vval.v_string = vim_strsave(buf);
7709}
7710
7711/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007712 * "search()" function
7713 */
7714 static void
7715f_search(typval_T *argvars, typval_T *rettv)
7716{
7717 int flags = 0;
7718
7719 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
7720}
7721
7722/*
7723 * "searchdecl()" function
7724 */
7725 static void
7726f_searchdecl(typval_T *argvars, typval_T *rettv)
7727{
Bram Moolenaar30788d32020-09-05 21:35:16 +02007728 int locally = TRUE;
7729 int thisblock = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007730 int error = FALSE;
7731 char_u *name;
7732
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007733 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007734
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007735 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007736 if (argvars[1].v_type != VAR_UNKNOWN)
7737 {
Bram Moolenaar30788d32020-09-05 21:35:16 +02007738 locally = !(int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007739 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar30788d32020-09-05 21:35:16 +02007740 thisblock = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007741 }
7742 if (!error && name != NULL)
7743 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
7744 locally, thisblock, SEARCH_KEEP) == FAIL;
7745}
7746
7747/*
7748 * Used by searchpair() and searchpairpos()
7749 */
7750 static int
7751searchpair_cmn(typval_T *argvars, pos_T *match_pos)
7752{
7753 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01007754 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007755 int save_p_ws = p_ws;
7756 int dir;
7757 int flags = 0;
7758 char_u nbuf1[NUMBUFLEN];
7759 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007760 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007761 long lnum_stop = 0;
7762 long time_limit = 0;
7763
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007764 // Get the three pattern arguments: start, middle, end. Will result in an
7765 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007766 spat = tv_get_string_chk(&argvars[0]);
7767 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
7768 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007769 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007770 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007771
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007772 // Handle the optional fourth argument: flags
7773 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007774 if (dir == 0)
7775 goto theend;
7776
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007777 // Don't accept SP_END or SP_SUBPAT.
7778 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007779 if ((flags & (SP_END | SP_SUBPAT)) != 0
7780 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7781 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007782 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007783 goto theend;
7784 }
7785
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007786 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007787 if (flags & SP_REPEAT)
7788 p_ws = FALSE;
7789
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007790 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007791 if (argvars[3].v_type == VAR_UNKNOWN
7792 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01007793 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007794 else
7795 {
Bram Moolenaara9c01042020-06-07 14:50:50 +02007796 // Type is checked later.
Bram Moolenaar48570482017-10-30 21:48:41 +01007797 skip = &argvars[4];
Bram Moolenaara9c01042020-06-07 14:50:50 +02007798
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007799 if (argvars[5].v_type != VAR_UNKNOWN)
7800 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007801 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007802 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007803 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007804 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007805 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007806 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007807#ifdef FEAT_RELTIME
7808 if (argvars[6].v_type != VAR_UNKNOWN)
7809 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007810 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007811 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007812 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007813 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007814 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007815 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007816 }
7817#endif
7818 }
7819 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007820
7821 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
7822 match_pos, lnum_stop, time_limit);
7823
7824theend:
7825 p_ws = save_p_ws;
7826
7827 return retval;
7828}
7829
7830/*
7831 * "searchpair()" function
7832 */
7833 static void
7834f_searchpair(typval_T *argvars, typval_T *rettv)
7835{
7836 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
7837}
7838
7839/*
7840 * "searchpairpos()" function
7841 */
7842 static void
7843f_searchpairpos(typval_T *argvars, typval_T *rettv)
7844{
7845 pos_T match_pos;
7846 int lnum = 0;
7847 int col = 0;
7848
7849 if (rettv_list_alloc(rettv) == FAIL)
7850 return;
7851
7852 if (searchpair_cmn(argvars, &match_pos) > 0)
7853 {
7854 lnum = match_pos.lnum;
7855 col = match_pos.col;
7856 }
7857
7858 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7859 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7860}
7861
7862/*
7863 * Search for a start/middle/end thing.
7864 * Used by searchpair(), see its documentation for the details.
7865 * Returns 0 or -1 for no match,
7866 */
7867 long
7868do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007869 char_u *spat, // start pattern
7870 char_u *mpat, // middle pattern
7871 char_u *epat, // end pattern
7872 int dir, // BACKWARD or FORWARD
7873 typval_T *skip, // skip expression
7874 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007875 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007876 linenr_T lnum_stop, // stop at this line if not zero
7877 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007878{
7879 char_u *save_cpo;
7880 char_u *pat, *pat2 = NULL, *pat3 = NULL;
7881 long retval = 0;
7882 pos_T pos;
7883 pos_T firstpos;
7884 pos_T foundpos;
7885 pos_T save_cursor;
7886 pos_T save_pos;
7887 int n;
7888 int r;
7889 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01007890 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007891 int err;
7892 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007893#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007894 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007895#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007896
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007897 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007898 save_cpo = p_cpo;
7899 p_cpo = empty_option;
7900
7901#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007902 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007903 profile_setlimit(time_limit, &tm);
7904#endif
7905
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007906 // Make two search patterns: start/end (pat2, for in nested pairs) and
7907 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02007908 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
7909 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007910 if (pat2 == NULL || pat3 == NULL)
7911 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01007912 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007913 if (*mpat == NUL)
7914 STRCPY(pat3, pat2);
7915 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01007916 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007917 spat, epat, mpat);
7918 if (flags & SP_START)
7919 options |= SEARCH_START;
7920
Bram Moolenaar48570482017-10-30 21:48:41 +01007921 if (skip != NULL)
Bram Moolenaara9c01042020-06-07 14:50:50 +02007922 use_skip = eval_expr_valid_arg(skip);
Bram Moolenaar48570482017-10-30 21:48:41 +01007923
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007924 save_cursor = curwin->w_cursor;
7925 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007926 CLEAR_POS(&firstpos);
7927 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007928 pat = pat3;
7929 for (;;)
7930 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007931 searchit_arg_T sia;
7932
Bram Moolenaara80faa82020-04-12 19:37:17 +02007933 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007934 sia.sa_stop_lnum = lnum_stop;
7935#ifdef FEAT_RELTIME
7936 sia.sa_tm = &tm;
7937#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01007938 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007939 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007940 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007941 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007942 break;
7943
7944 if (firstpos.lnum == 0)
7945 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007946 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007947 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007948 // Found the same position again. Can happen with a pattern that
7949 // has "\zs" at the end and searching backwards. Advance one
7950 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007951 if (dir == BACKWARD)
7952 decl(&pos);
7953 else
7954 incl(&pos);
7955 }
7956 foundpos = pos;
7957
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007958 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007959 options &= ~SEARCH_START;
7960
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007961 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01007962 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007963 {
7964 save_pos = curwin->w_cursor;
7965 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01007966 err = FALSE;
7967 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007968 curwin->w_cursor = save_pos;
7969 if (err)
7970 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007971 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007972 curwin->w_cursor = save_cursor;
7973 retval = -1;
7974 break;
7975 }
7976 if (r)
7977 continue;
7978 }
7979
7980 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
7981 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007982 // Found end when searching backwards or start when searching
7983 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007984 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007985 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007986 }
7987 else
7988 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007989 // Found end when searching forward or start when searching
7990 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007991 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007992 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007993 }
7994
7995 if (nest == 0)
7996 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007997 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007998 if (flags & SP_RETCOUNT)
7999 ++retval;
8000 else
8001 retval = pos.lnum;
8002 if (flags & SP_SETPCMARK)
8003 setpcmark();
8004 curwin->w_cursor = pos;
8005 if (!(flags & SP_REPEAT))
8006 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008007 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008008 }
8009 }
8010
8011 if (match_pos != NULL)
8012 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008013 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008014 match_pos->lnum = curwin->w_cursor.lnum;
8015 match_pos->col = curwin->w_cursor.col + 1;
8016 }
8017
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008018 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008019 if ((flags & SP_NOMOVE) || retval == 0)
8020 curwin->w_cursor = save_cursor;
8021
8022theend:
8023 vim_free(pat2);
8024 vim_free(pat3);
8025 if (p_cpo == empty_option)
8026 p_cpo = save_cpo;
8027 else
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008028 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008029 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008030 // If it's still empty it was changed and restored, need to restore in
8031 // the complicated way.
8032 if (*p_cpo == NUL)
8033 set_option_value((char_u *)"cpo", 0L, save_cpo, 0);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008034 free_string_option(save_cpo);
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008035 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008036
8037 return retval;
8038}
8039
8040/*
8041 * "searchpos()" function
8042 */
8043 static void
8044f_searchpos(typval_T *argvars, typval_T *rettv)
8045{
8046 pos_T match_pos;
8047 int lnum = 0;
8048 int col = 0;
8049 int n;
8050 int flags = 0;
8051
8052 if (rettv_list_alloc(rettv) == FAIL)
8053 return;
8054
8055 n = search_cmn(argvars, &match_pos, &flags);
8056 if (n > 0)
8057 {
8058 lnum = match_pos.lnum;
8059 col = match_pos.col;
8060 }
8061
8062 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
8063 list_append_number(rettv->vval.v_list, (varnumber_T)col);
8064 if (flags & SP_SUBPAT)
8065 list_append_number(rettv->vval.v_list, (varnumber_T)n);
8066}
8067
8068 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008069f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
8070{
8071 dict_T *d;
8072 dictitem_T *di;
8073 char_u *csearch;
8074
8075 if (argvars[0].v_type != VAR_DICT)
8076 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008077 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008078 return;
8079 }
8080
8081 if ((d = argvars[0].vval.v_dict) != NULL)
8082 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01008083 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008084 if (csearch != NULL)
8085 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008086 if (enc_utf8)
8087 {
8088 int pcc[MAX_MCO];
8089 int c = utfc_ptr2char(csearch, pcc);
8090
8091 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
8092 }
8093 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008094 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02008095 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008096 }
8097
8098 di = dict_find(d, (char_u *)"forward", -1);
8099 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008100 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008101 ? FORWARD : BACKWARD);
8102
8103 di = dict_find(d, (char_u *)"until", -1);
8104 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008105 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008106 }
8107}
8108
8109/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02008110 * "setenv()" function
8111 */
8112 static void
8113f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
8114{
8115 char_u namebuf[NUMBUFLEN];
8116 char_u valbuf[NUMBUFLEN];
8117 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
8118
8119 if (argvars[1].v_type == VAR_SPECIAL
8120 && argvars[1].vval.v_number == VVAL_NULL)
8121 vim_unsetenv(name);
8122 else
8123 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
8124}
8125
8126/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008127 * "setfperm({fname}, {mode})" function
8128 */
8129 static void
8130f_setfperm(typval_T *argvars, typval_T *rettv)
8131{
8132 char_u *fname;
8133 char_u modebuf[NUMBUFLEN];
8134 char_u *mode_str;
8135 int i;
8136 int mask;
8137 int mode = 0;
8138
8139 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008140 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008141 if (fname == NULL)
8142 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008143 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008144 if (mode_str == NULL)
8145 return;
8146 if (STRLEN(mode_str) != 9)
8147 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008148 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008149 return;
8150 }
8151
8152 mask = 1;
8153 for (i = 8; i >= 0; --i)
8154 {
8155 if (mode_str[i] != '-')
8156 mode |= mask;
8157 mask = mask << 1;
8158 }
8159 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
8160}
8161
8162/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008163 * "setpos()" function
8164 */
8165 static void
8166f_setpos(typval_T *argvars, typval_T *rettv)
8167{
8168 pos_T pos;
8169 int fnum;
8170 char_u *name;
8171 colnr_T curswant = -1;
8172
8173 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008174 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008175 if (name != NULL)
8176 {
8177 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
8178 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008179 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008180 pos.col = 0;
8181 if (name[0] == '.' && name[1] == NUL)
8182 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008183 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01008184 curwin->w_cursor = pos;
8185 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008186 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01008187 curwin->w_curswant = curswant - 1;
8188 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008189 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01008190 check_cursor();
8191 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008192 }
8193 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
8194 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008195 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008196 if (setmark_pos(name[1], &pos, fnum) == OK)
8197 rettv->vval.v_number = 0;
8198 }
8199 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008200 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008201 }
8202 }
8203}
8204
8205/*
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008206 * Translate a register type string to the yank type and block length
8207 */
8208 static int
8209get_yank_type(char_u **pp, char_u *yank_type, long *block_len)
8210{
8211 char_u *stropt = *pp;
8212 switch (*stropt)
8213 {
8214 case 'v': case 'c': // character-wise selection
8215 *yank_type = MCHAR;
8216 break;
8217 case 'V': case 'l': // line-wise selection
8218 *yank_type = MLINE;
8219 break;
8220 case 'b': case Ctrl_V: // block-wise selection
8221 *yank_type = MBLOCK;
8222 if (VIM_ISDIGIT(stropt[1]))
8223 {
8224 ++stropt;
8225 *block_len = getdigits(&stropt) - 1;
8226 --stropt;
8227 }
8228 break;
8229 default:
8230 return FAIL;
8231 }
8232 *pp = stropt;
8233 return OK;
8234}
8235
8236/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008237 * "setreg()" function
8238 */
8239 static void
8240f_setreg(typval_T *argvars, typval_T *rettv)
8241{
8242 int regname;
8243 char_u *strregname;
8244 char_u *stropt;
8245 char_u *strval;
8246 int append;
8247 char_u yank_type;
8248 long block_len;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008249 typval_T *regcontents;
8250 int pointreg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008251
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008252 pointreg = 0;
8253 regcontents = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008254 block_len = -1;
8255 yank_type = MAUTO;
8256 append = FALSE;
8257
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008258 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008259 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008260
8261 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008262 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008263 regname = *strregname;
8264 if (regname == 0 || regname == '@')
8265 regname = '"';
8266
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008267 if (argvars[1].v_type == VAR_DICT)
8268 {
8269 dict_T *d = argvars[1].vval.v_dict;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008270 dictitem_T *di;
8271
8272 if (d == NULL || d->dv_hashtab.ht_used == 0)
8273 {
8274 // Empty dict, clear the register (like setreg(0, []))
8275 char_u *lstval[2] = {NULL, NULL};
8276 write_reg_contents_lst(regname, lstval, 0, FALSE, MAUTO, -1);
8277 return;
8278 }
8279
8280 di = dict_find(d, (char_u *)"regcontents", -1);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008281 if (di != NULL)
8282 regcontents = &di->di_tv;
8283
8284 stropt = dict_get_string(d, (char_u *)"regtype", FALSE);
8285 if (stropt != NULL)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008286 {
8287 int ret = get_yank_type(&stropt, &yank_type, &block_len);
8288
8289 if (ret == FAIL || *++stropt != NUL)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008290 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008291 semsg(_(e_invargval), "value");
8292 return;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008293 }
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008294 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008295
8296 if (regname == '"')
8297 {
8298 stropt = dict_get_string(d, (char_u *)"points_to", FALSE);
8299 if (stropt != NULL)
8300 {
8301 pointreg = *stropt;
8302 regname = pointreg;
8303 }
8304 }
Bram Moolenaar6a950582020-08-28 16:39:33 +02008305 else if (dict_get_bool(d, (char_u *)"isunnamed", -1) > 0)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008306 pointreg = regname;
8307 }
8308 else
8309 regcontents = &argvars[1];
8310
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008311 if (argvars[2].v_type != VAR_UNKNOWN)
8312 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008313 if (yank_type != MAUTO)
8314 {
8315 semsg(_(e_toomanyarg), "setreg");
8316 return;
8317 }
8318
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008319 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008320 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008321 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008322 for (; *stropt != NUL; ++stropt)
8323 switch (*stropt)
8324 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008325 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008326 append = TRUE;
8327 break;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008328 default:
8329 get_yank_type(&stropt, &yank_type, &block_len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008330 }
8331 }
8332
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008333 if (regcontents && regcontents->v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008334 {
8335 char_u **lstval;
8336 char_u **allocval;
8337 char_u buf[NUMBUFLEN];
8338 char_u **curval;
8339 char_u **curallocval;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008340 list_T *ll = regcontents->vval.v_list;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008341 listitem_T *li;
8342 int len;
8343
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008344 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008345 len = ll == NULL ? 0 : ll->lv_len;
8346
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008347 // First half: use for pointers to result lines; second half: use for
8348 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02008349 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008350 if (lstval == NULL)
8351 return;
8352 curval = lstval;
8353 allocval = lstval + len + 2;
8354 curallocval = allocval;
8355
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008356 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008357 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02008358 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02008359 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008360 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008361 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008362 if (strval == NULL)
8363 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008364 if (strval == buf)
8365 {
8366 // Need to make a copy, next tv_get_string_buf_chk() will
8367 // overwrite the string.
8368 strval = vim_strsave(buf);
8369 if (strval == NULL)
8370 goto free_lstval;
8371 *curallocval++ = strval;
8372 }
8373 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008374 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008375 }
8376 *curval++ = NULL;
8377
8378 write_reg_contents_lst(regname, lstval, -1,
8379 append, yank_type, block_len);
8380free_lstval:
8381 while (curallocval > allocval)
8382 vim_free(*--curallocval);
8383 vim_free(lstval);
8384 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008385 else if (regcontents)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008386 {
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008387 strval = tv_get_string_chk(regcontents);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008388 if (strval == NULL)
8389 return;
8390 write_reg_contents_ex(regname, strval, -1,
8391 append, yank_type, block_len);
8392 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008393 if (pointreg != 0)
8394 get_yank_register(pointreg, TRUE);
8395
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008396 rettv->vval.v_number = 0;
8397}
8398
8399/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008400 * "settagstack()" function
8401 */
8402 static void
8403f_settagstack(typval_T *argvars, typval_T *rettv)
8404{
8405 static char *e_invact2 = N_("E962: Invalid action: '%s'");
8406 win_T *wp;
8407 dict_T *d;
8408 int action = 'r';
8409
8410 rettv->vval.v_number = -1;
8411
8412 // first argument: window number or id
8413 wp = find_win_by_nr_or_id(&argvars[0]);
8414 if (wp == NULL)
8415 return;
8416
8417 // second argument: dict with items to set in the tag stack
8418 if (argvars[1].v_type != VAR_DICT)
8419 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008420 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008421 return;
8422 }
8423 d = argvars[1].vval.v_dict;
8424 if (d == NULL)
8425 return;
8426
8427 // third argument: action - 'a' for append and 'r' for replace.
8428 // default is to replace the stack.
8429 if (argvars[2].v_type == VAR_UNKNOWN)
8430 action = 'r';
8431 else if (argvars[2].v_type == VAR_STRING)
8432 {
8433 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008434 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008435 if (actstr == NULL)
8436 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01008437 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
8438 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008439 action = *actstr;
8440 else
8441 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008442 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008443 return;
8444 }
8445 }
8446 else
8447 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008448 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008449 return;
8450 }
8451
8452 if (set_tagstack(wp, d, action) == OK)
8453 rettv->vval.v_number = 0;
8454}
8455
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008456#ifdef FEAT_CRYPT
8457/*
8458 * "sha256({string})" function
8459 */
8460 static void
8461f_sha256(typval_T *argvars, typval_T *rettv)
8462{
8463 char_u *p;
8464
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008465 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008466 rettv->vval.v_string = vim_strsave(
8467 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
8468 rettv->v_type = VAR_STRING;
8469}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008470#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008471
8472/*
8473 * "shellescape({string})" function
8474 */
8475 static void
8476f_shellescape(typval_T *argvars, typval_T *rettv)
8477{
Bram Moolenaar20615522017-06-05 18:46:26 +02008478 int do_special = non_zero_arg(&argvars[1]);
8479
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008480 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008481 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008482 rettv->v_type = VAR_STRING;
8483}
8484
8485/*
8486 * shiftwidth() function
8487 */
8488 static void
8489f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
8490{
Bram Moolenaarf9514162018-11-22 03:08:29 +01008491 rettv->vval.v_number = 0;
8492
8493 if (argvars[0].v_type != VAR_UNKNOWN)
8494 {
8495 long col;
8496
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008497 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01008498 if (col < 0)
8499 return; // type error; errmsg already given
8500#ifdef FEAT_VARTABS
8501 rettv->vval.v_number = get_sw_value_col(curbuf, col);
8502 return;
8503#endif
8504 }
8505
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008506 rettv->vval.v_number = get_sw_value(curbuf);
8507}
8508
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008509#ifdef FEAT_FLOAT
8510/*
8511 * "sin()" function
8512 */
8513 static void
8514f_sin(typval_T *argvars, typval_T *rettv)
8515{
8516 float_T f = 0.0;
8517
8518 rettv->v_type = VAR_FLOAT;
8519 if (get_float_arg(argvars, &f) == OK)
8520 rettv->vval.v_float = sin(f);
8521 else
8522 rettv->vval.v_float = 0.0;
8523}
8524
8525/*
8526 * "sinh()" function
8527 */
8528 static void
8529f_sinh(typval_T *argvars, typval_T *rettv)
8530{
8531 float_T f = 0.0;
8532
8533 rettv->v_type = VAR_FLOAT;
8534 if (get_float_arg(argvars, &f) == OK)
8535 rettv->vval.v_float = sinh(f);
8536 else
8537 rettv->vval.v_float = 0.0;
8538}
8539#endif
8540
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008541/*
8542 * "soundfold({word})" function
8543 */
8544 static void
8545f_soundfold(typval_T *argvars, typval_T *rettv)
8546{
8547 char_u *s;
8548
8549 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008550 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008551#ifdef FEAT_SPELL
8552 rettv->vval.v_string = eval_soundfold(s);
8553#else
8554 rettv->vval.v_string = vim_strsave(s);
8555#endif
8556}
8557
8558/*
8559 * "spellbadword()" function
8560 */
8561 static void
8562f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
8563{
8564 char_u *word = (char_u *)"";
8565 hlf_T attr = HLF_COUNT;
8566 int len = 0;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008567#ifdef FEAT_SPELL
8568 int wo_spell_save = curwin->w_p_spell;
8569
8570 if (!curwin->w_p_spell)
8571 {
8572 did_set_spelllang(curwin);
8573 curwin->w_p_spell = TRUE;
8574 }
8575
8576 if (*curwin->w_s->b_p_spl == NUL)
8577 {
8578 emsg(_(e_no_spell));
8579 curwin->w_p_spell = wo_spell_save;
8580 return;
8581 }
8582#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008583
8584 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008585 {
8586#ifdef FEAT_SPELL
8587 curwin->w_p_spell = wo_spell_save;
8588#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008589 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008590 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008591
8592#ifdef FEAT_SPELL
8593 if (argvars[0].v_type == VAR_UNKNOWN)
8594 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008595 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008596 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
8597 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008598 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008599 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008600 curwin->w_set_curswant = TRUE;
8601 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008602 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008603 else if (*curbuf->b_s.b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008604 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008605 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008606 int capcol = -1;
8607
8608 if (str != NULL)
8609 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008610 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008611 while (*str != NUL)
8612 {
8613 len = spell_check(curwin, str, &attr, &capcol, FALSE);
8614 if (attr != HLF_COUNT)
8615 {
8616 word = str;
8617 break;
8618 }
8619 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02008620 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02008621 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008622 }
8623 }
8624 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008625 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008626#endif
8627
8628 list_append_string(rettv->vval.v_list, word, len);
8629 list_append_string(rettv->vval.v_list, (char_u *)(
8630 attr == HLF_SPB ? "bad" :
8631 attr == HLF_SPR ? "rare" :
8632 attr == HLF_SPL ? "local" :
8633 attr == HLF_SPC ? "caps" :
8634 ""), -1);
8635}
8636
8637/*
8638 * "spellsuggest()" function
8639 */
8640 static void
8641f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
8642{
8643#ifdef FEAT_SPELL
8644 char_u *str;
8645 int typeerr = FALSE;
8646 int maxcount;
8647 garray_T ga;
8648 int i;
8649 listitem_T *li;
8650 int need_capital = FALSE;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008651 int wo_spell_save = curwin->w_p_spell;
8652
8653 if (!curwin->w_p_spell)
8654 {
8655 did_set_spelllang(curwin);
8656 curwin->w_p_spell = TRUE;
8657 }
8658
8659 if (*curwin->w_s->b_p_spl == NUL)
8660 {
8661 emsg(_(e_no_spell));
8662 curwin->w_p_spell = wo_spell_save;
8663 return;
8664 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008665#endif
8666
8667 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008668 {
8669#ifdef FEAT_SPELL
8670 curwin->w_p_spell = wo_spell_save;
8671#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008672 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008673 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008674
8675#ifdef FEAT_SPELL
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008676 if (*curwin->w_s->b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008677 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008678 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008679 if (argvars[1].v_type != VAR_UNKNOWN)
8680 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008681 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008682 if (maxcount <= 0)
8683 return;
8684 if (argvars[2].v_type != VAR_UNKNOWN)
8685 {
Bram Moolenaar7c27f332020-09-05 22:45:55 +02008686 need_capital = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008687 if (typeerr)
8688 return;
8689 }
8690 }
8691 else
8692 maxcount = 25;
8693
8694 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
8695
8696 for (i = 0; i < ga.ga_len; ++i)
8697 {
8698 str = ((char_u **)ga.ga_data)[i];
8699
8700 li = listitem_alloc();
8701 if (li == NULL)
8702 vim_free(str);
8703 else
8704 {
8705 li->li_tv.v_type = VAR_STRING;
8706 li->li_tv.v_lock = 0;
8707 li->li_tv.vval.v_string = str;
8708 list_append(rettv->vval.v_list, li);
8709 }
8710 }
8711 ga_clear(&ga);
8712 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008713 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008714#endif
8715}
8716
8717 static void
8718f_split(typval_T *argvars, typval_T *rettv)
8719{
8720 char_u *str;
8721 char_u *end;
8722 char_u *pat = NULL;
8723 regmatch_T regmatch;
8724 char_u patbuf[NUMBUFLEN];
8725 char_u *save_cpo;
8726 int match;
8727 colnr_T col = 0;
8728 int keepempty = FALSE;
8729 int typeerr = FALSE;
8730
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008731 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008732 save_cpo = p_cpo;
Bram Moolenaare5a2dc82021-01-03 19:52:05 +01008733 p_cpo = empty_option;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008734
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008735 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008736 if (argvars[1].v_type != VAR_UNKNOWN)
8737 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008738 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008739 if (pat == NULL)
8740 typeerr = TRUE;
8741 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02008742 keepempty = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008743 }
8744 if (pat == NULL || *pat == NUL)
8745 pat = (char_u *)"[\\x01- ]\\+";
8746
8747 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008748 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008749 if (typeerr)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008750 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008751
8752 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
8753 if (regmatch.regprog != NULL)
8754 {
8755 regmatch.rm_ic = FALSE;
8756 while (*str != NUL || keepempty)
8757 {
8758 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008759 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008760 else
8761 match = vim_regexec_nl(&regmatch, str, col);
8762 if (match)
8763 end = regmatch.startp[0];
8764 else
8765 end = str + STRLEN(str);
8766 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
8767 && *str != NUL && match && end < regmatch.endp[0]))
8768 {
8769 if (list_append_string(rettv->vval.v_list, str,
8770 (int)(end - str)) == FAIL)
8771 break;
8772 }
8773 if (!match)
8774 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01008775 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008776 if (regmatch.endp[0] > str)
8777 col = 0;
8778 else
Bram Moolenaar13505972019-01-24 15:04:48 +01008779 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008780 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008781 str = regmatch.endp[0];
8782 }
8783
8784 vim_regfree(regmatch.regprog);
8785 }
8786
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008787theend:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008788 p_cpo = save_cpo;
8789}
8790
8791#ifdef FEAT_FLOAT
8792/*
8793 * "sqrt()" function
8794 */
8795 static void
8796f_sqrt(typval_T *argvars, typval_T *rettv)
8797{
8798 float_T f = 0.0;
8799
8800 rettv->v_type = VAR_FLOAT;
8801 if (get_float_arg(argvars, &f) == OK)
8802 rettv->vval.v_float = sqrt(f);
8803 else
8804 rettv->vval.v_float = 0.0;
8805}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008806#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008807
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008808#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01008809/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008810 * "str2float()" function
8811 */
8812 static void
8813f_str2float(typval_T *argvars, typval_T *rettv)
8814{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008815 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008816 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008817
Bram Moolenaar08243d22017-01-10 16:12:29 +01008818 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008819 p = skipwhite(p + 1);
8820 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01008821 if (isneg)
8822 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008823 rettv->v_type = VAR_FLOAT;
8824}
8825#endif
8826
8827/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02008828 * "str2list()" function
8829 */
8830 static void
8831f_str2list(typval_T *argvars, typval_T *rettv)
8832{
8833 char_u *p;
8834 int utf8 = FALSE;
8835
8836 if (rettv_list_alloc(rettv) == FAIL)
8837 return;
8838
8839 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaara48f7862020-09-05 20:16:57 +02008840 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar9d401282019-04-06 13:18:12 +02008841
8842 p = tv_get_string(&argvars[0]);
8843
8844 if (has_mbyte || utf8)
8845 {
8846 int (*ptr2len)(char_u *);
8847 int (*ptr2char)(char_u *);
8848
8849 if (utf8 || enc_utf8)
8850 {
8851 ptr2len = utf_ptr2len;
8852 ptr2char = utf_ptr2char;
8853 }
8854 else
8855 {
8856 ptr2len = mb_ptr2len;
8857 ptr2char = mb_ptr2char;
8858 }
8859
8860 for ( ; *p != NUL; p += (*ptr2len)(p))
8861 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
8862 }
8863 else
8864 for ( ; *p != NUL; ++p)
8865 list_append_number(rettv->vval.v_list, *p);
8866}
8867
8868/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008869 * "str2nr()" function
8870 */
8871 static void
8872f_str2nr(typval_T *argvars, typval_T *rettv)
8873{
8874 int base = 10;
8875 char_u *p;
8876 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008877 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01008878 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008879
8880 if (argvars[1].v_type != VAR_UNKNOWN)
8881 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008882 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008883 if (base != 2 && base != 8 && base != 10 && base != 16)
8884 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008885 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008886 return;
8887 }
Bram Moolenaar3986b942020-09-06 16:09:04 +02008888 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[2]))
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008889 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008890 }
8891
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008892 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008893 isneg = (*p == '-');
8894 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008895 p = skipwhite(p + 1);
8896 switch (base)
8897 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008898 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02008899 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008900 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008901 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02008902 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
8903 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01008904 if (isneg)
8905 rettv->vval.v_number = -n;
8906 else
8907 rettv->vval.v_number = n;
8908
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008909}
8910
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008911/*
8912 * "strgetchar()" function
8913 */
8914 static void
8915f_strgetchar(typval_T *argvars, typval_T *rettv)
8916{
8917 char_u *str;
8918 int len;
8919 int error = FALSE;
8920 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01008921 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008922
8923 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008924 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008925 if (str == NULL)
8926 return;
8927 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008928 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008929 if (error)
8930 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008931
Bram Moolenaar13505972019-01-24 15:04:48 +01008932 while (charidx >= 0 && byteidx < len)
8933 {
8934 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008935 {
Bram Moolenaar13505972019-01-24 15:04:48 +01008936 rettv->vval.v_number = mb_ptr2char(str + byteidx);
8937 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008938 }
Bram Moolenaar13505972019-01-24 15:04:48 +01008939 --charidx;
8940 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008941 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008942}
8943
8944/*
8945 * "stridx()" function
8946 */
8947 static void
8948f_stridx(typval_T *argvars, typval_T *rettv)
8949{
8950 char_u buf[NUMBUFLEN];
8951 char_u *needle;
8952 char_u *haystack;
8953 char_u *save_haystack;
8954 char_u *pos;
8955 int start_idx;
8956
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008957 needle = tv_get_string_chk(&argvars[1]);
8958 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008959 rettv->vval.v_number = -1;
8960 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008961 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008962
8963 if (argvars[2].v_type != VAR_UNKNOWN)
8964 {
8965 int error = FALSE;
8966
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008967 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008968 if (error || start_idx >= (int)STRLEN(haystack))
8969 return;
8970 if (start_idx >= 0)
8971 haystack += start_idx;
8972 }
8973
8974 pos = (char_u *)strstr((char *)haystack, (char *)needle);
8975 if (pos != NULL)
8976 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
8977}
8978
8979/*
8980 * "string()" function
8981 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01008982 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008983f_string(typval_T *argvars, typval_T *rettv)
8984{
8985 char_u *tofree;
8986 char_u numbuf[NUMBUFLEN];
8987
8988 rettv->v_type = VAR_STRING;
8989 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
8990 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008991 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008992 if (rettv->vval.v_string != NULL && tofree == NULL)
8993 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
8994}
8995
8996/*
8997 * "strlen()" function
8998 */
8999 static void
9000f_strlen(typval_T *argvars, typval_T *rettv)
9001{
9002 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009003 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009004}
9005
9006/*
9007 * "strchars()" function
9008 */
9009 static void
9010f_strchars(typval_T *argvars, typval_T *rettv)
9011{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009012 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar3986b942020-09-06 16:09:04 +02009013 int skipcc = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009014 varnumber_T len = 0;
9015 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009016
9017 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02009018 skipcc = (int)tv_get_bool(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009019 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02009020 semsg(_(e_using_number_as_bool_nr), skipcc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009021 else
9022 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009023 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
9024 while (*s != NUL)
9025 {
9026 func_mb_ptr2char_adv(&s);
9027 ++len;
9028 }
9029 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009030 }
9031}
9032
9033/*
9034 * "strdisplaywidth()" function
9035 */
9036 static void
9037f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
9038{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009039 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009040 int col = 0;
9041
9042 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009043 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009044
9045 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
9046}
9047
9048/*
9049 * "strwidth()" function
9050 */
9051 static void
9052f_strwidth(typval_T *argvars, typval_T *rettv)
9053{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009054 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009055
Bram Moolenaar13505972019-01-24 15:04:48 +01009056 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009057}
9058
9059/*
9060 * "strcharpart()" function
9061 */
9062 static void
9063f_strcharpart(typval_T *argvars, typval_T *rettv)
9064{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009065 char_u *p;
9066 int nchar;
9067 int nbyte = 0;
9068 int charlen;
9069 int len = 0;
9070 int slen;
9071 int error = FALSE;
9072
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009073 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009074 slen = (int)STRLEN(p);
9075
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009076 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009077 if (!error)
9078 {
9079 if (nchar > 0)
9080 while (nchar > 0 && nbyte < slen)
9081 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009082 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009083 --nchar;
9084 }
9085 else
9086 nbyte = nchar;
9087 if (argvars[2].v_type != VAR_UNKNOWN)
9088 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009089 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009090 while (charlen > 0 && nbyte + len < slen)
9091 {
9092 int off = nbyte + len;
9093
9094 if (off < 0)
9095 len += 1;
9096 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02009097 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009098 --charlen;
9099 }
9100 }
9101 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009102 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009103 }
9104
9105 /*
9106 * Only return the overlap between the specified part and the actual
9107 * string.
9108 */
9109 if (nbyte < 0)
9110 {
9111 len += nbyte;
9112 nbyte = 0;
9113 }
9114 else if (nbyte > slen)
9115 nbyte = slen;
9116 if (len < 0)
9117 len = 0;
9118 else if (nbyte + len > slen)
9119 len = slen - nbyte;
9120
9121 rettv->v_type = VAR_STRING;
9122 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009123}
9124
9125/*
9126 * "strpart()" function
9127 */
9128 static void
9129f_strpart(typval_T *argvars, typval_T *rettv)
9130{
9131 char_u *p;
9132 int n;
9133 int len;
9134 int slen;
9135 int error = FALSE;
9136
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009137 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009138 slen = (int)STRLEN(p);
9139
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009140 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009141 if (error)
9142 len = 0;
9143 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009144 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009145 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009146 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009147
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009148 // Only return the overlap between the specified part and the actual
9149 // string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009150 if (n < 0)
9151 {
9152 len += n;
9153 n = 0;
9154 }
9155 else if (n > slen)
9156 n = slen;
9157 if (len < 0)
9158 len = 0;
9159 else if (n + len > slen)
9160 len = slen - n;
9161
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02009162 if (argvars[2].v_type != VAR_UNKNOWN && argvars[3].v_type != VAR_UNKNOWN)
9163 {
9164 int off;
9165
9166 // length in characters
9167 for (off = n; off < slen && len > 0; --len)
9168 off += mb_ptr2len(p + off);
9169 len = off - n;
9170 }
9171
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009172 rettv->v_type = VAR_STRING;
9173 rettv->vval.v_string = vim_strnsave(p + n, len);
9174}
9175
9176/*
9177 * "strridx()" function
9178 */
9179 static void
9180f_strridx(typval_T *argvars, typval_T *rettv)
9181{
9182 char_u buf[NUMBUFLEN];
9183 char_u *needle;
9184 char_u *haystack;
9185 char_u *rest;
9186 char_u *lastmatch = NULL;
9187 int haystack_len, end_idx;
9188
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009189 needle = tv_get_string_chk(&argvars[1]);
9190 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009191
9192 rettv->vval.v_number = -1;
9193 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009194 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009195
9196 haystack_len = (int)STRLEN(haystack);
9197 if (argvars[2].v_type != VAR_UNKNOWN)
9198 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009199 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009200 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009201 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009202 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009203 }
9204 else
9205 end_idx = haystack_len;
9206
9207 if (*needle == NUL)
9208 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009209 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009210 lastmatch = haystack + end_idx;
9211 }
9212 else
9213 {
9214 for (rest = haystack; *rest != '\0'; ++rest)
9215 {
9216 rest = (char_u *)strstr((char *)rest, (char *)needle);
9217 if (rest == NULL || rest > haystack + end_idx)
9218 break;
9219 lastmatch = rest;
9220 }
9221 }
9222
9223 if (lastmatch == NULL)
9224 rettv->vval.v_number = -1;
9225 else
9226 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
9227}
9228
9229/*
9230 * "strtrans()" function
9231 */
9232 static void
9233f_strtrans(typval_T *argvars, typval_T *rettv)
9234{
9235 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009236 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009237}
9238
9239/*
9240 * "submatch()" function
9241 */
9242 static void
9243f_submatch(typval_T *argvars, typval_T *rettv)
9244{
9245 int error = FALSE;
9246 int no;
9247 int retList = 0;
9248
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009249 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009250 if (error)
9251 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009252 if (no < 0 || no >= NSUBEXP)
9253 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009254 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01009255 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009256 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009257 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaarad304702020-09-06 18:22:53 +02009258 retList = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009259 if (error)
9260 return;
9261
9262 if (retList == 0)
9263 {
9264 rettv->v_type = VAR_STRING;
9265 rettv->vval.v_string = reg_submatch(no);
9266 }
9267 else
9268 {
9269 rettv->v_type = VAR_LIST;
9270 rettv->vval.v_list = reg_submatch_list(no);
9271 }
9272}
9273
9274/*
9275 * "substitute()" function
9276 */
9277 static void
9278f_substitute(typval_T *argvars, typval_T *rettv)
9279{
9280 char_u patbuf[NUMBUFLEN];
9281 char_u subbuf[NUMBUFLEN];
9282 char_u flagsbuf[NUMBUFLEN];
9283
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009284 char_u *str = tv_get_string_chk(&argvars[0]);
9285 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009286 char_u *sub = NULL;
9287 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009288 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009289
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009290 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
9291 expr = &argvars[2];
9292 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009293 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009294
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009295 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009296 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
9297 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009298 rettv->vval.v_string = NULL;
9299 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009300 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009301}
9302
9303/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009304 * "swapinfo(swap_filename)" function
9305 */
9306 static void
9307f_swapinfo(typval_T *argvars, typval_T *rettv)
9308{
9309 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009310 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009311}
9312
9313/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02009314 * "swapname(expr)" function
9315 */
9316 static void
9317f_swapname(typval_T *argvars, typval_T *rettv)
9318{
9319 buf_T *buf;
9320
9321 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01009322 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02009323 if (buf == NULL || buf->b_ml.ml_mfp == NULL
9324 || buf->b_ml.ml_mfp->mf_fname == NULL)
9325 rettv->vval.v_string = NULL;
9326 else
9327 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
9328}
9329
9330/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009331 * "synID(lnum, col, trans)" function
9332 */
9333 static void
9334f_synID(typval_T *argvars UNUSED, typval_T *rettv)
9335{
9336 int id = 0;
9337#ifdef FEAT_SYN_HL
9338 linenr_T lnum;
9339 colnr_T col;
9340 int trans;
9341 int transerr = FALSE;
9342
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009343 lnum = tv_get_lnum(argvars); // -1 on type error
9344 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaarfcb6d702020-09-05 21:41:56 +02009345 trans = (int)tv_get_bool_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009346
9347 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9348 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
9349 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
9350#endif
9351
9352 rettv->vval.v_number = id;
9353}
9354
9355/*
9356 * "synIDattr(id, what [, mode])" function
9357 */
9358 static void
9359f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
9360{
9361 char_u *p = NULL;
9362#ifdef FEAT_SYN_HL
9363 int id;
9364 char_u *what;
9365 char_u *mode;
9366 char_u modebuf[NUMBUFLEN];
9367 int modec;
9368
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009369 id = (int)tv_get_number(&argvars[0]);
9370 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009371 if (argvars[2].v_type != VAR_UNKNOWN)
9372 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009373 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009374 modec = TOLOWER_ASC(mode[0]);
9375 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009376 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009377 }
9378 else
9379 {
9380#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
9381 if (USE_24BIT)
9382 modec = 'g';
9383 else
9384#endif
9385 if (t_colors > 1)
9386 modec = 'c';
9387 else
9388 modec = 't';
9389 }
9390
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009391 switch (TOLOWER_ASC(what[0]))
9392 {
9393 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009394 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009395 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009396 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009397 p = highlight_has_attr(id, HL_BOLD, modec);
9398 break;
9399
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009400 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009401 p = highlight_color(id, what, modec);
9402 break;
9403
9404 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009405 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009406 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009407 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009408 p = highlight_has_attr(id, HL_ITALIC, modec);
9409 break;
9410
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009411 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02009412 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009413 break;
9414
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009415 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009416 p = highlight_has_attr(id, HL_INVERSE, modec);
9417 break;
9418
9419 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009420 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009421 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009422 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02009423 else if (TOLOWER_ASC(what[1]) == 't' &&
9424 TOLOWER_ASC(what[2]) == 'r')
9425 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009426 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009427 p = highlight_has_attr(id, HL_STANDOUT, modec);
9428 break;
9429
9430 case 'u':
Bram Moolenaar391c3622020-09-29 20:59:17 +02009431 if (TOLOWER_ASC(what[1]) == 'l') // ul
9432 p = highlight_color(id, what, modec);
9433 else if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009434 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009435 p = highlight_has_attr(id, HL_UNDERLINE, modec);
9436 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009437 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009438 p = highlight_has_attr(id, HL_UNDERCURL, modec);
9439 break;
9440 }
9441
9442 if (p != NULL)
9443 p = vim_strsave(p);
9444#endif
9445 rettv->v_type = VAR_STRING;
9446 rettv->vval.v_string = p;
9447}
9448
9449/*
9450 * "synIDtrans(id)" function
9451 */
9452 static void
9453f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
9454{
9455 int id;
9456
9457#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009458 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009459
9460 if (id > 0)
9461 id = syn_get_final_id(id);
9462 else
9463#endif
9464 id = 0;
9465
9466 rettv->vval.v_number = id;
9467}
9468
9469/*
9470 * "synconcealed(lnum, col)" function
9471 */
9472 static void
9473f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
9474{
9475#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
9476 linenr_T lnum;
9477 colnr_T col;
9478 int syntax_flags = 0;
9479 int cchar;
9480 int matchid = 0;
9481 char_u str[NUMBUFLEN];
9482#endif
9483
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009484 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009485
9486#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009487 lnum = tv_get_lnum(argvars); // -1 on type error
9488 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009489
Bram Moolenaara80faa82020-04-12 19:37:17 +02009490 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009491
9492 if (rettv_list_alloc(rettv) != FAIL)
9493 {
9494 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9495 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9496 && curwin->w_p_cole > 0)
9497 {
9498 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
9499 syntax_flags = get_syntax_info(&matchid);
9500
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009501 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009502 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
9503 {
9504 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02009505 if (cchar == NUL && curwin->w_p_cole == 1)
9506 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009507 if (cchar != NUL)
9508 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009509 if (has_mbyte)
9510 (*mb_char2bytes)(cchar, str);
9511 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009512 str[0] = cchar;
9513 }
9514 }
9515 }
9516
9517 list_append_number(rettv->vval.v_list,
9518 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009519 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009520 list_append_string(rettv->vval.v_list, str, -1);
9521 list_append_number(rettv->vval.v_list, matchid);
9522 }
9523#endif
9524}
9525
9526/*
9527 * "synstack(lnum, col)" function
9528 */
9529 static void
9530f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
9531{
9532#ifdef FEAT_SYN_HL
9533 linenr_T lnum;
9534 colnr_T col;
9535 int i;
9536 int id;
9537#endif
9538
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009539 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009540
9541#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009542 lnum = tv_get_lnum(argvars); // -1 on type error
9543 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009544
9545 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9546 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9547 && rettv_list_alloc(rettv) != FAIL)
9548 {
9549 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
9550 for (i = 0; ; ++i)
9551 {
9552 id = syn_get_stack_item(i);
9553 if (id < 0)
9554 break;
9555 if (list_append_number(rettv->vval.v_list, id) == FAIL)
9556 break;
9557 }
9558 }
9559#endif
9560}
9561
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009562/*
9563 * "tabpagebuflist()" function
9564 */
9565 static void
9566f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9567{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009568 tabpage_T *tp;
9569 win_T *wp = NULL;
9570
9571 if (argvars[0].v_type == VAR_UNKNOWN)
9572 wp = firstwin;
9573 else
9574 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009575 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009576 if (tp != NULL)
9577 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
9578 }
9579 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
9580 {
9581 for (; wp != NULL; wp = wp->w_next)
9582 if (list_append_number(rettv->vval.v_list,
9583 wp->w_buffer->b_fnum) == FAIL)
9584 break;
9585 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009586}
9587
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009588/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009589 * "tagfiles()" function
9590 */
9591 static void
9592f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
9593{
9594 char_u *fname;
9595 tagname_T tn;
9596 int first;
9597
9598 if (rettv_list_alloc(rettv) == FAIL)
9599 return;
9600 fname = alloc(MAXPATHL);
9601 if (fname == NULL)
9602 return;
9603
9604 for (first = TRUE; ; first = FALSE)
9605 if (get_tagfname(&tn, first, fname) == FAIL
9606 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
9607 break;
9608 tagname_free(&tn);
9609 vim_free(fname);
9610}
9611
9612/*
9613 * "taglist()" function
9614 */
9615 static void
9616f_taglist(typval_T *argvars, typval_T *rettv)
9617{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009618 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009619 char_u *tag_pattern;
9620
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009621 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009622
9623 rettv->vval.v_number = FALSE;
9624 if (*tag_pattern == NUL)
9625 return;
9626
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009627 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009628 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009629 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009630 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009631}
9632
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009633#ifdef FEAT_FLOAT
9634/*
9635 * "tan()" function
9636 */
9637 static void
9638f_tan(typval_T *argvars, typval_T *rettv)
9639{
9640 float_T f = 0.0;
9641
9642 rettv->v_type = VAR_FLOAT;
9643 if (get_float_arg(argvars, &f) == OK)
9644 rettv->vval.v_float = tan(f);
9645 else
9646 rettv->vval.v_float = 0.0;
9647}
9648
9649/*
9650 * "tanh()" function
9651 */
9652 static void
9653f_tanh(typval_T *argvars, typval_T *rettv)
9654{
9655 float_T f = 0.0;
9656
9657 rettv->v_type = VAR_FLOAT;
9658 if (get_float_arg(argvars, &f) == OK)
9659 rettv->vval.v_float = tanh(f);
9660 else
9661 rettv->vval.v_float = 0.0;
9662}
9663#endif
9664
9665/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009666 * "tolower(string)" function
9667 */
9668 static void
9669f_tolower(typval_T *argvars, typval_T *rettv)
9670{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009671 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009672 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009673}
9674
9675/*
9676 * "toupper(string)" function
9677 */
9678 static void
9679f_toupper(typval_T *argvars, typval_T *rettv)
9680{
9681 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009682 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009683}
9684
9685/*
9686 * "tr(string, fromstr, tostr)" function
9687 */
9688 static void
9689f_tr(typval_T *argvars, typval_T *rettv)
9690{
9691 char_u *in_str;
9692 char_u *fromstr;
9693 char_u *tostr;
9694 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009695 int inlen;
9696 int fromlen;
9697 int tolen;
9698 int idx;
9699 char_u *cpstr;
9700 int cplen;
9701 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009702 char_u buf[NUMBUFLEN];
9703 char_u buf2[NUMBUFLEN];
9704 garray_T ga;
9705
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009706 in_str = tv_get_string(&argvars[0]);
9707 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
9708 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009709
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009710 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009711 rettv->v_type = VAR_STRING;
9712 rettv->vval.v_string = NULL;
9713 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009714 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009715 ga_init2(&ga, (int)sizeof(char), 80);
9716
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009717 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009718 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009719 if (STRLEN(fromstr) != STRLEN(tostr))
9720 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009721error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009722 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009723 ga_clear(&ga);
9724 return;
9725 }
9726
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009727 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009728 while (*in_str != NUL)
9729 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009730 if (has_mbyte)
9731 {
9732 inlen = (*mb_ptr2len)(in_str);
9733 cpstr = in_str;
9734 cplen = inlen;
9735 idx = 0;
9736 for (p = fromstr; *p != NUL; p += fromlen)
9737 {
9738 fromlen = (*mb_ptr2len)(p);
9739 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
9740 {
9741 for (p = tostr; *p != NUL; p += tolen)
9742 {
9743 tolen = (*mb_ptr2len)(p);
9744 if (idx-- == 0)
9745 {
9746 cplen = tolen;
9747 cpstr = p;
9748 break;
9749 }
9750 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009751 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009752 goto error;
9753 break;
9754 }
9755 ++idx;
9756 }
9757
9758 if (first && cpstr == in_str)
9759 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009760 // Check that fromstr and tostr have the same number of
9761 // (multi-byte) characters. Done only once when a character
9762 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009763 first = FALSE;
9764 for (p = tostr; *p != NUL; p += tolen)
9765 {
9766 tolen = (*mb_ptr2len)(p);
9767 --idx;
9768 }
9769 if (idx != 0)
9770 goto error;
9771 }
9772
9773 (void)ga_grow(&ga, cplen);
9774 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
9775 ga.ga_len += cplen;
9776
9777 in_str += inlen;
9778 }
9779 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009780 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009781 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009782 p = vim_strchr(fromstr, *in_str);
9783 if (p != NULL)
9784 ga_append(&ga, tostr[p - fromstr]);
9785 else
9786 ga_append(&ga, *in_str);
9787 ++in_str;
9788 }
9789 }
9790
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009791 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009792 (void)ga_grow(&ga, 1);
9793 ga_append(&ga, NUL);
9794
9795 rettv->vval.v_string = ga.ga_data;
9796}
9797
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009798/*
9799 * "trim({expr})" function
9800 */
9801 static void
9802f_trim(typval_T *argvars, typval_T *rettv)
9803{
9804 char_u buf1[NUMBUFLEN];
9805 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009806 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009807 char_u *mask = NULL;
9808 char_u *tail;
9809 char_u *prev;
9810 char_u *p;
9811 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009812 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009813
9814 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009815 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009816 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009817 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009818
9819 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009820 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009821 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009822
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009823 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009824 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009825 int error = 0;
9826
9827 // leading or trailing characters to trim
9828 dir = (int)tv_get_number_chk(&argvars[2], &error);
9829 if (error)
9830 return;
9831 if (dir < 0 || dir > 2)
9832 {
9833 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
9834 return;
9835 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009836 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009837 }
9838
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009839 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009840 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009841 // Trim leading characters
9842 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009843 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009844 c1 = PTR2CHAR(head);
9845 if (mask == NULL)
9846 {
9847 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009848 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009849 }
9850 else
9851 {
9852 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9853 if (c1 == PTR2CHAR(p))
9854 break;
9855 if (*p == NUL)
9856 break;
9857 }
9858 MB_PTR_ADV(head);
9859 }
9860 }
9861
9862 tail = head + STRLEN(head);
9863 if (dir == 0 || dir == 2)
9864 {
9865 // Trim trailing characters
9866 for (; tail > head; tail = prev)
9867 {
9868 prev = tail;
9869 MB_PTR_BACK(head, prev);
9870 c1 = PTR2CHAR(prev);
9871 if (mask == NULL)
9872 {
9873 if (c1 > ' ' && c1 != 0xa0)
9874 break;
9875 }
9876 else
9877 {
9878 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9879 if (c1 == PTR2CHAR(p))
9880 break;
9881 if (*p == NUL)
9882 break;
9883 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009884 }
9885 }
Bram Moolenaardf44a272020-06-07 20:49:05 +02009886 rettv->vval.v_string = vim_strnsave(head, tail - head);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009887}
9888
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009889#ifdef FEAT_FLOAT
9890/*
9891 * "trunc({float})" function
9892 */
9893 static void
9894f_trunc(typval_T *argvars, typval_T *rettv)
9895{
9896 float_T f = 0.0;
9897
9898 rettv->v_type = VAR_FLOAT;
9899 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009900 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009901 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
9902 else
9903 rettv->vval.v_float = 0.0;
9904}
9905#endif
9906
9907/*
9908 * "type(expr)" function
9909 */
9910 static void
9911f_type(typval_T *argvars, typval_T *rettv)
9912{
9913 int n = -1;
9914
9915 switch (argvars[0].v_type)
9916 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01009917 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
9918 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009919 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01009920 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
9921 case VAR_LIST: n = VAR_TYPE_LIST; break;
9922 case VAR_DICT: n = VAR_TYPE_DICT; break;
9923 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
9924 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
9925 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02009926 case VAR_JOB: n = VAR_TYPE_JOB; break;
9927 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01009928 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009929 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02009930 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01009931 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01009932 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009933 n = -1;
9934 break;
9935 }
9936 rettv->vval.v_number = n;
9937}
9938
9939/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009940 * "virtcol(string)" function
9941 */
9942 static void
9943f_virtcol(typval_T *argvars, typval_T *rettv)
9944{
9945 colnr_T vcol = 0;
9946 pos_T *fp;
9947 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01009948 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009949
9950 fp = var2fpos(&argvars[0], FALSE, &fnum);
9951 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
9952 && fnum == curbuf->b_fnum)
9953 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01009954 // Limit the column to a valid value, getvvcol() doesn't check.
9955 if (fp->col < 0)
9956 fp->col = 0;
9957 else
9958 {
9959 len = (int)STRLEN(ml_get(fp->lnum));
9960 if (fp->col > len)
9961 fp->col = len;
9962 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009963 getvvcol(curwin, fp, NULL, NULL, &vcol);
9964 ++vcol;
9965 }
9966
9967 rettv->vval.v_number = vcol;
9968}
9969
9970/*
9971 * "visualmode()" function
9972 */
9973 static void
9974f_visualmode(typval_T *argvars, typval_T *rettv)
9975{
9976 char_u str[2];
9977
9978 rettv->v_type = VAR_STRING;
9979 str[0] = curbuf->b_visual_mode_eval;
9980 str[1] = NUL;
9981 rettv->vval.v_string = vim_strsave(str);
9982
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009983 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009984 if (non_zero_arg(&argvars[0]))
9985 curbuf->b_visual_mode_eval = NUL;
9986}
9987
9988/*
9989 * "wildmenumode()" function
9990 */
9991 static void
9992f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9993{
9994#ifdef FEAT_WILDMENU
9995 if (wild_menu_showing)
9996 rettv->vval.v_number = 1;
9997#endif
9998}
9999
10000/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +010010001 * "windowsversion()" function
10002 */
10003 static void
10004f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
10005{
10006 rettv->v_type = VAR_STRING;
10007 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
10008}
10009
10010/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010011 * "wordcount()" function
10012 */
10013 static void
10014f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
10015{
10016 if (rettv_dict_alloc(rettv) == FAIL)
10017 return;
10018 cursor_pos_info(rettv->vval.v_dict);
10019}
10020
10021/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010022 * "xor(expr, expr)" function
10023 */
10024 static void
10025f_xor(typval_T *argvars, typval_T *rettv)
10026{
Bram Moolenaard155d7a2018-12-21 16:04:21 +010010027 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
10028 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020010029}
10030
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010010031#endif // FEAT_EVAL