diff options
author | zeertzjq <zeertzjq@outlook.com> | 2025-03-27 08:02:45 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2025-03-27 08:02:45 +0800 |
commit | ce590e207720ec53a4592882840725bf8540c7d5 (patch) | |
tree | 6dc54fcd969c977ad35c02fc6ad3c5ce91f74e27 /src | |
parent | 9acb52c8f386ea0a026ba4e314e1294da66f8e79 (diff) | |
parent | 0af780e8df849f6d393873124afaa31681ab43ec (diff) | |
download | rneovim-ce590e207720ec53a4592882840725bf8540c7d5.tar.gz rneovim-ce590e207720ec53a4592882840725bf8540c7d5.tar.bz2 rneovim-ce590e207720ec53a4592882840725bf8540c7d5.zip |
Merge pull request #30189 from zeertzjq/vim-9.1.0598
vim-patch: 'completefuzzycollect' option
Diffstat (limited to 'src')
-rw-r--r-- | src/nvim/insexpand.c | 401 | ||||
-rw-r--r-- | src/nvim/option_vars.h | 4 | ||||
-rw-r--r-- | src/nvim/options.lua | 44 | ||||
-rw-r--r-- | src/nvim/optionstr.c | 1 | ||||
-rw-r--r-- | src/nvim/search.c | 175 | ||||
-rw-r--r-- | src/nvim/spell.c | 2 |
6 files changed, 571 insertions, 56 deletions
diff --git a/src/nvim/insexpand.c b/src/nvim/insexpand.c index 1be4f601bc..422eb409fe 100644 --- a/src/nvim/insexpand.c +++ b/src/nvim/insexpand.c @@ -215,6 +215,13 @@ static compl_T *compl_curr_match = NULL; static compl_T *compl_shown_match = NULL; static compl_T *compl_old_match = NULL; +/// list used to store the compl_T which have the max score +/// used for completefuzzycollect +static compl_T **compl_best_matches = NULL; +static int compl_num_bests = 0; +/// inserted a longest when completefuzzycollect enabled +static bool compl_cfc_longest_ins = false; + /// After using a cursor key <Enter> selects a match in the popup menu, /// otherwise it inserts a line break. static bool compl_enter_selects = false; @@ -284,6 +291,8 @@ static size_t spell_bad_len = 0; // length of located bad word static int compl_selected_item = -1; +static int *compl_fuzzy_scores; + // "compl_match_array" points the currently displayed list of entries in the // popup menu. It is NULL when there is no popup menu. static pumitem_T *compl_match_array = NULL; @@ -730,7 +739,7 @@ static char *ins_compl_infercase_gettext(const char *str, int char_len, int comp /// /// @param[in] cont_s_ipos next ^X<> will set initial_pos int ins_compl_add_infercase(char *str_arg, int len, bool icase, char *fname, Direction dir, - bool cont_s_ipos) + bool cont_s_ipos, int score) FUNC_ATTR_NONNULL_ARG(1) { char *str = str_arg; @@ -775,11 +784,25 @@ int ins_compl_add_infercase(char *str_arg, int len, bool icase, char *fname, Dir flags |= CP_ICASE; } - int res = ins_compl_add(str, len, fname, NULL, false, NULL, dir, flags, false, NULL); + int res = ins_compl_add(str, len, fname, NULL, false, NULL, dir, flags, false, NULL, score); xfree(tofree); return res; } +/// Check if ctrl_x_mode has been configured in 'completefuzzycollect' +static bool cfc_has_mode(void) +{ + if (ctrl_x_mode_normal() || ctrl_x_mode_dictionary()) { + return (cfc_flags & kOptCfcFlagKeyword) != 0; + } else if (ctrl_x_mode_files()) { + return (cfc_flags & kOptCfcFlagFiles) != 0; + } else if (ctrl_x_mode_whole_line()) { + return (cfc_flags & kOptCfcFlagWholeLine) != 0; + } else { + return false; + } +} + /// free cptext static inline void free_cptext(char *const *const cptext) { @@ -816,12 +839,13 @@ static inline void free_cptext(char *const *const cptext) /// returned in case of error. static int ins_compl_add(char *const str, int len, char *const fname, char *const *const cptext, const bool cptext_allocated, typval_T *user_data, const Direction cdir, - int flags_arg, const bool adup, const int *user_hl) + int flags_arg, const bool adup, const int *user_hl, const int score) FUNC_ATTR_NONNULL_ARG(1) { compl_T *match; const Direction dir = (cdir == kDirectionNotSet ? compl_direction : cdir); int flags = flags_arg; + bool inserted = false; if (flags & CP_FAST) { fast_breakcheck(); @@ -862,6 +886,7 @@ static int ins_compl_add(char *const str, int len, char *const fname, char *cons match = xcalloc(1, sizeof(compl_T)); match->cp_number = flags & CP_ORIGINAL_TEXT ? 0 : -1; match->cp_str = cbuf_to_string(str, (size_t)len); + match->cp_score = score; // match-fname is: // - compl_curr_match->cp_fname if it is a string equal to fname. @@ -905,6 +930,33 @@ static int ins_compl_add(char *const str, int len, char *const fname, char *cons // current match in the list of matches . if (compl_first_match == NULL) { match->cp_next = match->cp_prev = NULL; + } else if (cfc_has_mode() && score > 0 && compl_get_longest) { + compl_T *current = compl_first_match->cp_next; + compl_T *prev = compl_first_match; + inserted = false; + // The direction is ignored when using longest and + // completefuzzycollect, because matches are inserted + // and sorted by score. + while (current != NULL && current != compl_first_match) { + if (current->cp_score < score) { + match->cp_next = current; + match->cp_prev = current->cp_prev; + if (current->cp_prev) { + current->cp_prev->cp_next = match; + } + current->cp_prev = match; + inserted = true; + break; + } + prev = current; + current = current->cp_next; + } + if (!inserted) { + prev->cp_next = match; + match->cp_prev = prev; + match->cp_next = compl_first_match; + compl_first_match->cp_prev = match; + } } else if (dir == FORWARD) { match->cp_next = compl_curr_match->cp_next; match->cp_prev = compl_curr_match; @@ -923,7 +975,7 @@ static int ins_compl_add(char *const str, int len, char *const fname, char *cons compl_curr_match = match; // Find the longest common string if still doing that. - if (compl_get_longest && (flags & CP_ORIGINAL_TEXT) == 0) { + if (compl_get_longest && (flags & CP_ORIGINAL_TEXT) == 0 && !cfc_has_mode()) { ins_compl_longest_match(match); } @@ -1011,9 +1063,7 @@ static void ins_compl_longest_match(compl_T *match) compl_leader = copy_string(match->cp_str, NULL); bool had_match = (curwin->w_cursor.col > compl_col); - ins_compl_delete(false); - ins_compl_insert_bytes(compl_leader.data + get_compl_len(), -1); - ins_redraw(false); + ins_compl_longest_insert(compl_leader.data); // When the match isn't there (to avoid matching itself) remove it // again after redrawing. @@ -1047,9 +1097,7 @@ static void ins_compl_longest_match(compl_T *match) compl_leader.size = (size_t)(p - compl_leader.data); bool had_match = (curwin->w_cursor.col > compl_col); - ins_compl_delete(false); - ins_compl_insert_bytes(compl_leader.data + get_compl_len(), -1); - ins_redraw(false); + ins_compl_longest_insert(compl_leader.data); // When the match isn't there (to avoid matching itself) remove it // again after redrawing. @@ -1070,7 +1118,7 @@ static void ins_compl_add_matches(int num_matches, char **matches, int icase) for (int i = 0; i < num_matches && add_r != FAIL; i++) { add_r = ins_compl_add(matches[i], -1, NULL, NULL, false, NULL, dir, - CP_FAST | (icase ? CP_ICASE : 0), false, NULL); + CP_FAST | (icase ? CP_ICASE : 0), false, NULL, 0); if (add_r == OK) { // If dir was BACKWARD then honor it just once. dir = FORWARD; @@ -1236,6 +1284,7 @@ static int ins_compl_build_pum(void) bool compl_no_select = (cur_cot_flags & kOptCotFlagNoselect) != 0; bool fuzzy_filter = (cur_cot_flags & kOptCotFlagFuzzy) != 0; bool fuzzy_sort = fuzzy_filter && !(cur_cot_flags & kOptCotFlagNosort); + compl_T *match_head = NULL, *match_tail = NULL; // If the current match is the original text don't find the first @@ -1552,7 +1601,7 @@ static void ins_compl_dictionaries(char *dict_start, char *pat, int flags, bool spell_dump_compl(ptr, regmatch.rm_ic, &dir, 0); } else if (count > 0) { // avoid warning for using "files" uninit ins_compl_files(count, files, thesaurus, flags, - ®match, buf, &dir); + (cfc_has_mode() ? NULL : ®match), buf, &dir); if (flags != DICT_EXACT) { FreeWild(count, files); } @@ -1603,7 +1652,7 @@ static int thesaurus_add_words_in_line(char *fname, char **buf_arg, int dir, con // Add the word. Skip the regexp match. if (wstart != skip_word) { status = ins_compl_add_infercase(wstart, (int)(ptr - wstart), p_ic, - fname, dir, false); + fname, dir, false, 0); if (status == FAIL) { break; } @@ -1620,6 +1669,11 @@ static void ins_compl_files(int count, char **files, bool thesaurus, int flags, regmatch_T *regmatch, char *buf, Direction *dir) FUNC_ATTR_NONNULL_ARG(2, 7) { + bool in_fuzzy_collect = cfc_has_mode(); + + char *leader = in_fuzzy_collect ? ins_compl_leader() : NULL; + int leader_len = in_fuzzy_collect ? (int)ins_compl_leader_len() : 0; + for (int i = 0; i < count && !got_int && !compl_interrupted; i++) { FILE *fp = os_fopen(files[i], "r"); // open dictionary file if (flags != DICT_EXACT && !shortmess(SHM_COMPLETIONSCAN)) { @@ -1638,27 +1692,49 @@ static void ins_compl_files(int count, char **files, bool thesaurus, int flags, // Check each line for a match. while (!got_int && !compl_interrupted && !vim_fgets(buf, LSIZE, fp)) { char *ptr = buf; - while (vim_regexec(regmatch, buf, (colnr_T)(ptr - buf))) { - ptr = regmatch->startp[0]; - ptr = ctrl_x_mode_line_or_eval() ? find_line_end(ptr) : find_word_end(ptr); - int add_r = ins_compl_add_infercase(regmatch->startp[0], - (int)(ptr - regmatch->startp[0]), - p_ic, files[i], *dir, false); - if (thesaurus) { - // For a thesaurus, add all the words in the line - ptr = buf; - add_r = thesaurus_add_words_in_line(files[i], &ptr, *dir, regmatch->startp[0]); - } - if (add_r == OK) { - // if dir was BACKWARD then honor it just once - *dir = FORWARD; - } else if (add_r == FAIL) { - break; + if (regmatch != NULL) { + while (vim_regexec(regmatch, buf, (colnr_T)(ptr - buf))) { + ptr = regmatch->startp[0]; + ptr = ctrl_x_mode_line_or_eval() ? find_line_end(ptr) : find_word_end(ptr); + int add_r = ins_compl_add_infercase(regmatch->startp[0], + (int)(ptr - regmatch->startp[0]), + p_ic, files[i], *dir, false, 0); + if (thesaurus) { + // For a thesaurus, add all the words in the line + ptr = buf; + add_r = thesaurus_add_words_in_line(files[i], &ptr, *dir, regmatch->startp[0]); + } + if (add_r == OK) { + // if dir was BACKWARD then honor it just once + *dir = FORWARD; + } else if (add_r == FAIL) { + break; + } + // avoid expensive call to vim_regexec() when at end + // of line + if (*ptr == '\n' || got_int) { + break; + } } - // avoid expensive call to vim_regexec() when at end - // of line - if (*ptr == '\n' || got_int) { - break; + } else if (in_fuzzy_collect && leader_len > 0) { + char *line_end = find_line_end(ptr); + while (ptr < line_end) { + int score = 0; + int len = 0; + if (fuzzy_match_str_in_line(&ptr, leader, &len, NULL, &score)) { + char *end_ptr = ctrl_x_mode_line_or_eval() ? find_line_end(ptr) : find_word_end(ptr); + int add_r = ins_compl_add_infercase(ptr, (int)(end_ptr - ptr), + p_ic, files[i], *dir, false, score); + if (add_r == FAIL) { + break; + } + ptr = end_ptr; // start from next word + if (compl_get_longest && ctrl_x_mode_normal() + && compl_first_match->cp_next + && score == compl_first_match->cp_next->cp_score) { + compl_num_bests++; + } + } } } line_breakcheck(); @@ -1699,7 +1775,7 @@ char *find_word_end(char *ptr) /// Find the end of the line, omitting CR and NL at the end. /// /// @return a pointer to just after the line. -static char *find_line_end(char *ptr) +char *find_line_end(char *ptr) { char *s = ptr + strlen(ptr); while (s > ptr && (s[-1] == CAR || s[-1] == NL)) { @@ -1744,6 +1820,7 @@ void ins_compl_clear(void) { compl_cont_status = 0; compl_started = false; + compl_cfc_longest_ins = false; compl_matches = 0; compl_selected_item = -1; compl_ins_end_col = 0; @@ -2742,7 +2819,7 @@ static int ins_compl_add_tv(typval_T *const tv, const Direction dir, bool fast) return FAIL; } int status = ins_compl_add((char *)word, -1, NULL, cptext, true, - &user_data, dir, flags, dup, user_hl); + &user_data, dir, flags, dup, user_hl, 0); if (status != OK) { tv_clear(&user_data); } @@ -2838,7 +2915,7 @@ static void set_completion(colnr_T startcol, list_T *list) } if (ins_compl_add(compl_orig_text.data, (int)compl_orig_text.size, NULL, NULL, false, NULL, 0, - flags | CP_FAST, false, NULL) != OK) { + flags | CP_FAST, false, NULL, 0) != OK) { return; } @@ -3141,7 +3218,7 @@ enum { /// the "st->e_cpt" option value and process the next matching source. /// INS_COMPL_CPT_END if all the values in "st->e_cpt" are processed. static int process_next_cpt_value(ins_compl_next_state_T *st, int *compl_type_arg, - pos_T *start_match_pos) + pos_T *start_match_pos, bool fuzzy_collect) { int compl_type = -1; int status = INS_COMPL_CPT_OK; @@ -3157,7 +3234,7 @@ static int process_next_cpt_value(ins_compl_next_state_T *st, int *compl_type_ar st->first_match_pos = *start_match_pos; // Move the cursor back one character so that ^N can match the // word immediately after the cursor. - if (ctrl_x_mode_normal() && dec(&st->first_match_pos) < 0) { + if (ctrl_x_mode_normal() && (!fuzzy_collect && dec(&st->first_match_pos) < 0)) { // Move the cursor to after the last character in the // buffer, so that word at start of buffer is found // correctly. @@ -3297,11 +3374,157 @@ static void get_next_tag_completion(void) p_ic = save_p_ic; } +/// Compare function for qsort +static int compare_scores(const void *a, const void *b) +{ + int idx_a = *(const int *)a; + int idx_b = *(const int *)b; + int score_a = compl_fuzzy_scores[idx_a]; + int score_b = compl_fuzzy_scores[idx_b]; + return score_a == score_b ? (idx_a == idx_b ? 0 : (idx_a < idx_b ? -1 : 1)) + : (score_a > score_b ? -1 : 1); +} + +/// insert prefix with redraw +static void ins_compl_longest_insert(char *prefix) +{ + ins_compl_delete(false); + ins_compl_insert_bytes(prefix + get_compl_len(), -1); + ins_redraw(false); +} + +/// Calculate the longest common prefix among the best fuzzy matches +/// stored in compl_best_matches, and insert it as the longest. +static void fuzzy_longest_match(void) +{ + if (compl_num_bests == 0) { + return; + } + + compl_T *nn_compl = compl_first_match->cp_next->cp_next; + bool more_candidates = nn_compl && nn_compl != compl_first_match; + + compl_T *compl = ctrl_x_mode_whole_line() ? compl_first_match + : compl_first_match->cp_next; + if (compl_num_bests == 1) { + // no more candidates insert the match str + if (!more_candidates) { + ins_compl_longest_insert(compl->cp_str.data); + compl_num_bests = 0; + } + compl_num_bests = 0; + return; + } + + compl_best_matches = (compl_T **)xmalloc((size_t)compl_num_bests * sizeof(compl_T *)); + + for (int i = 0; compl != NULL && i < compl_num_bests; i++) { + compl_best_matches[i] = compl; + compl = compl->cp_next; + } + + char *prefix = compl_best_matches[0]->cp_str.data; + int prefix_len = (int)compl_best_matches[0]->cp_str.size; + + for (int i = 1; i < compl_num_bests; i++) { + char *match_str = compl_best_matches[i]->cp_str.data; + char *prefix_ptr = prefix; + char *match_ptr = match_str; + int j = 0; + + while (j < prefix_len && *match_ptr != NUL && *prefix_ptr != NUL) { + if (strncmp(prefix_ptr, match_ptr, (size_t)utfc_ptr2len(prefix_ptr)) != 0) { + break; + } + + MB_PTR_ADV(prefix_ptr); + MB_PTR_ADV(match_ptr); + j++; + } + + if (j > 0) { + prefix_len = j; + } + } + + char *leader = ins_compl_leader(); + size_t leader_len = ins_compl_leader_len(); + + // skip non-consecutive prefixes + if (leader_len > 0 && strncmp(prefix, leader, leader_len) != 0) { + goto end; + } + + prefix = xmemdupz(prefix, (size_t)prefix_len); + ins_compl_longest_insert(prefix); + compl_cfc_longest_ins = true; + xfree(prefix); + +end: + xfree(compl_best_matches); + compl_best_matches = NULL; + compl_num_bests = 0; +} + /// Get the next set of filename matching "compl_pattern". static void get_next_filename_completion(void) { char **matches; int num_matches; + char *leader = ins_compl_leader(); + size_t leader_len = ins_compl_leader_len(); + bool in_fuzzy_collect = (cfc_has_mode() && leader_len > 0); + bool need_collect_bests = in_fuzzy_collect && compl_get_longest; + int max_score = 0; + Direction dir = compl_direction; + +#ifdef BACKSLASH_IN_FILENAME + char pathsep = (curbuf->b_p_csl[0] == 's') + ? '/' : (curbuf->b_p_csl[0] == 'b') ? '\\' : PATHSEP; +#else + char pathsep = PATHSEP; +#endif + + if (in_fuzzy_collect) { +#ifdef BACKSLASH_IN_FILENAME + if (curbuf->b_p_csl[0] == 's') { + for (size_t i = 0; i < leader_len; i++) { + if (leader[i] == '\\') { + leader[i] = '/'; + } + } + } else if (curbuf->b_p_csl[0] == 'b') { + for (size_t i = 0; i < leader_len; i++) { + if (leader[i] == '/') { + leader[i] = '\\'; + } + } + } +#endif + char *last_sep = strrchr(leader, pathsep); + if (last_sep == NULL) { + // No path separator or separator is the last character, + // fuzzy match the whole leader + API_CLEAR_STRING(compl_pattern); + compl_pattern = cbuf_to_string("*", 1); + } else if (*(last_sep + 1) == NUL) { + in_fuzzy_collect = false; + } else { + // Split leader into path and file parts + size_t path_len = (size_t)(last_sep - leader) + 1; + char *path_with_wildcard = xmalloc(path_len + 2); + vim_snprintf(path_with_wildcard, path_len + 2, "%*.*s*", + (int)path_len, (int)path_len, leader); + API_CLEAR_STRING(compl_pattern); + compl_pattern.data = path_with_wildcard; + compl_pattern.size = path_len + 1; + + // Move leader to the file part + leader = last_sep + 1; + leader_len -= path_len; + } + } + if (expand_wildcards(1, &compl_pattern.data, &num_matches, &matches, EW_FILE|EW_DIR|EW_ADDSLASH|EW_SILENT) != OK) { return; @@ -3324,7 +3547,61 @@ static void get_next_filename_completion(void) } } #endif - ins_compl_add_matches(num_matches, matches, p_fic || p_wic); + + if (in_fuzzy_collect) { + garray_T fuzzy_indices; + ga_init(&fuzzy_indices, sizeof(int), 10); + compl_fuzzy_scores = (int *)xmalloc(sizeof(int) * (size_t)num_matches); + + for (int i = 0; i < num_matches; i++) { + char *ptr = matches[i]; + int score = fuzzy_match_str(ptr, leader); + if (score > 0) { + GA_APPEND(int, &fuzzy_indices, i); + compl_fuzzy_scores[i] = score; + } + } + + // prevent qsort from deref NULL pointer + if (fuzzy_indices.ga_len > 0) { + int *fuzzy_indices_data = (int *)fuzzy_indices.ga_data; + qsort(fuzzy_indices_data, (size_t)fuzzy_indices.ga_len, sizeof(int), compare_scores); + + for (int i = 0; i < fuzzy_indices.ga_len; i++) { + char *match = matches[fuzzy_indices_data[i]]; + int current_score = compl_fuzzy_scores[fuzzy_indices_data[i]]; + if (ins_compl_add(match, -1, NULL, NULL, false, NULL, dir, + CP_FAST | ((p_fic || p_wic) ? CP_ICASE : 0), + false, NULL, current_score) == OK) { + dir = FORWARD; + } + + if (need_collect_bests) { + if (i == 0 || current_score == max_score) { + compl_num_bests++; + max_score = current_score; + } + } + } + + FreeWild(num_matches, matches); + } else if (leader_len > 0) { + FreeWild(num_matches, matches); + num_matches = 0; + } + + xfree(compl_fuzzy_scores); + ga_clear(&fuzzy_indices); + + if (compl_num_bests > 0 && compl_get_longest) { + fuzzy_longest_match(); + } + return; + } + + if (num_matches > 0) { + ins_compl_add_matches(num_matches, matches, p_fic || p_wic); + } } /// Get the next set of command-line completions matching "compl_pattern". @@ -3447,6 +3724,12 @@ static char *ins_compl_get_next_word_or_line(buf_T *ins_buf, pos_T *cur_match_po /// @return OK if a new next match is found, otherwise FAIL. static int get_next_default_completion(ins_compl_next_state_T *st, pos_T *start_pos) { + char *ptr = NULL; + int len = 0; + bool in_collect = (cfc_has_mode() && compl_length > 0); + char *leader = ins_compl_leader(); + int score = 0; + // If 'infercase' is set, don't use 'smartcase' here const int save_p_scs = p_scs; assert(st->ins_buf); @@ -3470,9 +3753,15 @@ static int get_next_default_completion(ins_compl_next_state_T *st, pos_T *start_ bool cont_s_ipos = false; msg_silent++; // Don't want messages for wrapscan. - // ctrl_x_mode_line_or_eval() || word-wise search that - // has added a word that was at the beginning of the line. - if (ctrl_x_mode_line_or_eval() || (compl_cont_status & CONT_SOL)) { + + if (in_collect) { + found_new_match = search_for_fuzzy_match(st->ins_buf, + st->cur_match_pos, leader, compl_direction, + start_pos, &len, &ptr, &score); + // ctrl_x_mode_line_or_eval() || word-wise search that + // has added a word that was at the beginning of the line. + } else if (ctrl_x_mode_whole_line() || ctrl_x_mode_eval() + || (compl_cont_status & CONT_SOL)) { found_new_match = search_for_exact_line(st->ins_buf, st->cur_match_pos, compl_direction, compl_pattern.data); } else { @@ -3521,16 +3810,22 @@ static int get_next_default_completion(ins_compl_next_state_T *st, pos_T *start_ && start_pos->col == st->cur_match_pos->col) { continue; } - int len; - char *ptr = ins_compl_get_next_word_or_line(st->ins_buf, st->cur_match_pos, - &len, &cont_s_ipos); + + if (!in_collect) { + ptr = ins_compl_get_next_word_or_line(st->ins_buf, st->cur_match_pos, + &len, &cont_s_ipos); + } if (ptr == NULL || (ins_compl_has_preinsert() && strcmp(ptr, compl_pattern.data) == 0)) { continue; } + if (ins_compl_add_infercase(ptr, len, p_ic, st->ins_buf == curbuf ? NULL : st->ins_buf->b_sfname, - 0, cont_s_ipos) != NOTDONE) { + 0, cont_s_ipos, score) != NOTDONE) { + if (in_collect && score == compl_first_match->cp_next->cp_score) { + compl_num_bests++; + } found_new_match = OK; break; } @@ -3609,7 +3904,7 @@ static void get_next_bufname_token(void) char *tail = path_tail(b->b_sfname); if (strncmp(tail, compl_orig_text.data, compl_orig_text.size) == 0) { ins_compl_add(tail, (int)strlen(tail), NULL, NULL, false, NULL, 0, - p_ic ? CP_ICASE : 0, false, NULL); + p_ic ? CP_ICASE : 0, false, NULL, 0); } } } @@ -3664,7 +3959,7 @@ static int ins_compl_get_exp(pos_T *ini) // entries from 'complete' that look in loaded buffers. if ((ctrl_x_mode_normal() || ctrl_x_mode_line_or_eval()) && (!compl_started || st.found_all)) { - int status = process_next_cpt_value(&st, &type, ini); + int status = process_next_cpt_value(&st, &type, ini, cfc_has_mode()); if (status == INS_COMPL_CPT_END) { break; } @@ -3722,6 +4017,10 @@ static int ins_compl_get_exp(pos_T *ini) i = ins_compl_make_cyclic(); } + if (cfc_has_mode() && compl_get_longest && compl_num_bests > 0) { + fuzzy_longest_match(); + } + if (compl_old_match != NULL) { // If several matches were added (FORWARD) or the search failed and has // just been made cyclic then we have to move compl_curr_match to the @@ -4705,6 +5004,7 @@ static int ins_compl_start(void) line = ml_get(curwin->w_cursor.lnum); } + bool in_fuzzy = get_cot_flags() & kOptCotFlagFuzzy; if (compl_status_adding()) { edit_submode_pre = _(" Adding"); if (ctrl_x_mode_line_or_eval()) { @@ -4719,6 +5019,9 @@ static int ins_compl_start(void) compl_length = 0; compl_col = curwin->w_cursor.col; compl_lnum = curwin->w_cursor.lnum; + } else if (ctrl_x_mode_normal() && in_fuzzy) { + compl_startpos = curwin->w_cursor; + compl_cont_status &= CONT_S_IPOS; } } else { edit_submode_pre = NULL; @@ -4746,7 +5049,7 @@ static int ins_compl_start(void) } if (ins_compl_add(compl_orig_text.data, (int)compl_orig_text.size, NULL, NULL, false, NULL, 0, - flags, false, NULL) != OK) { + flags, false, NULL, 0) != OK) { API_CLEAR_STRING(compl_pattern); API_CLEAR_STRING(compl_orig_text); kv_destroy(compl_orig_extmarks); diff --git a/src/nvim/option_vars.h b/src/nvim/option_vars.h index 3f0445675d..2e5698870f 100644 --- a/src/nvim/option_vars.h +++ b/src/nvim/option_vars.h @@ -294,8 +294,10 @@ EXTERN char *p_cms; ///< 'commentstring' EXTERN char *p_cpt; ///< 'complete' EXTERN OptInt p_columns; ///< 'columns' EXTERN int p_confirm; ///< 'confirm' +EXTERN char *p_cfc; ///< 'completefuzzycollect' +EXTERN unsigned cfc_flags; ///< flags from 'completefuzzycollect' EXTERN char *p_cia; ///< 'completeitemalign' -EXTERN unsigned cia_flags; ///< order flags of 'completeitemalign' +EXTERN unsigned cia_flags; ///< order flags of 'completeitemalign' EXTERN char *p_cot; ///< 'completeopt' EXTERN unsigned cot_flags; ///< flags from 'completeopt' #ifdef BACKSLASH_IN_FILENAME diff --git a/src/nvim/options.lua b/src/nvim/options.lua index bb63ff638b..f566582c0c 100644 --- a/src/nvim/options.lua +++ b/src/nvim/options.lua @@ -1460,6 +1460,40 @@ local options = { varname = 'p_cfu', }, { + abbreviation = 'cfc', + defaults = '', + values = { 'keyword', 'files', 'whole_line' }, + flags = true, + deny_duplicates = true, + desc = [=[ + A comma-separated list of strings to enable fuzzy collection for + specific |ins-completion| modes, affecting how matches are gathered + during completion. For specified modes, fuzzy matching is used to + find completion candidates instead of the standard prefix-based + matching. This option can contain the following values: + + keyword keywords in the current file |i_CTRL-X_CTRL-N| + keywords with flags ".", "w", |i_CTRL-N| |i_CTRL-P| + "b", "u", "U" and "k{dict}" in 'complete' + keywords in 'dictionary' |i_CTRL-X_CTRL-K| + + files file names |i_CTRL-X_CTRL-F| + + whole_line whole lines |i_CTRL-X_CTRL-L| + + When using the 'completeopt' "longest" option value, fuzzy collection + can identify the longest common string among the best fuzzy matches + and insert it automatically. + ]=], + full_name = 'completefuzzycollect', + list = 'onecomma', + scope = { 'global' }, + short_desc = N_('use fuzzy collection for specific completion modes'), + type = 'string', + varname = 'p_cfc', + flags_varname = 'cfc_flags', + }, + { abbreviation = 'cia', cb = 'did_set_completeitemalign', defaults = 'abbr,kind,menu', @@ -1505,10 +1539,12 @@ local options = { fuzzy Enable |fuzzy-matching| for completion candidates. This allows for more flexible and intuitive matching, where characters can be skipped and matches can be found even - if the exact sequence is not typed. Only makes a - difference how completion candidates are reduced from the - list of alternatives, but not how the candidates are - collected (using different completion types). + if the exact sequence is not typed. Note: This option + does not affect the collection of candidate list, it only + controls how completion candidates are reduced from the + list of alternatives. If you want to use |fuzzy-matching| + to gather more alternatives for your candidate list, + see |'completefuzzycollect'|. longest Only insert the longest common text of the matches. If the menu is displayed you can use CTRL-L to add more diff --git a/src/nvim/optionstr.c b/src/nvim/optionstr.c index 3a6b4c9936..c6cc7af8cd 100644 --- a/src/nvim/optionstr.c +++ b/src/nvim/optionstr.c @@ -84,6 +84,7 @@ void didset_string_options(void) check_str_opt(kOptCasemap, NULL); check_str_opt(kOptBackupcopy, NULL); check_str_opt(kOptBelloff, NULL); + check_str_opt(kOptCompletefuzzycollect, NULL); check_str_opt(kOptCompleteopt, NULL); check_str_opt(kOptSessionoptions, NULL); check_str_opt(kOptViewoptions, NULL); diff --git a/src/nvim/search.c b/src/nvim/search.c index 04f33b9445..b38296ac5c 100644 --- a/src/nvim/search.c +++ b/src/nvim/search.c @@ -3620,6 +3620,179 @@ garray_T *fuzzy_match_str_with_pos(char *const str, const char *const pat) return match_positions; } +/// This function splits the line pointed to by `*ptr` into words and performs +/// a fuzzy match for the pattern `pat` on each word. It iterates through the +/// line, moving `*ptr` to the start of each word during the process. +/// +/// If a match is found: +/// - `*ptr` points to the start of the matched word. +/// - `*len` is set to the length of the matched word. +/// - `*score` contains the match score. +/// +/// If no match is found, `*ptr` is updated to the end of the line. +bool fuzzy_match_str_in_line(char **ptr, char *pat, int *len, pos_T *current_pos, int *score) +{ + char *str = *ptr; + char *strBegin = str; + char *end = NULL; + char *start = NULL; + bool found = false; + + if (str == NULL || pat == NULL) { + return found; + } + char *line_end = find_line_end(str); + + while (str < line_end) { + // Skip non-word characters + start = find_word_start(str); + if (*start == NUL) { + break; + } + end = find_word_end(start); + + // Extract the word from start to end + char save_end = *end; + *end = NUL; + + // Perform fuzzy match + *score = fuzzy_match_str(start, pat); + *end = save_end; + + if (*score > 0) { + *len = (int)(end - start); + found = true; + *ptr = start; + if (current_pos) { + current_pos->col += (int)(end - strBegin); + } + break; + } + + // Move to the end of the current word for the next iteration + str = end; + // Ensure we continue searching after the current word + while (*str != NUL && !vim_iswordp(str)) { + MB_PTR_ADV(str); + } + } + + if (!found) { + *ptr = line_end; + } + + return found; +} + +/// Search for the next fuzzy match in the specified buffer. +/// This function attempts to find the next occurrence of the given pattern +/// in the buffer, starting from the current position. It handles line wrapping +/// and direction of search. +/// +/// Return true if a match is found, otherwise false. +bool search_for_fuzzy_match(buf_T *buf, pos_T *pos, char *pattern, int dir, pos_T *start_pos, + int *len, char **ptr, int *score) +{ + pos_T current_pos = *pos; + pos_T circly_end; + bool found_new_match = false; + bool looped_around = false; + + bool whole_line = ctrl_x_mode_whole_line(); + + if (buf == curbuf) { + circly_end = *start_pos; + } else { + circly_end.lnum = buf->b_ml.ml_line_count; + circly_end.col = 0; + circly_end.coladd = 0; + } + + if (whole_line && start_pos->lnum != pos->lnum) { + current_pos.lnum += dir; + } + + while (true) { + // Check if looped around and back to start position + if (looped_around && equalpos(current_pos, circly_end)) { + break; + } + + // Ensure current_pos is valid + if (current_pos.lnum >= 1 && current_pos.lnum <= buf->b_ml.ml_line_count) { + // Get the current line buffer + *ptr = ml_get_buf(buf, current_pos.lnum); + if (!whole_line) { + *ptr += current_pos.col; + } + + // If ptr is end of line is reached, move to next line + // or previous line based on direction + if (*ptr != NULL && **ptr != NUL) { + if (!whole_line) { + // Try to find a fuzzy match in the current line starting + // from current position + found_new_match = fuzzy_match_str_in_line(ptr, pattern, + len, ¤t_pos, score); + if (found_new_match) { + if (ctrl_x_mode_normal()) { + if (strncmp(*ptr, pattern, (size_t)(*len)) == 0 && pattern[*len] == NUL) { + char *next_word_end = find_word_start(*ptr + *len); + if (*next_word_end != NUL && *next_word_end != NL) { + // Find end of the word. + while (*next_word_end != NUL) { + int l = utfc_ptr2len(next_word_end); + if (l < 2 && !vim_iswordc(*next_word_end)) { + break; + } + next_word_end += l; + } + } + *len = (int)(next_word_end - *ptr); + } + } + *pos = current_pos; + break; + } else if (looped_around && current_pos.lnum == circly_end.lnum) { + break; + } + } else { + if (fuzzy_match_str(*ptr, pattern) > 0) { + found_new_match = true; + *pos = current_pos; + *len = ml_get_buf_len(buf, current_pos.lnum); + break; + } + } + } + } + + // Move to the next line or previous line based on direction + if (dir == FORWARD) { + if (++current_pos.lnum > buf->b_ml.ml_line_count) { + if (p_ws) { + current_pos.lnum = 1; + looped_around = true; + } else { + break; + } + } + } else { + if (--current_pos.lnum < 1) { + if (p_ws) { + current_pos.lnum = buf->b_ml.ml_line_count; + looped_around = true; + } else { + break; + } + } + } + current_pos.col = 0; + } + + return found_new_match; +} + /// Copy a list of fuzzy matches into a string list after sorting the matches by /// the fuzzy score. Frees the memory allocated for "fuzmatch". void fuzzymatches_to_strmatches(fuzmatch_str_T *const fuzmatch, char ***const matches, @@ -4071,7 +4244,7 @@ search_line: const int add_r = ins_compl_add_infercase(aux, i, p_ic, curr_fname == curbuf->b_fname ? NULL : curr_fname, - dir, cont_s_ipos); + dir, cont_s_ipos, 0); if (add_r == OK) { // if dir was BACKWARD then honor it just once dir = FORWARD; diff --git a/src/nvim/spell.c b/src/nvim/spell.c index 7437a7a32f..c8c3622347 100644 --- a/src/nvim/spell.c +++ b/src/nvim/spell.c @@ -3483,7 +3483,7 @@ static void dump_word(slang_T *slang, char *word, char *pat, Direction *dir, int ? mb_strnicmp(p, pat, strlen(pat)) == 0 : strncmp(p, pat, strlen(pat)) == 0) && ins_compl_add_infercase(p, (int)strlen(p), - p_ic, NULL, *dir, false) == OK) { + p_ic, NULL, *dir, false, 0) == OK) { // if dir was BACKWARD then honor it just once *dir = FORWARD; } |