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