blob: 0994e513fc6cdc41cbc19ba15ce24ddb5b953c06 [file] [log] [blame]
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001/* 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 Moolenaar9b8d6222020-12-28 18:26:00 +010016#if defined(FEAT_EVAL)
17# include "vim9.h"
18#endif
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010019
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010020 int
21in_vim9script(void)
22{
Bram Moolenaareb6880b2020-07-12 17:07:05 +020023 // Do not go up the stack, a ":function" inside vim9script uses legacy
24 // syntax. "sc_version" is also set when compiling a ":def" function in
25 // legacy script.
Bram Moolenaar9979fcd2021-02-14 13:30:01 +010026 return current_sctx.sc_version == SCRIPT_VERSION_VIM9
27 || (cmdmod.cmod_flags & CMOD_VIM9CMD);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010028}
29
30/*
31 * ":vim9script".
32 */
33 void
Bram Moolenaar9b8d6222020-12-28 18:26:00 +010034ex_vim9script(exarg_T *eap UNUSED)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010035{
Bram Moolenaar9b8d6222020-12-28 18:26:00 +010036#ifdef FEAT_EVAL
Bram Moolenaar2b327002020-12-26 15:39:31 +010037 int sid = current_sctx.sc_sid;
Bram Moolenaar101f4812020-06-16 23:18:51 +020038 scriptitem_T *si;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010039
40 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
41 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +020042 emsg(_(e_vim9script_can_only_be_used_in_script));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010043 return;
44 }
Bram Moolenaar2b327002020-12-26 15:39:31 +010045
46 si = SCRIPT_ITEM(sid);
47 if (si->sn_state == SN_STATE_HAD_COMMAND)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010048 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +020049 emsg(_(e_vim9script_must_be_first_command_in_script));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010050 return;
51 }
Bram Moolenaar2b327002020-12-26 15:39:31 +010052 if (!IS_WHITE_OR_NUL(*eap->arg) && STRCMP(eap->arg, "noclear") != 0)
53 {
54 semsg(_(e_invarg2), eap->arg);
55 return;
56 }
57 if (si->sn_state == SN_STATE_RELOAD && IS_WHITE_OR_NUL(*eap->arg))
58 {
59 hashtab_T *ht = &SCRIPT_VARS(sid);
60
61 // Reloading a script without the "noclear" argument: clear
62 // script-local variables and functions.
63 hashtab_free_contents(ht);
64 hash_init(ht);
65 delete_script_functions(sid);
66
67 // old imports and script variables are no longer valid
68 free_imports_and_script_vars(sid);
69 }
70 si->sn_state = SN_STATE_HAD_COMMAND;
71
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010072 current_sctx.sc_version = SCRIPT_VERSION_VIM9;
73 si->sn_version = SCRIPT_VERSION_VIM9;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010074
75 if (STRCMP(p_cpo, CPO_VIM) != 0)
76 {
Bram Moolenaar39ca4122020-10-20 14:25:07 +020077 si->sn_save_cpo = vim_strsave(p_cpo);
78 set_option_value((char_u *)"cpo", 0L, (char_u *)CPO_VIM, 0);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010079 }
Bram Moolenaar9b8d6222020-12-28 18:26:00 +010080#else
81 // No check for this being the first command, it doesn't matter.
82 current_sctx.sc_version = SCRIPT_VERSION_VIM9;
83#endif
Bram Moolenaar8a7d6542020-01-26 15:56:19 +010084}
85
86/*
Bram Moolenaarae616492020-07-28 20:07:27 +020087 * When in Vim9 script give an error and return FAIL.
88 */
89 int
90not_in_vim9(exarg_T *eap)
91{
Bram Moolenaar68e30442020-07-28 21:15:07 +020092 if (in_vim9script())
93 switch (eap->cmdidx)
94 {
Bram Moolenaarada1d872021-02-20 08:16:51 +010095 case CMD_k:
96 if (eap->addr_count > 0)
97 {
98 emsg(_(e_norange));
99 return FAIL;
100 }
101 // FALLTHROUGH
Bram Moolenaar68e30442020-07-28 21:15:07 +0200102 case CMD_append:
103 case CMD_change:
Bram Moolenaarf5a48012020-08-01 17:00:03 +0200104 case CMD_insert:
105 case CMD_t:
Bram Moolenaar68e30442020-07-28 21:15:07 +0200106 case CMD_xit:
Bram Moolenaar9b8d6222020-12-28 18:26:00 +0100107 semsg(_(e_command_not_supported_in_vim9_script_missing_var_str), eap->cmd);
Bram Moolenaar68e30442020-07-28 21:15:07 +0200108 return FAIL;
109 default: break;
110 }
Bram Moolenaarae616492020-07-28 20:07:27 +0200111 return OK;
112}
113
Bram Moolenaardcc58e02020-12-28 20:53:21 +0100114/*
115 * Return TRUE if "p" points at a "#". Does not check for white space.
116 */
117 int
118vim9_comment_start(char_u *p)
119{
120 return p[0] == '#';
121}
122
Bram Moolenaar9b8d6222020-12-28 18:26:00 +0100123#if defined(FEAT_EVAL) || defined(PROTO)
124
Bram Moolenaarae616492020-07-28 20:07:27 +0200125/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100126 * ":export let Name: type"
127 * ":export const Name: type"
128 * ":export def Name(..."
129 * ":export class Name ..."
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100130 */
131 void
Bram Moolenaar09689a02020-05-09 22:50:08 +0200132ex_export(exarg_T *eap)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100133{
Bram Moolenaareb6880b2020-07-12 17:07:05 +0200134 if (!in_vim9script())
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100135 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200136 emsg(_(e_export_can_only_be_used_in_vim9script));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100137 return;
138 }
139
140 eap->cmd = eap->arg;
141 (void)find_ex_command(eap, NULL, lookup_scriptvar, NULL);
142 switch (eap->cmdidx)
143 {
144 case CMD_let:
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200145 case CMD_var:
146 case CMD_final:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100147 case CMD_const:
148 case CMD_def:
149 // case CMD_class:
150 is_export = TRUE;
151 do_cmdline(eap->cmd, eap->getline, eap->cookie,
152 DOCMD_VERBOSE + DOCMD_NOWAIT);
153
154 // The command will reset "is_export" when exporting an item.
155 if (is_export)
156 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200157 emsg(_(e_export_with_invalid_argument));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100158 is_export = FALSE;
159 }
160 break;
161 default:
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200162 emsg(_(e_invalid_command_after_export));
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100163 break;
164 }
165}
166
167/*
168 * Add a new imported item entry to the current script.
169 */
170 static imported_T *
171new_imported(garray_T *gap)
172{
173 if (ga_grow(gap, 1) == OK)
174 return ((imported_T *)gap->ga_data + gap->ga_len++);
175 return NULL;
176}
177
178/*
179 * Free all imported items in script "sid".
180 */
181 void
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200182free_imports_and_script_vars(int sid)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100183{
Bram Moolenaar21b9e972020-01-26 19:26:46 +0100184 scriptitem_T *si = SCRIPT_ITEM(sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100185 int idx;
186
187 for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
188 {
Bram Moolenaar20431c92020-03-20 18:39:46 +0100189 imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100190
191 vim_free(imp->imp_name);
192 }
193 ga_clear(&si->sn_imports);
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200194
195 free_all_script_vars(si);
196
Bram Moolenaar6110e792020-07-08 19:35:21 +0200197 clear_type_list(&si->sn_type_list);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100198}
199
200/*
Bram Moolenaara6294952020-12-27 13:39:50 +0100201 * Mark all imports as possible to redefine. Used when a script is loaded
202 * again but not cleared.
203 */
204 void
205mark_imports_for_reload(int sid)
206{
207 scriptitem_T *si = SCRIPT_ITEM(sid);
208 int idx;
209
210 for (idx = 0; idx < si->sn_imports.ga_len; ++idx)
211 {
212 imported_T *imp = ((imported_T *)si->sn_imports.ga_data) + idx;
213
214 imp->imp_flags |= IMP_FLAGS_RELOAD;
215 }
216}
217
218/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100219 * ":import Item from 'filename'"
220 * ":import Item as Alias from 'filename'"
221 * ":import {Item} from 'filename'".
222 * ":import {Item as Alias} from 'filename'"
223 * ":import {Item, Item} from 'filename'"
224 * ":import {Item, Item as Alias} from 'filename'"
225 *
226 * ":import * as Name from 'filename'"
227 */
228 void
229ex_import(exarg_T *eap)
230{
Bram Moolenaar1c991142020-07-04 13:15:31 +0200231 char_u *cmd_end;
232 evalarg_T evalarg;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100233
Bram Moolenaar101f4812020-06-16 23:18:51 +0200234 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
235 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200236 emsg(_(e_import_can_only_be_used_in_script));
Bram Moolenaar101f4812020-06-16 23:18:51 +0200237 return;
238 }
Bram Moolenaar1c991142020-07-04 13:15:31 +0200239 fill_evalarg_from_eap(&evalarg, eap, eap->skip);
Bram Moolenaar101f4812020-06-16 23:18:51 +0200240
Bram Moolenaar1c991142020-07-04 13:15:31 +0200241 cmd_end = handle_import(eap->arg, NULL, current_sctx.sc_sid,
242 &evalarg, NULL);
Bram Moolenaar9721fb42020-06-11 23:10:46 +0200243 if (cmd_end != NULL)
244 eap->nextcmd = check_nextcmd(cmd_end);
Bram Moolenaar1c991142020-07-04 13:15:31 +0200245 clear_evalarg(&evalarg, eap);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100246}
247
248/*
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100249 * Find an exported item in "sid" matching the name at "*argp".
250 * When it is a variable return the index.
251 * When it is a user function return "*ufunc".
252 * When not found returns -1 and "*ufunc" is NULL.
253 */
254 int
255find_exported(
256 int sid,
Bram Moolenaar1c991142020-07-04 13:15:31 +0200257 char_u *name,
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100258 ufunc_T **ufunc,
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200259 type_T **type,
260 cctx_T *cctx)
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100261{
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100262 int idx = -1;
263 svar_T *sv;
264 scriptitem_T *script = SCRIPT_ITEM(sid);
265
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100266 // find name in "script"
267 // TODO: also find script-local user function
Bram Moolenaar08251752021-01-11 21:20:18 +0100268 idx = get_script_item_idx(sid, name, 0, cctx);
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100269 if (idx >= 0)
270 {
271 sv = ((svar_T *)script->sn_var_vals.ga_data) + idx;
272 if (!sv->sv_export)
273 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200274 semsg(_(e_item_not_exported_in_script_str), name);
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100275 return -1;
276 }
277 *type = sv->sv_type;
278 *ufunc = NULL;
279 }
280 else
281 {
282 char_u buffer[200];
283 char_u *funcname;
284
285 // it could be a user function.
Bram Moolenaar5a67c372020-07-23 14:39:47 +0200286 if (STRLEN(name) < sizeof(buffer) - 15)
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100287 funcname = buffer;
288 else
289 {
Bram Moolenaar5a67c372020-07-23 14:39:47 +0200290 funcname = alloc(STRLEN(name) + 15);
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100291 if (funcname == NULL)
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100292 return -1;
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100293 }
294 funcname[0] = K_SPECIAL;
295 funcname[1] = KS_EXTRA;
296 funcname[2] = (int)KE_SNR;
297 sprintf((char *)funcname + 3, "%ld_%s", (long)sid, name);
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200298 *ufunc = find_func(funcname, FALSE, NULL);
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100299 if (funcname != buffer)
300 vim_free(funcname);
301
302 if (*ufunc == NULL)
303 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200304 semsg(_(e_item_not_found_in_script_str), name);
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100305 return -1;
306 }
307 }
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100308
309 return idx;
310}
311
312/*
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100313 * Handle an ":import" command and add the resulting imported_T to "gap", when
314 * not NULL, or script "import_sid" sn_imports.
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200315 * "cctx" is NULL at the script level.
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100316 * Returns a pointer to after the command or NULL in case of failure
317 */
318 char_u *
Bram Moolenaar1c991142020-07-04 13:15:31 +0200319handle_import(
320 char_u *arg_start,
321 garray_T *gap,
322 int import_sid,
323 evalarg_T *evalarg,
324 void *cctx)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100325{
326 char_u *arg = arg_start;
Bram Moolenaar1c991142020-07-04 13:15:31 +0200327 char_u *cmd_end = NULL;
328 char_u *as_name = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100329 int ret = FAIL;
330 typval_T tv;
331 int sid = -1;
332 int res;
Bram Moolenaar1c991142020-07-04 13:15:31 +0200333 garray_T names;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100334
Bram Moolenaar1c991142020-07-04 13:15:31 +0200335 ga_init2(&names, sizeof(char_u *), 10);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100336 if (*arg == '{')
337 {
Bram Moolenaar1c991142020-07-04 13:15:31 +0200338 // "import {item, item} from ..."
339 arg = skipwhite_and_linebreak(arg + 1, evalarg);
340 for (;;)
341 {
342 char_u *p = arg;
343 int had_comma = FALSE;
344
345 while (eval_isnamec(*arg))
346 ++arg;
347 if (p == arg)
348 break;
349 if (ga_grow(&names, 1) == FAIL)
350 goto erret;
351 ((char_u **)names.ga_data)[names.ga_len] =
352 vim_strnsave(p, arg - p);
353 ++names.ga_len;
354 if (*arg == ',')
355 {
356 had_comma = TRUE;
357 ++arg;
358 }
359 arg = skipwhite_and_linebreak(arg, evalarg);
360 if (*arg == '}')
361 {
362 arg = skipwhite_and_linebreak(arg + 1, evalarg);
363 break;
364 }
365 if (!had_comma)
366 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200367 emsg(_(e_missing_comma_in_import));
Bram Moolenaar1c991142020-07-04 13:15:31 +0200368 goto erret;
369 }
370 }
371 if (names.ga_len == 0)
372 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200373 emsg(_(e_syntax_error_in_import));
Bram Moolenaar1c991142020-07-04 13:15:31 +0200374 goto erret;
375 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100376 }
377 else
378 {
Bram Moolenaar1c991142020-07-04 13:15:31 +0200379 // "import Name from ..."
380 // "import * as Name from ..."
381 // "import item [as Name] from ..."
382 arg = skipwhite_and_linebreak(arg, evalarg);
383 if (arg[0] == '*' && IS_WHITE_OR_NUL(arg[1]))
384 arg = skipwhite_and_linebreak(arg + 1, evalarg);
Bram Moolenaarfa29c8a2020-02-23 22:35:05 +0100385 else if (eval_isnamec1(*arg))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100386 {
Bram Moolenaar1c991142020-07-04 13:15:31 +0200387 char_u *p = arg;
388
Bram Moolenaarfa29c8a2020-02-23 22:35:05 +0100389 while (eval_isnamec(*arg))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100390 ++arg;
Bram Moolenaar1c991142020-07-04 13:15:31 +0200391 if (ga_grow(&names, 1) == FAIL)
392 goto erret;
393 ((char_u **)names.ga_data)[names.ga_len] =
394 vim_strnsave(p, arg - p);
395 ++names.ga_len;
396 arg = skipwhite_and_linebreak(arg, evalarg);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100397 }
Bram Moolenaar1c991142020-07-04 13:15:31 +0200398 else
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100399 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200400 emsg(_(e_syntax_error_in_import));
Bram Moolenaar1c991142020-07-04 13:15:31 +0200401 goto erret;
402 }
403
404 if (STRNCMP("as", arg, 2) == 0 && IS_WHITE_OR_NUL(arg[2]))
405 {
406 char_u *p;
407
408 // skip over "as Name "; no line break allowed after "as"
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100409 arg = skipwhite(arg + 2);
Bram Moolenaar1c991142020-07-04 13:15:31 +0200410 p = arg;
Bram Moolenaarfa29c8a2020-02-23 22:35:05 +0100411 if (eval_isnamec1(*arg))
412 while (eval_isnamec(*arg))
413 ++arg;
Bram Moolenaarcbb6bdc2020-07-06 21:53:17 +0200414 if (check_defined(p, arg - p, cctx) == FAIL)
Bram Moolenaar1c991142020-07-04 13:15:31 +0200415 goto erret;
416 as_name = vim_strnsave(p, arg - p);
417 arg = skipwhite_and_linebreak(arg, evalarg);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100418 }
419 else if (*arg_start == '*')
420 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200421 emsg(_(e_missing_as_after_star));
Bram Moolenaar1c991142020-07-04 13:15:31 +0200422 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100423 }
424 }
Bram Moolenaar1c991142020-07-04 13:15:31 +0200425
426 if (STRNCMP("from", arg, 4) != 0 || !IS_WHITE_OR_NUL(arg[4]))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100427 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200428 emsg(_(e_missing_from));
Bram Moolenaar1c991142020-07-04 13:15:31 +0200429 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100430 }
Bram Moolenaar1c991142020-07-04 13:15:31 +0200431
Bram Moolenaar962d7212020-07-04 14:15:00 +0200432 arg = skipwhite_and_linebreak(arg + 4, evalarg);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100433 tv.v_type = VAR_UNKNOWN;
434 // TODO: should we accept any expression?
435 if (*arg == '\'')
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +0200436 ret = eval_lit_string(&arg, &tv, TRUE);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100437 else if (*arg == '"')
Bram Moolenaar9a78e6d2020-07-01 18:29:55 +0200438 ret = eval_string(&arg, &tv, TRUE);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100439 if (ret == FAIL || tv.vval.v_string == NULL || *tv.vval.v_string == NUL)
440 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200441 emsg(_(e_invalid_string_after_from));
Bram Moolenaar1c991142020-07-04 13:15:31 +0200442 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100443 }
444 cmd_end = arg;
445
Bram Moolenaar1c991142020-07-04 13:15:31 +0200446 /*
447 * find script file
448 */
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100449 if (*tv.vval.v_string == '.')
450 {
451 size_t len;
Bram Moolenaar21b9e972020-01-26 19:26:46 +0100452 scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100453 char_u *tail = gettail(si->sn_name);
454 char_u *from_name;
455
456 // Relative to current script: "./name.vim", "../../name.vim".
457 len = STRLEN(si->sn_name) - STRLEN(tail) + STRLEN(tv.vval.v_string) + 2;
458 from_name = alloc((int)len);
459 if (from_name == NULL)
460 {
461 clear_tv(&tv);
Bram Moolenaar1c991142020-07-04 13:15:31 +0200462 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100463 }
464 vim_strncpy(from_name, si->sn_name, tail - si->sn_name);
465 add_pathsep(from_name);
466 STRCAT(from_name, tv.vval.v_string);
467 simplify_filename(from_name);
468
469 res = do_source(from_name, FALSE, DOSO_NONE, &sid);
470 vim_free(from_name);
471 }
472 else if (mch_isFullName(tv.vval.v_string))
473 {
474 // Absolute path: "/tmp/name.vim"
475 res = do_source(tv.vval.v_string, FALSE, DOSO_NONE, &sid);
476 }
477 else
478 {
479 size_t len = 7 + STRLEN(tv.vval.v_string) + 1;
480 char_u *from_name;
481
482 // Find file in "import" subdirs in 'runtimepath'.
483 from_name = alloc((int)len);
484 if (from_name == NULL)
485 {
486 clear_tv(&tv);
Bram Moolenaar1c991142020-07-04 13:15:31 +0200487 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100488 }
489 vim_snprintf((char *)from_name, len, "import/%s", tv.vval.v_string);
490 res = source_in_path(p_rtp, from_name, DIP_NOAFTER, &sid);
491 vim_free(from_name);
492 }
493
494 if (res == FAIL || sid <= 0)
495 {
Bram Moolenaar451c2e32020-08-15 16:33:28 +0200496 semsg(_(e_could_not_import_str), tv.vval.v_string);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100497 clear_tv(&tv);
Bram Moolenaar1c991142020-07-04 13:15:31 +0200498 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100499 }
500 clear_tv(&tv);
501
502 if (*arg_start == '*')
503 {
Bram Moolenaara6294952020-12-27 13:39:50 +0100504 imported_T *imported;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100505
Bram Moolenaara6294952020-12-27 13:39:50 +0100506 imported = find_imported(as_name, STRLEN(as_name), cctx);
507 if (imported != NULL && imported->imp_sid == sid)
508 {
509 if (imported->imp_flags & IMP_FLAGS_RELOAD)
510 // import already defined on a previous script load
511 imported->imp_flags &= ~IMP_FLAGS_RELOAD;
512 else
513 {
514 semsg(_(e_name_already_defined_str), as_name);
515 goto erret;
516 }
517 }
518
519 imported = new_imported(gap != NULL ? gap
520 : &SCRIPT_ITEM(import_sid)->sn_imports);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100521 if (imported == NULL)
Bram Moolenaar1c991142020-07-04 13:15:31 +0200522 goto erret;
523 imported->imp_name = as_name;
524 as_name = NULL;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100525 imported->imp_sid = sid;
Bram Moolenaara6294952020-12-27 13:39:50 +0100526 imported->imp_flags = IMP_FLAGS_STAR;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100527 }
528 else
529 {
Bram Moolenaar1c991142020-07-04 13:15:31 +0200530 int i;
531
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100532 arg = arg_start;
533 if (*arg == '{')
534 arg = skipwhite(arg + 1);
Bram Moolenaar1c991142020-07-04 13:15:31 +0200535 for (i = 0; i < names.ga_len; ++i)
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100536 {
Bram Moolenaar1c991142020-07-04 13:15:31 +0200537 char_u *name = ((char_u **)names.ga_data)[i];
Bram Moolenaara6294952020-12-27 13:39:50 +0100538 size_t len = STRLEN(name);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100539 int idx;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100540 imported_T *imported;
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100541 ufunc_T *ufunc = NULL;
542 type_T *type;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100543
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200544 idx = find_exported(sid, name, &ufunc, &type, cctx);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100545
Bram Moolenaarf2d5c242020-02-23 21:25:54 +0100546 if (idx < 0 && ufunc == NULL)
Bram Moolenaar1c991142020-07-04 13:15:31 +0200547 goto erret;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100548
Bram Moolenaara6294952020-12-27 13:39:50 +0100549 // If already imported with the same propertis and the
550 // IMP_FLAGS_RELOAD set then we keep that entry. Otherwise create
551 // a new one (and give an error for an existing import).
552 imported = find_imported(name, len, cctx);
553 if (imported != NULL
554 && (imported->imp_flags & IMP_FLAGS_RELOAD)
555 && imported->imp_sid == sid
556 && (idx >= 0
557 ? (equal_type(imported->imp_type, type)
558 && imported->imp_var_vals_idx == idx)
559 : (equal_type(imported->imp_type, ufunc->uf_func_type)
560 && STRCMP(imported->imp_funcname,
561 ufunc->uf_name) == 0)))
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100562 {
Bram Moolenaara6294952020-12-27 13:39:50 +0100563 imported->imp_flags &= ~IMP_FLAGS_RELOAD;
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100564 }
565 else
Bram Moolenaar40f4f7a2020-07-23 22:41:43 +0200566 {
Bram Moolenaara6294952020-12-27 13:39:50 +0100567 if (check_defined(name, len, cctx) == FAIL)
568 goto erret;
569
570 imported = new_imported(gap != NULL ? gap
571 : &SCRIPT_ITEM(import_sid)->sn_imports);
572 if (imported == NULL)
573 goto erret;
574
575 // TODO: check for "as" following
576 // imported->imp_name = vim_strsave(as_name);
577 imported->imp_name = name;
578 ((char_u **)names.ga_data)[i] = NULL;
579 imported->imp_sid = sid;
580 if (idx >= 0)
581 {
582 imported->imp_type = type;
583 imported->imp_var_vals_idx = idx;
584 }
585 else
586 {
587 imported->imp_type = ufunc->uf_func_type;
588 imported->imp_funcname = ufunc->uf_name;
589 }
Bram Moolenaar40f4f7a2020-07-23 22:41:43 +0200590 }
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100591 }
592 }
Bram Moolenaar1c991142020-07-04 13:15:31 +0200593erret:
594 ga_clear_strings(&names);
595 vim_free(as_name);
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100596 return cmd_end;
597}
598
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200599/*
600 * Declare a script-local variable without init: "let var: type".
601 * "const" is an error since the value is missing.
602 * Returns a pointer to after the type.
603 */
604 char_u *
605vim9_declare_scriptvar(exarg_T *eap, char_u *arg)
606{
607 char_u *p;
608 char_u *name;
609 scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid);
610 type_T *type;
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200611 typval_T init_tv;
612
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200613 if (eap->cmdidx == CMD_final || eap->cmdidx == CMD_const)
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200614 {
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200615 if (eap->cmdidx == CMD_final)
616 emsg(_(e_final_requires_a_value));
617 else
618 emsg(_(e_const_requires_a_value));
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200619 return arg + STRLEN(arg);
620 }
621
622 // Check for valid starting character.
623 if (!eval_isnamec1(*arg))
624 {
625 semsg(_(e_invarg2), arg);
626 return arg + STRLEN(arg);
627 }
628
Bram Moolenaar984dddb2020-06-14 12:50:24 +0200629 for (p = arg + 1; *p != NUL && eval_isnamec(*p); MB_PTR_ADV(p))
Bram Moolenaar3b74b6b2020-06-19 19:01:43 +0200630 if (*p == ':' && (VIM_ISWHITE(p[1]) || p != arg + 1))
Bram Moolenaar984dddb2020-06-14 12:50:24 +0200631 break;
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200632
633 if (*p != ':')
634 {
Bram Moolenaarbc4c5052020-08-13 22:47:35 +0200635 emsg(_(e_type_or_initialization_required));
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200636 return arg + STRLEN(arg);
637 }
Bram Moolenaar984dddb2020-06-14 12:50:24 +0200638 if (!VIM_ISWHITE(p[1]))
639 {
Bram Moolenaarc3fc75d2021-02-07 15:28:09 +0100640 semsg(_(e_white_space_required_after_str_str), ":", p);
Bram Moolenaar984dddb2020-06-14 12:50:24 +0200641 return arg + STRLEN(arg);
642 }
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200643 name = vim_strnsave(arg, p - arg);
644
645 // parse type
646 p = skipwhite(p + 1);
Bram Moolenaar9e68c322020-12-25 12:38:04 +0100647 type = parse_type(&p, &si->sn_type_list, TRUE);
648 if (type == NULL)
Bram Moolenaarf3decc52020-06-13 19:56:38 +0200649 {
650 vim_free(name);
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200651 return p;
Bram Moolenaarf3decc52020-06-13 19:56:38 +0200652 }
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200653
654 // Create the variable with 0/NULL value.
655 CLEAR_FIELD(init_tv);
Bram Moolenaarf0afd9e2020-09-13 18:57:47 +0200656 if (type->tt_type == VAR_ANY)
657 // A variable of type "any" is not possible, just use zero instead
658 init_tv.v_type = VAR_NUMBER;
659 else
660 init_tv.v_type = type->tt_type;
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100661 set_var_const(name, type, &init_tv, FALSE, 0, 0);
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200662
663 vim_free(name);
664 return p;
665}
666
Bram Moolenaar34db91f2020-06-13 19:00:10 +0200667/*
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200668 * Vim9 part of adding a script variable: add it to sn_all_vars (lookup by name
669 * with a hashtable) and sn_var_vals (lookup by index).
Bram Moolenaar07a65d22020-12-26 20:09:15 +0100670 * When "create" is TRUE this is a new variable, otherwise find and update an
671 * existing variable.
Bram Moolenaar08251752021-01-11 21:20:18 +0100672 * "flags" can have ASSIGN_FINAL or ASSIGN_CONST.
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100673 * When "*type" is NULL use "tv" for the type and update "*type".
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200674 */
675 void
Bram Moolenaar08251752021-01-11 21:20:18 +0100676update_vim9_script_var(
677 int create,
678 dictitem_T *di,
679 int flags,
680 typval_T *tv,
681 type_T **type)
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200682{
Bram Moolenaar07a65d22020-12-26 20:09:15 +0100683 scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid);
684 hashitem_T *hi;
685 svar_T *sv;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200686
Bram Moolenaar07a65d22020-12-26 20:09:15 +0100687 if (create)
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200688 {
Bram Moolenaar07a65d22020-12-26 20:09:15 +0100689 sallvar_T *newsav;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200690
Bram Moolenaar07a65d22020-12-26 20:09:15 +0100691 // Store a pointer to the typval_T, so that it can be found by index
692 // instead of using a hastab lookup.
693 if (ga_grow(&si->sn_var_vals, 1) == FAIL)
694 return;
695
696 sv = ((svar_T *)si->sn_var_vals.ga_data) + si->sn_var_vals.ga_len;
697 newsav = (sallvar_T *)alloc_clear(
698 sizeof(sallvar_T) + STRLEN(di->di_key));
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200699 if (newsav == NULL)
700 return;
701
702 sv->sv_tv = &di->di_tv;
Bram Moolenaar08251752021-01-11 21:20:18 +0100703 sv->sv_const = (flags & ASSIGN_FINAL) ? ASSIGN_FINAL
704 : (flags & ASSIGN_CONST) ? ASSIGN_CONST : 0;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200705 sv->sv_export = is_export;
706 newsav->sav_var_vals_idx = si->sn_var_vals.ga_len;
707 ++si->sn_var_vals.ga_len;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200708 STRCPY(&newsav->sav_key, di->di_key);
709 sv->sv_name = newsav->sav_key;
710 newsav->sav_di = di;
711 newsav->sav_block_id = si->sn_current_block_id;
712
713 hi = hash_find(&si->sn_all_vars.dv_hashtab, newsav->sav_key);
714 if (!HASHITEM_EMPTY(hi))
715 {
716 sallvar_T *sav = HI2SAV(hi);
717
718 // variable with this name exists in another block
719 while (sav->sav_next != NULL)
720 sav = sav->sav_next;
721 sav->sav_next = newsav;
722 }
723 else
724 // new variable name
725 hash_add(&si->sn_all_vars.dv_hashtab, newsav->sav_key);
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200726 }
Bram Moolenaar07a65d22020-12-26 20:09:15 +0100727 else
728 {
729 sv = find_typval_in_script(&di->di_tv);
730 }
731 if (sv != NULL)
732 {
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100733 if (*type == NULL)
734 *type = typval2type(tv, &si->sn_type_list);
735 sv->sv_type = *type;
Bram Moolenaar07a65d22020-12-26 20:09:15 +0100736 }
737
738 // let ex_export() know the export worked.
739 is_export = FALSE;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200740}
741
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200742/*
743 * Hide a script variable when leaving a block.
744 * "idx" is de index in sn_var_vals.
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200745 * When "func_defined" is non-zero then a function was defined in this block,
746 * the variable may be accessed by it. Otherwise the variable can be cleared.
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200747 */
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200748 void
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200749hide_script_var(scriptitem_T *si, int idx, int func_defined)
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200750{
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200751 svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200752 hashtab_T *script_ht = get_script_local_ht();
753 hashtab_T *all_ht = &si->sn_all_vars.dv_hashtab;
754 hashitem_T *script_hi;
755 hashitem_T *all_hi;
756
757 // Remove a variable declared inside the block, if it still exists.
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200758 // If it was added in a nested block it will already have been removed.
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200759 // The typval is moved into the sallvar_T.
760 script_hi = hash_find(script_ht, sv->sv_name);
761 all_hi = hash_find(all_ht, sv->sv_name);
762 if (!HASHITEM_EMPTY(script_hi) && !HASHITEM_EMPTY(all_hi))
763 {
764 dictitem_T *di = HI2DI(script_hi);
765 sallvar_T *sav = HI2SAV(all_hi);
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200766 sallvar_T *sav_prev = NULL;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200767
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200768 // There can be multiple entries with the same name in different
769 // blocks, find the right one.
770 while (sav != NULL && sav->sav_var_vals_idx != idx)
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200771 {
772 sav_prev = sav;
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200773 sav = sav->sav_next;
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200774 }
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200775 if (sav != NULL)
776 {
Bram Moolenaar39ca4122020-10-20 14:25:07 +0200777 if (func_defined)
778 {
779 // move the typval from the dictitem to the sallvar
780 sav->sav_tv = di->di_tv;
781 di->di_tv.v_type = VAR_UNKNOWN;
782 sav->sav_flags = di->di_flags;
783 sav->sav_di = NULL;
784 sv->sv_tv = &sav->sav_tv;
785 }
786 else
787 {
788 if (sav_prev == NULL)
789 hash_remove(all_ht, all_hi);
790 else
791 sav_prev->sav_next = sav->sav_next;
792 sv->sv_name = NULL;
793 vim_free(sav);
794 }
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200795 delete_var(script_ht, script_hi);
Bram Moolenaarfbbcd002020-10-15 12:46:44 +0200796 }
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200797 }
798}
799
800/*
801 * Free the script variables from "sn_all_vars".
802 */
803 void
804free_all_script_vars(scriptitem_T *si)
805{
806 int todo;
807 hashtab_T *ht = &si->sn_all_vars.dv_hashtab;
808 hashitem_T *hi;
809 sallvar_T *sav;
810 sallvar_T *sav_next;
811
812 hash_lock(ht);
813 todo = (int)ht->ht_used;
814 for (hi = ht->ht_array; todo > 0; ++hi)
815 {
816 if (!HASHITEM_EMPTY(hi))
817 {
818 --todo;
819
820 // Free the variable. Don't remove it from the hashtab, ht_array
821 // might change then. hash_clear() takes care of it later.
822 sav = HI2SAV(hi);
823 while (sav != NULL)
824 {
825 sav_next = sav->sav_next;
826 if (sav->sav_di == NULL)
827 clear_tv(&sav->sav_tv);
828 vim_free(sav);
829 sav = sav_next;
830 }
831 }
832 }
833 hash_clear(ht);
834 hash_init(ht);
835
836 ga_clear(&si->sn_var_vals);
Bram Moolenaar2b327002020-12-26 15:39:31 +0100837
838 // existing commands using script variable indexes are no longer valid
839 si->sn_script_seq = current_sctx.sc_seq;
Bram Moolenaar8d739de2020-10-14 19:39:19 +0200840}
841
842/*
Bram Moolenaar10c65862020-10-08 21:16:42 +0200843 * Find the script-local variable that links to "dest".
844 * Returns NULL if not found.
Bram Moolenaar34db91f2020-06-13 19:00:10 +0200845 */
Bram Moolenaar10c65862020-10-08 21:16:42 +0200846 svar_T *
847find_typval_in_script(typval_T *dest)
Bram Moolenaar34db91f2020-06-13 19:00:10 +0200848{
849 scriptitem_T *si = SCRIPT_ITEM(current_sctx.sc_sid);
850 int idx;
851
Bram Moolenaar81e17fb2020-08-21 21:55:43 +0200852 if (si->sn_version != SCRIPT_VERSION_VIM9)
853 // legacy script doesn't store variable types
Bram Moolenaar10c65862020-10-08 21:16:42 +0200854 return NULL;
Bram Moolenaar81e17fb2020-08-21 21:55:43 +0200855
Bram Moolenaar34db91f2020-06-13 19:00:10 +0200856 // Find the svar_T in sn_var_vals.
857 for (idx = 0; idx < si->sn_var_vals.ga_len; ++idx)
858 {
859 svar_T *sv = ((svar_T *)si->sn_var_vals.ga_data) + idx;
860
Bram Moolenaarc0913d02020-12-05 14:44:37 +0100861 // If "sv_name" is NULL the variable was hidden when leaving a block,
862 // don't check "sv_tv" then, it might be used for another variable now.
863 if (sv->sv_name != NULL && sv->sv_tv == dest)
Bram Moolenaar10c65862020-10-08 21:16:42 +0200864 return sv;
Bram Moolenaar34db91f2020-06-13 19:00:10 +0200865 }
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100866 iemsg("find_typval_in_script(): not found");
Bram Moolenaar10c65862020-10-08 21:16:42 +0200867 return NULL;
868}
869
870/*
871 * Check if the type of script variable "dest" allows assigning "value".
872 * If needed convert "value" to a bool.
873 */
874 int
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100875check_script_var_type(
876 typval_T *dest,
877 typval_T *value,
878 char_u *name,
879 where_T where)
Bram Moolenaar10c65862020-10-08 21:16:42 +0200880{
881 svar_T *sv = find_typval_in_script(dest);
882 int ret;
883
884 if (sv != NULL)
885 {
Bram Moolenaar08251752021-01-11 21:20:18 +0100886 if (sv->sv_const != 0)
Bram Moolenaar10c65862020-10-08 21:16:42 +0200887 {
888 semsg(_(e_readonlyvar), name);
889 return FAIL;
890 }
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100891 ret = check_typval_type(sv->sv_type, value, where);
Bram Moolenaar10c65862020-10-08 21:16:42 +0200892 if (ret == OK && need_convert_to_bool(sv->sv_type, value))
893 {
894 int val = tv2bool(value);
895
896 clear_tv(value);
897 value->v_type = VAR_BOOL;
898 value->v_lock = 0;
899 value->vval.v_number = val ? VVAL_TRUE : VVAL_FALSE;
900 }
901 return ret;
902 }
903
Bram Moolenaarc785b9a2020-06-19 18:34:15 +0200904 return OK; // not really
Bram Moolenaar34db91f2020-06-13 19:00:10 +0200905}
Bram Moolenaarc82a5b52020-06-13 18:09:19 +0200906
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100907#endif // FEAT_EVAL