blob: a80fb352d93e988e48bad7c903f0df8566787664 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaarcd524592016-07-17 14:57:05 +02002 *
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 * dict.c: Dictionary support
12 */
Bram Moolenaarcd524592016-07-17 14:57:05 +020013
14#include "vim.h"
15
16#if defined(FEAT_EVAL) || defined(PROTO)
17
18/* List head for garbage collection. Although there can be a reference loop
19 * from partial to dict to partial, we don't need to keep track of the partial,
20 * since it will get freed when the dict is unused and gets freed. */
21static dict_T *first_dict = NULL; /* list of all dicts */
22
23/*
24 * Allocate an empty header for a dictionary.
25 */
26 dict_T *
27dict_alloc(void)
28{
29 dict_T *d;
30
Bram Moolenaaradc67142019-06-22 01:40:42 +020031 d = ALLOC_CLEAR_ONE(dict_T);
Bram Moolenaarcd524592016-07-17 14:57:05 +020032 if (d != NULL)
33 {
34 /* Add the dict to the list of dicts for garbage collection. */
35 if (first_dict != NULL)
36 first_dict->dv_used_prev = d;
37 d->dv_used_next = first_dict;
38 d->dv_used_prev = NULL;
39 first_dict = d;
40
41 hash_init(&d->dv_hashtab);
42 d->dv_lock = 0;
43 d->dv_scope = 0;
44 d->dv_refcount = 0;
45 d->dv_copyID = 0;
46 }
47 return d;
48}
49
Bram Moolenaarf49cc602018-11-11 15:21:05 +010050/*
51 * dict_alloc() with an ID for alloc_fail().
52 */
53 dict_T *
54dict_alloc_id(alloc_id_T id UNUSED)
55{
56#ifdef FEAT_EVAL
Bram Moolenaar51e14382019-05-25 20:21:28 +020057 if (alloc_fail_id == id && alloc_does_fail(sizeof(list_T)))
Bram Moolenaarf49cc602018-11-11 15:21:05 +010058 return NULL;
59#endif
60 return (dict_alloc());
61}
62
Bram Moolenaar7e1652c2017-12-16 18:27:02 +010063 dict_T *
64dict_alloc_lock(int lock)
65{
66 dict_T *d = dict_alloc();
67
68 if (d != NULL)
69 d->dv_lock = lock;
70 return d;
71}
72
Bram Moolenaarcd524592016-07-17 14:57:05 +020073/*
74 * Allocate an empty dict for a return value.
75 * Returns OK or FAIL.
76 */
77 int
78rettv_dict_alloc(typval_T *rettv)
79{
Bram Moolenaar7e1652c2017-12-16 18:27:02 +010080 dict_T *d = dict_alloc_lock(0);
Bram Moolenaarcd524592016-07-17 14:57:05 +020081
82 if (d == NULL)
83 return FAIL;
84
Bram Moolenaar45cf6e92017-04-30 20:25:19 +020085 rettv_dict_set(rettv, d);
Bram Moolenaarcd524592016-07-17 14:57:05 +020086 return OK;
87}
88
89/*
Bram Moolenaar45cf6e92017-04-30 20:25:19 +020090 * Set a dictionary as the return value
91 */
92 void
93rettv_dict_set(typval_T *rettv, dict_T *d)
94{
95 rettv->v_type = VAR_DICT;
96 rettv->vval.v_dict = d;
97 if (d != NULL)
98 ++d->dv_refcount;
99}
100
101/*
Bram Moolenaarcd524592016-07-17 14:57:05 +0200102 * Free a Dictionary, including all non-container items it contains.
103 * Ignores the reference count.
104 */
Bram Moolenaar7e1652c2017-12-16 18:27:02 +0100105 void
Bram Moolenaarcd524592016-07-17 14:57:05 +0200106dict_free_contents(dict_T *d)
107{
108 int todo;
109 hashitem_T *hi;
110 dictitem_T *di;
111
112 /* Lock the hashtab, we don't want it to resize while freeing items. */
113 hash_lock(&d->dv_hashtab);
114 todo = (int)d->dv_hashtab.ht_used;
115 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
116 {
117 if (!HASHITEM_EMPTY(hi))
118 {
119 /* Remove the item before deleting it, just in case there is
120 * something recursive causing trouble. */
121 di = HI2DI(hi);
122 hash_remove(&d->dv_hashtab, hi);
Bram Moolenaar95c526e2017-02-25 14:59:34 +0100123 dictitem_free(di);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200124 --todo;
125 }
126 }
Bram Moolenaar7e1652c2017-12-16 18:27:02 +0100127
128 /* The hashtab is still locked, it has to be re-initialized anyway */
Bram Moolenaarcd524592016-07-17 14:57:05 +0200129 hash_clear(&d->dv_hashtab);
130}
131
132 static void
133dict_free_dict(dict_T *d)
134{
135 /* Remove the dict from the list of dicts for garbage collection. */
136 if (d->dv_used_prev == NULL)
137 first_dict = d->dv_used_next;
138 else
139 d->dv_used_prev->dv_used_next = d->dv_used_next;
140 if (d->dv_used_next != NULL)
141 d->dv_used_next->dv_used_prev = d->dv_used_prev;
142 vim_free(d);
143}
144
145 static void
146dict_free(dict_T *d)
147{
148 if (!in_free_unref_items)
149 {
150 dict_free_contents(d);
151 dict_free_dict(d);
152 }
153}
154
155/*
156 * Unreference a Dictionary: decrement the reference count and free it when it
157 * becomes zero.
158 */
159 void
160dict_unref(dict_T *d)
161{
162 if (d != NULL && --d->dv_refcount <= 0)
163 dict_free(d);
164}
165
166/*
167 * Go through the list of dicts and free items without the copyID.
168 * Returns TRUE if something was freed.
169 */
170 int
171dict_free_nonref(int copyID)
172{
173 dict_T *dd;
174 int did_free = FALSE;
175
176 for (dd = first_dict; dd != NULL; dd = dd->dv_used_next)
177 if ((dd->dv_copyID & COPYID_MASK) != (copyID & COPYID_MASK))
178 {
179 /* Free the Dictionary and ordinary items it contains, but don't
180 * recurse into Lists and Dictionaries, they will be in the list
181 * of dicts or list of lists. */
182 dict_free_contents(dd);
183 did_free = TRUE;
184 }
185 return did_free;
186}
187
188 void
189dict_free_items(int copyID)
190{
191 dict_T *dd, *dd_next;
192
193 for (dd = first_dict; dd != NULL; dd = dd_next)
194 {
195 dd_next = dd->dv_used_next;
196 if ((dd->dv_copyID & COPYID_MASK) != (copyID & COPYID_MASK))
197 dict_free_dict(dd);
198 }
199}
200
201/*
202 * Allocate a Dictionary item.
203 * The "key" is copied to the new item.
Bram Moolenaarc89d4b32018-07-08 17:19:02 +0200204 * Note that the type and value of the item "di_tv" still needs to be
205 * initialized!
Bram Moolenaarcd524592016-07-17 14:57:05 +0200206 * Returns NULL when out of memory.
207 */
208 dictitem_T *
209dictitem_alloc(char_u *key)
210{
211 dictitem_T *di;
212
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200213 di = alloc(sizeof(dictitem_T) + STRLEN(key));
Bram Moolenaarcd524592016-07-17 14:57:05 +0200214 if (di != NULL)
215 {
216 STRCPY(di->di_key, key);
217 di->di_flags = DI_FLAGS_ALLOC;
Bram Moolenaarc89d4b32018-07-08 17:19:02 +0200218 di->di_tv.v_lock = 0;
Bram Moolenaarcd524592016-07-17 14:57:05 +0200219 }
220 return di;
221}
222
223/*
224 * Make a copy of a Dictionary item.
225 */
226 static dictitem_T *
227dictitem_copy(dictitem_T *org)
228{
229 dictitem_T *di;
230
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200231 di = alloc(sizeof(dictitem_T) + STRLEN(org->di_key));
Bram Moolenaarcd524592016-07-17 14:57:05 +0200232 if (di != NULL)
233 {
234 STRCPY(di->di_key, org->di_key);
235 di->di_flags = DI_FLAGS_ALLOC;
236 copy_tv(&org->di_tv, &di->di_tv);
237 }
238 return di;
239}
240
241/*
242 * Remove item "item" from Dictionary "dict" and free it.
243 */
244 void
245dictitem_remove(dict_T *dict, dictitem_T *item)
246{
247 hashitem_T *hi;
248
249 hi = hash_find(&dict->dv_hashtab, item->di_key);
250 if (HASHITEM_EMPTY(hi))
Bram Moolenaar95f09602016-11-10 20:01:45 +0100251 internal_error("dictitem_remove()");
Bram Moolenaarcd524592016-07-17 14:57:05 +0200252 else
253 hash_remove(&dict->dv_hashtab, hi);
254 dictitem_free(item);
255}
256
257/*
258 * Free a dict item. Also clears the value.
259 */
260 void
261dictitem_free(dictitem_T *item)
262{
263 clear_tv(&item->di_tv);
264 if (item->di_flags & DI_FLAGS_ALLOC)
265 vim_free(item);
266}
267
268/*
269 * Make a copy of dict "d". Shallow if "deep" is FALSE.
270 * The refcount of the new dict is set to 1.
271 * See item_copy() for "copyID".
272 * Returns NULL when out of memory.
273 */
274 dict_T *
275dict_copy(dict_T *orig, int deep, int copyID)
276{
277 dict_T *copy;
278 dictitem_T *di;
279 int todo;
280 hashitem_T *hi;
281
282 if (orig == NULL)
283 return NULL;
284
285 copy = dict_alloc();
286 if (copy != NULL)
287 {
288 if (copyID != 0)
289 {
290 orig->dv_copyID = copyID;
291 orig->dv_copydict = copy;
292 }
293 todo = (int)orig->dv_hashtab.ht_used;
294 for (hi = orig->dv_hashtab.ht_array; todo > 0 && !got_int; ++hi)
295 {
296 if (!HASHITEM_EMPTY(hi))
297 {
298 --todo;
299
300 di = dictitem_alloc(hi->hi_key);
301 if (di == NULL)
302 break;
303 if (deep)
304 {
305 if (item_copy(&HI2DI(hi)->di_tv, &di->di_tv, deep,
306 copyID) == FAIL)
307 {
308 vim_free(di);
309 break;
310 }
311 }
312 else
313 copy_tv(&HI2DI(hi)->di_tv, &di->di_tv);
314 if (dict_add(copy, di) == FAIL)
315 {
316 dictitem_free(di);
317 break;
318 }
319 }
320 }
321
322 ++copy->dv_refcount;
323 if (todo > 0)
324 {
325 dict_unref(copy);
326 copy = NULL;
327 }
328 }
329
330 return copy;
331}
332
333/*
334 * Add item "item" to Dictionary "d".
335 * Returns FAIL when out of memory and when key already exists.
336 */
337 int
338dict_add(dict_T *d, dictitem_T *item)
339{
340 return hash_add(&d->dv_hashtab, item->di_key);
341}
342
343/*
Bram Moolenaard7f246c2019-04-08 18:15:41 +0200344 * Add a number or special entry to dictionary "d".
Bram Moolenaarcd524592016-07-17 14:57:05 +0200345 * Returns FAIL when out of memory and when key already exists.
346 */
Bram Moolenaard7f246c2019-04-08 18:15:41 +0200347 static int
348dict_add_number_special(dict_T *d, char *key, varnumber_T nr, int special)
Bram Moolenaarcd524592016-07-17 14:57:05 +0200349{
350 dictitem_T *item;
351
352 item = dictitem_alloc((char_u *)key);
353 if (item == NULL)
354 return FAIL;
Bram Moolenaard7f246c2019-04-08 18:15:41 +0200355 item->di_tv.v_type = special ? VAR_SPECIAL : VAR_NUMBER;
Bram Moolenaare0be1672018-07-08 16:50:37 +0200356 item->di_tv.vval.v_number = nr;
357 if (dict_add(d, item) == FAIL)
Bram Moolenaarcd524592016-07-17 14:57:05 +0200358 {
Bram Moolenaare0be1672018-07-08 16:50:37 +0200359 dictitem_free(item);
360 return FAIL;
Bram Moolenaarcd524592016-07-17 14:57:05 +0200361 }
Bram Moolenaare0be1672018-07-08 16:50:37 +0200362 return OK;
363}
364
365/*
Bram Moolenaard7f246c2019-04-08 18:15:41 +0200366 * Add a number entry to dictionary "d".
367 * Returns FAIL when out of memory and when key already exists.
368 */
369 int
370dict_add_number(dict_T *d, char *key, varnumber_T nr)
371{
372 return dict_add_number_special(d, key, nr, FALSE);
373}
374
375/*
376 * Add a special entry to dictionary "d".
377 * Returns FAIL when out of memory and when key already exists.
378 */
379 int
380dict_add_special(dict_T *d, char *key, varnumber_T nr)
381{
382 return dict_add_number_special(d, key, nr, TRUE);
383}
384
385/*
Bram Moolenaare0be1672018-07-08 16:50:37 +0200386 * Add a string entry to dictionary "d".
387 * Returns FAIL when out of memory and when key already exists.
388 */
389 int
390dict_add_string(dict_T *d, char *key, char_u *str)
391{
Bram Moolenaare6fdf792018-12-26 22:57:42 +0100392 return dict_add_string_len(d, key, str, -1);
393}
394
395/*
396 * Add a string entry to dictionary "d".
397 * "str" will be copied to allocated memory.
398 * When "len" is -1 use the whole string, otherwise only this many bytes.
399 * Returns FAIL when out of memory and when key already exists.
400 */
401 int
402dict_add_string_len(dict_T *d, char *key, char_u *str, int len)
403{
Bram Moolenaare0be1672018-07-08 16:50:37 +0200404 dictitem_T *item;
Bram Moolenaare6fdf792018-12-26 22:57:42 +0100405 char_u *val = NULL;
Bram Moolenaare0be1672018-07-08 16:50:37 +0200406
407 item = dictitem_alloc((char_u *)key);
408 if (item == NULL)
409 return FAIL;
Bram Moolenaare0be1672018-07-08 16:50:37 +0200410 item->di_tv.v_type = VAR_STRING;
Bram Moolenaare6fdf792018-12-26 22:57:42 +0100411 if (str != NULL)
412 {
413 if (len == -1)
414 val = vim_strsave(str);
415 else
416 val = vim_strnsave(str, len);
417 }
418 item->di_tv.vval.v_string = val;
Bram Moolenaarcd524592016-07-17 14:57:05 +0200419 if (dict_add(d, item) == FAIL)
420 {
421 dictitem_free(item);
422 return FAIL;
423 }
424 return OK;
425}
426
427/*
428 * Add a list entry to dictionary "d".
429 * Returns FAIL when out of memory and when key already exists.
430 */
431 int
432dict_add_list(dict_T *d, char *key, list_T *list)
433{
434 dictitem_T *item;
435
436 item = dictitem_alloc((char_u *)key);
437 if (item == NULL)
438 return FAIL;
Bram Moolenaarcd524592016-07-17 14:57:05 +0200439 item->di_tv.v_type = VAR_LIST;
440 item->di_tv.vval.v_list = list;
Bram Moolenaar42f45b82017-03-14 22:17:14 +0100441 ++list->lv_refcount;
Bram Moolenaarcd524592016-07-17 14:57:05 +0200442 if (dict_add(d, item) == FAIL)
443 {
444 dictitem_free(item);
445 return FAIL;
446 }
Bram Moolenaarcd524592016-07-17 14:57:05 +0200447 return OK;
448}
449
450/*
Bram Moolenaarae943152019-06-16 22:54:14 +0200451 * Add a callback to dictionary "d".
452 * Returns FAIL when out of memory and when key already exists.
453 */
454 int
455dict_add_callback(dict_T *d, char *key, callback_T *cb)
456{
457 dictitem_T *item;
458
459 item = dictitem_alloc((char_u *)key);
460 if (item == NULL)
461 return FAIL;
462 put_callback(cb, &item->di_tv);
463 if (dict_add(d, item) == FAIL)
464 {
465 dictitem_free(item);
466 return FAIL;
467 }
468 return OK;
469}
470
471/*
Bram Moolenaar45e18cb2019-04-28 18:05:35 +0200472 * Initializes "iter" for iterating over dictionary items with
473 * dict_iterate_next().
474 * If "var" is not a Dict or an empty Dict then there will be nothing to
475 * iterate over, no error is given.
476 * NOTE: The dictionary must not change until iterating is finished!
477 */
478 void
479dict_iterate_start(typval_T *var, dict_iterator_T *iter)
480{
481 if (var->v_type != VAR_DICT || var->vval.v_dict == NULL)
482 iter->dit_todo = 0;
483 else
484 {
485 dict_T *d = var->vval.v_dict;
486
487 iter->dit_todo = d->dv_hashtab.ht_used;
488 iter->dit_hi = d->dv_hashtab.ht_array;
489 }
490}
491
492/*
493 * Iterate over the items referred to by "iter". It should be initialized with
494 * dict_iterate_start().
495 * Returns a pointer to the key.
496 * "*tv_result" is set to point to the value for that key.
497 * If there are no more items, NULL is returned.
498 */
499 char_u *
500dict_iterate_next(dict_iterator_T *iter, typval_T **tv_result)
501{
502 dictitem_T *di;
503 char_u *result;
504
505 if (iter->dit_todo == 0)
506 return NULL;
507
508 while (HASHITEM_EMPTY(iter->dit_hi))
509 ++iter->dit_hi;
510
511 di = HI2DI(iter->dit_hi);
512 result = di->di_key;
513 *tv_result = &di->di_tv;
514
515 --iter->dit_todo;
516 ++iter->dit_hi;
517 return result;
518}
519
520/*
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +0200521 * Add a dict entry to dictionary "d".
522 * Returns FAIL when out of memory and when key already exists.
523 */
524 int
525dict_add_dict(dict_T *d, char *key, dict_T *dict)
526{
527 dictitem_T *item;
528
529 item = dictitem_alloc((char_u *)key);
530 if (item == NULL)
531 return FAIL;
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +0200532 item->di_tv.v_type = VAR_DICT;
533 item->di_tv.vval.v_dict = dict;
Bram Moolenaar42f45b82017-03-14 22:17:14 +0100534 ++dict->dv_refcount;
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +0200535 if (dict_add(d, item) == FAIL)
536 {
537 dictitem_free(item);
538 return FAIL;
539 }
Bram Moolenaarb5ae48e2016-08-12 22:23:25 +0200540 return OK;
541}
542
543/*
Bram Moolenaarcd524592016-07-17 14:57:05 +0200544 * Get the number of items in a Dictionary.
545 */
546 long
547dict_len(dict_T *d)
548{
549 if (d == NULL)
550 return 0L;
551 return (long)d->dv_hashtab.ht_used;
552}
553
554/*
555 * Find item "key[len]" in Dictionary "d".
556 * If "len" is negative use strlen(key).
557 * Returns NULL when not found.
558 */
559 dictitem_T *
560dict_find(dict_T *d, char_u *key, int len)
561{
562#define AKEYLEN 200
563 char_u buf[AKEYLEN];
564 char_u *akey;
565 char_u *tofree = NULL;
566 hashitem_T *hi;
567
568 if (d == NULL)
569 return NULL;
570 if (len < 0)
571 akey = key;
572 else if (len >= AKEYLEN)
573 {
574 tofree = akey = vim_strnsave(key, len);
575 if (akey == NULL)
576 return NULL;
577 }
578 else
579 {
580 /* Avoid a malloc/free by using buf[]. */
581 vim_strncpy(buf, key, len);
582 akey = buf;
583 }
584
585 hi = hash_find(&d->dv_hashtab, akey);
586 vim_free(tofree);
587 if (HASHITEM_EMPTY(hi))
588 return NULL;
589 return HI2DI(hi);
590}
591
592/*
593 * Get a string item from a dictionary.
594 * When "save" is TRUE allocate memory for it.
Bram Moolenaar7dc5e2e2016-08-05 22:22:06 +0200595 * When FALSE a shared buffer is used, can only be used once!
Bram Moolenaarcd524592016-07-17 14:57:05 +0200596 * Returns NULL if the entry doesn't exist or out of memory.
597 */
598 char_u *
Bram Moolenaar8f667172018-12-14 15:38:31 +0100599dict_get_string(dict_T *d, char_u *key, int save)
Bram Moolenaarcd524592016-07-17 14:57:05 +0200600{
601 dictitem_T *di;
602 char_u *s;
603
604 di = dict_find(d, key, -1);
605 if (di == NULL)
606 return NULL;
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100607 s = tv_get_string(&di->di_tv);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200608 if (save && s != NULL)
609 s = vim_strsave(s);
610 return s;
611}
612
613/*
614 * Get a number item from a dictionary.
615 * Returns 0 if the entry doesn't exist.
616 */
617 varnumber_T
Bram Moolenaar8f667172018-12-14 15:38:31 +0100618dict_get_number(dict_T *d, char_u *key)
Bram Moolenaarcd524592016-07-17 14:57:05 +0200619{
620 dictitem_T *di;
621
622 di = dict_find(d, key, -1);
623 if (di == NULL)
624 return 0;
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100625 return tv_get_number(&di->di_tv);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200626}
627
628/*
Bram Moolenaarb0ebbda2019-06-02 16:51:21 +0200629 * Get a number item from a dictionary.
630 * Returns 0 if the entry doesn't exist.
631 * Give an error if the entry is not a number.
632 */
633 varnumber_T
634dict_get_number_check(dict_T *d, char_u *key)
635{
636 dictitem_T *di;
637
638 di = dict_find(d, key, -1);
639 if (di == NULL)
640 return 0;
641 if (di->di_tv.v_type != VAR_NUMBER)
642 {
643 semsg(_(e_invarg2), tv_get_string(&di->di_tv));
644 return 0;
645 }
646 return tv_get_number(&di->di_tv);
647}
648
649/*
Bram Moolenaarcd524592016-07-17 14:57:05 +0200650 * Return an allocated string with the string representation of a Dictionary.
651 * May return NULL.
652 */
653 char_u *
654dict2string(typval_T *tv, int copyID, int restore_copyID)
655{
656 garray_T ga;
657 int first = TRUE;
658 char_u *tofree;
659 char_u numbuf[NUMBUFLEN];
660 hashitem_T *hi;
661 char_u *s;
662 dict_T *d;
663 int todo;
664
665 if ((d = tv->vval.v_dict) == NULL)
666 return NULL;
667 ga_init2(&ga, (int)sizeof(char), 80);
668 ga_append(&ga, '{');
669
670 todo = (int)d->dv_hashtab.ht_used;
671 for (hi = d->dv_hashtab.ht_array; todo > 0 && !got_int; ++hi)
672 {
673 if (!HASHITEM_EMPTY(hi))
674 {
675 --todo;
676
677 if (first)
678 first = FALSE;
679 else
680 ga_concat(&ga, (char_u *)", ");
681
682 tofree = string_quote(hi->hi_key, FALSE);
683 if (tofree != NULL)
684 {
685 ga_concat(&ga, tofree);
686 vim_free(tofree);
687 }
688 ga_concat(&ga, (char_u *)": ");
689 s = echo_string_core(&HI2DI(hi)->di_tv, &tofree, numbuf, copyID,
690 FALSE, restore_copyID, TRUE);
691 if (s != NULL)
692 ga_concat(&ga, s);
693 vim_free(tofree);
694 if (s == NULL || did_echo_string_emsg)
695 break;
696 line_breakcheck();
697
698 }
699 }
700 if (todo > 0)
701 {
702 vim_free(ga.ga_data);
703 return NULL;
704 }
705
706 ga_append(&ga, '}');
707 ga_append(&ga, NUL);
708 return (char_u *)ga.ga_data;
709}
710
711/*
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200712 * Get the key for *{key: val} into "tv" and advance "arg".
713 * Return FAIL when there is no valid key.
714 */
715 static int
716get_literal_key(char_u **arg, typval_T *tv)
717{
718 char_u *p;
719
720 if (!ASCII_ISALNUM(**arg) && **arg != '_' && **arg != '-')
721 return FAIL;
722
723 for (p = *arg; ASCII_ISALNUM(*p) || *p == '_' || *p == '-'; ++p)
724 ;
725 tv->v_type = VAR_STRING;
726 tv->vval.v_string = vim_strnsave(*arg, (int)(p - *arg));
727
728 *arg = skipwhite(p);
729 return OK;
730}
731
732/*
Bram Moolenaarcd524592016-07-17 14:57:05 +0200733 * Allocate a variable for a Dictionary and fill it from "*arg".
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200734 * "literal" is TRUE for *{key: val}
Bram Moolenaarcd524592016-07-17 14:57:05 +0200735 * Return OK or FAIL. Returns NOTDONE for {expr}.
736 */
737 int
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200738dict_get_tv(char_u **arg, typval_T *rettv, int evaluate, int literal)
Bram Moolenaarcd524592016-07-17 14:57:05 +0200739{
740 dict_T *d = NULL;
741 typval_T tvkey;
742 typval_T tv;
743 char_u *key = NULL;
744 dictitem_T *item;
745 char_u *start = skipwhite(*arg + 1);
746 char_u buf[NUMBUFLEN];
747
748 /*
749 * First check if it's not a curly-braces thing: {expr}.
750 * Must do this without evaluating, otherwise a function may be called
751 * twice. Unfortunately this means we need to call eval1() twice for the
752 * first item.
753 * But {} is an empty Dictionary.
754 */
755 if (*start != '}')
756 {
757 if (eval1(&start, &tv, FALSE) == FAIL) /* recursive! */
758 return FAIL;
759 if (*start == '}')
760 return NOTDONE;
761 }
762
763 if (evaluate)
764 {
765 d = dict_alloc();
766 if (d == NULL)
767 return FAIL;
768 }
769 tvkey.v_type = VAR_UNKNOWN;
770 tv.v_type = VAR_UNKNOWN;
771
772 *arg = skipwhite(*arg + 1);
773 while (**arg != '}' && **arg != NUL)
774 {
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200775 if ((literal
776 ? get_literal_key(arg, &tvkey)
777 : eval1(arg, &tvkey, evaluate)) == FAIL) // recursive!
Bram Moolenaarcd524592016-07-17 14:57:05 +0200778 goto failret;
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200779
Bram Moolenaarcd524592016-07-17 14:57:05 +0200780 if (**arg != ':')
781 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100782 semsg(_("E720: Missing colon in Dictionary: %s"), *arg);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200783 clear_tv(&tvkey);
784 goto failret;
785 }
786 if (evaluate)
787 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100788 key = tv_get_string_buf_chk(&tvkey, buf);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200789 if (key == NULL)
790 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100791 /* "key" is NULL when tv_get_string_buf_chk() gave an errmsg */
Bram Moolenaarcd524592016-07-17 14:57:05 +0200792 clear_tv(&tvkey);
793 goto failret;
794 }
795 }
796
797 *arg = skipwhite(*arg + 1);
798 if (eval1(arg, &tv, evaluate) == FAIL) /* recursive! */
799 {
800 if (evaluate)
801 clear_tv(&tvkey);
802 goto failret;
803 }
804 if (evaluate)
805 {
806 item = dict_find(d, key, -1);
807 if (item != NULL)
808 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100809 semsg(_("E721: Duplicate key in Dictionary: \"%s\""), key);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200810 clear_tv(&tvkey);
811 clear_tv(&tv);
812 goto failret;
813 }
814 item = dictitem_alloc(key);
815 clear_tv(&tvkey);
816 if (item != NULL)
817 {
818 item->di_tv = tv;
819 item->di_tv.v_lock = 0;
820 if (dict_add(d, item) == FAIL)
821 dictitem_free(item);
822 }
823 }
824
825 if (**arg == '}')
826 break;
827 if (**arg != ',')
828 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100829 semsg(_("E722: Missing comma in Dictionary: %s"), *arg);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200830 goto failret;
831 }
832 *arg = skipwhite(*arg + 1);
833 }
834
835 if (**arg != '}')
836 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100837 semsg(_("E723: Missing end of Dictionary '}': %s"), *arg);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200838failret:
Bram Moolenaaradc67142019-06-22 01:40:42 +0200839 if (d != NULL)
Bram Moolenaarcd524592016-07-17 14:57:05 +0200840 dict_free(d);
841 return FAIL;
842 }
843
844 *arg = skipwhite(*arg + 1);
845 if (evaluate)
Bram Moolenaar45cf6e92017-04-30 20:25:19 +0200846 rettv_dict_set(rettv, d);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200847
848 return OK;
849}
850
851/*
852 * Go over all entries in "d2" and add them to "d1".
853 * When "action" is "error" then a duplicate key is an error.
854 * When "action" is "force" then a duplicate key is overwritten.
855 * Otherwise duplicate keys are ignored ("action" is "keep").
856 */
857 void
858dict_extend(dict_T *d1, dict_T *d2, char_u *action)
859{
860 dictitem_T *di1;
861 hashitem_T *hi2;
862 int todo;
863 char_u *arg_errmsg = (char_u *)N_("extend() argument");
864
865 todo = (int)d2->dv_hashtab.ht_used;
866 for (hi2 = d2->dv_hashtab.ht_array; todo > 0; ++hi2)
867 {
868 if (!HASHITEM_EMPTY(hi2))
869 {
870 --todo;
871 di1 = dict_find(d1, hi2->hi_key, -1);
872 if (d1->dv_scope != 0)
873 {
874 /* Disallow replacing a builtin function in l: and g:.
875 * Check the key to be valid when adding to any scope. */
876 if (d1->dv_scope == VAR_DEF_SCOPE
877 && HI2DI(hi2)->di_tv.v_type == VAR_FUNC
878 && var_check_func_name(hi2->hi_key, di1 == NULL))
879 break;
880 if (!valid_varname(hi2->hi_key))
881 break;
882 }
883 if (di1 == NULL)
884 {
885 di1 = dictitem_copy(HI2DI(hi2));
886 if (di1 != NULL && dict_add(d1, di1) == FAIL)
887 dictitem_free(di1);
888 }
889 else if (*action == 'e')
890 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100891 semsg(_("E737: Key already exists: %s"), hi2->hi_key);
Bram Moolenaarcd524592016-07-17 14:57:05 +0200892 break;
893 }
894 else if (*action == 'f' && HI2DI(hi2) != di1)
895 {
Bram Moolenaar05c00c02019-02-11 22:00:11 +0100896 if (var_check_lock(di1->di_tv.v_lock, arg_errmsg, TRUE)
897 || var_check_ro(di1->di_flags, arg_errmsg, TRUE))
Bram Moolenaarcd524592016-07-17 14:57:05 +0200898 break;
899 clear_tv(&di1->di_tv);
900 copy_tv(&HI2DI(hi2)->di_tv, &di1->di_tv);
901 }
902 }
903 }
904}
905
906/*
907 * Return the dictitem that an entry in a hashtable points to.
908 */
909 dictitem_T *
910dict_lookup(hashitem_T *hi)
911{
912 return HI2DI(hi);
913}
914
915/*
916 * Return TRUE when two dictionaries have exactly the same key/values.
917 */
918 int
919dict_equal(
920 dict_T *d1,
921 dict_T *d2,
922 int ic, /* ignore case for strings */
923 int recursive) /* TRUE when used recursively */
924{
925 hashitem_T *hi;
926 dictitem_T *item2;
927 int todo;
928
929 if (d1 == NULL && d2 == NULL)
930 return TRUE;
931 if (d1 == NULL || d2 == NULL)
932 return FALSE;
933 if (d1 == d2)
934 return TRUE;
935 if (dict_len(d1) != dict_len(d2))
936 return FALSE;
937
938 todo = (int)d1->dv_hashtab.ht_used;
939 for (hi = d1->dv_hashtab.ht_array; todo > 0; ++hi)
940 {
941 if (!HASHITEM_EMPTY(hi))
942 {
943 item2 = dict_find(d2, hi->hi_key, -1);
944 if (item2 == NULL)
945 return FALSE;
946 if (!tv_equal(&HI2DI(hi)->di_tv, &item2->di_tv, ic, recursive))
947 return FALSE;
948 --todo;
949 }
950 }
951 return TRUE;
952}
953
954/*
955 * Turn a dict into a list:
956 * "what" == 0: list of keys
957 * "what" == 1: list of values
958 * "what" == 2: list of items
959 */
960 void
961dict_list(typval_T *argvars, typval_T *rettv, int what)
962{
963 list_T *l2;
964 dictitem_T *di;
965 hashitem_T *hi;
966 listitem_T *li;
967 listitem_T *li2;
968 dict_T *d;
969 int todo;
970
971 if (argvars[0].v_type != VAR_DICT)
972 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100973 emsg(_(e_dictreq));
Bram Moolenaarcd524592016-07-17 14:57:05 +0200974 return;
975 }
976 if ((d = argvars[0].vval.v_dict) == NULL)
977 return;
978
979 if (rettv_list_alloc(rettv) == FAIL)
980 return;
981
982 todo = (int)d->dv_hashtab.ht_used;
983 for (hi = d->dv_hashtab.ht_array; todo > 0; ++hi)
984 {
985 if (!HASHITEM_EMPTY(hi))
986 {
987 --todo;
988 di = HI2DI(hi);
989
990 li = listitem_alloc();
991 if (li == NULL)
992 break;
993 list_append(rettv->vval.v_list, li);
994
995 if (what == 0)
996 {
997 /* keys() */
998 li->li_tv.v_type = VAR_STRING;
999 li->li_tv.v_lock = 0;
1000 li->li_tv.vval.v_string = vim_strsave(di->di_key);
1001 }
1002 else if (what == 1)
1003 {
1004 /* values() */
1005 copy_tv(&di->di_tv, &li->li_tv);
1006 }
1007 else
1008 {
1009 /* items() */
1010 l2 = list_alloc();
1011 li->li_tv.v_type = VAR_LIST;
1012 li->li_tv.v_lock = 0;
1013 li->li_tv.vval.v_list = l2;
1014 if (l2 == NULL)
1015 break;
1016 ++l2->lv_refcount;
1017
1018 li2 = listitem_alloc();
1019 if (li2 == NULL)
1020 break;
1021 list_append(l2, li2);
1022 li2->li_tv.v_type = VAR_STRING;
1023 li2->li_tv.v_lock = 0;
1024 li2->li_tv.vval.v_string = vim_strsave(di->di_key);
1025
1026 li2 = listitem_alloc();
1027 if (li2 == NULL)
1028 break;
1029 list_append(l2, li2);
1030 copy_tv(&di->di_tv, &li2->li_tv);
1031 }
1032 }
1033 }
1034}
1035
Bram Moolenaar7e1652c2017-12-16 18:27:02 +01001036/*
1037 * Make each item in the dict readonly (not the value of the item).
1038 */
1039 void
1040dict_set_items_ro(dict_T *di)
1041{
1042 int todo = (int)di->dv_hashtab.ht_used;
1043 hashitem_T *hi;
1044
1045 /* Set readonly */
1046 for (hi = di->dv_hashtab.ht_array; todo > 0 ; ++hi)
1047 {
1048 if (HASHITEM_EMPTY(hi))
1049 continue;
1050 --todo;
1051 HI2DI(hi)->di_flags |= DI_FLAGS_RO | DI_FLAGS_FIX;
1052 }
1053}
1054
Bram Moolenaarcd524592016-07-17 14:57:05 +02001055#endif /* defined(FEAT_EVAL) */