blob: d8e88e5c692f6d12cbf98df5856daf0f92e6bf38 [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 Moolenaar73dad1e2016-07-17 22:13:49 +020050static void f_col(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020051static void f_confirm(typval_T *argvars, typval_T *rettv);
52static void f_copy(typval_T *argvars, typval_T *rettv);
53#ifdef FEAT_FLOAT
54static void f_cos(typval_T *argvars, typval_T *rettv);
55static void f_cosh(typval_T *argvars, typval_T *rettv);
56#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020057static void f_cursor(typval_T *argsvars, typval_T *rettv);
Bram Moolenaar4f974752019-02-17 17:44:42 +010058#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +020059static void f_debugbreak(typval_T *argvars, typval_T *rettv);
60#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020061static void f_deepcopy(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020062static void f_did_filetype(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4132eb52020-02-14 16:53:00 +010063static void f_echoraw(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020064static void f_empty(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020065static void f_environ(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020066static void f_escape(typval_T *argvars, typval_T *rettv);
67static void f_eval(typval_T *argvars, typval_T *rettv);
68static void f_eventhandler(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020069static void f_execute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020070static void f_exists(typval_T *argvars, typval_T *rettv);
71#ifdef FEAT_FLOAT
72static void f_exp(typval_T *argvars, typval_T *rettv);
73#endif
74static void f_expand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar80dad482019-06-09 17:22:31 +020075static void f_expandcmd(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020076static void f_feedkeys(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020077#ifdef FEAT_FLOAT
78static void f_float2nr(typval_T *argvars, typval_T *rettv);
79static void f_floor(typval_T *argvars, typval_T *rettv);
80static void f_fmod(typval_T *argvars, typval_T *rettv);
81#endif
82static void f_fnameescape(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020083static void f_foreground(typval_T *argvars, typval_T *rettv);
Bram Moolenaar437bafe2016-08-01 15:40:54 +020084static void f_funcref(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020085static void f_function(typval_T *argvars, typval_T *rettv);
86static void f_garbagecollect(typval_T *argvars, typval_T *rettv);
87static void f_get(typval_T *argvars, typval_T *rettv);
Bram Moolenaar07ad8162018-02-13 13:59:59 +010088static void f_getchangelist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020089static void f_getcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +020090static void f_getenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020091static void f_getfontname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar4f505882018-02-10 21:06:32 +010092static void f_getjumplist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020093static void f_getpid(typval_T *argvars, typval_T *rettv);
94static void f_getcurpos(typval_T *argvars, typval_T *rettv);
95static void f_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020096static void f_getreg(typval_T *argvars, typval_T *rettv);
97static void f_getregtype(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +010098static void f_gettagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020099static void f_haslocaldir(typval_T *argvars, typval_T *rettv);
100static void f_hasmapto(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200101static void f_hlID(typval_T *argvars, typval_T *rettv);
102static void f_hlexists(typval_T *argvars, typval_T *rettv);
103static void f_hostname(typval_T *argvars, typval_T *rettv);
104static void f_iconv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200105static void f_index(typval_T *argvars, typval_T *rettv);
106static void f_input(typval_T *argvars, typval_T *rettv);
107static void f_inputdialog(typval_T *argvars, typval_T *rettv);
108static void f_inputlist(typval_T *argvars, typval_T *rettv);
109static void f_inputrestore(typval_T *argvars, typval_T *rettv);
110static void f_inputsave(typval_T *argvars, typval_T *rettv);
111static void f_inputsecret(typval_T *argvars, typval_T *rettv);
Bram Moolenaar67a2deb2019-11-25 00:05:32 +0100112static void f_interrupt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200113static void f_invert(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200114static void f_islocked(typval_T *argvars, typval_T *rettv);
115#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
Bram Moolenaarfda1bff2019-04-04 13:44:37 +0200116static void f_isinf(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200117static void f_isnan(typval_T *argvars, typval_T *rettv);
118#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200119static void f_last_buffer_nr(typval_T *argvars, typval_T *rettv);
120static void f_len(typval_T *argvars, typval_T *rettv);
121static void f_libcall(typval_T *argvars, typval_T *rettv);
122static void f_libcallnr(typval_T *argvars, typval_T *rettv);
123static void f_line(typval_T *argvars, typval_T *rettv);
124static void f_line2byte(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200125#ifdef FEAT_FLOAT
126static void f_log(typval_T *argvars, typval_T *rettv);
127static void f_log10(typval_T *argvars, typval_T *rettv);
128#endif
129#ifdef FEAT_LUA
130static void f_luaeval(typval_T *argvars, typval_T *rettv);
131#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200132static void f_maparg(typval_T *argvars, typval_T *rettv);
133static void f_mapcheck(typval_T *argvars, typval_T *rettv);
134static void f_match(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200135static void f_matchend(typval_T *argvars, typval_T *rettv);
136static void f_matchlist(typval_T *argvars, typval_T *rettv);
137static void f_matchstr(typval_T *argvars, typval_T *rettv);
138static void f_matchstrpos(typval_T *argvars, typval_T *rettv);
139static void f_max(typval_T *argvars, typval_T *rettv);
140static void f_min(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200141#ifdef FEAT_MZSCHEME
142static void f_mzeval(typval_T *argvars, typval_T *rettv);
143#endif
144static void f_nextnonblank(typval_T *argvars, typval_T *rettv);
145static void f_nr2char(typval_T *argvars, typval_T *rettv);
146static void f_or(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200147#ifdef FEAT_PERL
148static void f_perleval(typval_T *argvars, typval_T *rettv);
149#endif
150#ifdef FEAT_FLOAT
151static void f_pow(typval_T *argvars, typval_T *rettv);
152#endif
153static void f_prevnonblank(typval_T *argvars, typval_T *rettv);
154static void f_printf(typval_T *argvars, typval_T *rettv);
Bram Moolenaare9bd5722019-08-17 19:36:06 +0200155static void f_pum_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200156static void f_pumvisible(typval_T *argvars, typval_T *rettv);
157#ifdef FEAT_PYTHON3
158static void f_py3eval(typval_T *argvars, typval_T *rettv);
159#endif
160#ifdef FEAT_PYTHON
161static void f_pyeval(typval_T *argvars, typval_T *rettv);
162#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100163#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
164static void f_pyxeval(typval_T *argvars, typval_T *rettv);
165#endif
Bram Moolenaar4f645c52020-02-08 16:40:39 +0100166static void f_test_srand_seed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100167static void f_rand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200168static void f_range(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b6d9112018-05-22 20:35:17 +0200169static void f_reg_executing(typval_T *argvars, typval_T *rettv);
170static void f_reg_recording(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200171static void f_rename(typval_T *argvars, typval_T *rettv);
172static void f_repeat(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200173#ifdef FEAT_FLOAT
174static void f_round(typval_T *argvars, typval_T *rettv);
175#endif
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100176#ifdef FEAT_RUBY
177static void f_rubyeval(typval_T *argvars, typval_T *rettv);
178#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200179static void f_screenattr(typval_T *argvars, typval_T *rettv);
180static void f_screenchar(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100181static void f_screenchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200182static void f_screencol(typval_T *argvars, typval_T *rettv);
183static void f_screenrow(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100184static void f_screenstring(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200185static void f_search(typval_T *argvars, typval_T *rettv);
186static void f_searchdecl(typval_T *argvars, typval_T *rettv);
187static void f_searchpair(typval_T *argvars, typval_T *rettv);
188static void f_searchpairpos(typval_T *argvars, typval_T *rettv);
189static void f_searchpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200190static void f_setcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +0200191static void f_setenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200192static void f_setfperm(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200193static void f_setpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200194static void f_setreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100195static void f_settagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200196#ifdef FEAT_CRYPT
197static void f_sha256(typval_T *argvars, typval_T *rettv);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200198#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200199static void f_shellescape(typval_T *argvars, typval_T *rettv);
200static void f_shiftwidth(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200201#ifdef FEAT_FLOAT
202static void f_sin(typval_T *argvars, typval_T *rettv);
203static void f_sinh(typval_T *argvars, typval_T *rettv);
204#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200205static void f_soundfold(typval_T *argvars, typval_T *rettv);
206static void f_spellbadword(typval_T *argvars, typval_T *rettv);
207static void f_spellsuggest(typval_T *argvars, typval_T *rettv);
208static void f_split(typval_T *argvars, typval_T *rettv);
209#ifdef FEAT_FLOAT
210static void f_sqrt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100211#endif
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100212static void f_srand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100213#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200214static void f_str2float(typval_T *argvars, typval_T *rettv);
215#endif
Bram Moolenaar9d401282019-04-06 13:18:12 +0200216static void f_str2list(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200217static void f_str2nr(typval_T *argvars, typval_T *rettv);
218static void f_strchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200219static void f_strgetchar(typval_T *argvars, typval_T *rettv);
220static void f_stridx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200221static void f_strlen(typval_T *argvars, typval_T *rettv);
222static void f_strcharpart(typval_T *argvars, typval_T *rettv);
223static void f_strpart(typval_T *argvars, typval_T *rettv);
224static void f_strridx(typval_T *argvars, typval_T *rettv);
225static void f_strtrans(typval_T *argvars, typval_T *rettv);
226static void f_strdisplaywidth(typval_T *argvars, typval_T *rettv);
227static void f_strwidth(typval_T *argvars, typval_T *rettv);
228static void f_submatch(typval_T *argvars, typval_T *rettv);
229static void f_substitute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar00f123a2018-08-21 20:28:54 +0200230static void f_swapinfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar110bd602018-09-16 18:46:59 +0200231static void f_swapname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200232static void f_synID(typval_T *argvars, typval_T *rettv);
233static void f_synIDattr(typval_T *argvars, typval_T *rettv);
234static void f_synIDtrans(typval_T *argvars, typval_T *rettv);
235static void f_synstack(typval_T *argvars, typval_T *rettv);
236static void f_synconcealed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200237static void f_tabpagebuflist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200238static void f_taglist(typval_T *argvars, typval_T *rettv);
239static void f_tagfiles(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200240#ifdef FEAT_FLOAT
241static void f_tan(typval_T *argvars, typval_T *rettv);
242static void f_tanh(typval_T *argvars, typval_T *rettv);
243#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200244static void f_tolower(typval_T *argvars, typval_T *rettv);
245static void f_toupper(typval_T *argvars, typval_T *rettv);
246static void f_tr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +0100247static void f_trim(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200248#ifdef FEAT_FLOAT
249static void f_trunc(typval_T *argvars, typval_T *rettv);
250#endif
251static void f_type(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200252static void f_virtcol(typval_T *argvars, typval_T *rettv);
253static void f_visualmode(typval_T *argvars, typval_T *rettv);
254static void f_wildmenumode(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0c1e3742019-12-27 13:49:24 +0100255static void f_windowsversion(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200256static void f_wordcount(typval_T *argvars, typval_T *rettv);
257static void f_xor(typval_T *argvars, typval_T *rettv);
258
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100259
260 static type_T *
261ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
262{
263 return &t_void;
264}
265 static type_T *
266ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
267{
268 return &t_any;
269}
270 static type_T *
271ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
272{
273 return &t_number;
274}
275 static type_T *
276ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
277{
278 return &t_float;
279}
280 static type_T *
281ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
282{
283 return &t_string;
284}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200285 static type_T *
286ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100287{
288 return &t_list_any;
289}
290 static type_T *
291ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
292{
293 return &t_list_number;
294}
295 static type_T *
296ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
297{
298 return &t_list_string;
299}
300 static type_T *
301ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
302{
303 return &t_list_dict_any;
304}
305 static type_T *
306ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
307{
308 return &t_dict_any;
309}
310 static type_T *
311ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
312{
313 return &t_dict_number;
314}
315 static type_T *
316ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
317{
318 return &t_dict_string;
319}
320 static type_T *
321ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
322{
323 return &t_blob;
324}
325 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200326ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100327{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200328 return &t_func_any;
329}
330 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100331ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
332{
333 return &t_channel;
334}
335 static type_T *
336ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
337{
338 return &t_job;
339}
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100340
341static type_T *ret_f_function(int argcount, type_T **argtypes);
342
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200343/*
344 * Array with names and number of arguments of all internal functions
345 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
346 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200347typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200348{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200349 char *f_name; // function name
350 char f_min_argc; // minimal number of arguments
351 char f_max_argc; // maximal number of arguments
352 char f_argtype; // for method: FEARG_ values
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100353 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
354 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200355 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200356 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200357} funcentry_T;
358
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200359// values for f_argtype; zero means it cannot be used as a method
360#define FEARG_1 1 // base is the first argument
361#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200362#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200363#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200364#define FEARG_LAST 9 // base is the last argument
365
Bram Moolenaar15c47602020-03-26 22:16:48 +0100366#ifdef FEAT_FLOAT
367# define FLOAT_FUNC(name) name
368#else
369# define FLOAT_FUNC(name) NULL
370#endif
371#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
372# define MATH_FUNC(name) name
373#else
374# define MATH_FUNC(name) NULL
375#endif
376#ifdef FEAT_TIMERS
377# define TIMER_FUNC(name) name
378#else
379# define TIMER_FUNC(name) NULL
380#endif
381#ifdef FEAT_JOB_CHANNEL
382# define JOB_FUNC(name) name
383#else
384# define JOB_FUNC(name) NULL
385#endif
386#ifdef FEAT_PROP_POPUP
387# define PROP_FUNC(name) name
388#else
389# define PROP_FUNC(name) NULL
390#endif
391#ifdef FEAT_SIGNS
392# define SIGN_FUNC(name) name
393#else
394# define SIGN_FUNC(name) NULL
395#endif
396#ifdef FEAT_SOUND
397# define SOUND_FUNC(name) name
398#else
399# define SOUND_FUNC(name) NULL
400#endif
401#ifdef FEAT_TERMINAL
402# define TERM_FUNC(name) name
403#else
404# define TERM_FUNC(name) NULL
405#endif
406
Bram Moolenaarac92e252019-08-03 21:58:38 +0200407static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200408{
Bram Moolenaar15c47602020-03-26 22:16:48 +0100409 {"abs", 1, 1, FEARG_1, ret_any, FLOAT_FUNC(f_abs)},
410 {"acos", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_acos)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100411 {"add", 2, 2, FEARG_1, ret_any, f_add},
412 {"and", 2, 2, FEARG_1, ret_number, f_and},
413 {"append", 2, 2, FEARG_LAST, ret_number, f_append},
414 {"appendbufline", 3, 3, FEARG_LAST, ret_number, f_appendbufline},
415 {"argc", 0, 1, 0, ret_number, f_argc},
416 {"argidx", 0, 0, 0, ret_number, f_argidx},
417 {"arglistid", 0, 2, 0, ret_number, f_arglistid},
418 {"argv", 0, 2, 0, ret_any, f_argv},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100419 {"asin", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_asin)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100420 {"assert_beeps", 1, 2, FEARG_1, ret_number, f_assert_beeps},
421 {"assert_equal", 2, 3, FEARG_2, ret_number, f_assert_equal},
422 {"assert_equalfile", 2, 2, FEARG_1, ret_number, f_assert_equalfile},
423 {"assert_exception", 1, 2, 0, ret_number, f_assert_exception},
424 {"assert_fails", 1, 3, FEARG_1, ret_number, f_assert_fails},
425 {"assert_false", 1, 2, FEARG_1, ret_number, f_assert_false},
426 {"assert_inrange", 3, 4, FEARG_3, ret_number, f_assert_inrange},
427 {"assert_match", 2, 3, FEARG_2, ret_number, f_assert_match},
428 {"assert_notequal", 2, 3, FEARG_2, ret_number, f_assert_notequal},
429 {"assert_notmatch", 2, 3, FEARG_2, ret_number, f_assert_notmatch},
430 {"assert_report", 1, 1, FEARG_1, ret_number, f_assert_report},
431 {"assert_true", 1, 2, FEARG_1, ret_number, f_assert_true},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100432 {"atan", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_atan)},
433 {"atan2", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_atan2)},
434 {"balloon_gettext", 0, 0, 0, ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100435#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100436 f_balloon_gettext
437#else
438 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100439#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100440 },
441 {"balloon_show", 1, 1, FEARG_1, ret_void,
442#ifdef FEAT_BEVAL
443 f_balloon_show
444#else
445 NULL
446#endif
447 },
448 {"balloon_split", 1, 1, FEARG_1, ret_list_string,
449#if defined(FEAT_BEVAL_TERM)
450 f_balloon_split
451#else
452 NULL
453#endif
454 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100455 {"browse", 4, 4, 0, ret_string, f_browse},
456 {"browsedir", 2, 2, 0, ret_string, f_browsedir},
457 {"bufadd", 1, 1, FEARG_1, ret_number, f_bufadd},
458 {"bufexists", 1, 1, FEARG_1, ret_number, f_bufexists},
459 {"buffer_exists", 1, 1, FEARG_1, ret_number, f_bufexists}, // obsolete
460 {"buffer_name", 0, 1, FEARG_1, ret_string, f_bufname}, // obsolete
461 {"buffer_number", 0, 1, FEARG_1, ret_number, f_bufnr}, // obsolete
462 {"buflisted", 1, 1, FEARG_1, ret_number, f_buflisted},
463 {"bufload", 1, 1, FEARG_1, ret_void, f_bufload},
464 {"bufloaded", 1, 1, FEARG_1, ret_number, f_bufloaded},
465 {"bufname", 0, 1, FEARG_1, ret_string, f_bufname},
466 {"bufnr", 0, 2, FEARG_1, ret_number, f_bufnr},
467 {"bufwinid", 1, 1, FEARG_1, ret_number, f_bufwinid},
468 {"bufwinnr", 1, 1, FEARG_1, ret_number, f_bufwinnr},
469 {"byte2line", 1, 1, FEARG_1, ret_number, f_byte2line},
470 {"byteidx", 2, 2, FEARG_1, ret_number, f_byteidx},
471 {"byteidxcomp", 2, 2, FEARG_1, ret_number, f_byteidxcomp},
472 {"call", 2, 3, FEARG_1, ret_any, f_call},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100473 {"ceil", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_ceil)},
474 {"ch_canread", 1, 1, FEARG_1, ret_number, JOB_FUNC(f_ch_canread)},
475 {"ch_close", 1, 1, FEARG_1, ret_void, JOB_FUNC(f_ch_close)},
476 {"ch_close_in", 1, 1, FEARG_1, ret_void, JOB_FUNC(f_ch_close_in)},
477 {"ch_evalexpr", 2, 3, FEARG_1, ret_any, JOB_FUNC(f_ch_evalexpr)},
478 {"ch_evalraw", 2, 3, FEARG_1, ret_any, JOB_FUNC(f_ch_evalraw)},
479 {"ch_getbufnr", 2, 2, FEARG_1, ret_number, JOB_FUNC(f_ch_getbufnr)},
480 {"ch_getjob", 1, 1, FEARG_1, ret_job, JOB_FUNC(f_ch_getjob)},
481 {"ch_info", 1, 1, FEARG_1, ret_dict_any, JOB_FUNC(f_ch_info)},
482 {"ch_log", 1, 2, FEARG_1, ret_void, JOB_FUNC(f_ch_log)},
483 {"ch_logfile", 1, 2, FEARG_1, ret_void, JOB_FUNC(f_ch_logfile)},
484 {"ch_open", 1, 2, FEARG_1, ret_channel, JOB_FUNC(f_ch_open)},
485 {"ch_read", 1, 2, FEARG_1, ret_string, JOB_FUNC(f_ch_read)},
486 {"ch_readblob", 1, 2, FEARG_1, ret_blob, JOB_FUNC(f_ch_readblob)},
487 {"ch_readraw", 1, 2, FEARG_1, ret_string, JOB_FUNC(f_ch_readraw)},
488 {"ch_sendexpr", 2, 3, FEARG_1, ret_void, JOB_FUNC(f_ch_sendexpr)},
489 {"ch_sendraw", 2, 3, FEARG_1, ret_void, JOB_FUNC(f_ch_sendraw)},
490 {"ch_setoptions", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_ch_setoptions)},
491 {"ch_status", 1, 2, FEARG_1, ret_string, JOB_FUNC(f_ch_status)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100492 {"changenr", 0, 0, 0, ret_number, f_changenr},
493 {"char2nr", 1, 2, FEARG_1, ret_number, f_char2nr},
494 {"chdir", 1, 1, FEARG_1, ret_string, f_chdir},
495 {"cindent", 1, 1, FEARG_1, ret_number, f_cindent},
496 {"clearmatches", 0, 1, FEARG_1, ret_void, f_clearmatches},
497 {"col", 1, 1, FEARG_1, ret_number, f_col},
498 {"complete", 2, 2, FEARG_2, ret_void, f_complete},
499 {"complete_add", 1, 1, FEARG_1, ret_number, f_complete_add},
500 {"complete_check", 0, 0, 0, ret_number, f_complete_check},
501 {"complete_info", 0, 1, FEARG_1, ret_dict_any, f_complete_info},
502 {"confirm", 1, 4, FEARG_1, ret_number, f_confirm},
503 {"copy", 1, 1, FEARG_1, ret_any, f_copy},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100504 {"cos", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_cos)},
505 {"cosh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_cosh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100506 {"count", 2, 4, FEARG_1, ret_number, f_count},
507 {"cscope_connection",0,3, 0, ret_number, f_cscope_connection},
508 {"cursor", 1, 3, FEARG_1, ret_number, f_cursor},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100509 {"debugbreak", 1, 1, FEARG_1, ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100510#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100511 f_debugbreak
512#else
513 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200514#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100515 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100516 {"deepcopy", 1, 2, FEARG_1, ret_any, f_deepcopy},
517 {"delete", 1, 2, FEARG_1, ret_number, f_delete},
518 {"deletebufline", 2, 3, FEARG_1, ret_number, f_deletebufline},
519 {"did_filetype", 0, 0, 0, ret_number, f_did_filetype},
520 {"diff_filler", 1, 1, FEARG_1, ret_number, f_diff_filler},
521 {"diff_hlID", 2, 2, FEARG_1, ret_number, f_diff_hlID},
522 {"echoraw", 1, 1, FEARG_1, ret_number, f_echoraw},
523 {"empty", 1, 1, FEARG_1, ret_number, f_empty},
524 {"environ", 0, 0, 0, ret_dict_string, f_environ},
525 {"escape", 2, 2, FEARG_1, ret_string, f_escape},
526 {"eval", 1, 1, FEARG_1, ret_any, f_eval},
527 {"eventhandler", 0, 0, 0, ret_number, f_eventhandler},
528 {"executable", 1, 1, FEARG_1, ret_number, f_executable},
529 {"execute", 1, 2, FEARG_1, ret_string, f_execute},
530 {"exepath", 1, 1, FEARG_1, ret_string, f_exepath},
531 {"exists", 1, 1, FEARG_1, ret_number, f_exists},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100532 {"exp", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_exp)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100533 {"expand", 1, 3, FEARG_1, ret_any, f_expand},
534 {"expandcmd", 1, 1, FEARG_1, ret_string, f_expandcmd},
535 {"extend", 2, 3, FEARG_1, ret_any, f_extend},
536 {"feedkeys", 1, 2, FEARG_1, ret_void, f_feedkeys},
537 {"file_readable", 1, 1, FEARG_1, ret_number, f_filereadable}, // obsolete
538 {"filereadable", 1, 1, FEARG_1, ret_number, f_filereadable},
539 {"filewritable", 1, 1, FEARG_1, ret_number, f_filewritable},
540 {"filter", 2, 2, FEARG_1, ret_any, f_filter},
541 {"finddir", 1, 3, FEARG_1, ret_string, f_finddir},
542 {"findfile", 1, 3, FEARG_1, ret_string, f_findfile},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100543 {"float2nr", 1, 1, FEARG_1, ret_number, FLOAT_FUNC(f_float2nr)},
544 {"floor", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_floor)},
545 {"fmod", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_fmod)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100546 {"fnameescape", 1, 1, FEARG_1, ret_string, f_fnameescape},
547 {"fnamemodify", 2, 2, FEARG_1, ret_string, f_fnamemodify},
548 {"foldclosed", 1, 1, FEARG_1, ret_number, f_foldclosed},
549 {"foldclosedend", 1, 1, FEARG_1, ret_number, f_foldclosedend},
550 {"foldlevel", 1, 1, FEARG_1, ret_number, f_foldlevel},
551 {"foldtext", 0, 0, 0, ret_string, f_foldtext},
552 {"foldtextresult", 1, 1, FEARG_1, ret_string, f_foldtextresult},
553 {"foreground", 0, 0, 0, ret_void, f_foreground},
Bram Moolenaard77a8522020-04-03 21:59:57 +0200554 {"funcref", 1, 3, FEARG_1, ret_func_any, f_funcref},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100555 {"function", 1, 3, FEARG_1, ret_f_function, f_function},
556 {"garbagecollect", 0, 1, 0, ret_void, f_garbagecollect},
557 {"get", 2, 3, FEARG_1, ret_any, f_get},
558 {"getbufinfo", 0, 1, 0, ret_list_dict_any, f_getbufinfo},
559 {"getbufline", 2, 3, FEARG_1, ret_list_string, f_getbufline},
560 {"getbufvar", 2, 3, FEARG_1, ret_any, f_getbufvar},
561 {"getchangelist", 0, 1, FEARG_1, ret_list_any, f_getchangelist},
562 {"getchar", 0, 1, 0, ret_number, f_getchar},
563 {"getcharmod", 0, 0, 0, ret_number, f_getcharmod},
564 {"getcharsearch", 0, 0, 0, ret_dict_any, f_getcharsearch},
565 {"getcmdline", 0, 0, 0, ret_string, f_getcmdline},
566 {"getcmdpos", 0, 0, 0, ret_number, f_getcmdpos},
567 {"getcmdtype", 0, 0, 0, ret_string, f_getcmdtype},
568 {"getcmdwintype", 0, 0, 0, ret_string, f_getcmdwintype},
569 {"getcompletion", 2, 3, FEARG_1, ret_list_string, f_getcompletion},
570 {"getcurpos", 0, 0, 0, ret_list_number, f_getcurpos},
571 {"getcwd", 0, 2, FEARG_1, ret_string, f_getcwd},
572 {"getenv", 1, 1, FEARG_1, ret_string, f_getenv},
573 {"getfontname", 0, 1, 0, ret_string, f_getfontname},
574 {"getfperm", 1, 1, FEARG_1, ret_string, f_getfperm},
575 {"getfsize", 1, 1, FEARG_1, ret_number, f_getfsize},
576 {"getftime", 1, 1, FEARG_1, ret_number, f_getftime},
577 {"getftype", 1, 1, FEARG_1, ret_string, f_getftype},
578 {"getimstatus", 0, 0, 0, ret_number, f_getimstatus},
579 {"getjumplist", 0, 2, FEARG_1, ret_list_any, f_getjumplist},
580 {"getline", 1, 2, FEARG_1, ret_f_getline, f_getline},
581 {"getloclist", 1, 2, 0, ret_list_dict_any, f_getloclist},
Bram Moolenaarf17e7ea2020-06-01 14:14:44 +0200582 {"getmarklist", 0, 1, FEARG_1, ret_list_dict_any, f_getmarklist},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100583 {"getmatches", 0, 1, 0, ret_list_dict_any, f_getmatches},
584 {"getmousepos", 0, 0, 0, ret_dict_number, f_getmousepos},
585 {"getpid", 0, 0, 0, ret_number, f_getpid},
586 {"getpos", 1, 1, FEARG_1, ret_list_number, f_getpos},
587 {"getqflist", 0, 1, 0, ret_list_dict_any, f_getqflist},
588 {"getreg", 0, 3, FEARG_1, ret_string, f_getreg},
589 {"getregtype", 0, 1, FEARG_1, ret_string, f_getregtype},
590 {"gettabinfo", 0, 1, FEARG_1, ret_list_dict_any, f_gettabinfo},
591 {"gettabvar", 2, 3, FEARG_1, ret_any, f_gettabvar},
592 {"gettabwinvar", 3, 4, FEARG_1, ret_any, f_gettabwinvar},
593 {"gettagstack", 0, 1, FEARG_1, ret_dict_any, f_gettagstack},
594 {"getwininfo", 0, 1, FEARG_1, ret_list_dict_any, f_getwininfo},
595 {"getwinpos", 0, 1, FEARG_1, ret_list_number, f_getwinpos},
596 {"getwinposx", 0, 0, 0, ret_number, f_getwinposx},
597 {"getwinposy", 0, 0, 0, ret_number, f_getwinposy},
598 {"getwinvar", 2, 3, FEARG_1, ret_any, f_getwinvar},
599 {"glob", 1, 4, FEARG_1, ret_any, f_glob},
600 {"glob2regpat", 1, 1, FEARG_1, ret_string, f_glob2regpat},
601 {"globpath", 2, 5, FEARG_2, ret_any, f_globpath},
Bram Moolenaar79296512020-03-22 16:17:14 +0100602 {"has", 1, 2, 0, ret_number, f_has},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100603 {"has_key", 2, 2, FEARG_1, ret_number, f_has_key},
604 {"haslocaldir", 0, 2, FEARG_1, ret_number, f_haslocaldir},
605 {"hasmapto", 1, 3, FEARG_1, ret_number, f_hasmapto},
606 {"highlightID", 1, 1, FEARG_1, ret_number, f_hlID}, // obsolete
607 {"highlight_exists",1, 1, FEARG_1, ret_number, f_hlexists}, // obsolete
608 {"histadd", 2, 2, FEARG_2, ret_number, f_histadd},
609 {"histdel", 1, 2, FEARG_1, ret_number, f_histdel},
610 {"histget", 1, 2, FEARG_1, ret_string, f_histget},
611 {"histnr", 1, 1, FEARG_1, ret_number, f_histnr},
612 {"hlID", 1, 1, FEARG_1, ret_number, f_hlID},
613 {"hlexists", 1, 1, FEARG_1, ret_number, f_hlexists},
614 {"hostname", 0, 0, 0, ret_string, f_hostname},
615 {"iconv", 3, 3, FEARG_1, ret_string, f_iconv},
616 {"indent", 1, 1, FEARG_1, ret_number, f_indent},
617 {"index", 2, 4, FEARG_1, ret_number, f_index},
618 {"input", 1, 3, FEARG_1, ret_string, f_input},
619 {"inputdialog", 1, 3, FEARG_1, ret_string, f_inputdialog},
620 {"inputlist", 1, 1, FEARG_1, ret_number, f_inputlist},
621 {"inputrestore", 0, 0, 0, ret_number, f_inputrestore},
622 {"inputsave", 0, 0, 0, ret_number, f_inputsave},
623 {"inputsecret", 1, 2, FEARG_1, ret_string, f_inputsecret},
624 {"insert", 2, 3, FEARG_1, ret_any, f_insert},
625 {"interrupt", 0, 0, 0, ret_void, f_interrupt},
626 {"invert", 1, 1, FEARG_1, ret_number, f_invert},
627 {"isdirectory", 1, 1, FEARG_1, ret_number, f_isdirectory},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100628 {"isinf", 1, 1, FEARG_1, ret_number, MATH_FUNC(f_isinf)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100629 {"islocked", 1, 1, FEARG_1, ret_number, f_islocked},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100630 {"isnan", 1, 1, FEARG_1, ret_number, MATH_FUNC(f_isnan)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100631 {"items", 1, 1, FEARG_1, ret_list_any, f_items},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100632 {"job_getchannel", 1, 1, FEARG_1, ret_channel, JOB_FUNC(f_job_getchannel)},
633 {"job_info", 0, 1, FEARG_1, ret_dict_any, JOB_FUNC(f_job_info)},
634 {"job_setoptions", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_job_setoptions)},
635 {"job_start", 1, 2, FEARG_1, ret_job, JOB_FUNC(f_job_start)},
636 {"job_status", 1, 1, FEARG_1, ret_string, JOB_FUNC(f_job_status)},
637 {"job_stop", 1, 2, FEARG_1, ret_number, JOB_FUNC(f_job_stop)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100638 {"join", 1, 2, FEARG_1, ret_string, f_join},
639 {"js_decode", 1, 1, FEARG_1, ret_any, f_js_decode},
640 {"js_encode", 1, 1, FEARG_1, ret_string, f_js_encode},
641 {"json_decode", 1, 1, FEARG_1, ret_any, f_json_decode},
642 {"json_encode", 1, 1, FEARG_1, ret_string, f_json_encode},
643 {"keys", 1, 1, FEARG_1, ret_list_any, f_keys},
644 {"last_buffer_nr", 0, 0, 0, ret_number, f_last_buffer_nr}, // obsolete
645 {"len", 1, 1, FEARG_1, ret_number, f_len},
646 {"libcall", 3, 3, FEARG_3, ret_string, f_libcall},
647 {"libcallnr", 3, 3, FEARG_3, ret_number, f_libcallnr},
648 {"line", 1, 2, FEARG_1, ret_number, f_line},
649 {"line2byte", 1, 1, FEARG_1, ret_number, f_line2byte},
650 {"lispindent", 1, 1, FEARG_1, ret_number, f_lispindent},
651 {"list2str", 1, 2, FEARG_1, ret_string, f_list2str},
652 {"listener_add", 1, 2, FEARG_2, ret_number, f_listener_add},
653 {"listener_flush", 0, 1, FEARG_1, ret_void, f_listener_flush},
654 {"listener_remove", 1, 1, FEARG_1, ret_number, f_listener_remove},
655 {"localtime", 0, 0, 0, ret_number, f_localtime},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100656 {"log", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_log)},
657 {"log10", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_log10)},
658 {"luaeval", 1, 2, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200659#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +0100660 f_luaeval
661#else
662 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200663#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100664 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100665 {"map", 2, 2, FEARG_1, ret_any, f_map},
666 {"maparg", 1, 4, FEARG_1, ret_string, f_maparg},
667 {"mapcheck", 1, 3, FEARG_1, ret_string, f_mapcheck},
Bram Moolenaar4c9243f2020-05-22 13:10:44 +0200668 {"mapset", 3, 3, FEARG_1, ret_void, f_mapset},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100669 {"match", 2, 4, FEARG_1, ret_any, f_match},
670 {"matchadd", 2, 5, FEARG_1, ret_number, f_matchadd},
671 {"matchaddpos", 2, 5, FEARG_1, ret_number, f_matchaddpos},
672 {"matcharg", 1, 1, FEARG_1, ret_list_string, f_matcharg},
673 {"matchdelete", 1, 2, FEARG_1, ret_number, f_matchdelete},
674 {"matchend", 2, 4, FEARG_1, ret_number, f_matchend},
675 {"matchlist", 2, 4, FEARG_1, ret_list_string, f_matchlist},
676 {"matchstr", 2, 4, FEARG_1, ret_string, f_matchstr},
677 {"matchstrpos", 2, 4, FEARG_1, ret_list_any, f_matchstrpos},
678 {"max", 1, 1, FEARG_1, ret_any, f_max},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100679 {"menu_info", 1, 2, FEARG_1, ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +0100680#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +0100681 f_menu_info
682#else
683 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +0100684#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100685 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100686 {"min", 1, 1, FEARG_1, ret_any, f_min},
687 {"mkdir", 1, 3, FEARG_1, ret_number, f_mkdir},
688 {"mode", 0, 1, FEARG_1, ret_string, f_mode},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100689 {"mzeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200690#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100691 f_mzeval
692#else
693 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200694#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100695 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100696 {"nextnonblank", 1, 1, FEARG_1, ret_number, f_nextnonblank},
697 {"nr2char", 1, 2, FEARG_1, ret_string, f_nr2char},
698 {"or", 2, 2, FEARG_1, ret_number, f_or},
699 {"pathshorten", 1, 1, FEARG_1, ret_string, f_pathshorten},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100700 {"perleval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200701#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100702 f_perleval
703#else
704 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200705#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100706 },
707 {"popup_atcursor", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_atcursor)},
708 {"popup_beval", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_beval)},
Bram Moolenaar03a9f842020-05-13 13:40:16 +0200709 {"popup_clear", 0, 1, 0, ret_void, PROP_FUNC(f_popup_clear)},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100710 {"popup_close", 1, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_close)},
711 {"popup_create", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_create)},
712 {"popup_dialog", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_dialog)},
713 {"popup_filter_menu", 2, 2, 0, ret_number, PROP_FUNC(f_popup_filter_menu)},
714 {"popup_filter_yesno", 2, 2, 0, ret_number, PROP_FUNC(f_popup_filter_yesno)},
715 {"popup_findinfo", 0, 0, 0, ret_number, PROP_FUNC(f_popup_findinfo)},
716 {"popup_findpreview", 0, 0, 0, ret_number, PROP_FUNC(f_popup_findpreview)},
717 {"popup_getoptions", 1, 1, FEARG_1, ret_dict_any, PROP_FUNC(f_popup_getoptions)},
718 {"popup_getpos", 1, 1, FEARG_1, ret_dict_any, PROP_FUNC(f_popup_getpos)},
719 {"popup_hide", 1, 1, FEARG_1, ret_void, PROP_FUNC(f_popup_hide)},
Bram Moolenaaref6b9792020-05-13 16:34:15 +0200720 {"popup_list", 0, 0, 0, ret_list_number, PROP_FUNC(f_popup_list)},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100721 {"popup_locate", 2, 2, 0, ret_number, PROP_FUNC(f_popup_locate)},
722 {"popup_menu", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_menu)},
723 {"popup_move", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_move)},
724 {"popup_notification", 2, 2, FEARG_1, ret_number, PROP_FUNC(f_popup_notification)},
725 {"popup_setoptions", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_setoptions)},
726 {"popup_settext", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_popup_settext)},
727 {"popup_show", 1, 1, FEARG_1, ret_void, PROP_FUNC(f_popup_show)},
728 {"pow", 2, 2, FEARG_1, ret_float, FLOAT_FUNC(f_pow)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100729 {"prevnonblank", 1, 1, FEARG_1, ret_number, f_prevnonblank},
730 {"printf", 1, 19, FEARG_2, ret_string, f_printf},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100731 {"prompt_setcallback", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setcallback)},
732 {"prompt_setinterrupt", 2, 2, FEARG_1,ret_void, JOB_FUNC(f_prompt_setinterrupt)},
733 {"prompt_setprompt", 2, 2, FEARG_1, ret_void, JOB_FUNC(f_prompt_setprompt)},
734 {"prop_add", 3, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_add)},
735 {"prop_clear", 1, 3, FEARG_1, ret_void, PROP_FUNC(f_prop_clear)},
736 {"prop_find", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_find)},
737 {"prop_list", 1, 2, FEARG_1, ret_list_dict_any, PROP_FUNC(f_prop_list)},
738 {"prop_remove", 1, 3, FEARG_1, ret_number, PROP_FUNC(f_prop_remove)},
739 {"prop_type_add", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_add)},
740 {"prop_type_change", 2, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_change)},
741 {"prop_type_delete", 1, 2, FEARG_1, ret_void, PROP_FUNC(f_prop_type_delete)},
742 {"prop_type_get", 1, 2, FEARG_1, ret_dict_any, PROP_FUNC(f_prop_type_get)},
743 {"prop_type_list", 0, 1, FEARG_1, ret_list_string, PROP_FUNC(f_prop_type_list)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100744 {"pum_getpos", 0, 0, 0, ret_dict_number, f_pum_getpos},
745 {"pumvisible", 0, 0, 0, ret_number, f_pumvisible},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100746 {"py3eval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200747#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +0100748 f_py3eval
749#else
750 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200751#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100752 },
753 {"pyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200754#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +0100755 f_pyeval
756#else
757 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200758#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100759 },
760 {"pyxeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100761#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +0100762 f_pyxeval
763#else
764 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100765#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100766 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100767 {"rand", 0, 1, FEARG_1, ret_number, f_rand},
768 {"range", 1, 3, FEARG_1, ret_list_number, f_range},
769 {"readdir", 1, 2, FEARG_1, ret_list_string, f_readdir},
Bram Moolenaar6c9ba042020-06-01 16:09:41 +0200770 {"readdirex", 1, 2, FEARG_1, ret_list_dict_any, f_readdirex},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100771 {"readfile", 1, 3, FEARG_1, ret_any, f_readfile},
Bram Moolenaar85629982020-06-01 18:39:20 +0200772 {"reduce", 2, 3, FEARG_1, ret_any, f_reduce},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100773 {"reg_executing", 0, 0, 0, ret_string, f_reg_executing},
774 {"reg_recording", 0, 0, 0, ret_string, f_reg_recording},
775 {"reltime", 0, 2, FEARG_1, ret_list_any, f_reltime},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100776 {"reltimefloat", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_reltimefloat)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100777 {"reltimestr", 1, 1, FEARG_1, ret_string, f_reltimestr},
778 {"remote_expr", 2, 4, FEARG_1, ret_string, f_remote_expr},
779 {"remote_foreground", 1, 1, FEARG_1, ret_string, f_remote_foreground},
780 {"remote_peek", 1, 2, FEARG_1, ret_number, f_remote_peek},
781 {"remote_read", 1, 2, FEARG_1, ret_string, f_remote_read},
782 {"remote_send", 2, 3, FEARG_1, ret_string, f_remote_send},
783 {"remote_startserver", 1, 1, FEARG_1, ret_void, f_remote_startserver},
784 {"remove", 2, 3, FEARG_1, ret_any, f_remove},
785 {"rename", 2, 2, FEARG_1, ret_number, f_rename},
786 {"repeat", 2, 2, FEARG_1, ret_any, f_repeat},
787 {"resolve", 1, 1, FEARG_1, ret_string, f_resolve},
788 {"reverse", 1, 1, FEARG_1, ret_any, f_reverse},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100789 {"round", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_round)},
790 {"rubyeval", 1, 1, FEARG_1, ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100791#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +0100792 f_rubyeval
793#else
794 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100795#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100796 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100797 {"screenattr", 2, 2, FEARG_1, ret_number, f_screenattr},
798 {"screenchar", 2, 2, FEARG_1, ret_number, f_screenchar},
799 {"screenchars", 2, 2, FEARG_1, ret_list_number, f_screenchars},
800 {"screencol", 0, 0, 0, ret_number, f_screencol},
801 {"screenpos", 3, 3, FEARG_1, ret_dict_number, f_screenpos},
802 {"screenrow", 0, 0, 0, ret_number, f_screenrow},
803 {"screenstring", 2, 2, FEARG_1, ret_string, f_screenstring},
804 {"search", 1, 4, FEARG_1, ret_number, f_search},
Bram Moolenaare8f5ec02020-06-01 17:28:35 +0200805 {"searchcount", 0, 1, FEARG_1, ret_dict_any, f_searchcount},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100806 {"searchdecl", 1, 3, FEARG_1, ret_number, f_searchdecl},
807 {"searchpair", 3, 7, 0, ret_number, f_searchpair},
808 {"searchpairpos", 3, 7, 0, ret_list_number, f_searchpairpos},
809 {"searchpos", 1, 4, FEARG_1, ret_list_number, f_searchpos},
810 {"server2client", 2, 2, FEARG_1, ret_number, f_server2client},
811 {"serverlist", 0, 0, 0, ret_string, f_serverlist},
812 {"setbufline", 3, 3, FEARG_3, ret_number, f_setbufline},
813 {"setbufvar", 3, 3, FEARG_3, ret_void, f_setbufvar},
814 {"setcharsearch", 1, 1, FEARG_1, ret_void, f_setcharsearch},
815 {"setcmdpos", 1, 1, FEARG_1, ret_number, f_setcmdpos},
816 {"setenv", 2, 2, FEARG_2, ret_void, f_setenv},
817 {"setfperm", 2, 2, FEARG_1, ret_number, f_setfperm},
818 {"setline", 2, 2, FEARG_2, ret_number, f_setline},
819 {"setloclist", 2, 4, FEARG_2, ret_number, f_setloclist},
820 {"setmatches", 1, 2, FEARG_1, ret_number, f_setmatches},
821 {"setpos", 2, 2, FEARG_2, ret_number, f_setpos},
822 {"setqflist", 1, 3, FEARG_1, ret_number, f_setqflist},
823 {"setreg", 2, 3, FEARG_2, ret_number, f_setreg},
824 {"settabvar", 3, 3, FEARG_3, ret_void, f_settabvar},
825 {"settabwinvar", 4, 4, FEARG_4, ret_void, f_settabwinvar},
826 {"settagstack", 2, 3, FEARG_2, ret_number, f_settagstack},
827 {"setwinvar", 3, 3, FEARG_3, ret_void, f_setwinvar},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100828 {"sha256", 1, 1, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200829#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +0100830 f_sha256
831#else
832 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200833#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100834 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100835 {"shellescape", 1, 2, FEARG_1, ret_string, f_shellescape},
836 {"shiftwidth", 0, 1, FEARG_1, ret_number, f_shiftwidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100837 {"sign_define", 1, 2, FEARG_1, ret_any, SIGN_FUNC(f_sign_define)},
838 {"sign_getdefined", 0, 1, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
839 {"sign_getplaced", 0, 2, FEARG_1, ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
840 {"sign_jump", 3, 3, FEARG_1, ret_number, SIGN_FUNC(f_sign_jump)},
841 {"sign_place", 4, 5, FEARG_1, ret_number, SIGN_FUNC(f_sign_place)},
842 {"sign_placelist", 1, 1, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_placelist)},
843 {"sign_undefine", 0, 1, FEARG_1, ret_number, SIGN_FUNC(f_sign_undefine)},
844 {"sign_unplace", 1, 2, FEARG_1, ret_number, SIGN_FUNC(f_sign_unplace)},
845 {"sign_unplacelist", 1, 2, FEARG_1, ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
Bram Moolenaar7035fd92020-04-08 20:03:52 +0200846 {"simplify", 1, 1, FEARG_1, ret_string, f_simplify},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100847 {"sin", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sin)},
848 {"sinh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sinh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100849 {"sort", 1, 3, FEARG_1, ret_list_any, f_sort},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100850 {"sound_clear", 0, 0, 0, ret_void, SOUND_FUNC(f_sound_clear)},
851 {"sound_playevent", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playevent)},
852 {"sound_playfile", 1, 2, FEARG_1, ret_number, SOUND_FUNC(f_sound_playfile)},
853 {"sound_stop", 1, 1, FEARG_1, ret_void, SOUND_FUNC(f_sound_stop)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100854 {"soundfold", 1, 1, FEARG_1, ret_string, f_soundfold},
855 {"spellbadword", 0, 1, FEARG_1, ret_list_string, f_spellbadword},
856 {"spellsuggest", 1, 3, FEARG_1, ret_list_string, f_spellsuggest},
857 {"split", 1, 3, FEARG_1, ret_list_string, f_split},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100858 {"sqrt", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_sqrt)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100859 {"srand", 0, 1, FEARG_1, ret_list_number, f_srand},
860 {"state", 0, 1, FEARG_1, ret_string, f_state},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100861 {"str2float", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_str2float)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100862 {"str2list", 1, 2, FEARG_1, ret_list_number, f_str2list},
863 {"str2nr", 1, 3, FEARG_1, ret_number, f_str2nr},
864 {"strcharpart", 2, 3, FEARG_1, ret_string, f_strcharpart},
865 {"strchars", 1, 2, FEARG_1, ret_number, f_strchars},
866 {"strdisplaywidth", 1, 2, FEARG_1, ret_number, f_strdisplaywidth},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100867 {"strftime", 1, 2, FEARG_1, ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200868#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100869 f_strftime
870#else
871 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200872#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100873 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100874 {"strgetchar", 2, 2, FEARG_1, ret_number, f_strgetchar},
875 {"stridx", 2, 3, FEARG_1, ret_number, f_stridx},
876 {"string", 1, 1, FEARG_1, ret_string, f_string},
877 {"strlen", 1, 1, FEARG_1, ret_number, f_strlen},
878 {"strpart", 2, 3, FEARG_1, ret_string, f_strpart},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100879 {"strptime", 2, 2, FEARG_1, ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +0100880#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +0100881 f_strptime
882#else
883 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +0100884#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100885 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100886 {"strridx", 2, 3, FEARG_1, ret_number, f_strridx},
887 {"strtrans", 1, 1, FEARG_1, ret_string, f_strtrans},
888 {"strwidth", 1, 1, FEARG_1, ret_number, f_strwidth},
889 {"submatch", 1, 2, FEARG_1, ret_string, f_submatch},
890 {"substitute", 4, 4, FEARG_1, ret_string, f_substitute},
891 {"swapinfo", 1, 1, FEARG_1, ret_dict_any, f_swapinfo},
892 {"swapname", 1, 1, FEARG_1, ret_string, f_swapname},
893 {"synID", 3, 3, 0, ret_number, f_synID},
894 {"synIDattr", 2, 3, FEARG_1, ret_string, f_synIDattr},
895 {"synIDtrans", 1, 1, FEARG_1, ret_number, f_synIDtrans},
896 {"synconcealed", 2, 2, 0, ret_list_any, f_synconcealed},
897 {"synstack", 2, 2, 0, ret_list_number, f_synstack},
898 {"system", 1, 2, FEARG_1, ret_string, f_system},
899 {"systemlist", 1, 2, FEARG_1, ret_list_string, f_systemlist},
900 {"tabpagebuflist", 0, 1, FEARG_1, ret_list_number, f_tabpagebuflist},
901 {"tabpagenr", 0, 1, 0, ret_number, f_tabpagenr},
902 {"tabpagewinnr", 1, 2, FEARG_1, ret_number, f_tabpagewinnr},
903 {"tagfiles", 0, 0, 0, ret_list_string, f_tagfiles},
904 {"taglist", 1, 2, FEARG_1, ret_list_dict_any, f_taglist},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100905 {"tan", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tan)},
906 {"tanh", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_tanh)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100907 {"tempname", 0, 0, 0, ret_string, f_tempname},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100908 {"term_dumpdiff", 2, 3, FEARG_1, ret_number, TERM_FUNC(f_term_dumpdiff)},
909 {"term_dumpload", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_dumpload)},
910 {"term_dumpwrite", 2, 3, FEARG_2, ret_void, TERM_FUNC(f_term_dumpwrite)},
911 {"term_getaltscreen", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getaltscreen)},
912 {"term_getansicolors", 1, 1, FEARG_1, ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100913#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100914 f_term_getansicolors
915#else
916 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +0200917#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100918 },
919 {"term_getattr", 2, 2, FEARG_1, ret_number, TERM_FUNC(f_term_getattr)},
920 {"term_getcursor", 1, 1, FEARG_1, ret_list_any, TERM_FUNC(f_term_getcursor)},
921 {"term_getjob", 1, 1, FEARG_1, ret_job, TERM_FUNC(f_term_getjob)},
922 {"term_getline", 2, 2, FEARG_1, ret_string, TERM_FUNC(f_term_getline)},
923 {"term_getscrolled", 1, 1, FEARG_1, ret_number, TERM_FUNC(f_term_getscrolled)},
924 {"term_getsize", 1, 1, FEARG_1, ret_list_number, TERM_FUNC(f_term_getsize)},
925 {"term_getstatus", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_getstatus)},
926 {"term_gettitle", 1, 1, FEARG_1, ret_string, TERM_FUNC(f_term_gettitle)},
927 {"term_gettty", 1, 2, FEARG_1, ret_string, TERM_FUNC(f_term_gettty)},
928 {"term_list", 0, 0, 0, ret_list_number, TERM_FUNC(f_term_list)},
929 {"term_scrape", 2, 2, FEARG_1, ret_list_dict_any, TERM_FUNC(f_term_scrape)},
930 {"term_sendkeys", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_sendkeys)},
931 {"term_setansicolors", 2, 2, FEARG_1, ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +0100932#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +0100933 f_term_setansicolors
934#else
935 NULL
936#endif
937 },
938 {"term_setapi", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setapi)},
939 {"term_setkill", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setkill)},
940 {"term_setrestore", 2, 2, FEARG_1, ret_void, TERM_FUNC(f_term_setrestore)},
941 {"term_setsize", 3, 3, FEARG_1, ret_void, TERM_FUNC(f_term_setsize)},
942 {"term_start", 1, 2, FEARG_1, ret_number, TERM_FUNC(f_term_start)},
943 {"term_wait", 1, 2, FEARG_1, ret_void, TERM_FUNC(f_term_wait)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100944 {"test_alloc_fail", 3, 3, FEARG_1, ret_void, f_test_alloc_fail},
945 {"test_autochdir", 0, 0, 0, ret_void, f_test_autochdir},
946 {"test_feedinput", 1, 1, FEARG_1, ret_void, f_test_feedinput},
947 {"test_garbagecollect_now", 0, 0, 0, ret_void, f_test_garbagecollect_now},
948 {"test_garbagecollect_soon", 0, 0, 0, ret_void, f_test_garbagecollect_soon},
949 {"test_getvalue", 1, 1, FEARG_1, ret_number, f_test_getvalue},
950 {"test_ignore_error", 1, 1, FEARG_1, ret_void, f_test_ignore_error},
951 {"test_null_blob", 0, 0, 0, ret_blob, f_test_null_blob},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100952 {"test_null_channel", 0, 0, 0, ret_channel, JOB_FUNC(f_test_null_channel)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100953 {"test_null_dict", 0, 0, 0, ret_dict_any, f_test_null_dict},
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200954 {"test_null_function", 0, 0, 0, ret_func_any, f_test_null_function},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100955 {"test_null_job", 0, 0, 0, ret_job, JOB_FUNC(f_test_null_job)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100956 {"test_null_list", 0, 0, 0, ret_list_any, f_test_null_list},
Bram Moolenaard77a8522020-04-03 21:59:57 +0200957 {"test_null_partial", 0, 0, 0, ret_func_any, f_test_null_partial},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100958 {"test_null_string", 0, 0, 0, ret_string, f_test_null_string},
959 {"test_option_not_set", 1, 1, FEARG_1,ret_void, f_test_option_not_set},
960 {"test_override", 2, 2, FEARG_2, ret_void, f_test_override},
961 {"test_refcount", 1, 1, FEARG_1, ret_number, f_test_refcount},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100962 {"test_scrollbar", 3, 3, FEARG_2, ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +0200963#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +0100964 f_test_scrollbar
965#else
966 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +0200967#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100968 },
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100969 {"test_setmouse", 2, 2, 0, ret_void, f_test_setmouse},
970 {"test_settime", 1, 1, FEARG_1, ret_void, f_test_settime},
971 {"test_srand_seed", 0, 1, FEARG_1, ret_void, f_test_srand_seed},
972 {"test_unknown", 0, 0, 0, ret_any, f_test_unknown},
973 {"test_void", 0, 0, 0, ret_any, f_test_void},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100974 {"timer_info", 0, 1, FEARG_1, ret_list_dict_any, TIMER_FUNC(f_timer_info)},
975 {"timer_pause", 2, 2, FEARG_1, ret_void, TIMER_FUNC(f_timer_pause)},
976 {"timer_start", 2, 3, FEARG_1, ret_number, TIMER_FUNC(f_timer_start)},
977 {"timer_stop", 1, 1, FEARG_1, ret_void, TIMER_FUNC(f_timer_stop)},
978 {"timer_stopall", 0, 0, 0, ret_void, TIMER_FUNC(f_timer_stopall)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100979 {"tolower", 1, 1, FEARG_1, ret_string, f_tolower},
980 {"toupper", 1, 1, FEARG_1, ret_string, f_toupper},
981 {"tr", 3, 3, FEARG_1, ret_string, f_tr},
Bram Moolenaar2245ae12020-05-31 22:20:36 +0200982 {"trim", 1, 3, FEARG_1, ret_string, f_trim},
Bram Moolenaar15c47602020-03-26 22:16:48 +0100983 {"trunc", 1, 1, FEARG_1, ret_float, FLOAT_FUNC(f_trunc)},
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100984 {"type", 1, 1, FEARG_1, ret_number, f_type},
985 {"undofile", 1, 1, FEARG_1, ret_string, f_undofile},
986 {"undotree", 0, 0, 0, ret_dict_any, f_undotree},
987 {"uniq", 1, 3, FEARG_1, ret_list_any, f_uniq},
988 {"values", 1, 1, FEARG_1, ret_list_any, f_values},
989 {"virtcol", 1, 1, FEARG_1, ret_number, f_virtcol},
990 {"visualmode", 0, 1, 0, ret_string, f_visualmode},
991 {"wildmenumode", 0, 0, 0, ret_number, f_wildmenumode},
992 {"win_execute", 2, 3, FEARG_2, ret_string, f_win_execute},
993 {"win_findbuf", 1, 1, FEARG_1, ret_list_number, f_win_findbuf},
994 {"win_getid", 0, 2, FEARG_1, ret_number, f_win_getid},
995 {"win_gettype", 0, 1, FEARG_1, ret_string, f_win_gettype},
996 {"win_gotoid", 1, 1, FEARG_1, ret_number, f_win_gotoid},
997 {"win_id2tabwin", 1, 1, FEARG_1, ret_list_number, f_win_id2tabwin},
998 {"win_id2win", 1, 1, FEARG_1, ret_number, f_win_id2win},
999 {"win_screenpos", 1, 1, FEARG_1, ret_list_number, f_win_screenpos},
1000 {"win_splitmove", 2, 3, FEARG_1, ret_number, f_win_splitmove},
1001 {"winbufnr", 1, 1, FEARG_1, ret_number, f_winbufnr},
1002 {"wincol", 0, 0, 0, ret_number, f_wincol},
1003 {"windowsversion", 0, 0, 0, ret_string, f_windowsversion},
1004 {"winheight", 1, 1, FEARG_1, ret_number, f_winheight},
1005 {"winlayout", 0, 1, FEARG_1, ret_list_any, f_winlayout},
1006 {"winline", 0, 0, 0, ret_number, f_winline},
1007 {"winnr", 0, 1, FEARG_1, ret_number, f_winnr},
1008 {"winrestcmd", 0, 0, 0, ret_string, f_winrestcmd},
1009 {"winrestview", 1, 1, FEARG_1, ret_void, f_winrestview},
1010 {"winsaveview", 0, 0, 0, ret_dict_any, f_winsaveview},
1011 {"winwidth", 1, 1, FEARG_1, ret_number, f_winwidth},
1012 {"wordcount", 0, 0, 0, ret_dict_number, f_wordcount},
1013 {"writefile", 2, 3, FEARG_1, ret_number, f_writefile},
1014 {"xor", 2, 2, FEARG_1, ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001015};
1016
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001017/*
1018 * Function given to ExpandGeneric() to obtain the list of internal
1019 * or user defined function names.
1020 */
1021 char_u *
1022get_function_name(expand_T *xp, int idx)
1023{
1024 static int intidx = -1;
1025 char_u *name;
1026
1027 if (idx == 0)
1028 intidx = -1;
1029 if (intidx < 0)
1030 {
1031 name = get_user_func_name(xp, idx);
1032 if (name != NULL)
1033 return name;
1034 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001035 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001036 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001037 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001038 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001039 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001040 STRCAT(IObuff, ")");
1041 return IObuff;
1042 }
1043
1044 return NULL;
1045}
1046
1047/*
1048 * Function given to ExpandGeneric() to obtain the list of internal or
1049 * user defined variable or function names.
1050 */
1051 char_u *
1052get_expr_name(expand_T *xp, int idx)
1053{
1054 static int intidx = -1;
1055 char_u *name;
1056
1057 if (idx == 0)
1058 intidx = -1;
1059 if (intidx < 0)
1060 {
1061 name = get_function_name(xp, idx);
1062 if (name != NULL)
1063 return name;
1064 }
1065 return get_user_var_name(xp, ++intidx);
1066}
1067
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001068/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001069 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001070 * Return index, or -1 if not found or "implemented" is TRUE and the function
1071 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001072 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001073 static int
1074find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001075{
1076 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001077 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001078 int cmp;
1079 int x;
1080
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001081 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001082
1083 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001084 while (first <= last)
1085 {
1086 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001087 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001088 if (cmp < 0)
1089 last = x - 1;
1090 else if (cmp > 0)
1091 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001092 else if (implemented && global_functions[x].f_func == NULL)
1093 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001094 else
1095 return x;
1096 }
1097 return -1;
1098}
1099
Bram Moolenaar15c47602020-03-26 22:16:48 +01001100/*
1101 * Find internal function "name" in table "global_functions".
1102 * Return index, or -1 if not found or the function is not implemented.
1103 */
1104 int
1105find_internal_func(char_u *name)
1106{
1107 return find_internal_func_opt(name, TRUE);
1108}
1109
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001110 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001111has_internal_func(char_u *name)
1112{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001113 return find_internal_func_opt(name, TRUE) >= 0;
1114}
1115
1116 static int
1117has_internal_func_name(char_u *name)
1118{
1119 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001120}
1121
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001122 char *
1123internal_func_name(int idx)
1124{
1125 return global_functions[idx].f_name;
1126}
1127
1128 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001129internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001130{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001131 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001132}
1133
1134/*
1135 * Check the argument count to use for internal function "idx".
1136 * Returns OK or FAIL;
1137 */
1138 int
1139check_internal_func(int idx, int argcount)
1140{
1141 int res;
1142 char *name;
1143
1144 if (argcount < global_functions[idx].f_min_argc)
1145 res = FCERR_TOOFEW;
1146 else if (argcount > global_functions[idx].f_max_argc)
1147 res = FCERR_TOOMANY;
1148 else
1149 return OK;
1150
1151 name = internal_func_name(idx);
1152 if (res == FCERR_TOOMANY)
1153 semsg(_(e_toomanyarg), name);
1154 else
1155 semsg(_(e_toofewarg), name);
1156 return FAIL;
1157}
1158
Bram Moolenaarac92e252019-08-03 21:58:38 +02001159 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001160call_internal_func(
1161 char_u *name,
1162 int argcount,
1163 typval_T *argvars,
1164 typval_T *rettv)
1165{
1166 int i;
1167
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001168 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001169 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001170 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001171 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001172 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001173 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001174 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001175 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001176 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001177 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001178}
1179
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001180 void
1181call_internal_func_by_idx(
1182 int idx,
1183 typval_T *argvars,
1184 typval_T *rettv)
1185{
1186 global_functions[idx].f_func(argvars, rettv);
1187}
1188
Bram Moolenaarac92e252019-08-03 21:58:38 +02001189/*
1190 * Invoke a method for base->method().
1191 */
1192 int
1193call_internal_method(
1194 char_u *name,
1195 int argcount,
1196 typval_T *argvars,
1197 typval_T *rettv,
1198 typval_T *basetv)
1199{
1200 int i;
1201 int fi;
1202 typval_T argv[MAX_FUNC_ARGS + 1];
1203
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001204 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02001205 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001206 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02001207 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001208 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001209 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001210 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001211 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001212 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001213
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001214 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001215 {
1216 // base value goes last
1217 for (i = 0; i < argcount; ++i)
1218 argv[i] = argvars[i];
1219 argv[argcount] = *basetv;
1220 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001221 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001222 {
1223 // base value goes second
1224 argv[0] = argvars[0];
1225 argv[1] = *basetv;
1226 for (i = 1; i < argcount; ++i)
1227 argv[i + 1] = argvars[i];
1228 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02001229 else if (global_functions[fi].f_argtype == FEARG_3)
1230 {
1231 // base value goes third
1232 argv[0] = argvars[0];
1233 argv[1] = argvars[1];
1234 argv[2] = *basetv;
1235 for (i = 2; i < argcount; ++i)
1236 argv[i + 1] = argvars[i];
1237 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02001238 else if (global_functions[fi].f_argtype == FEARG_4)
1239 {
1240 // base value goes fourth
1241 argv[0] = argvars[0];
1242 argv[1] = argvars[1];
1243 argv[2] = argvars[2];
1244 argv[3] = *basetv;
1245 for (i = 3; i < argcount; ++i)
1246 argv[i + 1] = argvars[i];
1247 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02001248 else
1249 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001250 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02001251 argv[0] = *basetv;
1252 for (i = 0; i < argcount; ++i)
1253 argv[i + 1] = argvars[i];
1254 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001255 argv[argcount + 1].v_type = VAR_UNKNOWN;
1256
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001257 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001258 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001259}
1260
1261/*
1262 * Return TRUE for a non-zero Number and a non-empty String.
1263 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02001264 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001265non_zero_arg(typval_T *argvars)
1266{
1267 return ((argvars[0].v_type == VAR_NUMBER
1268 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001269 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001270 && argvars[0].vval.v_number == VVAL_TRUE)
1271 || (argvars[0].v_type == VAR_STRING
1272 && argvars[0].vval.v_string != NULL
1273 && *argvars[0].vval.v_string != NUL));
1274}
1275
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001276#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001277/*
1278 * Get the float value of "argvars[0]" into "f".
1279 * Returns FAIL when the argument is not a Number or Float.
1280 */
1281 static int
1282get_float_arg(typval_T *argvars, float_T *f)
1283{
1284 if (argvars[0].v_type == VAR_FLOAT)
1285 {
1286 *f = argvars[0].vval.v_float;
1287 return OK;
1288 }
1289 if (argvars[0].v_type == VAR_NUMBER)
1290 {
1291 *f = (float_T)argvars[0].vval.v_number;
1292 return OK;
1293 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001294 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001295 return FAIL;
1296}
1297
1298/*
1299 * "abs(expr)" function
1300 */
1301 static void
1302f_abs(typval_T *argvars, typval_T *rettv)
1303{
1304 if (argvars[0].v_type == VAR_FLOAT)
1305 {
1306 rettv->v_type = VAR_FLOAT;
1307 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
1308 }
1309 else
1310 {
1311 varnumber_T n;
1312 int error = FALSE;
1313
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001314 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001315 if (error)
1316 rettv->vval.v_number = -1;
1317 else if (n > 0)
1318 rettv->vval.v_number = n;
1319 else
1320 rettv->vval.v_number = -n;
1321 }
1322}
1323
1324/*
1325 * "acos()" function
1326 */
1327 static void
1328f_acos(typval_T *argvars, typval_T *rettv)
1329{
1330 float_T f = 0.0;
1331
1332 rettv->v_type = VAR_FLOAT;
1333 if (get_float_arg(argvars, &f) == OK)
1334 rettv->vval.v_float = acos(f);
1335 else
1336 rettv->vval.v_float = 0.0;
1337}
1338#endif
1339
1340/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001341 * "and(expr, expr)" function
1342 */
1343 static void
1344f_and(typval_T *argvars, typval_T *rettv)
1345{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001346 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
1347 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02001348}
1349
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001350#ifdef FEAT_FLOAT
1351/*
1352 * "asin()" function
1353 */
1354 static void
1355f_asin(typval_T *argvars, typval_T *rettv)
1356{
1357 float_T f = 0.0;
1358
1359 rettv->v_type = VAR_FLOAT;
1360 if (get_float_arg(argvars, &f) == OK)
1361 rettv->vval.v_float = asin(f);
1362 else
1363 rettv->vval.v_float = 0.0;
1364}
1365
1366/*
1367 * "atan()" function
1368 */
1369 static void
1370f_atan(typval_T *argvars, typval_T *rettv)
1371{
1372 float_T f = 0.0;
1373
1374 rettv->v_type = VAR_FLOAT;
1375 if (get_float_arg(argvars, &f) == OK)
1376 rettv->vval.v_float = atan(f);
1377 else
1378 rettv->vval.v_float = 0.0;
1379}
1380
1381/*
1382 * "atan2()" function
1383 */
1384 static void
1385f_atan2(typval_T *argvars, typval_T *rettv)
1386{
1387 float_T fx = 0.0, fy = 0.0;
1388
1389 rettv->v_type = VAR_FLOAT;
1390 if (get_float_arg(argvars, &fx) == OK
1391 && get_float_arg(&argvars[1], &fy) == OK)
1392 rettv->vval.v_float = atan2(fx, fy);
1393 else
1394 rettv->vval.v_float = 0.0;
1395}
1396#endif
1397
1398/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01001399 * "balloon_show()" function
1400 */
1401#ifdef FEAT_BEVAL
1402 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001403f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
1404{
1405 rettv->v_type = VAR_STRING;
1406 if (balloonEval != NULL)
1407 {
1408 if (balloonEval->msg == NULL)
1409 rettv->vval.v_string = NULL;
1410 else
1411 rettv->vval.v_string = vim_strsave(balloonEval->msg);
1412 }
1413}
1414
1415 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01001416f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
1417{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01001418 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001419 {
1420 if (argvars[0].v_type == VAR_LIST
1421# ifdef FEAT_GUI
1422 && !gui.in_use
1423# endif
1424 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001425 {
1426 list_T *l = argvars[0].vval.v_list;
1427
1428 // empty list removes the balloon
1429 post_balloon(balloonEval, NULL,
1430 l == NULL || l->lv_len == 0 ? NULL : l);
1431 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001432 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02001433 {
1434 char_u *mesg = tv_get_string_chk(&argvars[0]);
1435
1436 if (mesg != NULL)
1437 // empty string removes the balloon
1438 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
1439 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01001440 }
1441}
1442
Bram Moolenaar669a8282017-11-19 20:13:05 +01001443# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01001444 static void
1445f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
1446{
1447 if (rettv_list_alloc(rettv) == OK)
1448 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001449 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001450
1451 if (msg != NULL)
1452 {
1453 pumitem_T *array;
1454 int size = split_message(msg, &array);
1455 int i;
1456
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001457 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01001458 for (i = 1; i < size - 1; ++i)
1459 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01001460 while (size > 0)
1461 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01001462 vim_free(array);
1463 }
1464 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01001465}
Bram Moolenaar669a8282017-11-19 20:13:05 +01001466# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01001467#endif
1468
1469/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001470 * Get the buffer from "arg" and give an error and return NULL if it is not
1471 * valid.
1472 */
Bram Moolenaara3347722019-05-11 21:14:24 +02001473 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001474get_buf_arg(typval_T *arg)
1475{
1476 buf_T *buf;
1477
1478 ++emsg_off;
1479 buf = tv_get_buf(arg, FALSE);
1480 --emsg_off;
1481 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001482 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01001483 return buf;
1484}
1485
1486/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001487 * "byte2line(byte)" function
1488 */
1489 static void
1490f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
1491{
1492#ifndef FEAT_BYTEOFF
1493 rettv->vval.v_number = -1;
1494#else
1495 long boff = 0;
1496
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001497 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001498 if (boff < 0)
1499 rettv->vval.v_number = -1;
1500 else
1501 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
1502 (linenr_T)0, &boff);
1503#endif
1504}
1505
1506 static void
1507byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
1508{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001509 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001510 char_u *str;
1511 varnumber_T idx;
1512
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001513 str = tv_get_string_chk(&argvars[0]);
1514 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001515 rettv->vval.v_number = -1;
1516 if (str == NULL || idx < 0)
1517 return;
1518
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001519 t = str;
1520 for ( ; idx > 0; idx--)
1521 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001522 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001523 return;
1524 if (enc_utf8 && comp)
1525 t += utf_ptr2len(t);
1526 else
1527 t += (*mb_ptr2len)(t);
1528 }
1529 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001530}
1531
1532/*
1533 * "byteidx()" function
1534 */
1535 static void
1536f_byteidx(typval_T *argvars, typval_T *rettv)
1537{
1538 byteidx(argvars, rettv, FALSE);
1539}
1540
1541/*
1542 * "byteidxcomp()" function
1543 */
1544 static void
1545f_byteidxcomp(typval_T *argvars, typval_T *rettv)
1546{
1547 byteidx(argvars, rettv, TRUE);
1548}
1549
1550/*
1551 * "call(func, arglist [, dict])" function
1552 */
1553 static void
1554f_call(typval_T *argvars, typval_T *rettv)
1555{
1556 char_u *func;
1557 partial_T *partial = NULL;
1558 dict_T *selfdict = NULL;
1559
1560 if (argvars[1].v_type != VAR_LIST)
1561 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001562 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001563 return;
1564 }
1565 if (argvars[1].vval.v_list == NULL)
1566 return;
1567
1568 if (argvars[0].v_type == VAR_FUNC)
1569 func = argvars[0].vval.v_string;
1570 else if (argvars[0].v_type == VAR_PARTIAL)
1571 {
1572 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001573 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001574 }
1575 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001576 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001577 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001578 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001579
1580 if (argvars[2].v_type != VAR_UNKNOWN)
1581 {
1582 if (argvars[2].v_type != VAR_DICT)
1583 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001584 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001585 return;
1586 }
1587 selfdict = argvars[2].vval.v_dict;
1588 }
1589
1590 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
1591}
1592
1593#ifdef FEAT_FLOAT
1594/*
1595 * "ceil({float})" function
1596 */
1597 static void
1598f_ceil(typval_T *argvars, typval_T *rettv)
1599{
1600 float_T f = 0.0;
1601
1602 rettv->v_type = VAR_FLOAT;
1603 if (get_float_arg(argvars, &f) == OK)
1604 rettv->vval.v_float = ceil(f);
1605 else
1606 rettv->vval.v_float = 0.0;
1607}
1608#endif
1609
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001610/*
1611 * "changenr()" function
1612 */
1613 static void
1614f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
1615{
1616 rettv->vval.v_number = curbuf->b_u_seq_cur;
1617}
1618
1619/*
1620 * "char2nr(string)" function
1621 */
1622 static void
1623f_char2nr(typval_T *argvars, typval_T *rettv)
1624{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001625 if (has_mbyte)
1626 {
1627 int utf8 = 0;
1628
1629 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001630 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001631
1632 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01001633 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001634 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001635 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001636 }
1637 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001638 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001639}
1640
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02001641 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01001642get_optional_window(typval_T *argvars, int idx)
1643{
1644 win_T *win = curwin;
1645
1646 if (argvars[idx].v_type != VAR_UNKNOWN)
1647 {
1648 win = find_win_by_nr_or_id(&argvars[idx]);
1649 if (win == NULL)
1650 {
1651 emsg(_(e_invalwindow));
1652 return NULL;
1653 }
1654 }
1655 return win;
1656}
1657
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001658/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001659 * "col(string)" function
1660 */
1661 static void
1662f_col(typval_T *argvars, typval_T *rettv)
1663{
1664 colnr_T col = 0;
1665 pos_T *fp;
1666 int fnum = curbuf->b_fnum;
1667
1668 fp = var2fpos(&argvars[0], FALSE, &fnum);
1669 if (fp != NULL && fnum == curbuf->b_fnum)
1670 {
1671 if (fp->col == MAXCOL)
1672 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001673 // '> can be MAXCOL, get the length of the line then
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001674 if (fp->lnum <= curbuf->b_ml.ml_line_count)
1675 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
1676 else
1677 col = MAXCOL;
1678 }
1679 else
1680 {
1681 col = fp->col + 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001682 // col(".") when the cursor is on the NUL at the end of the line
1683 // because of "coladd" can be seen as an extra column.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001684 if (virtual_active() && fp == &curwin->w_cursor)
1685 {
1686 char_u *p = ml_get_cursor();
1687
1688 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
1689 curwin->w_virtcol - curwin->w_cursor.coladd))
1690 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001691 int l;
1692
1693 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
1694 col += l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001695 }
1696 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001697 }
1698 }
1699 rettv->vval.v_number = col;
1700}
1701
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001702/*
1703 * "confirm(message, buttons[, default [, type]])" function
1704 */
1705 static void
1706f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1707{
1708#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1709 char_u *message;
1710 char_u *buttons = NULL;
1711 char_u buf[NUMBUFLEN];
1712 char_u buf2[NUMBUFLEN];
1713 int def = 1;
1714 int type = VIM_GENERIC;
1715 char_u *typestr;
1716 int error = FALSE;
1717
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001718 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001719 if (message == NULL)
1720 error = TRUE;
1721 if (argvars[1].v_type != VAR_UNKNOWN)
1722 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001723 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001724 if (buttons == NULL)
1725 error = TRUE;
1726 if (argvars[2].v_type != VAR_UNKNOWN)
1727 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001728 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001729 if (argvars[3].v_type != VAR_UNKNOWN)
1730 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001731 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001732 if (typestr == NULL)
1733 error = TRUE;
1734 else
1735 {
1736 switch (TOUPPER_ASC(*typestr))
1737 {
1738 case 'E': type = VIM_ERROR; break;
1739 case 'Q': type = VIM_QUESTION; break;
1740 case 'I': type = VIM_INFO; break;
1741 case 'W': type = VIM_WARNING; break;
1742 case 'G': type = VIM_GENERIC; break;
1743 }
1744 }
1745 }
1746 }
1747 }
1748
1749 if (buttons == NULL || *buttons == NUL)
1750 buttons = (char_u *)_("&Ok");
1751
1752 if (!error)
1753 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
1754 def, NULL, FALSE);
1755#endif
1756}
1757
1758/*
1759 * "copy()" function
1760 */
1761 static void
1762f_copy(typval_T *argvars, typval_T *rettv)
1763{
1764 item_copy(&argvars[0], rettv, FALSE, 0);
1765}
1766
1767#ifdef FEAT_FLOAT
1768/*
1769 * "cos()" function
1770 */
1771 static void
1772f_cos(typval_T *argvars, typval_T *rettv)
1773{
1774 float_T f = 0.0;
1775
1776 rettv->v_type = VAR_FLOAT;
1777 if (get_float_arg(argvars, &f) == OK)
1778 rettv->vval.v_float = cos(f);
1779 else
1780 rettv->vval.v_float = 0.0;
1781}
1782
1783/*
1784 * "cosh()" function
1785 */
1786 static void
1787f_cosh(typval_T *argvars, typval_T *rettv)
1788{
1789 float_T f = 0.0;
1790
1791 rettv->v_type = VAR_FLOAT;
1792 if (get_float_arg(argvars, &f) == OK)
1793 rettv->vval.v_float = cosh(f);
1794 else
1795 rettv->vval.v_float = 0.0;
1796}
1797#endif
1798
1799/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001800 * "cursor(lnum, col)" function, or
1801 * "cursor(list)"
1802 *
1803 * Moves the cursor to the specified line and column.
1804 * Returns 0 when the position could be set, -1 otherwise.
1805 */
1806 static void
1807f_cursor(typval_T *argvars, typval_T *rettv)
1808{
1809 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001810 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001811 int set_curswant = TRUE;
1812
1813 rettv->vval.v_number = -1;
1814 if (argvars[1].v_type == VAR_UNKNOWN)
1815 {
1816 pos_T pos;
1817 colnr_T curswant = -1;
1818
1819 if (list2fpos(argvars, &pos, NULL, &curswant) == FAIL)
1820 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001821 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001822 return;
1823 }
1824 line = pos.lnum;
1825 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001826 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001827 if (curswant >= 0)
1828 {
1829 curwin->w_curswant = curswant - 1;
1830 set_curswant = FALSE;
1831 }
1832 }
1833 else
1834 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001835 line = tv_get_lnum(argvars);
1836 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001837 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001838 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001839 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01001840 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001841 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001842 if (line > 0)
1843 curwin->w_cursor.lnum = line;
1844 if (col > 0)
1845 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001846 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001847
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001848 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001849 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001850 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001851 if (has_mbyte)
1852 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001853
1854 curwin->w_set_curswant = set_curswant;
1855 rettv->vval.v_number = 0;
1856}
1857
Bram Moolenaar4f974752019-02-17 17:44:42 +01001858#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001859/*
1860 * "debugbreak()" function
1861 */
1862 static void
1863f_debugbreak(typval_T *argvars, typval_T *rettv)
1864{
1865 int pid;
1866
1867 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001868 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001869 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001870 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02001871 else
1872 {
1873 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
1874
1875 if (hProcess != NULL)
1876 {
1877 DebugBreakProcess(hProcess);
1878 CloseHandle(hProcess);
1879 rettv->vval.v_number = OK;
1880 }
1881 }
1882}
1883#endif
1884
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001885/*
1886 * "deepcopy()" function
1887 */
1888 static void
1889f_deepcopy(typval_T *argvars, typval_T *rettv)
1890{
1891 int noref = 0;
1892 int copyID;
1893
1894 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001895 noref = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001896 if (noref < 0 || noref > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001897 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001898 else
1899 {
1900 copyID = get_copyID();
1901 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
1902 }
1903}
1904
1905/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001906 * "did_filetype()" function
1907 */
1908 static void
1909f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1910{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001911 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001912}
1913
1914/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01001915 * "echoraw({expr})" function
1916 */
1917 static void
1918f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
1919{
1920 char_u *str = tv_get_string_chk(&argvars[0]);
1921
1922 if (str != NULL && *str != NUL)
1923 {
1924 out_str(str);
1925 out_flush();
1926 }
1927}
1928
1929/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001930 * "empty({expr})" function
1931 */
1932 static void
1933f_empty(typval_T *argvars, typval_T *rettv)
1934{
1935 int n = FALSE;
1936
1937 switch (argvars[0].v_type)
1938 {
1939 case VAR_STRING:
1940 case VAR_FUNC:
1941 n = argvars[0].vval.v_string == NULL
1942 || *argvars[0].vval.v_string == NUL;
1943 break;
1944 case VAR_PARTIAL:
1945 n = FALSE;
1946 break;
1947 case VAR_NUMBER:
1948 n = argvars[0].vval.v_number == 0;
1949 break;
1950 case VAR_FLOAT:
1951#ifdef FEAT_FLOAT
1952 n = argvars[0].vval.v_float == 0.0;
1953 break;
1954#endif
1955 case VAR_LIST:
1956 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01001957 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001958 break;
1959 case VAR_DICT:
1960 n = argvars[0].vval.v_dict == NULL
1961 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
1962 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001963 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001964 case VAR_SPECIAL:
1965 n = argvars[0].vval.v_number != VVAL_TRUE;
1966 break;
1967
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01001968 case VAR_BLOB:
1969 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01001970 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
1971 break;
1972
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001973 case VAR_JOB:
1974#ifdef FEAT_JOB_CHANNEL
1975 n = argvars[0].vval.v_job == NULL
1976 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
1977 break;
1978#endif
1979 case VAR_CHANNEL:
1980#ifdef FEAT_JOB_CHANNEL
1981 n = argvars[0].vval.v_channel == NULL
1982 || !channel_is_open(argvars[0].vval.v_channel);
1983 break;
1984#endif
1985 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02001986 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001987 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01001988 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001989 n = TRUE;
1990 break;
1991 }
1992
1993 rettv->vval.v_number = n;
1994}
1995
1996/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02001997 * "environ()" function
1998 */
1999 static void
2000f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2001{
2002#if !defined(AMIGA)
2003 int i = 0;
2004 char_u *entry, *value;
2005# ifdef MSWIN
2006 extern wchar_t **_wenviron;
2007# else
2008 extern char **environ;
2009# endif
2010
2011 if (rettv_dict_alloc(rettv) != OK)
2012 return;
2013
2014# ifdef MSWIN
2015 if (*_wenviron == NULL)
2016 return;
2017# else
2018 if (*environ == NULL)
2019 return;
2020# endif
2021
2022 for (i = 0; ; ++i)
2023 {
2024# ifdef MSWIN
2025 short_u *p;
2026
2027 if ((p = (short_u *)_wenviron[i]) == NULL)
2028 return;
2029 entry = utf16_to_enc(p, NULL);
2030# else
2031 if ((entry = (char_u *)environ[i]) == NULL)
2032 return;
2033 entry = vim_strsave(entry);
2034# endif
2035 if (entry == NULL) // out of memory
2036 return;
2037 if ((value = vim_strchr(entry, '=')) == NULL)
2038 {
2039 vim_free(entry);
2040 continue;
2041 }
2042 *value++ = NUL;
2043 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2044 vim_free(entry);
2045 }
2046#endif
2047}
2048
2049/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002050 * "escape({string}, {chars})" function
2051 */
2052 static void
2053f_escape(typval_T *argvars, typval_T *rettv)
2054{
2055 char_u buf[NUMBUFLEN];
2056
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002057 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2058 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002059 rettv->v_type = VAR_STRING;
2060}
2061
2062/*
2063 * "eval()" function
2064 */
2065 static void
2066f_eval(typval_T *argvars, typval_T *rettv)
2067{
2068 char_u *s, *p;
2069
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002070 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002071 if (s != NULL)
2072 s = skipwhite(s);
2073
2074 p = s;
Bram Moolenaar32e35112020-05-14 22:41:15 +02002075 if (s == NULL || eval1(&s, rettv, EVAL_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002076 {
2077 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002078 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002079 need_clr_eos = FALSE;
2080 rettv->v_type = VAR_NUMBER;
2081 rettv->vval.v_number = 0;
2082 }
2083 else if (*s != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002084 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002085}
2086
2087/*
2088 * "eventhandler()" function
2089 */
2090 static void
2091f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2092{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002093 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002094}
2095
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002096static garray_T redir_execute_ga;
2097
2098/*
2099 * Append "value[value_len]" to the execute() output.
2100 */
2101 void
2102execute_redir_str(char_u *value, int value_len)
2103{
2104 int len;
2105
2106 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002107 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002108 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002109 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002110 if (ga_grow(&redir_execute_ga, len) == OK)
2111 {
2112 mch_memmove((char *)redir_execute_ga.ga_data
2113 + redir_execute_ga.ga_len, value, len);
2114 redir_execute_ga.ga_len += len;
2115 }
2116}
2117
2118/*
2119 * Get next line from a list.
2120 * Called by do_cmdline() to get the next line.
2121 * Returns allocated string, or NULL for end of function.
2122 */
2123
2124 static char_u *
2125get_list_line(
2126 int c UNUSED,
2127 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002128 int indent UNUSED,
2129 int do_concat UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002130{
2131 listitem_T **p = (listitem_T **)cookie;
2132 listitem_T *item = *p;
2133 char_u buf[NUMBUFLEN];
2134 char_u *s;
2135
2136 if (item == NULL)
2137 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002138 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002139 *p = item->li_next;
2140 return s == NULL ? NULL : vim_strsave(s);
2141}
2142
2143/*
2144 * "execute()" function
2145 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002146 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002147execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002148{
2149 char_u *cmd = NULL;
2150 list_T *list = NULL;
2151 int save_msg_silent = msg_silent;
2152 int save_emsg_silent = emsg_silent;
2153 int save_emsg_noredir = emsg_noredir;
2154 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002155 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002156 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002157 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002158 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002159
2160 rettv->vval.v_string = NULL;
2161 rettv->v_type = VAR_STRING;
2162
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002163 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002164 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002165 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002166 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002167 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002168 return;
2169 ++list->lv_refcount;
2170 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01002171 else if (argvars[arg_off].v_type == VAR_JOB
2172 || argvars[arg_off].v_type == VAR_CHANNEL)
2173 {
2174 emsg(_(e_inval_string));
2175 return;
2176 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002177 else
2178 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002179 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002180 if (cmd == NULL)
2181 return;
2182 }
2183
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002184 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002185 {
2186 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002187 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002188
2189 if (s == NULL)
2190 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002191 if (*s == NUL)
2192 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002193 if (STRNCMP(s, "silent", 6) == 0)
2194 ++msg_silent;
2195 if (STRCMP(s, "silent!") == 0)
2196 {
2197 emsg_silent = TRUE;
2198 emsg_noredir = TRUE;
2199 }
2200 }
2201 else
2202 ++msg_silent;
2203
2204 if (redir_execute)
2205 save_ga = redir_execute_ga;
2206 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
2207 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01002208 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002209 if (!echo_output)
2210 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002211
2212 if (cmd != NULL)
2213 do_cmdline_cmd(cmd);
2214 else
2215 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002216 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002217
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002218 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002219 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002220 do_cmdline(NULL, get_list_line, (void *)&item,
2221 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
2222 --list->lv_refcount;
2223 }
2224
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002225 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01002226 if (ga_grow(&redir_execute_ga, 1) == OK)
2227 {
2228 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
2229 rettv->vval.v_string = redir_execute_ga.ga_data;
2230 }
2231 else
2232 {
2233 ga_clear(&redir_execute_ga);
2234 rettv->vval.v_string = NULL;
2235 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002236 msg_silent = save_msg_silent;
2237 emsg_silent = save_emsg_silent;
2238 emsg_noredir = save_emsg_noredir;
2239
2240 redir_execute = save_redir_execute;
2241 if (redir_execute)
2242 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01002243 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002244
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002245 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002246 if (echo_output)
2247 // When not working silently: put it in column zero. A following
2248 // "echon" will overwrite the message, unavoidably.
2249 msg_col = 0;
2250 else
2251 // When working silently: Put it back where it was, since nothing
2252 // should have been written.
2253 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002254}
2255
2256/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002257 * "execute()" function
2258 */
2259 static void
2260f_execute(typval_T *argvars, typval_T *rettv)
2261{
2262 execute_common(argvars, rettv, 0);
2263}
2264
2265/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002266 * "exists()" function
2267 */
2268 static void
2269f_exists(typval_T *argvars, typval_T *rettv)
2270{
2271 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002272 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002273
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002274 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002275 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002276 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002277 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002278 if (mch_getenv(p + 1) != NULL)
2279 n = TRUE;
2280 else
2281 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002282 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002283 p = expand_env_save(p);
2284 if (p != NULL && *p != '$')
2285 n = TRUE;
2286 vim_free(p);
2287 }
2288 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002289 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002290 {
2291 n = (get_option_tv(&p, NULL, TRUE) == OK);
2292 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002293 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002294 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002295 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002296 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02002297 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002298 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01002299 else if (*p == '?') // internal function only
2300 {
2301 n = has_internal_func_name(p + 1);
2302 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002303 else if (*p == ':')
2304 {
2305 n = cmd_exists(p + 1);
2306 }
2307 else if (*p == '#')
2308 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002309 if (p[1] == '#')
2310 n = autocmd_supported(p + 2);
2311 else
2312 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002313 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002314 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002315 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01002316 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002317 }
2318
2319 rettv->vval.v_number = n;
2320}
2321
2322#ifdef FEAT_FLOAT
2323/*
2324 * "exp()" function
2325 */
2326 static void
2327f_exp(typval_T *argvars, typval_T *rettv)
2328{
2329 float_T f = 0.0;
2330
2331 rettv->v_type = VAR_FLOAT;
2332 if (get_float_arg(argvars, &f) == OK)
2333 rettv->vval.v_float = exp(f);
2334 else
2335 rettv->vval.v_float = 0.0;
2336}
2337#endif
2338
2339/*
2340 * "expand()" function
2341 */
2342 static void
2343f_expand(typval_T *argvars, typval_T *rettv)
2344{
2345 char_u *s;
2346 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002347 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002348 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
2349 expand_T xpc;
2350 int error = FALSE;
2351 char_u *result;
2352
2353 rettv->v_type = VAR_STRING;
2354 if (argvars[1].v_type != VAR_UNKNOWN
2355 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002356 && tv_get_number_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002357 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02002358 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002359
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002360 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002361 if (*s == '%' || *s == '#' || *s == '<')
2362 {
2363 ++emsg_off;
2364 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
2365 --emsg_off;
2366 if (rettv->v_type == VAR_LIST)
2367 {
2368 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
2369 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02002370 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002371 }
2372 else
2373 rettv->vval.v_string = result;
2374 }
2375 else
2376 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002377 // When the optional second argument is non-zero, don't remove matches
2378 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002379 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002380 && tv_get_number_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002381 options |= WILD_KEEP_ALL;
2382 if (!error)
2383 {
2384 ExpandInit(&xpc);
2385 xpc.xp_context = EXPAND_FILES;
2386 if (p_wic)
2387 options += WILD_ICASE;
2388 if (rettv->v_type == VAR_STRING)
2389 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
2390 options, WILD_ALL);
2391 else if (rettv_list_alloc(rettv) != FAIL)
2392 {
2393 int i;
2394
2395 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
2396 for (i = 0; i < xpc.xp_numfiles; i++)
2397 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
2398 ExpandCleanup(&xpc);
2399 }
2400 }
2401 else
2402 rettv->vval.v_string = NULL;
2403 }
2404}
2405
2406/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02002407 * "expandcmd()" function
2408 * Expand all the special characters in a command string.
2409 */
2410 static void
2411f_expandcmd(typval_T *argvars, typval_T *rettv)
2412{
2413 exarg_T eap;
2414 char_u *cmdstr;
2415 char *errormsg = NULL;
2416
2417 rettv->v_type = VAR_STRING;
2418 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
2419
2420 memset(&eap, 0, sizeof(eap));
2421 eap.cmd = cmdstr;
2422 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02002423 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02002424 eap.usefilter = FALSE;
2425 eap.nextcmd = NULL;
2426 eap.cmdidx = CMD_USER;
2427
2428 expand_filename(&eap, &cmdstr, &errormsg);
2429 if (errormsg != NULL && *errormsg != NUL)
2430 emsg(errormsg);
2431
2432 rettv->vval.v_string = cmdstr;
2433}
2434
2435/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002436 * "feedkeys()" function
2437 */
2438 static void
2439f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
2440{
2441 int remap = TRUE;
2442 int insert = FALSE;
2443 char_u *keys, *flags;
2444 char_u nbuf[NUMBUFLEN];
2445 int typed = FALSE;
2446 int execute = FALSE;
2447 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002448 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002449 char_u *keys_esc;
2450
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002451 // This is not allowed in the sandbox. If the commands would still be
2452 // executed in the sandbox it would be OK, but it probably happens later,
2453 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002454 if (check_secure())
2455 return;
2456
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002457 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002458
2459 if (argvars[1].v_type != VAR_UNKNOWN)
2460 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002461 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002462 for ( ; *flags != NUL; ++flags)
2463 {
2464 switch (*flags)
2465 {
2466 case 'n': remap = FALSE; break;
2467 case 'm': remap = TRUE; break;
2468 case 't': typed = TRUE; break;
2469 case 'i': insert = TRUE; break;
2470 case 'x': execute = TRUE; break;
2471 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002472 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002473 }
2474 }
2475 }
2476
2477 if (*keys != NUL || execute)
2478 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002479 // Need to escape K_SPECIAL and CSI before putting the string in the
2480 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002481 keys_esc = vim_strsave_escape_csi(keys);
2482 if (keys_esc != NULL)
2483 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002484 if (lowlevel)
2485 {
2486#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01002487 int idx;
2488 int len = (int)STRLEN(keys);
2489
2490 for (idx = 0; idx < len; ++idx)
2491 {
2492 // if a CTRL-C was typed, set got_int, similar to what
2493 // happens in fill_input_buf()
2494 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
2495 got_int = TRUE;
2496 add_to_input_buf(keys + idx, 1);
2497 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002498#else
2499 emsg(_("E980: lowlevel input not supported"));
2500#endif
2501 }
2502 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002503 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01002504 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002505 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002506 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002507#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002508 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02002509#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002510 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01002511 typebuf_was_filled = TRUE;
2512 }
2513 vim_free(keys_esc);
2514
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002515 if (execute)
2516 {
2517 int save_msg_scroll = msg_scroll;
2518
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002519 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002520 msg_scroll = FALSE;
2521
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002522 if (!dangerous)
2523 ++ex_normal_busy;
Bram Moolenaar905dd902019-04-07 14:21:47 +02002524 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02002525 if (!dangerous)
2526 --ex_normal_busy;
2527
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002528 msg_scroll |= save_msg_scroll;
2529 }
2530 }
2531 }
2532}
2533
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002534#ifdef FEAT_FLOAT
2535/*
2536 * "float2nr({float})" function
2537 */
2538 static void
2539f_float2nr(typval_T *argvars, typval_T *rettv)
2540{
2541 float_T f = 0.0;
2542
2543 if (get_float_arg(argvars, &f) == OK)
2544 {
Bram Moolenaar37184272020-05-23 19:30:05 +02002545 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002546 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02002547 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01002548 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002549 else
2550 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002551 }
2552}
2553
2554/*
2555 * "floor({float})" function
2556 */
2557 static void
2558f_floor(typval_T *argvars, typval_T *rettv)
2559{
2560 float_T f = 0.0;
2561
2562 rettv->v_type = VAR_FLOAT;
2563 if (get_float_arg(argvars, &f) == OK)
2564 rettv->vval.v_float = floor(f);
2565 else
2566 rettv->vval.v_float = 0.0;
2567}
2568
2569/*
2570 * "fmod()" function
2571 */
2572 static void
2573f_fmod(typval_T *argvars, typval_T *rettv)
2574{
2575 float_T fx = 0.0, fy = 0.0;
2576
2577 rettv->v_type = VAR_FLOAT;
2578 if (get_float_arg(argvars, &fx) == OK
2579 && get_float_arg(&argvars[1], &fy) == OK)
2580 rettv->vval.v_float = fmod(fx, fy);
2581 else
2582 rettv->vval.v_float = 0.0;
2583}
2584#endif
2585
2586/*
2587 * "fnameescape({string})" function
2588 */
2589 static void
2590f_fnameescape(typval_T *argvars, typval_T *rettv)
2591{
2592 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002593 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002594 rettv->v_type = VAR_STRING;
2595}
2596
2597/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002598 * "foreground()" function
2599 */
2600 static void
2601f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2602{
2603#ifdef FEAT_GUI
2604 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002605 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002606 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02002607 return;
2608 }
2609#endif
2610#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002611 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002612#endif
2613}
2614
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002615 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002616common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002617{
2618 char_u *s;
2619 char_u *name;
2620 int use_string = FALSE;
2621 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002622 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002623 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002624
2625 if (argvars[0].v_type == VAR_FUNC)
2626 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002627 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002628 s = argvars[0].vval.v_string;
2629 }
2630 else if (argvars[0].v_type == VAR_PARTIAL
2631 && argvars[0].vval.v_partial != NULL)
2632 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002633 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002634 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002635 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002636 }
2637 else
2638 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002639 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002640 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002641 use_string = TRUE;
2642 }
2643
Bram Moolenaar843b8842016-08-21 14:36:15 +02002644 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002645 {
2646 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002647 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002648 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
2649 if (*name != NUL)
2650 s = NULL;
2651 }
2652
Bram Moolenaar843b8842016-08-21 14:36:15 +02002653 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
2654 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002655 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002656 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002657 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002658 ? find_func(trans_name, is_global, NULL) == NULL
2659 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002660 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002661 else
2662 {
2663 int dict_idx = 0;
2664 int arg_idx = 0;
2665 list_T *list = NULL;
2666
2667 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
2668 {
2669 char sid_buf[25];
2670 int off = *s == 's' ? 2 : 5;
2671
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002672 // Expand s: and <SID> into <SNR>nr_, so that the function can
2673 // also be called from another script. Using trans_function_name()
2674 // would also work, but some plugins depend on the name being
2675 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02002676 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02002677 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002678 if (name != NULL)
2679 {
2680 STRCPY(name, sid_buf);
2681 STRCAT(name, s + off);
2682 }
2683 }
2684 else
2685 name = vim_strsave(s);
2686
2687 if (argvars[1].v_type != VAR_UNKNOWN)
2688 {
2689 if (argvars[2].v_type != VAR_UNKNOWN)
2690 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002691 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002692 arg_idx = 1;
2693 dict_idx = 2;
2694 }
2695 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002696 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002697 dict_idx = 1;
2698 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002699 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002700 arg_idx = 1;
2701 if (dict_idx > 0)
2702 {
2703 if (argvars[dict_idx].v_type != VAR_DICT)
2704 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002705 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002706 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002707 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002708 }
2709 if (argvars[dict_idx].vval.v_dict == NULL)
2710 dict_idx = 0;
2711 }
2712 if (arg_idx > 0)
2713 {
2714 if (argvars[arg_idx].v_type != VAR_LIST)
2715 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002716 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002717 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002718 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002719 }
2720 list = argvars[arg_idx].vval.v_list;
2721 if (list == NULL || list->lv_len == 0)
2722 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002723 else if (list->lv_len > MAX_FUNC_ARGS)
2724 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01002725 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01002726 vim_free(name);
2727 goto theend;
2728 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002729 }
2730 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002731 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002732 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002733 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002734
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002735 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002736 if (pt == NULL)
2737 vim_free(name);
2738 else
2739 {
2740 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
2741 {
2742 listitem_T *li;
2743 int i = 0;
2744 int arg_len = 0;
2745 int lv_len = 0;
2746
2747 if (arg_pt != NULL)
2748 arg_len = arg_pt->pt_argc;
2749 if (list != NULL)
2750 lv_len = list->lv_len;
2751 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002752 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002753 if (pt->pt_argv == NULL)
2754 {
2755 vim_free(pt);
2756 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002757 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002758 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002759 for (i = 0; i < arg_len; i++)
2760 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
2761 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002762 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002763 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002764 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002765 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002766 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002767 }
2768
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002769 // For "function(dict.func, [], dict)" and "func" is a partial
2770 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002771 if (dict_idx > 0)
2772 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002773 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002774 pt->pt_dict = argvars[dict_idx].vval.v_dict;
2775 ++pt->pt_dict->dv_refcount;
2776 }
2777 else if (arg_pt != NULL)
2778 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002779 // If the dict was bound automatically the result is also
2780 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002781 pt->pt_dict = arg_pt->pt_dict;
2782 pt->pt_auto = arg_pt->pt_auto;
2783 if (pt->pt_dict != NULL)
2784 ++pt->pt_dict->dv_refcount;
2785 }
2786
2787 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002788 if (arg_pt != NULL && arg_pt->pt_func != NULL)
2789 {
2790 pt->pt_func = arg_pt->pt_func;
2791 func_ptr_ref(pt->pt_func);
2792 vim_free(name);
2793 }
2794 else if (is_funcref)
2795 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002796 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002797 func_ptr_ref(pt->pt_func);
2798 vim_free(name);
2799 }
2800 else
2801 {
2802 pt->pt_name = name;
2803 func_ref(name);
2804 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002805 }
2806 rettv->v_type = VAR_PARTIAL;
2807 rettv->vval.v_partial = pt;
2808 }
2809 else
2810 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002811 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002812 rettv->v_type = VAR_FUNC;
2813 rettv->vval.v_string = name;
2814 func_ref(name);
2815 }
2816 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002817theend:
2818 vim_free(trans_name);
2819}
2820
2821/*
2822 * "funcref()" function
2823 */
2824 static void
2825f_funcref(typval_T *argvars, typval_T *rettv)
2826{
2827 common_function(argvars, rettv, TRUE);
2828}
2829
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002830 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002831ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002832{
2833 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
2834 return &t_func_any;
Bram Moolenaard77a8522020-04-03 21:59:57 +02002835 return &t_func_void;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01002836}
2837
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002838/*
2839 * "function()" function
2840 */
2841 static void
2842f_function(typval_T *argvars, typval_T *rettv)
2843{
2844 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002845}
2846
2847/*
2848 * "garbagecollect()" function
2849 */
2850 static void
2851f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
2852{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002853 // This is postponed until we are back at the toplevel, because we may be
2854 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002855 want_garbage_collect = TRUE;
2856
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002857 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_number(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002858 garbage_collect_at_exit = TRUE;
2859}
2860
2861/*
2862 * "get()" function
2863 */
2864 static void
2865f_get(typval_T *argvars, typval_T *rettv)
2866{
2867 listitem_T *li;
2868 list_T *l;
2869 dictitem_T *di;
2870 dict_T *d;
2871 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002872 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002873
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002874 if (argvars[0].v_type == VAR_BLOB)
2875 {
2876 int error = FALSE;
2877 int idx = tv_get_number_chk(&argvars[1], &error);
2878
2879 if (!error)
2880 {
2881 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002882 if (idx < 0)
2883 idx = blob_len(argvars[0].vval.v_blob) + idx;
2884 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
2885 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002886 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002887 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002888 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01002889 tv = rettv;
2890 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002891 }
2892 }
2893 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002894 {
2895 if ((l = argvars[0].vval.v_list) != NULL)
2896 {
2897 int error = FALSE;
2898
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002899 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002900 if (!error && li != NULL)
2901 tv = &li->li_tv;
2902 }
2903 }
2904 else if (argvars[0].v_type == VAR_DICT)
2905 {
2906 if ((d = argvars[0].vval.v_dict) != NULL)
2907 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002908 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002909 if (di != NULL)
2910 tv = &di->di_tv;
2911 }
2912 }
2913 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
2914 {
2915 partial_T *pt;
2916 partial_T fref_pt;
2917
2918 if (argvars[0].v_type == VAR_PARTIAL)
2919 pt = argvars[0].vval.v_partial;
2920 else
2921 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02002922 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002923 fref_pt.pt_name = argvars[0].vval.v_string;
2924 pt = &fref_pt;
2925 }
2926
2927 if (pt != NULL)
2928 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002929 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002930 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002931
2932 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
2933 {
2934 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002935 n = partial_name(pt);
2936 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002937 rettv->vval.v_string = NULL;
2938 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002939 {
2940 rettv->vval.v_string = vim_strsave(n);
2941 if (rettv->v_type == VAR_FUNC)
2942 func_ref(rettv->vval.v_string);
2943 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002944 }
2945 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002946 {
2947 what_is_dict = TRUE;
2948 if (pt->pt_dict != NULL)
2949 rettv_dict_set(rettv, pt->pt_dict);
2950 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002951 else if (STRCMP(what, "args") == 0)
2952 {
2953 rettv->v_type = VAR_LIST;
2954 if (rettv_list_alloc(rettv) == OK)
2955 {
2956 int i;
2957
2958 for (i = 0; i < pt->pt_argc; ++i)
2959 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
2960 }
2961 }
2962 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002963 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02002964
2965 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
2966 // third argument
2967 if (!what_is_dict)
2968 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002969 }
2970 }
2971 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01002972 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002973
2974 if (tv == NULL)
2975 {
2976 if (argvars[2].v_type != VAR_UNKNOWN)
2977 copy_tv(&argvars[2], rettv);
2978 }
2979 else
2980 copy_tv(tv, rettv);
2981}
2982
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02002983/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01002984 * "getchangelist()" function
2985 */
2986 static void
2987f_getchangelist(typval_T *argvars, typval_T *rettv)
2988{
2989#ifdef FEAT_JUMPLIST
2990 buf_T *buf;
2991 int i;
2992 list_T *l;
2993 dict_T *d;
2994#endif
2995
2996 if (rettv_list_alloc(rettv) != OK)
2997 return;
2998
2999#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003000 if (argvars[0].v_type == VAR_UNKNOWN)
3001 buf = curbuf;
3002 else
3003 {
3004 (void)tv_get_number(&argvars[0]); // issue errmsg if type error
3005 ++emsg_off;
3006 buf = tv_get_buf(&argvars[0], FALSE);
3007 --emsg_off;
3008 }
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003009 if (buf == NULL)
3010 return;
3011
3012 l = list_alloc();
3013 if (l == NULL)
3014 return;
3015
3016 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3017 return;
3018 /*
3019 * The current window change list index tracks only the position in the
3020 * current buffer change list. For other buffers, use the change list
3021 * length as the current index.
3022 */
3023 list_append_number(rettv->vval.v_list,
3024 (varnumber_T)((buf == curwin->w_buffer)
3025 ? curwin->w_changelistidx : buf->b_changelistlen));
3026
3027 for (i = 0; i < buf->b_changelistlen; ++i)
3028 {
3029 if (buf->b_changelist[i].lnum == 0)
3030 continue;
3031 if ((d = dict_alloc()) == NULL)
3032 return;
3033 if (list_append_dict(l, d) == FAIL)
3034 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003035 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3036 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003037 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003038 }
3039#endif
3040}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003041
3042/*
3043 * "getcharsearch()" function
3044 */
3045 static void
3046f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3047{
3048 if (rettv_dict_alloc(rettv) != FAIL)
3049 {
3050 dict_T *dict = rettv->vval.v_dict;
3051
Bram Moolenaare0be1672018-07-08 16:50:37 +02003052 dict_add_string(dict, "char", last_csearch());
3053 dict_add_number(dict, "forward", last_csearch_forward());
3054 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003055 }
3056}
3057
3058/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003059 * "getenv()" function
3060 */
3061 static void
3062f_getenv(typval_T *argvars, typval_T *rettv)
3063{
3064 int mustfree = FALSE;
3065 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3066
3067 if (p == NULL)
3068 {
3069 rettv->v_type = VAR_SPECIAL;
3070 rettv->vval.v_number = VVAL_NULL;
3071 return;
3072 }
3073 if (!mustfree)
3074 p = vim_strsave(p);
3075 rettv->vval.v_string = p;
3076 rettv->v_type = VAR_STRING;
3077}
3078
3079/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003080 * "getfontname()" function
3081 */
3082 static void
3083f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3084{
3085 rettv->v_type = VAR_STRING;
3086 rettv->vval.v_string = NULL;
3087#ifdef FEAT_GUI
3088 if (gui.in_use)
3089 {
3090 GuiFont font;
3091 char_u *name = NULL;
3092
3093 if (argvars[0].v_type == VAR_UNKNOWN)
3094 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003095 // Get the "Normal" font. Either the name saved by
3096 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003097 font = gui.norm_font;
3098 name = hl_get_font_name();
3099 }
3100 else
3101 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003102 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003103 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003104 return;
3105 font = gui_mch_get_font(name, FALSE);
3106 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003107 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003108 }
3109 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3110 if (argvars[0].v_type != VAR_UNKNOWN)
3111 gui_mch_free_font(font);
3112 }
3113#endif
3114}
3115
3116/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003117 * "getjumplist()" function
3118 */
3119 static void
3120f_getjumplist(typval_T *argvars, typval_T *rettv)
3121{
3122#ifdef FEAT_JUMPLIST
3123 win_T *wp;
3124 int i;
3125 list_T *l;
3126 dict_T *d;
3127#endif
3128
3129 if (rettv_list_alloc(rettv) != OK)
3130 return;
3131
3132#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003133 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003134 if (wp == NULL)
3135 return;
3136
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003137 cleanup_jumplist(wp, TRUE);
3138
Bram Moolenaar4f505882018-02-10 21:06:32 +01003139 l = list_alloc();
3140 if (l == NULL)
3141 return;
3142
3143 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3144 return;
3145 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3146
3147 for (i = 0; i < wp->w_jumplistlen; ++i)
3148 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003149 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3150 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003151 if ((d = dict_alloc()) == NULL)
3152 return;
3153 if (list_append_dict(l, d) == FAIL)
3154 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003155 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
3156 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003157 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003158 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003159 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02003160 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003161 }
3162#endif
3163}
3164
3165/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003166 * "getpid()" function
3167 */
3168 static void
3169f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
3170{
3171 rettv->vval.v_number = mch_get_pid();
3172}
3173
3174 static void
3175getpos_both(
3176 typval_T *argvars,
3177 typval_T *rettv,
3178 int getcurpos)
3179{
3180 pos_T *fp;
3181 list_T *l;
3182 int fnum = -1;
3183
3184 if (rettv_list_alloc(rettv) == OK)
3185 {
3186 l = rettv->vval.v_list;
3187 if (getcurpos)
3188 fp = &curwin->w_cursor;
3189 else
3190 fp = var2fpos(&argvars[0], TRUE, &fnum);
3191 if (fnum != -1)
3192 list_append_number(l, (varnumber_T)fnum);
3193 else
3194 list_append_number(l, (varnumber_T)0);
3195 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
3196 : (varnumber_T)0);
3197 list_append_number(l, (fp != NULL)
3198 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
3199 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01003200 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003201 (varnumber_T)0);
3202 if (getcurpos)
3203 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01003204 int save_set_curswant = curwin->w_set_curswant;
3205 colnr_T save_curswant = curwin->w_curswant;
3206 colnr_T save_virtcol = curwin->w_virtcol;
3207
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003208 update_curswant();
3209 list_append_number(l, curwin->w_curswant == MAXCOL ?
3210 (varnumber_T)MAXCOL : (varnumber_T)curwin->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01003211
3212 // Do not change "curswant", as it is unexpected that a get
3213 // function has a side effect.
3214 if (save_set_curswant)
3215 {
3216 curwin->w_set_curswant = save_set_curswant;
3217 curwin->w_curswant = save_curswant;
3218 curwin->w_virtcol = save_virtcol;
3219 curwin->w_valid &= ~VALID_VIRTCOL;
3220 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003221 }
3222 }
3223 else
3224 rettv->vval.v_number = FALSE;
3225}
3226
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003227/*
3228 * "getcurpos()" function
3229 */
3230 static void
3231f_getcurpos(typval_T *argvars, typval_T *rettv)
3232{
3233 getpos_both(argvars, rettv, TRUE);
3234}
3235
3236/*
3237 * "getpos(string)" function
3238 */
3239 static void
3240f_getpos(typval_T *argvars, typval_T *rettv)
3241{
3242 getpos_both(argvars, rettv, FALSE);
3243}
3244
3245/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003246 * "getreg()" function
3247 */
3248 static void
3249f_getreg(typval_T *argvars, typval_T *rettv)
3250{
3251 char_u *strregname;
3252 int regname;
3253 int arg2 = FALSE;
3254 int return_list = FALSE;
3255 int error = FALSE;
3256
3257 if (argvars[0].v_type != VAR_UNKNOWN)
3258 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003259 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003260 error = strregname == NULL;
3261 if (argvars[1].v_type != VAR_UNKNOWN)
3262 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003263 arg2 = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003264 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003265 return_list = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003266 }
3267 }
3268 else
3269 strregname = get_vim_var_str(VV_REG);
3270
3271 if (error)
3272 return;
3273
3274 regname = (strregname == NULL ? '"' : *strregname);
3275 if (regname == 0)
3276 regname = '"';
3277
3278 if (return_list)
3279 {
3280 rettv->v_type = VAR_LIST;
3281 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
3282 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
3283 if (rettv->vval.v_list == NULL)
3284 (void)rettv_list_alloc(rettv);
3285 else
3286 ++rettv->vval.v_list->lv_refcount;
3287 }
3288 else
3289 {
3290 rettv->v_type = VAR_STRING;
3291 rettv->vval.v_string = get_reg_contents(regname,
3292 arg2 ? GREG_EXPR_SRC : 0);
3293 }
3294}
3295
3296/*
3297 * "getregtype()" function
3298 */
3299 static void
3300f_getregtype(typval_T *argvars, typval_T *rettv)
3301{
3302 char_u *strregname;
3303 int regname;
3304 char_u buf[NUMBUFLEN + 2];
3305 long reglen = 0;
3306
3307 if (argvars[0].v_type != VAR_UNKNOWN)
3308 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003309 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003310 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003311 {
3312 rettv->v_type = VAR_STRING;
3313 rettv->vval.v_string = NULL;
3314 return;
3315 }
3316 }
3317 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003318 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003319 strregname = get_vim_var_str(VV_REG);
3320
3321 regname = (strregname == NULL ? '"' : *strregname);
3322 if (regname == 0)
3323 regname = '"';
3324
3325 buf[0] = NUL;
3326 buf[1] = NUL;
3327 switch (get_reg_type(regname, &reglen))
3328 {
3329 case MLINE: buf[0] = 'V'; break;
3330 case MCHAR: buf[0] = 'v'; break;
3331 case MBLOCK:
3332 buf[0] = Ctrl_V;
3333 sprintf((char *)buf + 1, "%ld", reglen + 1);
3334 break;
3335 }
3336 rettv->v_type = VAR_STRING;
3337 rettv->vval.v_string = vim_strsave(buf);
3338}
3339
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003340/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01003341 * "gettagstack()" function
3342 */
3343 static void
3344f_gettagstack(typval_T *argvars, typval_T *rettv)
3345{
3346 win_T *wp = curwin; // default is current window
3347
3348 if (rettv_dict_alloc(rettv) != OK)
3349 return;
3350
3351 if (argvars[0].v_type != VAR_UNKNOWN)
3352 {
3353 wp = find_win_by_nr_or_id(&argvars[0]);
3354 if (wp == NULL)
3355 return;
3356 }
3357
3358 get_tagstack(wp, rettv->vval.v_dict);
3359}
3360
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003361// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003362#include "version.h"
3363
3364/*
3365 * "has()" function
3366 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01003367 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003368f_has(typval_T *argvars, typval_T *rettv)
3369{
3370 int i;
3371 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01003372 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003373 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01003374 typedef struct {
3375 char *name;
3376 short present;
3377 } has_item_T;
3378 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003379 {
Bram Moolenaar79296512020-03-22 16:17:14 +01003380 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003381#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01003382 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003383#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003384 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01003385#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003386 },
3387 {"arp",
3388#if defined(AMIGA) && defined(FEAT_ARP)
3389 1
3390#else
3391 0
3392#endif
3393 },
Bram Moolenaar79296512020-03-22 16:17:14 +01003394 {"haiku",
3395#ifdef __HAIKU__
3396 1
3397#else
3398 0
3399#endif
3400 },
3401 {"bsd",
3402#if defined(BSD) && !defined(MACOS_X)
3403 1
3404#else
3405 0
3406#endif
3407 },
3408 {"hpux",
3409#ifdef hpux
3410 1
3411#else
3412 0
3413#endif
3414 },
3415 {"linux",
3416#ifdef __linux__
3417 1
3418#else
3419 0
3420#endif
3421 },
3422 {"mac", // Mac OS X (and, once, Mac OS Classic)
3423#ifdef MACOS_X
3424 1
3425#else
3426 0
3427#endif
3428 },
3429 {"osx", // Mac OS X
3430#ifdef MACOS_X
3431 1
3432#else
3433 0
3434#endif
3435 },
3436 {"macunix", // Mac OS X, with the darwin feature
3437#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3438 1
3439#else
3440 0
3441#endif
3442 },
3443 {"osxdarwin", // synonym for macunix
3444#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
3445 1
3446#else
3447 0
3448#endif
3449 },
3450 {"qnx",
3451#ifdef __QNX__
3452 1
3453#else
3454 0
3455#endif
3456 },
3457 {"sun",
3458#ifdef SUN_SYSTEM
3459 1
3460#else
3461 0
3462#endif
3463 },
3464 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003465#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01003466 1
3467#else
3468 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003469#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003470 },
3471 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003472#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01003473 1
3474#else
3475 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003476#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003477 },
3478 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01003479#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01003480 1
3481#else
3482 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003483#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003484 },
3485 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01003486#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01003487 1
3488#else
3489 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003490#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003491 },
3492 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01003493#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01003494 1
3495#else
3496 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003497#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003498 },
3499 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003500#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003501 1
3502#else
3503 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003504#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003505 },
3506 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003507#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003508 1
3509#else
3510 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003511#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003512 },
3513 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003514#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01003515 1
3516#else
3517 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003518#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003519 },
3520 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003521#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01003522 1
3523#else
3524 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003525#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003526 },
3527 {"autocmd", 1},
3528 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003529#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01003530 1
3531#else
3532 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02003533#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003534 },
3535 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003536#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01003537 1
3538#else
3539 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01003540#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003541 },
3542 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003543#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01003544 1
3545#else
3546 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003547#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003548 },
3549 {"balloon_multiline",
3550#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
3551 // MS-Windows requires runtime check, see below
3552 1
3553#else
3554 0
3555#endif
3556 },
3557 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01003558#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01003559 1
3560#else
3561 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01003562#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003563 },
3564 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003565#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01003566 1
3567#else
3568 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003569#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003570 },
3571 {"all_builtin_terms",
3572#if defined(ALL_BUILTIN_TCAPS)
3573 1
3574#else
3575 0
3576#endif
3577 },
3578 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003579#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01003580 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003581 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01003582 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003583#else
Bram Moolenaar79296512020-03-22 16:17:14 +01003584 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003585#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01003586 },
3587 {"byte_offset",
3588#ifdef FEAT_BYTEOFF
3589 1
3590#else
3591 0
3592#endif
3593 },
3594 {"channel",
3595#ifdef FEAT_JOB_CHANNEL
3596 1
3597#else
3598 0
3599#endif
3600 },
3601 {"cindent",
3602#ifdef FEAT_CINDENT
3603 1
3604#else
3605 0
3606#endif
3607 },
3608 {"clientserver",
3609#ifdef FEAT_CLIENTSERVER
3610 1
3611#else
3612 0
3613#endif
3614 },
3615 {"clipboard",
3616#ifdef FEAT_CLIPBOARD
3617 1
3618#else
3619 0
3620#endif
3621 },
3622 {"cmdline_compl", 1},
3623 {"cmdline_hist", 1},
3624 {"comments", 1},
3625 {"conceal",
3626#ifdef FEAT_CONCEAL
3627 1
3628#else
3629 0
3630#endif
3631 },
3632 {"cryptv",
3633#ifdef FEAT_CRYPT
3634 1
3635#else
3636 0
3637#endif
3638 },
3639 {"crypt-blowfish",
3640#ifdef FEAT_CRYPT
3641 1
3642#else
3643 0
3644#endif
3645 },
3646 {"crypt-blowfish2",
3647#ifdef FEAT_CRYPT
3648 1
3649#else
3650 0
3651#endif
3652 },
3653 {"cscope",
3654#ifdef FEAT_CSCOPE
3655 1
3656#else
3657 0
3658#endif
3659 },
3660 {"cursorbind", 1},
3661 {"cursorshape",
3662#ifdef CURSOR_SHAPE
3663 1
3664#else
3665 0
3666#endif
3667 },
3668 {"debug",
3669#ifdef DEBUG
3670 1
3671#else
3672 0
3673#endif
3674 },
3675 {"dialog_con",
3676#ifdef FEAT_CON_DIALOG
3677 1
3678#else
3679 0
3680#endif
3681 },
3682 {"dialog_gui",
3683#ifdef FEAT_GUI_DIALOG
3684 1
3685#else
3686 0
3687#endif
3688 },
3689 {"diff",
3690#ifdef FEAT_DIFF
3691 1
3692#else
3693 0
3694#endif
3695 },
3696 {"digraphs",
3697#ifdef FEAT_DIGRAPHS
3698 1
3699#else
3700 0
3701#endif
3702 },
3703 {"directx",
3704#ifdef FEAT_DIRECTX
3705 1
3706#else
3707 0
3708#endif
3709 },
3710 {"dnd",
3711#ifdef FEAT_DND
3712 1
3713#else
3714 0
3715#endif
3716 },
3717 {"emacs_tags",
3718#ifdef FEAT_EMACS_TAGS
3719 1
3720#else
3721 0
3722#endif
3723 },
3724 {"eval", 1}, // always present, of course!
3725 {"ex_extra", 1}, // graduated feature
3726 {"extra_search",
3727#ifdef FEAT_SEARCH_EXTRA
3728 1
3729#else
3730 0
3731#endif
3732 },
3733 {"file_in_path",
3734#ifdef FEAT_SEARCHPATH
3735 1
3736#else
3737 0
3738#endif
3739 },
3740 {"filterpipe",
3741#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
3742 1
3743#else
3744 0
3745#endif
3746 },
3747 {"find_in_path",
3748#ifdef FEAT_FIND_ID
3749 1
3750#else
3751 0
3752#endif
3753 },
3754 {"float",
3755#ifdef FEAT_FLOAT
3756 1
3757#else
3758 0
3759#endif
3760 },
3761 {"folding",
3762#ifdef FEAT_FOLDING
3763 1
3764#else
3765 0
3766#endif
3767 },
3768 {"footer",
3769#ifdef FEAT_FOOTER
3770 1
3771#else
3772 0
3773#endif
3774 },
3775 {"fork",
3776#if !defined(USE_SYSTEM) && defined(UNIX)
3777 1
3778#else
3779 0
3780#endif
3781 },
3782 {"gettext",
3783#ifdef FEAT_GETTEXT
3784 1
3785#else
3786 0
3787#endif
3788 },
3789 {"gui",
3790#ifdef FEAT_GUI
3791 1
3792#else
3793 0
3794#endif
3795 },
3796 {"gui_neXtaw",
3797#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
3798 1
3799#else
3800 0
3801#endif
3802 },
3803 {"gui_athena",
3804#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
3805 1
3806#else
3807 0
3808#endif
3809 },
3810 {"gui_gtk",
3811#ifdef FEAT_GUI_GTK
3812 1
3813#else
3814 0
3815#endif
3816 },
3817 {"gui_gtk2",
3818#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
3819 1
3820#else
3821 0
3822#endif
3823 },
3824 {"gui_gtk3",
3825#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
3826 1
3827#else
3828 0
3829#endif
3830 },
3831 {"gui_gnome",
3832#ifdef FEAT_GUI_GNOME
3833 1
3834#else
3835 0
3836#endif
3837 },
3838 {"gui_haiku",
3839#ifdef FEAT_GUI_HAIKU
3840 1
3841#else
3842 0
3843#endif
3844 },
3845 {"gui_mac",
3846#ifdef FEAT_GUI_MAC
3847 1
3848#else
3849 0
3850#endif
3851 },
3852 {"gui_motif",
3853#ifdef FEAT_GUI_MOTIF
3854 1
3855#else
3856 0
3857#endif
3858 },
3859 {"gui_photon",
3860#ifdef FEAT_GUI_PHOTON
3861 1
3862#else
3863 0
3864#endif
3865 },
3866 {"gui_win32",
3867#ifdef FEAT_GUI_MSWIN
3868 1
3869#else
3870 0
3871#endif
3872 },
3873 {"iconv",
3874#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
3875 1
3876#else
3877 0
3878#endif
3879 },
3880 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02003881 {"ipv6",
3882#ifdef FEAT_IPV6
3883 1
3884#else
3885 0
3886#endif
3887 },
Bram Moolenaar79296512020-03-22 16:17:14 +01003888 {"job",
3889#ifdef FEAT_JOB_CHANNEL
3890 1
3891#else
3892 0
3893#endif
3894 },
3895 {"jumplist",
3896#ifdef FEAT_JUMPLIST
3897 1
3898#else
3899 0
3900#endif
3901 },
3902 {"keymap",
3903#ifdef FEAT_KEYMAP
3904 1
3905#else
3906 0
3907#endif
3908 },
3909 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
3910 {"langmap",
3911#ifdef FEAT_LANGMAP
3912 1
3913#else
3914 0
3915#endif
3916 },
3917 {"libcall",
3918#ifdef FEAT_LIBCALL
3919 1
3920#else
3921 0
3922#endif
3923 },
3924 {"linebreak",
3925#ifdef FEAT_LINEBREAK
3926 1
3927#else
3928 0
3929#endif
3930 },
3931 {"lispindent",
3932#ifdef FEAT_LISP
3933 1
3934#else
3935 0
3936#endif
3937 },
3938 {"listcmds", 1},
3939 {"localmap", 1},
3940 {"lua",
3941#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
3942 1
3943#else
3944 0
3945#endif
3946 },
3947 {"menu",
3948#ifdef FEAT_MENU
3949 1
3950#else
3951 0
3952#endif
3953 },
3954 {"mksession",
3955#ifdef FEAT_SESSION
3956 1
3957#else
3958 0
3959#endif
3960 },
3961 {"modify_fname", 1},
3962 {"mouse", 1},
3963 {"mouseshape",
3964#ifdef FEAT_MOUSESHAPE
3965 1
3966#else
3967 0
3968#endif
3969 },
3970 {"mouse_dec",
3971#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
3972 1
3973#else
3974 0
3975#endif
3976 },
3977 {"mouse_gpm",
3978#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
3979 1
3980#else
3981 0
3982#endif
3983 },
3984 {"mouse_jsbterm",
3985#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
3986 1
3987#else
3988 0
3989#endif
3990 },
3991 {"mouse_netterm",
3992#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
3993 1
3994#else
3995 0
3996#endif
3997 },
3998 {"mouse_pterm",
3999#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4000 1
4001#else
4002 0
4003#endif
4004 },
4005 {"mouse_sgr",
4006#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4007 1
4008#else
4009 0
4010#endif
4011 },
4012 {"mouse_sysmouse",
4013#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4014 1
4015#else
4016 0
4017#endif
4018 },
4019 {"mouse_urxvt",
4020#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4021 1
4022#else
4023 0
4024#endif
4025 },
4026 {"mouse_xterm",
4027#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4028 1
4029#else
4030 0
4031#endif
4032 },
4033 {"multi_byte", 1},
4034 {"multi_byte_ime",
4035#ifdef FEAT_MBYTE_IME
4036 1
4037#else
4038 0
4039#endif
4040 },
4041 {"multi_lang",
4042#ifdef FEAT_MULTI_LANG
4043 1
4044#else
4045 0
4046#endif
4047 },
4048 {"mzscheme",
4049#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4050 1
4051#else
4052 0
4053#endif
4054 },
4055 {"num64", 1},
4056 {"ole",
4057#ifdef FEAT_OLE
4058 1
4059#else
4060 0
4061#endif
4062 },
4063 {"packages",
4064#ifdef FEAT_EVAL
4065 1
4066#else
4067 0
4068#endif
4069 },
4070 {"path_extra",
4071#ifdef FEAT_PATH_EXTRA
4072 1
4073#else
4074 0
4075#endif
4076 },
4077 {"perl",
4078#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4079 1
4080#else
4081 0
4082#endif
4083 },
4084 {"persistent_undo",
4085#ifdef FEAT_PERSISTENT_UNDO
4086 1
4087#else
4088 0
4089#endif
4090 },
4091 {"python_compiled",
4092#if defined(FEAT_PYTHON)
4093 1
4094#else
4095 0
4096#endif
4097 },
4098 {"python_dynamic",
4099#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4100 1
4101#else
4102 0
4103#endif
4104 },
4105 {"python",
4106#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4107 1
4108#else
4109 0
4110#endif
4111 },
4112 {"pythonx",
4113#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4114 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4115 1
4116#else
4117 0
4118#endif
4119 },
4120 {"python3_compiled",
4121#if defined(FEAT_PYTHON3)
4122 1
4123#else
4124 0
4125#endif
4126 },
4127 {"python3_dynamic",
4128#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
4129 1
4130#else
4131 0
4132#endif
4133 },
4134 {"python3",
4135#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
4136 1
4137#else
4138 0
4139#endif
4140 },
4141 {"popupwin",
4142#ifdef FEAT_PROP_POPUP
4143 1
4144#else
4145 0
4146#endif
4147 },
4148 {"postscript",
4149#ifdef FEAT_POSTSCRIPT
4150 1
4151#else
4152 0
4153#endif
4154 },
4155 {"printer",
4156#ifdef FEAT_PRINTER
4157 1
4158#else
4159 0
4160#endif
4161 },
4162 {"profile",
4163#ifdef FEAT_PROFILE
4164 1
4165#else
4166 0
4167#endif
4168 },
4169 {"reltime",
4170#ifdef FEAT_RELTIME
4171 1
4172#else
4173 0
4174#endif
4175 },
4176 {"quickfix",
4177#ifdef FEAT_QUICKFIX
4178 1
4179#else
4180 0
4181#endif
4182 },
4183 {"rightleft",
4184#ifdef FEAT_RIGHTLEFT
4185 1
4186#else
4187 0
4188#endif
4189 },
4190 {"ruby",
4191#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
4192 1
4193#else
4194 0
4195#endif
4196 },
4197 {"scrollbind", 1},
4198 {"showcmd",
4199#ifdef FEAT_CMDL_INFO
4200 1
4201#else
4202 0
4203#endif
4204 },
4205 {"cmdline_info",
4206#ifdef FEAT_CMDL_INFO
4207 1
4208#else
4209 0
4210#endif
4211 },
4212 {"signs",
4213#ifdef FEAT_SIGNS
4214 1
4215#else
4216 0
4217#endif
4218 },
4219 {"smartindent",
4220#ifdef FEAT_SMARTINDENT
4221 1
4222#else
4223 0
4224#endif
4225 },
4226 {"startuptime",
4227#ifdef STARTUPTIME
4228 1
4229#else
4230 0
4231#endif
4232 },
4233 {"statusline",
4234#ifdef FEAT_STL_OPT
4235 1
4236#else
4237 0
4238#endif
4239 },
4240 {"netbeans_intg",
4241#ifdef FEAT_NETBEANS_INTG
4242 1
4243#else
4244 0
4245#endif
4246 },
4247 {"sound",
4248#ifdef FEAT_SOUND
4249 1
4250#else
4251 0
4252#endif
4253 },
4254 {"spell",
4255#ifdef FEAT_SPELL
4256 1
4257#else
4258 0
4259#endif
4260 },
4261 {"syntax",
4262#ifdef FEAT_SYN_HL
4263 1
4264#else
4265 0
4266#endif
4267 },
4268 {"system",
4269#if defined(USE_SYSTEM) || !defined(UNIX)
4270 1
4271#else
4272 0
4273#endif
4274 },
4275 {"tag_binary",
4276#ifdef FEAT_TAG_BINS
4277 1
4278#else
4279 0
4280#endif
4281 },
4282 {"tcl",
4283#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
4284 1
4285#else
4286 0
4287#endif
4288 },
4289 {"termguicolors",
4290#ifdef FEAT_TERMGUICOLORS
4291 1
4292#else
4293 0
4294#endif
4295 },
4296 {"terminal",
4297#if defined(FEAT_TERMINAL) && !defined(MSWIN)
4298 1
4299#else
4300 0
4301#endif
4302 },
4303 {"terminfo",
4304#ifdef TERMINFO
4305 1
4306#else
4307 0
4308#endif
4309 },
4310 {"termresponse",
4311#ifdef FEAT_TERMRESPONSE
4312 1
4313#else
4314 0
4315#endif
4316 },
4317 {"textobjects",
4318#ifdef FEAT_TEXTOBJ
4319 1
4320#else
4321 0
4322#endif
4323 },
4324 {"textprop",
4325#ifdef FEAT_PROP_POPUP
4326 1
4327#else
4328 0
4329#endif
4330 },
4331 {"tgetent",
4332#ifdef HAVE_TGETENT
4333 1
4334#else
4335 0
4336#endif
4337 },
4338 {"timers",
4339#ifdef FEAT_TIMERS
4340 1
4341#else
4342 0
4343#endif
4344 },
4345 {"title",
4346#ifdef FEAT_TITLE
4347 1
4348#else
4349 0
4350#endif
4351 },
4352 {"toolbar",
4353#ifdef FEAT_TOOLBAR
4354 1
4355#else
4356 0
4357#endif
4358 },
4359 {"unnamedplus",
4360#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
4361 1
4362#else
4363 0
4364#endif
4365 },
4366 {"user-commands", 1}, // was accidentally included in 5.4
4367 {"user_commands", 1},
4368 {"vartabs",
4369#ifdef FEAT_VARTABS
4370 1
4371#else
4372 0
4373#endif
4374 },
4375 {"vertsplit", 1},
4376 {"viminfo",
4377#ifdef FEAT_VIMINFO
4378 1
4379#else
4380 0
4381#endif
4382 },
4383 {"vimscript-1", 1},
4384 {"vimscript-2", 1},
4385 {"vimscript-3", 1},
4386 {"vimscript-4", 1},
4387 {"virtualedit", 1},
4388 {"visual", 1},
4389 {"visualextra", 1},
4390 {"vreplace", 1},
4391 {"vtp",
4392#ifdef FEAT_VTP
4393 1
4394#else
4395 0
4396#endif
4397 },
4398 {"wildignore",
4399#ifdef FEAT_WILDIGN
4400 1
4401#else
4402 0
4403#endif
4404 },
4405 {"wildmenu",
4406#ifdef FEAT_WILDMENU
4407 1
4408#else
4409 0
4410#endif
4411 },
4412 {"windows", 1},
4413 {"winaltkeys",
4414#ifdef FEAT_WAK
4415 1
4416#else
4417 0
4418#endif
4419 },
4420 {"writebackup",
4421#ifdef FEAT_WRITEBACKUP
4422 1
4423#else
4424 0
4425#endif
4426 },
4427 {"xim",
4428#ifdef FEAT_XIM
4429 1
4430#else
4431 0
4432#endif
4433 },
4434 {"xfontset",
4435#ifdef FEAT_XFONTSET
4436 1
4437#else
4438 0
4439#endif
4440 },
4441 {"xpm",
4442#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
4443 1
4444#else
4445 0
4446#endif
4447 },
4448 {"xpm_w32", // for backward compatibility
4449#ifdef FEAT_XPM_W32
4450 1
4451#else
4452 0
4453#endif
4454 },
4455 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004456#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01004457 1
4458#else
4459 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004460#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004461 },
4462 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004463#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01004464 1
4465#else
4466 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004467#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004468 },
4469 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004470#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01004471 1
4472#else
4473 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004474#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004475 },
4476 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004477#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01004478 1
4479#else
4480 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004481#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004482 },
4483 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004484#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01004485 1
4486#else
4487 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004488#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004489 },
4490 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004491 };
4492
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004493 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01004494 for (i = 0; has_list[i].name != NULL; ++i)
4495 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004496 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004497 x = TRUE;
4498 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004499 break;
4500 }
4501
Bram Moolenaar79296512020-03-22 16:17:14 +01004502 // features also in has_list[] but sometimes enabled at runtime
4503 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004504 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004505 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02004506 {
4507 // intentionally empty
4508 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01004509#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004510 else if (STRICMP(name, "balloon_multiline") == 0)
4511 n = multiline_balloon_available();
4512#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004513#ifdef VIMDLL
4514 else if (STRICMP(name, "filterpipe") == 0)
4515 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004516#endif
4517#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
4518 else if (STRICMP(name, "iconv") == 0)
4519 n = iconv_enabled(FALSE);
4520#endif
4521#ifdef DYNAMIC_LUA
4522 else if (STRICMP(name, "lua") == 0)
4523 n = lua_enabled(FALSE);
4524#endif
4525#ifdef DYNAMIC_MZSCHEME
4526 else if (STRICMP(name, "mzscheme") == 0)
4527 n = mzscheme_enabled(FALSE);
4528#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004529#ifdef DYNAMIC_PERL
4530 else if (STRICMP(name, "perl") == 0)
4531 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004532#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004533#ifdef DYNAMIC_PYTHON
4534 else if (STRICMP(name, "python") == 0)
4535 n = python_enabled(FALSE);
4536#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004537#ifdef DYNAMIC_PYTHON3
4538 else if (STRICMP(name, "python3") == 0)
4539 n = python3_enabled(FALSE);
4540#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01004541#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
4542 else if (STRICMP(name, "pythonx") == 0)
4543 {
4544# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
4545 if (p_pyx == 0)
4546 n = python3_enabled(FALSE) || python_enabled(FALSE);
4547 else if (p_pyx == 3)
4548 n = python3_enabled(FALSE);
4549 else if (p_pyx == 2)
4550 n = python_enabled(FALSE);
4551# elif defined(DYNAMIC_PYTHON)
4552 n = python_enabled(FALSE);
4553# elif defined(DYNAMIC_PYTHON3)
4554 n = python3_enabled(FALSE);
4555# endif
4556 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004557#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004558#ifdef DYNAMIC_RUBY
4559 else if (STRICMP(name, "ruby") == 0)
4560 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004561#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004562#ifdef DYNAMIC_TCL
4563 else if (STRICMP(name, "tcl") == 0)
4564 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02004565#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01004566#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02004567 else if (STRICMP(name, "terminal") == 0)
4568 n = terminal_enabled();
4569#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004570 }
4571
Bram Moolenaar79296512020-03-22 16:17:14 +01004572 // features not in has_list[]
4573 if (x == FALSE)
4574 {
4575 if (STRNICMP(name, "patch", 5) == 0)
4576 {
4577 x = TRUE;
4578 if (name[5] == '-'
4579 && STRLEN(name) >= 11
4580 && vim_isdigit(name[6])
4581 && vim_isdigit(name[8])
4582 && vim_isdigit(name[10]))
4583 {
4584 int major = atoi((char *)name + 6);
4585 int minor = atoi((char *)name + 8);
4586
4587 // Expect "patch-9.9.01234".
4588 n = (major < VIM_VERSION_MAJOR
4589 || (major == VIM_VERSION_MAJOR
4590 && (minor < VIM_VERSION_MINOR
4591 || (minor == VIM_VERSION_MINOR
4592 && has_patch(atoi((char *)name + 10))))));
4593 }
4594 else
4595 n = has_patch(atoi((char *)name + 5));
4596 }
4597 else if (STRICMP(name, "vim_starting") == 0)
4598 {
4599 x = TRUE;
4600 n = (starting != 0);
4601 }
4602 else if (STRICMP(name, "ttyin") == 0)
4603 {
4604 x = TRUE;
4605 n = mch_input_isatty();
4606 }
4607 else if (STRICMP(name, "ttyout") == 0)
4608 {
4609 x = TRUE;
4610 n = stdout_isatty;
4611 }
4612 else if (STRICMP(name, "multi_byte_encoding") == 0)
4613 {
4614 x = TRUE;
4615 n = has_mbyte;
4616 }
4617 else if (STRICMP(name, "gui_running") == 0)
4618 {
4619 x = TRUE;
4620#ifdef FEAT_GUI
4621 n = (gui.in_use || gui.starting);
4622#endif
4623 }
4624 else if (STRICMP(name, "browse") == 0)
4625 {
4626 x = TRUE;
4627#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
4628 n = gui.in_use; // gui_mch_browse() works when GUI is running
4629#endif
4630 }
4631 else if (STRICMP(name, "syntax_items") == 0)
4632 {
4633 x = TRUE;
4634#ifdef FEAT_SYN_HL
4635 n = syntax_present(curwin);
4636#endif
4637 }
4638 else if (STRICMP(name, "vcon") == 0)
4639 {
4640 x = TRUE;
4641#ifdef FEAT_VTP
4642 n = is_term_win32() && has_vtp_working();
4643#endif
4644 }
4645 else if (STRICMP(name, "netbeans_enabled") == 0)
4646 {
4647 x = TRUE;
4648#ifdef FEAT_NETBEANS_INTG
4649 n = netbeans_active();
4650#endif
4651 }
4652 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
4653 {
4654 x = TRUE;
4655#ifdef FEAT_MOUSE_GPM
4656 n = gpm_enabled();
4657#endif
4658 }
4659 else if (STRICMP(name, "conpty") == 0)
4660 {
4661 x = TRUE;
4662#if defined(FEAT_TERMINAL) && defined(MSWIN)
4663 n = use_conpty();
4664#endif
4665 }
4666 else if (STRICMP(name, "clipboard_working") == 0)
4667 {
4668 x = TRUE;
4669#ifdef FEAT_CLIPBOARD
4670 n = clip_star.available;
4671#endif
4672 }
4673 }
4674
4675 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_number(&argvars[1]) != 0)
4676 // return whether feature could ever be enabled
4677 rettv->vval.v_number = x;
4678 else
4679 // return whether feature is enabled
4680 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004681}
4682
4683/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004684 * "haslocaldir()" function
4685 */
4686 static void
4687f_haslocaldir(typval_T *argvars, typval_T *rettv)
4688{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004689 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004690 win_T *wp = NULL;
4691
Bram Moolenaar00aa0692019-04-27 20:37:57 +02004692 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
4693
4694 // Check for window-local and tab-local directories
4695 if (wp != NULL && wp->w_localdir != NULL)
4696 rettv->vval.v_number = 1;
4697 else if (tp != NULL && tp->tp_localdir != NULL)
4698 rettv->vval.v_number = 2;
4699 else
4700 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004701}
4702
4703/*
4704 * "hasmapto()" function
4705 */
4706 static void
4707f_hasmapto(typval_T *argvars, typval_T *rettv)
4708{
4709 char_u *name;
4710 char_u *mode;
4711 char_u buf[NUMBUFLEN];
4712 int abbr = FALSE;
4713
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004714 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004715 if (argvars[1].v_type == VAR_UNKNOWN)
4716 mode = (char_u *)"nvo";
4717 else
4718 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004719 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004720 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004721 abbr = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004722 }
4723
4724 if (map_to_exists(name, mode, abbr))
4725 rettv->vval.v_number = TRUE;
4726 else
4727 rettv->vval.v_number = FALSE;
4728}
4729
4730/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004731 * "highlightID(name)" function
4732 */
4733 static void
4734f_hlID(typval_T *argvars, typval_T *rettv)
4735{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004736 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004737}
4738
4739/*
4740 * "highlight_exists()" function
4741 */
4742 static void
4743f_hlexists(typval_T *argvars, typval_T *rettv)
4744{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004745 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004746}
4747
4748/*
4749 * "hostname()" function
4750 */
4751 static void
4752f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
4753{
4754 char_u hostname[256];
4755
4756 mch_get_host_name(hostname, 256);
4757 rettv->v_type = VAR_STRING;
4758 rettv->vval.v_string = vim_strsave(hostname);
4759}
4760
4761/*
4762 * iconv() function
4763 */
4764 static void
4765f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
4766{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004767 char_u buf1[NUMBUFLEN];
4768 char_u buf2[NUMBUFLEN];
4769 char_u *from, *to, *str;
4770 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004771
4772 rettv->v_type = VAR_STRING;
4773 rettv->vval.v_string = NULL;
4774
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004775 str = tv_get_string(&argvars[0]);
4776 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
4777 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004778 vimconv.vc_type = CONV_NONE;
4779 convert_setup(&vimconv, from, to);
4780
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004781 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004782 if (vimconv.vc_type == CONV_NONE)
4783 rettv->vval.v_string = vim_strsave(str);
4784 else
4785 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
4786
4787 convert_setup(&vimconv, NULL, NULL);
4788 vim_free(from);
4789 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004790}
4791
4792/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004793 * "index()" function
4794 */
4795 static void
4796f_index(typval_T *argvars, typval_T *rettv)
4797{
4798 list_T *l;
4799 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004800 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004801 long idx = 0;
4802 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004803 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004804
4805 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004806 if (argvars[0].v_type == VAR_BLOB)
4807 {
4808 typval_T tv;
4809 int start = 0;
4810
4811 if (argvars[2].v_type != VAR_UNKNOWN)
4812 {
4813 start = tv_get_number_chk(&argvars[2], &error);
4814 if (error)
4815 return;
4816 }
4817 b = argvars[0].vval.v_blob;
4818 if (b == NULL)
4819 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01004820 if (start < 0)
4821 {
4822 start = blob_len(b) + start;
4823 if (start < 0)
4824 start = 0;
4825 }
4826
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004827 for (idx = start; idx < blob_len(b); ++idx)
4828 {
4829 tv.v_type = VAR_NUMBER;
4830 tv.vval.v_number = blob_get(b, idx);
4831 if (tv_equal(&tv, &argvars[1], ic, FALSE))
4832 {
4833 rettv->vval.v_number = idx;
4834 return;
4835 }
4836 }
4837 return;
4838 }
4839 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004840 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01004841 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004842 return;
4843 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01004844
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004845 l = argvars[0].vval.v_list;
4846 if (l != NULL)
4847 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02004848 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004849 item = l->lv_first;
4850 if (argvars[2].v_type != VAR_UNKNOWN)
4851 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004852 // Start at specified item. Use the cached index that list_find()
4853 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004854 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01004855 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004856 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004857 ic = (int)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004858 if (error)
4859 item = NULL;
4860 }
4861
4862 for ( ; item != NULL; item = item->li_next, ++idx)
4863 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
4864 {
4865 rettv->vval.v_number = idx;
4866 break;
4867 }
4868 }
4869}
4870
4871static int inputsecret_flag = 0;
4872
4873/*
4874 * "input()" function
4875 * Also handles inputsecret() when inputsecret is set.
4876 */
4877 static void
4878f_input(typval_T *argvars, typval_T *rettv)
4879{
4880 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
4881}
4882
4883/*
4884 * "inputdialog()" function
4885 */
4886 static void
4887f_inputdialog(typval_T *argvars, typval_T *rettv)
4888{
4889#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004890 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004891 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
4892 {
4893 char_u *message;
4894 char_u buf[NUMBUFLEN];
4895 char_u *defstr = (char_u *)"";
4896
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004897 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004898 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004899 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004900 vim_strncpy(IObuff, defstr, IOSIZE - 1);
4901 else
4902 IObuff[0] = NUL;
4903 if (message != NULL && defstr != NULL
4904 && do_dialog(VIM_QUESTION, NULL, message,
4905 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
4906 rettv->vval.v_string = vim_strsave(IObuff);
4907 else
4908 {
4909 if (message != NULL && defstr != NULL
4910 && argvars[1].v_type != VAR_UNKNOWN
4911 && argvars[2].v_type != VAR_UNKNOWN)
4912 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004913 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004914 else
4915 rettv->vval.v_string = NULL;
4916 }
4917 rettv->v_type = VAR_STRING;
4918 }
4919 else
4920#endif
4921 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
4922}
4923
4924/*
4925 * "inputlist()" function
4926 */
4927 static void
4928f_inputlist(typval_T *argvars, typval_T *rettv)
4929{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004930 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004931 listitem_T *li;
4932 int selected;
4933 int mouse_used;
4934
4935#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004936 // While starting up, there is no place to enter text. When running tests
4937 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02004938 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004939 return;
4940#endif
4941 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
4942 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004943 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004944 return;
4945 }
4946
4947 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004948 msg_row = Rows - 1; // for when 'cmdheight' > 1
4949 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004950 msg_scroll = TRUE;
4951 msg_clr_eos();
4952
Bram Moolenaar50985eb2020-01-27 22:09:39 +01004953 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02004954 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02004955 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004956 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01004957 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004958 msg_putchar('\n');
4959 }
4960
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004961 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004962 selected = prompt_for_number(&mouse_used);
4963 if (mouse_used)
4964 selected -= lines_left;
4965
4966 rettv->vval.v_number = selected;
4967}
4968
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004969static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
4970
4971/*
4972 * "inputrestore()" function
4973 */
4974 static void
4975f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
4976{
4977 if (ga_userinput.ga_len > 0)
4978 {
4979 --ga_userinput.ga_len;
4980 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
4981 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004982 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004983 }
4984 else if (p_verbose > 1)
4985 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01004986 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004987 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004988 }
4989}
4990
4991/*
4992 * "inputsave()" function
4993 */
4994 static void
4995f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
4996{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004997 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004998 if (ga_grow(&ga_userinput, 1) == OK)
4999 {
5000 save_typeahead((tasave_T *)(ga_userinput.ga_data)
5001 + ga_userinput.ga_len);
5002 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005003 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005004 }
5005 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005006 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005007}
5008
5009/*
5010 * "inputsecret()" function
5011 */
5012 static void
5013f_inputsecret(typval_T *argvars, typval_T *rettv)
5014{
5015 ++cmdline_star;
5016 ++inputsecret_flag;
5017 f_input(argvars, rettv);
5018 --cmdline_star;
5019 --inputsecret_flag;
5020}
5021
5022/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005023 * "interrupt()" function
5024 */
5025 static void
5026f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5027{
5028 got_int = TRUE;
5029}
5030
5031/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005032 * "invert(expr)" function
5033 */
5034 static void
5035f_invert(typval_T *argvars, typval_T *rettv)
5036{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005037 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005038}
5039
5040/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005041 * "islocked()" function
5042 */
5043 static void
5044f_islocked(typval_T *argvars, typval_T *rettv)
5045{
5046 lval_T lv;
5047 char_u *end;
5048 dictitem_T *di;
5049
5050 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005051 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005052 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005053 if (end != NULL && lv.ll_name != NULL)
5054 {
5055 if (*end != NUL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005056 emsg(_(e_trailing));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005057 else
5058 {
5059 if (lv.ll_tv == NULL)
5060 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01005061 di = find_var(lv.ll_name, NULL, TRUE);
5062 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005063 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005064 // Consider a variable locked when:
5065 // 1. the variable itself is locked
5066 // 2. the value of the variable is locked.
5067 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01005068 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
5069 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005070 }
5071 }
5072 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005073 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005074 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005075 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005076 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005077 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005078 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
5079 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005080 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005081 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
5082 }
5083 }
5084
5085 clear_lval(&lv);
5086}
5087
5088#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
5089/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02005090 * "isinf()" function
5091 */
5092 static void
5093f_isinf(typval_T *argvars, typval_T *rettv)
5094{
5095 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
5096 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
5097}
5098
5099/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005100 * "isnan()" function
5101 */
5102 static void
5103f_isnan(typval_T *argvars, typval_T *rettv)
5104{
5105 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
5106 && isnan(argvars[0].vval.v_float);
5107}
5108#endif
5109
5110/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005111 * "last_buffer_nr()" function.
5112 */
5113 static void
5114f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
5115{
5116 int n = 0;
5117 buf_T *buf;
5118
Bram Moolenaar29323592016-07-24 22:04:11 +02005119 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005120 if (n < buf->b_fnum)
5121 n = buf->b_fnum;
5122
5123 rettv->vval.v_number = n;
5124}
5125
5126/*
5127 * "len()" function
5128 */
5129 static void
5130f_len(typval_T *argvars, typval_T *rettv)
5131{
5132 switch (argvars[0].v_type)
5133 {
5134 case VAR_STRING:
5135 case VAR_NUMBER:
5136 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005137 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005138 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005139 case VAR_BLOB:
5140 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
5141 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005142 case VAR_LIST:
5143 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
5144 break;
5145 case VAR_DICT:
5146 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
5147 break;
5148 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02005149 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01005150 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01005151 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005152 case VAR_SPECIAL:
5153 case VAR_FLOAT:
5154 case VAR_FUNC:
5155 case VAR_PARTIAL:
5156 case VAR_JOB:
5157 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005158 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005159 break;
5160 }
5161}
5162
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005163 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01005164libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005165{
5166#ifdef FEAT_LIBCALL
5167 char_u *string_in;
5168 char_u **string_result;
5169 int nr_result;
5170#endif
5171
5172 rettv->v_type = type;
5173 if (type != VAR_NUMBER)
5174 rettv->vval.v_string = NULL;
5175
5176 if (check_restricted() || check_secure())
5177 return;
5178
5179#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005180 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005181 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
5182 {
5183 string_in = NULL;
5184 if (argvars[2].v_type == VAR_STRING)
5185 string_in = argvars[2].vval.v_string;
5186 if (type == VAR_NUMBER)
5187 string_result = NULL;
5188 else
5189 string_result = &rettv->vval.v_string;
5190 if (mch_libcall(argvars[0].vval.v_string,
5191 argvars[1].vval.v_string,
5192 string_in,
5193 argvars[2].vval.v_number,
5194 string_result,
5195 &nr_result) == OK
5196 && type == VAR_NUMBER)
5197 rettv->vval.v_number = nr_result;
5198 }
5199#endif
5200}
5201
5202/*
5203 * "libcall()" function
5204 */
5205 static void
5206f_libcall(typval_T *argvars, typval_T *rettv)
5207{
5208 libcall_common(argvars, rettv, VAR_STRING);
5209}
5210
5211/*
5212 * "libcallnr()" function
5213 */
5214 static void
5215f_libcallnr(typval_T *argvars, typval_T *rettv)
5216{
5217 libcall_common(argvars, rettv, VAR_NUMBER);
5218}
5219
5220/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005221 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005222 */
5223 static void
5224f_line(typval_T *argvars, typval_T *rettv)
5225{
5226 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005227 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005228 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005229 int id;
5230 tabpage_T *tp;
5231 win_T *wp;
5232 win_T *save_curwin;
5233 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005234
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005235 if (argvars[1].v_type != VAR_UNKNOWN)
5236 {
5237 // use window specified in the second argument
5238 id = (int)tv_get_number(&argvars[1]);
5239 wp = win_id2wp_tp(id, &tp);
5240 if (wp != NULL && tp != NULL)
5241 {
5242 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
5243 == OK)
5244 {
5245 check_cursor();
5246 fp = var2fpos(&argvars[0], TRUE, &fnum);
5247 }
5248 restore_win_noblock(save_curwin, save_curtab, TRUE);
5249 }
5250 }
5251 else
5252 // use current window
5253 fp = var2fpos(&argvars[0], TRUE, &fnum);
5254
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005255 if (fp != NULL)
5256 lnum = fp->lnum;
5257 rettv->vval.v_number = lnum;
5258}
5259
5260/*
5261 * "line2byte(lnum)" function
5262 */
5263 static void
5264f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
5265{
5266#ifndef FEAT_BYTEOFF
5267 rettv->vval.v_number = -1;
5268#else
5269 linenr_T lnum;
5270
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005271 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005272 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
5273 rettv->vval.v_number = -1;
5274 else
5275 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
5276 if (rettv->vval.v_number >= 0)
5277 ++rettv->vval.v_number;
5278#endif
5279}
5280
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005281#ifdef FEAT_FLOAT
5282/*
5283 * "log()" function
5284 */
5285 static void
5286f_log(typval_T *argvars, typval_T *rettv)
5287{
5288 float_T f = 0.0;
5289
5290 rettv->v_type = VAR_FLOAT;
5291 if (get_float_arg(argvars, &f) == OK)
5292 rettv->vval.v_float = log(f);
5293 else
5294 rettv->vval.v_float = 0.0;
5295}
5296
5297/*
5298 * "log10()" function
5299 */
5300 static void
5301f_log10(typval_T *argvars, typval_T *rettv)
5302{
5303 float_T f = 0.0;
5304
5305 rettv->v_type = VAR_FLOAT;
5306 if (get_float_arg(argvars, &f) == OK)
5307 rettv->vval.v_float = log10(f);
5308 else
5309 rettv->vval.v_float = 0.0;
5310}
5311#endif
5312
5313#ifdef FEAT_LUA
5314/*
5315 * "luaeval()" function
5316 */
5317 static void
5318f_luaeval(typval_T *argvars, typval_T *rettv)
5319{
5320 char_u *str;
5321 char_u buf[NUMBUFLEN];
5322
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005323 if (check_restricted() || check_secure())
5324 return;
5325
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005326 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005327 do_luaeval(str, argvars + 1, rettv);
5328}
5329#endif
5330
5331/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005332 * "maparg()" function
5333 */
5334 static void
5335f_maparg(typval_T *argvars, typval_T *rettv)
5336{
5337 get_maparg(argvars, rettv, TRUE);
5338}
5339
5340/*
5341 * "mapcheck()" function
5342 */
5343 static void
5344f_mapcheck(typval_T *argvars, typval_T *rettv)
5345{
5346 get_maparg(argvars, rettv, FALSE);
5347}
5348
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005349typedef enum
5350{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005351 MATCH_END, // matchend()
5352 MATCH_MATCH, // match()
5353 MATCH_STR, // matchstr()
5354 MATCH_LIST, // matchlist()
5355 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005356} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005357
5358 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005359find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005360{
5361 char_u *str = NULL;
5362 long len = 0;
5363 char_u *expr = NULL;
5364 char_u *pat;
5365 regmatch_T regmatch;
5366 char_u patbuf[NUMBUFLEN];
5367 char_u strbuf[NUMBUFLEN];
5368 char_u *save_cpo;
5369 long start = 0;
5370 long nth = 1;
5371 colnr_T startcol = 0;
5372 int match = 0;
5373 list_T *l = NULL;
5374 listitem_T *li = NULL;
5375 long idx = 0;
5376 char_u *tofree = NULL;
5377
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005378 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005379 save_cpo = p_cpo;
5380 p_cpo = (char_u *)"";
5381
5382 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005383 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005384 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005385 // type MATCH_LIST: return empty list when there are no matches.
5386 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005387 if (rettv_list_alloc(rettv) == FAIL)
5388 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005389 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005390 && (list_append_string(rettv->vval.v_list,
5391 (char_u *)"", 0) == FAIL
5392 || list_append_number(rettv->vval.v_list,
5393 (varnumber_T)-1) == FAIL
5394 || list_append_number(rettv->vval.v_list,
5395 (varnumber_T)-1) == FAIL
5396 || list_append_number(rettv->vval.v_list,
5397 (varnumber_T)-1) == FAIL))
5398 {
5399 list_free(rettv->vval.v_list);
5400 rettv->vval.v_list = NULL;
5401 goto theend;
5402 }
5403 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005404 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005405 {
5406 rettv->v_type = VAR_STRING;
5407 rettv->vval.v_string = NULL;
5408 }
5409
5410 if (argvars[0].v_type == VAR_LIST)
5411 {
5412 if ((l = argvars[0].vval.v_list) == NULL)
5413 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005414 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005415 li = l->lv_first;
5416 }
5417 else
5418 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005419 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005420 len = (long)STRLEN(str);
5421 }
5422
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005423 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005424 if (pat == NULL)
5425 goto theend;
5426
5427 if (argvars[2].v_type != VAR_UNKNOWN)
5428 {
5429 int error = FALSE;
5430
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005431 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005432 if (error)
5433 goto theend;
5434 if (l != NULL)
5435 {
5436 li = list_find(l, start);
5437 if (li == NULL)
5438 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005439 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005440 }
5441 else
5442 {
5443 if (start < 0)
5444 start = 0;
5445 if (start > len)
5446 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005447 // When "count" argument is there ignore matches before "start",
5448 // otherwise skip part of the string. Differs when pattern is "^"
5449 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005450 if (argvars[3].v_type != VAR_UNKNOWN)
5451 startcol = start;
5452 else
5453 {
5454 str += start;
5455 len -= start;
5456 }
5457 }
5458
5459 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005460 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005461 if (error)
5462 goto theend;
5463 }
5464
5465 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
5466 if (regmatch.regprog != NULL)
5467 {
5468 regmatch.rm_ic = p_ic;
5469
5470 for (;;)
5471 {
5472 if (l != NULL)
5473 {
5474 if (li == NULL)
5475 {
5476 match = FALSE;
5477 break;
5478 }
5479 vim_free(tofree);
5480 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
5481 if (str == NULL)
5482 break;
5483 }
5484
5485 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
5486
5487 if (match && --nth <= 0)
5488 break;
5489 if (l == NULL && !match)
5490 break;
5491
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005492 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005493 if (l != NULL)
5494 {
5495 li = li->li_next;
5496 ++idx;
5497 }
5498 else
5499 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005500 startcol = (colnr_T)(regmatch.startp[0]
5501 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005502 if (startcol > (colnr_T)len
5503 || str + startcol <= regmatch.startp[0])
5504 {
5505 match = FALSE;
5506 break;
5507 }
5508 }
5509 }
5510
5511 if (match)
5512 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005513 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005514 {
5515 listitem_T *li1 = rettv->vval.v_list->lv_first;
5516 listitem_T *li2 = li1->li_next;
5517 listitem_T *li3 = li2->li_next;
5518 listitem_T *li4 = li3->li_next;
5519
5520 vim_free(li1->li_tv.vval.v_string);
5521 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
5522 (int)(regmatch.endp[0] - regmatch.startp[0]));
5523 li3->li_tv.vval.v_number =
5524 (varnumber_T)(regmatch.startp[0] - expr);
5525 li4->li_tv.vval.v_number =
5526 (varnumber_T)(regmatch.endp[0] - expr);
5527 if (l != NULL)
5528 li2->li_tv.vval.v_number = (varnumber_T)idx;
5529 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005530 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005531 {
5532 int i;
5533
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005534 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005535 for (i = 0; i < NSUBEXP; ++i)
5536 {
5537 if (regmatch.endp[i] == NULL)
5538 {
5539 if (list_append_string(rettv->vval.v_list,
5540 (char_u *)"", 0) == FAIL)
5541 break;
5542 }
5543 else if (list_append_string(rettv->vval.v_list,
5544 regmatch.startp[i],
5545 (int)(regmatch.endp[i] - regmatch.startp[i]))
5546 == FAIL)
5547 break;
5548 }
5549 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005550 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005551 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005552 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005553 if (l != NULL)
5554 copy_tv(&li->li_tv, rettv);
5555 else
5556 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
5557 (int)(regmatch.endp[0] - regmatch.startp[0]));
5558 }
5559 else if (l != NULL)
5560 rettv->vval.v_number = idx;
5561 else
5562 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005563 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005564 rettv->vval.v_number =
5565 (varnumber_T)(regmatch.startp[0] - str);
5566 else
5567 rettv->vval.v_number =
5568 (varnumber_T)(regmatch.endp[0] - str);
5569 rettv->vval.v_number += (varnumber_T)(str - expr);
5570 }
5571 }
5572 vim_regfree(regmatch.regprog);
5573 }
5574
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005575theend:
5576 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005577 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005578 listitem_remove(rettv->vval.v_list,
5579 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005580 vim_free(tofree);
5581 p_cpo = save_cpo;
5582}
5583
5584/*
5585 * "match()" function
5586 */
5587 static void
5588f_match(typval_T *argvars, typval_T *rettv)
5589{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005590 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005591}
5592
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005593/*
5594 * "matchend()" function
5595 */
5596 static void
5597f_matchend(typval_T *argvars, typval_T *rettv)
5598{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005599 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005600}
5601
5602/*
5603 * "matchlist()" function
5604 */
5605 static void
5606f_matchlist(typval_T *argvars, typval_T *rettv)
5607{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005608 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005609}
5610
5611/*
5612 * "matchstr()" function
5613 */
5614 static void
5615f_matchstr(typval_T *argvars, typval_T *rettv)
5616{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005617 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005618}
5619
5620/*
5621 * "matchstrpos()" function
5622 */
5623 static void
5624f_matchstrpos(typval_T *argvars, typval_T *rettv)
5625{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02005626 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005627}
5628
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005629 static void
5630max_min(typval_T *argvars, typval_T *rettv, int domax)
5631{
5632 varnumber_T n = 0;
5633 varnumber_T i;
5634 int error = FALSE;
5635
5636 if (argvars[0].v_type == VAR_LIST)
5637 {
5638 list_T *l;
5639 listitem_T *li;
5640
5641 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005642 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005643 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005644 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005645 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005646 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
5647 n = l->lv_u.nonmat.lv_start;
5648 else
5649 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
5650 * l->lv_u.nonmat.lv_stride;
5651 }
5652 else
5653 {
5654 li = l->lv_first;
5655 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005656 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01005657 n = tv_get_number_chk(&li->li_tv, &error);
5658 for (;;)
5659 {
5660 li = li->li_next;
5661 if (li == NULL)
5662 break;
5663 i = tv_get_number_chk(&li->li_tv, &error);
5664 if (domax ? i > n : i < n)
5665 n = i;
5666 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005667 }
5668 }
5669 }
5670 }
5671 else if (argvars[0].v_type == VAR_DICT)
5672 {
5673 dict_T *d;
5674 int first = TRUE;
5675 hashitem_T *hi;
5676 int todo;
5677
5678 d = argvars[0].vval.v_dict;
5679 if (d != NULL)
5680 {
5681 todo = (int)d->dv_hashtab.ht_used;
5682 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
5683 {
5684 if (!HASHITEM_EMPTY(hi))
5685 {
5686 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005687 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005688 if (first)
5689 {
5690 n = i;
5691 first = FALSE;
5692 }
5693 else if (domax ? i > n : i < n)
5694 n = i;
5695 }
5696 }
5697 }
5698 }
5699 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005700 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005701 rettv->vval.v_number = error ? 0 : n;
5702}
5703
5704/*
5705 * "max()" function
5706 */
5707 static void
5708f_max(typval_T *argvars, typval_T *rettv)
5709{
5710 max_min(argvars, rettv, TRUE);
5711}
5712
5713/*
5714 * "min()" function
5715 */
5716 static void
5717f_min(typval_T *argvars, typval_T *rettv)
5718{
5719 max_min(argvars, rettv, FALSE);
5720}
5721
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005722#if defined(FEAT_MZSCHEME) || defined(PROTO)
5723/*
5724 * "mzeval()" function
5725 */
5726 static void
5727f_mzeval(typval_T *argvars, typval_T *rettv)
5728{
5729 char_u *str;
5730 char_u buf[NUMBUFLEN];
5731
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005732 if (check_restricted() || check_secure())
5733 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005734 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005735 do_mzeval(str, rettv);
5736}
5737
5738 void
5739mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
5740{
5741 typval_T argvars[3];
5742
5743 argvars[0].v_type = VAR_STRING;
5744 argvars[0].vval.v_string = name;
5745 copy_tv(args, &argvars[1]);
5746 argvars[2].v_type = VAR_UNKNOWN;
5747 f_call(argvars, rettv);
5748 clear_tv(&argvars[1]);
5749}
5750#endif
5751
5752/*
5753 * "nextnonblank()" function
5754 */
5755 static void
5756f_nextnonblank(typval_T *argvars, typval_T *rettv)
5757{
5758 linenr_T lnum;
5759
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005760 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005761 {
5762 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
5763 {
5764 lnum = 0;
5765 break;
5766 }
5767 if (*skipwhite(ml_get(lnum)) != NUL)
5768 break;
5769 }
5770 rettv->vval.v_number = lnum;
5771}
5772
5773/*
5774 * "nr2char()" function
5775 */
5776 static void
5777f_nr2char(typval_T *argvars, typval_T *rettv)
5778{
5779 char_u buf[NUMBUFLEN];
5780
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005781 if (has_mbyte)
5782 {
5783 int utf8 = 0;
5784
5785 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005786 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005787 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01005788 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005789 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005790 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005791 }
5792 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005793 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005794 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005795 buf[1] = NUL;
5796 }
5797 rettv->v_type = VAR_STRING;
5798 rettv->vval.v_string = vim_strsave(buf);
5799}
5800
5801/*
5802 * "or(expr, expr)" function
5803 */
5804 static void
5805f_or(typval_T *argvars, typval_T *rettv)
5806{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005807 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
5808 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005809}
5810
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005811#ifdef FEAT_PERL
5812/*
5813 * "perleval()" function
5814 */
5815 static void
5816f_perleval(typval_T *argvars, typval_T *rettv)
5817{
5818 char_u *str;
5819 char_u buf[NUMBUFLEN];
5820
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005821 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005822 do_perleval(str, rettv);
5823}
5824#endif
5825
5826#ifdef FEAT_FLOAT
5827/*
5828 * "pow()" function
5829 */
5830 static void
5831f_pow(typval_T *argvars, typval_T *rettv)
5832{
5833 float_T fx = 0.0, fy = 0.0;
5834
5835 rettv->v_type = VAR_FLOAT;
5836 if (get_float_arg(argvars, &fx) == OK
5837 && get_float_arg(&argvars[1], &fy) == OK)
5838 rettv->vval.v_float = pow(fx, fy);
5839 else
5840 rettv->vval.v_float = 0.0;
5841}
5842#endif
5843
5844/*
5845 * "prevnonblank()" function
5846 */
5847 static void
5848f_prevnonblank(typval_T *argvars, typval_T *rettv)
5849{
5850 linenr_T lnum;
5851
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005852 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005853 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
5854 lnum = 0;
5855 else
5856 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
5857 --lnum;
5858 rettv->vval.v_number = lnum;
5859}
5860
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005861// This dummy va_list is here because:
5862// - passing a NULL pointer doesn't work when va_list isn't a pointer
5863// - locally in the function results in a "used before set" warning
5864// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005865static va_list ap;
5866
5867/*
5868 * "printf()" function
5869 */
5870 static void
5871f_printf(typval_T *argvars, typval_T *rettv)
5872{
5873 char_u buf[NUMBUFLEN];
5874 int len;
5875 char_u *s;
5876 int saved_did_emsg = did_emsg;
5877 char *fmt;
5878
5879 rettv->v_type = VAR_STRING;
5880 rettv->vval.v_string = NULL;
5881
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005882 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005883 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005884 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005885 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005886 if (!did_emsg)
5887 {
5888 s = alloc(len + 1);
5889 if (s != NULL)
5890 {
5891 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02005892 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
5893 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005894 }
5895 }
5896 did_emsg |= saved_did_emsg;
5897}
5898
5899/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005900 * "pum_getpos()" function
5901 */
5902 static void
5903f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5904{
5905 if (rettv_dict_alloc(rettv) != OK)
5906 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005907 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02005908}
5909
5910/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005911 * "pumvisible()" function
5912 */
5913 static void
5914f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5915{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005916 if (pum_visible())
5917 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005918}
5919
5920#ifdef FEAT_PYTHON3
5921/*
5922 * "py3eval()" function
5923 */
5924 static void
5925f_py3eval(typval_T *argvars, typval_T *rettv)
5926{
5927 char_u *str;
5928 char_u buf[NUMBUFLEN];
5929
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005930 if (check_restricted() || check_secure())
5931 return;
5932
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005933 if (p_pyx == 0)
5934 p_pyx = 3;
5935
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005936 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005937 do_py3eval(str, rettv);
5938}
5939#endif
5940
5941#ifdef FEAT_PYTHON
5942/*
5943 * "pyeval()" function
5944 */
5945 static void
5946f_pyeval(typval_T *argvars, typval_T *rettv)
5947{
5948 char_u *str;
5949 char_u buf[NUMBUFLEN];
5950
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005951 if (check_restricted() || check_secure())
5952 return;
5953
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005954 if (p_pyx == 0)
5955 p_pyx = 2;
5956
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005957 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005958 do_pyeval(str, rettv);
5959}
5960#endif
5961
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005962#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
5963/*
5964 * "pyxeval()" function
5965 */
5966 static void
5967f_pyxeval(typval_T *argvars, typval_T *rettv)
5968{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01005969 if (check_restricted() || check_secure())
5970 return;
5971
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005972# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
5973 init_pyxversion();
5974 if (p_pyx == 2)
5975 f_pyeval(argvars, rettv);
5976 else
5977 f_py3eval(argvars, rettv);
5978# elif defined(FEAT_PYTHON)
5979 f_pyeval(argvars, rettv);
5980# elif defined(FEAT_PYTHON3)
5981 f_py3eval(argvars, rettv);
5982# endif
5983}
5984#endif
5985
Bram Moolenaar4f645c52020-02-08 16:40:39 +01005986static UINT32_T srand_seed_for_testing = 0;
5987static int srand_seed_for_testing_is_used = FALSE;
5988
5989 static void
5990f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
5991{
5992 if (argvars[0].v_type == VAR_UNKNOWN)
5993 srand_seed_for_testing_is_used = FALSE;
5994 else
5995 {
5996 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
5997 srand_seed_for_testing_is_used = TRUE;
5998 }
5999}
6000
6001 static void
6002init_srand(UINT32_T *x)
6003{
6004#ifndef MSWIN
6005 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6006#endif
6007
6008 if (srand_seed_for_testing_is_used)
6009 {
6010 *x = srand_seed_for_testing;
6011 return;
6012 }
6013#ifndef MSWIN
6014 if (dev_urandom_state != FAIL)
6015 {
6016 int fd = open("/dev/urandom", O_RDONLY);
6017 struct {
6018 union {
6019 UINT32_T number;
6020 char bytes[sizeof(UINT32_T)];
6021 } contents;
6022 } buf;
6023
6024 // Attempt reading /dev/urandom.
6025 if (fd == -1)
6026 dev_urandom_state = FAIL;
6027 else
6028 {
6029 buf.contents.number = 0;
6030 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6031 != sizeof(UINT32_T))
6032 dev_urandom_state = FAIL;
6033 else
6034 {
6035 dev_urandom_state = OK;
6036 *x = buf.contents.number;
6037 }
6038 close(fd);
6039 }
6040 }
6041 if (dev_urandom_state != OK)
6042 // Reading /dev/urandom doesn't work, fall back to time().
6043#endif
6044 *x = vim_time();
6045}
6046
6047#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6048#define SPLITMIX32(x, z) ( \
6049 z = (x += 0x9e3779b9), \
6050 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6051 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6052 z ^ (z >> 16) \
6053 )
6054#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6055 result = ROTL(y * 5, 7) * 9; \
6056 t = y << 9; \
6057 z ^= x; \
6058 w ^= y; \
6059 y ^= z, x ^= w; \
6060 z ^= t; \
6061 w = ROTL(w, 11);
6062
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006063/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006064 * "rand()" function
6065 */
6066 static void
6067f_rand(typval_T *argvars, typval_T *rettv)
6068{
6069 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006070 static UINT32_T gx, gy, gz, gw;
6071 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006072 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006073 UINT32_T x, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006074
6075 if (argvars[0].v_type == VAR_UNKNOWN)
6076 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006077 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006078 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006079 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006080 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006081 init_srand(&x);
6082
6083 gx = SPLITMIX32(x, z);
6084 gy = SPLITMIX32(x, z);
6085 gz = SPLITMIX32(x, z);
6086 gw = SPLITMIX32(x, z);
6087 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006088 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006089
6090 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006091 }
6092 else if (argvars[0].v_type == VAR_LIST)
6093 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006094 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006095 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006096 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006097
6098 lx = list_find(l, 0L);
6099 ly = list_find(l, 1L);
6100 lz = list_find(l, 2L);
6101 lw = list_find(l, 3L);
6102 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
6103 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
6104 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
6105 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
6106 x = (UINT32_T)lx->li_tv.vval.v_number;
6107 y = (UINT32_T)ly->li_tv.vval.v_number;
6108 z = (UINT32_T)lz->li_tv.vval.v_number;
6109 w = (UINT32_T)lw->li_tv.vval.v_number;
6110
6111 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
6112
6113 lx->li_tv.vval.v_number = (varnumber_T)x;
6114 ly->li_tv.vval.v_number = (varnumber_T)y;
6115 lz->li_tv.vval.v_number = (varnumber_T)z;
6116 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006117 }
6118 else
6119 goto theend;
6120
6121 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006122 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006123 return;
6124
6125theend:
6126 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006127 rettv->v_type = VAR_NUMBER;
6128 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006129}
6130
6131/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006132 * "srand()" function
6133 */
6134 static void
6135f_srand(typval_T *argvars, typval_T *rettv)
6136{
6137 UINT32_T x = 0, z;
6138
6139 if (rettv_list_alloc(rettv) == FAIL)
6140 return;
6141 if (argvars[0].v_type == VAR_UNKNOWN)
6142 {
6143 init_srand(&x);
6144 }
6145 else
6146 {
6147 int error = FALSE;
6148
6149 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
6150 if (error)
6151 return;
6152 }
6153
6154 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6155 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6156 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6157 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6158}
6159
6160#undef ROTL
6161#undef SPLITMIX32
6162#undef SHUFFLE_XOSHIRO128STARSTAR
6163
6164/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006165 * "range()" function
6166 */
6167 static void
6168f_range(typval_T *argvars, typval_T *rettv)
6169{
6170 varnumber_T start;
6171 varnumber_T end;
6172 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006173 int error = FALSE;
6174
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006175 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006176 if (argvars[1].v_type == VAR_UNKNOWN)
6177 {
6178 end = start - 1;
6179 start = 0;
6180 }
6181 else
6182 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006183 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006184 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006185 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006186 }
6187
6188 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006189 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006190 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006191 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006192 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006193 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006194 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006195 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006196 list_T *list = rettv->vval.v_list;
6197
6198 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006199 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006200 // be called.
6201 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006202 list->lv_u.nonmat.lv_start = start;
6203 list->lv_u.nonmat.lv_end = end;
6204 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01006205 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006206 }
6207}
6208
6209/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006210 * Materialize "list".
6211 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006212 */
6213 void
6214range_list_materialize(list_T *list)
6215{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006216 varnumber_T start = list->lv_u.nonmat.lv_start;
6217 varnumber_T end = list->lv_u.nonmat.lv_end;
6218 int stride = list->lv_u.nonmat.lv_stride;
6219 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006220
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006221 list->lv_first = NULL;
6222 list->lv_u.mat.lv_last = NULL;
6223 list->lv_len = 0;
6224 list->lv_u.mat.lv_idx_item = NULL;
6225 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
6226 if (list_append_number(list, (varnumber_T)i) == FAIL)
6227 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006228}
6229
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02006230 static void
6231return_register(int regname, typval_T *rettv)
6232{
6233 char_u buf[2] = {0, 0};
6234
6235 buf[0] = (char_u)regname;
6236 rettv->v_type = VAR_STRING;
6237 rettv->vval.v_string = vim_strsave(buf);
6238}
6239
6240/*
6241 * "reg_executing()" function
6242 */
6243 static void
6244f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
6245{
6246 return_register(reg_executing, rettv);
6247}
6248
6249/*
6250 * "reg_recording()" function
6251 */
6252 static void
6253f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
6254{
6255 return_register(reg_recording, rettv);
6256}
6257
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01006258/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006259 * "rename({from}, {to})" function
6260 */
6261 static void
6262f_rename(typval_T *argvars, typval_T *rettv)
6263{
6264 char_u buf[NUMBUFLEN];
6265
6266 if (check_restricted() || check_secure())
6267 rettv->vval.v_number = -1;
6268 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006269 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
6270 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006271}
6272
6273/*
6274 * "repeat()" function
6275 */
6276 static void
6277f_repeat(typval_T *argvars, typval_T *rettv)
6278{
6279 char_u *p;
6280 int n;
6281 int slen;
6282 int len;
6283 char_u *r;
6284 int i;
6285
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006286 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006287 if (argvars[0].v_type == VAR_LIST)
6288 {
6289 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
6290 while (n-- > 0)
6291 if (list_extend(rettv->vval.v_list,
6292 argvars[0].vval.v_list, NULL) == FAIL)
6293 break;
6294 }
6295 else
6296 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006297 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006298 rettv->v_type = VAR_STRING;
6299 rettv->vval.v_string = NULL;
6300
6301 slen = (int)STRLEN(p);
6302 len = slen * n;
6303 if (len <= 0)
6304 return;
6305
6306 r = alloc(len + 1);
6307 if (r != NULL)
6308 {
6309 for (i = 0; i < n; i++)
6310 mch_memmove(r + i * slen, p, (size_t)slen);
6311 r[len] = NUL;
6312 }
6313
6314 rettv->vval.v_string = r;
6315 }
6316}
6317
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006318#define SP_NOMOVE 0x01 // don't move cursor
6319#define SP_REPEAT 0x02 // repeat to find outer pair
6320#define SP_RETCOUNT 0x04 // return matchcount
6321#define SP_SETPCMARK 0x08 // set previous context mark
6322#define SP_START 0x10 // accept match at start position
6323#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
6324#define SP_END 0x40 // leave cursor at end of match
6325#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006326
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006327/*
6328 * Get flags for a search function.
6329 * Possibly sets "p_ws".
6330 * Returns BACKWARD, FORWARD or zero (for an error).
6331 */
6332 static int
6333get_search_arg(typval_T *varp, int *flagsp)
6334{
6335 int dir = FORWARD;
6336 char_u *flags;
6337 char_u nbuf[NUMBUFLEN];
6338 int mask;
6339
6340 if (varp->v_type != VAR_UNKNOWN)
6341 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006342 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006343 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006344 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006345 while (*flags != NUL)
6346 {
6347 switch (*flags)
6348 {
6349 case 'b': dir = BACKWARD; break;
6350 case 'w': p_ws = TRUE; break;
6351 case 'W': p_ws = FALSE; break;
6352 default: mask = 0;
6353 if (flagsp != NULL)
6354 switch (*flags)
6355 {
6356 case 'c': mask = SP_START; break;
6357 case 'e': mask = SP_END; break;
6358 case 'm': mask = SP_RETCOUNT; break;
6359 case 'n': mask = SP_NOMOVE; break;
6360 case 'p': mask = SP_SUBPAT; break;
6361 case 'r': mask = SP_REPEAT; break;
6362 case 's': mask = SP_SETPCMARK; break;
6363 case 'z': mask = SP_COLUMN; break;
6364 }
6365 if (mask == 0)
6366 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006367 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006368 dir = 0;
6369 }
6370 else
6371 *flagsp |= mask;
6372 }
6373 if (dir == 0)
6374 break;
6375 ++flags;
6376 }
6377 }
6378 return dir;
6379}
6380
6381/*
6382 * Shared by search() and searchpos() functions.
6383 */
6384 static int
6385search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
6386{
6387 int flags;
6388 char_u *pat;
6389 pos_T pos;
6390 pos_T save_cursor;
6391 int save_p_ws = p_ws;
6392 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006393 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006394 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006395#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006396 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006397 long time_limit = 0;
6398#endif
6399 int options = SEARCH_KEEP;
6400 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006401 searchit_arg_T sia;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006402
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006403 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006404 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006405 if (dir == 0)
6406 goto theend;
6407 flags = *flagsp;
6408 if (flags & SP_START)
6409 options |= SEARCH_START;
6410 if (flags & SP_END)
6411 options |= SEARCH_END;
6412 if (flags & SP_COLUMN)
6413 options |= SEARCH_COL;
6414
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006415 // Optional arguments: line number to stop searching and timeout.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006416 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
6417 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006418 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006419 if (lnum_stop < 0)
6420 goto theend;
6421#ifdef FEAT_RELTIME
6422 if (argvars[3].v_type != VAR_UNKNOWN)
6423 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006424 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006425 if (time_limit < 0)
6426 goto theend;
6427 }
6428#endif
6429 }
6430
6431#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006432 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006433 profile_setlimit(time_limit, &tm);
6434#endif
6435
6436 /*
6437 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
6438 * Check to make sure only those flags are set.
6439 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
6440 * flags cannot be set. Check for that condition also.
6441 */
6442 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
6443 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6444 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006445 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006446 goto theend;
6447 }
6448
6449 pos = save_cursor = curwin->w_cursor;
Bram Moolenaara80faa82020-04-12 19:37:17 +02006450 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006451 sia.sa_stop_lnum = (linenr_T)lnum_stop;
6452#ifdef FEAT_RELTIME
6453 sia.sa_tm = &tm;
6454#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006455 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006456 options, RE_SEARCH, &sia);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006457 if (subpatnum != FAIL)
6458 {
6459 if (flags & SP_SUBPAT)
6460 retval = subpatnum;
6461 else
6462 retval = pos.lnum;
6463 if (flags & SP_SETPCMARK)
6464 setpcmark();
6465 curwin->w_cursor = pos;
6466 if (match_pos != NULL)
6467 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006468 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006469 match_pos->lnum = pos.lnum;
6470 match_pos->col = pos.col + 1;
6471 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006472 // "/$" will put the cursor after the end of the line, may need to
6473 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006474 check_cursor();
6475 }
6476
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006477 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006478 if (flags & SP_NOMOVE)
6479 curwin->w_cursor = save_cursor;
6480 else
6481 curwin->w_set_curswant = TRUE;
6482theend:
6483 p_ws = save_p_ws;
6484
6485 return retval;
6486}
6487
6488#ifdef FEAT_FLOAT
6489
6490/*
6491 * round() is not in C90, use ceil() or floor() instead.
6492 */
6493 float_T
6494vim_round(float_T f)
6495{
6496 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
6497}
6498
6499/*
6500 * "round({float})" function
6501 */
6502 static void
6503f_round(typval_T *argvars, typval_T *rettv)
6504{
6505 float_T f = 0.0;
6506
6507 rettv->v_type = VAR_FLOAT;
6508 if (get_float_arg(argvars, &f) == OK)
6509 rettv->vval.v_float = vim_round(f);
6510 else
6511 rettv->vval.v_float = 0.0;
6512}
6513#endif
6514
Bram Moolenaare99be0e2019-03-26 22:51:09 +01006515#ifdef FEAT_RUBY
6516/*
6517 * "rubyeval()" function
6518 */
6519 static void
6520f_rubyeval(typval_T *argvars, typval_T *rettv)
6521{
6522 char_u *str;
6523 char_u buf[NUMBUFLEN];
6524
6525 str = tv_get_string_buf(&argvars[0], buf);
6526 do_rubyeval(str, rettv);
6527}
6528#endif
6529
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006530/*
6531 * "screenattr()" function
6532 */
6533 static void
6534f_screenattr(typval_T *argvars, typval_T *rettv)
6535{
6536 int row;
6537 int col;
6538 int c;
6539
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006540 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6541 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006542 if (row < 0 || row >= screen_Rows
6543 || col < 0 || col >= screen_Columns)
6544 c = -1;
6545 else
6546 c = ScreenAttrs[LineOffset[row] + col];
6547 rettv->vval.v_number = c;
6548}
6549
6550/*
6551 * "screenchar()" function
6552 */
6553 static void
6554f_screenchar(typval_T *argvars, typval_T *rettv)
6555{
6556 int row;
6557 int col;
6558 int off;
6559 int c;
6560
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006561 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6562 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006563 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006564 c = -1;
6565 else
6566 {
6567 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006568 if (enc_utf8 && ScreenLinesUC[off] != 0)
6569 c = ScreenLinesUC[off];
6570 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006571 c = ScreenLines[off];
6572 }
6573 rettv->vval.v_number = c;
6574}
6575
6576/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006577 * "screenchars()" function
6578 */
6579 static void
6580f_screenchars(typval_T *argvars, typval_T *rettv)
6581{
6582 int row;
6583 int col;
6584 int off;
6585 int c;
6586 int i;
6587
6588 if (rettv_list_alloc(rettv) == FAIL)
6589 return;
6590 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6591 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6592 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6593 return;
6594
6595 off = LineOffset[row] + col;
6596 if (enc_utf8 && ScreenLinesUC[off] != 0)
6597 c = ScreenLinesUC[off];
6598 else
6599 c = ScreenLines[off];
6600 list_append_number(rettv->vval.v_list, (varnumber_T)c);
6601
6602 if (enc_utf8)
6603
6604 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6605 list_append_number(rettv->vval.v_list,
6606 (varnumber_T)ScreenLinesC[i][off]);
6607}
6608
6609/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006610 * "screencol()" function
6611 *
6612 * First column is 1 to be consistent with virtcol().
6613 */
6614 static void
6615f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
6616{
6617 rettv->vval.v_number = screen_screencol() + 1;
6618}
6619
6620/*
6621 * "screenrow()" function
6622 */
6623 static void
6624f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
6625{
6626 rettv->vval.v_number = screen_screenrow() + 1;
6627}
6628
6629/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01006630 * "screenstring()" function
6631 */
6632 static void
6633f_screenstring(typval_T *argvars, typval_T *rettv)
6634{
6635 int row;
6636 int col;
6637 int off;
6638 int c;
6639 int i;
6640 char_u buf[MB_MAXBYTES + 1];
6641 int buflen = 0;
6642
6643 rettv->vval.v_string = NULL;
6644 rettv->v_type = VAR_STRING;
6645
6646 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
6647 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
6648 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
6649 return;
6650
6651 off = LineOffset[row] + col;
6652 if (enc_utf8 && ScreenLinesUC[off] != 0)
6653 c = ScreenLinesUC[off];
6654 else
6655 c = ScreenLines[off];
6656 buflen += mb_char2bytes(c, buf);
6657
6658 if (enc_utf8)
6659 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
6660 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
6661
6662 buf[buflen] = NUL;
6663 rettv->vval.v_string = vim_strsave(buf);
6664}
6665
6666/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006667 * "search()" function
6668 */
6669 static void
6670f_search(typval_T *argvars, typval_T *rettv)
6671{
6672 int flags = 0;
6673
6674 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
6675}
6676
6677/*
6678 * "searchdecl()" function
6679 */
6680 static void
6681f_searchdecl(typval_T *argvars, typval_T *rettv)
6682{
6683 int locally = 1;
6684 int thisblock = 0;
6685 int error = FALSE;
6686 char_u *name;
6687
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006688 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006689
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006690 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006691 if (argvars[1].v_type != VAR_UNKNOWN)
6692 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006693 locally = (int)tv_get_number_chk(&argvars[1], &error) == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006694 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006695 thisblock = (int)tv_get_number_chk(&argvars[2], &error) != 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006696 }
6697 if (!error && name != NULL)
6698 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
6699 locally, thisblock, SEARCH_KEEP) == FAIL;
6700}
6701
6702/*
6703 * Used by searchpair() and searchpairpos()
6704 */
6705 static int
6706searchpair_cmn(typval_T *argvars, pos_T *match_pos)
6707{
6708 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01006709 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006710 int save_p_ws = p_ws;
6711 int dir;
6712 int flags = 0;
6713 char_u nbuf1[NUMBUFLEN];
6714 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006715 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006716 long lnum_stop = 0;
6717 long time_limit = 0;
6718
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006719 // Get the three pattern arguments: start, middle, end. Will result in an
6720 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006721 spat = tv_get_string_chk(&argvars[0]);
6722 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
6723 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006724 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006725 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006726
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006727 // Handle the optional fourth argument: flags
6728 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006729 if (dir == 0)
6730 goto theend;
6731
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006732 // Don't accept SP_END or SP_SUBPAT.
6733 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006734 if ((flags & (SP_END | SP_SUBPAT)) != 0
6735 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
6736 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006737 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006738 goto theend;
6739 }
6740
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006741 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006742 if (flags & SP_REPEAT)
6743 p_ws = FALSE;
6744
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006745 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006746 if (argvars[3].v_type == VAR_UNKNOWN
6747 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01006748 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006749 else
6750 {
Bram Moolenaar48570482017-10-30 21:48:41 +01006751 skip = &argvars[4];
6752 if (skip->v_type != VAR_FUNC && skip->v_type != VAR_PARTIAL
6753 && skip->v_type != VAR_STRING)
6754 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006755 // Type error
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006756 semsg(_(e_invarg2), tv_get_string(&argvars[4]));
Bram Moolenaar48570482017-10-30 21:48:41 +01006757 goto theend;
6758 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006759 if (argvars[5].v_type != VAR_UNKNOWN)
6760 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006761 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006762 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006763 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006764 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006765 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006766 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006767#ifdef FEAT_RELTIME
6768 if (argvars[6].v_type != VAR_UNKNOWN)
6769 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006770 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006771 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006772 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006773 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006774 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02006775 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006776 }
6777#endif
6778 }
6779 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006780
6781 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
6782 match_pos, lnum_stop, time_limit);
6783
6784theend:
6785 p_ws = save_p_ws;
6786
6787 return retval;
6788}
6789
6790/*
6791 * "searchpair()" function
6792 */
6793 static void
6794f_searchpair(typval_T *argvars, typval_T *rettv)
6795{
6796 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
6797}
6798
6799/*
6800 * "searchpairpos()" function
6801 */
6802 static void
6803f_searchpairpos(typval_T *argvars, typval_T *rettv)
6804{
6805 pos_T match_pos;
6806 int lnum = 0;
6807 int col = 0;
6808
6809 if (rettv_list_alloc(rettv) == FAIL)
6810 return;
6811
6812 if (searchpair_cmn(argvars, &match_pos) > 0)
6813 {
6814 lnum = match_pos.lnum;
6815 col = match_pos.col;
6816 }
6817
6818 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
6819 list_append_number(rettv->vval.v_list, (varnumber_T)col);
6820}
6821
6822/*
6823 * Search for a start/middle/end thing.
6824 * Used by searchpair(), see its documentation for the details.
6825 * Returns 0 or -1 for no match,
6826 */
6827 long
6828do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006829 char_u *spat, // start pattern
6830 char_u *mpat, // middle pattern
6831 char_u *epat, // end pattern
6832 int dir, // BACKWARD or FORWARD
6833 typval_T *skip, // skip expression
6834 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006835 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006836 linenr_T lnum_stop, // stop at this line if not zero
6837 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006838{
6839 char_u *save_cpo;
6840 char_u *pat, *pat2 = NULL, *pat3 = NULL;
6841 long retval = 0;
6842 pos_T pos;
6843 pos_T firstpos;
6844 pos_T foundpos;
6845 pos_T save_cursor;
6846 pos_T save_pos;
6847 int n;
6848 int r;
6849 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01006850 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006851 int err;
6852 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006853#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006854 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006855#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006856
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006857 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006858 save_cpo = p_cpo;
6859 p_cpo = empty_option;
6860
6861#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006862 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006863 profile_setlimit(time_limit, &tm);
6864#endif
6865
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006866 // Make two search patterns: start/end (pat2, for in nested pairs) and
6867 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02006868 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
6869 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006870 if (pat2 == NULL || pat3 == NULL)
6871 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006872 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006873 if (*mpat == NUL)
6874 STRCPY(pat3, pat2);
6875 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01006876 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006877 spat, epat, mpat);
6878 if (flags & SP_START)
6879 options |= SEARCH_START;
6880
Bram Moolenaar48570482017-10-30 21:48:41 +01006881 if (skip != NULL)
6882 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006883 // Empty string means to not use the skip expression.
Bram Moolenaar48570482017-10-30 21:48:41 +01006884 if (skip->v_type == VAR_STRING || skip->v_type == VAR_FUNC)
6885 use_skip = skip->vval.v_string != NULL
6886 && *skip->vval.v_string != NUL;
6887 }
6888
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006889 save_cursor = curwin->w_cursor;
6890 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006891 CLEAR_POS(&firstpos);
6892 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006893 pat = pat3;
6894 for (;;)
6895 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006896 searchit_arg_T sia;
6897
Bram Moolenaara80faa82020-04-12 19:37:17 +02006898 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006899 sia.sa_stop_lnum = lnum_stop;
6900#ifdef FEAT_RELTIME
6901 sia.sa_tm = &tm;
6902#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01006903 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02006904 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006905 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006906 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006907 break;
6908
6909 if (firstpos.lnum == 0)
6910 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01006911 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006912 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006913 // Found the same position again. Can happen with a pattern that
6914 // has "\zs" at the end and searching backwards. Advance one
6915 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006916 if (dir == BACKWARD)
6917 decl(&pos);
6918 else
6919 incl(&pos);
6920 }
6921 foundpos = pos;
6922
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006923 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006924 options &= ~SEARCH_START;
6925
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006926 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01006927 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006928 {
6929 save_pos = curwin->w_cursor;
6930 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01006931 err = FALSE;
6932 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006933 curwin->w_cursor = save_pos;
6934 if (err)
6935 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006936 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006937 curwin->w_cursor = save_cursor;
6938 retval = -1;
6939 break;
6940 }
6941 if (r)
6942 continue;
6943 }
6944
6945 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
6946 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006947 // Found end when searching backwards or start when searching
6948 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006949 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006950 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006951 }
6952 else
6953 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006954 // Found end when searching forward or start when searching
6955 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006956 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006957 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006958 }
6959
6960 if (nest == 0)
6961 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006962 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006963 if (flags & SP_RETCOUNT)
6964 ++retval;
6965 else
6966 retval = pos.lnum;
6967 if (flags & SP_SETPCMARK)
6968 setpcmark();
6969 curwin->w_cursor = pos;
6970 if (!(flags & SP_REPEAT))
6971 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006972 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006973 }
6974 }
6975
6976 if (match_pos != NULL)
6977 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006978 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006979 match_pos->lnum = curwin->w_cursor.lnum;
6980 match_pos->col = curwin->w_cursor.col + 1;
6981 }
6982
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006983 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006984 if ((flags & SP_NOMOVE) || retval == 0)
6985 curwin->w_cursor = save_cursor;
6986
6987theend:
6988 vim_free(pat2);
6989 vim_free(pat3);
6990 if (p_cpo == empty_option)
6991 p_cpo = save_cpo;
6992 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006993 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006994 free_string_option(save_cpo);
6995
6996 return retval;
6997}
6998
6999/*
7000 * "searchpos()" function
7001 */
7002 static void
7003f_searchpos(typval_T *argvars, typval_T *rettv)
7004{
7005 pos_T match_pos;
7006 int lnum = 0;
7007 int col = 0;
7008 int n;
7009 int flags = 0;
7010
7011 if (rettv_list_alloc(rettv) == FAIL)
7012 return;
7013
7014 n = search_cmn(argvars, &match_pos, &flags);
7015 if (n > 0)
7016 {
7017 lnum = match_pos.lnum;
7018 col = match_pos.col;
7019 }
7020
7021 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7022 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7023 if (flags & SP_SUBPAT)
7024 list_append_number(rettv->vval.v_list, (varnumber_T)n);
7025}
7026
7027 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007028f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
7029{
7030 dict_T *d;
7031 dictitem_T *di;
7032 char_u *csearch;
7033
7034 if (argvars[0].v_type != VAR_DICT)
7035 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007036 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007037 return;
7038 }
7039
7040 if ((d = argvars[0].vval.v_dict) != NULL)
7041 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01007042 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007043 if (csearch != NULL)
7044 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007045 if (enc_utf8)
7046 {
7047 int pcc[MAX_MCO];
7048 int c = utfc_ptr2char(csearch, pcc);
7049
7050 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
7051 }
7052 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007053 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02007054 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007055 }
7056
7057 di = dict_find(d, (char_u *)"forward", -1);
7058 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007059 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007060 ? FORWARD : BACKWARD);
7061
7062 di = dict_find(d, (char_u *)"until", -1);
7063 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007064 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007065 }
7066}
7067
7068/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02007069 * "setenv()" function
7070 */
7071 static void
7072f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
7073{
7074 char_u namebuf[NUMBUFLEN];
7075 char_u valbuf[NUMBUFLEN];
7076 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
7077
7078 if (argvars[1].v_type == VAR_SPECIAL
7079 && argvars[1].vval.v_number == VVAL_NULL)
7080 vim_unsetenv(name);
7081 else
7082 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
7083}
7084
7085/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007086 * "setfperm({fname}, {mode})" function
7087 */
7088 static void
7089f_setfperm(typval_T *argvars, typval_T *rettv)
7090{
7091 char_u *fname;
7092 char_u modebuf[NUMBUFLEN];
7093 char_u *mode_str;
7094 int i;
7095 int mask;
7096 int mode = 0;
7097
7098 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007099 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007100 if (fname == NULL)
7101 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007102 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007103 if (mode_str == NULL)
7104 return;
7105 if (STRLEN(mode_str) != 9)
7106 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007107 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007108 return;
7109 }
7110
7111 mask = 1;
7112 for (i = 8; i >= 0; --i)
7113 {
7114 if (mode_str[i] != '-')
7115 mode |= mask;
7116 mask = mask << 1;
7117 }
7118 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
7119}
7120
7121/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007122 * "setpos()" function
7123 */
7124 static void
7125f_setpos(typval_T *argvars, typval_T *rettv)
7126{
7127 pos_T pos;
7128 int fnum;
7129 char_u *name;
7130 colnr_T curswant = -1;
7131
7132 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007133 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007134 if (name != NULL)
7135 {
7136 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
7137 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01007138 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007139 pos.col = 0;
7140 if (name[0] == '.' && name[1] == NUL)
7141 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007142 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007143 curwin->w_cursor = pos;
7144 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007145 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007146 curwin->w_curswant = curswant - 1;
7147 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007148 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007149 check_cursor();
7150 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007151 }
7152 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
7153 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007154 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007155 if (setmark_pos(name[1], &pos, fnum) == OK)
7156 rettv->vval.v_number = 0;
7157 }
7158 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007159 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007160 }
7161 }
7162}
7163
7164/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007165 * "setreg()" function
7166 */
7167 static void
7168f_setreg(typval_T *argvars, typval_T *rettv)
7169{
7170 int regname;
7171 char_u *strregname;
7172 char_u *stropt;
7173 char_u *strval;
7174 int append;
7175 char_u yank_type;
7176 long block_len;
7177
7178 block_len = -1;
7179 yank_type = MAUTO;
7180 append = FALSE;
7181
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007182 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007183 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007184
7185 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007186 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007187 regname = *strregname;
7188 if (regname == 0 || regname == '@')
7189 regname = '"';
7190
7191 if (argvars[2].v_type != VAR_UNKNOWN)
7192 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007193 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007194 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007195 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007196 for (; *stropt != NUL; ++stropt)
7197 switch (*stropt)
7198 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007199 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007200 append = TRUE;
7201 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007202 case 'v': case 'c': // character-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007203 yank_type = MCHAR;
7204 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007205 case 'V': case 'l': // line-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007206 yank_type = MLINE;
7207 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007208 case 'b': case Ctrl_V: // block-wise selection
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007209 yank_type = MBLOCK;
7210 if (VIM_ISDIGIT(stropt[1]))
7211 {
7212 ++stropt;
7213 block_len = getdigits(&stropt) - 1;
7214 --stropt;
7215 }
7216 break;
7217 }
7218 }
7219
7220 if (argvars[1].v_type == VAR_LIST)
7221 {
7222 char_u **lstval;
7223 char_u **allocval;
7224 char_u buf[NUMBUFLEN];
7225 char_u **curval;
7226 char_u **curallocval;
7227 list_T *ll = argvars[1].vval.v_list;
7228 listitem_T *li;
7229 int len;
7230
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007231 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007232 len = ll == NULL ? 0 : ll->lv_len;
7233
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007234 // First half: use for pointers to result lines; second half: use for
7235 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02007236 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007237 if (lstval == NULL)
7238 return;
7239 curval = lstval;
7240 allocval = lstval + len + 2;
7241 curallocval = allocval;
7242
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007243 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007244 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02007245 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02007246 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007247 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007248 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007249 if (strval == NULL)
7250 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01007251 if (strval == buf)
7252 {
7253 // Need to make a copy, next tv_get_string_buf_chk() will
7254 // overwrite the string.
7255 strval = vim_strsave(buf);
7256 if (strval == NULL)
7257 goto free_lstval;
7258 *curallocval++ = strval;
7259 }
7260 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007261 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007262 }
7263 *curval++ = NULL;
7264
7265 write_reg_contents_lst(regname, lstval, -1,
7266 append, yank_type, block_len);
7267free_lstval:
7268 while (curallocval > allocval)
7269 vim_free(*--curallocval);
7270 vim_free(lstval);
7271 }
7272 else
7273 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007274 strval = tv_get_string_chk(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007275 if (strval == NULL)
7276 return;
7277 write_reg_contents_ex(regname, strval, -1,
7278 append, yank_type, block_len);
7279 }
7280 rettv->vval.v_number = 0;
7281}
7282
7283/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007284 * "settagstack()" function
7285 */
7286 static void
7287f_settagstack(typval_T *argvars, typval_T *rettv)
7288{
7289 static char *e_invact2 = N_("E962: Invalid action: '%s'");
7290 win_T *wp;
7291 dict_T *d;
7292 int action = 'r';
7293
7294 rettv->vval.v_number = -1;
7295
7296 // first argument: window number or id
7297 wp = find_win_by_nr_or_id(&argvars[0]);
7298 if (wp == NULL)
7299 return;
7300
7301 // second argument: dict with items to set in the tag stack
7302 if (argvars[1].v_type != VAR_DICT)
7303 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007304 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007305 return;
7306 }
7307 d = argvars[1].vval.v_dict;
7308 if (d == NULL)
7309 return;
7310
7311 // third argument: action - 'a' for append and 'r' for replace.
7312 // default is to replace the stack.
7313 if (argvars[2].v_type == VAR_UNKNOWN)
7314 action = 'r';
7315 else if (argvars[2].v_type == VAR_STRING)
7316 {
7317 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007318 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007319 if (actstr == NULL)
7320 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01007321 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
7322 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007323 action = *actstr;
7324 else
7325 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007326 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007327 return;
7328 }
7329 }
7330 else
7331 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007332 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01007333 return;
7334 }
7335
7336 if (set_tagstack(wp, d, action) == OK)
7337 rettv->vval.v_number = 0;
7338}
7339
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007340#ifdef FEAT_CRYPT
7341/*
7342 * "sha256({string})" function
7343 */
7344 static void
7345f_sha256(typval_T *argvars, typval_T *rettv)
7346{
7347 char_u *p;
7348
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007349 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007350 rettv->vval.v_string = vim_strsave(
7351 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
7352 rettv->v_type = VAR_STRING;
7353}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007354#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007355
7356/*
7357 * "shellescape({string})" function
7358 */
7359 static void
7360f_shellescape(typval_T *argvars, typval_T *rettv)
7361{
Bram Moolenaar20615522017-06-05 18:46:26 +02007362 int do_special = non_zero_arg(&argvars[1]);
7363
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007364 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007365 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007366 rettv->v_type = VAR_STRING;
7367}
7368
7369/*
7370 * shiftwidth() function
7371 */
7372 static void
7373f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
7374{
Bram Moolenaarf9514162018-11-22 03:08:29 +01007375 rettv->vval.v_number = 0;
7376
7377 if (argvars[0].v_type != VAR_UNKNOWN)
7378 {
7379 long col;
7380
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007381 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01007382 if (col < 0)
7383 return; // type error; errmsg already given
7384#ifdef FEAT_VARTABS
7385 rettv->vval.v_number = get_sw_value_col(curbuf, col);
7386 return;
7387#endif
7388 }
7389
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007390 rettv->vval.v_number = get_sw_value(curbuf);
7391}
7392
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007393#ifdef FEAT_FLOAT
7394/*
7395 * "sin()" function
7396 */
7397 static void
7398f_sin(typval_T *argvars, typval_T *rettv)
7399{
7400 float_T f = 0.0;
7401
7402 rettv->v_type = VAR_FLOAT;
7403 if (get_float_arg(argvars, &f) == OK)
7404 rettv->vval.v_float = sin(f);
7405 else
7406 rettv->vval.v_float = 0.0;
7407}
7408
7409/*
7410 * "sinh()" function
7411 */
7412 static void
7413f_sinh(typval_T *argvars, typval_T *rettv)
7414{
7415 float_T f = 0.0;
7416
7417 rettv->v_type = VAR_FLOAT;
7418 if (get_float_arg(argvars, &f) == OK)
7419 rettv->vval.v_float = sinh(f);
7420 else
7421 rettv->vval.v_float = 0.0;
7422}
7423#endif
7424
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007425/*
7426 * "soundfold({word})" function
7427 */
7428 static void
7429f_soundfold(typval_T *argvars, typval_T *rettv)
7430{
7431 char_u *s;
7432
7433 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007434 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007435#ifdef FEAT_SPELL
7436 rettv->vval.v_string = eval_soundfold(s);
7437#else
7438 rettv->vval.v_string = vim_strsave(s);
7439#endif
7440}
7441
7442/*
7443 * "spellbadword()" function
7444 */
7445 static void
7446f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
7447{
7448 char_u *word = (char_u *)"";
7449 hlf_T attr = HLF_COUNT;
7450 int len = 0;
7451
7452 if (rettv_list_alloc(rettv) == FAIL)
7453 return;
7454
7455#ifdef FEAT_SPELL
7456 if (argvars[0].v_type == VAR_UNKNOWN)
7457 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007458 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007459 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
7460 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007461 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007462 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01007463 curwin->w_set_curswant = TRUE;
7464 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007465 }
7466 else if (curwin->w_p_spell && *curbuf->b_s.b_p_spl != NUL)
7467 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007468 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007469 int capcol = -1;
7470
7471 if (str != NULL)
7472 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007473 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007474 while (*str != NUL)
7475 {
7476 len = spell_check(curwin, str, &attr, &capcol, FALSE);
7477 if (attr != HLF_COUNT)
7478 {
7479 word = str;
7480 break;
7481 }
7482 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02007483 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02007484 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007485 }
7486 }
7487 }
7488#endif
7489
7490 list_append_string(rettv->vval.v_list, word, len);
7491 list_append_string(rettv->vval.v_list, (char_u *)(
7492 attr == HLF_SPB ? "bad" :
7493 attr == HLF_SPR ? "rare" :
7494 attr == HLF_SPL ? "local" :
7495 attr == HLF_SPC ? "caps" :
7496 ""), -1);
7497}
7498
7499/*
7500 * "spellsuggest()" function
7501 */
7502 static void
7503f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
7504{
7505#ifdef FEAT_SPELL
7506 char_u *str;
7507 int typeerr = FALSE;
7508 int maxcount;
7509 garray_T ga;
7510 int i;
7511 listitem_T *li;
7512 int need_capital = FALSE;
7513#endif
7514
7515 if (rettv_list_alloc(rettv) == FAIL)
7516 return;
7517
7518#ifdef FEAT_SPELL
7519 if (curwin->w_p_spell && *curwin->w_s->b_p_spl != NUL)
7520 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007521 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007522 if (argvars[1].v_type != VAR_UNKNOWN)
7523 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007524 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007525 if (maxcount <= 0)
7526 return;
7527 if (argvars[2].v_type != VAR_UNKNOWN)
7528 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007529 need_capital = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007530 if (typeerr)
7531 return;
7532 }
7533 }
7534 else
7535 maxcount = 25;
7536
7537 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
7538
7539 for (i = 0; i < ga.ga_len; ++i)
7540 {
7541 str = ((char_u **)ga.ga_data)[i];
7542
7543 li = listitem_alloc();
7544 if (li == NULL)
7545 vim_free(str);
7546 else
7547 {
7548 li->li_tv.v_type = VAR_STRING;
7549 li->li_tv.v_lock = 0;
7550 li->li_tv.vval.v_string = str;
7551 list_append(rettv->vval.v_list, li);
7552 }
7553 }
7554 ga_clear(&ga);
7555 }
7556#endif
7557}
7558
7559 static void
7560f_split(typval_T *argvars, typval_T *rettv)
7561{
7562 char_u *str;
7563 char_u *end;
7564 char_u *pat = NULL;
7565 regmatch_T regmatch;
7566 char_u patbuf[NUMBUFLEN];
7567 char_u *save_cpo;
7568 int match;
7569 colnr_T col = 0;
7570 int keepempty = FALSE;
7571 int typeerr = FALSE;
7572
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007573 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007574 save_cpo = p_cpo;
7575 p_cpo = (char_u *)"";
7576
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007577 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007578 if (argvars[1].v_type != VAR_UNKNOWN)
7579 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007580 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007581 if (pat == NULL)
7582 typeerr = TRUE;
7583 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007584 keepempty = (int)tv_get_number_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007585 }
7586 if (pat == NULL || *pat == NUL)
7587 pat = (char_u *)"[\\x01- ]\\+";
7588
7589 if (rettv_list_alloc(rettv) == FAIL)
7590 return;
7591 if (typeerr)
7592 return;
7593
7594 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
7595 if (regmatch.regprog != NULL)
7596 {
7597 regmatch.rm_ic = FALSE;
7598 while (*str != NUL || keepempty)
7599 {
7600 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007601 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007602 else
7603 match = vim_regexec_nl(&regmatch, str, col);
7604 if (match)
7605 end = regmatch.startp[0];
7606 else
7607 end = str + STRLEN(str);
7608 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
7609 && *str != NUL && match && end < regmatch.endp[0]))
7610 {
7611 if (list_append_string(rettv->vval.v_list, str,
7612 (int)(end - str)) == FAIL)
7613 break;
7614 }
7615 if (!match)
7616 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01007617 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007618 if (regmatch.endp[0] > str)
7619 col = 0;
7620 else
Bram Moolenaar13505972019-01-24 15:04:48 +01007621 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007622 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007623 str = regmatch.endp[0];
7624 }
7625
7626 vim_regfree(regmatch.regprog);
7627 }
7628
7629 p_cpo = save_cpo;
7630}
7631
7632#ifdef FEAT_FLOAT
7633/*
7634 * "sqrt()" function
7635 */
7636 static void
7637f_sqrt(typval_T *argvars, typval_T *rettv)
7638{
7639 float_T f = 0.0;
7640
7641 rettv->v_type = VAR_FLOAT;
7642 if (get_float_arg(argvars, &f) == OK)
7643 rettv->vval.v_float = sqrt(f);
7644 else
7645 rettv->vval.v_float = 0.0;
7646}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007647#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007648
Bram Moolenaar0387cae2019-11-29 21:07:58 +01007649#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01007650/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007651 * "str2float()" function
7652 */
7653 static void
7654f_str2float(typval_T *argvars, typval_T *rettv)
7655{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007656 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007657 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007658
Bram Moolenaar08243d22017-01-10 16:12:29 +01007659 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007660 p = skipwhite(p + 1);
7661 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01007662 if (isneg)
7663 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007664 rettv->v_type = VAR_FLOAT;
7665}
7666#endif
7667
7668/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02007669 * "str2list()" function
7670 */
7671 static void
7672f_str2list(typval_T *argvars, typval_T *rettv)
7673{
7674 char_u *p;
7675 int utf8 = FALSE;
7676
7677 if (rettv_list_alloc(rettv) == FAIL)
7678 return;
7679
7680 if (argvars[1].v_type != VAR_UNKNOWN)
7681 utf8 = (int)tv_get_number_chk(&argvars[1], NULL);
7682
7683 p = tv_get_string(&argvars[0]);
7684
7685 if (has_mbyte || utf8)
7686 {
7687 int (*ptr2len)(char_u *);
7688 int (*ptr2char)(char_u *);
7689
7690 if (utf8 || enc_utf8)
7691 {
7692 ptr2len = utf_ptr2len;
7693 ptr2char = utf_ptr2char;
7694 }
7695 else
7696 {
7697 ptr2len = mb_ptr2len;
7698 ptr2char = mb_ptr2char;
7699 }
7700
7701 for ( ; *p != NUL; p += (*ptr2len)(p))
7702 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
7703 }
7704 else
7705 for ( ; *p != NUL; ++p)
7706 list_append_number(rettv->vval.v_list, *p);
7707}
7708
7709/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007710 * "str2nr()" function
7711 */
7712 static void
7713f_str2nr(typval_T *argvars, typval_T *rettv)
7714{
7715 int base = 10;
7716 char_u *p;
7717 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007718 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01007719 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007720
7721 if (argvars[1].v_type != VAR_UNKNOWN)
7722 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007723 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007724 if (base != 2 && base != 8 && base != 10 && base != 16)
7725 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007726 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007727 return;
7728 }
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007729 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_number(&argvars[2]))
7730 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007731 }
7732
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007733 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01007734 isneg = (*p == '-');
7735 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007736 p = skipwhite(p + 1);
7737 switch (base)
7738 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007739 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02007740 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02007741 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007742 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02007743 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
7744 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01007745 if (isneg)
7746 rettv->vval.v_number = -n;
7747 else
7748 rettv->vval.v_number = n;
7749
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007750}
7751
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007752/*
7753 * "strgetchar()" function
7754 */
7755 static void
7756f_strgetchar(typval_T *argvars, typval_T *rettv)
7757{
7758 char_u *str;
7759 int len;
7760 int error = FALSE;
7761 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01007762 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007763
7764 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007765 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007766 if (str == NULL)
7767 return;
7768 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007769 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007770 if (error)
7771 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007772
Bram Moolenaar13505972019-01-24 15:04:48 +01007773 while (charidx >= 0 && byteidx < len)
7774 {
7775 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007776 {
Bram Moolenaar13505972019-01-24 15:04:48 +01007777 rettv->vval.v_number = mb_ptr2char(str + byteidx);
7778 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007779 }
Bram Moolenaar13505972019-01-24 15:04:48 +01007780 --charidx;
7781 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007782 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007783}
7784
7785/*
7786 * "stridx()" function
7787 */
7788 static void
7789f_stridx(typval_T *argvars, typval_T *rettv)
7790{
7791 char_u buf[NUMBUFLEN];
7792 char_u *needle;
7793 char_u *haystack;
7794 char_u *save_haystack;
7795 char_u *pos;
7796 int start_idx;
7797
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007798 needle = tv_get_string_chk(&argvars[1]);
7799 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007800 rettv->vval.v_number = -1;
7801 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007802 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007803
7804 if (argvars[2].v_type != VAR_UNKNOWN)
7805 {
7806 int error = FALSE;
7807
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007808 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007809 if (error || start_idx >= (int)STRLEN(haystack))
7810 return;
7811 if (start_idx >= 0)
7812 haystack += start_idx;
7813 }
7814
7815 pos = (char_u *)strstr((char *)haystack, (char *)needle);
7816 if (pos != NULL)
7817 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
7818}
7819
7820/*
7821 * "string()" function
7822 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01007823 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007824f_string(typval_T *argvars, typval_T *rettv)
7825{
7826 char_u *tofree;
7827 char_u numbuf[NUMBUFLEN];
7828
7829 rettv->v_type = VAR_STRING;
7830 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
7831 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007832 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007833 if (rettv->vval.v_string != NULL && tofree == NULL)
7834 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
7835}
7836
7837/*
7838 * "strlen()" function
7839 */
7840 static void
7841f_strlen(typval_T *argvars, typval_T *rettv)
7842{
7843 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007844 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007845}
7846
7847/*
7848 * "strchars()" function
7849 */
7850 static void
7851f_strchars(typval_T *argvars, typval_T *rettv)
7852{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007853 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007854 int skipcc = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007855 varnumber_T len = 0;
7856 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007857
7858 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007859 skipcc = (int)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007860 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007861 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007862 else
7863 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007864 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
7865 while (*s != NUL)
7866 {
7867 func_mb_ptr2char_adv(&s);
7868 ++len;
7869 }
7870 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007871 }
7872}
7873
7874/*
7875 * "strdisplaywidth()" function
7876 */
7877 static void
7878f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
7879{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007880 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007881 int col = 0;
7882
7883 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007884 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007885
7886 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
7887}
7888
7889/*
7890 * "strwidth()" function
7891 */
7892 static void
7893f_strwidth(typval_T *argvars, typval_T *rettv)
7894{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007895 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007896
Bram Moolenaar13505972019-01-24 15:04:48 +01007897 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007898}
7899
7900/*
7901 * "strcharpart()" function
7902 */
7903 static void
7904f_strcharpart(typval_T *argvars, typval_T *rettv)
7905{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007906 char_u *p;
7907 int nchar;
7908 int nbyte = 0;
7909 int charlen;
7910 int len = 0;
7911 int slen;
7912 int error = FALSE;
7913
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007914 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007915 slen = (int)STRLEN(p);
7916
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007917 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007918 if (!error)
7919 {
7920 if (nchar > 0)
7921 while (nchar > 0 && nbyte < slen)
7922 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02007923 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007924 --nchar;
7925 }
7926 else
7927 nbyte = nchar;
7928 if (argvars[2].v_type != VAR_UNKNOWN)
7929 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007930 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007931 while (charlen > 0 && nbyte + len < slen)
7932 {
7933 int off = nbyte + len;
7934
7935 if (off < 0)
7936 len += 1;
7937 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02007938 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007939 --charlen;
7940 }
7941 }
7942 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007943 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007944 }
7945
7946 /*
7947 * Only return the overlap between the specified part and the actual
7948 * string.
7949 */
7950 if (nbyte < 0)
7951 {
7952 len += nbyte;
7953 nbyte = 0;
7954 }
7955 else if (nbyte > slen)
7956 nbyte = slen;
7957 if (len < 0)
7958 len = 0;
7959 else if (nbyte + len > slen)
7960 len = slen - nbyte;
7961
7962 rettv->v_type = VAR_STRING;
7963 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007964}
7965
7966/*
7967 * "strpart()" function
7968 */
7969 static void
7970f_strpart(typval_T *argvars, typval_T *rettv)
7971{
7972 char_u *p;
7973 int n;
7974 int len;
7975 int slen;
7976 int error = FALSE;
7977
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007978 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007979 slen = (int)STRLEN(p);
7980
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007981 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007982 if (error)
7983 len = 0;
7984 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007985 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007986 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007987 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007988
7989 /*
7990 * Only return the overlap between the specified part and the actual
7991 * string.
7992 */
7993 if (n < 0)
7994 {
7995 len += n;
7996 n = 0;
7997 }
7998 else if (n > slen)
7999 n = slen;
8000 if (len < 0)
8001 len = 0;
8002 else if (n + len > slen)
8003 len = slen - n;
8004
8005 rettv->v_type = VAR_STRING;
8006 rettv->vval.v_string = vim_strnsave(p + n, len);
8007}
8008
8009/*
8010 * "strridx()" function
8011 */
8012 static void
8013f_strridx(typval_T *argvars, typval_T *rettv)
8014{
8015 char_u buf[NUMBUFLEN];
8016 char_u *needle;
8017 char_u *haystack;
8018 char_u *rest;
8019 char_u *lastmatch = NULL;
8020 int haystack_len, end_idx;
8021
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008022 needle = tv_get_string_chk(&argvars[1]);
8023 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008024
8025 rettv->vval.v_number = -1;
8026 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008027 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008028
8029 haystack_len = (int)STRLEN(haystack);
8030 if (argvars[2].v_type != VAR_UNKNOWN)
8031 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008032 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008033 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008034 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008035 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008036 }
8037 else
8038 end_idx = haystack_len;
8039
8040 if (*needle == NUL)
8041 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008042 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008043 lastmatch = haystack + end_idx;
8044 }
8045 else
8046 {
8047 for (rest = haystack; *rest != '\0'; ++rest)
8048 {
8049 rest = (char_u *)strstr((char *)rest, (char *)needle);
8050 if (rest == NULL || rest > haystack + end_idx)
8051 break;
8052 lastmatch = rest;
8053 }
8054 }
8055
8056 if (lastmatch == NULL)
8057 rettv->vval.v_number = -1;
8058 else
8059 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
8060}
8061
8062/*
8063 * "strtrans()" function
8064 */
8065 static void
8066f_strtrans(typval_T *argvars, typval_T *rettv)
8067{
8068 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008069 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008070}
8071
8072/*
8073 * "submatch()" function
8074 */
8075 static void
8076f_submatch(typval_T *argvars, typval_T *rettv)
8077{
8078 int error = FALSE;
8079 int no;
8080 int retList = 0;
8081
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008082 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008083 if (error)
8084 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008085 if (no < 0 || no >= NSUBEXP)
8086 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008087 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01008088 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02008089 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008090 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008091 retList = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008092 if (error)
8093 return;
8094
8095 if (retList == 0)
8096 {
8097 rettv->v_type = VAR_STRING;
8098 rettv->vval.v_string = reg_submatch(no);
8099 }
8100 else
8101 {
8102 rettv->v_type = VAR_LIST;
8103 rettv->vval.v_list = reg_submatch_list(no);
8104 }
8105}
8106
8107/*
8108 * "substitute()" function
8109 */
8110 static void
8111f_substitute(typval_T *argvars, typval_T *rettv)
8112{
8113 char_u patbuf[NUMBUFLEN];
8114 char_u subbuf[NUMBUFLEN];
8115 char_u flagsbuf[NUMBUFLEN];
8116
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008117 char_u *str = tv_get_string_chk(&argvars[0]);
8118 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008119 char_u *sub = NULL;
8120 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008121 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008122
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008123 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
8124 expr = &argvars[2];
8125 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008126 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008127
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008128 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008129 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
8130 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008131 rettv->vval.v_string = NULL;
8132 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02008133 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008134}
8135
8136/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008137 * "swapinfo(swap_filename)" function
8138 */
8139 static void
8140f_swapinfo(typval_T *argvars, typval_T *rettv)
8141{
8142 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008143 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02008144}
8145
8146/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02008147 * "swapname(expr)" function
8148 */
8149 static void
8150f_swapname(typval_T *argvars, typval_T *rettv)
8151{
8152 buf_T *buf;
8153
8154 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01008155 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02008156 if (buf == NULL || buf->b_ml.ml_mfp == NULL
8157 || buf->b_ml.ml_mfp->mf_fname == NULL)
8158 rettv->vval.v_string = NULL;
8159 else
8160 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
8161}
8162
8163/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008164 * "synID(lnum, col, trans)" function
8165 */
8166 static void
8167f_synID(typval_T *argvars UNUSED, typval_T *rettv)
8168{
8169 int id = 0;
8170#ifdef FEAT_SYN_HL
8171 linenr_T lnum;
8172 colnr_T col;
8173 int trans;
8174 int transerr = FALSE;
8175
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008176 lnum = tv_get_lnum(argvars); // -1 on type error
8177 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008178 trans = (int)tv_get_number_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008179
8180 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8181 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
8182 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
8183#endif
8184
8185 rettv->vval.v_number = id;
8186}
8187
8188/*
8189 * "synIDattr(id, what [, mode])" function
8190 */
8191 static void
8192f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
8193{
8194 char_u *p = NULL;
8195#ifdef FEAT_SYN_HL
8196 int id;
8197 char_u *what;
8198 char_u *mode;
8199 char_u modebuf[NUMBUFLEN];
8200 int modec;
8201
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008202 id = (int)tv_get_number(&argvars[0]);
8203 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008204 if (argvars[2].v_type != VAR_UNKNOWN)
8205 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008206 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008207 modec = TOLOWER_ASC(mode[0]);
8208 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008209 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008210 }
8211 else
8212 {
8213#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
8214 if (USE_24BIT)
8215 modec = 'g';
8216 else
8217#endif
8218 if (t_colors > 1)
8219 modec = 'c';
8220 else
8221 modec = 't';
8222 }
8223
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008224 switch (TOLOWER_ASC(what[0]))
8225 {
8226 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008227 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008228 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008229 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008230 p = highlight_has_attr(id, HL_BOLD, modec);
8231 break;
8232
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008233 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008234 p = highlight_color(id, what, modec);
8235 break;
8236
8237 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008238 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008239 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008240 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008241 p = highlight_has_attr(id, HL_ITALIC, modec);
8242 break;
8243
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008244 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02008245 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008246 break;
8247
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008248 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008249 p = highlight_has_attr(id, HL_INVERSE, modec);
8250 break;
8251
8252 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008253 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008254 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008255 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02008256 else if (TOLOWER_ASC(what[1]) == 't' &&
8257 TOLOWER_ASC(what[2]) == 'r')
8258 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008259 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008260 p = highlight_has_attr(id, HL_STANDOUT, modec);
8261 break;
8262
8263 case 'u':
8264 if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008265 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008266 p = highlight_has_attr(id, HL_UNDERLINE, modec);
8267 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008268 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008269 p = highlight_has_attr(id, HL_UNDERCURL, modec);
8270 break;
8271 }
8272
8273 if (p != NULL)
8274 p = vim_strsave(p);
8275#endif
8276 rettv->v_type = VAR_STRING;
8277 rettv->vval.v_string = p;
8278}
8279
8280/*
8281 * "synIDtrans(id)" function
8282 */
8283 static void
8284f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
8285{
8286 int id;
8287
8288#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008289 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008290
8291 if (id > 0)
8292 id = syn_get_final_id(id);
8293 else
8294#endif
8295 id = 0;
8296
8297 rettv->vval.v_number = id;
8298}
8299
8300/*
8301 * "synconcealed(lnum, col)" function
8302 */
8303 static void
8304f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
8305{
8306#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
8307 linenr_T lnum;
8308 colnr_T col;
8309 int syntax_flags = 0;
8310 int cchar;
8311 int matchid = 0;
8312 char_u str[NUMBUFLEN];
8313#endif
8314
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008315 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008316
8317#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008318 lnum = tv_get_lnum(argvars); // -1 on type error
8319 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008320
Bram Moolenaara80faa82020-04-12 19:37:17 +02008321 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008322
8323 if (rettv_list_alloc(rettv) != FAIL)
8324 {
8325 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8326 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8327 && curwin->w_p_cole > 0)
8328 {
8329 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
8330 syntax_flags = get_syntax_info(&matchid);
8331
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008332 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008333 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
8334 {
8335 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02008336 if (cchar == NUL && curwin->w_p_cole == 1)
8337 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008338 if (cchar != NUL)
8339 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008340 if (has_mbyte)
8341 (*mb_char2bytes)(cchar, str);
8342 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008343 str[0] = cchar;
8344 }
8345 }
8346 }
8347
8348 list_append_number(rettv->vval.v_list,
8349 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008350 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008351 list_append_string(rettv->vval.v_list, str, -1);
8352 list_append_number(rettv->vval.v_list, matchid);
8353 }
8354#endif
8355}
8356
8357/*
8358 * "synstack(lnum, col)" function
8359 */
8360 static void
8361f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
8362{
8363#ifdef FEAT_SYN_HL
8364 linenr_T lnum;
8365 colnr_T col;
8366 int i;
8367 int id;
8368#endif
8369
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02008370 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008371
8372#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008373 lnum = tv_get_lnum(argvars); // -1 on type error
8374 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008375
8376 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
8377 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
8378 && rettv_list_alloc(rettv) != FAIL)
8379 {
8380 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
8381 for (i = 0; ; ++i)
8382 {
8383 id = syn_get_stack_item(i);
8384 if (id < 0)
8385 break;
8386 if (list_append_number(rettv->vval.v_list, id) == FAIL)
8387 break;
8388 }
8389 }
8390#endif
8391}
8392
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008393/*
8394 * "tabpagebuflist()" function
8395 */
8396 static void
8397f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8398{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008399 tabpage_T *tp;
8400 win_T *wp = NULL;
8401
8402 if (argvars[0].v_type == VAR_UNKNOWN)
8403 wp = firstwin;
8404 else
8405 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008406 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008407 if (tp != NULL)
8408 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
8409 }
8410 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
8411 {
8412 for (; wp != NULL; wp = wp->w_next)
8413 if (list_append_number(rettv->vval.v_list,
8414 wp->w_buffer->b_fnum) == FAIL)
8415 break;
8416 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008417}
8418
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008419/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008420 * "tagfiles()" function
8421 */
8422 static void
8423f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
8424{
8425 char_u *fname;
8426 tagname_T tn;
8427 int first;
8428
8429 if (rettv_list_alloc(rettv) == FAIL)
8430 return;
8431 fname = alloc(MAXPATHL);
8432 if (fname == NULL)
8433 return;
8434
8435 for (first = TRUE; ; first = FALSE)
8436 if (get_tagfname(&tn, first, fname) == FAIL
8437 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
8438 break;
8439 tagname_free(&tn);
8440 vim_free(fname);
8441}
8442
8443/*
8444 * "taglist()" function
8445 */
8446 static void
8447f_taglist(typval_T *argvars, typval_T *rettv)
8448{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008449 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008450 char_u *tag_pattern;
8451
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008452 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008453
8454 rettv->vval.v_number = FALSE;
8455 if (*tag_pattern == NUL)
8456 return;
8457
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008458 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008459 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008460 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01008461 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008462}
8463
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008464#ifdef FEAT_FLOAT
8465/*
8466 * "tan()" function
8467 */
8468 static void
8469f_tan(typval_T *argvars, typval_T *rettv)
8470{
8471 float_T f = 0.0;
8472
8473 rettv->v_type = VAR_FLOAT;
8474 if (get_float_arg(argvars, &f) == OK)
8475 rettv->vval.v_float = tan(f);
8476 else
8477 rettv->vval.v_float = 0.0;
8478}
8479
8480/*
8481 * "tanh()" function
8482 */
8483 static void
8484f_tanh(typval_T *argvars, typval_T *rettv)
8485{
8486 float_T f = 0.0;
8487
8488 rettv->v_type = VAR_FLOAT;
8489 if (get_float_arg(argvars, &f) == OK)
8490 rettv->vval.v_float = tanh(f);
8491 else
8492 rettv->vval.v_float = 0.0;
8493}
8494#endif
8495
8496/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008497 * "tolower(string)" function
8498 */
8499 static void
8500f_tolower(typval_T *argvars, typval_T *rettv)
8501{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008502 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008503 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008504}
8505
8506/*
8507 * "toupper(string)" function
8508 */
8509 static void
8510f_toupper(typval_T *argvars, typval_T *rettv)
8511{
8512 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008513 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008514}
8515
8516/*
8517 * "tr(string, fromstr, tostr)" function
8518 */
8519 static void
8520f_tr(typval_T *argvars, typval_T *rettv)
8521{
8522 char_u *in_str;
8523 char_u *fromstr;
8524 char_u *tostr;
8525 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008526 int inlen;
8527 int fromlen;
8528 int tolen;
8529 int idx;
8530 char_u *cpstr;
8531 int cplen;
8532 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008533 char_u buf[NUMBUFLEN];
8534 char_u buf2[NUMBUFLEN];
8535 garray_T ga;
8536
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008537 in_str = tv_get_string(&argvars[0]);
8538 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
8539 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008540
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008541 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008542 rettv->v_type = VAR_STRING;
8543 rettv->vval.v_string = NULL;
8544 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008545 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008546 ga_init2(&ga, (int)sizeof(char), 80);
8547
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008548 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008549 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008550 if (STRLEN(fromstr) != STRLEN(tostr))
8551 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008552error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008553 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008554 ga_clear(&ga);
8555 return;
8556 }
8557
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008558 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008559 while (*in_str != NUL)
8560 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008561 if (has_mbyte)
8562 {
8563 inlen = (*mb_ptr2len)(in_str);
8564 cpstr = in_str;
8565 cplen = inlen;
8566 idx = 0;
8567 for (p = fromstr; *p != NUL; p += fromlen)
8568 {
8569 fromlen = (*mb_ptr2len)(p);
8570 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
8571 {
8572 for (p = tostr; *p != NUL; p += tolen)
8573 {
8574 tolen = (*mb_ptr2len)(p);
8575 if (idx-- == 0)
8576 {
8577 cplen = tolen;
8578 cpstr = p;
8579 break;
8580 }
8581 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008582 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008583 goto error;
8584 break;
8585 }
8586 ++idx;
8587 }
8588
8589 if (first && cpstr == in_str)
8590 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008591 // Check that fromstr and tostr have the same number of
8592 // (multi-byte) characters. Done only once when a character
8593 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008594 first = FALSE;
8595 for (p = tostr; *p != NUL; p += tolen)
8596 {
8597 tolen = (*mb_ptr2len)(p);
8598 --idx;
8599 }
8600 if (idx != 0)
8601 goto error;
8602 }
8603
8604 (void)ga_grow(&ga, cplen);
8605 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
8606 ga.ga_len += cplen;
8607
8608 in_str += inlen;
8609 }
8610 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008611 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008612 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008613 p = vim_strchr(fromstr, *in_str);
8614 if (p != NULL)
8615 ga_append(&ga, tostr[p - fromstr]);
8616 else
8617 ga_append(&ga, *in_str);
8618 ++in_str;
8619 }
8620 }
8621
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008622 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008623 (void)ga_grow(&ga, 1);
8624 ga_append(&ga, NUL);
8625
8626 rettv->vval.v_string = ga.ga_data;
8627}
8628
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008629/*
8630 * "trim({expr})" function
8631 */
8632 static void
8633f_trim(typval_T *argvars, typval_T *rettv)
8634{
8635 char_u buf1[NUMBUFLEN];
8636 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008637 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008638 char_u *mask = NULL;
8639 char_u *tail;
8640 char_u *prev;
8641 char_u *p;
8642 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008643 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008644
8645 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008646 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008647 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008648 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008649
8650 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008651 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008652 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008653
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008654 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008655 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008656 int error = 0;
8657
8658 // leading or trailing characters to trim
8659 dir = (int)tv_get_number_chk(&argvars[2], &error);
8660 if (error)
8661 return;
8662 if (dir < 0 || dir > 2)
8663 {
8664 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
8665 return;
8666 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008667 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008668 }
8669
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008670 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008671 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008672 // Trim leading characters
8673 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008674 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008675 c1 = PTR2CHAR(head);
8676 if (mask == NULL)
8677 {
8678 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008679 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02008680 }
8681 else
8682 {
8683 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8684 if (c1 == PTR2CHAR(p))
8685 break;
8686 if (*p == NUL)
8687 break;
8688 }
8689 MB_PTR_ADV(head);
8690 }
8691 }
8692
8693 tail = head + STRLEN(head);
8694 if (dir == 0 || dir == 2)
8695 {
8696 // Trim trailing characters
8697 for (; tail > head; tail = prev)
8698 {
8699 prev = tail;
8700 MB_PTR_BACK(head, prev);
8701 c1 = PTR2CHAR(prev);
8702 if (mask == NULL)
8703 {
8704 if (c1 > ' ' && c1 != 0xa0)
8705 break;
8706 }
8707 else
8708 {
8709 for (p = mask; *p != NUL; MB_PTR_ADV(p))
8710 if (c1 == PTR2CHAR(p))
8711 break;
8712 if (*p == NUL)
8713 break;
8714 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01008715 }
8716 }
8717 rettv->vval.v_string = vim_strnsave(head, (int)(tail - head));
8718}
8719
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008720#ifdef FEAT_FLOAT
8721/*
8722 * "trunc({float})" function
8723 */
8724 static void
8725f_trunc(typval_T *argvars, typval_T *rettv)
8726{
8727 float_T f = 0.0;
8728
8729 rettv->v_type = VAR_FLOAT;
8730 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008731 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008732 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
8733 else
8734 rettv->vval.v_float = 0.0;
8735}
8736#endif
8737
8738/*
8739 * "type(expr)" function
8740 */
8741 static void
8742f_type(typval_T *argvars, typval_T *rettv)
8743{
8744 int n = -1;
8745
8746 switch (argvars[0].v_type)
8747 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008748 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
8749 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008750 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01008751 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
8752 case VAR_LIST: n = VAR_TYPE_LIST; break;
8753 case VAR_DICT: n = VAR_TYPE_DICT; break;
8754 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
8755 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
8756 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02008757 case VAR_JOB: n = VAR_TYPE_JOB; break;
8758 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01008759 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008760 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02008761 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01008762 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01008763 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008764 n = -1;
8765 break;
8766 }
8767 rettv->vval.v_number = n;
8768}
8769
8770/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008771 * "virtcol(string)" function
8772 */
8773 static void
8774f_virtcol(typval_T *argvars, typval_T *rettv)
8775{
8776 colnr_T vcol = 0;
8777 pos_T *fp;
8778 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008779 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008780
8781 fp = var2fpos(&argvars[0], FALSE, &fnum);
8782 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
8783 && fnum == curbuf->b_fnum)
8784 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01008785 // Limit the column to a valid value, getvvcol() doesn't check.
8786 if (fp->col < 0)
8787 fp->col = 0;
8788 else
8789 {
8790 len = (int)STRLEN(ml_get(fp->lnum));
8791 if (fp->col > len)
8792 fp->col = len;
8793 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008794 getvvcol(curwin, fp, NULL, NULL, &vcol);
8795 ++vcol;
8796 }
8797
8798 rettv->vval.v_number = vcol;
8799}
8800
8801/*
8802 * "visualmode()" function
8803 */
8804 static void
8805f_visualmode(typval_T *argvars, typval_T *rettv)
8806{
8807 char_u str[2];
8808
8809 rettv->v_type = VAR_STRING;
8810 str[0] = curbuf->b_visual_mode_eval;
8811 str[1] = NUL;
8812 rettv->vval.v_string = vim_strsave(str);
8813
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008814 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008815 if (non_zero_arg(&argvars[0]))
8816 curbuf->b_visual_mode_eval = NUL;
8817}
8818
8819/*
8820 * "wildmenumode()" function
8821 */
8822 static void
8823f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8824{
8825#ifdef FEAT_WILDMENU
8826 if (wild_menu_showing)
8827 rettv->vval.v_number = 1;
8828#endif
8829}
8830
8831/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +01008832 * "windowsversion()" function
8833 */
8834 static void
8835f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
8836{
8837 rettv->v_type = VAR_STRING;
8838 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
8839}
8840
8841/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008842 * "wordcount()" function
8843 */
8844 static void
8845f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
8846{
8847 if (rettv_dict_alloc(rettv) == FAIL)
8848 return;
8849 cursor_pos_info(rettv->vval.v_dict);
8850}
8851
8852/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008853 * "xor(expr, expr)" function
8854 */
8855 static void
8856f_xor(typval_T *argvars, typval_T *rettv)
8857{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008858 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
8859 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008860}
8861
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008862#endif // FEAT_EVAL