/* vi:set ts=8 sts=4 sw=4 noet: * * VIM - Vi IMproved by Bram Moolenaar * * Do ":help uganda" in Vim to read copying and usage conditions. * Do ":help credits" in Vim to see a list of people who contributed. * See README.txt for an overview of the Vim source code. */ /* * Text properties implementation. See ":help text-properties". */ #include "vim.h" #if defined(FEAT_PROP_POPUP) static void um_store_changes(unpacked_memline_T *um); /* * Free virtual text strings in a detached unpacked memline's props. */ static void um_free_vtext(textprop_T *props, int count) { for (int i = 0; i < count; ++i) if (props[i].tp_flags & TP_FLAG_VTEXT_PTR) VIM_CLEAR(props[i].u.tp_text); } /* * Free memory used by an unpacked memline. */ static void um_free(unpacked_memline_T *um) { if (um->detached) { VIM_CLEAR(um->text); um_free_vtext(um->props, um->prop_count); } VIM_CLEAR(um->props); um->prop_size = 0; um->prop_count = 0; um->lnum = 0; um->detached = false; } /* * Initialize an unpacked memline for the given buffer. * No line is loaded yet; use um_goto_line() to load one. */ unpacked_memline_T um_open(buf_T *buf) { unpacked_memline_T um; CLEAR_FIELD(um); um.buf = buf; return um; } /* * Load line "lnum" into an unpacked memline. If a previous line was * loaded and modified, it is stored first. * "extra_props" is the number of extra prop slots to pre-allocate. * Returns true on success, false on error (um becomes closed). */ bool um_goto_line(unpacked_memline_T *um, linenr_T lnum, int extra_props) { char_u *line; size_t textlen; size_t propdata_len; int proplen; if (um->buf == NULL) return false; // Store changes to the current line if any. if (um->lnum > 0 && um->detached) um_store_changes(um); um_free(um); if (lnum == 0) return true; // just unload line = ml_get_buf(um->buf, lnum, FALSE); textlen = ml_get_buf_len(um->buf, lnum) + 1; propdata_len = um->buf->b_ml.ml_line_len - textlen; um->lnum = lnum; um->text_size = (colnr_T)textlen; um->text = line; if (propdata_len == 0) return true; // New format: [prop_count (uint16)][textprop_T...][vtext...] if (propdata_len < PROP_COUNT_SIZE + sizeof(textprop_T)) { iemsg(e_text_property_info_corrupted); um->buf = NULL; return false; } uint16_t prop_count; char_u *count_ptr = line + textlen; char_u *props_start; mch_memmove(&prop_count, count_ptr, PROP_COUNT_SIZE); proplen = (int)prop_count; props_start = count_ptr + PROP_COUNT_SIZE; um->props = ALLOC_MULT(textprop_T, proplen + extra_props); if (um->props == NULL) { um->buf = NULL; return false; } um->prop_size = proplen + extra_props; um->prop_count = proplen; mch_memmove(um->props, props_start, proplen * sizeof(textprop_T)); // Convert tp_text_offset to tp_text pointer for virtual text // props. for (int i = 0; i < proplen; ++i) { if (um->props[i].tp_id < 0 && um->props[i].u.tp_text_offset > 0) { um->props[i].u.tp_text = count_ptr + um->props[i].u.tp_text_offset; um->props[i].tp_flags |= TP_FLAG_VTEXT_PTR; } else um->props[i].u.tp_text = NULL; } return true; } /* * Open an unpacked memline at a specific line. * Check um.buf != NULL for success. */ unpacked_memline_T um_open_at(buf_T *buf, linenr_T lnum, int extra_props) { unpacked_memline_T um = um_open(buf); if (!um_goto_line(&um, lnum, extra_props)) um.buf = NULL; return um; } /* * Make the unpacked memline writable by copying text and virtual text * strings to allocated memory (LOADED -> DETACHED). * Returns true on success. */ static bool um_detach(unpacked_memline_T *um) { char_u *newtext; if (um->detached || um->buf == NULL) return um->detached; newtext = vim_strnsave(um->text, um->text_size - 1); if (newtext == NULL) { um->buf = NULL; return false; } um->text = newtext; for (int i = 0; i < um->prop_count; ++i) { if (um->props[i].tp_flags & TP_FLAG_VTEXT_PTR) { char_u *copy = vim_strsave(um->props[i].u.tp_text); if (copy == NULL) { for (int j = 0; j < i; ++j) if (um->props[j].tp_id < 0) VIM_CLEAR(um->props[j].u.tp_text); VIM_CLEAR(um->text); um->buf = NULL; return false; } um->props[i].u.tp_text = copy; } } um->detached = true; return true; } /* * Set the text of an unpacked memline. "text" must be allocated memory; * ownership is transferred to the unpacked memline. */ bool um_set_text(unpacked_memline_T *um, char_u *text) { if (um->buf == NULL) return false; if (um->detached) vim_free(um->text); um->text = text; um->text_size = (colnr_T)STRLEN(text) + 1; um->text_changed = true; um->detached = true; return true; } /* * Ensure there is space for at least "needed" more properties. * Returns true on success. */ static bool um_grow_props(unpacked_memline_T *um, int needed) { if (um->prop_count + needed <= um->prop_size) return true; int new_size = um->prop_count + needed; textprop_T *newprops = vim_realloc(um->props, new_size * sizeof(textprop_T)); if (newprops == NULL) return false; um->props = newprops; um->prop_size = new_size; return true; } /* * Add a property to the end of the props array without sorting. * For virtual text props, u.tp_text ownership is transferred. */ static void um_add_prop(unpacked_memline_T *um, textprop_T *prop) { if (um->buf == NULL) return; if (!um->detached) um_detach(um); if (!um_grow_props(um, 1)) { um->buf = NULL; return; } um->props[um->prop_count++] = *prop; } /* * Reverse the order of all properties. */ void um_reverse_props(unpacked_memline_T *um) { for (int i = 0, j = um->prop_count - 1; i < j; ++i, --j) { textprop_T tmp = um->props[i]; um->props[i] = um->props[j]; um->props[j] = tmp; } } /* * Open an unpacked memline at "lnum" with no properties copied, * but with space for "prop_count" properties. The umemline is * immediately in the DETACHED state. */ unpacked_memline_T um_open_at_no_props(buf_T *buf, linenr_T lnum, int prop_count) { unpacked_memline_T um; CLEAR_FIELD(um); um.buf = buf; if (lnum < 1 || lnum > buf->b_ml.ml_line_count) { um.buf = NULL; return um; } char_u *line = ml_get_buf(buf, lnum, FALSE); char_u *text_copy = vim_strsave(line); if (text_copy == NULL) { um.buf = NULL; return um; } um.lnum = lnum; um.text = text_copy; um.text_size = (colnr_T)STRLEN(text_copy) + 1; um.detached = true; if (prop_count > 0) { um.props = ALLOC_MULT(textprop_T, prop_count); if (um.props == NULL) { vim_free(um.text); um.buf = NULL; return um; } um.prop_size = prop_count; } return um; } /* * Clear a continuation flag on a neighboring line's text property. * Find a property on "lnum" in "buf" that matches "tp_id" and "tp_type", * and clear "flag_to_clear" (TP_FLAG_CONT_NEXT or TP_FLAG_CONT_PREV). */ static void clear_cont_flag_on_neighbor(buf_T *buf, linenr_T lnum, int tp_id, int tp_type, int flag) { unpacked_memline_T neighbor; if (lnum < 1 || lnum > buf->b_ml.ml_line_count) return; neighbor = um_open_at(buf, lnum, 0); if (neighbor.buf == NULL) return; for (int i = 0; i < neighbor.prop_count; ++i) { textprop_T *p = &neighbor.props[i]; if (p->tp_id == tp_id && p->tp_type == tp_type && (p->tp_flags & flag)) { if (!neighbor.detached && !um_detach(&neighbor)) break; // Re-get pointer after detach. p = &neighbor.props[i]; p->tp_flags &= ~flag; break; } } um_close(&neighbor); } /* * Mark a property for deletion and free its virtual text if any. * Automatically detaches if needed. * Also adjusts continuation flags on neighboring lines. */ void um_delete_prop(unpacked_memline_T *um, int index) { textprop_T *prop; if (um->buf == NULL || index < 0 || index >= um->prop_count) return; if (!um->detached && !um_detach(um)) return; prop = &um->props[index]; // Adjust continuation flags on neighboring lines before deleting. if (prop->tp_flags & TP_FLAG_CONT_PREV) clear_cont_flag_on_neighbor(um->buf, um->lnum - 1, prop->tp_id, prop->tp_type, TP_FLAG_CONT_NEXT); if (prop->tp_flags & TP_FLAG_CONT_NEXT) clear_cont_flag_on_neighbor(um->buf, um->lnum + 1, prop->tp_id, prop->tp_type, TP_FLAG_CONT_PREV); if (prop->tp_flags & TP_FLAG_VTEXT_PTR) VIM_CLEAR(prop->u.tp_text); prop->tp_flags |= TP_FLAG_DELETED; } /* * Pack an unpacked memline into a newly allocated buffer. * Packed format: [text][NUL][textprop_T...][vtext strings...] * Returns the packed buffer, or NULL on failure. * "packed_len" is set to the total length. * * NOTE: Currently packs in the OLD format (no inline vtext) because * the read side has not been converted yet. */ static char_u * um_pack(unpacked_memline_T *um, int *packed_len) { uint16_t live_count = 0; int vtext_size = 0; int total_len; char_u *buf; char_u *count_dest; char_u *prop_dest; char_u *vtext_dest; for (int i = 0; i < um->prop_count; ++i) { textprop_T *prop = &um->props[i]; if (prop->tp_flags & TP_FLAG_DELETED) continue; ++live_count; if (prop->tp_flags & TP_FLAG_VTEXT_PTR) vtext_size += prop->tp_len + 1; } if (live_count == 0) { // No properties: just text, no prop_count header. total_len = um->text_size; buf = alloc(total_len); if (buf == NULL) return NULL; *packed_len = total_len; mch_memmove(buf, um->text, um->text_size); return buf; } // Format: [text][NUL][prop_count][textprop_T...][vtext...] total_len = um->text_size + (int)PROP_COUNT_SIZE + live_count * (int)sizeof(textprop_T) + vtext_size; buf = alloc(total_len); if (buf == NULL) return NULL; *packed_len = total_len; mch_memmove(buf, um->text, um->text_size); count_dest = buf + um->text_size; mch_memmove(count_dest, &live_count, PROP_COUNT_SIZE); prop_dest = count_dest + PROP_COUNT_SIZE; vtext_dest = prop_dest + live_count * sizeof(textprop_T); for (int i = 0; i < um->prop_count; ++i) { textprop_T prop = um->props[i]; if (prop.tp_flags & TP_FLAG_DELETED) continue; if (prop.tp_id < 0 && (prop.tp_flags & TP_FLAG_VTEXT_PTR)) { int copysize = prop.tp_len + 1; mch_memmove(vtext_dest, prop.u.tp_text, copysize); prop.u.tp_text_offset = (colnr_T)(vtext_dest - count_dest); vtext_dest += copysize; } else prop.u.tp_text_offset = 0; prop.tp_flags &= ~(TP_FLAG_DELETED | TP_FLAG_VTEXT_PTR); mch_memmove(prop_dest, &prop, sizeof(textprop_T)); prop_dest += sizeof(textprop_T); } return buf; } /* * Store changes to the current line back into the buffer's memline. */ static void um_store_changes(unpacked_memline_T *um) { char_u *packed; int packed_len; if (!um->detached || um->buf == NULL || um->lnum == 0) return; packed = um_pack(um, &packed_len); if (packed == NULL) { um->buf = NULL; return; } // Flush any other dirty line before setting ours. ml_get_buf() // with will_change=TRUE will flush the current dirty line and set // up for our line. if (um->buf->b_ml.ml_line_lnum != um->lnum) ml_get_buf(um->buf, um->lnum, TRUE); else if (!(um->buf->b_ml.ml_flags & ML_LINE_DIRTY)) (void)ml_get_buf(um->buf, um->lnum, TRUE); // Free the detached text and vtext before switching to packed. vim_free(um->text); um_free_vtext(um->props, um->prop_count); if (um->buf->b_ml.ml_flags & (ML_LINE_DIRTY | ML_ALLOCATED)) vim_free(um->buf->b_ml.ml_line_ptr); um->buf->b_ml.ml_line_ptr = packed; um->buf->b_ml.ml_line_len = packed_len; um->buf->b_ml.ml_flags |= ML_LINE_DIRTY; um->detached = false; um->text = packed; um->text_size = (colnr_T)STRLEN(packed) + 1; } /* * Close an unpacked memline, storing any changes. */ void um_close(unpacked_memline_T *um) { if (um->buf == NULL) return; if (um->detached && um->lnum > 0) um_store_changes(um); um_free(um); um->buf = NULL; } /* * Abort an unpacked memline, discarding any changes. */ void um_abort(unpacked_memline_T *um) { um_free(um); um->buf = NULL; } /* * In a hashtable item "hi_key" points to "pt_name" in a proptype_T. * This avoids adding a pointer to the hashtable item. * PT2HIKEY() converts a proptype pointer to a hashitem key pointer. * HIKEY2PT() converts a hashitem key pointer to a proptype pointer. * HI2PT() converts a hashitem pointer to a proptype pointer. */ #define PT2HIKEY(p) ((p)->pt_name) #define HIKEY2PT(p) ((proptype_T *)((p) - offsetof(proptype_T, pt_name))) #define HI2PT(hi) HIKEY2PT((hi)->hi_key) // The global text property types. static hashtab_T *global_proptypes = NULL; static proptype_T **global_proparray = NULL; // The last used text property type ID. static int proptype_id = 0; /* * Find a property type by name, return the hashitem. * Returns NULL if the item can't be found. */ static hashitem_T * find_prop_type_hi(char_u *name, buf_T *buf) { hashtab_T *ht; hashitem_T *hi; if (*name == NUL) return NULL; if (buf == NULL) ht = global_proptypes; else ht = buf->b_proptypes; if (ht == NULL) return NULL; hi = hash_find(ht, name); if (HASHITEM_EMPTY(hi)) return NULL; return hi; } /* * Like find_prop_type_hi() but return the property type. */ static proptype_T * find_prop_type(char_u *name, buf_T *buf) { hashitem_T *hi = find_prop_type_hi(name, buf); if (hi == NULL) return NULL; return HI2PT(hi); } /* * Get the prop type ID of "name". * When not found return zero. */ int find_prop_type_id(char_u *name, buf_T *buf) { proptype_T *pt = find_prop_type(name, buf); if (pt == NULL) return 0; return pt->pt_id; } /* * Lookup a property type by name. First in "buf" and when not found in the * global types. * When not found gives an error message and returns NULL. */ static proptype_T * lookup_prop_type(char_u *name, buf_T *buf) { proptype_T *type = find_prop_type(name, buf); if (type == NULL) type = find_prop_type(name, NULL); if (type == NULL) semsg(_(e_property_type_str_does_not_exist), name); return type; } /* * Get an optional "bufnr" item from the dict in "arg". * When the argument is not used or "bufnr" is not present then "buf" is * unchanged. * If "bufnr" is valid or not present return OK. * When "arg" is not a dict or "bufnr" is invalid return FAIL. */ static int get_bufnr_from_arg(typval_T *arg, buf_T **buf) { dictitem_T *di; if (arg->v_type != VAR_DICT) { emsg(_(e_dictionary_required)); return FAIL; } if (arg->vval.v_dict == NULL) return OK; // NULL dict is like an empty dict di = dict_find(arg->vval.v_dict, (char_u *)"bufnr", -1); if (di != NULL && (di->di_tv.v_type != VAR_NUMBER || di->di_tv.vval.v_number != 0)) { *buf = get_buf_arg(&di->di_tv); if (*buf == NULL) return FAIL; } return OK; } /* * prop_add({lnum}, {col}, {props}) */ void f_prop_add(typval_T *argvars, typval_T *rettv) { linenr_T start_lnum; colnr_T start_col; if (in_vim9script() && (check_for_number_arg(argvars, 0) == FAIL || check_for_number_arg(argvars, 1) == FAIL || check_for_dict_arg(argvars, 2) == FAIL)) return; start_lnum = tv_get_number(&argvars[0]); start_col = tv_get_number(&argvars[1]); if (check_for_dict_arg(argvars, 2) == FAIL) return; rettv->vval.v_number = prop_add_common(start_lnum, start_col, argvars[2].vval.v_dict, curbuf, &argvars[2]); } /* * Attach a text property 'type_name' to the text starting * at [start_lnum, start_col] and ending at [end_lnum, end_col] in * the buffer "buf" and assign identifier "id". * When "text" is not NULL store it as virtual text in the memline. */ static int prop_add_one( buf_T *buf, char_u *type_name, int id, char_u *text_arg, int text_padding_left, int text_flags, linenr_T start_lnum, linenr_T end_lnum, colnr_T start_col, colnr_T end_col) { proptype_T *type; linenr_T lnum; int proplen; char_u *props = NULL; char_u *newprops; size_t textlen; char_u *newtext; int i; textprop_T tmp_prop; char_u *text = text_arg; int res = FAIL; type = lookup_prop_type(type_name, buf); if (type == NULL) goto theend; if (start_lnum < 1 || start_lnum > buf->b_ml.ml_line_count) { semsg(_(e_invalid_line_number_nr), (long)start_lnum); goto theend; } if (end_lnum < start_lnum || end_lnum > buf->b_ml.ml_line_count) { semsg(_(e_invalid_line_number_nr), (long)end_lnum); goto theend; } if (buf->b_ml.ml_mfp == NULL) { emsg(_(e_cannot_add_text_property_to_unloaded_buffer)); goto theend; } if (text != NULL) { char_u *p; // change any control character (Tab, Newline, etc.) to a Space to make // it simpler to compute the size for (p = text; *p != NUL; MB_PTR_ADV(p)) if (*p < ' ') *p = ' '; } for (lnum = start_lnum; lnum <= end_lnum; ++lnum) { colnr_T col; // start column use in tp_col colnr_T sort_col; // column where it appears long length; // in bytes // Fetch the line to get the ml_line_len field updated. proplen = get_text_props(buf, lnum, &props, TRUE); textlen = ml_get_buf_len(buf, lnum) + 1; if (lnum == start_lnum) col = start_col; else col = 1; if (col - 1 > (colnr_T)textlen && !(col == 0 && text_arg != NULL)) { semsg(_(e_invalid_column_number_nr), (long)start_col); goto theend; } sort_col = col; if (lnum == end_lnum) length = end_col - col; else length = (int)textlen - col + 1; if (length > (long)textlen) length = (int)textlen; // can include the end-of-line if (length < 0) length = 0; // zero-width property if (text_arg != NULL) { length = (long)STRLEN(text_arg); if (col == 0) { col = MAXCOL; // before or after the line if ((text_flags & TP_FLAG_ALIGN_ABOVE) == 0) sort_col = MAXCOL; } } // Build the new property. CLEAR_FIELD(tmp_prop); tmp_prop.tp_col = col; tmp_prop.tp_len = length; tmp_prop.tp_id = id; tmp_prop.tp_type = type->pt_id; tmp_prop.tp_flags = text_flags | (lnum > start_lnum ? TP_FLAG_CONT_PREV : 0) | (lnum < end_lnum ? TP_FLAG_CONT_NEXT : 0) | ((type->pt_flags & PT_FLAG_INS_START_INCL) ? TP_FLAG_START_INCL : 0); tmp_prop.tp_padleft = text_padding_left; if (text_arg != NULL) tmp_prop.u.tp_text = text_arg; // Find the index where to insert the new property. for (i = 0; i < proplen; ++i) { textprop_T existing; colnr_T prop_col; mch_memmove(&existing, props + i * sizeof(textprop_T), sizeof(textprop_T)); prop_col = (existing.tp_flags & TP_FLAG_ALIGN_ABOVE) ? 0 : existing.tp_col; if (prop_col >= sort_col) break; } // Compute the sizes for the new memline format: // [text][NUL][prop_count][textprop_T...][vtext...] uint16_t new_propcount = (uint16_t)(proplen + 1); int vtext_total = 0; int new_line_len; char_u *count_dest; char_u *vtext_dest; int j; // Compute total vtext size from existing props. for (j = 0; j < proplen; ++j) { textprop_T ep; mch_memmove(&ep, props + j * sizeof(textprop_T), sizeof(textprop_T)); if (ep.tp_id < 0 && ep.tp_len > 0) vtext_total += ep.tp_len + 1; } // Add new vtext if this is a virtual text prop. if (text_arg != NULL) vtext_total += length + 1; new_line_len = (int)textlen + (int)PROP_COUNT_SIZE + new_propcount * (int)sizeof(textprop_T) + vtext_total; newtext = alloc(new_line_len); if (newtext == NULL) goto theend; // Copy line text. mch_memmove(newtext, buf->b_ml.ml_line_ptr, textlen); // Write prop_count. count_dest = newtext + textlen; mch_memmove(count_dest, &new_propcount, PROP_COUNT_SIZE); // Write properties: [0..i-1] existing, [i] new, [i..proplen-1] // existing. newprops = count_dest + PROP_COUNT_SIZE; if (i > 0) mch_memmove(newprops, props, sizeof(textprop_T) * i); // new prop is written after vtext offsets are computed if (i < proplen) mch_memmove(newprops + (i + 1) * sizeof(textprop_T), props + i * sizeof(textprop_T), sizeof(textprop_T) * (proplen - i)); // Write vtext strings and set offsets. vtext_dest = newprops + new_propcount * sizeof(textprop_T); for (j = 0; j < new_propcount; ++j) { textprop_T ep; if (j == i) continue; // handle new prop separately below mch_memmove(&ep, newprops + j * sizeof(textprop_T), sizeof(textprop_T)); if (ep.tp_id < 0 && ep.tp_len > 0) { // Copy existing vtext from old memline data. char_u *old_count = (char_u *)props - PROP_COUNT_SIZE; char_u *old_vtext = old_count + ep.u.tp_text_offset; mch_memmove(vtext_dest, old_vtext, ep.tp_len + 1); ep.u.tp_text_offset = (colnr_T)(vtext_dest - count_dest); mch_memmove(newprops + j * sizeof(textprop_T), &ep, sizeof(textprop_T)); vtext_dest += ep.tp_len + 1; } } // Write new prop with vtext. if (text_arg != NULL) { int copysize = tmp_prop.tp_len + 1; mch_memmove(vtext_dest, text_arg, copysize); tmp_prop.u.tp_text_offset = (colnr_T)(vtext_dest - count_dest); vtext_dest += copysize; } else tmp_prop.u.tp_text_offset = 0; mch_memmove(newprops + i * sizeof(textprop_T), &tmp_prop, sizeof(textprop_T)); if (buf->b_ml.ml_flags & (ML_LINE_DIRTY | ML_ALLOCATED)) vim_free(buf->b_ml.ml_line_ptr); buf->b_ml.ml_line_ptr = newtext; buf->b_ml.ml_line_len = new_line_len; buf->b_ml.ml_flags |= ML_LINE_DIRTY; } changed_line_display_buf(buf); changed_lines_buf(buf, start_lnum, end_lnum + 1, 0); res = OK; theend: vim_free(text); return res; } /* * prop_add_list() * First argument specifies the text property: * {'type': , 'id': , 'bufnr': } * Second argument is a List where each item is a List with the following * entries: [lnum, start_col, end_col] */ void f_prop_add_list(typval_T *argvars, typval_T *rettv UNUSED) { dict_T *dict; char_u *type_name; buf_T *buf = curbuf; int id = 0; listitem_T *li; list_T *pos_list; linenr_T start_lnum; colnr_T start_col; linenr_T end_lnum; colnr_T end_col; int error = FALSE; int prev_did_emsg = did_emsg; if (check_for_dict_arg(argvars, 0) == FAIL || check_for_list_arg(argvars, 1) == FAIL) return; if (check_for_nonnull_list_arg(argvars, 1) == FAIL) return; dict = argvars[0].vval.v_dict; if (dict == NULL || !dict_has_key(dict, "type")) { emsg(_(e_missing_property_type_name)); return; } type_name = dict_get_string(dict, "type", FALSE); if (dict_has_key(dict, "id")) { vimlong_T x; x = dict_get_number(dict, "id"); if (x > INT_MAX || x <= INT_MIN) { semsg(_(e_val_too_large), dict_get_string(dict, "id", FALSE)); return; } id = (int)x; } if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL) return; // This must be done _before_ we start adding properties because property // changes trigger buffer (memline) reorganisation, which needs this flag // to be correctly set. buf->b_has_textprop = true; // this is never reset FOR_ALL_LIST_ITEMS(argvars[1].vval.v_list, li) { if (li->li_tv.v_type != VAR_LIST || li->li_tv.vval.v_list == NULL) { emsg(_(e_list_required)); return; } pos_list = li->li_tv.vval.v_list; start_lnum = list_find_nr(pos_list, 0L, &error); if (!error) start_col = list_find_nr(pos_list, 1L, &error); if (!error) end_lnum = list_find_nr(pos_list, 2L, &error); if (!error) end_col = list_find_nr(pos_list, 3L, &error); int this_id = id; if (!error && pos_list->lv_len > 4) this_id = list_find_nr(pos_list, 4L, &error); if (error || start_lnum <= 0 || start_col <= 0 || end_lnum <= 0 || end_col <= 0) { if (prev_did_emsg == did_emsg) emsg(_(e_invalid_argument)); return; } if (prop_add_one(buf, type_name, this_id, NULL, 0, 0, start_lnum, end_lnum, start_col, end_col) == FAIL) return; } redraw_buf_later(buf, UPD_VALID); } // Counter for virtual text property IDs (decremented for each new one). static int vt_id_counter = 0; /* * Get the next ID to use for a textprop. */ static int get_textprop_id(void) { if (vt_id_counter > 0) vt_id_counter = 0; return --vt_id_counter; } /* * Shared between prop_add() and popup_create(). * "dict_arg" is the function argument of a dict containing "bufnr". * it is NULL for popup_create(). * Returns the "id" used for "text" or zero. */ int prop_add_common( linenr_T start_lnum, colnr_T start_col, dict_T *dict, buf_T *default_buf, typval_T *dict_arg) { linenr_T end_lnum; colnr_T end_col; char_u *type_name; buf_T *buf = default_buf; int id = 0; char_u *text = NULL; int text_padding_left = 0; int flags = 0; if (dict == NULL || !dict_has_key(dict, "type")) { emsg(_(e_missing_property_type_name)); goto theend; } type_name = dict_get_string(dict, "type", FALSE); if (dict_has_key(dict, "end_lnum")) { end_lnum = dict_get_number(dict, "end_lnum"); if (end_lnum < start_lnum) { semsg(_(e_invalid_value_for_argument_str), "end_lnum"); goto theend; } } else end_lnum = start_lnum; if (dict_has_key(dict, "length")) { long length = dict_get_number(dict, "length"); if (length < 0 || end_lnum > start_lnum) { semsg(_(e_invalid_value_for_argument_str), "length"); goto theend; } end_col = start_col + length; } else if (dict_has_key(dict, "end_col")) { end_col = dict_get_number(dict, "end_col"); if (end_col <= 0) { semsg(_(e_invalid_value_for_argument_str), "end_col"); goto theend; } } else if (start_lnum == end_lnum) end_col = start_col; else end_col = 1; if (dict_has_key(dict, "id")) { vimlong_T x; x = dict_get_number(dict, "id"); if (x > INT_MAX || x <= INT_MIN) { semsg(_(e_val_too_large), dict_get_string(dict, "id", FALSE)); goto theend; } id = (int)x; } if (dict_has_key(dict, "text")) { if (dict_has_key(dict, "length") || dict_has_key(dict, "end_col") || dict_has_key(dict, "end_lnum")) { emsg(_(e_cannot_use_length_endcol_and_endlnum_with_text)); goto theend; } text = dict_get_string(dict, "text", TRUE); if (text == NULL) goto theend; // use a default length of 1 to make multiple props show up end_col = start_col + 1; if (dict_has_key(dict, "text_align")) { char_u *p = dict_get_string(dict, "text_align", FALSE); if (p == NULL) goto theend; if (start_col != 0) { emsg(_(e_can_only_use_text_align_when_column_is_zero)); goto theend; } if (STRCMP(p, "right") == 0) flags |= TP_FLAG_ALIGN_RIGHT; else if (STRCMP(p, "above") == 0) flags |= TP_FLAG_ALIGN_ABOVE; else if (STRCMP(p, "below") == 0) flags |= TP_FLAG_ALIGN_BELOW; else if (STRCMP(p, "after") != 0) { semsg(_(e_invalid_value_for_argument_str_str), "text_align", p); goto theend; } } if (dict_has_key(dict, "text_padding_left")) { text_padding_left = dict_get_number(dict, "text_padding_left"); if (text_padding_left < 0) { semsg(_(e_argument_must_be_positive_str), "text_padding_left"); goto theend; } } if (dict_has_key(dict, "text_wrap")) { char_u *p = dict_get_string(dict, "text_wrap", FALSE); if (p == NULL) goto theend; if (STRCMP(p, "wrap") == 0) flags |= TP_FLAG_WRAP; else if (STRCMP(p, "truncate") != 0) { semsg(_(e_invalid_value_for_argument_str_str), "text_wrap", p); goto theend; } } } // Column must be 1 or more for a normal text property; when "text" is // present zero means it goes after the line. if (start_col < (text == NULL ? 1 : 0)) { semsg(_(e_invalid_column_number_nr), (long)start_col); goto theend; } if (start_col > 0 && text_padding_left > 0) { emsg(_(e_can_only_use_left_padding_when_column_is_zero)); goto theend; } if (dict_arg != NULL && get_bufnr_from_arg(dict_arg, &buf) == FAIL) goto theend; if (text != NULL) // Always assign an internal negative id; ignore any user-provided id. id = get_textprop_id(); else if (id < 0) { emsg(_(e_cannot_use_negative_id)); goto theend; } // This must be done _before_ we add the property because property changes // trigger buffer (memline) reorganisation, which needs this flag to be // correctly set. buf->b_has_textprop = true; // this is never reset prop_add_one(buf, type_name, id, text, text_padding_left, flags, start_lnum, end_lnum, start_col, end_col); text = NULL; redraw_buf_later(buf, UPD_VALID); theend: vim_free(text); return id; } /* * Fetch the text properties for line "lnum" in buffer "buf". * Returns the number of text properties and, when non-zero, a pointer to the * first one in "props" (note that it is not aligned, therefore the char_u * pointer). */ int get_text_props(buf_T *buf, linenr_T lnum, char_u **props, int will_change) { char_u *text; size_t textlen; size_t propdata_len; uint16_t prop_count; // Be quick when no text property types have been defined for the buffer, // unless we are adding one. if ((!buf->b_has_textprop && !will_change) || buf->b_ml.ml_mfp == NULL) return 0; // Fetch the line to get the ml_line_len field updated. text = ml_get_buf(buf, lnum, will_change); textlen = ml_get_buf_len(buf, lnum) + 1; propdata_len = buf->b_ml.ml_line_len - textlen; if (propdata_len == 0) return 0; // Format: [prop_count (uint16)][textprop_T...][vtext...] // Lines without properties have no prop data at all. // prop_count is never zero. if (propdata_len < PROP_COUNT_SIZE + sizeof(textprop_T)) { iemsg(e_text_property_info_corrupted); return 0; } mch_memmove(&prop_count, text + textlen, PROP_COUNT_SIZE); *props = text + textlen + PROP_COUNT_SIZE; return (int)prop_count; } /* * Return the number of text properties with "above" or "below" alignment in * line "lnum". A "right" aligned property also goes below after a "below" or * other "right" aligned property. */ int prop_count_above_below(buf_T *buf, linenr_T lnum) { char_u *props; int count = get_text_props(buf, lnum, &props, FALSE); int result = 0; textprop_T prop; int i; int next_right_goes_below = FALSE; if (count == 0) return 0; for (i = 0; i < count; ++i) { mch_memmove(&prop, props + i * sizeof(prop), sizeof(prop)); if (prop.tp_col == MAXCOL && text_prop_type_valid(buf, &prop)) { if ((prop.tp_flags & (TP_FLAG_ALIGN_ABOVE | TP_FLAG_ALIGN_BELOW)) || (next_right_goes_below && (prop.tp_flags & TP_FLAG_ALIGN_RIGHT))) { ++result; } else if (prop.tp_flags & TP_FLAG_ALIGN_RIGHT) next_right_goes_below = TRUE; } } return result; } /** * Return the number of text properties on line "lnum" in the current buffer. * When "only_starting" is true only text properties starting in this line will * be considered. * When "last_line" is FALSE then text properties after the line are not * counted. */ int count_props(linenr_T lnum, int only_starting, int last_line) { char_u *props; int proplen = get_text_props(curbuf, lnum, &props, 0); int result = proplen; textprop_T prop; for (int i = 0; i < proplen; ++i) { mch_memmove(&prop, props + i * sizeof(prop), sizeof(prop)); // A prop is dropped when in the first line and it continues from the // previous line, or when not in the last line and it is virtual text // after the line. if ((only_starting && (prop.tp_flags & TP_FLAG_CONT_PREV)) || (!last_line && prop.tp_col == MAXCOL)) --result; } return result; } static textprop_T *text_prop_compare_props; static buf_T *text_prop_compare_buf; /* * Score for sorting on position of the text property: 0: above, * 1: after (default), 2: right, 3: below (comes last) */ static int text_prop_order(int flags) { if (flags & TP_FLAG_ALIGN_ABOVE) return 0; if (flags & TP_FLAG_ALIGN_RIGHT) return 2; if (flags & TP_FLAG_ALIGN_BELOW) return 3; return 1; } /* * Function passed to qsort() to sort text properties. * Return 1 if "s1" has priority over "s2", -1 if the other way around, zero if * both have the same priority. */ static int text_prop_compare(const void *s1, const void *s2) { int idx1, idx2; textprop_T *tp1, *tp2; proptype_T *pt1, *pt2; colnr_T col1, col2; idx1 = *(int *)s1; idx2 = *(int *)s2; tp1 = &text_prop_compare_props[idx1]; tp2 = &text_prop_compare_props[idx2]; col1 = tp1->tp_col; col2 = tp2->tp_col; // property that inserts text has priority over one that doesn't if ((tp1->tp_id < 0) != (tp2->tp_id < 0)) return tp1->tp_id < 0 ? 1 : -1; if (col1 == MAXCOL || col2 == MAXCOL) { int order1 = text_prop_order(tp1->tp_flags); int order2 = text_prop_order(tp2->tp_flags); // sort on order where it is added if (order1 != order2) return order1 < order2 ? 1 : -1; } // check highest priority, defined by the type pt1 = text_prop_type_by_id(text_prop_compare_buf, tp1->tp_type); pt2 = text_prop_type_by_id(text_prop_compare_buf, tp2->tp_type); if (pt1 != pt2) { if (pt1 == NULL) return -1; if (pt2 == NULL) return 1; if (pt1->pt_priority != pt2->pt_priority) return pt1->pt_priority > pt2->pt_priority ? 1 : -1; } // same priority, one that starts first wins if (col1 != col2) return col1 < col2 ? 1 : -1; // for a property with text the id can be used as tie breaker if (tp1->tp_id < 0) return tp1->tp_id > tp2->tp_id ? 1 : -1; return 0; } /* * Sort "count" text properties using an array if indexes "idxs" into the list * of text props "props" for buffer "buf". */ void sort_text_props( buf_T *buf, textprop_T *props, int *idxs, int count) { text_prop_compare_buf = buf; text_prop_compare_props = props; qsort((void *)idxs, (size_t)count, sizeof(int), text_prop_compare); } /* * Find text property "type_id" in lines [first_lnum, last_lnum] of window * "wp"'s buffer. Match "id" when it is > 0. Returns false when not found. */ bool find_prop_in_lines( win_T *wp, int type_id, int id, textprop_T *prop, linenr_T *found_lnum, linenr_T first_lnum, linenr_T last_lnum) { if (text_prop_type_by_id(wp->w_buffer, type_id) == NULL) return false; if (first_lnum < 1) first_lnum = 1; if (last_lnum > wp->w_buffer->b_ml.ml_line_count) last_lnum = wp->w_buffer->b_ml.ml_line_count; for (linenr_T lnum = first_lnum; lnum <= last_lnum; ++lnum) { char_u *props; int count = get_text_props(wp->w_buffer, lnum, &props, FALSE); for (int i = 0; i < count; ++i) { mch_memmove(prop, props + i * sizeof(textprop_T), sizeof(textprop_T)); if (prop->tp_type == type_id && (id <= 0 || prop->tp_id == id)) { *found_lnum = lnum; return true; } } } return false; } /* * Find text property "type_id" in the visible lines of window "wp". * Match "id" when it is > 0. * Returns false when not found. */ bool find_visible_prop( win_T *wp, int type_id, int id, textprop_T *prop, linenr_T *found_lnum) { // w_botline may not have been updated yet. validate_botline_win(wp); return find_prop_in_lines(wp, type_id, id, prop, found_lnum, wp->w_topline, wp->w_botline - 1); } /* * Set the text properties for line "lnum" to "tps" array with "count" entries. * If "count" is zero text properties are removed. * Any existing text properties are dropped. * For virtual text props, u.tp_text must point to the vtext string. * Only works for the current buffer. */ static void set_text_props(linenr_T lnum, textprop_T *tps, int count) { char_u *text; char_u *newtext; int textlen; text = ml_get(lnum); textlen = ml_get_len(lnum) + 1; if (count == 0) { // No properties: just text. newtext = alloc(textlen); if (newtext == NULL) return; mch_memmove(newtext, text, textlen); if (curbuf->b_ml.ml_flags & (ML_LINE_DIRTY | ML_ALLOCATED)) vim_free(curbuf->b_ml.ml_line_ptr); curbuf->b_ml.ml_line_ptr = newtext; curbuf->b_ml.ml_line_len = textlen; curbuf->b_ml.ml_flags |= ML_LINE_DIRTY; } else { // Build new format: [text][NUL][prop_count][props...][vtext...] uint16_t prop_count = (uint16_t)count; int vtext_size = 0; int total_len; char_u *count_dest; char_u *prop_dest; char_u *vtext_dest; int i; for (i = 0; i < count; ++i) if (tps[i].tp_flags & TP_FLAG_VTEXT_PTR) vtext_size += tps[i].tp_len + 1; total_len = textlen + (int)PROP_COUNT_SIZE + count * (int)sizeof(textprop_T) + vtext_size; newtext = alloc(total_len); if (newtext == NULL) return; mch_memmove(newtext, text, textlen); count_dest = newtext + textlen; mch_memmove(count_dest, &prop_count, PROP_COUNT_SIZE); prop_dest = count_dest + PROP_COUNT_SIZE; vtext_dest = prop_dest + count * sizeof(textprop_T); for (i = 0; i < count; ++i) { textprop_T prop = tps[i]; if (prop.tp_flags & TP_FLAG_VTEXT_PTR) { int copysize = prop.tp_len + 1; mch_memmove(vtext_dest, prop.u.tp_text, copysize); vim_free(tps[i].u.tp_text); tps[i].u.tp_text = NULL; prop.u.tp_text_offset = (colnr_T)(vtext_dest - count_dest); vtext_dest += copysize; } else prop.u.tp_text_offset = 0; mch_memmove(prop_dest, &prop, sizeof(textprop_T)); prop_dest += sizeof(textprop_T); } if (curbuf->b_ml.ml_flags & (ML_LINE_DIRTY | ML_ALLOCATED)) vim_free(curbuf->b_ml.ml_line_ptr); curbuf->b_ml.ml_line_ptr = newtext; curbuf->b_ml.ml_line_len = total_len; curbuf->b_ml.ml_flags |= ML_LINE_DIRTY; } } /* * Convert a line buffer with text properties in the old format * [text][NUL][textprop_T...] to the new format * [text][NUL][prop_count][textprop_T...][vtext...]. * For virtual text props, u.tp_text is expected to be a pointer to the * vtext string (allocated). These strings are freed after copying. * Returns a newly allocated buffer, or NULL on failure. * "line_len" is the total length (text + props). "textlen" is the length * of the text including NUL. "*new_len" is set to the new total length. */ char_u * props_add_count_header(char_u *line, int line_len, int textlen, int *new_len) { int prop_bytes = line_len - textlen; uint16_t prop_count; int vtext_size = 0; int total; char_u *newline; char_u *count_dest; char_u *prop_dest; char_u *vtext_dest; int i; if (prop_bytes <= 0 || prop_bytes % sizeof(textprop_T) != 0) { *new_len = line_len; return NULL; } prop_count = (uint16_t)(prop_bytes / sizeof(textprop_T)); // Calculate total vtext size. for (i = 0; i < (int)prop_count; ++i) { textprop_T prop; mch_memmove(&prop, line + textlen + i * sizeof(textprop_T), sizeof(textprop_T)); if (prop.tp_flags & TP_FLAG_VTEXT_PTR) vtext_size += prop.tp_len + 1; } total = textlen + (int)PROP_COUNT_SIZE + prop_bytes + vtext_size; newline = alloc(total); if (newline == NULL) return NULL; *new_len = total; mch_memmove(newline, line, textlen); count_dest = newline + textlen; mch_memmove(count_dest, &prop_count, PROP_COUNT_SIZE); prop_dest = count_dest + PROP_COUNT_SIZE; vtext_dest = prop_dest + prop_bytes; for (i = 0; i < (int)prop_count; ++i) { textprop_T prop; mch_memmove(&prop, line + textlen + i * sizeof(textprop_T), sizeof(textprop_T)); if (prop.tp_flags & TP_FLAG_VTEXT_PTR) { int copysize = prop.tp_len + 1; mch_memmove(vtext_dest, prop.u.tp_text, copysize); vim_free(prop.u.tp_text); prop.u.tp_text_offset = (colnr_T)(vtext_dest - count_dest); vtext_dest += copysize; } else prop.u.tp_text_offset = 0; mch_memmove(prop_dest + i * sizeof(textprop_T), &prop, sizeof(textprop_T)); } return newline; } /* * Add "text_props" with "text_prop_count" text properties to line "lnum". */ void add_text_props(linenr_T lnum, textprop_T *text_props, int text_prop_count) { unpacked_memline_T um; int i; um = um_open_at(curbuf, lnum, text_prop_count); if (um.buf == NULL) return; if (!um_detach(&um)) { um_abort(&um); return; } // Grow props array if needed. if (um.prop_count + text_prop_count > um.prop_size) { textprop_T *newprops = vim_realloc(um.props, (um.prop_count + text_prop_count) * sizeof(textprop_T)); if (newprops == NULL) { um_abort(&um); return; } um.props = newprops; um.prop_size = um.prop_count + text_prop_count; } for (i = 0; i < text_prop_count; ++i) um.props[um.prop_count++] = text_props[i]; um_close(&um); } /* * Function passed to qsort() for sorting proptype_T on pt_id. */ static int compare_pt(const void *s1, const void *s2) { proptype_T *tp1 = *(proptype_T **)s1; proptype_T *tp2 = *(proptype_T **)s2; return tp1->pt_id == tp2->pt_id ? 0 : tp1->pt_id < tp2->pt_id ? -1 : 1; } static proptype_T * find_type_by_id(hashtab_T *ht, proptype_T ***array, int id) { int low = 0; int high; if (ht == NULL || ht->ht_used == 0) return NULL; // Make the lookup faster by creating an array with pointers to // hashtable entries, sorted on pt_id. if (*array == NULL) { long todo; hashitem_T *hi; int i = 0; *array = ALLOC_MULT(proptype_T *, ht->ht_used); if (*array == NULL) return NULL; todo = (long)ht->ht_used; FOR_ALL_HASHTAB_ITEMS(ht, hi, todo) { if (!HASHITEM_EMPTY(hi)) { (*array)[i++] = HI2PT(hi); --todo; } } qsort((void *)*array, ht->ht_used, sizeof(proptype_T *), compare_pt); } // binary search in the sorted array high = ht->ht_used; while (high > low) { int m = (high + low) / 2; if ((*array)[m]->pt_id == id) return (*array)[m]; if ((*array)[m]->pt_id > id) high = m; else low = m + 1; } return NULL; } /* * Fill 'dict' with text properties in 'prop'. */ static void prop_fill_dict(dict_T *dict, textprop_T *prop, buf_T *buf) { proptype_T *pt; int buflocal = TRUE; int virtualtext_prop = prop->tp_id < 0; dict_add_number(dict, "col", (prop->tp_col == MAXCOL) ? 0 : prop->tp_col); if (!virtualtext_prop) { dict_add_number(dict, "length", prop->tp_len); dict_add_number(dict, "id", prop->tp_id); } dict_add_number(dict, "start", !(prop->tp_flags & TP_FLAG_CONT_PREV)); dict_add_number(dict, "end", !(prop->tp_flags & TP_FLAG_CONT_NEXT)); pt = find_type_by_id(buf->b_proptypes, &buf->b_proparray, prop->tp_type); if (pt == NULL) { pt = find_type_by_id(global_proptypes, &global_proparray, prop->tp_type); buflocal = FALSE; } if (pt != NULL) dict_add_string(dict, "type", pt->pt_name); if (buflocal) dict_add_number(dict, "type_bufnr", buf->b_fnum); else dict_add_number(dict, "type_bufnr", 0); if (virtualtext_prop) { string_T text_align = {NULL, 0}; // virtual text property - u.tp_text must be set by caller dict_add_string(dict, "text", prop->u.tp_text); // text_align if (prop->tp_flags & TP_FLAG_ALIGN_RIGHT) STR_LITERAL_SET(text_align, "right"); else if (prop->tp_flags & TP_FLAG_ALIGN_ABOVE) STR_LITERAL_SET(text_align, "above"); else if (prop->tp_flags & TP_FLAG_ALIGN_BELOW) STR_LITERAL_SET(text_align, "below"); if (text_align.string != NULL) dict_add_string_len(dict, "text_align", text_align.string, (int)text_align.length); // text_wrap if (prop->tp_flags & TP_FLAG_WRAP) dict_add_string_len(dict, "text_wrap", (char_u *)"wrap", STRLEN_LITERAL("wrap")); if (prop->tp_padleft != 0) dict_add_number(dict, "text_padding_left", prop->tp_padleft); } } /* * Find a property type by ID in "buf" or globally. * Returns NULL if not found. */ proptype_T * text_prop_type_by_id(buf_T *buf, int id) { proptype_T *type; type = find_type_by_id(buf->b_proptypes, &buf->b_proparray, id); if (type == NULL) type = find_type_by_id(global_proptypes, &global_proparray, id); return type; } /* * Return true if "prop" is a valid text property type. */ bool text_prop_type_valid(buf_T *buf, textprop_T *prop) { return text_prop_type_by_id(buf, prop->tp_type) != NULL; } /* * prop_clear({lnum} [, {lnum_end} [, {bufnr}]]) */ void f_prop_clear(typval_T *argvars, typval_T *rettv UNUSED) { linenr_T start; linenr_T end; linenr_T lnum; buf_T *buf = curbuf; int did_clear = FALSE; if (in_vim9script() && (check_for_number_arg(argvars, 0) == FAIL || check_for_opt_number_arg(argvars, 1) == FAIL || (argvars[1].v_type != VAR_UNKNOWN && check_for_opt_dict_arg(argvars, 2) == FAIL))) return; start = tv_get_number(&argvars[0]); end = start; if (argvars[1].v_type != VAR_UNKNOWN) { end = tv_get_number(&argvars[1]); if (argvars[2].v_type != VAR_UNKNOWN) { if (get_bufnr_from_arg(&argvars[2], &buf) == FAIL) return; } } if (start < 1 || end < 1) { emsg(_(e_invalid_range)); return; } { unpacked_memline_T um = um_open(buf); for (lnum = start; lnum <= end; ++lnum) { int idx; if (lnum > buf->b_ml.ml_line_count) break; if (!um_goto_line(&um, lnum, 0)) break; if (um.prop_count > 0) { did_clear = TRUE; for (idx = um.prop_count - 1; idx >= 0; --idx) um_delete_prop(&um, idx); } } um_close(&um); } if (did_clear) redraw_buf_later(buf, UPD_NOT_VALID); } /* * prop_find({props} [, {direction}]) */ void f_prop_find(typval_T *argvars, typval_T *rettv) { pos_T *cursor = &curwin->w_cursor; dict_T *dict; buf_T *buf = curbuf; dictitem_T *di; int lnum_start; int start_pos_has_prop = 0; int seen_end = FALSE; int id = 0; int id_found = FALSE; int type_id = -1; int skipstart = FALSE; int lnum = -1; int col = -1; int dir = FORWARD; // FORWARD == 1, BACKWARD == -1 int both; if (in_vim9script() && (check_for_dict_arg(argvars, 0) == FAIL || check_for_opt_string_arg(argvars, 1) == FAIL)) return; if (check_for_nonnull_dict_arg(argvars, 0) == FAIL) return; dict = argvars[0].vval.v_dict; if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL) return; if (buf->b_ml.ml_mfp == NULL) return; if (argvars[1].v_type != VAR_UNKNOWN) { char_u *dir_s = tv_get_string(&argvars[1]); if (*dir_s == 'b') dir = BACKWARD; else if (*dir_s != 'f') { emsg(_(e_invalid_argument)); return; } } di = dict_find(dict, (char_u *)"lnum", -1); if (di != NULL) lnum = tv_get_number(&di->di_tv); di = dict_find(dict, (char_u *)"col", -1); if (di != NULL) col = tv_get_number(&di->di_tv); if (lnum == -1) { lnum = cursor->lnum; col = cursor->col + 1; } else if (col == -1) col = 1; if (lnum < 1 || lnum > buf->b_ml.ml_line_count) { emsg(_(e_invalid_range)); return; } skipstart = dict_get_bool(dict, "skipstart", 0); if (dict_has_key(dict, "id")) { id = dict_get_number(dict, "id"); id_found = TRUE; } if (dict_has_key(dict, "type")) { char_u *name = dict_get_string(dict, "type", FALSE); proptype_T *type = lookup_prop_type(name, buf); if (type == NULL) return; type_id = type->pt_id; } both = dict_get_bool(dict, "both", FALSE); if (!id_found && type_id == -1) { emsg(_(e_need_at_least_one_of_id_or_type)); return; } if (both && (!id_found || type_id == -1)) { emsg(_(e_need_id_and_type_or_types_with_both)); return; } lnum_start = lnum; if (rettv_dict_alloc(rettv) == FAIL) return; while (1) { char_u *prop_start_ptr; int count = get_text_props(buf, lnum, &prop_start_ptr, FALSE); int i; textprop_T prop; int prop_start; int prop_end; for (i = dir == BACKWARD ? count - 1 : 0; i >= 0 && i < count; i += dir) { mch_memmove(&prop, prop_start_ptr + i * sizeof(textprop_T), sizeof(textprop_T)); // Convert offset to pointer for virtual text props. if (prop.tp_id < 0 && prop.u.tp_text_offset > 0) { prop.u.tp_text = (prop_start_ptr - PROP_COUNT_SIZE) + prop.u.tp_text_offset; prop.tp_flags |= TP_FLAG_VTEXT_PTR; } else prop.u.tp_text = NULL; // For the very first line try to find the first property before or // after `col`, depending on the search direction. if (lnum == lnum_start) { bool is_floating_vtext = prop.tp_id < 0 && prop.tp_col == MAXCOL; if (dir == BACKWARD) { // Virtual text with MAXCOL always matches any column if (!is_floating_vtext && prop.tp_col > col) continue; } else if (!is_floating_vtext && prop.tp_col + prop.tp_len - (prop.tp_len != 0) < col) continue; } if (both ? prop.tp_id == id && prop.tp_type == type_id : (id_found && prop.tp_id == id) || prop.tp_type == type_id) { // Check if the starting position has text props. if (lnum_start == lnum && col >= prop.tp_col && (col <= prop.tp_col + prop.tp_len - (prop.tp_len != 0))) start_pos_has_prop = 1; // The property was not continued from last line, it starts on // this line. prop_start = !(prop.tp_flags & TP_FLAG_CONT_PREV); // The property does not continue on the next line, it ends on // this line. prop_end = !(prop.tp_flags & TP_FLAG_CONT_NEXT); if (!prop_start && prop_end && dir == FORWARD) seen_end = 1; // Skip lines without the start flag. if (!prop_start) { // Always search backwards for start when search started // on a prop and we're not skipping. if (start_pos_has_prop && !skipstart) dir = BACKWARD; continue; } // If skipstart is true, skip the prop at start pos (even if // continued from another line). if (start_pos_has_prop && skipstart && !seen_end) { start_pos_has_prop = 0; continue; } prop_fill_dict(rettv->vval.v_dict, &prop, buf); dict_add_number(rettv->vval.v_dict, "lnum", lnum); return; } } if (dir > 0) { if (lnum >= buf->b_ml.ml_line_count) break; lnum++; } else { if (lnum <= 1) break; lnum--; } } } /* * Returns TRUE if 'type_or_id' is in the 'types_or_ids' list. */ static int prop_type_or_id_in_list(int *types_or_ids, int len, int type_or_id) { int i; for (i = 0; i < len; i++) if (types_or_ids[i] == type_or_id) return TRUE; return FALSE; } /* * Return all the text properties in line 'lnum' in buffer 'buf' in 'retlist'. * If 'prop_types' is not NULL, then return only the text properties with * matching property type in the 'prop_types' array. * If 'prop_ids' is not NULL, then return only the text properties with * an identifier in the 'props_ids' array. * If 'add_lnum' is TRUE, then add the line number also to the text property * dictionary. */ static void get_props_in_line( buf_T *buf, linenr_T lnum, int *prop_types, int prop_types_len, int *prop_ids, int prop_ids_len, list_T *retlist, int add_lnum) { char_u *props; int count; int i; textprop_T prop; count = get_text_props(buf, lnum, &props, FALSE); for (i = 0; i < count; ++i) { mch_memmove(&prop, props + i * sizeof(textprop_T), sizeof(textprop_T)); // Convert offset to pointer for virtual text props. if (prop.tp_id < 0 && prop.u.tp_text_offset > 0) { prop.u.tp_text = (props - PROP_COUNT_SIZE) + prop.u.tp_text_offset; prop.tp_flags |= TP_FLAG_VTEXT_PTR; } else prop.u.tp_text = NULL; if ((prop_types == NULL || prop_type_or_id_in_list(prop_types, prop_types_len, prop.tp_type)) && (prop_ids == NULL || prop_type_or_id_in_list(prop_ids, prop_ids_len, prop.tp_id))) { dict_T *d = dict_alloc(); if (d == NULL) break; prop_fill_dict(d, &prop, buf); if (add_lnum) dict_add_number(d, "lnum", lnum); list_append_dict(retlist, d); } } } /* * Convert a List of property type names into an array of property type * identifiers. Returns a pointer to the allocated array. Returns NULL on * error. 'num_types' is set to the number of returned property types. */ static int * get_prop_types_from_names(list_T *l, buf_T *buf, int *num_types) { int *prop_types; listitem_T *li; int i; char_u *name; proptype_T *type; *num_types = 0; prop_types = ALLOC_MULT(int, list_len(l)); if (prop_types == NULL) return NULL; i = 0; FOR_ALL_LIST_ITEMS(l, li) { if (li->li_tv.v_type != VAR_STRING) { emsg(_(e_string_required)); goto errret; } name = li->li_tv.vval.v_string; if (name == NULL) goto errret; type = lookup_prop_type(name, buf); if (type == NULL) goto errret; prop_types[i++] = type->pt_id; } *num_types = i; return prop_types; errret: VIM_CLEAR(prop_types); return NULL; } /* * Convert a List of property identifiers into an array of property * identifiers. Returns a pointer to the allocated array. Returns NULL on * error. 'num_ids' is set to the number of returned property identifiers. */ static int * get_prop_ids_from_list(list_T *l, int *num_ids) { int *prop_ids; listitem_T *li; int i = 0; int id; int error; *num_ids = 0; prop_ids = ALLOC_MULT(int, list_len(l)); if (prop_ids == NULL) return NULL; CHECK_LIST_MATERIALIZE(l); FOR_ALL_LIST_ITEMS(l, li) { error = FALSE; id = tv_get_number_chk(&li->li_tv, &error); if (error) goto errret; prop_ids[i++] = id; } *num_ids = i; return prop_ids; errret: VIM_CLEAR(prop_ids); return NULL; } /* * prop_list({lnum} [, {bufnr}]) */ void f_prop_list(typval_T *argvars, typval_T *rettv) { linenr_T lnum; linenr_T start_lnum; linenr_T end_lnum; buf_T *buf = curbuf; int add_lnum = FALSE; int *prop_types = NULL; int prop_types_len = 0; int *prop_ids = NULL; int prop_ids_len = 0; list_T *l; dictitem_T *di; if (in_vim9script() && (check_for_number_arg(argvars, 0) == FAIL || check_for_opt_dict_arg(argvars, 1) == FAIL)) return; if (rettv_list_alloc(rettv) == FAIL) return; // default: get text properties on current line start_lnum = tv_get_number(&argvars[0]); end_lnum = start_lnum; if (argvars[1].v_type != VAR_UNKNOWN) { dict_T *d; if (check_for_dict_arg(argvars, 1) == FAIL) return; d = argvars[1].vval.v_dict; if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) return; if (d != NULL && (di = dict_find(d, (char_u *)"end_lnum", -1)) != NULL) { if (di->di_tv.v_type != VAR_NUMBER) { emsg(_(e_number_required)); return; } end_lnum = tv_get_number(&di->di_tv); if (end_lnum < 0) // negative end_lnum is used as an offset from the last buffer // line end_lnum = buf->b_ml.ml_line_count + end_lnum + 1; else if (end_lnum > buf->b_ml.ml_line_count) end_lnum = buf->b_ml.ml_line_count; add_lnum = TRUE; } if (d != NULL && (di = dict_find(d, (char_u *)"types", -1)) != NULL) { if (di->di_tv.v_type != VAR_LIST) { emsg(_(e_list_required)); return; } l = di->di_tv.vval.v_list; if (l != NULL && list_len(l) > 0) { prop_types = get_prop_types_from_names(l, buf, &prop_types_len); if (prop_types == NULL) return; } } if (d != NULL && (di = dict_find(d, (char_u *)"ids", -1)) != NULL) { if (di->di_tv.v_type != VAR_LIST) { emsg(_(e_list_required)); goto errret; } l = di->di_tv.vval.v_list; if (l != NULL && list_len(l) > 0) { prop_ids = get_prop_ids_from_list(l, &prop_ids_len); if (prop_ids == NULL) goto errret; } } } if (start_lnum < 1 || start_lnum > buf->b_ml.ml_line_count || end_lnum < 1 || end_lnum < start_lnum) emsg(_(e_invalid_range)); else for (lnum = start_lnum; lnum <= end_lnum; lnum++) get_props_in_line(buf, lnum, prop_types, prop_types_len, prop_ids, prop_ids_len, rettv->vval.v_list, add_lnum); errret: VIM_CLEAR(prop_types); VIM_CLEAR(prop_ids); } /* * prop_remove({props} [, {lnum} [, {lnum_end}]]) */ void f_prop_remove(typval_T *argvars, typval_T *rettv) { linenr_T start = 1; linenr_T end = 0; linenr_T lnum; linenr_T first_changed = 0; linenr_T last_changed = 0; dict_T *dict; buf_T *buf = curbuf; int do_all; int id = -MAXCOL; int type_id = -1; // for a single "type" int *type_ids = NULL; // array, for a list of "types", allocated int num_type_ids = 0; // number of elements in "type_ids" int both; rettv->vval.v_number = 0; if (in_vim9script() && (check_for_dict_arg(argvars, 0) == FAIL || check_for_opt_number_arg(argvars, 1) == FAIL || (argvars[1].v_type != VAR_UNKNOWN && check_for_opt_number_arg(argvars, 2) == FAIL))) return; if (check_for_nonnull_dict_arg(argvars, 0) == FAIL) return; if (argvars[1].v_type != VAR_UNKNOWN) { start = tv_get_number(&argvars[1]); end = start; if (argvars[2].v_type != VAR_UNKNOWN) end = tv_get_number(&argvars[2]); if (start < 1 || end < 1) { emsg(_(e_invalid_range)); return; } } dict = argvars[0].vval.v_dict; if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL) return; if (buf->b_ml.ml_mfp == NULL) return; do_all = dict_get_bool(dict, "all", FALSE); if (dict_has_key(dict, "id")) id = dict_get_number(dict, "id"); // if a specific type was supplied "type": check that (and ignore "types". // Otherwise check against the list of "types". if (dict_has_key(dict, "type")) { char_u *name = dict_get_string(dict, "type", FALSE); proptype_T *type = lookup_prop_type(name, buf); if (type == NULL) return; type_id = type->pt_id; } if (dict_has_key(dict, "types")) { typval_T types; listitem_T *li = NULL; dict_get_tv(dict, "types", &types); if (types.v_type == VAR_LIST && types.vval.v_list->lv_len > 0) { type_ids = alloc( sizeof(int) * types.vval.v_list->lv_len ); FOR_ALL_LIST_ITEMS(types.vval.v_list, li) { proptype_T *prop_type; if (li->li_tv.v_type != VAR_STRING) continue; prop_type = lookup_prop_type(li->li_tv.vval.v_string, buf); if (!prop_type) goto cleanup_prop_remove; type_ids[num_type_ids++] = prop_type->pt_id; } } } both = dict_get_bool(dict, "both", FALSE); if (id == -MAXCOL && (type_id == -1 && num_type_ids == 0)) { emsg(_(e_need_at_least_one_of_id_or_type)); goto cleanup_prop_remove; } if (both && (id == -MAXCOL || (type_id == -1 && num_type_ids == 0))) { emsg(_(e_need_id_and_type_or_types_with_both)); goto cleanup_prop_remove; } if (type_id != -1 && num_type_ids > 0) { emsg(_(e_cannot_specify_both_type_and_types)); goto cleanup_prop_remove; } if (end == 0) end = buf->b_ml.ml_line_count; { unpacked_memline_T um = um_open(buf); for (lnum = start; lnum <= end; ++lnum) { int idx; if (lnum > buf->b_ml.ml_line_count) break; if (!um_goto_line(&um, lnum, 0)) break; for (idx = 0; idx < um.prop_count; ++idx) { textprop_T *prop = &um.props[idx]; int matches_id = 0; int matches_type = 0; if (prop->tp_flags & TP_FLAG_DELETED) continue; matches_id = prop->tp_id == id; if (num_type_ids > 0) { int idx2; for (idx2 = 0; !matches_type && idx2 < num_type_ids; ++idx2) matches_type = prop->tp_type == type_ids[idx2]; } else matches_type = prop->tp_type == type_id; if (both ? matches_id && matches_type : matches_id || matches_type) { um_delete_prop(&um, idx); if (first_changed == 0) first_changed = lnum; last_changed = lnum; ++rettv->vval.v_number; if (!do_all) break; } } } um_close(&um); } if (first_changed > 0) { changed_line_display_buf(buf); changed_lines_buf(buf, first_changed, last_changed + 1, 0); redraw_buf_later(buf, UPD_VALID); } cleanup_prop_remove: vim_free(type_ids); } /* * Common for f_prop_type_add() and f_prop_type_change(). */ static void prop_type_set(typval_T *argvars, int add) { char_u *name; buf_T *buf = NULL; dict_T *dict; dictitem_T *di; proptype_T *prop; if (in_vim9script() && (check_for_string_arg(argvars, 0) == FAIL || check_for_dict_arg(argvars, 1) == FAIL)) return; name = tv_get_string(&argvars[0]); if (*name == NUL) { semsg(_(e_invalid_argument_str), "\"\""); return; } if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) return; dict = argvars[1].vval.v_dict; prop = find_prop_type(name, buf); if (add) { hashtab_T **htp; if (prop != NULL) { semsg(_(e_property_type_str_already_defined), name); return; } prop = alloc_clear(offsetof(proptype_T, pt_name) + STRLEN(name) + 1); if (prop == NULL) return; STRCPY(prop->pt_name, name); prop->pt_id = ++proptype_id; prop->pt_flags = PT_FLAG_COMBINE; if (buf == NULL) { htp = &global_proptypes; VIM_CLEAR(global_proparray); } else { htp = &buf->b_proptypes; VIM_CLEAR(buf->b_proparray); } if (*htp == NULL) { *htp = ALLOC_ONE(hashtab_T); if (*htp == NULL) { vim_free(prop); return; } hash_init(*htp); } hash_add(*htp, PT2HIKEY(prop), "prop type"); } else { if (prop == NULL) { semsg(_(e_property_type_str_does_not_exist), name); return; } } if (dict != NULL) { di = dict_find(dict, (char_u *)"highlight", -1); if (di != NULL) { char_u *highlight; int hl_id = 0; highlight = dict_get_string(dict, "highlight", FALSE); if (highlight != NULL && *highlight != NUL) hl_id = syn_name2id(highlight); if (hl_id <= 0) { semsg(_(e_unknown_highlight_group_name_str), highlight == NULL ? (char_u *)"" : highlight); return; } prop->pt_hl_id = hl_id; } di = dict_find(dict, (char_u *)"combine", -1); if (di != NULL) { if (tv_get_bool(&di->di_tv)) prop->pt_flags |= PT_FLAG_COMBINE; else prop->pt_flags &= ~PT_FLAG_COMBINE; } di = dict_find(dict, (char_u *)"override", -1); if (di != NULL) { if (tv_get_bool(&di->di_tv)) prop->pt_flags |= PT_FLAG_OVERRIDE; else prop->pt_flags &= ~PT_FLAG_OVERRIDE; } di = dict_find(dict, (char_u *)"priority", -1); if (di != NULL) prop->pt_priority = tv_get_number(&di->di_tv); di = dict_find(dict, (char_u *)"start_incl", -1); if (di != NULL) { if (tv_get_bool(&di->di_tv)) prop->pt_flags |= PT_FLAG_INS_START_INCL; else prop->pt_flags &= ~PT_FLAG_INS_START_INCL; } di = dict_find(dict, (char_u *)"end_incl", -1); if (di != NULL) { if (tv_get_bool(&di->di_tv)) prop->pt_flags |= PT_FLAG_INS_END_INCL; else prop->pt_flags &= ~PT_FLAG_INS_END_INCL; } } } /* * prop_type_add({name}, {props}) */ void f_prop_type_add(typval_T *argvars, typval_T *rettv UNUSED) { prop_type_set(argvars, TRUE); } /* * prop_type_change({name}, {props}) */ void f_prop_type_change(typval_T *argvars, typval_T *rettv UNUSED) { prop_type_set(argvars, FALSE); } /* * prop_type_delete({name} [, {bufnr}]) */ void f_prop_type_delete(typval_T *argvars, typval_T *rettv UNUSED) { char_u *name; buf_T *buf = NULL; hashitem_T *hi; if (in_vim9script() && (check_for_string_arg(argvars, 0) == FAIL || check_for_opt_dict_arg(argvars, 1) == FAIL)) return; name = tv_get_string(&argvars[0]); if (*name == NUL) { semsg(_(e_invalid_argument_str), "\"\""); return; } if (argvars[1].v_type != VAR_UNKNOWN) { if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) return; } hi = find_prop_type_hi(name, buf); if (hi == NULL) return; hashtab_T *ht; proptype_T *prop = HI2PT(hi); if (buf == NULL) { ht = global_proptypes; VIM_CLEAR(global_proparray); } else { ht = buf->b_proptypes; VIM_CLEAR(buf->b_proparray); } hash_remove(ht, hi, "prop type delete"); vim_free(prop); // currently visible text properties will disappear redraw_all_later(UPD_CLEAR); changed_window_setting_buf(buf == NULL ? curbuf : buf); } /* * prop_type_get({name} [, {props}]) */ void f_prop_type_get(typval_T *argvars, typval_T *rettv) { char_u *name; if (in_vim9script() && (check_for_string_arg(argvars, 0) == FAIL || check_for_opt_dict_arg(argvars, 1) == FAIL)) return; name = tv_get_string(&argvars[0]); if (*name == NUL) { semsg(_(e_invalid_argument_str), "\"\""); return; } if (rettv_dict_alloc(rettv) == FAIL) return; proptype_T *prop = NULL; buf_T *buf = NULL; if (argvars[1].v_type != VAR_UNKNOWN) { if (get_bufnr_from_arg(&argvars[1], &buf) == FAIL) return; } prop = find_prop_type(name, buf); if (prop == NULL) return; dict_T *d = rettv->vval.v_dict; if (prop->pt_hl_id > 0) dict_add_string(d, "highlight", syn_id2name(prop->pt_hl_id)); dict_add_number(d, "priority", prop->pt_priority); dict_add_number(d, "combine", (prop->pt_flags & PT_FLAG_COMBINE) ? 1 : 0); dict_add_number(d, "start_incl", (prop->pt_flags & PT_FLAG_INS_START_INCL) ? 1 : 0); dict_add_number(d, "end_incl", (prop->pt_flags & PT_FLAG_INS_END_INCL) ? 1 : 0); if (buf != NULL) dict_add_number(d, "bufnr", buf->b_fnum); } static void list_types(hashtab_T *ht, list_T *l) { long todo; hashitem_T *hi; todo = (long)ht->ht_used; FOR_ALL_HASHTAB_ITEMS(ht, hi, todo) { if (!HASHITEM_EMPTY(hi)) { proptype_T *prop = HI2PT(hi); list_append_string(l, prop->pt_name, -1); --todo; } } } /* * prop_type_list([{bufnr}]) */ void f_prop_type_list(typval_T *argvars, typval_T *rettv) { buf_T *buf = NULL; if (rettv_list_alloc(rettv) == FAIL) return; if (in_vim9script() && check_for_opt_dict_arg(argvars, 0) == FAIL) return; if (argvars[0].v_type != VAR_UNKNOWN) { if (get_bufnr_from_arg(&argvars[0], &buf) == FAIL) return; } if (buf == NULL) { if (global_proptypes != NULL) list_types(global_proptypes, rettv->vval.v_list); } else if (buf->b_proptypes != NULL) list_types(buf->b_proptypes, rettv->vval.v_list); } /* * Free all property types in "ht". */ static void clear_ht_prop_types(hashtab_T *ht) { long todo; hashitem_T *hi; if (ht == NULL) return; todo = (long)ht->ht_used; FOR_ALL_HASHTAB_ITEMS(ht, hi, todo) { if (!HASHITEM_EMPTY(hi)) { proptype_T *prop = HI2PT(hi); vim_free(prop); --todo; } } hash_clear(ht); vim_free(ht); } #if defined(EXITFREE) /* * Free all global property types. */ void clear_global_prop_types(void) { clear_ht_prop_types(global_proptypes); global_proptypes = NULL; VIM_CLEAR(global_proparray); } #endif /* * Free all property types for "buf". */ void clear_buf_prop_types(buf_T *buf) { clear_ht_prop_types(buf->b_proptypes); buf->b_proptypes = NULL; VIM_CLEAR(buf->b_proparray); } // Struct used to return two values from adjust_prop(). typedef struct { int dirty; // if the property was changed int can_drop; // whether after this change, the prop may be removed } adjustres_T; /* * Adjust the property for "added" bytes (can be negative) inserted at "col". * * Note that "col" is zero-based, while tp_col is one-based. * Only for the current buffer. * "flags" can have: * APC_SUBSTITUTE: Text is replaced, not inserted. * APC_INDENT: Text is inserted before virtual text prop */ static adjustres_T adjust_prop( textprop_T *prop, colnr_T col, int added, int flags) { proptype_T *pt; int start_incl; int end_incl; int droppable; adjustres_T res = {TRUE, FALSE}; // prop after end of the line doesn't move if (prop->tp_col == MAXCOL) { res.dirty = FALSE; return res; } pt = text_prop_type_by_id(curbuf, prop->tp_type); start_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_START_INCL)) || (flags & APC_SUBSTITUTE) || (prop->tp_flags & TP_FLAG_CONT_PREV); if (prop->tp_id < 0 && (flags & APC_INDENT)) // when inserting indent just before a character with virtual text // shift the text property start_incl = FALSE; end_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_END_INCL)) || (prop->tp_flags & TP_FLAG_CONT_NEXT); // do not drop zero-width props if they later can increase in size droppable = !(start_incl || end_incl); if (added > 0) { if (col + 1 <= prop->tp_col - (start_incl || (prop->tp_len == 0 && end_incl))) // Change is entirely before the text property: Only shift prop->tp_col += added; else if (col + 1 < prop->tp_col + prop->tp_len + end_incl && prop->tp_id >= 0) // don't change length for virtual text // Insertion was inside text property prop->tp_len += added; } else if (prop->tp_col > col + 1) { if (prop->tp_col + added < col + 1) { if (prop->tp_id < 0) { // Inline virtual text swallowed by the deletion. res.can_drop = TRUE; } else { prop->tp_len += (prop->tp_col - 1 - col) + added; prop->tp_col = col + 1; if (prop->tp_len <= 0) { prop->tp_len = 0; // Multiline properties with no text left should // also be dropped. res.can_drop = droppable || (prop->tp_flags & (TP_FLAG_CONT_PREV | TP_FLAG_CONT_NEXT)); } } } else prop->tp_col += added; } else if (prop->tp_len > 0 && prop->tp_col + prop->tp_len > col && prop->tp_id >= 0) // don't change length for virtual text { if (added <= -MAXCOL) // Delete everything from col to end of line. prop->tp_len = col - (prop->tp_col - 1); else { int after = col - added - (prop->tp_col - 1 + prop->tp_len); prop->tp_len += after > 0 ? added + after : added; } // A multiline property with no text left on this line // should also be dropped. When TP_FLAG_CONT_NEXT is set, // tp_len == 1 means only the newline remains. if (prop->tp_len <= 0 || ((prop->tp_flags & TP_FLAG_CONT_NEXT) && prop->tp_len <= 1)) res.can_drop = droppable || (prop->tp_flags & (TP_FLAG_CONT_PREV | TP_FLAG_CONT_NEXT)); } else res.dirty = FALSE; return res; } /* * Adjust the columns of text properties in line "lnum" after position "col" to * shift by "bytes_added" (can be negative). * Note that "col" is zero-based, while tp_col is one-based. * Only for the current buffer. * "flags" can have: * APC_SAVE_FOR_UNDO: Call u_savesub() before making changes to the line. * APC_SUBSTITUTE: Text is replaced, not inserted. * APC_INDENT: Text is inserted before virtual text prop * Caller is expected to check b_has_textprop and "bytes_added" being non-zero. * Returns TRUE when props were changed. */ int adjust_prop_columns( linenr_T lnum, colnr_T col, int bytes_added, int flags) { unpacked_memline_T um; int dirty = FALSE; int ri; if (text_prop_frozen > 0) return FALSE; um = um_open_at(curbuf, lnum, 0); if (um.buf == NULL || um.prop_count == 0) { um_abort(&um); return FALSE; } for (ri = 0; ri < um.prop_count; ++ri) { textprop_T *prop = &um.props[ri]; adjustres_T res; res = adjust_prop(prop, col, bytes_added, flags); if (res.dirty) { if (!dirty) { // Detach before u_savesub() so that all text and vtext // pointers are allocated copies, immune to memline // changes caused by u_savesub(). if (!um.detached && !um_detach(&um)) { um_abort(&um); return FALSE; } // Re-get prop pointer after detach (array may move). prop = &um.props[ri]; if ((flags & APC_SAVE_FOR_UNDO) && u_savesub(lnum) == FAIL) { um_abort(&um); return FALSE; } } dirty = TRUE; } if (res.can_drop) { um_delete_prop(&um, ri); continue; } } if (dirty) um_close(&um); else um_abort(&um); return dirty; } /* * Adjust text properties for a line that was split in two. * "lnum_props" is the line that has the properties from before the split. * "lnum_top" is the top line. * "kept" is the number of bytes kept in the first line, while * "deleted" is the number of bytes deleted. * "at_eol" is true if the split is after the end of the line. */ void adjust_props_for_split( linenr_T lnum_props, linenr_T lnum_top, int kept, int deleted, int at_eol) { char_u *props; int count; garray_T prevprop; garray_T nextprop; int skipped = kept + deleted; if (!curbuf->b_has_textprop) return; // Get the text properties from "lnum_props". count = get_text_props(curbuf, lnum_props, &props, FALSE); ga_init2(&prevprop, sizeof(textprop_T), 10); ga_init2(&nextprop, sizeof(textprop_T), 10); // count_ptr points to the prop_count field in the memline. char_u *count_ptr = props - PROP_COUNT_SIZE; // Keep the relevant ones in the first line, reducing the length if needed. // Copy the ones that include the split to the second line. // Move the ones after the split to the second line. for (int i = 0; i < count; ++i) { textprop_T prop; proptype_T *pt; int start_incl, end_incl; int cont_prev, cont_next; int prop_col; // copy the prop to an aligned structure mch_memmove(&prop, props + i * sizeof(textprop_T), sizeof(textprop_T)); // Convert offset to pointer and copy for virtual text props. // Must copy because set_text_props() may invalidate memline data. if (prop.tp_id < 0 && prop.u.tp_text_offset > 0) { prop.u.tp_text = vim_strsave(count_ptr + prop.u.tp_text_offset); prop.tp_flags |= TP_FLAG_VTEXT_PTR; } else prop.u.tp_text = NULL; pt = text_prop_type_by_id(curbuf, prop.tp_type); start_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_START_INCL)); end_incl = (pt != NULL && (pt->pt_flags & PT_FLAG_INS_END_INCL)); // a text prop "above" behaves like it is on the first text column prop_col = (prop.tp_flags & TP_FLAG_ALIGN_ABOVE) ? 1 : prop.tp_col; if (prop_col == MAXCOL) { cont_prev = at_eol; cont_next = !at_eol; } else { // Floating virtual text (tp_col == MAXCOL) should not use // tp_len for continuation since tp_len holds the vtext // string length. int is_floating_vtext = (prop.tp_id < 0 && prop.tp_col == MAXCOL); cont_prev = prop_col + !start_incl <= kept; if (is_floating_vtext) cont_next = FALSE; else cont_next = skipped <= prop_col + prop.tp_len - !end_incl; } // when a prop has text it is never copied if (prop.tp_id < 0 && cont_next) cont_prev = FALSE; if (cont_prev && ga_grow(&prevprop, 1) == OK) { textprop_T *p = ((textprop_T *)prevprop.ga_data) + prevprop.ga_len; int is_vtext = (prop.tp_id < 0); *p = prop; ++prevprop.ga_len; if (!is_vtext && p->tp_col + p->tp_len >= kept) p->tp_len = kept - p->tp_col; if (cont_next) p->tp_flags |= TP_FLAG_CONT_NEXT; } // Only add the property to the next line if the length is bigger than // zero. if (cont_next && ga_grow(&nextprop, 1) == OK) { textprop_T *p = ((textprop_T *)nextprop.ga_data) + nextprop.ga_len; int is_vtext = (prop.tp_id < 0); *p = prop; ++nextprop.ga_len; if (p->tp_col != MAXCOL) { if (p->tp_col > skipped) p->tp_col -= skipped - 1; else { if (!is_vtext) p->tp_len -= skipped - p->tp_col; p->tp_col = 1; } } if (cont_prev) p->tp_flags |= TP_FLAG_CONT_PREV; } } set_text_props(lnum_top, (textprop_T *)prevprop.ga_data, prevprop.ga_len); ga_clear(&prevprop); set_text_props(lnum_top + 1, (textprop_T *)nextprop.ga_data, nextprop.ga_len); ga_clear(&nextprop); } /* * Prepend properties of joined line "lnum" to the unpacked memline "um". * Properties are added in reverse order; caller should call * um_reverse_props() after all lines are processed. */ void prepend_joined_props( unpacked_memline_T *um, linenr_T lnum, int last_line, long col, int removed) { unpacked_memline_T r_um; if (um->buf == NULL) return; r_um = um_open_at(um->buf, lnum, 0); if (r_um.buf == NULL) return; if (!um_detach(&r_um)) { um_abort(&r_um); return; } for (int i = r_um.prop_count - 1; i >= 0; --i) { textprop_T *prop = &r_um.props[i]; int end; if (prop->tp_col == MAXCOL && !last_line) continue; // drop floating text for non-last lines end = !(prop->tp_flags & TP_FLAG_CONT_NEXT); adjust_prop(prop, 0, -removed, 0); adjust_prop(prop, -1, col, 0); if (last_line || end) { um_add_prop(um, prop); prop->u.tp_text = NULL; // ownership transferred } else { // Search for continuing prop in um. bool found = false; for (int j = 0; j < um->prop_count; ++j) { textprop_T *op = &um->props[j]; if ((op->tp_flags & TP_FLAG_CONT_PREV) && op->tp_id == prop->tp_id && op->tp_type == prop->tp_type) { found = true; op->tp_len += op->tp_col - prop->tp_col; op->tp_col = prop->tp_col; op->tp_flags = prop->tp_flags; break; } } if (!found) internal_error("text property above joined line not found"); } } um_abort(&r_um); } #endif // FEAT_PROP_POPUP