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