xref: /vim-8.2.3635/src/vim9script.c (revision 921ba529)
1 /* 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 /*
11  * vim9script.c: :vim9script, :import, :export and friends
12  */
13 
14 #include "vim.h"
15 
16 #if defined(FEAT_EVAL)
17 # include "vim9.h"
18 #endif
19 
20 /*
21  * Return TRUE when currently using Vim9 script syntax.
22  * Does not go up the stack, a ":function" inside vim9script uses legacy
23  * syntax.
24  */
25     int
26 in_vim9script(void)
27 {
28     // "sc_version" is also set when compiling a ":def" function in legacy
29     // script.
30     return (current_sctx.sc_version == SCRIPT_VERSION_VIM9
31 					 || (cmdmod.cmod_flags & CMOD_VIM9CMD))
32 		&& !(cmdmod.cmod_flags & CMOD_LEGACY);
33 }
34 
35 #if defined(FEAT_EVAL) || defined(PROTO)
36 /*
37  * Return TRUE if the current script is Vim9 script.
38  * This also returns TRUE in a legacy function in a Vim9 script.
39  */
40     int
41 current_script_is_vim9(void)
42 {
43     return SCRIPT_ID_VALID(current_sctx.sc_sid)
44 	    && SCRIPT_ITEM(current_sctx.sc_sid)->sn_version
45 						       == SCRIPT_VERSION_VIM9;
46 }
47 #endif
48 
49 /*
50  * ":vim9script".
51  */
52     void
53 ex_vim9script(exarg_T *eap UNUSED)
54 {
55 #ifdef FEAT_EVAL
56     int		    sid = current_sctx.sc_sid;
57     scriptitem_T    *si;
58 
59     if (!getline_equal(eap->getline, eap->cookie, getsourceline))
60     {
61 	emsg(_(e_vim9script_can_only_be_used_in_script));
62 	return;
63     }
64 
65     si = SCRIPT_ITEM(sid);
66     if (si->sn_state == SN_STATE_HAD_COMMAND)
67     {
68 	emsg(_(e_vim9script_must_be_first_command_in_script));
69 	return;
70     }
71     if (!IS_WHITE_OR_NUL(*eap->arg) && STRCMP(eap->arg, "noclear") != 0)
72     {
73 	semsg(_(e_invarg2), eap->arg);
74 	return;
75     }
76     if (si->sn_state == SN_STATE_RELOAD && IS_WHITE_OR_NUL(*eap->arg))
77     {
78 	hashtab_T	*ht = &SCRIPT_VARS(sid);
79 
80 	// Reloading a script without the "noclear" argument: clear
81 	// script-local variables and functions.
82 	hashtab_free_contents(ht);
83 	hash_init(ht);
84 	delete_script_functions(sid);
85 
86 	// old imports and script variables are no longer valid
87 	free_imports_and_script_vars(sid);
88     }
89     si->sn_state = SN_STATE_HAD_COMMAND;
90 
91     current_sctx.sc_version = SCRIPT_VERSION_VIM9;
92     si->sn_version = SCRIPT_VERSION_VIM9;
93 
94     if (STRCMP(p_cpo, CPO_VIM) != 0)
95     {
96 	si->sn_save_cpo = vim_strsave(p_cpo);
97 	set_option_value((char_u *)"cpo", 0L, (char_u *)CPO_VIM, OPT_NO_REDRAW);
98     }
99 #else
100     // No check for this being the first command, it doesn't matter.
101     current_sctx.sc_version = SCRIPT_VERSION_VIM9;
102 #endif
103 }
104 
105 /*
106  * When in Vim9 script give an error and return FAIL.
107  */
108     int
109 not_in_vim9(exarg_T *eap)
110 {
111     if (in_vim9script())
112 	switch (eap->cmdidx)
113 	{
114 	    case CMD_k:
115 		if (eap->addr_count > 0)
116 		{
117 		    emsg(_(e_norange));
118 		    return FAIL;
119 		}
120 		// FALLTHROUGH
121 	    case CMD_append:
122 	    case CMD_change:
123 	    case CMD_insert:
124 	    case CMD_open:
125 	    case CMD_t:
126 	    case CMD_xit:
127 		semsg(_(e_command_not_supported_in_vim9_script_missing_var_str), eap->cmd);
128 		return FAIL;
129 	    default: break;
130 	}
131     return OK;
132 }
133 
134 /*
135  * Give an error message if "p" points at "#{" and return TRUE.
136  * This avoids that using a legacy style #{} dictionary leads to difficult to
137  * understand errors.
138  */
139     int
140 vim9_bad_comment(char_u *p)
141 {
142     if (p[0] == '#' && p[1] == '{' && p[2] != '{')
143     {
144 	emsg(_(e_cannot_use_hash_curly_to_start_comment));
145 	return TRUE;
146     }
147     return FALSE;
148 }
149 
150 /*
151  * Return TRUE if "p" points at a "#" not followed by one '{'.
152  * Does not check for white space.
153  */
154     int
155 vim9_comment_start(char_u *p)
156 {
157     return p[0] == '#' && (p[1] != '{' || p[2] == '{');
158 }
159 
160 #if defined(FEAT_EVAL) || defined(PROTO)
161 
162 /*
163  * "++nr" and "--nr" commands.
164  */
165     void
166 ex_incdec(exarg_T *eap)
167 {
168     char_u	*cmd = eap->cmd;
169     char_u	*nextcmd = eap->nextcmd;
170     size_t	len = STRLEN(eap->cmd) + 8;
171 
172     if (VIM_ISWHITE(cmd[2]))
173     {
174 	semsg(_(e_no_white_space_allowed_after_str_str),
175 			 eap->cmdidx == CMD_increment ? "++" : "--", eap->cmd);
176 	return;
177     }
178 
179     // This works like "nr += 1" or "nr -= 1".
180     // Add a '|' to avoid looking in the next line.
181     eap->cmd = alloc(len);
182     if (eap->cmd == NULL)
183 	return;
184     vim_snprintf((char *)eap->cmd, len, "%s %c= 1 |", cmd + 2,
185 				     eap->cmdidx == CMD_increment ? '+' : '-');
186     eap->arg = eap->cmd;
187     eap->cmdidx = CMD_var;
188     eap->nextcmd = NULL;
189     ex_let(eap);
190     vim_free(eap->cmd);
191     eap->cmd = cmd;
192     eap->nextcmd = nextcmd;
193 }
194 
195 /*
196  * ":export let Name: type"
197  * ":export const Name: type"
198  * ":export def Name(..."
199  * ":export class Name ..."
200  */
201     void
202 ex_export(exarg_T *eap)
203 {
204     if (!in_vim9script())
205     {
206 	emsg(_(e_export_can_only_be_used_in_vim9script));
207 	return;
208     }
209 
210     eap->cmd = eap->arg;
211     (void)find_ex_command(eap, NULL, lookup_scriptitem, NULL);
212     switch (eap->cmdidx)
213     {
214 	case CMD_let:
215 	case CMD_var:
216 	case CMD_final:
217 	case CMD_const:
218 	case CMD_def:
219 	// case CMD_class:
220 	    is_export = TRUE;
221 	    do_cmdline(eap->cmd, eap->getline, eap->cookie,
222 						DOCMD_VERBOSE + DOCMD_NOWAIT);
223 
224 	    // The command will reset "is_export" when exporting an item.
225 	    if (is_export)
226 	    {
227 		emsg(_(e_export_with_invalid_argument));
228 		is_export = FALSE;
229 	    }
230 	    break;
231 	default:
232 	    emsg(_(e_invalid_command_after_export));
233 	    break;
234     }
235 }
236 
237 /*
238  * Add a new imported item entry to the current script.
239  */
240     static imported_T *
241 new_imported(garray_T *gap)
242 {
243     if (ga_grow(gap, 1) == OK)
244 	return ((imported_T *)gap->ga_data + gap->ga_len++);
245     return NULL;
246 }
247 
248 /*
249  * Free all imported items in script "sid".
250  */
251     void
252 free_imports_and_script_vars(int sid)
253 {
254     scriptitem_T    *si = SCRIPT_ITEM(sid);
255     int		    idx;
256 
257     for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
258     {
259 	imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx;
260 
261 	vim_free(imp->imp_name);
262     }
263     ga_clear(&si->sn_imports);
264 
265     free_all_script_vars(si);
266 
267     clear_type_list(&si->sn_type_list);
268 }
269 
270 /*
271  * Mark all imports as possible to redefine.  Used when a script is loaded
272  * again but not cleared.
273  */
274     void
275 mark_imports_for_reload(int sid)
276 {
277     scriptitem_T    *si = SCRIPT_ITEM(sid);
278     int		    idx;
279 
280     for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
281     {
282 	imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx;
283 
284 	imp->imp_flags |= IMP_FLAGS_RELOAD;
285     }
286 }
287 
288 /*
289  * ":import Item from 'filename'"
290  * ":import Item as Alias from 'filename'"
291  * ":import {Item} from 'filename'".
292  * ":import {Item as Alias} from 'filename'"
293  * ":import {Item, Item} from 'filename'"
294  * ":import {Item, Item as Alias} from 'filename'"
295  *
296  * ":import * as Name from 'filename'"
297  */
298     void
299 ex_import(exarg_T *eap)
300 {
301     char_u	*cmd_end;
302     evalarg_T	evalarg;
303 
304     if (!getline_equal(eap->getline, eap->cookie, getsourceline))
305     {
306 	emsg(_(e_import_can_only_be_used_in_script));
307 	return;
308     }
309     fill_evalarg_from_eap(&evalarg, eap, eap->skip);
310 
311     cmd_end = handle_import(eap->arg, NULL, current_sctx.sc_sid,
312 							       &evalarg, NULL);
313     if (cmd_end != NULL)
314 	eap->nextcmd = check_nextcmd(cmd_end);
315     clear_evalarg(&evalarg, eap);
316 }
317 
318 /*
319  * Find an exported item in "sid" matching the name at "*argp".
320  * When it is a variable return the index.
321  * When it is a user function return "*ufunc".
322  * When not found returns -1 and "*ufunc" is NULL.
323  */
324     int
325 find_exported(
326 	int	    sid,
327 	char_u	    *name,
328 	ufunc_T	    **ufunc,
329 	type_T	    **type,
330 	cctx_T	    *cctx,
331 	int	    verbose)
332 {
333     int		idx = -1;
334     svar_T	*sv;
335     scriptitem_T *script = SCRIPT_ITEM(sid);
336 
337     // Find name in "script".
338     idx = get_script_item_idx(sid, name, 0, cctx);
339     if (idx >= 0)
340     {
341 	sv = ((svar_T *)script->sn_var_vals.ga_data) + idx;
342 	if (!sv->sv_export)
343 	{
344 	    if (verbose)
345 		semsg(_(e_item_not_exported_in_script_str), name);
346 	    return -1;
347 	}
348 	*type = sv->sv_type;
349 	*ufunc = NULL;
350     }
351     else
352     {
353 	char_u	buffer[200];
354 	char_u	*funcname;
355 
356 	// it could be a user function.
357 	if (STRLEN(name) < sizeof(buffer) - 15)
358 	    funcname = buffer;
359 	else
360 	{
361 	    funcname = alloc(STRLEN(name) + 15);
362 	    if (funcname == NULL)
363 		return -1;
364 	}
365 	funcname[0] = K_SPECIAL;
366 	funcname[1] = KS_EXTRA;
367 	funcname[2] = (int)KE_SNR;
368 	sprintf((char *)funcname + 3, "%ld_%s", (long)sid, name);
369 	*ufunc = find_func(funcname, FALSE, NULL);
370 	if (funcname != buffer)
371 	    vim_free(funcname);
372 
373 	if (*ufunc == NULL)
374 	{
375 	    if (verbose)
376 		semsg(_(e_item_not_found_in_script_str), name);
377 	    return -1;
378 	}
379 	else if (((*ufunc)->uf_flags & FC_EXPORT) == 0)
380 	{
381 	    if (verbose)
382 		semsg(_(e_item_not_exported_in_script_str), name);
383 	    *ufunc = NULL;
384 	    return -1;
385 	}
386     }
387 
388     return idx;
389 }
390 
391 /*
392  * Handle an ":import" command and add the resulting imported_T to "gap", when
393  * not NULL, or script "import_sid" sn_imports.
394  * "cctx" is NULL at the script level.
395  * Returns a pointer to after the command or NULL in case of failure
396  */
397     char_u *
398 handle_import(
399 	char_u	    *arg_start,
400 	garray_T    *gap,
401 	int	    import_sid,
402 	evalarg_T   *evalarg,
403 	void	    *cctx)
404 {
405     char_u	*arg = arg_start;
406     char_u	*cmd_end = NULL;
407     int		ret = FAIL;
408     typval_T	tv;
409     int		sid = -1;
410     int		res;
411     int		mult = FALSE;
412     garray_T	names;
413     garray_T	as_names;
414     long	start_lnum = SOURCING_LNUM;
415 
416     tv.v_type = VAR_UNKNOWN;
417     ga_init2(&names, sizeof(char_u *), 10);
418     ga_init2(&as_names, sizeof(char_u *), 10);
419     if (*arg == '{')
420     {
421 	// "import {item, item} from ..."
422 	mult = TRUE;
423 	arg = skipwhite_and_linebreak(arg + 1, evalarg);
424     }
425 
426     for (;;)
427     {
428 	char_u	    *p = arg;
429 	int	    had_comma = FALSE;
430 	char_u	    *as_name = NULL;
431 
432 	// accept "*" or "Name"
433 	if (!mult && arg[0] == '*' && IS_WHITE_OR_NUL(arg[1]))
434 	    ++arg;
435 	else
436 	    while (eval_isnamec(*arg))
437 		++arg;
438 	if (p == arg)
439 	    break;
440 	if (ga_grow(&names, 1) == FAIL || ga_grow(&as_names, 1) == FAIL)
441 	    goto erret;
442 	((char_u **)names.ga_data)[names.ga_len] = vim_strnsave(p, arg - p);
443 	++names.ga_len;
444 
445 	arg = skipwhite_and_linebreak(arg, evalarg);
446 	if (STRNCMP("as", arg, 2) == 0 && IS_WHITE_OR_NUL(arg[2]))
447 	{
448 	    // skip over "as Name "; no line break allowed after "as"
449 	    arg = skipwhite(arg + 2);
450 	    p = arg;
451 	    if (eval_isnamec1(*arg))
452 		while (eval_isnamec(*arg))
453 		    ++arg;
454 	    if (check_defined(p, arg - p, cctx, FALSE) == FAIL)
455 		goto erret;
456 	    as_name = vim_strnsave(p, arg - p);
457 	    arg = skipwhite_and_linebreak(arg, evalarg);
458 	}
459 	else if (*arg_start == '*')
460 	{
461 	    emsg(_(e_missing_as_after_star));
462 	    goto erret;
463 	}
464 	// without "as Name" the as_names entry is NULL
465 	((char_u **)as_names.ga_data)[as_names.ga_len] = as_name;
466 	++as_names.ga_len;
467 
468 	if (!mult)
469 	    break;
470 	if (*arg == ',')
471 	{
472 	    had_comma = TRUE;
473 	    ++arg;
474 	}
475 	arg = skipwhite_and_linebreak(arg, evalarg);
476 	if (*arg == '}')
477 	{
478 	    ++arg;
479 	    break;
480 	}
481 	if (!had_comma)
482 	{
483 	    emsg(_(e_missing_comma_in_import));
484 	    goto erret;
485 	}
486     }
487     arg = skipwhite_and_linebreak(arg, evalarg);
488 
489     if (names.ga_len == 0)
490     {
491 	emsg(_(e_syntax_error_in_import));
492 	goto erret;
493     }
494 
495     if (STRNCMP("from", arg, 4) != 0 || !IS_WHITE_OR_NUL(arg[4]))
496     {
497 	emsg(_(e_missing_from));
498 	goto erret;
499     }
500 
501     // The name of the file can be an expression, which must evaluate to a
502     // string.
503     arg = skipwhite_and_linebreak(arg + 4, evalarg);
504     ret = eval0(arg, &tv, NULL, evalarg);
505     if (ret == FAIL)
506 	goto erret;
507     if (tv.v_type != VAR_STRING
508 		       || tv.vval.v_string == NULL || *tv.vval.v_string == NUL)
509     {
510 	emsg(_(e_invalid_string_after_from));
511 	goto erret;
512     }
513     cmd_end = arg;
514 
515     // Give error messages for the start of the line.
516     SOURCING_LNUM = start_lnum;
517 
518     /*
519      * find script file
520      */
521     if (*tv.vval.v_string == '.')
522     {
523 	size_t		len;
524 	scriptitem_T	*si = SCRIPT_ITEM(current_sctx.sc_sid);
525 	char_u		*tail = gettail(si->sn_name);
526 	char_u		*from_name;
527 
528 	// Relative to current script: "./name.vim", "../../name.vim".
529 	len = STRLEN(si->sn_name) - STRLEN(tail) + STRLEN(tv.vval.v_string) + 2;
530 	from_name = alloc((int)len);
531 	if (from_name == NULL)
532 	    goto erret;
533 	vim_strncpy(from_name, si->sn_name, tail - si->sn_name);
534 	add_pathsep(from_name);
535 	STRCAT(from_name, tv.vval.v_string);
536 	simplify_filename(from_name);
537 
538 	res = do_source(from_name, FALSE, DOSO_NONE, &sid);
539 	vim_free(from_name);
540     }
541     else if (mch_isFullName(tv.vval.v_string))
542     {
543 	// Absolute path: "/tmp/name.vim"
544 	res = do_source(tv.vval.v_string, FALSE, DOSO_NONE, &sid);
545     }
546     else
547     {
548 	size_t	    len = 7 + STRLEN(tv.vval.v_string) + 1;
549 	char_u	    *from_name;
550 
551 	// Find file in "import" subdirs in 'runtimepath'.
552 	from_name = alloc((int)len);
553 	if (from_name == NULL)
554 	{
555 	    goto erret;
556 	}
557 	vim_snprintf((char *)from_name, len, "import/%s", tv.vval.v_string);
558 	res = source_in_path(p_rtp, from_name, DIP_NOAFTER, &sid);
559 	vim_free(from_name);
560     }
561 
562     if (res == FAIL || sid <= 0)
563     {
564 	semsg(_(e_could_not_import_str), tv.vval.v_string);
565 	goto erret;
566     }
567 
568     if (*arg_start == '*')
569     {
570 	imported_T  *imported;
571 	char_u	    *as_name = ((char_u **)as_names.ga_data)[0];
572 
573 	// "import * as That"
574 	imported = find_imported(as_name, STRLEN(as_name), cctx);
575 	if (imported != NULL && imported->imp_sid == sid)
576 	{
577 	    if (imported->imp_flags & IMP_FLAGS_RELOAD)
578 		// import already defined on a previous script load
579 		imported->imp_flags &= ~IMP_FLAGS_RELOAD;
580 	    else
581 	    {
582 		semsg(_(e_name_already_defined_str), as_name);
583 		goto erret;
584 	    }
585 	}
586 
587 	imported = new_imported(gap != NULL ? gap
588 					: &SCRIPT_ITEM(import_sid)->sn_imports);
589 	if (imported == NULL)
590 	    goto erret;
591 	imported->imp_name = as_name;
592 	((char_u **)as_names.ga_data)[0] = NULL;
593 	imported->imp_sid = sid;
594 	imported->imp_flags = IMP_FLAGS_STAR;
595     }
596     else
597     {
598 	int i;
599 
600 	arg = arg_start;
601 	if (*arg == '{')
602 	    arg = skipwhite(arg + 1);
603 	for (i = 0; i < names.ga_len; ++i)
604 	{
605 	    char_u	*name = ((char_u **)names.ga_data)[i];
606 	    char_u	*as_name = ((char_u **)as_names.ga_data)[i];
607 	    size_t	len = STRLEN(name);
608 	    int		idx;
609 	    imported_T	*imported;
610 	    ufunc_T	*ufunc = NULL;
611 	    type_T	*type;
612 
613 	    idx = find_exported(sid, name, &ufunc, &type, cctx, TRUE);
614 
615 	    if (idx < 0 && ufunc == NULL)
616 		goto erret;
617 
618 	    // If already imported with the same properties and the
619 	    // IMP_FLAGS_RELOAD set then we keep that entry.  Otherwise create
620 	    // a new one (and give an error for an existing import).
621 	    imported = find_imported(name, len, cctx);
622 	    if (imported != NULL
623 		    && (imported->imp_flags & IMP_FLAGS_RELOAD)
624 		    && imported->imp_sid == sid
625 		    && (idx >= 0
626 			? (equal_type(imported->imp_type, type)
627 			    && imported->imp_var_vals_idx == idx)
628 			: (equal_type(imported->imp_type, ufunc->uf_func_type)
629 			    && STRCMP(imported->imp_funcname,
630 							ufunc->uf_name) == 0)))
631 	    {
632 		imported->imp_flags &= ~IMP_FLAGS_RELOAD;
633 	    }
634 	    else
635 	    {
636 		if (as_name == NULL
637 			      && check_defined(name, len, cctx, FALSE) == FAIL)
638 		    goto erret;
639 
640 		imported = new_imported(gap != NULL ? gap
641 				       : &SCRIPT_ITEM(import_sid)->sn_imports);
642 		if (imported == NULL)
643 		    goto erret;
644 
645 		if (as_name == NULL)
646 		{
647 		    imported->imp_name = name;
648 		    ((char_u **)names.ga_data)[i] = NULL;
649 		}
650 		else
651 		{
652 		    // "import This as That ..."
653 		    imported->imp_name = as_name;
654 		    ((char_u **)as_names.ga_data)[i] = NULL;
655 		}
656 		imported->imp_sid = sid;
657 		if (idx >= 0)
658 		{
659 		    imported->imp_type = type;
660 		    imported->imp_var_vals_idx = idx;
661 		}
662 		else
663 		{
664 		    imported->imp_type = ufunc->uf_func_type;
665 		    imported->imp_funcname = ufunc->uf_name;
666 		}
667 	    }
668 	}
669     }
670 erret:
671     clear_tv(&tv);
672     ga_clear_strings(&names);
673     ga_clear_strings(&as_names);
674     return cmd_end;
675 }
676 
677 /*
678  * Declare a script-local variable without init: "let var: type".
679  * "const" is an error since the value is missing.
680  * Returns a pointer to after the type.
681  */
682     char_u *
683 vim9_declare_scriptvar(exarg_T *eap, char_u *arg)
684 {
685     char_u	    *p;
686     char_u	    *name;
687     scriptitem_T    *si = SCRIPT_ITEM(current_sctx.sc_sid);
688     type_T	    *type;
689     typval_T	    init_tv;
690 
691     if (eap->cmdidx == CMD_final || eap->cmdidx == CMD_const)
692     {
693 	if (eap->cmdidx == CMD_final)
694 	    emsg(_(e_final_requires_a_value));
695 	else
696 	    emsg(_(e_const_requires_a_value));
697 	return arg + STRLEN(arg);
698     }
699 
700     // Check for valid starting character.
701     if (!eval_isnamec1(*arg))
702     {
703 	semsg(_(e_invarg2), arg);
704 	return arg + STRLEN(arg);
705     }
706 
707     for (p = arg + 1; *p != NUL && eval_isnamec(*p); MB_PTR_ADV(p))
708 	if (*p == ':' && (VIM_ISWHITE(p[1]) || p != arg + 1))
709 	    break;
710 
711     if (*p != ':')
712     {
713 	emsg(_(e_type_or_initialization_required));
714 	return arg + STRLEN(arg);
715     }
716     if (!VIM_ISWHITE(p[1]))
717     {
718 	semsg(_(e_white_space_required_after_str_str), ":", p);
719 	return arg + STRLEN(arg);
720     }
721     name = vim_strnsave(arg, p - arg);
722 
723     // parse type, check for reserved name
724     p = skipwhite(p + 1);
725     type = parse_type(&p, &si->sn_type_list, TRUE);
726     if (type == NULL || check_reserved_name(name) == FAIL)
727     {
728 	vim_free(name);
729 	return p;
730     }
731 
732     // Create the variable with 0/NULL value.
733     CLEAR_FIELD(init_tv);
734     if (type->tt_type == VAR_ANY)
735 	// A variable of type "any" is not possible, just use zero instead
736 	init_tv.v_type = VAR_NUMBER;
737     else
738 	init_tv.v_type = type->tt_type;
739     set_var_const(name, type, &init_tv, FALSE, 0, 0);
740 
741     vim_free(name);
742     return p;
743 }
744 
745 /*
746  * Vim9 part of adding a script variable: add it to sn_all_vars (lookup by name
747  * with a hashtable) and sn_var_vals (lookup by index).
748  * When "create" is TRUE this is a new variable, otherwise find and update an
749  * existing variable.
750  * "flags" can have ASSIGN_FINAL or ASSIGN_CONST.
751  * When "*type" is NULL use "tv" for the type and update "*type".  If
752  * "do_member" is TRUE also use the member type, otherwise use "any".
753  */
754     void
755 update_vim9_script_var(
756 	int	    create,
757 	dictitem_T  *di,
758 	int	    flags,
759 	typval_T    *tv,
760 	type_T	    **type,
761 	int	    do_member)
762 {
763     scriptitem_T    *si = SCRIPT_ITEM(current_sctx.sc_sid);
764     hashitem_T	    *hi;
765     svar_T	    *sv = NULL;
766 
767     if (create)
768     {
769 	sallvar_T   *newsav;
770 	sallvar_T   *sav = NULL;
771 
772 	// Store a pointer to the typval_T, so that it can be found by index
773 	// instead of using a hastab lookup.
774 	if (ga_grow(&si->sn_var_vals, 1) == FAIL)
775 	    return;
776 
777 	hi = hash_find(&si->sn_all_vars.dv_hashtab, di->di_key);
778 	if (!HASHITEM_EMPTY(hi))
779 	{
780 	    // Variable with this name exists, either in this block or in
781 	    // another block.
782 	    for (sav = HI2SAV(hi); ; sav = sav->sav_next)
783 	    {
784 		if (sav->sav_block_id == si->sn_current_block_id)
785 		{
786 		    // variable defined in a loop, re-use the entry
787 		    sv = ((svar_T *)si->sn_var_vals.ga_data)
788 						       + sav->sav_var_vals_idx;
789 		    // unhide the variable
790 		    if (sv->sv_tv == &sav->sav_tv)
791 		    {
792 			clear_tv(&sav->sav_tv);
793 			sv->sv_tv = &di->di_tv;
794 			sav->sav_di = di;
795 		    }
796 		    break;
797 		}
798 		if (sav->sav_next == NULL)
799 		    break;
800 	    }
801 	}
802 
803 	if (sv == NULL)
804 	{
805 	    // Variable not defined or not defined in current block: Add a
806 	    // svar_T and create a new sallvar_T.
807 	    sv = ((svar_T *)si->sn_var_vals.ga_data) + si->sn_var_vals.ga_len;
808 	    newsav = (sallvar_T *)alloc_clear(
809 				       sizeof(sallvar_T) + STRLEN(di->di_key));
810 	    if (newsav == NULL)
811 		return;
812 
813 	    sv->sv_tv = &di->di_tv;
814 	    sv->sv_const = (flags & ASSIGN_FINAL) ? ASSIGN_FINAL
815 				   : (flags & ASSIGN_CONST) ? ASSIGN_CONST : 0;
816 	    sv->sv_export = is_export;
817 	    newsav->sav_var_vals_idx = si->sn_var_vals.ga_len;
818 	    ++si->sn_var_vals.ga_len;
819 	    STRCPY(&newsav->sav_key, di->di_key);
820 	    sv->sv_name = newsav->sav_key;
821 	    newsav->sav_di = di;
822 	    newsav->sav_block_id = si->sn_current_block_id;
823 
824 	    if (HASHITEM_EMPTY(hi))
825 		// new variable name
826 		hash_add(&si->sn_all_vars.dv_hashtab, newsav->sav_key);
827 	    else if (sav != NULL)
828 		// existing name in a new block, append to the list
829 		sav->sav_next = newsav;
830 	}
831     }
832     else
833     {
834 	sv = find_typval_in_script(&di->di_tv);
835     }
836     if (sv != NULL)
837     {
838 	if (*type == NULL)
839 	    *type = typval2type(tv, get_copyID(), &si->sn_type_list, do_member);
840 	sv->sv_type = *type;
841     }
842 
843     // let ex_export() know the export worked.
844     is_export = FALSE;
845 }
846 
847 /*
848  * Hide a script variable when leaving a block.
849  * "idx" is de index in sn_var_vals.
850  * When "func_defined" is non-zero then a function was defined in this block,
851  * the variable may be accessed by it.  Otherwise the variable can be cleared.
852  */
853     void
854 hide_script_var(scriptitem_T *si, int idx, int func_defined)
855 {
856     svar_T	*sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
857     hashtab_T	*script_ht = get_script_local_ht();
858     hashtab_T	*all_ht = &si->sn_all_vars.dv_hashtab;
859     hashitem_T	*script_hi;
860     hashitem_T	*all_hi;
861 
862     // Remove a variable declared inside the block, if it still exists.
863     // If it was added in a nested block it will already have been removed.
864     // The typval is moved into the sallvar_T.
865     script_hi = hash_find(script_ht, sv->sv_name);
866     all_hi = hash_find(all_ht, sv->sv_name);
867     if (!HASHITEM_EMPTY(script_hi) && !HASHITEM_EMPTY(all_hi))
868     {
869 	dictitem_T	*di = HI2DI(script_hi);
870 	sallvar_T	*sav = HI2SAV(all_hi);
871 	sallvar_T	*sav_prev = NULL;
872 
873 	// There can be multiple entries with the same name in different
874 	// blocks, find the right one.
875 	while (sav != NULL && sav->sav_var_vals_idx != idx)
876 	{
877 	    sav_prev = sav;
878 	    sav = sav->sav_next;
879 	}
880 	if (sav != NULL)
881 	{
882 	    if (func_defined)
883 	    {
884 		// move the typval from the dictitem to the sallvar
885 		sav->sav_tv = di->di_tv;
886 		di->di_tv.v_type = VAR_UNKNOWN;
887 		sav->sav_flags = di->di_flags;
888 		sav->sav_di = NULL;
889 		sv->sv_tv = &sav->sav_tv;
890 	    }
891 	    else
892 	    {
893 		if (sav_prev == NULL)
894 		    hash_remove(all_ht, all_hi);
895 		else
896 		    sav_prev->sav_next = sav->sav_next;
897 		sv->sv_name = NULL;
898 		vim_free(sav);
899 	    }
900 	    delete_var(script_ht, script_hi);
901 	}
902     }
903 }
904 
905 /*
906  * Free the script variables from "sn_all_vars".
907  */
908     void
909 free_all_script_vars(scriptitem_T *si)
910 {
911     int		todo;
912     hashtab_T	*ht = &si->sn_all_vars.dv_hashtab;
913     hashitem_T	*hi;
914     sallvar_T	*sav;
915     sallvar_T	*sav_next;
916 
917     hash_lock(ht);
918     todo = (int)ht->ht_used;
919     for (hi = ht->ht_array; todo > 0; ++hi)
920     {
921 	if (!HASHITEM_EMPTY(hi))
922 	{
923 	    --todo;
924 
925 	    // Free the variable.  Don't remove it from the hashtab, ht_array
926 	    // might change then.  hash_clear() takes care of it later.
927 	    sav = HI2SAV(hi);
928 	    while (sav != NULL)
929 	    {
930 		sav_next = sav->sav_next;
931 		if (sav->sav_di == NULL)
932 		    clear_tv(&sav->sav_tv);
933 		vim_free(sav);
934 		sav = sav_next;
935 	    }
936 	}
937     }
938     hash_clear(ht);
939     hash_init(ht);
940 
941     ga_clear(&si->sn_var_vals);
942 
943     // existing commands using script variable indexes are no longer valid
944     si->sn_script_seq = current_sctx.sc_seq;
945 }
946 
947 /*
948  * Find the script-local variable that links to "dest".
949  * Returns NULL if not found and give an internal error.
950  */
951     svar_T *
952 find_typval_in_script(typval_T *dest)
953 {
954     scriptitem_T    *si = SCRIPT_ITEM(current_sctx.sc_sid);
955     int		    idx;
956 
957     if (si->sn_version != SCRIPT_VERSION_VIM9)
958 	// legacy script doesn't store variable types
959 	return NULL;
960 
961     // Find the svar_T in sn_var_vals.
962     for (idx = 0; idx < si->sn_var_vals.ga_len; ++idx)
963     {
964 	svar_T    *sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
965 
966 	// If "sv_name" is NULL the variable was hidden when leaving a block,
967 	// don't check "sv_tv" then, it might be used for another variable now.
968 	if (sv->sv_name != NULL && sv->sv_tv == dest)
969 	    return sv;
970     }
971     iemsg("find_typval_in_script(): not found");
972     return NULL;
973 }
974 
975 /*
976  * Check if the type of script variable "dest" allows assigning "value".
977  * If needed convert "value" to a bool.
978  */
979     int
980 check_script_var_type(
981 	typval_T    *dest,
982 	typval_T    *value,
983 	char_u	    *name,
984 	where_T	    where)
985 {
986     svar_T  *sv = find_typval_in_script(dest);
987     int	    ret;
988 
989     if (sv != NULL)
990     {
991 	if (sv->sv_const != 0)
992 	{
993 	    semsg(_(e_cannot_change_readonly_variable_str), name);
994 	    return FAIL;
995 	}
996 	ret = check_typval_type(sv->sv_type, value, where);
997 	if (ret == OK && need_convert_to_bool(sv->sv_type, value))
998 	{
999 	    int	val = tv2bool(value);
1000 
1001 	    clear_tv(value);
1002 	    value->v_type = VAR_BOOL;
1003 	    value->v_lock = 0;
1004 	    value->vval.v_number = val ? VVAL_TRUE : VVAL_FALSE;
1005 	}
1006 	return ret;
1007     }
1008 
1009     return OK; // not really
1010 }
1011 
1012 // words that cannot be used as a variable
1013 static char *reserved[] = {
1014     "true",
1015     "false",
1016     "null",
1017     "this",
1018     NULL
1019 };
1020 
1021     int
1022 check_reserved_name(char_u *name)
1023 {
1024     int idx;
1025 
1026     for (idx = 0; reserved[idx] != NULL; ++idx)
1027 	if (STRCMP(reserved[idx], name) == 0)
1028 	{
1029 	    semsg(_(e_cannot_use_reserved_name), name);
1030 	    return FAIL;
1031 	}
1032     return OK;
1033 }
1034 
1035 #endif // FEAT_EVAL
1036