blob: 241c4ae94ba1aa913178e0ae6fee38594603a4d3 [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
14#include "vim.h"
15
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +010016#if defined(FEAT_PROP_POPUP) || defined(PROTO)
Bram Moolenaar98aefe72018-12-13 22:20:09 +010017
18/*
19 * In a hashtable item "hi_key" points to "pt_name" in a proptype_T.
20 * This avoids adding a pointer to the hashtable item.
21 * PT2HIKEY() converts a proptype pointer to a hashitem key pointer.
22 * HIKEY2PT() converts a hashitem key pointer to a proptype pointer.
23 * HI2PT() converts a hashitem pointer to a proptype pointer.
24 */
25#define PT2HIKEY(p) ((p)->pt_name)
26#define HIKEY2PT(p) ((proptype_T *)((p) - offsetof(proptype_T, pt_name)))
27#define HI2PT(hi) HIKEY2PT((hi)->hi_key)
28
29// The global text property types.
30static hashtab_T *global_proptypes = NULL;
Bram Moolenaare44336b2022-08-07 18:20:08 +010031static proptype_T **global_proparray = NULL;
Bram Moolenaar98aefe72018-12-13 22:20:09 +010032
33// The last used text property type ID.
34static int proptype_id = 0;
35
Bram Moolenaar98aefe72018-12-13 22:20:09 +010036/*
37 * Find a property type by name, return the hashitem.
38 * Returns NULL if the item can't be found.
39 */
40 static hashitem_T *
Bram Moolenaare44336b2022-08-07 18:20:08 +010041find_prop_type_hi(char_u *name, buf_T *buf)
Bram Moolenaar98aefe72018-12-13 22:20:09 +010042{
43 hashtab_T *ht;
44 hashitem_T *hi;
45
46 if (*name == NUL)
47 return NULL;
48 if (buf == NULL)
49 ht = global_proptypes;
50 else
51 ht = buf->b_proptypes;
52
53 if (ht == NULL)
54 return NULL;
55 hi = hash_find(ht, name);
56 if (HASHITEM_EMPTY(hi))
57 return NULL;
58 return hi;
59}
60
61/*
Bram Moolenaare44336b2022-08-07 18:20:08 +010062 * Like find_prop_type_hi() but return the property type.
Bram Moolenaar98aefe72018-12-13 22:20:09 +010063 */
64 static proptype_T *
Bram Moolenaare44336b2022-08-07 18:20:08 +010065find_prop_type(char_u *name, buf_T *buf)
Bram Moolenaar98aefe72018-12-13 22:20:09 +010066{
Bram Moolenaare44336b2022-08-07 18:20:08 +010067 hashitem_T *hi = find_prop_type_hi(name, buf);
Bram Moolenaar98aefe72018-12-13 22:20:09 +010068
69 if (hi == NULL)
70 return NULL;
71 return HI2PT(hi);
72}
73
74/*
Bram Moolenaar12034e22019-08-25 22:25:02 +020075 * Get the prop type ID of "name".
76 * When not found return zero.
77 */
78 int
79find_prop_type_id(char_u *name, buf_T *buf)
80{
Bram Moolenaare44336b2022-08-07 18:20:08 +010081 proptype_T *pt = find_prop_type(name, buf);
Bram Moolenaar12034e22019-08-25 22:25:02 +020082
83 if (pt == NULL)
84 return 0;
85 return pt->pt_id;
86}
87
88/*
Bram Moolenaar98aefe72018-12-13 22:20:09 +010089 * Lookup a property type by name. First in "buf" and when not found in the
90 * global types.
91 * When not found gives an error message and returns NULL.
92 */
93 static proptype_T *
94lookup_prop_type(char_u *name, buf_T *buf)
95{
Bram Moolenaare44336b2022-08-07 18:20:08 +010096 proptype_T *type = find_prop_type(name, buf);
Bram Moolenaar98aefe72018-12-13 22:20:09 +010097
98 if (type == NULL)
Bram Moolenaare44336b2022-08-07 18:20:08 +010099 type = find_prop_type(name, NULL);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100100 if (type == NULL)
Bram Moolenaara9fa8c52023-01-02 18:10:04 +0000101 semsg(_(e_property_type_str_does_not_exist), name);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100102 return type;
103}
104
105/*
106 * Get an optional "bufnr" item from the dict in "arg".
107 * When the argument is not used or "bufnr" is not present then "buf" is
108 * unchanged.
109 * If "bufnr" is valid or not present return OK.
Bram Moolenaar32aa1022019-11-02 22:54:41 +0100110 * When "arg" is not a dict or "bufnr" is invalid return FAIL.
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100111 */
112 static int
113get_bufnr_from_arg(typval_T *arg, buf_T **buf)
114{
115 dictitem_T *di;
116
117 if (arg->v_type != VAR_DICT)
118 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +0000119 emsg(_(e_dictionary_required));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100120 return FAIL;
121 }
122 if (arg->vval.v_dict == NULL)
123 return OK; // NULL dict is like an empty dict
124 di = dict_find(arg->vval.v_dict, (char_u *)"bufnr", -1);
Martin Tournoije2390c72021-07-28 13:30:16 +0200125 if (di != NULL && (di->di_tv.v_type != VAR_NUMBER
126 || di->di_tv.vval.v_number != 0))
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100127 {
Bram Moolenaarf0884c52019-05-24 21:22:29 +0200128 *buf = get_buf_arg(&di->di_tv);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100129 if (*buf == NULL)
130 return FAIL;
131 }
132 return OK;
133}
134
135/*
136 * prop_add({lnum}, {col}, {props})
137 */
138 void
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100139f_prop_add(typval_T *argvars, typval_T *rettv)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100140{
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100141 linenr_T start_lnum;
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100142 colnr_T start_col;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100143
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +0200144 if (in_vim9script()
145 && (check_for_number_arg(argvars, 0) == FAIL
146 || check_for_number_arg(argvars, 1) == FAIL
147 || check_for_dict_arg(argvars, 2) == FAIL))
148 return;
149
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100150 start_lnum = tv_get_number(&argvars[0]);
151 start_col = tv_get_number(&argvars[1]);
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +0100152 if (check_for_dict_arg(argvars, 2) == FAIL)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100153 return;
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200154
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100155 rettv->vval.v_number = prop_add_common(start_lnum, start_col,
156 argvars[2].vval.v_dict, curbuf, &argvars[2]);
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200157}
158
159/*
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200160 * Attach a text property 'type_name' to the text starting
161 * at [start_lnum, start_col] and ending at [end_lnum, end_col] in
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100162 * the buffer "buf" and assign identifier "id".
163 * When "text" is not NULL add it to buf->b_textprop_text[-id - 1].
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200164 */
165 static int
166prop_add_one(
167 buf_T *buf,
168 char_u *type_name,
169 int id,
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100170 char_u *text_arg,
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100171 int text_padding_left,
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100172 int text_flags,
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200173 linenr_T start_lnum,
174 linenr_T end_lnum,
175 colnr_T start_col,
176 colnr_T end_col)
177{
178 proptype_T *type;
179 linenr_T lnum;
180 int proplen;
181 char_u *props = NULL;
182 char_u *newprops;
183 size_t textlen;
184 char_u *newtext;
185 int i;
186 textprop_T tmp_prop;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100187 char_u *text = text_arg;
188 int res = FAIL;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200189
190 type = lookup_prop_type(type_name, buf);
191 if (type == NULL)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100192 goto theend;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200193
194 if (start_lnum < 1 || start_lnum > buf->b_ml.ml_line_count)
195 {
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +0000196 semsg(_(e_invalid_line_number_nr), (long)start_lnum);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100197 goto theend;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200198 }
199 if (end_lnum < start_lnum || end_lnum > buf->b_ml.ml_line_count)
200 {
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +0000201 semsg(_(e_invalid_line_number_nr), (long)end_lnum);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100202 goto theend;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200203 }
204
205 if (buf->b_ml.ml_mfp == NULL)
206 {
Bram Moolenaar9a846fb2022-01-01 21:59:18 +0000207 emsg(_(e_cannot_add_text_property_to_unloaded_buffer));
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100208 goto theend;
209 }
210
211 if (text != NULL)
212 {
Bram Moolenaar783ef722022-08-01 16:11:06 +0100213 garray_T *gap = &buf->b_textprop_text;
214 char_u *p;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100215
216 // double check we got the right ID
217 if (-id - 1 != gap->ga_len)
218 iemsg("text prop ID mismatch");
219 if (gap->ga_growsize == 0)
220 ga_init2(gap, sizeof(char *), 50);
221 if (ga_grow(gap, 1) == FAIL)
222 goto theend;
223 ((char_u **)gap->ga_data)[gap->ga_len++] = text;
Bram Moolenaar783ef722022-08-01 16:11:06 +0100224
Bram Moolenaarf4ba8bc2022-08-05 17:05:04 +0100225 // change any control character (Tab, Newline, etc.) to a Space to make
226 // it simpler to compute the size
Bram Moolenaar783ef722022-08-01 16:11:06 +0100227 for (p = text; *p != NUL; MB_PTR_ADV(p))
Bram Moolenaarf4ba8bc2022-08-05 17:05:04 +0100228 if (*p < ' ')
Bram Moolenaar783ef722022-08-01 16:11:06 +0100229 *p = ' ';
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100230 text = NULL;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200231 }
232
233 for (lnum = start_lnum; lnum <= end_lnum; ++lnum)
234 {
Bram Moolenaar6eda17d2022-09-12 19:25:11 +0100235 colnr_T col; // start column use in tp_col
236 colnr_T sort_col; // column where it appears
237 long length; // in bytes
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200238
239 // Fetch the line to get the ml_line_len field updated.
240 proplen = get_text_props(buf, lnum, &props, TRUE);
241 textlen = buf->b_ml.ml_line_len - proplen * sizeof(textprop_T);
242
243 if (lnum == start_lnum)
244 col = start_col;
245 else
246 col = 1;
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100247 if (col - 1 > (colnr_T)textlen && !(col == 0 && text_arg != NULL))
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200248 {
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +0000249 semsg(_(e_invalid_column_number_nr), (long)start_col);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100250 goto theend;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200251 }
Bram Moolenaar6eda17d2022-09-12 19:25:11 +0100252 sort_col = col;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200253
254 if (lnum == end_lnum)
255 length = end_col - col;
256 else
257 length = (int)textlen - col + 1;
258 if (length > (long)textlen)
259 length = (int)textlen; // can include the end-of-line
260 if (length < 0)
261 length = 0; // zero-width property
262
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100263 if (text_arg != NULL)
264 {
265 length = 1; // text is placed on one character
266 if (col == 0)
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100267 {
Bram Moolenaar6eda17d2022-09-12 19:25:11 +0100268 col = MAXCOL; // before or after the line
269 if ((text_flags & TP_FLAG_ALIGN_ABOVE) == 0)
270 sort_col = MAXCOL;
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100271 length += text_padding_left;
272 }
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100273 }
274
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200275 // Allocate the new line with space for the new property.
276 newtext = alloc(buf->b_ml.ml_line_len + sizeof(textprop_T));
277 if (newtext == NULL)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100278 goto theend;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200279 // Copy the text, including terminating NUL.
280 mch_memmove(newtext, buf->b_ml.ml_line_ptr, textlen);
281
282 // Find the index where to insert the new property.
283 // Since the text properties are not aligned properly when stored with
284 // the text, we need to copy them as bytes before using it as a struct.
285 for (i = 0; i < proplen; ++i)
286 {
Bram Moolenaar6eda17d2022-09-12 19:25:11 +0100287 colnr_T prop_col;
288
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200289 mch_memmove(&tmp_prop, props + i * sizeof(textprop_T),
290 sizeof(textprop_T));
Bram Moolenaar6eda17d2022-09-12 19:25:11 +0100291 // col is MAXCOL when the text goes above or after the line, when
292 // above we should use column zero for sorting
293 prop_col = (tmp_prop.tp_flags & TP_FLAG_ALIGN_ABOVE)
294 ? 0 : tmp_prop.tp_col;
295 if (prop_col >= sort_col)
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200296 break;
297 }
298 newprops = newtext + textlen;
299 if (i > 0)
300 mch_memmove(newprops, props, sizeof(textprop_T) * i);
301
302 tmp_prop.tp_col = col;
303 tmp_prop.tp_len = length;
304 tmp_prop.tp_id = id;
305 tmp_prop.tp_type = type->pt_id;
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100306 tmp_prop.tp_flags = text_flags
307 | (lnum > start_lnum ? TP_FLAG_CONT_PREV : 0)
Bram Moolenaar28c9f892022-08-14 13:28:55 +0100308 | (lnum < end_lnum ? TP_FLAG_CONT_NEXT : 0)
309 | ((type->pt_flags & PT_FLAG_INS_START_INCL)
310 ? TP_FLAG_START_INCL : 0);
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200311 mch_memmove(newprops + i * sizeof(textprop_T), &tmp_prop,
312 sizeof(textprop_T));
313
314 if (i < proplen)
315 mch_memmove(newprops + (i + 1) * sizeof(textprop_T),
316 props + i * sizeof(textprop_T),
317 sizeof(textprop_T) * (proplen - i));
318
Bram Moolenaarfa4873c2022-06-30 22:13:59 +0100319 if (buf->b_ml.ml_flags & (ML_LINE_DIRTY | ML_ALLOCATED))
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200320 vim_free(buf->b_ml.ml_line_ptr);
321 buf->b_ml.ml_line_ptr = newtext;
322 buf->b_ml.ml_line_len += sizeof(textprop_T);
323 buf->b_ml.ml_flags |= ML_LINE_DIRTY;
324 }
325
Bram Moolenaar326c5d32022-08-12 13:05:49 +0100326 changed_line_display_buf(buf);
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200327 changed_lines_buf(buf, start_lnum, end_lnum + 1, 0);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100328 res = OK;
329
330theend:
331 vim_free(text);
332 return res;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200333}
334
335/*
336 * prop_add_list()
337 * First argument specifies the text property:
338 * {'type': <str>, 'id': <num>, 'bufnr': <num>}
339 * Second argument is a List where each item is a List with the following
340 * entries: [lnum, start_col, end_col]
341 */
342 void
343f_prop_add_list(typval_T *argvars, typval_T *rettv UNUSED)
344{
345 dict_T *dict;
346 char_u *type_name;
347 buf_T *buf = curbuf;
348 int id = 0;
349 listitem_T *li;
350 list_T *pos_list;
351 linenr_T start_lnum;
352 colnr_T start_col;
353 linenr_T end_lnum;
354 colnr_T end_col;
355 int error = FALSE;
Bram Moolenaar4997f2a2022-10-13 14:00:45 +0100356 int prev_did_emsg = did_emsg;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200357
358 if (check_for_dict_arg(argvars, 0) == FAIL
359 || check_for_list_arg(argvars, 1) == FAIL)
360 return;
361
Bram Moolenaard83392a2022-09-01 12:22:46 +0100362 if (check_for_nonnull_list_arg(argvars, 1) == FAIL)
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200363 return;
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200364
365 dict = argvars[0].vval.v_dict;
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +0100366 if (dict == NULL || !dict_has_key(dict, "type"))
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200367 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +0000368 emsg(_(e_missing_property_type_name));
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200369 return;
370 }
Bram Moolenaard61efa52022-07-23 09:52:04 +0100371 type_name = dict_get_string(dict, "type", FALSE);
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200372
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +0100373 if (dict_has_key(dict, "id"))
Bram Moolenaard61efa52022-07-23 09:52:04 +0100374 id = dict_get_number(dict, "id");
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200375
376 if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL)
377 return;
378
Paul Ollis4c3d21a2022-05-24 21:26:37 +0100379 // This must be done _before_ we start adding properties because property
380 // changes trigger buffer (memline) reorganisation, which needs this flag
381 // to be correctly set.
382 buf->b_has_textprop = TRUE; // this is never reset
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200383 FOR_ALL_LIST_ITEMS(argvars[1].vval.v_list, li)
384 {
385 if (li->li_tv.v_type != VAR_LIST || li->li_tv.vval.v_list == NULL)
386 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +0000387 emsg(_(e_list_required));
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200388 return;
389 }
390
391 pos_list = li->li_tv.vval.v_list;
392 start_lnum = list_find_nr(pos_list, 0L, &error);
Bram Moolenaar4997f2a2022-10-13 14:00:45 +0100393 if (!error)
394 start_col = list_find_nr(pos_list, 1L, &error);
395 if (!error)
396 end_lnum = list_find_nr(pos_list, 2L, &error);
397 if (!error)
398 end_col = list_find_nr(pos_list, 3L, &error);
Bram Moolenaard93009e2022-10-13 14:35:24 +0100399 int this_id = id;
400 if (!error && pos_list->lv_len > 4)
401 this_id = list_find_nr(pos_list, 4L, &error);
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200402 if (error || start_lnum <= 0 || start_col <= 0
Bram Moolenaar4997f2a2022-10-13 14:00:45 +0100403 || end_lnum <= 0 || end_col <= 0)
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200404 {
Bram Moolenaar4997f2a2022-10-13 14:00:45 +0100405 if (prev_did_emsg == did_emsg)
406 emsg(_(e_invalid_argument));
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200407 return;
408 }
Bram Moolenaard93009e2022-10-13 14:35:24 +0100409 if (prop_add_one(buf, type_name, this_id, NULL, 0, 0,
410 start_lnum, end_lnum, start_col, end_col) == FAIL)
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200411 return;
412 }
413
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100414 redraw_buf_later(buf, UPD_VALID);
Yegappan Lakshmananccfb7c62021-08-16 21:39:09 +0200415}
416
417/*
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100418 * Get the next ID to use for a textprop with text in buffer "buf".
419 */
420 static int
421get_textprop_id(buf_T *buf)
422{
423 // TODO: recycle deleted entries
424 return -(buf->b_textprop_text.ga_len + 1);
425}
426
Bram Moolenaar4ce1f992022-12-19 13:31:06 +0000427// Flag that is set when a negative ID isused for a normal text property.
428// It is then impossible to use virtual text properties.
429static int did_use_negative_pop_id = FALSE;
430
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100431/*
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200432 * Shared between prop_add() and popup_create().
433 * "dict_arg" is the function argument of a dict containing "bufnr".
434 * it is NULL for popup_create().
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100435 * Returns the "id" used for "text" or zero.
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200436 */
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100437 int
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200438prop_add_common(
439 linenr_T start_lnum,
440 colnr_T start_col,
441 dict_T *dict,
442 buf_T *default_buf,
443 typval_T *dict_arg)
444{
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200445 linenr_T end_lnum;
446 colnr_T end_col;
447 char_u *type_name;
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200448 buf_T *buf = default_buf;
449 int id = 0;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100450 char_u *text = NULL;
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100451 int text_padding_left = 0;
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100452 int flags = 0;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100453
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +0100454 if (dict == NULL || !dict_has_key(dict, "type"))
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100455 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +0000456 emsg(_(e_missing_property_type_name));
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100457 goto theend;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100458 }
Bram Moolenaard61efa52022-07-23 09:52:04 +0100459 type_name = dict_get_string(dict, "type", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100460
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +0100461 if (dict_has_key(dict, "end_lnum"))
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100462 {
Bram Moolenaard61efa52022-07-23 09:52:04 +0100463 end_lnum = dict_get_number(dict, "end_lnum");
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100464 if (end_lnum < start_lnum)
465 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000466 semsg(_(e_invalid_value_for_argument_str), "end_lnum");
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100467 goto theend;
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100468 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100469 }
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100470 else
471 end_lnum = start_lnum;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100472
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +0100473 if (dict_has_key(dict, "length"))
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100474 {
Bram Moolenaard61efa52022-07-23 09:52:04 +0100475 long length = dict_get_number(dict, "length");
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100476
Bram Moolenaarb9c67a52019-01-01 19:49:20 +0100477 if (length < 0 || end_lnum > start_lnum)
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100478 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000479 semsg(_(e_invalid_value_for_argument_str), "length");
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100480 goto theend;
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100481 }
Bram Moolenaarb9c67a52019-01-01 19:49:20 +0100482 end_col = start_col + length;
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100483 }
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +0100484 else if (dict_has_key(dict, "end_col"))
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100485 {
Bram Moolenaard61efa52022-07-23 09:52:04 +0100486 end_col = dict_get_number(dict, "end_col");
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100487 if (end_col <= 0)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100488 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000489 semsg(_(e_invalid_value_for_argument_str), "end_col");
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100490 goto theend;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100491 }
492 }
Bram Moolenaare3d31b02018-12-24 23:07:04 +0100493 else if (start_lnum == end_lnum)
494 end_col = start_col;
495 else
496 end_col = 1;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100497
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +0100498 if (dict_has_key(dict, "id"))
Bram Moolenaard61efa52022-07-23 09:52:04 +0100499 id = dict_get_number(dict, "id");
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100500
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100501 if (dict_has_key(dict, "text"))
502 {
Bram Moolenaarfb593c52022-09-17 18:57:36 +0100503 if (dict_has_key(dict, "length")
504 || dict_has_key(dict, "end_col")
505 || dict_has_key(dict, "end_lnum"))
506 {
507 emsg(_(e_cannot_use_length_endcol_and_endlnum_with_text));
508 goto theend;
509 }
510
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100511 text = dict_get_string(dict, "text", TRUE);
512 if (text == NULL)
513 goto theend;
514 // use a default length of 1 to make multiple props show up
515 end_col = start_col + 1;
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100516
517 if (dict_has_key(dict, "text_align"))
518 {
519 char_u *p = dict_get_string(dict, "text_align", FALSE);
520
521 if (p == NULL)
522 goto theend;
Bram Moolenaar82b14c12022-08-10 19:50:47 +0100523 if (start_col != 0)
524 {
525 emsg(_(e_can_only_use_text_align_when_column_is_zero));
526 goto theend;
527 }
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100528 if (STRCMP(p, "right") == 0)
529 flags |= TP_FLAG_ALIGN_RIGHT;
Bram Moolenaar04e0ed12022-09-10 20:00:56 +0100530 else if (STRCMP(p, "above") == 0)
531 flags |= TP_FLAG_ALIGN_ABOVE;
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100532 else if (STRCMP(p, "below") == 0)
533 flags |= TP_FLAG_ALIGN_BELOW;
534 else if (STRCMP(p, "after") != 0)
535 {
536 semsg(_(e_invalid_value_for_argument_str_str), "text_align", p);
537 goto theend;
538 }
539 }
540
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100541 if (dict_has_key(dict, "text_padding_left"))
542 {
543 text_padding_left = dict_get_number(dict, "text_padding_left");
544 if (text_padding_left < 0)
545 {
546 semsg(_(e_argument_must_be_positive_str), "text_padding_left");
547 goto theend;
548 }
549 }
550
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100551 if (dict_has_key(dict, "text_wrap"))
552 {
553 char_u *p = dict_get_string(dict, "text_wrap", FALSE);
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100554
Bram Moolenaarb7963df2022-07-31 17:12:43 +0100555 if (p == NULL)
556 goto theend;
557 if (STRCMP(p, "wrap") == 0)
558 flags |= TP_FLAG_WRAP;
559 else if (STRCMP(p, "truncate") != 0)
560 {
561 semsg(_(e_invalid_value_for_argument_str_str), "text_wrap", p);
562 goto theend;
563 }
564 }
565 }
566
567 // Column must be 1 or more for a normal text property; when "text" is
568 // present zero means it goes after the line.
569 if (start_col < (text == NULL ? 1 : 0))
570 {
571 semsg(_(e_invalid_column_number_nr), (long)start_col);
572 goto theend;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100573 }
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100574 if (start_col > 0 && text_padding_left > 0)
575 {
576 emsg(_(e_can_only_use_left_padding_when_column_is_zero));
577 goto theend;
578 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100579
Bram Moolenaar7a8d0272019-05-26 23:32:06 +0200580 if (dict_arg != NULL && get_bufnr_from_arg(dict_arg, &buf) == FAIL)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100581 goto theend;
582
Bram Moolenaar4ce1f992022-12-19 13:31:06 +0000583 if (id < 0)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100584 {
Bram Moolenaar4ce1f992022-12-19 13:31:06 +0000585 if (buf->b_textprop_text.ga_len > 0)
586 {
587 emsg(_(e_cannot_use_negative_id_after_adding_textprop_with_text));
588 goto theend;
589 }
590 did_use_negative_pop_id = TRUE;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100591 }
Bram Moolenaar4ce1f992022-12-19 13:31:06 +0000592
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100593 if (text != NULL)
Bram Moolenaar4ce1f992022-12-19 13:31:06 +0000594 {
595 if (did_use_negative_pop_id)
596 {
597 emsg(_(e_cannot_add_textprop_with_text_after_using_textprop_with_negative_id));
598 goto theend;
599 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100600 id = get_textprop_id(buf);
Bram Moolenaar4ce1f992022-12-19 13:31:06 +0000601 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100602
Paul Ollis4c3d21a2022-05-24 21:26:37 +0100603 // This must be done _before_ we add the property because property changes
604 // trigger buffer (memline) reorganisation, which needs this flag to be
605 // correctly set.
606 buf->b_has_textprop = TRUE; // this is never reset
607
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100608 prop_add_one(buf, type_name, id, text, text_padding_left, flags,
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100609 start_lnum, end_lnum, start_col, end_col);
610 text = NULL;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100611
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100612 redraw_buf_later(buf, UPD_VALID);
Bram Moolenaar7f9969c2022-07-25 18:13:54 +0100613
614theend:
615 vim_free(text);
616 return id;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100617}
618
619/*
Bram Moolenaarfb95e212018-12-14 12:18:11 +0100620 * Fetch the text properties for line "lnum" in buffer "buf".
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100621 * Returns the number of text properties and, when non-zero, a pointer to the
622 * first one in "props" (note that it is not aligned, therefore the char_u
623 * pointer).
624 */
625 int
626get_text_props(buf_T *buf, linenr_T lnum, char_u **props, int will_change)
627{
628 char_u *text;
629 size_t textlen;
630 size_t proplen;
631
Bram Moolenaarb413d2e2018-12-25 23:15:46 +0100632 // Be quick when no text property types have been defined or the buffer,
633 // unless we are adding one.
Bram Moolenaard79eef22019-05-24 20:41:55 +0200634 if ((!buf->b_has_textprop && !will_change) || buf->b_ml.ml_mfp == NULL)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100635 return 0;
636
637 // Fetch the line to get the ml_line_len field updated.
638 text = ml_get_buf(buf, lnum, will_change);
639 textlen = STRLEN(text) + 1;
640 proplen = buf->b_ml.ml_line_len - textlen;
Bram Moolenaar38ea2732022-08-07 22:04:56 +0100641 if (proplen == 0)
642 return 0;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100643 if (proplen % sizeof(textprop_T) != 0)
644 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +0000645 iemsg(_(e_text_property_info_corrupted));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100646 return 0;
647 }
Bram Moolenaar38ea2732022-08-07 22:04:56 +0100648 *props = text + textlen;
Bram Moolenaar4efe73b2018-12-16 14:37:39 +0100649 return (int)(proplen / sizeof(textprop_T));
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100650}
651
Bram Moolenaar4d91d342022-08-06 13:48:20 +0100652/*
Bram Moolenaarc9dc03f2022-09-12 17:51:07 +0100653 * Return the number of text properties with "above" or "below" alignment in
654 * line "lnum". A "right" aligned property also goes below after a "below" or
655 * other "right" aligned property.
Bram Moolenaar4d91d342022-08-06 13:48:20 +0100656 */
657 int
Bram Moolenaarc9dc03f2022-09-12 17:51:07 +0100658prop_count_above_below(buf_T *buf, linenr_T lnum)
Bram Moolenaar4d91d342022-08-06 13:48:20 +0100659{
660 char_u *props;
661 int count = get_text_props(buf, lnum, &props, FALSE);
662 int result = 0;
663 textprop_T prop;
664 int i;
Bram Moolenaar8f369fb2022-08-13 19:35:05 +0100665 int next_right_goes_below = FALSE;
Bram Moolenaar4d91d342022-08-06 13:48:20 +0100666
667 if (count == 0)
668 return 0;
669 for (i = 0; i < count; ++i)
670 {
671 mch_memmove(&prop, props + i * sizeof(prop), sizeof(prop));
Bram Moolenaar89469d12022-12-02 20:46:26 +0000672 if (prop.tp_col == MAXCOL && text_prop_type_valid(buf, &prop))
Bram Moolenaar8f369fb2022-08-13 19:35:05 +0100673 {
674 if ((prop.tp_flags & TP_FLAG_ALIGN_BELOW)
675 || (next_right_goes_below
676 && (prop.tp_flags & TP_FLAG_ALIGN_RIGHT)))
677 {
678 next_right_goes_below = TRUE;
679 ++result;
680 }
Bram Moolenaarc9dc03f2022-09-12 17:51:07 +0100681 else if (prop.tp_flags & TP_FLAG_ALIGN_ABOVE)
682 {
683 next_right_goes_below = FALSE;
684 ++result;
685 }
Bram Moolenaar8f369fb2022-08-13 19:35:05 +0100686 else if (prop.tp_flags & TP_FLAG_ALIGN_RIGHT)
687 next_right_goes_below = TRUE;
688 }
Bram Moolenaar4d91d342022-08-06 13:48:20 +0100689 }
690 return result;
691}
692
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200693/**
694 * Return the number of text properties on line "lnum" in the current buffer.
695 * When "only_starting" is true only text properties starting in this line will
696 * be considered.
Bram Moolenaar7d0f7e92022-08-06 17:10:57 +0100697 * When "last_line" is FALSE then text properties after the line are not
698 * counted.
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200699 */
700 int
Bram Moolenaare175dc62022-08-01 22:18:50 +0100701count_props(linenr_T lnum, int only_starting, int last_line)
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200702{
703 char_u *props;
704 int proplen = get_text_props(curbuf, lnum, &props, 0);
705 int result = proplen;
706 int i;
707 textprop_T prop;
708
Bram Moolenaare175dc62022-08-01 22:18:50 +0100709 for (i = 0; i < proplen; ++i)
710 {
711 mch_memmove(&prop, props + i * sizeof(prop), sizeof(prop));
Bram Moolenaar7d0f7e92022-08-06 17:10:57 +0100712 // A prop is dropped when in the first line and it continues from the
Bram Moolenaare175dc62022-08-01 22:18:50 +0100713 // previous line, or when not in the last line and it is virtual text
714 // after the line.
715 if ((only_starting && (prop.tp_flags & TP_FLAG_CONT_PREV))
Bram Moolenaar89469d12022-12-02 20:46:26 +0000716 || (!last_line && prop.tp_col == MAXCOL)
717 || !text_prop_type_valid(curbuf, &prop))
Bram Moolenaare175dc62022-08-01 22:18:50 +0100718 --result;
719 }
Bram Moolenaar87be9be2020-05-30 15:32:02 +0200720 return result;
721}
722
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100723static textprop_T *text_prop_compare_props;
724static buf_T *text_prop_compare_buf;
725
Bram Moolenaar04e0ed12022-09-10 20:00:56 +0100726/* Score for sorting on position of the text property: 0: above,
727 * 1: after (default), 2: right, 3: below (comes last)
728 */
729 static int
730text_prop_order(int flags)
731{
732 if (flags & TP_FLAG_ALIGN_ABOVE)
733 return 0;
734 if (flags & TP_FLAG_ALIGN_RIGHT)
735 return 2;
736 if (flags & TP_FLAG_ALIGN_BELOW)
737 return 3;
738 return 1;
739}
740
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100741/*
742 * Function passed to qsort() to sort text properties.
743 * Return 1 if "s1" has priority over "s2", -1 if the other way around, zero if
744 * both have the same priority.
745 */
746 static int
747text_prop_compare(const void *s1, const void *s2)
748{
749 int idx1, idx2;
750 textprop_T *tp1, *tp2;
751 proptype_T *pt1, *pt2;
752 colnr_T col1, col2;
753
754 idx1 = *(int *)s1;
755 idx2 = *(int *)s2;
756 tp1 = &text_prop_compare_props[idx1];
757 tp2 = &text_prop_compare_props[idx2];
758 col1 = tp1->tp_col;
759 col2 = tp2->tp_col;
760 if (col1 == MAXCOL && col2 == MAXCOL)
761 {
Bram Moolenaar04e0ed12022-09-10 20:00:56 +0100762 int order1 = text_prop_order(tp1->tp_flags);
763 int order2 = text_prop_order(tp2->tp_flags);
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100764
Bram Moolenaar04e0ed12022-09-10 20:00:56 +0100765 // both props add text before or after the line, sort on order where it
766 // is added
767 if (order1 != order2)
768 return order1 < order2 ? 1 : -1;
Bram Moolenaarf396ce82022-08-23 18:39:37 +0100769 }
770
771 // property that inserts text has priority over one that doesn't
772 if ((tp1->tp_id < 0) != (tp2->tp_id < 0))
773 return tp1->tp_id < 0 ? 1 : -1;
774
775 // check highest priority, defined by the type
776 pt1 = text_prop_type_by_id(text_prop_compare_buf, tp1->tp_type);
777 pt2 = text_prop_type_by_id(text_prop_compare_buf, tp2->tp_type);
778 if (pt1 != pt2)
779 {
780 if (pt1 == NULL)
781 return -1;
782 if (pt2 == NULL)
783 return 1;
784 if (pt1->pt_priority != pt2->pt_priority)
785 return pt1->pt_priority > pt2->pt_priority ? 1 : -1;
786 }
787
788 // same priority, one that starts first wins
789 if (col1 != col2)
790 return col1 < col2 ? 1 : -1;
791
792 // for a property with text the id can be used as tie breaker
793 if (tp1->tp_id < 0)
794 return tp1->tp_id > tp2->tp_id ? 1 : -1;
795
796 return 0;
797}
798
799/*
800 * Sort "count" text properties using an array if indexes "idxs" into the list
801 * of text props "props" for buffer "buf".
802 */
803 void
804sort_text_props(
805 buf_T *buf,
806 textprop_T *props,
807 int *idxs,
808 int count)
809{
810 text_prop_compare_buf = buf;
811 text_prop_compare_props = props;
812 qsort((void *)idxs, (size_t)count, sizeof(int), text_prop_compare);
813}
814
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100815/*
Bram Moolenaar12034e22019-08-25 22:25:02 +0200816 * Find text property "type_id" in the visible lines of window "wp".
817 * Match "id" when it is > 0.
818 * Returns FAIL when not found.
819 */
820 int
Bram Moolenaar89469d12022-12-02 20:46:26 +0000821find_visible_prop(
822 win_T *wp,
823 int type_id,
824 int id,
825 textprop_T *prop,
826 linenr_T *found_lnum)
Bram Moolenaar12034e22019-08-25 22:25:02 +0200827{
Bram Moolenaar89469d12022-12-02 20:46:26 +0000828 // return when "type_id" no longer exists
829 if (text_prop_type_by_id(wp->w_buffer, type_id) == NULL)
830 return FAIL;
Bram Moolenaar12034e22019-08-25 22:25:02 +0200831
832 // w_botline may not have been updated yet.
Bram Moolenaar23999d72020-12-23 14:36:00 +0100833 validate_botline_win(wp);
Bram Moolenaar89469d12022-12-02 20:46:26 +0000834 for (linenr_T lnum = wp->w_topline; lnum < wp->w_botline; ++lnum)
Bram Moolenaar12034e22019-08-25 22:25:02 +0200835 {
Bram Moolenaar89469d12022-12-02 20:46:26 +0000836 char_u *props;
837 int count = get_text_props(wp->w_buffer, lnum, &props, FALSE);
838 for (int i = 0; i < count; ++i)
Bram Moolenaar12034e22019-08-25 22:25:02 +0200839 {
840 mch_memmove(prop, props + i * sizeof(textprop_T),
841 sizeof(textprop_T));
842 if (prop->tp_type == type_id && (id <= 0 || prop->tp_id == id))
843 {
844 *found_lnum = lnum;
845 return OK;
846 }
847 }
848 }
849 return FAIL;
850}
851
852/*
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100853 * Set the text properties for line "lnum" to "props" with length "len".
854 * If "len" is zero text properties are removed, "props" is not used.
855 * Any existing text properties are dropped.
856 * Only works for the current buffer.
857 */
858 static void
859set_text_props(linenr_T lnum, char_u *props, int len)
860{
Bram Moolenaar8aef43b2019-01-08 20:14:35 +0100861 char_u *text;
862 char_u *newtext;
863 int textlen;
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100864
865 text = ml_get(lnum);
Bram Moolenaar8aef43b2019-01-08 20:14:35 +0100866 textlen = (int)STRLEN(text) + 1;
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100867 newtext = alloc(textlen + len);
868 if (newtext == NULL)
869 return;
870 mch_memmove(newtext, text, textlen);
871 if (len > 0)
872 mch_memmove(newtext + textlen, props, len);
Bram Moolenaarfa4873c2022-06-30 22:13:59 +0100873 if (curbuf->b_ml.ml_flags & (ML_LINE_DIRTY | ML_ALLOCATED))
Bram Moolenaar4164bb22019-01-04 23:09:49 +0100874 vim_free(curbuf->b_ml.ml_line_ptr);
875 curbuf->b_ml.ml_line_ptr = newtext;
876 curbuf->b_ml.ml_line_len = textlen + len;
877 curbuf->b_ml.ml_flags |= ML_LINE_DIRTY;
878}
879
Bram Moolenaar213bbaf2022-08-05 19:46:48 +0100880/*
Bram Moolenaar7d0f7e92022-08-06 17:10:57 +0100881 * Add "text_props" with "text_prop_count" text properties to line "lnum".
Bram Moolenaar213bbaf2022-08-05 19:46:48 +0100882 */
883 void
884add_text_props(linenr_T lnum, textprop_T *text_props, int text_prop_count)
885{
886 char_u *text;
887 char_u *newtext;
888 int proplen = text_prop_count * (int)sizeof(textprop_T);
889
890 text = ml_get(lnum);
891 newtext = alloc(curbuf->b_ml.ml_line_len + proplen);
892 if (newtext == NULL)
893 return;
894 mch_memmove(newtext, text, curbuf->b_ml.ml_line_len);
895 mch_memmove(newtext + curbuf->b_ml.ml_line_len, text_props, proplen);
896 if (curbuf->b_ml.ml_flags & (ML_LINE_DIRTY | ML_ALLOCATED))
897 vim_free(curbuf->b_ml.ml_line_ptr);
898 curbuf->b_ml.ml_line_ptr = newtext;
899 curbuf->b_ml.ml_line_len += proplen;
900 curbuf->b_ml.ml_flags |= ML_LINE_DIRTY;
901}
902
Bram Moolenaare44336b2022-08-07 18:20:08 +0100903/*
904 * Function passed to qsort() for sorting proptype_T on pt_id.
905 */
906 static int
907compare_pt(const void *s1, const void *s2)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100908{
Bram Moolenaare44336b2022-08-07 18:20:08 +0100909 proptype_T *tp1 = *(proptype_T **)s1;
910 proptype_T *tp2 = *(proptype_T **)s2;
911
912 return tp1->pt_id == tp2->pt_id ? 0 : tp1->pt_id < tp2->pt_id ? -1 : 1;
913}
914
915 static proptype_T *
916find_type_by_id(hashtab_T *ht, proptype_T ***array, int id)
917{
918 int low = 0;
919 int high;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100920
Bram Moolenaar10246902022-08-08 17:08:05 +0100921 if (ht == NULL || ht->ht_used == 0)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100922 return NULL;
923
dundargocc57b5bc2022-11-02 13:30:51 +0000924 // Make the lookup faster by creating an array with pointers to
Bram Moolenaare44336b2022-08-07 18:20:08 +0100925 // hashtable entries, sorted on pt_id.
926 if (*array == NULL)
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100927 {
Bram Moolenaare44336b2022-08-07 18:20:08 +0100928 long todo;
929 hashitem_T *hi;
930 int i = 0;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100931
Bram Moolenaare44336b2022-08-07 18:20:08 +0100932 *array = ALLOC_MULT(proptype_T *, ht->ht_used);
933 if (*array == NULL)
934 return NULL;
935 todo = (long)ht->ht_used;
936 for (hi = ht->ht_array; todo > 0; ++hi)
937 {
938 if (!HASHITEM_EMPTY(hi))
939 {
940 (*array)[i++] = HI2PT(hi);
941 --todo;
942 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100943 }
Bram Moolenaare44336b2022-08-07 18:20:08 +0100944 qsort((void *)*array, ht->ht_used, sizeof(proptype_T *), compare_pt);
945 }
946
947 // binary search in the sorted array
948 high = ht->ht_used;
949 while (high > low)
950 {
951 int m = (high + low) / 2;
952
953 if ((*array)[m]->pt_id == id)
954 return (*array)[m];
955 if ((*array)[m]->pt_id > id)
956 high = m;
957 else
958 low = m + 1;
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100959 }
960 return NULL;
961}
962
963/*
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100964 * Fill 'dict' with text properties in 'prop'.
965 */
966 static void
967prop_fill_dict(dict_T *dict, textprop_T *prop, buf_T *buf)
968{
969 proptype_T *pt;
Martin Tournoije2390c72021-07-28 13:30:16 +0200970 int buflocal = TRUE;
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100971
972 dict_add_number(dict, "col", prop->tp_col);
973 dict_add_number(dict, "length", prop->tp_len);
974 dict_add_number(dict, "id", prop->tp_id);
975 dict_add_number(dict, "start", !(prop->tp_flags & TP_FLAG_CONT_PREV));
976 dict_add_number(dict, "end", !(prop->tp_flags & TP_FLAG_CONT_NEXT));
Martin Tournoije2390c72021-07-28 13:30:16 +0200977
Bram Moolenaare44336b2022-08-07 18:20:08 +0100978 pt = find_type_by_id(buf->b_proptypes, &buf->b_proparray, prop->tp_type);
Martin Tournoije2390c72021-07-28 13:30:16 +0200979 if (pt == NULL)
980 {
Bram Moolenaare44336b2022-08-07 18:20:08 +0100981 pt = find_type_by_id(global_proptypes, &global_proparray,
982 prop->tp_type);
Martin Tournoije2390c72021-07-28 13:30:16 +0200983 buflocal = FALSE;
984 }
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100985 if (pt != NULL)
986 dict_add_string(dict, "type", pt->pt_name);
Martin Tournoije2390c72021-07-28 13:30:16 +0200987
988 if (buflocal)
989 dict_add_number(dict, "type_bufnr", buf->b_fnum);
990 else
991 dict_add_number(dict, "type_bufnr", 0);
Bram Moolenaare05a89a2020-01-10 19:56:46 +0100992}
993
994/*
Bram Moolenaar98aefe72018-12-13 22:20:09 +0100995 * Find a property type by ID in "buf" or globally.
996 * Returns NULL if not found.
997 */
998 proptype_T *
999text_prop_type_by_id(buf_T *buf, int id)
1000{
1001 proptype_T *type;
1002
Bram Moolenaare44336b2022-08-07 18:20:08 +01001003 type = find_type_by_id(buf->b_proptypes, &buf->b_proparray, id);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001004 if (type == NULL)
Bram Moolenaare44336b2022-08-07 18:20:08 +01001005 type = find_type_by_id(global_proptypes, &global_proparray, id);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001006 return type;
1007}
1008
1009/*
Bram Moolenaar89469d12022-12-02 20:46:26 +00001010 * Return TRUE if "prop" is a valid text property type.
1011 */
1012 int
1013text_prop_type_valid(buf_T *buf, textprop_T *prop)
1014{
1015 return text_prop_type_by_id(buf, prop->tp_type) != NULL;
1016}
1017
1018/*
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001019 * prop_clear({lnum} [, {lnum_end} [, {bufnr}]])
1020 */
1021 void
1022f_prop_clear(typval_T *argvars, typval_T *rettv UNUSED)
1023{
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001024 linenr_T start;
1025 linenr_T end;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001026 linenr_T lnum;
1027 buf_T *buf = curbuf;
Bram Moolenaarda1dbed2021-03-22 19:43:34 +01001028 int did_clear = FALSE;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001029
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001030 if (in_vim9script()
1031 && (check_for_number_arg(argvars, 0) == FAIL
1032 || check_for_opt_number_arg(argvars, 1) == FAIL
1033 || (argvars[1].v_type != VAR_UNKNOWN
1034 && check_for_opt_dict_arg(argvars, 2) == FAIL)))
1035 return;
1036
1037 start = tv_get_number(&argvars[0]);
1038 end = start;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001039 if (argvars[1].v_type != VAR_UNKNOWN)
1040 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001041 end = tv_get_number(&argvars[1]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001042 if (argvars[2].v_type != VAR_UNKNOWN)
1043 {
1044 if (get_bufnr_from_arg(&argvars[2], &buf) == FAIL)
1045 return;
1046 }
1047 }
1048 if (start < 1 || end < 1)
1049 {
Bram Moolenaar108010a2021-06-27 22:03:33 +02001050 emsg(_(e_invalid_range));
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001051 return;
1052 }
1053
1054 for (lnum = start; lnum <= end; ++lnum)
1055 {
1056 char_u *text;
1057 size_t len;
1058
1059 if (lnum > buf->b_ml.ml_line_count)
1060 break;
1061 text = ml_get_buf(buf, lnum, FALSE);
1062 len = STRLEN(text) + 1;
1063 if ((size_t)buf->b_ml.ml_line_len > len)
1064 {
Bram Moolenaarda1dbed2021-03-22 19:43:34 +01001065 did_clear = TRUE;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001066 if (!(buf->b_ml.ml_flags & ML_LINE_DIRTY))
1067 {
1068 char_u *newtext = vim_strsave(text);
1069
1070 // need to allocate the line now
1071 if (newtext == NULL)
1072 return;
Bram Moolenaarfa4873c2022-06-30 22:13:59 +01001073 if (buf->b_ml.ml_flags & ML_ALLOCATED)
1074 vim_free(buf->b_ml.ml_line_ptr);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001075 buf->b_ml.ml_line_ptr = newtext;
1076 buf->b_ml.ml_flags |= ML_LINE_DIRTY;
1077 }
Bram Moolenaar4efe73b2018-12-16 14:37:39 +01001078 buf->b_ml.ml_line_len = (int)len;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001079 }
1080 }
Bram Moolenaarda1dbed2021-03-22 19:43:34 +01001081 if (did_clear)
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001082 redraw_buf_later(buf, UPD_NOT_VALID);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001083}
1084
1085/*
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001086 * prop_find({props} [, {direction}])
1087 */
1088 void
1089f_prop_find(typval_T *argvars, typval_T *rettv)
1090{
1091 pos_T *cursor = &curwin->w_cursor;
1092 dict_T *dict;
1093 buf_T *buf = curbuf;
1094 dictitem_T *di;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001095 int lnum_start;
1096 int start_pos_has_prop = 0;
LemonBoy9bd3ce22022-04-18 21:54:02 +01001097 int seen_end = FALSE;
Bram Moolenaar0d4d9ee2021-08-01 19:28:15 +02001098 int id = 0;
1099 int id_found = FALSE;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001100 int type_id = -1;
LemonBoy9bd3ce22022-04-18 21:54:02 +01001101 int skipstart = FALSE;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001102 int lnum = -1;
1103 int col = -1;
LemonBoy9bd3ce22022-04-18 21:54:02 +01001104 int dir = FORWARD; // FORWARD == 1, BACKWARD == -1
Bram Moolenaar24f21fd2021-03-27 22:07:29 +01001105 int both;
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001106
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001107 if (in_vim9script()
1108 && (check_for_dict_arg(argvars, 0) == FAIL
1109 || check_for_opt_string_arg(argvars, 1) == FAIL))
1110 return;
1111
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +01001112 if (check_for_nonnull_dict_arg(argvars, 0) == FAIL)
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001113 return;
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001114 dict = argvars[0].vval.v_dict;
1115
1116 if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL)
1117 return;
1118 if (buf->b_ml.ml_mfp == NULL)
1119 return;
1120
1121 if (argvars[1].v_type != VAR_UNKNOWN)
1122 {
1123 char_u *dir_s = tv_get_string(&argvars[1]);
1124
1125 if (*dir_s == 'b')
LemonBoy9bd3ce22022-04-18 21:54:02 +01001126 dir = BACKWARD;
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001127 else if (*dir_s != 'f')
1128 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001129 emsg(_(e_invalid_argument));
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001130 return;
1131 }
1132 }
1133
1134 di = dict_find(dict, (char_u *)"lnum", -1);
1135 if (di != NULL)
1136 lnum = tv_get_number(&di->di_tv);
1137
1138 di = dict_find(dict, (char_u *)"col", -1);
1139 if (di != NULL)
1140 col = tv_get_number(&di->di_tv);
1141
1142 if (lnum == -1)
1143 {
1144 lnum = cursor->lnum;
1145 col = cursor->col + 1;
1146 }
1147 else if (col == -1)
1148 col = 1;
1149
1150 if (lnum < 1 || lnum > buf->b_ml.ml_line_count)
1151 {
Bram Moolenaar108010a2021-06-27 22:03:33 +02001152 emsg(_(e_invalid_range));
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001153 return;
1154 }
1155
Bram Moolenaard61efa52022-07-23 09:52:04 +01001156 skipstart = dict_get_bool(dict, "skipstart", 0);
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001157
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001158 if (dict_has_key(dict, "id"))
Bram Moolenaar8e3fc132021-07-31 18:33:57 +02001159 {
Bram Moolenaard61efa52022-07-23 09:52:04 +01001160 id = dict_get_number(dict, "id");
Bram Moolenaare041dde2021-08-01 21:30:12 +02001161 id_found = TRUE;
Bram Moolenaar8e3fc132021-07-31 18:33:57 +02001162 }
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001163 if (dict_has_key(dict, "type"))
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001164 {
Bram Moolenaard61efa52022-07-23 09:52:04 +01001165 char_u *name = dict_get_string(dict, "type", FALSE);
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001166 proptype_T *type = lookup_prop_type(name, buf);
1167
1168 if (type == NULL)
1169 return;
1170 type_id = type->pt_id;
1171 }
Bram Moolenaard61efa52022-07-23 09:52:04 +01001172 both = dict_get_bool(dict, "both", FALSE);
Bram Moolenaar0d4d9ee2021-08-01 19:28:15 +02001173 if (!id_found && type_id == -1)
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001174 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001175 emsg(_(e_need_at_least_one_of_id_or_type));
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001176 return;
1177 }
Bram Moolenaar0d4d9ee2021-08-01 19:28:15 +02001178 if (both && (!id_found || type_id == -1))
Bram Moolenaar24f21fd2021-03-27 22:07:29 +01001179 {
Ben Jacksona7704222022-08-20 20:54:51 +01001180 emsg(_(e_need_id_and_type_or_types_with_both));
Bram Moolenaar24f21fd2021-03-27 22:07:29 +01001181 return;
1182 }
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001183
1184 lnum_start = lnum;
1185
1186 if (rettv_dict_alloc(rettv) == FAIL)
1187 return;
1188
1189 while (1)
1190 {
1191 char_u *text = ml_get_buf(buf, lnum, FALSE);
1192 size_t textlen = STRLEN(text) + 1;
1193 int count = (int)((buf->b_ml.ml_line_len - textlen)
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001194 / sizeof(textprop_T));
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001195 int i;
1196 textprop_T prop;
1197 int prop_start;
1198 int prop_end;
1199
LemonBoy9bd3ce22022-04-18 21:54:02 +01001200 for (i = dir == BACKWARD ? count - 1 : 0; i >= 0 && i < count; i += dir)
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001201 {
1202 mch_memmove(&prop, text + textlen + i * sizeof(textprop_T),
LemonBoy9bd3ce22022-04-18 21:54:02 +01001203 sizeof(textprop_T));
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001204
LemonBoy9bd3ce22022-04-18 21:54:02 +01001205 // For the very first line try to find the first property before or
1206 // after `col`, depending on the search direction.
Bram Moolenaar346f18e2020-03-13 21:36:40 +01001207 if (lnum == lnum_start)
Bram Moolenaar965fd8d2020-03-14 07:46:40 +01001208 {
LemonBoy9bd3ce22022-04-18 21:54:02 +01001209 if (dir == BACKWARD)
Bram Moolenaar346f18e2020-03-13 21:36:40 +01001210 {
LemonBoy9bd3ce22022-04-18 21:54:02 +01001211 if (prop.tp_col > col)
1212 continue;
Bram Moolenaar346f18e2020-03-13 21:36:40 +01001213 }
1214 else if (prop.tp_col + prop.tp_len - (prop.tp_len != 0) < col)
1215 continue;
Bram Moolenaar965fd8d2020-03-14 07:46:40 +01001216 }
Bram Moolenaar24f21fd2021-03-27 22:07:29 +01001217 if (both ? prop.tp_id == id && prop.tp_type == type_id
Bram Moolenaar0d4d9ee2021-08-01 19:28:15 +02001218 : (id_found && prop.tp_id == id)
1219 || prop.tp_type == type_id)
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001220 {
1221 // Check if the starting position has text props.
Bram Moolenaar66b98852020-03-11 19:15:52 +01001222 if (lnum_start == lnum
1223 && col >= prop.tp_col
1224 && (col <= prop.tp_col + prop.tp_len
1225 - (prop.tp_len != 0)))
1226 start_pos_has_prop = 1;
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001227
LemonBoy9bd3ce22022-04-18 21:54:02 +01001228 // The property was not continued from last line, it starts on
1229 // this line.
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001230 prop_start = !(prop.tp_flags & TP_FLAG_CONT_PREV);
LemonBoy9bd3ce22022-04-18 21:54:02 +01001231 // The property does not continue on the next line, it ends on
1232 // this line.
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001233 prop_end = !(prop.tp_flags & TP_FLAG_CONT_NEXT);
LemonBoy9bd3ce22022-04-18 21:54:02 +01001234 if (!prop_start && prop_end && dir == FORWARD)
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001235 seen_end = 1;
1236
1237 // Skip lines without the start flag.
1238 if (!prop_start)
1239 {
1240 // Always search backwards for start when search started
1241 // on a prop and we're not skipping.
1242 if (start_pos_has_prop && !skipstart)
LemonBoy9bd3ce22022-04-18 21:54:02 +01001243 dir = BACKWARD;
Bram Moolenaar4da7a252020-09-02 19:59:00 +02001244 continue;
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001245 }
1246
1247 // If skipstart is true, skip the prop at start pos (even if
1248 // continued from another line).
1249 if (start_pos_has_prop && skipstart && !seen_end)
1250 {
1251 start_pos_has_prop = 0;
Bram Moolenaar4da7a252020-09-02 19:59:00 +02001252 continue;
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001253 }
1254
Bram Moolenaare05a89a2020-01-10 19:56:46 +01001255 prop_fill_dict(rettv->vval.v_dict, &prop, buf);
1256 dict_add_number(rettv->vval.v_dict, "lnum", lnum);
1257
1258 return;
1259 }
1260 }
1261
1262 if (dir > 0)
1263 {
1264 if (lnum >= buf->b_ml.ml_line_count)
1265 break;
1266 lnum++;
1267 }
1268 else
1269 {
1270 if (lnum <= 1)
1271 break;
1272 lnum--;
1273 }
1274 }
1275}
1276
1277/*
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001278 * Returns TRUE if 'type_or_id' is in the 'types_or_ids' list.
1279 */
1280 static int
1281prop_type_or_id_in_list(int *types_or_ids, int len, int type_or_id)
1282{
1283 int i;
1284
1285 for (i = 0; i < len; i++)
1286 if (types_or_ids[i] == type_or_id)
1287 return TRUE;
1288
1289 return FALSE;
1290}
1291
1292/*
1293 * Return all the text properties in line 'lnum' in buffer 'buf' in 'retlist'.
1294 * If 'prop_types' is not NULL, then return only the text properties with
1295 * matching property type in the 'prop_types' array.
1296 * If 'prop_ids' is not NULL, then return only the text properties with
1297 * an identifier in the 'props_ids' array.
1298 * If 'add_lnum' is TRUE, then add the line number also to the text property
1299 * dictionary.
1300 */
1301 static void
1302get_props_in_line(
1303 buf_T *buf,
1304 linenr_T lnum,
1305 int *prop_types,
1306 int prop_types_len,
1307 int *prop_ids,
1308 int prop_ids_len,
1309 list_T *retlist,
1310 int add_lnum)
1311{
1312 char_u *text = ml_get_buf(buf, lnum, FALSE);
1313 size_t textlen = STRLEN(text) + 1;
1314 int count;
1315 int i;
1316 textprop_T prop;
1317
1318 count = (int)((buf->b_ml.ml_line_len - textlen) / sizeof(textprop_T));
1319 for (i = 0; i < count; ++i)
1320 {
1321 mch_memmove(&prop, text + textlen + i * sizeof(textprop_T),
1322 sizeof(textprop_T));
1323 if ((prop_types == NULL
1324 || prop_type_or_id_in_list(prop_types, prop_types_len,
1325 prop.tp_type))
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001326 && (prop_ids == NULL
1327 || prop_type_or_id_in_list(prop_ids, prop_ids_len,
1328 prop.tp_id)))
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001329 {
1330 dict_T *d = dict_alloc();
1331
1332 if (d == NULL)
1333 break;
1334 prop_fill_dict(d, &prop, buf);
1335 if (add_lnum)
1336 dict_add_number(d, "lnum", lnum);
1337 list_append_dict(retlist, d);
1338 }
1339 }
1340}
1341
1342/*
1343 * Convert a List of property type names into an array of property type
1344 * identifiers. Returns a pointer to the allocated array. Returns NULL on
1345 * error. 'num_types' is set to the number of returned property types.
1346 */
1347 static int *
1348get_prop_types_from_names(list_T *l, buf_T *buf, int *num_types)
1349{
1350 int *prop_types;
1351 listitem_T *li;
1352 int i;
1353 char_u *name;
1354 proptype_T *type;
1355
1356 *num_types = 0;
1357
1358 prop_types = ALLOC_MULT(int, list_len(l));
1359 if (prop_types == NULL)
1360 return NULL;
1361
1362 i = 0;
1363 FOR_ALL_LIST_ITEMS(l, li)
1364 {
1365 if (li->li_tv.v_type != VAR_STRING)
1366 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001367 emsg(_(e_string_required));
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001368 goto errret;
1369 }
1370 name = li->li_tv.vval.v_string;
1371 if (name == NULL)
1372 goto errret;
1373
1374 type = lookup_prop_type(name, buf);
1375 if (type == NULL)
1376 goto errret;
1377 prop_types[i++] = type->pt_id;
1378 }
1379
1380 *num_types = i;
1381 return prop_types;
1382
1383errret:
1384 VIM_CLEAR(prop_types);
1385 return NULL;
1386}
1387
1388/*
1389 * Convert a List of property identifiers into an array of property
1390 * identifiers. Returns a pointer to the allocated array. Returns NULL on
1391 * error. 'num_ids' is set to the number of returned property identifiers.
1392 */
1393 static int *
1394get_prop_ids_from_list(list_T *l, int *num_ids)
1395{
1396 int *prop_ids;
1397 listitem_T *li;
1398 int i;
1399 int id;
1400 int error;
1401
1402 *num_ids = 0;
1403
1404 prop_ids = ALLOC_MULT(int, list_len(l));
1405 if (prop_ids == NULL)
1406 return NULL;
1407
1408 i = 0;
1409 FOR_ALL_LIST_ITEMS(l, li)
1410 {
1411 error = FALSE;
1412 id = tv_get_number_chk(&li->li_tv, &error);
1413 if (error)
1414 goto errret;
1415
1416 prop_ids[i++] = id;
1417 }
1418
1419 *num_ids = i;
1420 return prop_ids;
1421
1422errret:
1423 VIM_CLEAR(prop_ids);
1424 return NULL;
1425}
1426
1427/*
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001428 * prop_list({lnum} [, {bufnr}])
1429 */
1430 void
1431f_prop_list(typval_T *argvars, typval_T *rettv)
1432{
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001433 linenr_T lnum;
1434 linenr_T start_lnum;
1435 linenr_T end_lnum;
1436 buf_T *buf = curbuf;
1437 int add_lnum = FALSE;
1438 int *prop_types = NULL;
1439 int prop_types_len = 0;
1440 int *prop_ids = NULL;
1441 int prop_ids_len = 0;
1442 list_T *l;
1443 dictitem_T *di;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001444
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001445 if (in_vim9script()
1446 && (check_for_number_arg(argvars, 0) == FAIL
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001447 || check_for_opt_dict_arg(argvars, 1) == FAIL))
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001448 return;
1449
Bram Moolenaar93a10962022-06-16 11:42:09 +01001450 if (rettv_list_alloc(rettv) == FAIL)
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001451 return;
1452
1453 // default: get text properties on current line
1454 start_lnum = tv_get_number(&argvars[0]);
1455 end_lnum = start_lnum;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001456 if (argvars[1].v_type != VAR_UNKNOWN)
1457 {
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001458 dict_T *d;
1459
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +01001460 if (check_for_dict_arg(argvars, 1) == FAIL)
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001461 return;
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001462 d = argvars[1].vval.v_dict;
1463
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001464 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
1465 return;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001466
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001467 if (d != NULL && (di = dict_find(d, (char_u *)"end_lnum", -1)) != NULL)
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001468 {
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001469 if (di->di_tv.v_type != VAR_NUMBER)
1470 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001471 emsg(_(e_number_required));
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001472 return;
1473 }
1474 end_lnum = tv_get_number(&di->di_tv);
1475 if (end_lnum < 0)
1476 // negative end_lnum is used as an offset from the last buffer
1477 // line
1478 end_lnum = buf->b_ml.ml_line_count + end_lnum + 1;
1479 else if (end_lnum > buf->b_ml.ml_line_count)
1480 end_lnum = buf->b_ml.ml_line_count;
1481 add_lnum = TRUE;
1482 }
1483 if (d != NULL && (di = dict_find(d, (char_u *)"types", -1)) != NULL)
1484 {
1485 if (di->di_tv.v_type != VAR_LIST)
1486 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001487 emsg(_(e_list_required));
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001488 return;
1489 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001490
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001491 l = di->di_tv.vval.v_list;
1492 if (l != NULL && list_len(l) > 0)
1493 {
1494 prop_types = get_prop_types_from_names(l, buf, &prop_types_len);
1495 if (prop_types == NULL)
1496 return;
1497 }
1498 }
1499 if (d != NULL && (di = dict_find(d, (char_u *)"ids", -1)) != NULL)
1500 {
1501 if (di->di_tv.v_type != VAR_LIST)
1502 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001503 emsg(_(e_list_required));
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001504 goto errret;
1505 }
1506
1507 l = di->di_tv.vval.v_list;
1508 if (l != NULL && list_len(l) > 0)
1509 {
1510 prop_ids = get_prop_ids_from_list(l, &prop_ids_len);
1511 if (prop_ids == NULL)
1512 goto errret;
1513 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001514 }
1515 }
Yegappan Lakshmanane0216622021-11-23 11:46:32 +00001516 if (start_lnum < 1 || start_lnum > buf->b_ml.ml_line_count
1517 || end_lnum < 1 || end_lnum < start_lnum)
1518 emsg(_(e_invalid_range));
1519 else
1520 for (lnum = start_lnum; lnum <= end_lnum; lnum++)
1521 get_props_in_line(buf, lnum, prop_types, prop_types_len,
1522 prop_ids, prop_ids_len,
1523 rettv->vval.v_list, add_lnum);
1524
1525errret:
1526 VIM_CLEAR(prop_types);
1527 VIM_CLEAR(prop_ids);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001528}
1529
1530/*
1531 * prop_remove({props} [, {lnum} [, {lnum_end}]])
1532 */
1533 void
1534f_prop_remove(typval_T *argvars, typval_T *rettv)
1535{
1536 linenr_T start = 1;
1537 linenr_T end = 0;
1538 linenr_T lnum;
Bram Moolenaar965c0442021-05-17 00:22:06 +02001539 linenr_T first_changed = 0;
1540 linenr_T last_changed = 0;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001541 dict_T *dict;
1542 buf_T *buf = curbuf;
Bram Moolenaara5a40c52020-09-05 20:50:49 +02001543 int do_all;
Bram Moolenaar3a4cd392022-07-30 22:17:18 +01001544 int id = -MAXCOL;
Ben Jacksona7704222022-08-20 20:54:51 +01001545 int type_id = -1; // for a single "type"
1546 int *type_ids = NULL; // array, for a list of "types", allocated
1547 int num_type_ids = 0; // number of elements in "type_ids"
Bram Moolenaara5a40c52020-09-05 20:50:49 +02001548 int both;
Bram Moolenaar3a4cd392022-07-30 22:17:18 +01001549 int did_remove_text = FALSE;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001550
1551 rettv->vval.v_number = 0;
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02001552
1553 if (in_vim9script()
1554 && (check_for_dict_arg(argvars, 0) == FAIL
1555 || check_for_opt_number_arg(argvars, 1) == FAIL
1556 || (argvars[1].v_type != VAR_UNKNOWN
1557 && check_for_opt_number_arg(argvars, 2) == FAIL)))
1558 return;
1559
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +01001560 if (check_for_nonnull_dict_arg(argvars, 0) == FAIL)
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001561 return;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001562
1563 if (argvars[1].v_type != VAR_UNKNOWN)
1564 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001565 start = tv_get_number(&argvars[1]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001566 end = start;
1567 if (argvars[2].v_type != VAR_UNKNOWN)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001568 end = tv_get_number(&argvars[2]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001569 if (start < 1 || end < 1)
1570 {
Bram Moolenaar108010a2021-06-27 22:03:33 +02001571 emsg(_(e_invalid_range));
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001572 return;
1573 }
1574 }
1575
1576 dict = argvars[0].vval.v_dict;
Bram Moolenaarf0884c52019-05-24 21:22:29 +02001577 if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL)
1578 return;
1579 if (buf->b_ml.ml_mfp == NULL)
1580 return;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001581
Bram Moolenaard61efa52022-07-23 09:52:04 +01001582 do_all = dict_get_bool(dict, "all", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001583
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001584 if (dict_has_key(dict, "id"))
Bram Moolenaard61efa52022-07-23 09:52:04 +01001585 id = dict_get_number(dict, "id");
Ben Jacksona7704222022-08-20 20:54:51 +01001586
1587 // if a specific type was supplied "type": check that (and ignore "types".
1588 // Otherwise check against the list of "types".
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001589 if (dict_has_key(dict, "type"))
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001590 {
Bram Moolenaard61efa52022-07-23 09:52:04 +01001591 char_u *name = dict_get_string(dict, "type", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001592 proptype_T *type = lookup_prop_type(name, buf);
1593
1594 if (type == NULL)
1595 return;
1596 type_id = type->pt_id;
1597 }
Ben Jacksona7704222022-08-20 20:54:51 +01001598 if (dict_has_key(dict, "types"))
1599 {
1600 typval_T types;
1601 listitem_T *li = NULL;
1602
1603 dict_get_tv(dict, "types", &types);
1604 if (types.v_type == VAR_LIST && types.vval.v_list->lv_len > 0)
1605 {
1606 type_ids = alloc( sizeof(int) * types.vval.v_list->lv_len );
1607
1608 FOR_ALL_LIST_ITEMS(types.vval.v_list, li)
1609 {
1610 proptype_T *prop_type;
1611
1612 if (li->li_tv.v_type != VAR_STRING)
1613 continue;
1614
1615 prop_type = lookup_prop_type(li->li_tv.vval.v_string, buf);
1616
1617 if (!prop_type)
1618 goto cleanup_prop_remove;
1619
1620 type_ids[num_type_ids++] = prop_type->pt_id;
1621 }
1622 }
1623 }
Bram Moolenaard61efa52022-07-23 09:52:04 +01001624 both = dict_get_bool(dict, "both", FALSE);
Bram Moolenaara5a40c52020-09-05 20:50:49 +02001625
Ben Jacksona7704222022-08-20 20:54:51 +01001626 if (id == -MAXCOL && (type_id == -1 && num_type_ids == 0))
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001627 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001628 emsg(_(e_need_at_least_one_of_id_or_type));
Ben Jacksona7704222022-08-20 20:54:51 +01001629 goto cleanup_prop_remove;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001630 }
Ben Jacksona7704222022-08-20 20:54:51 +01001631 if (both && (id == -MAXCOL || (type_id == -1 && num_type_ids == 0)))
Bram Moolenaar49b79bd2020-03-05 21:52:55 +01001632 {
Ben Jacksona7704222022-08-20 20:54:51 +01001633 emsg(_(e_need_id_and_type_or_types_with_both));
1634 goto cleanup_prop_remove;
1635 }
1636 if (type_id != -1 && num_type_ids > 0)
1637 {
1638 emsg(_(e_cannot_specify_both_type_and_types));
1639 goto cleanup_prop_remove;
Bram Moolenaar49b79bd2020-03-05 21:52:55 +01001640 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001641
1642 if (end == 0)
1643 end = buf->b_ml.ml_line_count;
1644 for (lnum = start; lnum <= end; ++lnum)
1645 {
1646 char_u *text;
1647 size_t len;
1648
1649 if (lnum > buf->b_ml.ml_line_count)
1650 break;
1651 text = ml_get_buf(buf, lnum, FALSE);
1652 len = STRLEN(text) + 1;
1653 if ((size_t)buf->b_ml.ml_line_len > len)
1654 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02001655 static textprop_T textprop; // static because of alignment
1656 unsigned idx;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001657
1658 for (idx = 0; idx < (buf->b_ml.ml_line_len - len)
1659 / sizeof(textprop_T); ++idx)
1660 {
1661 char_u *cur_prop = buf->b_ml.ml_line_ptr + len
1662 + idx * sizeof(textprop_T);
1663 size_t taillen;
Ben Jacksona7704222022-08-20 20:54:51 +01001664 int matches_id = 0;
1665 int matches_type = 0;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001666
1667 mch_memmove(&textprop, cur_prop, sizeof(textprop_T));
Ben Jacksona7704222022-08-20 20:54:51 +01001668
1669 matches_id = textprop.tp_id == id;
1670 if (num_type_ids > 0)
1671 {
1672 int idx2;
1673
1674 for (idx2 = 0; !matches_type && idx2 < num_type_ids; ++idx2)
1675 matches_type = textprop.tp_type == type_ids[idx2];
1676 }
1677 else
1678 {
1679 matches_type = textprop.tp_type == type_id;
1680 }
1681
1682 if (both ? matches_id && matches_type
1683 : matches_id || matches_type)
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001684 {
1685 if (!(buf->b_ml.ml_flags & ML_LINE_DIRTY))
1686 {
1687 char_u *newptr = alloc(buf->b_ml.ml_line_len);
1688
1689 // need to allocate the line to be able to change it
1690 if (newptr == NULL)
Ben Jacksona7704222022-08-20 20:54:51 +01001691 goto cleanup_prop_remove;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001692 mch_memmove(newptr, buf->b_ml.ml_line_ptr,
1693 buf->b_ml.ml_line_len);
Bram Moolenaarfa4873c2022-06-30 22:13:59 +01001694 if (buf->b_ml.ml_flags & ML_ALLOCATED)
1695 vim_free(buf->b_ml.ml_line_ptr);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001696 buf->b_ml.ml_line_ptr = newptr;
Bram Moolenaar0a2f5782019-03-22 13:20:43 +01001697 buf->b_ml.ml_flags |= ML_LINE_DIRTY;
1698
1699 cur_prop = buf->b_ml.ml_line_ptr + len
Bram Moolenaarf0884c52019-05-24 21:22:29 +02001700 + idx * sizeof(textprop_T);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001701 }
1702
1703 taillen = buf->b_ml.ml_line_len - len
1704 - (idx + 1) * sizeof(textprop_T);
1705 if (taillen > 0)
1706 mch_memmove(cur_prop, cur_prop + sizeof(textprop_T),
1707 taillen);
1708 buf->b_ml.ml_line_len -= sizeof(textprop_T);
1709 --idx;
1710
Bram Moolenaar3a4cd392022-07-30 22:17:18 +01001711 if (textprop.tp_id < 0)
1712 {
1713 garray_T *gap = &buf->b_textprop_text;
1714 int ii = -textprop.tp_id - 1;
1715
1716 // negative ID: property with text - free the text
1717 if (ii < gap->ga_len)
1718 {
1719 char_u **p = ((char_u **)gap->ga_data) + ii;
1720 vim_free(*p);
1721 *p = NULL;
1722 did_remove_text = TRUE;
1723 }
1724 }
1725
Bram Moolenaar965c0442021-05-17 00:22:06 +02001726 if (first_changed == 0)
1727 first_changed = lnum;
1728 last_changed = lnum;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001729 ++rettv->vval.v_number;
1730 if (!do_all)
1731 break;
1732 }
1733 }
1734 }
1735 }
Bram Moolenaar3a4cd392022-07-30 22:17:18 +01001736
Bram Moolenaar965c0442021-05-17 00:22:06 +02001737 if (first_changed > 0)
Bram Moolenaarfc643e62021-05-17 00:15:18 +02001738 {
Bram Moolenaar326c5d32022-08-12 13:05:49 +01001739 changed_line_display_buf(buf);
Bram Moolenaar965c0442021-05-17 00:22:06 +02001740 changed_lines_buf(buf, first_changed, last_changed + 1, 0);
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001741 redraw_buf_later(buf, UPD_VALID);
Bram Moolenaarfc643e62021-05-17 00:15:18 +02001742 }
Bram Moolenaar3a4cd392022-07-30 22:17:18 +01001743
1744 if (did_remove_text)
1745 {
1746 garray_T *gap = &buf->b_textprop_text;
1747
1748 // Reduce the growarray size for NULL pointers at the end.
1749 while (gap->ga_len > 0
1750 && ((char_u **)gap->ga_data)[gap->ga_len - 1] == NULL)
1751 --gap->ga_len;
1752 }
Ben Jacksona7704222022-08-20 20:54:51 +01001753
1754cleanup_prop_remove:
1755 vim_free(type_ids);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001756}
1757
1758/*
1759 * Common for f_prop_type_add() and f_prop_type_change().
1760 */
Bram Moolenaar5843f5f2019-08-20 20:13:45 +02001761 static void
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001762prop_type_set(typval_T *argvars, int add)
1763{
1764 char_u *name;
1765 buf_T *buf = NULL;
1766 dict_T *dict;
1767 dictitem_T *di;
1768 proptype_T *prop;
1769
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001770 if (in_vim9script()
1771 && (check_for_string_arg(argvars, 0) == FAIL
1772 || check_for_dict_arg(argvars, 1) == FAIL))
1773 return;
1774
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001775 name = tv_get_string(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001776 if (*name == NUL)
1777 {
Bram Moolenaar89469d12022-12-02 20:46:26 +00001778 semsg(_(e_invalid_argument_str), "\"\"");
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001779 return;
1780 }
1781
1782 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
1783 return;
1784 dict = argvars[1].vval.v_dict;
1785
Bram Moolenaare44336b2022-08-07 18:20:08 +01001786 prop = find_prop_type(name, buf);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001787 if (add)
1788 {
1789 hashtab_T **htp;
1790
1791 if (prop != NULL)
1792 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001793 semsg(_(e_property_type_str_already_defined), name);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001794 return;
1795 }
Bram Moolenaar47ed5532019-08-08 20:49:14 +02001796 prop = alloc_clear(offsetof(proptype_T, pt_name) + STRLEN(name) + 1);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001797 if (prop == NULL)
1798 return;
1799 STRCPY(prop->pt_name, name);
1800 prop->pt_id = ++proptype_id;
Bram Moolenaar0743ef92019-11-13 16:37:31 +01001801 prop->pt_flags = PT_FLAG_COMBINE;
Bram Moolenaare44336b2022-08-07 18:20:08 +01001802 if (buf == NULL)
1803 {
1804 htp = &global_proptypes;
1805 VIM_CLEAR(global_proparray);
1806 }
1807 else
1808 {
1809 htp = &buf->b_proptypes;
1810 VIM_CLEAR(buf->b_proparray);
1811 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001812 if (*htp == NULL)
1813 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02001814 *htp = ALLOC_ONE(hashtab_T);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001815 if (*htp == NULL)
Bram Moolenaarfb95e212018-12-14 12:18:11 +01001816 {
1817 vim_free(prop);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001818 return;
Bram Moolenaarfb95e212018-12-14 12:18:11 +01001819 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001820 hash_init(*htp);
1821 }
Bram Moolenaaref2c3252022-11-25 16:31:51 +00001822 hash_add(*htp, PT2HIKEY(prop), "prop type");
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001823 }
1824 else
1825 {
1826 if (prop == NULL)
1827 {
Bram Moolenaara9fa8c52023-01-02 18:10:04 +00001828 semsg(_(e_property_type_str_does_not_exist), name);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001829 return;
1830 }
1831 }
1832
1833 if (dict != NULL)
1834 {
1835 di = dict_find(dict, (char_u *)"highlight", -1);
1836 if (di != NULL)
1837 {
1838 char_u *highlight;
1839 int hl_id = 0;
1840
Bram Moolenaard61efa52022-07-23 09:52:04 +01001841 highlight = dict_get_string(dict, "highlight", FALSE);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001842 if (highlight != NULL && *highlight != NUL)
1843 hl_id = syn_name2id(highlight);
1844 if (hl_id <= 0)
1845 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001846 semsg(_(e_unknown_highlight_group_name_str),
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001847 highlight == NULL ? (char_u *)"" : highlight);
1848 return;
1849 }
1850 prop->pt_hl_id = hl_id;
1851 }
1852
Bram Moolenaar58187f12019-05-05 16:33:47 +02001853 di = dict_find(dict, (char_u *)"combine", -1);
1854 if (di != NULL)
1855 {
Bram Moolenaarfa2e38d2020-09-05 21:00:00 +02001856 if (tv_get_bool(&di->di_tv))
Bram Moolenaar58187f12019-05-05 16:33:47 +02001857 prop->pt_flags |= PT_FLAG_COMBINE;
1858 else
1859 prop->pt_flags &= ~PT_FLAG_COMBINE;
1860 }
1861
Bram Moolenaarf4ba8bc2022-08-05 17:05:04 +01001862 di = dict_find(dict, (char_u *)"override", -1);
1863 if (di != NULL)
1864 {
1865 if (tv_get_bool(&di->di_tv))
1866 prop->pt_flags |= PT_FLAG_OVERRIDE;
1867 else
1868 prop->pt_flags &= ~PT_FLAG_OVERRIDE;
1869 }
1870
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001871 di = dict_find(dict, (char_u *)"priority", -1);
1872 if (di != NULL)
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001873 prop->pt_priority = tv_get_number(&di->di_tv);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001874
1875 di = dict_find(dict, (char_u *)"start_incl", -1);
1876 if (di != NULL)
1877 {
Bram Moolenaarfa2e38d2020-09-05 21:00:00 +02001878 if (tv_get_bool(&di->di_tv))
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001879 prop->pt_flags |= PT_FLAG_INS_START_INCL;
1880 else
1881 prop->pt_flags &= ~PT_FLAG_INS_START_INCL;
1882 }
1883
1884 di = dict_find(dict, (char_u *)"end_incl", -1);
1885 if (di != NULL)
1886 {
Bram Moolenaarfa2e38d2020-09-05 21:00:00 +02001887 if (tv_get_bool(&di->di_tv))
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001888 prop->pt_flags |= PT_FLAG_INS_END_INCL;
1889 else
1890 prop->pt_flags &= ~PT_FLAG_INS_END_INCL;
1891 }
1892 }
1893}
1894
1895/*
1896 * prop_type_add({name}, {props})
1897 */
1898 void
1899f_prop_type_add(typval_T *argvars, typval_T *rettv UNUSED)
1900{
1901 prop_type_set(argvars, TRUE);
1902}
1903
1904/*
1905 * prop_type_change({name}, {props})
1906 */
1907 void
1908f_prop_type_change(typval_T *argvars, typval_T *rettv UNUSED)
1909{
1910 prop_type_set(argvars, FALSE);
1911}
1912
1913/*
1914 * prop_type_delete({name} [, {bufnr}])
1915 */
1916 void
1917f_prop_type_delete(typval_T *argvars, typval_T *rettv UNUSED)
1918{
1919 char_u *name;
1920 buf_T *buf = NULL;
1921 hashitem_T *hi;
1922
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001923 if (in_vim9script()
1924 && (check_for_string_arg(argvars, 0) == FAIL
1925 || check_for_opt_dict_arg(argvars, 1) == FAIL))
1926 return;
1927
Bram Moolenaard155d7a2018-12-21 16:04:21 +01001928 name = tv_get_string(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001929 if (*name == NUL)
1930 {
Bram Moolenaar89469d12022-12-02 20:46:26 +00001931 semsg(_(e_invalid_argument_str), "\"\"");
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001932 return;
1933 }
1934
1935 if (argvars[1].v_type != VAR_UNKNOWN)
1936 {
1937 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
1938 return;
1939 }
1940
Bram Moolenaare44336b2022-08-07 18:20:08 +01001941 hi = find_prop_type_hi(name, buf);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001942 if (hi != NULL)
1943 {
1944 hashtab_T *ht;
Bram Moolenaarfb95e212018-12-14 12:18:11 +01001945 proptype_T *prop = HI2PT(hi);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001946
1947 if (buf == NULL)
Bram Moolenaare44336b2022-08-07 18:20:08 +01001948 {
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001949 ht = global_proptypes;
Bram Moolenaare44336b2022-08-07 18:20:08 +01001950 VIM_CLEAR(global_proparray);
1951 }
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001952 else
Bram Moolenaare44336b2022-08-07 18:20:08 +01001953 {
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001954 ht = buf->b_proptypes;
Bram Moolenaare44336b2022-08-07 18:20:08 +01001955 VIM_CLEAR(buf->b_proparray);
1956 }
Bram Moolenaaref2c3252022-11-25 16:31:51 +00001957 hash_remove(ht, hi, "prop type delete");
Bram Moolenaarfb95e212018-12-14 12:18:11 +01001958 vim_free(prop);
Bram Moolenaar89469d12022-12-02 20:46:26 +00001959
1960 // currently visibile text properties will disappear
1961 redraw_all_later(UPD_CLEAR);
1962 changed_window_setting_buf(buf == NULL ? curbuf : buf);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001963 }
1964}
1965
1966/*
Martin Tournoije2390c72021-07-28 13:30:16 +02001967 * prop_type_get({name} [, {props}])
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001968 */
1969 void
Bram Moolenaar3d8a5132020-01-04 16:13:49 +01001970f_prop_type_get(typval_T *argvars, typval_T *rettv)
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001971{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001972 char_u *name;
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001973
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001974 if (in_vim9script()
1975 && (check_for_string_arg(argvars, 0) == FAIL
1976 || check_for_opt_dict_arg(argvars, 1) == FAIL))
1977 return;
1978
1979 name = tv_get_string(&argvars[0]);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001980 if (*name == NUL)
1981 {
Bram Moolenaar89469d12022-12-02 20:46:26 +00001982 semsg(_(e_invalid_argument_str), "\"\"");
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001983 return;
1984 }
1985 if (rettv_dict_alloc(rettv) == OK)
1986 {
1987 proptype_T *prop = NULL;
1988 buf_T *buf = NULL;
1989
1990 if (argvars[1].v_type != VAR_UNKNOWN)
1991 {
1992 if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL)
1993 return;
1994 }
1995
Bram Moolenaare44336b2022-08-07 18:20:08 +01001996 prop = find_prop_type(name, buf);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001997 if (prop != NULL)
1998 {
1999 dict_T *d = rettv->vval.v_dict;
2000
2001 if (prop->pt_hl_id > 0)
2002 dict_add_string(d, "highlight", syn_id2name(prop->pt_hl_id));
2003 dict_add_number(d, "priority", prop->pt_priority);
Bram Moolenaar58187f12019-05-05 16:33:47 +02002004 dict_add_number(d, "combine",
2005 (prop->pt_flags & PT_FLAG_COMBINE) ? 1 : 0);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01002006 dict_add_number(d, "start_incl",
2007 (prop->pt_flags & PT_FLAG_INS_START_INCL) ? 1 : 0);
2008 dict_add_number(d, "end_incl",
2009 (prop->pt_flags & PT_FLAG_INS_END_INCL) ? 1 : 0);
2010 if (buf != NULL)
2011 dict_add_number(d, "bufnr", buf->b_fnum);
2012 }
2013 }
2014}
2015
2016 static void
2017list_types(hashtab_T *ht, list_T *l)
2018{
2019 long todo;
2020 hashitem_T *hi;
2021
2022 todo = (long)ht->ht_used;
2023 for (hi = ht->ht_array; todo > 0; ++hi)
2024 {
2025 if (!HASHITEM_EMPTY(hi))
2026 {
2027 proptype_T *prop = HI2PT(hi);
2028
2029 list_append_string(l, prop->pt_name, -1);
2030 --todo;
2031 }
2032 }
2033}
2034
2035/*
2036 * prop_type_list([{bufnr}])
2037 */
2038 void
2039f_prop_type_list(typval_T *argvars, typval_T *rettv UNUSED)
2040{
2041 buf_T *buf = NULL;
2042
2043 if (rettv_list_alloc(rettv) == OK)
2044 {
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02002045 if (in_vim9script() && check_for_opt_dict_arg(argvars, 0) == FAIL)
2046 return;
2047
Bram Moolenaar98aefe72018-12-13 22:20:09 +01002048 if (argvars[0].v_type != VAR_UNKNOWN)
2049 {
2050 if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL)
2051 return;
2052 }
2053 if (buf == NULL)
2054 {
2055 if (global_proptypes != NULL)
2056 list_types(global_proptypes, rettv->vval.v_list);
2057 }
2058 else if (buf->b_proptypes != NULL)
2059 list_types(buf->b_proptypes, rettv->vval.v_list);
2060 }
2061}
2062
2063/*
2064 * Free all property types in "ht".
2065 */
2066 static void
2067clear_ht_prop_types(hashtab_T *ht)
2068{
2069 long todo;
2070 hashitem_T *hi;
2071
2072 if (ht == NULL)
2073 return;
2074
2075 todo = (long)ht->ht_used;
2076 for (hi = ht->ht_array; todo > 0; ++hi)
2077 {
2078 if (!HASHITEM_EMPTY(hi))
2079 {
2080 proptype_T *prop = HI2PT(hi);
2081
2082 vim_free(prop);
2083 --todo;
2084 }
2085 }
2086
2087 hash_clear(ht);
2088 vim_free(ht);
2089}
2090
2091#if defined(EXITFREE) || defined(PROTO)
2092/*
Bram Moolenaarfb95e212018-12-14 12:18:11 +01002093 * Free all global property types.
Bram Moolenaar98aefe72018-12-13 22:20:09 +01002094 */
2095 void
2096clear_global_prop_types(void)
2097{
2098 clear_ht_prop_types(global_proptypes);
2099 global_proptypes = NULL;
Bram Moolenaare44336b2022-08-07 18:20:08 +01002100 VIM_CLEAR(global_proparray);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01002101}
2102#endif
2103
2104/*
2105 * Free all property types for "buf".
2106 */
2107 void
2108clear_buf_prop_types(buf_T *buf)
2109{
2110 clear_ht_prop_types(buf->b_proptypes);
2111 buf->b_proptypes = NULL;
Bram Moolenaare44336b2022-08-07 18:20:08 +01002112 VIM_CLEAR(buf->b_proparray);
Bram Moolenaar98aefe72018-12-13 22:20:09 +01002113}
2114
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002115// Struct used to return two values from adjust_prop().
2116typedef struct
2117{
2118 int dirty; // if the property was changed
2119 int can_drop; // whether after this change, the prop may be removed
2120} adjustres_T;
2121
2122/*
2123 * Adjust the property for "added" bytes (can be negative) inserted at "col".
2124 *
2125 * Note that "col" is zero-based, while tp_col is one-based.
2126 * Only for the current buffer.
2127 * "flags" can have:
2128 * APC_SUBSTITUTE: Text is replaced, not inserted.
Bram Moolenaard8d4cfc2022-08-15 15:55:10 +01002129 * APC_INDENT: Text is inserted before virtual text prop
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002130 */
2131 static adjustres_T
2132adjust_prop(
Bram Moolenaarb7963df2022-07-31 17:12:43 +01002133 textprop_T *prop,
2134 colnr_T col,
2135 int added,
2136 int flags)
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002137{
Bram Moolenaarb7963df2022-07-31 17:12:43 +01002138 proptype_T *pt;
2139 int start_incl;
2140 int end_incl;
2141 int droppable;
2142 adjustres_T res = {TRUE, FALSE};
2143
2144 // prop after end of the line doesn't move
2145 if (prop->tp_col == MAXCOL)
2146 {
2147 res.dirty = FALSE;
2148 return res;
2149 }
2150
2151 pt = text_prop_type_by_id(curbuf, prop->tp_type);
2152 start_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_START_INCL))
LemonBoy698cb4c2022-05-14 18:10:15 +01002153 || (flags & APC_SUBSTITUTE)
2154 || (prop->tp_flags & TP_FLAG_CONT_PREV);
Bram Moolenaard8d4cfc2022-08-15 15:55:10 +01002155 if (prop->tp_id < 0 && (flags & APC_INDENT))
2156 // when inserting indent just before a character with virtual text
2157 // shift the text property
2158 start_incl = FALSE;
Bram Moolenaarb7963df2022-07-31 17:12:43 +01002159 end_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_END_INCL))
LemonBoy698cb4c2022-05-14 18:10:15 +01002160 || (prop->tp_flags & TP_FLAG_CONT_NEXT);
Bram Moolenaarb7963df2022-07-31 17:12:43 +01002161 // do not drop zero-width props if they later can increase in size
2162 droppable = !(start_incl || end_incl);
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002163
2164 if (added > 0)
2165 {
2166 if (col + 1 <= prop->tp_col
2167 - (start_incl || (prop->tp_len == 0 && end_incl)))
2168 // Change is entirely before the text property: Only shift
2169 prop->tp_col += added;
2170 else if (col + 1 < prop->tp_col + prop->tp_len + end_incl)
2171 // Insertion was inside text property
2172 prop->tp_len += added;
2173 }
2174 else if (prop->tp_col > col + 1)
2175 {
2176 if (prop->tp_col + added < col + 1)
2177 {
2178 prop->tp_len += (prop->tp_col - 1 - col) + added;
2179 prop->tp_col = col + 1;
2180 if (prop->tp_len <= 0)
2181 {
2182 prop->tp_len = 0;
2183 res.can_drop = droppable;
2184 }
2185 }
2186 else
2187 prop->tp_col += added;
2188 }
Bram Moolenaarf5240b92022-08-24 12:24:37 +01002189 else if (prop->tp_len > 0 && prop->tp_col + prop->tp_len > col
2190 && prop->tp_id >= 0) // don't change length for virtual text
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002191 {
2192 int after = col - added - (prop->tp_col - 1 + prop->tp_len);
2193
2194 prop->tp_len += after > 0 ? added + after : added;
2195 res.can_drop = prop->tp_len <= 0 && droppable;
2196 }
2197 else
2198 res.dirty = FALSE;
2199
2200 return res;
2201}
2202
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01002203/*
2204 * Adjust the columns of text properties in line "lnum" after position "col" to
2205 * shift by "bytes_added" (can be negative).
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002206 * Note that "col" is zero-based, while tp_col is one-based.
2207 * Only for the current buffer.
Bram Moolenaarf3333b02019-05-19 22:53:40 +02002208 * "flags" can have:
2209 * APC_SAVE_FOR_UNDO: Call u_savesub() before making changes to the line.
2210 * APC_SUBSTITUTE: Text is replaced, not inserted.
Bram Moolenaard8d4cfc2022-08-15 15:55:10 +01002211 * APC_INDENT: Text is inserted before virtual text prop
Bram Moolenaar8055d172019-05-17 22:57:26 +02002212 * Caller is expected to check b_has_textprop and "bytes_added" being non-zero.
Bram Moolenaar338dfda2019-05-19 15:19:57 +02002213 * Returns TRUE when props were changed.
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01002214 */
Bram Moolenaar338dfda2019-05-19 15:19:57 +02002215 int
Bram Moolenaar196d1572019-01-02 23:47:18 +01002216adjust_prop_columns(
2217 linenr_T lnum,
2218 colnr_T col,
Bram Moolenaar338dfda2019-05-19 15:19:57 +02002219 int bytes_added,
Bram Moolenaarf3333b02019-05-19 22:53:40 +02002220 int flags)
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01002221{
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002222 int proplen;
2223 char_u *props;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002224 int dirty = FALSE;
Bram Moolenaar196d1572019-01-02 23:47:18 +01002225 int ri, wi;
2226 size_t textlen;
2227
2228 if (text_prop_frozen > 0)
Bram Moolenaar338dfda2019-05-19 15:19:57 +02002229 return FALSE;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002230
2231 proplen = get_text_props(curbuf, lnum, &props, TRUE);
2232 if (proplen == 0)
Bram Moolenaar338dfda2019-05-19 15:19:57 +02002233 return FALSE;
Bram Moolenaar196d1572019-01-02 23:47:18 +01002234 textlen = curbuf->b_ml.ml_line_len - proplen * sizeof(textprop_T);
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002235
Bram Moolenaar196d1572019-01-02 23:47:18 +01002236 wi = 0; // write index
2237 for (ri = 0; ri < proplen; ++ri)
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002238 {
Bram Moolenaar12f20032020-02-26 22:06:00 +01002239 textprop_T prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002240 adjustres_T res;
Bram Moolenaarf3333b02019-05-19 22:53:40 +02002241
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002242 mch_memmove(&prop, props + ri * sizeof(prop), sizeof(prop));
2243 res = adjust_prop(&prop, col, bytes_added, flags);
2244 if (res.dirty)
Bram Moolenaarf3333b02019-05-19 22:53:40 +02002245 {
Bram Moolenaar338dfda2019-05-19 15:19:57 +02002246 // Save for undo if requested and not done yet.
Bram Moolenaarcf070112020-06-29 23:02:21 +02002247 if ((flags & APC_SAVE_FOR_UNDO) && !dirty
2248 && u_savesub(lnum) == FAIL)
2249 return FALSE;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002250 dirty = TRUE;
Bram Moolenaar8902b312020-09-20 21:04:35 +02002251
2252 // u_savesub() may have updated curbuf->b_ml, fetch it again
2253 if (curbuf->b_ml.ml_line_lnum != lnum)
2254 proplen = get_text_props(curbuf, lnum, &props, TRUE);
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002255 }
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002256 if (res.can_drop)
2257 continue; // Drop this text property
Bram Moolenaar12f20032020-02-26 22:06:00 +01002258 mch_memmove(props + wi * sizeof(textprop_T), &prop, sizeof(textprop_T));
Bram Moolenaar196d1572019-01-02 23:47:18 +01002259 ++wi;
2260 }
2261 if (dirty)
2262 {
Bram Moolenaar4614f532019-01-06 12:54:55 +01002263 colnr_T newlen = (int)textlen + wi * (colnr_T)sizeof(textprop_T);
2264
2265 if ((curbuf->b_ml.ml_flags & ML_LINE_DIRTY) == 0)
Bram Moolenaarfa4873c2022-06-30 22:13:59 +01002266 {
2267 char_u *p = vim_memsave(curbuf->b_ml.ml_line_ptr, newlen);
2268
2269 if (curbuf->b_ml.ml_flags & ML_ALLOCATED)
2270 vim_free(curbuf->b_ml.ml_line_ptr);
2271 curbuf->b_ml.ml_line_ptr = p;
2272 }
Bram Moolenaar196d1572019-01-02 23:47:18 +01002273 curbuf->b_ml.ml_flags |= ML_LINE_DIRTY;
Bram Moolenaar4614f532019-01-06 12:54:55 +01002274 curbuf->b_ml.ml_line_len = newlen;
Bram Moolenaar44746aa2019-01-02 00:02:11 +01002275 }
Bram Moolenaar338dfda2019-05-19 15:19:57 +02002276 return dirty;
Bram Moolenaarb9c67a52019-01-01 19:49:20 +01002277}
2278
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002279/*
2280 * Adjust text properties for a line that was split in two.
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02002281 * "lnum_props" is the line that has the properties from before the split.
2282 * "lnum_top" is the top line.
2283 * "kept" is the number of bytes kept in the first line, while
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002284 * "deleted" is the number of bytes deleted.
Bram Moolenaarebd0e8b2022-09-14 22:13:59 +01002285 * "at_eol" is true if the split is after the end of the line.
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002286 */
2287 void
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02002288adjust_props_for_split(
Bram Moolenaarebd0e8b2022-09-14 22:13:59 +01002289 linenr_T lnum_props,
2290 linenr_T lnum_top,
2291 int kept,
2292 int deleted,
2293 int at_eol)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002294{
2295 char_u *props;
2296 int count;
2297 garray_T prevprop;
2298 garray_T nextprop;
2299 int i;
2300 int skipped = kept + deleted;
2301
2302 if (!curbuf->b_has_textprop)
2303 return;
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02002304
2305 // Get the text properties from "lnum_props".
2306 count = get_text_props(curbuf, lnum_props, &props, FALSE);
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002307 ga_init2(&prevprop, sizeof(textprop_T), 10);
2308 ga_init2(&nextprop, sizeof(textprop_T), 10);
2309
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002310 // Keep the relevant ones in the first line, reducing the length if needed.
2311 // Copy the ones that include the split to the second line.
2312 // Move the ones after the split to the second line.
2313 for (i = 0; i < count; ++i)
2314 {
2315 textprop_T prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002316 proptype_T *pt;
2317 int start_incl, end_incl;
2318 int cont_prev, cont_next;
Bram Moolenaar3b93cf22022-09-13 18:34:18 +01002319 int prop_col;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002320
2321 // copy the prop to an aligned structure
2322 mch_memmove(&prop, props + i * sizeof(textprop_T), sizeof(textprop_T));
2323
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002324 pt = text_prop_type_by_id(curbuf, prop.tp_type);
2325 start_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_START_INCL));
2326 end_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_END_INCL));
Bram Moolenaar3b93cf22022-09-13 18:34:18 +01002327
2328 // a text prop "above" behaves like it is on the first text column
2329 prop_col = (prop.tp_flags & TP_FLAG_ALIGN_ABOVE) ? 1 : prop.tp_col;
2330
Bram Moolenaarebd0e8b2022-09-14 22:13:59 +01002331 if (prop_col == MAXCOL)
2332 {
2333 cont_prev = at_eol;
2334 cont_next = !at_eol;
2335 }
2336 else
2337 {
2338 cont_prev = prop_col + !start_incl <= kept;
2339 cont_next = skipped <= prop_col + prop.tp_len - !end_incl;
2340 }
Bram Moolenaard8d4cfc2022-08-15 15:55:10 +01002341 // when a prop has text it is never copied
2342 if (prop.tp_id < 0 && cont_next)
2343 cont_prev = FALSE;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002344
2345 if (cont_prev && ga_grow(&prevprop, 1) == OK)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002346 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002347 textprop_T *p = ((textprop_T *)prevprop.ga_data) + prevprop.ga_len;
2348
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002349 *p = prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002350 ++prevprop.ga_len;
Bram Moolenaar98b37302022-09-14 12:06:53 +01002351 if (p->tp_col != MAXCOL && p->tp_col + p->tp_len >= kept)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002352 p->tp_len = kept - p->tp_col;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002353 if (cont_next)
2354 p->tp_flags |= TP_FLAG_CONT_NEXT;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002355 }
2356
Bram Moolenaar5c65e6a2019-05-17 11:08:56 +02002357 // Only add the property to the next line if the length is bigger than
2358 // zero.
Bram Moolenaarebd0e8b2022-09-14 22:13:59 +01002359 if (cont_next && ga_grow(&nextprop, 1) == OK)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002360 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002361 textprop_T *p = ((textprop_T *)nextprop.ga_data) + nextprop.ga_len;
Bram Moolenaar7d0f7e92022-08-06 17:10:57 +01002362
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002363 *p = prop;
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002364 ++nextprop.ga_len;
Bram Moolenaar7d0f7e92022-08-06 17:10:57 +01002365 if (p->tp_col != MAXCOL)
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002366 {
Bram Moolenaar7d0f7e92022-08-06 17:10:57 +01002367 if (p->tp_col > skipped)
2368 p->tp_col -= skipped - 1;
2369 else
2370 {
2371 p->tp_len -= skipped - p->tp_col;
2372 p->tp_col = 1;
2373 }
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002374 }
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002375 if (cont_prev)
2376 p->tp_flags |= TP_FLAG_CONT_PREV;
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002377 }
2378 }
2379
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02002380 set_text_props(lnum_top, prevprop.ga_data,
2381 prevprop.ga_len * sizeof(textprop_T));
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002382 ga_clear(&prevprop);
Bram Moolenaar45dd07f2019-05-15 22:45:37 +02002383 set_text_props(lnum_top + 1, nextprop.ga_data,
2384 nextprop.ga_len * sizeof(textprop_T));
Bram Moolenaar4164bb22019-01-04 23:09:49 +01002385 ga_clear(&nextprop);
2386}
2387
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002388/*
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002389 * Prepend properties of joined line "lnum" to "new_props".
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002390 */
2391 void
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002392prepend_joined_props(
2393 char_u *new_props,
2394 int propcount,
2395 int *props_remaining,
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002396 linenr_T lnum,
Bram Moolenaare175dc62022-08-01 22:18:50 +01002397 int last_line,
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002398 long col,
2399 int removed)
2400{
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002401 char_u *props;
2402 int proplen = get_text_props(curbuf, lnum, &props, FALSE);
2403 int i;
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002404
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002405 for (i = proplen; i-- > 0; )
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002406 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002407 textprop_T prop;
2408 int end;
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002409
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002410 mch_memmove(&prop, props + i * sizeof(prop), sizeof(prop));
Bram Moolenaare175dc62022-08-01 22:18:50 +01002411 if (prop.tp_col == MAXCOL && !last_line)
2412 continue; // drop property with text after the line
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002413 end = !(prop.tp_flags & TP_FLAG_CONT_NEXT);
2414
2415 adjust_prop(&prop, 0, -removed, 0); // Remove leading spaces
Bram Moolenaar8e7d6222020-12-18 19:49:56 +01002416 adjust_prop(&prop, -1, col, 0); // Make line start at its final column
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002417
Bram Moolenaare175dc62022-08-01 22:18:50 +01002418 if (last_line || end)
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002419 mch_memmove(new_props + --(*props_remaining) * sizeof(prop),
2420 &prop, sizeof(prop));
2421 else
2422 {
2423 int j;
2424 int found = FALSE;
2425
2426 // Search for continuing prop.
2427 for (j = *props_remaining; j < propcount; ++j)
2428 {
2429 textprop_T op;
2430
2431 mch_memmove(&op, new_props + j * sizeof(op), sizeof(op));
2432 if ((op.tp_flags & TP_FLAG_CONT_PREV)
2433 && op.tp_id == prop.tp_id && op.tp_type == prop.tp_type)
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002434 {
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002435 found = TRUE;
2436 op.tp_len += op.tp_col - prop.tp_col;
2437 op.tp_col = prop.tp_col;
2438 // Start/end is taken care of when deleting joined lines
2439 op.tp_flags = prop.tp_flags;
2440 mch_memmove(new_props + j * sizeof(op), &op, sizeof(op));
2441 break;
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002442 }
2443 }
Bram Moolenaar87be9be2020-05-30 15:32:02 +02002444 if (!found)
2445 internal_error("text property above joined line not found");
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002446 }
Bram Moolenaar80e737c2019-05-17 19:56:34 +02002447 }
2448}
2449
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01002450#endif // FEAT_PROP_POPUP