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 208a7d6542SBram Moolenaar int 218a7d6542SBram Moolenaar in_vim9script(void) 228a7d6542SBram Moolenaar { 23eb6880b6SBram Moolenaar // Do not go up the stack, a ":function" inside vim9script uses legacy 24eb6880b6SBram Moolenaar // syntax. "sc_version" is also set when compiling a ":def" function in 25eb6880b6SBram Moolenaar // legacy script. 26*9979fcd7SBram Moolenaar return current_sctx.sc_version == SCRIPT_VERSION_VIM9 27*9979fcd7SBram Moolenaar || (cmdmod.cmod_flags & CMOD_VIM9CMD); 288a7d6542SBram Moolenaar } 298a7d6542SBram Moolenaar 308a7d6542SBram Moolenaar /* 318a7d6542SBram Moolenaar * ":vim9script". 328a7d6542SBram Moolenaar */ 338a7d6542SBram Moolenaar void 349b8d6226SBram Moolenaar ex_vim9script(exarg_T *eap UNUSED) 358a7d6542SBram Moolenaar { 369b8d6226SBram Moolenaar #ifdef FEAT_EVAL 372b32700dSBram Moolenaar int sid = current_sctx.sc_sid; 38101f4810SBram Moolenaar scriptitem_T *si; 398a7d6542SBram Moolenaar 408a7d6542SBram Moolenaar if (!getline_equal(eap->getline, eap->cookie, getsourceline)) 418a7d6542SBram Moolenaar { 42451c2e35SBram Moolenaar emsg(_(e_vim9script_can_only_be_used_in_script)); 438a7d6542SBram Moolenaar return; 448a7d6542SBram Moolenaar } 452b32700dSBram Moolenaar 462b32700dSBram Moolenaar si = SCRIPT_ITEM(sid); 472b32700dSBram Moolenaar if (si->sn_state == SN_STATE_HAD_COMMAND) 488a7d6542SBram Moolenaar { 49451c2e35SBram Moolenaar emsg(_(e_vim9script_must_be_first_command_in_script)); 508a7d6542SBram Moolenaar return; 518a7d6542SBram Moolenaar } 522b32700dSBram Moolenaar if (!IS_WHITE_OR_NUL(*eap->arg) && STRCMP(eap->arg, "noclear") != 0) 532b32700dSBram Moolenaar { 542b32700dSBram Moolenaar semsg(_(e_invarg2), eap->arg); 552b32700dSBram Moolenaar return; 562b32700dSBram Moolenaar } 572b32700dSBram Moolenaar if (si->sn_state == SN_STATE_RELOAD && IS_WHITE_OR_NUL(*eap->arg)) 582b32700dSBram Moolenaar { 592b32700dSBram Moolenaar hashtab_T *ht = &SCRIPT_VARS(sid); 602b32700dSBram Moolenaar 612b32700dSBram Moolenaar // Reloading a script without the "noclear" argument: clear 622b32700dSBram Moolenaar // script-local variables and functions. 632b32700dSBram Moolenaar hashtab_free_contents(ht); 642b32700dSBram Moolenaar hash_init(ht); 652b32700dSBram Moolenaar delete_script_functions(sid); 662b32700dSBram Moolenaar 672b32700dSBram Moolenaar // old imports and script variables are no longer valid 682b32700dSBram Moolenaar free_imports_and_script_vars(sid); 692b32700dSBram Moolenaar } 702b32700dSBram Moolenaar si->sn_state = SN_STATE_HAD_COMMAND; 712b32700dSBram Moolenaar 728a7d6542SBram Moolenaar current_sctx.sc_version = SCRIPT_VERSION_VIM9; 738a7d6542SBram Moolenaar si->sn_version = SCRIPT_VERSION_VIM9; 748a7d6542SBram Moolenaar 758a7d6542SBram Moolenaar if (STRCMP(p_cpo, CPO_VIM) != 0) 768a7d6542SBram Moolenaar { 7739ca4127SBram Moolenaar si->sn_save_cpo = vim_strsave(p_cpo); 7839ca4127SBram Moolenaar set_option_value((char_u *)"cpo", 0L, (char_u *)CPO_VIM, 0); 798a7d6542SBram Moolenaar } 809b8d6226SBram Moolenaar #else 819b8d6226SBram Moolenaar // No check for this being the first command, it doesn't matter. 829b8d6226SBram Moolenaar current_sctx.sc_version = SCRIPT_VERSION_VIM9; 839b8d6226SBram Moolenaar #endif 84227a69deSBram Moolenaar } 858a7d6542SBram Moolenaar 868a7d6542SBram Moolenaar /* 87ae616494SBram Moolenaar * When in Vim9 script give an error and return FAIL. 88ae616494SBram Moolenaar */ 89ae616494SBram Moolenaar int 90ae616494SBram Moolenaar not_in_vim9(exarg_T *eap) 91ae616494SBram Moolenaar { 9268e30449SBram Moolenaar if (in_vim9script()) 93ae616494SBram Moolenaar switch (eap->cmdidx) 94ae616494SBram Moolenaar { 95ae616494SBram Moolenaar case CMD_append: 96ae616494SBram Moolenaar case CMD_change: 97f5a48010SBram Moolenaar case CMD_insert: 98f5a48010SBram Moolenaar case CMD_t: 99ae616494SBram Moolenaar case CMD_xit: 1009b8d6226SBram Moolenaar semsg(_(e_command_not_supported_in_vim9_script_missing_var_str), eap->cmd); 101ae616494SBram Moolenaar return FAIL; 102ae616494SBram Moolenaar default: break; 103ae616494SBram Moolenaar } 104ae616494SBram Moolenaar return OK; 105ae616494SBram Moolenaar } 106ae616494SBram Moolenaar 107dcc58e03SBram Moolenaar /* 108dcc58e03SBram Moolenaar * Return TRUE if "p" points at a "#". Does not check for white space. 109dcc58e03SBram Moolenaar */ 110dcc58e03SBram Moolenaar int 111dcc58e03SBram Moolenaar vim9_comment_start(char_u *p) 112dcc58e03SBram Moolenaar { 113dcc58e03SBram Moolenaar return p[0] == '#'; 114dcc58e03SBram Moolenaar } 115dcc58e03SBram Moolenaar 1169b8d6226SBram Moolenaar #if defined(FEAT_EVAL) || defined(PROTO) 1179b8d6226SBram Moolenaar 118ae616494SBram Moolenaar /* 1198a7d6542SBram Moolenaar * ":export let Name: type" 1208a7d6542SBram Moolenaar * ":export const Name: type" 1218a7d6542SBram Moolenaar * ":export def Name(..." 1228a7d6542SBram Moolenaar * ":export class Name ..." 1238a7d6542SBram Moolenaar */ 1248a7d6542SBram Moolenaar void 12509689a02SBram Moolenaar ex_export(exarg_T *eap) 1268a7d6542SBram Moolenaar { 127eb6880b6SBram Moolenaar if (!in_vim9script()) 1288a7d6542SBram Moolenaar { 129451c2e35SBram Moolenaar emsg(_(e_export_can_only_be_used_in_vim9script)); 1308a7d6542SBram Moolenaar return; 1318a7d6542SBram Moolenaar } 1328a7d6542SBram Moolenaar 1338a7d6542SBram Moolenaar eap->cmd = eap->arg; 1348a7d6542SBram Moolenaar (void)find_ex_command(eap, NULL, lookup_scriptvar, NULL); 1358a7d6542SBram Moolenaar switch (eap->cmdidx) 1368a7d6542SBram Moolenaar { 1378a7d6542SBram Moolenaar case CMD_let: 13830fd8204SBram Moolenaar case CMD_var: 13930fd8204SBram Moolenaar case CMD_final: 1408a7d6542SBram Moolenaar case CMD_const: 1418a7d6542SBram Moolenaar case CMD_def: 1428a7d6542SBram Moolenaar // case CMD_class: 1438a7d6542SBram Moolenaar is_export = TRUE; 1448a7d6542SBram Moolenaar do_cmdline(eap->cmd, eap->getline, eap->cookie, 1458a7d6542SBram Moolenaar DOCMD_VERBOSE + DOCMD_NOWAIT); 1468a7d6542SBram Moolenaar 1478a7d6542SBram Moolenaar // The command will reset "is_export" when exporting an item. 1488a7d6542SBram Moolenaar if (is_export) 1498a7d6542SBram Moolenaar { 150451c2e35SBram Moolenaar emsg(_(e_export_with_invalid_argument)); 1518a7d6542SBram Moolenaar is_export = FALSE; 1528a7d6542SBram Moolenaar } 1538a7d6542SBram Moolenaar break; 1548a7d6542SBram Moolenaar default: 155451c2e35SBram Moolenaar emsg(_(e_invalid_command_after_export)); 1568a7d6542SBram Moolenaar break; 1578a7d6542SBram Moolenaar } 1588a7d6542SBram Moolenaar } 1598a7d6542SBram Moolenaar 1608a7d6542SBram Moolenaar /* 1618a7d6542SBram Moolenaar * Add a new imported item entry to the current script. 1628a7d6542SBram Moolenaar */ 1638a7d6542SBram Moolenaar static imported_T * 1648a7d6542SBram Moolenaar new_imported(garray_T *gap) 1658a7d6542SBram Moolenaar { 1668a7d6542SBram Moolenaar if (ga_grow(gap, 1) == OK) 1678a7d6542SBram Moolenaar return ((imported_T *)gap->ga_data + gap->ga_len++); 1688a7d6542SBram Moolenaar return NULL; 1698a7d6542SBram Moolenaar } 1708a7d6542SBram Moolenaar 1718a7d6542SBram Moolenaar /* 1728a7d6542SBram Moolenaar * Free all imported items in script "sid". 1738a7d6542SBram Moolenaar */ 1748a7d6542SBram Moolenaar void 1758d739de4SBram Moolenaar free_imports_and_script_vars(int sid) 1768a7d6542SBram Moolenaar { 17721b9e977SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM(sid); 1788a7d6542SBram Moolenaar int idx; 1798a7d6542SBram Moolenaar 1808a7d6542SBram Moolenaar for (idx = 0; idx < si->sn_imports.ga_len; ++idx) 1818a7d6542SBram Moolenaar { 18220431c9dSBram Moolenaar imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx; 1838a7d6542SBram Moolenaar 1848a7d6542SBram Moolenaar vim_free(imp->imp_name); 1858a7d6542SBram Moolenaar } 1868a7d6542SBram Moolenaar ga_clear(&si->sn_imports); 1878d739de4SBram Moolenaar 1888d739de4SBram Moolenaar free_all_script_vars(si); 1898d739de4SBram Moolenaar 1906110e79aSBram Moolenaar clear_type_list(&si->sn_type_list); 1918a7d6542SBram Moolenaar } 1928a7d6542SBram Moolenaar 1938a7d6542SBram Moolenaar /* 194a6294955SBram Moolenaar * Mark all imports as possible to redefine. Used when a script is loaded 195a6294955SBram Moolenaar * again but not cleared. 196a6294955SBram Moolenaar */ 197a6294955SBram Moolenaar void 198a6294955SBram Moolenaar mark_imports_for_reload(int sid) 199a6294955SBram Moolenaar { 200a6294955SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM(sid); 201a6294955SBram Moolenaar int idx; 202a6294955SBram Moolenaar 203a6294955SBram Moolenaar for (idx = 0; idx < si->sn_imports.ga_len; ++idx) 204a6294955SBram Moolenaar { 205a6294955SBram Moolenaar imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx; 206a6294955SBram Moolenaar 207a6294955SBram Moolenaar imp->imp_flags |= IMP_FLAGS_RELOAD; 208a6294955SBram Moolenaar } 209a6294955SBram Moolenaar } 210a6294955SBram Moolenaar 211a6294955SBram Moolenaar /* 2128a7d6542SBram Moolenaar * ":import Item from 'filename'" 2138a7d6542SBram Moolenaar * ":import Item as Alias from 'filename'" 2148a7d6542SBram Moolenaar * ":import {Item} from 'filename'". 2158a7d6542SBram Moolenaar * ":import {Item as Alias} from 'filename'" 2168a7d6542SBram Moolenaar * ":import {Item, Item} from 'filename'" 2178a7d6542SBram Moolenaar * ":import {Item, Item as Alias} from 'filename'" 2188a7d6542SBram Moolenaar * 2198a7d6542SBram Moolenaar * ":import * as Name from 'filename'" 2208a7d6542SBram Moolenaar */ 2218a7d6542SBram Moolenaar void 2228a7d6542SBram Moolenaar ex_import(exarg_T *eap) 2238a7d6542SBram Moolenaar { 224101f4810SBram Moolenaar char_u *cmd_end; 2251c991144SBram Moolenaar evalarg_T evalarg; 2268a7d6542SBram Moolenaar 227101f4810SBram Moolenaar if (!getline_equal(eap->getline, eap->cookie, getsourceline)) 228101f4810SBram Moolenaar { 229451c2e35SBram Moolenaar emsg(_(e_import_can_only_be_used_in_script)); 230101f4810SBram Moolenaar return; 231101f4810SBram Moolenaar } 2321c991144SBram Moolenaar fill_evalarg_from_eap(&evalarg, eap, eap->skip); 233101f4810SBram Moolenaar 2341c991144SBram Moolenaar cmd_end = handle_import(eap->arg, NULL, current_sctx.sc_sid, 2351c991144SBram Moolenaar &evalarg, NULL); 2368a7d6542SBram Moolenaar if (cmd_end != NULL) 2378a7d6542SBram Moolenaar eap->nextcmd = check_nextcmd(cmd_end); 2381c991144SBram Moolenaar clear_evalarg(&evalarg, eap); 2398a7d6542SBram Moolenaar } 2408a7d6542SBram Moolenaar 2418a7d6542SBram Moolenaar /* 242f2d5c240SBram Moolenaar * Find an exported item in "sid" matching the name at "*argp". 243f2d5c240SBram Moolenaar * When it is a variable return the index. 244f2d5c240SBram Moolenaar * When it is a user function return "*ufunc". 245f2d5c240SBram Moolenaar * When not found returns -1 and "*ufunc" is NULL. 246f2d5c240SBram Moolenaar */ 247f2d5c240SBram Moolenaar int 248f2d5c240SBram Moolenaar find_exported( 249f2d5c240SBram Moolenaar int sid, 2501c991144SBram Moolenaar char_u *name, 251f2d5c240SBram Moolenaar ufunc_T **ufunc, 252fbbcd003SBram Moolenaar type_T **type, 253fbbcd003SBram Moolenaar cctx_T *cctx) 254f2d5c240SBram Moolenaar { 255f2d5c240SBram Moolenaar int idx = -1; 256f2d5c240SBram Moolenaar svar_T *sv; 257f2d5c240SBram Moolenaar scriptitem_T *script = SCRIPT_ITEM(sid); 258f2d5c240SBram Moolenaar 259f2d5c240SBram Moolenaar // find name in "script" 260f2d5c240SBram Moolenaar // TODO: also find script-local user function 26108251757SBram Moolenaar idx = get_script_item_idx(sid, name, 0, cctx); 262f2d5c240SBram Moolenaar if (idx >= 0) 263f2d5c240SBram Moolenaar { 264f2d5c240SBram Moolenaar sv = ((svar_T *)script->sn_var_vals.ga_data) + idx; 265f2d5c240SBram Moolenaar if (!sv->sv_export) 266f2d5c240SBram Moolenaar { 267451c2e35SBram Moolenaar semsg(_(e_item_not_exported_in_script_str), name); 268f2d5c240SBram Moolenaar return -1; 269f2d5c240SBram Moolenaar } 270f2d5c240SBram Moolenaar *type = sv->sv_type; 271f2d5c240SBram Moolenaar *ufunc = NULL; 272f2d5c240SBram Moolenaar } 273f2d5c240SBram Moolenaar else 274f2d5c240SBram Moolenaar { 275f2d5c240SBram Moolenaar char_u buffer[200]; 276f2d5c240SBram Moolenaar char_u *funcname; 277f2d5c240SBram Moolenaar 278f2d5c240SBram Moolenaar // it could be a user function. 2795a67c37aSBram Moolenaar if (STRLEN(name) < sizeof(buffer) - 15) 280f2d5c240SBram Moolenaar funcname = buffer; 281f2d5c240SBram Moolenaar else 282f2d5c240SBram Moolenaar { 2835a67c37aSBram Moolenaar funcname = alloc(STRLEN(name) + 15); 284f2d5c240SBram Moolenaar if (funcname == NULL) 285f2d5c240SBram Moolenaar return -1; 286f2d5c240SBram Moolenaar } 287f2d5c240SBram Moolenaar funcname[0] = K_SPECIAL; 288f2d5c240SBram Moolenaar funcname[1] = KS_EXTRA; 289f2d5c240SBram Moolenaar funcname[2] = (int)KE_SNR; 290f2d5c240SBram Moolenaar sprintf((char *)funcname + 3, "%ld_%s", (long)sid, name); 2914c17ad94SBram Moolenaar *ufunc = find_func(funcname, FALSE, NULL); 292f2d5c240SBram Moolenaar if (funcname != buffer) 293f2d5c240SBram Moolenaar vim_free(funcname); 294f2d5c240SBram Moolenaar 295f2d5c240SBram Moolenaar if (*ufunc == NULL) 296f2d5c240SBram Moolenaar { 297451c2e35SBram Moolenaar semsg(_(e_item_not_found_in_script_str), name); 298f2d5c240SBram Moolenaar return -1; 299f2d5c240SBram Moolenaar } 300f2d5c240SBram Moolenaar } 301f2d5c240SBram Moolenaar 302f2d5c240SBram Moolenaar return idx; 303f2d5c240SBram Moolenaar } 304f2d5c240SBram Moolenaar 305f2d5c240SBram Moolenaar /* 3068a7d6542SBram Moolenaar * Handle an ":import" command and add the resulting imported_T to "gap", when 3078a7d6542SBram Moolenaar * not NULL, or script "import_sid" sn_imports. 308fbbcd003SBram Moolenaar * "cctx" is NULL at the script level. 3098a7d6542SBram Moolenaar * Returns a pointer to after the command or NULL in case of failure 3108a7d6542SBram Moolenaar */ 3118a7d6542SBram Moolenaar char_u * 3121c991144SBram Moolenaar handle_import( 3131c991144SBram Moolenaar char_u *arg_start, 3141c991144SBram Moolenaar garray_T *gap, 3151c991144SBram Moolenaar int import_sid, 3161c991144SBram Moolenaar evalarg_T *evalarg, 3171c991144SBram Moolenaar void *cctx) 3188a7d6542SBram Moolenaar { 3198a7d6542SBram Moolenaar char_u *arg = arg_start; 3201c991144SBram Moolenaar char_u *cmd_end = NULL; 3211c991144SBram Moolenaar char_u *as_name = NULL; 3228a7d6542SBram Moolenaar int ret = FAIL; 3238a7d6542SBram Moolenaar typval_T tv; 3248a7d6542SBram Moolenaar int sid = -1; 3258a7d6542SBram Moolenaar int res; 3261c991144SBram Moolenaar garray_T names; 3278a7d6542SBram Moolenaar 3281c991144SBram Moolenaar ga_init2(&names, sizeof(char_u *), 10); 3298a7d6542SBram Moolenaar if (*arg == '{') 3308a7d6542SBram Moolenaar { 3311c991144SBram Moolenaar // "import {item, item} from ..." 3321c991144SBram Moolenaar arg = skipwhite_and_linebreak(arg + 1, evalarg); 3331c991144SBram Moolenaar for (;;) 3341c991144SBram Moolenaar { 3351c991144SBram Moolenaar char_u *p = arg; 3361c991144SBram Moolenaar int had_comma = FALSE; 3371c991144SBram Moolenaar 3381c991144SBram Moolenaar while (eval_isnamec(*arg)) 3398a7d6542SBram Moolenaar ++arg; 3401c991144SBram Moolenaar if (p == arg) 3411c991144SBram Moolenaar break; 3421c991144SBram Moolenaar if (ga_grow(&names, 1) == FAIL) 3431c991144SBram Moolenaar goto erret; 3441c991144SBram Moolenaar ((char_u **)names.ga_data)[names.ga_len] = 3451c991144SBram Moolenaar vim_strnsave(p, arg - p); 3461c991144SBram Moolenaar ++names.ga_len; 3471c991144SBram Moolenaar if (*arg == ',') 3481c991144SBram Moolenaar { 3491c991144SBram Moolenaar had_comma = TRUE; 3501c991144SBram Moolenaar ++arg; 3511c991144SBram Moolenaar } 3521c991144SBram Moolenaar arg = skipwhite_and_linebreak(arg, evalarg); 3538a7d6542SBram Moolenaar if (*arg == '}') 3541c991144SBram Moolenaar { 3551c991144SBram Moolenaar arg = skipwhite_and_linebreak(arg + 1, evalarg); 3561c991144SBram Moolenaar break; 3571c991144SBram Moolenaar } 3581c991144SBram Moolenaar if (!had_comma) 3591c991144SBram Moolenaar { 360451c2e35SBram Moolenaar emsg(_(e_missing_comma_in_import)); 3611c991144SBram Moolenaar goto erret; 3621c991144SBram Moolenaar } 3631c991144SBram Moolenaar } 3641c991144SBram Moolenaar if (names.ga_len == 0) 3651c991144SBram Moolenaar { 366451c2e35SBram Moolenaar emsg(_(e_syntax_error_in_import)); 3671c991144SBram Moolenaar goto erret; 3681c991144SBram Moolenaar } 3698a7d6542SBram Moolenaar } 3708a7d6542SBram Moolenaar else 3718a7d6542SBram Moolenaar { 3721c991144SBram Moolenaar // "import Name from ..." 3731c991144SBram Moolenaar // "import * as Name from ..." 3741c991144SBram Moolenaar // "import item [as Name] from ..." 3751c991144SBram Moolenaar arg = skipwhite_and_linebreak(arg, evalarg); 3761c991144SBram Moolenaar if (arg[0] == '*' && IS_WHITE_OR_NUL(arg[1])) 3771c991144SBram Moolenaar arg = skipwhite_and_linebreak(arg + 1, evalarg); 378fa29c8abSBram Moolenaar else if (eval_isnamec1(*arg)) 3798a7d6542SBram Moolenaar { 3801c991144SBram Moolenaar char_u *p = arg; 3811c991144SBram Moolenaar 382fa29c8abSBram Moolenaar while (eval_isnamec(*arg)) 3838a7d6542SBram Moolenaar ++arg; 3841c991144SBram Moolenaar if (ga_grow(&names, 1) == FAIL) 3851c991144SBram Moolenaar goto erret; 3861c991144SBram Moolenaar ((char_u **)names.ga_data)[names.ga_len] = 3871c991144SBram Moolenaar vim_strnsave(p, arg - p); 3881c991144SBram Moolenaar ++names.ga_len; 3891c991144SBram Moolenaar arg = skipwhite_and_linebreak(arg, evalarg); 3908a7d6542SBram Moolenaar } 3911c991144SBram Moolenaar else 3928a7d6542SBram Moolenaar { 393451c2e35SBram Moolenaar emsg(_(e_syntax_error_in_import)); 3941c991144SBram Moolenaar goto erret; 3951c991144SBram Moolenaar } 3961c991144SBram Moolenaar 3971c991144SBram Moolenaar if (STRNCMP("as", arg, 2) == 0 && IS_WHITE_OR_NUL(arg[2])) 3981c991144SBram Moolenaar { 3991c991144SBram Moolenaar char_u *p; 4001c991144SBram Moolenaar 4011c991144SBram Moolenaar // skip over "as Name "; no line break allowed after "as" 4028a7d6542SBram Moolenaar arg = skipwhite(arg + 2); 4031c991144SBram Moolenaar p = arg; 404fa29c8abSBram Moolenaar if (eval_isnamec1(*arg)) 405fa29c8abSBram Moolenaar while (eval_isnamec(*arg)) 4068a7d6542SBram Moolenaar ++arg; 407cbb6bdcdSBram Moolenaar if (check_defined(p, arg - p, cctx) == FAIL) 4081c991144SBram Moolenaar goto erret; 4091c991144SBram Moolenaar as_name = vim_strnsave(p, arg - p); 4101c991144SBram Moolenaar arg = skipwhite_and_linebreak(arg, evalarg); 4118a7d6542SBram Moolenaar } 4128a7d6542SBram Moolenaar else if (*arg_start == '*') 4138a7d6542SBram Moolenaar { 414451c2e35SBram Moolenaar emsg(_(e_missing_as_after_star)); 4151c991144SBram Moolenaar goto erret; 4168a7d6542SBram Moolenaar } 4178a7d6542SBram Moolenaar } 4181c991144SBram Moolenaar 4191c991144SBram Moolenaar if (STRNCMP("from", arg, 4) != 0 || !IS_WHITE_OR_NUL(arg[4])) 4208a7d6542SBram Moolenaar { 421451c2e35SBram Moolenaar emsg(_(e_missing_from)); 4221c991144SBram Moolenaar goto erret; 4238a7d6542SBram Moolenaar } 4241c991144SBram Moolenaar 425962d7213SBram Moolenaar arg = skipwhite_and_linebreak(arg + 4, evalarg); 4268a7d6542SBram Moolenaar tv.v_type = VAR_UNKNOWN; 4278a7d6542SBram Moolenaar // TODO: should we accept any expression? 4288a7d6542SBram Moolenaar if (*arg == '\'') 4299a78e6dfSBram Moolenaar ret = eval_lit_string(&arg, &tv, TRUE); 4308a7d6542SBram Moolenaar else if (*arg == '"') 4319a78e6dfSBram Moolenaar ret = eval_string(&arg, &tv, TRUE); 4328a7d6542SBram Moolenaar if (ret == FAIL || tv.vval.v_string == NULL || *tv.vval.v_string == NUL) 4338a7d6542SBram Moolenaar { 434451c2e35SBram Moolenaar emsg(_(e_invalid_string_after_from)); 4351c991144SBram Moolenaar goto erret; 4368a7d6542SBram Moolenaar } 4378a7d6542SBram Moolenaar cmd_end = arg; 4388a7d6542SBram Moolenaar 4391c991144SBram Moolenaar /* 4401c991144SBram Moolenaar * find script file 4411c991144SBram Moolenaar */ 4428a7d6542SBram Moolenaar if (*tv.vval.v_string == '.') 4438a7d6542SBram Moolenaar { 4448a7d6542SBram Moolenaar size_t len; 44521b9e977SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid); 4468a7d6542SBram Moolenaar char_u *tail = gettail(si->sn_name); 4478a7d6542SBram Moolenaar char_u *from_name; 4488a7d6542SBram Moolenaar 4498a7d6542SBram Moolenaar // Relative to current script: "./name.vim", "../../name.vim". 4508a7d6542SBram Moolenaar len = STRLEN(si->sn_name) - STRLEN(tail) + STRLEN(tv.vval.v_string) + 2; 4518a7d6542SBram Moolenaar from_name = alloc((int)len); 4528a7d6542SBram Moolenaar if (from_name == NULL) 4538a7d6542SBram Moolenaar { 4548a7d6542SBram Moolenaar clear_tv(&tv); 4551c991144SBram Moolenaar goto erret; 4568a7d6542SBram Moolenaar } 4578a7d6542SBram Moolenaar vim_strncpy(from_name, si->sn_name, tail - si->sn_name); 4588a7d6542SBram Moolenaar add_pathsep(from_name); 4598a7d6542SBram Moolenaar STRCAT(from_name, tv.vval.v_string); 4608a7d6542SBram Moolenaar simplify_filename(from_name); 4618a7d6542SBram Moolenaar 4628a7d6542SBram Moolenaar res = do_source(from_name, FALSE, DOSO_NONE, &sid); 4638a7d6542SBram Moolenaar vim_free(from_name); 4648a7d6542SBram Moolenaar } 4658a7d6542SBram Moolenaar else if (mch_isFullName(tv.vval.v_string)) 4668a7d6542SBram Moolenaar { 4678a7d6542SBram Moolenaar // Absolute path: "/tmp/name.vim" 4688a7d6542SBram Moolenaar res = do_source(tv.vval.v_string, FALSE, DOSO_NONE, &sid); 4698a7d6542SBram Moolenaar } 4708a7d6542SBram Moolenaar else 4718a7d6542SBram Moolenaar { 4728a7d6542SBram Moolenaar size_t len = 7 + STRLEN(tv.vval.v_string) + 1; 4738a7d6542SBram Moolenaar char_u *from_name; 4748a7d6542SBram Moolenaar 4758a7d6542SBram Moolenaar // Find file in "import" subdirs in 'runtimepath'. 4768a7d6542SBram Moolenaar from_name = alloc((int)len); 4778a7d6542SBram Moolenaar if (from_name == NULL) 4788a7d6542SBram Moolenaar { 4798a7d6542SBram Moolenaar clear_tv(&tv); 4801c991144SBram Moolenaar goto erret; 4818a7d6542SBram Moolenaar } 4828a7d6542SBram Moolenaar vim_snprintf((char *)from_name, len, "import/%s", tv.vval.v_string); 4838a7d6542SBram Moolenaar res = source_in_path(p_rtp, from_name, DIP_NOAFTER, &sid); 4848a7d6542SBram Moolenaar vim_free(from_name); 4858a7d6542SBram Moolenaar } 4868a7d6542SBram Moolenaar 4878a7d6542SBram Moolenaar if (res == FAIL || sid <= 0) 4888a7d6542SBram Moolenaar { 489451c2e35SBram Moolenaar semsg(_(e_could_not_import_str), tv.vval.v_string); 4908a7d6542SBram Moolenaar clear_tv(&tv); 4911c991144SBram Moolenaar goto erret; 4928a7d6542SBram Moolenaar } 4938a7d6542SBram Moolenaar clear_tv(&tv); 4948a7d6542SBram Moolenaar 4958a7d6542SBram Moolenaar if (*arg_start == '*') 4968a7d6542SBram Moolenaar { 497a6294955SBram Moolenaar imported_T *imported; 4988a7d6542SBram Moolenaar 499a6294955SBram Moolenaar imported = find_imported(as_name, STRLEN(as_name), cctx); 500a6294955SBram Moolenaar if (imported != NULL && imported->imp_sid == sid) 501a6294955SBram Moolenaar { 502a6294955SBram Moolenaar if (imported->imp_flags & IMP_FLAGS_RELOAD) 503a6294955SBram Moolenaar // import already defined on a previous script load 504a6294955SBram Moolenaar imported->imp_flags &= ~IMP_FLAGS_RELOAD; 505a6294955SBram Moolenaar else 506a6294955SBram Moolenaar { 507a6294955SBram Moolenaar semsg(_(e_name_already_defined_str), as_name); 508a6294955SBram Moolenaar goto erret; 509a6294955SBram Moolenaar } 510a6294955SBram Moolenaar } 511a6294955SBram Moolenaar 512a6294955SBram Moolenaar imported = new_imported(gap != NULL ? gap 513a6294955SBram Moolenaar : &SCRIPT_ITEM(import_sid)->sn_imports); 5148a7d6542SBram Moolenaar if (imported == NULL) 5151c991144SBram Moolenaar goto erret; 5161c991144SBram Moolenaar imported->imp_name = as_name; 5171c991144SBram Moolenaar as_name = NULL; 5188a7d6542SBram Moolenaar imported->imp_sid = sid; 519a6294955SBram Moolenaar imported->imp_flags = IMP_FLAGS_STAR; 5208a7d6542SBram Moolenaar } 5218a7d6542SBram Moolenaar else 5228a7d6542SBram Moolenaar { 5231c991144SBram Moolenaar int i; 5241c991144SBram Moolenaar 5258a7d6542SBram Moolenaar arg = arg_start; 5268a7d6542SBram Moolenaar if (*arg == '{') 5278a7d6542SBram Moolenaar arg = skipwhite(arg + 1); 5281c991144SBram Moolenaar for (i = 0; i < names.ga_len; ++i) 5298a7d6542SBram Moolenaar { 5301c991144SBram Moolenaar char_u *name = ((char_u **)names.ga_data)[i]; 531a6294955SBram Moolenaar size_t len = STRLEN(name); 5328a7d6542SBram Moolenaar int idx; 5338a7d6542SBram Moolenaar imported_T *imported; 534f2d5c240SBram Moolenaar ufunc_T *ufunc = NULL; 535f2d5c240SBram Moolenaar type_T *type; 5368a7d6542SBram Moolenaar 537fbbcd003SBram Moolenaar idx = find_exported(sid, name, &ufunc, &type, cctx); 5388a7d6542SBram Moolenaar 539f2d5c240SBram Moolenaar if (idx < 0 && ufunc == NULL) 5401c991144SBram Moolenaar goto erret; 5418a7d6542SBram Moolenaar 542a6294955SBram Moolenaar // If already imported with the same propertis and the 543a6294955SBram Moolenaar // IMP_FLAGS_RELOAD set then we keep that entry. Otherwise create 544a6294955SBram Moolenaar // a new one (and give an error for an existing import). 545a6294955SBram Moolenaar imported = find_imported(name, len, cctx); 546a6294955SBram Moolenaar if (imported != NULL 547a6294955SBram Moolenaar && (imported->imp_flags & IMP_FLAGS_RELOAD) 548a6294955SBram Moolenaar && imported->imp_sid == sid 549a6294955SBram Moolenaar && (idx >= 0 550a6294955SBram Moolenaar ? (equal_type(imported->imp_type, type) 551a6294955SBram Moolenaar && imported->imp_var_vals_idx == idx) 552a6294955SBram Moolenaar : (equal_type(imported->imp_type, ufunc->uf_func_type) 553a6294955SBram Moolenaar && STRCMP(imported->imp_funcname, 554a6294955SBram Moolenaar ufunc->uf_name) == 0))) 555a6294955SBram Moolenaar { 556a6294955SBram Moolenaar imported->imp_flags &= ~IMP_FLAGS_RELOAD; 557a6294955SBram Moolenaar } 558a6294955SBram Moolenaar else 559a6294955SBram Moolenaar { 560a6294955SBram Moolenaar if (check_defined(name, len, cctx) == FAIL) 5611c991144SBram Moolenaar goto erret; 5625269bd2aSBram Moolenaar 5638a7d6542SBram Moolenaar imported = new_imported(gap != NULL ? gap 56421b9e977SBram Moolenaar : &SCRIPT_ITEM(import_sid)->sn_imports); 5658a7d6542SBram Moolenaar if (imported == NULL) 5661c991144SBram Moolenaar goto erret; 5678a7d6542SBram Moolenaar 5688a7d6542SBram Moolenaar // TODO: check for "as" following 5691c991144SBram Moolenaar // imported->imp_name = vim_strsave(as_name); 5701c991144SBram Moolenaar imported->imp_name = name; 5711c991144SBram Moolenaar ((char_u **)names.ga_data)[i] = NULL; 5728a7d6542SBram Moolenaar imported->imp_sid = sid; 5738a7d6542SBram Moolenaar if (idx >= 0) 5748a7d6542SBram Moolenaar { 575f2d5c240SBram Moolenaar imported->imp_type = type; 5768a7d6542SBram Moolenaar imported->imp_var_vals_idx = idx; 5778a7d6542SBram Moolenaar } 5788a7d6542SBram Moolenaar else 57940f4f7a4SBram Moolenaar { 58040f4f7a4SBram Moolenaar imported->imp_type = ufunc->uf_func_type; 5818a7d6542SBram Moolenaar imported->imp_funcname = ufunc->uf_name; 5828a7d6542SBram Moolenaar } 5838a7d6542SBram Moolenaar } 58440f4f7a4SBram Moolenaar } 585a6294955SBram Moolenaar } 5861c991144SBram Moolenaar erret: 5871c991144SBram Moolenaar ga_clear_strings(&names); 5881c991144SBram Moolenaar vim_free(as_name); 5898a7d6542SBram Moolenaar return cmd_end; 5908a7d6542SBram Moolenaar } 5918a7d6542SBram Moolenaar 592c82a5b5dSBram Moolenaar /* 593c82a5b5dSBram Moolenaar * Declare a script-local variable without init: "let var: type". 594c82a5b5dSBram Moolenaar * "const" is an error since the value is missing. 595c82a5b5dSBram Moolenaar * Returns a pointer to after the type. 596c82a5b5dSBram Moolenaar */ 597c82a5b5dSBram Moolenaar char_u * 598c82a5b5dSBram Moolenaar vim9_declare_scriptvar(exarg_T *eap, char_u *arg) 599c82a5b5dSBram Moolenaar { 600c82a5b5dSBram Moolenaar char_u *p; 601c82a5b5dSBram Moolenaar char_u *name; 602c82a5b5dSBram Moolenaar scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid); 603c82a5b5dSBram Moolenaar type_T *type; 604c82a5b5dSBram Moolenaar typval_T init_tv; 605c82a5b5dSBram Moolenaar 60630fd8204SBram Moolenaar if (eap->cmdidx == CMD_final || eap->cmdidx == CMD_const) 607c82a5b5dSBram Moolenaar { 60830fd8204SBram Moolenaar if (eap->cmdidx == CMD_final) 60930fd8204SBram Moolenaar emsg(_(e_final_requires_a_value)); 61030fd8204SBram Moolenaar else 611bc4c5051SBram Moolenaar emsg(_(e_const_requires_a_value)); 612c82a5b5dSBram Moolenaar return arg + STRLEN(arg); 613c82a5b5dSBram Moolenaar } 614c82a5b5dSBram Moolenaar 615c82a5b5dSBram Moolenaar // Check for valid starting character. 616c82a5b5dSBram Moolenaar if (!eval_isnamec1(*arg)) 617c82a5b5dSBram Moolenaar { 618c82a5b5dSBram Moolenaar semsg(_(e_invarg2), arg); 619c82a5b5dSBram Moolenaar return arg + STRLEN(arg); 620c82a5b5dSBram Moolenaar } 621c82a5b5dSBram Moolenaar 622984dddbeSBram Moolenaar for (p = arg + 1; *p != NUL && eval_isnamec(*p); MB_PTR_ADV(p)) 6233b74b6b4SBram Moolenaar if (*p == ':' && (VIM_ISWHITE(p[1]) || p != arg + 1)) 624984dddbeSBram Moolenaar break; 625c82a5b5dSBram Moolenaar 626c82a5b5dSBram Moolenaar if (*p != ':') 627c82a5b5dSBram Moolenaar { 628bc4c5051SBram Moolenaar emsg(_(e_type_or_initialization_required)); 629c82a5b5dSBram Moolenaar return arg + STRLEN(arg); 630c82a5b5dSBram Moolenaar } 631984dddbeSBram Moolenaar if (!VIM_ISWHITE(p[1])) 632984dddbeSBram Moolenaar { 633c3fc75dbSBram Moolenaar semsg(_(e_white_space_required_after_str_str), ":", p); 634984dddbeSBram Moolenaar return arg + STRLEN(arg); 635984dddbeSBram Moolenaar } 636c82a5b5dSBram Moolenaar name = vim_strnsave(arg, p - arg); 637c82a5b5dSBram Moolenaar 638c82a5b5dSBram Moolenaar // parse type 639c82a5b5dSBram Moolenaar p = skipwhite(p + 1); 6409e68c325SBram Moolenaar type = parse_type(&p, &si->sn_type_list, TRUE); 6419e68c325SBram Moolenaar if (type == NULL) 642f3decc58SBram Moolenaar { 643f3decc58SBram Moolenaar vim_free(name); 644c82a5b5dSBram Moolenaar return p; 645f3decc58SBram Moolenaar } 646c82a5b5dSBram Moolenaar 647c82a5b5dSBram Moolenaar // Create the variable with 0/NULL value. 648c82a5b5dSBram Moolenaar CLEAR_FIELD(init_tv); 649f0afd9e1SBram Moolenaar if (type->tt_type == VAR_ANY) 650f0afd9e1SBram Moolenaar // A variable of type "any" is not possible, just use zero instead 651f0afd9e1SBram Moolenaar init_tv.v_type = VAR_NUMBER; 652f0afd9e1SBram Moolenaar else 653c82a5b5dSBram Moolenaar init_tv.v_type = type->tt_type; 654f785aa13SBram Moolenaar set_var_const(name, type, &init_tv, FALSE, 0, 0); 655c82a5b5dSBram Moolenaar 656c82a5b5dSBram Moolenaar vim_free(name); 657c82a5b5dSBram Moolenaar return p; 658c82a5b5dSBram Moolenaar } 659c82a5b5dSBram Moolenaar 66034db91f7SBram Moolenaar /* 66139ca4127SBram Moolenaar * Vim9 part of adding a script variable: add it to sn_all_vars (lookup by name 66239ca4127SBram Moolenaar * with a hashtable) and sn_var_vals (lookup by index). 66307a65d26SBram Moolenaar * When "create" is TRUE this is a new variable, otherwise find and update an 66407a65d26SBram Moolenaar * existing variable. 66508251757SBram Moolenaar * "flags" can have ASSIGN_FINAL or ASSIGN_CONST. 666aa210a3aSBram Moolenaar * When "*type" is NULL use "tv" for the type and update "*type". 6678d739de4SBram Moolenaar */ 6688d739de4SBram Moolenaar void 66908251757SBram Moolenaar update_vim9_script_var( 67008251757SBram Moolenaar int create, 67108251757SBram Moolenaar dictitem_T *di, 67208251757SBram Moolenaar int flags, 67308251757SBram Moolenaar typval_T *tv, 67408251757SBram Moolenaar type_T **type) 6758d739de4SBram Moolenaar { 6768d739de4SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid); 6778d739de4SBram Moolenaar hashitem_T *hi; 67807a65d26SBram Moolenaar svar_T *sv; 6798d739de4SBram Moolenaar 68007a65d26SBram Moolenaar if (create) 68107a65d26SBram Moolenaar { 68207a65d26SBram Moolenaar sallvar_T *newsav; 68307a65d26SBram Moolenaar 68407a65d26SBram Moolenaar // Store a pointer to the typval_T, so that it can be found by index 68507a65d26SBram Moolenaar // instead of using a hastab lookup. 68607a65d26SBram Moolenaar if (ga_grow(&si->sn_var_vals, 1) == FAIL) 68707a65d26SBram Moolenaar return; 68807a65d26SBram Moolenaar 68907a65d26SBram Moolenaar sv = ((svar_T *)si->sn_var_vals.ga_data) + si->sn_var_vals.ga_len; 69007a65d26SBram Moolenaar newsav = (sallvar_T *)alloc_clear( 69107a65d26SBram Moolenaar sizeof(sallvar_T) + STRLEN(di->di_key)); 6928d739de4SBram Moolenaar if (newsav == NULL) 6938d739de4SBram Moolenaar return; 6948d739de4SBram Moolenaar 6958d739de4SBram Moolenaar sv->sv_tv = &di->di_tv; 69608251757SBram Moolenaar sv->sv_const = (flags & ASSIGN_FINAL) ? ASSIGN_FINAL 69708251757SBram Moolenaar : (flags & ASSIGN_CONST) ? ASSIGN_CONST : 0; 6988d739de4SBram Moolenaar sv->sv_export = is_export; 6998d739de4SBram Moolenaar newsav->sav_var_vals_idx = si->sn_var_vals.ga_len; 7008d739de4SBram Moolenaar ++si->sn_var_vals.ga_len; 7018d739de4SBram Moolenaar STRCPY(&newsav->sav_key, di->di_key); 7028d739de4SBram Moolenaar sv->sv_name = newsav->sav_key; 7038d739de4SBram Moolenaar newsav->sav_di = di; 7048d739de4SBram Moolenaar newsav->sav_block_id = si->sn_current_block_id; 7058d739de4SBram Moolenaar 7068d739de4SBram Moolenaar hi = hash_find(&si->sn_all_vars.dv_hashtab, newsav->sav_key); 7078d739de4SBram Moolenaar if (!HASHITEM_EMPTY(hi)) 7088d739de4SBram Moolenaar { 7098d739de4SBram Moolenaar sallvar_T *sav = HI2SAV(hi); 7108d739de4SBram Moolenaar 7118d739de4SBram Moolenaar // variable with this name exists in another block 7128d739de4SBram Moolenaar while (sav->sav_next != NULL) 7138d739de4SBram Moolenaar sav = sav->sav_next; 7148d739de4SBram Moolenaar sav->sav_next = newsav; 7158d739de4SBram Moolenaar } 7168d739de4SBram Moolenaar else 7178d739de4SBram Moolenaar // new variable name 7188d739de4SBram Moolenaar hash_add(&si->sn_all_vars.dv_hashtab, newsav->sav_key); 71907a65d26SBram Moolenaar } 72007a65d26SBram Moolenaar else 72107a65d26SBram Moolenaar { 72207a65d26SBram Moolenaar sv = find_typval_in_script(&di->di_tv); 72307a65d26SBram Moolenaar } 72407a65d26SBram Moolenaar if (sv != NULL) 72507a65d26SBram Moolenaar { 726aa210a3aSBram Moolenaar if (*type == NULL) 727aa210a3aSBram Moolenaar *type = typval2type(tv, &si->sn_type_list); 728aa210a3aSBram Moolenaar sv->sv_type = *type; 72907a65d26SBram Moolenaar } 7308d739de4SBram Moolenaar 7318d739de4SBram Moolenaar // let ex_export() know the export worked. 7328d739de4SBram Moolenaar is_export = FALSE; 7338d739de4SBram Moolenaar } 7348d739de4SBram Moolenaar 735fbbcd003SBram Moolenaar /* 736fbbcd003SBram Moolenaar * Hide a script variable when leaving a block. 737fbbcd003SBram Moolenaar * "idx" is de index in sn_var_vals. 73839ca4127SBram Moolenaar * When "func_defined" is non-zero then a function was defined in this block, 73939ca4127SBram Moolenaar * the variable may be accessed by it. Otherwise the variable can be cleared. 740fbbcd003SBram Moolenaar */ 7418d739de4SBram Moolenaar void 74239ca4127SBram Moolenaar hide_script_var(scriptitem_T *si, int idx, int func_defined) 7438d739de4SBram Moolenaar { 744fbbcd003SBram Moolenaar svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data) + idx; 7458d739de4SBram Moolenaar hashtab_T *script_ht = get_script_local_ht(); 7468d739de4SBram Moolenaar hashtab_T *all_ht = &si->sn_all_vars.dv_hashtab; 7478d739de4SBram Moolenaar hashitem_T *script_hi; 7488d739de4SBram Moolenaar hashitem_T *all_hi; 7498d739de4SBram Moolenaar 7508d739de4SBram Moolenaar // Remove a variable declared inside the block, if it still exists. 75139ca4127SBram Moolenaar // If it was added in a nested block it will already have been removed. 7528d739de4SBram Moolenaar // The typval is moved into the sallvar_T. 7538d739de4SBram Moolenaar script_hi = hash_find(script_ht, sv->sv_name); 7548d739de4SBram Moolenaar all_hi = hash_find(all_ht, sv->sv_name); 7558d739de4SBram Moolenaar if (!HASHITEM_EMPTY(script_hi) && !HASHITEM_EMPTY(all_hi)) 7568d739de4SBram Moolenaar { 7578d739de4SBram Moolenaar dictitem_T *di = HI2DI(script_hi); 7588d739de4SBram Moolenaar sallvar_T *sav = HI2SAV(all_hi); 75939ca4127SBram Moolenaar sallvar_T *sav_prev = NULL; 7608d739de4SBram Moolenaar 761fbbcd003SBram Moolenaar // There can be multiple entries with the same name in different 762fbbcd003SBram Moolenaar // blocks, find the right one. 763fbbcd003SBram Moolenaar while (sav != NULL && sav->sav_var_vals_idx != idx) 76439ca4127SBram Moolenaar { 76539ca4127SBram Moolenaar sav_prev = sav; 766fbbcd003SBram Moolenaar sav = sav->sav_next; 76739ca4127SBram Moolenaar } 768fbbcd003SBram Moolenaar if (sav != NULL) 769fbbcd003SBram Moolenaar { 77039ca4127SBram Moolenaar if (func_defined) 77139ca4127SBram Moolenaar { 77239ca4127SBram Moolenaar // move the typval from the dictitem to the sallvar 7738d739de4SBram Moolenaar sav->sav_tv = di->di_tv; 7748d739de4SBram Moolenaar di->di_tv.v_type = VAR_UNKNOWN; 7758d739de4SBram Moolenaar sav->sav_flags = di->di_flags; 7768d739de4SBram Moolenaar sav->sav_di = NULL; 777fbbcd003SBram Moolenaar sv->sv_tv = &sav->sav_tv; 778fbbcd003SBram Moolenaar } 77939ca4127SBram Moolenaar else 78039ca4127SBram Moolenaar { 78139ca4127SBram Moolenaar if (sav_prev == NULL) 78239ca4127SBram Moolenaar hash_remove(all_ht, all_hi); 78339ca4127SBram Moolenaar else 78439ca4127SBram Moolenaar sav_prev->sav_next = sav->sav_next; 78539ca4127SBram Moolenaar sv->sv_name = NULL; 78639ca4127SBram Moolenaar vim_free(sav); 78739ca4127SBram Moolenaar } 78839ca4127SBram Moolenaar delete_var(script_ht, script_hi); 78939ca4127SBram Moolenaar } 7908d739de4SBram Moolenaar } 7918d739de4SBram Moolenaar } 7928d739de4SBram Moolenaar 7938d739de4SBram Moolenaar /* 7948d739de4SBram Moolenaar * Free the script variables from "sn_all_vars". 7958d739de4SBram Moolenaar */ 7968d739de4SBram Moolenaar void 7978d739de4SBram Moolenaar free_all_script_vars(scriptitem_T *si) 7988d739de4SBram Moolenaar { 7998d739de4SBram Moolenaar int todo; 8008d739de4SBram Moolenaar hashtab_T *ht = &si->sn_all_vars.dv_hashtab; 8018d739de4SBram Moolenaar hashitem_T *hi; 8028d739de4SBram Moolenaar sallvar_T *sav; 8038d739de4SBram Moolenaar sallvar_T *sav_next; 8048d739de4SBram Moolenaar 8058d739de4SBram Moolenaar hash_lock(ht); 8068d739de4SBram Moolenaar todo = (int)ht->ht_used; 8078d739de4SBram Moolenaar for (hi = ht->ht_array; todo > 0; ++hi) 8088d739de4SBram Moolenaar { 8098d739de4SBram Moolenaar if (!HASHITEM_EMPTY(hi)) 8108d739de4SBram Moolenaar { 8118d739de4SBram Moolenaar --todo; 8128d739de4SBram Moolenaar 8138d739de4SBram Moolenaar // Free the variable. Don't remove it from the hashtab, ht_array 8148d739de4SBram Moolenaar // might change then. hash_clear() takes care of it later. 8158d739de4SBram Moolenaar sav = HI2SAV(hi); 8168d739de4SBram Moolenaar while (sav != NULL) 8178d739de4SBram Moolenaar { 8188d739de4SBram Moolenaar sav_next = sav->sav_next; 8198d739de4SBram Moolenaar if (sav->sav_di == NULL) 8208d739de4SBram Moolenaar clear_tv(&sav->sav_tv); 8218d739de4SBram Moolenaar vim_free(sav); 8228d739de4SBram Moolenaar sav = sav_next; 8238d739de4SBram Moolenaar } 8248d739de4SBram Moolenaar } 8258d739de4SBram Moolenaar } 8268d739de4SBram Moolenaar hash_clear(ht); 8278d739de4SBram Moolenaar hash_init(ht); 8288d739de4SBram Moolenaar 8298d739de4SBram Moolenaar ga_clear(&si->sn_var_vals); 8302b32700dSBram Moolenaar 8312b32700dSBram Moolenaar // existing commands using script variable indexes are no longer valid 8322b32700dSBram Moolenaar si->sn_script_seq = current_sctx.sc_seq; 8338d739de4SBram Moolenaar } 8348d739de4SBram Moolenaar 8358d739de4SBram Moolenaar /* 83610c65860SBram Moolenaar * Find the script-local variable that links to "dest". 83710c65860SBram Moolenaar * Returns NULL if not found. 83834db91f7SBram Moolenaar */ 83910c65860SBram Moolenaar svar_T * 84010c65860SBram Moolenaar find_typval_in_script(typval_T *dest) 84134db91f7SBram Moolenaar { 84234db91f7SBram Moolenaar scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid); 84334db91f7SBram Moolenaar int idx; 84434db91f7SBram Moolenaar 84581e17fbeSBram Moolenaar if (si->sn_version != SCRIPT_VERSION_VIM9) 84681e17fbeSBram Moolenaar // legacy script doesn't store variable types 84710c65860SBram Moolenaar return NULL; 84881e17fbeSBram Moolenaar 84934db91f7SBram Moolenaar // Find the svar_T in sn_var_vals. 85034db91f7SBram Moolenaar for (idx = 0; idx < si->sn_var_vals.ga_len; ++idx) 85134db91f7SBram Moolenaar { 85234db91f7SBram Moolenaar svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data) + idx; 85334db91f7SBram Moolenaar 854c0913d02SBram Moolenaar // If "sv_name" is NULL the variable was hidden when leaving a block, 855c0913d02SBram Moolenaar // don't check "sv_tv" then, it might be used for another variable now. 856c0913d02SBram Moolenaar if (sv->sv_name != NULL && sv->sv_tv == dest) 85710c65860SBram Moolenaar return sv; 85810c65860SBram Moolenaar } 859f785aa13SBram Moolenaar iemsg("find_typval_in_script(): not found"); 86010c65860SBram Moolenaar return NULL; 86110c65860SBram Moolenaar } 86210c65860SBram Moolenaar 86310c65860SBram Moolenaar /* 86410c65860SBram Moolenaar * Check if the type of script variable "dest" allows assigning "value". 86510c65860SBram Moolenaar * If needed convert "value" to a bool. 86610c65860SBram Moolenaar */ 86710c65860SBram Moolenaar int 868f785aa13SBram Moolenaar check_script_var_type( 869f785aa13SBram Moolenaar typval_T *dest, 870f785aa13SBram Moolenaar typval_T *value, 871f785aa13SBram Moolenaar char_u *name, 872f785aa13SBram Moolenaar where_T where) 87334db91f7SBram Moolenaar { 87410c65860SBram Moolenaar svar_T *sv = find_typval_in_script(dest); 875c1ec0422SBram Moolenaar int ret; 876c1ec0422SBram Moolenaar 87710c65860SBram Moolenaar if (sv != NULL) 87810c65860SBram Moolenaar { 87908251757SBram Moolenaar if (sv->sv_const != 0) 880c785b9a7SBram Moolenaar { 88134db91f7SBram Moolenaar semsg(_(e_readonlyvar), name); 882c785b9a7SBram Moolenaar return FAIL; 883c785b9a7SBram Moolenaar } 884f785aa13SBram Moolenaar ret = check_typval_type(sv->sv_type, value, where); 885c1ec0422SBram Moolenaar if (ret == OK && need_convert_to_bool(sv->sv_type, value)) 886c1ec0422SBram Moolenaar { 887c1ec0422SBram Moolenaar int val = tv2bool(value); 888c1ec0422SBram Moolenaar 889c1ec0422SBram Moolenaar clear_tv(value); 890c1ec0422SBram Moolenaar value->v_type = VAR_BOOL; 891c1ec0422SBram Moolenaar value->v_lock = 0; 892c1ec0422SBram Moolenaar value->vval.v_number = val ? VVAL_TRUE : VVAL_FALSE; 893c1ec0422SBram Moolenaar } 894c1ec0422SBram Moolenaar return ret; 89534db91f7SBram Moolenaar } 89610c65860SBram Moolenaar 897c785b9a7SBram Moolenaar return OK; // not really 89834db91f7SBram Moolenaar } 899c82a5b5dSBram Moolenaar 9008a7d6542SBram Moolenaar #endif // FEAT_EVAL 901