blob: 041837830550846865aa228e07e92acf72efc58c [file] [log] [blame]
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
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/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +000011 * vim9compile.c: compiling a :def function
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010012 */
13
14#define USING_FLOAT_STUFF
15#include "vim.h"
16
17#if defined(FEAT_EVAL) || defined(PROTO)
18
Bram Moolenaardc7c3662021-12-20 15:04:29 +000019// When not generating protos this is included in proto.h
20#ifdef PROTO
21# include "vim9.h"
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010022#endif
23
Bram Moolenaardc7c3662021-12-20 15:04:29 +000024// Functions defined with :def are stored in this growarray.
25// They are never removed, so that they can be found by index.
26// Deleted functions have the df_deleted flag set.
27garray_T def_functions = {0, 0, sizeof(dfunc_T), 50, NULL};
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010028
Bram Moolenaarcdc40c42020-12-26 17:43:08 +010029static void delete_def_function_contents(dfunc_T *dfunc, int mark_deleted);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010030
31/*
Bram Moolenaar709664c2020-12-12 14:33:41 +010032 * Lookup variable "name" in the local scope and return it in "lvar".
Bram Moolenaarab360522021-01-10 14:02:28 +010033 * "lvar->lv_from_outer" is incremented accordingly.
Bram Moolenaar709664c2020-12-12 14:33:41 +010034 * If "lvar" is NULL only check if the variable can be found.
35 * Return FAIL if not found.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010036 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +000037 int
Bram Moolenaar709664c2020-12-12 14:33:41 +010038lookup_local(char_u *name, size_t len, lvar_T *lvar, cctx_T *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010039{
40 int idx;
Bram Moolenaar709664c2020-12-12 14:33:41 +010041 lvar_T *lvp;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010042
Bram Moolenaarae8d2de2020-02-13 21:42:24 +010043 if (len == 0)
Bram Moolenaar709664c2020-12-12 14:33:41 +010044 return FAIL;
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +020045
46 // Find local in current function scope.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010047 for (idx = 0; idx < cctx->ctx_locals.ga_len; ++idx)
48 {
Bram Moolenaar709664c2020-12-12 14:33:41 +010049 lvp = ((lvar_T *)cctx->ctx_locals.ga_data) + idx;
50 if (STRNCMP(name, lvp->lv_name, len) == 0
51 && STRLEN(lvp->lv_name) == len)
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +020052 {
Bram Moolenaar709664c2020-12-12 14:33:41 +010053 if (lvar != NULL)
54 {
55 *lvar = *lvp;
Bram Moolenaarab360522021-01-10 14:02:28 +010056 lvar->lv_from_outer = 0;
Bram Moolenaar709664c2020-12-12 14:33:41 +010057 }
58 return OK;
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +020059 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010060 }
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +020061
62 // Find local in outer function scope.
63 if (cctx->ctx_outer != NULL)
64 {
Bram Moolenaar709664c2020-12-12 14:33:41 +010065 if (lookup_local(name, len, lvar, cctx->ctx_outer) == OK)
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +020066 {
Bram Moolenaar709664c2020-12-12 14:33:41 +010067 if (lvar != NULL)
68 {
69 cctx->ctx_outer_used = TRUE;
Bram Moolenaarab360522021-01-10 14:02:28 +010070 ++lvar->lv_from_outer;
Bram Moolenaar709664c2020-12-12 14:33:41 +010071 }
72 return OK;
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +020073 }
74 }
75
Bram Moolenaar709664c2020-12-12 14:33:41 +010076 return FAIL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010077}
78
79/*
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +020080 * Lookup an argument in the current function and an enclosing function.
81 * Returns the argument index in "idxp"
82 * Returns the argument type in "type"
83 * Sets "gen_load_outer" to TRUE if found in outer scope.
84 * Returns OK when found, FAIL otherwise.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010085 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +000086 int
Bram Moolenaarfbbcd002020-10-15 12:46:44 +020087arg_exists(
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +020088 char_u *name,
89 size_t len,
90 int *idxp,
91 type_T **type,
92 int *gen_load_outer,
93 cctx_T *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010094{
95 int idx;
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +020096 char_u *va_name;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010097
Bram Moolenaarae8d2de2020-02-13 21:42:24 +010098 if (len == 0)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +020099 return FAIL;
Bram Moolenaare28d9b32021-07-03 18:56:53 +0200100 for (idx = 0; idx < cctx->ctx_ufunc->uf_args_visible; ++idx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100101 {
102 char_u *arg = FUNCARG(cctx->ctx_ufunc, idx);
103
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200104 if (STRNCMP(name, arg, len) == 0 && arg[len] == NUL)
105 {
106 if (idxp != NULL)
107 {
108 // Arguments are located above the frame pointer. One further
109 // if there is a vararg argument
110 *idxp = idx - (cctx->ctx_ufunc->uf_args.ga_len
111 + STACK_FRAME_SIZE)
112 + (cctx->ctx_ufunc->uf_va_name != NULL ? -1 : 0);
113
114 if (cctx->ctx_ufunc->uf_arg_types != NULL)
115 *type = cctx->ctx_ufunc->uf_arg_types[idx];
116 else
117 *type = &t_any;
118 }
119 return OK;
120 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100121 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100122
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200123 va_name = cctx->ctx_ufunc->uf_va_name;
124 if (va_name != NULL
125 && STRNCMP(name, va_name, len) == 0 && va_name[len] == NUL)
126 {
127 if (idxp != NULL)
128 {
129 // varargs is always the last argument
130 *idxp = -STACK_FRAME_SIZE - 1;
131 *type = cctx->ctx_ufunc->uf_va_type;
132 }
133 return OK;
134 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100135
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200136 if (cctx->ctx_outer != NULL)
137 {
138 // Lookup the name for an argument of the outer function.
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200139 if (arg_exists(name, len, idxp, type, gen_load_outer, cctx->ctx_outer)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200140 == OK)
141 {
Bram Moolenaar44ec21c2021-02-12 21:50:57 +0100142 if (gen_load_outer != NULL)
143 ++*gen_load_outer;
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200144 return OK;
145 }
146 }
147
148 return FAIL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100149}
150
151/*
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200152 * Lookup a script-local variable in the current script, possibly defined in a
153 * block that contains the function "cctx->ctx_ufunc".
154 * "cctx" is NULL at the script level.
Bram Moolenaar18062fc2021-03-05 21:35:47 +0100155 * If "len" is <= 0 "name" must be NUL terminated.
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200156 * Return NULL when not found.
157 */
158 static sallvar_T *
159find_script_var(char_u *name, size_t len, cctx_T *cctx)
160{
161 scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid);
162 hashitem_T *hi;
163 int cc;
164 sallvar_T *sav;
Bram Moolenaar88421d62021-07-24 14:14:52 +0200165 sallvar_T *found_sav;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200166 ufunc_T *ufunc;
167
168 // Find the list of all script variables with the right name.
169 if (len > 0)
170 {
171 cc = name[len];
172 name[len] = NUL;
173 }
174 hi = hash_find(&si->sn_all_vars.dv_hashtab, name);
175 if (len > 0)
176 name[len] = cc;
177 if (HASHITEM_EMPTY(hi))
178 return NULL;
179
180 sav = HI2SAV(hi);
Bram Moolenaar3c77b6a2021-07-25 18:07:00 +0200181 if (sav->sav_block_id == 0)
182 // variable defined in the top script scope is always visible
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200183 return sav;
184
Bram Moolenaar3c77b6a2021-07-25 18:07:00 +0200185 if (cctx == NULL)
186 {
187 // Not in a function scope, find variable with block id equal to or
188 // smaller than the current block id.
189 while (sav != NULL)
190 {
191 if (sav->sav_block_id <= si->sn_current_block_id)
192 break;
193 sav = sav->sav_next;
194 }
195 return sav;
196 }
197
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200198 // Go over the variables with this name and find one that was visible
199 // from the function.
200 ufunc = cctx->ctx_ufunc;
Bram Moolenaar88421d62021-07-24 14:14:52 +0200201 found_sav = sav;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200202 while (sav != NULL)
203 {
204 int idx;
205
206 // Go over the blocks that this function was defined in. If the
207 // variable block ID matches it was visible to the function.
208 for (idx = 0; idx < ufunc->uf_block_depth; ++idx)
209 if (ufunc->uf_block_ids[idx] == sav->sav_block_id)
210 return sav;
211 sav = sav->sav_next;
212 }
213
Bram Moolenaar88421d62021-07-24 14:14:52 +0200214 // Not found, assume variable at script level was visible.
215 return found_sav;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200216}
217
218/*
Bram Moolenaar8e7d6222020-12-18 19:49:56 +0100219 * Return TRUE if the script context is Vim9 script.
Bram Moolenaar84367732020-08-23 15:21:55 +0200220 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000221 int
Bram Moolenaar84367732020-08-23 15:21:55 +0200222script_is_vim9()
223{
224 return SCRIPT_ITEM(current_sctx.sc_sid)->sn_version == SCRIPT_VERSION_VIM9;
225}
226
227/*
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200228 * Lookup a variable (without s: prefix) in the current script.
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200229 * "cctx" is NULL at the script level.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100230 * Returns OK or FAIL.
231 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000232 int
Bram Moolenaar15e5e532021-04-07 21:21:13 +0200233script_var_exists(char_u *name, size_t len, cctx_T *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100234{
Bram Moolenaar9c4f5522020-09-25 21:47:28 +0200235 if (current_sctx.sc_sid <= 0)
236 return FAIL;
Bram Moolenaar15e5e532021-04-07 21:21:13 +0200237 if (script_is_vim9())
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200238 {
239 // Check script variables that were visible where the function was
240 // defined.
241 if (find_script_var(name, len, cctx) != NULL)
242 return OK;
243 }
244 else
245 {
246 hashtab_T *ht = &SCRIPT_VARS(current_sctx.sc_sid);
247 dictitem_T *di;
248 int cc;
249
250 // Check script variables that are currently visible
251 cc = name[len];
252 name[len] = NUL;
253 di = find_var_in_ht(ht, 0, name, TRUE);
254 name[len] = cc;
255 if (di != NULL)
256 return OK;
257 }
258
259 return FAIL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100260}
261
Bram Moolenaar5269bd22020-03-09 19:25:27 +0100262/*
Bram Moolenaare0890d62021-02-17 14:52:14 +0100263 * Return TRUE if "name" is a local variable, argument, script variable or
264 * imported.
265 */
266 static int
267variable_exists(char_u *name, size_t len, cctx_T *cctx)
268{
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100269 return (cctx != NULL
270 && (lookup_local(name, len, NULL, cctx) == OK
271 || arg_exists(name, len, NULL, NULL, NULL, cctx) == OK))
Bram Moolenaar15e5e532021-04-07 21:21:13 +0200272 || script_var_exists(name, len, cctx) == OK
Bram Moolenaare0890d62021-02-17 14:52:14 +0100273 || find_imported(name, len, cctx) != NULL;
274}
275
276/*
Bram Moolenaar6914e872021-03-06 21:01:09 +0100277 * Return TRUE if "name" is a local variable, argument, script variable,
278 * imported or function.
279 */
280 static int
Bram Moolenaar77b10ff2021-03-14 13:21:35 +0100281item_exists(char_u *name, size_t len, int cmd UNUSED, cctx_T *cctx)
Bram Moolenaar6914e872021-03-06 21:01:09 +0100282{
283 int is_global;
Bram Moolenaar77b10ff2021-03-14 13:21:35 +0100284 char_u *p;
Bram Moolenaar6914e872021-03-06 21:01:09 +0100285
286 if (variable_exists(name, len, cctx))
287 return TRUE;
288
Bram Moolenaar77b10ff2021-03-14 13:21:35 +0100289 // This is similar to what is in lookup_scriptitem():
290 // Find a function, so that a following "->" works.
291 // Require "(" or "->" to follow, "Cmd" is a user command while "Cmd()" is
292 // a function call.
293 p = skipwhite(name + len);
294
295 if (name[len] == '(' || (p[0] == '-' && p[1] == '>'))
296 {
297 // Do not check for an internal function, since it might also be a
Yegappan Lakshmanan2d6d7182021-06-13 21:52:48 +0200298 // valid command, such as ":split" versus "split()".
Bram Moolenaar77b10ff2021-03-14 13:21:35 +0100299 // Skip "g:" before a function name.
300 is_global = (name[0] == 'g' && name[1] == ':');
301 return find_func(is_global ? name + 2 : name, is_global, cctx) != NULL;
302 }
303 return FALSE;
Bram Moolenaar6914e872021-03-06 21:01:09 +0100304}
305
306/*
Bram Moolenaar5269bd22020-03-09 19:25:27 +0100307 * Check if "p[len]" is already defined, either in script "import_sid" or in
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200308 * compilation context "cctx". "cctx" is NULL at the script level.
Bram Moolenaar0f769812020-09-12 18:32:34 +0200309 * Does not check the global namespace.
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100310 * If "is_arg" is TRUE the error message is for an argument name.
Bram Moolenaar5269bd22020-03-09 19:25:27 +0100311 * Return FAIL and give an error if it defined.
312 */
313 int
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100314check_defined(char_u *p, size_t len, cctx_T *cctx, int is_arg)
Bram Moolenaar5269bd22020-03-09 19:25:27 +0100315{
Bram Moolenaar0f769812020-09-12 18:32:34 +0200316 int c = p[len];
317 ufunc_T *ufunc = NULL;
Bram Moolenaarad486a02020-08-01 23:22:18 +0200318
Bram Moolenaarda479c72021-04-10 21:01:38 +0200319 // underscore argument is OK
320 if (len == 1 && *p == '_')
321 return OK;
322
Bram Moolenaar15e5e532021-04-07 21:21:13 +0200323 if (script_var_exists(p, len, cctx) == OK)
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100324 {
325 if (is_arg)
326 semsg(_(e_argument_already_declared_in_script_str), p);
327 else
328 semsg(_(e_variable_already_declared_in_script_str), p);
329 return FAIL;
330 }
331
Bram Moolenaarad486a02020-08-01 23:22:18 +0200332 p[len] = NUL;
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100333 if ((cctx != NULL
Bram Moolenaar709664c2020-12-12 14:33:41 +0100334 && (lookup_local(p, len, NULL, cctx) == OK
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200335 || arg_exists(p, len, NULL, NULL, NULL, cctx) == OK))
Bram Moolenaarad486a02020-08-01 23:22:18 +0200336 || find_imported(p, len, cctx) != NULL
Bram Moolenaar0f769812020-09-12 18:32:34 +0200337 || (ufunc = find_func_even_dead(p, FALSE, cctx)) != NULL)
Bram Moolenaar5269bd22020-03-09 19:25:27 +0100338 {
Bram Moolenaar0f769812020-09-12 18:32:34 +0200339 // A local or script-local function can shadow a global function.
Bram Moolenaar577dc932021-06-27 15:35:40 +0200340 if (ufunc == NULL || ((ufunc->uf_flags & FC_DEAD) == 0
341 && (!func_is_global(ufunc)
342 || (p[0] == 'g' && p[1] == ':'))))
Bram Moolenaar0f769812020-09-12 18:32:34 +0200343 {
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100344 if (is_arg)
345 semsg(_(e_argument_name_shadows_existing_variable_str), p);
346 else
347 semsg(_(e_name_already_defined_str), p);
Bram Moolenaar0f769812020-09-12 18:32:34 +0200348 p[len] = c;
Bram Moolenaar0f769812020-09-12 18:32:34 +0200349 return FAIL;
350 }
Bram Moolenaar5269bd22020-03-09 19:25:27 +0100351 }
Bram Moolenaarad486a02020-08-01 23:22:18 +0200352 p[len] = c;
Bram Moolenaar5269bd22020-03-09 19:25:27 +0100353 return OK;
354}
355
Bram Moolenaar65b95452020-07-19 14:03:09 +0200356
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100357/*
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200358 * Return TRUE if "actual" could be "expected" and a runtime typecheck is to be
359 * used. Return FALSE if the types will never match.
360 */
Yegappan Lakshmanan8ee52af2021-08-09 19:59:06 +0200361 static int
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200362use_typecheck(type_T *actual, type_T *expected)
363{
364 if (actual->tt_type == VAR_ANY
365 || actual->tt_type == VAR_UNKNOWN
366 || (actual->tt_type == VAR_FUNC
367 && (expected->tt_type == VAR_FUNC
368 || expected->tt_type == VAR_PARTIAL)
Bram Moolenaarfa46ead2021-12-22 13:18:39 +0000369 && (actual->tt_member == &t_any
370 || actual->tt_member == &t_unknown
371 || actual->tt_argcount < 0)
372 && (actual->tt_member == &t_unknown ||
373 (actual->tt_member == &t_void)
Bram Moolenaar328eac22021-01-07 19:23:08 +0100374 == (expected->tt_member == &t_void))))
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200375 return TRUE;
376 if ((actual->tt_type == VAR_LIST || actual->tt_type == VAR_DICT)
377 && actual->tt_type == expected->tt_type)
378 // This takes care of a nested list or dict.
379 return use_typecheck(actual->tt_member, expected->tt_member);
380 return FALSE;
381}
382
383/*
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200384 * Check that
Bram Moolenaar5e654232020-09-16 15:22:00 +0200385 * - "actual" matches "expected" type or
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200386 * - "actual" is a type that can be "expected" type: add a runtime check; or
387 * - return FAIL.
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200388 * If "actual_is_const" is TRUE then the type won't change at runtime, do not
389 * generate a TYPECHECK.
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200390 */
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200391 static int
392need_type_where(
393 type_T *actual,
394 type_T *expected,
395 int offset,
396 where_T where,
397 cctx_T *cctx,
398 int silent,
399 int actual_is_const)
400{
Bram Moolenaar44a89772021-12-18 12:31:33 +0000401 int ret;
402
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200403 if (expected == &t_bool && actual != &t_bool
404 && (actual->tt_flags & TTFLAG_BOOL_OK))
405 {
406 // Using "0", "1" or the result of an expression with "&&" or "||" as a
407 // boolean is OK but requires a conversion.
408 generate_2BOOL(cctx, FALSE, offset);
409 return OK;
410 }
411
Bram Moolenaar44a89772021-12-18 12:31:33 +0000412 ret = check_type_maybe(expected, actual, FALSE, where);
413 if (ret == OK)
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200414 return OK;
415
416 // If the actual type can be the expected type add a runtime check.
417 // If it's a constant a runtime check makes no sense.
Bram Moolenaarfa46ead2021-12-22 13:18:39 +0000418 if (!actual_is_const && ret == MAYBE && use_typecheck(actual, expected))
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200419 {
420 generate_TYPECHECK(cctx, expected, offset, where.wt_index);
421 return OK;
422 }
423
424 if (!silent)
425 type_mismatch_where(expected, actual, where);
426 return FAIL;
427}
428
Bram Moolenaar351ead02021-01-16 16:07:01 +0100429 int
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200430need_type(
431 type_T *actual,
432 type_T *expected,
433 int offset,
Bram Moolenaar351ead02021-01-16 16:07:01 +0100434 int arg_idx,
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200435 cctx_T *cctx,
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200436 int silent,
437 int actual_is_const)
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200438{
Bram Moolenaar7a3fe3e2021-07-22 14:58:47 +0200439 where_T where = WHERE_INIT;
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100440
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100441 where.wt_index = arg_idx;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200442 return need_type_where(actual, expected, offset, where,
443 cctx, silent, actual_is_const);
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200444}
445
446/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100447 * Reserve space for a local variable.
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200448 * Return the variable or NULL if it failed.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100449 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000450 lvar_T *
Bram Moolenaare8211a32020-10-09 22:04:29 +0200451reserve_local(
452 cctx_T *cctx,
453 char_u *name,
454 size_t len,
455 int isConst,
456 type_T *type)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100457{
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100458 lvar_T *lvar;
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200459 dfunc_T *dfunc;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100460
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200461 if (arg_exists(name, len, NULL, NULL, NULL, cctx) == OK)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100462 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200463 emsg_namelen(_(e_str_is_used_as_argument), name, (int)len);
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200464 return NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100465 }
466
Bram Moolenaar35578162021-08-02 19:10:38 +0200467 if (GA_GROW_FAILS(&cctx->ctx_locals, 1))
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200468 return NULL;
469 lvar = ((lvar_T *)cctx->ctx_locals.ga_data) + cctx->ctx_locals.ga_len++;
Bram Moolenaare8211a32020-10-09 22:04:29 +0200470 CLEAR_POINTER(lvar);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100471
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200472 // Every local variable uses the next entry on the stack. We could re-use
473 // the last ones when leaving a scope, but then variables used in a closure
474 // might get overwritten. To keep things simple do not re-use stack
475 // entries. This is less efficient, but memory is cheap these days.
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200476 dfunc = ((dfunc_T *)def_functions.ga_data) + cctx->ctx_ufunc->uf_dfunc_idx;
477 lvar->lv_idx = dfunc->df_var_names.ga_len;
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200478
Bram Moolenaar71ccd032020-06-12 22:59:11 +0200479 lvar->lv_name = vim_strnsave(name, len == 0 ? STRLEN(name) : len);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100480 lvar->lv_const = isConst;
481 lvar->lv_type = type;
482
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200483 // Remember the name for debugging.
Bram Moolenaar35578162021-08-02 19:10:38 +0200484 if (GA_GROW_FAILS(&dfunc->df_var_names, 1))
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200485 return NULL;
486 ((char_u **)dfunc->df_var_names.ga_data)[lvar->lv_idx] =
487 vim_strsave(lvar->lv_name);
488 ++dfunc->df_var_names.ga_len;
489
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200490 return lvar;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100491}
492
493/*
Bram Moolenaar08251752021-01-11 21:20:18 +0100494 * If "check_writable" is ASSIGN_CONST give an error if the variable was
495 * defined with :final or :const, if "check_writable" is ASSIGN_FINAL give an
496 * error if the variable was defined with :const.
497 */
498 static int
499check_item_writable(svar_T *sv, int check_writable, char_u *name)
500{
501 if ((check_writable == ASSIGN_CONST && sv->sv_const != 0)
502 || (check_writable == ASSIGN_FINAL
503 && sv->sv_const == ASSIGN_CONST))
504 {
Bram Moolenaard8e44472021-07-21 22:20:33 +0200505 semsg(_(e_cannot_change_readonly_variable_str), name);
Bram Moolenaar08251752021-01-11 21:20:18 +0100506 return FAIL;
507 }
508 return OK;
509}
510
511/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100512 * Find "name" in script-local items of script "sid".
Bram Moolenaar08251752021-01-11 21:20:18 +0100513 * Pass "check_writable" to check_item_writable().
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100514 * Returns the index in "sn_var_vals" if found.
515 * If found but not in "sn_var_vals" returns -1.
Bram Moolenaar08251752021-01-11 21:20:18 +0100516 * If not found or the variable is not writable returns -2.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100517 */
518 int
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200519get_script_item_idx(int sid, char_u *name, int check_writable, cctx_T *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100520{
521 hashtab_T *ht;
522 dictitem_T *di;
Bram Moolenaar21b9e972020-01-26 19:26:46 +0100523 scriptitem_T *si = SCRIPT_ITEM(sid);
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200524 svar_T *sv;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100525 int idx;
526
Bram Moolenaare3d46852020-08-29 13:39:17 +0200527 if (!SCRIPT_ID_VALID(sid))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100528 return -1;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200529 if (sid == current_sctx.sc_sid)
530 {
Bram Moolenaar209f0202020-10-15 13:57:56 +0200531 sallvar_T *sav = find_script_var(name, 0, cctx);
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200532
533 if (sav == NULL)
534 return -2;
535 idx = sav->sav_var_vals_idx;
536 sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
Bram Moolenaar08251752021-01-11 21:20:18 +0100537 if (check_item_writable(sv, check_writable, name) == FAIL)
538 return -2;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200539 return idx;
540 }
541
542 // First look the name up in the hashtable.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100543 ht = &SCRIPT_VARS(sid);
544 di = find_var_in_ht(ht, 0, name, TRUE);
545 if (di == NULL)
546 return -2;
547
548 // Now find the svar_T index in sn_var_vals.
549 for (idx = 0; idx < si->sn_var_vals.ga_len; ++idx)
550 {
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200551 sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100552 if (sv->sv_tv == &di->di_tv)
553 {
Bram Moolenaar08251752021-01-11 21:20:18 +0100554 if (check_item_writable(sv, check_writable, name) == FAIL)
555 return -2;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100556 return idx;
557 }
558 }
559 return -1;
560}
561
562/*
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200563 * Find "name" in imported items of the current script or in "cctx" if not
564 * NULL.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100565 */
566 imported_T *
Bram Moolenaar4e12a5d2020-02-03 20:50:59 +0100567find_imported(char_u *name, size_t len, cctx_T *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100568{
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100569 int idx;
570
Bram Moolenaare3d46852020-08-29 13:39:17 +0200571 if (!SCRIPT_ID_VALID(current_sctx.sc_sid))
Bram Moolenaar8e6cbb72020-07-01 14:38:12 +0200572 return NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100573 if (cctx != NULL)
574 for (idx = 0; idx < cctx->ctx_imports.ga_len; ++idx)
575 {
576 imported_T *import = ((imported_T *)cctx->ctx_imports.ga_data)
577 + idx;
578
Bram Moolenaar4e12a5d2020-02-03 20:50:59 +0100579 if (len == 0 ? STRCMP(name, import->imp_name) == 0
580 : STRLEN(import->imp_name) == len
581 && STRNCMP(name, import->imp_name, len) == 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100582 return import;
583 }
584
Bram Moolenaarefa94442020-08-08 22:16:00 +0200585 return find_imported_in_script(name, len, current_sctx.sc_sid);
586}
587
588 imported_T *
589find_imported_in_script(char_u *name, size_t len, int sid)
590{
Bram Moolenaare3d46852020-08-29 13:39:17 +0200591 scriptitem_T *si;
Bram Moolenaarefa94442020-08-08 22:16:00 +0200592 int idx;
593
Bram Moolenaare3d46852020-08-29 13:39:17 +0200594 if (!SCRIPT_ID_VALID(sid))
595 return NULL;
596 si = SCRIPT_ITEM(sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100597 for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
598 {
599 imported_T *import = ((imported_T *)si->sn_imports.ga_data) + idx;
600
Bram Moolenaar4e12a5d2020-02-03 20:50:59 +0100601 if (len == 0 ? STRCMP(name, import->imp_name) == 0
602 : STRLEN(import->imp_name) == len
603 && STRNCMP(name, import->imp_name, len) == 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100604 return import;
605 }
606 return NULL;
607}
608
609/*
Bram Moolenaar20431c92020-03-20 18:39:46 +0100610 * Free all imported variables.
611 */
612 static void
613free_imported(cctx_T *cctx)
614{
615 int idx;
616
617 for (idx = 0; idx < cctx->ctx_imports.ga_len; ++idx)
618 {
619 imported_T *import = ((imported_T *)cctx->ctx_imports.ga_data) + idx;
620
621 vim_free(import->imp_name);
622 }
623 ga_clear(&cctx->ctx_imports);
624}
625
626/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000627 * Called when checking for a following operator at "arg". When the rest of
628 * the line is empty or only a comment, peek the next line. If there is a next
629 * line return a pointer to it and set "nextp".
630 * Otherwise skip over white space.
631 */
632 char_u *
633may_peek_next_line(cctx_T *cctx, char_u *arg, char_u **nextp)
634{
635 char_u *p = skipwhite(arg);
636
637 *nextp = NULL;
638 if (*p == NUL || (VIM_ISWHITE(*arg) && vim9_comment_start(p)))
639 {
640 *nextp = peek_next_line_from_context(cctx);
641 if (*nextp != NULL)
642 return *nextp;
643 }
644 return p;
645}
646
647/*
Bram Moolenaar23c55272020-06-21 16:58:13 +0200648 * Return a pointer to the next line that isn't empty or only contains a
649 * comment. Skips over white space.
650 * Returns NULL if there is none.
651 */
Bram Moolenaar7a4b8982020-07-08 17:36:21 +0200652 char_u *
653peek_next_line_from_context(cctx_T *cctx)
Bram Moolenaar23c55272020-06-21 16:58:13 +0200654{
655 int lnum = cctx->ctx_lnum;
656
657 while (++lnum < cctx->ctx_ufunc->uf_lines.ga_len)
658 {
659 char_u *line = ((char_u **)cctx->ctx_ufunc->uf_lines.ga_data)[lnum];
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +0200660 char_u *p;
Bram Moolenaar23c55272020-06-21 16:58:13 +0200661
Bram Moolenaarba60cc42020-08-12 19:15:33 +0200662 // ignore NULLs inserted for continuation lines
663 if (line != NULL)
664 {
665 p = skipwhite(line);
Bram Moolenaar4b3e1962021-03-18 21:37:55 +0100666 if (vim9_bad_comment(p))
667 return NULL;
Bram Moolenaarba60cc42020-08-12 19:15:33 +0200668 if (*p != NUL && !vim9_comment_start(p))
669 return p;
670 }
Bram Moolenaar23c55272020-06-21 16:58:13 +0200671 }
672 return NULL;
673}
674
675/*
Bram Moolenaare6085c52020-04-12 20:19:16 +0200676 * Get the next line of the function from "cctx".
Bram Moolenaar23c55272020-06-21 16:58:13 +0200677 * Skips over empty lines. Skips over comment lines if "skip_comment" is TRUE.
Bram Moolenaare6085c52020-04-12 20:19:16 +0200678 * Returns NULL when at the end.
679 */
Bram Moolenaar7a4b8982020-07-08 17:36:21 +0200680 char_u *
Bram Moolenaar23c55272020-06-21 16:58:13 +0200681next_line_from_context(cctx_T *cctx, int skip_comment)
Bram Moolenaare6085c52020-04-12 20:19:16 +0200682{
Bram Moolenaar7a092242020-04-16 22:10:49 +0200683 char_u *line;
Bram Moolenaare6085c52020-04-12 20:19:16 +0200684
685 do
686 {
687 ++cctx->ctx_lnum;
688 if (cctx->ctx_lnum >= cctx->ctx_ufunc->uf_lines.ga_len)
Bram Moolenaar7a092242020-04-16 22:10:49 +0200689 {
690 line = NULL;
Bram Moolenaare6085c52020-04-12 20:19:16 +0200691 break;
Bram Moolenaar7a092242020-04-16 22:10:49 +0200692 }
Bram Moolenaare6085c52020-04-12 20:19:16 +0200693 line = ((char_u **)cctx->ctx_ufunc->uf_lines.ga_data)[cctx->ctx_lnum];
Bram Moolenaar7a092242020-04-16 22:10:49 +0200694 cctx->ctx_line_start = line;
Bram Moolenaar25e0f582020-05-25 22:36:50 +0200695 SOURCING_LNUM = cctx->ctx_lnum + 1;
Bram Moolenaar23c55272020-06-21 16:58:13 +0200696 } while (line == NULL || *skipwhite(line) == NUL
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +0200697 || (skip_comment && vim9_comment_start(skipwhite(line))));
Bram Moolenaare6085c52020-04-12 20:19:16 +0200698 return line;
699}
700
701/*
Bram Moolenaar5afd0812021-01-03 18:33:13 +0100702 * Skip over white space at "whitep" and assign to "*arg".
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200703 * If "*arg" is at the end of the line, advance to the next line.
Bram Moolenaar2c330432020-04-13 14:41:35 +0200704 * Also when "whitep" points to white space and "*arg" is on a "#".
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200705 * Return FAIL if beyond the last line, "*arg" is unmodified then.
706 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000707 int
Bram Moolenaar2c330432020-04-13 14:41:35 +0200708may_get_next_line(char_u *whitep, char_u **arg, cctx_T *cctx)
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200709{
Bram Moolenaar5afd0812021-01-03 18:33:13 +0100710 *arg = skipwhite(whitep);
Bram Moolenaar4b3e1962021-03-18 21:37:55 +0100711 if (vim9_bad_comment(*arg))
712 return FAIL;
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200713 if (**arg == NUL || (VIM_ISWHITE(*whitep) && vim9_comment_start(*arg)))
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200714 {
Bram Moolenaar23c55272020-06-21 16:58:13 +0200715 char_u *next = next_line_from_context(cctx, TRUE);
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200716
717 if (next == NULL)
718 return FAIL;
719 *arg = skipwhite(next);
720 }
721 return OK;
722}
723
Bram Moolenaara7eedf32020-07-10 21:50:41 +0200724/*
725 * Idem, and give an error when failed.
726 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000727 int
Bram Moolenaara7eedf32020-07-10 21:50:41 +0200728may_get_next_line_error(char_u *whitep, char_u **arg, cctx_T *cctx)
729{
730 if (may_get_next_line(whitep, arg, cctx) == FAIL)
731 {
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100732 SOURCING_LNUM = cctx->ctx_lnum + 1;
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200733 emsg(_(e_line_incomplete));
Bram Moolenaara7eedf32020-07-10 21:50:41 +0200734 return FAIL;
735 }
736 return OK;
737}
738
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100739/*
Bram Moolenaar04b12692020-05-04 23:24:44 +0200740 * Get a line from the compilation context, compatible with exarg_T getline().
741 * Return a pointer to the line in allocated memory.
742 * Return NULL for end-of-file or some error.
743 */
744 static char_u *
745exarg_getline(
746 int c UNUSED,
747 void *cookie,
748 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +0200749 getline_opt_T options UNUSED)
Bram Moolenaar04b12692020-05-04 23:24:44 +0200750{
751 cctx_T *cctx = (cctx_T *)cookie;
Bram Moolenaar66250c92020-08-20 15:02:42 +0200752 char_u *p;
Bram Moolenaar04b12692020-05-04 23:24:44 +0200753
Bram Moolenaar66250c92020-08-20 15:02:42 +0200754 for (;;)
Bram Moolenaar04b12692020-05-04 23:24:44 +0200755 {
Bram Moolenaar2914a202020-09-27 18:24:03 +0200756 if (cctx->ctx_lnum >= cctx->ctx_ufunc->uf_lines.ga_len - 1)
Bram Moolenaar66250c92020-08-20 15:02:42 +0200757 return NULL;
758 ++cctx->ctx_lnum;
759 p = ((char_u **)cctx->ctx_ufunc->uf_lines.ga_data)[cctx->ctx_lnum];
760 // Comment lines result in NULL pointers, skip them.
761 if (p != NULL)
762 return vim_strsave(p);
Bram Moolenaar04b12692020-05-04 23:24:44 +0200763 }
Bram Moolenaar04b12692020-05-04 23:24:44 +0200764}
765
Bram Moolenaar7a6eaa02021-03-21 20:53:29 +0100766 void
767fill_exarg_from_cctx(exarg_T *eap, cctx_T *cctx)
768{
769 eap->getline = exarg_getline;
770 eap->cookie = cctx;
771}
772
Bram Moolenaar04b12692020-05-04 23:24:44 +0200773/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000774 * Return TRUE if "ufunc" should be compiled, taking into account whether
775 * "profile" indicates profiling is to be done.
776 */
777 int
778func_needs_compiling(ufunc_T *ufunc, compiletype_T compile_type)
779{
780 switch (ufunc->uf_def_status)
781 {
782 case UF_TO_BE_COMPILED:
783 return TRUE;
784
785 case UF_COMPILED:
786 {
787 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
788 + ufunc->uf_dfunc_idx;
789
790 switch (compile_type)
791 {
792 case CT_PROFILE:
793#ifdef FEAT_PROFILE
794 return dfunc->df_instr_prof == NULL;
795#endif
796 case CT_NONE:
797 return dfunc->df_instr == NULL;
798 case CT_DEBUG:
799 return dfunc->df_instr_debug == NULL;
800 }
801 }
802
803 case UF_NOT_COMPILED:
804 case UF_COMPILE_ERROR:
805 case UF_COMPILING:
806 break;
807 }
808 return FALSE;
809}
810
811/*
Bram Moolenaar04b12692020-05-04 23:24:44 +0200812 * Compile a nested :def command.
813 */
814 static char_u *
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +0000815compile_nested_function(exarg_T *eap, cctx_T *cctx, char_u **line_to_free)
Bram Moolenaar04b12692020-05-04 23:24:44 +0200816{
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200817 int is_global = *eap->arg == 'g' && eap->arg[1] == ':';
Bram Moolenaar04b12692020-05-04 23:24:44 +0200818 char_u *name_start = eap->arg;
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200819 char_u *name_end = to_name_end(eap->arg, TRUE);
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +0000820 int off;
821 char_u *func_name;
Bram Moolenaareef21022020-08-01 22:16:43 +0200822 char_u *lambda_name;
Bram Moolenaar04b12692020-05-04 23:24:44 +0200823 ufunc_T *ufunc;
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100824 int r = FAIL;
Bram Moolenaarffcfddc2021-07-11 20:22:30 +0200825 compiletype_T compile_type;
Bram Moolenaar04b12692020-05-04 23:24:44 +0200826
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +0200827 if (eap->forceit)
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200828 {
829 emsg(_(e_cannot_use_bang_with_nested_def));
830 return NULL;
831 }
832
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100833 if (*name_start == '/')
834 {
835 name_end = skip_regexp(name_start + 1, '/', TRUE);
836 if (*name_end == '/')
837 ++name_end;
Bram Moolenaar63b91732021-08-05 20:40:03 +0200838 set_nextcmd(eap, name_end);
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100839 }
840 if (name_end == name_start || *skipwhite(name_end) != '(')
841 {
842 if (!ends_excmd2(name_start, name_end))
843 {
844 semsg(_(e_invalid_command_str), eap->cmd);
845 return NULL;
846 }
847
848 // "def" or "def Name": list functions
849 if (generate_DEF(cctx, name_start, name_end - name_start) == FAIL)
850 return NULL;
851 return eap->nextcmd == NULL ? (char_u *)"" : eap->nextcmd;
852 }
853
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200854 // Only g:Func() can use a namespace.
855 if (name_start[1] == ':' && !is_global)
856 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200857 semsg(_(e_namespace_not_supported_str), name_start);
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200858 return NULL;
859 }
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100860 if (check_defined(name_start, name_end - name_start, cctx, FALSE) == FAIL)
Bram Moolenaareef21022020-08-01 22:16:43 +0200861 return NULL;
862
Bram Moolenaar04b12692020-05-04 23:24:44 +0200863 eap->arg = name_end;
Bram Moolenaar7a6eaa02021-03-21 20:53:29 +0100864 fill_exarg_from_cctx(eap, cctx);
865
Bram Moolenaar04b12692020-05-04 23:24:44 +0200866 eap->forceit = FALSE;
Bram Moolenaar38453522021-11-28 22:00:12 +0000867 // We use the special <Lamba>99 name, but it's not really a lambda.
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100868 lambda_name = vim_strsave(get_lambda_name());
869 if (lambda_name == NULL)
870 return NULL;
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +0000871
872 // This may free the current line, make a copy of the name.
873 off = is_global ? 2 : 0;
874 func_name = vim_strnsave(name_start + off, name_end - name_start - off);
875 if (func_name == NULL)
876 {
877 r = FAIL;
878 goto theend;
879 }
880
881 ufunc = define_function(eap, lambda_name, line_to_free);
Bram Moolenaar822ba242020-05-24 23:00:18 +0200882 if (ufunc == NULL)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100883 {
884 r = eap->skip ? OK : FAIL;
885 goto theend;
886 }
Bram Moolenaar7473a842021-12-28 17:55:26 +0000887 if (eap->nextcmd != NULL)
888 {
889 semsg(_(e_text_found_after_str_str),
890 eap->cmdidx == CMD_def ? "enddef" : "endfunction", eap->nextcmd);
891 r = FAIL;
Bram Moolenaard2939812021-12-30 17:09:05 +0000892 func_ptr_unref(ufunc);
Bram Moolenaar7473a842021-12-28 17:55:26 +0000893 goto theend;
894 }
Bram Moolenaar8863bda2021-03-17 18:42:08 +0100895
896 // copy over the block scope IDs before compiling
897 if (!is_global && cctx->ctx_ufunc->uf_block_depth > 0)
898 {
899 int block_depth = cctx->ctx_ufunc->uf_block_depth;
900
901 ufunc->uf_block_ids = ALLOC_MULT(int, block_depth);
902 if (ufunc->uf_block_ids != NULL)
903 {
904 mch_memmove(ufunc->uf_block_ids, cctx->ctx_ufunc->uf_block_ids,
905 sizeof(int) * block_depth);
906 ufunc->uf_block_depth = block_depth;
907 }
908 }
909
Bram Moolenaarffcfddc2021-07-11 20:22:30 +0200910 compile_type = COMPILE_TYPE(ufunc);
911#ifdef FEAT_PROFILE
912 // If the outer function is profiled, also compile the nested function for
913 // profiling.
914 if (cctx->ctx_compile_type == CT_PROFILE)
915 compile_type = CT_PROFILE;
916#endif
917 if (func_needs_compiling(ufunc, compile_type)
918 && compile_def_function(ufunc, TRUE, compile_type, cctx) == FAIL)
Bram Moolenaar4ee711f2020-09-23 18:51:11 +0200919 {
920 func_ptr_unref(ufunc);
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100921 goto theend;
Bram Moolenaar4ee711f2020-09-23 18:51:11 +0200922 }
Bram Moolenaar04b12692020-05-04 23:24:44 +0200923
Bram Moolenaar648594e2021-07-11 17:55:01 +0200924#ifdef FEAT_PROFILE
925 // When the outer function is compiled for profiling, the nested function
926 // may be called without profiling. Compile it here in the right context.
Bram Moolenaarffcfddc2021-07-11 20:22:30 +0200927 if (compile_type == CT_PROFILE && func_needs_compiling(ufunc, CT_NONE))
Bram Moolenaar648594e2021-07-11 17:55:01 +0200928 compile_def_function(ufunc, FALSE, CT_NONE, cctx);
929#endif
930
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200931 if (is_global)
932 {
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +0000933 r = generate_NEWFUNC(cctx, lambda_name, func_name);
934 func_name = NULL;
935 lambda_name = NULL;
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200936 }
937 else
938 {
939 // Define a local variable for the function reference.
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +0000940 lvar_T *lvar = reserve_local(cctx, func_name, name_end - name_start,
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200941 TRUE, ufunc->uf_func_type);
Bram Moolenaare8211a32020-10-09 22:04:29 +0200942
Bram Moolenaareef21022020-08-01 22:16:43 +0200943 if (lvar == NULL)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100944 goto theend;
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200945 if (generate_FUNCREF(cctx, ufunc) == FAIL)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100946 goto theend;
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200947 r = generate_STORE(cctx, ISN_STORE, lvar->lv_idx, NULL);
948 }
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100949
950theend:
951 vim_free(lambda_name);
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +0000952 vim_free(func_name);
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200953 return r == FAIL ? NULL : (char_u *)"";
Bram Moolenaar04b12692020-05-04 23:24:44 +0200954}
955
956/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100957 * Return the length of an assignment operator, or zero if there isn't one.
958 */
959 int
960assignment_len(char_u *p, int *heredoc)
961{
962 if (*p == '=')
963 {
964 if (p[1] == '<' && p[2] == '<')
965 {
966 *heredoc = TRUE;
967 return 3;
968 }
969 return 1;
970 }
971 if (vim_strchr((char_u *)"+-*/%", *p) != NULL && p[1] == '=')
972 return 2;
973 if (STRNCMP(p, "..=", 3) == 0)
974 return 3;
975 return 0;
976}
977
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100978/*
Bram Moolenaar1cc2a942020-05-10 19:10:31 +0200979 * Generate the load instruction for "name".
980 */
981 static void
982generate_loadvar(
983 cctx_T *cctx,
984 assign_dest_T dest,
985 char_u *name,
986 lvar_T *lvar,
987 type_T *type)
988{
989 switch (dest)
990 {
991 case dest_option:
Bram Moolenaardcb53be2021-12-09 14:23:43 +0000992 case dest_func_option:
Bram Moolenaar1cc2a942020-05-10 19:10:31 +0200993 generate_LOAD(cctx, ISN_LOADOPT, 0, name, type);
994 break;
995 case dest_global:
Bram Moolenaar03290b82020-12-19 16:30:44 +0100996 if (vim_strchr(name, AUTOLOAD_CHAR) == NULL)
997 generate_LOAD(cctx, ISN_LOADG, 0, name + 2, type);
998 else
999 generate_LOAD(cctx, ISN_LOADAUTO, 0, name, type);
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001000 break;
1001 case dest_buffer:
1002 generate_LOAD(cctx, ISN_LOADB, 0, name + 2, type);
1003 break;
1004 case dest_window:
1005 generate_LOAD(cctx, ISN_LOADW, 0, name + 2, type);
1006 break;
1007 case dest_tab:
1008 generate_LOAD(cctx, ISN_LOADT, 0, name + 2, type);
1009 break;
1010 case dest_script:
1011 compile_load_scriptvar(cctx,
1012 name + (name[1] == ':' ? 2 : 0), NULL, NULL, TRUE);
1013 break;
1014 case dest_env:
1015 // Include $ in the name here
1016 generate_LOAD(cctx, ISN_LOADENV, 0, name, type);
1017 break;
1018 case dest_reg:
1019 generate_LOAD(cctx, ISN_LOADREG, name[1], NULL, &t_string);
1020 break;
1021 case dest_vimvar:
1022 generate_LOADV(cctx, name + 2, TRUE);
1023 break;
1024 case dest_local:
Bram Moolenaarab360522021-01-10 14:02:28 +01001025 if (lvar->lv_from_outer > 0)
1026 generate_LOADOUTER(cctx, lvar->lv_idx, lvar->lv_from_outer,
1027 type);
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001028 else
1029 generate_LOAD(cctx, ISN_LOAD, lvar->lv_idx, NULL, type);
1030 break;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001031 case dest_expr:
1032 // list or dict value should already be on the stack.
1033 break;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001034 }
1035}
1036
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001037/*
1038 * Skip over "[expr]" or ".member".
1039 * Does not check for any errors.
1040 */
1041 static char_u *
1042skip_index(char_u *start)
1043{
1044 char_u *p = start;
1045
1046 if (*p == '[')
1047 {
1048 p = skipwhite(p + 1);
1049 (void)skip_expr(&p, NULL);
1050 p = skipwhite(p);
1051 if (*p == ']')
1052 return p + 1;
1053 return p;
1054 }
1055 // if (*p == '.')
1056 return to_name_end(p + 1, TRUE);
1057}
1058
Bram Moolenaare55b1c02020-06-21 15:52:59 +02001059 void
1060vim9_declare_error(char_u *name)
1061{
1062 char *scope = "";
1063
1064 switch (*name)
1065 {
Bram Moolenaar7fe87552020-06-21 20:38:28 +02001066 case 'g': scope = _("global"); break;
1067 case 'b': scope = _("buffer"); break;
1068 case 'w': scope = _("window"); break;
1069 case 't': scope = _("tab"); break;
1070 case 'v': scope = "v:"; break;
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02001071 case '$': semsg(_(e_cannot_declare_an_environment_variable), name);
Bram Moolenaarc2ee44c2020-08-02 16:59:00 +02001072 return;
Bram Moolenaar451c2e32020-08-15 16:33:28 +02001073 case '&': semsg(_(e_cannot_declare_an_option), name);
Bram Moolenaarc2ee44c2020-08-02 16:59:00 +02001074 return;
Bram Moolenaar7cb6fc22020-08-21 22:36:47 +02001075 case '@': semsg(_(e_cannot_declare_a_register_str), name);
Bram Moolenaarc2ee44c2020-08-02 16:59:00 +02001076 return;
Bram Moolenaar7fe87552020-06-21 20:38:28 +02001077 default: return;
Bram Moolenaare55b1c02020-06-21 15:52:59 +02001078 }
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02001079 semsg(_(e_cannot_declare_a_scope_variable), scope, name);
Bram Moolenaare55b1c02020-06-21 15:52:59 +02001080}
1081
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001082/*
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001083 * For one assignment figure out the type of destination. Return it in "dest".
1084 * When not recognized "dest" is not set.
Yegappan Lakshmanan64095532021-12-06 11:03:55 +00001085 * For an option "option_scope" is set.
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001086 * For a v:var "vimvaridx" is set.
1087 * "type" is set to the destination type if known, unchanted otherwise.
1088 * Return FAIL if an error message was given.
1089 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001090 int
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001091get_var_dest(
1092 char_u *name,
1093 assign_dest_T *dest,
1094 int cmdidx,
Yegappan Lakshmanan64095532021-12-06 11:03:55 +00001095 int *option_scope,
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001096 int *vimvaridx,
1097 type_T **type,
1098 cctx_T *cctx)
1099{
1100 char_u *p;
1101
1102 if (*name == '&')
1103 {
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001104 int cc;
1105 long numval;
1106 getoption_T opt_type;
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001107 int opt_p_flags;
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001108
1109 *dest = dest_option;
1110 if (cmdidx == CMD_final || cmdidx == CMD_const)
1111 {
1112 emsg(_(e_const_option));
1113 return FAIL;
1114 }
1115 p = name;
Yegappan Lakshmanan64095532021-12-06 11:03:55 +00001116 p = find_option_end(&p, option_scope);
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001117 if (p == NULL)
1118 {
1119 // cannot happen?
Bram Moolenaar108010a2021-06-27 22:03:33 +02001120 emsg(_(e_unexpected_characters_in_assignment));
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001121 return FAIL;
1122 }
1123 cc = *p;
1124 *p = NUL;
1125 opt_type = get_option_value(skip_option_env_lead(name),
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001126 &numval, NULL, &opt_p_flags, *option_scope);
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001127 *p = cc;
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001128 switch (opt_type)
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001129 {
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001130 case gov_unknown:
Bram Moolenaare1242042021-12-16 20:56:57 +00001131 semsg(_(e_unknown_option_str), name);
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001132 return FAIL;
1133 case gov_string:
1134 case gov_hidden_string:
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001135 if (opt_p_flags & P_FUNC)
1136 {
1137 // might be a Funcref, check the type later
1138 *type = &t_any;
1139 *dest = dest_func_option;
1140 }
1141 else
1142 {
1143 *type = &t_string;
1144 }
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001145 break;
1146 case gov_bool:
1147 case gov_hidden_bool:
1148 *type = &t_bool;
1149 break;
1150 case gov_number:
1151 case gov_hidden_number:
1152 *type = &t_number;
1153 break;
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001154 }
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001155 }
1156 else if (*name == '$')
1157 {
1158 *dest = dest_env;
1159 *type = &t_string;
1160 }
1161 else if (*name == '@')
1162 {
Bram Moolenaar13e45d12021-06-26 13:28:35 +02001163 if (name[1] != '@'
1164 && (!valid_yank_reg(name[1], FALSE) || name[1] == '.'))
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001165 {
1166 emsg_invreg(name[1]);
1167 return FAIL;
1168 }
1169 *dest = dest_reg;
Bram Moolenaar74f4a962021-06-17 21:03:07 +02001170 *type = name[1] == '#' ? &t_number_or_string : &t_string;
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001171 }
1172 else if (STRNCMP(name, "g:", 2) == 0)
1173 {
1174 *dest = dest_global;
1175 }
1176 else if (STRNCMP(name, "b:", 2) == 0)
1177 {
1178 *dest = dest_buffer;
1179 }
1180 else if (STRNCMP(name, "w:", 2) == 0)
1181 {
1182 *dest = dest_window;
1183 }
1184 else if (STRNCMP(name, "t:", 2) == 0)
1185 {
1186 *dest = dest_tab;
1187 }
1188 else if (STRNCMP(name, "v:", 2) == 0)
1189 {
1190 typval_T *vtv;
1191 int di_flags;
1192
1193 *vimvaridx = find_vim_var(name + 2, &di_flags);
1194 if (*vimvaridx < 0)
1195 {
1196 semsg(_(e_variable_not_found_str), name);
1197 return FAIL;
1198 }
1199 // We use the current value of "sandbox" here, is that OK?
1200 if (var_check_ro(di_flags, name, FALSE))
1201 return FAIL;
1202 *dest = dest_vimvar;
1203 vtv = get_vim_var_tv(*vimvaridx);
1204 *type = typval2type_vimvar(vtv, cctx->ctx_type_list);
1205 }
1206 return OK;
1207}
1208
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001209 static int
Bram Moolenaar752fc692021-01-04 21:57:11 +01001210is_decl_command(int cmdidx)
1211{
1212 return cmdidx == CMD_let || cmdidx == CMD_var
1213 || cmdidx == CMD_final || cmdidx == CMD_const;
1214}
1215
1216/*
1217 * Figure out the LHS type and other properties for an assignment or one item
1218 * of ":unlet" with an index.
1219 * Returns OK or FAIL.
1220 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001221 int
Bram Moolenaar752fc692021-01-04 21:57:11 +01001222compile_lhs(
1223 char_u *var_start,
1224 lhs_T *lhs,
1225 int cmdidx,
1226 int heredoc,
1227 int oplen,
1228 cctx_T *cctx)
1229{
1230 char_u *var_end;
1231 int is_decl = is_decl_command(cmdidx);
1232
1233 CLEAR_POINTER(lhs);
1234 lhs->lhs_dest = dest_local;
1235 lhs->lhs_vimvaridx = -1;
1236 lhs->lhs_scriptvar_idx = -1;
1237
1238 // "dest_end" is the end of the destination, including "[expr]" or
1239 // ".name".
1240 // "var_end" is the end of the variable/option/etc. name.
1241 lhs->lhs_dest_end = skip_var_one(var_start, FALSE);
1242 if (*var_start == '@')
1243 var_end = var_start + 2;
1244 else
1245 {
1246 // skip over the leading "&", "&l:", "&g:" and "$"
1247 var_end = skip_option_env_lead(var_start);
1248 var_end = to_name_end(var_end, TRUE);
1249 }
1250
1251 // "a: type" is declaring variable "a" with a type, not dict "a:".
1252 if (is_decl && lhs->lhs_dest_end == var_start + 2
1253 && lhs->lhs_dest_end[-1] == ':')
1254 --lhs->lhs_dest_end;
1255 if (is_decl && var_end == var_start + 2 && var_end[-1] == ':')
1256 --var_end;
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001257 lhs->lhs_end = lhs->lhs_dest_end;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001258
1259 // compute the length of the destination without "[expr]" or ".name"
1260 lhs->lhs_varlen = var_end - var_start;
Bram Moolenaar753bcf82021-04-21 14:24:24 +02001261 lhs->lhs_varlen_total = lhs->lhs_varlen;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001262 lhs->lhs_name = vim_strnsave(var_start, lhs->lhs_varlen);
1263 if (lhs->lhs_name == NULL)
1264 return FAIL;
Bram Moolenaar08251752021-01-11 21:20:18 +01001265
1266 if (lhs->lhs_dest_end > var_start + lhs->lhs_varlen)
1267 // Something follows after the variable: "var[idx]" or "var.key".
1268 lhs->lhs_has_index = TRUE;
1269
Bram Moolenaar752fc692021-01-04 21:57:11 +01001270 if (heredoc)
1271 lhs->lhs_type = &t_list_string;
1272 else
1273 lhs->lhs_type = &t_any;
1274
1275 if (cctx->ctx_skip != SKIP_YES)
1276 {
1277 int declare_error = FALSE;
1278
1279 if (get_var_dest(lhs->lhs_name, &lhs->lhs_dest, cmdidx,
1280 &lhs->lhs_opt_flags, &lhs->lhs_vimvaridx,
1281 &lhs->lhs_type, cctx) == FAIL)
1282 return FAIL;
Bram Moolenaard877a572021-04-01 19:42:48 +02001283 if (lhs->lhs_dest != dest_local
1284 && cmdidx != CMD_const && cmdidx != CMD_final)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001285 {
1286 // Specific kind of variable recognized.
1287 declare_error = is_decl;
1288 }
1289 else
1290 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01001291 // No specific kind of variable recognized, just a name.
Bram Moolenaard0edaf92021-05-28 21:06:08 +02001292 if (check_reserved_name(lhs->lhs_name) == FAIL)
1293 return FAIL;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001294
1295 if (lookup_local(var_start, lhs->lhs_varlen,
1296 &lhs->lhs_local_lvar, cctx) == OK)
1297 lhs->lhs_lvar = &lhs->lhs_local_lvar;
1298 else
1299 {
1300 CLEAR_FIELD(lhs->lhs_arg_lvar);
1301 if (arg_exists(var_start, lhs->lhs_varlen,
1302 &lhs->lhs_arg_lvar.lv_idx, &lhs->lhs_arg_lvar.lv_type,
1303 &lhs->lhs_arg_lvar.lv_from_outer, cctx) == OK)
1304 {
1305 if (is_decl)
1306 {
1307 semsg(_(e_str_is_used_as_argument), lhs->lhs_name);
1308 return FAIL;
1309 }
1310 lhs->lhs_lvar = &lhs->lhs_arg_lvar;
1311 }
1312 }
1313 if (lhs->lhs_lvar != NULL)
1314 {
1315 if (is_decl)
1316 {
1317 semsg(_(e_variable_already_declared), lhs->lhs_name);
1318 return FAIL;
1319 }
1320 }
1321 else
1322 {
1323 int script_namespace = lhs->lhs_varlen > 1
1324 && STRNCMP(var_start, "s:", 2) == 0;
1325 int script_var = (script_namespace
1326 ? script_var_exists(var_start + 2, lhs->lhs_varlen - 2,
Bram Moolenaar15e5e532021-04-07 21:21:13 +02001327 cctx)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001328 : script_var_exists(var_start, lhs->lhs_varlen,
Bram Moolenaar15e5e532021-04-07 21:21:13 +02001329 cctx)) == OK;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001330 imported_T *import =
1331 find_imported(var_start, lhs->lhs_varlen, cctx);
1332
1333 if (script_namespace || script_var || import != NULL)
1334 {
1335 char_u *rawname = lhs->lhs_name
1336 + (lhs->lhs_name[1] == ':' ? 2 : 0);
1337
1338 if (is_decl)
1339 {
1340 if (script_namespace)
1341 semsg(_(e_cannot_declare_script_variable_in_function),
1342 lhs->lhs_name);
1343 else
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001344 semsg(_(e_variable_already_declared_in_script_str),
Bram Moolenaar752fc692021-01-04 21:57:11 +01001345 lhs->lhs_name);
1346 return FAIL;
1347 }
1348 else if (cctx->ctx_ufunc->uf_script_ctx_version
1349 == SCRIPT_VERSION_VIM9
1350 && script_namespace
1351 && !script_var && import == NULL)
1352 {
1353 semsg(_(e_unknown_variable_str), lhs->lhs_name);
1354 return FAIL;
1355 }
1356
1357 lhs->lhs_dest = dest_script;
1358
1359 // existing script-local variables should have a type
1360 lhs->lhs_scriptvar_sid = current_sctx.sc_sid;
1361 if (import != NULL)
1362 lhs->lhs_scriptvar_sid = import->imp_sid;
1363 if (SCRIPT_ID_VALID(lhs->lhs_scriptvar_sid))
1364 {
Bram Moolenaar08251752021-01-11 21:20:18 +01001365 // Check writable only when no index follows.
Bram Moolenaar752fc692021-01-04 21:57:11 +01001366 lhs->lhs_scriptvar_idx = get_script_item_idx(
Bram Moolenaar08251752021-01-11 21:20:18 +01001367 lhs->lhs_scriptvar_sid, rawname,
1368 lhs->lhs_has_index ? ASSIGN_FINAL : ASSIGN_CONST,
1369 cctx);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001370 if (lhs->lhs_scriptvar_idx >= 0)
1371 {
1372 scriptitem_T *si = SCRIPT_ITEM(
1373 lhs->lhs_scriptvar_sid);
1374 svar_T *sv =
1375 ((svar_T *)si->sn_var_vals.ga_data)
1376 + lhs->lhs_scriptvar_idx;
1377 lhs->lhs_type = sv->sv_type;
1378 }
1379 }
1380 }
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001381 else if (check_defined(var_start, lhs->lhs_varlen, cctx, FALSE)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001382 == FAIL)
1383 return FAIL;
1384 }
1385 }
1386
1387 if (declare_error)
1388 {
1389 vim9_declare_error(lhs->lhs_name);
1390 return FAIL;
1391 }
1392 }
1393
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001394 // handle "a:name" as a name, not index "name" in "a"
Bram Moolenaar752fc692021-01-04 21:57:11 +01001395 if (lhs->lhs_varlen > 1 || var_start[lhs->lhs_varlen] != ':')
1396 var_end = lhs->lhs_dest_end;
1397
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001398 if (lhs->lhs_dest != dest_option && lhs->lhs_dest != dest_func_option)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001399 {
1400 if (is_decl && *var_end == ':')
1401 {
1402 char_u *p;
1403
1404 // parse optional type: "let var: type = expr"
1405 if (!VIM_ISWHITE(var_end[1]))
1406 {
Bram Moolenaarc3fc75d2021-02-07 15:28:09 +01001407 semsg(_(e_white_space_required_after_str_str), ":", var_end);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001408 return FAIL;
1409 }
1410 p = skipwhite(var_end + 1);
1411 lhs->lhs_type = parse_type(&p, cctx->ctx_type_list, TRUE);
1412 if (lhs->lhs_type == NULL)
1413 return FAIL;
1414 lhs->lhs_has_type = TRUE;
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001415 lhs->lhs_end = p;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001416 }
1417 else if (lhs->lhs_lvar != NULL)
1418 lhs->lhs_type = lhs->lhs_lvar->lv_type;
1419 }
1420
Bram Moolenaare42939a2021-04-05 17:11:17 +02001421 if (oplen == 3 && !heredoc
1422 && lhs->lhs_dest != dest_global
1423 && !lhs->lhs_has_index
1424 && lhs->lhs_type->tt_type != VAR_STRING
1425 && lhs->lhs_type->tt_type != VAR_ANY)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001426 {
1427 emsg(_(e_can_only_concatenate_to_string));
1428 return FAIL;
1429 }
1430
1431 if (lhs->lhs_lvar == NULL && lhs->lhs_dest == dest_local
1432 && cctx->ctx_skip != SKIP_YES)
1433 {
1434 if (oplen > 1 && !heredoc)
1435 {
1436 // +=, /=, etc. require an existing variable
1437 semsg(_(e_cannot_use_operator_on_new_variable), lhs->lhs_name);
1438 return FAIL;
1439 }
1440 if (!is_decl)
1441 {
1442 semsg(_(e_unknown_variable_str), lhs->lhs_name);
1443 return FAIL;
1444 }
1445
Bram Moolenaar3f327882021-03-17 20:56:38 +01001446 // Check the name is valid for a funcref.
Bram Moolenaar752fc692021-01-04 21:57:11 +01001447 if ((lhs->lhs_type->tt_type == VAR_FUNC
1448 || lhs->lhs_type->tt_type == VAR_PARTIAL)
Bram Moolenaar3f327882021-03-17 20:56:38 +01001449 && var_wrong_func_name(lhs->lhs_name, TRUE))
Bram Moolenaar752fc692021-01-04 21:57:11 +01001450 return FAIL;
Bram Moolenaar3f327882021-03-17 20:56:38 +01001451
1452 // New local variable.
Bram Moolenaar752fc692021-01-04 21:57:11 +01001453 lhs->lhs_lvar = reserve_local(cctx, var_start, lhs->lhs_varlen,
1454 cmdidx == CMD_final || cmdidx == CMD_const, lhs->lhs_type);
1455 if (lhs->lhs_lvar == NULL)
1456 return FAIL;
1457 lhs->lhs_new_local = TRUE;
1458 }
1459
1460 lhs->lhs_member_type = lhs->lhs_type;
Bram Moolenaar08251752021-01-11 21:20:18 +01001461 if (lhs->lhs_has_index)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001462 {
Bram Moolenaarc750d912021-11-29 22:02:12 +00001463 char_u *after = var_start + lhs->lhs_varlen;
1464 char_u *p;
1465
Bram Moolenaar752fc692021-01-04 21:57:11 +01001466 // Something follows after the variable: "var[idx]" or "var.key".
Bram Moolenaar752fc692021-01-04 21:57:11 +01001467 if (is_decl)
1468 {
1469 emsg(_(e_cannot_use_index_when_declaring_variable));
1470 return FAIL;
1471 }
1472
Bram Moolenaarc750d912021-11-29 22:02:12 +00001473 // Now: var_start[lhs->lhs_varlen] is '[' or '.'
1474 // Only the last index is used below, if there are others
1475 // before it generate code for the expression. Thus for
1476 // "ll[1][2]" the expression is "ll[1]" and "[2]" is the index.
1477 for (;;)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001478 {
Bram Moolenaarc750d912021-11-29 22:02:12 +00001479 p = skip_index(after);
1480 if (*p != '[' && *p != '.')
Bram Moolenaar752fc692021-01-04 21:57:11 +01001481 {
Bram Moolenaarc750d912021-11-29 22:02:12 +00001482 lhs->lhs_varlen_total = p - var_start;
1483 break;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001484 }
Bram Moolenaarc750d912021-11-29 22:02:12 +00001485 after = p;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001486 }
Bram Moolenaarc750d912021-11-29 22:02:12 +00001487 if (after > var_start + lhs->lhs_varlen)
1488 {
1489 lhs->lhs_varlen = after - var_start;
1490 lhs->lhs_dest = dest_expr;
1491 // We don't know the type before evaluating the expression,
1492 // use "any" until then.
1493 lhs->lhs_type = &t_any;
1494 }
1495
1496 if (lhs->lhs_type->tt_member == NULL)
1497 lhs->lhs_member_type = &t_any;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001498 else
Bram Moolenaarc750d912021-11-29 22:02:12 +00001499 lhs->lhs_member_type = lhs->lhs_type->tt_member;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001500 }
1501 return OK;
1502}
1503
1504/*
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001505 * Figure out the LHS and check a few errors.
1506 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001507 int
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001508compile_assign_lhs(
1509 char_u *var_start,
1510 lhs_T *lhs,
1511 int cmdidx,
1512 int is_decl,
1513 int heredoc,
1514 int oplen,
1515 cctx_T *cctx)
1516{
1517 if (compile_lhs(var_start, lhs, cmdidx, heredoc, oplen, cctx) == FAIL)
1518 return FAIL;
1519
1520 if (!lhs->lhs_has_index && lhs->lhs_lvar == &lhs->lhs_arg_lvar)
1521 {
1522 semsg(_(e_cannot_assign_to_argument), lhs->lhs_name);
1523 return FAIL;
1524 }
1525 if (!is_decl && lhs->lhs_lvar != NULL
1526 && lhs->lhs_lvar->lv_const && !lhs->lhs_has_index)
1527 {
1528 semsg(_(e_cannot_assign_to_constant), lhs->lhs_name);
1529 return FAIL;
1530 }
1531 return OK;
1532}
1533
1534/*
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02001535 * Return TRUE if "lhs" has a range index: "[expr : expr]".
1536 */
1537 static int
1538has_list_index(char_u *idx_start, cctx_T *cctx)
1539{
1540 char_u *p = idx_start;
1541 int save_skip;
1542
1543 if (*p != '[')
1544 return FALSE;
1545
1546 p = skipwhite(p + 1);
1547 if (*p == ':')
1548 return TRUE;
1549
1550 save_skip = cctx->ctx_skip;
1551 cctx->ctx_skip = SKIP_YES;
1552 (void)compile_expr0(&p, cctx);
1553 cctx->ctx_skip = save_skip;
1554 return *skipwhite(p) == ':';
1555}
1556
1557/*
Bram Moolenaare42939a2021-04-05 17:11:17 +02001558 * For an assignment with an index, compile the "idx" in "var[idx]" or "key" in
1559 * "var.key".
Bram Moolenaar752fc692021-01-04 21:57:11 +01001560 */
1561 static int
Bram Moolenaare42939a2021-04-05 17:11:17 +02001562compile_assign_index(
Bram Moolenaar752fc692021-01-04 21:57:11 +01001563 char_u *var_start,
1564 lhs_T *lhs,
Bram Moolenaare42939a2021-04-05 17:11:17 +02001565 int *range,
Bram Moolenaar752fc692021-01-04 21:57:11 +01001566 cctx_T *cctx)
1567{
Bram Moolenaar752fc692021-01-04 21:57:11 +01001568 size_t varlen = lhs->lhs_varlen;
Bram Moolenaare42939a2021-04-05 17:11:17 +02001569 char_u *p;
1570 int r = OK;
Bram Moolenaar68452172021-04-12 21:21:02 +02001571 int need_white_before = TRUE;
1572 int empty_second;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001573
Bram Moolenaar752fc692021-01-04 21:57:11 +01001574 p = var_start + varlen;
1575 if (*p == '[')
1576 {
1577 p = skipwhite(p + 1);
Bram Moolenaar68452172021-04-12 21:21:02 +02001578 if (*p == ':')
1579 {
1580 // empty first index, push zero
1581 r = generate_PUSHNR(cctx, 0);
1582 need_white_before = FALSE;
1583 }
1584 else
1585 r = compile_expr0(&p, cctx);
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001586
1587 if (r == OK && *skipwhite(p) == ':')
1588 {
1589 // unlet var[idx : idx]
Bram Moolenaar68452172021-04-12 21:21:02 +02001590 // blob[idx : idx] = value
Bram Moolenaare42939a2021-04-05 17:11:17 +02001591 *range = TRUE;
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001592 p = skipwhite(p);
Bram Moolenaar68452172021-04-12 21:21:02 +02001593 empty_second = *skipwhite(p + 1) == ']';
1594 if ((need_white_before && !IS_WHITE_OR_NUL(p[-1]))
1595 || (!empty_second && !IS_WHITE_OR_NUL(p[1])))
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001596 {
1597 semsg(_(e_white_space_required_before_and_after_str_at_str),
1598 ":", p);
1599 return FAIL;
1600 }
1601 p = skipwhite(p + 1);
Bram Moolenaar68452172021-04-12 21:21:02 +02001602 if (*p == ']')
1603 // empty second index, push "none"
1604 r = generate_PUSHSPEC(cctx, VVAL_NONE);
1605 else
1606 r = compile_expr0(&p, cctx);
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001607 }
1608
Bram Moolenaar752fc692021-01-04 21:57:11 +01001609 if (r == OK && *skipwhite(p) != ']')
1610 {
1611 // this should not happen
Bram Moolenaare1242042021-12-16 20:56:57 +00001612 emsg(_(e_missing_closing_square_brace));
Bram Moolenaar752fc692021-01-04 21:57:11 +01001613 r = FAIL;
1614 }
1615 }
1616 else // if (*p == '.')
1617 {
1618 char_u *key_end = to_name_end(p + 1, TRUE);
1619 char_u *key = vim_strnsave(p + 1, key_end - p - 1);
1620
Zdenek Dohnal9fe17d42021-08-04 22:30:52 +02001621 r = generate_PUSHS(cctx, &key);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001622 }
Bram Moolenaare42939a2021-04-05 17:11:17 +02001623 return r;
1624}
1625
1626/*
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001627 * For a LHS with an index, load the variable to be indexed.
1628 */
1629 static int
1630compile_load_lhs(
1631 lhs_T *lhs,
1632 char_u *var_start,
1633 type_T *rhs_type,
1634 cctx_T *cctx)
1635{
1636 if (lhs->lhs_dest == dest_expr)
1637 {
1638 size_t varlen = lhs->lhs_varlen;
1639 int c = var_start[varlen];
Bram Moolenaare97976b2021-08-01 13:17:17 +02001640 int lines_len = cctx->ctx_ufunc->uf_lines.ga_len;
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001641 char_u *p = var_start;
1642 garray_T *stack = &cctx->ctx_type_stack;
Bram Moolenaare97976b2021-08-01 13:17:17 +02001643 int res;
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001644
Bram Moolenaare97976b2021-08-01 13:17:17 +02001645 // Evaluate "ll[expr]" of "ll[expr][idx]". End the line with a NUL and
1646 // limit the lines array length to avoid skipping to a following line.
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001647 var_start[varlen] = NUL;
Bram Moolenaare97976b2021-08-01 13:17:17 +02001648 cctx->ctx_ufunc->uf_lines.ga_len = cctx->ctx_lnum + 1;
1649 res = compile_expr0(&p, cctx);
1650 var_start[varlen] = c;
1651 cctx->ctx_ufunc->uf_lines.ga_len = lines_len;
1652 if (res == FAIL || p != var_start + varlen)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001653 {
1654 // this should not happen
Bram Moolenaare97976b2021-08-01 13:17:17 +02001655 if (res != FAIL)
Bram Moolenaare1242042021-12-16 20:56:57 +00001656 emsg(_(e_missing_closing_square_brace));
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001657 return FAIL;
1658 }
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001659
1660 lhs->lhs_type = stack->ga_len == 0 ? &t_void
1661 : ((type_T **)stack->ga_data)[stack->ga_len - 1];
1662 // now we can properly check the type
1663 if (rhs_type != NULL && lhs->lhs_type->tt_member != NULL
1664 && rhs_type != &t_void
1665 && need_type(rhs_type, lhs->lhs_type->tt_member, -2, 0, cctx,
1666 FALSE, FALSE) == FAIL)
1667 return FAIL;
1668 }
1669 else
1670 generate_loadvar(cctx, lhs->lhs_dest, lhs->lhs_name,
1671 lhs->lhs_lvar, lhs->lhs_type);
1672 return OK;
1673}
1674
1675/*
Bram Moolenaara369c3d2021-04-21 16:00:10 +02001676 * Produce code for loading "lhs" and also take care of an index.
1677 * Return OK/FAIL.
1678 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001679 int
Bram Moolenaara369c3d2021-04-21 16:00:10 +02001680compile_load_lhs_with_index(lhs_T *lhs, char_u *var_start, cctx_T *cctx)
1681{
1682 compile_load_lhs(lhs, var_start, NULL, cctx);
1683
1684 if (lhs->lhs_has_index)
1685 {
1686 int range = FALSE;
1687
1688 // Get member from list or dict. First compile the
1689 // index value.
1690 if (compile_assign_index(var_start, lhs, &range, cctx) == FAIL)
1691 return FAIL;
1692 if (range)
1693 {
1694 semsg(_(e_cannot_use_range_with_assignment_operator_str),
1695 var_start);
1696 return FAIL;
1697 }
1698
1699 // Get the member.
Bram Moolenaarb1b6f4d2021-09-13 18:25:54 +02001700 if (compile_member(FALSE, NULL, cctx) == FAIL)
Bram Moolenaara369c3d2021-04-21 16:00:10 +02001701 return FAIL;
1702 }
1703 return OK;
1704}
1705
1706/*
Bram Moolenaare42939a2021-04-05 17:11:17 +02001707 * Assignment to a list or dict member, or ":unlet" for the item, using the
1708 * information in "lhs".
1709 * Returns OK or FAIL.
1710 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001711 int
Bram Moolenaare42939a2021-04-05 17:11:17 +02001712compile_assign_unlet(
1713 char_u *var_start,
1714 lhs_T *lhs,
1715 int is_assign,
1716 type_T *rhs_type,
1717 cctx_T *cctx)
1718{
Bram Moolenaare42939a2021-04-05 17:11:17 +02001719 vartype_T dest_type;
Bram Moolenaare42939a2021-04-05 17:11:17 +02001720 garray_T *stack = &cctx->ctx_type_stack;
1721 int range = FALSE;
1722
Bram Moolenaar68452172021-04-12 21:21:02 +02001723 if (compile_assign_index(var_start, lhs, &range, cctx) == FAIL)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001724 return FAIL;
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02001725 if (is_assign && range
1726 && lhs->lhs_type->tt_type != VAR_LIST
1727 && lhs->lhs_type != &t_blob
1728 && lhs->lhs_type != &t_any)
Bram Moolenaar68452172021-04-12 21:21:02 +02001729 {
1730 semsg(_(e_cannot_use_range_with_assignment_str), var_start);
1731 return FAIL;
1732 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001733
1734 if (lhs->lhs_type == &t_any)
1735 {
1736 // Index on variable of unknown type: check at runtime.
1737 dest_type = VAR_ANY;
1738 }
1739 else
1740 {
1741 dest_type = lhs->lhs_type->tt_type;
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001742 if (dest_type == VAR_DICT && range)
1743 {
1744 emsg(e_cannot_use_range_with_dictionary);
1745 return FAIL;
1746 }
Bram Moolenaar5fa9b242021-06-04 21:00:32 +02001747 if (dest_type == VAR_DICT
1748 && may_generate_2STRING(-1, FALSE, cctx) == FAIL)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001749 return FAIL;
Bram Moolenaar51e93322021-04-17 20:44:56 +02001750 if (dest_type == VAR_LIST || dest_type == VAR_BLOB)
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001751 {
Bram Moolenaar51e93322021-04-17 20:44:56 +02001752 type_T *type;
1753
1754 if (range)
1755 {
1756 type = ((type_T **)stack->ga_data)[stack->ga_len - 2];
1757 if (need_type(type, &t_number,
1758 -1, 0, cctx, FALSE, FALSE) == FAIL)
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001759 return FAIL;
Bram Moolenaar51e93322021-04-17 20:44:56 +02001760 }
1761 type = ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaar63cb6562021-07-20 22:21:59 +02001762 if ((dest_type != VAR_BLOB && type != &t_special)
Bram Moolenaar51e93322021-04-17 20:44:56 +02001763 && need_type(type, &t_number,
1764 -1, 0, cctx, FALSE, FALSE) == FAIL)
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001765 return FAIL;
1766 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001767 }
1768
1769 // Load the dict or list. On the stack we then have:
1770 // - value (for assignment, not for :unlet)
1771 // - index
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001772 // - for [a : b] second index
Bram Moolenaar752fc692021-01-04 21:57:11 +01001773 // - variable
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001774 if (compile_load_lhs(lhs, var_start, rhs_type, cctx) == FAIL)
1775 return FAIL;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001776
Bram Moolenaar68452172021-04-12 21:21:02 +02001777 if (dest_type == VAR_LIST || dest_type == VAR_DICT
1778 || dest_type == VAR_BLOB || dest_type == VAR_ANY)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001779 {
1780 if (is_assign)
1781 {
Bram Moolenaar68452172021-04-12 21:21:02 +02001782 if (range)
1783 {
1784 if (generate_instr_drop(cctx, ISN_STORERANGE, 4) == NULL)
1785 return FAIL;
1786 }
1787 else
1788 {
1789 isn_T *isn = generate_instr_drop(cctx, ISN_STOREINDEX, 3);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001790
Bram Moolenaar68452172021-04-12 21:21:02 +02001791 if (isn == NULL)
1792 return FAIL;
1793 isn->isn_arg.vartype = dest_type;
1794 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001795 }
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001796 else if (range)
1797 {
1798 if (generate_instr_drop(cctx, ISN_UNLETRANGE, 3) == NULL)
1799 return FAIL;
1800 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001801 else
1802 {
1803 if (generate_instr_drop(cctx, ISN_UNLETINDEX, 2) == NULL)
1804 return FAIL;
1805 }
1806 }
1807 else
1808 {
1809 emsg(_(e_indexable_type_required));
1810 return FAIL;
1811 }
1812
1813 return OK;
1814}
1815
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001816/*
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001817 * Compile declaration and assignment:
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001818 * "let name"
1819 * "var name = expr"
1820 * "final name = expr"
1821 * "const name = expr"
1822 * "name = expr"
1823 * "arg" points to "name".
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001824 * "++arg" and "--arg"
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001825 * Return NULL for an error.
1826 * Return "arg" if it does not look like a variable list.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001827 */
1828 static char_u *
1829compile_assignment(char_u *arg, exarg_T *eap, cmdidx_T cmdidx, cctx_T *cctx)
1830{
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001831 char_u *var_start;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001832 char_u *p;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001833 char_u *end = arg;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001834 char_u *ret = NULL;
1835 int var_count = 0;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001836 int var_idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001837 int semicolon = 0;
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +02001838 int did_generate_slice = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001839 garray_T *instr = &cctx->ctx_instr;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001840 garray_T *stack = &cctx->ctx_type_stack;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001841 char_u *op;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001842 int oplen = 0;
1843 int heredoc = FALSE;
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001844 int incdec = FALSE;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001845 type_T *rhs_type = &t_any;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001846 char_u *sp;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001847 int is_decl = is_decl_command(cmdidx);
1848 lhs_T lhs;
Bram Moolenaar77709b12021-04-03 21:01:01 +02001849 long start_lnum = SOURCING_LNUM;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001850
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001851 // Skip over the "var" or "[var, var]" to get to any "=".
1852 p = skip_var_list(arg, TRUE, &var_count, &semicolon, TRUE);
1853 if (p == NULL)
1854 return *arg == '[' ? arg : NULL;
1855
Bram Moolenaar752fc692021-01-04 21:57:11 +01001856 lhs.lhs_name = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001857
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001858 sp = p;
1859 p = skipwhite(p);
1860 op = p;
1861 oplen = assignment_len(p, &heredoc);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001862
1863 if (var_count > 0 && oplen == 0)
1864 // can be something like "[1, 2]->func()"
1865 return arg;
1866
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001867 if (oplen > 0 && (!VIM_ISWHITE(*sp) || !IS_WHITE_OR_NUL(op[oplen])))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001868 {
Bram Moolenaarbb1b5e22020-08-05 10:53:21 +02001869 error_white_both(op, oplen);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001870 return NULL;
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001871 }
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001872 if (eap->cmdidx == CMD_increment || eap->cmdidx == CMD_decrement)
1873 {
Bram Moolenaar22480d12021-06-25 21:31:09 +02001874 if (VIM_ISWHITE(eap->cmd[2]))
1875 {
1876 semsg(_(e_no_white_space_allowed_after_str_str),
1877 eap->cmdidx == CMD_increment ? "++" : "--", eap->cmd);
1878 return NULL;
1879 }
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001880 op = (char_u *)(eap->cmdidx == CMD_increment ? "+=" : "-=");
1881 oplen = 2;
1882 incdec = TRUE;
1883 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001884
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001885 if (heredoc)
1886 {
1887 list_T *l;
1888 listitem_T *li;
1889
1890 // [let] varname =<< [trim] {end}
Bram Moolenaar04b12692020-05-04 23:24:44 +02001891 eap->getline = exarg_getline;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001892 eap->cookie = cctx;
Bram Moolenaar6c2b7b82020-04-14 20:15:49 +02001893 l = heredoc_get(eap, op + 3, FALSE);
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001894 if (l == NULL)
1895 return NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001896
Bram Moolenaar078269b2020-09-21 20:35:55 +02001897 if (cctx->ctx_skip != SKIP_YES)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001898 {
Bram Moolenaar078269b2020-09-21 20:35:55 +02001899 // Push each line and the create the list.
1900 FOR_ALL_LIST_ITEMS(l, li)
1901 {
Zdenek Dohnal9fe17d42021-08-04 22:30:52 +02001902 generate_PUSHS(cctx, &li->li_tv.vval.v_string);
Bram Moolenaar078269b2020-09-21 20:35:55 +02001903 li->li_tv.vval.v_string = NULL;
1904 }
1905 generate_NEWLIST(cctx, l->lv_len);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001906 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001907 list_free(l);
1908 p += STRLEN(p);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001909 end = p;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001910 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001911 else if (var_count > 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001912 {
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001913 char_u *wp;
1914
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001915 // for "[var, var] = expr" evaluate the expression here, loop over the
1916 // list of variables below.
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001917 // A line break may follow the "=".
Bram Moolenaard25ec2c2020-03-30 21:05:45 +02001918
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001919 wp = op + oplen;
Bram Moolenaar8ff16e02020-12-07 21:49:52 +01001920 if (may_get_next_line_error(wp, &p, cctx) == FAIL)
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001921 return FAIL;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001922 if (compile_expr0(&p, cctx) == FAIL)
1923 return NULL;
1924 end = p;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001925
Bram Moolenaar9b68c822020-06-18 19:31:08 +02001926 if (cctx->ctx_skip != SKIP_YES)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001927 {
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001928 type_T *stacktype;
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +00001929 int needed_list_len;
1930 int did_check = FALSE;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001931
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001932 stacktype = stack->ga_len == 0 ? &t_void
1933 : ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001934 if (stacktype->tt_type == VAR_VOID)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001935 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02001936 emsg(_(e_cannot_use_void_value));
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001937 goto theend;
1938 }
Bram Moolenaar351ead02021-01-16 16:07:01 +01001939 if (need_type(stacktype, &t_list_any, -1, 0, cctx,
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001940 FALSE, FALSE) == FAIL)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001941 goto theend;
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +00001942 // If a constant list was used we can check the length right here.
1943 needed_list_len = semicolon ? var_count - 1 : var_count;
1944 if (instr->ga_len > 0)
1945 {
1946 isn_T *isn = ((isn_T *)instr->ga_data) + instr->ga_len - 1;
1947
1948 if (isn->isn_type == ISN_NEWLIST)
1949 {
1950 did_check = TRUE;
1951 if (semicolon ? isn->isn_arg.number < needed_list_len
1952 : isn->isn_arg.number != needed_list_len)
1953 {
1954 semsg(_(e_expected_nr_items_but_got_nr),
1955 needed_list_len, isn->isn_arg.number);
1956 goto theend;
1957 }
1958 }
1959 }
1960 if (!did_check)
1961 generate_CHECKLEN(cctx, needed_list_len, semicolon);
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001962 if (stacktype->tt_member != NULL)
1963 rhs_type = stacktype->tt_member;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001964 }
1965 }
1966
1967 /*
1968 * Loop over variables in "[var, var] = expr".
1969 * For "var = expr" and "let var: type" this is done only once.
1970 */
1971 if (var_count > 0)
1972 var_start = skipwhite(arg + 1); // skip over the "["
1973 else
1974 var_start = arg;
1975 for (var_idx = 0; var_idx == 0 || var_idx < var_count; var_idx++)
1976 {
Bram Moolenaar6977dba2021-07-04 22:48:12 +02001977 int instr_count = -1;
1978 int save_lnum;
Bram Moolenaar5cd64792021-12-25 18:23:24 +00001979 int skip_store = FALSE;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001980
Bram Moolenaarf93bbd02021-04-10 22:35:43 +02001981 if (var_start[0] == '_' && !eval_isnamec(var_start[1]))
1982 {
1983 // Ignore underscore in "[a, _, b] = list".
1984 if (var_count > 0)
1985 {
1986 var_start = skipwhite(var_start + 2);
1987 continue;
1988 }
1989 emsg(_(e_cannot_use_underscore_here));
1990 goto theend;
1991 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001992 vim_free(lhs.lhs_name);
1993
1994 /*
1995 * Figure out the LHS type and other properties.
1996 */
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001997 if (compile_assign_lhs(var_start, &lhs, cmdidx,
1998 is_decl, heredoc, oplen, cctx) == FAIL)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001999 goto theend;
Bram Moolenaar81530e32021-07-28 21:25:49 +02002000 if (heredoc)
2001 {
2002 SOURCING_LNUM = start_lnum;
2003 if (lhs.lhs_has_type
2004 && need_type(&t_list_string, lhs.lhs_type,
2005 -1, 0, cctx, FALSE, FALSE) == FAIL)
2006 goto theend;
2007 }
2008 else
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002009 {
Bram Moolenaar9b68c822020-06-18 19:31:08 +02002010 if (cctx->ctx_skip == SKIP_YES)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002011 {
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002012 if (oplen > 0 && var_count == 0)
2013 {
2014 // skip over the "=" and the expression
2015 p = skipwhite(op + oplen);
Bram Moolenaar169502f2021-04-21 12:19:35 +02002016 (void)compile_expr0(&p, cctx);
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002017 }
2018 }
2019 else if (oplen > 0)
2020 {
Bram Moolenaar334a8b42020-10-19 16:07:42 +02002021 int is_const = FALSE;
Bram Moolenaar7f764942020-12-02 15:11:18 +01002022 char_u *wp;
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002023
Bram Moolenaar035bd1c2021-06-21 19:44:11 +02002024 // for "+=", "*=", "..=" etc. first load the current value
2025 if (*op != '='
2026 && compile_load_lhs_with_index(&lhs, var_start,
2027 cctx) == FAIL)
2028 goto theend;
2029
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002030 // For "var = expr" evaluate the expression.
2031 if (var_count == 0)
2032 {
2033 int r;
2034
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002035 // Compile the expression.
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002036 instr_count = instr->ga_len;
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002037 if (incdec)
Bram Moolenaar21e51222020-12-04 12:43:29 +01002038 {
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002039 r = generate_PUSHNR(cctx, 1);
2040 }
2041 else
2042 {
2043 // Temporarily hide the new local variable here, it is
2044 // not available to this expression.
2045 if (lhs.lhs_new_local)
2046 --cctx->ctx_locals.ga_len;
2047 wp = op + oplen;
2048 if (may_get_next_line_error(wp, &p, cctx) == FAIL)
2049 {
2050 if (lhs.lhs_new_local)
2051 ++cctx->ctx_locals.ga_len;
2052 goto theend;
2053 }
2054 r = compile_expr0_ext(&p, cctx, &is_const);
Bram Moolenaar752fc692021-01-04 21:57:11 +01002055 if (lhs.lhs_new_local)
Bram Moolenaar21e51222020-12-04 12:43:29 +01002056 ++cctx->ctx_locals.ga_len;
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002057 if (r == FAIL)
2058 goto theend;
Bram Moolenaar21e51222020-12-04 12:43:29 +01002059 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002060 }
Bram Moolenaar9af78762020-06-16 11:34:42 +02002061 else if (semicolon && var_idx == var_count - 1)
2062 {
2063 // For "[var; var] = expr" get the rest of the list
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +02002064 did_generate_slice = TRUE;
Bram Moolenaar9af78762020-06-16 11:34:42 +02002065 if (generate_SLICE(cctx, var_count - 1) == FAIL)
2066 goto theend;
2067 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002068 else
2069 {
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002070 // For "[var, var] = expr" get the "var_idx" item from the
2071 // list.
Bram Moolenaar035bd1c2021-06-21 19:44:11 +02002072 if (generate_GETITEM(cctx, var_idx, *op != '=') == FAIL)
Bram Moolenaar7f764942020-12-02 15:11:18 +01002073 goto theend;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002074 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002075
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002076 rhs_type = stack->ga_len == 0 ? &t_void
Bram Moolenaar6802cce2020-07-19 15:49:49 +02002077 : ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaar752fc692021-01-04 21:57:11 +01002078 if (lhs.lhs_lvar != NULL && (is_decl || !lhs.lhs_has_type))
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002079 {
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002080 if ((rhs_type->tt_type == VAR_FUNC
2081 || rhs_type->tt_type == VAR_PARTIAL)
Bram Moolenaar3f327882021-03-17 20:56:38 +01002082 && !lhs.lhs_has_index
Bram Moolenaar752fc692021-01-04 21:57:11 +01002083 && var_wrong_func_name(lhs.lhs_name, TRUE))
Bram Moolenaar0f769812020-09-12 18:32:34 +02002084 goto theend;
2085
Bram Moolenaar752fc692021-01-04 21:57:11 +01002086 if (lhs.lhs_new_local && !lhs.lhs_has_type)
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002087 {
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002088 if (rhs_type->tt_type == VAR_VOID)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002089 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02002090 emsg(_(e_cannot_use_void_value));
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002091 goto theend;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002092 }
2093 else
2094 {
Bram Moolenaar04bdd572020-09-23 13:25:32 +02002095 // An empty list or dict has a &t_unknown member,
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002096 // for a variable that implies &t_any.
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002097 if (rhs_type == &t_list_empty)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002098 lhs.lhs_lvar->lv_type = &t_list_any;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002099 else if (rhs_type == &t_dict_empty)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002100 lhs.lhs_lvar->lv_type = &t_dict_any;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002101 else if (rhs_type == &t_unknown)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002102 lhs.lhs_lvar->lv_type = &t_any;
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002103 else
Bram Moolenaar752fc692021-01-04 21:57:11 +01002104 lhs.lhs_lvar->lv_type = rhs_type;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002105 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002106 }
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002107 else if (*op == '=')
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002108 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002109 type_T *use_type = lhs.lhs_lvar->lv_type;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +02002110 where_T where = WHERE_INIT;
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002111
Bram Moolenaar77709b12021-04-03 21:01:01 +02002112 // Without operator check type here, otherwise below.
2113 // Use the line number of the assignment.
2114 SOURCING_LNUM = start_lnum;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +02002115 where.wt_index = var_count > 0 ? var_idx + 1 : 0;
2116 where.wt_variable = var_count > 0;
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02002117 // If assigning to a list or dict member, use the
2118 // member type. Not for "list[:] =".
2119 if (lhs.lhs_has_index
2120 && !has_list_index(var_start + lhs.lhs_varlen,
2121 cctx))
Bram Moolenaar752fc692021-01-04 21:57:11 +01002122 use_type = lhs.lhs_member_type;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +02002123 if (need_type_where(rhs_type, use_type, -1, where,
2124 cctx, FALSE, is_const) == FAIL)
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002125 goto theend;
2126 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002127 }
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002128 else
2129 {
2130 type_T *lhs_type = lhs.lhs_member_type;
2131
2132 // Special case: assigning to @# can use a number or a
2133 // string.
Bram Moolenaaraf647e72021-08-05 19:01:17 +02002134 // Also: can assign a number to a float.
2135 if ((lhs_type == &t_number_or_string
2136 || lhs_type == &t_float)
2137 && rhs_type->tt_type == VAR_NUMBER)
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002138 lhs_type = &t_number;
2139 if (*p != '=' && need_type(rhs_type, lhs_type,
Bram Moolenaar351ead02021-01-16 16:07:01 +01002140 -1, 0, cctx, FALSE, FALSE) == FAIL)
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002141 goto theend;
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002142 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002143 }
Bram Moolenaar30fd8202020-09-26 15:09:30 +02002144 else if (cmdidx == CMD_final)
2145 {
2146 emsg(_(e_final_requires_a_value));
2147 goto theend;
2148 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002149 else if (cmdidx == CMD_const)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002150 {
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02002151 emsg(_(e_const_requires_a_value));
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002152 goto theend;
2153 }
Bram Moolenaardcb53be2021-12-09 14:23:43 +00002154 else if (!lhs.lhs_has_type || lhs.lhs_dest == dest_option
2155 || lhs.lhs_dest == dest_func_option)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002156 {
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02002157 emsg(_(e_type_or_initialization_required));
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002158 goto theend;
2159 }
2160 else
2161 {
2162 // variables are always initialized
Bram Moolenaar35578162021-08-02 19:10:38 +02002163 if (GA_GROW_FAILS(instr, 1))
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002164 goto theend;
Bram Moolenaar752fc692021-01-04 21:57:11 +01002165 switch (lhs.lhs_member_type->tt_type)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002166 {
2167 case VAR_BOOL:
2168 generate_PUSHBOOL(cctx, VVAL_FALSE);
2169 break;
2170 case VAR_FLOAT:
2171#ifdef FEAT_FLOAT
2172 generate_PUSHF(cctx, 0.0);
2173#endif
2174 break;
2175 case VAR_STRING:
2176 generate_PUSHS(cctx, NULL);
2177 break;
2178 case VAR_BLOB:
Bram Moolenaar80b0e5e2020-10-19 20:45:36 +02002179 generate_PUSHBLOB(cctx, blob_alloc());
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002180 break;
2181 case VAR_FUNC:
2182 generate_PUSHFUNC(cctx, NULL, &t_func_void);
2183 break;
2184 case VAR_LIST:
2185 generate_NEWLIST(cctx, 0);
2186 break;
2187 case VAR_DICT:
2188 generate_NEWDICT(cctx, 0);
2189 break;
2190 case VAR_JOB:
2191 generate_PUSHJOB(cctx, NULL);
2192 break;
2193 case VAR_CHANNEL:
2194 generate_PUSHCHANNEL(cctx, NULL);
2195 break;
2196 case VAR_NUMBER:
2197 case VAR_UNKNOWN:
2198 case VAR_ANY:
2199 case VAR_PARTIAL:
2200 case VAR_VOID:
Bram Moolenaarf18332f2021-05-07 17:55:55 +02002201 case VAR_INSTR:
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002202 case VAR_SPECIAL: // cannot happen
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002203 // This is skipped for local variables, they are
2204 // always initialized to zero.
2205 if (lhs.lhs_dest == dest_local)
2206 skip_store = TRUE;
2207 else
2208 generate_PUSHNR(cctx, 0);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002209 break;
2210 }
2211 }
2212 if (var_count == 0)
2213 end = p;
2214 }
2215
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002216 // no need to parse more when skipping
Bram Moolenaar9b68c822020-06-18 19:31:08 +02002217 if (cctx->ctx_skip == SKIP_YES)
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002218 break;
2219
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002220 if (oplen > 0 && *op != '=')
2221 {
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002222 type_T *expected;
Bram Moolenaarc3160722021-08-02 21:12:05 +02002223 type_T *stacktype = NULL;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002224
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002225 if (*op == '.')
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002226 {
2227 if (may_generate_2STRING(-1, FALSE, cctx) == FAIL)
2228 goto theend;
2229 }
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002230 else
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002231 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002232 expected = lhs.lhs_member_type;
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002233 stacktype = ((type_T **)stack->ga_data)[stack->ga_len - 1];
2234 if (
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002235#ifdef FEAT_FLOAT
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002236 // If variable is float operation with number is OK.
Bram Moolenaar7bf9a072021-08-02 21:55:15 +02002237 !(expected == &t_float && (stacktype == &t_number
2238 || stacktype == &t_number_bool)) &&
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002239#endif
Bram Moolenaar351ead02021-01-16 16:07:01 +01002240 need_type(stacktype, expected, -1, 0, cctx,
Bram Moolenaar334a8b42020-10-19 16:07:42 +02002241 FALSE, FALSE) == FAIL)
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002242 goto theend;
2243 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002244
2245 if (*op == '.')
Bram Moolenaardd29f1b2020-08-07 20:46:20 +02002246 {
2247 if (generate_instr_drop(cctx, ISN_CONCAT, 1) == NULL)
2248 goto theend;
2249 }
2250 else if (*op == '+')
2251 {
2252 if (generate_add_instr(cctx,
Bram Moolenaar752fc692021-01-04 21:57:11 +01002253 operator_type(lhs.lhs_member_type, stacktype),
Bram Moolenaar07802042021-09-09 23:01:14 +02002254 lhs.lhs_member_type, stacktype,
2255 EXPR_APPEND) == FAIL)
Bram Moolenaardd29f1b2020-08-07 20:46:20 +02002256 goto theend;
2257 }
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002258 else if (generate_two_op(cctx, op) == FAIL)
2259 goto theend;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002260 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002261
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002262 // Use the line number of the assignment for store instruction.
2263 save_lnum = cctx->ctx_lnum;
2264 cctx->ctx_lnum = start_lnum - 1;
2265
Bram Moolenaar752fc692021-01-04 21:57:11 +01002266 if (lhs.lhs_has_index)
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002267 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002268 // Use the info in "lhs" to store the value at the index in the
2269 // list or dict.
2270 if (compile_assign_unlet(var_start, &lhs, TRUE, rhs_type, cctx)
2271 == FAIL)
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002272 {
2273 cctx->ctx_lnum = save_lnum;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002274 goto theend;
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002275 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002276 }
2277 else
2278 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002279 if (is_decl && cmdidx == CMD_const && (lhs.lhs_dest == dest_script
Bram Moolenaard877a572021-04-01 19:42:48 +02002280 || lhs.lhs_dest == dest_global
Bram Moolenaar752fc692021-01-04 21:57:11 +01002281 || lhs.lhs_dest == dest_local))
Bram Moolenaar30fd8202020-09-26 15:09:30 +02002282 // ":const var": lock the value, but not referenced variables
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +02002283 generate_LOCKCONST(cctx);
2284
Bram Moolenaaraa210a32021-01-02 15:41:03 +01002285 if (is_decl
Bram Moolenaar752fc692021-01-04 21:57:11 +01002286 && (lhs.lhs_type->tt_type == VAR_DICT
2287 || lhs.lhs_type->tt_type == VAR_LIST)
2288 && lhs.lhs_type->tt_member != NULL
Bram Moolenaar6e48b842021-08-10 22:52:02 +02002289 && !(lhs.lhs_type->tt_member == &t_any
2290 && oplen > 0
2291 && rhs_type != NULL
2292 && rhs_type->tt_type == lhs.lhs_type->tt_type
2293 && rhs_type->tt_member != &t_unknown)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002294 && lhs.lhs_type->tt_member != &t_unknown)
Bram Moolenaaraa210a32021-01-02 15:41:03 +01002295 // Set the type in the list or dict, so that it can be checked,
Bram Moolenaar6e48b842021-08-10 22:52:02 +02002296 // also in legacy script. Not for "list<any> = val", then the
2297 // type of "val" is used.
Bram Moolenaar752fc692021-01-04 21:57:11 +01002298 generate_SETTYPE(cctx, lhs.lhs_type);
Bram Moolenaaraa210a32021-01-02 15:41:03 +01002299
Bram Moolenaar5cd64792021-12-25 18:23:24 +00002300 if (!skip_store && generate_store_lhs(cctx, &lhs,
2301 instr_count, is_decl) == FAIL)
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002302 {
2303 cctx->ctx_lnum = save_lnum;
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02002304 goto theend;
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002305 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002306 }
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002307 cctx->ctx_lnum = save_lnum;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002308
2309 if (var_idx + 1 < var_count)
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00002310 var_start = skipwhite(lhs.lhs_end + 1);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002311 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002312
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +02002313 // For "[var, var] = expr" drop the "expr" value.
2314 // Also for "[var, var; _] = expr".
2315 if (var_count > 0 && (!semicolon || !did_generate_slice))
Bram Moolenaar9af78762020-06-16 11:34:42 +02002316 {
Bram Moolenaarec792292020-12-13 21:26:56 +01002317 if (generate_instr_drop(cctx, ISN_DROP, 1) == NULL)
Bram Moolenaar9af78762020-06-16 11:34:42 +02002318 goto theend;
2319 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002320
Bram Moolenaarb2097502020-07-19 17:17:02 +02002321 ret = skipwhite(end);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002322
2323theend:
Bram Moolenaar752fc692021-01-04 21:57:11 +01002324 vim_free(lhs.lhs_name);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002325 return ret;
2326}
2327
2328/*
Bram Moolenaar17126b12021-01-07 22:03:02 +01002329 * Check for an assignment at "eap->cmd", compile it if found.
2330 * Return NOTDONE if there is none, FAIL for failure, OK if done.
2331 */
2332 static int
2333may_compile_assignment(exarg_T *eap, char_u **line, cctx_T *cctx)
2334{
2335 char_u *pskip;
2336 char_u *p;
2337
2338 // Assuming the command starts with a variable or function name,
2339 // find what follows.
2340 // Skip over "var.member", "var[idx]" and the like.
2341 // Also "&opt = val", "$ENV = val" and "@r = val".
2342 pskip = (*eap->cmd == '&' || *eap->cmd == '$' || *eap->cmd == '@')
2343 ? eap->cmd + 1 : eap->cmd;
2344 p = to_name_end(pskip, TRUE);
2345 if (p > eap->cmd && *p != NUL)
2346 {
2347 char_u *var_end;
2348 int oplen;
2349 int heredoc;
2350
2351 if (eap->cmd[0] == '@')
2352 var_end = eap->cmd + 2;
2353 else
2354 var_end = find_name_end(pskip, NULL, NULL,
2355 FNE_CHECK_START | FNE_INCL_BR);
2356 oplen = assignment_len(skipwhite(var_end), &heredoc);
2357 if (oplen > 0)
2358 {
2359 size_t len = p - eap->cmd;
2360
2361 // Recognize an assignment if we recognize the variable
2362 // name:
2363 // "g:var = expr"
2364 // "local = expr" where "local" is a local var.
2365 // "script = expr" where "script" is a script-local var.
2366 // "import = expr" where "import" is an imported var
2367 // "&opt = expr"
2368 // "$ENV = expr"
2369 // "@r = expr"
2370 if (*eap->cmd == '&'
2371 || *eap->cmd == '$'
2372 || *eap->cmd == '@'
2373 || ((len) > 2 && eap->cmd[1] == ':')
Bram Moolenaare0890d62021-02-17 14:52:14 +01002374 || variable_exists(eap->cmd, len, cctx))
Bram Moolenaar17126b12021-01-07 22:03:02 +01002375 {
2376 *line = compile_assignment(eap->cmd, eap, CMD_SIZE, cctx);
2377 if (*line == NULL || *line == eap->cmd)
2378 return FAIL;
2379 return OK;
2380 }
2381 }
2382 }
2383
2384 if (*eap->cmd == '[')
2385 {
2386 // [var, var] = expr
2387 *line = compile_assignment(eap->cmd, eap, CMD_SIZE, cctx);
2388 if (*line == NULL)
2389 return FAIL;
2390 if (*line != eap->cmd)
2391 return OK;
2392 }
2393 return NOTDONE;
2394}
2395
Bram Moolenaar9a015112021-12-31 14:06:45 +00002396/*
2397 * Check if arguments of "ufunc" shadow variables in "cctx".
2398 * Return OK or FAIL.
2399 */
2400 static int
2401check_args_shadowing(ufunc_T *ufunc, cctx_T *cctx)
2402{
2403 int i;
2404 char_u *arg;
2405 int r = OK;
2406
2407 // Make sure arguments are not found when compiling a second time.
2408 ufunc->uf_args_visible = 0;
2409
2410 // Check for arguments shadowing variables from the context.
2411 for (i = 0; i < ufunc->uf_args.ga_len; ++i)
2412 {
2413 arg = ((char_u **)(ufunc->uf_args.ga_data))[i];
2414 if (check_defined(arg, STRLEN(arg), cctx, TRUE) == FAIL)
2415 {
2416 r = FAIL;
2417 break;
2418 }
2419 }
2420 ufunc->uf_args_visible = ufunc->uf_args.ga_len;
2421 return r;
2422}
2423
Bram Moolenaar7b829262021-10-13 15:04:34 +01002424
2425/*
Bram Moolenaar09689a02020-05-09 22:50:08 +02002426 * Add a function to the list of :def functions.
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002427 * This sets "ufunc->uf_dfunc_idx" but the function isn't compiled yet.
Bram Moolenaar09689a02020-05-09 22:50:08 +02002428 */
Bram Moolenaar822ba242020-05-24 23:00:18 +02002429 static int
Bram Moolenaar09689a02020-05-09 22:50:08 +02002430add_def_function(ufunc_T *ufunc)
2431{
2432 dfunc_T *dfunc;
2433
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002434 if (def_functions.ga_len == 0)
2435 {
2436 // The first position is not used, so that a zero uf_dfunc_idx means it
2437 // wasn't set.
Bram Moolenaar35578162021-08-02 19:10:38 +02002438 if (GA_GROW_FAILS(&def_functions, 1))
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002439 return FAIL;
2440 ++def_functions.ga_len;
2441 }
2442
Bram Moolenaar09689a02020-05-09 22:50:08 +02002443 // Add the function to "def_functions".
Bram Moolenaar35578162021-08-02 19:10:38 +02002444 if (GA_GROW_FAILS(&def_functions, 1))
Bram Moolenaar09689a02020-05-09 22:50:08 +02002445 return FAIL;
2446 dfunc = ((dfunc_T *)def_functions.ga_data) + def_functions.ga_len;
2447 CLEAR_POINTER(dfunc);
2448 dfunc->df_idx = def_functions.ga_len;
2449 ufunc->uf_dfunc_idx = dfunc->df_idx;
2450 dfunc->df_ufunc = ufunc;
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002451 dfunc->df_name = vim_strsave(ufunc->uf_name);
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02002452 ga_init2(&dfunc->df_var_names, sizeof(char_u *), 10);
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002453 ++dfunc->df_refcount;
Bram Moolenaar09689a02020-05-09 22:50:08 +02002454 ++def_functions.ga_len;
2455 return OK;
2456}
2457
2458/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002459 * After ex_function() has collected all the function lines: parse and compile
2460 * the lines into instructions.
2461 * Adds the function to "def_functions".
Bram Moolenaar9e68c322020-12-25 12:38:04 +01002462 * When "check_return_type" is set then set ufunc->uf_ret_type to the type of
2463 * the return statement (used for lambda). When uf_ret_type is already set
2464 * then check that it matches.
Bram Moolenaarb2049902021-01-24 12:53:53 +01002465 * When "profiling" is true add ISN_PROF_START instructions.
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002466 * "outer_cctx" is set for a nested function.
Bram Moolenaar05afcee2020-03-31 23:32:31 +02002467 * This can be used recursively through compile_lambda(), which may reallocate
2468 * "def_functions".
Bram Moolenaar822ba242020-05-24 23:00:18 +02002469 * Returns OK or FAIL.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002470 */
Bram Moolenaar822ba242020-05-24 23:00:18 +02002471 int
Bram Moolenaarb2049902021-01-24 12:53:53 +01002472compile_def_function(
Bram Moolenaare99d4222021-06-13 14:01:26 +02002473 ufunc_T *ufunc,
2474 int check_return_type,
2475 compiletype_T compile_type,
2476 cctx_T *outer_cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002477{
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002478 char_u *line = NULL;
Bram Moolenaarf62d7392021-04-14 12:40:00 +02002479 char_u *line_to_free = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002480 char_u *p;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002481 char *errormsg = NULL; // error message
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002482 cctx_T cctx;
2483 garray_T *instr;
Bram Moolenaard66960b2020-10-30 20:46:26 +01002484 int did_emsg_before = did_emsg;
Bram Moolenaar599410c2021-04-10 14:03:43 +02002485 int did_emsg_silent_before = did_emsg_silent;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002486 int ret = FAIL;
2487 sctx_T save_current_sctx = current_sctx;
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +02002488 int save_estack_compiling = estack_compiling;
Bram Moolenaardc4c2302021-04-25 13:54:42 +02002489 int save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002490 int do_estack_push;
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02002491 int new_def_function = FALSE;
Bram Moolenaarf002a412021-01-24 13:34:18 +01002492#ifdef FEAT_PROFILE
Bram Moolenaarb2049902021-01-24 12:53:53 +01002493 int prof_lnum = -1;
Bram Moolenaarf002a412021-01-24 13:34:18 +01002494#endif
Bram Moolenaare99d4222021-06-13 14:01:26 +02002495 int debug_lnum = -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002496
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002497 // When using a function that was compiled before: Free old instructions.
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01002498 // The index is reused. Otherwise add a new entry in "def_functions".
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002499 if (ufunc->uf_dfunc_idx > 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002500 {
Bram Moolenaar09689a02020-05-09 22:50:08 +02002501 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
2502 + ufunc->uf_dfunc_idx;
Bram Moolenaar3b814af2021-06-15 10:22:17 +02002503 isn_T *instr_dest = NULL;
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02002504
2505 switch (compile_type)
2506 {
2507 case CT_PROFILE:
2508#ifdef FEAT_PROFILE
2509 instr_dest = dfunc->df_instr_prof; break;
2510#endif
2511 case CT_NONE: instr_dest = dfunc->df_instr; break;
2512 case CT_DEBUG: instr_dest = dfunc->df_instr_debug; break;
2513 }
2514 if (instr_dest != NULL)
2515 // Was compiled in this mode before: Free old instructions.
2516 delete_def_function_contents(dfunc, FALSE);
2517 ga_clear_strings(&dfunc->df_var_names);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002518 }
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02002519 else
2520 {
2521 if (add_def_function(ufunc) == FAIL)
2522 return FAIL;
2523 new_def_function = TRUE;
2524 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002525
Bram Moolenaar985116a2020-07-12 17:31:09 +02002526 ufunc->uf_def_status = UF_COMPILING;
2527
Bram Moolenaara80faa82020-04-12 19:37:17 +02002528 CLEAR_FIELD(cctx);
Bram Moolenaarb2049902021-01-24 12:53:53 +01002529
Bram Moolenaare99d4222021-06-13 14:01:26 +02002530 cctx.ctx_compile_type = compile_type;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002531 cctx.ctx_ufunc = ufunc;
2532 cctx.ctx_lnum = -1;
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002533 cctx.ctx_outer = outer_cctx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002534 ga_init2(&cctx.ctx_locals, sizeof(lvar_T), 10);
2535 ga_init2(&cctx.ctx_type_stack, sizeof(type_T *), 50);
2536 ga_init2(&cctx.ctx_imports, sizeof(imported_T), 10);
2537 cctx.ctx_type_list = &ufunc->uf_type_list;
2538 ga_init2(&cctx.ctx_instr, sizeof(isn_T), 50);
2539 instr = &cctx.ctx_instr;
2540
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002541 // Set the context to the function, it may be compiled when called from
2542 // another script. Set the script version to the most modern one.
2543 // The line number will be set in next_line_from_context().
2544 current_sctx = ufunc->uf_script_ctx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002545 current_sctx.sc_version = SCRIPT_VERSION_VIM9;
2546
Bram Moolenaardc4c2302021-04-25 13:54:42 +02002547 // Don't use the flag from ":legacy" here.
2548 cmdmod.cmod_flags &= ~CMOD_LEGACY;
2549
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002550 // Make sure error messages are OK.
2551 do_estack_push = !estack_top_is_ufunc(ufunc, 1);
2552 if (do_estack_push)
2553 estack_push_ufunc(ufunc, 1);
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +02002554 estack_compiling = TRUE;
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002555
Bram Moolenaar9a015112021-12-31 14:06:45 +00002556 if (check_args_shadowing(ufunc, &cctx) == FAIL)
2557 goto erret;
2558
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002559 if (ufunc->uf_def_args.ga_len > 0)
2560 {
2561 int count = ufunc->uf_def_args.ga_len;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002562 int first_def_arg = ufunc->uf_args.ga_len - count;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002563 int i;
2564 char_u *arg;
2565 int off = STACK_FRAME_SIZE + (ufunc->uf_va_name != NULL ? 1 : 0);
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002566 int did_set_arg_type = FALSE;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002567
2568 // Produce instructions for the default values of optional arguments.
Bram Moolenaar12bce952021-03-11 20:04:04 +01002569 SOURCING_LNUM = 0; // line number unknown
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002570 for (i = 0; i < count; ++i)
2571 {
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002572 garray_T *stack = &cctx.ctx_type_stack;
2573 type_T *val_type;
2574 int arg_idx = first_def_arg + i;
Bram Moolenaar7a3fe3e2021-07-22 14:58:47 +02002575 where_T where = WHERE_INIT;
Bram Moolenaar12bce952021-03-11 20:04:04 +01002576 int r;
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +02002577 int jump_instr_idx = instr->ga_len;
2578 isn_T *isn;
2579
2580 // Use a JUMP_IF_ARG_SET instruction to skip if the value was given.
2581 if (generate_JUMP_IF_ARG_SET(&cctx, i - count - off) == FAIL)
2582 goto erret;
Bram Moolenaar12bce952021-03-11 20:04:04 +01002583
2584 // Make sure later arguments are not found.
Bram Moolenaare28d9b32021-07-03 18:56:53 +02002585 ufunc->uf_args_visible = arg_idx;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002586
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002587 arg = ((char_u **)(ufunc->uf_def_args.ga_data))[i];
Bram Moolenaar12bce952021-03-11 20:04:04 +01002588 r = compile_expr0(&arg, &cctx);
2589
Bram Moolenaar12bce952021-03-11 20:04:04 +01002590 if (r == FAIL)
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002591 goto erret;
2592
2593 // If no type specified use the type of the default value.
2594 // Otherwise check that the default value type matches the
2595 // specified type.
2596 val_type = ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaarf785aa12021-02-11 21:19:34 +01002597 where.wt_index = arg_idx + 1;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002598 if (ufunc->uf_arg_types[arg_idx] == &t_unknown)
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002599 {
2600 did_set_arg_type = TRUE;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002601 ufunc->uf_arg_types[arg_idx] = val_type;
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002602 }
Bram Moolenaarfa46ead2021-12-22 13:18:39 +00002603 else if (need_type_where(val_type, ufunc->uf_arg_types[arg_idx],
2604 -1, where, &cctx, FALSE, FALSE) == FAIL)
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002605 goto erret;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002606
2607 if (generate_STORE(&cctx, ISN_STORE, i - count - off, NULL) == FAIL)
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002608 goto erret;
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +02002609
2610 // set instruction index in JUMP_IF_ARG_SET to here
2611 isn = ((isn_T *)instr->ga_data) + jump_instr_idx;
2612 isn->isn_arg.jumparg.jump_where = instr->ga_len;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002613 }
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002614
2615 if (did_set_arg_type)
2616 set_function_type(ufunc);
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002617 }
Bram Moolenaare28d9b32021-07-03 18:56:53 +02002618 ufunc->uf_args_visible = ufunc->uf_args.ga_len;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002619
2620 /*
2621 * Loop over all the lines of the function and generate instructions.
2622 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002623 for (;;)
2624 {
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002625 exarg_T ea;
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002626 int starts_with_colon = FALSE;
2627 char_u *cmd;
Bram Moolenaar02194d22020-10-24 23:08:38 +02002628 cmdmod_T local_cmdmod;
Bram Moolenaar5b1c8fe2020-02-21 18:42:43 +01002629
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002630 // Bail out on the first error to avoid a flood of errors and report
2631 // the right line number when inside try/catch.
Bram Moolenaard66960b2020-10-30 20:46:26 +01002632 if (did_emsg_before != did_emsg)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002633 goto erret;
2634
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002635 if (line != NULL && *line == '|')
2636 // the line continues after a '|'
2637 ++line;
Bram Moolenaara3b7fdc2020-06-21 14:12:17 +02002638 else if (line != NULL && *skipwhite(line) != NUL
Bram Moolenaar7a092242020-04-16 22:10:49 +02002639 && !(*line == '#' && (line == cctx.ctx_line_start
2640 || VIM_ISWHITE(line[-1]))))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002641 {
Bram Moolenaardd1a9af2020-07-23 15:38:03 +02002642 semsg(_(e_trailing_arg), line);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002643 goto erret;
2644 }
Bram Moolenaar4b3e1962021-03-18 21:37:55 +01002645 else if (line != NULL && vim9_bad_comment(skipwhite(line)))
2646 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002647 else
2648 {
Bram Moolenaar23c55272020-06-21 16:58:13 +02002649 line = next_line_from_context(&cctx, FALSE);
Bram Moolenaar4fdae992020-04-12 16:38:57 +02002650 if (cctx.ctx_lnum >= ufunc->uf_lines.ga_len)
Bram Moolenaarb2049902021-01-24 12:53:53 +01002651 {
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002652 // beyond the last line
Bram Moolenaarf002a412021-01-24 13:34:18 +01002653#ifdef FEAT_PROFILE
Bram Moolenaarced68a02021-01-24 17:53:47 +01002654 if (cctx.ctx_skip != SKIP_YES)
2655 may_generate_prof_end(&cctx, prof_lnum);
Bram Moolenaarf002a412021-01-24 13:34:18 +01002656#endif
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002657 break;
Bram Moolenaarb2049902021-01-24 12:53:53 +01002658 }
Bram Moolenaarf62d7392021-04-14 12:40:00 +02002659 // Make a copy, splitting off nextcmd and removing trailing spaces
2660 // may change it.
2661 if (line != NULL)
2662 {
2663 line = vim_strsave(line);
2664 vim_free(line_to_free);
2665 line_to_free = line;
2666 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002667 }
2668
Bram Moolenaara80faa82020-04-12 19:37:17 +02002669 CLEAR_FIELD(ea);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002670 ea.cmdlinep = &line;
2671 ea.cmd = skipwhite(line);
2672
Bram Moolenaarb2049902021-01-24 12:53:53 +01002673 if (*ea.cmd == '#')
2674 {
2675 // "#" starts a comment
2676 line = (char_u *)"";
2677 continue;
2678 }
2679
Bram Moolenaarf002a412021-01-24 13:34:18 +01002680#ifdef FEAT_PROFILE
Bram Moolenaare99d4222021-06-13 14:01:26 +02002681 if (cctx.ctx_compile_type == CT_PROFILE && cctx.ctx_lnum != prof_lnum
2682 && cctx.ctx_skip != SKIP_YES)
Bram Moolenaarb2049902021-01-24 12:53:53 +01002683 {
2684 may_generate_prof_end(&cctx, prof_lnum);
2685
2686 prof_lnum = cctx.ctx_lnum;
2687 generate_instr(&cctx, ISN_PROF_START);
2688 }
Bram Moolenaarf002a412021-01-24 13:34:18 +01002689#endif
Bram Moolenaare99d4222021-06-13 14:01:26 +02002690 if (cctx.ctx_compile_type == CT_DEBUG && cctx.ctx_lnum != debug_lnum
2691 && cctx.ctx_skip != SKIP_YES)
2692 {
2693 debug_lnum = cctx.ctx_lnum;
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02002694 generate_instr_debug(&cctx);
Bram Moolenaare99d4222021-06-13 14:01:26 +02002695 }
Bram Moolenaar8cec9272021-06-23 20:20:53 +02002696 cctx.ctx_prev_lnum = cctx.ctx_lnum + 1;
Bram Moolenaarb2049902021-01-24 12:53:53 +01002697
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002698 // Some things can be recognized by the first character.
2699 switch (*ea.cmd)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002700 {
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002701 case '}':
2702 {
2703 // "}" ends a block scope
2704 scopetype_T stype = cctx.ctx_scope == NULL
2705 ? NO_SCOPE : cctx.ctx_scope->se_type;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002706
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002707 if (stype == BLOCK_SCOPE)
2708 {
2709 compile_endblock(&cctx);
2710 line = ea.cmd;
2711 }
2712 else
2713 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02002714 emsg(_(e_using_rcurly_outside_if_block_scope));
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002715 goto erret;
2716 }
2717 if (line != NULL)
2718 line = skipwhite(ea.cmd + 1);
2719 continue;
2720 }
2721
2722 case '{':
2723 // "{" starts a block scope
2724 // "{'a': 1}->func() is something else
2725 if (ends_excmd(*skipwhite(ea.cmd + 1)))
2726 {
2727 line = compile_block(ea.cmd, &cctx);
2728 continue;
2729 }
2730 break;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002731 }
2732
2733 /*
2734 * COMMAND MODIFIERS
2735 */
Bram Moolenaar02194d22020-10-24 23:08:38 +02002736 cctx.ctx_has_cmdmod = FALSE;
Bram Moolenaare1004402020-10-24 20:49:43 +02002737 if (parse_command_modifiers(&ea, &errormsg, &local_cmdmod, FALSE)
2738 == FAIL)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002739 {
2740 if (errormsg != NULL)
2741 goto erret;
2742 // empty line or comment
2743 line = (char_u *)"";
2744 continue;
2745 }
Bram Moolenaare1004402020-10-24 20:49:43 +02002746 generate_cmdmods(&cctx, &local_cmdmod);
2747 undo_cmdmod(&local_cmdmod);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002748
Bram Moolenaare88c8e82020-11-01 17:03:37 +01002749 // Check if there was a colon after the last command modifier or before
2750 // the current position.
2751 for (p = ea.cmd; p >= line; --p)
2752 {
2753 if (*p == ':')
2754 starts_with_colon = TRUE;
2755 if (p < ea.cmd && !VIM_ISWHITE(*p))
2756 break;
2757 }
2758
Bram Moolenaarce024c32021-06-26 13:00:49 +02002759 // Skip ":call" to get to the function name, unless using :legacy
Bram Moolenaar575f24b2020-08-12 14:21:11 +02002760 p = ea.cmd;
Bram Moolenaarce024c32021-06-26 13:00:49 +02002761 if (!(local_cmdmod.cmod_flags & CMOD_LEGACY))
Bram Moolenaar575f24b2020-08-12 14:21:11 +02002762 {
Bram Moolenaarce024c32021-06-26 13:00:49 +02002763 if (checkforcmd(&ea.cmd, "call", 3))
2764 {
2765 if (*ea.cmd == '(')
2766 // not for "call()"
2767 ea.cmd = p;
2768 else
2769 ea.cmd = skipwhite(ea.cmd);
2770 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002771
Bram Moolenaarce024c32021-06-26 13:00:49 +02002772 if (!starts_with_colon)
2773 {
2774 int assign;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002775
Bram Moolenaarce024c32021-06-26 13:00:49 +02002776 // Check for assignment after command modifiers.
2777 assign = may_compile_assignment(&ea, &line, &cctx);
2778 if (assign == OK)
2779 goto nextline;
2780 if (assign == FAIL)
2781 goto erret;
2782 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002783 }
2784
2785 /*
2786 * COMMAND after range
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002787 * 'text'->func() should not be confused with 'a mark
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002788 * "++nr" and "--nr" are eval commands
Bram Moolenaar5ca5cc62021-08-24 21:56:03 +02002789 * in "$ENV->func()" the "$" is not a range
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002790 */
Bram Moolenaardf069ee2020-06-22 23:02:51 +02002791 cmd = ea.cmd;
Bram Moolenaarb579f6e2021-12-04 11:57:00 +00002792 if ((*cmd != '$' || starts_with_colon)
Bram Moolenaarce024c32021-06-26 13:00:49 +02002793 && (starts_with_colon || !(*cmd == '\''
2794 || (cmd[0] == cmd[1] && (*cmd == '+' || *cmd == '-')))))
Bram Moolenaardf069ee2020-06-22 23:02:51 +02002795 {
Bram Moolenaar3bd8de42020-09-14 16:37:34 +02002796 ea.cmd = skip_range(ea.cmd, TRUE, NULL);
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002797 if (ea.cmd > cmd)
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002798 {
Bram Moolenaarb579f6e2021-12-04 11:57:00 +00002799 if (!starts_with_colon
2800 && !(local_cmdmod.cmod_flags & CMOD_LEGACY))
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002801 {
Bram Moolenaar6e2c2c52020-12-25 19:25:45 +01002802 semsg(_(e_colon_required_before_range_str), cmd);
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002803 goto erret;
2804 }
Bram Moolenaarada1d872021-02-20 08:16:51 +01002805 ea.addr_count = 1;
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002806 if (ends_excmd2(line, ea.cmd))
2807 {
2808 // A range without a command: jump to the line.
Bram Moolenaare4eed8c2021-12-01 15:22:56 +00002809 generate_EXEC(&cctx, ISN_EXECRANGE,
Bram Moolenaarb579f6e2021-12-04 11:57:00 +00002810 vim_strnsave(cmd, ea.cmd - cmd));
Bram Moolenaare4eed8c2021-12-01 15:22:56 +00002811 line = ea.cmd;
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002812 goto nextline;
2813 }
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002814 }
Bram Moolenaardf069ee2020-06-22 23:02:51 +02002815 }
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02002816 p = find_ex_command(&ea, NULL,
2817 starts_with_colon || (local_cmdmod.cmod_flags & CMOD_LEGACY)
Bram Moolenaar77b10ff2021-03-14 13:21:35 +01002818 ? NULL : item_exists, &cctx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002819
Bram Moolenaard1510ee2021-01-04 16:15:58 +01002820 if (p == NULL)
2821 {
2822 if (cctx.ctx_skip != SKIP_YES)
2823 emsg(_(e_ambiguous_use_of_user_defined_command));
2824 goto erret;
2825 }
2826
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002827 // When using ":legacy cmd" always use compile_exec().
2828 if (local_cmdmod.cmod_flags & CMOD_LEGACY)
Bram Moolenaar3b1373b2021-05-17 00:01:42 +02002829 {
2830 char_u *start = ea.cmd;
2831
Bram Moolenaarc3cb1c92021-06-02 16:47:53 +02002832 switch (ea.cmdidx)
2833 {
2834 case CMD_if:
2835 case CMD_elseif:
2836 case CMD_else:
2837 case CMD_endif:
2838 case CMD_for:
2839 case CMD_endfor:
2840 case CMD_continue:
2841 case CMD_break:
2842 case CMD_while:
2843 case CMD_endwhile:
2844 case CMD_try:
2845 case CMD_catch:
2846 case CMD_finally:
2847 case CMD_endtry:
2848 semsg(_(e_cannot_use_legacy_with_command_str), ea.cmd);
2849 goto erret;
2850 default: break;
2851 }
2852
Bram Moolenaar3b1373b2021-05-17 00:01:42 +02002853 // ":legacy return expr" needs to be handled differently.
2854 if (checkforcmd(&start, "return", 4))
2855 ea.cmdidx = CMD_return;
2856 else
2857 ea.cmdidx = CMD_legacy;
2858 }
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002859
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002860 if (p == ea.cmd && ea.cmdidx != CMD_SIZE)
2861 {
Bram Moolenaare525bdd2021-08-07 18:12:40 +02002862 if (cctx.ctx_skip == SKIP_YES && ea.cmdidx != CMD_eval)
Bram Moolenaara259d8d2020-01-31 20:10:50 +01002863 {
2864 line += STRLEN(line);
Bram Moolenaarf665e972020-12-05 19:17:16 +01002865 goto nextline;
Bram Moolenaara259d8d2020-01-31 20:10:50 +01002866 }
Bram Moolenaare525bdd2021-08-07 18:12:40 +02002867 else if (ea.cmdidx != CMD_eval)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002868 {
Bram Moolenaar52c124d2020-12-20 21:43:35 +01002869 // CMD_var cannot happen, compile_assignment() above would be
2870 // used. Most likely an assignment to a non-existing variable.
2871 semsg(_(e_command_not_recognized_str), ea.cmd);
Bram Moolenaar007f9d62020-07-06 23:04:49 +02002872 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002873 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002874 }
2875
Bram Moolenaar3988f642020-08-27 22:43:03 +02002876 if (cctx.ctx_had_return
Bram Moolenaara259d8d2020-01-31 20:10:50 +01002877 && ea.cmdidx != CMD_elseif
2878 && ea.cmdidx != CMD_else
Bram Moolenaarefd88552020-06-18 20:50:10 +02002879 && ea.cmdidx != CMD_endif
2880 && ea.cmdidx != CMD_endfor
2881 && ea.cmdidx != CMD_endwhile
2882 && ea.cmdidx != CMD_catch
2883 && ea.cmdidx != CMD_finally
2884 && ea.cmdidx != CMD_endtry)
2885 {
Bram Moolenaar3988f642020-08-27 22:43:03 +02002886 emsg(_(e_unreachable_code_after_return));
2887 goto erret;
Bram Moolenaarefd88552020-06-18 20:50:10 +02002888 }
2889
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +02002890 p = skipwhite(p);
2891 if (ea.cmdidx != CMD_SIZE
2892 && ea.cmdidx != CMD_write && ea.cmdidx != CMD_read)
2893 {
Bram Moolenaar9b123d82020-09-14 22:39:11 +02002894 if (ea.cmdidx >= 0)
2895 ea.argt = excmd_get_argt(ea.cmdidx);
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +02002896 if ((ea.argt & EX_BANG) && *p == '!')
2897 {
2898 ea.forceit = TRUE;
2899 p = skipwhite(p + 1);
2900 }
2901 }
2902
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002903 switch (ea.cmdidx)
2904 {
2905 case CMD_def:
Bram Moolenaar38453522021-11-28 22:00:12 +00002906 case CMD_function:
Bram Moolenaar04b12692020-05-04 23:24:44 +02002907 ea.arg = p;
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +00002908 line = compile_nested_function(&ea, &cctx, &line_to_free);
Bram Moolenaar04b12692020-05-04 23:24:44 +02002909 break;
2910
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002911 case CMD_return:
Bram Moolenaar3b1373b2021-05-17 00:01:42 +02002912 line = compile_return(p, check_return_type,
2913 local_cmdmod.cmod_flags & CMOD_LEGACY, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002914 cctx.ctx_had_return = TRUE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002915 break;
2916
2917 case CMD_let:
Bram Moolenaarc58f5452020-10-21 20:58:52 +02002918 emsg(_(e_cannot_use_let_in_vim9_script));
2919 break;
Bram Moolenaar30fd8202020-09-26 15:09:30 +02002920 case CMD_var:
2921 case CMD_final:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002922 case CMD_const:
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002923 case CMD_increment:
2924 case CMD_decrement:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002925 line = compile_assignment(p, &ea, ea.cmdidx, &cctx);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002926 if (line == p)
2927 line = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002928 break;
2929
Bram Moolenaard72c1bf2020-04-19 16:28:59 +02002930 case CMD_unlet:
2931 case CMD_unlockvar:
2932 case CMD_lockvar:
2933 line = compile_unletlock(p, &ea, &cctx);
2934 break;
2935
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002936 case CMD_import:
Bram Moolenaar4db572e2021-07-18 18:21:38 +02002937 emsg(_(e_import_can_only_be_used_in_script));
2938 line = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002939 break;
2940
2941 case CMD_if:
2942 line = compile_if(p, &cctx);
2943 break;
2944 case CMD_elseif:
2945 line = compile_elseif(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002946 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002947 break;
2948 case CMD_else:
2949 line = compile_else(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002950 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002951 break;
2952 case CMD_endif:
2953 line = compile_endif(p, &cctx);
2954 break;
2955
2956 case CMD_while:
2957 line = compile_while(p, &cctx);
2958 break;
2959 case CMD_endwhile:
2960 line = compile_endwhile(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002961 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002962 break;
2963
2964 case CMD_for:
2965 line = compile_for(p, &cctx);
2966 break;
2967 case CMD_endfor:
2968 line = compile_endfor(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002969 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002970 break;
2971 case CMD_continue:
2972 line = compile_continue(p, &cctx);
2973 break;
2974 case CMD_break:
2975 line = compile_break(p, &cctx);
2976 break;
2977
2978 case CMD_try:
2979 line = compile_try(p, &cctx);
2980 break;
2981 case CMD_catch:
2982 line = compile_catch(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002983 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002984 break;
2985 case CMD_finally:
2986 line = compile_finally(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002987 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002988 break;
2989 case CMD_endtry:
2990 line = compile_endtry(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002991 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002992 break;
2993 case CMD_throw:
2994 line = compile_throw(p, &cctx);
2995 break;
2996
Bram Moolenaar007f9d62020-07-06 23:04:49 +02002997 case CMD_eval:
Bram Moolenaarc3235272021-07-10 19:42:03 +02002998 line = compile_eval(p, &cctx);
Bram Moolenaar007f9d62020-07-06 23:04:49 +02002999 break;
3000
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003001 case CMD_echo:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003002 case CMD_echon:
Bram Moolenaarad39c092020-02-26 18:23:43 +01003003 case CMD_execute:
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02003004 case CMD_echomsg:
3005 case CMD_echoerr:
Bram Moolenaar7de62622021-08-07 15:05:47 +02003006 case CMD_echoconsole:
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02003007 line = compile_mult_expr(p, ea.cmdidx, &cctx);
Bram Moolenaarad39c092020-02-26 18:23:43 +01003008 break;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003009
Bram Moolenaarc3516f72020-09-08 22:45:35 +02003010 case CMD_put:
3011 ea.cmd = cmd;
3012 line = compile_put(p, &ea, &cctx);
3013 break;
3014
Bram Moolenaar4c137212021-04-19 16:48:48 +02003015 case CMD_substitute:
Bram Moolenaar7b829262021-10-13 15:04:34 +01003016 if (check_global_and_subst(ea.cmd, p) == FAIL)
3017 goto erret;
Bram Moolenaar4c137212021-04-19 16:48:48 +02003018 if (cctx.ctx_skip == SKIP_YES)
3019 line = (char_u *)"";
3020 else
3021 {
3022 ea.arg = p;
3023 line = compile_substitute(line, &ea, &cctx);
3024 }
3025 break;
3026
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02003027 case CMD_redir:
3028 ea.arg = p;
3029 line = compile_redir(line, &ea, &cctx);
3030 break;
3031
Bram Moolenaar5f7d4c02021-05-05 21:31:39 +02003032 case CMD_cexpr:
3033 case CMD_lexpr:
3034 case CMD_caddexpr:
3035 case CMD_laddexpr:
3036 case CMD_cgetexpr:
3037 case CMD_lgetexpr:
Bram Moolenaarb7c97812021-05-05 22:51:39 +02003038#ifdef FEAT_QUICKFIX
Bram Moolenaar5f7d4c02021-05-05 21:31:39 +02003039 ea.arg = p;
3040 line = compile_cexpr(line, &ea, &cctx);
Bram Moolenaarb7c97812021-05-05 22:51:39 +02003041#else
3042 ex_ni(&ea);
3043 line = NULL;
3044#endif
Bram Moolenaar5f7d4c02021-05-05 21:31:39 +02003045 break;
3046
Bram Moolenaarae616492020-07-28 20:07:27 +02003047 case CMD_append:
3048 case CMD_change:
3049 case CMD_insert:
Bram Moolenaar10b94212021-02-19 21:42:57 +01003050 case CMD_k:
Bram Moolenaarf5a48012020-08-01 17:00:03 +02003051 case CMD_t:
Bram Moolenaarae616492020-07-28 20:07:27 +02003052 case CMD_xit:
3053 not_in_vim9(&ea);
3054 goto erret;
3055
Bram Moolenaar002262f2020-07-08 17:47:57 +02003056 case CMD_SIZE:
Bram Moolenaar3988f642020-08-27 22:43:03 +02003057 if (cctx.ctx_skip != SKIP_YES)
3058 {
3059 semsg(_(e_invalid_command_str), ea.cmd);
3060 goto erret;
3061 }
3062 // We don't check for a next command here.
3063 line = (char_u *)"";
3064 break;
Bram Moolenaar002262f2020-07-08 17:47:57 +02003065
Bram Moolenaar20677332021-06-06 17:02:53 +02003066 case CMD_lua:
3067 case CMD_mzscheme:
3068 case CMD_perl:
3069 case CMD_py3:
3070 case CMD_python3:
3071 case CMD_python:
3072 case CMD_pythonx:
3073 case CMD_ruby:
3074 case CMD_tcl:
3075 ea.arg = p;
3076 if (vim_strchr(line, '\n') == NULL)
Bram Moolenaar5163fcc2020-08-27 23:37:09 +02003077 line = compile_exec(line, &ea, &cctx);
Bram Moolenaar20677332021-06-06 17:02:53 +02003078 else
3079 // heredoc lines have been concatenated with NL
3080 // characters in get_function_body()
3081 line = compile_script(line, &cctx);
3082 break;
3083
Bram Moolenaar7b829262021-10-13 15:04:34 +01003084 case CMD_global:
3085 if (check_global_and_subst(ea.cmd, p) == FAIL)
3086 goto erret;
3087 // FALLTHROUGH
Bram Moolenaar20677332021-06-06 17:02:53 +02003088 default:
3089 // Not recognized, execute with do_cmdline_cmd().
3090 ea.arg = p;
3091 line = compile_exec(line, &ea, &cctx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003092 break;
3093 }
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02003094nextline:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003095 if (line == NULL)
3096 goto erret;
Bram Moolenaar585fea72020-04-02 22:33:21 +02003097 line = skipwhite(line);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003098
Bram Moolenaarf4c6e1e2020-10-23 18:02:32 +02003099 // Undo any command modifiers.
Bram Moolenaar02194d22020-10-24 23:08:38 +02003100 generate_undo_cmdmods(&cctx);
Bram Moolenaarf4c6e1e2020-10-23 18:02:32 +02003101
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003102 if (cctx.ctx_type_stack.ga_len < 0)
3103 {
3104 iemsg("Type stack underflow");
3105 goto erret;
3106 }
3107 }
3108
3109 if (cctx.ctx_scope != NULL)
3110 {
3111 if (cctx.ctx_scope->se_type == IF_SCOPE)
3112 emsg(_(e_endif));
3113 else if (cctx.ctx_scope->se_type == WHILE_SCOPE)
3114 emsg(_(e_endwhile));
3115 else if (cctx.ctx_scope->se_type == FOR_SCOPE)
3116 emsg(_(e_endfor));
3117 else
Bram Moolenaar451c2e32020-08-15 16:33:28 +02003118 emsg(_(e_missing_rcurly));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003119 goto erret;
3120 }
3121
Bram Moolenaarefd88552020-06-18 20:50:10 +02003122 if (!cctx.ctx_had_return)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003123 {
Bram Moolenaare6174fd2021-06-12 18:30:56 +02003124 if (ufunc->uf_ret_type->tt_type == VAR_UNKNOWN)
3125 ufunc->uf_ret_type = &t_void;
3126 else if (ufunc->uf_ret_type->tt_type != VAR_VOID)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003127 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02003128 emsg(_(e_missing_return_statement));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003129 goto erret;
3130 }
3131
Bram Moolenaarf57b43c2021-06-15 22:13:27 +02003132 // Return void if there is no return at the end.
3133 generate_instr(&cctx, ISN_RETURN_VOID);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003134 }
3135
Bram Moolenaar599410c2021-04-10 14:03:43 +02003136 // When compiled with ":silent!" and there was an error don't consider the
3137 // function compiled.
3138 if (emsg_silent == 0 || did_emsg_silent == did_emsg_silent_before)
Bram Moolenaar05afcee2020-03-31 23:32:31 +02003139 {
3140 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3141 + ufunc->uf_dfunc_idx;
3142 dfunc->df_deleted = FALSE;
Bram Moolenaar4aab88d2020-12-24 21:56:41 +01003143 dfunc->df_script_seq = current_sctx.sc_seq;
Bram Moolenaarf002a412021-01-24 13:34:18 +01003144#ifdef FEAT_PROFILE
Bram Moolenaare99d4222021-06-13 14:01:26 +02003145 if (cctx.ctx_compile_type == CT_PROFILE)
Bram Moolenaarb2049902021-01-24 12:53:53 +01003146 {
3147 dfunc->df_instr_prof = instr->ga_data;
3148 dfunc->df_instr_prof_count = instr->ga_len;
3149 }
3150 else
Bram Moolenaarf002a412021-01-24 13:34:18 +01003151#endif
Bram Moolenaare99d4222021-06-13 14:01:26 +02003152 if (cctx.ctx_compile_type == CT_DEBUG)
3153 {
3154 dfunc->df_instr_debug = instr->ga_data;
3155 dfunc->df_instr_debug_count = instr->ga_len;
3156 }
3157 else
Bram Moolenaarb2049902021-01-24 12:53:53 +01003158 {
3159 dfunc->df_instr = instr->ga_data;
3160 dfunc->df_instr_count = instr->ga_len;
3161 }
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02003162 dfunc->df_varcount = dfunc->df_var_names.ga_len;
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02003163 dfunc->df_has_closure = cctx.ctx_has_closure;
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02003164 if (cctx.ctx_outer_used)
3165 ufunc->uf_flags |= FC_CLOSURE;
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003166 ufunc->uf_def_status = UF_COMPILED;
Bram Moolenaar05afcee2020-03-31 23:32:31 +02003167 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003168
3169 ret = OK;
3170
3171erret:
Bram Moolenaar599410c2021-04-10 14:03:43 +02003172 if (ufunc->uf_def_status == UF_COMPILING)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003173 {
Bram Moolenaar05afcee2020-03-31 23:32:31 +02003174 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3175 + ufunc->uf_dfunc_idx;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003176
Bram Moolenaarcaf1a2f2021-06-15 11:27:21 +02003177 // Compiling aborted, free the generated instructions.
Bram Moolenaar8238f082021-04-20 21:10:48 +02003178 clear_instr_ga(instr);
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003179 VIM_CLEAR(dfunc->df_name);
Bram Moolenaarcaf1a2f2021-06-15 11:27:21 +02003180 ga_clear_strings(&dfunc->df_var_names);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003181
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02003182 // If using the last entry in the table and it was added above, we
3183 // might as well remove it.
3184 if (!dfunc->df_deleted && new_def_function
Bram Moolenaar45a15082020-05-25 00:28:33 +02003185 && ufunc->uf_dfunc_idx == def_functions.ga_len - 1)
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02003186 {
Bram Moolenaar20431c92020-03-20 18:39:46 +01003187 --def_functions.ga_len;
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02003188 ufunc->uf_dfunc_idx = 0;
3189 }
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003190 ufunc->uf_def_status = UF_COMPILE_ERROR;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003191
Bram Moolenaar3cca2992020-04-02 22:57:36 +02003192 while (cctx.ctx_scope != NULL)
3193 drop_scope(&cctx);
3194
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003195 if (errormsg != NULL)
3196 emsg(errormsg);
Bram Moolenaard66960b2020-10-30 20:46:26 +01003197 else if (did_emsg == did_emsg_before)
Bram Moolenaare13bdec2020-10-16 23:16:47 +02003198 emsg(_(e_compiling_def_function_failed));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003199 }
3200
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02003201 if (cctx.ctx_redir_lhs.lhs_name != NULL)
3202 {
3203 if (ret == OK)
3204 {
3205 emsg(_(e_missing_redir_end));
3206 ret = FAIL;
3207 }
3208 vim_free(cctx.ctx_redir_lhs.lhs_name);
Bram Moolenaar753bcf82021-04-21 14:24:24 +02003209 vim_free(cctx.ctx_redir_lhs.lhs_whole);
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02003210 }
3211
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003212 current_sctx = save_current_sctx;
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +02003213 estack_compiling = save_estack_compiling;
Bram Moolenaardc4c2302021-04-25 13:54:42 +02003214 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar25e0f582020-05-25 22:36:50 +02003215 if (do_estack_push)
3216 estack_pop();
3217
Bram Moolenaarf62d7392021-04-14 12:40:00 +02003218 vim_free(line_to_free);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003219 free_imported(&cctx);
Bram Moolenaarb84a3812020-05-01 15:44:29 +02003220 free_locals(&cctx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003221 ga_clear(&cctx.ctx_type_stack);
Bram Moolenaar822ba242020-05-24 23:00:18 +02003222 return ret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003223}
3224
Bram Moolenaar6ff71d82020-05-24 23:45:24 +02003225 void
3226set_function_type(ufunc_T *ufunc)
3227{
3228 int varargs = ufunc->uf_va_name != NULL;
3229 int argcount = ufunc->uf_args.ga_len;
3230
3231 // Create a type for the function, with the return type and any
3232 // argument types.
3233 // A vararg is included in uf_args.ga_len but not in uf_arg_types.
3234 // The type is included in "tt_args".
3235 if (argcount > 0 || varargs)
3236 {
Bram Moolenaar18062fc2021-03-05 21:35:47 +01003237 if (ufunc->uf_type_list.ga_itemsize == 0)
3238 ga_init2(&ufunc->uf_type_list, sizeof(type_T *), 10);
Bram Moolenaar6ff71d82020-05-24 23:45:24 +02003239 ufunc->uf_func_type = alloc_func_type(ufunc->uf_ret_type,
3240 argcount, &ufunc->uf_type_list);
3241 // Add argument types to the function type.
3242 if (func_type_add_arg_types(ufunc->uf_func_type,
3243 argcount + varargs,
3244 &ufunc->uf_type_list) == FAIL)
3245 return;
3246 ufunc->uf_func_type->tt_argcount = argcount + varargs;
3247 ufunc->uf_func_type->tt_min_argcount =
3248 argcount - ufunc->uf_def_args.ga_len;
3249 if (ufunc->uf_arg_types == NULL)
3250 {
3251 int i;
3252
3253 // lambda does not have argument types.
3254 for (i = 0; i < argcount; ++i)
3255 ufunc->uf_func_type->tt_args[i] = &t_any;
3256 }
3257 else
3258 mch_memmove(ufunc->uf_func_type->tt_args,
3259 ufunc->uf_arg_types, sizeof(type_T *) * argcount);
3260 if (varargs)
3261 {
3262 ufunc->uf_func_type->tt_args[argcount] =
Bram Moolenaar2a389082021-04-09 20:24:31 +02003263 ufunc->uf_va_type == NULL ? &t_list_any : ufunc->uf_va_type;
Bram Moolenaar6ff71d82020-05-24 23:45:24 +02003264 ufunc->uf_func_type->tt_flags = TTFLAG_VARARGS;
3265 }
3266 }
3267 else
3268 // No arguments, can use a predefined type.
3269 ufunc->uf_func_type = get_func_type(ufunc->uf_ret_type,
3270 argcount, &ufunc->uf_type_list);
3271}
3272
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003273/*
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003274 * Free all instructions for "dfunc" except df_name.
Bram Moolenaar20431c92020-03-20 18:39:46 +01003275 */
3276 static void
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003277delete_def_function_contents(dfunc_T *dfunc, int mark_deleted)
Bram Moolenaar20431c92020-03-20 18:39:46 +01003278{
3279 int idx;
3280
3281 ga_clear(&dfunc->df_def_args_isn);
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02003282 ga_clear_strings(&dfunc->df_var_names);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003283
3284 if (dfunc->df_instr != NULL)
3285 {
3286 for (idx = 0; idx < dfunc->df_instr_count; ++idx)
3287 delete_instr(dfunc->df_instr + idx);
3288 VIM_CLEAR(dfunc->df_instr);
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003289 dfunc->df_instr = NULL;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003290 }
Bram Moolenaarc2dec4c2021-06-13 15:39:00 +02003291 if (dfunc->df_instr_debug != NULL)
3292 {
3293 for (idx = 0; idx < dfunc->df_instr_debug_count; ++idx)
3294 delete_instr(dfunc->df_instr_debug + idx);
3295 VIM_CLEAR(dfunc->df_instr_debug);
3296 dfunc->df_instr_debug = NULL;
3297 }
Bram Moolenaarc05fe072021-01-24 21:30:48 +01003298#ifdef FEAT_PROFILE
3299 if (dfunc->df_instr_prof != NULL)
3300 {
3301 for (idx = 0; idx < dfunc->df_instr_prof_count; ++idx)
3302 delete_instr(dfunc->df_instr_prof + idx);
3303 VIM_CLEAR(dfunc->df_instr_prof);
3304 dfunc->df_instr_prof = NULL;
3305 }
3306#endif
Bram Moolenaar20431c92020-03-20 18:39:46 +01003307
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003308 if (mark_deleted)
3309 dfunc->df_deleted = TRUE;
3310 if (dfunc->df_ufunc != NULL)
3311 dfunc->df_ufunc->uf_def_status = UF_NOT_COMPILED;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003312}
3313
3314/*
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003315 * When a user function is deleted, clear the contents of any associated def
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003316 * function, unless another user function still uses it.
3317 * The position in def_functions can be re-used.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003318 */
3319 void
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003320unlink_def_function(ufunc_T *ufunc)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003321{
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003322 if (ufunc->uf_dfunc_idx > 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003323 {
3324 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3325 + ufunc->uf_dfunc_idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003326
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003327 if (--dfunc->df_refcount <= 0)
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003328 delete_def_function_contents(dfunc, TRUE);
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003329 ufunc->uf_def_status = UF_NOT_COMPILED;
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003330 ufunc->uf_dfunc_idx = 0;
3331 if (dfunc->df_ufunc == ufunc)
3332 dfunc->df_ufunc = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003333 }
3334}
3335
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003336/*
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003337 * Used when a user function refers to an existing dfunc.
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003338 */
3339 void
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003340link_def_function(ufunc_T *ufunc)
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003341{
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003342 if (ufunc->uf_dfunc_idx > 0)
3343 {
3344 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3345 + ufunc->uf_dfunc_idx;
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003346
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003347 ++dfunc->df_refcount;
3348 }
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003349}
3350
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003351#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaar20431c92020-03-20 18:39:46 +01003352/*
3353 * Free all functions defined with ":def".
3354 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003355 void
3356free_def_functions(void)
3357{
Bram Moolenaar20431c92020-03-20 18:39:46 +01003358 int idx;
3359
3360 for (idx = 0; idx < def_functions.ga_len; ++idx)
3361 {
3362 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) + idx;
3363
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003364 delete_def_function_contents(dfunc, TRUE);
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003365 vim_free(dfunc->df_name);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003366 }
3367
3368 ga_clear(&def_functions);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003369}
3370#endif
3371
3372
3373#endif // FEAT_EVAL