xref: /vim-8.2.3635/src/vim9script.c (revision 529fb5a5)
18a7d6542SBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet:
28a7d6542SBram Moolenaar  *
38a7d6542SBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
48a7d6542SBram Moolenaar  *
58a7d6542SBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
68a7d6542SBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
78a7d6542SBram Moolenaar  * See README.txt for an overview of the Vim source code.
88a7d6542SBram Moolenaar  */
98a7d6542SBram Moolenaar 
108a7d6542SBram Moolenaar /*
118a7d6542SBram Moolenaar  * vim9script.c: :vim9script, :import, :export and friends
128a7d6542SBram Moolenaar  */
138a7d6542SBram Moolenaar 
148a7d6542SBram Moolenaar #include "vim.h"
158a7d6542SBram Moolenaar 
169b8d6226SBram Moolenaar #if defined(FEAT_EVAL)
178a7d6542SBram Moolenaar # include "vim9.h"
189b8d6226SBram Moolenaar #endif
198a7d6542SBram Moolenaar 
20e535db86SBram Moolenaar /*
21e535db86SBram Moolenaar  * Return TRUE when currently using Vim9 script syntax.
22e535db86SBram Moolenaar  * Does not go up the stack, a ":function" inside vim9script uses legacy
23e535db86SBram Moolenaar  * syntax.
24e535db86SBram Moolenaar  */
258a7d6542SBram Moolenaar     int
268a7d6542SBram Moolenaar in_vim9script(void)
278a7d6542SBram Moolenaar {
28e535db86SBram Moolenaar     // "sc_version" is also set when compiling a ":def" function in legacy
29e535db86SBram Moolenaar     // script.
309979fcd7SBram Moolenaar     return current_sctx.sc_version == SCRIPT_VERSION_VIM9
319979fcd7SBram Moolenaar 		|| (cmdmod.cmod_flags & CMOD_VIM9CMD);
328a7d6542SBram Moolenaar }
338a7d6542SBram Moolenaar 
348a7d6542SBram Moolenaar /*
35e535db86SBram Moolenaar  * Return TRUE if the current script is Vim9 script.
36e535db86SBram Moolenaar  * This also returns TRUE in a legacy function in a Vim9 script.
37e535db86SBram Moolenaar  */
38e535db86SBram Moolenaar     int
39e535db86SBram Moolenaar current_script_is_vim9(void)
40e535db86SBram Moolenaar {
41e535db86SBram Moolenaar     return SCRIPT_ID_VALID(current_sctx.sc_sid)
42e535db86SBram Moolenaar 	    && SCRIPT_ITEM(current_sctx.sc_sid)->sn_version
43e535db86SBram Moolenaar 						       == SCRIPT_VERSION_VIM9;
44e535db86SBram Moolenaar }
45e535db86SBram Moolenaar 
46e535db86SBram Moolenaar /*
478a7d6542SBram Moolenaar  * ":vim9script".
488a7d6542SBram Moolenaar  */
498a7d6542SBram Moolenaar     void
509b8d6226SBram Moolenaar ex_vim9script(exarg_T *eap UNUSED)
518a7d6542SBram Moolenaar {
529b8d6226SBram Moolenaar #ifdef FEAT_EVAL
532b32700dSBram Moolenaar     int		    sid = current_sctx.sc_sid;
54101f4810SBram Moolenaar     scriptitem_T    *si;
558a7d6542SBram Moolenaar 
568a7d6542SBram Moolenaar     if (!getline_equal(eap->getline, eap->cookie, getsourceline))
578a7d6542SBram Moolenaar     {
58451c2e35SBram Moolenaar 	emsg(_(e_vim9script_can_only_be_used_in_script));
598a7d6542SBram Moolenaar 	return;
608a7d6542SBram Moolenaar     }
612b32700dSBram Moolenaar 
622b32700dSBram Moolenaar     si = SCRIPT_ITEM(sid);
632b32700dSBram Moolenaar     if (si->sn_state == SN_STATE_HAD_COMMAND)
648a7d6542SBram Moolenaar     {
65451c2e35SBram Moolenaar 	emsg(_(e_vim9script_must_be_first_command_in_script));
668a7d6542SBram Moolenaar 	return;
678a7d6542SBram Moolenaar     }
682b32700dSBram Moolenaar     if (!IS_WHITE_OR_NUL(*eap->arg) && STRCMP(eap->arg, "noclear") != 0)
692b32700dSBram Moolenaar     {
702b32700dSBram Moolenaar 	semsg(_(e_invarg2), eap->arg);
712b32700dSBram Moolenaar 	return;
722b32700dSBram Moolenaar     }
732b32700dSBram Moolenaar     if (si->sn_state == SN_STATE_RELOAD && IS_WHITE_OR_NUL(*eap->arg))
742b32700dSBram Moolenaar     {
752b32700dSBram Moolenaar 	hashtab_T	*ht = &SCRIPT_VARS(sid);
762b32700dSBram Moolenaar 
772b32700dSBram Moolenaar 	// Reloading a script without the "noclear" argument: clear
782b32700dSBram Moolenaar 	// script-local variables and functions.
792b32700dSBram Moolenaar 	hashtab_free_contents(ht);
802b32700dSBram Moolenaar 	hash_init(ht);
812b32700dSBram Moolenaar 	delete_script_functions(sid);
822b32700dSBram Moolenaar 
832b32700dSBram Moolenaar 	// old imports and script variables are no longer valid
842b32700dSBram Moolenaar 	free_imports_and_script_vars(sid);
852b32700dSBram Moolenaar     }
862b32700dSBram Moolenaar     si->sn_state = SN_STATE_HAD_COMMAND;
872b32700dSBram Moolenaar 
888a7d6542SBram Moolenaar     current_sctx.sc_version = SCRIPT_VERSION_VIM9;
898a7d6542SBram Moolenaar     si->sn_version = SCRIPT_VERSION_VIM9;
908a7d6542SBram Moolenaar 
918a7d6542SBram Moolenaar     if (STRCMP(p_cpo, CPO_VIM) != 0)
928a7d6542SBram Moolenaar     {
9339ca4127SBram Moolenaar 	si->sn_save_cpo = vim_strsave(p_cpo);
9437294bd6SBram Moolenaar 	set_option_value((char_u *)"cpo", 0L, (char_u *)CPO_VIM, OPT_NO_REDRAW);
958a7d6542SBram Moolenaar     }
969b8d6226SBram Moolenaar #else
979b8d6226SBram Moolenaar     // No check for this being the first command, it doesn't matter.
989b8d6226SBram Moolenaar     current_sctx.sc_version = SCRIPT_VERSION_VIM9;
999b8d6226SBram Moolenaar #endif
100227a69deSBram Moolenaar }
1018a7d6542SBram Moolenaar 
1028a7d6542SBram Moolenaar /*
103ae616494SBram Moolenaar  * When in Vim9 script give an error and return FAIL.
104ae616494SBram Moolenaar  */
105ae616494SBram Moolenaar     int
106ae616494SBram Moolenaar not_in_vim9(exarg_T *eap)
107ae616494SBram Moolenaar {
10868e30449SBram Moolenaar     if (in_vim9script())
109ae616494SBram Moolenaar 	switch (eap->cmdidx)
110ae616494SBram Moolenaar 	{
111ada1d870SBram Moolenaar 	    case CMD_k:
112ada1d870SBram Moolenaar 		if (eap->addr_count > 0)
113ada1d870SBram Moolenaar 		{
114ada1d870SBram Moolenaar 		    emsg(_(e_norange));
115ada1d870SBram Moolenaar 		    return FAIL;
116ada1d870SBram Moolenaar 		}
117ada1d870SBram Moolenaar 		// FALLTHROUGH
118ae616494SBram Moolenaar 	    case CMD_append:
119ae616494SBram Moolenaar 	    case CMD_change:
120f5a48010SBram Moolenaar 	    case CMD_insert:
1216508880dSBram Moolenaar 	    case CMD_open:
122f5a48010SBram Moolenaar 	    case CMD_t:
123ae616494SBram Moolenaar 	    case CMD_xit:
1249b8d6226SBram Moolenaar 		semsg(_(e_command_not_supported_in_vim9_script_missing_var_str), eap->cmd);
125ae616494SBram Moolenaar 		return FAIL;
126ae616494SBram Moolenaar 	    default: break;
127ae616494SBram Moolenaar 	}
128ae616494SBram Moolenaar     return OK;
129ae616494SBram Moolenaar }
130ae616494SBram Moolenaar 
131dcc58e03SBram Moolenaar /*
1324b3e1964SBram Moolenaar  * Give an error message if "p" points at "#{" and return TRUE.
1334b3e1964SBram Moolenaar  * This avoids that using a legacy style #{} dictionary leads to difficult to
1344b3e1964SBram Moolenaar  * understand errors.
1354b3e1964SBram Moolenaar  */
1364b3e1964SBram Moolenaar     int
1374b3e1964SBram Moolenaar vim9_bad_comment(char_u *p)
1384b3e1964SBram Moolenaar {
139a0399efaSBram Moolenaar     if (p[0] == '#' && p[1] == '{' && p[2] != '{')
1404b3e1964SBram Moolenaar     {
1414b3e1964SBram Moolenaar 	emsg(_(e_cannot_use_hash_curly_to_start_comment));
1424b3e1964SBram Moolenaar 	return TRUE;
1434b3e1964SBram Moolenaar     }
1444b3e1964SBram Moolenaar     return FALSE;
1454b3e1964SBram Moolenaar }
1464b3e1964SBram Moolenaar 
1474b3e1964SBram Moolenaar /*
148a0399efaSBram Moolenaar  * Return TRUE if "p" points at a "#" not followed by one '{'.
1494b3e1964SBram Moolenaar  * Does not check for white space.
150dcc58e03SBram Moolenaar  */
151dcc58e03SBram Moolenaar     int
152dcc58e03SBram Moolenaar vim9_comment_start(char_u *p)
153dcc58e03SBram Moolenaar {
154a0399efaSBram Moolenaar     return p[0] == '#' && (p[1] != '{' || p[2] == '{');
155dcc58e03SBram Moolenaar }
156dcc58e03SBram Moolenaar 
1579b8d6226SBram Moolenaar #if defined(FEAT_EVAL) || defined(PROTO)
1589b8d6226SBram Moolenaar 
159ae616494SBram Moolenaar /*
1608a7d6542SBram Moolenaar  * ":export let Name: type"
1618a7d6542SBram Moolenaar  * ":export const Name: type"
1628a7d6542SBram Moolenaar  * ":export def Name(..."
1638a7d6542SBram Moolenaar  * ":export class Name ..."
1648a7d6542SBram Moolenaar  */
1658a7d6542SBram Moolenaar     void
16609689a02SBram Moolenaar ex_export(exarg_T *eap)
1678a7d6542SBram Moolenaar {
168eb6880b6SBram Moolenaar     if (!in_vim9script())
1698a7d6542SBram Moolenaar     {
170451c2e35SBram Moolenaar 	emsg(_(e_export_can_only_be_used_in_vim9script));
1718a7d6542SBram Moolenaar 	return;
1728a7d6542SBram Moolenaar     }
1738a7d6542SBram Moolenaar 
1748a7d6542SBram Moolenaar     eap->cmd = eap->arg;
1752e2d7589SBram Moolenaar     (void)find_ex_command(eap, NULL, lookup_scriptitem, NULL);
1768a7d6542SBram Moolenaar     switch (eap->cmdidx)
1778a7d6542SBram Moolenaar     {
1788a7d6542SBram Moolenaar 	case CMD_let:
17930fd8204SBram Moolenaar 	case CMD_var:
18030fd8204SBram Moolenaar 	case CMD_final:
1818a7d6542SBram Moolenaar 	case CMD_const:
1828a7d6542SBram Moolenaar 	case CMD_def:
1838a7d6542SBram Moolenaar 	// case CMD_class:
1848a7d6542SBram Moolenaar 	    is_export = TRUE;
1858a7d6542SBram Moolenaar 	    do_cmdline(eap->cmd, eap->getline, eap->cookie,
1868a7d6542SBram Moolenaar 						DOCMD_VERBOSE + DOCMD_NOWAIT);
1878a7d6542SBram Moolenaar 
1888a7d6542SBram Moolenaar 	    // The command will reset "is_export" when exporting an item.
1898a7d6542SBram Moolenaar 	    if (is_export)
1908a7d6542SBram Moolenaar 	    {
191451c2e35SBram Moolenaar 		emsg(_(e_export_with_invalid_argument));
1928a7d6542SBram Moolenaar 		is_export = FALSE;
1938a7d6542SBram Moolenaar 	    }
1948a7d6542SBram Moolenaar 	    break;
1958a7d6542SBram Moolenaar 	default:
196451c2e35SBram Moolenaar 	    emsg(_(e_invalid_command_after_export));
1978a7d6542SBram Moolenaar 	    break;
1988a7d6542SBram Moolenaar     }
1998a7d6542SBram Moolenaar }
2008a7d6542SBram Moolenaar 
2018a7d6542SBram Moolenaar /*
2028a7d6542SBram Moolenaar  * Add a new imported item entry to the current script.
2038a7d6542SBram Moolenaar  */
2048a7d6542SBram Moolenaar     static imported_T *
2058a7d6542SBram Moolenaar new_imported(garray_T *gap)
2068a7d6542SBram Moolenaar {
2078a7d6542SBram Moolenaar     if (ga_grow(gap, 1) == OK)
2088a7d6542SBram Moolenaar 	return ((imported_T *)gap->ga_data + gap->ga_len++);
2098a7d6542SBram Moolenaar     return NULL;
2108a7d6542SBram Moolenaar }
2118a7d6542SBram Moolenaar 
2128a7d6542SBram Moolenaar /*
2138a7d6542SBram Moolenaar  * Free all imported items in script "sid".
2148a7d6542SBram Moolenaar  */
2158a7d6542SBram Moolenaar     void
2168d739de4SBram Moolenaar free_imports_and_script_vars(int sid)
2178a7d6542SBram Moolenaar {
21821b9e977SBram Moolenaar     scriptitem_T    *si = SCRIPT_ITEM(sid);
2198a7d6542SBram Moolenaar     int		    idx;
2208a7d6542SBram Moolenaar 
2218a7d6542SBram Moolenaar     for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
2228a7d6542SBram Moolenaar     {
22320431c9dSBram Moolenaar 	imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx;
2248a7d6542SBram Moolenaar 
2258a7d6542SBram Moolenaar 	vim_free(imp->imp_name);
2268a7d6542SBram Moolenaar     }
2278a7d6542SBram Moolenaar     ga_clear(&si->sn_imports);
2288d739de4SBram Moolenaar 
2298d739de4SBram Moolenaar     free_all_script_vars(si);
2308d739de4SBram Moolenaar 
2316110e79aSBram Moolenaar     clear_type_list(&si->sn_type_list);
2328a7d6542SBram Moolenaar }
2338a7d6542SBram Moolenaar 
2348a7d6542SBram Moolenaar /*
235a6294955SBram Moolenaar  * Mark all imports as possible to redefine.  Used when a script is loaded
236a6294955SBram Moolenaar  * again but not cleared.
237a6294955SBram Moolenaar  */
238a6294955SBram Moolenaar     void
239a6294955SBram Moolenaar mark_imports_for_reload(int sid)
240a6294955SBram Moolenaar {
241a6294955SBram Moolenaar     scriptitem_T    *si = SCRIPT_ITEM(sid);
242a6294955SBram Moolenaar     int		    idx;
243a6294955SBram Moolenaar 
244a6294955SBram Moolenaar     for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
245a6294955SBram Moolenaar     {
246a6294955SBram Moolenaar 	imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx;
247a6294955SBram Moolenaar 
248a6294955SBram Moolenaar 	imp->imp_flags |= IMP_FLAGS_RELOAD;
249a6294955SBram Moolenaar     }
250a6294955SBram Moolenaar }
251a6294955SBram Moolenaar 
252a6294955SBram Moolenaar /*
2538a7d6542SBram Moolenaar  * ":import Item from 'filename'"
2548a7d6542SBram Moolenaar  * ":import Item as Alias from 'filename'"
2558a7d6542SBram Moolenaar  * ":import {Item} from 'filename'".
2568a7d6542SBram Moolenaar  * ":import {Item as Alias} from 'filename'"
2578a7d6542SBram Moolenaar  * ":import {Item, Item} from 'filename'"
2588a7d6542SBram Moolenaar  * ":import {Item, Item as Alias} from 'filename'"
2598a7d6542SBram Moolenaar  *
2608a7d6542SBram Moolenaar  * ":import * as Name from 'filename'"
2618a7d6542SBram Moolenaar  */
2628a7d6542SBram Moolenaar     void
2638a7d6542SBram Moolenaar ex_import(exarg_T *eap)
2648a7d6542SBram Moolenaar {
265101f4810SBram Moolenaar     char_u	*cmd_end;
2661c991144SBram Moolenaar     evalarg_T	evalarg;
2678a7d6542SBram Moolenaar 
268101f4810SBram Moolenaar     if (!getline_equal(eap->getline, eap->cookie, getsourceline))
269101f4810SBram Moolenaar     {
270451c2e35SBram Moolenaar 	emsg(_(e_import_can_only_be_used_in_script));
271101f4810SBram Moolenaar 	return;
272101f4810SBram Moolenaar     }
2731c991144SBram Moolenaar     fill_evalarg_from_eap(&evalarg, eap, eap->skip);
274101f4810SBram Moolenaar 
2751c991144SBram Moolenaar     cmd_end = handle_import(eap->arg, NULL, current_sctx.sc_sid,
2761c991144SBram Moolenaar 							       &evalarg, NULL);
2778a7d6542SBram Moolenaar     if (cmd_end != NULL)
2788a7d6542SBram Moolenaar 	eap->nextcmd = check_nextcmd(cmd_end);
2791c991144SBram Moolenaar     clear_evalarg(&evalarg, eap);
2808a7d6542SBram Moolenaar }
2818a7d6542SBram Moolenaar 
2828a7d6542SBram Moolenaar /*
283f2d5c240SBram Moolenaar  * Find an exported item in "sid" matching the name at "*argp".
284f2d5c240SBram Moolenaar  * When it is a variable return the index.
285f2d5c240SBram Moolenaar  * When it is a user function return "*ufunc".
286f2d5c240SBram Moolenaar  * When not found returns -1 and "*ufunc" is NULL.
287f2d5c240SBram Moolenaar  */
288f2d5c240SBram Moolenaar     int
289f2d5c240SBram Moolenaar find_exported(
290f2d5c240SBram Moolenaar 	int	    sid,
2911c991144SBram Moolenaar 	char_u	    *name,
292f2d5c240SBram Moolenaar 	ufunc_T	    **ufunc,
293fbbcd003SBram Moolenaar 	type_T	    **type,
294cb4e80faSBram Moolenaar 	cctx_T	    *cctx,
295cb4e80faSBram Moolenaar 	int	    verbose)
296f2d5c240SBram Moolenaar {
297f2d5c240SBram Moolenaar     int		idx = -1;
298f2d5c240SBram Moolenaar     svar_T	*sv;
299f2d5c240SBram Moolenaar     scriptitem_T *script = SCRIPT_ITEM(sid);
300f2d5c240SBram Moolenaar 
301*529fb5a5SBram Moolenaar     // Find name in "script".
30208251757SBram Moolenaar     idx = get_script_item_idx(sid, name, 0, cctx);
303f2d5c240SBram Moolenaar     if (idx >= 0)
304f2d5c240SBram Moolenaar     {
305f2d5c240SBram Moolenaar 	sv = ((svar_T *)script->sn_var_vals.ga_data) + idx;
306f2d5c240SBram Moolenaar 	if (!sv->sv_export)
307f2d5c240SBram Moolenaar 	{
308cb4e80faSBram Moolenaar 	    if (verbose)
309451c2e35SBram Moolenaar 		semsg(_(e_item_not_exported_in_script_str), name);
310f2d5c240SBram Moolenaar 	    return -1;
311f2d5c240SBram Moolenaar 	}
312f2d5c240SBram Moolenaar 	*type = sv->sv_type;
313f2d5c240SBram Moolenaar 	*ufunc = NULL;
314f2d5c240SBram Moolenaar     }
315f2d5c240SBram Moolenaar     else
316f2d5c240SBram Moolenaar     {
317f2d5c240SBram Moolenaar 	char_u	buffer[200];
318f2d5c240SBram Moolenaar 	char_u	*funcname;
319f2d5c240SBram Moolenaar 
320f2d5c240SBram Moolenaar 	// it could be a user function.
3215a67c37aSBram Moolenaar 	if (STRLEN(name) < sizeof(buffer) - 15)
322f2d5c240SBram Moolenaar 	    funcname = buffer;
323f2d5c240SBram Moolenaar 	else
324f2d5c240SBram Moolenaar 	{
3255a67c37aSBram Moolenaar 	    funcname = alloc(STRLEN(name) + 15);
326f2d5c240SBram Moolenaar 	    if (funcname == NULL)
327f2d5c240SBram Moolenaar 		return -1;
328f2d5c240SBram Moolenaar 	}
329f2d5c240SBram Moolenaar 	funcname[0] = K_SPECIAL;
330f2d5c240SBram Moolenaar 	funcname[1] = KS_EXTRA;
331f2d5c240SBram Moolenaar 	funcname[2] = (int)KE_SNR;
332f2d5c240SBram Moolenaar 	sprintf((char *)funcname + 3, "%ld_%s", (long)sid, name);
3334c17ad94SBram Moolenaar 	*ufunc = find_func(funcname, FALSE, NULL);
334f2d5c240SBram Moolenaar 	if (funcname != buffer)
335f2d5c240SBram Moolenaar 	    vim_free(funcname);
336f2d5c240SBram Moolenaar 
337f2d5c240SBram Moolenaar 	if (*ufunc == NULL)
338f2d5c240SBram Moolenaar 	{
339cb4e80faSBram Moolenaar 	    if (verbose)
340451c2e35SBram Moolenaar 		semsg(_(e_item_not_found_in_script_str), name);
341f2d5c240SBram Moolenaar 	    return -1;
342f2d5c240SBram Moolenaar 	}
343*529fb5a5SBram Moolenaar 	else if (((*ufunc)->uf_flags & FC_EXPORT) == 0)
344*529fb5a5SBram Moolenaar 	{
345*529fb5a5SBram Moolenaar 	    if (verbose)
346*529fb5a5SBram Moolenaar 		semsg(_(e_item_not_exported_in_script_str), name);
347*529fb5a5SBram Moolenaar 	    *ufunc = NULL;
348*529fb5a5SBram Moolenaar 	    return -1;
349*529fb5a5SBram Moolenaar 	}
350f2d5c240SBram Moolenaar     }
351f2d5c240SBram Moolenaar 
352f2d5c240SBram Moolenaar     return idx;
353f2d5c240SBram Moolenaar }
354f2d5c240SBram Moolenaar 
355f2d5c240SBram Moolenaar /*
3568a7d6542SBram Moolenaar  * Handle an ":import" command and add the resulting imported_T to "gap", when
3578a7d6542SBram Moolenaar  * not NULL, or script "import_sid" sn_imports.
358fbbcd003SBram Moolenaar  * "cctx" is NULL at the script level.
3598a7d6542SBram Moolenaar  * Returns a pointer to after the command or NULL in case of failure
3608a7d6542SBram Moolenaar  */
3618a7d6542SBram Moolenaar     char_u *
3621c991144SBram Moolenaar handle_import(
3631c991144SBram Moolenaar 	char_u	    *arg_start,
3641c991144SBram Moolenaar 	garray_T    *gap,
3651c991144SBram Moolenaar 	int	    import_sid,
3661c991144SBram Moolenaar 	evalarg_T   *evalarg,
3671c991144SBram Moolenaar 	void	    *cctx)
3688a7d6542SBram Moolenaar {
3698a7d6542SBram Moolenaar     char_u	*arg = arg_start;
3701c991144SBram Moolenaar     char_u	*cmd_end = NULL;
3718a7d6542SBram Moolenaar     int		ret = FAIL;
3728a7d6542SBram Moolenaar     typval_T	tv;
3738a7d6542SBram Moolenaar     int		sid = -1;
3748a7d6542SBram Moolenaar     int		res;
3750a84284eSBram Moolenaar     int		mult = FALSE;
3761c991144SBram Moolenaar     garray_T	names;
3770a84284eSBram Moolenaar     garray_T	as_names;
3788a7d6542SBram Moolenaar 
3791c991144SBram Moolenaar     ga_init2(&names, sizeof(char_u *), 10);
3800a84284eSBram Moolenaar     ga_init2(&as_names, sizeof(char_u *), 10);
3818a7d6542SBram Moolenaar     if (*arg == '{')
3828a7d6542SBram Moolenaar     {
3831c991144SBram Moolenaar 	// "import {item, item} from ..."
3840a84284eSBram Moolenaar 	mult = TRUE;
3851c991144SBram Moolenaar 	arg = skipwhite_and_linebreak(arg + 1, evalarg);
3860a84284eSBram Moolenaar     }
3870a84284eSBram Moolenaar 
3881c991144SBram Moolenaar     for (;;)
3891c991144SBram Moolenaar     {
3901c991144SBram Moolenaar 	char_u	    *p = arg;
3911c991144SBram Moolenaar 	int	    had_comma = FALSE;
3920a84284eSBram Moolenaar 	char_u	    *as_name = NULL;
3931c991144SBram Moolenaar 
3940a84284eSBram Moolenaar 	// accept "*" or "Name"
3950a84284eSBram Moolenaar 	if (!mult && arg[0] == '*' && IS_WHITE_OR_NUL(arg[1]))
3960a84284eSBram Moolenaar 	    ++arg;
3970a84284eSBram Moolenaar 	else
3981c991144SBram Moolenaar 	    while (eval_isnamec(*arg))
3998a7d6542SBram Moolenaar 		++arg;
4001c991144SBram Moolenaar 	if (p == arg)
4011c991144SBram Moolenaar 	    break;
4020a84284eSBram Moolenaar 	if (ga_grow(&names, 1) == FAIL || ga_grow(&as_names, 1) == FAIL)
4031c991144SBram Moolenaar 	    goto erret;
4040a84284eSBram Moolenaar 	((char_u **)names.ga_data)[names.ga_len] = vim_strnsave(p, arg - p);
4051c991144SBram Moolenaar 	++names.ga_len;
4061c991144SBram Moolenaar 
4071c991144SBram Moolenaar 	arg = skipwhite_and_linebreak(arg, evalarg);
4081c991144SBram Moolenaar 	if (STRNCMP("as", arg, 2) == 0 && IS_WHITE_OR_NUL(arg[2]))
4091c991144SBram Moolenaar 	{
4101c991144SBram Moolenaar 	    // skip over "as Name "; no line break allowed after "as"
4118a7d6542SBram Moolenaar 	    arg = skipwhite(arg + 2);
4121c991144SBram Moolenaar 	    p = arg;
413fa29c8abSBram Moolenaar 	    if (eval_isnamec1(*arg))
414fa29c8abSBram Moolenaar 		while (eval_isnamec(*arg))
4158a7d6542SBram Moolenaar 		    ++arg;
416057e84afSBram Moolenaar 	    if (check_defined(p, arg - p, cctx, FALSE) == FAIL)
4171c991144SBram Moolenaar 		goto erret;
4181c991144SBram Moolenaar 	    as_name = vim_strnsave(p, arg - p);
4191c991144SBram Moolenaar 	    arg = skipwhite_and_linebreak(arg, evalarg);
4208a7d6542SBram Moolenaar 	}
4218a7d6542SBram Moolenaar 	else if (*arg_start == '*')
4228a7d6542SBram Moolenaar 	{
423451c2e35SBram Moolenaar 	    emsg(_(e_missing_as_after_star));
4241c991144SBram Moolenaar 	    goto erret;
4258a7d6542SBram Moolenaar 	}
4260a84284eSBram Moolenaar 	// without "as Name" the as_names entry is NULL
4270a84284eSBram Moolenaar 	((char_u **)as_names.ga_data)[as_names.ga_len] = as_name;
4280a84284eSBram Moolenaar 	++as_names.ga_len;
4290a84284eSBram Moolenaar 
4300a84284eSBram Moolenaar 	if (!mult)
4310a84284eSBram Moolenaar 	    break;
4320a84284eSBram Moolenaar 	if (*arg == ',')
4330a84284eSBram Moolenaar 	{
4340a84284eSBram Moolenaar 	    had_comma = TRUE;
4350a84284eSBram Moolenaar 	    ++arg;
4360a84284eSBram Moolenaar 	}
4370a84284eSBram Moolenaar 	arg = skipwhite_and_linebreak(arg, evalarg);
4380a84284eSBram Moolenaar 	if (*arg == '}')
4390a84284eSBram Moolenaar 	{
4400a84284eSBram Moolenaar 	    ++arg;
4410a84284eSBram Moolenaar 	    break;
4420a84284eSBram Moolenaar 	}
4430a84284eSBram Moolenaar 	if (!had_comma)
4440a84284eSBram Moolenaar 	{
4450a84284eSBram Moolenaar 	    emsg(_(e_missing_comma_in_import));
4460a84284eSBram Moolenaar 	    goto erret;
4470a84284eSBram Moolenaar 	}
4480a84284eSBram Moolenaar     }
4490a84284eSBram Moolenaar     arg = skipwhite_and_linebreak(arg, evalarg);
4500a84284eSBram Moolenaar 
4510a84284eSBram Moolenaar     if (names.ga_len == 0)
4520a84284eSBram Moolenaar     {
4530a84284eSBram Moolenaar 	emsg(_(e_syntax_error_in_import));
4540a84284eSBram Moolenaar 	goto erret;
4558a7d6542SBram Moolenaar     }
4561c991144SBram Moolenaar 
4571c991144SBram Moolenaar     if (STRNCMP("from", arg, 4) != 0 || !IS_WHITE_OR_NUL(arg[4]))
4588a7d6542SBram Moolenaar     {
459451c2e35SBram Moolenaar 	emsg(_(e_missing_from));
4601c991144SBram Moolenaar 	goto erret;
4618a7d6542SBram Moolenaar     }
4621c991144SBram Moolenaar 
463962d7213SBram Moolenaar     arg = skipwhite_and_linebreak(arg + 4, evalarg);
4648a7d6542SBram Moolenaar     tv.v_type = VAR_UNKNOWN;
4658a7d6542SBram Moolenaar     // TODO: should we accept any expression?
4668a7d6542SBram Moolenaar     if (*arg == '\'')
4679a78e6dfSBram Moolenaar 	ret = eval_lit_string(&arg, &tv, TRUE);
4688a7d6542SBram Moolenaar     else if (*arg == '"')
4699a78e6dfSBram Moolenaar 	ret = eval_string(&arg, &tv, TRUE);
4708a7d6542SBram Moolenaar     if (ret == FAIL || tv.vval.v_string == NULL || *tv.vval.v_string == NUL)
4718a7d6542SBram Moolenaar     {
472451c2e35SBram Moolenaar 	emsg(_(e_invalid_string_after_from));
4731c991144SBram Moolenaar 	goto erret;
4748a7d6542SBram Moolenaar     }
4758a7d6542SBram Moolenaar     cmd_end = arg;
4768a7d6542SBram Moolenaar 
4771c991144SBram Moolenaar     /*
4781c991144SBram Moolenaar      * find script file
4791c991144SBram Moolenaar      */
4808a7d6542SBram Moolenaar     if (*tv.vval.v_string == '.')
4818a7d6542SBram Moolenaar     {
4828a7d6542SBram Moolenaar 	size_t		len;
48321b9e977SBram Moolenaar 	scriptitem_T	*si = SCRIPT_ITEM(current_sctx.sc_sid);
4848a7d6542SBram Moolenaar 	char_u		*tail = gettail(si->sn_name);
4858a7d6542SBram Moolenaar 	char_u		*from_name;
4868a7d6542SBram Moolenaar 
4878a7d6542SBram Moolenaar 	// Relative to current script: "./name.vim", "../../name.vim".
4888a7d6542SBram Moolenaar 	len = STRLEN(si->sn_name) - STRLEN(tail) + STRLEN(tv.vval.v_string) + 2;
4898a7d6542SBram Moolenaar 	from_name = alloc((int)len);
4908a7d6542SBram Moolenaar 	if (from_name == NULL)
4918a7d6542SBram Moolenaar 	{
4928a7d6542SBram Moolenaar 	    clear_tv(&tv);
4931c991144SBram Moolenaar 	    goto erret;
4948a7d6542SBram Moolenaar 	}
4958a7d6542SBram Moolenaar 	vim_strncpy(from_name, si->sn_name, tail - si->sn_name);
4968a7d6542SBram Moolenaar 	add_pathsep(from_name);
4978a7d6542SBram Moolenaar 	STRCAT(from_name, tv.vval.v_string);
4988a7d6542SBram Moolenaar 	simplify_filename(from_name);
4998a7d6542SBram Moolenaar 
5008a7d6542SBram Moolenaar 	res = do_source(from_name, FALSE, DOSO_NONE, &sid);
5018a7d6542SBram Moolenaar 	vim_free(from_name);
5028a7d6542SBram Moolenaar     }
5038a7d6542SBram Moolenaar     else if (mch_isFullName(tv.vval.v_string))
5048a7d6542SBram Moolenaar     {
5058a7d6542SBram Moolenaar 	// Absolute path: "/tmp/name.vim"
5068a7d6542SBram Moolenaar 	res = do_source(tv.vval.v_string, FALSE, DOSO_NONE, &sid);
5078a7d6542SBram Moolenaar     }
5088a7d6542SBram Moolenaar     else
5098a7d6542SBram Moolenaar     {
5108a7d6542SBram Moolenaar 	size_t	    len = 7 + STRLEN(tv.vval.v_string) + 1;
5118a7d6542SBram Moolenaar 	char_u	    *from_name;
5128a7d6542SBram Moolenaar 
5138a7d6542SBram Moolenaar 	// Find file in "import" subdirs in 'runtimepath'.
5148a7d6542SBram Moolenaar 	from_name = alloc((int)len);
5158a7d6542SBram Moolenaar 	if (from_name == NULL)
5168a7d6542SBram Moolenaar 	{
5178a7d6542SBram Moolenaar 	    clear_tv(&tv);
5181c991144SBram Moolenaar 	    goto erret;
5198a7d6542SBram Moolenaar 	}
5208a7d6542SBram Moolenaar 	vim_snprintf((char *)from_name, len, "import/%s", tv.vval.v_string);
5218a7d6542SBram Moolenaar 	res = source_in_path(p_rtp, from_name, DIP_NOAFTER, &sid);
5228a7d6542SBram Moolenaar 	vim_free(from_name);
5238a7d6542SBram Moolenaar     }
5248a7d6542SBram Moolenaar 
5258a7d6542SBram Moolenaar     if (res == FAIL || sid <= 0)
5268a7d6542SBram Moolenaar     {
527451c2e35SBram Moolenaar 	semsg(_(e_could_not_import_str), tv.vval.v_string);
5288a7d6542SBram Moolenaar 	clear_tv(&tv);
5291c991144SBram Moolenaar 	goto erret;
5308a7d6542SBram Moolenaar     }
5318a7d6542SBram Moolenaar     clear_tv(&tv);
5328a7d6542SBram Moolenaar 
5338a7d6542SBram Moolenaar     if (*arg_start == '*')
5348a7d6542SBram Moolenaar     {
535a6294955SBram Moolenaar 	imported_T  *imported;
5360a84284eSBram Moolenaar 	char_u	    *as_name = ((char_u **)as_names.ga_data)[0];
5378a7d6542SBram Moolenaar 
5380a84284eSBram Moolenaar 	// "import * as That"
539a6294955SBram Moolenaar 	imported = find_imported(as_name, STRLEN(as_name), cctx);
540a6294955SBram Moolenaar 	if (imported != NULL && imported->imp_sid == sid)
541a6294955SBram Moolenaar 	{
542a6294955SBram Moolenaar 	    if (imported->imp_flags & IMP_FLAGS_RELOAD)
543a6294955SBram Moolenaar 		// import already defined on a previous script load
544a6294955SBram Moolenaar 		imported->imp_flags &= ~IMP_FLAGS_RELOAD;
545a6294955SBram Moolenaar 	    else
546a6294955SBram Moolenaar 	    {
547a6294955SBram Moolenaar 		semsg(_(e_name_already_defined_str), as_name);
548a6294955SBram Moolenaar 		goto erret;
549a6294955SBram Moolenaar 	    }
550a6294955SBram Moolenaar 	}
551a6294955SBram Moolenaar 
552a6294955SBram Moolenaar 	imported = new_imported(gap != NULL ? gap
553a6294955SBram Moolenaar 					: &SCRIPT_ITEM(import_sid)->sn_imports);
5548a7d6542SBram Moolenaar 	if (imported == NULL)
5551c991144SBram Moolenaar 	    goto erret;
5561c991144SBram Moolenaar 	imported->imp_name = as_name;
5570a84284eSBram Moolenaar 	((char_u **)as_names.ga_data)[0] = NULL;
5588a7d6542SBram Moolenaar 	imported->imp_sid = sid;
559a6294955SBram Moolenaar 	imported->imp_flags = IMP_FLAGS_STAR;
5608a7d6542SBram Moolenaar     }
5618a7d6542SBram Moolenaar     else
5628a7d6542SBram Moolenaar     {
5631c991144SBram Moolenaar 	int i;
5641c991144SBram Moolenaar 
5658a7d6542SBram Moolenaar 	arg = arg_start;
5668a7d6542SBram Moolenaar 	if (*arg == '{')
5678a7d6542SBram Moolenaar 	    arg = skipwhite(arg + 1);
5681c991144SBram Moolenaar 	for (i = 0; i < names.ga_len; ++i)
5698a7d6542SBram Moolenaar 	{
5701c991144SBram Moolenaar 	    char_u	*name = ((char_u **)names.ga_data)[i];
5710a84284eSBram Moolenaar 	    char_u	*as_name = ((char_u **)as_names.ga_data)[i];
572a6294955SBram Moolenaar 	    size_t	len = STRLEN(name);
5738a7d6542SBram Moolenaar 	    int		idx;
5748a7d6542SBram Moolenaar 	    imported_T	*imported;
575f2d5c240SBram Moolenaar 	    ufunc_T	*ufunc = NULL;
576f2d5c240SBram Moolenaar 	    type_T	*type;
5778a7d6542SBram Moolenaar 
578cb4e80faSBram Moolenaar 	    idx = find_exported(sid, name, &ufunc, &type, cctx, TRUE);
5798a7d6542SBram Moolenaar 
580f2d5c240SBram Moolenaar 	    if (idx < 0 && ufunc == NULL)
5811c991144SBram Moolenaar 		goto erret;
5828a7d6542SBram Moolenaar 
583a6294955SBram Moolenaar 	    // If already imported with the same propertis and the
584a6294955SBram Moolenaar 	    // IMP_FLAGS_RELOAD set then we keep that entry.  Otherwise create
585a6294955SBram Moolenaar 	    // a new one (and give an error for an existing import).
586a6294955SBram Moolenaar 	    imported = find_imported(name, len, cctx);
587a6294955SBram Moolenaar 	    if (imported != NULL
588a6294955SBram Moolenaar 		    && (imported->imp_flags & IMP_FLAGS_RELOAD)
589a6294955SBram Moolenaar 		    && imported->imp_sid == sid
590a6294955SBram Moolenaar 		    && (idx >= 0
591a6294955SBram Moolenaar 			? (equal_type(imported->imp_type, type)
592a6294955SBram Moolenaar 			    && imported->imp_var_vals_idx == idx)
593a6294955SBram Moolenaar 			: (equal_type(imported->imp_type, ufunc->uf_func_type)
594a6294955SBram Moolenaar 			    && STRCMP(imported->imp_funcname,
595a6294955SBram Moolenaar 							ufunc->uf_name) == 0)))
596a6294955SBram Moolenaar 	    {
597a6294955SBram Moolenaar 		imported->imp_flags &= ~IMP_FLAGS_RELOAD;
598a6294955SBram Moolenaar 	    }
599a6294955SBram Moolenaar 	    else
600a6294955SBram Moolenaar 	    {
601057e84afSBram Moolenaar 		if (check_defined(name, len, cctx, FALSE) == FAIL)
6021c991144SBram Moolenaar 		    goto erret;
6035269bd2aSBram Moolenaar 
6048a7d6542SBram Moolenaar 		imported = new_imported(gap != NULL ? gap
60521b9e977SBram Moolenaar 				       : &SCRIPT_ITEM(import_sid)->sn_imports);
6068a7d6542SBram Moolenaar 		if (imported == NULL)
6071c991144SBram Moolenaar 		    goto erret;
6088a7d6542SBram Moolenaar 
6090a84284eSBram Moolenaar 		if (as_name == NULL)
6100a84284eSBram Moolenaar 		{
6111c991144SBram Moolenaar 		    imported->imp_name = name;
6121c991144SBram Moolenaar 		    ((char_u **)names.ga_data)[i] = NULL;
6130a84284eSBram Moolenaar 		}
6140a84284eSBram Moolenaar 		else
6150a84284eSBram Moolenaar 		{
6160a84284eSBram Moolenaar 		    // "import This as That ..."
6170a84284eSBram Moolenaar 		    imported->imp_name = as_name;
6180a84284eSBram Moolenaar 		    ((char_u **)as_names.ga_data)[i] = NULL;
6190a84284eSBram Moolenaar 		}
6208a7d6542SBram Moolenaar 		imported->imp_sid = sid;
6218a7d6542SBram Moolenaar 		if (idx >= 0)
6228a7d6542SBram Moolenaar 		{
623f2d5c240SBram Moolenaar 		    imported->imp_type = type;
6248a7d6542SBram Moolenaar 		    imported->imp_var_vals_idx = idx;
6258a7d6542SBram Moolenaar 		}
6268a7d6542SBram Moolenaar 		else
62740f4f7a4SBram Moolenaar 		{
62840f4f7a4SBram Moolenaar 		    imported->imp_type = ufunc->uf_func_type;
6298a7d6542SBram Moolenaar 		    imported->imp_funcname = ufunc->uf_name;
6308a7d6542SBram Moolenaar 		}
6318a7d6542SBram Moolenaar 	    }
63240f4f7a4SBram Moolenaar 	}
633a6294955SBram Moolenaar     }
6341c991144SBram Moolenaar erret:
6351c991144SBram Moolenaar     ga_clear_strings(&names);
6360a84284eSBram Moolenaar     ga_clear_strings(&as_names);
6378a7d6542SBram Moolenaar     return cmd_end;
6388a7d6542SBram Moolenaar }
6398a7d6542SBram Moolenaar 
640c82a5b5dSBram Moolenaar /*
641c82a5b5dSBram Moolenaar  * Declare a script-local variable without init: "let var: type".
642c82a5b5dSBram Moolenaar  * "const" is an error since the value is missing.
643c82a5b5dSBram Moolenaar  * Returns a pointer to after the type.
644c82a5b5dSBram Moolenaar  */
645c82a5b5dSBram Moolenaar     char_u *
646c82a5b5dSBram Moolenaar vim9_declare_scriptvar(exarg_T *eap, char_u *arg)
647c82a5b5dSBram Moolenaar {
648c82a5b5dSBram Moolenaar     char_u	    *p;
649c82a5b5dSBram Moolenaar     char_u	    *name;
650c82a5b5dSBram Moolenaar     scriptitem_T    *si = SCRIPT_ITEM(current_sctx.sc_sid);
651c82a5b5dSBram Moolenaar     type_T	    *type;
652c82a5b5dSBram Moolenaar     typval_T	    init_tv;
653c82a5b5dSBram Moolenaar 
65430fd8204SBram Moolenaar     if (eap->cmdidx == CMD_final || eap->cmdidx == CMD_const)
655c82a5b5dSBram Moolenaar     {
65630fd8204SBram Moolenaar 	if (eap->cmdidx == CMD_final)
65730fd8204SBram Moolenaar 	    emsg(_(e_final_requires_a_value));
65830fd8204SBram Moolenaar 	else
659bc4c5051SBram Moolenaar 	    emsg(_(e_const_requires_a_value));
660c82a5b5dSBram Moolenaar 	return arg + STRLEN(arg);
661c82a5b5dSBram Moolenaar     }
662c82a5b5dSBram Moolenaar 
663c82a5b5dSBram Moolenaar     // Check for valid starting character.
664c82a5b5dSBram Moolenaar     if (!eval_isnamec1(*arg))
665c82a5b5dSBram Moolenaar     {
666c82a5b5dSBram Moolenaar 	semsg(_(e_invarg2), arg);
667c82a5b5dSBram Moolenaar 	return arg + STRLEN(arg);
668c82a5b5dSBram Moolenaar     }
669c82a5b5dSBram Moolenaar 
670984dddbeSBram Moolenaar     for (p = arg + 1; *p != NUL && eval_isnamec(*p); MB_PTR_ADV(p))
6713b74b6b4SBram Moolenaar 	if (*p == ':' && (VIM_ISWHITE(p[1]) || p != arg + 1))
672984dddbeSBram Moolenaar 	    break;
673c82a5b5dSBram Moolenaar 
674c82a5b5dSBram Moolenaar     if (*p != ':')
675c82a5b5dSBram Moolenaar     {
676bc4c5051SBram Moolenaar 	emsg(_(e_type_or_initialization_required));
677c82a5b5dSBram Moolenaar 	return arg + STRLEN(arg);
678c82a5b5dSBram Moolenaar     }
679984dddbeSBram Moolenaar     if (!VIM_ISWHITE(p[1]))
680984dddbeSBram Moolenaar     {
681c3fc75dbSBram Moolenaar 	semsg(_(e_white_space_required_after_str_str), ":", p);
682984dddbeSBram Moolenaar 	return arg + STRLEN(arg);
683984dddbeSBram Moolenaar     }
684c82a5b5dSBram Moolenaar     name = vim_strnsave(arg, p - arg);
685c82a5b5dSBram Moolenaar 
686c82a5b5dSBram Moolenaar     // parse type
687c82a5b5dSBram Moolenaar     p = skipwhite(p + 1);
6889e68c325SBram Moolenaar     type = parse_type(&p, &si->sn_type_list, TRUE);
6899e68c325SBram Moolenaar     if (type == NULL)
690f3decc58SBram Moolenaar     {
691f3decc58SBram Moolenaar 	vim_free(name);
692c82a5b5dSBram Moolenaar 	return p;
693f3decc58SBram Moolenaar     }
694c82a5b5dSBram Moolenaar 
695c82a5b5dSBram Moolenaar     // Create the variable with 0/NULL value.
696c82a5b5dSBram Moolenaar     CLEAR_FIELD(init_tv);
697f0afd9e1SBram Moolenaar     if (type->tt_type == VAR_ANY)
698f0afd9e1SBram Moolenaar 	// A variable of type "any" is not possible, just use zero instead
699f0afd9e1SBram Moolenaar 	init_tv.v_type = VAR_NUMBER;
700f0afd9e1SBram Moolenaar     else
701c82a5b5dSBram Moolenaar 	init_tv.v_type = type->tt_type;
702f785aa13SBram Moolenaar     set_var_const(name, type, &init_tv, FALSE, 0, 0);
703c82a5b5dSBram Moolenaar 
704c82a5b5dSBram Moolenaar     vim_free(name);
705c82a5b5dSBram Moolenaar     return p;
706c82a5b5dSBram Moolenaar }
707c82a5b5dSBram Moolenaar 
70834db91f7SBram Moolenaar /*
70939ca4127SBram Moolenaar  * Vim9 part of adding a script variable: add it to sn_all_vars (lookup by name
71039ca4127SBram Moolenaar  * with a hashtable) and sn_var_vals (lookup by index).
71107a65d26SBram Moolenaar  * When "create" is TRUE this is a new variable, otherwise find and update an
71207a65d26SBram Moolenaar  * existing variable.
71308251757SBram Moolenaar  * "flags" can have ASSIGN_FINAL or ASSIGN_CONST.
714aa210a3aSBram Moolenaar  * When "*type" is NULL use "tv" for the type and update "*type".
7158d739de4SBram Moolenaar  */
7168d739de4SBram Moolenaar     void
71708251757SBram Moolenaar update_vim9_script_var(
71808251757SBram Moolenaar 	int	    create,
71908251757SBram Moolenaar 	dictitem_T  *di,
72008251757SBram Moolenaar 	int	    flags,
72108251757SBram Moolenaar 	typval_T    *tv,
72208251757SBram Moolenaar 	type_T	    **type)
7238d739de4SBram Moolenaar {
7248d739de4SBram Moolenaar     scriptitem_T    *si = SCRIPT_ITEM(current_sctx.sc_sid);
7258d739de4SBram Moolenaar     hashitem_T	    *hi;
72607a65d26SBram Moolenaar     svar_T	    *sv;
7278d739de4SBram Moolenaar 
72807a65d26SBram Moolenaar     if (create)
72907a65d26SBram Moolenaar     {
73007a65d26SBram Moolenaar 	sallvar_T	    *newsav;
73107a65d26SBram Moolenaar 
73207a65d26SBram Moolenaar 	// Store a pointer to the typval_T, so that it can be found by index
73307a65d26SBram Moolenaar 	// instead of using a hastab lookup.
73407a65d26SBram Moolenaar 	if (ga_grow(&si->sn_var_vals, 1) == FAIL)
73507a65d26SBram Moolenaar 	    return;
73607a65d26SBram Moolenaar 
73707a65d26SBram Moolenaar 	sv = ((svar_T *)si->sn_var_vals.ga_data) + si->sn_var_vals.ga_len;
73807a65d26SBram Moolenaar 	newsav = (sallvar_T *)alloc_clear(
73907a65d26SBram Moolenaar 				       sizeof(sallvar_T) + STRLEN(di->di_key));
7408d739de4SBram Moolenaar 	if (newsav == NULL)
7418d739de4SBram Moolenaar 	    return;
7428d739de4SBram Moolenaar 
7438d739de4SBram Moolenaar 	sv->sv_tv = &di->di_tv;
74408251757SBram Moolenaar 	sv->sv_const = (flags & ASSIGN_FINAL) ? ASSIGN_FINAL
74508251757SBram Moolenaar 				   : (flags & ASSIGN_CONST) ? ASSIGN_CONST : 0;
7468d739de4SBram Moolenaar 	sv->sv_export = is_export;
7478d739de4SBram Moolenaar 	newsav->sav_var_vals_idx = si->sn_var_vals.ga_len;
7488d739de4SBram Moolenaar 	++si->sn_var_vals.ga_len;
7498d739de4SBram Moolenaar 	STRCPY(&newsav->sav_key, di->di_key);
7508d739de4SBram Moolenaar 	sv->sv_name = newsav->sav_key;
7518d739de4SBram Moolenaar 	newsav->sav_di = di;
7528d739de4SBram Moolenaar 	newsav->sav_block_id = si->sn_current_block_id;
7538d739de4SBram Moolenaar 
7548d739de4SBram Moolenaar 	hi = hash_find(&si->sn_all_vars.dv_hashtab, newsav->sav_key);
7558d739de4SBram Moolenaar 	if (!HASHITEM_EMPTY(hi))
7568d739de4SBram Moolenaar 	{
7578d739de4SBram Moolenaar 	    sallvar_T *sav = HI2SAV(hi);
7588d739de4SBram Moolenaar 
7598d739de4SBram Moolenaar 	    // variable with this name exists in another block
7608d739de4SBram Moolenaar 	    while (sav->sav_next != NULL)
7618d739de4SBram Moolenaar 		sav = sav->sav_next;
7628d739de4SBram Moolenaar 	    sav->sav_next = newsav;
7638d739de4SBram Moolenaar 	}
7648d739de4SBram Moolenaar 	else
7658d739de4SBram Moolenaar 	    // new variable name
7668d739de4SBram Moolenaar 	    hash_add(&si->sn_all_vars.dv_hashtab, newsav->sav_key);
76707a65d26SBram Moolenaar     }
76807a65d26SBram Moolenaar     else
76907a65d26SBram Moolenaar     {
77007a65d26SBram Moolenaar 	sv = find_typval_in_script(&di->di_tv);
77107a65d26SBram Moolenaar     }
77207a65d26SBram Moolenaar     if (sv != NULL)
77307a65d26SBram Moolenaar     {
774aa210a3aSBram Moolenaar 	if (*type == NULL)
775108cf015SBram Moolenaar 	    *type = typval2type(tv, get_copyID(), &si->sn_type_list);
776aa210a3aSBram Moolenaar 	sv->sv_type = *type;
77707a65d26SBram Moolenaar     }
7788d739de4SBram Moolenaar 
7798d739de4SBram Moolenaar     // let ex_export() know the export worked.
7808d739de4SBram Moolenaar     is_export = FALSE;
7818d739de4SBram Moolenaar }
7828d739de4SBram Moolenaar 
783fbbcd003SBram Moolenaar /*
784fbbcd003SBram Moolenaar  * Hide a script variable when leaving a block.
785fbbcd003SBram Moolenaar  * "idx" is de index in sn_var_vals.
78639ca4127SBram Moolenaar  * When "func_defined" is non-zero then a function was defined in this block,
78739ca4127SBram Moolenaar  * the variable may be accessed by it.  Otherwise the variable can be cleared.
788fbbcd003SBram Moolenaar  */
7898d739de4SBram Moolenaar     void
79039ca4127SBram Moolenaar hide_script_var(scriptitem_T *si, int idx, int func_defined)
7918d739de4SBram Moolenaar {
792fbbcd003SBram Moolenaar     svar_T	*sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
7938d739de4SBram Moolenaar     hashtab_T	*script_ht = get_script_local_ht();
7948d739de4SBram Moolenaar     hashtab_T	*all_ht = &si->sn_all_vars.dv_hashtab;
7958d739de4SBram Moolenaar     hashitem_T	*script_hi;
7968d739de4SBram Moolenaar     hashitem_T	*all_hi;
7978d739de4SBram Moolenaar 
7988d739de4SBram Moolenaar     // Remove a variable declared inside the block, if it still exists.
79939ca4127SBram Moolenaar     // If it was added in a nested block it will already have been removed.
8008d739de4SBram Moolenaar     // The typval is moved into the sallvar_T.
8018d739de4SBram Moolenaar     script_hi = hash_find(script_ht, sv->sv_name);
8028d739de4SBram Moolenaar     all_hi = hash_find(all_ht, sv->sv_name);
8038d739de4SBram Moolenaar     if (!HASHITEM_EMPTY(script_hi) && !HASHITEM_EMPTY(all_hi))
8048d739de4SBram Moolenaar     {
8058d739de4SBram Moolenaar 	dictitem_T	*di = HI2DI(script_hi);
8068d739de4SBram Moolenaar 	sallvar_T	*sav = HI2SAV(all_hi);
80739ca4127SBram Moolenaar 	sallvar_T	*sav_prev = NULL;
8088d739de4SBram Moolenaar 
809fbbcd003SBram Moolenaar 	// There can be multiple entries with the same name in different
810fbbcd003SBram Moolenaar 	// blocks, find the right one.
811fbbcd003SBram Moolenaar 	while (sav != NULL && sav->sav_var_vals_idx != idx)
81239ca4127SBram Moolenaar 	{
81339ca4127SBram Moolenaar 	    sav_prev = sav;
814fbbcd003SBram Moolenaar 	    sav = sav->sav_next;
81539ca4127SBram Moolenaar 	}
816fbbcd003SBram Moolenaar 	if (sav != NULL)
817fbbcd003SBram Moolenaar 	{
81839ca4127SBram Moolenaar 	    if (func_defined)
81939ca4127SBram Moolenaar 	    {
82039ca4127SBram Moolenaar 		// move the typval from the dictitem to the sallvar
8218d739de4SBram Moolenaar 		sav->sav_tv = di->di_tv;
8228d739de4SBram Moolenaar 		di->di_tv.v_type = VAR_UNKNOWN;
8238d739de4SBram Moolenaar 		sav->sav_flags = di->di_flags;
8248d739de4SBram Moolenaar 		sav->sav_di = NULL;
825fbbcd003SBram Moolenaar 		sv->sv_tv = &sav->sav_tv;
826fbbcd003SBram Moolenaar 	    }
82739ca4127SBram Moolenaar 	    else
82839ca4127SBram Moolenaar 	    {
82939ca4127SBram Moolenaar 		if (sav_prev == NULL)
83039ca4127SBram Moolenaar 		    hash_remove(all_ht, all_hi);
83139ca4127SBram Moolenaar 		else
83239ca4127SBram Moolenaar 		    sav_prev->sav_next = sav->sav_next;
83339ca4127SBram Moolenaar 		sv->sv_name = NULL;
83439ca4127SBram Moolenaar 		vim_free(sav);
83539ca4127SBram Moolenaar 	    }
83639ca4127SBram Moolenaar 	    delete_var(script_ht, script_hi);
83739ca4127SBram Moolenaar 	}
8388d739de4SBram Moolenaar     }
8398d739de4SBram Moolenaar }
8408d739de4SBram Moolenaar 
8418d739de4SBram Moolenaar /*
8428d739de4SBram Moolenaar  * Free the script variables from "sn_all_vars".
8438d739de4SBram Moolenaar  */
8448d739de4SBram Moolenaar     void
8458d739de4SBram Moolenaar free_all_script_vars(scriptitem_T *si)
8468d739de4SBram Moolenaar {
8478d739de4SBram Moolenaar     int		todo;
8488d739de4SBram Moolenaar     hashtab_T	*ht = &si->sn_all_vars.dv_hashtab;
8498d739de4SBram Moolenaar     hashitem_T	*hi;
8508d739de4SBram Moolenaar     sallvar_T	*sav;
8518d739de4SBram Moolenaar     sallvar_T	*sav_next;
8528d739de4SBram Moolenaar 
8538d739de4SBram Moolenaar     hash_lock(ht);
8548d739de4SBram Moolenaar     todo = (int)ht->ht_used;
8558d739de4SBram Moolenaar     for (hi = ht->ht_array; todo > 0; ++hi)
8568d739de4SBram Moolenaar     {
8578d739de4SBram Moolenaar 	if (!HASHITEM_EMPTY(hi))
8588d739de4SBram Moolenaar 	{
8598d739de4SBram Moolenaar 	    --todo;
8608d739de4SBram Moolenaar 
8618d739de4SBram Moolenaar 	    // Free the variable.  Don't remove it from the hashtab, ht_array
8628d739de4SBram Moolenaar 	    // might change then.  hash_clear() takes care of it later.
8638d739de4SBram Moolenaar 	    sav = HI2SAV(hi);
8648d739de4SBram Moolenaar 	    while (sav != NULL)
8658d739de4SBram Moolenaar 	    {
8668d739de4SBram Moolenaar 		sav_next = sav->sav_next;
8678d739de4SBram Moolenaar 		if (sav->sav_di == NULL)
8688d739de4SBram Moolenaar 		    clear_tv(&sav->sav_tv);
8698d739de4SBram Moolenaar 		vim_free(sav);
8708d739de4SBram Moolenaar 		sav = sav_next;
8718d739de4SBram Moolenaar 	    }
8728d739de4SBram Moolenaar 	}
8738d739de4SBram Moolenaar     }
8748d739de4SBram Moolenaar     hash_clear(ht);
8758d739de4SBram Moolenaar     hash_init(ht);
8768d739de4SBram Moolenaar 
8778d739de4SBram Moolenaar     ga_clear(&si->sn_var_vals);
8782b32700dSBram Moolenaar 
8792b32700dSBram Moolenaar     // existing commands using script variable indexes are no longer valid
8802b32700dSBram Moolenaar     si->sn_script_seq = current_sctx.sc_seq;
8818d739de4SBram Moolenaar }
8828d739de4SBram Moolenaar 
8838d739de4SBram Moolenaar /*
88410c65860SBram Moolenaar  * Find the script-local variable that links to "dest".
88510c65860SBram Moolenaar  * Returns NULL if not found.
88634db91f7SBram Moolenaar  */
88710c65860SBram Moolenaar     svar_T *
88810c65860SBram Moolenaar find_typval_in_script(typval_T *dest)
88934db91f7SBram Moolenaar {
89034db91f7SBram Moolenaar     scriptitem_T    *si = SCRIPT_ITEM(current_sctx.sc_sid);
89134db91f7SBram Moolenaar     int		    idx;
89234db91f7SBram Moolenaar 
89381e17fbeSBram Moolenaar     if (si->sn_version != SCRIPT_VERSION_VIM9)
89481e17fbeSBram Moolenaar 	// legacy script doesn't store variable types
89510c65860SBram Moolenaar 	return NULL;
89681e17fbeSBram Moolenaar 
89734db91f7SBram Moolenaar     // Find the svar_T in sn_var_vals.
89834db91f7SBram Moolenaar     for (idx = 0; idx < si->sn_var_vals.ga_len; ++idx)
89934db91f7SBram Moolenaar     {
90034db91f7SBram Moolenaar 	svar_T    *sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
90134db91f7SBram Moolenaar 
902c0913d02SBram Moolenaar 	// If "sv_name" is NULL the variable was hidden when leaving a block,
903c0913d02SBram Moolenaar 	// don't check "sv_tv" then, it might be used for another variable now.
904c0913d02SBram Moolenaar 	if (sv->sv_name != NULL && sv->sv_tv == dest)
90510c65860SBram Moolenaar 	    return sv;
90610c65860SBram Moolenaar     }
907f785aa13SBram Moolenaar     iemsg("find_typval_in_script(): not found");
90810c65860SBram Moolenaar     return NULL;
90910c65860SBram Moolenaar }
91010c65860SBram Moolenaar 
91110c65860SBram Moolenaar /*
91210c65860SBram Moolenaar  * Check if the type of script variable "dest" allows assigning "value".
91310c65860SBram Moolenaar  * If needed convert "value" to a bool.
91410c65860SBram Moolenaar  */
91510c65860SBram Moolenaar     int
916f785aa13SBram Moolenaar check_script_var_type(
917f785aa13SBram Moolenaar 	typval_T    *dest,
918f785aa13SBram Moolenaar 	typval_T    *value,
919f785aa13SBram Moolenaar 	char_u	    *name,
920f785aa13SBram Moolenaar 	where_T	    where)
92134db91f7SBram Moolenaar {
92210c65860SBram Moolenaar     svar_T  *sv = find_typval_in_script(dest);
923c1ec0422SBram Moolenaar     int	    ret;
924c1ec0422SBram Moolenaar 
92510c65860SBram Moolenaar     if (sv != NULL)
92610c65860SBram Moolenaar     {
92708251757SBram Moolenaar 	if (sv->sv_const != 0)
928c785b9a7SBram Moolenaar 	{
92934db91f7SBram Moolenaar 	    semsg(_(e_readonlyvar), name);
930c785b9a7SBram Moolenaar 	    return FAIL;
931c785b9a7SBram Moolenaar 	}
932f785aa13SBram Moolenaar 	ret = check_typval_type(sv->sv_type, value, where);
933c1ec0422SBram Moolenaar 	if (ret == OK && need_convert_to_bool(sv->sv_type, value))
934c1ec0422SBram Moolenaar 	{
935c1ec0422SBram Moolenaar 	    int	val = tv2bool(value);
936c1ec0422SBram Moolenaar 
937c1ec0422SBram Moolenaar 	    clear_tv(value);
938c1ec0422SBram Moolenaar 	    value->v_type = VAR_BOOL;
939c1ec0422SBram Moolenaar 	    value->v_lock = 0;
940c1ec0422SBram Moolenaar 	    value->vval.v_number = val ? VVAL_TRUE : VVAL_FALSE;
941c1ec0422SBram Moolenaar 	}
942c1ec0422SBram Moolenaar 	return ret;
94334db91f7SBram Moolenaar     }
94410c65860SBram Moolenaar 
945c785b9a7SBram Moolenaar     return OK; // not really
94634db91f7SBram Moolenaar }
947c82a5b5dSBram Moolenaar 
9488a7d6542SBram Moolenaar #endif // FEAT_EVAL
949