blob: adfc226a05a244bdf9c11242d16b58684c878f48 [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);
Bram Moolenaarbb861e22020-06-07 18:16:36 +020097static void f_getreginfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +020098static void f_getregtype(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +010099static void f_gettagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +0200100static void f_gettext(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200101static void f_haslocaldir(typval_T *argvars, typval_T *rettv);
102static void f_hasmapto(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200103static void f_hlID(typval_T *argvars, typval_T *rettv);
104static void f_hlexists(typval_T *argvars, typval_T *rettv);
105static void f_hostname(typval_T *argvars, typval_T *rettv);
106static void f_iconv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200107static void f_index(typval_T *argvars, typval_T *rettv);
108static void f_input(typval_T *argvars, typval_T *rettv);
109static void f_inputdialog(typval_T *argvars, typval_T *rettv);
110static void f_inputlist(typval_T *argvars, typval_T *rettv);
111static void f_inputrestore(typval_T *argvars, typval_T *rettv);
112static void f_inputsave(typval_T *argvars, typval_T *rettv);
113static void f_inputsecret(typval_T *argvars, typval_T *rettv);
Bram Moolenaar67a2deb2019-11-25 00:05:32 +0100114static void f_interrupt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200115static void f_invert(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200116static void f_islocked(typval_T *argvars, typval_T *rettv);
117#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
Bram Moolenaarfda1bff2019-04-04 13:44:37 +0200118static void f_isinf(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200119static void f_isnan(typval_T *argvars, typval_T *rettv);
120#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200121static void f_last_buffer_nr(typval_T *argvars, typval_T *rettv);
122static void f_len(typval_T *argvars, typval_T *rettv);
123static void f_libcall(typval_T *argvars, typval_T *rettv);
124static void f_libcallnr(typval_T *argvars, typval_T *rettv);
125static void f_line(typval_T *argvars, typval_T *rettv);
126static void f_line2byte(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200127#ifdef FEAT_FLOAT
128static void f_log(typval_T *argvars, typval_T *rettv);
129static void f_log10(typval_T *argvars, typval_T *rettv);
130#endif
131#ifdef FEAT_LUA
132static void f_luaeval(typval_T *argvars, typval_T *rettv);
133#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200134static void f_maparg(typval_T *argvars, typval_T *rettv);
135static void f_mapcheck(typval_T *argvars, typval_T *rettv);
136static void f_match(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200137static void f_matchend(typval_T *argvars, typval_T *rettv);
138static void f_matchlist(typval_T *argvars, typval_T *rettv);
139static void f_matchstr(typval_T *argvars, typval_T *rettv);
140static void f_matchstrpos(typval_T *argvars, typval_T *rettv);
141static void f_max(typval_T *argvars, typval_T *rettv);
142static void f_min(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200143#ifdef FEAT_MZSCHEME
144static void f_mzeval(typval_T *argvars, typval_T *rettv);
145#endif
146static void f_nextnonblank(typval_T *argvars, typval_T *rettv);
147static void f_nr2char(typval_T *argvars, typval_T *rettv);
148static void f_or(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200149#ifdef FEAT_PERL
150static void f_perleval(typval_T *argvars, typval_T *rettv);
151#endif
152#ifdef FEAT_FLOAT
153static void f_pow(typval_T *argvars, typval_T *rettv);
154#endif
155static void f_prevnonblank(typval_T *argvars, typval_T *rettv);
156static void f_printf(typval_T *argvars, typval_T *rettv);
Bram Moolenaare9bd5722019-08-17 19:36:06 +0200157static void f_pum_getpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200158static void f_pumvisible(typval_T *argvars, typval_T *rettv);
159#ifdef FEAT_PYTHON3
160static void f_py3eval(typval_T *argvars, typval_T *rettv);
161#endif
162#ifdef FEAT_PYTHON
163static void f_pyeval(typval_T *argvars, typval_T *rettv);
164#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +0100165#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
166static void f_pyxeval(typval_T *argvars, typval_T *rettv);
167#endif
Bram Moolenaar4f645c52020-02-08 16:40:39 +0100168static void f_test_srand_seed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100169static void f_rand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200170static void f_range(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0b6d9112018-05-22 20:35:17 +0200171static void f_reg_executing(typval_T *argvars, typval_T *rettv);
172static void f_reg_recording(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200173static void f_rename(typval_T *argvars, typval_T *rettv);
174static void f_repeat(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200175#ifdef FEAT_FLOAT
176static void f_round(typval_T *argvars, typval_T *rettv);
177#endif
Bram Moolenaare99be0e2019-03-26 22:51:09 +0100178#ifdef FEAT_RUBY
179static void f_rubyeval(typval_T *argvars, typval_T *rettv);
180#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200181static void f_screenattr(typval_T *argvars, typval_T *rettv);
182static void f_screenchar(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100183static void f_screenchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200184static void f_screencol(typval_T *argvars, typval_T *rettv);
185static void f_screenrow(typval_T *argvars, typval_T *rettv);
Bram Moolenaar2912abb2019-03-29 14:16:42 +0100186static void f_screenstring(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200187static void f_search(typval_T *argvars, typval_T *rettv);
188static void f_searchdecl(typval_T *argvars, typval_T *rettv);
189static void f_searchpair(typval_T *argvars, typval_T *rettv);
190static void f_searchpairpos(typval_T *argvars, typval_T *rettv);
191static void f_searchpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200192static void f_setcharsearch(typval_T *argvars, typval_T *rettv);
Bram Moolenaar691ddee2019-05-09 14:52:41 +0200193static void f_setenv(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200194static void f_setfperm(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200195static void f_setpos(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200196static void f_setreg(typval_T *argvars, typval_T *rettv);
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100197static void f_settagstack(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200198#ifdef FEAT_CRYPT
199static void f_sha256(typval_T *argvars, typval_T *rettv);
Bram Moolenaarb005cd82019-09-04 15:54:55 +0200200#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200201static void f_shellescape(typval_T *argvars, typval_T *rettv);
202static void f_shiftwidth(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200203#ifdef FEAT_FLOAT
204static void f_sin(typval_T *argvars, typval_T *rettv);
205static void f_sinh(typval_T *argvars, typval_T *rettv);
206#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200207static void f_soundfold(typval_T *argvars, typval_T *rettv);
208static void f_spellbadword(typval_T *argvars, typval_T *rettv);
209static void f_spellsuggest(typval_T *argvars, typval_T *rettv);
210static void f_split(typval_T *argvars, typval_T *rettv);
211#ifdef FEAT_FLOAT
212static void f_sqrt(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100213#endif
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +0100214static void f_srand(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0387cae2019-11-29 21:07:58 +0100215#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200216static void f_str2float(typval_T *argvars, typval_T *rettv);
217#endif
Bram Moolenaar9d401282019-04-06 13:18:12 +0200218static void f_str2list(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200219static void f_str2nr(typval_T *argvars, typval_T *rettv);
220static void f_strchars(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200221static void f_strgetchar(typval_T *argvars, typval_T *rettv);
222static void f_stridx(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200223static void f_strlen(typval_T *argvars, typval_T *rettv);
224static void f_strcharpart(typval_T *argvars, typval_T *rettv);
225static void f_strpart(typval_T *argvars, typval_T *rettv);
226static void f_strridx(typval_T *argvars, typval_T *rettv);
227static void f_strtrans(typval_T *argvars, typval_T *rettv);
228static void f_strdisplaywidth(typval_T *argvars, typval_T *rettv);
229static void f_strwidth(typval_T *argvars, typval_T *rettv);
230static void f_submatch(typval_T *argvars, typval_T *rettv);
231static void f_substitute(typval_T *argvars, typval_T *rettv);
Bram Moolenaar00f123a2018-08-21 20:28:54 +0200232static void f_swapinfo(typval_T *argvars, typval_T *rettv);
Bram Moolenaar110bd602018-09-16 18:46:59 +0200233static void f_swapname(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200234static void f_synID(typval_T *argvars, typval_T *rettv);
235static void f_synIDattr(typval_T *argvars, typval_T *rettv);
236static void f_synIDtrans(typval_T *argvars, typval_T *rettv);
237static void f_synstack(typval_T *argvars, typval_T *rettv);
238static void f_synconcealed(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200239static void f_tabpagebuflist(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200240static void f_taglist(typval_T *argvars, typval_T *rettv);
241static void f_tagfiles(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200242#ifdef FEAT_FLOAT
243static void f_tan(typval_T *argvars, typval_T *rettv);
244static void f_tanh(typval_T *argvars, typval_T *rettv);
245#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200246static void f_tolower(typval_T *argvars, typval_T *rettv);
247static void f_toupper(typval_T *argvars, typval_T *rettv);
248static void f_tr(typval_T *argvars, typval_T *rettv);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +0100249static void f_trim(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200250#ifdef FEAT_FLOAT
251static void f_trunc(typval_T *argvars, typval_T *rettv);
252#endif
253static void f_type(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200254static void f_virtcol(typval_T *argvars, typval_T *rettv);
255static void f_visualmode(typval_T *argvars, typval_T *rettv);
256static void f_wildmenumode(typval_T *argvars, typval_T *rettv);
Bram Moolenaar0c1e3742019-12-27 13:49:24 +0100257static void f_windowsversion(typval_T *argvars, typval_T *rettv);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200258static void f_wordcount(typval_T *argvars, typval_T *rettv);
259static void f_xor(typval_T *argvars, typval_T *rettv);
260
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100261
Bram Moolenaar94738d82020-10-21 14:25:07 +0200262/*
263 * Functions that check the argument type of a builtin function.
264 * Each function returns FAIL and gives an error message if the type is wrong.
265 */
266
267// Context passed to an arg_ function.
268typedef struct {
269 int arg_count; // actual argument count
270 int arg_idx; // current argument index (first arg is zero)
271} argcontext_T;
272
273// A function to check one argument type. The first argument is the type to
274// check. If needed, other argument types can be obtained with the context.
275// E.g. if "arg_idx" is 1, then (type - 1) is the first argument type.
276typedef int (*argcheck_T)(type_T *, argcontext_T *);
277
278 static int
279arg_float_or_nr(type_T *type, argcontext_T *context)
280{
281 if (type->tt_type == VAR_FLOAT || type->tt_type == VAR_NUMBER)
282 return OK;
283 arg_type_mismatch(&t_number, type, context->arg_idx + 1);
284 return FAIL;
285}
286
287/*
288 * Lists of functions that check the argument types of a builtin function.
289 */
290argcheck_T arg1_float_or_nr[] = {arg_float_or_nr};
291
292/*
293 * Functions that return the return type of a builtin function.
294 */
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100295 static type_T *
296ret_void(int argcount UNUSED, type_T **argtypes UNUSED)
297{
298 return &t_void;
299}
300 static type_T *
301ret_any(int argcount UNUSED, type_T **argtypes UNUSED)
302{
303 return &t_any;
304}
305 static type_T *
Bram Moolenaar403dc312020-10-17 19:29:51 +0200306ret_bool(int argcount UNUSED, type_T **argtypes UNUSED)
307{
308 return &t_bool;
309}
310 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100311ret_number(int argcount UNUSED, type_T **argtypes UNUSED)
312{
313 return &t_number;
314}
315 static type_T *
316ret_float(int argcount UNUSED, type_T **argtypes UNUSED)
317{
318 return &t_float;
319}
320 static type_T *
321ret_string(int argcount UNUSED, type_T **argtypes UNUSED)
322{
323 return &t_string;
324}
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200325 static type_T *
326ret_list_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100327{
328 return &t_list_any;
329}
330 static type_T *
331ret_list_number(int argcount UNUSED, type_T **argtypes UNUSED)
332{
333 return &t_list_number;
334}
335 static type_T *
336ret_list_string(int argcount UNUSED, type_T **argtypes UNUSED)
337{
338 return &t_list_string;
339}
340 static type_T *
341ret_list_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
342{
343 return &t_list_dict_any;
344}
345 static type_T *
346ret_dict_any(int argcount UNUSED, type_T **argtypes UNUSED)
347{
348 return &t_dict_any;
349}
350 static type_T *
351ret_dict_number(int argcount UNUSED, type_T **argtypes UNUSED)
352{
353 return &t_dict_number;
354}
355 static type_T *
356ret_dict_string(int argcount UNUSED, type_T **argtypes UNUSED)
357{
358 return &t_dict_string;
359}
360 static type_T *
361ret_blob(int argcount UNUSED, type_T **argtypes UNUSED)
362{
363 return &t_blob;
364}
365 static type_T *
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200366ret_func_any(int argcount UNUSED, type_T **argtypes UNUSED)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100367{
Bram Moolenaare69f6d02020-04-01 22:11:01 +0200368 return &t_func_any;
369}
370 static type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100371ret_channel(int argcount UNUSED, type_T **argtypes UNUSED)
372{
373 return &t_channel;
374}
375 static type_T *
376ret_job(int argcount UNUSED, type_T **argtypes UNUSED)
377{
378 return &t_job;
379}
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100380
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200381 static type_T *
382ret_first_arg(int argcount, type_T **argtypes)
383{
384 if (argcount > 0)
385 return argtypes[0];
386 return &t_void;
387}
388
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200389/*
390 * Used for getqflist(): returns list if there is no argument, dict if there is
391 * one.
392 */
393 static type_T *
394ret_list_or_dict_0(int argcount, type_T **argtypes UNUSED)
395{
396 if (argcount > 0)
397 return &t_dict_any;
398 return &t_list_dict_any;
399}
400
401/*
402 * Used for getloclist(): returns list if there is one argument, dict if there
403 * are two.
404 */
405 static type_T *
406ret_list_or_dict_1(int argcount, type_T **argtypes UNUSED)
407{
408 if (argcount > 1)
409 return &t_dict_any;
410 return &t_list_dict_any;
411}
412
Bram Moolenaar846178a2020-07-05 17:04:13 +0200413 static type_T *
414ret_argv(int argcount, type_T **argtypes UNUSED)
415{
416 // argv() returns list of strings
417 if (argcount == 0)
418 return &t_list_string;
419
420 // argv(0) returns a string, but argv(-1] returns a list
421 return &t_any;
422}
423
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200424 static type_T *
425ret_remove(int argcount UNUSED, type_T **argtypes)
426{
Bram Moolenaar5e654232020-09-16 15:22:00 +0200427 if (argtypes != NULL)
428 {
429 if (argtypes[0]->tt_type == VAR_LIST
430 || argtypes[0]->tt_type == VAR_DICT)
431 return argtypes[0]->tt_member;
432 if (argtypes[0]->tt_type == VAR_BLOB)
433 return &t_number;
434 }
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200435 return &t_any;
436}
437
Bram Moolenaar3d945cc2020-08-06 21:26:59 +0200438 static type_T *
439ret_getreg(int argcount, type_T **argtypes UNUSED)
440{
441 // Assume that if the third argument is passed it's non-zero
442 if (argcount == 3)
443 return &t_list_string;
444 return &t_string;
445}
446
Bram Moolenaar4a6d1b62020-08-08 17:55:49 +0200447 static type_T *
448ret_maparg(int argcount, type_T **argtypes UNUSED)
449{
450 // Assume that if the fourth argument is passed it's non-zero
451 if (argcount == 4)
452 return &t_dict_any;
453 return &t_string;
454}
455
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100456static type_T *ret_f_function(int argcount, type_T **argtypes);
457
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200458/*
459 * Array with names and number of arguments of all internal functions
460 * MUST BE KEPT SORTED IN strcmp() ORDER FOR BINARY SEARCH!
461 */
Bram Moolenaarac92e252019-08-03 21:58:38 +0200462typedef struct
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200463{
Bram Moolenaar25e42232019-08-04 15:04:10 +0200464 char *f_name; // function name
465 char f_min_argc; // minimal number of arguments
466 char f_max_argc; // maximal number of arguments
467 char f_argtype; // for method: FEARG_ values
Bram Moolenaar94738d82020-10-21 14:25:07 +0200468 argcheck_T *f_argcheck; // list of functions to check argument types
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +0100469 type_T *(*f_retfunc)(int argcount, type_T **argtypes);
470 // return type function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200471 void (*f_func)(typval_T *args, typval_T *rvar);
Bram Moolenaar25e42232019-08-04 15:04:10 +0200472 // implementation of function
Bram Moolenaarac92e252019-08-03 21:58:38 +0200473} funcentry_T;
474
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200475// values for f_argtype; zero means it cannot be used as a method
476#define FEARG_1 1 // base is the first argument
477#define FEARG_2 2 // base is the second argument
Bram Moolenaar24278d22019-08-16 21:49:22 +0200478#define FEARG_3 3 // base is the third argument
Bram Moolenaaraad222c2019-09-06 22:46:09 +0200479#define FEARG_4 4 // base is the fourth argument
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +0200480#define FEARG_LAST 9 // base is the last argument
481
Bram Moolenaar15c47602020-03-26 22:16:48 +0100482#ifdef FEAT_FLOAT
483# define FLOAT_FUNC(name) name
484#else
485# define FLOAT_FUNC(name) NULL
486#endif
487#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
488# define MATH_FUNC(name) name
489#else
490# define MATH_FUNC(name) NULL
491#endif
492#ifdef FEAT_TIMERS
493# define TIMER_FUNC(name) name
494#else
495# define TIMER_FUNC(name) NULL
496#endif
497#ifdef FEAT_JOB_CHANNEL
498# define JOB_FUNC(name) name
499#else
500# define JOB_FUNC(name) NULL
501#endif
502#ifdef FEAT_PROP_POPUP
503# define PROP_FUNC(name) name
504#else
505# define PROP_FUNC(name) NULL
506#endif
507#ifdef FEAT_SIGNS
508# define SIGN_FUNC(name) name
509#else
510# define SIGN_FUNC(name) NULL
511#endif
512#ifdef FEAT_SOUND
513# define SOUND_FUNC(name) name
514#else
515# define SOUND_FUNC(name) NULL
516#endif
517#ifdef FEAT_TERMINAL
518# define TERM_FUNC(name) name
519#else
520# define TERM_FUNC(name) NULL
521#endif
522
Bram Moolenaarac92e252019-08-03 21:58:38 +0200523static funcentry_T global_functions[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +0200524{
Bram Moolenaar94738d82020-10-21 14:25:07 +0200525 {"abs", 1, 1, FEARG_1, arg1_float_or_nr,
526 ret_any, FLOAT_FUNC(f_abs)},
527 {"acos", 1, 1, FEARG_1, NULL,
528 ret_float, FLOAT_FUNC(f_acos)},
529 {"add", 2, 2, FEARG_1, NULL,
530 ret_first_arg, f_add},
531 {"and", 2, 2, FEARG_1, NULL,
532 ret_number, f_and},
533 {"append", 2, 2, FEARG_2, NULL,
534 ret_number, f_append},
535 {"appendbufline", 3, 3, FEARG_3, NULL,
536 ret_number, f_appendbufline},
537 {"argc", 0, 1, 0, NULL,
538 ret_number, f_argc},
539 {"argidx", 0, 0, 0, NULL,
540 ret_number, f_argidx},
541 {"arglistid", 0, 2, 0, NULL,
542 ret_number, f_arglistid},
543 {"argv", 0, 2, 0, NULL,
544 ret_argv, f_argv},
545 {"asin", 1, 1, FEARG_1, NULL,
546 ret_float, FLOAT_FUNC(f_asin)},
547 {"assert_beeps", 1, 2, FEARG_1, NULL,
548 ret_number, f_assert_beeps},
549 {"assert_equal", 2, 3, FEARG_2, NULL,
550 ret_number, f_assert_equal},
551 {"assert_equalfile", 2, 3, FEARG_1, NULL,
552 ret_number, f_assert_equalfile},
553 {"assert_exception", 1, 2, 0, NULL,
554 ret_number, f_assert_exception},
555 {"assert_fails", 1, 5, FEARG_1, NULL,
556 ret_number, f_assert_fails},
557 {"assert_false", 1, 2, FEARG_1, NULL,
558 ret_number, f_assert_false},
559 {"assert_inrange", 3, 4, FEARG_3, NULL,
560 ret_number, f_assert_inrange},
561 {"assert_match", 2, 3, FEARG_2, NULL,
562 ret_number, f_assert_match},
563 {"assert_notequal", 2, 3, FEARG_2, NULL,
564 ret_number, f_assert_notequal},
565 {"assert_notmatch", 2, 3, FEARG_2, NULL,
566 ret_number, f_assert_notmatch},
567 {"assert_report", 1, 1, FEARG_1, NULL,
568 ret_number, f_assert_report},
569 {"assert_true", 1, 2, FEARG_1, NULL,
570 ret_number, f_assert_true},
571 {"atan", 1, 1, FEARG_1, NULL,
572 ret_float, FLOAT_FUNC(f_atan)},
573 {"atan2", 2, 2, FEARG_1, NULL,
574 ret_float, FLOAT_FUNC(f_atan2)},
575 {"balloon_gettext", 0, 0, 0, NULL,
576 ret_string,
Bram Moolenaar59716a22017-03-01 20:32:44 +0100577#ifdef FEAT_BEVAL
Bram Moolenaar15c47602020-03-26 22:16:48 +0100578 f_balloon_gettext
579#else
580 NULL
Bram Moolenaar59716a22017-03-01 20:32:44 +0100581#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100582 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200583 {"balloon_show", 1, 1, FEARG_1, NULL,
584 ret_void,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100585#ifdef FEAT_BEVAL
586 f_balloon_show
587#else
588 NULL
589#endif
590 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200591 {"balloon_split", 1, 1, FEARG_1, NULL,
592 ret_list_string,
Bram Moolenaar15c47602020-03-26 22:16:48 +0100593#if defined(FEAT_BEVAL_TERM)
594 f_balloon_split
595#else
596 NULL
597#endif
598 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200599 {"browse", 4, 4, 0, NULL,
600 ret_string, f_browse},
601 {"browsedir", 2, 2, 0, NULL,
602 ret_string, f_browsedir},
603 {"bufadd", 1, 1, FEARG_1, NULL,
604 ret_number, f_bufadd},
605 {"bufexists", 1, 1, FEARG_1, NULL,
606 ret_number, f_bufexists},
607 {"buffer_exists", 1, 1, FEARG_1, NULL,
608 ret_number, f_bufexists}, // obsolete
609 {"buffer_name", 0, 1, FEARG_1, NULL,
610 ret_string, f_bufname}, // obsolete
611 {"buffer_number", 0, 1, FEARG_1, NULL,
612 ret_number, f_bufnr}, // obsolete
613 {"buflisted", 1, 1, FEARG_1, NULL,
614 ret_number, f_buflisted},
615 {"bufload", 1, 1, FEARG_1, NULL,
616 ret_void, f_bufload},
617 {"bufloaded", 1, 1, FEARG_1, NULL,
618 ret_number, f_bufloaded},
619 {"bufname", 0, 1, FEARG_1, NULL,
620 ret_string, f_bufname},
621 {"bufnr", 0, 2, FEARG_1, NULL,
622 ret_number, f_bufnr},
623 {"bufwinid", 1, 1, FEARG_1, NULL,
624 ret_number, f_bufwinid},
625 {"bufwinnr", 1, 1, FEARG_1, NULL,
626 ret_number, f_bufwinnr},
627 {"byte2line", 1, 1, FEARG_1, NULL,
628 ret_number, f_byte2line},
629 {"byteidx", 2, 2, FEARG_1, NULL,
630 ret_number, f_byteidx},
631 {"byteidxcomp", 2, 2, FEARG_1, NULL,
632 ret_number, f_byteidxcomp},
633 {"call", 2, 3, FEARG_1, NULL,
634 ret_any, f_call},
635 {"ceil", 1, 1, FEARG_1, NULL,
636 ret_float, FLOAT_FUNC(f_ceil)},
637 {"ch_canread", 1, 1, FEARG_1, NULL,
638 ret_number, JOB_FUNC(f_ch_canread)},
639 {"ch_close", 1, 1, FEARG_1, NULL,
640 ret_void, JOB_FUNC(f_ch_close)},
641 {"ch_close_in", 1, 1, FEARG_1, NULL,
642 ret_void, JOB_FUNC(f_ch_close_in)},
643 {"ch_evalexpr", 2, 3, FEARG_1, NULL,
644 ret_any, JOB_FUNC(f_ch_evalexpr)},
645 {"ch_evalraw", 2, 3, FEARG_1, NULL,
646 ret_any, JOB_FUNC(f_ch_evalraw)},
647 {"ch_getbufnr", 2, 2, FEARG_1, NULL,
648 ret_number, JOB_FUNC(f_ch_getbufnr)},
649 {"ch_getjob", 1, 1, FEARG_1, NULL,
650 ret_job, JOB_FUNC(f_ch_getjob)},
651 {"ch_info", 1, 1, FEARG_1, NULL,
652 ret_dict_any, JOB_FUNC(f_ch_info)},
653 {"ch_log", 1, 2, FEARG_1, NULL,
654 ret_void, JOB_FUNC(f_ch_log)},
655 {"ch_logfile", 1, 2, FEARG_1, NULL,
656 ret_void, JOB_FUNC(f_ch_logfile)},
657 {"ch_open", 1, 2, FEARG_1, NULL,
658 ret_channel, JOB_FUNC(f_ch_open)},
659 {"ch_read", 1, 2, FEARG_1, NULL,
660 ret_string, JOB_FUNC(f_ch_read)},
661 {"ch_readblob", 1, 2, FEARG_1, NULL,
662 ret_blob, JOB_FUNC(f_ch_readblob)},
663 {"ch_readraw", 1, 2, FEARG_1, NULL,
664 ret_string, JOB_FUNC(f_ch_readraw)},
665 {"ch_sendexpr", 2, 3, FEARG_1, NULL,
666 ret_void, JOB_FUNC(f_ch_sendexpr)},
667 {"ch_sendraw", 2, 3, FEARG_1, NULL,
668 ret_void, JOB_FUNC(f_ch_sendraw)},
669 {"ch_setoptions", 2, 2, FEARG_1, NULL,
670 ret_void, JOB_FUNC(f_ch_setoptions)},
671 {"ch_status", 1, 2, FEARG_1, NULL,
672 ret_string, JOB_FUNC(f_ch_status)},
673 {"changenr", 0, 0, 0, NULL,
674 ret_number, f_changenr},
675 {"char2nr", 1, 2, FEARG_1, NULL,
676 ret_number, f_char2nr},
677 {"charclass", 1, 1, FEARG_1, NULL,
678 ret_number, f_charclass},
679 {"chdir", 1, 1, FEARG_1, NULL,
680 ret_string, f_chdir},
681 {"cindent", 1, 1, FEARG_1, NULL,
682 ret_number, f_cindent},
683 {"clearmatches", 0, 1, FEARG_1, NULL,
684 ret_void, f_clearmatches},
685 {"col", 1, 1, FEARG_1, NULL,
686 ret_number, f_col},
687 {"complete", 2, 2, FEARG_2, NULL,
688 ret_void, f_complete},
689 {"complete_add", 1, 1, FEARG_1, NULL,
690 ret_number, f_complete_add},
691 {"complete_check", 0, 0, 0, NULL,
692 ret_number, f_complete_check},
693 {"complete_info", 0, 1, FEARG_1, NULL,
694 ret_dict_any, f_complete_info},
695 {"confirm", 1, 4, FEARG_1, NULL,
696 ret_number, f_confirm},
697 {"copy", 1, 1, FEARG_1, NULL,
698 ret_first_arg, f_copy},
699 {"cos", 1, 1, FEARG_1, NULL,
700 ret_float, FLOAT_FUNC(f_cos)},
701 {"cosh", 1, 1, FEARG_1, NULL,
702 ret_float, FLOAT_FUNC(f_cosh)},
703 {"count", 2, 4, FEARG_1, NULL,
704 ret_number, f_count},
705 {"cscope_connection",0,3, 0, NULL,
706 ret_number, f_cscope_connection},
707 {"cursor", 1, 3, FEARG_1, NULL,
708 ret_number, f_cursor},
709 {"debugbreak", 1, 1, FEARG_1, NULL,
710 ret_number,
Bram Moolenaar4f974752019-02-17 17:44:42 +0100711#ifdef MSWIN
Bram Moolenaar15c47602020-03-26 22:16:48 +0100712 f_debugbreak
713#else
714 NULL
Bram Moolenaar4551c0a2018-06-20 22:38:21 +0200715#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +0100716 },
Bram Moolenaar94738d82020-10-21 14:25:07 +0200717 {"deepcopy", 1, 2, FEARG_1, NULL,
718 ret_first_arg, f_deepcopy},
719 {"delete", 1, 2, FEARG_1, NULL,
720 ret_number, f_delete},
721 {"deletebufline", 2, 3, FEARG_1, NULL,
722 ret_number, f_deletebufline},
723 {"did_filetype", 0, 0, 0, NULL,
724 ret_number, f_did_filetype},
725 {"diff_filler", 1, 1, FEARG_1, NULL,
726 ret_number, f_diff_filler},
727 {"diff_hlID", 2, 2, FEARG_1, NULL,
728 ret_number, f_diff_hlID},
729 {"echoraw", 1, 1, FEARG_1, NULL,
730 ret_number, f_echoraw},
731 {"empty", 1, 1, FEARG_1, NULL,
732 ret_number, f_empty},
733 {"environ", 0, 0, 0, NULL,
734 ret_dict_string, f_environ},
735 {"escape", 2, 2, FEARG_1, NULL,
736 ret_string, f_escape},
737 {"eval", 1, 1, FEARG_1, NULL,
738 ret_any, f_eval},
739 {"eventhandler", 0, 0, 0, NULL,
740 ret_number, f_eventhandler},
741 {"executable", 1, 1, FEARG_1, NULL,
742 ret_number, f_executable},
743 {"execute", 1, 2, FEARG_1, NULL,
744 ret_string, f_execute},
745 {"exepath", 1, 1, FEARG_1, NULL,
746 ret_string, f_exepath},
747 {"exists", 1, 1, FEARG_1, NULL,
748 ret_number, f_exists},
749 {"exp", 1, 1, FEARG_1, NULL,
750 ret_float, FLOAT_FUNC(f_exp)},
751 {"expand", 1, 3, FEARG_1, NULL,
752 ret_any, f_expand},
753 {"expandcmd", 1, 1, FEARG_1, NULL,
754 ret_string, f_expandcmd},
755 {"extend", 2, 3, FEARG_1, NULL,
756 ret_first_arg, f_extend},
757 {"feedkeys", 1, 2, FEARG_1, NULL,
758 ret_void, f_feedkeys},
759 {"file_readable", 1, 1, FEARG_1, NULL,
760 ret_number, f_filereadable}, // obsolete
761 {"filereadable", 1, 1, FEARG_1, NULL,
762 ret_number, f_filereadable},
763 {"filewritable", 1, 1, FEARG_1, NULL,
764 ret_number, f_filewritable},
765 {"filter", 2, 2, FEARG_1, NULL,
766 ret_first_arg, f_filter},
767 {"finddir", 1, 3, FEARG_1, NULL,
768 ret_string, f_finddir},
769 {"findfile", 1, 3, FEARG_1, NULL,
770 ret_string, f_findfile},
771 {"flatten", 1, 2, FEARG_1, NULL,
772 ret_list_any, f_flatten},
773 {"float2nr", 1, 1, FEARG_1, NULL,
774 ret_number, FLOAT_FUNC(f_float2nr)},
775 {"floor", 1, 1, FEARG_1, NULL,
776 ret_float, FLOAT_FUNC(f_floor)},
777 {"fmod", 2, 2, FEARG_1, NULL,
778 ret_float, FLOAT_FUNC(f_fmod)},
779 {"fnameescape", 1, 1, FEARG_1, NULL,
780 ret_string, f_fnameescape},
781 {"fnamemodify", 2, 2, FEARG_1, NULL,
782 ret_string, f_fnamemodify},
783 {"foldclosed", 1, 1, FEARG_1, NULL,
784 ret_number, f_foldclosed},
785 {"foldclosedend", 1, 1, FEARG_1, NULL,
786 ret_number, f_foldclosedend},
787 {"foldlevel", 1, 1, FEARG_1, NULL,
788 ret_number, f_foldlevel},
789 {"foldtext", 0, 0, 0, NULL,
790 ret_string, f_foldtext},
791 {"foldtextresult", 1, 1, FEARG_1, NULL,
792 ret_string, f_foldtextresult},
793 {"foreground", 0, 0, 0, NULL,
794 ret_void, f_foreground},
795 {"funcref", 1, 3, FEARG_1, NULL,
796 ret_func_any, f_funcref},
797 {"function", 1, 3, FEARG_1, NULL,
798 ret_f_function, f_function},
799 {"garbagecollect", 0, 1, 0, NULL,
800 ret_void, f_garbagecollect},
801 {"get", 2, 3, FEARG_1, NULL,
802 ret_any, f_get},
803 {"getbufinfo", 0, 1, FEARG_1, NULL,
804 ret_list_dict_any, f_getbufinfo},
805 {"getbufline", 2, 3, FEARG_1, NULL,
806 ret_list_string, f_getbufline},
807 {"getbufvar", 2, 3, FEARG_1, NULL,
808 ret_any, f_getbufvar},
809 {"getchangelist", 0, 1, FEARG_1, NULL,
810 ret_list_any, f_getchangelist},
811 {"getchar", 0, 1, 0, NULL,
812 ret_number, f_getchar},
813 {"getcharmod", 0, 0, 0, NULL,
814 ret_number, f_getcharmod},
815 {"getcharsearch", 0, 0, 0, NULL,
816 ret_dict_any, f_getcharsearch},
817 {"getcmdline", 0, 0, 0, NULL,
818 ret_string, f_getcmdline},
819 {"getcmdpos", 0, 0, 0, NULL,
820 ret_number, f_getcmdpos},
821 {"getcmdtype", 0, 0, 0, NULL,
822 ret_string, f_getcmdtype},
823 {"getcmdwintype", 0, 0, 0, NULL,
824 ret_string, f_getcmdwintype},
825 {"getcompletion", 2, 3, FEARG_1, NULL,
826 ret_list_string, f_getcompletion},
827 {"getcurpos", 0, 1, FEARG_1, NULL,
828 ret_list_number, f_getcurpos},
829 {"getcwd", 0, 2, FEARG_1, NULL,
830 ret_string, f_getcwd},
831 {"getenv", 1, 1, FEARG_1, NULL,
832 ret_string, f_getenv},
833 {"getfontname", 0, 1, 0, NULL,
834 ret_string, f_getfontname},
835 {"getfperm", 1, 1, FEARG_1, NULL,
836 ret_string, f_getfperm},
837 {"getfsize", 1, 1, FEARG_1, NULL,
838 ret_number, f_getfsize},
839 {"getftime", 1, 1, FEARG_1, NULL,
840 ret_number, f_getftime},
841 {"getftype", 1, 1, FEARG_1, NULL,
842 ret_string, f_getftype},
843 {"getimstatus", 0, 0, 0, NULL,
844 ret_number, f_getimstatus},
845 {"getjumplist", 0, 2, FEARG_1, NULL,
846 ret_list_any, f_getjumplist},
847 {"getline", 1, 2, FEARG_1, NULL,
848 ret_f_getline, f_getline},
849 {"getloclist", 1, 2, 0, NULL,
850 ret_list_or_dict_1, f_getloclist},
851 {"getmarklist", 0, 1, FEARG_1, NULL,
852 ret_list_dict_any, f_getmarklist},
853 {"getmatches", 0, 1, 0, NULL,
854 ret_list_dict_any, f_getmatches},
855 {"getmousepos", 0, 0, 0, NULL,
856 ret_dict_number, f_getmousepos},
857 {"getpid", 0, 0, 0, NULL,
858 ret_number, f_getpid},
859 {"getpos", 1, 1, FEARG_1, NULL,
860 ret_list_number, f_getpos},
861 {"getqflist", 0, 1, 0, NULL,
862 ret_list_or_dict_0, f_getqflist},
863 {"getreg", 0, 3, FEARG_1, NULL,
864 ret_getreg, f_getreg},
865 {"getreginfo", 0, 1, FEARG_1, NULL,
866 ret_dict_any, f_getreginfo},
867 {"getregtype", 0, 1, FEARG_1, NULL,
868 ret_string, f_getregtype},
869 {"gettabinfo", 0, 1, FEARG_1, NULL,
870 ret_list_dict_any, f_gettabinfo},
871 {"gettabvar", 2, 3, FEARG_1, NULL,
872 ret_any, f_gettabvar},
873 {"gettabwinvar", 3, 4, FEARG_1, NULL,
874 ret_any, f_gettabwinvar},
875 {"gettagstack", 0, 1, FEARG_1, NULL,
876 ret_dict_any, f_gettagstack},
877 {"gettext", 1, 1, FEARG_1, NULL,
878 ret_string, f_gettext},
879 {"getwininfo", 0, 1, FEARG_1, NULL,
880 ret_list_dict_any, f_getwininfo},
881 {"getwinpos", 0, 1, FEARG_1, NULL,
882 ret_list_number, f_getwinpos},
883 {"getwinposx", 0, 0, 0, NULL,
884 ret_number, f_getwinposx},
885 {"getwinposy", 0, 0, 0, NULL,
886 ret_number, f_getwinposy},
887 {"getwinvar", 2, 3, FEARG_1, NULL,
888 ret_any, f_getwinvar},
889 {"glob", 1, 4, FEARG_1, NULL,
890 ret_any, f_glob},
891 {"glob2regpat", 1, 1, FEARG_1, NULL,
892 ret_string, f_glob2regpat},
893 {"globpath", 2, 5, FEARG_2, NULL,
894 ret_any, f_globpath},
895 {"has", 1, 2, 0, NULL,
896 ret_number, f_has},
897 {"has_key", 2, 2, FEARG_1, NULL,
898 ret_number, f_has_key},
899 {"haslocaldir", 0, 2, FEARG_1, NULL,
900 ret_number, f_haslocaldir},
901 {"hasmapto", 1, 3, FEARG_1, NULL,
902 ret_number, f_hasmapto},
903 {"highlightID", 1, 1, FEARG_1, NULL,
904 ret_number, f_hlID}, // obsolete
905 {"highlight_exists",1, 1, FEARG_1, NULL,
906 ret_number, f_hlexists}, // obsolete
907 {"histadd", 2, 2, FEARG_2, NULL,
908 ret_number, f_histadd},
909 {"histdel", 1, 2, FEARG_1, NULL,
910 ret_number, f_histdel},
911 {"histget", 1, 2, FEARG_1, NULL,
912 ret_string, f_histget},
913 {"histnr", 1, 1, FEARG_1, NULL,
914 ret_number, f_histnr},
915 {"hlID", 1, 1, FEARG_1, NULL,
916 ret_number, f_hlID},
917 {"hlexists", 1, 1, FEARG_1, NULL,
918 ret_number, f_hlexists},
919 {"hostname", 0, 0, 0, NULL,
920 ret_string, f_hostname},
921 {"iconv", 3, 3, FEARG_1, NULL,
922 ret_string, f_iconv},
923 {"indent", 1, 1, FEARG_1, NULL,
924 ret_number, f_indent},
925 {"index", 2, 4, FEARG_1, NULL,
926 ret_number, f_index},
927 {"input", 1, 3, FEARG_1, NULL,
928 ret_string, f_input},
929 {"inputdialog", 1, 3, FEARG_1, NULL,
930 ret_string, f_inputdialog},
931 {"inputlist", 1, 1, FEARG_1, NULL,
932 ret_number, f_inputlist},
933 {"inputrestore", 0, 0, 0, NULL,
934 ret_number, f_inputrestore},
935 {"inputsave", 0, 0, 0, NULL,
936 ret_number, f_inputsave},
937 {"inputsecret", 1, 2, FEARG_1, NULL,
938 ret_string, f_inputsecret},
939 {"insert", 2, 3, FEARG_1, NULL,
940 ret_first_arg, f_insert},
941 {"interrupt", 0, 0, 0, NULL,
942 ret_void, f_interrupt},
943 {"invert", 1, 1, FEARG_1, NULL,
944 ret_number, f_invert},
945 {"isdirectory", 1, 1, FEARG_1, NULL,
946 ret_number, f_isdirectory},
947 {"isinf", 1, 1, FEARG_1, NULL,
948 ret_number, MATH_FUNC(f_isinf)},
949 {"islocked", 1, 1, FEARG_1, NULL,
950 ret_number, f_islocked},
951 {"isnan", 1, 1, FEARG_1, NULL,
952 ret_number, MATH_FUNC(f_isnan)},
953 {"items", 1, 1, FEARG_1, NULL,
954 ret_list_any, f_items},
955 {"job_getchannel", 1, 1, FEARG_1, NULL,
956 ret_channel, JOB_FUNC(f_job_getchannel)},
957 {"job_info", 0, 1, FEARG_1, NULL,
958 ret_dict_any, JOB_FUNC(f_job_info)},
959 {"job_setoptions", 2, 2, FEARG_1, NULL,
960 ret_void, JOB_FUNC(f_job_setoptions)},
961 {"job_start", 1, 2, FEARG_1, NULL,
962 ret_job, JOB_FUNC(f_job_start)},
963 {"job_status", 1, 1, FEARG_1, NULL,
964 ret_string, JOB_FUNC(f_job_status)},
965 {"job_stop", 1, 2, FEARG_1, NULL,
966 ret_number, JOB_FUNC(f_job_stop)},
967 {"join", 1, 2, FEARG_1, NULL,
968 ret_string, f_join},
969 {"js_decode", 1, 1, FEARG_1, NULL,
970 ret_any, f_js_decode},
971 {"js_encode", 1, 1, FEARG_1, NULL,
972 ret_string, f_js_encode},
973 {"json_decode", 1, 1, FEARG_1, NULL,
974 ret_any, f_json_decode},
975 {"json_encode", 1, 1, FEARG_1, NULL,
976 ret_string, f_json_encode},
977 {"keys", 1, 1, FEARG_1, NULL,
978 ret_list_string, f_keys},
979 {"last_buffer_nr", 0, 0, 0, NULL,
980 ret_number, f_last_buffer_nr}, // obsolete
981 {"len", 1, 1, FEARG_1, NULL,
982 ret_number, f_len},
983 {"libcall", 3, 3, FEARG_3, NULL,
984 ret_string, f_libcall},
985 {"libcallnr", 3, 3, FEARG_3, NULL,
986 ret_number, f_libcallnr},
987 {"line", 1, 2, FEARG_1, NULL,
988 ret_number, f_line},
989 {"line2byte", 1, 1, FEARG_1, NULL,
990 ret_number, f_line2byte},
991 {"lispindent", 1, 1, FEARG_1, NULL,
992 ret_number, f_lispindent},
993 {"list2str", 1, 2, FEARG_1, NULL,
994 ret_string, f_list2str},
995 {"listener_add", 1, 2, FEARG_2, NULL,
996 ret_number, f_listener_add},
997 {"listener_flush", 0, 1, FEARG_1, NULL,
998 ret_void, f_listener_flush},
999 {"listener_remove", 1, 1, FEARG_1, NULL,
1000 ret_number, f_listener_remove},
1001 {"localtime", 0, 0, 0, NULL,
1002 ret_number, f_localtime},
1003 {"log", 1, 1, FEARG_1, NULL,
1004 ret_float, FLOAT_FUNC(f_log)},
1005 {"log10", 1, 1, FEARG_1, NULL,
1006 ret_float, FLOAT_FUNC(f_log10)},
1007 {"luaeval", 1, 2, FEARG_1, NULL,
1008 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001009#ifdef FEAT_LUA
Bram Moolenaar15c47602020-03-26 22:16:48 +01001010 f_luaeval
1011#else
1012 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001013#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001014 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001015 {"map", 2, 2, FEARG_1, NULL,
1016 ret_any, f_map},
1017 {"maparg", 1, 4, FEARG_1, NULL,
1018 ret_maparg, f_maparg},
1019 {"mapcheck", 1, 3, FEARG_1, NULL,
1020 ret_string, f_mapcheck},
1021 {"mapset", 3, 3, FEARG_1, NULL,
1022 ret_void, f_mapset},
1023 {"match", 2, 4, FEARG_1, NULL,
1024 ret_any, f_match},
1025 {"matchadd", 2, 5, FEARG_1, NULL,
1026 ret_number, f_matchadd},
1027 {"matchaddpos", 2, 5, FEARG_1, NULL,
1028 ret_number, f_matchaddpos},
1029 {"matcharg", 1, 1, FEARG_1, NULL,
1030 ret_list_string, f_matcharg},
1031 {"matchdelete", 1, 2, FEARG_1, NULL,
1032 ret_number, f_matchdelete},
1033 {"matchend", 2, 4, FEARG_1, NULL,
1034 ret_number, f_matchend},
1035 {"matchfuzzy", 2, 3, FEARG_1, NULL,
1036 ret_list_string, f_matchfuzzy},
1037 {"matchfuzzypos", 2, 3, FEARG_1, NULL,
1038 ret_list_any, f_matchfuzzypos},
1039 {"matchlist", 2, 4, FEARG_1, NULL,
1040 ret_list_string, f_matchlist},
1041 {"matchstr", 2, 4, FEARG_1, NULL,
1042 ret_string, f_matchstr},
1043 {"matchstrpos", 2, 4, FEARG_1, NULL,
1044 ret_list_any, f_matchstrpos},
1045 {"max", 1, 1, FEARG_1, NULL,
1046 ret_any, f_max},
1047 {"menu_info", 1, 2, FEARG_1, NULL,
1048 ret_dict_any,
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001049#ifdef FEAT_MENU
Bram Moolenaar15c47602020-03-26 22:16:48 +01001050 f_menu_info
1051#else
1052 NULL
Bram Moolenaara2cbdea2020-03-16 21:08:31 +01001053#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001054 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001055 {"min", 1, 1, FEARG_1, NULL,
1056 ret_any, f_min},
1057 {"mkdir", 1, 3, FEARG_1, NULL,
1058 ret_number, f_mkdir},
1059 {"mode", 0, 1, FEARG_1, NULL,
1060 ret_string, f_mode},
1061 {"mzeval", 1, 1, FEARG_1, NULL,
1062 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001063#ifdef FEAT_MZSCHEME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001064 f_mzeval
1065#else
1066 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001067#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001068 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001069 {"nextnonblank", 1, 1, FEARG_1, NULL,
1070 ret_number, f_nextnonblank},
1071 {"nr2char", 1, 2, FEARG_1, NULL,
1072 ret_string, f_nr2char},
1073 {"or", 2, 2, FEARG_1, NULL,
1074 ret_number, f_or},
1075 {"pathshorten", 1, 2, FEARG_1, NULL,
1076 ret_string, f_pathshorten},
1077 {"perleval", 1, 1, FEARG_1, NULL,
1078 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001079#ifdef FEAT_PERL
Bram Moolenaar15c47602020-03-26 22:16:48 +01001080 f_perleval
1081#else
1082 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001083#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001084 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001085 {"popup_atcursor", 2, 2, FEARG_1, NULL,
1086 ret_number, PROP_FUNC(f_popup_atcursor)},
1087 {"popup_beval", 2, 2, FEARG_1, NULL,
1088 ret_number, PROP_FUNC(f_popup_beval)},
1089 {"popup_clear", 0, 1, 0, NULL,
1090 ret_void, PROP_FUNC(f_popup_clear)},
1091 {"popup_close", 1, 2, FEARG_1, NULL,
1092 ret_void, PROP_FUNC(f_popup_close)},
1093 {"popup_create", 2, 2, FEARG_1, NULL,
1094 ret_number, PROP_FUNC(f_popup_create)},
1095 {"popup_dialog", 2, 2, FEARG_1, NULL,
1096 ret_number, PROP_FUNC(f_popup_dialog)},
1097 {"popup_filter_menu", 2, 2, 0, NULL,
1098 ret_bool, PROP_FUNC(f_popup_filter_menu)},
1099 {"popup_filter_yesno", 2, 2, 0, NULL,
1100 ret_bool, PROP_FUNC(f_popup_filter_yesno)},
1101 {"popup_findinfo", 0, 0, 0, NULL,
1102 ret_number, PROP_FUNC(f_popup_findinfo)},
1103 {"popup_findpreview", 0, 0, 0, NULL,
1104 ret_number, PROP_FUNC(f_popup_findpreview)},
1105 {"popup_getoptions", 1, 1, FEARG_1, NULL,
1106 ret_dict_any, PROP_FUNC(f_popup_getoptions)},
1107 {"popup_getpos", 1, 1, FEARG_1, NULL,
1108 ret_dict_any, PROP_FUNC(f_popup_getpos)},
1109 {"popup_hide", 1, 1, FEARG_1, NULL,
1110 ret_void, PROP_FUNC(f_popup_hide)},
1111 {"popup_list", 0, 0, 0, NULL,
1112 ret_list_number, PROP_FUNC(f_popup_list)},
1113 {"popup_locate", 2, 2, 0, NULL,
1114 ret_number, PROP_FUNC(f_popup_locate)},
1115 {"popup_menu", 2, 2, FEARG_1, NULL,
1116 ret_number, PROP_FUNC(f_popup_menu)},
1117 {"popup_move", 2, 2, FEARG_1, NULL,
1118 ret_void, PROP_FUNC(f_popup_move)},
1119 {"popup_notification", 2, 2, FEARG_1, NULL,
1120 ret_number, PROP_FUNC(f_popup_notification)},
1121 {"popup_setoptions", 2, 2, FEARG_1, NULL,
1122 ret_void, PROP_FUNC(f_popup_setoptions)},
1123 {"popup_settext", 2, 2, FEARG_1, NULL,
1124 ret_void, PROP_FUNC(f_popup_settext)},
1125 {"popup_show", 1, 1, FEARG_1, NULL,
1126 ret_void, PROP_FUNC(f_popup_show)},
1127 {"pow", 2, 2, FEARG_1, NULL,
1128 ret_float, FLOAT_FUNC(f_pow)},
1129 {"prevnonblank", 1, 1, FEARG_1, NULL,
1130 ret_number, f_prevnonblank},
1131 {"printf", 1, 19, FEARG_2, NULL,
1132 ret_string, f_printf},
1133 {"prompt_getprompt", 1, 1, FEARG_1, NULL,
1134 ret_string, JOB_FUNC(f_prompt_getprompt)},
1135 {"prompt_setcallback", 2, 2, FEARG_1, NULL,
1136 ret_void, JOB_FUNC(f_prompt_setcallback)},
1137 {"prompt_setinterrupt", 2, 2, FEARG_1, NULL,
1138 ret_void, JOB_FUNC(f_prompt_setinterrupt)},
1139 {"prompt_setprompt", 2, 2, FEARG_1, NULL,
1140 ret_void, JOB_FUNC(f_prompt_setprompt)},
1141 {"prop_add", 3, 3, FEARG_1, NULL,
1142 ret_void, PROP_FUNC(f_prop_add)},
1143 {"prop_clear", 1, 3, FEARG_1, NULL,
1144 ret_void, PROP_FUNC(f_prop_clear)},
1145 {"prop_find", 1, 2, FEARG_1, NULL,
1146 ret_dict_any, PROP_FUNC(f_prop_find)},
1147 {"prop_list", 1, 2, FEARG_1, NULL,
1148 ret_list_dict_any, PROP_FUNC(f_prop_list)},
1149 {"prop_remove", 1, 3, FEARG_1, NULL,
1150 ret_number, PROP_FUNC(f_prop_remove)},
1151 {"prop_type_add", 2, 2, FEARG_1, NULL,
1152 ret_void, PROP_FUNC(f_prop_type_add)},
1153 {"prop_type_change", 2, 2, FEARG_1, NULL,
1154 ret_void, PROP_FUNC(f_prop_type_change)},
1155 {"prop_type_delete", 1, 2, FEARG_1, NULL,
1156 ret_void, PROP_FUNC(f_prop_type_delete)},
1157 {"prop_type_get", 1, 2, FEARG_1, NULL,
1158 ret_dict_any, PROP_FUNC(f_prop_type_get)},
1159 {"prop_type_list", 0, 1, FEARG_1, NULL,
1160 ret_list_string, PROP_FUNC(f_prop_type_list)},
1161 {"pum_getpos", 0, 0, 0, NULL,
1162 ret_dict_number, f_pum_getpos},
1163 {"pumvisible", 0, 0, 0, NULL,
1164 ret_number, f_pumvisible},
1165 {"py3eval", 1, 1, FEARG_1, NULL,
1166 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001167#ifdef FEAT_PYTHON3
Bram Moolenaar15c47602020-03-26 22:16:48 +01001168 f_py3eval
1169#else
1170 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001171#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001172 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001173 {"pyeval", 1, 1, FEARG_1, NULL,
1174 ret_any,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001175#ifdef FEAT_PYTHON
Bram Moolenaar15c47602020-03-26 22:16:48 +01001176 f_pyeval
1177#else
1178 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001179#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001180 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001181 {"pyxeval", 1, 1, FEARG_1, NULL,
1182 ret_any,
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001183#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
Bram Moolenaar15c47602020-03-26 22:16:48 +01001184 f_pyxeval
1185#else
1186 NULL
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01001187#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001188 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001189 {"rand", 0, 1, FEARG_1, NULL,
1190 ret_number, f_rand},
1191 {"range", 1, 3, FEARG_1, NULL,
1192 ret_list_number, f_range},
1193 {"readdir", 1, 3, FEARG_1, NULL,
1194 ret_list_string, f_readdir},
1195 {"readdirex", 1, 3, FEARG_1, NULL,
1196 ret_list_dict_any, f_readdirex},
1197 {"readfile", 1, 3, FEARG_1, NULL,
1198 ret_any, f_readfile},
1199 {"reduce", 2, 3, FEARG_1, NULL,
1200 ret_any, f_reduce},
1201 {"reg_executing", 0, 0, 0, NULL,
1202 ret_string, f_reg_executing},
1203 {"reg_recording", 0, 0, 0, NULL,
1204 ret_string, f_reg_recording},
1205 {"reltime", 0, 2, FEARG_1, NULL,
1206 ret_list_any, f_reltime},
1207 {"reltimefloat", 1, 1, FEARG_1, NULL,
1208 ret_float, FLOAT_FUNC(f_reltimefloat)},
1209 {"reltimestr", 1, 1, FEARG_1, NULL,
1210 ret_string, f_reltimestr},
1211 {"remote_expr", 2, 4, FEARG_1, NULL,
1212 ret_string, f_remote_expr},
1213 {"remote_foreground", 1, 1, FEARG_1, NULL,
1214 ret_string, f_remote_foreground},
1215 {"remote_peek", 1, 2, FEARG_1, NULL,
1216 ret_number, f_remote_peek},
1217 {"remote_read", 1, 2, FEARG_1, NULL,
1218 ret_string, f_remote_read},
1219 {"remote_send", 2, 3, FEARG_1, NULL,
1220 ret_string, f_remote_send},
1221 {"remote_startserver", 1, 1, FEARG_1, NULL,
1222 ret_void, f_remote_startserver},
1223 {"remove", 2, 3, FEARG_1, NULL,
1224 ret_remove, f_remove},
1225 {"rename", 2, 2, FEARG_1, NULL,
1226 ret_number, f_rename},
1227 {"repeat", 2, 2, FEARG_1, NULL,
1228 ret_first_arg, f_repeat},
1229 {"resolve", 1, 1, FEARG_1, NULL,
1230 ret_string, f_resolve},
1231 {"reverse", 1, 1, FEARG_1, NULL,
1232 ret_first_arg, f_reverse},
1233 {"round", 1, 1, FEARG_1, NULL,
1234 ret_float, FLOAT_FUNC(f_round)},
1235 {"rubyeval", 1, 1, FEARG_1, NULL,
1236 ret_any,
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001237#ifdef FEAT_RUBY
Bram Moolenaar15c47602020-03-26 22:16:48 +01001238 f_rubyeval
1239#else
1240 NULL
Bram Moolenaare99be0e2019-03-26 22:51:09 +01001241#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001242 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001243 {"screenattr", 2, 2, FEARG_1, NULL,
1244 ret_number, f_screenattr},
1245 {"screenchar", 2, 2, FEARG_1, NULL,
1246 ret_number, f_screenchar},
1247 {"screenchars", 2, 2, FEARG_1, NULL,
1248 ret_list_number, f_screenchars},
1249 {"screencol", 0, 0, 0, NULL,
1250 ret_number, f_screencol},
1251 {"screenpos", 3, 3, FEARG_1, NULL,
1252 ret_dict_number, f_screenpos},
1253 {"screenrow", 0, 0, 0, NULL,
1254 ret_number, f_screenrow},
1255 {"screenstring", 2, 2, FEARG_1, NULL,
1256 ret_string, f_screenstring},
1257 {"search", 1, 5, FEARG_1, NULL,
1258 ret_number, f_search},
1259 {"searchcount", 0, 1, FEARG_1, NULL,
1260 ret_dict_any, f_searchcount},
1261 {"searchdecl", 1, 3, FEARG_1, NULL,
1262 ret_number, f_searchdecl},
1263 {"searchpair", 3, 7, 0, NULL,
1264 ret_number, f_searchpair},
1265 {"searchpairpos", 3, 7, 0, NULL,
1266 ret_list_number, f_searchpairpos},
1267 {"searchpos", 1, 5, FEARG_1, NULL,
1268 ret_list_number, f_searchpos},
1269 {"server2client", 2, 2, FEARG_1, NULL,
1270 ret_number, f_server2client},
1271 {"serverlist", 0, 0, 0, NULL,
1272 ret_string, f_serverlist},
1273 {"setbufline", 3, 3, FEARG_3, NULL,
1274 ret_number, f_setbufline},
1275 {"setbufvar", 3, 3, FEARG_3, NULL,
1276 ret_void, f_setbufvar},
1277 {"setcellwidths", 1, 1, FEARG_1, NULL,
1278 ret_void, f_setcellwidths},
1279 {"setcharsearch", 1, 1, FEARG_1, NULL,
1280 ret_void, f_setcharsearch},
1281 {"setcmdpos", 1, 1, FEARG_1, NULL,
1282 ret_number, f_setcmdpos},
1283 {"setenv", 2, 2, FEARG_2, NULL,
1284 ret_void, f_setenv},
1285 {"setfperm", 2, 2, FEARG_1, NULL,
1286 ret_number, f_setfperm},
1287 {"setline", 2, 2, FEARG_2, NULL,
1288 ret_number, f_setline},
1289 {"setloclist", 2, 4, FEARG_2, NULL,
1290 ret_number, f_setloclist},
1291 {"setmatches", 1, 2, FEARG_1, NULL,
1292 ret_number, f_setmatches},
1293 {"setpos", 2, 2, FEARG_2, NULL,
1294 ret_number, f_setpos},
1295 {"setqflist", 1, 3, FEARG_1, NULL,
1296 ret_number, f_setqflist},
1297 {"setreg", 2, 3, FEARG_2, NULL,
1298 ret_number, f_setreg},
1299 {"settabvar", 3, 3, FEARG_3, NULL,
1300 ret_void, f_settabvar},
1301 {"settabwinvar", 4, 4, FEARG_4, NULL,
1302 ret_void, f_settabwinvar},
1303 {"settagstack", 2, 3, FEARG_2, NULL,
1304 ret_number, f_settagstack},
1305 {"setwinvar", 3, 3, FEARG_3, NULL,
1306 ret_void, f_setwinvar},
1307 {"sha256", 1, 1, FEARG_1, NULL,
1308 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001309#ifdef FEAT_CRYPT
Bram Moolenaar15c47602020-03-26 22:16:48 +01001310 f_sha256
1311#else
1312 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001313#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001314 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001315 {"shellescape", 1, 2, FEARG_1, NULL,
1316 ret_string, f_shellescape},
1317 {"shiftwidth", 0, 1, FEARG_1, NULL,
1318 ret_number, f_shiftwidth},
1319 {"sign_define", 1, 2, FEARG_1, NULL,
1320 ret_any, SIGN_FUNC(f_sign_define)},
1321 {"sign_getdefined", 0, 1, FEARG_1, NULL,
1322 ret_list_dict_any, SIGN_FUNC(f_sign_getdefined)},
1323 {"sign_getplaced", 0, 2, FEARG_1, NULL,
1324 ret_list_dict_any, SIGN_FUNC(f_sign_getplaced)},
1325 {"sign_jump", 3, 3, FEARG_1, NULL,
1326 ret_number, SIGN_FUNC(f_sign_jump)},
1327 {"sign_place", 4, 5, FEARG_1, NULL,
1328 ret_number, SIGN_FUNC(f_sign_place)},
1329 {"sign_placelist", 1, 1, FEARG_1, NULL,
1330 ret_list_number, SIGN_FUNC(f_sign_placelist)},
1331 {"sign_undefine", 0, 1, FEARG_1, NULL,
1332 ret_number, SIGN_FUNC(f_sign_undefine)},
1333 {"sign_unplace", 1, 2, FEARG_1, NULL,
1334 ret_number, SIGN_FUNC(f_sign_unplace)},
1335 {"sign_unplacelist", 1, 2, FEARG_1, NULL,
1336 ret_list_number, SIGN_FUNC(f_sign_unplacelist)},
1337 {"simplify", 1, 1, FEARG_1, NULL,
1338 ret_string, f_simplify},
1339 {"sin", 1, 1, FEARG_1, NULL,
1340 ret_float, FLOAT_FUNC(f_sin)},
1341 {"sinh", 1, 1, FEARG_1, NULL,
1342 ret_float, FLOAT_FUNC(f_sinh)},
1343 {"sort", 1, 3, FEARG_1, NULL,
1344 ret_first_arg, f_sort},
1345 {"sound_clear", 0, 0, 0, NULL,
1346 ret_void, SOUND_FUNC(f_sound_clear)},
1347 {"sound_playevent", 1, 2, FEARG_1, NULL,
1348 ret_number, SOUND_FUNC(f_sound_playevent)},
1349 {"sound_playfile", 1, 2, FEARG_1, NULL,
1350 ret_number, SOUND_FUNC(f_sound_playfile)},
1351 {"sound_stop", 1, 1, FEARG_1, NULL,
1352 ret_void, SOUND_FUNC(f_sound_stop)},
1353 {"soundfold", 1, 1, FEARG_1, NULL,
1354 ret_string, f_soundfold},
1355 {"spellbadword", 0, 1, FEARG_1, NULL,
1356 ret_list_string, f_spellbadword},
1357 {"spellsuggest", 1, 3, FEARG_1, NULL,
1358 ret_list_string, f_spellsuggest},
1359 {"split", 1, 3, FEARG_1, NULL,
1360 ret_list_string, f_split},
1361 {"sqrt", 1, 1, FEARG_1, NULL,
1362 ret_float, FLOAT_FUNC(f_sqrt)},
1363 {"srand", 0, 1, FEARG_1, NULL,
1364 ret_list_number, f_srand},
1365 {"state", 0, 1, FEARG_1, NULL,
1366 ret_string, f_state},
1367 {"str2float", 1, 1, FEARG_1, NULL,
1368 ret_float, FLOAT_FUNC(f_str2float)},
1369 {"str2list", 1, 2, FEARG_1, NULL,
1370 ret_list_number, f_str2list},
1371 {"str2nr", 1, 3, FEARG_1, NULL,
1372 ret_number, f_str2nr},
1373 {"strcharpart", 2, 3, FEARG_1, NULL,
1374 ret_string, f_strcharpart},
1375 {"strchars", 1, 2, FEARG_1, NULL,
1376 ret_number, f_strchars},
1377 {"strdisplaywidth", 1, 2, FEARG_1, NULL,
1378 ret_number, f_strdisplaywidth},
1379 {"strftime", 1, 2, FEARG_1, NULL,
1380 ret_string,
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001381#ifdef HAVE_STRFTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001382 f_strftime
1383#else
1384 NULL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001385#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001386 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001387 {"strgetchar", 2, 2, FEARG_1, NULL,
1388 ret_number, f_strgetchar},
1389 {"stridx", 2, 3, FEARG_1, NULL,
1390 ret_number, f_stridx},
1391 {"string", 1, 1, FEARG_1, NULL,
1392 ret_string, f_string},
1393 {"strlen", 1, 1, FEARG_1, NULL,
1394 ret_number, f_strlen},
1395 {"strpart", 2, 4, FEARG_1, NULL,
1396 ret_string, f_strpart},
1397 {"strptime", 2, 2, FEARG_1, NULL,
1398 ret_number,
Bram Moolenaar10455d42019-11-21 15:36:18 +01001399#ifdef HAVE_STRPTIME
Bram Moolenaar15c47602020-03-26 22:16:48 +01001400 f_strptime
1401#else
1402 NULL
Bram Moolenaar10455d42019-11-21 15:36:18 +01001403#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001404 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001405 {"strridx", 2, 3, FEARG_1, NULL,
1406 ret_number, f_strridx},
1407 {"strtrans", 1, 1, FEARG_1, NULL,
1408 ret_string, f_strtrans},
1409 {"strwidth", 1, 1, FEARG_1, NULL,
1410 ret_number, f_strwidth},
1411 {"submatch", 1, 2, FEARG_1, NULL,
1412 ret_string, f_submatch},
1413 {"substitute", 4, 4, FEARG_1, NULL,
1414 ret_string, f_substitute},
1415 {"swapinfo", 1, 1, FEARG_1, NULL,
1416 ret_dict_any, f_swapinfo},
1417 {"swapname", 1, 1, FEARG_1, NULL,
1418 ret_string, f_swapname},
1419 {"synID", 3, 3, 0, NULL,
1420 ret_number, f_synID},
1421 {"synIDattr", 2, 3, FEARG_1, NULL,
1422 ret_string, f_synIDattr},
1423 {"synIDtrans", 1, 1, FEARG_1, NULL,
1424 ret_number, f_synIDtrans},
1425 {"synconcealed", 2, 2, 0, NULL,
1426 ret_list_any, f_synconcealed},
1427 {"synstack", 2, 2, 0, NULL,
1428 ret_list_number, f_synstack},
1429 {"system", 1, 2, FEARG_1, NULL,
1430 ret_string, f_system},
1431 {"systemlist", 1, 2, FEARG_1, NULL,
1432 ret_list_string, f_systemlist},
1433 {"tabpagebuflist", 0, 1, FEARG_1, NULL,
1434 ret_list_number, f_tabpagebuflist},
1435 {"tabpagenr", 0, 1, 0, NULL,
1436 ret_number, f_tabpagenr},
1437 {"tabpagewinnr", 1, 2, FEARG_1, NULL,
1438 ret_number, f_tabpagewinnr},
1439 {"tagfiles", 0, 0, 0, NULL,
1440 ret_list_string, f_tagfiles},
1441 {"taglist", 1, 2, FEARG_1, NULL,
1442 ret_list_dict_any, f_taglist},
1443 {"tan", 1, 1, FEARG_1, NULL,
1444 ret_float, FLOAT_FUNC(f_tan)},
1445 {"tanh", 1, 1, FEARG_1, NULL,
1446 ret_float, FLOAT_FUNC(f_tanh)},
1447 {"tempname", 0, 0, 0, NULL,
1448 ret_string, f_tempname},
1449 {"term_dumpdiff", 2, 3, FEARG_1, NULL,
1450 ret_number, TERM_FUNC(f_term_dumpdiff)},
1451 {"term_dumpload", 1, 2, FEARG_1, NULL,
1452 ret_number, TERM_FUNC(f_term_dumpload)},
1453 {"term_dumpwrite", 2, 3, FEARG_2, NULL,
1454 ret_void, TERM_FUNC(f_term_dumpwrite)},
1455 {"term_getaltscreen", 1, 1, FEARG_1, NULL,
1456 ret_number, TERM_FUNC(f_term_getaltscreen)},
1457 {"term_getansicolors", 1, 1, FEARG_1, NULL,
1458 ret_list_string,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001459#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001460 f_term_getansicolors
1461#else
1462 NULL
Bram Moolenaarc6df10e2017-07-29 20:15:08 +02001463#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001464 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001465 {"term_getattr", 2, 2, FEARG_1, NULL,
1466 ret_number, TERM_FUNC(f_term_getattr)},
1467 {"term_getcursor", 1, 1, FEARG_1, NULL,
1468 ret_list_any, TERM_FUNC(f_term_getcursor)},
1469 {"term_getjob", 1, 1, FEARG_1, NULL,
1470 ret_job, TERM_FUNC(f_term_getjob)},
1471 {"term_getline", 2, 2, FEARG_1, NULL,
1472 ret_string, TERM_FUNC(f_term_getline)},
1473 {"term_getscrolled", 1, 1, FEARG_1, NULL,
1474 ret_number, TERM_FUNC(f_term_getscrolled)},
1475 {"term_getsize", 1, 1, FEARG_1, NULL,
1476 ret_list_number, TERM_FUNC(f_term_getsize)},
1477 {"term_getstatus", 1, 1, FEARG_1, NULL,
1478 ret_string, TERM_FUNC(f_term_getstatus)},
1479 {"term_gettitle", 1, 1, FEARG_1, NULL,
1480 ret_string, TERM_FUNC(f_term_gettitle)},
1481 {"term_gettty", 1, 2, FEARG_1, NULL,
1482 ret_string, TERM_FUNC(f_term_gettty)},
1483 {"term_list", 0, 0, 0, NULL,
1484 ret_list_number, TERM_FUNC(f_term_list)},
1485 {"term_scrape", 2, 2, FEARG_1, NULL,
1486 ret_list_dict_any, TERM_FUNC(f_term_scrape)},
1487 {"term_sendkeys", 2, 2, FEARG_1, NULL,
1488 ret_void, TERM_FUNC(f_term_sendkeys)},
1489 {"term_setansicolors", 2, 2, FEARG_1, NULL,
1490 ret_void,
Bram Moolenaarbd5e6222020-03-26 23:13:34 +01001491#if defined(FEAT_TERMINAL) && (defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS))
Bram Moolenaar15c47602020-03-26 22:16:48 +01001492 f_term_setansicolors
1493#else
1494 NULL
1495#endif
1496 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001497 {"term_setapi", 2, 2, FEARG_1, NULL,
1498 ret_void, TERM_FUNC(f_term_setapi)},
1499 {"term_setkill", 2, 2, FEARG_1, NULL,
1500 ret_void, TERM_FUNC(f_term_setkill)},
1501 {"term_setrestore", 2, 2, FEARG_1, NULL,
1502 ret_void, TERM_FUNC(f_term_setrestore)},
1503 {"term_setsize", 3, 3, FEARG_1, NULL,
1504 ret_void, TERM_FUNC(f_term_setsize)},
1505 {"term_start", 1, 2, FEARG_1, NULL,
1506 ret_number, TERM_FUNC(f_term_start)},
1507 {"term_wait", 1, 2, FEARG_1, NULL,
1508 ret_void, TERM_FUNC(f_term_wait)},
1509 {"terminalprops", 0, 0, 0, NULL,
1510 ret_dict_string, f_terminalprops},
1511 {"test_alloc_fail", 3, 3, FEARG_1, NULL,
1512 ret_void, f_test_alloc_fail},
1513 {"test_autochdir", 0, 0, 0, NULL,
1514 ret_void, f_test_autochdir},
1515 {"test_feedinput", 1, 1, FEARG_1, NULL,
1516 ret_void, f_test_feedinput},
1517 {"test_garbagecollect_now", 0, 0, 0, NULL,
1518 ret_void, f_test_garbagecollect_now},
1519 {"test_garbagecollect_soon", 0, 0, 0, NULL,
1520 ret_void, f_test_garbagecollect_soon},
1521 {"test_getvalue", 1, 1, FEARG_1, NULL,
1522 ret_number, f_test_getvalue},
1523 {"test_ignore_error", 1, 1, FEARG_1, NULL,
1524 ret_void, f_test_ignore_error},
1525 {"test_null_blob", 0, 0, 0, NULL,
1526 ret_blob, f_test_null_blob},
1527 {"test_null_channel", 0, 0, 0, NULL,
1528 ret_channel, JOB_FUNC(f_test_null_channel)},
1529 {"test_null_dict", 0, 0, 0, NULL,
1530 ret_dict_any, f_test_null_dict},
1531 {"test_null_function", 0, 0, 0, NULL,
1532 ret_func_any, f_test_null_function},
1533 {"test_null_job", 0, 0, 0, NULL,
1534 ret_job, JOB_FUNC(f_test_null_job)},
1535 {"test_null_list", 0, 0, 0, NULL,
1536 ret_list_any, f_test_null_list},
1537 {"test_null_partial", 0, 0, 0, NULL,
1538 ret_func_any, f_test_null_partial},
1539 {"test_null_string", 0, 0, 0, NULL,
1540 ret_string, f_test_null_string},
1541 {"test_option_not_set", 1, 1, FEARG_1, NULL,
1542 ret_void, f_test_option_not_set},
1543 {"test_override", 2, 2, FEARG_2, NULL,
1544 ret_void, f_test_override},
1545 {"test_refcount", 1, 1, FEARG_1, NULL,
1546 ret_number, f_test_refcount},
1547 {"test_scrollbar", 3, 3, FEARG_2, NULL,
1548 ret_void,
Bram Moolenaarab186732018-09-14 21:27:06 +02001549#ifdef FEAT_GUI
Bram Moolenaar15c47602020-03-26 22:16:48 +01001550 f_test_scrollbar
1551#else
1552 NULL
Bram Moolenaarab186732018-09-14 21:27:06 +02001553#endif
Bram Moolenaar15c47602020-03-26 22:16:48 +01001554 },
Bram Moolenaar94738d82020-10-21 14:25:07 +02001555 {"test_setmouse", 2, 2, 0, NULL,
1556 ret_void, f_test_setmouse},
1557 {"test_settime", 1, 1, FEARG_1, NULL,
1558 ret_void, f_test_settime},
1559 {"test_srand_seed", 0, 1, FEARG_1, NULL,
1560 ret_void, f_test_srand_seed},
1561 {"test_unknown", 0, 0, 0, NULL,
1562 ret_any, f_test_unknown},
1563 {"test_void", 0, 0, 0, NULL,
1564 ret_void, f_test_void},
1565 {"timer_info", 0, 1, FEARG_1, NULL,
1566 ret_list_dict_any, TIMER_FUNC(f_timer_info)},
1567 {"timer_pause", 2, 2, FEARG_1, NULL,
1568 ret_void, TIMER_FUNC(f_timer_pause)},
1569 {"timer_start", 2, 3, FEARG_1, NULL,
1570 ret_number, TIMER_FUNC(f_timer_start)},
1571 {"timer_stop", 1, 1, FEARG_1, NULL,
1572 ret_void, TIMER_FUNC(f_timer_stop)},
1573 {"timer_stopall", 0, 0, 0, NULL,
1574 ret_void, TIMER_FUNC(f_timer_stopall)},
1575 {"tolower", 1, 1, FEARG_1, NULL,
1576 ret_string, f_tolower},
1577 {"toupper", 1, 1, FEARG_1, NULL,
1578 ret_string, f_toupper},
1579 {"tr", 3, 3, FEARG_1, NULL,
1580 ret_string, f_tr},
1581 {"trim", 1, 3, FEARG_1, NULL,
1582 ret_string, f_trim},
1583 {"trunc", 1, 1, FEARG_1, NULL,
1584 ret_float, FLOAT_FUNC(f_trunc)},
1585 {"type", 1, 1, FEARG_1, NULL,
1586 ret_number, f_type},
1587 {"undofile", 1, 1, FEARG_1, NULL,
1588 ret_string, f_undofile},
1589 {"undotree", 0, 0, 0, NULL,
1590 ret_dict_any, f_undotree},
1591 {"uniq", 1, 3, FEARG_1, NULL,
1592 ret_list_any, f_uniq},
1593 {"values", 1, 1, FEARG_1, NULL,
1594 ret_list_any, f_values},
1595 {"virtcol", 1, 1, FEARG_1, NULL,
1596 ret_number, f_virtcol},
1597 {"visualmode", 0, 1, 0, NULL,
1598 ret_string, f_visualmode},
1599 {"wildmenumode", 0, 0, 0, NULL,
1600 ret_number, f_wildmenumode},
1601 {"win_execute", 2, 3, FEARG_2, NULL,
1602 ret_string, f_win_execute},
1603 {"win_findbuf", 1, 1, FEARG_1, NULL,
1604 ret_list_number, f_win_findbuf},
1605 {"win_getid", 0, 2, FEARG_1, NULL,
1606 ret_number, f_win_getid},
1607 {"win_gettype", 0, 1, FEARG_1, NULL,
1608 ret_string, f_win_gettype},
1609 {"win_gotoid", 1, 1, FEARG_1, NULL,
1610 ret_number, f_win_gotoid},
1611 {"win_id2tabwin", 1, 1, FEARG_1, NULL,
1612 ret_list_number, f_win_id2tabwin},
1613 {"win_id2win", 1, 1, FEARG_1, NULL,
1614 ret_number, f_win_id2win},
1615 {"win_screenpos", 1, 1, FEARG_1, NULL,
1616 ret_list_number, f_win_screenpos},
1617 {"win_splitmove", 2, 3, FEARG_1, NULL,
1618 ret_number, f_win_splitmove},
1619 {"winbufnr", 1, 1, FEARG_1, NULL,
1620 ret_number, f_winbufnr},
1621 {"wincol", 0, 0, 0, NULL,
1622 ret_number, f_wincol},
1623 {"windowsversion", 0, 0, 0, NULL,
1624 ret_string, f_windowsversion},
1625 {"winheight", 1, 1, FEARG_1, NULL,
1626 ret_number, f_winheight},
1627 {"winlayout", 0, 1, FEARG_1, NULL,
1628 ret_list_any, f_winlayout},
1629 {"winline", 0, 0, 0, NULL,
1630 ret_number, f_winline},
1631 {"winnr", 0, 1, FEARG_1, NULL,
1632 ret_number, f_winnr},
1633 {"winrestcmd", 0, 0, 0, NULL,
1634 ret_string, f_winrestcmd},
1635 {"winrestview", 1, 1, FEARG_1, NULL,
1636 ret_void, f_winrestview},
1637 {"winsaveview", 0, 0, 0, NULL,
1638 ret_dict_any, f_winsaveview},
1639 {"winwidth", 1, 1, FEARG_1, NULL,
1640 ret_number, f_winwidth},
1641 {"wordcount", 0, 0, 0, NULL,
1642 ret_dict_number, f_wordcount},
1643 {"writefile", 2, 3, FEARG_1, NULL,
1644 ret_number, f_writefile},
1645 {"xor", 2, 2, FEARG_1, NULL,
1646 ret_number, f_xor},
Bram Moolenaarac92e252019-08-03 21:58:38 +02001647};
1648
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001649/*
1650 * Function given to ExpandGeneric() to obtain the list of internal
1651 * or user defined function names.
1652 */
1653 char_u *
1654get_function_name(expand_T *xp, int idx)
1655{
1656 static int intidx = -1;
1657 char_u *name;
1658
1659 if (idx == 0)
1660 intidx = -1;
1661 if (intidx < 0)
1662 {
1663 name = get_user_func_name(xp, idx);
1664 if (name != NULL)
1665 return name;
1666 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001667 if (++intidx < (int)(sizeof(global_functions) / sizeof(funcentry_T)))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001668 {
Bram Moolenaarac92e252019-08-03 21:58:38 +02001669 STRCPY(IObuff, global_functions[intidx].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001670 STRCAT(IObuff, "(");
Bram Moolenaarac92e252019-08-03 21:58:38 +02001671 if (global_functions[intidx].f_max_argc == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001672 STRCAT(IObuff, ")");
1673 return IObuff;
1674 }
1675
1676 return NULL;
1677}
1678
1679/*
1680 * Function given to ExpandGeneric() to obtain the list of internal or
1681 * user defined variable or function names.
1682 */
1683 char_u *
1684get_expr_name(expand_T *xp, int idx)
1685{
1686 static int intidx = -1;
1687 char_u *name;
1688
1689 if (idx == 0)
1690 intidx = -1;
1691 if (intidx < 0)
1692 {
1693 name = get_function_name(xp, idx);
1694 if (name != NULL)
1695 return name;
1696 }
1697 return get_user_var_name(xp, ++intidx);
1698}
1699
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001700/*
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001701 * Find internal function "name" in table "global_functions".
Bram Moolenaar15c47602020-03-26 22:16:48 +01001702 * Return index, or -1 if not found or "implemented" is TRUE and the function
1703 * is not implemented.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001704 */
Bram Moolenaar15c47602020-03-26 22:16:48 +01001705 static int
1706find_internal_func_opt(char_u *name, int implemented)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001707{
1708 int first = 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001709 int last;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001710 int cmp;
1711 int x;
1712
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001713 last = (int)(sizeof(global_functions) / sizeof(funcentry_T)) - 1;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001714
1715 // Find the function name in the table. Binary search.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001716 while (first <= last)
1717 {
1718 x = first + ((unsigned)(last - first) >> 1);
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001719 cmp = STRCMP(name, global_functions[x].f_name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001720 if (cmp < 0)
1721 last = x - 1;
1722 else if (cmp > 0)
1723 first = x + 1;
Bram Moolenaar15c47602020-03-26 22:16:48 +01001724 else if (implemented && global_functions[x].f_func == NULL)
1725 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001726 else
1727 return x;
1728 }
1729 return -1;
1730}
1731
Bram Moolenaar15c47602020-03-26 22:16:48 +01001732/*
1733 * Find internal function "name" in table "global_functions".
1734 * Return index, or -1 if not found or the function is not implemented.
1735 */
1736 int
1737find_internal_func(char_u *name)
1738{
1739 return find_internal_func_opt(name, TRUE);
1740}
1741
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001742 int
Bram Moolenaarac92e252019-08-03 21:58:38 +02001743has_internal_func(char_u *name)
1744{
Bram Moolenaar15c47602020-03-26 22:16:48 +01001745 return find_internal_func_opt(name, TRUE) >= 0;
1746}
1747
1748 static int
1749has_internal_func_name(char_u *name)
1750{
1751 return find_internal_func_opt(name, FALSE) >= 0;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001752}
1753
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001754 char *
1755internal_func_name(int idx)
1756{
1757 return global_functions[idx].f_name;
1758}
1759
Bram Moolenaar94738d82020-10-21 14:25:07 +02001760/*
1761 * Check the argument types for builting function "idx".
1762 * Uses the list of types on the type stack: "types".
1763 * Return FAIL and gives an error message when a type is wrong.
1764 */
1765 int
1766internal_func_check_arg_types(type_T *types, int idx, int argcount)
1767{
1768 argcheck_T *argchecks = global_functions[idx].f_argcheck;
1769 int i;
1770
1771 if (argchecks != NULL)
1772 {
1773 argcontext_T context;
1774
1775 context.arg_count = argcount;
1776 for (i = 0; i < argcount; ++i)
1777 if (argchecks[i] != NULL)
1778 {
1779 context.arg_idx = i;
1780 if (argchecks[i](types + i, &context) == FAIL)
1781 return FAIL;
1782 }
1783 }
1784 return OK;
1785}
1786
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001787 type_T *
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001788internal_func_ret_type(int idx, int argcount, type_T **argtypes)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001789{
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01001790 return global_functions[idx].f_retfunc(argcount, argtypes);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001791}
1792
1793/*
1794 * Check the argument count to use for internal function "idx".
Bram Moolenaar389df252020-07-09 21:20:47 +02001795 * Returns -1 for failure, 0 if no method base accepted, 1 if method base is
1796 * first argument, 2 if method base is second argument, etc. 9 if method base
1797 * is last argument.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001798 */
1799 int
1800check_internal_func(int idx, int argcount)
1801{
1802 int res;
1803 char *name;
1804
1805 if (argcount < global_functions[idx].f_min_argc)
1806 res = FCERR_TOOFEW;
1807 else if (argcount > global_functions[idx].f_max_argc)
1808 res = FCERR_TOOMANY;
1809 else
Bram Moolenaar389df252020-07-09 21:20:47 +02001810 return global_functions[idx].f_argtype;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001811
1812 name = internal_func_name(idx);
1813 if (res == FCERR_TOOMANY)
1814 semsg(_(e_toomanyarg), name);
1815 else
1816 semsg(_(e_toofewarg), name);
Bram Moolenaar389df252020-07-09 21:20:47 +02001817 return -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001818}
1819
Bram Moolenaarac92e252019-08-03 21:58:38 +02001820 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001821call_internal_func(
1822 char_u *name,
1823 int argcount,
1824 typval_T *argvars,
1825 typval_T *rettv)
1826{
1827 int i;
1828
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001829 i = find_internal_func(name);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001830 if (i < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001831 return FCERR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001832 if (argcount < global_functions[i].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001833 return FCERR_TOOFEW;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001834 if (argcount > global_functions[i].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001835 return FCERR_TOOMANY;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001836 argvars[argcount].v_type = VAR_UNKNOWN;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001837 global_functions[i].f_func(argvars, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001838 return FCERR_NONE;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001839}
1840
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001841 void
1842call_internal_func_by_idx(
1843 int idx,
1844 typval_T *argvars,
1845 typval_T *rettv)
1846{
1847 global_functions[idx].f_func(argvars, rettv);
1848}
1849
Bram Moolenaarac92e252019-08-03 21:58:38 +02001850/*
1851 * Invoke a method for base->method().
1852 */
1853 int
1854call_internal_method(
1855 char_u *name,
1856 int argcount,
1857 typval_T *argvars,
1858 typval_T *rettv,
1859 typval_T *basetv)
1860{
1861 int i;
1862 int fi;
1863 typval_T argv[MAX_FUNC_ARGS + 1];
1864
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001865 fi = find_internal_func(name);
Bram Moolenaar91746392019-08-16 22:22:31 +02001866 if (fi < 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001867 return FCERR_UNKNOWN;
Bram Moolenaar91746392019-08-16 22:22:31 +02001868 if (global_functions[fi].f_argtype == 0)
Bram Moolenaaref140542019-12-31 21:27:13 +01001869 return FCERR_NOTMETHOD;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001870 if (argcount + 1 < global_functions[fi].f_min_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001871 return FCERR_TOOFEW;
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001872 if (argcount + 1 > global_functions[fi].f_max_argc)
Bram Moolenaaref140542019-12-31 21:27:13 +01001873 return FCERR_TOOMANY;
Bram Moolenaarac92e252019-08-03 21:58:38 +02001874
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001875 if (global_functions[fi].f_argtype == FEARG_LAST)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001876 {
1877 // base value goes last
1878 for (i = 0; i < argcount; ++i)
1879 argv[i] = argvars[i];
1880 argv[argcount] = *basetv;
1881 }
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001882 else if (global_functions[fi].f_argtype == FEARG_2)
Bram Moolenaar25e42232019-08-04 15:04:10 +02001883 {
1884 // base value goes second
1885 argv[0] = argvars[0];
1886 argv[1] = *basetv;
1887 for (i = 1; i < argcount; ++i)
1888 argv[i + 1] = argvars[i];
1889 }
Bram Moolenaar24278d22019-08-16 21:49:22 +02001890 else if (global_functions[fi].f_argtype == FEARG_3)
1891 {
1892 // base value goes third
1893 argv[0] = argvars[0];
1894 argv[1] = argvars[1];
1895 argv[2] = *basetv;
1896 for (i = 2; i < argcount; ++i)
1897 argv[i + 1] = argvars[i];
1898 }
Bram Moolenaaraad222c2019-09-06 22:46:09 +02001899 else if (global_functions[fi].f_argtype == FEARG_4)
1900 {
1901 // base value goes fourth
1902 argv[0] = argvars[0];
1903 argv[1] = argvars[1];
1904 argv[2] = argvars[2];
1905 argv[3] = *basetv;
1906 for (i = 3; i < argcount; ++i)
1907 argv[i + 1] = argvars[i];
1908 }
Bram Moolenaar25e42232019-08-04 15:04:10 +02001909 else
1910 {
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001911 // FEARG_1: base value goes first
Bram Moolenaar25e42232019-08-04 15:04:10 +02001912 argv[0] = *basetv;
1913 for (i = 0; i < argcount; ++i)
1914 argv[i + 1] = argvars[i];
1915 }
Bram Moolenaarac92e252019-08-03 21:58:38 +02001916 argv[argcount + 1].v_type = VAR_UNKNOWN;
1917
Bram Moolenaar7a4ea1d2019-08-04 21:35:12 +02001918 global_functions[fi].f_func(argv, rettv);
Bram Moolenaaref140542019-12-31 21:27:13 +01001919 return FCERR_NONE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001920}
1921
1922/*
1923 * Return TRUE for a non-zero Number and a non-empty String.
1924 */
Bram Moolenaar0e57dd82019-09-16 22:56:03 +02001925 int
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001926non_zero_arg(typval_T *argvars)
1927{
1928 return ((argvars[0].v_type == VAR_NUMBER
1929 && argvars[0].vval.v_number != 0)
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001930 || (argvars[0].v_type == VAR_BOOL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001931 && argvars[0].vval.v_number == VVAL_TRUE)
1932 || (argvars[0].v_type == VAR_STRING
1933 && argvars[0].vval.v_string != NULL
1934 && *argvars[0].vval.v_string != NUL));
1935}
1936
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001937#ifdef FEAT_FLOAT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001938/*
1939 * Get the float value of "argvars[0]" into "f".
1940 * Returns FAIL when the argument is not a Number or Float.
1941 */
1942 static int
1943get_float_arg(typval_T *argvars, float_T *f)
1944{
1945 if (argvars[0].v_type == VAR_FLOAT)
1946 {
1947 *f = argvars[0].vval.v_float;
1948 return OK;
1949 }
1950 if (argvars[0].v_type == VAR_NUMBER)
1951 {
1952 *f = (float_T)argvars[0].vval.v_number;
1953 return OK;
1954 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001955 emsg(_("E808: Number or Float required"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001956 return FAIL;
1957}
1958
1959/*
1960 * "abs(expr)" function
1961 */
1962 static void
1963f_abs(typval_T *argvars, typval_T *rettv)
1964{
1965 if (argvars[0].v_type == VAR_FLOAT)
1966 {
1967 rettv->v_type = VAR_FLOAT;
1968 rettv->vval.v_float = fabs(argvars[0].vval.v_float);
1969 }
1970 else
1971 {
1972 varnumber_T n;
1973 int error = FALSE;
1974
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001975 n = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02001976 if (error)
1977 rettv->vval.v_number = -1;
1978 else if (n > 0)
1979 rettv->vval.v_number = n;
1980 else
1981 rettv->vval.v_number = -n;
1982 }
1983}
1984
1985/*
1986 * "acos()" function
1987 */
1988 static void
1989f_acos(typval_T *argvars, typval_T *rettv)
1990{
1991 float_T f = 0.0;
1992
1993 rettv->v_type = VAR_FLOAT;
1994 if (get_float_arg(argvars, &f) == OK)
1995 rettv->vval.v_float = acos(f);
1996 else
1997 rettv->vval.v_float = 0.0;
1998}
1999#endif
2000
2001/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002002 * "and(expr, expr)" function
2003 */
2004 static void
2005f_and(typval_T *argvars, typval_T *rettv)
2006{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002007 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
2008 & tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaarca851592018-06-06 21:04:07 +02002009}
2010
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002011#ifdef FEAT_FLOAT
2012/*
2013 * "asin()" function
2014 */
2015 static void
2016f_asin(typval_T *argvars, typval_T *rettv)
2017{
2018 float_T f = 0.0;
2019
2020 rettv->v_type = VAR_FLOAT;
2021 if (get_float_arg(argvars, &f) == OK)
2022 rettv->vval.v_float = asin(f);
2023 else
2024 rettv->vval.v_float = 0.0;
2025}
2026
2027/*
2028 * "atan()" function
2029 */
2030 static void
2031f_atan(typval_T *argvars, typval_T *rettv)
2032{
2033 float_T f = 0.0;
2034
2035 rettv->v_type = VAR_FLOAT;
2036 if (get_float_arg(argvars, &f) == OK)
2037 rettv->vval.v_float = atan(f);
2038 else
2039 rettv->vval.v_float = 0.0;
2040}
2041
2042/*
2043 * "atan2()" function
2044 */
2045 static void
2046f_atan2(typval_T *argvars, typval_T *rettv)
2047{
2048 float_T fx = 0.0, fy = 0.0;
2049
2050 rettv->v_type = VAR_FLOAT;
2051 if (get_float_arg(argvars, &fx) == OK
2052 && get_float_arg(&argvars[1], &fy) == OK)
2053 rettv->vval.v_float = atan2(fx, fy);
2054 else
2055 rettv->vval.v_float = 0.0;
2056}
2057#endif
2058
2059/*
Bram Moolenaar59716a22017-03-01 20:32:44 +01002060 * "balloon_show()" function
2061 */
2062#ifdef FEAT_BEVAL
2063 static void
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002064f_balloon_gettext(typval_T *argvars UNUSED, typval_T *rettv)
2065{
2066 rettv->v_type = VAR_STRING;
2067 if (balloonEval != NULL)
2068 {
2069 if (balloonEval->msg == NULL)
2070 rettv->vval.v_string = NULL;
2071 else
2072 rettv->vval.v_string = vim_strsave(balloonEval->msg);
2073 }
2074}
2075
2076 static void
Bram Moolenaar59716a22017-03-01 20:32:44 +01002077f_balloon_show(typval_T *argvars, typval_T *rettv UNUSED)
2078{
Bram Moolenaarcaf64342017-03-02 22:11:33 +01002079 if (balloonEval != NULL)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002080 {
2081 if (argvars[0].v_type == VAR_LIST
2082# ifdef FEAT_GUI
2083 && !gui.in_use
2084# endif
2085 )
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002086 {
2087 list_T *l = argvars[0].vval.v_list;
2088
2089 // empty list removes the balloon
2090 post_balloon(balloonEval, NULL,
2091 l == NULL || l->lv_len == 0 ? NULL : l);
2092 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002093 else
Bram Moolenaarbe0a2592019-05-09 13:50:16 +02002094 {
2095 char_u *mesg = tv_get_string_chk(&argvars[0]);
2096
2097 if (mesg != NULL)
2098 // empty string removes the balloon
2099 post_balloon(balloonEval, *mesg == NUL ? NULL : mesg, NULL);
2100 }
Bram Moolenaar246fe032017-11-19 19:56:27 +01002101 }
2102}
2103
Bram Moolenaar669a8282017-11-19 20:13:05 +01002104# if defined(FEAT_BEVAL_TERM)
Bram Moolenaar246fe032017-11-19 19:56:27 +01002105 static void
2106f_balloon_split(typval_T *argvars, typval_T *rettv UNUSED)
2107{
2108 if (rettv_list_alloc(rettv) == OK)
2109 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002110 char_u *msg = tv_get_string_chk(&argvars[0]);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002111
2112 if (msg != NULL)
2113 {
2114 pumitem_T *array;
2115 int size = split_message(msg, &array);
2116 int i;
2117
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002118 // Skip the first and last item, they are always empty.
Bram Moolenaar246fe032017-11-19 19:56:27 +01002119 for (i = 1; i < size - 1; ++i)
2120 list_append_string(rettv->vval.v_list, array[i].pum_text, -1);
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01002121 while (size > 0)
2122 vim_free(array[--size].pum_text);
Bram Moolenaar246fe032017-11-19 19:56:27 +01002123 vim_free(array);
2124 }
2125 }
Bram Moolenaar59716a22017-03-01 20:32:44 +01002126}
Bram Moolenaar669a8282017-11-19 20:13:05 +01002127# endif
Bram Moolenaar59716a22017-03-01 20:32:44 +01002128#endif
2129
2130/*
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002131 * Get the buffer from "arg" and give an error and return NULL if it is not
2132 * valid.
2133 */
Bram Moolenaara3347722019-05-11 21:14:24 +02002134 buf_T *
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002135get_buf_arg(typval_T *arg)
2136{
2137 buf_T *buf;
2138
2139 ++emsg_off;
2140 buf = tv_get_buf(arg, FALSE);
2141 --emsg_off;
2142 if (buf == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002143 semsg(_("E158: Invalid buffer name: %s"), tv_get_string(arg));
Bram Moolenaar6b7b7192019-01-11 13:42:41 +01002144 return buf;
2145}
2146
2147/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002148 * "byte2line(byte)" function
2149 */
2150 static void
2151f_byte2line(typval_T *argvars UNUSED, typval_T *rettv)
2152{
2153#ifndef FEAT_BYTEOFF
2154 rettv->vval.v_number = -1;
2155#else
2156 long boff = 0;
2157
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002158 boff = tv_get_number(&argvars[0]) - 1; // boff gets -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002159 if (boff < 0)
2160 rettv->vval.v_number = -1;
2161 else
2162 rettv->vval.v_number = ml_find_line_or_offset(curbuf,
2163 (linenr_T)0, &boff);
2164#endif
2165}
2166
2167 static void
2168byteidx(typval_T *argvars, typval_T *rettv, int comp UNUSED)
2169{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002170 char_u *t;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002171 char_u *str;
2172 varnumber_T idx;
2173
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002174 str = tv_get_string_chk(&argvars[0]);
2175 idx = tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002176 rettv->vval.v_number = -1;
2177 if (str == NULL || idx < 0)
2178 return;
2179
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002180 t = str;
2181 for ( ; idx > 0; idx--)
2182 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002183 if (*t == NUL) // EOL reached
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002184 return;
2185 if (enc_utf8 && comp)
2186 t += utf_ptr2len(t);
2187 else
2188 t += (*mb_ptr2len)(t);
2189 }
2190 rettv->vval.v_number = (varnumber_T)(t - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002191}
2192
2193/*
2194 * "byteidx()" function
2195 */
2196 static void
2197f_byteidx(typval_T *argvars, typval_T *rettv)
2198{
2199 byteidx(argvars, rettv, FALSE);
2200}
2201
2202/*
2203 * "byteidxcomp()" function
2204 */
2205 static void
2206f_byteidxcomp(typval_T *argvars, typval_T *rettv)
2207{
2208 byteidx(argvars, rettv, TRUE);
2209}
2210
2211/*
2212 * "call(func, arglist [, dict])" function
2213 */
2214 static void
2215f_call(typval_T *argvars, typval_T *rettv)
2216{
2217 char_u *func;
2218 partial_T *partial = NULL;
2219 dict_T *selfdict = NULL;
2220
2221 if (argvars[1].v_type != VAR_LIST)
2222 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002223 emsg(_(e_listreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002224 return;
2225 }
2226 if (argvars[1].vval.v_list == NULL)
2227 return;
2228
2229 if (argvars[0].v_type == VAR_FUNC)
2230 func = argvars[0].vval.v_string;
2231 else if (argvars[0].v_type == VAR_PARTIAL)
2232 {
2233 partial = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02002234 func = partial_name(partial);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002235 }
2236 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002237 func = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002238 if (*func == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002239 return; // type error or empty name
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002240
2241 if (argvars[2].v_type != VAR_UNKNOWN)
2242 {
2243 if (argvars[2].v_type != VAR_DICT)
2244 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002245 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002246 return;
2247 }
2248 selfdict = argvars[2].vval.v_dict;
2249 }
2250
2251 (void)func_call(func, &argvars[1], partial, selfdict, rettv);
2252}
2253
2254#ifdef FEAT_FLOAT
2255/*
2256 * "ceil({float})" function
2257 */
2258 static void
2259f_ceil(typval_T *argvars, typval_T *rettv)
2260{
2261 float_T f = 0.0;
2262
2263 rettv->v_type = VAR_FLOAT;
2264 if (get_float_arg(argvars, &f) == OK)
2265 rettv->vval.v_float = ceil(f);
2266 else
2267 rettv->vval.v_float = 0.0;
2268}
2269#endif
2270
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002271/*
2272 * "changenr()" function
2273 */
2274 static void
2275f_changenr(typval_T *argvars UNUSED, typval_T *rettv)
2276{
2277 rettv->vval.v_number = curbuf->b_u_seq_cur;
2278}
2279
2280/*
2281 * "char2nr(string)" function
2282 */
2283 static void
2284f_char2nr(typval_T *argvars, typval_T *rettv)
2285{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002286 if (has_mbyte)
2287 {
2288 int utf8 = 0;
2289
2290 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar24f77502020-09-04 19:50:57 +02002291 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002292
2293 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01002294 rettv->vval.v_number = utf_ptr2char(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002295 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002296 rettv->vval.v_number = (*mb_ptr2char)(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002297 }
2298 else
Bram Moolenaar13505972019-01-24 15:04:48 +01002299 rettv->vval.v_number = tv_get_string(&argvars[0])[0];
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002300}
2301
Bram Moolenaar29b7d7a2019-07-22 23:03:57 +02002302 win_T *
Bram Moolenaaraff74912019-03-30 18:11:49 +01002303get_optional_window(typval_T *argvars, int idx)
2304{
2305 win_T *win = curwin;
2306
2307 if (argvars[idx].v_type != VAR_UNKNOWN)
2308 {
2309 win = find_win_by_nr_or_id(&argvars[idx]);
2310 if (win == NULL)
2311 {
2312 emsg(_(e_invalwindow));
2313 return NULL;
2314 }
2315 }
2316 return win;
2317}
2318
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002319/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002320 * "col(string)" function
2321 */
2322 static void
2323f_col(typval_T *argvars, typval_T *rettv)
2324{
2325 colnr_T col = 0;
2326 pos_T *fp;
2327 int fnum = curbuf->b_fnum;
2328
2329 fp = var2fpos(&argvars[0], FALSE, &fnum);
2330 if (fp != NULL && fnum == curbuf->b_fnum)
2331 {
2332 if (fp->col == MAXCOL)
2333 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002334 // '> can be MAXCOL, get the length of the line then
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002335 if (fp->lnum <= curbuf->b_ml.ml_line_count)
2336 col = (colnr_T)STRLEN(ml_get(fp->lnum)) + 1;
2337 else
2338 col = MAXCOL;
2339 }
2340 else
2341 {
2342 col = fp->col + 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002343 // col(".") when the cursor is on the NUL at the end of the line
2344 // because of "coladd" can be seen as an extra column.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002345 if (virtual_active() && fp == &curwin->w_cursor)
2346 {
2347 char_u *p = ml_get_cursor();
2348
2349 if (curwin->w_cursor.coladd >= (colnr_T)chartabsize(p,
2350 curwin->w_virtcol - curwin->w_cursor.coladd))
2351 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002352 int l;
2353
2354 if (*p != NUL && p[(l = (*mb_ptr2len)(p))] == NUL)
2355 col += l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002356 }
2357 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002358 }
2359 }
2360 rettv->vval.v_number = col;
2361}
2362
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002363/*
2364 * "confirm(message, buttons[, default [, type]])" function
2365 */
2366 static void
2367f_confirm(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2368{
2369#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2370 char_u *message;
2371 char_u *buttons = NULL;
2372 char_u buf[NUMBUFLEN];
2373 char_u buf2[NUMBUFLEN];
2374 int def = 1;
2375 int type = VIM_GENERIC;
2376 char_u *typestr;
2377 int error = FALSE;
2378
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002379 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002380 if (message == NULL)
2381 error = TRUE;
2382 if (argvars[1].v_type != VAR_UNKNOWN)
2383 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002384 buttons = tv_get_string_buf_chk(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002385 if (buttons == NULL)
2386 error = TRUE;
2387 if (argvars[2].v_type != VAR_UNKNOWN)
2388 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002389 def = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002390 if (argvars[3].v_type != VAR_UNKNOWN)
2391 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002392 typestr = tv_get_string_buf_chk(&argvars[3], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002393 if (typestr == NULL)
2394 error = TRUE;
2395 else
2396 {
2397 switch (TOUPPER_ASC(*typestr))
2398 {
2399 case 'E': type = VIM_ERROR; break;
2400 case 'Q': type = VIM_QUESTION; break;
2401 case 'I': type = VIM_INFO; break;
2402 case 'W': type = VIM_WARNING; break;
2403 case 'G': type = VIM_GENERIC; break;
2404 }
2405 }
2406 }
2407 }
2408 }
2409
2410 if (buttons == NULL || *buttons == NUL)
2411 buttons = (char_u *)_("&Ok");
2412
2413 if (!error)
2414 rettv->vval.v_number = do_dialog(type, NULL, message, buttons,
2415 def, NULL, FALSE);
2416#endif
2417}
2418
2419/*
2420 * "copy()" function
2421 */
2422 static void
2423f_copy(typval_T *argvars, typval_T *rettv)
2424{
2425 item_copy(&argvars[0], rettv, FALSE, 0);
2426}
2427
2428#ifdef FEAT_FLOAT
2429/*
2430 * "cos()" function
2431 */
2432 static void
2433f_cos(typval_T *argvars, typval_T *rettv)
2434{
2435 float_T f = 0.0;
2436
2437 rettv->v_type = VAR_FLOAT;
2438 if (get_float_arg(argvars, &f) == OK)
2439 rettv->vval.v_float = cos(f);
2440 else
2441 rettv->vval.v_float = 0.0;
2442}
2443
2444/*
2445 * "cosh()" function
2446 */
2447 static void
2448f_cosh(typval_T *argvars, typval_T *rettv)
2449{
2450 float_T f = 0.0;
2451
2452 rettv->v_type = VAR_FLOAT;
2453 if (get_float_arg(argvars, &f) == OK)
2454 rettv->vval.v_float = cosh(f);
2455 else
2456 rettv->vval.v_float = 0.0;
2457}
2458#endif
2459
2460/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002461 * "cursor(lnum, col)" function, or
2462 * "cursor(list)"
2463 *
2464 * Moves the cursor to the specified line and column.
2465 * Returns 0 when the position could be set, -1 otherwise.
2466 */
2467 static void
2468f_cursor(typval_T *argvars, typval_T *rettv)
2469{
2470 long line, col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002471 long coladd = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002472 int set_curswant = TRUE;
2473
2474 rettv->vval.v_number = -1;
2475 if (argvars[1].v_type == VAR_UNKNOWN)
2476 {
2477 pos_T pos;
2478 colnr_T curswant = -1;
2479
2480 if (list2fpos(argvars, &pos, NULL, &curswant) == FAIL)
2481 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002482 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002483 return;
2484 }
2485 line = pos.lnum;
2486 col = pos.col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002487 coladd = pos.coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002488 if (curswant >= 0)
2489 {
2490 curwin->w_curswant = curswant - 1;
2491 set_curswant = FALSE;
2492 }
2493 }
2494 else
2495 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002496 line = tv_get_lnum(argvars);
2497 col = (long)tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002498 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002499 coladd = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002500 }
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002501 if (line < 0 || col < 0 || coladd < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002502 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002503 if (line > 0)
2504 curwin->w_cursor.lnum = line;
2505 if (col > 0)
2506 curwin->w_cursor.col = col - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002507 curwin->w_cursor.coladd = coladd;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002508
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002509 // Make sure the cursor is in a valid position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002510 check_cursor();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002511 // Correct cursor for multi-byte character.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002512 if (has_mbyte)
2513 mb_adjust_cursor();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002514
2515 curwin->w_set_curswant = set_curswant;
2516 rettv->vval.v_number = 0;
2517}
2518
Bram Moolenaar4f974752019-02-17 17:44:42 +01002519#ifdef MSWIN
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002520/*
2521 * "debugbreak()" function
2522 */
2523 static void
2524f_debugbreak(typval_T *argvars, typval_T *rettv)
2525{
2526 int pid;
2527
2528 rettv->vval.v_number = FAIL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002529 pid = (int)tv_get_number(&argvars[0]);
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002530 if (pid == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002531 emsg(_(e_invarg));
Bram Moolenaar4551c0a2018-06-20 22:38:21 +02002532 else
2533 {
2534 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, 0, pid);
2535
2536 if (hProcess != NULL)
2537 {
2538 DebugBreakProcess(hProcess);
2539 CloseHandle(hProcess);
2540 rettv->vval.v_number = OK;
2541 }
2542 }
2543}
2544#endif
2545
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002546/*
2547 * "deepcopy()" function
2548 */
2549 static void
2550f_deepcopy(typval_T *argvars, typval_T *rettv)
2551{
2552 int noref = 0;
2553 int copyID;
2554
2555 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar44b4a242020-09-05 17:18:28 +02002556 noref = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002557 if (noref < 0 || noref > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02002558 semsg(_(e_using_number_as_bool_nr), noref);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002559 else
2560 {
2561 copyID = get_copyID();
2562 item_copy(&argvars[0], rettv, TRUE, noref == 0 ? copyID : 0);
2563 }
2564}
2565
2566/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002567 * "did_filetype()" function
2568 */
2569 static void
2570f_did_filetype(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
2571{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002572 rettv->vval.v_number = did_filetype;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002573}
2574
2575/*
Bram Moolenaar4132eb52020-02-14 16:53:00 +01002576 * "echoraw({expr})" function
2577 */
2578 static void
2579f_echoraw(typval_T *argvars, typval_T *rettv UNUSED)
2580{
2581 char_u *str = tv_get_string_chk(&argvars[0]);
2582
2583 if (str != NULL && *str != NUL)
2584 {
2585 out_str(str);
2586 out_flush();
2587 }
2588}
2589
2590/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002591 * "empty({expr})" function
2592 */
2593 static void
2594f_empty(typval_T *argvars, typval_T *rettv)
2595{
2596 int n = FALSE;
2597
2598 switch (argvars[0].v_type)
2599 {
2600 case VAR_STRING:
2601 case VAR_FUNC:
2602 n = argvars[0].vval.v_string == NULL
2603 || *argvars[0].vval.v_string == NUL;
2604 break;
2605 case VAR_PARTIAL:
2606 n = FALSE;
2607 break;
2608 case VAR_NUMBER:
2609 n = argvars[0].vval.v_number == 0;
2610 break;
2611 case VAR_FLOAT:
2612#ifdef FEAT_FLOAT
2613 n = argvars[0].vval.v_float == 0.0;
2614 break;
2615#endif
2616 case VAR_LIST:
2617 n = argvars[0].vval.v_list == NULL
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002618 || argvars[0].vval.v_list->lv_len == 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002619 break;
2620 case VAR_DICT:
2621 n = argvars[0].vval.v_dict == NULL
2622 || argvars[0].vval.v_dict->dv_hashtab.ht_used == 0;
2623 break;
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01002624 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002625 case VAR_SPECIAL:
2626 n = argvars[0].vval.v_number != VVAL_TRUE;
2627 break;
2628
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002629 case VAR_BLOB:
2630 n = argvars[0].vval.v_blob == NULL
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01002631 || argvars[0].vval.v_blob->bv_ga.ga_len == 0;
2632 break;
2633
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002634 case VAR_JOB:
2635#ifdef FEAT_JOB_CHANNEL
2636 n = argvars[0].vval.v_job == NULL
2637 || argvars[0].vval.v_job->jv_status != JOB_STARTED;
2638 break;
2639#endif
2640 case VAR_CHANNEL:
2641#ifdef FEAT_JOB_CHANNEL
2642 n = argvars[0].vval.v_channel == NULL
2643 || !channel_is_open(argvars[0].vval.v_channel);
2644 break;
2645#endif
2646 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02002647 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002648 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01002649 internal_error_no_abort("f_empty(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002650 n = TRUE;
2651 break;
2652 }
2653
2654 rettv->vval.v_number = n;
2655}
2656
2657/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02002658 * "environ()" function
2659 */
2660 static void
2661f_environ(typval_T *argvars UNUSED, typval_T *rettv)
2662{
2663#if !defined(AMIGA)
2664 int i = 0;
2665 char_u *entry, *value;
2666# ifdef MSWIN
2667 extern wchar_t **_wenviron;
2668# else
2669 extern char **environ;
2670# endif
2671
2672 if (rettv_dict_alloc(rettv) != OK)
2673 return;
2674
2675# ifdef MSWIN
2676 if (*_wenviron == NULL)
2677 return;
2678# else
2679 if (*environ == NULL)
2680 return;
2681# endif
2682
2683 for (i = 0; ; ++i)
2684 {
2685# ifdef MSWIN
2686 short_u *p;
2687
2688 if ((p = (short_u *)_wenviron[i]) == NULL)
2689 return;
2690 entry = utf16_to_enc(p, NULL);
2691# else
2692 if ((entry = (char_u *)environ[i]) == NULL)
2693 return;
2694 entry = vim_strsave(entry);
2695# endif
2696 if (entry == NULL) // out of memory
2697 return;
2698 if ((value = vim_strchr(entry, '=')) == NULL)
2699 {
2700 vim_free(entry);
2701 continue;
2702 }
2703 *value++ = NUL;
2704 dict_add_string(rettv->vval.v_dict, (char *)entry, value);
2705 vim_free(entry);
2706 }
2707#endif
2708}
2709
2710/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002711 * "escape({string}, {chars})" function
2712 */
2713 static void
2714f_escape(typval_T *argvars, typval_T *rettv)
2715{
2716 char_u buf[NUMBUFLEN];
2717
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002718 rettv->vval.v_string = vim_strsave_escaped(tv_get_string(&argvars[0]),
2719 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002720 rettv->v_type = VAR_STRING;
2721}
2722
2723/*
2724 * "eval()" function
2725 */
2726 static void
2727f_eval(typval_T *argvars, typval_T *rettv)
2728{
2729 char_u *s, *p;
2730
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002731 s = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002732 if (s != NULL)
2733 s = skipwhite(s);
2734
2735 p = s;
Bram Moolenaar5409f5d2020-06-24 18:37:35 +02002736 if (s == NULL || eval1(&s, rettv, &EVALARG_EVALUATE) == FAIL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002737 {
2738 if (p != NULL && !aborting())
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002739 semsg(_(e_invexpr2), p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002740 need_clr_eos = FALSE;
2741 rettv->v_type = VAR_NUMBER;
2742 rettv->vval.v_number = 0;
2743 }
2744 else if (*s != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02002745 semsg(_(e_trailing_arg), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002746}
2747
2748/*
2749 * "eventhandler()" function
2750 */
2751 static void
2752f_eventhandler(typval_T *argvars UNUSED, typval_T *rettv)
2753{
Bram Moolenaardfc33a62020-04-29 22:30:13 +02002754 rettv->vval.v_number = vgetc_busy || input_busy;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002755}
2756
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002757static garray_T redir_execute_ga;
2758
2759/*
2760 * Append "value[value_len]" to the execute() output.
2761 */
2762 void
2763execute_redir_str(char_u *value, int value_len)
2764{
2765 int len;
2766
2767 if (value_len == -1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002768 len = (int)STRLEN(value); // Append the entire string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002769 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002770 len = value_len; // Append only "value_len" characters
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002771 if (ga_grow(&redir_execute_ga, len) == OK)
2772 {
2773 mch_memmove((char *)redir_execute_ga.ga_data
2774 + redir_execute_ga.ga_len, value, len);
2775 redir_execute_ga.ga_len += len;
2776 }
2777}
2778
2779/*
2780 * Get next line from a list.
2781 * Called by do_cmdline() to get the next line.
2782 * Returns allocated string, or NULL for end of function.
2783 */
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002784 static char_u *
2785get_list_line(
2786 int c UNUSED,
2787 void *cookie,
Bram Moolenaare96a2492019-06-25 04:12:16 +02002788 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +02002789 getline_opt_T options UNUSED)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002790{
2791 listitem_T **p = (listitem_T **)cookie;
2792 listitem_T *item = *p;
2793 char_u buf[NUMBUFLEN];
2794 char_u *s;
2795
2796 if (item == NULL)
2797 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002798 s = tv_get_string_buf_chk(&item->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002799 *p = item->li_next;
2800 return s == NULL ? NULL : vim_strsave(s);
2801}
2802
2803/*
2804 * "execute()" function
2805 */
Bram Moolenaar261f3462019-09-07 15:45:32 +02002806 void
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002807execute_common(typval_T *argvars, typval_T *rettv, int arg_off)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002808{
2809 char_u *cmd = NULL;
2810 list_T *list = NULL;
2811 int save_msg_silent = msg_silent;
2812 int save_emsg_silent = emsg_silent;
2813 int save_emsg_noredir = emsg_noredir;
2814 int save_redir_execute = redir_execute;
Bram Moolenaar20951482017-12-25 13:44:43 +01002815 int save_redir_off = redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002816 garray_T save_ga;
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002817 int save_msg_col = msg_col;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002818 int echo_output = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002819
2820 rettv->vval.v_string = NULL;
2821 rettv->v_type = VAR_STRING;
2822
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002823 if (argvars[arg_off].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002824 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002825 list = argvars[arg_off].vval.v_list;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002826 if (list == NULL || list->lv_len == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002827 // empty list, no commands, empty output
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002828 return;
2829 ++list->lv_refcount;
2830 }
Bram Moolenaare2a8f072020-01-08 19:32:18 +01002831 else if (argvars[arg_off].v_type == VAR_JOB
2832 || argvars[arg_off].v_type == VAR_CHANNEL)
2833 {
2834 emsg(_(e_inval_string));
2835 return;
2836 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002837 else
2838 {
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002839 cmd = tv_get_string_chk(&argvars[arg_off]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002840 if (cmd == NULL)
2841 return;
2842 }
2843
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002844 if (argvars[arg_off + 1].v_type != VAR_UNKNOWN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002845 {
2846 char_u buf[NUMBUFLEN];
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002847 char_u *s = tv_get_string_buf_chk(&argvars[arg_off + 1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002848
2849 if (s == NULL)
2850 return;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002851 if (*s == NUL)
2852 echo_output = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002853 if (STRNCMP(s, "silent", 6) == 0)
2854 ++msg_silent;
2855 if (STRCMP(s, "silent!") == 0)
2856 {
2857 emsg_silent = TRUE;
2858 emsg_noredir = TRUE;
2859 }
2860 }
2861 else
2862 ++msg_silent;
2863
2864 if (redir_execute)
2865 save_ga = redir_execute_ga;
2866 ga_init2(&redir_execute_ga, (int)sizeof(char), 500);
2867 redir_execute = TRUE;
Bram Moolenaar20951482017-12-25 13:44:43 +01002868 redir_off = FALSE;
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002869 if (!echo_output)
2870 msg_col = 0; // prevent leading spaces
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002871
2872 if (cmd != NULL)
2873 do_cmdline_cmd(cmd);
2874 else
2875 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002876 listitem_T *item;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002877
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002878 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01002879 item = list->lv_first;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002880 do_cmdline(NULL, get_list_line, (void *)&item,
2881 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT|DOCMD_KEYTYPED);
2882 --list->lv_refcount;
2883 }
2884
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002885 // Need to append a NUL to the result.
Bram Moolenaard297f352017-01-29 20:31:21 +01002886 if (ga_grow(&redir_execute_ga, 1) == OK)
2887 {
2888 ((char *)redir_execute_ga.ga_data)[redir_execute_ga.ga_len] = NUL;
2889 rettv->vval.v_string = redir_execute_ga.ga_data;
2890 }
2891 else
2892 {
2893 ga_clear(&redir_execute_ga);
2894 rettv->vval.v_string = NULL;
2895 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002896 msg_silent = save_msg_silent;
2897 emsg_silent = save_emsg_silent;
2898 emsg_noredir = save_emsg_noredir;
2899
2900 redir_execute = save_redir_execute;
2901 if (redir_execute)
2902 redir_execute_ga = save_ga;
Bram Moolenaar20951482017-12-25 13:44:43 +01002903 redir_off = save_redir_off;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002904
Bram Moolenaar10ccaa12018-12-07 16:38:23 +01002905 // "silent reg" or "silent echo x" leaves msg_col somewhere in the line.
Bram Moolenaar446e7a32018-12-08 13:57:42 +01002906 if (echo_output)
2907 // When not working silently: put it in column zero. A following
2908 // "echon" will overwrite the message, unavoidably.
2909 msg_col = 0;
2910 else
2911 // When working silently: Put it back where it was, since nothing
2912 // should have been written.
2913 msg_col = save_msg_col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002914}
2915
2916/*
Bram Moolenaar868b7b62019-05-29 21:44:40 +02002917 * "execute()" function
2918 */
2919 static void
2920f_execute(typval_T *argvars, typval_T *rettv)
2921{
2922 execute_common(argvars, rettv, 0);
2923}
2924
2925/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002926 * "exists()" function
2927 */
2928 static void
2929f_exists(typval_T *argvars, typval_T *rettv)
2930{
2931 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002932 int n = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002933
Bram Moolenaard155d7a2018-12-21 16:04:21 +01002934 p = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002935 if (*p == '$') // environment variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002936 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002937 // first try "normal" environment variables (fast)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002938 if (mch_getenv(p + 1) != NULL)
2939 n = TRUE;
2940 else
2941 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002942 // try expanding things like $VIM and ${HOME}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002943 p = expand_env_save(p);
2944 if (p != NULL && *p != '$')
2945 n = TRUE;
2946 vim_free(p);
2947 }
2948 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002949 else if (*p == '&' || *p == '+') // option
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002950 {
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +02002951 n = (eval_option(&p, NULL, TRUE) == OK);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002952 if (*skipwhite(p) != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002953 n = FALSE; // trailing garbage
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002954 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002955 else if (*p == '*') // internal or user defined function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002956 {
Bram Moolenaarb54c3ff2016-07-31 14:11:58 +02002957 n = function_exists(p + 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002958 }
Bram Moolenaar15c47602020-03-26 22:16:48 +01002959 else if (*p == '?') // internal function only
2960 {
2961 n = has_internal_func_name(p + 1);
2962 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002963 else if (*p == ':')
2964 {
2965 n = cmd_exists(p + 1);
2966 }
2967 else if (*p == '#')
2968 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002969 if (p[1] == '#')
2970 n = autocmd_supported(p + 2);
2971 else
2972 n = au_exists(p + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002973 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002974 else // internal variable
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002975 {
Bram Moolenaarc6f9f732018-02-11 19:06:26 +01002976 n = var_exists(p);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02002977 }
2978
2979 rettv->vval.v_number = n;
2980}
2981
2982#ifdef FEAT_FLOAT
2983/*
2984 * "exp()" function
2985 */
2986 static void
2987f_exp(typval_T *argvars, typval_T *rettv)
2988{
2989 float_T f = 0.0;
2990
2991 rettv->v_type = VAR_FLOAT;
2992 if (get_float_arg(argvars, &f) == OK)
2993 rettv->vval.v_float = exp(f);
2994 else
2995 rettv->vval.v_float = 0.0;
2996}
2997#endif
2998
2999/*
3000 * "expand()" function
3001 */
3002 static void
3003f_expand(typval_T *argvars, typval_T *rettv)
3004{
3005 char_u *s;
3006 int len;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003007 char *errormsg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003008 int options = WILD_SILENT|WILD_USE_NL|WILD_LIST_NOTFOUND;
3009 expand_T xpc;
3010 int error = FALSE;
3011 char_u *result;
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003012#ifdef BACKSLASH_IN_FILENAME
3013 char_u *p_csl_save = p_csl;
3014
3015 // avoid using 'completeslash' here
3016 p_csl = empty_option;
3017#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003018
3019 rettv->v_type = VAR_STRING;
3020 if (argvars[1].v_type != VAR_UNKNOWN
3021 && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003022 && tv_get_bool_chk(&argvars[2], &error)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003023 && !error)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02003024 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003025
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003026 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003027 if (*s == '%' || *s == '#' || *s == '<')
3028 {
3029 ++emsg_off;
3030 result = eval_vars(s, s, &len, NULL, &errormsg, NULL);
3031 --emsg_off;
3032 if (rettv->v_type == VAR_LIST)
3033 {
3034 if (rettv_list_alloc(rettv) != FAIL && result != NULL)
3035 list_append_string(rettv->vval.v_list, result, -1);
Bram Moolenaar86173482019-10-01 17:02:16 +02003036 vim_free(result);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003037 }
3038 else
3039 rettv->vval.v_string = result;
3040 }
3041 else
3042 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003043 // When the optional second argument is non-zero, don't remove matches
3044 // for 'wildignore' and don't put matches for 'suffixes' at the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003045 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaar551d25e2020-09-02 21:37:56 +02003046 && tv_get_bool_chk(&argvars[1], &error))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003047 options |= WILD_KEEP_ALL;
3048 if (!error)
3049 {
3050 ExpandInit(&xpc);
3051 xpc.xp_context = EXPAND_FILES;
3052 if (p_wic)
3053 options += WILD_ICASE;
3054 if (rettv->v_type == VAR_STRING)
3055 rettv->vval.v_string = ExpandOne(&xpc, s, NULL,
3056 options, WILD_ALL);
3057 else if (rettv_list_alloc(rettv) != FAIL)
3058 {
3059 int i;
3060
3061 ExpandOne(&xpc, s, NULL, options, WILD_ALL_KEEP);
3062 for (i = 0; i < xpc.xp_numfiles; i++)
3063 list_append_string(rettv->vval.v_list, xpc.xp_files[i], -1);
3064 ExpandCleanup(&xpc);
3065 }
3066 }
3067 else
3068 rettv->vval.v_string = NULL;
3069 }
Bram Moolenaar8f187fc2020-09-26 18:47:11 +02003070#ifdef BACKSLASH_IN_FILENAME
3071 p_csl = p_csl_save;
3072#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003073}
3074
3075/*
Bram Moolenaar80dad482019-06-09 17:22:31 +02003076 * "expandcmd()" function
3077 * Expand all the special characters in a command string.
3078 */
3079 static void
3080f_expandcmd(typval_T *argvars, typval_T *rettv)
3081{
3082 exarg_T eap;
3083 char_u *cmdstr;
3084 char *errormsg = NULL;
3085
3086 rettv->v_type = VAR_STRING;
3087 cmdstr = vim_strsave(tv_get_string(&argvars[0]));
3088
3089 memset(&eap, 0, sizeof(eap));
3090 eap.cmd = cmdstr;
3091 eap.arg = cmdstr;
Bram Moolenaar8071cb22019-07-12 17:58:01 +02003092 eap.argt |= EX_NOSPC;
Bram Moolenaar80dad482019-06-09 17:22:31 +02003093 eap.usefilter = FALSE;
3094 eap.nextcmd = NULL;
3095 eap.cmdidx = CMD_USER;
3096
3097 expand_filename(&eap, &cmdstr, &errormsg);
3098 if (errormsg != NULL && *errormsg != NUL)
3099 emsg(errormsg);
3100
3101 rettv->vval.v_string = cmdstr;
3102}
3103
3104/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003105 * "feedkeys()" function
3106 */
3107 static void
3108f_feedkeys(typval_T *argvars, typval_T *rettv UNUSED)
3109{
3110 int remap = TRUE;
3111 int insert = FALSE;
3112 char_u *keys, *flags;
3113 char_u nbuf[NUMBUFLEN];
3114 int typed = FALSE;
3115 int execute = FALSE;
3116 int dangerous = FALSE;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003117 int lowlevel = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003118 char_u *keys_esc;
3119
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003120 // This is not allowed in the sandbox. If the commands would still be
3121 // executed in the sandbox it would be OK, but it probably happens later,
3122 // when "sandbox" is no longer set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003123 if (check_secure())
3124 return;
3125
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003126 keys = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003127
3128 if (argvars[1].v_type != VAR_UNKNOWN)
3129 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003130 flags = tv_get_string_buf(&argvars[1], nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003131 for ( ; *flags != NUL; ++flags)
3132 {
3133 switch (*flags)
3134 {
3135 case 'n': remap = FALSE; break;
3136 case 'm': remap = TRUE; break;
3137 case 't': typed = TRUE; break;
3138 case 'i': insert = TRUE; break;
3139 case 'x': execute = TRUE; break;
3140 case '!': dangerous = TRUE; break;
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003141 case 'L': lowlevel = TRUE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003142 }
3143 }
3144 }
3145
3146 if (*keys != NUL || execute)
3147 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003148 // Need to escape K_SPECIAL and CSI before putting the string in the
3149 // typeahead buffer.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003150 keys_esc = vim_strsave_escape_csi(keys);
3151 if (keys_esc != NULL)
3152 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003153 if (lowlevel)
3154 {
3155#ifdef USE_INPUT_BUF
Bram Moolenaar9645e2d2020-03-20 20:48:49 +01003156 int idx;
3157 int len = (int)STRLEN(keys);
3158
3159 for (idx = 0; idx < len; ++idx)
3160 {
3161 // if a CTRL-C was typed, set got_int, similar to what
3162 // happens in fill_input_buf()
3163 if (keys[idx] == 3 && ctrl_c_interrupts && typed)
3164 got_int = TRUE;
3165 add_to_input_buf(keys + idx, 1);
3166 }
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003167#else
3168 emsg(_("E980: lowlevel input not supported"));
3169#endif
3170 }
3171 else
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003172 {
Bram Moolenaar5e66b422019-01-24 21:58:10 +01003173 ins_typebuf(keys_esc, (remap ? REMAP_YES : REMAP_NONE),
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003174 insert ? 0 : typebuf.tb_len, !typed, FALSE);
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003175 if (vgetc_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003176#ifdef FEAT_TIMERS
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003177 || timer_busy
Bram Moolenaar5d7be4f2017-06-25 13:40:17 +02003178#endif
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003179 || input_busy)
Bram Moolenaar8d4ce562019-01-30 22:01:40 +01003180 typebuf_was_filled = TRUE;
3181 }
3182 vim_free(keys_esc);
3183
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003184 if (execute)
3185 {
3186 int save_msg_scroll = msg_scroll;
3187
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003188 // Avoid a 1 second delay when the keys start Insert mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003189 msg_scroll = FALSE;
3190
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003191 if (!dangerous)
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003192 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003193 ++ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003194 ++in_feedkeys;
3195 }
Bram Moolenaar905dd902019-04-07 14:21:47 +02003196 exec_normal(TRUE, lowlevel, TRUE);
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003197 if (!dangerous)
Bram Moolenaar189832b2020-09-23 12:29:11 +02003198 {
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003199 --ex_normal_busy;
Bram Moolenaar4934ad02020-09-28 22:29:58 +02003200 --in_feedkeys;
Bram Moolenaar189832b2020-09-23 12:29:11 +02003201 }
Bram Moolenaarb2ac14c2018-05-01 18:47:59 +02003202
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003203 msg_scroll |= save_msg_scroll;
3204 }
3205 }
3206 }
3207}
3208
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003209#ifdef FEAT_FLOAT
3210/*
3211 * "float2nr({float})" function
3212 */
3213 static void
3214f_float2nr(typval_T *argvars, typval_T *rettv)
3215{
3216 float_T f = 0.0;
3217
3218 if (get_float_arg(argvars, &f) == OK)
3219 {
Bram Moolenaar37184272020-05-23 19:30:05 +02003220 if (f <= (float_T)-VARNUM_MAX + DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003221 rettv->vval.v_number = -VARNUM_MAX;
Bram Moolenaar37184272020-05-23 19:30:05 +02003222 else if (f >= (float_T)VARNUM_MAX - DBL_EPSILON)
Bram Moolenaar7a40ea22017-01-22 18:34:57 +01003223 rettv->vval.v_number = VARNUM_MAX;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003224 else
3225 rettv->vval.v_number = (varnumber_T)f;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003226 }
3227}
3228
3229/*
3230 * "floor({float})" function
3231 */
3232 static void
3233f_floor(typval_T *argvars, typval_T *rettv)
3234{
3235 float_T f = 0.0;
3236
3237 rettv->v_type = VAR_FLOAT;
3238 if (get_float_arg(argvars, &f) == OK)
3239 rettv->vval.v_float = floor(f);
3240 else
3241 rettv->vval.v_float = 0.0;
3242}
3243
3244/*
3245 * "fmod()" function
3246 */
3247 static void
3248f_fmod(typval_T *argvars, typval_T *rettv)
3249{
3250 float_T fx = 0.0, fy = 0.0;
3251
3252 rettv->v_type = VAR_FLOAT;
3253 if (get_float_arg(argvars, &fx) == OK
3254 && get_float_arg(&argvars[1], &fy) == OK)
3255 rettv->vval.v_float = fmod(fx, fy);
3256 else
3257 rettv->vval.v_float = 0.0;
3258}
3259#endif
3260
3261/*
3262 * "fnameescape({string})" function
3263 */
3264 static void
3265f_fnameescape(typval_T *argvars, typval_T *rettv)
3266{
3267 rettv->vval.v_string = vim_strsave_fnameescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003268 tv_get_string(&argvars[0]), FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003269 rettv->v_type = VAR_STRING;
3270}
3271
3272/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003273 * "foreground()" function
3274 */
3275 static void
3276f_foreground(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3277{
3278#ifdef FEAT_GUI
3279 if (gui.in_use)
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003280 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003281 gui_mch_set_foreground();
Bram Moolenaarafde13b2019-04-28 19:46:49 +02003282 return;
3283 }
3284#endif
3285#if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003286 win32_set_foreground();
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003287#endif
3288}
3289
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003290 static void
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003291common_function(typval_T *argvars, typval_T *rettv, int is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003292{
3293 char_u *s;
3294 char_u *name;
3295 int use_string = FALSE;
3296 partial_T *arg_pt = NULL;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003297 char_u *trans_name = NULL;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003298 int is_global = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003299
3300 if (argvars[0].v_type == VAR_FUNC)
3301 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003302 // function(MyFunc, [arg], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003303 s = argvars[0].vval.v_string;
3304 }
3305 else if (argvars[0].v_type == VAR_PARTIAL
3306 && argvars[0].vval.v_partial != NULL)
3307 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003308 // function(dict.MyFunc, [arg])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003309 arg_pt = argvars[0].vval.v_partial;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003310 s = partial_name(arg_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003311 }
3312 else
3313 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003314 // function('MyFunc', [arg], dict)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003315 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003316 use_string = TRUE;
3317 }
3318
Bram Moolenaar843b8842016-08-21 14:36:15 +02003319 if ((use_string && vim_strchr(s, AUTOLOAD_CHAR) == NULL) || is_funcref)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003320 {
3321 name = s;
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003322 trans_name = trans_function_name(&name, &is_global, FALSE,
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003323 TFN_INT | TFN_QUIET | TFN_NO_AUTOLOAD | TFN_NO_DEREF, NULL, NULL);
3324 if (*name != NUL)
3325 s = NULL;
3326 }
3327
Bram Moolenaar843b8842016-08-21 14:36:15 +02003328 if (s == NULL || *s == NUL || (use_string && VIM_ISDIGIT(*s))
3329 || (is_funcref && trans_name == NULL))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003330 semsg(_(e_invarg2), use_string ? tv_get_string(&argvars[0]) : s);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003331 // Don't check an autoload name for existence here.
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003332 else if (trans_name != NULL && (is_funcref
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003333 ? find_func(trans_name, is_global, NULL) == NULL
3334 : !translated_function_exists(trans_name, is_global)))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003335 semsg(_("E700: Unknown function: %s"), s);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003336 else
3337 {
3338 int dict_idx = 0;
3339 int arg_idx = 0;
3340 list_T *list = NULL;
3341
3342 if (STRNCMP(s, "s:", 2) == 0 || STRNCMP(s, "<SID>", 5) == 0)
3343 {
3344 char sid_buf[25];
3345 int off = *s == 's' ? 2 : 5;
3346
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003347 // Expand s: and <SID> into <SNR>nr_, so that the function can
3348 // also be called from another script. Using trans_function_name()
3349 // would also work, but some plugins depend on the name being
3350 // printable text.
Bram Moolenaarf29c1c62018-09-10 21:05:02 +02003351 sprintf(sid_buf, "<SNR>%ld_", (long)current_sctx.sc_sid);
Bram Moolenaar51e14382019-05-25 20:21:28 +02003352 name = alloc(STRLEN(sid_buf) + STRLEN(s + off) + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003353 if (name != NULL)
3354 {
3355 STRCPY(name, sid_buf);
3356 STRCAT(name, s + off);
3357 }
3358 }
3359 else
3360 name = vim_strsave(s);
3361
3362 if (argvars[1].v_type != VAR_UNKNOWN)
3363 {
3364 if (argvars[2].v_type != VAR_UNKNOWN)
3365 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003366 // function(name, [args], dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003367 arg_idx = 1;
3368 dict_idx = 2;
3369 }
3370 else if (argvars[1].v_type == VAR_DICT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003371 // function(name, dict)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003372 dict_idx = 1;
3373 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003374 // function(name, [args])
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003375 arg_idx = 1;
3376 if (dict_idx > 0)
3377 {
3378 if (argvars[dict_idx].v_type != VAR_DICT)
3379 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003380 emsg(_("E922: expected a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003381 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003382 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003383 }
3384 if (argvars[dict_idx].vval.v_dict == NULL)
3385 dict_idx = 0;
3386 }
3387 if (arg_idx > 0)
3388 {
3389 if (argvars[arg_idx].v_type != VAR_LIST)
3390 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003391 emsg(_("E923: Second argument of function() must be a list or a dict"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003392 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003393 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003394 }
3395 list = argvars[arg_idx].vval.v_list;
3396 if (list == NULL || list->lv_len == 0)
3397 arg_idx = 0;
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003398 else if (list->lv_len > MAX_FUNC_ARGS)
3399 {
Bram Moolenaar2118a302019-11-22 19:29:45 +01003400 emsg_funcname((char *)e_toomanyarg, s);
Bram Moolenaar4c054e92019-11-10 00:13:50 +01003401 vim_free(name);
3402 goto theend;
3403 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003404 }
3405 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003406 if (dict_idx > 0 || arg_idx > 0 || arg_pt != NULL || is_funcref)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003407 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003408 partial_T *pt = ALLOC_CLEAR_ONE(partial_T);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003409
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003410 // result is a VAR_PARTIAL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003411 if (pt == NULL)
3412 vim_free(name);
3413 else
3414 {
3415 if (arg_idx > 0 || (arg_pt != NULL && arg_pt->pt_argc > 0))
3416 {
3417 listitem_T *li;
3418 int i = 0;
3419 int arg_len = 0;
3420 int lv_len = 0;
3421
3422 if (arg_pt != NULL)
3423 arg_len = arg_pt->pt_argc;
3424 if (list != NULL)
3425 lv_len = list->lv_len;
3426 pt->pt_argc = arg_len + lv_len;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02003427 pt->pt_argv = ALLOC_MULT(typval_T, pt->pt_argc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003428 if (pt->pt_argv == NULL)
3429 {
3430 vim_free(pt);
3431 vim_free(name);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003432 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003433 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003434 for (i = 0; i < arg_len; i++)
3435 copy_tv(&arg_pt->pt_argv[i], &pt->pt_argv[i]);
3436 if (lv_len > 0)
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003437 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003438 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003439 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003440 copy_tv(&li->li_tv, &pt->pt_argv[i++]);
Bram Moolenaar50985eb2020-01-27 22:09:39 +01003441 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003442 }
3443
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003444 // For "function(dict.func, [], dict)" and "func" is a partial
3445 // use "dict". That is backwards compatible.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003446 if (dict_idx > 0)
3447 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003448 // The dict is bound explicitly, pt_auto is FALSE.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003449 pt->pt_dict = argvars[dict_idx].vval.v_dict;
3450 ++pt->pt_dict->dv_refcount;
3451 }
3452 else if (arg_pt != NULL)
3453 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003454 // If the dict was bound automatically the result is also
3455 // bound automatically.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003456 pt->pt_dict = arg_pt->pt_dict;
3457 pt->pt_auto = arg_pt->pt_auto;
3458 if (pt->pt_dict != NULL)
3459 ++pt->pt_dict->dv_refcount;
3460 }
3461
3462 pt->pt_refcount = 1;
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003463 if (arg_pt != NULL && arg_pt->pt_func != NULL)
3464 {
3465 pt->pt_func = arg_pt->pt_func;
3466 func_ptr_ref(pt->pt_func);
3467 vim_free(name);
3468 }
3469 else if (is_funcref)
3470 {
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02003471 pt->pt_func = find_func(trans_name, is_global, NULL);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003472 func_ptr_ref(pt->pt_func);
3473 vim_free(name);
3474 }
3475 else
3476 {
3477 pt->pt_name = name;
3478 func_ref(name);
3479 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003480 }
3481 rettv->v_type = VAR_PARTIAL;
3482 rettv->vval.v_partial = pt;
3483 }
3484 else
3485 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003486 // result is a VAR_FUNC
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003487 rettv->v_type = VAR_FUNC;
3488 rettv->vval.v_string = name;
3489 func_ref(name);
3490 }
3491 }
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003492theend:
3493 vim_free(trans_name);
3494}
3495
3496/*
3497 * "funcref()" function
3498 */
3499 static void
3500f_funcref(typval_T *argvars, typval_T *rettv)
3501{
3502 common_function(argvars, rettv, TRUE);
3503}
3504
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003505 static type_T *
Bram Moolenaardfc33a62020-04-29 22:30:13 +02003506ret_f_function(int argcount, type_T **argtypes)
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003507{
3508 if (argcount == 1 && argtypes[0]->tt_type == VAR_STRING)
3509 return &t_func_any;
Bram Moolenaar5e654232020-09-16 15:22:00 +02003510 return &t_func_unknown;
Bram Moolenaarfbdd08e2020-03-01 14:04:46 +01003511}
3512
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003513/*
3514 * "function()" function
3515 */
3516 static void
3517f_function(typval_T *argvars, typval_T *rettv)
3518{
3519 common_function(argvars, rettv, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003520}
3521
3522/*
3523 * "garbagecollect()" function
3524 */
3525 static void
3526f_garbagecollect(typval_T *argvars, typval_T *rettv UNUSED)
3527{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003528 // This is postponed until we are back at the toplevel, because we may be
3529 // using Lists and Dicts internally. E.g.: ":echo [garbagecollect()]".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003530 want_garbage_collect = TRUE;
3531
Bram Moolenaar2df47312020-09-05 17:30:44 +02003532 if (argvars[0].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[0]) == 1)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003533 garbage_collect_at_exit = TRUE;
3534}
3535
3536/*
3537 * "get()" function
3538 */
3539 static void
3540f_get(typval_T *argvars, typval_T *rettv)
3541{
3542 listitem_T *li;
3543 list_T *l;
3544 dictitem_T *di;
3545 dict_T *d;
3546 typval_T *tv = NULL;
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003547 int what_is_dict = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003548
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003549 if (argvars[0].v_type == VAR_BLOB)
3550 {
3551 int error = FALSE;
3552 int idx = tv_get_number_chk(&argvars[1], &error);
3553
3554 if (!error)
3555 {
3556 rettv->v_type = VAR_NUMBER;
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003557 if (idx < 0)
3558 idx = blob_len(argvars[0].vval.v_blob) + idx;
3559 if (idx < 0 || idx >= blob_len(argvars[0].vval.v_blob))
3560 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003561 else
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003562 {
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003563 rettv->vval.v_number = blob_get(argvars[0].vval.v_blob, idx);
Bram Moolenaar2ea773b2019-01-15 22:16:42 +01003564 tv = rettv;
3565 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01003566 }
3567 }
3568 else if (argvars[0].v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003569 {
3570 if ((l = argvars[0].vval.v_list) != NULL)
3571 {
3572 int error = FALSE;
3573
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003574 li = list_find(l, (long)tv_get_number_chk(&argvars[1], &error));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003575 if (!error && li != NULL)
3576 tv = &li->li_tv;
3577 }
3578 }
3579 else if (argvars[0].v_type == VAR_DICT)
3580 {
3581 if ((d = argvars[0].vval.v_dict) != NULL)
3582 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003583 di = dict_find(d, tv_get_string(&argvars[1]), -1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003584 if (di != NULL)
3585 tv = &di->di_tv;
3586 }
3587 }
3588 else if (argvars[0].v_type == VAR_PARTIAL || argvars[0].v_type == VAR_FUNC)
3589 {
3590 partial_T *pt;
3591 partial_T fref_pt;
3592
3593 if (argvars[0].v_type == VAR_PARTIAL)
3594 pt = argvars[0].vval.v_partial;
3595 else
3596 {
Bram Moolenaara80faa82020-04-12 19:37:17 +02003597 CLEAR_FIELD(fref_pt);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003598 fref_pt.pt_name = argvars[0].vval.v_string;
3599 pt = &fref_pt;
3600 }
3601
3602 if (pt != NULL)
3603 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003604 char_u *what = tv_get_string(&argvars[1]);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003605 char_u *n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003606
3607 if (STRCMP(what, "func") == 0 || STRCMP(what, "name") == 0)
3608 {
3609 rettv->v_type = (*what == 'f' ? VAR_FUNC : VAR_STRING);
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003610 n = partial_name(pt);
3611 if (n == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003612 rettv->vval.v_string = NULL;
3613 else
Bram Moolenaar437bafe2016-08-01 15:40:54 +02003614 {
3615 rettv->vval.v_string = vim_strsave(n);
3616 if (rettv->v_type == VAR_FUNC)
3617 func_ref(rettv->vval.v_string);
3618 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003619 }
3620 else if (STRCMP(what, "dict") == 0)
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003621 {
3622 what_is_dict = TRUE;
3623 if (pt->pt_dict != NULL)
3624 rettv_dict_set(rettv, pt->pt_dict);
3625 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003626 else if (STRCMP(what, "args") == 0)
3627 {
3628 rettv->v_type = VAR_LIST;
3629 if (rettv_list_alloc(rettv) == OK)
3630 {
3631 int i;
3632
3633 for (i = 0; i < pt->pt_argc; ++i)
3634 list_append_tv(rettv->vval.v_list, &pt->pt_argv[i]);
3635 }
3636 }
3637 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003638 semsg(_(e_invarg2), what);
Bram Moolenaarf91aac52019-07-28 13:21:01 +02003639
3640 // When {what} == "dict" and pt->pt_dict == NULL, evaluate the
3641 // third argument
3642 if (!what_is_dict)
3643 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003644 }
3645 }
3646 else
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01003647 semsg(_(e_listdictblobarg), "get()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003648
3649 if (tv == NULL)
3650 {
3651 if (argvars[2].v_type != VAR_UNKNOWN)
3652 copy_tv(&argvars[2], rettv);
3653 }
3654 else
3655 copy_tv(tv, rettv);
3656}
3657
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02003658/*
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003659 * "getchangelist()" function
3660 */
3661 static void
3662f_getchangelist(typval_T *argvars, typval_T *rettv)
3663{
3664#ifdef FEAT_JUMPLIST
3665 buf_T *buf;
3666 int i;
3667 list_T *l;
3668 dict_T *d;
3669#endif
3670
3671 if (rettv_list_alloc(rettv) != OK)
3672 return;
3673
3674#ifdef FEAT_JUMPLIST
Bram Moolenaar4c313b12019-08-24 22:58:31 +02003675 if (argvars[0].v_type == VAR_UNKNOWN)
3676 buf = curbuf;
3677 else
Bram Moolenaara5d38412020-09-02 21:02:35 +02003678 buf = tv_get_buf_from_arg(&argvars[0]);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003679 if (buf == NULL)
3680 return;
3681
3682 l = list_alloc();
3683 if (l == NULL)
3684 return;
3685
3686 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3687 return;
3688 /*
3689 * The current window change list index tracks only the position in the
3690 * current buffer change list. For other buffers, use the change list
3691 * length as the current index.
3692 */
3693 list_append_number(rettv->vval.v_list,
3694 (varnumber_T)((buf == curwin->w_buffer)
3695 ? curwin->w_changelistidx : buf->b_changelistlen));
3696
3697 for (i = 0; i < buf->b_changelistlen; ++i)
3698 {
3699 if (buf->b_changelist[i].lnum == 0)
3700 continue;
3701 if ((d = dict_alloc()) == NULL)
3702 return;
3703 if (list_append_dict(l, d) == FAIL)
3704 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003705 dict_add_number(d, "lnum", (long)buf->b_changelist[i].lnum);
3706 dict_add_number(d, "col", (long)buf->b_changelist[i].col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003707 dict_add_number(d, "coladd", (long)buf->b_changelist[i].coladd);
Bram Moolenaar07ad8162018-02-13 13:59:59 +01003708 }
3709#endif
3710}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003711
3712/*
3713 * "getcharsearch()" function
3714 */
3715 static void
3716f_getcharsearch(typval_T *argvars UNUSED, typval_T *rettv)
3717{
3718 if (rettv_dict_alloc(rettv) != FAIL)
3719 {
3720 dict_T *dict = rettv->vval.v_dict;
3721
Bram Moolenaare0be1672018-07-08 16:50:37 +02003722 dict_add_string(dict, "char", last_csearch());
3723 dict_add_number(dict, "forward", last_csearch_forward());
3724 dict_add_number(dict, "until", last_csearch_until());
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003725 }
3726}
3727
3728/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02003729 * "getenv()" function
3730 */
3731 static void
3732f_getenv(typval_T *argvars, typval_T *rettv)
3733{
3734 int mustfree = FALSE;
3735 char_u *p = vim_getenv(tv_get_string(&argvars[0]), &mustfree);
3736
3737 if (p == NULL)
3738 {
3739 rettv->v_type = VAR_SPECIAL;
3740 rettv->vval.v_number = VVAL_NULL;
3741 return;
3742 }
3743 if (!mustfree)
3744 p = vim_strsave(p);
3745 rettv->vval.v_string = p;
3746 rettv->v_type = VAR_STRING;
3747}
3748
3749/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003750 * "getfontname()" function
3751 */
3752 static void
3753f_getfontname(typval_T *argvars UNUSED, typval_T *rettv)
3754{
3755 rettv->v_type = VAR_STRING;
3756 rettv->vval.v_string = NULL;
3757#ifdef FEAT_GUI
3758 if (gui.in_use)
3759 {
3760 GuiFont font;
3761 char_u *name = NULL;
3762
3763 if (argvars[0].v_type == VAR_UNKNOWN)
3764 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003765 // Get the "Normal" font. Either the name saved by
3766 // hl_set_font_name() or from the font ID.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003767 font = gui.norm_font;
3768 name = hl_get_font_name();
3769 }
3770 else
3771 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003772 name = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003773 if (STRCMP(name, "*") == 0) // don't use font dialog
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003774 return;
3775 font = gui_mch_get_font(name, FALSE);
3776 if (font == NOFONT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003777 return; // Invalid font name, return empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003778 }
3779 rettv->vval.v_string = gui_mch_get_fontname(font, name);
3780 if (argvars[0].v_type != VAR_UNKNOWN)
3781 gui_mch_free_font(font);
3782 }
3783#endif
3784}
3785
3786/*
Bram Moolenaar4f505882018-02-10 21:06:32 +01003787 * "getjumplist()" function
3788 */
3789 static void
3790f_getjumplist(typval_T *argvars, typval_T *rettv)
3791{
3792#ifdef FEAT_JUMPLIST
3793 win_T *wp;
3794 int i;
3795 list_T *l;
3796 dict_T *d;
3797#endif
3798
3799 if (rettv_list_alloc(rettv) != OK)
3800 return;
3801
3802#ifdef FEAT_JUMPLIST
Bram Moolenaar00aa0692019-04-27 20:37:57 +02003803 wp = find_tabwin(&argvars[0], &argvars[1], NULL);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003804 if (wp == NULL)
3805 return;
3806
Bram Moolenaar57ee2b62019-02-12 22:15:06 +01003807 cleanup_jumplist(wp, TRUE);
3808
Bram Moolenaar4f505882018-02-10 21:06:32 +01003809 l = list_alloc();
3810 if (l == NULL)
3811 return;
3812
3813 if (list_append_list(rettv->vval.v_list, l) == FAIL)
3814 return;
3815 list_append_number(rettv->vval.v_list, (varnumber_T)wp->w_jumplistidx);
3816
3817 for (i = 0; i < wp->w_jumplistlen; ++i)
3818 {
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003819 if (wp->w_jumplist[i].fmark.mark.lnum == 0)
3820 continue;
Bram Moolenaar4f505882018-02-10 21:06:32 +01003821 if ((d = dict_alloc()) == NULL)
3822 return;
3823 if (list_append_dict(l, d) == FAIL)
3824 return;
Bram Moolenaare0be1672018-07-08 16:50:37 +02003825 dict_add_number(d, "lnum", (long)wp->w_jumplist[i].fmark.mark.lnum);
3826 dict_add_number(d, "col", (long)wp->w_jumplist[i].fmark.mark.col);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003827 dict_add_number(d, "coladd", (long)wp->w_jumplist[i].fmark.mark.coladd);
Bram Moolenaare0be1672018-07-08 16:50:37 +02003828 dict_add_number(d, "bufnr", (long)wp->w_jumplist[i].fmark.fnum);
Bram Moolenaara7e18d22018-02-11 14:29:49 +01003829 if (wp->w_jumplist[i].fname != NULL)
Bram Moolenaare0be1672018-07-08 16:50:37 +02003830 dict_add_string(d, "filename", wp->w_jumplist[i].fname);
Bram Moolenaar4f505882018-02-10 21:06:32 +01003831 }
3832#endif
3833}
3834
3835/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003836 * "getpid()" function
3837 */
3838 static void
3839f_getpid(typval_T *argvars UNUSED, typval_T *rettv)
3840{
3841 rettv->vval.v_number = mch_get_pid();
3842}
3843
3844 static void
3845getpos_both(
3846 typval_T *argvars,
3847 typval_T *rettv,
3848 int getcurpos)
3849{
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02003850 pos_T *fp = NULL;
3851 win_T *wp = curwin;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003852 list_T *l;
3853 int fnum = -1;
3854
3855 if (rettv_list_alloc(rettv) == OK)
3856 {
3857 l = rettv->vval.v_list;
3858 if (getcurpos)
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02003859 {
3860 if (argvars[0].v_type != VAR_UNKNOWN)
3861 {
3862 wp = find_win_by_nr_or_id(&argvars[0]);
3863 if (wp != NULL)
3864 fp = &wp->w_cursor;
3865 }
3866 else
3867 fp = &curwin->w_cursor;
3868 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003869 else
3870 fp = var2fpos(&argvars[0], TRUE, &fnum);
3871 if (fnum != -1)
3872 list_append_number(l, (varnumber_T)fnum);
3873 else
3874 list_append_number(l, (varnumber_T)0);
3875 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->lnum
3876 : (varnumber_T)0);
3877 list_append_number(l, (fp != NULL)
3878 ? (varnumber_T)(fp->col == MAXCOL ? MAXCOL : fp->col + 1)
3879 : (varnumber_T)0);
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01003880 list_append_number(l, (fp != NULL) ? (varnumber_T)fp->coladd :
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003881 (varnumber_T)0);
3882 if (getcurpos)
3883 {
Bram Moolenaar19a66852019-03-07 11:25:32 +01003884 int save_set_curswant = curwin->w_set_curswant;
3885 colnr_T save_curswant = curwin->w_curswant;
3886 colnr_T save_virtcol = curwin->w_virtcol;
3887
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02003888 if (wp == curwin)
3889 update_curswant();
3890 list_append_number(l, wp == NULL ? 0 : wp->w_curswant == MAXCOL
3891 ? (varnumber_T)MAXCOL : (varnumber_T)wp->w_curswant + 1);
Bram Moolenaar19a66852019-03-07 11:25:32 +01003892
3893 // Do not change "curswant", as it is unexpected that a get
3894 // function has a side effect.
Bram Moolenaar99ca9c42020-09-22 21:55:41 +02003895 if (wp == curwin && save_set_curswant)
Bram Moolenaar19a66852019-03-07 11:25:32 +01003896 {
3897 curwin->w_set_curswant = save_set_curswant;
3898 curwin->w_curswant = save_curswant;
3899 curwin->w_virtcol = save_virtcol;
3900 curwin->w_valid &= ~VALID_VIRTCOL;
3901 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003902 }
3903 }
3904 else
3905 rettv->vval.v_number = FALSE;
3906}
3907
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003908/*
3909 * "getcurpos()" function
3910 */
3911 static void
3912f_getcurpos(typval_T *argvars, typval_T *rettv)
3913{
3914 getpos_both(argvars, rettv, TRUE);
3915}
3916
3917/*
3918 * "getpos(string)" function
3919 */
3920 static void
3921f_getpos(typval_T *argvars, typval_T *rettv)
3922{
3923 getpos_both(argvars, rettv, FALSE);
3924}
3925
3926/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003927 * "getreg()" function
3928 */
3929 static void
3930f_getreg(typval_T *argvars, typval_T *rettv)
3931{
3932 char_u *strregname;
3933 int regname;
3934 int arg2 = FALSE;
3935 int return_list = FALSE;
3936 int error = FALSE;
3937
3938 if (argvars[0].v_type != VAR_UNKNOWN)
3939 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003940 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003941 error = strregname == NULL;
3942 if (argvars[1].v_type != VAR_UNKNOWN)
3943 {
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02003944 arg2 = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003945 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02003946 return_list = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003947 }
3948 }
3949 else
3950 strregname = get_vim_var_str(VV_REG);
3951
3952 if (error)
3953 return;
3954
3955 regname = (strregname == NULL ? '"' : *strregname);
3956 if (regname == 0)
3957 regname = '"';
3958
3959 if (return_list)
3960 {
3961 rettv->v_type = VAR_LIST;
3962 rettv->vval.v_list = (list_T *)get_reg_contents(regname,
3963 (arg2 ? GREG_EXPR_SRC : 0) | GREG_LIST);
3964 if (rettv->vval.v_list == NULL)
3965 (void)rettv_list_alloc(rettv);
3966 else
3967 ++rettv->vval.v_list->lv_refcount;
3968 }
3969 else
3970 {
3971 rettv->v_type = VAR_STRING;
3972 rettv->vval.v_string = get_reg_contents(regname,
3973 arg2 ? GREG_EXPR_SRC : 0);
3974 }
3975}
3976
3977/*
3978 * "getregtype()" function
3979 */
3980 static void
3981f_getregtype(typval_T *argvars, typval_T *rettv)
3982{
3983 char_u *strregname;
3984 int regname;
3985 char_u buf[NUMBUFLEN + 2];
3986 long reglen = 0;
3987
3988 if (argvars[0].v_type != VAR_UNKNOWN)
3989 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01003990 strregname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003991 if (strregname == NULL) // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02003992 {
3993 rettv->v_type = VAR_STRING;
3994 rettv->vval.v_string = NULL;
3995 return;
3996 }
3997 }
3998 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003999 // Default to v:register
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004000 strregname = get_vim_var_str(VV_REG);
4001
4002 regname = (strregname == NULL ? '"' : *strregname);
4003 if (regname == 0)
4004 regname = '"';
4005
4006 buf[0] = NUL;
4007 buf[1] = NUL;
4008 switch (get_reg_type(regname, &reglen))
4009 {
4010 case MLINE: buf[0] = 'V'; break;
4011 case MCHAR: buf[0] = 'v'; break;
4012 case MBLOCK:
4013 buf[0] = Ctrl_V;
4014 sprintf((char *)buf + 1, "%ld", reglen + 1);
4015 break;
4016 }
4017 rettv->v_type = VAR_STRING;
4018 rettv->vval.v_string = vim_strsave(buf);
4019}
4020
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +02004021/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01004022 * "gettagstack()" function
4023 */
4024 static void
4025f_gettagstack(typval_T *argvars, typval_T *rettv)
4026{
4027 win_T *wp = curwin; // default is current window
4028
4029 if (rettv_dict_alloc(rettv) != OK)
4030 return;
4031
4032 if (argvars[0].v_type != VAR_UNKNOWN)
4033 {
4034 wp = find_win_by_nr_or_id(&argvars[0]);
4035 if (wp == NULL)
4036 return;
4037 }
4038
4039 get_tagstack(wp, rettv->vval.v_dict);
4040}
4041
Bram Moolenaar0b39c3f2020-08-30 15:52:10 +02004042/*
4043 * "gettext()" function
4044 */
4045 static void
4046f_gettext(typval_T *argvars, typval_T *rettv)
4047{
4048 if (argvars[0].v_type != VAR_STRING
4049 || argvars[0].vval.v_string == NULL
4050 || *argvars[0].vval.v_string == NUL)
4051 {
4052 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
4053 }
4054 else
4055 {
4056 rettv->v_type = VAR_STRING;
4057 rettv->vval.v_string = vim_strsave(
4058 (char_u *)_(argvars[0].vval.v_string));
4059 }
4060}
4061
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01004062// for VIM_VERSION_ defines
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004063#include "version.h"
4064
4065/*
4066 * "has()" function
4067 */
Bram Moolenaara259d8d2020-01-31 20:10:50 +01004068 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004069f_has(typval_T *argvars, typval_T *rettv)
4070{
4071 int i;
4072 char_u *name;
Bram Moolenaar79296512020-03-22 16:17:14 +01004073 int x = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004074 int n = FALSE;
Bram Moolenaar79296512020-03-22 16:17:14 +01004075 typedef struct {
4076 char *name;
4077 short present;
4078 } has_item_T;
4079 static has_item_T has_list[] =
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004080 {
Bram Moolenaar79296512020-03-22 16:17:14 +01004081 {"amiga",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004082#ifdef AMIGA
Bram Moolenaar79296512020-03-22 16:17:14 +01004083 1
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004084#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004085 0
Bram Moolenaar39536dd2019-01-29 22:58:21 +01004086#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004087 },
4088 {"arp",
4089#if defined(AMIGA) && defined(FEAT_ARP)
4090 1
4091#else
4092 0
4093#endif
4094 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004095 {"haiku",
4096#ifdef __HAIKU__
4097 1
4098#else
4099 0
4100#endif
4101 },
4102 {"bsd",
4103#if defined(BSD) && !defined(MACOS_X)
4104 1
4105#else
4106 0
4107#endif
4108 },
4109 {"hpux",
4110#ifdef hpux
4111 1
4112#else
4113 0
4114#endif
4115 },
4116 {"linux",
4117#ifdef __linux__
4118 1
4119#else
4120 0
4121#endif
4122 },
4123 {"mac", // Mac OS X (and, once, Mac OS Classic)
4124#ifdef MACOS_X
4125 1
4126#else
4127 0
4128#endif
4129 },
4130 {"osx", // Mac OS X
4131#ifdef MACOS_X
4132 1
4133#else
4134 0
4135#endif
4136 },
4137 {"macunix", // Mac OS X, with the darwin feature
4138#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4139 1
4140#else
4141 0
4142#endif
4143 },
4144 {"osxdarwin", // synonym for macunix
4145#if defined(MACOS_X) && defined(MACOS_X_DARWIN)
4146 1
4147#else
4148 0
4149#endif
4150 },
4151 {"qnx",
4152#ifdef __QNX__
4153 1
4154#else
4155 0
4156#endif
4157 },
4158 {"sun",
4159#ifdef SUN_SYSTEM
4160 1
4161#else
4162 0
4163#endif
4164 },
4165 {"unix",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004166#ifdef UNIX
Bram Moolenaar79296512020-03-22 16:17:14 +01004167 1
4168#else
4169 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004170#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004171 },
4172 {"vms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004173#ifdef VMS
Bram Moolenaar79296512020-03-22 16:17:14 +01004174 1
4175#else
4176 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004177#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004178 },
4179 {"win32",
Bram Moolenaar4f974752019-02-17 17:44:42 +01004180#ifdef MSWIN
Bram Moolenaar79296512020-03-22 16:17:14 +01004181 1
4182#else
4183 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004184#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004185 },
4186 {"win32unix",
Bram Moolenaar1eed5322019-02-26 17:03:54 +01004187#if defined(UNIX) && defined(__CYGWIN__)
Bram Moolenaar79296512020-03-22 16:17:14 +01004188 1
4189#else
4190 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004191#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004192 },
4193 {"win64",
Bram Moolenaar44b443c2019-02-18 22:14:18 +01004194#ifdef _WIN64
Bram Moolenaar79296512020-03-22 16:17:14 +01004195 1
4196#else
4197 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004198#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004199 },
4200 {"ebcdic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004201#ifdef EBCDIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004202 1
4203#else
4204 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004205#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004206 },
4207 {"fname_case",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004208#ifndef CASE_INSENSITIVE_FILENAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004209 1
4210#else
4211 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004212#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004213 },
4214 {"acl",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004215#ifdef HAVE_ACL
Bram Moolenaar79296512020-03-22 16:17:14 +01004216 1
4217#else
4218 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004219#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004220 },
4221 {"arabic",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004222#ifdef FEAT_ARABIC
Bram Moolenaar79296512020-03-22 16:17:14 +01004223 1
4224#else
4225 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004226#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004227 },
4228 {"autocmd", 1},
4229 {"autochdir",
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004230#ifdef FEAT_AUTOCHDIR
Bram Moolenaar79296512020-03-22 16:17:14 +01004231 1
4232#else
4233 0
Bram Moolenaar83ec2a72018-07-27 22:08:59 +02004234#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004235 },
4236 {"autoservername",
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004237#ifdef FEAT_AUTOSERVERNAME
Bram Moolenaar79296512020-03-22 16:17:14 +01004238 1
4239#else
4240 0
Bram Moolenaare42a6d22017-11-12 19:21:51 +01004241#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004242 },
4243 {"balloon_eval",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004244#ifdef FEAT_BEVAL_GUI
Bram Moolenaar79296512020-03-22 16:17:14 +01004245 1
4246#else
4247 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004248#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004249 },
4250 {"balloon_multiline",
4251#if defined(FEAT_BEVAL_GUI) && !defined(FEAT_GUI_MSWIN)
4252 // MS-Windows requires runtime check, see below
4253 1
4254#else
4255 0
4256#endif
4257 },
4258 {"balloon_eval_term",
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01004259#ifdef FEAT_BEVAL_TERM
Bram Moolenaar79296512020-03-22 16:17:14 +01004260 1
4261#else
4262 0
Bram Moolenaar51b0f372017-11-18 18:52:04 +01004263#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004264 },
4265 {"builtin_terms",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004266#if defined(SOME_BUILTIN_TCAPS) || defined(ALL_BUILTIN_TCAPS)
Bram Moolenaar79296512020-03-22 16:17:14 +01004267 1
4268#else
4269 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004270#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004271 },
4272 {"all_builtin_terms",
4273#if defined(ALL_BUILTIN_TCAPS)
4274 1
4275#else
4276 0
4277#endif
4278 },
4279 {"browsefilter",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004280#if defined(FEAT_BROWSE) && (defined(USE_FILE_CHOOSER) \
Bram Moolenaar4f974752019-02-17 17:44:42 +01004281 || defined(FEAT_GUI_MSWIN) \
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004282 || defined(FEAT_GUI_MOTIF))
Bram Moolenaar79296512020-03-22 16:17:14 +01004283 1
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004284#else
Bram Moolenaar79296512020-03-22 16:17:14 +01004285 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02004286#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01004287 },
4288 {"byte_offset",
4289#ifdef FEAT_BYTEOFF
4290 1
4291#else
4292 0
4293#endif
4294 },
4295 {"channel",
4296#ifdef FEAT_JOB_CHANNEL
4297 1
4298#else
4299 0
4300#endif
4301 },
4302 {"cindent",
4303#ifdef FEAT_CINDENT
4304 1
4305#else
4306 0
4307#endif
4308 },
4309 {"clientserver",
4310#ifdef FEAT_CLIENTSERVER
4311 1
4312#else
4313 0
4314#endif
4315 },
4316 {"clipboard",
4317#ifdef FEAT_CLIPBOARD
4318 1
4319#else
4320 0
4321#endif
4322 },
4323 {"cmdline_compl", 1},
4324 {"cmdline_hist", 1},
4325 {"comments", 1},
4326 {"conceal",
4327#ifdef FEAT_CONCEAL
4328 1
4329#else
4330 0
4331#endif
4332 },
4333 {"cryptv",
4334#ifdef FEAT_CRYPT
4335 1
4336#else
4337 0
4338#endif
4339 },
4340 {"crypt-blowfish",
4341#ifdef FEAT_CRYPT
4342 1
4343#else
4344 0
4345#endif
4346 },
4347 {"crypt-blowfish2",
4348#ifdef FEAT_CRYPT
4349 1
4350#else
4351 0
4352#endif
4353 },
4354 {"cscope",
4355#ifdef FEAT_CSCOPE
4356 1
4357#else
4358 0
4359#endif
4360 },
4361 {"cursorbind", 1},
4362 {"cursorshape",
4363#ifdef CURSOR_SHAPE
4364 1
4365#else
4366 0
4367#endif
4368 },
4369 {"debug",
4370#ifdef DEBUG
4371 1
4372#else
4373 0
4374#endif
4375 },
4376 {"dialog_con",
4377#ifdef FEAT_CON_DIALOG
4378 1
4379#else
4380 0
4381#endif
4382 },
4383 {"dialog_gui",
4384#ifdef FEAT_GUI_DIALOG
4385 1
4386#else
4387 0
4388#endif
4389 },
4390 {"diff",
4391#ifdef FEAT_DIFF
4392 1
4393#else
4394 0
4395#endif
4396 },
4397 {"digraphs",
4398#ifdef FEAT_DIGRAPHS
4399 1
4400#else
4401 0
4402#endif
4403 },
4404 {"directx",
4405#ifdef FEAT_DIRECTX
4406 1
4407#else
4408 0
4409#endif
4410 },
4411 {"dnd",
4412#ifdef FEAT_DND
4413 1
4414#else
4415 0
4416#endif
4417 },
4418 {"emacs_tags",
4419#ifdef FEAT_EMACS_TAGS
4420 1
4421#else
4422 0
4423#endif
4424 },
4425 {"eval", 1}, // always present, of course!
4426 {"ex_extra", 1}, // graduated feature
4427 {"extra_search",
4428#ifdef FEAT_SEARCH_EXTRA
4429 1
4430#else
4431 0
4432#endif
4433 },
4434 {"file_in_path",
4435#ifdef FEAT_SEARCHPATH
4436 1
4437#else
4438 0
4439#endif
4440 },
4441 {"filterpipe",
4442#if defined(FEAT_FILTERPIPE) && !defined(VIMDLL)
4443 1
4444#else
4445 0
4446#endif
4447 },
4448 {"find_in_path",
4449#ifdef FEAT_FIND_ID
4450 1
4451#else
4452 0
4453#endif
4454 },
4455 {"float",
4456#ifdef FEAT_FLOAT
4457 1
4458#else
4459 0
4460#endif
4461 },
4462 {"folding",
4463#ifdef FEAT_FOLDING
4464 1
4465#else
4466 0
4467#endif
4468 },
4469 {"footer",
4470#ifdef FEAT_FOOTER
4471 1
4472#else
4473 0
4474#endif
4475 },
4476 {"fork",
4477#if !defined(USE_SYSTEM) && defined(UNIX)
4478 1
4479#else
4480 0
4481#endif
4482 },
4483 {"gettext",
4484#ifdef FEAT_GETTEXT
4485 1
4486#else
4487 0
4488#endif
4489 },
4490 {"gui",
4491#ifdef FEAT_GUI
4492 1
4493#else
4494 0
4495#endif
4496 },
4497 {"gui_neXtaw",
4498#if defined(FEAT_GUI_ATHENA) && defined(FEAT_GUI_NEXTAW)
4499 1
4500#else
4501 0
4502#endif
4503 },
4504 {"gui_athena",
4505#if defined(FEAT_GUI_ATHENA) && !defined(FEAT_GUI_NEXTAW)
4506 1
4507#else
4508 0
4509#endif
4510 },
4511 {"gui_gtk",
4512#ifdef FEAT_GUI_GTK
4513 1
4514#else
4515 0
4516#endif
4517 },
4518 {"gui_gtk2",
4519#if defined(FEAT_GUI_GTK) && !defined(USE_GTK3)
4520 1
4521#else
4522 0
4523#endif
4524 },
4525 {"gui_gtk3",
4526#if defined(FEAT_GUI_GTK) && defined(USE_GTK3)
4527 1
4528#else
4529 0
4530#endif
4531 },
4532 {"gui_gnome",
4533#ifdef FEAT_GUI_GNOME
4534 1
4535#else
4536 0
4537#endif
4538 },
4539 {"gui_haiku",
4540#ifdef FEAT_GUI_HAIKU
4541 1
4542#else
4543 0
4544#endif
4545 },
Bram Moolenaar097148e2020-08-11 21:58:20 +02004546 {"gui_mac", 0},
Bram Moolenaar79296512020-03-22 16:17:14 +01004547 {"gui_motif",
4548#ifdef FEAT_GUI_MOTIF
4549 1
4550#else
4551 0
4552#endif
4553 },
4554 {"gui_photon",
4555#ifdef FEAT_GUI_PHOTON
4556 1
4557#else
4558 0
4559#endif
4560 },
4561 {"gui_win32",
4562#ifdef FEAT_GUI_MSWIN
4563 1
4564#else
4565 0
4566#endif
4567 },
4568 {"iconv",
4569#if defined(HAVE_ICONV_H) && defined(USE_ICONV)
4570 1
4571#else
4572 0
4573#endif
4574 },
4575 {"insert_expand", 1},
Bram Moolenaarbfe13cc2020-04-12 17:53:12 +02004576 {"ipv6",
4577#ifdef FEAT_IPV6
4578 1
4579#else
4580 0
4581#endif
4582 },
Bram Moolenaar79296512020-03-22 16:17:14 +01004583 {"job",
4584#ifdef FEAT_JOB_CHANNEL
4585 1
4586#else
4587 0
4588#endif
4589 },
4590 {"jumplist",
4591#ifdef FEAT_JUMPLIST
4592 1
4593#else
4594 0
4595#endif
4596 },
4597 {"keymap",
4598#ifdef FEAT_KEYMAP
4599 1
4600#else
4601 0
4602#endif
4603 },
4604 {"lambda", 1}, // always with FEAT_EVAL, since 7.4.2120 with closure
4605 {"langmap",
4606#ifdef FEAT_LANGMAP
4607 1
4608#else
4609 0
4610#endif
4611 },
4612 {"libcall",
4613#ifdef FEAT_LIBCALL
4614 1
4615#else
4616 0
4617#endif
4618 },
4619 {"linebreak",
4620#ifdef FEAT_LINEBREAK
4621 1
4622#else
4623 0
4624#endif
4625 },
4626 {"lispindent",
4627#ifdef FEAT_LISP
4628 1
4629#else
4630 0
4631#endif
4632 },
4633 {"listcmds", 1},
4634 {"localmap", 1},
4635 {"lua",
4636#if defined(FEAT_LUA) && !defined(DYNAMIC_LUA)
4637 1
4638#else
4639 0
4640#endif
4641 },
4642 {"menu",
4643#ifdef FEAT_MENU
4644 1
4645#else
4646 0
4647#endif
4648 },
4649 {"mksession",
4650#ifdef FEAT_SESSION
4651 1
4652#else
4653 0
4654#endif
4655 },
4656 {"modify_fname", 1},
4657 {"mouse", 1},
4658 {"mouseshape",
4659#ifdef FEAT_MOUSESHAPE
4660 1
4661#else
4662 0
4663#endif
4664 },
4665 {"mouse_dec",
4666#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_DEC)
4667 1
4668#else
4669 0
4670#endif
4671 },
4672 {"mouse_gpm",
4673#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_GPM)
4674 1
4675#else
4676 0
4677#endif
4678 },
4679 {"mouse_jsbterm",
4680#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_JSB)
4681 1
4682#else
4683 0
4684#endif
4685 },
4686 {"mouse_netterm",
4687#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_NET)
4688 1
4689#else
4690 0
4691#endif
4692 },
4693 {"mouse_pterm",
4694#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_PTERM)
4695 1
4696#else
4697 0
4698#endif
4699 },
4700 {"mouse_sgr",
4701#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4702 1
4703#else
4704 0
4705#endif
4706 },
4707 {"mouse_sysmouse",
4708#if (defined(UNIX) || defined(VMS)) && defined(FEAT_SYSMOUSE)
4709 1
4710#else
4711 0
4712#endif
4713 },
4714 {"mouse_urxvt",
4715#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_URXVT)
4716 1
4717#else
4718 0
4719#endif
4720 },
4721 {"mouse_xterm",
4722#if (defined(UNIX) || defined(VMS)) && defined(FEAT_MOUSE_XTERM)
4723 1
4724#else
4725 0
4726#endif
4727 },
4728 {"multi_byte", 1},
4729 {"multi_byte_ime",
4730#ifdef FEAT_MBYTE_IME
4731 1
4732#else
4733 0
4734#endif
4735 },
4736 {"multi_lang",
4737#ifdef FEAT_MULTI_LANG
4738 1
4739#else
4740 0
4741#endif
4742 },
4743 {"mzscheme",
4744#if defined(FEAT_MZSCHEME) && !defined(DYNAMIC_MZSCHEME)
4745 1
4746#else
4747 0
4748#endif
4749 },
4750 {"num64", 1},
4751 {"ole",
4752#ifdef FEAT_OLE
4753 1
4754#else
4755 0
4756#endif
4757 },
4758 {"packages",
4759#ifdef FEAT_EVAL
4760 1
4761#else
4762 0
4763#endif
4764 },
4765 {"path_extra",
4766#ifdef FEAT_PATH_EXTRA
4767 1
4768#else
4769 0
4770#endif
4771 },
4772 {"perl",
4773#if defined(FEAT_PERL) && !defined(DYNAMIC_PERL)
4774 1
4775#else
4776 0
4777#endif
4778 },
4779 {"persistent_undo",
4780#ifdef FEAT_PERSISTENT_UNDO
4781 1
4782#else
4783 0
4784#endif
4785 },
4786 {"python_compiled",
4787#if defined(FEAT_PYTHON)
4788 1
4789#else
4790 0
4791#endif
4792 },
4793 {"python_dynamic",
4794#if defined(FEAT_PYTHON) && defined(DYNAMIC_PYTHON)
4795 1
4796#else
4797 0
4798#endif
4799 },
4800 {"python",
4801#if defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)
4802 1
4803#else
4804 0
4805#endif
4806 },
4807 {"pythonx",
4808#if (defined(FEAT_PYTHON) && !defined(DYNAMIC_PYTHON)) \
4809 || (defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3))
4810 1
4811#else
4812 0
4813#endif
4814 },
4815 {"python3_compiled",
4816#if defined(FEAT_PYTHON3)
4817 1
4818#else
4819 0
4820#endif
4821 },
4822 {"python3_dynamic",
4823#if defined(FEAT_PYTHON3) && defined(DYNAMIC_PYTHON3)
4824 1
4825#else
4826 0
4827#endif
4828 },
4829 {"python3",
4830#if defined(FEAT_PYTHON3) && !defined(DYNAMIC_PYTHON3)
4831 1
4832#else
4833 0
4834#endif
4835 },
4836 {"popupwin",
4837#ifdef FEAT_PROP_POPUP
4838 1
4839#else
4840 0
4841#endif
4842 },
4843 {"postscript",
4844#ifdef FEAT_POSTSCRIPT
4845 1
4846#else
4847 0
4848#endif
4849 },
4850 {"printer",
4851#ifdef FEAT_PRINTER
4852 1
4853#else
4854 0
4855#endif
4856 },
4857 {"profile",
4858#ifdef FEAT_PROFILE
4859 1
4860#else
4861 0
4862#endif
4863 },
4864 {"reltime",
4865#ifdef FEAT_RELTIME
4866 1
4867#else
4868 0
4869#endif
4870 },
4871 {"quickfix",
4872#ifdef FEAT_QUICKFIX
4873 1
4874#else
4875 0
4876#endif
4877 },
4878 {"rightleft",
4879#ifdef FEAT_RIGHTLEFT
4880 1
4881#else
4882 0
4883#endif
4884 },
4885 {"ruby",
4886#if defined(FEAT_RUBY) && !defined(DYNAMIC_RUBY)
4887 1
4888#else
4889 0
4890#endif
4891 },
4892 {"scrollbind", 1},
4893 {"showcmd",
4894#ifdef FEAT_CMDL_INFO
4895 1
4896#else
4897 0
4898#endif
4899 },
4900 {"cmdline_info",
4901#ifdef FEAT_CMDL_INFO
4902 1
4903#else
4904 0
4905#endif
4906 },
4907 {"signs",
4908#ifdef FEAT_SIGNS
4909 1
4910#else
4911 0
4912#endif
4913 },
4914 {"smartindent",
4915#ifdef FEAT_SMARTINDENT
4916 1
4917#else
4918 0
4919#endif
4920 },
4921 {"startuptime",
4922#ifdef STARTUPTIME
4923 1
4924#else
4925 0
4926#endif
4927 },
4928 {"statusline",
4929#ifdef FEAT_STL_OPT
4930 1
4931#else
4932 0
4933#endif
4934 },
4935 {"netbeans_intg",
4936#ifdef FEAT_NETBEANS_INTG
4937 1
4938#else
4939 0
4940#endif
4941 },
4942 {"sound",
4943#ifdef FEAT_SOUND
4944 1
4945#else
4946 0
4947#endif
4948 },
4949 {"spell",
4950#ifdef FEAT_SPELL
4951 1
4952#else
4953 0
4954#endif
4955 },
4956 {"syntax",
4957#ifdef FEAT_SYN_HL
4958 1
4959#else
4960 0
4961#endif
4962 },
4963 {"system",
4964#if defined(USE_SYSTEM) || !defined(UNIX)
4965 1
4966#else
4967 0
4968#endif
4969 },
4970 {"tag_binary",
4971#ifdef FEAT_TAG_BINS
4972 1
4973#else
4974 0
4975#endif
4976 },
4977 {"tcl",
4978#if defined(FEAT_TCL) && !defined(DYNAMIC_TCL)
4979 1
4980#else
4981 0
4982#endif
4983 },
4984 {"termguicolors",
4985#ifdef FEAT_TERMGUICOLORS
4986 1
4987#else
4988 0
4989#endif
4990 },
4991 {"terminal",
4992#if defined(FEAT_TERMINAL) && !defined(MSWIN)
4993 1
4994#else
4995 0
4996#endif
4997 },
4998 {"terminfo",
4999#ifdef TERMINFO
5000 1
5001#else
5002 0
5003#endif
5004 },
5005 {"termresponse",
5006#ifdef FEAT_TERMRESPONSE
5007 1
5008#else
5009 0
5010#endif
5011 },
5012 {"textobjects",
5013#ifdef FEAT_TEXTOBJ
5014 1
5015#else
5016 0
5017#endif
5018 },
5019 {"textprop",
5020#ifdef FEAT_PROP_POPUP
5021 1
5022#else
5023 0
5024#endif
5025 },
5026 {"tgetent",
5027#ifdef HAVE_TGETENT
5028 1
5029#else
5030 0
5031#endif
5032 },
5033 {"timers",
5034#ifdef FEAT_TIMERS
5035 1
5036#else
5037 0
5038#endif
5039 },
5040 {"title",
5041#ifdef FEAT_TITLE
5042 1
5043#else
5044 0
5045#endif
5046 },
5047 {"toolbar",
5048#ifdef FEAT_TOOLBAR
5049 1
5050#else
5051 0
5052#endif
5053 },
5054 {"unnamedplus",
5055#if defined(FEAT_CLIPBOARD) && defined(FEAT_X11)
5056 1
5057#else
5058 0
5059#endif
5060 },
5061 {"user-commands", 1}, // was accidentally included in 5.4
5062 {"user_commands", 1},
5063 {"vartabs",
5064#ifdef FEAT_VARTABS
5065 1
5066#else
5067 0
5068#endif
5069 },
5070 {"vertsplit", 1},
5071 {"viminfo",
5072#ifdef FEAT_VIMINFO
5073 1
5074#else
5075 0
5076#endif
5077 },
5078 {"vimscript-1", 1},
5079 {"vimscript-2", 1},
5080 {"vimscript-3", 1},
5081 {"vimscript-4", 1},
5082 {"virtualedit", 1},
5083 {"visual", 1},
5084 {"visualextra", 1},
5085 {"vreplace", 1},
5086 {"vtp",
5087#ifdef FEAT_VTP
5088 1
5089#else
5090 0
5091#endif
5092 },
5093 {"wildignore",
5094#ifdef FEAT_WILDIGN
5095 1
5096#else
5097 0
5098#endif
5099 },
5100 {"wildmenu",
5101#ifdef FEAT_WILDMENU
5102 1
5103#else
5104 0
5105#endif
5106 },
5107 {"windows", 1},
5108 {"winaltkeys",
5109#ifdef FEAT_WAK
5110 1
5111#else
5112 0
5113#endif
5114 },
5115 {"writebackup",
5116#ifdef FEAT_WRITEBACKUP
5117 1
5118#else
5119 0
5120#endif
5121 },
5122 {"xim",
5123#ifdef FEAT_XIM
5124 1
5125#else
5126 0
5127#endif
5128 },
5129 {"xfontset",
5130#ifdef FEAT_XFONTSET
5131 1
5132#else
5133 0
5134#endif
5135 },
5136 {"xpm",
5137#if defined(FEAT_XPM_W32) || defined(HAVE_XPM)
5138 1
5139#else
5140 0
5141#endif
5142 },
5143 {"xpm_w32", // for backward compatibility
5144#ifdef FEAT_XPM_W32
5145 1
5146#else
5147 0
5148#endif
5149 },
5150 {"xsmp",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005151#ifdef USE_XSMP
Bram Moolenaar79296512020-03-22 16:17:14 +01005152 1
5153#else
5154 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005155#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005156 },
5157 {"xsmp_interact",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005158#ifdef USE_XSMP_INTERACT
Bram Moolenaar79296512020-03-22 16:17:14 +01005159 1
5160#else
5161 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005162#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005163 },
5164 {"xterm_clipboard",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005165#ifdef FEAT_XCLIPBOARD
Bram Moolenaar79296512020-03-22 16:17:14 +01005166 1
5167#else
5168 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005169#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005170 },
5171 {"xterm_save",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005172#ifdef FEAT_XTERM_SAVE
Bram Moolenaar79296512020-03-22 16:17:14 +01005173 1
5174#else
5175 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005176#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005177 },
5178 {"X11",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005179#if defined(UNIX) && defined(FEAT_X11)
Bram Moolenaar79296512020-03-22 16:17:14 +01005180 1
5181#else
5182 0
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005183#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005184 },
5185 {NULL, 0}
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005186 };
5187
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005188 name = tv_get_string(&argvars[0]);
Bram Moolenaar79296512020-03-22 16:17:14 +01005189 for (i = 0; has_list[i].name != NULL; ++i)
5190 if (STRICMP(name, has_list[i].name) == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005191 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005192 x = TRUE;
5193 n = has_list[i].present;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005194 break;
5195 }
5196
Bram Moolenaar79296512020-03-22 16:17:14 +01005197 // features also in has_list[] but sometimes enabled at runtime
5198 if (x == TRUE && n == FALSE)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005199 {
Bram Moolenaar79296512020-03-22 16:17:14 +01005200 if (0)
Bram Moolenaar86b9a3e2020-04-07 19:57:29 +02005201 {
5202 // intentionally empty
5203 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01005204#if defined(FEAT_BEVAL) && defined(FEAT_GUI_MSWIN)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005205 else if (STRICMP(name, "balloon_multiline") == 0)
5206 n = multiline_balloon_available();
5207#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005208#ifdef VIMDLL
5209 else if (STRICMP(name, "filterpipe") == 0)
5210 n = gui.in_use || gui.starting;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005211#endif
5212#if defined(USE_ICONV) && defined(DYNAMIC_ICONV)
5213 else if (STRICMP(name, "iconv") == 0)
5214 n = iconv_enabled(FALSE);
5215#endif
5216#ifdef DYNAMIC_LUA
5217 else if (STRICMP(name, "lua") == 0)
5218 n = lua_enabled(FALSE);
5219#endif
5220#ifdef DYNAMIC_MZSCHEME
5221 else if (STRICMP(name, "mzscheme") == 0)
5222 n = mzscheme_enabled(FALSE);
5223#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005224#ifdef DYNAMIC_PERL
5225 else if (STRICMP(name, "perl") == 0)
5226 n = perl_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005227#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005228#ifdef DYNAMIC_PYTHON
5229 else if (STRICMP(name, "python") == 0)
5230 n = python_enabled(FALSE);
5231#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005232#ifdef DYNAMIC_PYTHON3
5233 else if (STRICMP(name, "python3") == 0)
5234 n = python3_enabled(FALSE);
5235#endif
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01005236#if defined(DYNAMIC_PYTHON) || defined(DYNAMIC_PYTHON3)
5237 else if (STRICMP(name, "pythonx") == 0)
5238 {
5239# if defined(DYNAMIC_PYTHON) && defined(DYNAMIC_PYTHON3)
5240 if (p_pyx == 0)
5241 n = python3_enabled(FALSE) || python_enabled(FALSE);
5242 else if (p_pyx == 3)
5243 n = python3_enabled(FALSE);
5244 else if (p_pyx == 2)
5245 n = python_enabled(FALSE);
5246# elif defined(DYNAMIC_PYTHON)
5247 n = python_enabled(FALSE);
5248# elif defined(DYNAMIC_PYTHON3)
5249 n = python3_enabled(FALSE);
5250# endif
5251 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005252#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005253#ifdef DYNAMIC_RUBY
5254 else if (STRICMP(name, "ruby") == 0)
5255 n = ruby_enabled(FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005256#endif
Bram Moolenaar79296512020-03-22 16:17:14 +01005257#ifdef DYNAMIC_TCL
5258 else if (STRICMP(name, "tcl") == 0)
5259 n = tcl_enabled(FALSE);
Bram Moolenaar4b8366b2019-05-04 17:34:34 +02005260#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01005261#if defined(FEAT_TERMINAL) && defined(MSWIN)
Bram Moolenaara83e3962017-08-17 14:39:07 +02005262 else if (STRICMP(name, "terminal") == 0)
5263 n = terminal_enabled();
5264#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005265 }
5266
Bram Moolenaar79296512020-03-22 16:17:14 +01005267 // features not in has_list[]
5268 if (x == FALSE)
5269 {
5270 if (STRNICMP(name, "patch", 5) == 0)
5271 {
5272 x = TRUE;
5273 if (name[5] == '-'
5274 && STRLEN(name) >= 11
5275 && vim_isdigit(name[6])
5276 && vim_isdigit(name[8])
5277 && vim_isdigit(name[10]))
5278 {
5279 int major = atoi((char *)name + 6);
5280 int minor = atoi((char *)name + 8);
5281
5282 // Expect "patch-9.9.01234".
5283 n = (major < VIM_VERSION_MAJOR
5284 || (major == VIM_VERSION_MAJOR
5285 && (minor < VIM_VERSION_MINOR
5286 || (minor == VIM_VERSION_MINOR
5287 && has_patch(atoi((char *)name + 10))))));
5288 }
5289 else
5290 n = has_patch(atoi((char *)name + 5));
5291 }
5292 else if (STRICMP(name, "vim_starting") == 0)
5293 {
5294 x = TRUE;
5295 n = (starting != 0);
5296 }
5297 else if (STRICMP(name, "ttyin") == 0)
5298 {
5299 x = TRUE;
5300 n = mch_input_isatty();
5301 }
5302 else if (STRICMP(name, "ttyout") == 0)
5303 {
5304 x = TRUE;
5305 n = stdout_isatty;
5306 }
5307 else if (STRICMP(name, "multi_byte_encoding") == 0)
5308 {
5309 x = TRUE;
5310 n = has_mbyte;
5311 }
5312 else if (STRICMP(name, "gui_running") == 0)
5313 {
5314 x = TRUE;
5315#ifdef FEAT_GUI
5316 n = (gui.in_use || gui.starting);
5317#endif
5318 }
5319 else if (STRICMP(name, "browse") == 0)
5320 {
5321 x = TRUE;
5322#if defined(FEAT_GUI) && defined(FEAT_BROWSE)
5323 n = gui.in_use; // gui_mch_browse() works when GUI is running
5324#endif
5325 }
5326 else if (STRICMP(name, "syntax_items") == 0)
5327 {
5328 x = TRUE;
5329#ifdef FEAT_SYN_HL
5330 n = syntax_present(curwin);
5331#endif
5332 }
5333 else if (STRICMP(name, "vcon") == 0)
5334 {
5335 x = TRUE;
5336#ifdef FEAT_VTP
5337 n = is_term_win32() && has_vtp_working();
5338#endif
5339 }
5340 else if (STRICMP(name, "netbeans_enabled") == 0)
5341 {
5342 x = TRUE;
5343#ifdef FEAT_NETBEANS_INTG
5344 n = netbeans_active();
5345#endif
5346 }
5347 else if (STRICMP(name, "mouse_gpm_enabled") == 0)
5348 {
5349 x = TRUE;
5350#ifdef FEAT_MOUSE_GPM
5351 n = gpm_enabled();
5352#endif
5353 }
5354 else if (STRICMP(name, "conpty") == 0)
5355 {
5356 x = TRUE;
5357#if defined(FEAT_TERMINAL) && defined(MSWIN)
5358 n = use_conpty();
5359#endif
5360 }
5361 else if (STRICMP(name, "clipboard_working") == 0)
5362 {
5363 x = TRUE;
5364#ifdef FEAT_CLIPBOARD
5365 n = clip_star.available;
5366#endif
5367 }
5368 }
5369
Bram Moolenaar04637e22020-09-05 18:45:29 +02005370 if (argvars[1].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[1]))
Bram Moolenaar79296512020-03-22 16:17:14 +01005371 // return whether feature could ever be enabled
5372 rettv->vval.v_number = x;
5373 else
5374 // return whether feature is enabled
5375 rettv->vval.v_number = n;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005376}
5377
5378/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005379 * "haslocaldir()" function
5380 */
5381 static void
5382f_haslocaldir(typval_T *argvars, typval_T *rettv)
5383{
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005384 tabpage_T *tp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005385 win_T *wp = NULL;
5386
Bram Moolenaar00aa0692019-04-27 20:37:57 +02005387 wp = find_tabwin(&argvars[0], &argvars[1], &tp);
5388
5389 // Check for window-local and tab-local directories
5390 if (wp != NULL && wp->w_localdir != NULL)
5391 rettv->vval.v_number = 1;
5392 else if (tp != NULL && tp->tp_localdir != NULL)
5393 rettv->vval.v_number = 2;
5394 else
5395 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005396}
5397
5398/*
5399 * "hasmapto()" function
5400 */
5401 static void
5402f_hasmapto(typval_T *argvars, typval_T *rettv)
5403{
5404 char_u *name;
5405 char_u *mode;
5406 char_u buf[NUMBUFLEN];
5407 int abbr = FALSE;
5408
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005409 name = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005410 if (argvars[1].v_type == VAR_UNKNOWN)
5411 mode = (char_u *)"nvo";
5412 else
5413 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005414 mode = tv_get_string_buf(&argvars[1], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005415 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02005416 abbr = (int)tv_get_bool(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005417 }
5418
5419 if (map_to_exists(name, mode, abbr))
5420 rettv->vval.v_number = TRUE;
5421 else
5422 rettv->vval.v_number = FALSE;
5423}
5424
5425/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005426 * "highlightID(name)" function
5427 */
5428 static void
5429f_hlID(typval_T *argvars, typval_T *rettv)
5430{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005431 rettv->vval.v_number = syn_name2id(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005432}
5433
5434/*
5435 * "highlight_exists()" function
5436 */
5437 static void
5438f_hlexists(typval_T *argvars, typval_T *rettv)
5439{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005440 rettv->vval.v_number = highlight_exists(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005441}
5442
5443/*
5444 * "hostname()" function
5445 */
5446 static void
5447f_hostname(typval_T *argvars UNUSED, typval_T *rettv)
5448{
5449 char_u hostname[256];
5450
5451 mch_get_host_name(hostname, 256);
5452 rettv->v_type = VAR_STRING;
5453 rettv->vval.v_string = vim_strsave(hostname);
5454}
5455
5456/*
5457 * iconv() function
5458 */
5459 static void
5460f_iconv(typval_T *argvars UNUSED, typval_T *rettv)
5461{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005462 char_u buf1[NUMBUFLEN];
5463 char_u buf2[NUMBUFLEN];
5464 char_u *from, *to, *str;
5465 vimconv_T vimconv;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005466
5467 rettv->v_type = VAR_STRING;
5468 rettv->vval.v_string = NULL;
5469
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005470 str = tv_get_string(&argvars[0]);
5471 from = enc_canonize(enc_skip(tv_get_string_buf(&argvars[1], buf1)));
5472 to = enc_canonize(enc_skip(tv_get_string_buf(&argvars[2], buf2)));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005473 vimconv.vc_type = CONV_NONE;
5474 convert_setup(&vimconv, from, to);
5475
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005476 // If the encodings are equal, no conversion needed.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005477 if (vimconv.vc_type == CONV_NONE)
5478 rettv->vval.v_string = vim_strsave(str);
5479 else
5480 rettv->vval.v_string = string_convert(&vimconv, str, NULL);
5481
5482 convert_setup(&vimconv, NULL, NULL);
5483 vim_free(from);
5484 vim_free(to);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005485}
5486
5487/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005488 * "index()" function
5489 */
5490 static void
5491f_index(typval_T *argvars, typval_T *rettv)
5492{
5493 list_T *l;
5494 listitem_T *item;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005495 blob_T *b;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005496 long idx = 0;
5497 int ic = FALSE;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005498 int error = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005499
5500 rettv->vval.v_number = -1;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005501 if (argvars[0].v_type == VAR_BLOB)
5502 {
5503 typval_T tv;
5504 int start = 0;
5505
5506 if (argvars[2].v_type != VAR_UNKNOWN)
5507 {
5508 start = tv_get_number_chk(&argvars[2], &error);
5509 if (error)
5510 return;
5511 }
5512 b = argvars[0].vval.v_blob;
5513 if (b == NULL)
5514 return;
Bram Moolenaar05500ec2019-01-13 19:10:33 +01005515 if (start < 0)
5516 {
5517 start = blob_len(b) + start;
5518 if (start < 0)
5519 start = 0;
5520 }
5521
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005522 for (idx = start; idx < blob_len(b); ++idx)
5523 {
5524 tv.v_type = VAR_NUMBER;
5525 tv.vval.v_number = blob_get(b, idx);
5526 if (tv_equal(&tv, &argvars[1], ic, FALSE))
5527 {
5528 rettv->vval.v_number = idx;
5529 return;
5530 }
5531 }
5532 return;
5533 }
5534 else if (argvars[0].v_type != VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005535 {
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +01005536 emsg(_(e_listblobreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005537 return;
5538 }
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005539
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005540 l = argvars[0].vval.v_list;
5541 if (l != NULL)
5542 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005543 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005544 item = l->lv_first;
5545 if (argvars[2].v_type != VAR_UNKNOWN)
5546 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005547 // Start at specified item. Use the cached index that list_find()
5548 // sets, so that a negative number also works.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005549 item = list_find(l, (long)tv_get_number_chk(&argvars[2], &error));
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01005550 idx = l->lv_u.mat.lv_idx;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005551 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar6c553f92020-09-02 22:10:34 +02005552 ic = (int)tv_get_bool_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005553 if (error)
5554 item = NULL;
5555 }
5556
5557 for ( ; item != NULL; item = item->li_next, ++idx)
5558 if (tv_equal(&item->li_tv, &argvars[1], ic, FALSE))
5559 {
5560 rettv->vval.v_number = idx;
5561 break;
5562 }
5563 }
5564}
5565
5566static int inputsecret_flag = 0;
5567
5568/*
5569 * "input()" function
5570 * Also handles inputsecret() when inputsecret is set.
5571 */
5572 static void
5573f_input(typval_T *argvars, typval_T *rettv)
5574{
5575 get_user_input(argvars, rettv, FALSE, inputsecret_flag);
5576}
5577
5578/*
5579 * "inputdialog()" function
5580 */
5581 static void
5582f_inputdialog(typval_T *argvars, typval_T *rettv)
5583{
5584#if defined(FEAT_GUI_TEXTDIALOG)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005585 // Use a GUI dialog if the GUI is running and 'c' is not in 'guioptions'
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005586 if (gui.in_use && vim_strchr(p_go, GO_CONDIALOG) == NULL)
5587 {
5588 char_u *message;
5589 char_u buf[NUMBUFLEN];
5590 char_u *defstr = (char_u *)"";
5591
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005592 message = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005593 if (argvars[1].v_type != VAR_UNKNOWN
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005594 && (defstr = tv_get_string_buf_chk(&argvars[1], buf)) != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005595 vim_strncpy(IObuff, defstr, IOSIZE - 1);
5596 else
5597 IObuff[0] = NUL;
5598 if (message != NULL && defstr != NULL
5599 && do_dialog(VIM_QUESTION, NULL, message,
5600 (char_u *)_("&OK\n&Cancel"), 1, IObuff, FALSE) == 1)
5601 rettv->vval.v_string = vim_strsave(IObuff);
5602 else
5603 {
5604 if (message != NULL && defstr != NULL
5605 && argvars[1].v_type != VAR_UNKNOWN
5606 && argvars[2].v_type != VAR_UNKNOWN)
5607 rettv->vval.v_string = vim_strsave(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005608 tv_get_string_buf(&argvars[2], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005609 else
5610 rettv->vval.v_string = NULL;
5611 }
5612 rettv->v_type = VAR_STRING;
5613 }
5614 else
5615#endif
5616 get_user_input(argvars, rettv, TRUE, inputsecret_flag);
5617}
5618
5619/*
5620 * "inputlist()" function
5621 */
5622 static void
5623f_inputlist(typval_T *argvars, typval_T *rettv)
5624{
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005625 list_T *l;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005626 listitem_T *li;
5627 int selected;
5628 int mouse_used;
5629
5630#ifdef NO_CONSOLE_INPUT
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005631 // While starting up, there is no place to enter text. When running tests
5632 // with --not-a-term we assume feedkeys() will be used.
Bram Moolenaar91d348a2017-07-29 20:16:03 +02005633 if (no_console_input() && !is_not_a_term())
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005634 return;
5635#endif
5636 if (argvars[0].v_type != VAR_LIST || argvars[0].vval.v_list == NULL)
5637 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005638 semsg(_(e_listarg), "inputlist()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005639 return;
5640 }
5641
5642 msg_start();
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005643 msg_row = Rows - 1; // for when 'cmdheight' > 1
5644 lines_left = Rows; // avoid more prompt
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005645 msg_scroll = TRUE;
5646 msg_clr_eos();
5647
Bram Moolenaar50985eb2020-01-27 22:09:39 +01005648 l = argvars[0].vval.v_list;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02005649 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02005650 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005651 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005652 msg_puts((char *)tv_get_string(&li->li_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005653 msg_putchar('\n');
5654 }
5655
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005656 // Ask for choice.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005657 selected = prompt_for_number(&mouse_used);
5658 if (mouse_used)
5659 selected -= lines_left;
5660
5661 rettv->vval.v_number = selected;
5662}
5663
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005664static garray_T ga_userinput = {0, 0, sizeof(tasave_T), 4, NULL};
5665
5666/*
5667 * "inputrestore()" function
5668 */
5669 static void
5670f_inputrestore(typval_T *argvars UNUSED, typval_T *rettv)
5671{
5672 if (ga_userinput.ga_len > 0)
5673 {
5674 --ga_userinput.ga_len;
5675 restore_typeahead((tasave_T *)(ga_userinput.ga_data)
5676 + ga_userinput.ga_len);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005677 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005678 }
5679 else if (p_verbose > 1)
5680 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005681 verb_msg(_("called inputrestore() more often than inputsave()"));
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005682 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005683 }
5684}
5685
5686/*
5687 * "inputsave()" function
5688 */
5689 static void
5690f_inputsave(typval_T *argvars UNUSED, typval_T *rettv)
5691{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005692 // Add an entry to the stack of typeahead storage.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005693 if (ga_grow(&ga_userinput, 1) == OK)
5694 {
5695 save_typeahead((tasave_T *)(ga_userinput.ga_data)
5696 + ga_userinput.ga_len);
5697 ++ga_userinput.ga_len;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005698 // default return is zero == OK
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005699 }
5700 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005701 rettv->vval.v_number = 1; // Failed
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005702}
5703
5704/*
5705 * "inputsecret()" function
5706 */
5707 static void
5708f_inputsecret(typval_T *argvars, typval_T *rettv)
5709{
5710 ++cmdline_star;
5711 ++inputsecret_flag;
5712 f_input(argvars, rettv);
5713 --cmdline_star;
5714 --inputsecret_flag;
5715}
5716
5717/*
Bram Moolenaar67a2deb2019-11-25 00:05:32 +01005718 * "interrupt()" function
5719 */
5720 static void
5721f_interrupt(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
5722{
5723 got_int = TRUE;
5724}
5725
5726/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005727 * "invert(expr)" function
5728 */
5729 static void
5730f_invert(typval_T *argvars, typval_T *rettv)
5731{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005732 rettv->vval.v_number = ~tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005733}
5734
5735/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005736 * "islocked()" function
5737 */
5738 static void
5739f_islocked(typval_T *argvars, typval_T *rettv)
5740{
5741 lval_T lv;
5742 char_u *end;
5743 dictitem_T *di;
5744
5745 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005746 end = get_lval(tv_get_string(&argvars[0]), NULL, &lv, FALSE, FALSE,
Bram Moolenaar3a257732017-02-21 20:47:13 +01005747 GLV_NO_AUTOLOAD | GLV_READ_ONLY, FNE_CHECK_START);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005748 if (end != NULL && lv.ll_name != NULL)
5749 {
5750 if (*end != NUL)
Bram Moolenaar2d06bfd2020-07-23 17:16:18 +02005751 semsg(_(e_trailing_arg), end);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005752 else
5753 {
5754 if (lv.ll_tv == NULL)
5755 {
Bram Moolenaar79518e22017-02-17 16:31:35 +01005756 di = find_var(lv.ll_name, NULL, TRUE);
5757 if (di != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005758 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005759 // Consider a variable locked when:
5760 // 1. the variable itself is locked
5761 // 2. the value of the variable is locked.
5762 // 3. the List or Dict value is locked.
Bram Moolenaar79518e22017-02-17 16:31:35 +01005763 rettv->vval.v_number = ((di->di_flags & DI_FLAGS_LOCK)
5764 || tv_islocked(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005765 }
5766 }
5767 else if (lv.ll_range)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005768 emsg(_("E786: Range not allowed"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005769 else if (lv.ll_newkey != NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005770 semsg(_(e_dictkey), lv.ll_newkey);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005771 else if (lv.ll_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005772 // List item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005773 rettv->vval.v_number = tv_islocked(&lv.ll_li->li_tv);
5774 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005775 // Dictionary item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005776 rettv->vval.v_number = tv_islocked(&lv.ll_di->di_tv);
5777 }
5778 }
5779
5780 clear_lval(&lv);
5781}
5782
5783#if defined(FEAT_FLOAT) && defined(HAVE_MATH_H)
5784/*
Bram Moolenaarfda1bff2019-04-04 13:44:37 +02005785 * "isinf()" function
5786 */
5787 static void
5788f_isinf(typval_T *argvars, typval_T *rettv)
5789{
5790 if (argvars[0].v_type == VAR_FLOAT && isinf(argvars[0].vval.v_float))
5791 rettv->vval.v_number = argvars[0].vval.v_float > 0.0 ? 1 : -1;
5792}
5793
5794/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005795 * "isnan()" function
5796 */
5797 static void
5798f_isnan(typval_T *argvars, typval_T *rettv)
5799{
5800 rettv->vval.v_number = argvars[0].v_type == VAR_FLOAT
5801 && isnan(argvars[0].vval.v_float);
5802}
5803#endif
5804
5805/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005806 * "last_buffer_nr()" function.
5807 */
5808 static void
5809f_last_buffer_nr(typval_T *argvars UNUSED, typval_T *rettv)
5810{
5811 int n = 0;
5812 buf_T *buf;
5813
Bram Moolenaar29323592016-07-24 22:04:11 +02005814 FOR_ALL_BUFFERS(buf)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005815 if (n < buf->b_fnum)
5816 n = buf->b_fnum;
5817
5818 rettv->vval.v_number = n;
5819}
5820
5821/*
5822 * "len()" function
5823 */
5824 static void
5825f_len(typval_T *argvars, typval_T *rettv)
5826{
5827 switch (argvars[0].v_type)
5828 {
5829 case VAR_STRING:
5830 case VAR_NUMBER:
5831 rettv->vval.v_number = (varnumber_T)STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005832 tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005833 break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01005834 case VAR_BLOB:
5835 rettv->vval.v_number = blob_len(argvars[0].vval.v_blob);
5836 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005837 case VAR_LIST:
5838 rettv->vval.v_number = list_len(argvars[0].vval.v_list);
5839 break;
5840 case VAR_DICT:
5841 rettv->vval.v_number = dict_len(argvars[0].vval.v_dict);
5842 break;
5843 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02005844 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01005845 case VAR_VOID:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01005846 case VAR_BOOL:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005847 case VAR_SPECIAL:
5848 case VAR_FLOAT:
5849 case VAR_FUNC:
5850 case VAR_PARTIAL:
5851 case VAR_JOB:
5852 case VAR_CHANNEL:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005853 emsg(_("E701: Invalid type for len()"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005854 break;
5855 }
5856}
5857
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005858 static void
Bram Moolenaar6d721c72017-01-17 16:56:28 +01005859libcall_common(typval_T *argvars UNUSED, typval_T *rettv, int type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005860{
5861#ifdef FEAT_LIBCALL
5862 char_u *string_in;
5863 char_u **string_result;
5864 int nr_result;
5865#endif
5866
5867 rettv->v_type = type;
5868 if (type != VAR_NUMBER)
5869 rettv->vval.v_string = NULL;
5870
5871 if (check_restricted() || check_secure())
5872 return;
5873
5874#ifdef FEAT_LIBCALL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01005875 // The first two args must be strings, otherwise it's meaningless
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005876 if (argvars[0].v_type == VAR_STRING && argvars[1].v_type == VAR_STRING)
5877 {
5878 string_in = NULL;
5879 if (argvars[2].v_type == VAR_STRING)
5880 string_in = argvars[2].vval.v_string;
5881 if (type == VAR_NUMBER)
5882 string_result = NULL;
5883 else
5884 string_result = &rettv->vval.v_string;
5885 if (mch_libcall(argvars[0].vval.v_string,
5886 argvars[1].vval.v_string,
5887 string_in,
5888 argvars[2].vval.v_number,
5889 string_result,
5890 &nr_result) == OK
5891 && type == VAR_NUMBER)
5892 rettv->vval.v_number = nr_result;
5893 }
5894#endif
5895}
5896
5897/*
5898 * "libcall()" function
5899 */
5900 static void
5901f_libcall(typval_T *argvars, typval_T *rettv)
5902{
5903 libcall_common(argvars, rettv, VAR_STRING);
5904}
5905
5906/*
5907 * "libcallnr()" function
5908 */
5909 static void
5910f_libcallnr(typval_T *argvars, typval_T *rettv)
5911{
5912 libcall_common(argvars, rettv, VAR_NUMBER);
5913}
5914
5915/*
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005916 * "line(string, [winid])" function
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005917 */
5918 static void
5919f_line(typval_T *argvars, typval_T *rettv)
5920{
5921 linenr_T lnum = 0;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005922 pos_T *fp = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005923 int fnum;
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005924 int id;
5925 tabpage_T *tp;
5926 win_T *wp;
5927 win_T *save_curwin;
5928 tabpage_T *save_curtab;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005929
Bram Moolenaar8e0a8e72019-09-02 22:56:24 +02005930 if (argvars[1].v_type != VAR_UNKNOWN)
5931 {
5932 // use window specified in the second argument
5933 id = (int)tv_get_number(&argvars[1]);
5934 wp = win_id2wp_tp(id, &tp);
5935 if (wp != NULL && tp != NULL)
5936 {
5937 if (switch_win_noblock(&save_curwin, &save_curtab, wp, tp, TRUE)
5938 == OK)
5939 {
5940 check_cursor();
5941 fp = var2fpos(&argvars[0], TRUE, &fnum);
5942 }
5943 restore_win_noblock(save_curwin, save_curtab, TRUE);
5944 }
5945 }
5946 else
5947 // use current window
5948 fp = var2fpos(&argvars[0], TRUE, &fnum);
5949
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005950 if (fp != NULL)
5951 lnum = fp->lnum;
5952 rettv->vval.v_number = lnum;
5953}
5954
5955/*
5956 * "line2byte(lnum)" function
5957 */
5958 static void
5959f_line2byte(typval_T *argvars UNUSED, typval_T *rettv)
5960{
5961#ifndef FEAT_BYTEOFF
5962 rettv->vval.v_number = -1;
5963#else
5964 linenr_T lnum;
5965
Bram Moolenaard155d7a2018-12-21 16:04:21 +01005966 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005967 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count + 1)
5968 rettv->vval.v_number = -1;
5969 else
5970 rettv->vval.v_number = ml_find_line_or_offset(curbuf, lnum, NULL);
5971 if (rettv->vval.v_number >= 0)
5972 ++rettv->vval.v_number;
5973#endif
5974}
5975
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02005976#ifdef FEAT_FLOAT
5977/*
5978 * "log()" function
5979 */
5980 static void
5981f_log(typval_T *argvars, typval_T *rettv)
5982{
5983 float_T f = 0.0;
5984
5985 rettv->v_type = VAR_FLOAT;
5986 if (get_float_arg(argvars, &f) == OK)
5987 rettv->vval.v_float = log(f);
5988 else
5989 rettv->vval.v_float = 0.0;
5990}
5991
5992/*
5993 * "log10()" function
5994 */
5995 static void
5996f_log10(typval_T *argvars, typval_T *rettv)
5997{
5998 float_T f = 0.0;
5999
6000 rettv->v_type = VAR_FLOAT;
6001 if (get_float_arg(argvars, &f) == OK)
6002 rettv->vval.v_float = log10(f);
6003 else
6004 rettv->vval.v_float = 0.0;
6005}
6006#endif
6007
6008#ifdef FEAT_LUA
6009/*
6010 * "luaeval()" function
6011 */
6012 static void
6013f_luaeval(typval_T *argvars, typval_T *rettv)
6014{
6015 char_u *str;
6016 char_u buf[NUMBUFLEN];
6017
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006018 if (check_restricted() || check_secure())
6019 return;
6020
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006021 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006022 do_luaeval(str, argvars + 1, rettv);
6023}
6024#endif
6025
6026/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006027 * "maparg()" function
6028 */
6029 static void
6030f_maparg(typval_T *argvars, typval_T *rettv)
6031{
6032 get_maparg(argvars, rettv, TRUE);
6033}
6034
6035/*
6036 * "mapcheck()" function
6037 */
6038 static void
6039f_mapcheck(typval_T *argvars, typval_T *rettv)
6040{
6041 get_maparg(argvars, rettv, FALSE);
6042}
6043
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006044typedef enum
6045{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006046 MATCH_END, // matchend()
6047 MATCH_MATCH, // match()
6048 MATCH_STR, // matchstr()
6049 MATCH_LIST, // matchlist()
6050 MATCH_POS // matchstrpos()
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006051} matchtype_T;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006052
6053 static void
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006054find_some_match(typval_T *argvars, typval_T *rettv, matchtype_T type)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006055{
6056 char_u *str = NULL;
6057 long len = 0;
6058 char_u *expr = NULL;
6059 char_u *pat;
6060 regmatch_T regmatch;
6061 char_u patbuf[NUMBUFLEN];
6062 char_u strbuf[NUMBUFLEN];
6063 char_u *save_cpo;
6064 long start = 0;
6065 long nth = 1;
6066 colnr_T startcol = 0;
6067 int match = 0;
6068 list_T *l = NULL;
6069 listitem_T *li = NULL;
6070 long idx = 0;
6071 char_u *tofree = NULL;
6072
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006073 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006074 save_cpo = p_cpo;
6075 p_cpo = (char_u *)"";
6076
6077 rettv->vval.v_number = -1;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006078 if (type == MATCH_LIST || type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006079 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006080 // type MATCH_LIST: return empty list when there are no matches.
6081 // type MATCH_POS: return ["", -1, -1, -1]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006082 if (rettv_list_alloc(rettv) == FAIL)
6083 goto theend;
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006084 if (type == MATCH_POS
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006085 && (list_append_string(rettv->vval.v_list,
6086 (char_u *)"", 0) == FAIL
6087 || list_append_number(rettv->vval.v_list,
6088 (varnumber_T)-1) == FAIL
6089 || list_append_number(rettv->vval.v_list,
6090 (varnumber_T)-1) == FAIL
6091 || list_append_number(rettv->vval.v_list,
6092 (varnumber_T)-1) == FAIL))
6093 {
6094 list_free(rettv->vval.v_list);
6095 rettv->vval.v_list = NULL;
6096 goto theend;
6097 }
6098 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006099 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006100 {
6101 rettv->v_type = VAR_STRING;
6102 rettv->vval.v_string = NULL;
6103 }
6104
6105 if (argvars[0].v_type == VAR_LIST)
6106 {
6107 if ((l = argvars[0].vval.v_list) == NULL)
6108 goto theend;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006109 CHECK_LIST_MATERIALIZE(l);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006110 li = l->lv_first;
6111 }
6112 else
6113 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006114 expr = str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006115 len = (long)STRLEN(str);
6116 }
6117
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006118 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006119 if (pat == NULL)
6120 goto theend;
6121
6122 if (argvars[2].v_type != VAR_UNKNOWN)
6123 {
6124 int error = FALSE;
6125
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006126 start = (long)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006127 if (error)
6128 goto theend;
6129 if (l != NULL)
6130 {
6131 li = list_find(l, start);
6132 if (li == NULL)
6133 goto theend;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006134 idx = l->lv_u.mat.lv_idx; // use the cached index
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006135 }
6136 else
6137 {
6138 if (start < 0)
6139 start = 0;
6140 if (start > len)
6141 goto theend;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006142 // When "count" argument is there ignore matches before "start",
6143 // otherwise skip part of the string. Differs when pattern is "^"
6144 // or "\<".
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006145 if (argvars[3].v_type != VAR_UNKNOWN)
6146 startcol = start;
6147 else
6148 {
6149 str += start;
6150 len -= start;
6151 }
6152 }
6153
6154 if (argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006155 nth = (long)tv_get_number_chk(&argvars[3], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006156 if (error)
6157 goto theend;
6158 }
6159
6160 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
6161 if (regmatch.regprog != NULL)
6162 {
6163 regmatch.rm_ic = p_ic;
6164
6165 for (;;)
6166 {
6167 if (l != NULL)
6168 {
6169 if (li == NULL)
6170 {
6171 match = FALSE;
6172 break;
6173 }
6174 vim_free(tofree);
6175 expr = str = echo_string(&li->li_tv, &tofree, strbuf, 0);
6176 if (str == NULL)
6177 break;
6178 }
6179
6180 match = vim_regexec_nl(&regmatch, str, (colnr_T)startcol);
6181
6182 if (match && --nth <= 0)
6183 break;
6184 if (l == NULL && !match)
6185 break;
6186
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006187 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006188 if (l != NULL)
6189 {
6190 li = li->li_next;
6191 ++idx;
6192 }
6193 else
6194 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006195 startcol = (colnr_T)(regmatch.startp[0]
6196 + (*mb_ptr2len)(regmatch.startp[0]) - str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006197 if (startcol > (colnr_T)len
6198 || str + startcol <= regmatch.startp[0])
6199 {
6200 match = FALSE;
6201 break;
6202 }
6203 }
6204 }
6205
6206 if (match)
6207 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006208 if (type == MATCH_POS)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006209 {
6210 listitem_T *li1 = rettv->vval.v_list->lv_first;
6211 listitem_T *li2 = li1->li_next;
6212 listitem_T *li3 = li2->li_next;
6213 listitem_T *li4 = li3->li_next;
6214
6215 vim_free(li1->li_tv.vval.v_string);
6216 li1->li_tv.vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006217 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006218 li3->li_tv.vval.v_number =
6219 (varnumber_T)(regmatch.startp[0] - expr);
6220 li4->li_tv.vval.v_number =
6221 (varnumber_T)(regmatch.endp[0] - expr);
6222 if (l != NULL)
6223 li2->li_tv.vval.v_number = (varnumber_T)idx;
6224 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006225 else if (type == MATCH_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006226 {
6227 int i;
6228
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006229 // return list with matched string and submatches
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006230 for (i = 0; i < NSUBEXP; ++i)
6231 {
6232 if (regmatch.endp[i] == NULL)
6233 {
6234 if (list_append_string(rettv->vval.v_list,
6235 (char_u *)"", 0) == FAIL)
6236 break;
6237 }
6238 else if (list_append_string(rettv->vval.v_list,
6239 regmatch.startp[i],
6240 (int)(regmatch.endp[i] - regmatch.startp[i]))
6241 == FAIL)
6242 break;
6243 }
6244 }
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006245 else if (type == MATCH_STR)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006246 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006247 // return matched string
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006248 if (l != NULL)
6249 copy_tv(&li->li_tv, rettv);
6250 else
6251 rettv->vval.v_string = vim_strnsave(regmatch.startp[0],
Bram Moolenaardf44a272020-06-07 20:49:05 +02006252 regmatch.endp[0] - regmatch.startp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006253 }
6254 else if (l != NULL)
6255 rettv->vval.v_number = idx;
6256 else
6257 {
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006258 if (type != MATCH_END)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006259 rettv->vval.v_number =
6260 (varnumber_T)(regmatch.startp[0] - str);
6261 else
6262 rettv->vval.v_number =
6263 (varnumber_T)(regmatch.endp[0] - str);
6264 rettv->vval.v_number += (varnumber_T)(str - expr);
6265 }
6266 }
6267 vim_regfree(regmatch.regprog);
6268 }
6269
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006270theend:
6271 if (type == MATCH_POS && l == NULL && rettv->vval.v_list != NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006272 // matchstrpos() without a list: drop the second item.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006273 listitem_remove(rettv->vval.v_list,
6274 rettv->vval.v_list->lv_first->li_next);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006275 vim_free(tofree);
6276 p_cpo = save_cpo;
6277}
6278
6279/*
6280 * "match()" function
6281 */
6282 static void
6283f_match(typval_T *argvars, typval_T *rettv)
6284{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006285 find_some_match(argvars, rettv, MATCH_MATCH);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006286}
6287
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006288/*
6289 * "matchend()" function
6290 */
6291 static void
6292f_matchend(typval_T *argvars, typval_T *rettv)
6293{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006294 find_some_match(argvars, rettv, MATCH_END);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006295}
6296
6297/*
6298 * "matchlist()" function
6299 */
6300 static void
6301f_matchlist(typval_T *argvars, typval_T *rettv)
6302{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006303 find_some_match(argvars, rettv, MATCH_LIST);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006304}
6305
6306/*
6307 * "matchstr()" function
6308 */
6309 static void
6310f_matchstr(typval_T *argvars, typval_T *rettv)
6311{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006312 find_some_match(argvars, rettv, MATCH_STR);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006313}
6314
6315/*
6316 * "matchstrpos()" function
6317 */
6318 static void
6319f_matchstrpos(typval_T *argvars, typval_T *rettv)
6320{
Bram Moolenaar8d9f0ef2017-08-27 13:51:01 +02006321 find_some_match(argvars, rettv, MATCH_POS);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006322}
6323
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006324 static void
6325max_min(typval_T *argvars, typval_T *rettv, int domax)
6326{
6327 varnumber_T n = 0;
6328 varnumber_T i;
6329 int error = FALSE;
6330
6331 if (argvars[0].v_type == VAR_LIST)
6332 {
6333 list_T *l;
6334 listitem_T *li;
6335
6336 l = argvars[0].vval.v_list;
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006337 if (l != NULL && l->lv_len > 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006338 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006339 if (l->lv_first == &range_list_item)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006340 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006341 if ((l->lv_u.nonmat.lv_stride > 0) ^ domax)
6342 n = l->lv_u.nonmat.lv_start;
6343 else
6344 n = l->lv_u.nonmat.lv_start + (l->lv_len - 1)
6345 * l->lv_u.nonmat.lv_stride;
6346 }
6347 else
6348 {
6349 li = l->lv_first;
6350 if (li != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006351 {
Bram Moolenaar9f2d0202020-01-30 16:40:10 +01006352 n = tv_get_number_chk(&li->li_tv, &error);
6353 for (;;)
6354 {
6355 li = li->li_next;
6356 if (li == NULL)
6357 break;
6358 i = tv_get_number_chk(&li->li_tv, &error);
6359 if (domax ? i > n : i < n)
6360 n = i;
6361 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006362 }
6363 }
6364 }
6365 }
6366 else if (argvars[0].v_type == VAR_DICT)
6367 {
6368 dict_T *d;
6369 int first = TRUE;
6370 hashitem_T *hi;
6371 int todo;
6372
6373 d = argvars[0].vval.v_dict;
6374 if (d != NULL)
6375 {
6376 todo = (int)d->dv_hashtab.ht_used;
6377 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
6378 {
6379 if (!HASHITEM_EMPTY(hi))
6380 {
6381 --todo;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006382 i = tv_get_number_chk(&HI2DI(hi)->di_tv, &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006383 if (first)
6384 {
6385 n = i;
6386 first = FALSE;
6387 }
6388 else if (domax ? i > n : i < n)
6389 n = i;
6390 }
6391 }
6392 }
6393 }
6394 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006395 semsg(_(e_listdictarg), domax ? "max()" : "min()");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006396 rettv->vval.v_number = error ? 0 : n;
6397}
6398
6399/*
6400 * "max()" function
6401 */
6402 static void
6403f_max(typval_T *argvars, typval_T *rettv)
6404{
6405 max_min(argvars, rettv, TRUE);
6406}
6407
6408/*
6409 * "min()" function
6410 */
6411 static void
6412f_min(typval_T *argvars, typval_T *rettv)
6413{
6414 max_min(argvars, rettv, FALSE);
6415}
6416
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006417#if defined(FEAT_MZSCHEME) || defined(PROTO)
6418/*
6419 * "mzeval()" function
6420 */
6421 static void
6422f_mzeval(typval_T *argvars, typval_T *rettv)
6423{
6424 char_u *str;
6425 char_u buf[NUMBUFLEN];
6426
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006427 if (check_restricted() || check_secure())
6428 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006429 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006430 do_mzeval(str, rettv);
6431}
6432
6433 void
6434mzscheme_call_vim(char_u *name, typval_T *args, typval_T *rettv)
6435{
6436 typval_T argvars[3];
6437
6438 argvars[0].v_type = VAR_STRING;
6439 argvars[0].vval.v_string = name;
6440 copy_tv(args, &argvars[1]);
6441 argvars[2].v_type = VAR_UNKNOWN;
6442 f_call(argvars, rettv);
6443 clear_tv(&argvars[1]);
6444}
6445#endif
6446
6447/*
6448 * "nextnonblank()" function
6449 */
6450 static void
6451f_nextnonblank(typval_T *argvars, typval_T *rettv)
6452{
6453 linenr_T lnum;
6454
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006455 for (lnum = tv_get_lnum(argvars); ; ++lnum)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006456 {
6457 if (lnum < 0 || lnum > curbuf->b_ml.ml_line_count)
6458 {
6459 lnum = 0;
6460 break;
6461 }
6462 if (*skipwhite(ml_get(lnum)) != NUL)
6463 break;
6464 }
6465 rettv->vval.v_number = lnum;
6466}
6467
6468/*
6469 * "nr2char()" function
6470 */
6471 static void
6472f_nr2char(typval_T *argvars, typval_T *rettv)
6473{
6474 char_u buf[NUMBUFLEN];
6475
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006476 if (has_mbyte)
6477 {
6478 int utf8 = 0;
6479
6480 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaared6a4302020-09-05 20:29:41 +02006481 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006482 if (utf8)
Bram Moolenaarbdace832019-03-02 10:13:42 +01006483 buf[utf_char2bytes((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006484 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006485 buf[(*mb_char2bytes)((int)tv_get_number(&argvars[0]), buf)] = NUL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006486 }
6487 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006488 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006489 buf[0] = (char_u)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006490 buf[1] = NUL;
6491 }
6492 rettv->v_type = VAR_STRING;
6493 rettv->vval.v_string = vim_strsave(buf);
6494}
6495
6496/*
6497 * "or(expr, expr)" function
6498 */
6499 static void
6500f_or(typval_T *argvars, typval_T *rettv)
6501{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006502 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
6503 | tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006504}
6505
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006506#ifdef FEAT_PERL
6507/*
6508 * "perleval()" function
6509 */
6510 static void
6511f_perleval(typval_T *argvars, typval_T *rettv)
6512{
6513 char_u *str;
6514 char_u buf[NUMBUFLEN];
6515
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006516 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006517 do_perleval(str, rettv);
6518}
6519#endif
6520
6521#ifdef FEAT_FLOAT
6522/*
6523 * "pow()" function
6524 */
6525 static void
6526f_pow(typval_T *argvars, typval_T *rettv)
6527{
6528 float_T fx = 0.0, fy = 0.0;
6529
6530 rettv->v_type = VAR_FLOAT;
6531 if (get_float_arg(argvars, &fx) == OK
6532 && get_float_arg(&argvars[1], &fy) == OK)
6533 rettv->vval.v_float = pow(fx, fy);
6534 else
6535 rettv->vval.v_float = 0.0;
6536}
6537#endif
6538
6539/*
6540 * "prevnonblank()" function
6541 */
6542 static void
6543f_prevnonblank(typval_T *argvars, typval_T *rettv)
6544{
6545 linenr_T lnum;
6546
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006547 lnum = tv_get_lnum(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006548 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
6549 lnum = 0;
6550 else
6551 while (lnum >= 1 && *skipwhite(ml_get(lnum)) == NUL)
6552 --lnum;
6553 rettv->vval.v_number = lnum;
6554}
6555
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006556// This dummy va_list is here because:
6557// - passing a NULL pointer doesn't work when va_list isn't a pointer
6558// - locally in the function results in a "used before set" warning
6559// - using va_start() to initialize it gives "function with fixed args" error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006560static va_list ap;
6561
6562/*
6563 * "printf()" function
6564 */
6565 static void
6566f_printf(typval_T *argvars, typval_T *rettv)
6567{
6568 char_u buf[NUMBUFLEN];
6569 int len;
6570 char_u *s;
6571 int saved_did_emsg = did_emsg;
6572 char *fmt;
6573
6574 rettv->v_type = VAR_STRING;
6575 rettv->vval.v_string = NULL;
6576
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006577 // Get the required length, allocate the buffer and do it for real.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006578 did_emsg = FALSE;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006579 fmt = (char *)tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006580 len = vim_vsnprintf_typval(NULL, 0, fmt, ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006581 if (!did_emsg)
6582 {
6583 s = alloc(len + 1);
6584 if (s != NULL)
6585 {
6586 rettv->vval.v_string = s;
Bram Moolenaar8327d1d2017-07-11 22:34:51 +02006587 (void)vim_vsnprintf_typval((char *)s, len + 1, fmt,
6588 ap, argvars + 1);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006589 }
6590 }
6591 did_emsg |= saved_did_emsg;
6592}
6593
6594/*
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006595 * "pum_getpos()" function
6596 */
6597 static void
6598f_pum_getpos(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6599{
6600 if (rettv_dict_alloc(rettv) != OK)
6601 return;
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006602 pum_set_event_info(rettv->vval.v_dict);
Bram Moolenaare9bd5722019-08-17 19:36:06 +02006603}
6604
6605/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006606 * "pumvisible()" function
6607 */
6608 static void
6609f_pumvisible(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
6610{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006611 if (pum_visible())
6612 rettv->vval.v_number = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006613}
6614
6615#ifdef FEAT_PYTHON3
6616/*
6617 * "py3eval()" function
6618 */
6619 static void
6620f_py3eval(typval_T *argvars, typval_T *rettv)
6621{
6622 char_u *str;
6623 char_u buf[NUMBUFLEN];
6624
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006625 if (check_restricted() || check_secure())
6626 return;
6627
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006628 if (p_pyx == 0)
6629 p_pyx = 3;
6630
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006631 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006632 do_py3eval(str, rettv);
6633}
6634#endif
6635
6636#ifdef FEAT_PYTHON
6637/*
6638 * "pyeval()" function
6639 */
6640 static void
6641f_pyeval(typval_T *argvars, typval_T *rettv)
6642{
6643 char_u *str;
6644 char_u buf[NUMBUFLEN];
6645
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006646 if (check_restricted() || check_secure())
6647 return;
6648
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006649 if (p_pyx == 0)
6650 p_pyx = 2;
6651
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006652 str = tv_get_string_buf(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006653 do_pyeval(str, rettv);
6654}
6655#endif
6656
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006657#if defined(FEAT_PYTHON) || defined(FEAT_PYTHON3)
6658/*
6659 * "pyxeval()" function
6660 */
6661 static void
6662f_pyxeval(typval_T *argvars, typval_T *rettv)
6663{
Bram Moolenaar8c62a082019-02-08 14:34:10 +01006664 if (check_restricted() || check_secure())
6665 return;
6666
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01006667# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
6668 init_pyxversion();
6669 if (p_pyx == 2)
6670 f_pyeval(argvars, rettv);
6671 else
6672 f_py3eval(argvars, rettv);
6673# elif defined(FEAT_PYTHON)
6674 f_pyeval(argvars, rettv);
6675# elif defined(FEAT_PYTHON3)
6676 f_py3eval(argvars, rettv);
6677# endif
6678}
6679#endif
6680
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006681static UINT32_T srand_seed_for_testing = 0;
6682static int srand_seed_for_testing_is_used = FALSE;
6683
6684 static void
6685f_test_srand_seed(typval_T *argvars, typval_T *rettv UNUSED)
6686{
6687 if (argvars[0].v_type == VAR_UNKNOWN)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006688 srand_seed_for_testing_is_used = FALSE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006689 else
6690 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006691 srand_seed_for_testing = (UINT32_T)tv_get_number(&argvars[0]);
6692 srand_seed_for_testing_is_used = TRUE;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006693 }
6694}
6695
6696 static void
6697init_srand(UINT32_T *x)
6698{
6699#ifndef MSWIN
6700 static int dev_urandom_state = NOTDONE; // FAIL or OK once tried
6701#endif
6702
6703 if (srand_seed_for_testing_is_used)
6704 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02006705 *x = srand_seed_for_testing;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006706 return;
6707 }
6708#ifndef MSWIN
6709 if (dev_urandom_state != FAIL)
6710 {
6711 int fd = open("/dev/urandom", O_RDONLY);
6712 struct {
6713 union {
6714 UINT32_T number;
6715 char bytes[sizeof(UINT32_T)];
6716 } contents;
6717 } buf;
6718
6719 // Attempt reading /dev/urandom.
6720 if (fd == -1)
6721 dev_urandom_state = FAIL;
6722 else
6723 {
6724 buf.contents.number = 0;
6725 if (read(fd, buf.contents.bytes, sizeof(UINT32_T))
6726 != sizeof(UINT32_T))
6727 dev_urandom_state = FAIL;
6728 else
6729 {
6730 dev_urandom_state = OK;
6731 *x = buf.contents.number;
6732 }
6733 close(fd);
6734 }
6735 }
6736 if (dev_urandom_state != OK)
6737 // Reading /dev/urandom doesn't work, fall back to time().
6738#endif
6739 *x = vim_time();
6740}
6741
6742#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
6743#define SPLITMIX32(x, z) ( \
6744 z = (x += 0x9e3779b9), \
6745 z = (z ^ (z >> 16)) * 0x85ebca6b, \
6746 z = (z ^ (z >> 13)) * 0xc2b2ae35, \
6747 z ^ (z >> 16) \
6748 )
6749#define SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w) \
6750 result = ROTL(y * 5, 7) * 9; \
6751 t = y << 9; \
6752 z ^= x; \
6753 w ^= y; \
6754 y ^= z, x ^= w; \
6755 z ^= t; \
6756 w = ROTL(w, 11);
6757
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006758/*
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006759 * "rand()" function
6760 */
6761 static void
6762f_rand(typval_T *argvars, typval_T *rettv)
6763{
6764 list_T *l = NULL;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006765 static UINT32_T gx, gy, gz, gw;
6766 static int initialized = FALSE;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006767 listitem_T *lx, *ly, *lz, *lw;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006768 UINT32_T x, y, z, w, t, result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006769
6770 if (argvars[0].v_type == VAR_UNKNOWN)
6771 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006772 // When no argument is given use the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006773 if (initialized == FALSE)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006774 {
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006775 // Initialize the global seed list.
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006776 init_srand(&x);
6777
6778 gx = SPLITMIX32(x, z);
6779 gy = SPLITMIX32(x, z);
6780 gz = SPLITMIX32(x, z);
6781 gw = SPLITMIX32(x, z);
6782 initialized = TRUE;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006783 }
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006784
6785 SHUFFLE_XOSHIRO128STARSTAR(gx, gy, gz, gw);
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006786 }
6787 else if (argvars[0].v_type == VAR_LIST)
6788 {
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006789 l = argvars[0].vval.v_list;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006790 if (l == NULL || list_len(l) != 4)
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006791 goto theend;
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006792
6793 lx = list_find(l, 0L);
6794 ly = list_find(l, 1L);
6795 lz = list_find(l, 2L);
6796 lw = list_find(l, 3L);
6797 if (lx->li_tv.v_type != VAR_NUMBER) goto theend;
6798 if (ly->li_tv.v_type != VAR_NUMBER) goto theend;
6799 if (lz->li_tv.v_type != VAR_NUMBER) goto theend;
6800 if (lw->li_tv.v_type != VAR_NUMBER) goto theend;
6801 x = (UINT32_T)lx->li_tv.vval.v_number;
6802 y = (UINT32_T)ly->li_tv.vval.v_number;
6803 z = (UINT32_T)lz->li_tv.vval.v_number;
6804 w = (UINT32_T)lw->li_tv.vval.v_number;
6805
6806 SHUFFLE_XOSHIRO128STARSTAR(x, y, z, w);
6807
6808 lx->li_tv.vval.v_number = (varnumber_T)x;
6809 ly->li_tv.vval.v_number = (varnumber_T)y;
6810 lz->li_tv.vval.v_number = (varnumber_T)z;
6811 lw->li_tv.vval.v_number = (varnumber_T)w;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006812 }
6813 else
6814 goto theend;
6815
6816 rettv->v_type = VAR_NUMBER;
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006817 rettv->vval.v_number = (varnumber_T)result;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006818 return;
6819
6820theend:
6821 semsg(_(e_invarg2), tv_get_string(&argvars[0]));
Bram Moolenaarf8c1f922019-11-28 22:13:14 +01006822 rettv->v_type = VAR_NUMBER;
6823 rettv->vval.v_number = -1;
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01006824}
6825
6826/*
Bram Moolenaar4f645c52020-02-08 16:40:39 +01006827 * "srand()" function
6828 */
6829 static void
6830f_srand(typval_T *argvars, typval_T *rettv)
6831{
6832 UINT32_T x = 0, z;
6833
6834 if (rettv_list_alloc(rettv) == FAIL)
6835 return;
6836 if (argvars[0].v_type == VAR_UNKNOWN)
6837 {
6838 init_srand(&x);
6839 }
6840 else
6841 {
6842 int error = FALSE;
6843
6844 x = (UINT32_T)tv_get_number_chk(&argvars[0], &error);
6845 if (error)
6846 return;
6847 }
6848
6849 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6850 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6851 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6852 list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32(x, z));
6853}
6854
6855#undef ROTL
6856#undef SPLITMIX32
6857#undef SHUFFLE_XOSHIRO128STARSTAR
6858
6859/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006860 * "range()" function
6861 */
6862 static void
6863f_range(typval_T *argvars, typval_T *rettv)
6864{
6865 varnumber_T start;
6866 varnumber_T end;
6867 varnumber_T stride = 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006868 int error = FALSE;
6869
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006870 start = tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006871 if (argvars[1].v_type == VAR_UNKNOWN)
6872 {
6873 end = start - 1;
6874 start = 0;
6875 }
6876 else
6877 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006878 end = tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006879 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01006880 stride = tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006881 }
6882
6883 if (error)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01006884 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006885 if (stride == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006886 emsg(_("E726: Stride is zero"));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006887 else if (stride > 0 ? end + 1 < start : end - 1 > start)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006888 emsg(_("E727: Start past end"));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006889 else if (rettv_list_alloc(rettv) == OK)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006890 {
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006891 list_T *list = rettv->vval.v_list;
6892
6893 // Create a non-materialized list. This is much more efficient and
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006894 // works with ":for". If used otherwise CHECK_LIST_MATERIALIZE() must
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006895 // be called.
6896 list->lv_first = &range_list_item;
Bram Moolenaar0ff6aad2020-01-29 21:27:21 +01006897 list->lv_u.nonmat.lv_start = start;
6898 list->lv_u.nonmat.lv_end = end;
6899 list->lv_u.nonmat.lv_stride = stride;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01006900 list->lv_len = (end - start) / stride + 1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006901 }
6902}
6903
6904/*
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006905 * Materialize "list".
6906 * Do not call directly, use CHECK_LIST_MATERIALIZE()
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006907 */
6908 void
6909range_list_materialize(list_T *list)
6910{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006911 varnumber_T start = list->lv_u.nonmat.lv_start;
6912 varnumber_T end = list->lv_u.nonmat.lv_end;
6913 int stride = list->lv_u.nonmat.lv_stride;
6914 varnumber_T i;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01006915
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02006916 list->lv_first = NULL;
6917 list->lv_u.mat.lv_last = NULL;
6918 list->lv_len = 0;
6919 list->lv_u.mat.lv_idx_item = NULL;
6920 for (i = start; stride > 0 ? i <= end : i >= end; i += stride)
6921 if (list_append_number(list, (varnumber_T)i) == FAIL)
6922 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02006923}
6924
Bram Moolenaarbb861e22020-06-07 18:16:36 +02006925/*
6926 * "getreginfo()" function
6927 */
6928 static void
6929f_getreginfo(typval_T *argvars, typval_T *rettv)
6930{
6931 char_u *strregname;
6932 int regname;
6933 char_u buf[NUMBUFLEN + 2];
6934 long reglen = 0;
6935 dict_T *dict;
6936 list_T *list;
6937
6938 if (argvars[0].v_type != VAR_UNKNOWN)
6939 {
6940 strregname = tv_get_string_chk(&argvars[0]);
6941 if (strregname == NULL)
6942 return;
6943 }
6944 else
6945 strregname = get_vim_var_str(VV_REG);
6946
6947 regname = (strregname == NULL ? '"' : *strregname);
6948 if (regname == 0 || regname == '@')
6949 regname = '"';
6950
6951 if (rettv_dict_alloc(rettv) == FAIL)
6952 return;
6953 dict = rettv->vval.v_dict;
6954
6955 list = (list_T *)get_reg_contents(regname, GREG_EXPR_SRC | GREG_LIST);
6956 if (list == NULL)
6957 return;
Bram Moolenaar91639192020-06-29 19:55:58 +02006958 (void)dict_add_list(dict, "regcontents", list);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02006959
6960 buf[0] = NUL;
6961 buf[1] = NUL;
6962 switch (get_reg_type(regname, &reglen))
6963 {
6964 case MLINE: buf[0] = 'V'; break;
6965 case MCHAR: buf[0] = 'v'; break;
6966 case MBLOCK:
6967 vim_snprintf((char *)buf, sizeof(buf), "%c%ld", Ctrl_V,
6968 reglen + 1);
6969 break;
6970 }
Bram Moolenaar91639192020-06-29 19:55:58 +02006971 (void)dict_add_string(dict, (char *)"regtype", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02006972
6973 buf[0] = get_register_name(get_unname_register());
6974 buf[1] = NUL;
6975 if (regname == '"')
Bram Moolenaar91639192020-06-29 19:55:58 +02006976 (void)dict_add_string(dict, (char *)"points_to", buf);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02006977 else
6978 {
6979 dictitem_T *item = dictitem_alloc((char_u *)"isunnamed");
6980
6981 if (item != NULL)
6982 {
6983 item->di_tv.v_type = VAR_SPECIAL;
6984 item->di_tv.vval.v_number = regname == buf[0]
6985 ? VVAL_TRUE : VVAL_FALSE;
Bram Moolenaar91639192020-06-29 19:55:58 +02006986 (void)dict_add(dict, item);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02006987 }
6988 }
6989}
6990
Bram Moolenaar0b6d9112018-05-22 20:35:17 +02006991 static void
6992return_register(int regname, typval_T *rettv)
6993{
6994 char_u buf[2] = {0, 0};
6995
6996 buf[0] = (char_u)regname;
6997 rettv->v_type = VAR_STRING;
6998 rettv->vval.v_string = vim_strsave(buf);
6999}
7000
7001/*
7002 * "reg_executing()" function
7003 */
7004 static void
7005f_reg_executing(typval_T *argvars UNUSED, typval_T *rettv)
7006{
7007 return_register(reg_executing, rettv);
7008}
7009
7010/*
7011 * "reg_recording()" function
7012 */
7013 static void
7014f_reg_recording(typval_T *argvars UNUSED, typval_T *rettv)
7015{
7016 return_register(reg_recording, rettv);
7017}
7018
Bram Moolenaar7416f3e2017-03-18 18:10:13 +01007019/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007020 * "rename({from}, {to})" function
7021 */
7022 static void
7023f_rename(typval_T *argvars, typval_T *rettv)
7024{
7025 char_u buf[NUMBUFLEN];
7026
7027 if (check_restricted() || check_secure())
7028 rettv->vval.v_number = -1;
7029 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007030 rettv->vval.v_number = vim_rename(tv_get_string(&argvars[0]),
7031 tv_get_string_buf(&argvars[1], buf));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007032}
7033
7034/*
7035 * "repeat()" function
7036 */
7037 static void
7038f_repeat(typval_T *argvars, typval_T *rettv)
7039{
7040 char_u *p;
7041 int n;
7042 int slen;
7043 int len;
7044 char_u *r;
7045 int i;
7046
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007047 n = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007048 if (argvars[0].v_type == VAR_LIST)
7049 {
7050 if (rettv_list_alloc(rettv) == OK && argvars[0].vval.v_list != NULL)
7051 while (n-- > 0)
7052 if (list_extend(rettv->vval.v_list,
7053 argvars[0].vval.v_list, NULL) == FAIL)
7054 break;
7055 }
7056 else
7057 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007058 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007059 rettv->v_type = VAR_STRING;
7060 rettv->vval.v_string = NULL;
7061
7062 slen = (int)STRLEN(p);
7063 len = slen * n;
7064 if (len <= 0)
7065 return;
7066
7067 r = alloc(len + 1);
7068 if (r != NULL)
7069 {
7070 for (i = 0; i < n; i++)
7071 mch_memmove(r + i * slen, p, (size_t)slen);
7072 r[len] = NUL;
7073 }
7074
7075 rettv->vval.v_string = r;
7076 }
7077}
7078
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007079#define SP_NOMOVE 0x01 // don't move cursor
7080#define SP_REPEAT 0x02 // repeat to find outer pair
7081#define SP_RETCOUNT 0x04 // return matchcount
7082#define SP_SETPCMARK 0x08 // set previous context mark
7083#define SP_START 0x10 // accept match at start position
7084#define SP_SUBPAT 0x20 // return nr of matching sub-pattern
7085#define SP_END 0x40 // leave cursor at end of match
7086#define SP_COLUMN 0x80 // start at cursor column
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007087
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007088/*
7089 * Get flags for a search function.
7090 * Possibly sets "p_ws".
7091 * Returns BACKWARD, FORWARD or zero (for an error).
7092 */
7093 static int
7094get_search_arg(typval_T *varp, int *flagsp)
7095{
7096 int dir = FORWARD;
7097 char_u *flags;
7098 char_u nbuf[NUMBUFLEN];
7099 int mask;
7100
7101 if (varp->v_type != VAR_UNKNOWN)
7102 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007103 flags = tv_get_string_buf_chk(varp, nbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007104 if (flags == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007105 return 0; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007106 while (*flags != NUL)
7107 {
7108 switch (*flags)
7109 {
7110 case 'b': dir = BACKWARD; break;
7111 case 'w': p_ws = TRUE; break;
7112 case 'W': p_ws = FALSE; break;
7113 default: mask = 0;
7114 if (flagsp != NULL)
7115 switch (*flags)
7116 {
7117 case 'c': mask = SP_START; break;
7118 case 'e': mask = SP_END; break;
7119 case 'm': mask = SP_RETCOUNT; break;
7120 case 'n': mask = SP_NOMOVE; break;
7121 case 'p': mask = SP_SUBPAT; break;
7122 case 'r': mask = SP_REPEAT; break;
7123 case 's': mask = SP_SETPCMARK; break;
7124 case 'z': mask = SP_COLUMN; break;
7125 }
7126 if (mask == 0)
7127 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007128 semsg(_(e_invarg2), flags);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007129 dir = 0;
7130 }
7131 else
7132 *flagsp |= mask;
7133 }
7134 if (dir == 0)
7135 break;
7136 ++flags;
7137 }
7138 }
7139 return dir;
7140}
7141
7142/*
7143 * Shared by search() and searchpos() functions.
7144 */
7145 static int
7146search_cmn(typval_T *argvars, pos_T *match_pos, int *flagsp)
7147{
7148 int flags;
7149 char_u *pat;
7150 pos_T pos;
7151 pos_T save_cursor;
7152 int save_p_ws = p_ws;
7153 int dir;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007154 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007155 long lnum_stop = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007156#ifdef FEAT_RELTIME
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007157 proftime_T tm;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007158 long time_limit = 0;
7159#endif
7160 int options = SEARCH_KEEP;
7161 int subpatnum;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007162 searchit_arg_T sia;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007163 int use_skip = FALSE;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007164 pos_T firstpos;
7165
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007166 pat = tv_get_string(&argvars[0]);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007167 dir = get_search_arg(&argvars[1], flagsp); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007168 if (dir == 0)
7169 goto theend;
7170 flags = *flagsp;
7171 if (flags & SP_START)
7172 options |= SEARCH_START;
7173 if (flags & SP_END)
7174 options |= SEARCH_END;
7175 if (flags & SP_COLUMN)
7176 options |= SEARCH_COL;
7177
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007178 // Optional arguments: line number to stop searching, timeout and skip.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007179 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
7180 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007181 lnum_stop = (long)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007182 if (lnum_stop < 0)
7183 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007184 if (argvars[3].v_type != VAR_UNKNOWN)
7185 {
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007186#ifdef FEAT_RELTIME
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007187 time_limit = (long)tv_get_number_chk(&argvars[3], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007188 if (time_limit < 0)
7189 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007190#endif
Bram Moolenaara9c01042020-06-07 14:50:50 +02007191 use_skip = eval_expr_valid_arg(&argvars[4]);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007192 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007193 }
7194
7195#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007196 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007197 profile_setlimit(time_limit, &tm);
7198#endif
7199
7200 /*
7201 * This function does not accept SP_REPEAT and SP_RETCOUNT flags.
7202 * Check to make sure only those flags are set.
7203 * Also, Only the SP_NOMOVE or the SP_SETPCMARK flag can be set. Both
7204 * flags cannot be set. Check for that condition also.
7205 */
7206 if (((flags & (SP_REPEAT | SP_RETCOUNT)) != 0)
7207 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7208 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007209 semsg(_(e_invarg2), tv_get_string(&argvars[1]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007210 goto theend;
7211 }
7212
7213 pos = save_cursor = curwin->w_cursor;
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007214 CLEAR_FIELD(firstpos);
Bram Moolenaara80faa82020-04-12 19:37:17 +02007215 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007216 sia.sa_stop_lnum = (linenr_T)lnum_stop;
7217#ifdef FEAT_RELTIME
7218 sia.sa_tm = &tm;
7219#endif
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007220
7221 // Repeat until {skip} returns FALSE.
7222 for (;;)
7223 {
7224 subpatnum = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007225 options, RE_SEARCH, &sia);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007226 // finding the first match again means there is no match where {skip}
7227 // evaluates to zero.
7228 if (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos))
7229 subpatnum = FAIL;
7230
Bram Moolenaara9c01042020-06-07 14:50:50 +02007231 if (subpatnum == FAIL || !use_skip)
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007232 // didn't find it or no skip argument
7233 break;
7234 firstpos = pos;
7235
Bram Moolenaara9c01042020-06-07 14:50:50 +02007236 // If the skip expression matches, ignore this match.
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007237 {
7238 int do_skip;
7239 int err;
7240 pos_T save_pos = curwin->w_cursor;
7241
7242 curwin->w_cursor = pos;
Bram Moolenaara9c01042020-06-07 14:50:50 +02007243 err = FALSE;
7244 do_skip = eval_expr_to_bool(&argvars[4], &err);
Bram Moolenaaradc17a52020-06-06 18:37:51 +02007245 curwin->w_cursor = save_pos;
7246 if (err)
7247 {
7248 // Evaluating {skip} caused an error, break here.
7249 subpatnum = FAIL;
7250 break;
7251 }
7252 if (!do_skip)
7253 break;
7254 }
7255 }
7256
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007257 if (subpatnum != FAIL)
7258 {
7259 if (flags & SP_SUBPAT)
7260 retval = subpatnum;
7261 else
7262 retval = pos.lnum;
7263 if (flags & SP_SETPCMARK)
7264 setpcmark();
7265 curwin->w_cursor = pos;
7266 if (match_pos != NULL)
7267 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007268 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007269 match_pos->lnum = pos.lnum;
7270 match_pos->col = pos.col + 1;
7271 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007272 // "/$" will put the cursor after the end of the line, may need to
7273 // correct that here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007274 check_cursor();
7275 }
7276
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007277 // If 'n' flag is used: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007278 if (flags & SP_NOMOVE)
7279 curwin->w_cursor = save_cursor;
7280 else
7281 curwin->w_set_curswant = TRUE;
7282theend:
7283 p_ws = save_p_ws;
7284
7285 return retval;
7286}
7287
7288#ifdef FEAT_FLOAT
7289
7290/*
7291 * round() is not in C90, use ceil() or floor() instead.
7292 */
7293 float_T
7294vim_round(float_T f)
7295{
7296 return f > 0 ? floor(f + 0.5) : ceil(f - 0.5);
7297}
7298
7299/*
7300 * "round({float})" function
7301 */
7302 static void
7303f_round(typval_T *argvars, typval_T *rettv)
7304{
7305 float_T f = 0.0;
7306
7307 rettv->v_type = VAR_FLOAT;
7308 if (get_float_arg(argvars, &f) == OK)
7309 rettv->vval.v_float = vim_round(f);
7310 else
7311 rettv->vval.v_float = 0.0;
7312}
7313#endif
7314
Bram Moolenaare99be0e2019-03-26 22:51:09 +01007315#ifdef FEAT_RUBY
7316/*
7317 * "rubyeval()" function
7318 */
7319 static void
7320f_rubyeval(typval_T *argvars, typval_T *rettv)
7321{
7322 char_u *str;
7323 char_u buf[NUMBUFLEN];
7324
7325 str = tv_get_string_buf(&argvars[0], buf);
7326 do_rubyeval(str, rettv);
7327}
7328#endif
7329
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007330/*
7331 * "screenattr()" function
7332 */
7333 static void
7334f_screenattr(typval_T *argvars, typval_T *rettv)
7335{
7336 int row;
7337 int col;
7338 int c;
7339
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007340 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7341 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007342 if (row < 0 || row >= screen_Rows
7343 || col < 0 || col >= screen_Columns)
7344 c = -1;
7345 else
7346 c = ScreenAttrs[LineOffset[row] + col];
7347 rettv->vval.v_number = c;
7348}
7349
7350/*
7351 * "screenchar()" function
7352 */
7353 static void
7354f_screenchar(typval_T *argvars, typval_T *rettv)
7355{
7356 int row;
7357 int col;
7358 int off;
7359 int c;
7360
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007361 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7362 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007363 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007364 c = -1;
7365 else
7366 {
7367 off = LineOffset[row] + col;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007368 if (enc_utf8 && ScreenLinesUC[off] != 0)
7369 c = ScreenLinesUC[off];
7370 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007371 c = ScreenLines[off];
7372 }
7373 rettv->vval.v_number = c;
7374}
7375
7376/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007377 * "screenchars()" function
7378 */
7379 static void
7380f_screenchars(typval_T *argvars, typval_T *rettv)
7381{
7382 int row;
7383 int col;
7384 int off;
7385 int c;
7386 int i;
7387
7388 if (rettv_list_alloc(rettv) == FAIL)
7389 return;
7390 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7391 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7392 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7393 return;
7394
7395 off = LineOffset[row] + col;
7396 if (enc_utf8 && ScreenLinesUC[off] != 0)
7397 c = ScreenLinesUC[off];
7398 else
7399 c = ScreenLines[off];
7400 list_append_number(rettv->vval.v_list, (varnumber_T)c);
7401
7402 if (enc_utf8)
7403
7404 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7405 list_append_number(rettv->vval.v_list,
7406 (varnumber_T)ScreenLinesC[i][off]);
7407}
7408
7409/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007410 * "screencol()" function
7411 *
7412 * First column is 1 to be consistent with virtcol().
7413 */
7414 static void
7415f_screencol(typval_T *argvars UNUSED, typval_T *rettv)
7416{
7417 rettv->vval.v_number = screen_screencol() + 1;
7418}
7419
7420/*
7421 * "screenrow()" function
7422 */
7423 static void
7424f_screenrow(typval_T *argvars UNUSED, typval_T *rettv)
7425{
7426 rettv->vval.v_number = screen_screenrow() + 1;
7427}
7428
7429/*
Bram Moolenaar2912abb2019-03-29 14:16:42 +01007430 * "screenstring()" function
7431 */
7432 static void
7433f_screenstring(typval_T *argvars, typval_T *rettv)
7434{
7435 int row;
7436 int col;
7437 int off;
7438 int c;
7439 int i;
7440 char_u buf[MB_MAXBYTES + 1];
7441 int buflen = 0;
7442
7443 rettv->vval.v_string = NULL;
7444 rettv->v_type = VAR_STRING;
7445
7446 row = (int)tv_get_number_chk(&argvars[0], NULL) - 1;
7447 col = (int)tv_get_number_chk(&argvars[1], NULL) - 1;
7448 if (row < 0 || row >= screen_Rows || col < 0 || col >= screen_Columns)
7449 return;
7450
7451 off = LineOffset[row] + col;
7452 if (enc_utf8 && ScreenLinesUC[off] != 0)
7453 c = ScreenLinesUC[off];
7454 else
7455 c = ScreenLines[off];
7456 buflen += mb_char2bytes(c, buf);
7457
7458 if (enc_utf8)
7459 for (i = 0; i < Screen_mco && ScreenLinesC[i][off] != 0; ++i)
7460 buflen += mb_char2bytes(ScreenLinesC[i][off], buf + buflen);
7461
7462 buf[buflen] = NUL;
7463 rettv->vval.v_string = vim_strsave(buf);
7464}
7465
7466/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007467 * "search()" function
7468 */
7469 static void
7470f_search(typval_T *argvars, typval_T *rettv)
7471{
7472 int flags = 0;
7473
7474 rettv->vval.v_number = search_cmn(argvars, NULL, &flags);
7475}
7476
7477/*
7478 * "searchdecl()" function
7479 */
7480 static void
7481f_searchdecl(typval_T *argvars, typval_T *rettv)
7482{
Bram Moolenaar30788d32020-09-05 21:35:16 +02007483 int locally = TRUE;
7484 int thisblock = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007485 int error = FALSE;
7486 char_u *name;
7487
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007488 rettv->vval.v_number = 1; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007489
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007490 name = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007491 if (argvars[1].v_type != VAR_UNKNOWN)
7492 {
Bram Moolenaar30788d32020-09-05 21:35:16 +02007493 locally = !(int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007494 if (!error && argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar30788d32020-09-05 21:35:16 +02007495 thisblock = (int)tv_get_bool_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007496 }
7497 if (!error && name != NULL)
7498 rettv->vval.v_number = find_decl(name, (int)STRLEN(name),
7499 locally, thisblock, SEARCH_KEEP) == FAIL;
7500}
7501
7502/*
7503 * Used by searchpair() and searchpairpos()
7504 */
7505 static int
7506searchpair_cmn(typval_T *argvars, pos_T *match_pos)
7507{
7508 char_u *spat, *mpat, *epat;
Bram Moolenaar48570482017-10-30 21:48:41 +01007509 typval_T *skip;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007510 int save_p_ws = p_ws;
7511 int dir;
7512 int flags = 0;
7513 char_u nbuf1[NUMBUFLEN];
7514 char_u nbuf2[NUMBUFLEN];
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007515 int retval = 0; // default: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007516 long lnum_stop = 0;
7517 long time_limit = 0;
7518
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007519 // Get the three pattern arguments: start, middle, end. Will result in an
7520 // error if not a valid argument.
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007521 spat = tv_get_string_chk(&argvars[0]);
7522 mpat = tv_get_string_buf_chk(&argvars[1], nbuf1);
7523 epat = tv_get_string_buf_chk(&argvars[2], nbuf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007524 if (spat == NULL || mpat == NULL || epat == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007525 goto theend; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007526
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007527 // Handle the optional fourth argument: flags
7528 dir = get_search_arg(&argvars[3], &flags); // may set p_ws
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007529 if (dir == 0)
7530 goto theend;
7531
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007532 // Don't accept SP_END or SP_SUBPAT.
7533 // Only one of the SP_NOMOVE or SP_SETPCMARK flags can be set.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007534 if ((flags & (SP_END | SP_SUBPAT)) != 0
7535 || ((flags & SP_NOMOVE) && (flags & SP_SETPCMARK)))
7536 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007537 semsg(_(e_invarg2), tv_get_string(&argvars[3]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007538 goto theend;
7539 }
7540
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007541 // Using 'r' implies 'W', otherwise it doesn't work.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007542 if (flags & SP_REPEAT)
7543 p_ws = FALSE;
7544
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007545 // Optional fifth argument: skip expression
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007546 if (argvars[3].v_type == VAR_UNKNOWN
7547 || argvars[4].v_type == VAR_UNKNOWN)
Bram Moolenaar48570482017-10-30 21:48:41 +01007548 skip = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007549 else
7550 {
Bram Moolenaara9c01042020-06-07 14:50:50 +02007551 // Type is checked later.
Bram Moolenaar48570482017-10-30 21:48:41 +01007552 skip = &argvars[4];
Bram Moolenaara9c01042020-06-07 14:50:50 +02007553
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007554 if (argvars[5].v_type != VAR_UNKNOWN)
7555 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007556 lnum_stop = (long)tv_get_number_chk(&argvars[5], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007557 if (lnum_stop < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007558 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007559 semsg(_(e_invarg2), tv_get_string(&argvars[5]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007560 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007561 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007562#ifdef FEAT_RELTIME
7563 if (argvars[6].v_type != VAR_UNKNOWN)
7564 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007565 time_limit = (long)tv_get_number_chk(&argvars[6], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007566 if (time_limit < 0)
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007567 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007568 semsg(_(e_invarg2), tv_get_string(&argvars[6]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007569 goto theend;
Bram Moolenaar3dddb092018-06-24 19:01:59 +02007570 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007571 }
7572#endif
7573 }
7574 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007575
7576 retval = do_searchpair(spat, mpat, epat, dir, skip, flags,
7577 match_pos, lnum_stop, time_limit);
7578
7579theend:
7580 p_ws = save_p_ws;
7581
7582 return retval;
7583}
7584
7585/*
7586 * "searchpair()" function
7587 */
7588 static void
7589f_searchpair(typval_T *argvars, typval_T *rettv)
7590{
7591 rettv->vval.v_number = searchpair_cmn(argvars, NULL);
7592}
7593
7594/*
7595 * "searchpairpos()" function
7596 */
7597 static void
7598f_searchpairpos(typval_T *argvars, typval_T *rettv)
7599{
7600 pos_T match_pos;
7601 int lnum = 0;
7602 int col = 0;
7603
7604 if (rettv_list_alloc(rettv) == FAIL)
7605 return;
7606
7607 if (searchpair_cmn(argvars, &match_pos) > 0)
7608 {
7609 lnum = match_pos.lnum;
7610 col = match_pos.col;
7611 }
7612
7613 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7614 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7615}
7616
7617/*
7618 * Search for a start/middle/end thing.
7619 * Used by searchpair(), see its documentation for the details.
7620 * Returns 0 or -1 for no match,
7621 */
7622 long
7623do_searchpair(
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007624 char_u *spat, // start pattern
7625 char_u *mpat, // middle pattern
7626 char_u *epat, // end pattern
7627 int dir, // BACKWARD or FORWARD
7628 typval_T *skip, // skip expression
7629 int flags, // SP_SETPCMARK and other SP_ values
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007630 pos_T *match_pos,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007631 linenr_T lnum_stop, // stop at this line if not zero
7632 long time_limit UNUSED) // stop after this many msec
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007633{
7634 char_u *save_cpo;
7635 char_u *pat, *pat2 = NULL, *pat3 = NULL;
7636 long retval = 0;
7637 pos_T pos;
7638 pos_T firstpos;
7639 pos_T foundpos;
7640 pos_T save_cursor;
7641 pos_T save_pos;
7642 int n;
7643 int r;
7644 int nest = 1;
Bram Moolenaar48570482017-10-30 21:48:41 +01007645 int use_skip = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007646 int err;
7647 int options = SEARCH_KEEP;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007648#ifdef FEAT_RELTIME
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007649 proftime_T tm;
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007650#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007651
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007652 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007653 save_cpo = p_cpo;
7654 p_cpo = empty_option;
7655
7656#ifdef FEAT_RELTIME
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007657 // Set the time limit, if there is one.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007658 profile_setlimit(time_limit, &tm);
7659#endif
7660
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007661 // Make two search patterns: start/end (pat2, for in nested pairs) and
7662 // start/middle/end (pat3, for the top pair).
Bram Moolenaar964b3742019-05-24 18:54:09 +02007663 pat2 = alloc(STRLEN(spat) + STRLEN(epat) + 17);
7664 pat3 = alloc(STRLEN(spat) + STRLEN(mpat) + STRLEN(epat) + 25);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007665 if (pat2 == NULL || pat3 == NULL)
7666 goto theend;
Bram Moolenaar6e450a52017-01-06 20:03:58 +01007667 sprintf((char *)pat2, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)", spat, epat);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007668 if (*mpat == NUL)
7669 STRCPY(pat3, pat2);
7670 else
Bram Moolenaar6e450a52017-01-06 20:03:58 +01007671 sprintf((char *)pat3, "\\m\\(%s\\m\\)\\|\\(%s\\m\\)\\|\\(%s\\m\\)",
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007672 spat, epat, mpat);
7673 if (flags & SP_START)
7674 options |= SEARCH_START;
7675
Bram Moolenaar48570482017-10-30 21:48:41 +01007676 if (skip != NULL)
Bram Moolenaara9c01042020-06-07 14:50:50 +02007677 use_skip = eval_expr_valid_arg(skip);
Bram Moolenaar48570482017-10-30 21:48:41 +01007678
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007679 save_cursor = curwin->w_cursor;
7680 pos = curwin->w_cursor;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007681 CLEAR_POS(&firstpos);
7682 CLEAR_POS(&foundpos);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007683 pat = pat3;
7684 for (;;)
7685 {
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007686 searchit_arg_T sia;
7687
Bram Moolenaara80faa82020-04-12 19:37:17 +02007688 CLEAR_FIELD(sia);
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007689 sia.sa_stop_lnum = lnum_stop;
7690#ifdef FEAT_RELTIME
7691 sia.sa_tm = &tm;
7692#endif
Bram Moolenaar5d24a222018-12-23 19:10:09 +01007693 n = searchit(curwin, curbuf, &pos, NULL, dir, pat, 1L,
Bram Moolenaar92ea26b2019-10-18 20:53:34 +02007694 options, RE_SEARCH, &sia);
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007695 if (n == FAIL || (firstpos.lnum != 0 && EQUAL_POS(pos, firstpos)))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007696 // didn't find it or found the first match again: FAIL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007697 break;
7698
7699 if (firstpos.lnum == 0)
7700 firstpos = pos;
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007701 if (EQUAL_POS(pos, foundpos))
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007702 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007703 // Found the same position again. Can happen with a pattern that
7704 // has "\zs" at the end and searching backwards. Advance one
7705 // character and try again.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007706 if (dir == BACKWARD)
7707 decl(&pos);
7708 else
7709 incl(&pos);
7710 }
7711 foundpos = pos;
7712
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007713 // clear the start flag to avoid getting stuck here
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007714 options &= ~SEARCH_START;
7715
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007716 // If the skip pattern matches, ignore this match.
Bram Moolenaar48570482017-10-30 21:48:41 +01007717 if (use_skip)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007718 {
7719 save_pos = curwin->w_cursor;
7720 curwin->w_cursor = pos;
Bram Moolenaar48570482017-10-30 21:48:41 +01007721 err = FALSE;
7722 r = eval_expr_to_bool(skip, &err);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007723 curwin->w_cursor = save_pos;
7724 if (err)
7725 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007726 // Evaluating {skip} caused an error, break here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007727 curwin->w_cursor = save_cursor;
7728 retval = -1;
7729 break;
7730 }
7731 if (r)
7732 continue;
7733 }
7734
7735 if ((dir == BACKWARD && n == 3) || (dir == FORWARD && n == 2))
7736 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007737 // Found end when searching backwards or start when searching
7738 // forward: nested pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007739 ++nest;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007740 pat = pat2; // nested, don't search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007741 }
7742 else
7743 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007744 // Found end when searching forward or start when searching
7745 // backward: end of (nested) pair; or found middle in outer pair.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007746 if (--nest == 1)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007747 pat = pat3; // outer level, search for middle
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007748 }
7749
7750 if (nest == 0)
7751 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007752 // Found the match: return matchcount or line number.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007753 if (flags & SP_RETCOUNT)
7754 ++retval;
7755 else
7756 retval = pos.lnum;
7757 if (flags & SP_SETPCMARK)
7758 setpcmark();
7759 curwin->w_cursor = pos;
7760 if (!(flags & SP_REPEAT))
7761 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007762 nest = 1; // search for next unmatched
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007763 }
7764 }
7765
7766 if (match_pos != NULL)
7767 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007768 // Store the match cursor position
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007769 match_pos->lnum = curwin->w_cursor.lnum;
7770 match_pos->col = curwin->w_cursor.col + 1;
7771 }
7772
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007773 // If 'n' flag is used or search failed: restore cursor position.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007774 if ((flags & SP_NOMOVE) || retval == 0)
7775 curwin->w_cursor = save_cursor;
7776
7777theend:
7778 vim_free(pat2);
7779 vim_free(pat3);
7780 if (p_cpo == empty_option)
7781 p_cpo = save_cpo;
7782 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007783 // Darn, evaluating the {skip} expression changed the value.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007784 free_string_option(save_cpo);
7785
7786 return retval;
7787}
7788
7789/*
7790 * "searchpos()" function
7791 */
7792 static void
7793f_searchpos(typval_T *argvars, typval_T *rettv)
7794{
7795 pos_T match_pos;
7796 int lnum = 0;
7797 int col = 0;
7798 int n;
7799 int flags = 0;
7800
7801 if (rettv_list_alloc(rettv) == FAIL)
7802 return;
7803
7804 n = search_cmn(argvars, &match_pos, &flags);
7805 if (n > 0)
7806 {
7807 lnum = match_pos.lnum;
7808 col = match_pos.col;
7809 }
7810
7811 list_append_number(rettv->vval.v_list, (varnumber_T)lnum);
7812 list_append_number(rettv->vval.v_list, (varnumber_T)col);
7813 if (flags & SP_SUBPAT)
7814 list_append_number(rettv->vval.v_list, (varnumber_T)n);
7815}
7816
7817 static void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007818f_setcharsearch(typval_T *argvars, typval_T *rettv UNUSED)
7819{
7820 dict_T *d;
7821 dictitem_T *di;
7822 char_u *csearch;
7823
7824 if (argvars[0].v_type != VAR_DICT)
7825 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007826 emsg(_(e_dictreq));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007827 return;
7828 }
7829
7830 if ((d = argvars[0].vval.v_dict) != NULL)
7831 {
Bram Moolenaar8f667172018-12-14 15:38:31 +01007832 csearch = dict_get_string(d, (char_u *)"char", FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007833 if (csearch != NULL)
7834 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007835 if (enc_utf8)
7836 {
7837 int pcc[MAX_MCO];
7838 int c = utfc_ptr2char(csearch, pcc);
7839
7840 set_last_csearch(c, csearch, utfc_ptr2len(csearch));
7841 }
7842 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007843 set_last_csearch(PTR2CHAR(csearch),
Bram Moolenaar1614a142019-10-06 22:00:13 +02007844 csearch, mb_ptr2len(csearch));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007845 }
7846
7847 di = dict_find(d, (char_u *)"forward", -1);
7848 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007849 set_csearch_direction((int)tv_get_number(&di->di_tv)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007850 ? FORWARD : BACKWARD);
7851
7852 di = dict_find(d, (char_u *)"until", -1);
7853 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007854 set_csearch_until(!!tv_get_number(&di->di_tv));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007855 }
7856}
7857
7858/*
Bram Moolenaar691ddee2019-05-09 14:52:41 +02007859 * "setenv()" function
7860 */
7861 static void
7862f_setenv(typval_T *argvars, typval_T *rettv UNUSED)
7863{
7864 char_u namebuf[NUMBUFLEN];
7865 char_u valbuf[NUMBUFLEN];
7866 char_u *name = tv_get_string_buf(&argvars[0], namebuf);
7867
7868 if (argvars[1].v_type == VAR_SPECIAL
7869 && argvars[1].vval.v_number == VVAL_NULL)
7870 vim_unsetenv(name);
7871 else
7872 vim_setenv(name, tv_get_string_buf(&argvars[1], valbuf));
7873}
7874
7875/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007876 * "setfperm({fname}, {mode})" function
7877 */
7878 static void
7879f_setfperm(typval_T *argvars, typval_T *rettv)
7880{
7881 char_u *fname;
7882 char_u modebuf[NUMBUFLEN];
7883 char_u *mode_str;
7884 int i;
7885 int mask;
7886 int mode = 0;
7887
7888 rettv->vval.v_number = 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007889 fname = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007890 if (fname == NULL)
7891 return;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007892 mode_str = tv_get_string_buf_chk(&argvars[1], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007893 if (mode_str == NULL)
7894 return;
7895 if (STRLEN(mode_str) != 9)
7896 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007897 semsg(_(e_invarg2), mode_str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007898 return;
7899 }
7900
7901 mask = 1;
7902 for (i = 8; i >= 0; --i)
7903 {
7904 if (mode_str[i] != '-')
7905 mode |= mask;
7906 mask = mask << 1;
7907 }
7908 rettv->vval.v_number = mch_setperm(fname, mode) == OK;
7909}
7910
7911/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007912 * "setpos()" function
7913 */
7914 static void
7915f_setpos(typval_T *argvars, typval_T *rettv)
7916{
7917 pos_T pos;
7918 int fnum;
7919 char_u *name;
7920 colnr_T curswant = -1;
7921
7922 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01007923 name = tv_get_string_chk(argvars);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007924 if (name != NULL)
7925 {
7926 if (list2fpos(&argvars[1], &pos, &fnum, &curswant) == OK)
7927 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01007928 if (pos.col != MAXCOL && --pos.col < 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007929 pos.col = 0;
7930 if (name[0] == '.' && name[1] == NUL)
7931 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007932 // set cursor; "fnum" is ignored
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007933 curwin->w_cursor = pos;
7934 if (curswant >= 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007935 {
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007936 curwin->w_curswant = curswant - 1;
7937 curwin->w_set_curswant = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007938 }
Bram Moolenaar3a29abc2017-01-28 18:31:41 +01007939 check_cursor();
7940 rettv->vval.v_number = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007941 }
7942 else if (name[0] == '\'' && name[1] != NUL && name[2] == NUL)
7943 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01007944 // set mark
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007945 if (setmark_pos(name[1], &pos, fnum) == OK)
7946 rettv->vval.v_number = 0;
7947 }
7948 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007949 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007950 }
7951 }
7952}
7953
7954/*
Bram Moolenaar7633fe52020-06-22 19:10:56 +02007955 * Translate a register type string to the yank type and block length
7956 */
7957 static int
7958get_yank_type(char_u **pp, char_u *yank_type, long *block_len)
7959{
7960 char_u *stropt = *pp;
7961 switch (*stropt)
7962 {
7963 case 'v': case 'c': // character-wise selection
7964 *yank_type = MCHAR;
7965 break;
7966 case 'V': case 'l': // line-wise selection
7967 *yank_type = MLINE;
7968 break;
7969 case 'b': case Ctrl_V: // block-wise selection
7970 *yank_type = MBLOCK;
7971 if (VIM_ISDIGIT(stropt[1]))
7972 {
7973 ++stropt;
7974 *block_len = getdigits(&stropt) - 1;
7975 --stropt;
7976 }
7977 break;
7978 default:
7979 return FAIL;
7980 }
7981 *pp = stropt;
7982 return OK;
7983}
7984
7985/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02007986 * "setreg()" function
7987 */
7988 static void
7989f_setreg(typval_T *argvars, typval_T *rettv)
7990{
7991 int regname;
7992 char_u *strregname;
7993 char_u *stropt;
7994 char_u *strval;
7995 int append;
7996 char_u yank_type;
7997 long block_len;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02007998 typval_T *regcontents;
7999 int pointreg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008000
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008001 pointreg = 0;
8002 regcontents = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008003 block_len = -1;
8004 yank_type = MAUTO;
8005 append = FALSE;
8006
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008007 strregname = tv_get_string_chk(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008008 rettv->vval.v_number = 1; // FAIL is default
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008009
8010 if (strregname == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008011 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008012 regname = *strregname;
8013 if (regname == 0 || regname == '@')
8014 regname = '"';
8015
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008016 if (argvars[1].v_type == VAR_DICT)
8017 {
8018 dict_T *d = argvars[1].vval.v_dict;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008019 dictitem_T *di;
8020
8021 if (d == NULL || d->dv_hashtab.ht_used == 0)
8022 {
8023 // Empty dict, clear the register (like setreg(0, []))
8024 char_u *lstval[2] = {NULL, NULL};
8025 write_reg_contents_lst(regname, lstval, 0, FALSE, MAUTO, -1);
8026 return;
8027 }
8028
8029 di = dict_find(d, (char_u *)"regcontents", -1);
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008030 if (di != NULL)
8031 regcontents = &di->di_tv;
8032
8033 stropt = dict_get_string(d, (char_u *)"regtype", FALSE);
8034 if (stropt != NULL)
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008035 {
8036 int ret = get_yank_type(&stropt, &yank_type, &block_len);
8037
8038 if (ret == FAIL || *++stropt != NUL)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008039 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008040 semsg(_(e_invargval), "value");
8041 return;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008042 }
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008043 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008044
8045 if (regname == '"')
8046 {
8047 stropt = dict_get_string(d, (char_u *)"points_to", FALSE);
8048 if (stropt != NULL)
8049 {
8050 pointreg = *stropt;
8051 regname = pointreg;
8052 }
8053 }
Bram Moolenaar6a950582020-08-28 16:39:33 +02008054 else if (dict_get_bool(d, (char_u *)"isunnamed", -1) > 0)
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008055 pointreg = regname;
8056 }
8057 else
8058 regcontents = &argvars[1];
8059
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008060 if (argvars[2].v_type != VAR_UNKNOWN)
8061 {
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008062 if (yank_type != MAUTO)
8063 {
8064 semsg(_(e_toomanyarg), "setreg");
8065 return;
8066 }
8067
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008068 stropt = tv_get_string_chk(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008069 if (stropt == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008070 return; // type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008071 for (; *stropt != NUL; ++stropt)
8072 switch (*stropt)
8073 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008074 case 'a': case 'A': // append
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008075 append = TRUE;
8076 break;
Bram Moolenaar7633fe52020-06-22 19:10:56 +02008077 default:
8078 get_yank_type(&stropt, &yank_type, &block_len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008079 }
8080 }
8081
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008082 if (regcontents && regcontents->v_type == VAR_LIST)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008083 {
8084 char_u **lstval;
8085 char_u **allocval;
8086 char_u buf[NUMBUFLEN];
8087 char_u **curval;
8088 char_u **curallocval;
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008089 list_T *ll = regcontents->vval.v_list;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008090 listitem_T *li;
8091 int len;
8092
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008093 // If the list is NULL handle like an empty list.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008094 len = ll == NULL ? 0 : ll->lv_len;
8095
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008096 // First half: use for pointers to result lines; second half: use for
8097 // pointers to allocated copies.
Bram Moolenaarc799fe22019-05-28 23:08:19 +02008098 lstval = ALLOC_MULT(char_u *, (len + 1) * 2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008099 if (lstval == NULL)
8100 return;
8101 curval = lstval;
8102 allocval = lstval + len + 2;
8103 curallocval = allocval;
8104
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008105 if (ll != NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008106 {
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02008107 CHECK_LIST_MATERIALIZE(ll);
Bram Moolenaar00d253e2020-04-06 22:13:01 +02008108 FOR_ALL_LIST_ITEMS(ll, li)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008109 {
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008110 strval = tv_get_string_buf_chk(&li->li_tv, buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008111 if (strval == NULL)
8112 goto free_lstval;
Bram Moolenaar50985eb2020-01-27 22:09:39 +01008113 if (strval == buf)
8114 {
8115 // Need to make a copy, next tv_get_string_buf_chk() will
8116 // overwrite the string.
8117 strval = vim_strsave(buf);
8118 if (strval == NULL)
8119 goto free_lstval;
8120 *curallocval++ = strval;
8121 }
8122 *curval++ = strval;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008123 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008124 }
8125 *curval++ = NULL;
8126
8127 write_reg_contents_lst(regname, lstval, -1,
8128 append, yank_type, block_len);
8129free_lstval:
8130 while (curallocval > allocval)
8131 vim_free(*--curallocval);
8132 vim_free(lstval);
8133 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008134 else if (regcontents)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008135 {
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008136 strval = tv_get_string_chk(regcontents);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008137 if (strval == NULL)
8138 return;
8139 write_reg_contents_ex(regname, strval, -1,
8140 append, yank_type, block_len);
8141 }
Bram Moolenaarbb861e22020-06-07 18:16:36 +02008142 if (pointreg != 0)
8143 get_yank_register(pointreg, TRUE);
8144
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008145 rettv->vval.v_number = 0;
8146}
8147
8148/*
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008149 * "settagstack()" function
8150 */
8151 static void
8152f_settagstack(typval_T *argvars, typval_T *rettv)
8153{
8154 static char *e_invact2 = N_("E962: Invalid action: '%s'");
8155 win_T *wp;
8156 dict_T *d;
8157 int action = 'r';
8158
8159 rettv->vval.v_number = -1;
8160
8161 // first argument: window number or id
8162 wp = find_win_by_nr_or_id(&argvars[0]);
8163 if (wp == NULL)
8164 return;
8165
8166 // second argument: dict with items to set in the tag stack
8167 if (argvars[1].v_type != VAR_DICT)
8168 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008169 emsg(_(e_dictreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008170 return;
8171 }
8172 d = argvars[1].vval.v_dict;
8173 if (d == NULL)
8174 return;
8175
8176 // third argument: action - 'a' for append and 'r' for replace.
8177 // default is to replace the stack.
8178 if (argvars[2].v_type == VAR_UNKNOWN)
8179 action = 'r';
8180 else if (argvars[2].v_type == VAR_STRING)
8181 {
8182 char_u *actstr;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008183 actstr = tv_get_string_chk(&argvars[2]);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008184 if (actstr == NULL)
8185 return;
Bram Moolenaar271fa082020-01-02 14:02:16 +01008186 if ((*actstr == 'r' || *actstr == 'a' || *actstr == 't')
8187 && actstr[1] == NUL)
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008188 action = *actstr;
8189 else
8190 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008191 semsg(_(e_invact2), actstr);
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008192 return;
8193 }
8194 }
8195 else
8196 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008197 emsg(_(e_stringreq));
Bram Moolenaarf49cc602018-11-11 15:21:05 +01008198 return;
8199 }
8200
8201 if (set_tagstack(wp, d, action) == OK)
8202 rettv->vval.v_number = 0;
8203}
8204
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008205#ifdef FEAT_CRYPT
8206/*
8207 * "sha256({string})" function
8208 */
8209 static void
8210f_sha256(typval_T *argvars, typval_T *rettv)
8211{
8212 char_u *p;
8213
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008214 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008215 rettv->vval.v_string = vim_strsave(
8216 sha256_bytes(p, (int)STRLEN(p), NULL, 0));
8217 rettv->v_type = VAR_STRING;
8218}
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008219#endif // FEAT_CRYPT
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008220
8221/*
8222 * "shellescape({string})" function
8223 */
8224 static void
8225f_shellescape(typval_T *argvars, typval_T *rettv)
8226{
Bram Moolenaar20615522017-06-05 18:46:26 +02008227 int do_special = non_zero_arg(&argvars[1]);
8228
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008229 rettv->vval.v_string = vim_strsave_shellescape(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008230 tv_get_string(&argvars[0]), do_special, do_special);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008231 rettv->v_type = VAR_STRING;
8232}
8233
8234/*
8235 * shiftwidth() function
8236 */
8237 static void
8238f_shiftwidth(typval_T *argvars UNUSED, typval_T *rettv)
8239{
Bram Moolenaarf9514162018-11-22 03:08:29 +01008240 rettv->vval.v_number = 0;
8241
8242 if (argvars[0].v_type != VAR_UNKNOWN)
8243 {
8244 long col;
8245
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008246 col = (long)tv_get_number_chk(argvars, NULL);
Bram Moolenaarf9514162018-11-22 03:08:29 +01008247 if (col < 0)
8248 return; // type error; errmsg already given
8249#ifdef FEAT_VARTABS
8250 rettv->vval.v_number = get_sw_value_col(curbuf, col);
8251 return;
8252#endif
8253 }
8254
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008255 rettv->vval.v_number = get_sw_value(curbuf);
8256}
8257
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008258#ifdef FEAT_FLOAT
8259/*
8260 * "sin()" function
8261 */
8262 static void
8263f_sin(typval_T *argvars, typval_T *rettv)
8264{
8265 float_T f = 0.0;
8266
8267 rettv->v_type = VAR_FLOAT;
8268 if (get_float_arg(argvars, &f) == OK)
8269 rettv->vval.v_float = sin(f);
8270 else
8271 rettv->vval.v_float = 0.0;
8272}
8273
8274/*
8275 * "sinh()" function
8276 */
8277 static void
8278f_sinh(typval_T *argvars, typval_T *rettv)
8279{
8280 float_T f = 0.0;
8281
8282 rettv->v_type = VAR_FLOAT;
8283 if (get_float_arg(argvars, &f) == OK)
8284 rettv->vval.v_float = sinh(f);
8285 else
8286 rettv->vval.v_float = 0.0;
8287}
8288#endif
8289
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008290/*
8291 * "soundfold({word})" function
8292 */
8293 static void
8294f_soundfold(typval_T *argvars, typval_T *rettv)
8295{
8296 char_u *s;
8297
8298 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008299 s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008300#ifdef FEAT_SPELL
8301 rettv->vval.v_string = eval_soundfold(s);
8302#else
8303 rettv->vval.v_string = vim_strsave(s);
8304#endif
8305}
8306
8307/*
8308 * "spellbadword()" function
8309 */
8310 static void
8311f_spellbadword(typval_T *argvars UNUSED, typval_T *rettv)
8312{
8313 char_u *word = (char_u *)"";
8314 hlf_T attr = HLF_COUNT;
8315 int len = 0;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008316#ifdef FEAT_SPELL
8317 int wo_spell_save = curwin->w_p_spell;
8318
8319 if (!curwin->w_p_spell)
8320 {
8321 did_set_spelllang(curwin);
8322 curwin->w_p_spell = TRUE;
8323 }
8324
8325 if (*curwin->w_s->b_p_spl == NUL)
8326 {
8327 emsg(_(e_no_spell));
8328 curwin->w_p_spell = wo_spell_save;
8329 return;
8330 }
8331#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008332
8333 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008334 {
8335#ifdef FEAT_SPELL
8336 curwin->w_p_spell = wo_spell_save;
8337#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008338 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008339 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008340
8341#ifdef FEAT_SPELL
8342 if (argvars[0].v_type == VAR_UNKNOWN)
8343 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008344 // Find the start and length of the badly spelled word.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008345 len = spell_move_to(curwin, FORWARD, TRUE, TRUE, &attr);
8346 if (len != 0)
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008347 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008348 word = ml_get_cursor();
Bram Moolenaarb73fa622017-12-21 20:27:47 +01008349 curwin->w_set_curswant = TRUE;
8350 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008351 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008352 else if (*curbuf->b_s.b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008353 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008354 char_u *str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008355 int capcol = -1;
8356
8357 if (str != NULL)
8358 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008359 // Check the argument for spelling.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008360 while (*str != NUL)
8361 {
8362 len = spell_check(curwin, str, &attr, &capcol, FALSE);
8363 if (attr != HLF_COUNT)
8364 {
8365 word = str;
8366 break;
8367 }
8368 str += len;
Bram Moolenaar66ab9162018-07-20 20:28:48 +02008369 capcol -= len;
Bram Moolenaar0c779e82019-08-09 17:01:02 +02008370 len = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008371 }
8372 }
8373 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008374 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008375#endif
8376
8377 list_append_string(rettv->vval.v_list, word, len);
8378 list_append_string(rettv->vval.v_list, (char_u *)(
8379 attr == HLF_SPB ? "bad" :
8380 attr == HLF_SPR ? "rare" :
8381 attr == HLF_SPL ? "local" :
8382 attr == HLF_SPC ? "caps" :
8383 ""), -1);
8384}
8385
8386/*
8387 * "spellsuggest()" function
8388 */
8389 static void
8390f_spellsuggest(typval_T *argvars UNUSED, typval_T *rettv)
8391{
8392#ifdef FEAT_SPELL
8393 char_u *str;
8394 int typeerr = FALSE;
8395 int maxcount;
8396 garray_T ga;
8397 int i;
8398 listitem_T *li;
8399 int need_capital = FALSE;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008400 int wo_spell_save = curwin->w_p_spell;
8401
8402 if (!curwin->w_p_spell)
8403 {
8404 did_set_spelllang(curwin);
8405 curwin->w_p_spell = TRUE;
8406 }
8407
8408 if (*curwin->w_s->b_p_spl == NUL)
8409 {
8410 emsg(_(e_no_spell));
8411 curwin->w_p_spell = wo_spell_save;
8412 return;
8413 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008414#endif
8415
8416 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008417 {
8418#ifdef FEAT_SPELL
8419 curwin->w_p_spell = wo_spell_save;
8420#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008421 return;
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008422 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008423
8424#ifdef FEAT_SPELL
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008425 if (*curwin->w_s->b_p_spl != NUL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008426 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008427 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008428 if (argvars[1].v_type != VAR_UNKNOWN)
8429 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008430 maxcount = (int)tv_get_number_chk(&argvars[1], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008431 if (maxcount <= 0)
8432 return;
8433 if (argvars[2].v_type != VAR_UNKNOWN)
8434 {
Bram Moolenaar7c27f332020-09-05 22:45:55 +02008435 need_capital = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008436 if (typeerr)
8437 return;
8438 }
8439 }
8440 else
8441 maxcount = 25;
8442
8443 spell_suggest_list(&ga, str, maxcount, need_capital, FALSE);
8444
8445 for (i = 0; i < ga.ga_len; ++i)
8446 {
8447 str = ((char_u **)ga.ga_data)[i];
8448
8449 li = listitem_alloc();
8450 if (li == NULL)
8451 vim_free(str);
8452 else
8453 {
8454 li->li_tv.v_type = VAR_STRING;
8455 li->li_tv.v_lock = 0;
8456 li->li_tv.vval.v_string = str;
8457 list_append(rettv->vval.v_list, li);
8458 }
8459 }
8460 ga_clear(&ga);
8461 }
Bram Moolenaar152e79e2020-06-10 15:32:08 +02008462 curwin->w_p_spell = wo_spell_save;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008463#endif
8464}
8465
8466 static void
8467f_split(typval_T *argvars, typval_T *rettv)
8468{
8469 char_u *str;
8470 char_u *end;
8471 char_u *pat = NULL;
8472 regmatch_T regmatch;
8473 char_u patbuf[NUMBUFLEN];
8474 char_u *save_cpo;
8475 int match;
8476 colnr_T col = 0;
8477 int keepempty = FALSE;
8478 int typeerr = FALSE;
8479
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008480 // Make 'cpoptions' empty, the 'l' flag should not be used here.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008481 save_cpo = p_cpo;
8482 p_cpo = (char_u *)"";
8483
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008484 str = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008485 if (argvars[1].v_type != VAR_UNKNOWN)
8486 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008487 pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008488 if (pat == NULL)
8489 typeerr = TRUE;
8490 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02008491 keepempty = (int)tv_get_bool_chk(&argvars[2], &typeerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008492 }
8493 if (pat == NULL || *pat == NUL)
8494 pat = (char_u *)"[\\x01- ]\\+";
8495
8496 if (rettv_list_alloc(rettv) == FAIL)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008497 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008498 if (typeerr)
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008499 goto theend;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008500
8501 regmatch.regprog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
8502 if (regmatch.regprog != NULL)
8503 {
8504 regmatch.rm_ic = FALSE;
8505 while (*str != NUL || keepempty)
8506 {
8507 if (*str == NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008508 match = FALSE; // empty item at the end
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008509 else
8510 match = vim_regexec_nl(&regmatch, str, col);
8511 if (match)
8512 end = regmatch.startp[0];
8513 else
8514 end = str + STRLEN(str);
8515 if (keepempty || end > str || (rettv->vval.v_list->lv_len > 0
8516 && *str != NUL && match && end < regmatch.endp[0]))
8517 {
8518 if (list_append_string(rettv->vval.v_list, str,
8519 (int)(end - str)) == FAIL)
8520 break;
8521 }
8522 if (!match)
8523 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01008524 // Advance to just after the match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008525 if (regmatch.endp[0] > str)
8526 col = 0;
8527 else
Bram Moolenaar13505972019-01-24 15:04:48 +01008528 // Don't get stuck at the same match.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008529 col = (*mb_ptr2len)(regmatch.endp[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008530 str = regmatch.endp[0];
8531 }
8532
8533 vim_regfree(regmatch.regprog);
8534 }
8535
Bram Moolenaar7d5e7442020-07-21 22:25:51 +02008536theend:
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008537 p_cpo = save_cpo;
8538}
8539
8540#ifdef FEAT_FLOAT
8541/*
8542 * "sqrt()" function
8543 */
8544 static void
8545f_sqrt(typval_T *argvars, typval_T *rettv)
8546{
8547 float_T f = 0.0;
8548
8549 rettv->v_type = VAR_FLOAT;
8550 if (get_float_arg(argvars, &f) == OK)
8551 rettv->vval.v_float = sqrt(f);
8552 else
8553 rettv->vval.v_float = 0.0;
8554}
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008555#endif
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008556
Bram Moolenaar0387cae2019-11-29 21:07:58 +01008557#ifdef FEAT_FLOAT
Bram Moolenaar06b0b4b2019-11-25 15:40:55 +01008558/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008559 * "str2float()" function
8560 */
8561 static void
8562f_str2float(typval_T *argvars, typval_T *rettv)
8563{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008564 char_u *p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008565 int isneg = (*p == '-');
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008566
Bram Moolenaar08243d22017-01-10 16:12:29 +01008567 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008568 p = skipwhite(p + 1);
8569 (void)string2float(p, &rettv->vval.v_float);
Bram Moolenaar08243d22017-01-10 16:12:29 +01008570 if (isneg)
8571 rettv->vval.v_float *= -1;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008572 rettv->v_type = VAR_FLOAT;
8573}
8574#endif
8575
8576/*
Bram Moolenaar9d401282019-04-06 13:18:12 +02008577 * "str2list()" function
8578 */
8579 static void
8580f_str2list(typval_T *argvars, typval_T *rettv)
8581{
8582 char_u *p;
8583 int utf8 = FALSE;
8584
8585 if (rettv_list_alloc(rettv) == FAIL)
8586 return;
8587
8588 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaara48f7862020-09-05 20:16:57 +02008589 utf8 = (int)tv_get_bool_chk(&argvars[1], NULL);
Bram Moolenaar9d401282019-04-06 13:18:12 +02008590
8591 p = tv_get_string(&argvars[0]);
8592
8593 if (has_mbyte || utf8)
8594 {
8595 int (*ptr2len)(char_u *);
8596 int (*ptr2char)(char_u *);
8597
8598 if (utf8 || enc_utf8)
8599 {
8600 ptr2len = utf_ptr2len;
8601 ptr2char = utf_ptr2char;
8602 }
8603 else
8604 {
8605 ptr2len = mb_ptr2len;
8606 ptr2char = mb_ptr2char;
8607 }
8608
8609 for ( ; *p != NUL; p += (*ptr2len)(p))
8610 list_append_number(rettv->vval.v_list, (*ptr2char)(p));
8611 }
8612 else
8613 for ( ; *p != NUL; ++p)
8614 list_append_number(rettv->vval.v_list, *p);
8615}
8616
8617/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008618 * "str2nr()" function
8619 */
8620 static void
8621f_str2nr(typval_T *argvars, typval_T *rettv)
8622{
8623 int base = 10;
8624 char_u *p;
8625 varnumber_T n;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008626 int what = 0;
Bram Moolenaar08243d22017-01-10 16:12:29 +01008627 int isneg;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008628
8629 if (argvars[1].v_type != VAR_UNKNOWN)
8630 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008631 base = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008632 if (base != 2 && base != 8 && base != 10 && base != 16)
8633 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01008634 emsg(_(e_invarg));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008635 return;
8636 }
Bram Moolenaar3986b942020-09-06 16:09:04 +02008637 if (argvars[2].v_type != VAR_UNKNOWN && tv_get_bool(&argvars[2]))
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008638 what |= STR2NR_QUOTE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008639 }
8640
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008641 p = skipwhite(tv_get_string(&argvars[0]));
Bram Moolenaar08243d22017-01-10 16:12:29 +01008642 isneg = (*p == '-');
8643 if (*p == '+' || *p == '-')
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008644 p = skipwhite(p + 1);
8645 switch (base)
8646 {
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008647 case 2: what |= STR2NR_BIN + STR2NR_FORCE; break;
Bram Moolenaarc17e66c2020-06-02 21:38:22 +02008648 case 8: what |= STR2NR_OCT + STR2NR_OOCT + STR2NR_FORCE; break;
Bram Moolenaar60a8de22019-09-15 14:33:22 +02008649 case 16: what |= STR2NR_HEX + STR2NR_FORCE; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008650 }
Bram Moolenaar16e9b852019-05-19 19:59:35 +02008651 vim_str2nr(p, NULL, NULL, what, &n, NULL, 0, FALSE);
8652 // Text after the number is silently ignored.
Bram Moolenaar08243d22017-01-10 16:12:29 +01008653 if (isneg)
8654 rettv->vval.v_number = -n;
8655 else
8656 rettv->vval.v_number = n;
8657
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008658}
8659
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008660/*
8661 * "strgetchar()" function
8662 */
8663 static void
8664f_strgetchar(typval_T *argvars, typval_T *rettv)
8665{
8666 char_u *str;
8667 int len;
8668 int error = FALSE;
8669 int charidx;
Bram Moolenaar13505972019-01-24 15:04:48 +01008670 int byteidx = 0;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008671
8672 rettv->vval.v_number = -1;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008673 str = tv_get_string_chk(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008674 if (str == NULL)
8675 return;
8676 len = (int)STRLEN(str);
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008677 charidx = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008678 if (error)
8679 return;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008680
Bram Moolenaar13505972019-01-24 15:04:48 +01008681 while (charidx >= 0 && byteidx < len)
8682 {
8683 if (charidx == 0)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008684 {
Bram Moolenaar13505972019-01-24 15:04:48 +01008685 rettv->vval.v_number = mb_ptr2char(str + byteidx);
8686 break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008687 }
Bram Moolenaar13505972019-01-24 15:04:48 +01008688 --charidx;
8689 byteidx += MB_CPTR2LEN(str + byteidx);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008690 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008691}
8692
8693/*
8694 * "stridx()" function
8695 */
8696 static void
8697f_stridx(typval_T *argvars, typval_T *rettv)
8698{
8699 char_u buf[NUMBUFLEN];
8700 char_u *needle;
8701 char_u *haystack;
8702 char_u *save_haystack;
8703 char_u *pos;
8704 int start_idx;
8705
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008706 needle = tv_get_string_chk(&argvars[1]);
8707 save_haystack = haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008708 rettv->vval.v_number = -1;
8709 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008710 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008711
8712 if (argvars[2].v_type != VAR_UNKNOWN)
8713 {
8714 int error = FALSE;
8715
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008716 start_idx = (int)tv_get_number_chk(&argvars[2], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008717 if (error || start_idx >= (int)STRLEN(haystack))
8718 return;
8719 if (start_idx >= 0)
8720 haystack += start_idx;
8721 }
8722
8723 pos = (char_u *)strstr((char *)haystack, (char *)needle);
8724 if (pos != NULL)
8725 rettv->vval.v_number = (varnumber_T)(pos - save_haystack);
8726}
8727
8728/*
8729 * "string()" function
8730 */
Bram Moolenaar461a7fc2018-12-22 13:28:07 +01008731 void
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008732f_string(typval_T *argvars, typval_T *rettv)
8733{
8734 char_u *tofree;
8735 char_u numbuf[NUMBUFLEN];
8736
8737 rettv->v_type = VAR_STRING;
8738 rettv->vval.v_string = tv2string(&argvars[0], &tofree, numbuf,
8739 get_copyID());
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008740 // Make a copy if we have a value but it's not in allocated memory.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008741 if (rettv->vval.v_string != NULL && tofree == NULL)
8742 rettv->vval.v_string = vim_strsave(rettv->vval.v_string);
8743}
8744
8745/*
8746 * "strlen()" function
8747 */
8748 static void
8749f_strlen(typval_T *argvars, typval_T *rettv)
8750{
8751 rettv->vval.v_number = (varnumber_T)(STRLEN(
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008752 tv_get_string(&argvars[0])));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008753}
8754
8755/*
8756 * "strchars()" function
8757 */
8758 static void
8759f_strchars(typval_T *argvars, typval_T *rettv)
8760{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008761 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar3986b942020-09-06 16:09:04 +02008762 int skipcc = FALSE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008763 varnumber_T len = 0;
8764 int (*func_mb_ptr2char_adv)(char_u **pp);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008765
8766 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaar3986b942020-09-06 16:09:04 +02008767 skipcc = (int)tv_get_bool(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008768 if (skipcc < 0 || skipcc > 1)
Bram Moolenaarbade44e2020-09-26 22:39:24 +02008769 semsg(_(e_using_number_as_bool_nr), skipcc);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008770 else
8771 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008772 func_mb_ptr2char_adv = skipcc ? mb_ptr2char_adv : mb_cptr2char_adv;
8773 while (*s != NUL)
8774 {
8775 func_mb_ptr2char_adv(&s);
8776 ++len;
8777 }
8778 rettv->vval.v_number = len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008779 }
8780}
8781
8782/*
8783 * "strdisplaywidth()" function
8784 */
8785 static void
8786f_strdisplaywidth(typval_T *argvars, typval_T *rettv)
8787{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008788 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008789 int col = 0;
8790
8791 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008792 col = (int)tv_get_number(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008793
8794 rettv->vval.v_number = (varnumber_T)(linetabsize_col(col, s) - col);
8795}
8796
8797/*
8798 * "strwidth()" function
8799 */
8800 static void
8801f_strwidth(typval_T *argvars, typval_T *rettv)
8802{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008803 char_u *s = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008804
Bram Moolenaar13505972019-01-24 15:04:48 +01008805 rettv->vval.v_number = (varnumber_T)(mb_string2cells(s, -1));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008806}
8807
8808/*
8809 * "strcharpart()" function
8810 */
8811 static void
8812f_strcharpart(typval_T *argvars, typval_T *rettv)
8813{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008814 char_u *p;
8815 int nchar;
8816 int nbyte = 0;
8817 int charlen;
8818 int len = 0;
8819 int slen;
8820 int error = FALSE;
8821
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008822 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008823 slen = (int)STRLEN(p);
8824
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008825 nchar = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008826 if (!error)
8827 {
8828 if (nchar > 0)
8829 while (nchar > 0 && nbyte < slen)
8830 {
Bram Moolenaard3c907b2016-08-17 21:32:09 +02008831 nbyte += MB_CPTR2LEN(p + nbyte);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008832 --nchar;
8833 }
8834 else
8835 nbyte = nchar;
8836 if (argvars[2].v_type != VAR_UNKNOWN)
8837 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008838 charlen = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008839 while (charlen > 0 && nbyte + len < slen)
8840 {
8841 int off = nbyte + len;
8842
8843 if (off < 0)
8844 len += 1;
8845 else
Bram Moolenaard3c907b2016-08-17 21:32:09 +02008846 len += MB_CPTR2LEN(p + off);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008847 --charlen;
8848 }
8849 }
8850 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008851 len = slen - nbyte; // default: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008852 }
8853
8854 /*
8855 * Only return the overlap between the specified part and the actual
8856 * string.
8857 */
8858 if (nbyte < 0)
8859 {
8860 len += nbyte;
8861 nbyte = 0;
8862 }
8863 else if (nbyte > slen)
8864 nbyte = slen;
8865 if (len < 0)
8866 len = 0;
8867 else if (nbyte + len > slen)
8868 len = slen - nbyte;
8869
8870 rettv->v_type = VAR_STRING;
8871 rettv->vval.v_string = vim_strnsave(p + nbyte, len);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008872}
8873
8874/*
8875 * "strpart()" function
8876 */
8877 static void
8878f_strpart(typval_T *argvars, typval_T *rettv)
8879{
8880 char_u *p;
8881 int n;
8882 int len;
8883 int slen;
8884 int error = FALSE;
8885
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008886 p = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008887 slen = (int)STRLEN(p);
8888
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008889 n = (int)tv_get_number_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008890 if (error)
8891 len = 0;
8892 else if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008893 len = (int)tv_get_number(&argvars[2]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008894 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008895 len = slen - n; // default len: all bytes that are available.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008896
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02008897 // Only return the overlap between the specified part and the actual
8898 // string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008899 if (n < 0)
8900 {
8901 len += n;
8902 n = 0;
8903 }
8904 else if (n > slen)
8905 n = slen;
8906 if (len < 0)
8907 len = 0;
8908 else if (n + len > slen)
8909 len = slen - n;
8910
Bram Moolenaar6c53fca2020-08-23 17:34:46 +02008911 if (argvars[2].v_type != VAR_UNKNOWN && argvars[3].v_type != VAR_UNKNOWN)
8912 {
8913 int off;
8914
8915 // length in characters
8916 for (off = n; off < slen && len > 0; --len)
8917 off += mb_ptr2len(p + off);
8918 len = off - n;
8919 }
8920
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008921 rettv->v_type = VAR_STRING;
8922 rettv->vval.v_string = vim_strnsave(p + n, len);
8923}
8924
8925/*
8926 * "strridx()" function
8927 */
8928 static void
8929f_strridx(typval_T *argvars, typval_T *rettv)
8930{
8931 char_u buf[NUMBUFLEN];
8932 char_u *needle;
8933 char_u *haystack;
8934 char_u *rest;
8935 char_u *lastmatch = NULL;
8936 int haystack_len, end_idx;
8937
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008938 needle = tv_get_string_chk(&argvars[1]);
8939 haystack = tv_get_string_buf_chk(&argvars[0], buf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008940
8941 rettv->vval.v_number = -1;
8942 if (needle == NULL || haystack == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008943 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008944
8945 haystack_len = (int)STRLEN(haystack);
8946 if (argvars[2].v_type != VAR_UNKNOWN)
8947 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008948 // Third argument: upper limit for index
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008949 end_idx = (int)tv_get_number_chk(&argvars[2], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008950 if (end_idx < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008951 return; // can never find a match
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008952 }
8953 else
8954 end_idx = haystack_len;
8955
8956 if (*needle == NUL)
8957 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01008958 // Empty string matches past the end.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008959 lastmatch = haystack + end_idx;
8960 }
8961 else
8962 {
8963 for (rest = haystack; *rest != '\0'; ++rest)
8964 {
8965 rest = (char_u *)strstr((char *)rest, (char *)needle);
8966 if (rest == NULL || rest > haystack + end_idx)
8967 break;
8968 lastmatch = rest;
8969 }
8970 }
8971
8972 if (lastmatch == NULL)
8973 rettv->vval.v_number = -1;
8974 else
8975 rettv->vval.v_number = (varnumber_T)(lastmatch - haystack);
8976}
8977
8978/*
8979 * "strtrans()" function
8980 */
8981 static void
8982f_strtrans(typval_T *argvars, typval_T *rettv)
8983{
8984 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008985 rettv->vval.v_string = transstr(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008986}
8987
8988/*
8989 * "submatch()" function
8990 */
8991 static void
8992f_submatch(typval_T *argvars, typval_T *rettv)
8993{
8994 int error = FALSE;
8995 int no;
8996 int retList = 0;
8997
Bram Moolenaard155d7a2018-12-21 16:04:21 +01008998 no = (int)tv_get_number_chk(&argvars[0], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02008999 if (error)
9000 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009001 if (no < 0 || no >= NSUBEXP)
9002 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009003 semsg(_("E935: invalid submatch number: %d"), no);
Bram Moolenaar79518e22017-02-17 16:31:35 +01009004 return;
Bram Moolenaar989f5922016-08-21 15:26:54 +02009005 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009006 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaarad304702020-09-06 18:22:53 +02009007 retList = (int)tv_get_bool_chk(&argvars[1], &error);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009008 if (error)
9009 return;
9010
9011 if (retList == 0)
9012 {
9013 rettv->v_type = VAR_STRING;
9014 rettv->vval.v_string = reg_submatch(no);
9015 }
9016 else
9017 {
9018 rettv->v_type = VAR_LIST;
9019 rettv->vval.v_list = reg_submatch_list(no);
9020 }
9021}
9022
9023/*
9024 * "substitute()" function
9025 */
9026 static void
9027f_substitute(typval_T *argvars, typval_T *rettv)
9028{
9029 char_u patbuf[NUMBUFLEN];
9030 char_u subbuf[NUMBUFLEN];
9031 char_u flagsbuf[NUMBUFLEN];
9032
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009033 char_u *str = tv_get_string_chk(&argvars[0]);
9034 char_u *pat = tv_get_string_buf_chk(&argvars[1], patbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009035 char_u *sub = NULL;
9036 typval_T *expr = NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009037 char_u *flg = tv_get_string_buf_chk(&argvars[3], flagsbuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009038
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009039 if (argvars[2].v_type == VAR_FUNC || argvars[2].v_type == VAR_PARTIAL)
9040 expr = &argvars[2];
9041 else
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009042 sub = tv_get_string_buf_chk(&argvars[2], subbuf);
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009043
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009044 rettv->v_type = VAR_STRING;
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009045 if (str == NULL || pat == NULL || (sub == NULL && expr == NULL)
9046 || flg == NULL)
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009047 rettv->vval.v_string = NULL;
9048 else
Bram Moolenaar72ab7292016-07-19 19:10:51 +02009049 rettv->vval.v_string = do_string_sub(str, pat, sub, expr, flg);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009050}
9051
9052/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009053 * "swapinfo(swap_filename)" function
9054 */
9055 static void
9056f_swapinfo(typval_T *argvars, typval_T *rettv)
9057{
9058 if (rettv_dict_alloc(rettv) == OK)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009059 get_b0_dict(tv_get_string(argvars), rettv->vval.v_dict);
Bram Moolenaar00f123a2018-08-21 20:28:54 +02009060}
9061
9062/*
Bram Moolenaar110bd602018-09-16 18:46:59 +02009063 * "swapname(expr)" function
9064 */
9065 static void
9066f_swapname(typval_T *argvars, typval_T *rettv)
9067{
9068 buf_T *buf;
9069
9070 rettv->v_type = VAR_STRING;
Bram Moolenaarf2d79fa2019-01-03 22:19:27 +01009071 buf = tv_get_buf(&argvars[0], FALSE);
Bram Moolenaar110bd602018-09-16 18:46:59 +02009072 if (buf == NULL || buf->b_ml.ml_mfp == NULL
9073 || buf->b_ml.ml_mfp->mf_fname == NULL)
9074 rettv->vval.v_string = NULL;
9075 else
9076 rettv->vval.v_string = vim_strsave(buf->b_ml.ml_mfp->mf_fname);
9077}
9078
9079/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009080 * "synID(lnum, col, trans)" function
9081 */
9082 static void
9083f_synID(typval_T *argvars UNUSED, typval_T *rettv)
9084{
9085 int id = 0;
9086#ifdef FEAT_SYN_HL
9087 linenr_T lnum;
9088 colnr_T col;
9089 int trans;
9090 int transerr = FALSE;
9091
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009092 lnum = tv_get_lnum(argvars); // -1 on type error
9093 col = (linenr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaarfcb6d702020-09-05 21:41:56 +02009094 trans = (int)tv_get_bool_chk(&argvars[2], &transerr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009095
9096 if (!transerr && lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9097 && col >= 0 && col < (long)STRLEN(ml_get(lnum)))
9098 id = syn_get_id(curwin, lnum, (colnr_T)col, trans, NULL, FALSE);
9099#endif
9100
9101 rettv->vval.v_number = id;
9102}
9103
9104/*
9105 * "synIDattr(id, what [, mode])" function
9106 */
9107 static void
9108f_synIDattr(typval_T *argvars UNUSED, typval_T *rettv)
9109{
9110 char_u *p = NULL;
9111#ifdef FEAT_SYN_HL
9112 int id;
9113 char_u *what;
9114 char_u *mode;
9115 char_u modebuf[NUMBUFLEN];
9116 int modec;
9117
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009118 id = (int)tv_get_number(&argvars[0]);
9119 what = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009120 if (argvars[2].v_type != VAR_UNKNOWN)
9121 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009122 mode = tv_get_string_buf(&argvars[2], modebuf);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009123 modec = TOLOWER_ASC(mode[0]);
9124 if (modec != 't' && modec != 'c' && modec != 'g')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009125 modec = 0; // replace invalid with current
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009126 }
9127 else
9128 {
9129#if defined(FEAT_GUI) || defined(FEAT_TERMGUICOLORS)
9130 if (USE_24BIT)
9131 modec = 'g';
9132 else
9133#endif
9134 if (t_colors > 1)
9135 modec = 'c';
9136 else
9137 modec = 't';
9138 }
9139
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009140 switch (TOLOWER_ASC(what[0]))
9141 {
9142 case 'b':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009143 if (TOLOWER_ASC(what[1]) == 'g') // bg[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009144 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009145 else // bold
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009146 p = highlight_has_attr(id, HL_BOLD, modec);
9147 break;
9148
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009149 case 'f': // fg[#] or font
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009150 p = highlight_color(id, what, modec);
9151 break;
9152
9153 case 'i':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009154 if (TOLOWER_ASC(what[1]) == 'n') // inverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009155 p = highlight_has_attr(id, HL_INVERSE, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009156 else // italic
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009157 p = highlight_has_attr(id, HL_ITALIC, modec);
9158 break;
9159
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009160 case 'n': // name
Bram Moolenaarc96272e2017-03-26 13:50:09 +02009161 p = get_highlight_name_ext(NULL, id - 1, FALSE);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009162 break;
9163
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009164 case 'r': // reverse
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009165 p = highlight_has_attr(id, HL_INVERSE, modec);
9166 break;
9167
9168 case 's':
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009169 if (TOLOWER_ASC(what[1]) == 'p') // sp[#]
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009170 p = highlight_color(id, what, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009171 // strikeout
Bram Moolenaarcf4b00c2017-09-02 18:33:56 +02009172 else if (TOLOWER_ASC(what[1]) == 't' &&
9173 TOLOWER_ASC(what[2]) == 'r')
9174 p = highlight_has_attr(id, HL_STRIKETHROUGH, modec);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009175 else // standout
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009176 p = highlight_has_attr(id, HL_STANDOUT, modec);
9177 break;
9178
9179 case 'u':
Bram Moolenaar391c3622020-09-29 20:59:17 +02009180 if (TOLOWER_ASC(what[1]) == 'l') // ul
9181 p = highlight_color(id, what, modec);
9182 else if (STRLEN(what) <= 5 || TOLOWER_ASC(what[5]) != 'c')
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009183 // underline
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009184 p = highlight_has_attr(id, HL_UNDERLINE, modec);
9185 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009186 // undercurl
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009187 p = highlight_has_attr(id, HL_UNDERCURL, modec);
9188 break;
9189 }
9190
9191 if (p != NULL)
9192 p = vim_strsave(p);
9193#endif
9194 rettv->v_type = VAR_STRING;
9195 rettv->vval.v_string = p;
9196}
9197
9198/*
9199 * "synIDtrans(id)" function
9200 */
9201 static void
9202f_synIDtrans(typval_T *argvars UNUSED, typval_T *rettv)
9203{
9204 int id;
9205
9206#ifdef FEAT_SYN_HL
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009207 id = (int)tv_get_number(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009208
9209 if (id > 0)
9210 id = syn_get_final_id(id);
9211 else
9212#endif
9213 id = 0;
9214
9215 rettv->vval.v_number = id;
9216}
9217
9218/*
9219 * "synconcealed(lnum, col)" function
9220 */
9221 static void
9222f_synconcealed(typval_T *argvars UNUSED, typval_T *rettv)
9223{
9224#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
9225 linenr_T lnum;
9226 colnr_T col;
9227 int syntax_flags = 0;
9228 int cchar;
9229 int matchid = 0;
9230 char_u str[NUMBUFLEN];
9231#endif
9232
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009233 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009234
9235#if defined(FEAT_SYN_HL) && defined(FEAT_CONCEAL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009236 lnum = tv_get_lnum(argvars); // -1 on type error
9237 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009238
Bram Moolenaara80faa82020-04-12 19:37:17 +02009239 CLEAR_FIELD(str);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009240
9241 if (rettv_list_alloc(rettv) != FAIL)
9242 {
9243 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9244 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9245 && curwin->w_p_cole > 0)
9246 {
9247 (void)syn_get_id(curwin, lnum, col, FALSE, NULL, FALSE);
9248 syntax_flags = get_syntax_info(&matchid);
9249
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009250 // get the conceal character
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009251 if ((syntax_flags & HL_CONCEAL) && curwin->w_p_cole < 3)
9252 {
9253 cchar = syn_get_sub_char();
Bram Moolenaar4d785892017-06-22 22:00:50 +02009254 if (cchar == NUL && curwin->w_p_cole == 1)
9255 cchar = (lcs_conceal == NUL) ? ' ' : lcs_conceal;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009256 if (cchar != NUL)
9257 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009258 if (has_mbyte)
9259 (*mb_char2bytes)(cchar, str);
9260 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009261 str[0] = cchar;
9262 }
9263 }
9264 }
9265
9266 list_append_number(rettv->vval.v_list,
9267 (syntax_flags & HL_CONCEAL) != 0);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009268 // -1 to auto-determine strlen
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009269 list_append_string(rettv->vval.v_list, str, -1);
9270 list_append_number(rettv->vval.v_list, matchid);
9271 }
9272#endif
9273}
9274
9275/*
9276 * "synstack(lnum, col)" function
9277 */
9278 static void
9279f_synstack(typval_T *argvars UNUSED, typval_T *rettv)
9280{
9281#ifdef FEAT_SYN_HL
9282 linenr_T lnum;
9283 colnr_T col;
9284 int i;
9285 int id;
9286#endif
9287
Bram Moolenaar45cf6e92017-04-30 20:25:19 +02009288 rettv_list_set(rettv, NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009289
9290#ifdef FEAT_SYN_HL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009291 lnum = tv_get_lnum(argvars); // -1 on type error
9292 col = (colnr_T)tv_get_number(&argvars[1]) - 1; // -1 on type error
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009293
9294 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count
9295 && col >= 0 && col <= (long)STRLEN(ml_get(lnum))
9296 && rettv_list_alloc(rettv) != FAIL)
9297 {
9298 (void)syn_get_id(curwin, lnum, (colnr_T)col, FALSE, NULL, TRUE);
9299 for (i = 0; ; ++i)
9300 {
9301 id = syn_get_stack_item(i);
9302 if (id < 0)
9303 break;
9304 if (list_append_number(rettv->vval.v_list, id) == FAIL)
9305 break;
9306 }
9307 }
9308#endif
9309}
9310
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009311/*
9312 * "tabpagebuflist()" function
9313 */
9314 static void
9315f_tabpagebuflist(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9316{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009317 tabpage_T *tp;
9318 win_T *wp = NULL;
9319
9320 if (argvars[0].v_type == VAR_UNKNOWN)
9321 wp = firstwin;
9322 else
9323 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009324 tp = find_tabpage((int)tv_get_number(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009325 if (tp != NULL)
9326 wp = (tp == curtab) ? firstwin : tp->tp_firstwin;
9327 }
9328 if (wp != NULL && rettv_list_alloc(rettv) != FAIL)
9329 {
9330 for (; wp != NULL; wp = wp->w_next)
9331 if (list_append_number(rettv->vval.v_list,
9332 wp->w_buffer->b_fnum) == FAIL)
9333 break;
9334 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009335}
9336
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009337/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009338 * "tagfiles()" function
9339 */
9340 static void
9341f_tagfiles(typval_T *argvars UNUSED, typval_T *rettv)
9342{
9343 char_u *fname;
9344 tagname_T tn;
9345 int first;
9346
9347 if (rettv_list_alloc(rettv) == FAIL)
9348 return;
9349 fname = alloc(MAXPATHL);
9350 if (fname == NULL)
9351 return;
9352
9353 for (first = TRUE; ; first = FALSE)
9354 if (get_tagfname(&tn, first, fname) == FAIL
9355 || list_append_string(rettv->vval.v_list, fname, -1) == FAIL)
9356 break;
9357 tagname_free(&tn);
9358 vim_free(fname);
9359}
9360
9361/*
9362 * "taglist()" function
9363 */
9364 static void
9365f_taglist(typval_T *argvars, typval_T *rettv)
9366{
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009367 char_u *fname = NULL;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009368 char_u *tag_pattern;
9369
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009370 tag_pattern = tv_get_string(&argvars[0]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009371
9372 rettv->vval.v_number = FALSE;
9373 if (*tag_pattern == NUL)
9374 return;
9375
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009376 if (argvars[1].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009377 fname = tv_get_string(&argvars[1]);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009378 if (rettv_list_alloc(rettv) == OK)
Bram Moolenaarc6aafba2017-03-21 17:09:10 +01009379 (void)get_tags(rettv->vval.v_list, tag_pattern, fname);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009380}
9381
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009382#ifdef FEAT_FLOAT
9383/*
9384 * "tan()" function
9385 */
9386 static void
9387f_tan(typval_T *argvars, typval_T *rettv)
9388{
9389 float_T f = 0.0;
9390
9391 rettv->v_type = VAR_FLOAT;
9392 if (get_float_arg(argvars, &f) == OK)
9393 rettv->vval.v_float = tan(f);
9394 else
9395 rettv->vval.v_float = 0.0;
9396}
9397
9398/*
9399 * "tanh()" function
9400 */
9401 static void
9402f_tanh(typval_T *argvars, typval_T *rettv)
9403{
9404 float_T f = 0.0;
9405
9406 rettv->v_type = VAR_FLOAT;
9407 if (get_float_arg(argvars, &f) == OK)
9408 rettv->vval.v_float = tanh(f);
9409 else
9410 rettv->vval.v_float = 0.0;
9411}
9412#endif
9413
9414/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009415 * "tolower(string)" function
9416 */
9417 static void
9418f_tolower(typval_T *argvars, typval_T *rettv)
9419{
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009420 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009421 rettv->vval.v_string = strlow_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009422}
9423
9424/*
9425 * "toupper(string)" function
9426 */
9427 static void
9428f_toupper(typval_T *argvars, typval_T *rettv)
9429{
9430 rettv->v_type = VAR_STRING;
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009431 rettv->vval.v_string = strup_save(tv_get_string(&argvars[0]));
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009432}
9433
9434/*
9435 * "tr(string, fromstr, tostr)" function
9436 */
9437 static void
9438f_tr(typval_T *argvars, typval_T *rettv)
9439{
9440 char_u *in_str;
9441 char_u *fromstr;
9442 char_u *tostr;
9443 char_u *p;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009444 int inlen;
9445 int fromlen;
9446 int tolen;
9447 int idx;
9448 char_u *cpstr;
9449 int cplen;
9450 int first = TRUE;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009451 char_u buf[NUMBUFLEN];
9452 char_u buf2[NUMBUFLEN];
9453 garray_T ga;
9454
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009455 in_str = tv_get_string(&argvars[0]);
9456 fromstr = tv_get_string_buf_chk(&argvars[1], buf);
9457 tostr = tv_get_string_buf_chk(&argvars[2], buf2);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009458
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009459 // Default return value: empty string.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009460 rettv->v_type = VAR_STRING;
9461 rettv->vval.v_string = NULL;
9462 if (fromstr == NULL || tostr == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009463 return; // type error; errmsg already given
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009464 ga_init2(&ga, (int)sizeof(char), 80);
9465
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009466 if (!has_mbyte)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009467 // not multi-byte: fromstr and tostr must be the same length
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009468 if (STRLEN(fromstr) != STRLEN(tostr))
9469 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009470error:
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01009471 semsg(_(e_invarg2), fromstr);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009472 ga_clear(&ga);
9473 return;
9474 }
9475
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009476 // fromstr and tostr have to contain the same number of chars
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009477 while (*in_str != NUL)
9478 {
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009479 if (has_mbyte)
9480 {
9481 inlen = (*mb_ptr2len)(in_str);
9482 cpstr = in_str;
9483 cplen = inlen;
9484 idx = 0;
9485 for (p = fromstr; *p != NUL; p += fromlen)
9486 {
9487 fromlen = (*mb_ptr2len)(p);
9488 if (fromlen == inlen && STRNCMP(in_str, p, inlen) == 0)
9489 {
9490 for (p = tostr; *p != NUL; p += tolen)
9491 {
9492 tolen = (*mb_ptr2len)(p);
9493 if (idx-- == 0)
9494 {
9495 cplen = tolen;
9496 cpstr = p;
9497 break;
9498 }
9499 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009500 if (*p == NUL) // tostr is shorter than fromstr
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009501 goto error;
9502 break;
9503 }
9504 ++idx;
9505 }
9506
9507 if (first && cpstr == in_str)
9508 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009509 // Check that fromstr and tostr have the same number of
9510 // (multi-byte) characters. Done only once when a character
9511 // of in_str doesn't appear in fromstr.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009512 first = FALSE;
9513 for (p = tostr; *p != NUL; p += tolen)
9514 {
9515 tolen = (*mb_ptr2len)(p);
9516 --idx;
9517 }
9518 if (idx != 0)
9519 goto error;
9520 }
9521
9522 (void)ga_grow(&ga, cplen);
9523 mch_memmove((char *)ga.ga_data + ga.ga_len, cpstr, (size_t)cplen);
9524 ga.ga_len += cplen;
9525
9526 in_str += inlen;
9527 }
9528 else
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009529 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009530 // When not using multi-byte chars we can do it faster.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009531 p = vim_strchr(fromstr, *in_str);
9532 if (p != NULL)
9533 ga_append(&ga, tostr[p - fromstr]);
9534 else
9535 ga_append(&ga, *in_str);
9536 ++in_str;
9537 }
9538 }
9539
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009540 // add a terminating NUL
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009541 (void)ga_grow(&ga, 1);
9542 ga_append(&ga, NUL);
9543
9544 rettv->vval.v_string = ga.ga_data;
9545}
9546
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009547/*
9548 * "trim({expr})" function
9549 */
9550 static void
9551f_trim(typval_T *argvars, typval_T *rettv)
9552{
9553 char_u buf1[NUMBUFLEN];
9554 char_u buf2[NUMBUFLEN];
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009555 char_u *head = tv_get_string_buf_chk(&argvars[0], buf1);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009556 char_u *mask = NULL;
9557 char_u *tail;
9558 char_u *prev;
9559 char_u *p;
9560 int c1;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009561 int dir = 0;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009562
9563 rettv->v_type = VAR_STRING;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009564 rettv->vval.v_string = NULL;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009565 if (head == NULL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009566 return;
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009567
9568 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009569 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009570 mask = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009571
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009572 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009573 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009574 int error = 0;
9575
9576 // leading or trailing characters to trim
9577 dir = (int)tv_get_number_chk(&argvars[2], &error);
9578 if (error)
9579 return;
9580 if (dir < 0 || dir > 2)
9581 {
9582 semsg(_(e_invarg2), tv_get_string(&argvars[2]));
9583 return;
9584 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009585 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009586 }
9587
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009588 if (dir == 0 || dir == 1)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009589 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009590 // Trim leading characters
9591 while (*head != NUL)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009592 {
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009593 c1 = PTR2CHAR(head);
9594 if (mask == NULL)
9595 {
9596 if (c1 > ' ' && c1 != 0xa0)
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009597 break;
Bram Moolenaar2245ae12020-05-31 22:20:36 +02009598 }
9599 else
9600 {
9601 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9602 if (c1 == PTR2CHAR(p))
9603 break;
9604 if (*p == NUL)
9605 break;
9606 }
9607 MB_PTR_ADV(head);
9608 }
9609 }
9610
9611 tail = head + STRLEN(head);
9612 if (dir == 0 || dir == 2)
9613 {
9614 // Trim trailing characters
9615 for (; tail > head; tail = prev)
9616 {
9617 prev = tail;
9618 MB_PTR_BACK(head, prev);
9619 c1 = PTR2CHAR(prev);
9620 if (mask == NULL)
9621 {
9622 if (c1 > ' ' && c1 != 0xa0)
9623 break;
9624 }
9625 else
9626 {
9627 for (p = mask; *p != NUL; MB_PTR_ADV(p))
9628 if (c1 == PTR2CHAR(p))
9629 break;
9630 if (*p == NUL)
9631 break;
9632 }
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009633 }
9634 }
Bram Moolenaardf44a272020-06-07 20:49:05 +02009635 rettv->vval.v_string = vim_strnsave(head, tail - head);
Bram Moolenaar295ac5a2018-03-22 23:04:02 +01009636}
9637
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009638#ifdef FEAT_FLOAT
9639/*
9640 * "trunc({float})" function
9641 */
9642 static void
9643f_trunc(typval_T *argvars, typval_T *rettv)
9644{
9645 float_T f = 0.0;
9646
9647 rettv->v_type = VAR_FLOAT;
9648 if (get_float_arg(argvars, &f) == OK)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009649 // trunc() is not in C90, use floor() or ceil() instead.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009650 rettv->vval.v_float = f > 0 ? floor(f) : ceil(f);
9651 else
9652 rettv->vval.v_float = 0.0;
9653}
9654#endif
9655
9656/*
9657 * "type(expr)" function
9658 */
9659 static void
9660f_type(typval_T *argvars, typval_T *rettv)
9661{
9662 int n = -1;
9663
9664 switch (argvars[0].v_type)
9665 {
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01009666 case VAR_NUMBER: n = VAR_TYPE_NUMBER; break;
9667 case VAR_STRING: n = VAR_TYPE_STRING; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009668 case VAR_PARTIAL:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01009669 case VAR_FUNC: n = VAR_TYPE_FUNC; break;
9670 case VAR_LIST: n = VAR_TYPE_LIST; break;
9671 case VAR_DICT: n = VAR_TYPE_DICT; break;
9672 case VAR_FLOAT: n = VAR_TYPE_FLOAT; break;
9673 case VAR_BOOL: n = VAR_TYPE_BOOL; break;
9674 case VAR_SPECIAL: n = VAR_TYPE_NONE; break;
Bram Moolenaarf562e722016-07-19 17:25:25 +02009675 case VAR_JOB: n = VAR_TYPE_JOB; break;
9676 case VAR_CHANNEL: n = VAR_TYPE_CHANNEL; break;
Bram Moolenaar6e5ea8d2019-01-12 22:47:31 +01009677 case VAR_BLOB: n = VAR_TYPE_BLOB; break;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009678 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02009679 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01009680 case VAR_VOID:
Bram Moolenaardd589232020-02-29 17:38:12 +01009681 internal_error_no_abort("f_type(UNKNOWN)");
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009682 n = -1;
9683 break;
9684 }
9685 rettv->vval.v_number = n;
9686}
9687
9688/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009689 * "virtcol(string)" function
9690 */
9691 static void
9692f_virtcol(typval_T *argvars, typval_T *rettv)
9693{
9694 colnr_T vcol = 0;
9695 pos_T *fp;
9696 int fnum = curbuf->b_fnum;
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01009697 int len;
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009698
9699 fp = var2fpos(&argvars[0], FALSE, &fnum);
9700 if (fp != NULL && fp->lnum <= curbuf->b_ml.ml_line_count
9701 && fnum == curbuf->b_fnum)
9702 {
Bram Moolenaarb3d33d82020-01-15 20:36:55 +01009703 // Limit the column to a valid value, getvvcol() doesn't check.
9704 if (fp->col < 0)
9705 fp->col = 0;
9706 else
9707 {
9708 len = (int)STRLEN(ml_get(fp->lnum));
9709 if (fp->col > len)
9710 fp->col = len;
9711 }
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009712 getvvcol(curwin, fp, NULL, NULL, &vcol);
9713 ++vcol;
9714 }
9715
9716 rettv->vval.v_number = vcol;
9717}
9718
9719/*
9720 * "visualmode()" function
9721 */
9722 static void
9723f_visualmode(typval_T *argvars, typval_T *rettv)
9724{
9725 char_u str[2];
9726
9727 rettv->v_type = VAR_STRING;
9728 str[0] = curbuf->b_visual_mode_eval;
9729 str[1] = NUL;
9730 rettv->vval.v_string = vim_strsave(str);
9731
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009732 // A non-zero number or non-empty string argument: reset mode.
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009733 if (non_zero_arg(&argvars[0]))
9734 curbuf->b_visual_mode_eval = NUL;
9735}
9736
9737/*
9738 * "wildmenumode()" function
9739 */
9740 static void
9741f_wildmenumode(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9742{
9743#ifdef FEAT_WILDMENU
9744 if (wild_menu_showing)
9745 rettv->vval.v_number = 1;
9746#endif
9747}
9748
9749/*
Bram Moolenaar0c1e3742019-12-27 13:49:24 +01009750 * "windowsversion()" function
9751 */
9752 static void
9753f_windowsversion(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
9754{
9755 rettv->v_type = VAR_STRING;
9756 rettv->vval.v_string = vim_strsave((char_u *)windowsVersion);
9757}
9758
9759/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009760 * "wordcount()" function
9761 */
9762 static void
9763f_wordcount(typval_T *argvars UNUSED, typval_T *rettv)
9764{
9765 if (rettv_dict_alloc(rettv) == FAIL)
9766 return;
9767 cursor_pos_info(rettv->vval.v_dict);
9768}
9769
9770/*
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009771 * "xor(expr, expr)" function
9772 */
9773 static void
9774f_xor(typval_T *argvars, typval_T *rettv)
9775{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01009776 rettv->vval.v_number = tv_get_number_chk(&argvars[0], NULL)
9777 ^ tv_get_number_chk(&argvars[1], NULL);
Bram Moolenaar73dad1e2016-07-17 22:13:49 +02009778}
9779
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01009780#endif // FEAT_EVAL