blob: 50fe7e257ba700c36b6cb20b299976e41b37c608 [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 Moolenaar328eac22021-01-07 19:23:08 +0100369 && (actual->tt_member == &t_any || actual->tt_argcount < 0)
370 && ((actual->tt_member == &t_void)
371 == (expected->tt_member == &t_void))))
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200372 return TRUE;
373 if ((actual->tt_type == VAR_LIST || actual->tt_type == VAR_DICT)
374 && actual->tt_type == expected->tt_type)
375 // This takes care of a nested list or dict.
376 return use_typecheck(actual->tt_member, expected->tt_member);
377 return FALSE;
378}
379
380/*
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200381 * Check that
Bram Moolenaar5e654232020-09-16 15:22:00 +0200382 * - "actual" matches "expected" type or
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200383 * - "actual" is a type that can be "expected" type: add a runtime check; or
384 * - return FAIL.
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200385 * If "actual_is_const" is TRUE then the type won't change at runtime, do not
386 * generate a TYPECHECK.
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200387 */
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200388 static int
389need_type_where(
390 type_T *actual,
391 type_T *expected,
392 int offset,
393 where_T where,
394 cctx_T *cctx,
395 int silent,
396 int actual_is_const)
397{
Bram Moolenaar44a89772021-12-18 12:31:33 +0000398 int ret;
399
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200400 if (expected == &t_bool && actual != &t_bool
401 && (actual->tt_flags & TTFLAG_BOOL_OK))
402 {
403 // Using "0", "1" or the result of an expression with "&&" or "||" as a
404 // boolean is OK but requires a conversion.
405 generate_2BOOL(cctx, FALSE, offset);
406 return OK;
407 }
408
Bram Moolenaar44a89772021-12-18 12:31:33 +0000409 ret = check_type_maybe(expected, actual, FALSE, where);
410 if (ret == OK)
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200411 return OK;
412
413 // If the actual type can be the expected type add a runtime check.
414 // If it's a constant a runtime check makes no sense.
Bram Moolenaar44a89772021-12-18 12:31:33 +0000415 if (ret == MAYBE || ((!actual_is_const || actual == &t_any)
416 && use_typecheck(actual, expected)))
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200417 {
418 generate_TYPECHECK(cctx, expected, offset, where.wt_index);
419 return OK;
420 }
421
422 if (!silent)
423 type_mismatch_where(expected, actual, where);
424 return FAIL;
425}
426
Bram Moolenaar351ead02021-01-16 16:07:01 +0100427 int
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200428need_type(
429 type_T *actual,
430 type_T *expected,
431 int offset,
Bram Moolenaar351ead02021-01-16 16:07:01 +0100432 int arg_idx,
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200433 cctx_T *cctx,
Bram Moolenaar334a8b42020-10-19 16:07:42 +0200434 int silent,
435 int actual_is_const)
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200436{
Bram Moolenaar7a3fe3e2021-07-22 14:58:47 +0200437 where_T where = WHERE_INIT;
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100438
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100439 where.wt_index = arg_idx;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200440 return need_type_where(actual, expected, offset, where,
441 cctx, silent, actual_is_const);
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200442}
443
444/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100445 * Reserve space for a local variable.
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200446 * Return the variable or NULL if it failed.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100447 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000448 lvar_T *
Bram Moolenaare8211a32020-10-09 22:04:29 +0200449reserve_local(
450 cctx_T *cctx,
451 char_u *name,
452 size_t len,
453 int isConst,
454 type_T *type)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100455{
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100456 lvar_T *lvar;
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200457 dfunc_T *dfunc;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100458
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200459 if (arg_exists(name, len, NULL, NULL, NULL, cctx) == OK)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100460 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200461 emsg_namelen(_(e_str_is_used_as_argument), name, (int)len);
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200462 return NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100463 }
464
Bram Moolenaar35578162021-08-02 19:10:38 +0200465 if (GA_GROW_FAILS(&cctx->ctx_locals, 1))
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200466 return NULL;
467 lvar = ((lvar_T *)cctx->ctx_locals.ga_data) + cctx->ctx_locals.ga_len++;
Bram Moolenaare8211a32020-10-09 22:04:29 +0200468 CLEAR_POINTER(lvar);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100469
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200470 // Every local variable uses the next entry on the stack. We could re-use
471 // the last ones when leaving a scope, but then variables used in a closure
472 // might get overwritten. To keep things simple do not re-use stack
473 // entries. This is less efficient, but memory is cheap these days.
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200474 dfunc = ((dfunc_T *)def_functions.ga_data) + cctx->ctx_ufunc->uf_dfunc_idx;
475 lvar->lv_idx = dfunc->df_var_names.ga_len;
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200476
Bram Moolenaar71ccd032020-06-12 22:59:11 +0200477 lvar->lv_name = vim_strnsave(name, len == 0 ? STRLEN(name) : len);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100478 lvar->lv_const = isConst;
479 lvar->lv_type = type;
480
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200481 // Remember the name for debugging.
Bram Moolenaar35578162021-08-02 19:10:38 +0200482 if (GA_GROW_FAILS(&dfunc->df_var_names, 1))
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +0200483 return NULL;
484 ((char_u **)dfunc->df_var_names.ga_data)[lvar->lv_idx] =
485 vim_strsave(lvar->lv_name);
486 ++dfunc->df_var_names.ga_len;
487
Bram Moolenaarb84a3812020-05-01 15:44:29 +0200488 return lvar;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100489}
490
491/*
Bram Moolenaar08251752021-01-11 21:20:18 +0100492 * If "check_writable" is ASSIGN_CONST give an error if the variable was
493 * defined with :final or :const, if "check_writable" is ASSIGN_FINAL give an
494 * error if the variable was defined with :const.
495 */
496 static int
497check_item_writable(svar_T *sv, int check_writable, char_u *name)
498{
499 if ((check_writable == ASSIGN_CONST && sv->sv_const != 0)
500 || (check_writable == ASSIGN_FINAL
501 && sv->sv_const == ASSIGN_CONST))
502 {
Bram Moolenaard8e44472021-07-21 22:20:33 +0200503 semsg(_(e_cannot_change_readonly_variable_str), name);
Bram Moolenaar08251752021-01-11 21:20:18 +0100504 return FAIL;
505 }
506 return OK;
507}
508
509/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100510 * Find "name" in script-local items of script "sid".
Bram Moolenaar08251752021-01-11 21:20:18 +0100511 * Pass "check_writable" to check_item_writable().
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100512 * Returns the index in "sn_var_vals" if found.
513 * If found but not in "sn_var_vals" returns -1.
Bram Moolenaar08251752021-01-11 21:20:18 +0100514 * If not found or the variable is not writable returns -2.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100515 */
516 int
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200517get_script_item_idx(int sid, char_u *name, int check_writable, cctx_T *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100518{
519 hashtab_T *ht;
520 dictitem_T *di;
Bram Moolenaar21b9e972020-01-26 19:26:46 +0100521 scriptitem_T *si = SCRIPT_ITEM(sid);
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200522 svar_T *sv;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100523 int idx;
524
Bram Moolenaare3d46852020-08-29 13:39:17 +0200525 if (!SCRIPT_ID_VALID(sid))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100526 return -1;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200527 if (sid == current_sctx.sc_sid)
528 {
Bram Moolenaar209f0202020-10-15 13:57:56 +0200529 sallvar_T *sav = find_script_var(name, 0, cctx);
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200530
531 if (sav == NULL)
532 return -2;
533 idx = sav->sav_var_vals_idx;
534 sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
Bram Moolenaar08251752021-01-11 21:20:18 +0100535 if (check_item_writable(sv, check_writable, name) == FAIL)
536 return -2;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200537 return idx;
538 }
539
540 // First look the name up in the hashtable.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100541 ht = &SCRIPT_VARS(sid);
542 di = find_var_in_ht(ht, 0, name, TRUE);
543 if (di == NULL)
544 return -2;
545
546 // Now find the svar_T index in sn_var_vals.
547 for (idx = 0; idx < si->sn_var_vals.ga_len; ++idx)
548 {
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200549 sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100550 if (sv->sv_tv == &di->di_tv)
551 {
Bram Moolenaar08251752021-01-11 21:20:18 +0100552 if (check_item_writable(sv, check_writable, name) == FAIL)
553 return -2;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100554 return idx;
555 }
556 }
557 return -1;
558}
559
560/*
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200561 * Find "name" in imported items of the current script or in "cctx" if not
562 * NULL.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100563 */
564 imported_T *
Bram Moolenaar4e12a5d2020-02-03 20:50:59 +0100565find_imported(char_u *name, size_t len, cctx_T *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100566{
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100567 int idx;
568
Bram Moolenaare3d46852020-08-29 13:39:17 +0200569 if (!SCRIPT_ID_VALID(current_sctx.sc_sid))
Bram Moolenaar8e6cbb72020-07-01 14:38:12 +0200570 return NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100571 if (cctx != NULL)
572 for (idx = 0; idx < cctx->ctx_imports.ga_len; ++idx)
573 {
574 imported_T *import = ((imported_T *)cctx->ctx_imports.ga_data)
575 + idx;
576
Bram Moolenaar4e12a5d2020-02-03 20:50:59 +0100577 if (len == 0 ? STRCMP(name, import->imp_name) == 0
578 : STRLEN(import->imp_name) == len
579 && STRNCMP(name, import->imp_name, len) == 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100580 return import;
581 }
582
Bram Moolenaarefa94442020-08-08 22:16:00 +0200583 return find_imported_in_script(name, len, current_sctx.sc_sid);
584}
585
586 imported_T *
587find_imported_in_script(char_u *name, size_t len, int sid)
588{
Bram Moolenaare3d46852020-08-29 13:39:17 +0200589 scriptitem_T *si;
Bram Moolenaarefa94442020-08-08 22:16:00 +0200590 int idx;
591
Bram Moolenaare3d46852020-08-29 13:39:17 +0200592 if (!SCRIPT_ID_VALID(sid))
593 return NULL;
594 si = SCRIPT_ITEM(sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100595 for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
596 {
597 imported_T *import = ((imported_T *)si->sn_imports.ga_data) + idx;
598
Bram Moolenaar4e12a5d2020-02-03 20:50:59 +0100599 if (len == 0 ? STRCMP(name, import->imp_name) == 0
600 : STRLEN(import->imp_name) == len
601 && STRNCMP(name, import->imp_name, len) == 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100602 return import;
603 }
604 return NULL;
605}
606
607/*
Bram Moolenaar20431c92020-03-20 18:39:46 +0100608 * Free all imported variables.
609 */
610 static void
611free_imported(cctx_T *cctx)
612{
613 int idx;
614
615 for (idx = 0; idx < cctx->ctx_imports.ga_len; ++idx)
616 {
617 imported_T *import = ((imported_T *)cctx->ctx_imports.ga_data) + idx;
618
619 vim_free(import->imp_name);
620 }
621 ga_clear(&cctx->ctx_imports);
622}
623
624/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000625 * Called when checking for a following operator at "arg". When the rest of
626 * the line is empty or only a comment, peek the next line. If there is a next
627 * line return a pointer to it and set "nextp".
628 * Otherwise skip over white space.
629 */
630 char_u *
631may_peek_next_line(cctx_T *cctx, char_u *arg, char_u **nextp)
632{
633 char_u *p = skipwhite(arg);
634
635 *nextp = NULL;
636 if (*p == NUL || (VIM_ISWHITE(*arg) && vim9_comment_start(p)))
637 {
638 *nextp = peek_next_line_from_context(cctx);
639 if (*nextp != NULL)
640 return *nextp;
641 }
642 return p;
643}
644
645/*
Bram Moolenaar23c55272020-06-21 16:58:13 +0200646 * Return a pointer to the next line that isn't empty or only contains a
647 * comment. Skips over white space.
648 * Returns NULL if there is none.
649 */
Bram Moolenaar7a4b8982020-07-08 17:36:21 +0200650 char_u *
651peek_next_line_from_context(cctx_T *cctx)
Bram Moolenaar23c55272020-06-21 16:58:13 +0200652{
653 int lnum = cctx->ctx_lnum;
654
655 while (++lnum < cctx->ctx_ufunc->uf_lines.ga_len)
656 {
657 char_u *line = ((char_u **)cctx->ctx_ufunc->uf_lines.ga_data)[lnum];
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +0200658 char_u *p;
Bram Moolenaar23c55272020-06-21 16:58:13 +0200659
Bram Moolenaarba60cc42020-08-12 19:15:33 +0200660 // ignore NULLs inserted for continuation lines
661 if (line != NULL)
662 {
663 p = skipwhite(line);
Bram Moolenaar4b3e1962021-03-18 21:37:55 +0100664 if (vim9_bad_comment(p))
665 return NULL;
Bram Moolenaarba60cc42020-08-12 19:15:33 +0200666 if (*p != NUL && !vim9_comment_start(p))
667 return p;
668 }
Bram Moolenaar23c55272020-06-21 16:58:13 +0200669 }
670 return NULL;
671}
672
673/*
Bram Moolenaare6085c52020-04-12 20:19:16 +0200674 * Get the next line of the function from "cctx".
Bram Moolenaar23c55272020-06-21 16:58:13 +0200675 * Skips over empty lines. Skips over comment lines if "skip_comment" is TRUE.
Bram Moolenaare6085c52020-04-12 20:19:16 +0200676 * Returns NULL when at the end.
677 */
Bram Moolenaar7a4b8982020-07-08 17:36:21 +0200678 char_u *
Bram Moolenaar23c55272020-06-21 16:58:13 +0200679next_line_from_context(cctx_T *cctx, int skip_comment)
Bram Moolenaare6085c52020-04-12 20:19:16 +0200680{
Bram Moolenaar7a092242020-04-16 22:10:49 +0200681 char_u *line;
Bram Moolenaare6085c52020-04-12 20:19:16 +0200682
683 do
684 {
685 ++cctx->ctx_lnum;
686 if (cctx->ctx_lnum >= cctx->ctx_ufunc->uf_lines.ga_len)
Bram Moolenaar7a092242020-04-16 22:10:49 +0200687 {
688 line = NULL;
Bram Moolenaare6085c52020-04-12 20:19:16 +0200689 break;
Bram Moolenaar7a092242020-04-16 22:10:49 +0200690 }
Bram Moolenaare6085c52020-04-12 20:19:16 +0200691 line = ((char_u **)cctx->ctx_ufunc->uf_lines.ga_data)[cctx->ctx_lnum];
Bram Moolenaar7a092242020-04-16 22:10:49 +0200692 cctx->ctx_line_start = line;
Bram Moolenaar25e0f582020-05-25 22:36:50 +0200693 SOURCING_LNUM = cctx->ctx_lnum + 1;
Bram Moolenaar23c55272020-06-21 16:58:13 +0200694 } while (line == NULL || *skipwhite(line) == NUL
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +0200695 || (skip_comment && vim9_comment_start(skipwhite(line))));
Bram Moolenaare6085c52020-04-12 20:19:16 +0200696 return line;
697}
698
699/*
Bram Moolenaar5afd0812021-01-03 18:33:13 +0100700 * Skip over white space at "whitep" and assign to "*arg".
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200701 * If "*arg" is at the end of the line, advance to the next line.
Bram Moolenaar2c330432020-04-13 14:41:35 +0200702 * Also when "whitep" points to white space and "*arg" is on a "#".
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200703 * Return FAIL if beyond the last line, "*arg" is unmodified then.
704 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000705 int
Bram Moolenaar2c330432020-04-13 14:41:35 +0200706may_get_next_line(char_u *whitep, char_u **arg, cctx_T *cctx)
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200707{
Bram Moolenaar5afd0812021-01-03 18:33:13 +0100708 *arg = skipwhite(whitep);
Bram Moolenaar4b3e1962021-03-18 21:37:55 +0100709 if (vim9_bad_comment(*arg))
710 return FAIL;
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200711 if (**arg == NUL || (VIM_ISWHITE(*whitep) && vim9_comment_start(*arg)))
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200712 {
Bram Moolenaar23c55272020-06-21 16:58:13 +0200713 char_u *next = next_line_from_context(cctx, TRUE);
Bram Moolenaar9c7e6dd2020-04-12 20:55:20 +0200714
715 if (next == NULL)
716 return FAIL;
717 *arg = skipwhite(next);
718 }
719 return OK;
720}
721
Bram Moolenaara7eedf32020-07-10 21:50:41 +0200722/*
723 * Idem, and give an error when failed.
724 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000725 int
Bram Moolenaara7eedf32020-07-10 21:50:41 +0200726may_get_next_line_error(char_u *whitep, char_u **arg, cctx_T *cctx)
727{
728 if (may_get_next_line(whitep, arg, cctx) == FAIL)
729 {
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100730 SOURCING_LNUM = cctx->ctx_lnum + 1;
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200731 emsg(_(e_line_incomplete));
Bram Moolenaara7eedf32020-07-10 21:50:41 +0200732 return FAIL;
733 }
734 return OK;
735}
736
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100737/*
Bram Moolenaar04b12692020-05-04 23:24:44 +0200738 * Get a line from the compilation context, compatible with exarg_T getline().
739 * Return a pointer to the line in allocated memory.
740 * Return NULL for end-of-file or some error.
741 */
742 static char_u *
743exarg_getline(
744 int c UNUSED,
745 void *cookie,
746 int indent UNUSED,
Bram Moolenaar66250c92020-08-20 15:02:42 +0200747 getline_opt_T options UNUSED)
Bram Moolenaar04b12692020-05-04 23:24:44 +0200748{
749 cctx_T *cctx = (cctx_T *)cookie;
Bram Moolenaar66250c92020-08-20 15:02:42 +0200750 char_u *p;
Bram Moolenaar04b12692020-05-04 23:24:44 +0200751
Bram Moolenaar66250c92020-08-20 15:02:42 +0200752 for (;;)
Bram Moolenaar04b12692020-05-04 23:24:44 +0200753 {
Bram Moolenaar2914a202020-09-27 18:24:03 +0200754 if (cctx->ctx_lnum >= cctx->ctx_ufunc->uf_lines.ga_len - 1)
Bram Moolenaar66250c92020-08-20 15:02:42 +0200755 return NULL;
756 ++cctx->ctx_lnum;
757 p = ((char_u **)cctx->ctx_ufunc->uf_lines.ga_data)[cctx->ctx_lnum];
758 // Comment lines result in NULL pointers, skip them.
759 if (p != NULL)
760 return vim_strsave(p);
Bram Moolenaar04b12692020-05-04 23:24:44 +0200761 }
Bram Moolenaar04b12692020-05-04 23:24:44 +0200762}
763
Bram Moolenaar7a6eaa02021-03-21 20:53:29 +0100764 void
765fill_exarg_from_cctx(exarg_T *eap, cctx_T *cctx)
766{
767 eap->getline = exarg_getline;
768 eap->cookie = cctx;
769}
770
Bram Moolenaar04b12692020-05-04 23:24:44 +0200771/*
Bram Moolenaardc7c3662021-12-20 15:04:29 +0000772 * Return TRUE if "ufunc" should be compiled, taking into account whether
773 * "profile" indicates profiling is to be done.
774 */
775 int
776func_needs_compiling(ufunc_T *ufunc, compiletype_T compile_type)
777{
778 switch (ufunc->uf_def_status)
779 {
780 case UF_TO_BE_COMPILED:
781 return TRUE;
782
783 case UF_COMPILED:
784 {
785 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
786 + ufunc->uf_dfunc_idx;
787
788 switch (compile_type)
789 {
790 case CT_PROFILE:
791#ifdef FEAT_PROFILE
792 return dfunc->df_instr_prof == NULL;
793#endif
794 case CT_NONE:
795 return dfunc->df_instr == NULL;
796 case CT_DEBUG:
797 return dfunc->df_instr_debug == NULL;
798 }
799 }
800
801 case UF_NOT_COMPILED:
802 case UF_COMPILE_ERROR:
803 case UF_COMPILING:
804 break;
805 }
806 return FALSE;
807}
808
809/*
Bram Moolenaar04b12692020-05-04 23:24:44 +0200810 * Compile a nested :def command.
811 */
812 static char_u *
813compile_nested_function(exarg_T *eap, cctx_T *cctx)
814{
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200815 int is_global = *eap->arg == 'g' && eap->arg[1] == ':';
Bram Moolenaar04b12692020-05-04 23:24:44 +0200816 char_u *name_start = eap->arg;
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200817 char_u *name_end = to_name_end(eap->arg, TRUE);
Bram Moolenaareef21022020-08-01 22:16:43 +0200818 char_u *lambda_name;
Bram Moolenaar04b12692020-05-04 23:24:44 +0200819 ufunc_T *ufunc;
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100820 int r = FAIL;
Bram Moolenaarffcfddc2021-07-11 20:22:30 +0200821 compiletype_T compile_type;
Bram Moolenaar04b12692020-05-04 23:24:44 +0200822
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +0200823 if (eap->forceit)
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200824 {
825 emsg(_(e_cannot_use_bang_with_nested_def));
826 return NULL;
827 }
828
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100829 if (*name_start == '/')
830 {
831 name_end = skip_regexp(name_start + 1, '/', TRUE);
832 if (*name_end == '/')
833 ++name_end;
Bram Moolenaar63b91732021-08-05 20:40:03 +0200834 set_nextcmd(eap, name_end);
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100835 }
836 if (name_end == name_start || *skipwhite(name_end) != '(')
837 {
838 if (!ends_excmd2(name_start, name_end))
839 {
840 semsg(_(e_invalid_command_str), eap->cmd);
841 return NULL;
842 }
843
844 // "def" or "def Name": list functions
845 if (generate_DEF(cctx, name_start, name_end - name_start) == FAIL)
846 return NULL;
847 return eap->nextcmd == NULL ? (char_u *)"" : eap->nextcmd;
848 }
849
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200850 // Only g:Func() can use a namespace.
851 if (name_start[1] == ':' && !is_global)
852 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200853 semsg(_(e_namespace_not_supported_str), name_start);
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200854 return NULL;
855 }
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100856 if (check_defined(name_start, name_end - name_start, cctx, FALSE) == FAIL)
Bram Moolenaareef21022020-08-01 22:16:43 +0200857 return NULL;
858
Bram Moolenaar04b12692020-05-04 23:24:44 +0200859 eap->arg = name_end;
Bram Moolenaar7a6eaa02021-03-21 20:53:29 +0100860 fill_exarg_from_cctx(eap, cctx);
861
Bram Moolenaar04b12692020-05-04 23:24:44 +0200862 eap->forceit = FALSE;
Bram Moolenaar38453522021-11-28 22:00:12 +0000863 // We use the special <Lamba>99 name, but it's not really a lambda.
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100864 lambda_name = vim_strsave(get_lambda_name());
865 if (lambda_name == NULL)
866 return NULL;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200867 ufunc = define_function(eap, lambda_name);
Bram Moolenaar04b12692020-05-04 23:24:44 +0200868
Bram Moolenaar822ba242020-05-24 23:00:18 +0200869 if (ufunc == NULL)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100870 {
871 r = eap->skip ? OK : FAIL;
872 goto theend;
873 }
Bram Moolenaar8863bda2021-03-17 18:42:08 +0100874
875 // copy over the block scope IDs before compiling
876 if (!is_global && cctx->ctx_ufunc->uf_block_depth > 0)
877 {
878 int block_depth = cctx->ctx_ufunc->uf_block_depth;
879
880 ufunc->uf_block_ids = ALLOC_MULT(int, block_depth);
881 if (ufunc->uf_block_ids != NULL)
882 {
883 mch_memmove(ufunc->uf_block_ids, cctx->ctx_ufunc->uf_block_ids,
884 sizeof(int) * block_depth);
885 ufunc->uf_block_depth = block_depth;
886 }
887 }
888
Bram Moolenaarffcfddc2021-07-11 20:22:30 +0200889 compile_type = COMPILE_TYPE(ufunc);
890#ifdef FEAT_PROFILE
891 // If the outer function is profiled, also compile the nested function for
892 // profiling.
893 if (cctx->ctx_compile_type == CT_PROFILE)
894 compile_type = CT_PROFILE;
895#endif
896 if (func_needs_compiling(ufunc, compile_type)
897 && compile_def_function(ufunc, TRUE, compile_type, cctx) == FAIL)
Bram Moolenaar4ee711f2020-09-23 18:51:11 +0200898 {
899 func_ptr_unref(ufunc);
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100900 goto theend;
Bram Moolenaar4ee711f2020-09-23 18:51:11 +0200901 }
Bram Moolenaar04b12692020-05-04 23:24:44 +0200902
Bram Moolenaar648594e2021-07-11 17:55:01 +0200903#ifdef FEAT_PROFILE
904 // When the outer function is compiled for profiling, the nested function
905 // may be called without profiling. Compile it here in the right context.
Bram Moolenaarffcfddc2021-07-11 20:22:30 +0200906 if (compile_type == CT_PROFILE && func_needs_compiling(ufunc, CT_NONE))
Bram Moolenaar648594e2021-07-11 17:55:01 +0200907 compile_def_function(ufunc, FALSE, CT_NONE, cctx);
908#endif
909
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200910 if (is_global)
911 {
912 char_u *func_name = vim_strnsave(name_start + 2,
913 name_end - name_start - 2);
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200914
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200915 if (func_name == NULL)
916 r = FAIL;
917 else
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100918 {
Bram Moolenaareef21022020-08-01 22:16:43 +0200919 r = generate_NEWFUNC(cctx, lambda_name, func_name);
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100920 lambda_name = NULL;
921 }
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200922 }
923 else
924 {
925 // Define a local variable for the function reference.
Bram Moolenaare8211a32020-10-09 22:04:29 +0200926 lvar_T *lvar = reserve_local(cctx, name_start, name_end - name_start,
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200927 TRUE, ufunc->uf_func_type);
Bram Moolenaare8211a32020-10-09 22:04:29 +0200928
Bram Moolenaareef21022020-08-01 22:16:43 +0200929 if (lvar == NULL)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100930 goto theend;
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200931 if (generate_FUNCREF(cctx, ufunc) == FAIL)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100932 goto theend;
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200933 r = generate_STORE(cctx, ISN_STORE, lvar->lv_idx, NULL);
934 }
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100935
936theend:
937 vim_free(lambda_name);
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200938 return r == FAIL ? NULL : (char_u *)"";
Bram Moolenaar04b12692020-05-04 23:24:44 +0200939}
940
941/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100942 * Return the length of an assignment operator, or zero if there isn't one.
943 */
944 int
945assignment_len(char_u *p, int *heredoc)
946{
947 if (*p == '=')
948 {
949 if (p[1] == '<' && p[2] == '<')
950 {
951 *heredoc = TRUE;
952 return 3;
953 }
954 return 1;
955 }
956 if (vim_strchr((char_u *)"+-*/%", *p) != NULL && p[1] == '=')
957 return 2;
958 if (STRNCMP(p, "..=", 3) == 0)
959 return 3;
960 return 0;
961}
962
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100963/*
Bram Moolenaar1cc2a942020-05-10 19:10:31 +0200964 * Generate the load instruction for "name".
965 */
966 static void
967generate_loadvar(
968 cctx_T *cctx,
969 assign_dest_T dest,
970 char_u *name,
971 lvar_T *lvar,
972 type_T *type)
973{
974 switch (dest)
975 {
976 case dest_option:
Bram Moolenaardcb53be2021-12-09 14:23:43 +0000977 case dest_func_option:
Bram Moolenaar1cc2a942020-05-10 19:10:31 +0200978 generate_LOAD(cctx, ISN_LOADOPT, 0, name, type);
979 break;
980 case dest_global:
Bram Moolenaar03290b82020-12-19 16:30:44 +0100981 if (vim_strchr(name, AUTOLOAD_CHAR) == NULL)
982 generate_LOAD(cctx, ISN_LOADG, 0, name + 2, type);
983 else
984 generate_LOAD(cctx, ISN_LOADAUTO, 0, name, type);
Bram Moolenaar1cc2a942020-05-10 19:10:31 +0200985 break;
986 case dest_buffer:
987 generate_LOAD(cctx, ISN_LOADB, 0, name + 2, type);
988 break;
989 case dest_window:
990 generate_LOAD(cctx, ISN_LOADW, 0, name + 2, type);
991 break;
992 case dest_tab:
993 generate_LOAD(cctx, ISN_LOADT, 0, name + 2, type);
994 break;
995 case dest_script:
996 compile_load_scriptvar(cctx,
997 name + (name[1] == ':' ? 2 : 0), NULL, NULL, TRUE);
998 break;
999 case dest_env:
1000 // Include $ in the name here
1001 generate_LOAD(cctx, ISN_LOADENV, 0, name, type);
1002 break;
1003 case dest_reg:
1004 generate_LOAD(cctx, ISN_LOADREG, name[1], NULL, &t_string);
1005 break;
1006 case dest_vimvar:
1007 generate_LOADV(cctx, name + 2, TRUE);
1008 break;
1009 case dest_local:
Bram Moolenaarab360522021-01-10 14:02:28 +01001010 if (lvar->lv_from_outer > 0)
1011 generate_LOADOUTER(cctx, lvar->lv_idx, lvar->lv_from_outer,
1012 type);
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001013 else
1014 generate_LOAD(cctx, ISN_LOAD, lvar->lv_idx, NULL, type);
1015 break;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001016 case dest_expr:
1017 // list or dict value should already be on the stack.
1018 break;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001019 }
1020}
1021
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001022/*
1023 * Skip over "[expr]" or ".member".
1024 * Does not check for any errors.
1025 */
1026 static char_u *
1027skip_index(char_u *start)
1028{
1029 char_u *p = start;
1030
1031 if (*p == '[')
1032 {
1033 p = skipwhite(p + 1);
1034 (void)skip_expr(&p, NULL);
1035 p = skipwhite(p);
1036 if (*p == ']')
1037 return p + 1;
1038 return p;
1039 }
1040 // if (*p == '.')
1041 return to_name_end(p + 1, TRUE);
1042}
1043
Bram Moolenaare55b1c02020-06-21 15:52:59 +02001044 void
1045vim9_declare_error(char_u *name)
1046{
1047 char *scope = "";
1048
1049 switch (*name)
1050 {
Bram Moolenaar7fe87552020-06-21 20:38:28 +02001051 case 'g': scope = _("global"); break;
1052 case 'b': scope = _("buffer"); break;
1053 case 'w': scope = _("window"); break;
1054 case 't': scope = _("tab"); break;
1055 case 'v': scope = "v:"; break;
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02001056 case '$': semsg(_(e_cannot_declare_an_environment_variable), name);
Bram Moolenaarc2ee44c2020-08-02 16:59:00 +02001057 return;
Bram Moolenaar451c2e32020-08-15 16:33:28 +02001058 case '&': semsg(_(e_cannot_declare_an_option), name);
Bram Moolenaarc2ee44c2020-08-02 16:59:00 +02001059 return;
Bram Moolenaar7cb6fc22020-08-21 22:36:47 +02001060 case '@': semsg(_(e_cannot_declare_a_register_str), name);
Bram Moolenaarc2ee44c2020-08-02 16:59:00 +02001061 return;
Bram Moolenaar7fe87552020-06-21 20:38:28 +02001062 default: return;
Bram Moolenaare55b1c02020-06-21 15:52:59 +02001063 }
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02001064 semsg(_(e_cannot_declare_a_scope_variable), scope, name);
Bram Moolenaare55b1c02020-06-21 15:52:59 +02001065}
1066
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001067/*
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001068 * For one assignment figure out the type of destination. Return it in "dest".
1069 * When not recognized "dest" is not set.
Yegappan Lakshmanan64095532021-12-06 11:03:55 +00001070 * For an option "option_scope" is set.
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001071 * For a v:var "vimvaridx" is set.
1072 * "type" is set to the destination type if known, unchanted otherwise.
1073 * Return FAIL if an error message was given.
1074 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001075 int
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001076get_var_dest(
1077 char_u *name,
1078 assign_dest_T *dest,
1079 int cmdidx,
Yegappan Lakshmanan64095532021-12-06 11:03:55 +00001080 int *option_scope,
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001081 int *vimvaridx,
1082 type_T **type,
1083 cctx_T *cctx)
1084{
1085 char_u *p;
1086
1087 if (*name == '&')
1088 {
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001089 int cc;
1090 long numval;
1091 getoption_T opt_type;
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001092 int opt_p_flags;
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001093
1094 *dest = dest_option;
1095 if (cmdidx == CMD_final || cmdidx == CMD_const)
1096 {
1097 emsg(_(e_const_option));
1098 return FAIL;
1099 }
1100 p = name;
Yegappan Lakshmanan64095532021-12-06 11:03:55 +00001101 p = find_option_end(&p, option_scope);
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001102 if (p == NULL)
1103 {
1104 // cannot happen?
Bram Moolenaar108010a2021-06-27 22:03:33 +02001105 emsg(_(e_unexpected_characters_in_assignment));
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001106 return FAIL;
1107 }
1108 cc = *p;
1109 *p = NUL;
1110 opt_type = get_option_value(skip_option_env_lead(name),
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001111 &numval, NULL, &opt_p_flags, *option_scope);
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001112 *p = cc;
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001113 switch (opt_type)
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001114 {
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001115 case gov_unknown:
Bram Moolenaare1242042021-12-16 20:56:57 +00001116 semsg(_(e_unknown_option_str), name);
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001117 return FAIL;
1118 case gov_string:
1119 case gov_hidden_string:
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001120 if (opt_p_flags & P_FUNC)
1121 {
1122 // might be a Funcref, check the type later
1123 *type = &t_any;
1124 *dest = dest_func_option;
1125 }
1126 else
1127 {
1128 *type = &t_string;
1129 }
Bram Moolenaardd1f4262020-12-31 17:41:01 +01001130 break;
1131 case gov_bool:
1132 case gov_hidden_bool:
1133 *type = &t_bool;
1134 break;
1135 case gov_number:
1136 case gov_hidden_number:
1137 *type = &t_number;
1138 break;
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001139 }
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001140 }
1141 else if (*name == '$')
1142 {
1143 *dest = dest_env;
1144 *type = &t_string;
1145 }
1146 else if (*name == '@')
1147 {
Bram Moolenaar13e45d12021-06-26 13:28:35 +02001148 if (name[1] != '@'
1149 && (!valid_yank_reg(name[1], FALSE) || name[1] == '.'))
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001150 {
1151 emsg_invreg(name[1]);
1152 return FAIL;
1153 }
1154 *dest = dest_reg;
Bram Moolenaar74f4a962021-06-17 21:03:07 +02001155 *type = name[1] == '#' ? &t_number_or_string : &t_string;
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001156 }
1157 else if (STRNCMP(name, "g:", 2) == 0)
1158 {
1159 *dest = dest_global;
1160 }
1161 else if (STRNCMP(name, "b:", 2) == 0)
1162 {
1163 *dest = dest_buffer;
1164 }
1165 else if (STRNCMP(name, "w:", 2) == 0)
1166 {
1167 *dest = dest_window;
1168 }
1169 else if (STRNCMP(name, "t:", 2) == 0)
1170 {
1171 *dest = dest_tab;
1172 }
1173 else if (STRNCMP(name, "v:", 2) == 0)
1174 {
1175 typval_T *vtv;
1176 int di_flags;
1177
1178 *vimvaridx = find_vim_var(name + 2, &di_flags);
1179 if (*vimvaridx < 0)
1180 {
1181 semsg(_(e_variable_not_found_str), name);
1182 return FAIL;
1183 }
1184 // We use the current value of "sandbox" here, is that OK?
1185 if (var_check_ro(di_flags, name, FALSE))
1186 return FAIL;
1187 *dest = dest_vimvar;
1188 vtv = get_vim_var_tv(*vimvaridx);
1189 *type = typval2type_vimvar(vtv, cctx->ctx_type_list);
1190 }
1191 return OK;
1192}
1193
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001194 static int
Bram Moolenaar752fc692021-01-04 21:57:11 +01001195is_decl_command(int cmdidx)
1196{
1197 return cmdidx == CMD_let || cmdidx == CMD_var
1198 || cmdidx == CMD_final || cmdidx == CMD_const;
1199}
1200
1201/*
1202 * Figure out the LHS type and other properties for an assignment or one item
1203 * of ":unlet" with an index.
1204 * Returns OK or FAIL.
1205 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001206 int
Bram Moolenaar752fc692021-01-04 21:57:11 +01001207compile_lhs(
1208 char_u *var_start,
1209 lhs_T *lhs,
1210 int cmdidx,
1211 int heredoc,
1212 int oplen,
1213 cctx_T *cctx)
1214{
1215 char_u *var_end;
1216 int is_decl = is_decl_command(cmdidx);
1217
1218 CLEAR_POINTER(lhs);
1219 lhs->lhs_dest = dest_local;
1220 lhs->lhs_vimvaridx = -1;
1221 lhs->lhs_scriptvar_idx = -1;
1222
1223 // "dest_end" is the end of the destination, including "[expr]" or
1224 // ".name".
1225 // "var_end" is the end of the variable/option/etc. name.
1226 lhs->lhs_dest_end = skip_var_one(var_start, FALSE);
1227 if (*var_start == '@')
1228 var_end = var_start + 2;
1229 else
1230 {
1231 // skip over the leading "&", "&l:", "&g:" and "$"
1232 var_end = skip_option_env_lead(var_start);
1233 var_end = to_name_end(var_end, TRUE);
1234 }
1235
1236 // "a: type" is declaring variable "a" with a type, not dict "a:".
1237 if (is_decl && lhs->lhs_dest_end == var_start + 2
1238 && lhs->lhs_dest_end[-1] == ':')
1239 --lhs->lhs_dest_end;
1240 if (is_decl && var_end == var_start + 2 && var_end[-1] == ':')
1241 --var_end;
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001242 lhs->lhs_end = lhs->lhs_dest_end;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001243
1244 // compute the length of the destination without "[expr]" or ".name"
1245 lhs->lhs_varlen = var_end - var_start;
Bram Moolenaar753bcf82021-04-21 14:24:24 +02001246 lhs->lhs_varlen_total = lhs->lhs_varlen;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001247 lhs->lhs_name = vim_strnsave(var_start, lhs->lhs_varlen);
1248 if (lhs->lhs_name == NULL)
1249 return FAIL;
Bram Moolenaar08251752021-01-11 21:20:18 +01001250
1251 if (lhs->lhs_dest_end > var_start + lhs->lhs_varlen)
1252 // Something follows after the variable: "var[idx]" or "var.key".
1253 lhs->lhs_has_index = TRUE;
1254
Bram Moolenaar752fc692021-01-04 21:57:11 +01001255 if (heredoc)
1256 lhs->lhs_type = &t_list_string;
1257 else
1258 lhs->lhs_type = &t_any;
1259
1260 if (cctx->ctx_skip != SKIP_YES)
1261 {
1262 int declare_error = FALSE;
1263
1264 if (get_var_dest(lhs->lhs_name, &lhs->lhs_dest, cmdidx,
1265 &lhs->lhs_opt_flags, &lhs->lhs_vimvaridx,
1266 &lhs->lhs_type, cctx) == FAIL)
1267 return FAIL;
Bram Moolenaard877a572021-04-01 19:42:48 +02001268 if (lhs->lhs_dest != dest_local
1269 && cmdidx != CMD_const && cmdidx != CMD_final)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001270 {
1271 // Specific kind of variable recognized.
1272 declare_error = is_decl;
1273 }
1274 else
1275 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01001276 // No specific kind of variable recognized, just a name.
Bram Moolenaard0edaf92021-05-28 21:06:08 +02001277 if (check_reserved_name(lhs->lhs_name) == FAIL)
1278 return FAIL;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001279
1280 if (lookup_local(var_start, lhs->lhs_varlen,
1281 &lhs->lhs_local_lvar, cctx) == OK)
1282 lhs->lhs_lvar = &lhs->lhs_local_lvar;
1283 else
1284 {
1285 CLEAR_FIELD(lhs->lhs_arg_lvar);
1286 if (arg_exists(var_start, lhs->lhs_varlen,
1287 &lhs->lhs_arg_lvar.lv_idx, &lhs->lhs_arg_lvar.lv_type,
1288 &lhs->lhs_arg_lvar.lv_from_outer, cctx) == OK)
1289 {
1290 if (is_decl)
1291 {
1292 semsg(_(e_str_is_used_as_argument), lhs->lhs_name);
1293 return FAIL;
1294 }
1295 lhs->lhs_lvar = &lhs->lhs_arg_lvar;
1296 }
1297 }
1298 if (lhs->lhs_lvar != NULL)
1299 {
1300 if (is_decl)
1301 {
1302 semsg(_(e_variable_already_declared), lhs->lhs_name);
1303 return FAIL;
1304 }
1305 }
1306 else
1307 {
1308 int script_namespace = lhs->lhs_varlen > 1
1309 && STRNCMP(var_start, "s:", 2) == 0;
1310 int script_var = (script_namespace
1311 ? script_var_exists(var_start + 2, lhs->lhs_varlen - 2,
Bram Moolenaar15e5e532021-04-07 21:21:13 +02001312 cctx)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001313 : script_var_exists(var_start, lhs->lhs_varlen,
Bram Moolenaar15e5e532021-04-07 21:21:13 +02001314 cctx)) == OK;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001315 imported_T *import =
1316 find_imported(var_start, lhs->lhs_varlen, cctx);
1317
1318 if (script_namespace || script_var || import != NULL)
1319 {
1320 char_u *rawname = lhs->lhs_name
1321 + (lhs->lhs_name[1] == ':' ? 2 : 0);
1322
1323 if (is_decl)
1324 {
1325 if (script_namespace)
1326 semsg(_(e_cannot_declare_script_variable_in_function),
1327 lhs->lhs_name);
1328 else
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001329 semsg(_(e_variable_already_declared_in_script_str),
Bram Moolenaar752fc692021-01-04 21:57:11 +01001330 lhs->lhs_name);
1331 return FAIL;
1332 }
1333 else if (cctx->ctx_ufunc->uf_script_ctx_version
1334 == SCRIPT_VERSION_VIM9
1335 && script_namespace
1336 && !script_var && import == NULL)
1337 {
1338 semsg(_(e_unknown_variable_str), lhs->lhs_name);
1339 return FAIL;
1340 }
1341
1342 lhs->lhs_dest = dest_script;
1343
1344 // existing script-local variables should have a type
1345 lhs->lhs_scriptvar_sid = current_sctx.sc_sid;
1346 if (import != NULL)
1347 lhs->lhs_scriptvar_sid = import->imp_sid;
1348 if (SCRIPT_ID_VALID(lhs->lhs_scriptvar_sid))
1349 {
Bram Moolenaar08251752021-01-11 21:20:18 +01001350 // Check writable only when no index follows.
Bram Moolenaar752fc692021-01-04 21:57:11 +01001351 lhs->lhs_scriptvar_idx = get_script_item_idx(
Bram Moolenaar08251752021-01-11 21:20:18 +01001352 lhs->lhs_scriptvar_sid, rawname,
1353 lhs->lhs_has_index ? ASSIGN_FINAL : ASSIGN_CONST,
1354 cctx);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001355 if (lhs->lhs_scriptvar_idx >= 0)
1356 {
1357 scriptitem_T *si = SCRIPT_ITEM(
1358 lhs->lhs_scriptvar_sid);
1359 svar_T *sv =
1360 ((svar_T *)si->sn_var_vals.ga_data)
1361 + lhs->lhs_scriptvar_idx;
1362 lhs->lhs_type = sv->sv_type;
1363 }
1364 }
1365 }
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001366 else if (check_defined(var_start, lhs->lhs_varlen, cctx, FALSE)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001367 == FAIL)
1368 return FAIL;
1369 }
1370 }
1371
1372 if (declare_error)
1373 {
1374 vim9_declare_error(lhs->lhs_name);
1375 return FAIL;
1376 }
1377 }
1378
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001379 // handle "a:name" as a name, not index "name" in "a"
Bram Moolenaar752fc692021-01-04 21:57:11 +01001380 if (lhs->lhs_varlen > 1 || var_start[lhs->lhs_varlen] != ':')
1381 var_end = lhs->lhs_dest_end;
1382
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001383 if (lhs->lhs_dest != dest_option && lhs->lhs_dest != dest_func_option)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001384 {
1385 if (is_decl && *var_end == ':')
1386 {
1387 char_u *p;
1388
1389 // parse optional type: "let var: type = expr"
1390 if (!VIM_ISWHITE(var_end[1]))
1391 {
Bram Moolenaarc3fc75d2021-02-07 15:28:09 +01001392 semsg(_(e_white_space_required_after_str_str), ":", var_end);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001393 return FAIL;
1394 }
1395 p = skipwhite(var_end + 1);
1396 lhs->lhs_type = parse_type(&p, cctx->ctx_type_list, TRUE);
1397 if (lhs->lhs_type == NULL)
1398 return FAIL;
1399 lhs->lhs_has_type = TRUE;
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001400 lhs->lhs_end = p;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001401 }
1402 else if (lhs->lhs_lvar != NULL)
1403 lhs->lhs_type = lhs->lhs_lvar->lv_type;
1404 }
1405
Bram Moolenaare42939a2021-04-05 17:11:17 +02001406 if (oplen == 3 && !heredoc
1407 && lhs->lhs_dest != dest_global
1408 && !lhs->lhs_has_index
1409 && lhs->lhs_type->tt_type != VAR_STRING
1410 && lhs->lhs_type->tt_type != VAR_ANY)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001411 {
1412 emsg(_(e_can_only_concatenate_to_string));
1413 return FAIL;
1414 }
1415
1416 if (lhs->lhs_lvar == NULL && lhs->lhs_dest == dest_local
1417 && cctx->ctx_skip != SKIP_YES)
1418 {
1419 if (oplen > 1 && !heredoc)
1420 {
1421 // +=, /=, etc. require an existing variable
1422 semsg(_(e_cannot_use_operator_on_new_variable), lhs->lhs_name);
1423 return FAIL;
1424 }
1425 if (!is_decl)
1426 {
1427 semsg(_(e_unknown_variable_str), lhs->lhs_name);
1428 return FAIL;
1429 }
1430
Bram Moolenaar3f327882021-03-17 20:56:38 +01001431 // Check the name is valid for a funcref.
Bram Moolenaar752fc692021-01-04 21:57:11 +01001432 if ((lhs->lhs_type->tt_type == VAR_FUNC
1433 || lhs->lhs_type->tt_type == VAR_PARTIAL)
Bram Moolenaar3f327882021-03-17 20:56:38 +01001434 && var_wrong_func_name(lhs->lhs_name, TRUE))
Bram Moolenaar752fc692021-01-04 21:57:11 +01001435 return FAIL;
Bram Moolenaar3f327882021-03-17 20:56:38 +01001436
1437 // New local variable.
Bram Moolenaar752fc692021-01-04 21:57:11 +01001438 lhs->lhs_lvar = reserve_local(cctx, var_start, lhs->lhs_varlen,
1439 cmdidx == CMD_final || cmdidx == CMD_const, lhs->lhs_type);
1440 if (lhs->lhs_lvar == NULL)
1441 return FAIL;
1442 lhs->lhs_new_local = TRUE;
1443 }
1444
1445 lhs->lhs_member_type = lhs->lhs_type;
Bram Moolenaar08251752021-01-11 21:20:18 +01001446 if (lhs->lhs_has_index)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001447 {
Bram Moolenaarc750d912021-11-29 22:02:12 +00001448 char_u *after = var_start + lhs->lhs_varlen;
1449 char_u *p;
1450
Bram Moolenaar752fc692021-01-04 21:57:11 +01001451 // Something follows after the variable: "var[idx]" or "var.key".
Bram Moolenaar752fc692021-01-04 21:57:11 +01001452 if (is_decl)
1453 {
1454 emsg(_(e_cannot_use_index_when_declaring_variable));
1455 return FAIL;
1456 }
1457
Bram Moolenaarc750d912021-11-29 22:02:12 +00001458 // Now: var_start[lhs->lhs_varlen] is '[' or '.'
1459 // Only the last index is used below, if there are others
1460 // before it generate code for the expression. Thus for
1461 // "ll[1][2]" the expression is "ll[1]" and "[2]" is the index.
1462 for (;;)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001463 {
Bram Moolenaarc750d912021-11-29 22:02:12 +00001464 p = skip_index(after);
1465 if (*p != '[' && *p != '.')
Bram Moolenaar752fc692021-01-04 21:57:11 +01001466 {
Bram Moolenaarc750d912021-11-29 22:02:12 +00001467 lhs->lhs_varlen_total = p - var_start;
1468 break;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001469 }
Bram Moolenaarc750d912021-11-29 22:02:12 +00001470 after = p;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001471 }
Bram Moolenaarc750d912021-11-29 22:02:12 +00001472 if (after > var_start + lhs->lhs_varlen)
1473 {
1474 lhs->lhs_varlen = after - var_start;
1475 lhs->lhs_dest = dest_expr;
1476 // We don't know the type before evaluating the expression,
1477 // use "any" until then.
1478 lhs->lhs_type = &t_any;
1479 }
1480
1481 if (lhs->lhs_type->tt_member == NULL)
1482 lhs->lhs_member_type = &t_any;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001483 else
Bram Moolenaarc750d912021-11-29 22:02:12 +00001484 lhs->lhs_member_type = lhs->lhs_type->tt_member;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001485 }
1486 return OK;
1487}
1488
1489/*
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001490 * Figure out the LHS and check a few errors.
1491 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001492 int
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001493compile_assign_lhs(
1494 char_u *var_start,
1495 lhs_T *lhs,
1496 int cmdidx,
1497 int is_decl,
1498 int heredoc,
1499 int oplen,
1500 cctx_T *cctx)
1501{
1502 if (compile_lhs(var_start, lhs, cmdidx, heredoc, oplen, cctx) == FAIL)
1503 return FAIL;
1504
1505 if (!lhs->lhs_has_index && lhs->lhs_lvar == &lhs->lhs_arg_lvar)
1506 {
1507 semsg(_(e_cannot_assign_to_argument), lhs->lhs_name);
1508 return FAIL;
1509 }
1510 if (!is_decl && lhs->lhs_lvar != NULL
1511 && lhs->lhs_lvar->lv_const && !lhs->lhs_has_index)
1512 {
1513 semsg(_(e_cannot_assign_to_constant), lhs->lhs_name);
1514 return FAIL;
1515 }
1516 return OK;
1517}
1518
1519/*
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02001520 * Return TRUE if "lhs" has a range index: "[expr : expr]".
1521 */
1522 static int
1523has_list_index(char_u *idx_start, cctx_T *cctx)
1524{
1525 char_u *p = idx_start;
1526 int save_skip;
1527
1528 if (*p != '[')
1529 return FALSE;
1530
1531 p = skipwhite(p + 1);
1532 if (*p == ':')
1533 return TRUE;
1534
1535 save_skip = cctx->ctx_skip;
1536 cctx->ctx_skip = SKIP_YES;
1537 (void)compile_expr0(&p, cctx);
1538 cctx->ctx_skip = save_skip;
1539 return *skipwhite(p) == ':';
1540}
1541
1542/*
Bram Moolenaare42939a2021-04-05 17:11:17 +02001543 * For an assignment with an index, compile the "idx" in "var[idx]" or "key" in
1544 * "var.key".
Bram Moolenaar752fc692021-01-04 21:57:11 +01001545 */
1546 static int
Bram Moolenaare42939a2021-04-05 17:11:17 +02001547compile_assign_index(
Bram Moolenaar752fc692021-01-04 21:57:11 +01001548 char_u *var_start,
1549 lhs_T *lhs,
Bram Moolenaare42939a2021-04-05 17:11:17 +02001550 int *range,
Bram Moolenaar752fc692021-01-04 21:57:11 +01001551 cctx_T *cctx)
1552{
Bram Moolenaar752fc692021-01-04 21:57:11 +01001553 size_t varlen = lhs->lhs_varlen;
Bram Moolenaare42939a2021-04-05 17:11:17 +02001554 char_u *p;
1555 int r = OK;
Bram Moolenaar68452172021-04-12 21:21:02 +02001556 int need_white_before = TRUE;
1557 int empty_second;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001558
Bram Moolenaar752fc692021-01-04 21:57:11 +01001559 p = var_start + varlen;
1560 if (*p == '[')
1561 {
1562 p = skipwhite(p + 1);
Bram Moolenaar68452172021-04-12 21:21:02 +02001563 if (*p == ':')
1564 {
1565 // empty first index, push zero
1566 r = generate_PUSHNR(cctx, 0);
1567 need_white_before = FALSE;
1568 }
1569 else
1570 r = compile_expr0(&p, cctx);
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001571
1572 if (r == OK && *skipwhite(p) == ':')
1573 {
1574 // unlet var[idx : idx]
Bram Moolenaar68452172021-04-12 21:21:02 +02001575 // blob[idx : idx] = value
Bram Moolenaare42939a2021-04-05 17:11:17 +02001576 *range = TRUE;
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001577 p = skipwhite(p);
Bram Moolenaar68452172021-04-12 21:21:02 +02001578 empty_second = *skipwhite(p + 1) == ']';
1579 if ((need_white_before && !IS_WHITE_OR_NUL(p[-1]))
1580 || (!empty_second && !IS_WHITE_OR_NUL(p[1])))
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001581 {
1582 semsg(_(e_white_space_required_before_and_after_str_at_str),
1583 ":", p);
1584 return FAIL;
1585 }
1586 p = skipwhite(p + 1);
Bram Moolenaar68452172021-04-12 21:21:02 +02001587 if (*p == ']')
1588 // empty second index, push "none"
1589 r = generate_PUSHSPEC(cctx, VVAL_NONE);
1590 else
1591 r = compile_expr0(&p, cctx);
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001592 }
1593
Bram Moolenaar752fc692021-01-04 21:57:11 +01001594 if (r == OK && *skipwhite(p) != ']')
1595 {
1596 // this should not happen
Bram Moolenaare1242042021-12-16 20:56:57 +00001597 emsg(_(e_missing_closing_square_brace));
Bram Moolenaar752fc692021-01-04 21:57:11 +01001598 r = FAIL;
1599 }
1600 }
1601 else // if (*p == '.')
1602 {
1603 char_u *key_end = to_name_end(p + 1, TRUE);
1604 char_u *key = vim_strnsave(p + 1, key_end - p - 1);
1605
Zdenek Dohnal9fe17d42021-08-04 22:30:52 +02001606 r = generate_PUSHS(cctx, &key);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001607 }
Bram Moolenaare42939a2021-04-05 17:11:17 +02001608 return r;
1609}
1610
1611/*
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001612 * For a LHS with an index, load the variable to be indexed.
1613 */
1614 static int
1615compile_load_lhs(
1616 lhs_T *lhs,
1617 char_u *var_start,
1618 type_T *rhs_type,
1619 cctx_T *cctx)
1620{
1621 if (lhs->lhs_dest == dest_expr)
1622 {
1623 size_t varlen = lhs->lhs_varlen;
1624 int c = var_start[varlen];
Bram Moolenaare97976b2021-08-01 13:17:17 +02001625 int lines_len = cctx->ctx_ufunc->uf_lines.ga_len;
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001626 char_u *p = var_start;
1627 garray_T *stack = &cctx->ctx_type_stack;
Bram Moolenaare97976b2021-08-01 13:17:17 +02001628 int res;
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001629
Bram Moolenaare97976b2021-08-01 13:17:17 +02001630 // Evaluate "ll[expr]" of "ll[expr][idx]". End the line with a NUL and
1631 // limit the lines array length to avoid skipping to a following line.
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001632 var_start[varlen] = NUL;
Bram Moolenaare97976b2021-08-01 13:17:17 +02001633 cctx->ctx_ufunc->uf_lines.ga_len = cctx->ctx_lnum + 1;
1634 res = compile_expr0(&p, cctx);
1635 var_start[varlen] = c;
1636 cctx->ctx_ufunc->uf_lines.ga_len = lines_len;
1637 if (res == FAIL || p != var_start + varlen)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001638 {
1639 // this should not happen
Bram Moolenaare97976b2021-08-01 13:17:17 +02001640 if (res != FAIL)
Bram Moolenaare1242042021-12-16 20:56:57 +00001641 emsg(_(e_missing_closing_square_brace));
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001642 return FAIL;
1643 }
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001644
1645 lhs->lhs_type = stack->ga_len == 0 ? &t_void
1646 : ((type_T **)stack->ga_data)[stack->ga_len - 1];
1647 // now we can properly check the type
1648 if (rhs_type != NULL && lhs->lhs_type->tt_member != NULL
1649 && rhs_type != &t_void
1650 && need_type(rhs_type, lhs->lhs_type->tt_member, -2, 0, cctx,
1651 FALSE, FALSE) == FAIL)
1652 return FAIL;
1653 }
1654 else
1655 generate_loadvar(cctx, lhs->lhs_dest, lhs->lhs_name,
1656 lhs->lhs_lvar, lhs->lhs_type);
1657 return OK;
1658}
1659
1660/*
Bram Moolenaara369c3d2021-04-21 16:00:10 +02001661 * Produce code for loading "lhs" and also take care of an index.
1662 * Return OK/FAIL.
1663 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001664 int
Bram Moolenaara369c3d2021-04-21 16:00:10 +02001665compile_load_lhs_with_index(lhs_T *lhs, char_u *var_start, cctx_T *cctx)
1666{
1667 compile_load_lhs(lhs, var_start, NULL, cctx);
1668
1669 if (lhs->lhs_has_index)
1670 {
1671 int range = FALSE;
1672
1673 // Get member from list or dict. First compile the
1674 // index value.
1675 if (compile_assign_index(var_start, lhs, &range, cctx) == FAIL)
1676 return FAIL;
1677 if (range)
1678 {
1679 semsg(_(e_cannot_use_range_with_assignment_operator_str),
1680 var_start);
1681 return FAIL;
1682 }
1683
1684 // Get the member.
Bram Moolenaarb1b6f4d2021-09-13 18:25:54 +02001685 if (compile_member(FALSE, NULL, cctx) == FAIL)
Bram Moolenaara369c3d2021-04-21 16:00:10 +02001686 return FAIL;
1687 }
1688 return OK;
1689}
1690
1691/*
Bram Moolenaare42939a2021-04-05 17:11:17 +02001692 * Assignment to a list or dict member, or ":unlet" for the item, using the
1693 * information in "lhs".
1694 * Returns OK or FAIL.
1695 */
Bram Moolenaardc7c3662021-12-20 15:04:29 +00001696 int
Bram Moolenaare42939a2021-04-05 17:11:17 +02001697compile_assign_unlet(
1698 char_u *var_start,
1699 lhs_T *lhs,
1700 int is_assign,
1701 type_T *rhs_type,
1702 cctx_T *cctx)
1703{
Bram Moolenaare42939a2021-04-05 17:11:17 +02001704 vartype_T dest_type;
Bram Moolenaare42939a2021-04-05 17:11:17 +02001705 garray_T *stack = &cctx->ctx_type_stack;
1706 int range = FALSE;
1707
Bram Moolenaar68452172021-04-12 21:21:02 +02001708 if (compile_assign_index(var_start, lhs, &range, cctx) == FAIL)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001709 return FAIL;
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02001710 if (is_assign && range
1711 && lhs->lhs_type->tt_type != VAR_LIST
1712 && lhs->lhs_type != &t_blob
1713 && lhs->lhs_type != &t_any)
Bram Moolenaar68452172021-04-12 21:21:02 +02001714 {
1715 semsg(_(e_cannot_use_range_with_assignment_str), var_start);
1716 return FAIL;
1717 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001718
1719 if (lhs->lhs_type == &t_any)
1720 {
1721 // Index on variable of unknown type: check at runtime.
1722 dest_type = VAR_ANY;
1723 }
1724 else
1725 {
1726 dest_type = lhs->lhs_type->tt_type;
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001727 if (dest_type == VAR_DICT && range)
1728 {
1729 emsg(e_cannot_use_range_with_dictionary);
1730 return FAIL;
1731 }
Bram Moolenaar5fa9b242021-06-04 21:00:32 +02001732 if (dest_type == VAR_DICT
1733 && may_generate_2STRING(-1, FALSE, cctx) == FAIL)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001734 return FAIL;
Bram Moolenaar51e93322021-04-17 20:44:56 +02001735 if (dest_type == VAR_LIST || dest_type == VAR_BLOB)
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001736 {
Bram Moolenaar51e93322021-04-17 20:44:56 +02001737 type_T *type;
1738
1739 if (range)
1740 {
1741 type = ((type_T **)stack->ga_data)[stack->ga_len - 2];
1742 if (need_type(type, &t_number,
1743 -1, 0, cctx, FALSE, FALSE) == FAIL)
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001744 return FAIL;
Bram Moolenaar51e93322021-04-17 20:44:56 +02001745 }
1746 type = ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaar63cb6562021-07-20 22:21:59 +02001747 if ((dest_type != VAR_BLOB && type != &t_special)
Bram Moolenaar51e93322021-04-17 20:44:56 +02001748 && need_type(type, &t_number,
1749 -1, 0, cctx, FALSE, FALSE) == FAIL)
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001750 return FAIL;
1751 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001752 }
1753
1754 // Load the dict or list. On the stack we then have:
1755 // - value (for assignment, not for :unlet)
1756 // - index
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001757 // - for [a : b] second index
Bram Moolenaar752fc692021-01-04 21:57:11 +01001758 // - variable
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001759 if (compile_load_lhs(lhs, var_start, rhs_type, cctx) == FAIL)
1760 return FAIL;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001761
Bram Moolenaar68452172021-04-12 21:21:02 +02001762 if (dest_type == VAR_LIST || dest_type == VAR_DICT
1763 || dest_type == VAR_BLOB || dest_type == VAR_ANY)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001764 {
1765 if (is_assign)
1766 {
Bram Moolenaar68452172021-04-12 21:21:02 +02001767 if (range)
1768 {
1769 if (generate_instr_drop(cctx, ISN_STORERANGE, 4) == NULL)
1770 return FAIL;
1771 }
1772 else
1773 {
1774 isn_T *isn = generate_instr_drop(cctx, ISN_STOREINDEX, 3);
Bram Moolenaar752fc692021-01-04 21:57:11 +01001775
Bram Moolenaar68452172021-04-12 21:21:02 +02001776 if (isn == NULL)
1777 return FAIL;
1778 isn->isn_arg.vartype = dest_type;
1779 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001780 }
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001781 else if (range)
1782 {
1783 if (generate_instr_drop(cctx, ISN_UNLETRANGE, 3) == NULL)
1784 return FAIL;
1785 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001786 else
1787 {
1788 if (generate_instr_drop(cctx, ISN_UNLETINDEX, 2) == NULL)
1789 return FAIL;
1790 }
1791 }
1792 else
1793 {
1794 emsg(_(e_indexable_type_required));
1795 return FAIL;
1796 }
1797
1798 return OK;
1799}
1800
Bram Moolenaar4b8a0652020-12-01 16:30:44 +01001801/*
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001802 * Compile declaration and assignment:
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001803 * "let name"
1804 * "var name = expr"
1805 * "final name = expr"
1806 * "const name = expr"
1807 * "name = expr"
1808 * "arg" points to "name".
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001809 * "++arg" and "--arg"
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001810 * Return NULL for an error.
1811 * Return "arg" if it does not look like a variable list.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001812 */
1813 static char_u *
1814compile_assignment(char_u *arg, exarg_T *eap, cmdidx_T cmdidx, cctx_T *cctx)
1815{
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001816 char_u *var_start;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001817 char_u *p;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001818 char_u *end = arg;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001819 char_u *ret = NULL;
1820 int var_count = 0;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001821 int var_idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001822 int semicolon = 0;
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +02001823 int did_generate_slice = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001824 garray_T *instr = &cctx->ctx_instr;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001825 garray_T *stack = &cctx->ctx_type_stack;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001826 char_u *op;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001827 int oplen = 0;
1828 int heredoc = FALSE;
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001829 int incdec = FALSE;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001830 type_T *rhs_type = &t_any;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001831 char_u *sp;
Bram Moolenaar752fc692021-01-04 21:57:11 +01001832 int is_decl = is_decl_command(cmdidx);
1833 lhs_T lhs;
Bram Moolenaar77709b12021-04-03 21:01:01 +02001834 long start_lnum = SOURCING_LNUM;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001835
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001836 // Skip over the "var" or "[var, var]" to get to any "=".
1837 p = skip_var_list(arg, TRUE, &var_count, &semicolon, TRUE);
1838 if (p == NULL)
1839 return *arg == '[' ? arg : NULL;
1840
Bram Moolenaar752fc692021-01-04 21:57:11 +01001841 lhs.lhs_name = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001842
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001843 sp = p;
1844 p = skipwhite(p);
1845 op = p;
1846 oplen = assignment_len(p, &heredoc);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001847
1848 if (var_count > 0 && oplen == 0)
1849 // can be something like "[1, 2]->func()"
1850 return arg;
1851
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001852 if (oplen > 0 && (!VIM_ISWHITE(*sp) || !IS_WHITE_OR_NUL(op[oplen])))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001853 {
Bram Moolenaarbb1b5e22020-08-05 10:53:21 +02001854 error_white_both(op, oplen);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001855 return NULL;
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001856 }
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001857 if (eap->cmdidx == CMD_increment || eap->cmdidx == CMD_decrement)
1858 {
Bram Moolenaar22480d12021-06-25 21:31:09 +02001859 if (VIM_ISWHITE(eap->cmd[2]))
1860 {
1861 semsg(_(e_no_white_space_allowed_after_str_str),
1862 eap->cmdidx == CMD_increment ? "++" : "--", eap->cmd);
1863 return NULL;
1864 }
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001865 op = (char_u *)(eap->cmdidx == CMD_increment ? "+=" : "-=");
1866 oplen = 2;
1867 incdec = TRUE;
1868 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001869
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001870 if (heredoc)
1871 {
1872 list_T *l;
1873 listitem_T *li;
1874
1875 // [let] varname =<< [trim] {end}
Bram Moolenaar04b12692020-05-04 23:24:44 +02001876 eap->getline = exarg_getline;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001877 eap->cookie = cctx;
Bram Moolenaar6c2b7b82020-04-14 20:15:49 +02001878 l = heredoc_get(eap, op + 3, FALSE);
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001879 if (l == NULL)
1880 return NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001881
Bram Moolenaar078269b2020-09-21 20:35:55 +02001882 if (cctx->ctx_skip != SKIP_YES)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001883 {
Bram Moolenaar078269b2020-09-21 20:35:55 +02001884 // Push each line and the create the list.
1885 FOR_ALL_LIST_ITEMS(l, li)
1886 {
Zdenek Dohnal9fe17d42021-08-04 22:30:52 +02001887 generate_PUSHS(cctx, &li->li_tv.vval.v_string);
Bram Moolenaar078269b2020-09-21 20:35:55 +02001888 li->li_tv.vval.v_string = NULL;
1889 }
1890 generate_NEWLIST(cctx, l->lv_len);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001891 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001892 list_free(l);
1893 p += STRLEN(p);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001894 end = p;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001895 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001896 else if (var_count > 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001897 {
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001898 char_u *wp;
1899
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001900 // for "[var, var] = expr" evaluate the expression here, loop over the
1901 // list of variables below.
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001902 // A line break may follow the "=".
Bram Moolenaard25ec2c2020-03-30 21:05:45 +02001903
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001904 wp = op + oplen;
Bram Moolenaar8ff16e02020-12-07 21:49:52 +01001905 if (may_get_next_line_error(wp, &p, cctx) == FAIL)
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001906 return FAIL;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001907 if (compile_expr0(&p, cctx) == FAIL)
1908 return NULL;
1909 end = p;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001910
Bram Moolenaar9b68c822020-06-18 19:31:08 +02001911 if (cctx->ctx_skip != SKIP_YES)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001912 {
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001913 type_T *stacktype;
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +00001914 int needed_list_len;
1915 int did_check = FALSE;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02001916
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001917 stacktype = stack->ga_len == 0 ? &t_void
1918 : ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001919 if (stacktype->tt_type == VAR_VOID)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001920 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02001921 emsg(_(e_cannot_use_void_value));
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001922 goto theend;
1923 }
Bram Moolenaar351ead02021-01-16 16:07:01 +01001924 if (need_type(stacktype, &t_list_any, -1, 0, cctx,
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001925 FALSE, FALSE) == FAIL)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001926 goto theend;
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +00001927 // If a constant list was used we can check the length right here.
1928 needed_list_len = semicolon ? var_count - 1 : var_count;
1929 if (instr->ga_len > 0)
1930 {
1931 isn_T *isn = ((isn_T *)instr->ga_data) + instr->ga_len - 1;
1932
1933 if (isn->isn_type == ISN_NEWLIST)
1934 {
1935 did_check = TRUE;
1936 if (semicolon ? isn->isn_arg.number < needed_list_len
1937 : isn->isn_arg.number != needed_list_len)
1938 {
1939 semsg(_(e_expected_nr_items_but_got_nr),
1940 needed_list_len, isn->isn_arg.number);
1941 goto theend;
1942 }
1943 }
1944 }
1945 if (!did_check)
1946 generate_CHECKLEN(cctx, needed_list_len, semicolon);
Bram Moolenaardc234ca2020-11-28 18:52:33 +01001947 if (stacktype->tt_member != NULL)
1948 rhs_type = stacktype->tt_member;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001949 }
1950 }
1951
1952 /*
1953 * Loop over variables in "[var, var] = expr".
1954 * For "var = expr" and "let var: type" this is done only once.
1955 */
1956 if (var_count > 0)
1957 var_start = skipwhite(arg + 1); // skip over the "["
1958 else
1959 var_start = arg;
1960 for (var_idx = 0; var_idx == 0 || var_idx < var_count; var_idx++)
1961 {
Bram Moolenaar6977dba2021-07-04 22:48:12 +02001962 int instr_count = -1;
1963 int save_lnum;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001964
Bram Moolenaarf93bbd02021-04-10 22:35:43 +02001965 if (var_start[0] == '_' && !eval_isnamec(var_start[1]))
1966 {
1967 // Ignore underscore in "[a, _, b] = list".
1968 if (var_count > 0)
1969 {
1970 var_start = skipwhite(var_start + 2);
1971 continue;
1972 }
1973 emsg(_(e_cannot_use_underscore_here));
1974 goto theend;
1975 }
Bram Moolenaar752fc692021-01-04 21:57:11 +01001976 vim_free(lhs.lhs_name);
1977
1978 /*
1979 * Figure out the LHS type and other properties.
1980 */
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02001981 if (compile_assign_lhs(var_start, &lhs, cmdidx,
1982 is_decl, heredoc, oplen, cctx) == FAIL)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001983 goto theend;
Bram Moolenaar81530e32021-07-28 21:25:49 +02001984 if (heredoc)
1985 {
1986 SOURCING_LNUM = start_lnum;
1987 if (lhs.lhs_has_type
1988 && need_type(&t_list_string, lhs.lhs_type,
1989 -1, 0, cctx, FALSE, FALSE) == FAIL)
1990 goto theend;
1991 }
1992 else
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001993 {
Bram Moolenaar9b68c822020-06-18 19:31:08 +02001994 if (cctx->ctx_skip == SKIP_YES)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02001995 {
Bram Moolenaar72abcf42020-06-18 18:26:24 +02001996 if (oplen > 0 && var_count == 0)
1997 {
1998 // skip over the "=" and the expression
1999 p = skipwhite(op + oplen);
Bram Moolenaar169502f2021-04-21 12:19:35 +02002000 (void)compile_expr0(&p, cctx);
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002001 }
2002 }
2003 else if (oplen > 0)
2004 {
Bram Moolenaar334a8b42020-10-19 16:07:42 +02002005 int is_const = FALSE;
Bram Moolenaar7f764942020-12-02 15:11:18 +01002006 char_u *wp;
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002007
Bram Moolenaar035bd1c2021-06-21 19:44:11 +02002008 // for "+=", "*=", "..=" etc. first load the current value
2009 if (*op != '='
2010 && compile_load_lhs_with_index(&lhs, var_start,
2011 cctx) == FAIL)
2012 goto theend;
2013
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002014 // For "var = expr" evaluate the expression.
2015 if (var_count == 0)
2016 {
2017 int r;
2018
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002019 // Compile the expression.
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002020 instr_count = instr->ga_len;
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002021 if (incdec)
Bram Moolenaar21e51222020-12-04 12:43:29 +01002022 {
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002023 r = generate_PUSHNR(cctx, 1);
2024 }
2025 else
2026 {
2027 // Temporarily hide the new local variable here, it is
2028 // not available to this expression.
2029 if (lhs.lhs_new_local)
2030 --cctx->ctx_locals.ga_len;
2031 wp = op + oplen;
2032 if (may_get_next_line_error(wp, &p, cctx) == FAIL)
2033 {
2034 if (lhs.lhs_new_local)
2035 ++cctx->ctx_locals.ga_len;
2036 goto theend;
2037 }
2038 r = compile_expr0_ext(&p, cctx, &is_const);
Bram Moolenaar752fc692021-01-04 21:57:11 +01002039 if (lhs.lhs_new_local)
Bram Moolenaar21e51222020-12-04 12:43:29 +01002040 ++cctx->ctx_locals.ga_len;
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002041 if (r == FAIL)
2042 goto theend;
Bram Moolenaar21e51222020-12-04 12:43:29 +01002043 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002044 }
Bram Moolenaar9af78762020-06-16 11:34:42 +02002045 else if (semicolon && var_idx == var_count - 1)
2046 {
2047 // For "[var; var] = expr" get the rest of the list
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +02002048 did_generate_slice = TRUE;
Bram Moolenaar9af78762020-06-16 11:34:42 +02002049 if (generate_SLICE(cctx, var_count - 1) == FAIL)
2050 goto theend;
2051 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002052 else
2053 {
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002054 // For "[var, var] = expr" get the "var_idx" item from the
2055 // list.
Bram Moolenaar035bd1c2021-06-21 19:44:11 +02002056 if (generate_GETITEM(cctx, var_idx, *op != '=') == FAIL)
Bram Moolenaar7f764942020-12-02 15:11:18 +01002057 goto theend;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002058 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002059
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002060 rhs_type = stack->ga_len == 0 ? &t_void
Bram Moolenaar6802cce2020-07-19 15:49:49 +02002061 : ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaar752fc692021-01-04 21:57:11 +01002062 if (lhs.lhs_lvar != NULL && (is_decl || !lhs.lhs_has_type))
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002063 {
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002064 if ((rhs_type->tt_type == VAR_FUNC
2065 || rhs_type->tt_type == VAR_PARTIAL)
Bram Moolenaar3f327882021-03-17 20:56:38 +01002066 && !lhs.lhs_has_index
Bram Moolenaar752fc692021-01-04 21:57:11 +01002067 && var_wrong_func_name(lhs.lhs_name, TRUE))
Bram Moolenaar0f769812020-09-12 18:32:34 +02002068 goto theend;
2069
Bram Moolenaar752fc692021-01-04 21:57:11 +01002070 if (lhs.lhs_new_local && !lhs.lhs_has_type)
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002071 {
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002072 if (rhs_type->tt_type == VAR_VOID)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002073 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02002074 emsg(_(e_cannot_use_void_value));
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002075 goto theend;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002076 }
2077 else
2078 {
Bram Moolenaar04bdd572020-09-23 13:25:32 +02002079 // An empty list or dict has a &t_unknown member,
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002080 // for a variable that implies &t_any.
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002081 if (rhs_type == &t_list_empty)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002082 lhs.lhs_lvar->lv_type = &t_list_any;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002083 else if (rhs_type == &t_dict_empty)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002084 lhs.lhs_lvar->lv_type = &t_dict_any;
Bram Moolenaardc234ca2020-11-28 18:52:33 +01002085 else if (rhs_type == &t_unknown)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002086 lhs.lhs_lvar->lv_type = &t_any;
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002087 else
Bram Moolenaar752fc692021-01-04 21:57:11 +01002088 lhs.lhs_lvar->lv_type = rhs_type;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002089 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002090 }
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002091 else if (*op == '=')
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002092 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002093 type_T *use_type = lhs.lhs_lvar->lv_type;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +02002094 where_T where = WHERE_INIT;
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002095
Bram Moolenaar77709b12021-04-03 21:01:01 +02002096 // Without operator check type here, otherwise below.
2097 // Use the line number of the assignment.
2098 SOURCING_LNUM = start_lnum;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +02002099 where.wt_index = var_count > 0 ? var_idx + 1 : 0;
2100 where.wt_variable = var_count > 0;
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02002101 // If assigning to a list or dict member, use the
2102 // member type. Not for "list[:] =".
2103 if (lhs.lhs_has_index
2104 && !has_list_index(var_start + lhs.lhs_varlen,
2105 cctx))
Bram Moolenaar752fc692021-01-04 21:57:11 +01002106 use_type = lhs.lhs_member_type;
Bram Moolenaar4270d8b2021-08-07 16:30:42 +02002107 if (need_type_where(rhs_type, use_type, -1, where,
2108 cctx, FALSE, is_const) == FAIL)
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002109 goto theend;
2110 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002111 }
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002112 else
2113 {
2114 type_T *lhs_type = lhs.lhs_member_type;
2115
2116 // Special case: assigning to @# can use a number or a
2117 // string.
Bram Moolenaaraf647e72021-08-05 19:01:17 +02002118 // Also: can assign a number to a float.
2119 if ((lhs_type == &t_number_or_string
2120 || lhs_type == &t_float)
2121 && rhs_type->tt_type == VAR_NUMBER)
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002122 lhs_type = &t_number;
2123 if (*p != '=' && need_type(rhs_type, lhs_type,
Bram Moolenaar351ead02021-01-16 16:07:01 +01002124 -1, 0, cctx, FALSE, FALSE) == FAIL)
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002125 goto theend;
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002126 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002127 }
Bram Moolenaar30fd8202020-09-26 15:09:30 +02002128 else if (cmdidx == CMD_final)
2129 {
2130 emsg(_(e_final_requires_a_value));
2131 goto theend;
2132 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002133 else if (cmdidx == CMD_const)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002134 {
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02002135 emsg(_(e_const_requires_a_value));
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002136 goto theend;
2137 }
Bram Moolenaardcb53be2021-12-09 14:23:43 +00002138 else if (!lhs.lhs_has_type || lhs.lhs_dest == dest_option
2139 || lhs.lhs_dest == dest_func_option)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002140 {
Bram Moolenaarbc4c5052020-08-13 22:47:35 +02002141 emsg(_(e_type_or_initialization_required));
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002142 goto theend;
2143 }
2144 else
2145 {
2146 // variables are always initialized
Bram Moolenaar35578162021-08-02 19:10:38 +02002147 if (GA_GROW_FAILS(instr, 1))
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002148 goto theend;
Bram Moolenaar752fc692021-01-04 21:57:11 +01002149 switch (lhs.lhs_member_type->tt_type)
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002150 {
2151 case VAR_BOOL:
2152 generate_PUSHBOOL(cctx, VVAL_FALSE);
2153 break;
2154 case VAR_FLOAT:
2155#ifdef FEAT_FLOAT
2156 generate_PUSHF(cctx, 0.0);
2157#endif
2158 break;
2159 case VAR_STRING:
2160 generate_PUSHS(cctx, NULL);
2161 break;
2162 case VAR_BLOB:
Bram Moolenaar80b0e5e2020-10-19 20:45:36 +02002163 generate_PUSHBLOB(cctx, blob_alloc());
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002164 break;
2165 case VAR_FUNC:
2166 generate_PUSHFUNC(cctx, NULL, &t_func_void);
2167 break;
2168 case VAR_LIST:
2169 generate_NEWLIST(cctx, 0);
2170 break;
2171 case VAR_DICT:
2172 generate_NEWDICT(cctx, 0);
2173 break;
2174 case VAR_JOB:
2175 generate_PUSHJOB(cctx, NULL);
2176 break;
2177 case VAR_CHANNEL:
2178 generate_PUSHCHANNEL(cctx, NULL);
2179 break;
2180 case VAR_NUMBER:
2181 case VAR_UNKNOWN:
2182 case VAR_ANY:
2183 case VAR_PARTIAL:
2184 case VAR_VOID:
Bram Moolenaarf18332f2021-05-07 17:55:55 +02002185 case VAR_INSTR:
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002186 case VAR_SPECIAL: // cannot happen
2187 generate_PUSHNR(cctx, 0);
2188 break;
2189 }
2190 }
2191 if (var_count == 0)
2192 end = p;
2193 }
2194
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002195 // no need to parse more when skipping
Bram Moolenaar9b68c822020-06-18 19:31:08 +02002196 if (cctx->ctx_skip == SKIP_YES)
Bram Moolenaar72abcf42020-06-18 18:26:24 +02002197 break;
2198
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002199 if (oplen > 0 && *op != '=')
2200 {
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002201 type_T *expected;
Bram Moolenaarc3160722021-08-02 21:12:05 +02002202 type_T *stacktype = NULL;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002203
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002204 if (*op == '.')
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002205 {
2206 if (may_generate_2STRING(-1, FALSE, cctx) == FAIL)
2207 goto theend;
2208 }
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002209 else
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002210 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002211 expected = lhs.lhs_member_type;
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002212 stacktype = ((type_T **)stack->ga_data)[stack->ga_len - 1];
2213 if (
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002214#ifdef FEAT_FLOAT
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002215 // If variable is float operation with number is OK.
Bram Moolenaar7bf9a072021-08-02 21:55:15 +02002216 !(expected == &t_float && (stacktype == &t_number
2217 || stacktype == &t_number_bool)) &&
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002218#endif
Bram Moolenaar351ead02021-01-16 16:07:01 +01002219 need_type(stacktype, expected, -1, 0, cctx,
Bram Moolenaar334a8b42020-10-19 16:07:42 +02002220 FALSE, FALSE) == FAIL)
Bram Moolenaarf5d52c92021-07-31 22:51:10 +02002221 goto theend;
2222 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002223
2224 if (*op == '.')
Bram Moolenaardd29f1b2020-08-07 20:46:20 +02002225 {
2226 if (generate_instr_drop(cctx, ISN_CONCAT, 1) == NULL)
2227 goto theend;
2228 }
2229 else if (*op == '+')
2230 {
2231 if (generate_add_instr(cctx,
Bram Moolenaar752fc692021-01-04 21:57:11 +01002232 operator_type(lhs.lhs_member_type, stacktype),
Bram Moolenaar07802042021-09-09 23:01:14 +02002233 lhs.lhs_member_type, stacktype,
2234 EXPR_APPEND) == FAIL)
Bram Moolenaardd29f1b2020-08-07 20:46:20 +02002235 goto theend;
2236 }
Bram Moolenaar93ad1472020-08-19 22:02:41 +02002237 else if (generate_two_op(cctx, op) == FAIL)
2238 goto theend;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002239 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002240
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002241 // Use the line number of the assignment for store instruction.
2242 save_lnum = cctx->ctx_lnum;
2243 cctx->ctx_lnum = start_lnum - 1;
2244
Bram Moolenaar752fc692021-01-04 21:57:11 +01002245 if (lhs.lhs_has_index)
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002246 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002247 // Use the info in "lhs" to store the value at the index in the
2248 // list or dict.
2249 if (compile_assign_unlet(var_start, &lhs, TRUE, rhs_type, cctx)
2250 == FAIL)
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002251 {
2252 cctx->ctx_lnum = save_lnum;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002253 goto theend;
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002254 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002255 }
2256 else
2257 {
Bram Moolenaar752fc692021-01-04 21:57:11 +01002258 if (is_decl && cmdidx == CMD_const && (lhs.lhs_dest == dest_script
Bram Moolenaard877a572021-04-01 19:42:48 +02002259 || lhs.lhs_dest == dest_global
Bram Moolenaar752fc692021-01-04 21:57:11 +01002260 || lhs.lhs_dest == dest_local))
Bram Moolenaar30fd8202020-09-26 15:09:30 +02002261 // ":const var": lock the value, but not referenced variables
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +02002262 generate_LOCKCONST(cctx);
2263
Bram Moolenaaraa210a32021-01-02 15:41:03 +01002264 if (is_decl
Bram Moolenaar752fc692021-01-04 21:57:11 +01002265 && (lhs.lhs_type->tt_type == VAR_DICT
2266 || lhs.lhs_type->tt_type == VAR_LIST)
2267 && lhs.lhs_type->tt_member != NULL
Bram Moolenaar6e48b842021-08-10 22:52:02 +02002268 && !(lhs.lhs_type->tt_member == &t_any
2269 && oplen > 0
2270 && rhs_type != NULL
2271 && rhs_type->tt_type == lhs.lhs_type->tt_type
2272 && rhs_type->tt_member != &t_unknown)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002273 && lhs.lhs_type->tt_member != &t_unknown)
Bram Moolenaaraa210a32021-01-02 15:41:03 +01002274 // Set the type in the list or dict, so that it can be checked,
Bram Moolenaar6e48b842021-08-10 22:52:02 +02002275 // also in legacy script. Not for "list<any> = val", then the
2276 // type of "val" is used.
Bram Moolenaar752fc692021-01-04 21:57:11 +01002277 generate_SETTYPE(cctx, lhs.lhs_type);
Bram Moolenaaraa210a32021-01-02 15:41:03 +01002278
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02002279 if (generate_store_lhs(cctx, &lhs, instr_count) == FAIL)
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002280 {
2281 cctx->ctx_lnum = save_lnum;
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02002282 goto theend;
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002283 }
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002284 }
Bram Moolenaar6977dba2021-07-04 22:48:12 +02002285 cctx->ctx_lnum = save_lnum;
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002286
2287 if (var_idx + 1 < var_count)
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00002288 var_start = skipwhite(lhs.lhs_end + 1);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002289 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002290
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +02002291 // For "[var, var] = expr" drop the "expr" value.
2292 // Also for "[var, var; _] = expr".
2293 if (var_count > 0 && (!semicolon || !did_generate_slice))
Bram Moolenaar9af78762020-06-16 11:34:42 +02002294 {
Bram Moolenaarec792292020-12-13 21:26:56 +01002295 if (generate_instr_drop(cctx, ISN_DROP, 1) == NULL)
Bram Moolenaar9af78762020-06-16 11:34:42 +02002296 goto theend;
2297 }
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002298
Bram Moolenaarb2097502020-07-19 17:17:02 +02002299 ret = skipwhite(end);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002300
2301theend:
Bram Moolenaar752fc692021-01-04 21:57:11 +01002302 vim_free(lhs.lhs_name);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002303 return ret;
2304}
2305
2306/*
Bram Moolenaar17126b12021-01-07 22:03:02 +01002307 * Check for an assignment at "eap->cmd", compile it if found.
2308 * Return NOTDONE if there is none, FAIL for failure, OK if done.
2309 */
2310 static int
2311may_compile_assignment(exarg_T *eap, char_u **line, cctx_T *cctx)
2312{
2313 char_u *pskip;
2314 char_u *p;
2315
2316 // Assuming the command starts with a variable or function name,
2317 // find what follows.
2318 // Skip over "var.member", "var[idx]" and the like.
2319 // Also "&opt = val", "$ENV = val" and "@r = val".
2320 pskip = (*eap->cmd == '&' || *eap->cmd == '$' || *eap->cmd == '@')
2321 ? eap->cmd + 1 : eap->cmd;
2322 p = to_name_end(pskip, TRUE);
2323 if (p > eap->cmd && *p != NUL)
2324 {
2325 char_u *var_end;
2326 int oplen;
2327 int heredoc;
2328
2329 if (eap->cmd[0] == '@')
2330 var_end = eap->cmd + 2;
2331 else
2332 var_end = find_name_end(pskip, NULL, NULL,
2333 FNE_CHECK_START | FNE_INCL_BR);
2334 oplen = assignment_len(skipwhite(var_end), &heredoc);
2335 if (oplen > 0)
2336 {
2337 size_t len = p - eap->cmd;
2338
2339 // Recognize an assignment if we recognize the variable
2340 // name:
2341 // "g:var = expr"
2342 // "local = expr" where "local" is a local var.
2343 // "script = expr" where "script" is a script-local var.
2344 // "import = expr" where "import" is an imported var
2345 // "&opt = expr"
2346 // "$ENV = expr"
2347 // "@r = expr"
2348 if (*eap->cmd == '&'
2349 || *eap->cmd == '$'
2350 || *eap->cmd == '@'
2351 || ((len) > 2 && eap->cmd[1] == ':')
Bram Moolenaare0890d62021-02-17 14:52:14 +01002352 || variable_exists(eap->cmd, len, cctx))
Bram Moolenaar17126b12021-01-07 22:03:02 +01002353 {
2354 *line = compile_assignment(eap->cmd, eap, CMD_SIZE, cctx);
2355 if (*line == NULL || *line == eap->cmd)
2356 return FAIL;
2357 return OK;
2358 }
2359 }
2360 }
2361
2362 if (*eap->cmd == '[')
2363 {
2364 // [var, var] = expr
2365 *line = compile_assignment(eap->cmd, eap, CMD_SIZE, cctx);
2366 if (*line == NULL)
2367 return FAIL;
2368 if (*line != eap->cmd)
2369 return OK;
2370 }
2371 return NOTDONE;
2372}
2373
Bram Moolenaar7b829262021-10-13 15:04:34 +01002374
2375/*
Bram Moolenaar09689a02020-05-09 22:50:08 +02002376 * Add a function to the list of :def functions.
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002377 * This sets "ufunc->uf_dfunc_idx" but the function isn't compiled yet.
Bram Moolenaar09689a02020-05-09 22:50:08 +02002378 */
Bram Moolenaar822ba242020-05-24 23:00:18 +02002379 static int
Bram Moolenaar09689a02020-05-09 22:50:08 +02002380add_def_function(ufunc_T *ufunc)
2381{
2382 dfunc_T *dfunc;
2383
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002384 if (def_functions.ga_len == 0)
2385 {
2386 // The first position is not used, so that a zero uf_dfunc_idx means it
2387 // wasn't set.
Bram Moolenaar35578162021-08-02 19:10:38 +02002388 if (GA_GROW_FAILS(&def_functions, 1))
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002389 return FAIL;
2390 ++def_functions.ga_len;
2391 }
2392
Bram Moolenaar09689a02020-05-09 22:50:08 +02002393 // Add the function to "def_functions".
Bram Moolenaar35578162021-08-02 19:10:38 +02002394 if (GA_GROW_FAILS(&def_functions, 1))
Bram Moolenaar09689a02020-05-09 22:50:08 +02002395 return FAIL;
2396 dfunc = ((dfunc_T *)def_functions.ga_data) + def_functions.ga_len;
2397 CLEAR_POINTER(dfunc);
2398 dfunc->df_idx = def_functions.ga_len;
2399 ufunc->uf_dfunc_idx = dfunc->df_idx;
2400 dfunc->df_ufunc = ufunc;
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002401 dfunc->df_name = vim_strsave(ufunc->uf_name);
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02002402 ga_init2(&dfunc->df_var_names, sizeof(char_u *), 10);
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002403 ++dfunc->df_refcount;
Bram Moolenaar09689a02020-05-09 22:50:08 +02002404 ++def_functions.ga_len;
2405 return OK;
2406}
2407
2408/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002409 * After ex_function() has collected all the function lines: parse and compile
2410 * the lines into instructions.
2411 * Adds the function to "def_functions".
Bram Moolenaar9e68c322020-12-25 12:38:04 +01002412 * When "check_return_type" is set then set ufunc->uf_ret_type to the type of
2413 * the return statement (used for lambda). When uf_ret_type is already set
2414 * then check that it matches.
Bram Moolenaarb2049902021-01-24 12:53:53 +01002415 * When "profiling" is true add ISN_PROF_START instructions.
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002416 * "outer_cctx" is set for a nested function.
Bram Moolenaar05afcee2020-03-31 23:32:31 +02002417 * This can be used recursively through compile_lambda(), which may reallocate
2418 * "def_functions".
Bram Moolenaar822ba242020-05-24 23:00:18 +02002419 * Returns OK or FAIL.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002420 */
Bram Moolenaar822ba242020-05-24 23:00:18 +02002421 int
Bram Moolenaarb2049902021-01-24 12:53:53 +01002422compile_def_function(
Bram Moolenaare99d4222021-06-13 14:01:26 +02002423 ufunc_T *ufunc,
2424 int check_return_type,
2425 compiletype_T compile_type,
2426 cctx_T *outer_cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002427{
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002428 char_u *line = NULL;
Bram Moolenaarf62d7392021-04-14 12:40:00 +02002429 char_u *line_to_free = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002430 char_u *p;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002431 char *errormsg = NULL; // error message
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002432 cctx_T cctx;
2433 garray_T *instr;
Bram Moolenaard66960b2020-10-30 20:46:26 +01002434 int did_emsg_before = did_emsg;
Bram Moolenaar599410c2021-04-10 14:03:43 +02002435 int did_emsg_silent_before = did_emsg_silent;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002436 int ret = FAIL;
2437 sctx_T save_current_sctx = current_sctx;
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +02002438 int save_estack_compiling = estack_compiling;
Bram Moolenaardc4c2302021-04-25 13:54:42 +02002439 int save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002440 int do_estack_push;
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02002441 int new_def_function = FALSE;
Bram Moolenaarf002a412021-01-24 13:34:18 +01002442#ifdef FEAT_PROFILE
Bram Moolenaarb2049902021-01-24 12:53:53 +01002443 int prof_lnum = -1;
Bram Moolenaarf002a412021-01-24 13:34:18 +01002444#endif
Bram Moolenaare99d4222021-06-13 14:01:26 +02002445 int debug_lnum = -1;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002446
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002447 // When using a function that was compiled before: Free old instructions.
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01002448 // The index is reused. Otherwise add a new entry in "def_functions".
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02002449 if (ufunc->uf_dfunc_idx > 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002450 {
Bram Moolenaar09689a02020-05-09 22:50:08 +02002451 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
2452 + ufunc->uf_dfunc_idx;
Bram Moolenaar3b814af2021-06-15 10:22:17 +02002453 isn_T *instr_dest = NULL;
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02002454
2455 switch (compile_type)
2456 {
2457 case CT_PROFILE:
2458#ifdef FEAT_PROFILE
2459 instr_dest = dfunc->df_instr_prof; break;
2460#endif
2461 case CT_NONE: instr_dest = dfunc->df_instr; break;
2462 case CT_DEBUG: instr_dest = dfunc->df_instr_debug; break;
2463 }
2464 if (instr_dest != NULL)
2465 // Was compiled in this mode before: Free old instructions.
2466 delete_def_function_contents(dfunc, FALSE);
2467 ga_clear_strings(&dfunc->df_var_names);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002468 }
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02002469 else
2470 {
2471 if (add_def_function(ufunc) == FAIL)
2472 return FAIL;
2473 new_def_function = TRUE;
2474 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002475
Bram Moolenaar985116a2020-07-12 17:31:09 +02002476 ufunc->uf_def_status = UF_COMPILING;
2477
Bram Moolenaara80faa82020-04-12 19:37:17 +02002478 CLEAR_FIELD(cctx);
Bram Moolenaarb2049902021-01-24 12:53:53 +01002479
Bram Moolenaare99d4222021-06-13 14:01:26 +02002480 cctx.ctx_compile_type = compile_type;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002481 cctx.ctx_ufunc = ufunc;
2482 cctx.ctx_lnum = -1;
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002483 cctx.ctx_outer = outer_cctx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002484 ga_init2(&cctx.ctx_locals, sizeof(lvar_T), 10);
2485 ga_init2(&cctx.ctx_type_stack, sizeof(type_T *), 50);
2486 ga_init2(&cctx.ctx_imports, sizeof(imported_T), 10);
2487 cctx.ctx_type_list = &ufunc->uf_type_list;
2488 ga_init2(&cctx.ctx_instr, sizeof(isn_T), 50);
2489 instr = &cctx.ctx_instr;
2490
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002491 // Set the context to the function, it may be compiled when called from
2492 // another script. Set the script version to the most modern one.
2493 // The line number will be set in next_line_from_context().
2494 current_sctx = ufunc->uf_script_ctx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002495 current_sctx.sc_version = SCRIPT_VERSION_VIM9;
2496
Bram Moolenaardc4c2302021-04-25 13:54:42 +02002497 // Don't use the flag from ":legacy" here.
2498 cmdmod.cmod_flags &= ~CMOD_LEGACY;
2499
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002500 // Make sure error messages are OK.
2501 do_estack_push = !estack_top_is_ufunc(ufunc, 1);
2502 if (do_estack_push)
2503 estack_push_ufunc(ufunc, 1);
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +02002504 estack_compiling = TRUE;
Bram Moolenaar25e0f582020-05-25 22:36:50 +02002505
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002506 if (ufunc->uf_def_args.ga_len > 0)
2507 {
2508 int count = ufunc->uf_def_args.ga_len;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002509 int first_def_arg = ufunc->uf_args.ga_len - count;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002510 int i;
2511 char_u *arg;
2512 int off = STACK_FRAME_SIZE + (ufunc->uf_va_name != NULL ? 1 : 0);
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002513 int did_set_arg_type = FALSE;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002514
2515 // Produce instructions for the default values of optional arguments.
Bram Moolenaar12bce952021-03-11 20:04:04 +01002516 SOURCING_LNUM = 0; // line number unknown
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002517 for (i = 0; i < count; ++i)
2518 {
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002519 garray_T *stack = &cctx.ctx_type_stack;
2520 type_T *val_type;
2521 int arg_idx = first_def_arg + i;
Bram Moolenaar7a3fe3e2021-07-22 14:58:47 +02002522 where_T where = WHERE_INIT;
Bram Moolenaar12bce952021-03-11 20:04:04 +01002523 int r;
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +02002524 int jump_instr_idx = instr->ga_len;
2525 isn_T *isn;
2526
2527 // Use a JUMP_IF_ARG_SET instruction to skip if the value was given.
2528 if (generate_JUMP_IF_ARG_SET(&cctx, i - count - off) == FAIL)
2529 goto erret;
Bram Moolenaar12bce952021-03-11 20:04:04 +01002530
2531 // Make sure later arguments are not found.
Bram Moolenaare28d9b32021-07-03 18:56:53 +02002532 ufunc->uf_args_visible = arg_idx;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002533
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002534 arg = ((char_u **)(ufunc->uf_def_args.ga_data))[i];
Bram Moolenaar12bce952021-03-11 20:04:04 +01002535 r = compile_expr0(&arg, &cctx);
2536
Bram Moolenaar12bce952021-03-11 20:04:04 +01002537 if (r == FAIL)
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002538 goto erret;
2539
2540 // If no type specified use the type of the default value.
2541 // Otherwise check that the default value type matches the
2542 // specified type.
2543 val_type = ((type_T **)stack->ga_data)[stack->ga_len - 1];
Bram Moolenaarf785aa12021-02-11 21:19:34 +01002544 where.wt_index = arg_idx + 1;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002545 if (ufunc->uf_arg_types[arg_idx] == &t_unknown)
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002546 {
2547 did_set_arg_type = TRUE;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002548 ufunc->uf_arg_types[arg_idx] = val_type;
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002549 }
Bram Moolenaar8b565c22020-08-30 23:24:20 +02002550 else if (check_type(ufunc->uf_arg_types[arg_idx], val_type,
Bram Moolenaarf785aa12021-02-11 21:19:34 +01002551 TRUE, where) == FAIL)
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002552 goto erret;
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +02002553
2554 if (generate_STORE(&cctx, ISN_STORE, i - count - off, NULL) == FAIL)
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002555 goto erret;
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +02002556
2557 // set instruction index in JUMP_IF_ARG_SET to here
2558 isn = ((isn_T *)instr->ga_data) + jump_instr_idx;
2559 isn->isn_arg.jumparg.jump_where = instr->ga_len;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002560 }
Bram Moolenaarb8070e32020-07-23 20:56:04 +02002561
2562 if (did_set_arg_type)
2563 set_function_type(ufunc);
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002564 }
Bram Moolenaare28d9b32021-07-03 18:56:53 +02002565 ufunc->uf_args_visible = ufunc->uf_args.ga_len;
Bram Moolenaar170fcfc2020-02-06 17:51:35 +01002566
2567 /*
2568 * Loop over all the lines of the function and generate instructions.
2569 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002570 for (;;)
2571 {
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002572 exarg_T ea;
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002573 int starts_with_colon = FALSE;
2574 char_u *cmd;
Bram Moolenaar02194d22020-10-24 23:08:38 +02002575 cmdmod_T local_cmdmod;
Bram Moolenaar5b1c8fe2020-02-21 18:42:43 +01002576
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002577 // Bail out on the first error to avoid a flood of errors and report
2578 // the right line number when inside try/catch.
Bram Moolenaard66960b2020-10-30 20:46:26 +01002579 if (did_emsg_before != did_emsg)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002580 goto erret;
2581
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002582 if (line != NULL && *line == '|')
2583 // the line continues after a '|'
2584 ++line;
Bram Moolenaara3b7fdc2020-06-21 14:12:17 +02002585 else if (line != NULL && *skipwhite(line) != NUL
Bram Moolenaar7a092242020-04-16 22:10:49 +02002586 && !(*line == '#' && (line == cctx.ctx_line_start
2587 || VIM_ISWHITE(line[-1]))))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002588 {
Bram Moolenaardd1a9af2020-07-23 15:38:03 +02002589 semsg(_(e_trailing_arg), line);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002590 goto erret;
2591 }
Bram Moolenaar4b3e1962021-03-18 21:37:55 +01002592 else if (line != NULL && vim9_bad_comment(skipwhite(line)))
2593 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002594 else
2595 {
Bram Moolenaar23c55272020-06-21 16:58:13 +02002596 line = next_line_from_context(&cctx, FALSE);
Bram Moolenaar4fdae992020-04-12 16:38:57 +02002597 if (cctx.ctx_lnum >= ufunc->uf_lines.ga_len)
Bram Moolenaarb2049902021-01-24 12:53:53 +01002598 {
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002599 // beyond the last line
Bram Moolenaarf002a412021-01-24 13:34:18 +01002600#ifdef FEAT_PROFILE
Bram Moolenaarced68a02021-01-24 17:53:47 +01002601 if (cctx.ctx_skip != SKIP_YES)
2602 may_generate_prof_end(&cctx, prof_lnum);
Bram Moolenaarf002a412021-01-24 13:34:18 +01002603#endif
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002604 break;
Bram Moolenaarb2049902021-01-24 12:53:53 +01002605 }
Bram Moolenaarf62d7392021-04-14 12:40:00 +02002606 // Make a copy, splitting off nextcmd and removing trailing spaces
2607 // may change it.
2608 if (line != NULL)
2609 {
2610 line = vim_strsave(line);
2611 vim_free(line_to_free);
2612 line_to_free = line;
2613 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002614 }
2615
Bram Moolenaara80faa82020-04-12 19:37:17 +02002616 CLEAR_FIELD(ea);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002617 ea.cmdlinep = &line;
2618 ea.cmd = skipwhite(line);
2619
Bram Moolenaarb2049902021-01-24 12:53:53 +01002620 if (*ea.cmd == '#')
2621 {
2622 // "#" starts a comment
2623 line = (char_u *)"";
2624 continue;
2625 }
2626
Bram Moolenaarf002a412021-01-24 13:34:18 +01002627#ifdef FEAT_PROFILE
Bram Moolenaare99d4222021-06-13 14:01:26 +02002628 if (cctx.ctx_compile_type == CT_PROFILE && cctx.ctx_lnum != prof_lnum
2629 && cctx.ctx_skip != SKIP_YES)
Bram Moolenaarb2049902021-01-24 12:53:53 +01002630 {
2631 may_generate_prof_end(&cctx, prof_lnum);
2632
2633 prof_lnum = cctx.ctx_lnum;
2634 generate_instr(&cctx, ISN_PROF_START);
2635 }
Bram Moolenaarf002a412021-01-24 13:34:18 +01002636#endif
Bram Moolenaare99d4222021-06-13 14:01:26 +02002637 if (cctx.ctx_compile_type == CT_DEBUG && cctx.ctx_lnum != debug_lnum
2638 && cctx.ctx_skip != SKIP_YES)
2639 {
2640 debug_lnum = cctx.ctx_lnum;
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02002641 generate_instr_debug(&cctx);
Bram Moolenaare99d4222021-06-13 14:01:26 +02002642 }
Bram Moolenaar8cec9272021-06-23 20:20:53 +02002643 cctx.ctx_prev_lnum = cctx.ctx_lnum + 1;
Bram Moolenaarb2049902021-01-24 12:53:53 +01002644
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002645 // Some things can be recognized by the first character.
2646 switch (*ea.cmd)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002647 {
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002648 case '}':
2649 {
2650 // "}" ends a block scope
2651 scopetype_T stype = cctx.ctx_scope == NULL
2652 ? NO_SCOPE : cctx.ctx_scope->se_type;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002653
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002654 if (stype == BLOCK_SCOPE)
2655 {
2656 compile_endblock(&cctx);
2657 line = ea.cmd;
2658 }
2659 else
2660 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02002661 emsg(_(e_using_rcurly_outside_if_block_scope));
Bram Moolenaarcb711ab2020-04-16 13:00:29 +02002662 goto erret;
2663 }
2664 if (line != NULL)
2665 line = skipwhite(ea.cmd + 1);
2666 continue;
2667 }
2668
2669 case '{':
2670 // "{" starts a block scope
2671 // "{'a': 1}->func() is something else
2672 if (ends_excmd(*skipwhite(ea.cmd + 1)))
2673 {
2674 line = compile_block(ea.cmd, &cctx);
2675 continue;
2676 }
2677 break;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002678 }
2679
2680 /*
2681 * COMMAND MODIFIERS
2682 */
Bram Moolenaar02194d22020-10-24 23:08:38 +02002683 cctx.ctx_has_cmdmod = FALSE;
Bram Moolenaare1004402020-10-24 20:49:43 +02002684 if (parse_command_modifiers(&ea, &errormsg, &local_cmdmod, FALSE)
2685 == FAIL)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002686 {
2687 if (errormsg != NULL)
2688 goto erret;
2689 // empty line or comment
2690 line = (char_u *)"";
2691 continue;
2692 }
Bram Moolenaare1004402020-10-24 20:49:43 +02002693 generate_cmdmods(&cctx, &local_cmdmod);
2694 undo_cmdmod(&local_cmdmod);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002695
Bram Moolenaare88c8e82020-11-01 17:03:37 +01002696 // Check if there was a colon after the last command modifier or before
2697 // the current position.
2698 for (p = ea.cmd; p >= line; --p)
2699 {
2700 if (*p == ':')
2701 starts_with_colon = TRUE;
2702 if (p < ea.cmd && !VIM_ISWHITE(*p))
2703 break;
2704 }
2705
Bram Moolenaarce024c32021-06-26 13:00:49 +02002706 // Skip ":call" to get to the function name, unless using :legacy
Bram Moolenaar575f24b2020-08-12 14:21:11 +02002707 p = ea.cmd;
Bram Moolenaarce024c32021-06-26 13:00:49 +02002708 if (!(local_cmdmod.cmod_flags & CMOD_LEGACY))
Bram Moolenaar575f24b2020-08-12 14:21:11 +02002709 {
Bram Moolenaarce024c32021-06-26 13:00:49 +02002710 if (checkforcmd(&ea.cmd, "call", 3))
2711 {
2712 if (*ea.cmd == '(')
2713 // not for "call()"
2714 ea.cmd = p;
2715 else
2716 ea.cmd = skipwhite(ea.cmd);
2717 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002718
Bram Moolenaarce024c32021-06-26 13:00:49 +02002719 if (!starts_with_colon)
2720 {
2721 int assign;
Bram Moolenaar1cc2a942020-05-10 19:10:31 +02002722
Bram Moolenaarce024c32021-06-26 13:00:49 +02002723 // Check for assignment after command modifiers.
2724 assign = may_compile_assignment(&ea, &line, &cctx);
2725 if (assign == OK)
2726 goto nextline;
2727 if (assign == FAIL)
2728 goto erret;
2729 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002730 }
2731
2732 /*
2733 * COMMAND after range
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002734 * 'text'->func() should not be confused with 'a mark
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002735 * "++nr" and "--nr" are eval commands
Bram Moolenaar5ca5cc62021-08-24 21:56:03 +02002736 * in "$ENV->func()" the "$" is not a range
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002737 */
Bram Moolenaardf069ee2020-06-22 23:02:51 +02002738 cmd = ea.cmd;
Bram Moolenaarb579f6e2021-12-04 11:57:00 +00002739 if ((*cmd != '$' || starts_with_colon)
Bram Moolenaarce024c32021-06-26 13:00:49 +02002740 && (starts_with_colon || !(*cmd == '\''
2741 || (cmd[0] == cmd[1] && (*cmd == '+' || *cmd == '-')))))
Bram Moolenaardf069ee2020-06-22 23:02:51 +02002742 {
Bram Moolenaar3bd8de42020-09-14 16:37:34 +02002743 ea.cmd = skip_range(ea.cmd, TRUE, NULL);
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002744 if (ea.cmd > cmd)
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002745 {
Bram Moolenaarb579f6e2021-12-04 11:57:00 +00002746 if (!starts_with_colon
2747 && !(local_cmdmod.cmod_flags & CMOD_LEGACY))
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002748 {
Bram Moolenaar6e2c2c52020-12-25 19:25:45 +01002749 semsg(_(e_colon_required_before_range_str), cmd);
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002750 goto erret;
2751 }
Bram Moolenaarada1d872021-02-20 08:16:51 +01002752 ea.addr_count = 1;
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002753 if (ends_excmd2(line, ea.cmd))
2754 {
2755 // A range without a command: jump to the line.
Bram Moolenaare4eed8c2021-12-01 15:22:56 +00002756 generate_EXEC(&cctx, ISN_EXECRANGE,
Bram Moolenaarb579f6e2021-12-04 11:57:00 +00002757 vim_strnsave(cmd, ea.cmd - cmd));
Bram Moolenaare4eed8c2021-12-01 15:22:56 +00002758 line = ea.cmd;
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02002759 goto nextline;
2760 }
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002761 }
Bram Moolenaardf069ee2020-06-22 23:02:51 +02002762 }
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02002763 p = find_ex_command(&ea, NULL,
2764 starts_with_colon || (local_cmdmod.cmod_flags & CMOD_LEGACY)
Bram Moolenaar77b10ff2021-03-14 13:21:35 +01002765 ? NULL : item_exists, &cctx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002766
Bram Moolenaard1510ee2021-01-04 16:15:58 +01002767 if (p == NULL)
2768 {
2769 if (cctx.ctx_skip != SKIP_YES)
2770 emsg(_(e_ambiguous_use_of_user_defined_command));
2771 goto erret;
2772 }
2773
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002774 // When using ":legacy cmd" always use compile_exec().
2775 if (local_cmdmod.cmod_flags & CMOD_LEGACY)
Bram Moolenaar3b1373b2021-05-17 00:01:42 +02002776 {
2777 char_u *start = ea.cmd;
2778
Bram Moolenaarc3cb1c92021-06-02 16:47:53 +02002779 switch (ea.cmdidx)
2780 {
2781 case CMD_if:
2782 case CMD_elseif:
2783 case CMD_else:
2784 case CMD_endif:
2785 case CMD_for:
2786 case CMD_endfor:
2787 case CMD_continue:
2788 case CMD_break:
2789 case CMD_while:
2790 case CMD_endwhile:
2791 case CMD_try:
2792 case CMD_catch:
2793 case CMD_finally:
2794 case CMD_endtry:
2795 semsg(_(e_cannot_use_legacy_with_command_str), ea.cmd);
2796 goto erret;
2797 default: break;
2798 }
2799
Bram Moolenaar3b1373b2021-05-17 00:01:42 +02002800 // ":legacy return expr" needs to be handled differently.
2801 if (checkforcmd(&start, "return", 4))
2802 ea.cmdidx = CMD_return;
2803 else
2804 ea.cmdidx = CMD_legacy;
2805 }
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002806
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002807 if (p == ea.cmd && ea.cmdidx != CMD_SIZE)
2808 {
Bram Moolenaare525bdd2021-08-07 18:12:40 +02002809 if (cctx.ctx_skip == SKIP_YES && ea.cmdidx != CMD_eval)
Bram Moolenaara259d8d2020-01-31 20:10:50 +01002810 {
2811 line += STRLEN(line);
Bram Moolenaarf665e972020-12-05 19:17:16 +01002812 goto nextline;
Bram Moolenaara259d8d2020-01-31 20:10:50 +01002813 }
Bram Moolenaare525bdd2021-08-07 18:12:40 +02002814 else if (ea.cmdidx != CMD_eval)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002815 {
Bram Moolenaar52c124d2020-12-20 21:43:35 +01002816 // CMD_var cannot happen, compile_assignment() above would be
2817 // used. Most likely an assignment to a non-existing variable.
2818 semsg(_(e_command_not_recognized_str), ea.cmd);
Bram Moolenaar007f9d62020-07-06 23:04:49 +02002819 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002820 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002821 }
2822
Bram Moolenaar3988f642020-08-27 22:43:03 +02002823 if (cctx.ctx_had_return
Bram Moolenaara259d8d2020-01-31 20:10:50 +01002824 && ea.cmdidx != CMD_elseif
2825 && ea.cmdidx != CMD_else
Bram Moolenaarefd88552020-06-18 20:50:10 +02002826 && ea.cmdidx != CMD_endif
2827 && ea.cmdidx != CMD_endfor
2828 && ea.cmdidx != CMD_endwhile
2829 && ea.cmdidx != CMD_catch
2830 && ea.cmdidx != CMD_finally
2831 && ea.cmdidx != CMD_endtry)
2832 {
Bram Moolenaar3988f642020-08-27 22:43:03 +02002833 emsg(_(e_unreachable_code_after_return));
2834 goto erret;
Bram Moolenaarefd88552020-06-18 20:50:10 +02002835 }
2836
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +02002837 p = skipwhite(p);
2838 if (ea.cmdidx != CMD_SIZE
2839 && ea.cmdidx != CMD_write && ea.cmdidx != CMD_read)
2840 {
Bram Moolenaar9b123d82020-09-14 22:39:11 +02002841 if (ea.cmdidx >= 0)
2842 ea.argt = excmd_get_argt(ea.cmdidx);
Bram Moolenaar0b4c66c2020-09-14 21:39:44 +02002843 if ((ea.argt & EX_BANG) && *p == '!')
2844 {
2845 ea.forceit = TRUE;
2846 p = skipwhite(p + 1);
2847 }
2848 }
2849
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002850 switch (ea.cmdidx)
2851 {
2852 case CMD_def:
Bram Moolenaar38453522021-11-28 22:00:12 +00002853 case CMD_function:
Bram Moolenaar04b12692020-05-04 23:24:44 +02002854 ea.arg = p;
2855 line = compile_nested_function(&ea, &cctx);
2856 break;
2857
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002858 case CMD_return:
Bram Moolenaar3b1373b2021-05-17 00:01:42 +02002859 line = compile_return(p, check_return_type,
2860 local_cmdmod.cmod_flags & CMOD_LEGACY, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002861 cctx.ctx_had_return = TRUE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002862 break;
2863
2864 case CMD_let:
Bram Moolenaarc58f5452020-10-21 20:58:52 +02002865 emsg(_(e_cannot_use_let_in_vim9_script));
2866 break;
Bram Moolenaar30fd8202020-09-26 15:09:30 +02002867 case CMD_var:
2868 case CMD_final:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002869 case CMD_const:
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002870 case CMD_increment:
2871 case CMD_decrement:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002872 line = compile_assignment(p, &ea, ea.cmdidx, &cctx);
Bram Moolenaar47a519a2020-06-14 23:05:10 +02002873 if (line == p)
2874 line = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002875 break;
2876
Bram Moolenaard72c1bf2020-04-19 16:28:59 +02002877 case CMD_unlet:
2878 case CMD_unlockvar:
2879 case CMD_lockvar:
2880 line = compile_unletlock(p, &ea, &cctx);
2881 break;
2882
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002883 case CMD_import:
Bram Moolenaar4db572e2021-07-18 18:21:38 +02002884 emsg(_(e_import_can_only_be_used_in_script));
2885 line = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002886 break;
2887
2888 case CMD_if:
2889 line = compile_if(p, &cctx);
2890 break;
2891 case CMD_elseif:
2892 line = compile_elseif(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002893 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002894 break;
2895 case CMD_else:
2896 line = compile_else(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002897 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002898 break;
2899 case CMD_endif:
2900 line = compile_endif(p, &cctx);
2901 break;
2902
2903 case CMD_while:
2904 line = compile_while(p, &cctx);
2905 break;
2906 case CMD_endwhile:
2907 line = compile_endwhile(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002908 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002909 break;
2910
2911 case CMD_for:
2912 line = compile_for(p, &cctx);
2913 break;
2914 case CMD_endfor:
2915 line = compile_endfor(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002916 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002917 break;
2918 case CMD_continue:
2919 line = compile_continue(p, &cctx);
2920 break;
2921 case CMD_break:
2922 line = compile_break(p, &cctx);
2923 break;
2924
2925 case CMD_try:
2926 line = compile_try(p, &cctx);
2927 break;
2928 case CMD_catch:
2929 line = compile_catch(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002930 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002931 break;
2932 case CMD_finally:
2933 line = compile_finally(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002934 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002935 break;
2936 case CMD_endtry:
2937 line = compile_endtry(p, &cctx);
Bram Moolenaarefd88552020-06-18 20:50:10 +02002938 cctx.ctx_had_return = FALSE;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002939 break;
2940 case CMD_throw:
2941 line = compile_throw(p, &cctx);
2942 break;
2943
Bram Moolenaar007f9d62020-07-06 23:04:49 +02002944 case CMD_eval:
Bram Moolenaarc3235272021-07-10 19:42:03 +02002945 line = compile_eval(p, &cctx);
Bram Moolenaar007f9d62020-07-06 23:04:49 +02002946 break;
2947
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002948 case CMD_echo:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002949 case CMD_echon:
Bram Moolenaarad39c092020-02-26 18:23:43 +01002950 case CMD_execute:
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02002951 case CMD_echomsg:
2952 case CMD_echoerr:
Bram Moolenaar7de62622021-08-07 15:05:47 +02002953 case CMD_echoconsole:
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02002954 line = compile_mult_expr(p, ea.cmdidx, &cctx);
Bram Moolenaarad39c092020-02-26 18:23:43 +01002955 break;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01002956
Bram Moolenaarc3516f72020-09-08 22:45:35 +02002957 case CMD_put:
2958 ea.cmd = cmd;
2959 line = compile_put(p, &ea, &cctx);
2960 break;
2961
Bram Moolenaar4c137212021-04-19 16:48:48 +02002962 case CMD_substitute:
Bram Moolenaar7b829262021-10-13 15:04:34 +01002963 if (check_global_and_subst(ea.cmd, p) == FAIL)
2964 goto erret;
Bram Moolenaar4c137212021-04-19 16:48:48 +02002965 if (cctx.ctx_skip == SKIP_YES)
2966 line = (char_u *)"";
2967 else
2968 {
2969 ea.arg = p;
2970 line = compile_substitute(line, &ea, &cctx);
2971 }
2972 break;
2973
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02002974 case CMD_redir:
2975 ea.arg = p;
2976 line = compile_redir(line, &ea, &cctx);
2977 break;
2978
Bram Moolenaar5f7d4c02021-05-05 21:31:39 +02002979 case CMD_cexpr:
2980 case CMD_lexpr:
2981 case CMD_caddexpr:
2982 case CMD_laddexpr:
2983 case CMD_cgetexpr:
2984 case CMD_lgetexpr:
Bram Moolenaarb7c97812021-05-05 22:51:39 +02002985#ifdef FEAT_QUICKFIX
Bram Moolenaar5f7d4c02021-05-05 21:31:39 +02002986 ea.arg = p;
2987 line = compile_cexpr(line, &ea, &cctx);
Bram Moolenaarb7c97812021-05-05 22:51:39 +02002988#else
2989 ex_ni(&ea);
2990 line = NULL;
2991#endif
Bram Moolenaar5f7d4c02021-05-05 21:31:39 +02002992 break;
2993
Bram Moolenaarae616492020-07-28 20:07:27 +02002994 case CMD_append:
2995 case CMD_change:
2996 case CMD_insert:
Bram Moolenaar10b94212021-02-19 21:42:57 +01002997 case CMD_k:
Bram Moolenaarf5a48012020-08-01 17:00:03 +02002998 case CMD_t:
Bram Moolenaarae616492020-07-28 20:07:27 +02002999 case CMD_xit:
3000 not_in_vim9(&ea);
3001 goto erret;
3002
Bram Moolenaar002262f2020-07-08 17:47:57 +02003003 case CMD_SIZE:
Bram Moolenaar3988f642020-08-27 22:43:03 +02003004 if (cctx.ctx_skip != SKIP_YES)
3005 {
3006 semsg(_(e_invalid_command_str), ea.cmd);
3007 goto erret;
3008 }
3009 // We don't check for a next command here.
3010 line = (char_u *)"";
3011 break;
Bram Moolenaar002262f2020-07-08 17:47:57 +02003012
Bram Moolenaar20677332021-06-06 17:02:53 +02003013 case CMD_lua:
3014 case CMD_mzscheme:
3015 case CMD_perl:
3016 case CMD_py3:
3017 case CMD_python3:
3018 case CMD_python:
3019 case CMD_pythonx:
3020 case CMD_ruby:
3021 case CMD_tcl:
3022 ea.arg = p;
3023 if (vim_strchr(line, '\n') == NULL)
Bram Moolenaar5163fcc2020-08-27 23:37:09 +02003024 line = compile_exec(line, &ea, &cctx);
Bram Moolenaar20677332021-06-06 17:02:53 +02003025 else
3026 // heredoc lines have been concatenated with NL
3027 // characters in get_function_body()
3028 line = compile_script(line, &cctx);
3029 break;
3030
Bram Moolenaar7b829262021-10-13 15:04:34 +01003031 case CMD_global:
3032 if (check_global_and_subst(ea.cmd, p) == FAIL)
3033 goto erret;
3034 // FALLTHROUGH
Bram Moolenaar20677332021-06-06 17:02:53 +02003035 default:
3036 // Not recognized, execute with do_cmdline_cmd().
3037 ea.arg = p;
3038 line = compile_exec(line, &ea, &cctx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003039 break;
3040 }
Bram Moolenaar5d72ce62020-08-20 23:04:06 +02003041nextline:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003042 if (line == NULL)
3043 goto erret;
Bram Moolenaar585fea72020-04-02 22:33:21 +02003044 line = skipwhite(line);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003045
Bram Moolenaarf4c6e1e2020-10-23 18:02:32 +02003046 // Undo any command modifiers.
Bram Moolenaar02194d22020-10-24 23:08:38 +02003047 generate_undo_cmdmods(&cctx);
Bram Moolenaarf4c6e1e2020-10-23 18:02:32 +02003048
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003049 if (cctx.ctx_type_stack.ga_len < 0)
3050 {
3051 iemsg("Type stack underflow");
3052 goto erret;
3053 }
3054 }
3055
3056 if (cctx.ctx_scope != NULL)
3057 {
3058 if (cctx.ctx_scope->se_type == IF_SCOPE)
3059 emsg(_(e_endif));
3060 else if (cctx.ctx_scope->se_type == WHILE_SCOPE)
3061 emsg(_(e_endwhile));
3062 else if (cctx.ctx_scope->se_type == FOR_SCOPE)
3063 emsg(_(e_endfor));
3064 else
Bram Moolenaar451c2e32020-08-15 16:33:28 +02003065 emsg(_(e_missing_rcurly));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003066 goto erret;
3067 }
3068
Bram Moolenaarefd88552020-06-18 20:50:10 +02003069 if (!cctx.ctx_had_return)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003070 {
Bram Moolenaare6174fd2021-06-12 18:30:56 +02003071 if (ufunc->uf_ret_type->tt_type == VAR_UNKNOWN)
3072 ufunc->uf_ret_type = &t_void;
3073 else if (ufunc->uf_ret_type->tt_type != VAR_VOID)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003074 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +02003075 emsg(_(e_missing_return_statement));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003076 goto erret;
3077 }
3078
Bram Moolenaarf57b43c2021-06-15 22:13:27 +02003079 // Return void if there is no return at the end.
3080 generate_instr(&cctx, ISN_RETURN_VOID);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003081 }
3082
Bram Moolenaar599410c2021-04-10 14:03:43 +02003083 // When compiled with ":silent!" and there was an error don't consider the
3084 // function compiled.
3085 if (emsg_silent == 0 || did_emsg_silent == did_emsg_silent_before)
Bram Moolenaar05afcee2020-03-31 23:32:31 +02003086 {
3087 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3088 + ufunc->uf_dfunc_idx;
3089 dfunc->df_deleted = FALSE;
Bram Moolenaar4aab88d2020-12-24 21:56:41 +01003090 dfunc->df_script_seq = current_sctx.sc_seq;
Bram Moolenaarf002a412021-01-24 13:34:18 +01003091#ifdef FEAT_PROFILE
Bram Moolenaare99d4222021-06-13 14:01:26 +02003092 if (cctx.ctx_compile_type == CT_PROFILE)
Bram Moolenaarb2049902021-01-24 12:53:53 +01003093 {
3094 dfunc->df_instr_prof = instr->ga_data;
3095 dfunc->df_instr_prof_count = instr->ga_len;
3096 }
3097 else
Bram Moolenaarf002a412021-01-24 13:34:18 +01003098#endif
Bram Moolenaare99d4222021-06-13 14:01:26 +02003099 if (cctx.ctx_compile_type == CT_DEBUG)
3100 {
3101 dfunc->df_instr_debug = instr->ga_data;
3102 dfunc->df_instr_debug_count = instr->ga_len;
3103 }
3104 else
Bram Moolenaarb2049902021-01-24 12:53:53 +01003105 {
3106 dfunc->df_instr = instr->ga_data;
3107 dfunc->df_instr_count = instr->ga_len;
3108 }
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02003109 dfunc->df_varcount = dfunc->df_var_names.ga_len;
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02003110 dfunc->df_has_closure = cctx.ctx_has_closure;
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02003111 if (cctx.ctx_outer_used)
3112 ufunc->uf_flags |= FC_CLOSURE;
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003113 ufunc->uf_def_status = UF_COMPILED;
Bram Moolenaar05afcee2020-03-31 23:32:31 +02003114 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003115
3116 ret = OK;
3117
3118erret:
Bram Moolenaar599410c2021-04-10 14:03:43 +02003119 if (ufunc->uf_def_status == UF_COMPILING)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003120 {
Bram Moolenaar05afcee2020-03-31 23:32:31 +02003121 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3122 + ufunc->uf_dfunc_idx;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003123
Bram Moolenaarcaf1a2f2021-06-15 11:27:21 +02003124 // Compiling aborted, free the generated instructions.
Bram Moolenaar8238f082021-04-20 21:10:48 +02003125 clear_instr_ga(instr);
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003126 VIM_CLEAR(dfunc->df_name);
Bram Moolenaarcaf1a2f2021-06-15 11:27:21 +02003127 ga_clear_strings(&dfunc->df_var_names);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003128
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02003129 // If using the last entry in the table and it was added above, we
3130 // might as well remove it.
3131 if (!dfunc->df_deleted && new_def_function
Bram Moolenaar45a15082020-05-25 00:28:33 +02003132 && ufunc->uf_dfunc_idx == def_functions.ga_len - 1)
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02003133 {
Bram Moolenaar20431c92020-03-20 18:39:46 +01003134 --def_functions.ga_len;
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02003135 ufunc->uf_dfunc_idx = 0;
3136 }
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003137 ufunc->uf_def_status = UF_COMPILE_ERROR;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003138
Bram Moolenaar3cca2992020-04-02 22:57:36 +02003139 while (cctx.ctx_scope != NULL)
3140 drop_scope(&cctx);
3141
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003142 if (errormsg != NULL)
3143 emsg(errormsg);
Bram Moolenaard66960b2020-10-30 20:46:26 +01003144 else if (did_emsg == did_emsg_before)
Bram Moolenaare13bdec2020-10-16 23:16:47 +02003145 emsg(_(e_compiling_def_function_failed));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003146 }
3147
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02003148 if (cctx.ctx_redir_lhs.lhs_name != NULL)
3149 {
3150 if (ret == OK)
3151 {
3152 emsg(_(e_missing_redir_end));
3153 ret = FAIL;
3154 }
3155 vim_free(cctx.ctx_redir_lhs.lhs_name);
Bram Moolenaar753bcf82021-04-21 14:24:24 +02003156 vim_free(cctx.ctx_redir_lhs.lhs_whole);
Bram Moolenaar2d1c57e2021-04-19 20:50:03 +02003157 }
3158
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003159 current_sctx = save_current_sctx;
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +02003160 estack_compiling = save_estack_compiling;
Bram Moolenaardc4c2302021-04-25 13:54:42 +02003161 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar25e0f582020-05-25 22:36:50 +02003162 if (do_estack_push)
3163 estack_pop();
3164
Bram Moolenaarf62d7392021-04-14 12:40:00 +02003165 vim_free(line_to_free);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003166 free_imported(&cctx);
Bram Moolenaarb84a3812020-05-01 15:44:29 +02003167 free_locals(&cctx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003168 ga_clear(&cctx.ctx_type_stack);
Bram Moolenaar822ba242020-05-24 23:00:18 +02003169 return ret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003170}
3171
Bram Moolenaar6ff71d82020-05-24 23:45:24 +02003172 void
3173set_function_type(ufunc_T *ufunc)
3174{
3175 int varargs = ufunc->uf_va_name != NULL;
3176 int argcount = ufunc->uf_args.ga_len;
3177
3178 // Create a type for the function, with the return type and any
3179 // argument types.
3180 // A vararg is included in uf_args.ga_len but not in uf_arg_types.
3181 // The type is included in "tt_args".
3182 if (argcount > 0 || varargs)
3183 {
Bram Moolenaar18062fc2021-03-05 21:35:47 +01003184 if (ufunc->uf_type_list.ga_itemsize == 0)
3185 ga_init2(&ufunc->uf_type_list, sizeof(type_T *), 10);
Bram Moolenaar6ff71d82020-05-24 23:45:24 +02003186 ufunc->uf_func_type = alloc_func_type(ufunc->uf_ret_type,
3187 argcount, &ufunc->uf_type_list);
3188 // Add argument types to the function type.
3189 if (func_type_add_arg_types(ufunc->uf_func_type,
3190 argcount + varargs,
3191 &ufunc->uf_type_list) == FAIL)
3192 return;
3193 ufunc->uf_func_type->tt_argcount = argcount + varargs;
3194 ufunc->uf_func_type->tt_min_argcount =
3195 argcount - ufunc->uf_def_args.ga_len;
3196 if (ufunc->uf_arg_types == NULL)
3197 {
3198 int i;
3199
3200 // lambda does not have argument types.
3201 for (i = 0; i < argcount; ++i)
3202 ufunc->uf_func_type->tt_args[i] = &t_any;
3203 }
3204 else
3205 mch_memmove(ufunc->uf_func_type->tt_args,
3206 ufunc->uf_arg_types, sizeof(type_T *) * argcount);
3207 if (varargs)
3208 {
3209 ufunc->uf_func_type->tt_args[argcount] =
Bram Moolenaar2a389082021-04-09 20:24:31 +02003210 ufunc->uf_va_type == NULL ? &t_list_any : ufunc->uf_va_type;
Bram Moolenaar6ff71d82020-05-24 23:45:24 +02003211 ufunc->uf_func_type->tt_flags = TTFLAG_VARARGS;
3212 }
3213 }
3214 else
3215 // No arguments, can use a predefined type.
3216 ufunc->uf_func_type = get_func_type(ufunc->uf_ret_type,
3217 argcount, &ufunc->uf_type_list);
3218}
3219
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003220/*
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003221 * Free all instructions for "dfunc" except df_name.
Bram Moolenaar20431c92020-03-20 18:39:46 +01003222 */
3223 static void
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003224delete_def_function_contents(dfunc_T *dfunc, int mark_deleted)
Bram Moolenaar20431c92020-03-20 18:39:46 +01003225{
3226 int idx;
3227
3228 ga_clear(&dfunc->df_def_args_isn);
Bram Moolenaarb69c6fb2021-06-14 20:40:37 +02003229 ga_clear_strings(&dfunc->df_var_names);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003230
3231 if (dfunc->df_instr != NULL)
3232 {
3233 for (idx = 0; idx < dfunc->df_instr_count; ++idx)
3234 delete_instr(dfunc->df_instr + idx);
3235 VIM_CLEAR(dfunc->df_instr);
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003236 dfunc->df_instr = NULL;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003237 }
Bram Moolenaarc2dec4c2021-06-13 15:39:00 +02003238 if (dfunc->df_instr_debug != NULL)
3239 {
3240 for (idx = 0; idx < dfunc->df_instr_debug_count; ++idx)
3241 delete_instr(dfunc->df_instr_debug + idx);
3242 VIM_CLEAR(dfunc->df_instr_debug);
3243 dfunc->df_instr_debug = NULL;
3244 }
Bram Moolenaarc05fe072021-01-24 21:30:48 +01003245#ifdef FEAT_PROFILE
3246 if (dfunc->df_instr_prof != NULL)
3247 {
3248 for (idx = 0; idx < dfunc->df_instr_prof_count; ++idx)
3249 delete_instr(dfunc->df_instr_prof + idx);
3250 VIM_CLEAR(dfunc->df_instr_prof);
3251 dfunc->df_instr_prof = NULL;
3252 }
3253#endif
Bram Moolenaar20431c92020-03-20 18:39:46 +01003254
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003255 if (mark_deleted)
3256 dfunc->df_deleted = TRUE;
3257 if (dfunc->df_ufunc != NULL)
3258 dfunc->df_ufunc->uf_def_status = UF_NOT_COMPILED;
Bram Moolenaar20431c92020-03-20 18:39:46 +01003259}
3260
3261/*
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003262 * When a user function is deleted, clear the contents of any associated def
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003263 * function, unless another user function still uses it.
3264 * The position in def_functions can be re-used.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003265 */
3266 void
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003267unlink_def_function(ufunc_T *ufunc)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003268{
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003269 if (ufunc->uf_dfunc_idx > 0)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003270 {
3271 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3272 + ufunc->uf_dfunc_idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003273
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003274 if (--dfunc->df_refcount <= 0)
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003275 delete_def_function_contents(dfunc, TRUE);
Bram Moolenaar0cb5bcf2020-06-20 18:19:09 +02003276 ufunc->uf_def_status = UF_NOT_COMPILED;
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003277 ufunc->uf_dfunc_idx = 0;
3278 if (dfunc->df_ufunc == ufunc)
3279 dfunc->df_ufunc = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003280 }
3281}
3282
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003283/*
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003284 * Used when a user function refers to an existing dfunc.
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003285 */
3286 void
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003287link_def_function(ufunc_T *ufunc)
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003288{
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003289 if (ufunc->uf_dfunc_idx > 0)
3290 {
3291 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data)
3292 + ufunc->uf_dfunc_idx;
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003293
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003294 ++dfunc->df_refcount;
3295 }
Bram Moolenaarfdeab652020-09-19 15:16:50 +02003296}
3297
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003298#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaar20431c92020-03-20 18:39:46 +01003299/*
3300 * Free all functions defined with ":def".
3301 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003302 void
3303free_def_functions(void)
3304{
Bram Moolenaar20431c92020-03-20 18:39:46 +01003305 int idx;
3306
3307 for (idx = 0; idx < def_functions.ga_len; ++idx)
3308 {
3309 dfunc_T *dfunc = ((dfunc_T *)def_functions.ga_data) + idx;
3310
Bram Moolenaarcdc40c42020-12-26 17:43:08 +01003311 delete_def_function_contents(dfunc, TRUE);
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01003312 vim_free(dfunc->df_name);
Bram Moolenaar20431c92020-03-20 18:39:46 +01003313 }
3314
3315 ga_clear(&def_functions);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01003316}
3317#endif
3318
3319
3320#endif // FEAT_EVAL