blob: 5a18f6c734e07c41de818990c8a01302a22b1e23 [file] [log] [blame]
Bram Moolenaar98aefe72018-12-13 22:20:09 +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/*
Bram Moolenaar45dd07f2019-05-15 22:45:37 +020011 * Text properties implementation. See ":help text-properties".
Bram Moolenaar98aefe72018-12-13 22:20:09 +010012 *
13 * TODO:
Bram Moolenaarb9c67a52019-01-01 19:49:20 +010014 * - Adjust text property column and length when text is inserted/deleted.
Bram Moolenaar4164bb22019-01-04 23:09:49 +010015 * -> a :substitute with a multi-line match
Bram Moolenaarf9e3e092019-01-13 23:38:42 +010016 * -> search for changed_bytes() from misc1.c
Bram Moolenaar80e737c2019-05-17 19:56:34 +020017 * -> search for mark_col_adjust()
Bram Moolenaarb9c67a52019-01-01 19:49:20 +010018 * - Perhaps we only need TP_FLAG_CONT_NEXT and can drop TP_FLAG_CONT_PREV?
Bram Moolenaar32aa1022019-11-02 22:54:41 +010019 * - Add an array for global_proptypes, to quickly lookup a prop type by ID
20 * - Add an array for b_proptypes, to quickly lookup a prop type by ID
Bram Moolenaarb56ac042018-12-28 23:22:40 +010021 * - Checking the text length to detect text properties is slow. Use a flag in
22 * the index, like DB_MARKED?
Bram Moolenaarb413d2e2018-12-25 23:15:46 +010023 * - Also test line2byte() with many lines, so that ml_updatechunk() is taken
24 * into account.
Bram Moolenaarc6663882019-02-22 19:14:54 +010025 * - Perhaps have a window-local option to disable highlighting from text
26 * properties?
Bram Moolenaar98aefe72018-12-13 22:20:09 +010027 */
28
29#include "vim.h"
30
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +010031#if defined(FEAT_PROP_POPUP) || defined(PROTO)
Bram Moolenaar98aefe72018-12-13 22:20:09 +010032
33/*
34 * In a hashtable item "hi_key" points to "pt_name" in a proptype_T.
35 * This avoids adding a pointer to the hashtable item.
36 * PT2HIKEY() converts a proptype pointer to a hashitem key pointer.
37 * HIKEY2PT() converts a hashitem key pointer to a proptype pointer.
38 * HI2PT() converts a hashitem pointer to a proptype pointer.
39 */
40#define PT2HIKEY(p) ((p)->pt_name)
41#define HIKEY2PT(p) ((proptype_T *)((p) - offsetof(proptype_T, pt_name)))
42#define HI2PT(hi) HIKEY2PT((hi)->hi_key)
43
44// The global text property types.
45static hashtab_T *global_proptypes = NULL;
46
47// The last used text property type ID.
48static int proptype_id = 0;
49
50static char_u e_type_not_exist[] = N_("E971: Property type %s does not exist");
51static char_u e_invalid_col[] = N_("E964: Invalid column number: %ld");
Bram Moolenaare3d31b02018-12-24 23:07:04 +010052static char_u e_invalid_lnum[] = N_("E966: Invalid line number: %ld");
Bram Moolenaar98aefe72018-12-13 22:20:09 +010053
54/*
55 * Find a property type by name, return the hashitem.
56 * Returns NULL if the item can't be found.
57 */
58 static hashitem_T *
59find_prop_hi(char_u *name, buf_T *buf)
60{
61 hashtab_T *ht;
62 hashitem_T *hi;
63
64 if (*name == NUL)
65 return NULL;
66 if (buf == NULL)
67 ht = global_proptypes;
68 else
69 ht = buf->b_proptypes;
70
71 if (ht == NULL)
72 return NULL;
73 hi = hash_find(ht, name);
74 if (HASHITEM_EMPTY(hi))
75 return NULL;
76 return hi;
77}
78
79/*
80 * Like find_prop_hi() but return the property type.
81 */
82 static proptype_T *
83find_prop(char_u *name, buf_T *buf)
84{
85 hashitem_T *hi = find_prop_hi(name, buf);
86
87 if (hi == NULL)
88 return NULL;
89 return HI2PT(hi);
90}
91
92/*
Bram Moolenaar12034e22019-08-25 22:25:02 +020093 * Get the prop type ID of "name".
94 * When not found return zero.
95 */
96 int
97find_prop_type_id(char_u *name, buf_T *buf)
98{
99 proptype_T *pt = find_prop(name, buf);
100
101 if (pt == NULL)
102 return 0;
103 return pt->pt_id;
104}
105
106/*
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100107 * Lookup a property type by name. First in "buf" and when not found in the
108 * global types.
109 * When not found gives an error message and returns NULL.
110 */
111 static proptype_T *
112lookup_prop_type(char_u *name, buf_T *buf)
113{
114 proptype_T *type = find_prop(name, buf);
115
116 if (type == NULL)
117 type = find_prop(name, NULL);
118 if (type == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100119 semsg(_(e_type_not_exist), name);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100120 return type;
121}
122
123/*
124 * Get an optional "bufnr" item from the dict in "arg".
125 * When the argument is not used or "bufnr" is not present then "buf" is
126 * unchanged.
127 * If "bufnr" is valid or not present return OK.
Bram Moolenaar32aa1022019-11-02 22:54:41 +0100128 * When "arg" is not a dict or "bufnr" is invalid return FAIL.
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100129 */
130 static int
131get_bufnr_from_arg(typval_T *arg, buf_T **buf)
132{
133 dictitem_T *di;
134
135 if (arg->v_type != VAR_DICT)
136 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100137 emsg(_(e_dictreq));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100138 return FAIL;
139 }
140 if (arg->vval.v_dict == NULL)
141 return OK; // NULL dict is like an empty dict
142 di = dict_find(arg->vval.v_dict, (char_u *)"bufnr", -1);
143 if (di != NULL)
144 {
Bram Moolenaarf0884c52019-05-24 21:22:29 +0200145 *buf = get_buf_arg(&di->di_tv);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100146 if (*buf == NULL)
147 return FAIL;
148 }
149 return OK;
150}
151
152/*
153 * prop_add({lnum}, {col}, {props})
154 */
155 void
156f_prop_add(typval_T *argvars, typval_T *rettv UNUSED)
157{
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100158 linenr_T start_lnum;
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100159 colnr_T start_col;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100160
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100161 start_lnum = tv_get_number(&argvars[0]);
162 start_col = tv_get_number(&argvars[1]);
163 if (start_col < 1)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100164 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100165 semsg(_(e_invalid_col), (long)start_col);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100166 return;
167 }
168 if (argvars[2].v_type != VAR_DICT)
169 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100170 emsg(_(e_dictreq));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100171 return;
172 }
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200173
174 prop_add_common(start_lnum, start_col, argvars[2].vval.v_dict,
175 curbuf, &argvars[2]);
176}
177
178/*
179 * Shared between prop_add() and popup_create().
180 * "dict_arg" is the function argument of a dict containing "bufnr".
181 * it is NULL for popup_create().
182 */
183 void
184prop_add_common(
185 linenr_T start_lnum,
186 colnr_T start_col,
187 dict_T *dict,
188 buf_T *default_buf,
189 typval_T *dict_arg)
190{
191 linenr_T lnum;
192 linenr_T end_lnum;
193 colnr_T end_col;
194 char_u *type_name;
195 proptype_T *type;
196 buf_T *buf = default_buf;
197 int id = 0;
198 char_u *newtext;
199 int proplen;
200 size_t textlen;
201 char_u *props = NULL;
202 char_u *newprops;
203 textprop_T tmp_prop;
204 int i;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100205
206 if (dict == NULL || dict_find(dict, (char_u *)"type", -1) == NULL)
207 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100208 emsg(_("E965: missing property type name"));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100209 return;
210 }
Bram Moolenaar8f667172018-12-14 15:38:31 +0100211 type_name = dict_get_string(dict, (char_u *)"type", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100212
213 if (dict_find(dict, (char_u *)"end_lnum", -1) != NULL)
214 {
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100215 end_lnum = dict_get_number(dict, (char_u *)"end_lnum");
216 if (end_lnum < start_lnum)
217 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100218 semsg(_(e_invargval), "end_lnum");
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100219 return;
220 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100221 }
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100222 else
223 end_lnum = start_lnum;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100224
225 if (dict_find(dict, (char_u *)"length", -1) != NULL)
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100226 {
227 long length = dict_get_number(dict, (char_u *)"length");
228
Bram Moolenaarb9c67a52019-01-01 19:49:20 +0100229 if (length < 0 || end_lnum > start_lnum)
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100230 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100231 semsg(_(e_invargval), "length");
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100232 return;
233 }
Bram Moolenaarb9c67a52019-01-01 19:49:20 +0100234 end_col = start_col + length;
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100235 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100236 else if (dict_find(dict, (char_u *)"end_col", -1) != NULL)
237 {
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100238 end_col = dict_get_number(dict, (char_u *)"end_col");
239 if (end_col <= 0)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100240 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100241 semsg(_(e_invargval), "end_col");
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100242 return;
243 }
244 }
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100245 else if (start_lnum == end_lnum)
246 end_col = start_col;
247 else
248 end_col = 1;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100249
250 if (dict_find(dict, (char_u *)"id", -1) != NULL)
Bram Moolenaar8f667172018-12-14 15:38:31 +0100251 id = dict_get_number(dict, (char_u *)"id");
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100252
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200253 if (dict_arg != NULL && get_bufnr_from_arg(dict_arg, &buf) == FAIL)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100254 return;
255
256 type = lookup_prop_type(type_name, buf);
257 if (type == NULL)
258 return;
259
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100260 if (start_lnum < 1 || start_lnum > buf->b_ml.ml_line_count)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100261 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100262 semsg(_(e_invalid_lnum), (long)start_lnum);
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100263 return;
264 }
265 if (end_lnum < start_lnum || end_lnum > buf->b_ml.ml_line_count)
266 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100267 semsg(_(e_invalid_lnum), (long)end_lnum);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100268 return;
269 }
270
Bram Moolenaard79eef22019-05-24 20:41:55 +0200271 if (buf->b_ml.ml_mfp == NULL)
Bram Moolenaar45311b52019-08-13 22:27:32 +0200272 {
273 emsg(_("E275: Cannot add text property to unloaded buffer"));
274 return;
275 }
Bram Moolenaard79eef22019-05-24 20:41:55 +0200276
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100277 for (lnum = start_lnum; lnum <= end_lnum; ++lnum)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100278 {
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100279 colnr_T col; // start column
280 long length; // in bytes
281
282 // Fetch the line to get the ml_line_len field updated.
283 proplen = get_text_props(buf, lnum, &props, TRUE);
284 textlen = buf->b_ml.ml_line_len - proplen * sizeof(textprop_T);
285
286 if (lnum == start_lnum)
287 col = start_col;
288 else
289 col = 1;
290 if (col - 1 > (colnr_T)textlen)
291 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100292 semsg(_(e_invalid_col), (long)start_col);
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100293 return;
294 }
295
296 if (lnum == end_lnum)
Bram Moolenaarb9c67a52019-01-01 19:49:20 +0100297 length = end_col - col;
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100298 else
Bram Moolenaar4b7214e2019-01-03 21:55:32 +0100299 length = (int)textlen - col + 1;
Bram Moolenaarb413d2e2018-12-25 23:15:46 +0100300 if (length > (long)textlen)
Bram Moolenaar4b7214e2019-01-03 21:55:32 +0100301 length = (int)textlen; // can include the end-of-line
Bram Moolenaarb9c67a52019-01-01 19:49:20 +0100302 if (length < 0)
303 length = 0; // zero-width property
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100304
Bram Moolenaar8e7d6222020-12-18 19:49:56 +0100305 // Allocate the new line with space for the new property.
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100306 newtext = alloc(buf->b_ml.ml_line_len + sizeof(textprop_T));
307 if (newtext == NULL)
308 return;
309 // Copy the text, including terminating NUL.
310 mch_memmove(newtext, buf->b_ml.ml_line_ptr, textlen);
311
312 // Find the index where to insert the new property.
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200313 // Since the text properties are not aligned properly when stored with
314 // the text, we need to copy them as bytes before using it as a struct.
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100315 for (i = 0; i < proplen; ++i)
316 {
317 mch_memmove(&tmp_prop, props + i * sizeof(textprop_T),
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200318 sizeof(textprop_T));
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100319 if (tmp_prop.tp_col >= col)
320 break;
321 }
322 newprops = newtext + textlen;
323 if (i > 0)
324 mch_memmove(newprops, props, sizeof(textprop_T) * i);
325
326 tmp_prop.tp_col = col;
327 tmp_prop.tp_len = length;
328 tmp_prop.tp_id = id;
329 tmp_prop.tp_type = type->pt_id;
330 tmp_prop.tp_flags = (lnum > start_lnum ? TP_FLAG_CONT_PREV : 0)
331 | (lnum < end_lnum ? TP_FLAG_CONT_NEXT : 0);
332 mch_memmove(newprops + i * sizeof(textprop_T), &tmp_prop,
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200333 sizeof(textprop_T));
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100334
335 if (i < proplen)
336 mch_memmove(newprops + (i + 1) * sizeof(textprop_T),
337 props + i * sizeof(textprop_T),
338 sizeof(textprop_T) * (proplen - i));
339
340 if (buf->b_ml.ml_flags & ML_LINE_DIRTY)
341 vim_free(buf->b_ml.ml_line_ptr);
342 buf->b_ml.ml_line_ptr = newtext;
343 buf->b_ml.ml_line_len += sizeof(textprop_T);
344 buf->b_ml.ml_flags |= ML_LINE_DIRTY;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100345 }
346
Bram Moolenaarb413d2e2018-12-25 23:15:46 +0100347 buf->b_has_textprop = TRUE; // this is never reset
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100348 redraw_buf_later(buf, NOT_VALID);
349}
350
351/*
Bram Moolenaarfb95e212018-12-14 12:18:11 +0100352 * Fetch the text properties for line "lnum" in buffer "buf".
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100353 * Returns the number of text properties and, when non-zero, a pointer to the
354 * first one in "props" (note that it is not aligned, therefore the char_u
355 * pointer).
356 */
357 int
358get_text_props(buf_T *buf, linenr_T lnum, char_u **props, int will_change)
359{
360 char_u *text;
361 size_t textlen;
362 size_t proplen;
363
Bram Moolenaarb413d2e2018-12-25 23:15:46 +0100364 // Be quick when no text property types have been defined or the buffer,
365 // unless we are adding one.
Bram Moolenaard79eef22019-05-24 20:41:55 +0200366 if ((!buf->b_has_textprop && !will_change) || buf->b_ml.ml_mfp == NULL)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100367 return 0;
368
369 // Fetch the line to get the ml_line_len field updated.
370 text = ml_get_buf(buf, lnum, will_change);
371 textlen = STRLEN(text) + 1;
372 proplen = buf->b_ml.ml_line_len - textlen;
373 if (proplen % sizeof(textprop_T) != 0)
374 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100375 iemsg(_("E967: text property info corrupted"));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100376 return 0;
377 }
378 if (proplen > 0)
379 *props = text + textlen;
Bram Moolenaar4efe73b2018-12-16 14:37:39 +0100380 return (int)(proplen / sizeof(textprop_T));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100381}
382
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200383/**
384 * Return the number of text properties on line "lnum" in the current buffer.
385 * When "only_starting" is true only text properties starting in this line will
386 * be considered.
387 */
388 int
389count_props(linenr_T lnum, int only_starting)
390{
391 char_u *props;
392 int proplen = get_text_props(curbuf, lnum, &props, 0);
393 int result = proplen;
394 int i;
395 textprop_T prop;
396
397 if (only_starting)
398 for (i = 0; i < proplen; ++i)
399 {
400 mch_memmove(&prop, props + i * sizeof(prop), sizeof(prop));
401 if (prop.tp_flags & TP_FLAG_CONT_PREV)
402 --result;
403 }
404 return result;
405}
406
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100407/*
Bram Moolenaar12034e22019-08-25 22:25:02 +0200408 * Find text property "type_id" in the visible lines of window "wp".
409 * Match "id" when it is > 0.
410 * Returns FAIL when not found.
411 */
412 int
413find_visible_prop(win_T *wp, int type_id, int id, textprop_T *prop,
414 linenr_T *found_lnum)
415{
416 linenr_T lnum;
417 char_u *props;
418 int count;
419 int i;
420
421 // w_botline may not have been updated yet.
Bram Moolenaar8e956362019-08-25 23:08:17 +0200422 validate_botline();
Bram Moolenaar12034e22019-08-25 22:25:02 +0200423 for (lnum = wp->w_topline; lnum < wp->w_botline; ++lnum)
424 {
425 count = get_text_props(wp->w_buffer, lnum, &props, FALSE);
426 for (i = 0; i < count; ++i)
427 {
428 mch_memmove(prop, props + i * sizeof(textprop_T),
429 sizeof(textprop_T));
430 if (prop->tp_type == type_id && (id <= 0 || prop->tp_id == id))
431 {
432 *found_lnum = lnum;
433 return OK;
434 }
435 }
436 }
437 return FAIL;
438}
439
440/*
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100441 * Set the text properties for line "lnum" to "props" with length "len".
442 * If "len" is zero text properties are removed, "props" is not used.
443 * Any existing text properties are dropped.
444 * Only works for the current buffer.
445 */
446 static void
447set_text_props(linenr_T lnum, char_u *props, int len)
448{
Bram Moolenaar8aef43b2019-01-08 20:14:35 +0100449 char_u *text;
450 char_u *newtext;
451 int textlen;
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100452
453 text = ml_get(lnum);
Bram Moolenaar8aef43b2019-01-08 20:14:35 +0100454 textlen = (int)STRLEN(text) + 1;
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100455 newtext = alloc(textlen + len);
456 if (newtext == NULL)
457 return;
458 mch_memmove(newtext, text, textlen);
459 if (len > 0)
460 mch_memmove(newtext + textlen, props, len);
461 if (curbuf->b_ml.ml_flags & ML_LINE_DIRTY)
462 vim_free(curbuf->b_ml.ml_line_ptr);
463 curbuf->b_ml.ml_line_ptr = newtext;
464 curbuf->b_ml.ml_line_len = textlen + len;
465 curbuf->b_ml.ml_flags |= ML_LINE_DIRTY;
466}
467
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100468 static proptype_T *
469find_type_by_id(hashtab_T *ht, int id)
470{
471 long todo;
472 hashitem_T *hi;
473
474 if (ht == NULL)
475 return NULL;
476
477 // TODO: Make this faster by keeping a list of types sorted on ID and use
478 // a binary search.
479
480 todo = (long)ht->ht_used;
481 for (hi = ht->ht_array; todo > 0; ++hi)
482 {
483 if (!HASHITEM_EMPTY(hi))
484 {
485 proptype_T *prop = HI2PT(hi);
486
487 if (prop->pt_id == id)
488 return prop;
489 --todo;
490 }
491 }
492 return NULL;
493}
494
495/*
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100496 * Fill 'dict' with text properties in 'prop'.
497 */
498 static void
499prop_fill_dict(dict_T *dict, textprop_T *prop, buf_T *buf)
500{
501 proptype_T *pt;
502
503 dict_add_number(dict, "col", prop->tp_col);
504 dict_add_number(dict, "length", prop->tp_len);
505 dict_add_number(dict, "id", prop->tp_id);
506 dict_add_number(dict, "start", !(prop->tp_flags & TP_FLAG_CONT_PREV));
507 dict_add_number(dict, "end", !(prop->tp_flags & TP_FLAG_CONT_NEXT));
508 pt = text_prop_type_by_id(buf, prop->tp_type);
509 if (pt != NULL)
510 dict_add_string(dict, "type", pt->pt_name);
511}
512
513/*
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100514 * Find a property type by ID in "buf" or globally.
515 * Returns NULL if not found.
516 */
517 proptype_T *
518text_prop_type_by_id(buf_T *buf, int id)
519{
520 proptype_T *type;
521
522 type = find_type_by_id(buf->b_proptypes, id);
523 if (type == NULL)
524 type = find_type_by_id(global_proptypes, id);
525 return type;
526}
527
528/*
529 * prop_clear({lnum} [, {lnum_end} [, {bufnr}]])
530 */
531 void
532f_prop_clear(typval_T *argvars, typval_T *rettv UNUSED)
533{
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100534 linenr_T start = tv_get_number(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100535 linenr_T end = start;
536 linenr_T lnum;
537 buf_T *buf = curbuf;
538
539 if (argvars[1].v_type != VAR_UNKNOWN)
540 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100541 end = tv_get_number(&argvars[1]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100542 if (argvars[2].v_type != VAR_UNKNOWN)
543 {
544 if (get_bufnr_from_arg(&argvars[2], &buf) == FAIL)
545 return;
546 }
547 }
548 if (start < 1 || end < 1)
549 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100550 emsg(_(e_invrange));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100551 return;
552 }
553
554 for (lnum = start; lnum <= end; ++lnum)
555 {
556 char_u *text;
557 size_t len;
558
559 if (lnum > buf->b_ml.ml_line_count)
560 break;
561 text = ml_get_buf(buf, lnum, FALSE);
562 len = STRLEN(text) + 1;
563 if ((size_t)buf->b_ml.ml_line_len > len)
564 {
565 if (!(buf->b_ml.ml_flags & ML_LINE_DIRTY))
566 {
567 char_u *newtext = vim_strsave(text);
568
569 // need to allocate the line now
570 if (newtext == NULL)
571 return;
572 buf->b_ml.ml_line_ptr = newtext;
573 buf->b_ml.ml_flags |= ML_LINE_DIRTY;
574 }
Bram Moolenaar4efe73b2018-12-16 14:37:39 +0100575 buf->b_ml.ml_line_len = (int)len;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100576 }
577 }
578 redraw_buf_later(buf, NOT_VALID);
579}
580
581/*
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100582 * prop_find({props} [, {direction}])
583 */
584 void
585f_prop_find(typval_T *argvars, typval_T *rettv)
586{
587 pos_T *cursor = &curwin->w_cursor;
588 dict_T *dict;
589 buf_T *buf = curbuf;
590 dictitem_T *di;
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200591 int lnum_start;
592 int start_pos_has_prop = 0;
593 int seen_end = 0;
594 int id = -1;
595 int type_id = -1;
596 int skipstart = 0;
597 int lnum = -1;
598 int col = -1;
599 int dir = 1; // 1 = forward, -1 = backward
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100600
601 if (argvars[0].v_type != VAR_DICT || argvars[0].vval.v_dict == NULL)
602 {
603 emsg(_(e_invarg));
604 return;
605 }
606 dict = argvars[0].vval.v_dict;
607
608 if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL)
609 return;
610 if (buf->b_ml.ml_mfp == NULL)
611 return;
612
613 if (argvars[1].v_type != VAR_UNKNOWN)
614 {
615 char_u *dir_s = tv_get_string(&argvars[1]);
616
617 if (*dir_s == 'b')
618 dir = -1;
619 else if (*dir_s != 'f')
620 {
621 emsg(_(e_invarg));
622 return;
623 }
624 }
625
626 di = dict_find(dict, (char_u *)"lnum", -1);
627 if (di != NULL)
628 lnum = tv_get_number(&di->di_tv);
629
630 di = dict_find(dict, (char_u *)"col", -1);
631 if (di != NULL)
632 col = tv_get_number(&di->di_tv);
633
634 if (lnum == -1)
635 {
636 lnum = cursor->lnum;
637 col = cursor->col + 1;
638 }
639 else if (col == -1)
640 col = 1;
641
642 if (lnum < 1 || lnum > buf->b_ml.ml_line_count)
643 {
644 emsg(_(e_invrange));
645 return;
646 }
647
Bram Moolenaareb245562020-09-03 22:33:44 +0200648 skipstart = dict_get_bool(dict, (char_u *)"skipstart", 0);
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100649
650 if (dict_find(dict, (char_u *)"id", -1) != NULL)
651 id = dict_get_number(dict, (char_u *)"id");
652 if (dict_find(dict, (char_u *)"type", -1))
653 {
654 char_u *name = dict_get_string(dict, (char_u *)"type", FALSE);
655 proptype_T *type = lookup_prop_type(name, buf);
656
657 if (type == NULL)
658 return;
659 type_id = type->pt_id;
660 }
661 if (id == -1 && type_id == -1)
662 {
663 emsg(_("E968: Need at least one of 'id' or 'type'"));
664 return;
665 }
666
667 lnum_start = lnum;
668
669 if (rettv_dict_alloc(rettv) == FAIL)
670 return;
671
672 while (1)
673 {
674 char_u *text = ml_get_buf(buf, lnum, FALSE);
675 size_t textlen = STRLEN(text) + 1;
676 int count = (int)((buf->b_ml.ml_line_len - textlen)
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200677 / sizeof(textprop_T));
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100678 int i;
679 textprop_T prop;
680 int prop_start;
681 int prop_end;
682
683 for (i = 0; i < count; ++i)
684 {
685 mch_memmove(&prop, text + textlen + i * sizeof(textprop_T),
686 sizeof(textprop_T));
687
Bram Moolenaar346f18e2020-03-13 21:36:40 +0100688 if (lnum == lnum_start)
Bram Moolenaar965fd8d2020-03-14 07:46:40 +0100689 {
Bram Moolenaar346f18e2020-03-13 21:36:40 +0100690 if (dir < 0)
691 {
692 if (col < prop.tp_col)
693 break;
694 }
695 else if (prop.tp_col + prop.tp_len - (prop.tp_len != 0) < col)
696 continue;
Bram Moolenaar965fd8d2020-03-14 07:46:40 +0100697 }
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100698 if (prop.tp_id == id || prop.tp_type == type_id)
699 {
700 // Check if the starting position has text props.
Bram Moolenaar66b98852020-03-11 19:15:52 +0100701 if (lnum_start == lnum
702 && col >= prop.tp_col
703 && (col <= prop.tp_col + prop.tp_len
704 - (prop.tp_len != 0)))
705 start_pos_has_prop = 1;
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100706
707 prop_start = !(prop.tp_flags & TP_FLAG_CONT_PREV);
708 prop_end = !(prop.tp_flags & TP_FLAG_CONT_NEXT);
709 if (!prop_start && prop_end && dir > 0)
710 seen_end = 1;
711
712 // Skip lines without the start flag.
713 if (!prop_start)
714 {
715 // Always search backwards for start when search started
716 // on a prop and we're not skipping.
717 if (start_pos_has_prop && !skipstart)
718 dir = -1;
Bram Moolenaar4da7a252020-09-02 19:59:00 +0200719 continue;
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100720 }
721
722 // If skipstart is true, skip the prop at start pos (even if
723 // continued from another line).
724 if (start_pos_has_prop && skipstart && !seen_end)
725 {
726 start_pos_has_prop = 0;
Bram Moolenaar4da7a252020-09-02 19:59:00 +0200727 continue;
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100728 }
729
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100730 prop_fill_dict(rettv->vval.v_dict, &prop, buf);
731 dict_add_number(rettv->vval.v_dict, "lnum", lnum);
732
733 return;
734 }
735 }
736
737 if (dir > 0)
738 {
739 if (lnum >= buf->b_ml.ml_line_count)
740 break;
741 lnum++;
742 }
743 else
744 {
745 if (lnum <= 1)
746 break;
747 lnum--;
748 }
Bram Moolenaar66b98852020-03-11 19:15:52 +0100749 // Adjust col to indicate that we're continuing from prev/next line.
750 col = dir < 0 ? buf->b_ml.ml_line_len : 1;
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100751 }
752}
753
754/*
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100755 * prop_list({lnum} [, {bufnr}])
756 */
757 void
758f_prop_list(typval_T *argvars, typval_T *rettv)
759{
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100760 linenr_T lnum = tv_get_number(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100761 buf_T *buf = curbuf;
762
763 if (argvars[1].v_type != VAR_UNKNOWN)
764 {
765 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
766 return;
767 }
768 if (lnum < 1 || lnum > buf->b_ml.ml_line_count)
769 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100770 emsg(_(e_invrange));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100771 return;
772 }
773
774 if (rettv_list_alloc(rettv) == OK)
775 {
776 char_u *text = ml_get_buf(buf, lnum, FALSE);
777 size_t textlen = STRLEN(text) + 1;
Bram Moolenaar4efe73b2018-12-16 14:37:39 +0100778 int count = (int)((buf->b_ml.ml_line_len - textlen)
779 / sizeof(textprop_T));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100780 int i;
781 textprop_T prop;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100782
783 for (i = 0; i < count; ++i)
784 {
785 dict_T *d = dict_alloc();
786
787 if (d == NULL)
788 break;
789 mch_memmove(&prop, text + textlen + i * sizeof(textprop_T),
790 sizeof(textprop_T));
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100791 prop_fill_dict(d, &prop, buf);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100792 list_append_dict(rettv->vval.v_list, d);
793 }
794 }
795}
796
797/*
798 * prop_remove({props} [, {lnum} [, {lnum_end}]])
799 */
800 void
801f_prop_remove(typval_T *argvars, typval_T *rettv)
802{
803 linenr_T start = 1;
804 linenr_T end = 0;
805 linenr_T lnum;
806 dict_T *dict;
807 buf_T *buf = curbuf;
Bram Moolenaara5a40c52020-09-05 20:50:49 +0200808 int do_all;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100809 int id = -1;
810 int type_id = -1;
Bram Moolenaara5a40c52020-09-05 20:50:49 +0200811 int both;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100812
813 rettv->vval.v_number = 0;
814 if (argvars[0].v_type != VAR_DICT || argvars[0].vval.v_dict == NULL)
815 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100816 emsg(_(e_invarg));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100817 return;
818 }
819
820 if (argvars[1].v_type != VAR_UNKNOWN)
821 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100822 start = tv_get_number(&argvars[1]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100823 end = start;
824 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100825 end = tv_get_number(&argvars[2]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100826 if (start < 1 || end < 1)
827 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100828 emsg(_(e_invrange));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100829 return;
830 }
831 }
832
833 dict = argvars[0].vval.v_dict;
Bram Moolenaarf0884c52019-05-24 21:22:29 +0200834 if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL)
835 return;
836 if (buf->b_ml.ml_mfp == NULL)
837 return;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100838
Bram Moolenaara5a40c52020-09-05 20:50:49 +0200839 do_all = dict_get_bool(dict, (char_u *)"all", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100840
841 if (dict_find(dict, (char_u *)"id", -1) != NULL)
Bram Moolenaar8f667172018-12-14 15:38:31 +0100842 id = dict_get_number(dict, (char_u *)"id");
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100843 if (dict_find(dict, (char_u *)"type", -1))
844 {
Bram Moolenaar8f667172018-12-14 15:38:31 +0100845 char_u *name = dict_get_string(dict, (char_u *)"type", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100846 proptype_T *type = lookup_prop_type(name, buf);
847
848 if (type == NULL)
849 return;
850 type_id = type->pt_id;
851 }
Bram Moolenaara5a40c52020-09-05 20:50:49 +0200852 both = dict_get_bool(dict, (char_u *)"both", FALSE);
853
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100854 if (id == -1 && type_id == -1)
855 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100856 emsg(_("E968: Need at least one of 'id' or 'type'"));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100857 return;
858 }
Bram Moolenaar49b79bd2020-03-05 21:52:55 +0100859 if (both && (id == -1 || type_id == -1))
860 {
861 emsg(_("E860: Need 'id' and 'type' with 'both'"));
862 return;
863 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100864
865 if (end == 0)
866 end = buf->b_ml.ml_line_count;
867 for (lnum = start; lnum <= end; ++lnum)
868 {
869 char_u *text;
870 size_t len;
871
872 if (lnum > buf->b_ml.ml_line_count)
873 break;
874 text = ml_get_buf(buf, lnum, FALSE);
875 len = STRLEN(text) + 1;
876 if ((size_t)buf->b_ml.ml_line_len > len)
877 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200878 static textprop_T textprop; // static because of alignment
879 unsigned idx;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100880
881 for (idx = 0; idx < (buf->b_ml.ml_line_len - len)
882 / sizeof(textprop_T); ++idx)
883 {
884 char_u *cur_prop = buf->b_ml.ml_line_ptr + len
885 + idx * sizeof(textprop_T);
886 size_t taillen;
887
888 mch_memmove(&textprop, cur_prop, sizeof(textprop_T));
Bram Moolenaar49b79bd2020-03-05 21:52:55 +0100889 if (both ? textprop.tp_id == id && textprop.tp_type == type_id
890 : textprop.tp_id == id || textprop.tp_type == type_id)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100891 {
892 if (!(buf->b_ml.ml_flags & ML_LINE_DIRTY))
893 {
894 char_u *newptr = alloc(buf->b_ml.ml_line_len);
895
896 // need to allocate the line to be able to change it
897 if (newptr == NULL)
898 return;
899 mch_memmove(newptr, buf->b_ml.ml_line_ptr,
900 buf->b_ml.ml_line_len);
901 buf->b_ml.ml_line_ptr = newptr;
Bram Moolenaar0a2f5782019-03-22 13:20:43 +0100902 buf->b_ml.ml_flags |= ML_LINE_DIRTY;
903
904 cur_prop = buf->b_ml.ml_line_ptr + len
Bram Moolenaarf0884c52019-05-24 21:22:29 +0200905 + idx * sizeof(textprop_T);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100906 }
907
908 taillen = buf->b_ml.ml_line_len - len
909 - (idx + 1) * sizeof(textprop_T);
910 if (taillen > 0)
911 mch_memmove(cur_prop, cur_prop + sizeof(textprop_T),
912 taillen);
913 buf->b_ml.ml_line_len -= sizeof(textprop_T);
914 --idx;
915
916 ++rettv->vval.v_number;
917 if (!do_all)
918 break;
919 }
920 }
921 }
922 }
923 redraw_buf_later(buf, NOT_VALID);
924}
925
926/*
927 * Common for f_prop_type_add() and f_prop_type_change().
928 */
Bram Moolenaar5843f5f2019-08-20 20:13:45 +0200929 static void
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100930prop_type_set(typval_T *argvars, int add)
931{
932 char_u *name;
933 buf_T *buf = NULL;
934 dict_T *dict;
935 dictitem_T *di;
936 proptype_T *prop;
937
Bram Moolenaard155d7a2018-12-21 16:04:21 +0100938 name = tv_get_string(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100939 if (*name == NUL)
940 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100941 emsg(_(e_invarg));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100942 return;
943 }
944
945 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
946 return;
947 dict = argvars[1].vval.v_dict;
948
949 prop = find_prop(name, buf);
950 if (add)
951 {
952 hashtab_T **htp;
953
954 if (prop != NULL)
955 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100956 semsg(_("E969: Property type %s already defined"), name);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100957 return;
958 }
Bram Moolenaar47ed5532019-08-08 20:49:14 +0200959 prop = alloc_clear(offsetof(proptype_T, pt_name) + STRLEN(name) + 1);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100960 if (prop == NULL)
961 return;
962 STRCPY(prop->pt_name, name);
963 prop->pt_id = ++proptype_id;
Bram Moolenaar0743ef92019-11-13 16:37:31 +0100964 prop->pt_flags = PT_FLAG_COMBINE;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100965 htp = buf == NULL ? &global_proptypes : &buf->b_proptypes;
966 if (*htp == NULL)
967 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200968 *htp = ALLOC_ONE(hashtab_T);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100969 if (*htp == NULL)
Bram Moolenaarfb95e212018-12-14 12:18:11 +0100970 {
971 vim_free(prop);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100972 return;
Bram Moolenaarfb95e212018-12-14 12:18:11 +0100973 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100974 hash_init(*htp);
975 }
Bram Moolenaarfb95e212018-12-14 12:18:11 +0100976 hash_add(*htp, PT2HIKEY(prop));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100977 }
978 else
979 {
980 if (prop == NULL)
981 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100982 semsg(_(e_type_not_exist), name);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100983 return;
984 }
985 }
986
987 if (dict != NULL)
988 {
989 di = dict_find(dict, (char_u *)"highlight", -1);
990 if (di != NULL)
991 {
992 char_u *highlight;
993 int hl_id = 0;
994
Bram Moolenaar8f667172018-12-14 15:38:31 +0100995 highlight = dict_get_string(dict, (char_u *)"highlight", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100996 if (highlight != NULL && *highlight != NUL)
997 hl_id = syn_name2id(highlight);
998 if (hl_id <= 0)
999 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001000 semsg(_("E970: Unknown highlight group name: '%s'"),
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001001 highlight == NULL ? (char_u *)"" : highlight);
1002 return;
1003 }
1004 prop->pt_hl_id = hl_id;
1005 }
1006
Bram Moolenaar58187f12019-05-05 16:33:47 +02001007 di = dict_find(dict, (char_u *)"combine", -1);
1008 if (di != NULL)
1009 {
Bram Moolenaarfa2e38d2020-09-05 21:00:00 +02001010 if (tv_get_bool(&di->di_tv))
Bram Moolenaar58187f12019-05-05 16:33:47 +02001011 prop->pt_flags |= PT_FLAG_COMBINE;
1012 else
1013 prop->pt_flags &= ~PT_FLAG_COMBINE;
1014 }
1015
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001016 di = dict_find(dict, (char_u *)"priority", -1);
1017 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001018 prop->pt_priority = tv_get_number(&di->di_tv);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001019
1020 di = dict_find(dict, (char_u *)"start_incl", -1);
1021 if (di != NULL)
1022 {
Bram Moolenaarfa2e38d2020-09-05 21:00:00 +02001023 if (tv_get_bool(&di->di_tv))
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001024 prop->pt_flags |= PT_FLAG_INS_START_INCL;
1025 else
1026 prop->pt_flags &= ~PT_FLAG_INS_START_INCL;
1027 }
1028
1029 di = dict_find(dict, (char_u *)"end_incl", -1);
1030 if (di != NULL)
1031 {
Bram Moolenaarfa2e38d2020-09-05 21:00:00 +02001032 if (tv_get_bool(&di->di_tv))
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001033 prop->pt_flags |= PT_FLAG_INS_END_INCL;
1034 else
1035 prop->pt_flags &= ~PT_FLAG_INS_END_INCL;
1036 }
1037 }
1038}
1039
1040/*
1041 * prop_type_add({name}, {props})
1042 */
1043 void
1044f_prop_type_add(typval_T *argvars, typval_T *rettv UNUSED)
1045{
1046 prop_type_set(argvars, TRUE);
1047}
1048
1049/*
1050 * prop_type_change({name}, {props})
1051 */
1052 void
1053f_prop_type_change(typval_T *argvars, typval_T *rettv UNUSED)
1054{
1055 prop_type_set(argvars, FALSE);
1056}
1057
1058/*
1059 * prop_type_delete({name} [, {bufnr}])
1060 */
1061 void
1062f_prop_type_delete(typval_T *argvars, typval_T *rettv UNUSED)
1063{
1064 char_u *name;
1065 buf_T *buf = NULL;
1066 hashitem_T *hi;
1067
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001068 name = tv_get_string(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001069 if (*name == NUL)
1070 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001071 emsg(_(e_invarg));
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001072 return;
1073 }
1074
1075 if (argvars[1].v_type != VAR_UNKNOWN)
1076 {
1077 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
1078 return;
1079 }
1080
1081 hi = find_prop_hi(name, buf);
1082 if (hi != NULL)
1083 {
1084 hashtab_T *ht;
Bram Moolenaarfb95e212018-12-14 12:18:11 +01001085 proptype_T *prop = HI2PT(hi);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001086
1087 if (buf == NULL)
1088 ht = global_proptypes;
1089 else
1090 ht = buf->b_proptypes;
1091 hash_remove(ht, hi);
Bram Moolenaarfb95e212018-12-14 12:18:11 +01001092 vim_free(prop);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001093 }
1094}
1095
1096/*
1097 * prop_type_get({name} [, {bufnr}])
1098 */
1099 void
Bram Moolenaar3d8a5132020-01-04 16:13:49 +01001100f_prop_type_get(typval_T *argvars, typval_T *rettv)
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001101{
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001102 char_u *name = tv_get_string(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001103
1104 if (*name == NUL)
1105 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001106 emsg(_(e_invarg));
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001107 return;
1108 }
1109 if (rettv_dict_alloc(rettv) == OK)
1110 {
1111 proptype_T *prop = NULL;
1112 buf_T *buf = NULL;
1113
1114 if (argvars[1].v_type != VAR_UNKNOWN)
1115 {
1116 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
1117 return;
1118 }
1119
1120 prop = find_prop(name, buf);
1121 if (prop != NULL)
1122 {
1123 dict_T *d = rettv->vval.v_dict;
1124
1125 if (prop->pt_hl_id > 0)
1126 dict_add_string(d, "highlight", syn_id2name(prop->pt_hl_id));
1127 dict_add_number(d, "priority", prop->pt_priority);
Bram Moolenaar58187f12019-05-05 16:33:47 +02001128 dict_add_number(d, "combine",
1129 (prop->pt_flags & PT_FLAG_COMBINE) ? 1 : 0);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001130 dict_add_number(d, "start_incl",
1131 (prop->pt_flags & PT_FLAG_INS_START_INCL) ? 1 : 0);
1132 dict_add_number(d, "end_incl",
1133 (prop->pt_flags & PT_FLAG_INS_END_INCL) ? 1 : 0);
1134 if (buf != NULL)
1135 dict_add_number(d, "bufnr", buf->b_fnum);
1136 }
1137 }
1138}
1139
1140 static void
1141list_types(hashtab_T *ht, list_T *l)
1142{
1143 long todo;
1144 hashitem_T *hi;
1145
1146 todo = (long)ht->ht_used;
1147 for (hi = ht->ht_array; todo > 0; ++hi)
1148 {
1149 if (!HASHITEM_EMPTY(hi))
1150 {
1151 proptype_T *prop = HI2PT(hi);
1152
1153 list_append_string(l, prop->pt_name, -1);
1154 --todo;
1155 }
1156 }
1157}
1158
1159/*
1160 * prop_type_list([{bufnr}])
1161 */
1162 void
1163f_prop_type_list(typval_T *argvars, typval_T *rettv UNUSED)
1164{
1165 buf_T *buf = NULL;
1166
1167 if (rettv_list_alloc(rettv) == OK)
1168 {
1169 if (argvars[0].v_type != VAR_UNKNOWN)
1170 {
1171 if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL)
1172 return;
1173 }
1174 if (buf == NULL)
1175 {
1176 if (global_proptypes != NULL)
1177 list_types(global_proptypes, rettv->vval.v_list);
1178 }
1179 else if (buf->b_proptypes != NULL)
1180 list_types(buf->b_proptypes, rettv->vval.v_list);
1181 }
1182}
1183
1184/*
1185 * Free all property types in "ht".
1186 */
1187 static void
1188clear_ht_prop_types(hashtab_T *ht)
1189{
1190 long todo;
1191 hashitem_T *hi;
1192
1193 if (ht == NULL)
1194 return;
1195
1196 todo = (long)ht->ht_used;
1197 for (hi = ht->ht_array; todo > 0; ++hi)
1198 {
1199 if (!HASHITEM_EMPTY(hi))
1200 {
1201 proptype_T *prop = HI2PT(hi);
1202
1203 vim_free(prop);
1204 --todo;
1205 }
1206 }
1207
1208 hash_clear(ht);
1209 vim_free(ht);
1210}
1211
1212#if defined(EXITFREE) || defined(PROTO)
1213/*
Bram Moolenaarfb95e212018-12-14 12:18:11 +01001214 * Free all global property types.
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001215 */
1216 void
1217clear_global_prop_types(void)
1218{
1219 clear_ht_prop_types(global_proptypes);
1220 global_proptypes = NULL;
1221}
1222#endif
1223
1224/*
1225 * Free all property types for "buf".
1226 */
1227 void
1228clear_buf_prop_types(buf_T *buf)
1229{
1230 clear_ht_prop_types(buf->b_proptypes);
1231 buf->b_proptypes = NULL;
1232}
1233
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001234// Struct used to return two values from adjust_prop().
1235typedef struct
1236{
1237 int dirty; // if the property was changed
1238 int can_drop; // whether after this change, the prop may be removed
1239} adjustres_T;
1240
1241/*
1242 * Adjust the property for "added" bytes (can be negative) inserted at "col".
1243 *
1244 * Note that "col" is zero-based, while tp_col is one-based.
1245 * Only for the current buffer.
1246 * "flags" can have:
1247 * APC_SUBSTITUTE: Text is replaced, not inserted.
1248 */
1249 static adjustres_T
1250adjust_prop(
1251 textprop_T *prop,
1252 colnr_T col,
1253 int added,
1254 int flags)
1255{
1256 proptype_T *pt = text_prop_type_by_id(curbuf, prop->tp_type);
1257 int start_incl = (pt != NULL
1258 && (pt->pt_flags & PT_FLAG_INS_START_INCL))
1259 || (flags & APC_SUBSTITUTE);
1260 int end_incl = (pt != NULL
1261 && (pt->pt_flags & PT_FLAG_INS_END_INCL));
1262 // Do not drop zero-width props if they later can increase in
1263 // size.
1264 int droppable = !(start_incl || end_incl);
1265 adjustres_T res = {TRUE, FALSE};
1266
1267 if (added > 0)
1268 {
1269 if (col + 1 <= prop->tp_col
1270 - (start_incl || (prop->tp_len == 0 && end_incl)))
1271 // Change is entirely before the text property: Only shift
1272 prop->tp_col += added;
1273 else if (col + 1 < prop->tp_col + prop->tp_len + end_incl)
1274 // Insertion was inside text property
1275 prop->tp_len += added;
1276 }
1277 else if (prop->tp_col > col + 1)
1278 {
1279 if (prop->tp_col + added < col + 1)
1280 {
1281 prop->tp_len += (prop->tp_col - 1 - col) + added;
1282 prop->tp_col = col + 1;
1283 if (prop->tp_len <= 0)
1284 {
1285 prop->tp_len = 0;
1286 res.can_drop = droppable;
1287 }
1288 }
1289 else
1290 prop->tp_col += added;
1291 }
1292 else if (prop->tp_len > 0 && prop->tp_col + prop->tp_len > col)
1293 {
1294 int after = col - added - (prop->tp_col - 1 + prop->tp_len);
1295
1296 prop->tp_len += after > 0 ? added + after : added;
1297 res.can_drop = prop->tp_len <= 0 && droppable;
1298 }
1299 else
1300 res.dirty = FALSE;
1301
1302 return res;
1303}
1304
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01001305/*
1306 * Adjust the columns of text properties in line "lnum" after position "col" to
1307 * shift by "bytes_added" (can be negative).
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001308 * Note that "col" is zero-based, while tp_col is one-based.
1309 * Only for the current buffer.
Bram Moolenaarf3333b02019-05-19 22:53:40 +02001310 * "flags" can have:
1311 * APC_SAVE_FOR_UNDO: Call u_savesub() before making changes to the line.
1312 * APC_SUBSTITUTE: Text is replaced, not inserted.
Bram Moolenaar8055d172019-05-17 22:57:26 +02001313 * Caller is expected to check b_has_textprop and "bytes_added" being non-zero.
Bram Moolenaar338dfda2019-05-19 15:19:57 +02001314 * Returns TRUE when props were changed.
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01001315 */
Bram Moolenaar338dfda2019-05-19 15:19:57 +02001316 int
Bram Moolenaar196d1572019-01-02 23:47:18 +01001317adjust_prop_columns(
1318 linenr_T lnum,
1319 colnr_T col,
Bram Moolenaar338dfda2019-05-19 15:19:57 +02001320 int bytes_added,
Bram Moolenaarf3333b02019-05-19 22:53:40 +02001321 int flags)
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01001322{
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001323 int proplen;
1324 char_u *props;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001325 int dirty = FALSE;
Bram Moolenaar196d1572019-01-02 23:47:18 +01001326 int ri, wi;
1327 size_t textlen;
1328
1329 if (text_prop_frozen > 0)
Bram Moolenaar338dfda2019-05-19 15:19:57 +02001330 return FALSE;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001331
1332 proplen = get_text_props(curbuf, lnum, &props, TRUE);
1333 if (proplen == 0)
Bram Moolenaar338dfda2019-05-19 15:19:57 +02001334 return FALSE;
Bram Moolenaar196d1572019-01-02 23:47:18 +01001335 textlen = curbuf->b_ml.ml_line_len - proplen * sizeof(textprop_T);
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001336
Bram Moolenaar196d1572019-01-02 23:47:18 +01001337 wi = 0; // write index
1338 for (ri = 0; ri < proplen; ++ri)
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001339 {
Bram Moolenaar12f20032020-02-26 22:06:00 +01001340 textprop_T prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001341 adjustres_T res;
Bram Moolenaarf3333b02019-05-19 22:53:40 +02001342
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001343 mch_memmove(&prop, props + ri * sizeof(prop), sizeof(prop));
1344 res = adjust_prop(&prop, col, bytes_added, flags);
1345 if (res.dirty)
Bram Moolenaarf3333b02019-05-19 22:53:40 +02001346 {
Bram Moolenaar338dfda2019-05-19 15:19:57 +02001347 // Save for undo if requested and not done yet.
Bram Moolenaarcf070112020-06-29 23:02:21 +02001348 if ((flags & APC_SAVE_FOR_UNDO) && !dirty
1349 && u_savesub(lnum) == FAIL)
1350 return FALSE;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001351 dirty = TRUE;
Bram Moolenaar8902b312020-09-20 21:04:35 +02001352
1353 // u_savesub() may have updated curbuf->b_ml, fetch it again
1354 if (curbuf->b_ml.ml_line_lnum != lnum)
1355 proplen = get_text_props(curbuf, lnum, &props, TRUE);
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001356 }
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001357 if (res.can_drop)
1358 continue; // Drop this text property
Bram Moolenaar12f20032020-02-26 22:06:00 +01001359 mch_memmove(props + wi * sizeof(textprop_T), &prop, sizeof(textprop_T));
Bram Moolenaar196d1572019-01-02 23:47:18 +01001360 ++wi;
1361 }
1362 if (dirty)
1363 {
Bram Moolenaar4614f532019-01-06 12:54:55 +01001364 colnr_T newlen = (int)textlen + wi * (colnr_T)sizeof(textprop_T);
1365
1366 if ((curbuf->b_ml.ml_flags & ML_LINE_DIRTY) == 0)
1367 curbuf->b_ml.ml_line_ptr =
1368 vim_memsave(curbuf->b_ml.ml_line_ptr, newlen);
Bram Moolenaar196d1572019-01-02 23:47:18 +01001369 curbuf->b_ml.ml_flags |= ML_LINE_DIRTY;
Bram Moolenaar4614f532019-01-06 12:54:55 +01001370 curbuf->b_ml.ml_line_len = newlen;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01001371 }
Bram Moolenaar338dfda2019-05-19 15:19:57 +02001372 return dirty;
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01001373}
1374
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001375/*
1376 * Adjust text properties for a line that was split in two.
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02001377 * "lnum_props" is the line that has the properties from before the split.
1378 * "lnum_top" is the top line.
1379 * "kept" is the number of bytes kept in the first line, while
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001380 * "deleted" is the number of bytes deleted.
1381 */
1382 void
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02001383adjust_props_for_split(
1384 linenr_T lnum_props,
1385 linenr_T lnum_top,
1386 int kept,
1387 int deleted)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001388{
1389 char_u *props;
1390 int count;
1391 garray_T prevprop;
1392 garray_T nextprop;
1393 int i;
1394 int skipped = kept + deleted;
1395
1396 if (!curbuf->b_has_textprop)
1397 return;
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02001398
1399 // Get the text properties from "lnum_props".
1400 count = get_text_props(curbuf, lnum_props, &props, FALSE);
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001401 ga_init2(&prevprop, sizeof(textprop_T), 10);
1402 ga_init2(&nextprop, sizeof(textprop_T), 10);
1403
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001404 // Keep the relevant ones in the first line, reducing the length if needed.
1405 // Copy the ones that include the split to the second line.
1406 // Move the ones after the split to the second line.
1407 for (i = 0; i < count; ++i)
1408 {
1409 textprop_T prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001410 proptype_T *pt;
1411 int start_incl, end_incl;
1412 int cont_prev, cont_next;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001413
1414 // copy the prop to an aligned structure
1415 mch_memmove(&prop, props + i * sizeof(textprop_T), sizeof(textprop_T));
1416
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001417 pt = text_prop_type_by_id(curbuf, prop.tp_type);
1418 start_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_START_INCL));
1419 end_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_END_INCL));
1420 cont_prev = prop.tp_col + !start_incl <= kept;
1421 cont_next = skipped <= prop.tp_col + prop.tp_len - !end_incl;
1422
1423 if (cont_prev && ga_grow(&prevprop, 1) == OK)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001424 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001425 textprop_T *p = ((textprop_T *)prevprop.ga_data) + prevprop.ga_len;
1426
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001427 *p = prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001428 ++prevprop.ga_len;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001429 if (p->tp_col + p->tp_len >= kept)
1430 p->tp_len = kept - p->tp_col;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001431 if (cont_next)
1432 p->tp_flags |= TP_FLAG_CONT_NEXT;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001433 }
1434
Bram Moolenaar5c65e6a2019-05-17 11:08:56 +02001435 // Only add the property to the next line if the length is bigger than
1436 // zero.
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001437 if (cont_next && ga_grow(&nextprop, 1) == OK)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001438 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001439 textprop_T *p = ((textprop_T *)nextprop.ga_data) + nextprop.ga_len;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001440 *p = prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001441 ++nextprop.ga_len;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001442 if (p->tp_col > skipped)
1443 p->tp_col -= skipped - 1;
1444 else
1445 {
1446 p->tp_len -= skipped - p->tp_col;
1447 p->tp_col = 1;
1448 }
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001449 if (cont_prev)
1450 p->tp_flags |= TP_FLAG_CONT_PREV;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001451 }
1452 }
1453
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02001454 set_text_props(lnum_top, prevprop.ga_data,
1455 prevprop.ga_len * sizeof(textprop_T));
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001456 ga_clear(&prevprop);
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02001457 set_text_props(lnum_top + 1, nextprop.ga_data,
1458 nextprop.ga_len * sizeof(textprop_T));
Bram Moolenaar4164bb22019-01-04 23:09:49 +01001459 ga_clear(&nextprop);
1460}
1461
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001462/*
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001463 * Prepend properties of joined line "lnum" to "new_props".
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001464 */
1465 void
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001466prepend_joined_props(
1467 char_u *new_props,
1468 int propcount,
1469 int *props_remaining,
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001470 linenr_T lnum,
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001471 int add_all,
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001472 long col,
1473 int removed)
1474{
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001475 char_u *props;
1476 int proplen = get_text_props(curbuf, lnum, &props, FALSE);
1477 int i;
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001478
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001479 for (i = proplen; i-- > 0; )
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001480 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001481 textprop_T prop;
1482 int end;
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001483
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001484 mch_memmove(&prop, props + i * sizeof(prop), sizeof(prop));
1485 end = !(prop.tp_flags & TP_FLAG_CONT_NEXT);
1486
1487 adjust_prop(&prop, 0, -removed, 0); // Remove leading spaces
Bram Moolenaar8e7d6222020-12-18 19:49:56 +01001488 adjust_prop(&prop, -1, col, 0); // Make line start at its final column
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001489
1490 if (add_all || end)
1491 mch_memmove(new_props + --(*props_remaining) * sizeof(prop),
1492 &prop, sizeof(prop));
1493 else
1494 {
1495 int j;
1496 int found = FALSE;
1497
1498 // Search for continuing prop.
1499 for (j = *props_remaining; j < propcount; ++j)
1500 {
1501 textprop_T op;
1502
1503 mch_memmove(&op, new_props + j * sizeof(op), sizeof(op));
1504 if ((op.tp_flags & TP_FLAG_CONT_PREV)
1505 && op.tp_id == prop.tp_id && op.tp_type == prop.tp_type)
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001506 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001507 found = TRUE;
1508 op.tp_len += op.tp_col - prop.tp_col;
1509 op.tp_col = prop.tp_col;
1510 // Start/end is taken care of when deleting joined lines
1511 op.tp_flags = prop.tp_flags;
1512 mch_memmove(new_props + j * sizeof(op), &op, sizeof(op));
1513 break;
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001514 }
1515 }
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001516 if (!found)
1517 internal_error("text property above joined line not found");
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001518 }
Bram Moolenaar80e737c2019-05-17 19:56:34 +02001519 }
1520}
1521
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01001522#endif // FEAT_PROP_POPUP