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