blob: 0c920709e409d8ec77ebbf9e7ab603e10fd1a0d5 [file] [log] [blame]
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * insexpand.c: functions for Insert mode completion
12 */
13
14#include "vim.h"
15
Bram Moolenaar7591bb32019-03-30 13:53:47 +010016/*
17 * Definitions used for CTRL-X submode.
18 * Note: If you change CTRL-X submode, you must also maintain ctrl_x_msgs[] and
19 * ctrl_x_mode_names[] below.
20 */
21# define CTRL_X_WANT_IDENT 0x100
22
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +010023# define CTRL_X_NORMAL 0 // CTRL-N CTRL-P completion, default
Bram Moolenaar7591bb32019-03-30 13:53:47 +010024# define CTRL_X_NOT_DEFINED_YET 1
25# define CTRL_X_SCROLL 2
26# define CTRL_X_WHOLE_LINE 3
27# define CTRL_X_FILES 4
28# define CTRL_X_TAGS (5 + CTRL_X_WANT_IDENT)
29# define CTRL_X_PATH_PATTERNS (6 + CTRL_X_WANT_IDENT)
30# define CTRL_X_PATH_DEFINES (7 + CTRL_X_WANT_IDENT)
31# define CTRL_X_FINISHED 8
32# define CTRL_X_DICTIONARY (9 + CTRL_X_WANT_IDENT)
33# define CTRL_X_THESAURUS (10 + CTRL_X_WANT_IDENT)
34# define CTRL_X_CMDLINE 11
Bram Moolenaar9810cfb2019-12-11 21:23:00 +010035# define CTRL_X_FUNCTION 12
Bram Moolenaar7591bb32019-03-30 13:53:47 +010036# define CTRL_X_OMNI 13
37# define CTRL_X_SPELL 14
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +010038# define CTRL_X_LOCAL_MSG 15 // only used in "ctrl_x_msgs"
39# define CTRL_X_EVAL 16 // for builtin function complete()
zeertzjqdca29d92021-08-31 19:12:51 +020040# define CTRL_X_CMDLINE_CTRL_X 17 // CTRL-X typed in CTRL_X_CMDLINE
glepnir05460682025-05-26 18:23:27 +020041# define CTRL_X_REGISTER 18 // complete words from registers
Bram Moolenaar7591bb32019-03-30 13:53:47 +010042
43# define CTRL_X_MSG(i) ctrl_x_msgs[(i) & ~CTRL_X_WANT_IDENT]
44
45// Message for CTRL-X mode, index is ctrl_x_mode.
46static char *ctrl_x_msgs[] =
47{
48 N_(" Keyword completion (^N^P)"), // CTRL_X_NORMAL, ^P/^N compl.
glepnir05460682025-05-26 18:23:27 +020049 N_(" ^X mode (^]^D^E^F^I^K^L^N^O^P^Rs^U^V^Y)"),
Bram Moolenaar7591bb32019-03-30 13:53:47 +010050 NULL, // CTRL_X_SCROLL: depends on state
51 N_(" Whole line completion (^L^N^P)"),
52 N_(" File name completion (^F^N^P)"),
53 N_(" Tag completion (^]^N^P)"),
54 N_(" Path pattern completion (^N^P)"),
55 N_(" Definition completion (^D^N^P)"),
56 NULL, // CTRL_X_FINISHED
57 N_(" Dictionary completion (^K^N^P)"),
58 N_(" Thesaurus completion (^T^N^P)"),
59 N_(" Command-line completion (^V^N^P)"),
60 N_(" User defined completion (^U^N^P)"),
61 N_(" Omni completion (^O^N^P)"),
zeertzjq7002c052024-06-21 07:55:07 +020062 N_(" Spelling suggestion (^S^N^P)"),
Bram Moolenaar7591bb32019-03-30 13:53:47 +010063 N_(" Keyword Local completion (^N^P)"),
64 NULL, // CTRL_X_EVAL doesn't use msg.
zeertzjqdca29d92021-08-31 19:12:51 +020065 N_(" Command-line completion (^V^N^P)"),
glepnir05460682025-05-26 18:23:27 +020066 N_(" Register completion (^N^P)"),
Bram Moolenaar7591bb32019-03-30 13:53:47 +010067};
68
Bram Moolenaar9cb698d2019-08-21 15:30:45 +020069#if defined(FEAT_COMPL_FUNC) || defined(FEAT_EVAL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +010070static char *ctrl_x_mode_names[] = {
Hirohito Higashi355db992025-04-28 18:07:02 +020071 "keyword",
72 "ctrl_x",
73 "scroll",
74 "whole_line",
75 "files",
76 "tags",
77 "path_patterns",
78 "path_defines",
79 "unknown", // CTRL_X_FINISHED
80 "dictionary",
81 "thesaurus",
82 "cmdline",
83 "function",
84 "omni",
85 "spell",
86 NULL, // CTRL_X_LOCAL_MSG only used in "ctrl_x_msgs"
87 "eval",
88 "cmdline",
glepnir05460682025-05-26 18:23:27 +020089 "register",
Bram Moolenaar7591bb32019-03-30 13:53:47 +010090};
Bram Moolenaar9cb698d2019-08-21 15:30:45 +020091#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +010092
93/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +010094 * Structure used to store one match for insert completion.
95 */
96typedef struct compl_S compl_T;
97struct compl_S
98{
99 compl_T *cp_next;
100 compl_T *cp_prev;
glepnir80b66202024-11-27 21:53:53 +0100101 compl_T *cp_match_next; // matched next compl_T
John Marriott5e6ea922024-11-23 14:01:57 +0100102 string_T cp_str; // matched text
Bram Moolenaar73655cf2019-04-06 13:45:55 +0200103 char_u *(cp_text[CPT_COUNT]); // text for the menu
Bram Moolenaarab782c52020-01-04 19:00:11 +0100104#ifdef FEAT_EVAL
Bram Moolenaar08928322020-01-04 14:32:48 +0100105 typval_T cp_user_data;
Bram Moolenaarab782c52020-01-04 19:00:11 +0100106#endif
glepnir0fe17f82024-10-08 22:26:44 +0200107 char_u *cp_fname; // file containing the match, allocated when
108 // cp_flags has CP_FREE_FNAME
109 int cp_flags; // CP_ values
110 int cp_number; // sequence number
Girish Palyab1565882025-04-15 20:16:00 +0200111 int cp_score; // fuzzy match score or proximity score
glepnird4088ed2024-12-31 10:55:22 +0100112 int cp_in_match_array; // collected by compl_match_array
glepnir7baa0142024-10-09 20:19:25 +0200113 int cp_user_abbr_hlattr; // highlight attribute for abbr
glepnir0fe17f82024-10-08 22:26:44 +0200114 int cp_user_kind_hlattr; // highlight attribute for kind
Girish Palya0ac1eb32025-04-16 20:18:33 +0200115 int cp_cpt_source_idx; // index of this match's source in 'cpt' option
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100116};
117
Bram Moolenaard9eefe32019-04-06 14:22:21 +0200118// values for cp_flags
119# define CP_ORIGINAL_TEXT 1 // the original text when the expansion begun
120# define CP_FREE_FNAME 2 // cp_fname is allocated
121# define CP_CONT_S_IPOS 4 // use CONT_S_IPOS for compl_cont_status
122# define CP_EQUAL 8 // ins_compl_equal() always returns TRUE
123# define CP_ICASE 16 // ins_compl_equal() ignores case
Bram Moolenaar440cf092021-04-03 20:13:30 +0200124# define CP_FAST 32 // use fast_breakcheck instead of ui_breakcheck
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100125
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100126/*
127 * All the current matches are stored in a list.
128 * "compl_first_match" points to the start of the list.
129 * "compl_curr_match" points to the currently selected entry.
130 * "compl_shown_match" is different from compl_curr_match during
Girish Palyacbe53192025-04-14 22:13:15 +0200131 * ins_compl_get_exp(), when new matches are added to the list.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000132 * "compl_old_match" points to previous "compl_curr_match".
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100133 */
134static compl_T *compl_first_match = NULL;
135static compl_T *compl_curr_match = NULL;
136static compl_T *compl_shown_match = NULL;
137static compl_T *compl_old_match = NULL;
138
glepnirf31cfa22025-03-06 21:59:13 +0100139// list used to store the compl_T which have the max score
140// used for completefuzzycollect
141static compl_T **compl_best_matches = NULL;
142static int compl_num_bests = 0;
143// inserted a longest when completefuzzycollect enabled
144static int compl_cfc_longest_ins = FALSE;
145
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100146// After using a cursor key <Enter> selects a match in the popup menu,
147// otherwise it inserts a line break.
148static int compl_enter_selects = FALSE;
149
150// When "compl_leader" is not NULL only matches that start with this string
151// are used.
John Marriott5e6ea922024-11-23 14:01:57 +0100152static string_T compl_leader = {NULL, 0};
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100153
154static int compl_get_longest = FALSE; // put longest common string
155 // in compl_leader
156
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100157// Selected one of the matches. When FALSE the match was edited or using the
158// longest common string.
159static int compl_used_match;
160
161// didn't finish finding completions.
162static int compl_was_interrupted = FALSE;
163
164// Set when character typed while looking for matches and it means we should
165// stop looking for matches.
166static int compl_interrupted = FALSE;
167
168static int compl_restarting = FALSE; // don't insert match
169
170// When the first completion is done "compl_started" is set. When it's
171// FALSE the word to be completed must be located.
172static int compl_started = FALSE;
173
174// Which Ctrl-X mode are we in?
175static int ctrl_x_mode = CTRL_X_NORMAL;
176
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000177static int compl_matches = 0; // number of completion matches
Girish Palyacbe53192025-04-14 22:13:15 +0200178static string_T compl_pattern = {NULL, 0}; // search pattern for matching items
179#ifdef FEAT_COMPL_FUNC
180static string_T cpt_compl_pattern = {NULL, 0}; // pattern returned by func in 'cpt'
181#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100182static int compl_direction = FORWARD;
183static int compl_shows_dir = FORWARD;
184static int compl_pending = 0; // > 1 for postponed CTRL-N
185static pos_T compl_startpos;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000186// Length in bytes of the text being completed (this is deleted to be replaced
187// by the match.)
188static int compl_length = 0;
glepnir76bdb822025-02-08 19:04:51 +0100189static linenr_T compl_lnum = 0; // lnum where the completion start
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100190static colnr_T compl_col = 0; // column where the text starts
191 // that is being completed
glepnir6a38aff2024-12-16 21:56:16 +0100192static colnr_T compl_ins_end_col = 0;
John Marriott5e6ea922024-11-23 14:01:57 +0100193static string_T compl_orig_text = {NULL, 0}; // text as it was before
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100194 // completion started
195static int compl_cont_mode = 0;
196static expand_T compl_xp;
197
glepnircf7f0122025-04-15 19:02:00 +0200198static win_T *compl_curr_win = NULL; // win where completion is active
199static buf_T *compl_curr_buf = NULL; // buf where completion is active
200
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000201// List of flags for method of completion.
202static int compl_cont_status = 0;
203# define CONT_ADDING 1 // "normal" or "adding" expansion
204# define CONT_INTRPT (2 + 4) // a ^X interrupted the current expansion
205 // it's set only iff N_ADDS is set
206# define CONT_N_ADDS 4 // next ^X<> will add-new or expand-current
207# define CONT_S_IPOS 8 // next ^X<> will set initial_pos?
208 // if so, word-wise-expansion will set SOL
209# define CONT_SOL 16 // pattern includes start of line, just for
210 // word-wise expansion, not set for ^X^L
211# define CONT_LOCAL 32 // for ctrl_x_mode 0, ^X^P/^X^N do a local
212 // expansion, (eg use complete=.)
213
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100214static int compl_opt_refresh_always = FALSE;
215static int compl_opt_suppress_empty = FALSE;
216
glepnira218cc62024-06-03 19:32:39 +0200217static int compl_selected_item = -1;
218
glepnir8159fb12024-07-17 20:32:54 +0200219static int *compl_fuzzy_scores;
220
Girish Palya0ac1eb32025-04-16 20:18:33 +0200221// Define the structure for completion source (in 'cpt' option) information
222typedef struct cpt_source_T
223{
Girish Palya98c29db2025-06-01 19:40:00 +0200224 int cs_refresh_always; // Whether 'refresh:always' is set for func
225 int cs_startcol; // Start column returned by func
226 int cs_max_matches; // Max items to display from this source
Girish Palya0ac1eb32025-04-16 20:18:33 +0200227} cpt_source_T;
228
229static cpt_source_T *cpt_sources_array; // Pointer to the array of completion sources
230static int cpt_sources_count; // Total number of completion sources specified in the 'cpt' option
Girish Palya7c621052025-05-26 19:41:59 +0200231static int cpt_sources_index = -1; // Index of the current completion source being expanded
Girish Palyacbe53192025-04-14 22:13:15 +0200232
glepnir6a38aff2024-12-16 21:56:16 +0100233// "compl_match_array" points the currently displayed list of entries in the
234// popup menu. It is NULL when there is no popup menu.
235static pumitem_T *compl_match_array = NULL;
236static int compl_match_arraysize;
237
glepnirf31cfa22025-03-06 21:59:13 +0100238static int ins_compl_add(char_u *str, int len, char_u *fname, char_u **cptext, typval_T *user_data, int cdir, int flags, int adup, int *user_hl, int score);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100239static void ins_compl_longest_match(compl_T *match);
240static void ins_compl_del_pum(void);
241static void ins_compl_files(int count, char_u **files, int thesaurus, int flags, regmatch_T *regmatch, char_u *buf, int *dir);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100242static void ins_compl_free(void);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100243static int ins_compl_need_restart(void);
244static void ins_compl_new_leader(void);
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000245static int get_compl_len(void);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100246static void ins_compl_restart(void);
John Marriott5e6ea922024-11-23 14:01:57 +0100247static void ins_compl_set_original_text(char_u *str, size_t len);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100248static void ins_compl_fixRedoBufForLeader(char_u *ptr_arg);
249# if defined(FEAT_COMPL_FUNC) || defined(FEAT_EVAL)
250static void ins_compl_add_list(list_T *list);
251static void ins_compl_add_dict(dict_T *dict);
Girish Palyacbe53192025-04-14 22:13:15 +0200252static int get_userdefined_compl_info(colnr_T curs_col, callback_T *cb, int *startcol);
Girish Palyafa16c7a2025-06-30 20:14:43 +0200253static void get_cpt_func_completion_matches(callback_T *cb, int restore_leader);
Girish Palya98c29db2025-06-01 19:40:00 +0200254static callback_T *get_callback_if_cpt_func(char_u *p);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100255# endif
Girish Palya98c29db2025-06-01 19:40:00 +0200256static int setup_cpt_sources(void);
Girish Palyacbe53192025-04-14 22:13:15 +0200257static int is_cpt_func_refresh_always(void);
Girish Palya0ac1eb32025-04-16 20:18:33 +0200258static void cpt_sources_clear(void);
Girish Palyacbe53192025-04-14 22:13:15 +0200259static void cpt_compl_refresh(void);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100260static int ins_compl_key2dir(int c);
261static int ins_compl_pum_key(int c);
262static int ins_compl_key2count(int c);
263static void show_pum(int prev_w_wrow, int prev_w_leftcol);
264static unsigned quote_meta(char_u *dest, char_u *str, int len);
glepnir76bdb822025-02-08 19:04:51 +0100265static int ins_compl_has_multiple(void);
266static void ins_compl_expand_multiple(char_u *str);
glepnirf31cfa22025-03-06 21:59:13 +0100267static void ins_compl_longest_insert(char_u *prefix);
Girish Palya8cd42a52025-06-05 21:04:29 +0200268static void ins_compl_make_linear(void);
269static int ins_compl_make_cyclic(void);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100270
271#ifdef FEAT_SPELL
272static void spell_back_to_badword(void);
273static int spell_bad_len = 0; // length of located bad word
274#endif
275
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100276/*
277 * CTRL-X pressed in Insert mode.
278 */
279 void
280ins_ctrl_x(void)
281{
zeertzjqdca29d92021-08-31 19:12:51 +0200282 if (!ctrl_x_mode_cmdline())
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100283 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000284 // if the next ^X<> won't ADD nothing, then reset compl_cont_status
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100285 if (compl_cont_status & CONT_N_ADDS)
286 compl_cont_status |= CONT_INTRPT;
287 else
288 compl_cont_status = 0;
289 // We're not sure which CTRL-X mode it will be yet
290 ctrl_x_mode = CTRL_X_NOT_DEFINED_YET;
291 edit_submode = (char_u *)_(CTRL_X_MSG(ctrl_x_mode));
292 edit_submode_pre = NULL;
293 showmode();
294 }
zeertzjqdca29d92021-08-31 19:12:51 +0200295 else
296 // CTRL-X in CTRL-X CTRL-V mode behaves differently to make CTRL-X
297 // CTRL-V look like CTRL-N
298 ctrl_x_mode = CTRL_X_CMDLINE_CTRL_X;
=?UTF-8?q?Magnus=20Gro=C3=9F?=25def2c2021-10-22 18:56:39 +0100299
LemonBoy2bf52dd2022-04-09 18:17:34 +0100300 may_trigger_modechanged();
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100301}
302
303/*
304 * Functions to check the current CTRL-X mode.
305 */
Bram Moolenaarebfec1c2023-01-22 21:14:53 +0000306int ctrl_x_mode_none(void)
307 { return ctrl_x_mode == 0; }
308int ctrl_x_mode_normal(void)
309 { return ctrl_x_mode == CTRL_X_NORMAL; }
310int ctrl_x_mode_scroll(void)
311 { return ctrl_x_mode == CTRL_X_SCROLL; }
312int ctrl_x_mode_whole_line(void)
313 { return ctrl_x_mode == CTRL_X_WHOLE_LINE; }
314int ctrl_x_mode_files(void)
315 { return ctrl_x_mode == CTRL_X_FILES; }
316int ctrl_x_mode_tags(void)
317 { return ctrl_x_mode == CTRL_X_TAGS; }
318int ctrl_x_mode_path_patterns(void)
319 { return ctrl_x_mode == CTRL_X_PATH_PATTERNS; }
320int ctrl_x_mode_path_defines(void)
321 { return ctrl_x_mode == CTRL_X_PATH_DEFINES; }
322int ctrl_x_mode_dictionary(void)
323 { return ctrl_x_mode == CTRL_X_DICTIONARY; }
324int ctrl_x_mode_thesaurus(void)
325 { return ctrl_x_mode == CTRL_X_THESAURUS; }
326int ctrl_x_mode_cmdline(void)
327 { return ctrl_x_mode == CTRL_X_CMDLINE
zeertzjqdca29d92021-08-31 19:12:51 +0200328 || ctrl_x_mode == CTRL_X_CMDLINE_CTRL_X; }
Bram Moolenaarebfec1c2023-01-22 21:14:53 +0000329int ctrl_x_mode_function(void)
330 { return ctrl_x_mode == CTRL_X_FUNCTION; }
331int ctrl_x_mode_omni(void)
332 { return ctrl_x_mode == CTRL_X_OMNI; }
333int ctrl_x_mode_spell(void)
334 { return ctrl_x_mode == CTRL_X_SPELL; }
335static int ctrl_x_mode_eval(void)
336 { return ctrl_x_mode == CTRL_X_EVAL; }
337int ctrl_x_mode_line_or_eval(void)
338 { return ctrl_x_mode == CTRL_X_WHOLE_LINE || ctrl_x_mode == CTRL_X_EVAL; }
glepnir05460682025-05-26 18:23:27 +0200339int ctrl_x_mode_register(void)
340 { return ctrl_x_mode == CTRL_X_REGISTER; }
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100341
342/*
343 * Whether other than default completion has been selected.
344 */
345 int
346ctrl_x_mode_not_default(void)
347{
348 return ctrl_x_mode != CTRL_X_NORMAL;
349}
350
351/*
zeertzjqdca29d92021-08-31 19:12:51 +0200352 * Whether CTRL-X was typed without a following character,
353 * not including when in CTRL-X CTRL-V mode.
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100354 */
355 int
356ctrl_x_mode_not_defined_yet(void)
357{
358 return ctrl_x_mode == CTRL_X_NOT_DEFINED_YET;
359}
360
361/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000362 * Return TRUE if currently in "normal" or "adding" insert completion matches
363 * state
364 */
365 int
366compl_status_adding(void)
367{
368 return compl_cont_status & CONT_ADDING;
369}
370
371/*
372 * Return TRUE if the completion pattern includes start of line, just for
373 * word-wise expansion.
374 */
375 int
376compl_status_sol(void)
377{
378 return compl_cont_status & CONT_SOL;
379}
380
381/*
382 * Return TRUE if ^X^P/^X^N will do a local completion (i.e. use complete=.)
383 */
384 int
385compl_status_local(void)
386{
387 return compl_cont_status & CONT_LOCAL;
388}
389
390/*
391 * Clear the completion status flags
392 */
393 void
394compl_status_clear(void)
395{
396 compl_cont_status = 0;
397}
398
399/*
400 * Return TRUE if completion is using the forward direction matches
401 */
402 static int
403compl_dir_forward(void)
404{
405 return compl_direction == FORWARD;
406}
407
408/*
409 * Return TRUE if currently showing forward completion matches
410 */
411 static int
412compl_shows_dir_forward(void)
413{
414 return compl_shows_dir == FORWARD;
415}
416
417/*
418 * Return TRUE if currently showing backward completion matches
419 */
420 static int
421compl_shows_dir_backward(void)
422{
423 return compl_shows_dir == BACKWARD;
424}
425
426/*
427 * Return TRUE if the 'dictionary' or 'thesaurus' option can be used.
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100428 */
429 int
430has_compl_option(int dict_opt)
431{
432 if (dict_opt ? (*curbuf->b_p_dict == NUL && *p_dict == NUL
Bram Moolenaare2c453d2019-08-21 14:37:09 +0200433#ifdef FEAT_SPELL
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100434 && !curwin->w_p_spell
Bram Moolenaare2c453d2019-08-21 14:37:09 +0200435#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100436 )
Yegappan Lakshmanan160e9942021-10-16 15:41:29 +0100437 : (*curbuf->b_p_tsr == NUL && *p_tsr == NUL
438#ifdef FEAT_COMPL_FUNC
Bram Moolenaarf4d8b762021-10-17 14:13:09 +0100439 && *curbuf->b_p_tsrfu == NUL && *p_tsrfu == NUL
Yegappan Lakshmanan160e9942021-10-16 15:41:29 +0100440#endif
441 ))
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100442 {
443 ctrl_x_mode = CTRL_X_NORMAL;
444 edit_submode = NULL;
445 msg_attr(dict_opt ? _("'dictionary' option is empty")
446 : _("'thesaurus' option is empty"),
447 HL_ATTR(HLF_E));
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100448 if (emsg_silent == 0 && !in_assert_fails)
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100449 {
450 vim_beep(BO_COMPL);
451 setcursor();
452 out_flush();
453#ifdef FEAT_EVAL
454 if (!get_vim_var_nr(VV_TESTING))
455#endif
Bram Moolenaareda1da02019-11-17 17:06:33 +0100456 ui_delay(2004L, FALSE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100457 }
458 return FALSE;
459 }
460 return TRUE;
461}
462
463/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000464 * Is the character "c" a valid key to go to or keep us in CTRL-X mode?
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100465 * This depends on the current mode.
466 */
467 int
468vim_is_ctrl_x_key(int c)
469{
470 // Always allow ^R - let its results then be checked
glepnir05460682025-05-26 18:23:27 +0200471 if (c == Ctrl_R && ctrl_x_mode != CTRL_X_REGISTER)
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100472 return TRUE;
473
474 // Accept <PageUp> and <PageDown> if the popup menu is visible.
475 if (ins_compl_pum_key(c))
476 return TRUE;
477
478 switch (ctrl_x_mode)
479 {
480 case 0: // Not in any CTRL-X mode
481 return (c == Ctrl_N || c == Ctrl_P || c == Ctrl_X);
482 case CTRL_X_NOT_DEFINED_YET:
zeertzjqdca29d92021-08-31 19:12:51 +0200483 case CTRL_X_CMDLINE_CTRL_X:
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100484 return ( c == Ctrl_X || c == Ctrl_Y || c == Ctrl_E
485 || c == Ctrl_L || c == Ctrl_F || c == Ctrl_RSB
486 || c == Ctrl_I || c == Ctrl_D || c == Ctrl_P
487 || c == Ctrl_N || c == Ctrl_T || c == Ctrl_V
488 || c == Ctrl_Q || c == Ctrl_U || c == Ctrl_O
zeertzjqdca29d92021-08-31 19:12:51 +0200489 || c == Ctrl_S || c == Ctrl_K || c == 's'
glepnir05460682025-05-26 18:23:27 +0200490 || c == Ctrl_Z || c == Ctrl_R);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100491 case CTRL_X_SCROLL:
492 return (c == Ctrl_Y || c == Ctrl_E);
493 case CTRL_X_WHOLE_LINE:
494 return (c == Ctrl_L || c == Ctrl_P || c == Ctrl_N);
495 case CTRL_X_FILES:
496 return (c == Ctrl_F || c == Ctrl_P || c == Ctrl_N);
497 case CTRL_X_DICTIONARY:
498 return (c == Ctrl_K || c == Ctrl_P || c == Ctrl_N);
499 case CTRL_X_THESAURUS:
500 return (c == Ctrl_T || c == Ctrl_P || c == Ctrl_N);
501 case CTRL_X_TAGS:
502 return (c == Ctrl_RSB || c == Ctrl_P || c == Ctrl_N);
503#ifdef FEAT_FIND_ID
504 case CTRL_X_PATH_PATTERNS:
505 return (c == Ctrl_P || c == Ctrl_N);
506 case CTRL_X_PATH_DEFINES:
507 return (c == Ctrl_D || c == Ctrl_P || c == Ctrl_N);
508#endif
509 case CTRL_X_CMDLINE:
510 return (c == Ctrl_V || c == Ctrl_Q || c == Ctrl_P || c == Ctrl_N
511 || c == Ctrl_X);
512#ifdef FEAT_COMPL_FUNC
513 case CTRL_X_FUNCTION:
514 return (c == Ctrl_U || c == Ctrl_P || c == Ctrl_N);
515 case CTRL_X_OMNI:
516 return (c == Ctrl_O || c == Ctrl_P || c == Ctrl_N);
517#endif
518 case CTRL_X_SPELL:
519 return (c == Ctrl_S || c == Ctrl_P || c == Ctrl_N);
520 case CTRL_X_EVAL:
521 return (c == Ctrl_P || c == Ctrl_N);
glepnir05460682025-05-26 18:23:27 +0200522 case CTRL_X_REGISTER:
523 return (c == Ctrl_R || c == Ctrl_P || c == Ctrl_N);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100524 }
525 internal_error("vim_is_ctrl_x_key()");
526 return FALSE;
527}
528
529/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000530 * Return TRUE if "match" is the original text when the completion began.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000531 */
532 static int
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000533match_at_original_text(compl_T *match)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000534{
535 return match->cp_flags & CP_ORIGINAL_TEXT;
536}
537
538/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000539 * Returns TRUE if "match" is the first match in the completion list.
540 */
541 static int
542is_first_match(compl_T *match)
543{
544 return match == compl_first_match;
545}
546
547/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100548 * Return TRUE when character "c" is part of the item currently being
549 * completed. Used to decide whether to abandon complete mode when the menu
550 * is visible.
551 */
552 int
553ins_compl_accept_char(int c)
554{
555 if (ctrl_x_mode & CTRL_X_WANT_IDENT)
556 // When expanding an identifier only accept identifier chars.
557 return vim_isIDc(c);
558
559 switch (ctrl_x_mode)
560 {
561 case CTRL_X_FILES:
562 // When expanding file name only accept file name chars. But not
563 // path separators, so that "proto/<Tab>" expands files in
564 // "proto", not "proto/" as a whole
565 return vim_isfilec(c) && !vim_ispathsep(c);
566
567 case CTRL_X_CMDLINE:
zeertzjqdca29d92021-08-31 19:12:51 +0200568 case CTRL_X_CMDLINE_CTRL_X:
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100569 case CTRL_X_OMNI:
570 // Command line and Omni completion can work with just about any
571 // printable character, but do stop at white space.
572 return vim_isprintc(c) && !VIM_ISWHITE(c);
573
574 case CTRL_X_WHOLE_LINE:
575 // For while line completion a space can be part of the line.
576 return vim_isprintc(c);
577 }
578 return vim_iswordc(c);
579}
580
581/*
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000582 * Get the completed text by inferring the case of the originally typed text.
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100583 * If the result is in allocated memory "tofree" is set to it.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000584 */
585 static char_u *
586ins_compl_infercase_gettext(
glepnir19e1dd62025-05-08 22:50:38 +0200587 char_u *str,
588 int char_len,
589 int compl_char_len,
590 int min_len,
591 char_u **tofree)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000592{
593 int *wca; // Wide character array.
594 char_u *p;
595 int i, c;
596 int has_lower = FALSE;
597 int was_letter = FALSE;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100598 garray_T gap;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000599
600 IObuff[0] = NUL;
601
602 // Allocate wide character array for the completion and fill it.
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100603 wca = ALLOC_MULT(int, char_len);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000604 if (wca == NULL)
605 return IObuff;
606
607 p = str;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100608 for (i = 0; i < char_len; ++i)
glepnir6e199932024-12-14 21:13:27 +0100609 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000610 if (has_mbyte)
611 wca[i] = mb_ptr2char_adv(&p);
612 else
613 wca[i] = *(p++);
glepnir6e199932024-12-14 21:13:27 +0100614 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000615
616 // Rule 1: Were any chars converted to lower?
John Marriott5e6ea922024-11-23 14:01:57 +0100617 p = compl_orig_text.string;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000618 for (i = 0; i < min_len; ++i)
619 {
glepnir6e199932024-12-14 21:13:27 +0100620 c = has_mbyte ? mb_ptr2char_adv(&p) : *(p++);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000621 if (MB_ISLOWER(c))
622 {
623 has_lower = TRUE;
624 if (MB_ISUPPER(wca[i]))
625 {
626 // Rule 1 is satisfied.
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100627 for (i = compl_char_len; i < char_len; ++i)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000628 wca[i] = MB_TOLOWER(wca[i]);
629 break;
630 }
631 }
632 }
633
634 // Rule 2: No lower case, 2nd consecutive letter converted to
635 // upper case.
636 if (!has_lower)
637 {
John Marriott5e6ea922024-11-23 14:01:57 +0100638 p = compl_orig_text.string;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000639 for (i = 0; i < min_len; ++i)
640 {
glepnir6e199932024-12-14 21:13:27 +0100641 c = has_mbyte ? mb_ptr2char_adv(&p) : *(p++);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000642 if (was_letter && MB_ISUPPER(c) && MB_ISLOWER(wca[i]))
643 {
644 // Rule 2 is satisfied.
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100645 for (i = compl_char_len; i < char_len; ++i)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000646 wca[i] = MB_TOUPPER(wca[i]);
647 break;
648 }
649 was_letter = MB_ISLOWER(c) || MB_ISUPPER(c);
650 }
651 }
652
653 // Copy the original case of the part we typed.
John Marriott5e6ea922024-11-23 14:01:57 +0100654 p = compl_orig_text.string;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000655 for (i = 0; i < min_len; ++i)
656 {
glepnir6e199932024-12-14 21:13:27 +0100657 c = has_mbyte ? mb_ptr2char_adv(&p) : *(p++);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000658 if (MB_ISLOWER(c))
659 wca[i] = MB_TOLOWER(wca[i]);
660 else if (MB_ISUPPER(c))
661 wca[i] = MB_TOUPPER(wca[i]);
662 }
663
664 // Generate encoding specific output from wide character array.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000665 p = IObuff;
666 i = 0;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100667 ga_init2(&gap, 1, 500);
668 while (i < char_len)
669 {
670 if (gap.ga_data != NULL)
671 {
672 if (ga_grow(&gap, 10) == FAIL)
673 {
674 ga_clear(&gap);
675 return (char_u *)"[failed]";
676 }
677 p = (char_u *)gap.ga_data + gap.ga_len;
678 if (has_mbyte)
679 gap.ga_len += (*mb_char2bytes)(wca[i++], p);
680 else
681 {
682 *p = wca[i++];
683 ++gap.ga_len;
684 }
685 }
686 else if ((p - IObuff) + 6 >= IOSIZE)
687 {
688 // Multi-byte characters can occupy up to five bytes more than
689 // ASCII characters, and we also need one byte for NUL, so when
690 // getting to six bytes from the edge of IObuff switch to using a
691 // growarray. Add the character in the next round.
692 if (ga_grow(&gap, IOSIZE) == FAIL)
zeertzjq70e566b2024-03-21 07:11:58 +0100693 {
694 vim_free(wca);
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100695 return (char_u *)"[failed]";
zeertzjq70e566b2024-03-21 07:11:58 +0100696 }
Bram Moolenaarb9e71732022-07-23 06:53:08 +0100697 *p = NUL;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100698 STRCPY(gap.ga_data, IObuff);
John Marriott5e6ea922024-11-23 14:01:57 +0100699 gap.ga_len = (int)(p - IObuff);
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100700 }
701 else if (has_mbyte)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000702 p += (*mb_char2bytes)(wca[i++], p);
703 else
704 *(p++) = wca[i++];
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100705 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000706 vim_free(wca);
707
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100708 if (gap.ga_data != NULL)
709 {
710 *tofree = gap.ga_data;
711 return gap.ga_data;
712 }
713
714 *p = NUL;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +0000715 return IObuff;
716}
717
718/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100719 * This is like ins_compl_add(), but if 'ic' and 'inf' are set, then the
720 * case of the originally typed text is used, and the case of the completed
721 * text is inferred, ie this tries to work out what case you probably wanted
722 * the rest of the word to be in -- webb
723 */
724 int
725ins_compl_add_infercase(
Bram Moolenaar73655cf2019-04-06 13:45:55 +0200726 char_u *str_arg,
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100727 int len,
728 int icase,
729 char_u *fname,
730 int dir,
glepnirf31cfa22025-03-06 21:59:13 +0100731 int cont_s_ipos, // next ^X<> will set initial_pos
732 int score)
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100733{
Bram Moolenaar73655cf2019-04-06 13:45:55 +0200734 char_u *str = str_arg;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100735 char_u *p;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100736 int char_len; // count multi-byte characters
737 int compl_char_len;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100738 int min_len;
Bram Moolenaard9eefe32019-04-06 14:22:21 +0200739 int flags = 0;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100740 int res;
741 char_u *tofree = NULL;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100742
743 if (p_ic && curbuf->b_p_inf && len > 0)
744 {
745 // Infer case of completed part.
746
747 // Find actual length of completion.
748 if (has_mbyte)
749 {
750 p = str;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100751 char_len = 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100752 while (*p != NUL)
753 {
754 MB_PTR_ADV(p);
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100755 ++char_len;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100756 }
757 }
758 else
glepnir6e199932024-12-14 21:13:27 +0100759 {
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100760 char_len = len;
glepnir6e199932024-12-14 21:13:27 +0100761 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100762
763 // Find actual length of original text.
764 if (has_mbyte)
765 {
John Marriott5e6ea922024-11-23 14:01:57 +0100766 p = compl_orig_text.string;
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100767 compl_char_len = 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100768 while (*p != NUL)
769 {
770 MB_PTR_ADV(p);
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100771 ++compl_char_len;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100772 }
773 }
774 else
glepnir6e199932024-12-14 21:13:27 +0100775 {
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100776 compl_char_len = compl_length;
glepnir6e199932024-12-14 21:13:27 +0100777 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100778
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100779 // "char_len" may be smaller than "compl_char_len" when using
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100780 // thesaurus, only use the minimum when comparing.
glepnir6e199932024-12-14 21:13:27 +0100781 min_len = MIN(char_len, compl_char_len);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100782
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100783 str = ins_compl_infercase_gettext(str, char_len,
784 compl_char_len, min_len, &tofree);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100785 }
Bram Moolenaard9eefe32019-04-06 14:22:21 +0200786 if (cont_s_ipos)
787 flags |= CP_CONT_S_IPOS;
788 if (icase)
789 flags |= CP_ICASE;
Bram Moolenaar73655cf2019-04-06 13:45:55 +0200790
glepnirf31cfa22025-03-06 21:59:13 +0100791 res = ins_compl_add(str, len, fname, NULL, NULL, dir, flags, FALSE, NULL, score);
Bram Moolenaarcaea6642022-07-07 19:42:04 +0100792 vim_free(tofree);
793 return res;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100794}
795
796/*
glepnirf31cfa22025-03-06 21:59:13 +0100797 * Check if ctrl_x_mode has been configured in 'completefuzzycollect'
798 */
799 static int
800cfc_has_mode(void)
801{
glepnir58760162025-03-13 21:39:51 +0100802 if (ctrl_x_mode_normal() || ctrl_x_mode_dictionary())
803 return (cfc_flags & CFC_KEYWORD) != 0;
804 else if (ctrl_x_mode_files())
805 return (cfc_flags & CFC_FILES) != 0;
806 else if (ctrl_x_mode_whole_line())
807 return (cfc_flags & CFC_WHOLELINE) != 0;
808 else
809 return FALSE;
glepnirf31cfa22025-03-06 21:59:13 +0100810}
811
812/*
Girish Palyab1565882025-04-15 20:16:00 +0200813 * Returns TRUE if matches should be sorted based on proximity to the cursor.
814 */
815 static int
816is_nearest_active(void)
817{
glepnirc3fbaa02025-05-08 23:05:10 +0200818 return (get_cot_flags() & (COT_NEAREST | COT_FUZZY)) == COT_NEAREST;
Girish Palyab1565882025-04-15 20:16:00 +0200819}
820
821/*
Yegappan Lakshmanan37079142022-01-08 10:38:48 +0000822 * Add a match to the list of matches. The arguments are:
823 * str - text of the match to add
824 * len - length of "str". If -1, then the length of "str" is
825 * computed.
826 * fname - file name to associate with this match.
827 * cptext - list of strings to use with this match (for abbr, menu, info
828 * and kind)
829 * user_data - user supplied data (any vim type) for this match
830 * cdir - match direction. If 0, use "compl_direction".
831 * flags_arg - match flags (cp_flags)
832 * adup - accept this match even if it is already present.
glepnir80b66202024-11-27 21:53:53 +0100833 * *user_hl - list of extra highlight attributes for abbr kind.
Yegappan Lakshmanan37079142022-01-08 10:38:48 +0000834 * If "cdir" is FORWARD, then the match is added after the current match.
835 * Otherwise, it is added before the current match.
836 *
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100837 * If the given string is already in the list of completions, then return
838 * NOTDONE, otherwise add it to the list and return OK. If there is an error,
839 * maybe because alloc() returns NULL, then FAIL is returned.
840 */
841 static int
842ins_compl_add(
843 char_u *str,
844 int len,
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100845 char_u *fname,
Bram Moolenaar8a7d6542020-01-26 15:56:19 +0100846 char_u **cptext, // extra text for popup menu or NULL
847 typval_T *user_data UNUSED, // "user_data" entry or NULL
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100848 int cdir,
Bram Moolenaard9eefe32019-04-06 14:22:21 +0200849 int flags_arg,
glepnir508e7852024-07-25 21:39:08 +0200850 int adup, // accept duplicate match
glepnirf31cfa22025-03-06 21:59:13 +0100851 int *user_hl, // user abbr/kind hlattr
852 int score)
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100853{
glepnirf31cfa22025-03-06 21:59:13 +0100854 compl_T *match, *current, *prev;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100855 int dir = (cdir == 0 ? compl_direction : cdir);
Bram Moolenaard9eefe32019-04-06 14:22:21 +0200856 int flags = flags_arg;
glepnirf31cfa22025-03-06 21:59:13 +0100857 int inserted = FALSE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100858
Bram Moolenaarceb06192021-04-04 15:05:22 +0200859 if (flags & CP_FAST)
860 fast_breakcheck();
861 else
862 ui_breakcheck();
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100863 if (got_int)
864 return FAIL;
865 if (len < 0)
866 len = (int)STRLEN(str);
867
868 // If the same match is already present, don't add it.
869 if (compl_first_match != NULL && !adup)
870 {
871 match = compl_first_match;
872 do
873 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000874 if (!match_at_original_text(match)
John Marriott5e6ea922024-11-23 14:01:57 +0100875 && STRNCMP(match->cp_str.string, str, len) == 0
876 && ((int)match->cp_str.length <= len
877 || match->cp_str.string[len] == NUL))
Girish Palyab1565882025-04-15 20:16:00 +0200878 {
879 if (is_nearest_active() && score > 0 && score < match->cp_score)
Girish Palyab1565882025-04-15 20:16:00 +0200880 match->cp_score = score;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100881 return NOTDONE;
Girish Palyab1565882025-04-15 20:16:00 +0200882 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100883 match = match->cp_next;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +0000884 } while (match != NULL && !is_first_match(match));
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100885 }
886
887 // Remove any popup menu before changing the list of matches.
888 ins_compl_del_pum();
889
890 // Allocate a new match structure.
891 // Copy the values to the new match structure.
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200892 match = ALLOC_CLEAR_ONE(compl_T);
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100893 if (match == NULL)
894 return FAIL;
glepnir40891ba2025-02-10 22:18:00 +0100895 match->cp_number = flags & CP_ORIGINAL_TEXT ? 0 : -1;
John Marriott5e6ea922024-11-23 14:01:57 +0100896 if ((match->cp_str.string = vim_strnsave(str, len)) == NULL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100897 {
898 vim_free(match);
899 return FAIL;
900 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100901
John Marriott5e6ea922024-11-23 14:01:57 +0100902 match->cp_str.length = len;
903
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100904 // match-fname is:
905 // - compl_curr_match->cp_fname if it is a string equal to fname.
Bram Moolenaard9eefe32019-04-06 14:22:21 +0200906 // - a copy of fname, CP_FREE_FNAME is set to free later THE allocated mem.
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100907 // - NULL otherwise. --Acevedo
908 if (fname != NULL
909 && compl_curr_match != NULL
910 && compl_curr_match->cp_fname != NULL
911 && STRCMP(fname, compl_curr_match->cp_fname) == 0)
912 match->cp_fname = compl_curr_match->cp_fname;
913 else if (fname != NULL)
914 {
915 match->cp_fname = vim_strsave(fname);
Bram Moolenaard9eefe32019-04-06 14:22:21 +0200916 flags |= CP_FREE_FNAME;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100917 }
918 else
919 match->cp_fname = NULL;
920 match->cp_flags = flags;
glepnir80b66202024-11-27 21:53:53 +0100921 match->cp_user_abbr_hlattr = user_hl ? user_hl[0] : -1;
922 match->cp_user_kind_hlattr = user_hl ? user_hl[1] : -1;
glepnirf31cfa22025-03-06 21:59:13 +0100923 match->cp_score = score;
Girish Palya0ac1eb32025-04-16 20:18:33 +0200924 match->cp_cpt_source_idx = cpt_sources_index;
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100925
926 if (cptext != NULL)
927 {
glepnir19e1dd62025-05-08 22:50:38 +0200928 for (int i = 0; i < CPT_COUNT; ++i)
glepnir6e199932024-12-14 21:13:27 +0100929 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100930 if (cptext[i] != NULL && *cptext[i] != NUL)
931 match->cp_text[i] = vim_strsave(cptext[i]);
glepnir6e199932024-12-14 21:13:27 +0100932 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100933 }
Bram Moolenaarab782c52020-01-04 19:00:11 +0100934#ifdef FEAT_EVAL
Bram Moolenaar08928322020-01-04 14:32:48 +0100935 if (user_data != NULL)
936 match->cp_user_data = *user_data;
Bram Moolenaarab782c52020-01-04 19:00:11 +0100937#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100938
Yegappan Lakshmanan37079142022-01-08 10:38:48 +0000939 // Link the new match structure after (FORWARD) or before (BACKWARD) the
940 // current match in the list of matches .
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100941 if (compl_first_match == NULL)
942 match->cp_next = match->cp_prev = NULL;
glepnirf31cfa22025-03-06 21:59:13 +0100943 else if (cfc_has_mode() && score > 0 && compl_get_longest)
944 {
945 current = compl_first_match->cp_next;
946 prev = compl_first_match;
947 inserted = FALSE;
948 // The direction is ignored when using longest and
949 // completefuzzycollect, because matches are inserted
950 // and sorted by score.
951 while (current != NULL && current != compl_first_match)
952 {
953 if (current->cp_score < score)
954 {
Hirohito Higashi355db992025-04-28 18:07:02 +0200955 match->cp_next = current;
956 match->cp_prev = current->cp_prev;
957 if (current->cp_prev)
glepnirf31cfa22025-03-06 21:59:13 +0100958 current->cp_prev->cp_next = match;
Hirohito Higashi355db992025-04-28 18:07:02 +0200959 current->cp_prev = match;
960 inserted = TRUE;
961 break;
glepnirf31cfa22025-03-06 21:59:13 +0100962 }
963 prev = current;
964 current = current->cp_next;
965 }
966 if (!inserted)
967 {
968 prev->cp_next = match;
969 match->cp_prev = prev;
970 match->cp_next = compl_first_match;
971 compl_first_match->cp_prev = match;
972 }
973 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100974 else if (dir == FORWARD)
975 {
976 match->cp_next = compl_curr_match->cp_next;
977 match->cp_prev = compl_curr_match;
978 }
979 else // BACKWARD
980 {
981 match->cp_next = compl_curr_match;
982 match->cp_prev = compl_curr_match->cp_prev;
983 }
984 if (match->cp_next)
985 match->cp_next->cp_prev = match;
986 if (match->cp_prev)
987 match->cp_prev->cp_next = match;
988 else // if there's nothing before, it is the first match
989 compl_first_match = match;
990 compl_curr_match = match;
991
992 // Find the longest common string if still doing that.
glepnirf31cfa22025-03-06 21:59:13 +0100993 if (compl_get_longest && (flags & CP_ORIGINAL_TEXT) == 0 && !cfc_has_mode())
Bram Moolenaar7591bb32019-03-30 13:53:47 +0100994 ins_compl_longest_match(match);
995
996 return OK;
997}
998
999/*
1000 * Return TRUE if "str[len]" matches with match->cp_str, considering
Bram Moolenaard9eefe32019-04-06 14:22:21 +02001001 * match->cp_flags.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001002 */
1003 static int
1004ins_compl_equal(compl_T *match, char_u *str, int len)
1005{
Bram Moolenaard9eefe32019-04-06 14:22:21 +02001006 if (match->cp_flags & CP_EQUAL)
Bram Moolenaar73655cf2019-04-06 13:45:55 +02001007 return TRUE;
Bram Moolenaard9eefe32019-04-06 14:22:21 +02001008 if (match->cp_flags & CP_ICASE)
John Marriott5e6ea922024-11-23 14:01:57 +01001009 return STRNICMP(match->cp_str.string, str, (size_t)len) == 0;
1010 return STRNCMP(match->cp_str.string, str, (size_t)len) == 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001011}
1012
1013/*
glepnir6a38aff2024-12-16 21:56:16 +01001014 * when len is -1 mean use whole length of p otherwise part of p
1015 */
1016 static void
1017ins_compl_insert_bytes(char_u *p, int len)
1018{
1019 if (len == -1)
1020 len = (int)STRLEN(p);
1021 ins_bytes_len(p, len);
zeertzjqf25d8f92024-12-18 21:12:25 +01001022 compl_ins_end_col = curwin->w_cursor.col;
glepnir6a38aff2024-12-16 21:56:16 +01001023}
1024
1025/*
zeertzjqd32bf0a2024-12-17 20:55:13 +01001026 * Checks if the column is within the currently inserted completion text
1027 * column range. If it is, it returns a special highlight attribute.
glepnir76bdb822025-02-08 19:04:51 +01001028 * -1 means normal item.
glepnir6a38aff2024-12-16 21:56:16 +01001029 */
1030 int
glepnir76bdb822025-02-08 19:04:51 +01001031ins_compl_col_range_attr(linenr_T lnum, int col)
glepnir6a38aff2024-12-16 21:56:16 +01001032{
glepnir76bdb822025-02-08 19:04:51 +01001033 int start_col;
1034 int attr;
1035
1036 if ((get_cot_flags() & COT_FUZZY)
1037 || (attr = syn_name2attr((char_u *)"ComplMatchIns")) == 0)
glepnire8908872025-01-08 18:30:45 +01001038 return -1;
1039
glepnir76bdb822025-02-08 19:04:51 +01001040 start_col = compl_col + (int)ins_compl_leader_len();
1041 if (!ins_compl_has_multiple())
1042 return (col >= start_col && col < compl_ins_end_col) ? attr : -1;
1043
1044 // Multiple lines
1045 if ((lnum == compl_lnum && col >= start_col && col < MAXCOL) ||
1046 (lnum > compl_lnum && lnum < curwin->w_cursor.lnum) ||
1047 (lnum == curwin->w_cursor.lnum && col <= compl_ins_end_col))
1048 return attr;
glepnir6a38aff2024-12-16 21:56:16 +01001049
1050 return -1;
1051}
1052
1053/*
glepnir76bdb822025-02-08 19:04:51 +01001054 * Returns TRUE if the current completion string contains newline characters,
1055 * indicating it's a multi-line completion.
1056 */
1057 static int
1058ins_compl_has_multiple(void)
1059{
1060 return vim_strchr(compl_shown_match->cp_str.string, '\n') != NULL;
1061}
1062
1063/*
1064 * Returns TRUE if the given line number falls within the range of a multi-line
1065 * completion, i.e. between the starting line (compl_lnum) and current cursor
1066 * line. Always returns FALSE for single-line completions.
1067 */
1068 int
1069ins_compl_lnum_in_range(linenr_T lnum)
1070{
1071 if (!ins_compl_has_multiple())
1072 return FALSE;
1073 return lnum >= compl_lnum && lnum <= curwin->w_cursor.lnum;
1074}
1075
1076/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001077 * Reduce the longest common string for match "match".
1078 */
1079 static void
1080ins_compl_longest_match(compl_T *match)
1081{
1082 char_u *p, *s;
1083 int c1, c2;
1084 int had_match;
1085
John Marriott5e6ea922024-11-23 14:01:57 +01001086 if (compl_leader.string == NULL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001087 {
1088 // First match, use it as a whole.
John Marriott5e6ea922024-11-23 14:01:57 +01001089 compl_leader.string = vim_strnsave(match->cp_str.string, match->cp_str.length);
1090 if (compl_leader.string == NULL)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001091 return;
1092
John Marriott5e6ea922024-11-23 14:01:57 +01001093 compl_leader.length = match->cp_str.length;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001094 had_match = (curwin->w_cursor.col > compl_col);
glepnirf31cfa22025-03-06 21:59:13 +01001095 ins_compl_longest_insert(compl_leader.string);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001096
1097 // When the match isn't there (to avoid matching itself) remove it
1098 // again after redrawing.
1099 if (!had_match)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001100 ins_compl_delete();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001101 compl_used_match = FALSE;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001102
1103 return;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001104 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001105
1106 // Reduce the text if this match differs from compl_leader.
John Marriott5e6ea922024-11-23 14:01:57 +01001107 p = compl_leader.string;
1108 s = match->cp_str.string;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001109 while (*p != NUL)
1110 {
1111 if (has_mbyte)
1112 {
1113 c1 = mb_ptr2char(p);
1114 c2 = mb_ptr2char(s);
1115 }
1116 else
1117 {
1118 c1 = *p;
1119 c2 = *s;
1120 }
1121 if ((match->cp_flags & CP_ICASE)
1122 ? (MB_TOLOWER(c1) != MB_TOLOWER(c2)) : (c1 != c2))
1123 break;
1124 if (has_mbyte)
1125 {
1126 MB_PTR_ADV(p);
1127 MB_PTR_ADV(s);
1128 }
1129 else
1130 {
1131 ++p;
1132 ++s;
1133 }
1134 }
1135
1136 if (*p != NUL)
1137 {
1138 // Leader was shortened, need to change the inserted text.
1139 *p = NUL;
John Marriott5e6ea922024-11-23 14:01:57 +01001140 compl_leader.length = (size_t)(p - compl_leader.string);
1141
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001142 had_match = (curwin->w_cursor.col > compl_col);
glepnirf31cfa22025-03-06 21:59:13 +01001143 ins_compl_longest_insert(compl_leader.string);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001144
1145 // When the match isn't there (to avoid matching itself) remove it
1146 // again after redrawing.
1147 if (!had_match)
1148 ins_compl_delete();
1149 }
1150
1151 compl_used_match = FALSE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001152}
1153
1154/*
1155 * Add an array of matches to the list of matches.
1156 * Frees matches[].
1157 */
1158 static void
1159ins_compl_add_matches(
1160 int num_matches,
1161 char_u **matches,
1162 int icase)
1163{
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001164 int add_r = OK;
1165 int dir = compl_direction;
1166
glepnir19e1dd62025-05-08 22:50:38 +02001167 for (int i = 0; i < num_matches && add_r != FAIL; i++)
glepnir6e199932024-12-14 21:13:27 +01001168 {
1169 add_r = ins_compl_add(matches[i], -1, NULL, NULL, NULL, dir,
glepnirf31cfa22025-03-06 21:59:13 +01001170 CP_FAST | (icase ? CP_ICASE : 0), FALSE, NULL, 0);
glepnir6e199932024-12-14 21:13:27 +01001171 if (add_r == OK)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001172 // if dir was BACKWARD then honor it just once
1173 dir = FORWARD;
glepnir6e199932024-12-14 21:13:27 +01001174 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001175 FreeWild(num_matches, matches);
1176}
1177
1178/*
1179 * Make the completion list cyclic.
1180 * Return the number of matches (excluding the original).
1181 */
1182 static int
1183ins_compl_make_cyclic(void)
1184{
1185 compl_T *match;
1186 int count = 0;
1187
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001188 if (compl_first_match == NULL)
1189 return 0;
1190
1191 // Find the end of the list.
1192 match = compl_first_match;
1193 // there's always an entry for the compl_orig_text, it doesn't count.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001194 while (match->cp_next != NULL && !is_first_match(match->cp_next))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001195 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001196 match = match->cp_next;
1197 ++count;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001198 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001199 match->cp_next = compl_first_match;
1200 compl_first_match->cp_prev = match;
1201
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001202 return count;
1203}
1204
1205/*
1206 * Return whether there currently is a shown match.
1207 */
1208 int
1209ins_compl_has_shown_match(void)
1210{
1211 return compl_shown_match == NULL
1212 || compl_shown_match != compl_shown_match->cp_next;
1213}
1214
1215/*
1216 * Return whether the shown match is long enough.
1217 */
1218 int
1219ins_compl_long_shown_match(void)
1220{
John Marriott5e6ea922024-11-23 14:01:57 +01001221 return (int)compl_shown_match->cp_str.length
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001222 > curwin->w_cursor.col - compl_col;
1223}
1224
1225/*
zeertzjq529b9ad2024-06-05 20:27:06 +02001226 * Get the local or global value of 'completeopt' flags.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001227 */
zeertzjq529b9ad2024-06-05 20:27:06 +02001228 unsigned int
1229get_cot_flags(void)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001230{
zeertzjq529b9ad2024-06-05 20:27:06 +02001231 return curbuf->b_cot_flags != 0 ? curbuf->b_cot_flags : cot_flags;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001232}
1233
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001234/*
1235 * Update the screen and when there is any scrolling remove the popup menu.
1236 */
1237 static void
1238ins_compl_upd_pum(void)
1239{
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001240 if (compl_match_array == NULL)
1241 return;
1242
glepnir19e1dd62025-05-08 22:50:38 +02001243 int h = curwin->w_cline_height;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001244 // Update the screen later, before drawing the popup menu over it.
1245 pum_call_update_screen();
1246 if (h != curwin->w_cline_height)
1247 ins_compl_del_pum();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001248}
1249
1250/*
1251 * Remove any popup menu.
1252 */
1253 static void
1254ins_compl_del_pum(void)
1255{
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001256 if (compl_match_array == NULL)
1257 return;
1258
1259 pum_undisplay();
1260 VIM_CLEAR(compl_match_array);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001261}
1262
1263/*
1264 * Return TRUE if the popup menu should be displayed.
1265 */
1266 int
1267pum_wanted(void)
1268{
1269 // 'completeopt' must contain "menu" or "menuone"
zeertzjq529b9ad2024-06-05 20:27:06 +02001270 if ((get_cot_flags() & COT_ANY_MENU) == 0)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001271 return FALSE;
1272
1273 // The display looks bad on a B&W display.
1274 if (t_colors < 8
1275#ifdef FEAT_GUI
1276 && !gui.in_use
1277#endif
1278 )
1279 return FALSE;
1280 return TRUE;
1281}
1282
1283/*
1284 * Return TRUE if there are two or more matches to be shown in the popup menu.
1285 * One if 'completopt' contains "menuone".
1286 */
1287 static int
1288pum_enough_matches(void)
1289{
1290 compl_T *compl;
glepnir6e199932024-12-14 21:13:27 +01001291 int i = 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001292
1293 // Don't display the popup menu if there are no matches or there is only
1294 // one (ignoring the original text).
1295 compl = compl_first_match;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001296 do
1297 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001298 if (compl == NULL || (!match_at_original_text(compl) && ++i == 2))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001299 break;
1300 compl = compl->cp_next;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001301 } while (!is_first_match(compl));
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001302
zeertzjq529b9ad2024-06-05 20:27:06 +02001303 if (get_cot_flags() & COT_MENUONE)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001304 return (i >= 1);
1305 return (i >= 2);
1306}
1307
Bram Moolenaar3075a452021-11-17 15:51:52 +00001308#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02001309/*
1310 * Allocate Dict for the completed item.
1311 * { word, abbr, menu, kind, info }
1312 */
1313 static dict_T *
1314ins_compl_dict_alloc(compl_T *match)
1315{
1316 dict_T *dict = dict_alloc_lock(VAR_FIXED);
1317
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001318 if (dict == NULL)
1319 return NULL;
1320
John Marriott5e6ea922024-11-23 14:01:57 +01001321 dict_add_string(dict, "word", match->cp_str.string);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001322 dict_add_string(dict, "abbr", match->cp_text[CPT_ABBR]);
1323 dict_add_string(dict, "menu", match->cp_text[CPT_MENU]);
1324 dict_add_string(dict, "kind", match->cp_text[CPT_KIND]);
1325 dict_add_string(dict, "info", match->cp_text[CPT_INFO]);
1326 if (match->cp_user_data.v_type == VAR_UNKNOWN)
1327 dict_add_string(dict, "user_data", (char_u *)"");
1328 else
1329 dict_add_tv(dict, "user_data", &match->cp_user_data);
1330
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02001331 return dict;
1332}
1333
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00001334/*
1335 * Trigger the CompleteChanged autocmd event. Invoked each time the Insert mode
1336 * completion menu is changed.
1337 */
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001338 static void
1339trigger_complete_changed_event(int cur)
1340{
1341 dict_T *v_event;
1342 dict_T *item;
1343 static int recursive = FALSE;
Bram Moolenaar3075a452021-11-17 15:51:52 +00001344 save_v_event_T save_v_event;
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001345
1346 if (recursive)
1347 return;
1348
glepnir40891ba2025-02-10 22:18:00 +01001349 item = cur < 0 ? dict_alloc() : ins_compl_dict_alloc(compl_curr_match);
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001350 if (item == NULL)
1351 return;
Bram Moolenaar3075a452021-11-17 15:51:52 +00001352 v_event = get_v_event(&save_v_event);
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001353 dict_add_dict(v_event, "completed_item", item);
1354 pum_set_event_info(v_event);
1355 dict_set_items_ro(v_event);
1356
1357 recursive = TRUE;
zeertzjqcfe45652022-05-27 17:26:55 +01001358 textlock++;
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001359 apply_autocmds(EVENT_COMPLETECHANGED, NULL, NULL, FALSE, curbuf);
zeertzjqcfe45652022-05-27 17:26:55 +01001360 textlock--;
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001361 recursive = FALSE;
1362
Bram Moolenaar3075a452021-11-17 15:51:52 +00001363 restore_v_event(v_event, &save_v_event);
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001364}
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02001365#endif
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001366
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001367/*
Girish Palya8cd42a52025-06-05 21:04:29 +02001368 * Helper functions for mergesort_list().
Girish Palya19ef6b02025-05-28 20:28:21 +02001369 */
Girish Palya8cd42a52025-06-05 21:04:29 +02001370 static void*
1371cp_get_next(void *node)
Girish Palya19ef6b02025-05-28 20:28:21 +02001372{
Girish Palya8cd42a52025-06-05 21:04:29 +02001373 return ((compl_T*)node)->cp_next;
Girish Palya19ef6b02025-05-28 20:28:21 +02001374}
1375
Girish Palya8cd42a52025-06-05 21:04:29 +02001376 static void
1377cp_set_next(void *node, void *next)
glepnira218cc62024-06-03 19:32:39 +02001378{
Girish Palya8cd42a52025-06-05 21:04:29 +02001379 ((compl_T*)node)->cp_next = (compl_T*)next;
1380}
1381
1382 static void*
1383cp_get_prev(void* node)
1384{
1385 return ((compl_T*)node)->cp_prev;
1386}
1387
1388 static void
1389cp_set_prev(void* node, void* prev)
1390{
1391 ((compl_T*)node)->cp_prev = (compl_T*)prev;
1392}
1393
1394 static int
1395cp_compare_fuzzy(const void* a, const void* b)
1396{
1397 int score_a = ((compl_T*)a)->cp_score;
1398 int score_b = ((compl_T*)b)->cp_score;
1399 return (score_b > score_a) ? 1 : (score_b < score_a) ? -1 : 0;
glepnira218cc62024-06-03 19:32:39 +02001400}
1401
Girish Palyab8ee1cf2025-06-08 16:20:06 +02001402 static int
1403cp_compare_nearest(const void* a, const void* b)
1404{
1405 int score_a = ((compl_T*)a)->cp_score;
1406 int score_b = ((compl_T*)b)->cp_score;
Girish Palyacd68f212025-06-22 20:23:54 +02001407 if (score_a == 0 || score_b == 0)
Girish Palyab8ee1cf2025-06-08 16:20:06 +02001408 return 0;
1409 return (score_a > score_b) ? 1 : (score_a < score_b) ? -1 : 0;
1410}
1411
1412/*
1413 * Set fuzzy score.
1414 */
1415 static void
1416set_fuzzy_score(void)
1417{
zeertzjq91782b42025-06-10 20:06:53 +02001418 compl_T *compl;
zeertzjqde1c7ac2025-06-09 20:34:57 +02001419
zeertzjq91782b42025-06-10 20:06:53 +02001420 if (!compl_first_match
1421 || compl_leader.string == NULL || compl_leader.length == 0)
1422 return;
1423
1424 compl = compl_first_match;
1425 do
1426 {
1427 compl->cp_score = fuzzy_match_str(compl->cp_str.string,
1428 compl_leader.string);
1429 compl = compl->cp_next;
1430 } while (compl != NULL && !is_first_match(compl));
Girish Palyab8ee1cf2025-06-08 16:20:06 +02001431}
1432
1433/*
1434 * Sort completion matches, excluding the node that contains the leader.
1435 */
1436 static void
1437sort_compl_match_list(int (*compare)(const void *, const void *))
1438{
zeertzjq91782b42025-06-10 20:06:53 +02001439 compl_T *compl;
Girish Palyab8ee1cf2025-06-08 16:20:06 +02001440
1441 if (!compl_first_match || is_first_match(compl_first_match->cp_next))
1442 return;
1443
zeertzjq91782b42025-06-10 20:06:53 +02001444 compl = compl_first_match->cp_prev;
Girish Palyab8ee1cf2025-06-08 16:20:06 +02001445 ins_compl_make_linear();
1446 if (compl_shows_dir_forward())
1447 {
1448 compl_first_match->cp_next->cp_prev = NULL;
1449 compl_first_match->cp_next = mergesort_list(compl_first_match->cp_next,
1450 cp_get_next, cp_set_next, cp_get_prev, cp_set_prev, compare);
1451 compl_first_match->cp_next->cp_prev = compl_first_match;
1452 }
1453 else
1454 {
1455 compl->cp_prev->cp_next = NULL;
1456 compl_first_match = mergesort_list(compl_first_match, cp_get_next,
1457 cp_set_next, cp_get_prev, cp_set_prev, compare);
1458 compl_T *tail = compl_first_match;
1459 while (tail->cp_next != NULL)
1460 tail = tail->cp_next;
1461 tail->cp_next = compl;
1462 compl->cp_prev = tail;
1463 }
1464 (void)ins_compl_make_cyclic();
1465}
1466
glepnira218cc62024-06-03 19:32:39 +02001467/*
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001468 * Build a popup menu to show the completion matches.
1469 * Returns the popup menu entry that should be selected. Returns -1 if nothing
1470 * should be selected.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001471 */
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001472 static int
1473ins_compl_build_pum(void)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001474{
1475 compl_T *compl;
1476 compl_T *shown_compl = NULL;
1477 int did_find_shown_match = FALSE;
1478 int shown_match_ok = FALSE;
glepnira49c0772024-11-30 10:56:30 +01001479 int i = 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001480 int cur = -1;
zeertzjqaa925ee2024-06-09 18:24:05 +02001481 unsigned int cur_cot_flags = get_cot_flags();
zeertzjq529b9ad2024-06-05 20:27:06 +02001482 int compl_no_select = (cur_cot_flags & COT_NOSELECT) != 0;
zeertzjqd65aa1b2025-01-25 15:29:03 +01001483 int fuzzy_filter = (cur_cot_flags & COT_FUZZY) != 0;
glepnir80b66202024-11-27 21:53:53 +01001484 compl_T *match_head = NULL;
1485 compl_T *match_tail = NULL;
1486 compl_T *match_next = NULL;
Girish Palya8cd42a52025-06-05 21:04:29 +02001487 int *match_count = NULL;
Girish Palya19ef6b02025-05-28 20:28:21 +02001488 int is_forward = compl_shows_dir_forward();
Girish Palya8cd42a52025-06-05 21:04:29 +02001489 int is_cpt_completion = (cpt_sources_array != NULL);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001490
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001491 // Need to build the popup menu list.
1492 compl_match_arraysize = 0;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001493
glepnira49c0772024-11-30 10:56:30 +01001494 // If the current match is the original text don't find the first
1495 // match after it, don't highlight anything.
1496 if (match_at_original_text(compl_shown_match))
1497 shown_match_ok = TRUE;
1498
1499 if (compl_leader.string != NULL
1500 && STRCMP(compl_leader.string, compl_orig_text.string) == 0
1501 && shown_match_ok == FALSE)
1502 compl_shown_match = compl_no_select ? compl_first_match
1503 : compl_first_match->cp_next;
1504
Girish Palya8cd42a52025-06-05 21:04:29 +02001505 if (is_cpt_completion)
1506 {
1507 match_count = ALLOC_CLEAR_MULT(int, cpt_sources_count);
1508 if (match_count == NULL)
1509 return -1;
1510 }
1511
1512 compl = compl_first_match;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001513 do
1514 {
glepnird4088ed2024-12-31 10:55:22 +01001515 compl->cp_in_match_array = FALSE;
Girish Palya0ac1eb32025-04-16 20:18:33 +02001516
Girish Palyadc314052025-05-08 23:28:52 +02001517 // Apply 'smartcase' behavior during normal mode
1518 if (ctrl_x_mode_normal() && !p_inf && compl_leader.string
1519 && !ignorecase(compl_leader.string) && !fuzzy_filter)
1520 compl->cp_flags &= ~CP_ICASE;
1521
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001522 if (!match_at_original_text(compl)
John Marriott5e6ea922024-11-23 14:01:57 +01001523 && (compl_leader.string == NULL
Girish Palyadc314052025-05-08 23:28:52 +02001524 || ins_compl_equal(compl, compl_leader.string,
1525 (int)compl_leader.length)
glepnirf400a0c2025-01-23 19:55:14 +01001526 || (fuzzy_filter && compl->cp_score > 0)))
glepnir80b66202024-11-27 21:53:53 +01001527 {
Girish Palya8cd42a52025-06-05 21:04:29 +02001528 // Limit number of items from each source if max_items is set.
1529 int match_limit_exceeded = FALSE;
1530 int cur_source = compl->cp_cpt_source_idx;
1531 if (is_forward && cur_source != -1 && is_cpt_completion)
glepnira49c0772024-11-30 10:56:30 +01001532 {
Girish Palya8cd42a52025-06-05 21:04:29 +02001533 match_count[cur_source]++;
1534 int max_matches = cpt_sources_array[cur_source].cs_max_matches;
1535 if (max_matches > 0 && match_count[cur_source] > max_matches)
1536 match_limit_exceeded = TRUE;
1537 }
1538
1539 if (!match_limit_exceeded)
1540 {
1541 ++compl_match_arraysize;
1542 compl->cp_in_match_array = TRUE;
1543 if (match_head == NULL)
1544 match_head = compl;
glepnira49c0772024-11-30 10:56:30 +01001545 else
Girish Palya8cd42a52025-06-05 21:04:29 +02001546 match_tail->cp_match_next = compl;
1547 match_tail = compl;
glepnira49c0772024-11-30 10:56:30 +01001548
Girish Palya8cd42a52025-06-05 21:04:29 +02001549 if (!shown_match_ok && !fuzzy_filter)
glepnira49c0772024-11-30 10:56:30 +01001550 {
Girish Palya8cd42a52025-06-05 21:04:29 +02001551 if (compl == compl_shown_match || did_find_shown_match)
1552 {
1553 // This item is the shown match or this is the
1554 // first displayed item after the shown match.
1555 compl_shown_match = compl;
1556 did_find_shown_match = TRUE;
1557 shown_match_ok = TRUE;
1558 }
1559 else
1560 // Remember this displayed match for when the
1561 // shown match is just below it.
1562 shown_compl = compl;
glepnira49c0772024-11-30 10:56:30 +01001563 cur = i;
glepnira49c0772024-11-30 10:56:30 +01001564 }
Girish Palya8cd42a52025-06-05 21:04:29 +02001565 else if (fuzzy_filter)
1566 {
1567 if (i == 0)
1568 shown_compl = compl;
1569
1570 if (!shown_match_ok && compl == compl_shown_match)
1571 {
1572 cur = i;
1573 shown_match_ok = TRUE;
1574 }
1575 }
1576 i++;
glepnira49c0772024-11-30 10:56:30 +01001577 }
glepnir80b66202024-11-27 21:53:53 +01001578 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001579
glepnirf400a0c2025-01-23 19:55:14 +01001580 if (compl == compl_shown_match && !fuzzy_filter)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001581 {
1582 did_find_shown_match = TRUE;
1583
1584 // When the original text is the shown match don't set
1585 // compl_shown_match.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00001586 if (match_at_original_text(compl))
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001587 shown_match_ok = TRUE;
1588
1589 if (!shown_match_ok && shown_compl != NULL)
1590 {
1591 // The shown match isn't displayed, set it to the
1592 // previously displayed match.
1593 compl_shown_match = shown_compl;
1594 shown_match_ok = TRUE;
1595 }
1596 }
glepnira49c0772024-11-30 10:56:30 +01001597 compl = compl->cp_next;
1598 } while (compl != NULL && !is_first_match(compl));
1599
Girish Palya8cd42a52025-06-05 21:04:29 +02001600 vim_free(match_count);
1601
glepnira49c0772024-11-30 10:56:30 +01001602 if (compl_match_arraysize == 0)
1603 return -1;
1604
Girish Palya8cd42a52025-06-05 21:04:29 +02001605 if (fuzzy_filter && !compl_no_select && !shown_match_ok)
glepnirc0b7ca42025-02-13 20:27:44 +01001606 {
1607 compl_shown_match = shown_compl;
1608 shown_match_ok = TRUE;
1609 cur = 0;
1610 }
1611
glepnira49c0772024-11-30 10:56:30 +01001612 compl_match_array = ALLOC_CLEAR_MULT(pumitem_T, compl_match_arraysize);
1613 if (compl_match_array == NULL)
1614 return -1;
1615
1616 compl = match_head;
1617 i = 0;
1618 while (compl != NULL)
1619 {
glepnir6e199932024-12-14 21:13:27 +01001620 compl_match_array[i].pum_text = compl->cp_text[CPT_ABBR] != NULL
1621 ? compl->cp_text[CPT_ABBR] : compl->cp_str.string;
glepnira49c0772024-11-30 10:56:30 +01001622 compl_match_array[i].pum_kind = compl->cp_text[CPT_KIND];
1623 compl_match_array[i].pum_info = compl->cp_text[CPT_INFO];
Girish Palya19ef6b02025-05-28 20:28:21 +02001624 compl_match_array[i].pum_cpt_source_idx = compl->cp_cpt_source_idx;
glepnira49c0772024-11-30 10:56:30 +01001625 compl_match_array[i].pum_user_abbr_hlattr = compl->cp_user_abbr_hlattr;
1626 compl_match_array[i].pum_user_kind_hlattr = compl->cp_user_kind_hlattr;
glepnir6e199932024-12-14 21:13:27 +01001627 compl_match_array[i++].pum_extra = compl->cp_text[CPT_MENU] != NULL
1628 ? compl->cp_text[CPT_MENU] : compl->cp_fname;
glepnir80b66202024-11-27 21:53:53 +01001629 match_next = compl->cp_match_next;
1630 compl->cp_match_next = NULL;
1631 compl = match_next;
1632 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001633
1634 if (!shown_match_ok) // no displayed match at all
1635 cur = -1;
1636
1637 return cur;
1638}
1639
1640/*
1641 * Show the popup menu for the list of matches.
1642 * Also adjusts "compl_shown_match" to an entry that is actually displayed.
1643 */
1644 void
1645ins_compl_show_pum(void)
1646{
1647 int i;
1648 int cur = -1;
1649 colnr_T col;
1650
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001651 if (!pum_wanted() || !pum_enough_matches())
1652 return;
1653
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001654 // Update the screen later, before drawing the popup menu over it.
1655 pum_call_update_screen();
1656
1657 if (compl_match_array == NULL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001658 // Need to build the popup menu list.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001659 cur = ins_compl_build_pum();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001660 else
1661 {
1662 // popup menu already exists, only need to find the current item.
1663 for (i = 0; i < compl_match_arraysize; ++i)
glepnir19e1dd62025-05-08 22:50:38 +02001664 {
John Marriott5e6ea922024-11-23 14:01:57 +01001665 if (compl_match_array[i].pum_text == compl_shown_match->cp_str.string
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001666 || compl_match_array[i].pum_text
1667 == compl_shown_match->cp_text[CPT_ABBR])
1668 {
1669 cur = i;
1670 break;
1671 }
glepnir19e1dd62025-05-08 22:50:38 +02001672 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001673 }
1674
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001675 if (compl_match_array == NULL)
glepnir0d3c0a62024-02-11 17:52:40 +01001676 {
1677#ifdef FEAT_EVAL
1678 if (compl_started && has_completechanged())
1679 trigger_complete_changed_event(cur);
1680#endif
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001681 return;
glepnir0d3c0a62024-02-11 17:52:40 +01001682 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001683
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001684 // In Replace mode when a $ is displayed at the end of the line only
1685 // part of the screen would be updated. We do need to redraw here.
1686 dollar_vcol = -1;
1687
1688 // Compute the screen column of the start of the completed text.
1689 // Use the cursor to get all wrapping and other settings right.
1690 col = curwin->w_cursor.col;
1691 curwin->w_cursor.col = compl_col;
glepnira218cc62024-06-03 19:32:39 +02001692 compl_selected_item = cur;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001693 pum_display(compl_match_array, compl_match_arraysize, cur);
1694 curwin->w_cursor.col = col;
Bram Moolenaard7f246c2019-04-08 18:15:41 +02001695
glepnircbb46b42024-02-03 18:11:13 +01001696 // After adding leader, set the current match to shown match.
1697 if (compl_started && compl_curr_match != compl_shown_match)
1698 compl_curr_match = compl_shown_match;
1699
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02001700#ifdef FEAT_EVAL
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001701 if (has_completechanged())
1702 trigger_complete_changed_event(cur);
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02001703#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001704}
1705
1706#define DICT_FIRST (1) // use just first element in "dict"
1707#define DICT_EXACT (2) // "dict" is the exact name of a file
1708
1709/*
glepnir40c1c332024-06-11 19:37:04 +02001710 * Get current completion leader
1711 */
1712 char_u *
1713ins_compl_leader(void)
1714{
John Marriott5e6ea922024-11-23 14:01:57 +01001715 return compl_leader.string != NULL ? compl_leader.string : compl_orig_text.string;
1716}
1717
1718/*
1719 * Get current completion leader length
1720 */
1721 size_t
1722ins_compl_leader_len(void)
1723{
1724 return compl_leader.string != NULL ? compl_leader.length : compl_orig_text.length;
glepnir40c1c332024-06-11 19:37:04 +02001725}
1726
1727/*
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00001728 * Add any identifiers that match the given pattern "pat" in the list of
1729 * dictionary files "dict_start" to the list of completions.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001730 */
1731 static void
1732ins_compl_dictionaries(
1733 char_u *dict_start,
1734 char_u *pat,
1735 int flags, // DICT_FIRST and/or DICT_EXACT
1736 int thesaurus) // Thesaurus completion
1737{
1738 char_u *dict = dict_start;
1739 char_u *ptr;
1740 char_u *buf;
1741 regmatch_T regmatch;
1742 char_u **files;
1743 int count;
1744 int save_p_scs;
1745 int dir = compl_direction;
1746
1747 if (*dict == NUL)
1748 {
1749#ifdef FEAT_SPELL
1750 // When 'dictionary' is empty and spell checking is enabled use
1751 // "spell".
1752 if (!thesaurus && curwin->w_p_spell)
1753 dict = (char_u *)"spell";
1754 else
1755#endif
1756 return;
1757 }
1758
1759 buf = alloc(LSIZE);
1760 if (buf == NULL)
1761 return;
1762 regmatch.regprog = NULL; // so that we can goto theend
1763
1764 // If 'infercase' is set, don't use 'smartcase' here
1765 save_p_scs = p_scs;
1766 if (curbuf->b_p_inf)
1767 p_scs = FALSE;
1768
1769 // When invoked to match whole lines for CTRL-X CTRL-L adjust the pattern
1770 // to only match at the start of a line. Otherwise just match the
1771 // pattern. Also need to double backslashes.
1772 if (ctrl_x_mode_line_or_eval())
1773 {
1774 char_u *pat_esc = vim_strsave_escaped(pat, (char_u *)"\\");
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001775
1776 if (pat_esc == NULL)
1777 goto theend;
glepnir19e1dd62025-05-08 22:50:38 +02001778 size_t len = STRLEN(pat_esc) + 10;
Bram Moolenaar964b3742019-05-24 18:54:09 +02001779 ptr = alloc(len);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001780 if (ptr == NULL)
1781 {
1782 vim_free(pat_esc);
1783 goto theend;
1784 }
1785 vim_snprintf((char *)ptr, len, "^\\s*\\zs\\V%s", pat_esc);
1786 regmatch.regprog = vim_regcomp(ptr, RE_MAGIC);
1787 vim_free(pat_esc);
1788 vim_free(ptr);
1789 }
1790 else
1791 {
Bram Moolenaarf4e20992020-12-21 19:59:08 +01001792 regmatch.regprog = vim_regcomp(pat, magic_isset() ? RE_MAGIC : 0);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001793 if (regmatch.regprog == NULL)
1794 goto theend;
1795 }
1796
1797 // ignore case depends on 'ignorecase', 'smartcase' and "pat"
1798 regmatch.rm_ic = ignorecase(pat);
1799 while (*dict != NUL && !got_int && !compl_interrupted)
1800 {
1801 // copy one dictionary file name into buf
1802 if (flags == DICT_EXACT)
1803 {
1804 count = 1;
1805 files = &dict;
1806 }
1807 else
1808 {
1809 // Expand wildcards in the dictionary name, but do not allow
1810 // backticks (for security, the 'dict' option may have been set in
1811 // a modeline).
1812 copy_option_part(&dict, buf, LSIZE, ",");
1813# ifdef FEAT_SPELL
1814 if (!thesaurus && STRCMP(buf, "spell") == 0)
1815 count = -1;
1816 else
1817# endif
1818 if (vim_strchr(buf, '`') != NULL
1819 || expand_wildcards(1, &buf, &count, &files,
1820 EW_FILE|EW_SILENT) != OK)
1821 count = 0;
1822 }
1823
1824# ifdef FEAT_SPELL
1825 if (count == -1)
1826 {
1827 // Complete from active spelling. Skip "\<" in the pattern, we
1828 // don't use it as a RE.
1829 if (pat[0] == '\\' && pat[1] == '<')
1830 ptr = pat + 2;
1831 else
1832 ptr = pat;
1833 spell_dump_compl(ptr, regmatch.rm_ic, &dir, 0);
1834 }
1835 else
1836# endif
1837 if (count > 0) // avoid warning for using "files" uninit
1838 {
1839 ins_compl_files(count, files, thesaurus, flags,
glepnirf31cfa22025-03-06 21:59:13 +01001840 (cfc_has_mode() ? NULL : &regmatch), buf, &dir);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001841 if (flags != DICT_EXACT)
1842 FreeWild(count, files);
1843 }
1844 if (flags != 0)
1845 break;
1846 }
1847
1848theend:
1849 p_scs = save_p_scs;
1850 vim_regfree(regmatch.regprog);
1851 vim_free(buf);
1852}
1853
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00001854/*
1855 * Add all the words in the line "*buf_arg" from the thesaurus file "fname"
1856 * skipping the word at 'skip_word'. Returns OK on success.
1857 */
1858 static int
zeertzjq5fb3aab2022-08-24 16:48:23 +01001859thesaurus_add_words_in_line(
glepnir19e1dd62025-05-08 22:50:38 +02001860 char_u *fname,
1861 char_u **buf_arg,
1862 int dir,
1863 char_u *skip_word)
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00001864{
1865 int status = OK;
1866 char_u *ptr;
1867 char_u *wstart;
1868
1869 // Add the other matches on the line
1870 ptr = *buf_arg;
1871 while (!got_int)
1872 {
1873 // Find start of the next word. Skip white
1874 // space and punctuation.
1875 ptr = find_word_start(ptr);
1876 if (*ptr == NUL || *ptr == NL)
1877 break;
1878 wstart = ptr;
1879
1880 // Find end of the word.
1881 if (has_mbyte)
1882 // Japanese words may have characters in
1883 // different classes, only separate words
1884 // with single-byte non-word characters.
1885 while (*ptr != NUL)
1886 {
1887 int l = (*mb_ptr2len)(ptr);
1888
1889 if (l < 2 && !vim_iswordc(*ptr))
1890 break;
1891 ptr += l;
1892 }
1893 else
1894 ptr = find_word_end(ptr);
1895
1896 // Add the word. Skip the regexp match.
1897 if (wstart != skip_word)
1898 {
1899 status = ins_compl_add_infercase(wstart, (int)(ptr - wstart), p_ic,
glepnirf31cfa22025-03-06 21:59:13 +01001900 fname, dir, FALSE, 0);
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00001901 if (status == FAIL)
1902 break;
1903 }
1904 }
1905
1906 *buf_arg = ptr;
1907 return status;
1908}
1909
1910/*
1911 * Process "count" dictionary/thesaurus "files" and add the text matching
1912 * "regmatch".
1913 */
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001914 static void
1915ins_compl_files(
1916 int count,
1917 char_u **files,
1918 int thesaurus,
1919 int flags,
1920 regmatch_T *regmatch,
1921 char_u *buf,
1922 int *dir)
1923{
1924 char_u *ptr;
1925 int i;
1926 FILE *fp;
1927 int add_r;
glepnirf31cfa22025-03-06 21:59:13 +01001928 char_u *leader = NULL;
1929 int leader_len = 0;
glepnir58760162025-03-13 21:39:51 +01001930 int in_fuzzy_collect = cfc_has_mode();
glepnirf31cfa22025-03-06 21:59:13 +01001931 int score = 0;
1932 int len = 0;
1933 char_u *line_end = NULL;
1934
1935 if (in_fuzzy_collect)
1936 {
1937 leader = ins_compl_leader();
Yegappan Lakshmanan7b6add02025-04-01 20:38:37 +02001938 leader_len = (int)ins_compl_leader_len();
glepnirf31cfa22025-03-06 21:59:13 +01001939 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001940
1941 for (i = 0; i < count && !got_int && !compl_interrupted; i++)
1942 {
1943 fp = mch_fopen((char *)files[i], "r"); // open dictionary file
=?UTF-8?q?Bj=C3=B6rn=20Linse?=91ccbad2022-10-13 12:51:13 +01001944 if (flags != DICT_EXACT && !shortmess(SHM_COMPLETIONSCAN))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001945 {
Bram Moolenaarcc233582020-12-12 13:32:07 +01001946 msg_hist_off = TRUE; // reset in msg_trunc_attr()
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001947 vim_snprintf((char *)IObuff, IOSIZE,
1948 _("Scanning dictionary: %s"), (char *)files[i]);
1949 (void)msg_trunc_attr((char *)IObuff, TRUE, HL_ATTR(HLF_R));
1950 }
1951
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001952 if (fp == NULL)
1953 continue;
1954
1955 // Read dictionary file line by line.
1956 // Check each line for a match.
1957 while (!got_int && !compl_interrupted && !vim_fgets(buf, LSIZE, fp))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001958 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00001959 ptr = buf;
glepnirf31cfa22025-03-06 21:59:13 +01001960 if (regmatch != NULL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001961 {
glepnirf31cfa22025-03-06 21:59:13 +01001962 while (vim_regexec(regmatch, buf, (colnr_T)(ptr - buf)))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001963 {
glepnirf31cfa22025-03-06 21:59:13 +01001964 ptr = regmatch->startp[0];
1965 ptr = ctrl_x_mode_line_or_eval() ? find_line_end(ptr)
1966 : find_word_end(ptr);
1967 add_r = ins_compl_add_infercase(regmatch->startp[0],
1968 (int)(ptr - regmatch->startp[0]),
1969 p_ic, files[i], *dir, FALSE, 0);
1970 if (thesaurus)
1971 {
1972 // For a thesaurus, add all the words in the line
1973 ptr = buf;
1974 add_r = thesaurus_add_words_in_line(files[i], &ptr, *dir,
1975 regmatch->startp[0]);
1976 }
1977 if (add_r == OK)
1978 // if dir was BACKWARD then honor it just once
1979 *dir = FORWARD;
1980 else if (add_r == FAIL)
1981 break;
1982 // avoid expensive call to vim_regexec() when at end
1983 // of line
1984 if (*ptr == '\n' || got_int)
1985 break;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001986 }
glepnirf31cfa22025-03-06 21:59:13 +01001987 }
1988 else if (in_fuzzy_collect && leader_len > 0)
1989 {
1990 line_end = find_line_end(ptr);
1991 while (ptr < line_end)
1992 {
1993 if (fuzzy_match_str_in_line(&ptr, leader, &len, NULL, &score))
1994 {
1995 char_u *end_ptr = ctrl_x_mode_line_or_eval()
1996 ? find_line_end(ptr) : find_word_end(ptr);
1997 add_r = ins_compl_add_infercase(ptr, (int)(end_ptr - ptr),
1998 p_ic, files[i], *dir, FALSE, score);
1999 if (add_r == FAIL)
2000 break;
2001 ptr = end_ptr; // start from next word
2002 if (compl_get_longest && ctrl_x_mode_normal()
2003 && compl_first_match->cp_next
2004 && score == compl_first_match->cp_next->cp_score)
2005 compl_num_bests++;
2006 }
glepnirf31cfa22025-03-06 21:59:13 +01002007 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002008 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00002009 line_breakcheck();
2010 ins_compl_check_keys(50, FALSE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002011 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00002012 fclose(fp);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002013 }
2014}
2015
2016/*
2017 * Find the start of the next word.
2018 * Returns a pointer to the first char of the word. Also stops at a NUL.
2019 */
2020 char_u *
2021find_word_start(char_u *ptr)
2022{
2023 if (has_mbyte)
glepnir19e1dd62025-05-08 22:50:38 +02002024 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002025 while (*ptr != NUL && *ptr != '\n' && mb_get_class(ptr) <= 1)
2026 ptr += (*mb_ptr2len)(ptr);
glepnir19e1dd62025-05-08 22:50:38 +02002027 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002028 else
glepnir19e1dd62025-05-08 22:50:38 +02002029 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002030 while (*ptr != NUL && *ptr != '\n' && !vim_iswordc(*ptr))
2031 ++ptr;
glepnir19e1dd62025-05-08 22:50:38 +02002032 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002033 return ptr;
2034}
2035
2036/*
2037 * Find the end of the word. Assumes it starts inside a word.
2038 * Returns a pointer to just after the word.
2039 */
2040 char_u *
2041find_word_end(char_u *ptr)
2042{
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002043 if (has_mbyte)
2044 {
glepnir19e1dd62025-05-08 22:50:38 +02002045 int start_class = mb_get_class(ptr);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002046 if (start_class > 1)
2047 while (*ptr != NUL)
2048 {
2049 ptr += (*mb_ptr2len)(ptr);
2050 if (mb_get_class(ptr) != start_class)
2051 break;
2052 }
2053 }
2054 else
2055 while (vim_iswordc(*ptr))
2056 ++ptr;
2057 return ptr;
2058}
2059
2060/*
2061 * Find the end of the line, omitting CR and NL at the end.
2062 * Returns a pointer to just after the line.
2063 */
glepnirdd42b052025-03-08 16:52:55 +01002064 char_u *
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002065find_line_end(char_u *ptr)
2066{
glepnir19e1dd62025-05-08 22:50:38 +02002067 char_u *s = ptr + STRLEN(ptr);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002068 while (s > ptr && (s[-1] == CAR || s[-1] == NL))
2069 --s;
2070 return s;
2071}
2072
2073/*
Girish Palyacbe53192025-04-14 22:13:15 +02002074 * Free a completion item in the list
2075 */
2076 static void
2077ins_compl_item_free(compl_T *match)
2078{
Girish Palyacbe53192025-04-14 22:13:15 +02002079 VIM_CLEAR_STRING(match->cp_str);
2080 // several entries may use the same fname, free it just once.
2081 if (match->cp_flags & CP_FREE_FNAME)
2082 vim_free(match->cp_fname);
glepnir19e1dd62025-05-08 22:50:38 +02002083 for (int i = 0; i < CPT_COUNT; ++i)
Girish Palyacbe53192025-04-14 22:13:15 +02002084 vim_free(match->cp_text[i]);
2085#ifdef FEAT_EVAL
2086 clear_tv(&match->cp_user_data);
2087#endif
2088 vim_free(match);
2089}
2090
2091/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002092 * Free the list of completions
2093 */
2094 static void
2095ins_compl_free(void)
2096{
2097 compl_T *match;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002098
John Marriott5e6ea922024-11-23 14:01:57 +01002099 VIM_CLEAR_STRING(compl_pattern);
2100 VIM_CLEAR_STRING(compl_leader);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002101
2102 if (compl_first_match == NULL)
2103 return;
2104
2105 ins_compl_del_pum();
2106 pum_clear();
2107
2108 compl_curr_match = compl_first_match;
2109 do
2110 {
2111 match = compl_curr_match;
2112 compl_curr_match = compl_curr_match->cp_next;
Girish Palyacbe53192025-04-14 22:13:15 +02002113 ins_compl_item_free(match);
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002114 } while (compl_curr_match != NULL && !is_first_match(compl_curr_match));
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002115 compl_first_match = compl_curr_match = NULL;
2116 compl_shown_match = NULL;
2117 compl_old_match = NULL;
2118}
2119
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002120/*
2121 * Reset/clear the completion state.
2122 */
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002123 void
2124ins_compl_clear(void)
2125{
2126 compl_cont_status = 0;
2127 compl_started = FALSE;
glepnirf31cfa22025-03-06 21:59:13 +01002128 compl_cfc_longest_ins = FALSE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002129 compl_matches = 0;
glepnir07f0dbe2025-02-18 20:27:30 +01002130 compl_selected_item = -1;
glepnir6a38aff2024-12-16 21:56:16 +01002131 compl_ins_end_col = 0;
glepnircf7f0122025-04-15 19:02:00 +02002132 compl_curr_win = NULL;
2133 compl_curr_buf = NULL;
John Marriott5e6ea922024-11-23 14:01:57 +01002134 VIM_CLEAR_STRING(compl_pattern);
2135 VIM_CLEAR_STRING(compl_leader);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002136 edit_submode_extra = NULL;
John Marriott5e6ea922024-11-23 14:01:57 +01002137 VIM_CLEAR_STRING(compl_orig_text);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002138 compl_enter_selects = FALSE;
Girish Palya0ac1eb32025-04-16 20:18:33 +02002139 cpt_sources_clear();
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02002140#ifdef FEAT_EVAL
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002141 // clear v:completed_item
2142 set_vim_var_dict(VV_COMPLETED_ITEM, dict_alloc_lock(VAR_FIXED));
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02002143#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002144}
2145
2146/*
2147 * Return TRUE when Insert completion is active.
2148 */
2149 int
2150ins_compl_active(void)
2151{
2152 return compl_started;
2153}
2154
2155/*
glepnir8d0bb6d2024-12-24 09:44:35 +01002156 * Return True when wp is the actual completion window
2157 */
2158 int
glepnircf7f0122025-04-15 19:02:00 +02002159ins_compl_win_active(win_T *wp)
glepnir8d0bb6d2024-12-24 09:44:35 +01002160{
glepnircf7f0122025-04-15 19:02:00 +02002161 return ins_compl_active() && wp == compl_curr_win
2162 && wp->w_buffer == compl_curr_buf;
glepnir8d0bb6d2024-12-24 09:44:35 +01002163}
2164
2165/*
Girish Palyacbe53192025-04-14 22:13:15 +02002166 * Selected one of the matches. When FALSE, the match was either edited or
2167 * using the longest common string.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002168 */
2169 int
2170ins_compl_used_match(void)
2171{
2172 return compl_used_match;
2173}
2174
2175/*
2176 * Initialize get longest common string.
2177 */
2178 void
2179ins_compl_init_get_longest(void)
2180{
2181 compl_get_longest = FALSE;
2182}
2183
2184/*
2185 * Returns TRUE when insert completion is interrupted.
2186 */
2187 int
2188ins_compl_interrupted(void)
2189{
2190 return compl_interrupted;
2191}
2192
2193/*
2194 * Returns TRUE if the <Enter> key selects a match in the completion popup
2195 * menu.
2196 */
2197 int
2198ins_compl_enter_selects(void)
2199{
2200 return compl_enter_selects;
2201}
2202
2203/*
2204 * Return the column where the text starts that is being completed
2205 */
2206 colnr_T
2207ins_compl_col(void)
2208{
2209 return compl_col;
2210}
2211
2212/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002213 * Return the length in bytes of the text being completed
2214 */
2215 int
2216ins_compl_len(void)
2217{
2218 return compl_length;
2219}
2220
2221/*
glepnir94a045e2025-03-01 16:12:23 +01002222 * Return TRUE when the 'completeopt' "preinsert" flag is in effect,
2223 * otherwise return FALSE.
glepniredd4ac32025-01-29 18:53:51 +01002224 */
2225 static int
2226ins_compl_has_preinsert(void)
2227{
glepnira2c55592025-02-28 17:43:42 +01002228 int cur_cot_flags = get_cot_flags();
glepnir94a045e2025-03-01 16:12:23 +01002229 return (cur_cot_flags & (COT_PREINSERT | COT_FUZZY | COT_MENUONE))
2230 == (COT_PREINSERT | COT_MENUONE);
glepniredd4ac32025-01-29 18:53:51 +01002231}
2232
2233/*
2234 * Returns TRUE if the pre-insert effect is valid and the cursor is within
2235 * the `compl_ins_end_col` range.
2236 */
2237 int
2238ins_compl_preinsert_effect(void)
2239{
2240 if (!ins_compl_has_preinsert())
2241 return FALSE;
2242
2243 return curwin->w_cursor.col < compl_ins_end_col;
2244}
2245
2246/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002247 * Delete one character before the cursor and show the subset of the matches
2248 * that match the word that is now before the cursor.
2249 * Returns the character to be used, NUL if the work is done and another char
2250 * to be got from the user.
2251 */
2252 int
2253ins_compl_bs(void)
2254{
2255 char_u *line;
2256 char_u *p;
2257
glepniredd4ac32025-01-29 18:53:51 +01002258 if (ins_compl_preinsert_effect())
2259 ins_compl_delete();
2260
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002261 line = ml_get_curline();
2262 p = line + curwin->w_cursor.col;
2263 MB_PTR_BACK(line, p);
2264
2265 // Stop completion when the whole word was deleted. For Omni completion
2266 // allow the word to be deleted, we won't match everything.
2267 // Respect the 'backspace' option.
2268 if ((int)(p - line) - (int)compl_col < 0
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002269 || ((int)(p - line) - (int)compl_col == 0 && !ctrl_x_mode_omni())
2270 || ctrl_x_mode_eval()
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002271 || (!can_bs(BS_START) && (int)(p - line) - (int)compl_col
2272 - compl_length < 0))
2273 return K_BS;
2274
2275 // Deleted more than what was used to find matches or didn't finish
2276 // finding all matches: need to look for matches all over again.
2277 if (curwin->w_cursor.col <= compl_col + compl_length
2278 || ins_compl_need_restart())
2279 ins_compl_restart();
2280
John Marriott5e6ea922024-11-23 14:01:57 +01002281 VIM_CLEAR_STRING(compl_leader);
2282 compl_leader.length = (size_t)((p - line) - compl_col);
2283 compl_leader.string = vim_strnsave(line + compl_col, compl_leader.length);
2284 if (compl_leader.string == NULL)
2285 {
2286 compl_leader.length = 0;
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00002287 return K_BS;
John Marriott5e6ea922024-11-23 14:01:57 +01002288 }
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00002289
2290 ins_compl_new_leader();
2291 if (compl_shown_match != NULL)
2292 // Make sure current match is not a hidden item.
2293 compl_curr_match = compl_shown_match;
2294 return NUL;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002295}
2296
2297/*
2298 * Return TRUE when we need to find matches again, ins_compl_restart() is to
2299 * be called.
2300 */
2301 static int
2302ins_compl_need_restart(void)
2303{
2304 // Return TRUE if we didn't complete finding matches or when the
2305 // 'completefunc' returned "always" in the "refresh" dictionary item.
2306 return compl_was_interrupted
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002307 || ((ctrl_x_mode_function() || ctrl_x_mode_omni())
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002308 && compl_opt_refresh_always);
2309}
2310
2311/*
2312 * Called after changing "compl_leader".
2313 * Show the popup menu with a different set of matches.
2314 * May also search for matches again if the previous search was interrupted.
2315 */
2316 static void
2317ins_compl_new_leader(void)
2318{
glepnirecf8f152025-06-10 20:52:41 +02002319 int cur_cot_flags = get_cot_flags();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002320 ins_compl_del_pum();
2321 ins_compl_delete();
glepnir6a38aff2024-12-16 21:56:16 +01002322 ins_compl_insert_bytes(compl_leader.string + get_compl_len(), -1);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002323 compl_used_match = FALSE;
2324
2325 if (compl_started)
Girish Palyacbe53192025-04-14 22:13:15 +02002326 {
John Marriott5e6ea922024-11-23 14:01:57 +01002327 ins_compl_set_original_text(compl_leader.string, compl_leader.length);
Girish Palyacbe53192025-04-14 22:13:15 +02002328 if (is_cpt_func_refresh_always())
2329 cpt_compl_refresh();
2330 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002331 else
2332 {
2333#ifdef FEAT_SPELL
2334 spell_bad_len = 0; // need to redetect bad word
2335#endif
Bram Moolenaar32aa1022019-11-02 22:54:41 +01002336 // Matches were cleared, need to search for them now. Before drawing
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002337 // the popup menu display the changed text before the cursor. Set
2338 // "compl_restarting" to avoid that the first match is inserted.
2339 pum_call_update_screen();
2340#ifdef FEAT_GUI
2341 if (gui.in_use)
2342 {
2343 // Show the cursor after the match, not after the redrawn text.
2344 setcursor();
2345 out_flush_cursor(FALSE, FALSE);
2346 }
2347#endif
2348 compl_restarting = TRUE;
2349 if (ins_complete(Ctrl_N, TRUE) == FAIL)
2350 compl_cont_status = 0;
2351 compl_restarting = FALSE;
2352 }
2353
glepnirecf8f152025-06-10 20:52:41 +02002354 // When 'cot' contains "fuzzy" set the cp_score and maybe sort
2355 if (cur_cot_flags & COT_FUZZY)
Girish Palyab8ee1cf2025-06-08 16:20:06 +02002356 {
glepnirecf8f152025-06-10 20:52:41 +02002357 set_fuzzy_score();
2358 // Sort the matches linked list based on fuzzy score
2359 if (!(cur_cot_flags & COT_NOSORT))
Girish Palyab8ee1cf2025-06-08 16:20:06 +02002360 {
glepnirecf8f152025-06-10 20:52:41 +02002361 sort_compl_match_list(cp_compare_fuzzy);
2362 if ((cur_cot_flags & (COT_NOINSERT | COT_NOSELECT)) == COT_NOINSERT
2363 && compl_first_match)
2364 {
2365 compl_shown_match = compl_first_match;
2366 if (compl_shows_dir_forward())
2367 compl_shown_match = compl_first_match->cp_next;
2368 }
Girish Palyab8ee1cf2025-06-08 16:20:06 +02002369 }
2370 }
2371
glepnir44180412025-02-20 22:09:48 +01002372 compl_enter_selects = !compl_used_match && compl_selected_item != -1;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002373
2374 // Show the popup menu with a different set of matches.
2375 ins_compl_show_pum();
2376
2377 // Don't let Enter select the original text when there is no popup menu.
2378 if (compl_match_array == NULL)
2379 compl_enter_selects = FALSE;
glepniredd4ac32025-01-29 18:53:51 +01002380 else if (ins_compl_has_preinsert() && compl_leader.length > 0)
Girish Palya5fbe72e2025-06-18 19:15:45 +02002381 ins_compl_insert(TRUE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002382}
2383
2384/*
2385 * Return the length of the completion, from the completion start column to
2386 * the cursor column. Making sure it never goes below zero.
2387 */
2388 static int
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002389get_compl_len(void)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002390{
2391 int off = (int)curwin->w_cursor.col - (int)compl_col;
glepnir40891ba2025-02-10 22:18:00 +01002392 return MAX(0, off);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002393}
2394
2395/*
2396 * Append one character to the match leader. May reduce the number of
2397 * matches.
2398 */
2399 void
2400ins_compl_addleader(int c)
2401{
glepnir19e1dd62025-05-08 22:50:38 +02002402 int cc;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002403
glepniredd4ac32025-01-29 18:53:51 +01002404 if (ins_compl_preinsert_effect())
2405 ins_compl_delete();
2406
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002407 if (stop_arrow() == FAIL)
2408 return;
2409 if (has_mbyte && (cc = (*mb_char2len)(c)) > 1)
2410 {
2411 char_u buf[MB_MAXBYTES + 1];
2412
2413 (*mb_char2bytes)(c, buf);
2414 buf[cc] = NUL;
2415 ins_char_bytes(buf, cc);
2416 if (compl_opt_refresh_always)
2417 AppendToRedobuff(buf);
2418 }
2419 else
2420 {
2421 ins_char(c);
2422 if (compl_opt_refresh_always)
2423 AppendCharToRedobuff(c);
2424 }
2425
2426 // If we didn't complete finding matches we must search again.
2427 if (ins_compl_need_restart())
2428 ins_compl_restart();
2429
2430 // When 'always' is set, don't reset compl_leader. While completing,
2431 // cursor doesn't point original position, changing compl_leader would
2432 // break redo.
2433 if (!compl_opt_refresh_always)
2434 {
John Marriott5e6ea922024-11-23 14:01:57 +01002435 VIM_CLEAR_STRING(compl_leader);
2436 compl_leader.length = (size_t)(curwin->w_cursor.col - compl_col);
2437 compl_leader.string = vim_strnsave(ml_get_curline() + compl_col,
2438 compl_leader.length);
2439 if (compl_leader.string == NULL)
2440 {
2441 compl_leader.length = 0;
2442 return;
2443 }
2444
2445 ins_compl_new_leader();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002446 }
2447}
2448
2449/*
2450 * Setup for finding completions again without leaving CTRL-X mode. Used when
2451 * BS or a key was typed while still searching for matches.
2452 */
2453 static void
2454ins_compl_restart(void)
2455{
2456 ins_compl_free();
2457 compl_started = FALSE;
2458 compl_matches = 0;
2459 compl_cont_status = 0;
2460 compl_cont_mode = 0;
Girish Palya0ac1eb32025-04-16 20:18:33 +02002461 cpt_sources_clear();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002462}
2463
2464/*
2465 * Set the first match, the original text.
2466 */
2467 static void
John Marriott5e6ea922024-11-23 14:01:57 +01002468ins_compl_set_original_text(char_u *str, size_t len)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002469{
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002470 // Replace the original text entry.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00002471 // The CP_ORIGINAL_TEXT flag is either at the first item or might possibly
2472 // be at the last item for backward completion
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002473 if (match_at_original_text(compl_first_match)) // safety check
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002474 {
John Marriott5e6ea922024-11-23 14:01:57 +01002475 char_u *p = vim_strnsave(str, len);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002476 if (p != NULL)
2477 {
John Marriott5e6ea922024-11-23 14:01:57 +01002478 VIM_CLEAR_STRING(compl_first_match->cp_str);
2479 compl_first_match->cp_str.string = p;
2480 compl_first_match->cp_str.length = len;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002481 }
2482 }
2483 else if (compl_first_match->cp_prev != NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002484 && match_at_original_text(compl_first_match->cp_prev))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002485 {
John Marriott5e6ea922024-11-23 14:01:57 +01002486 char_u *p = vim_strnsave(str, len);
2487 if (p != NULL)
2488 {
2489 VIM_CLEAR_STRING(compl_first_match->cp_prev->cp_str);
2490 compl_first_match->cp_prev->cp_str.string = p;
2491 compl_first_match->cp_prev->cp_str.length = len;
2492 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002493 }
2494}
2495
2496/*
2497 * Append one character to the match leader. May reduce the number of
2498 * matches.
2499 */
2500 void
2501ins_compl_addfrommatch(void)
2502{
2503 char_u *p;
2504 int len = (int)curwin->w_cursor.col - (int)compl_col;
2505 int c;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002506
John Marriott5e6ea922024-11-23 14:01:57 +01002507 p = compl_shown_match->cp_str.string;
2508 if ((int)compl_shown_match->cp_str.length <= len) // the match is too short
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002509 {
John Marriott5e6ea922024-11-23 14:01:57 +01002510 size_t plen;
2511 compl_T *cp;
2512
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002513 // When still at the original match use the first entry that matches
2514 // the leader.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002515 if (!match_at_original_text(compl_shown_match))
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00002516 return;
2517
2518 p = NULL;
John Marriott5e6ea922024-11-23 14:01:57 +01002519 plen = 0;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00002520 for (cp = compl_shown_match->cp_next; cp != NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002521 && !is_first_match(cp); cp = cp->cp_next)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002522 {
John Marriott5e6ea922024-11-23 14:01:57 +01002523 if (compl_leader.string == NULL
2524 || ins_compl_equal(cp, compl_leader.string,
2525 (int)compl_leader.length))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002526 {
John Marriott5e6ea922024-11-23 14:01:57 +01002527 p = cp->cp_str.string;
2528 plen = cp->cp_str.length;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00002529 break;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002530 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002531 }
John Marriott5e6ea922024-11-23 14:01:57 +01002532 if (p == NULL || (int)plen <= len)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002533 return;
2534 }
2535 p += len;
2536 c = PTR2CHAR(p);
2537 ins_compl_addleader(c);
2538}
2539
2540/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002541 * Set the CTRL-X completion mode based on the key "c" typed after a CTRL-X.
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002542 * Uses the global variables: ctrl_x_mode, edit_submode, edit_submode_pre,
2543 * compl_cont_mode and compl_cont_status.
2544 * Returns TRUE when the character is not to be inserted.
2545 */
2546 static int
2547set_ctrl_x_mode(int c)
2548{
glepnir05460682025-05-26 18:23:27 +02002549 int retval = FALSE;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002550
2551 switch (c)
2552 {
2553 case Ctrl_E:
2554 case Ctrl_Y:
2555 // scroll the window one line up or down
2556 ctrl_x_mode = CTRL_X_SCROLL;
2557 if (!(State & REPLACE_FLAG))
2558 edit_submode = (char_u *)_(" (insert) Scroll (^E/^Y)");
2559 else
2560 edit_submode = (char_u *)_(" (replace) Scroll (^E/^Y)");
2561 edit_submode_pre = NULL;
2562 showmode();
2563 break;
2564 case Ctrl_L:
2565 // complete whole line
2566 ctrl_x_mode = CTRL_X_WHOLE_LINE;
2567 break;
2568 case Ctrl_F:
2569 // complete filenames
2570 ctrl_x_mode = CTRL_X_FILES;
2571 break;
2572 case Ctrl_K:
zeertzjq5fb3aab2022-08-24 16:48:23 +01002573 // complete words from a dictionary
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002574 ctrl_x_mode = CTRL_X_DICTIONARY;
2575 break;
2576 case Ctrl_R:
glepnir05460682025-05-26 18:23:27 +02002577 // When CTRL-R is followed by '=', don't trigger register completion
2578 // This allows expressions like <C-R>=func()<CR> to work normally
2579 if (vpeekc() == '=')
2580 break;
2581 ctrl_x_mode = CTRL_X_REGISTER;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002582 break;
2583 case Ctrl_T:
2584 // complete words from a thesaurus
2585 ctrl_x_mode = CTRL_X_THESAURUS;
2586 break;
2587#ifdef FEAT_COMPL_FUNC
2588 case Ctrl_U:
2589 // user defined completion
2590 ctrl_x_mode = CTRL_X_FUNCTION;
2591 break;
2592 case Ctrl_O:
2593 // omni completion
2594 ctrl_x_mode = CTRL_X_OMNI;
2595 break;
2596#endif
2597 case 's':
2598 case Ctrl_S:
2599 // complete spelling suggestions
2600 ctrl_x_mode = CTRL_X_SPELL;
2601#ifdef FEAT_SPELL
2602 ++emsg_off; // Avoid getting the E756 error twice.
2603 spell_back_to_badword();
2604 --emsg_off;
2605#endif
2606 break;
2607 case Ctrl_RSB:
2608 // complete tag names
2609 ctrl_x_mode = CTRL_X_TAGS;
2610 break;
2611#ifdef FEAT_FIND_ID
2612 case Ctrl_I:
2613 case K_S_TAB:
2614 // complete keywords from included files
2615 ctrl_x_mode = CTRL_X_PATH_PATTERNS;
2616 break;
2617 case Ctrl_D:
2618 // complete definitions from included files
2619 ctrl_x_mode = CTRL_X_PATH_DEFINES;
2620 break;
2621#endif
2622 case Ctrl_V:
2623 case Ctrl_Q:
2624 // complete vim commands
2625 ctrl_x_mode = CTRL_X_CMDLINE;
2626 break;
2627 case Ctrl_Z:
2628 // stop completion
2629 ctrl_x_mode = CTRL_X_NORMAL;
2630 edit_submode = NULL;
2631 showmode();
2632 retval = TRUE;
2633 break;
2634 case Ctrl_P:
2635 case Ctrl_N:
2636 // ^X^P means LOCAL expansion if nothing interrupted (eg we
2637 // just started ^X mode, or there were enough ^X's to cancel
2638 // the previous mode, say ^X^F^X^X^P or ^P^X^X^X^P, see below)
2639 // do normal expansion when interrupting a different mode (say
2640 // ^X^F^X^P or ^P^X^X^P, see below)
2641 // nothing changes if interrupting mode 0, (eg, the flag
2642 // doesn't change when going to ADDING mode -- Acevedo
2643 if (!(compl_cont_status & CONT_INTRPT))
2644 compl_cont_status |= CONT_LOCAL;
2645 else if (compl_cont_mode != 0)
2646 compl_cont_status &= ~CONT_LOCAL;
2647 // FALLTHROUGH
2648 default:
2649 // If we have typed at least 2 ^X's... for modes != 0, we set
2650 // compl_cont_status = 0 (eg, as if we had just started ^X
2651 // mode).
2652 // For mode 0, we set "compl_cont_mode" to an impossible
2653 // value, in both cases ^X^X can be used to restart the same
2654 // mode (avoiding ADDING mode).
2655 // Undocumented feature: In a mode != 0 ^X^P and ^X^X^P start
2656 // 'complete' and local ^P expansions respectively.
2657 // In mode 0 an extra ^X is needed since ^X^P goes to ADDING
2658 // mode -- Acevedo
2659 if (c == Ctrl_X)
2660 {
2661 if (compl_cont_mode != 0)
2662 compl_cont_status = 0;
2663 else
2664 compl_cont_mode = CTRL_X_NOT_DEFINED_YET;
2665 }
2666 ctrl_x_mode = CTRL_X_NORMAL;
2667 edit_submode = NULL;
2668 showmode();
2669 break;
2670 }
2671
2672 return retval;
2673}
2674
2675/*
glepnir1c5a1202024-12-04 20:27:34 +01002676 * Trigger CompleteDone event and adds relevant information to v:event
2677 */
2678 static void
2679trigger_complete_done_event(int mode UNUSED, char_u *word UNUSED)
2680{
2681#if defined(FEAT_EVAL)
2682 save_v_event_T save_v_event;
2683 dict_T *v_event = get_v_event(&save_v_event);
2684 char_u *mode_str = NULL;
2685
2686 mode = mode & ~CTRL_X_WANT_IDENT;
2687 if (ctrl_x_mode_names[mode])
2688 mode_str = (char_u *)ctrl_x_mode_names[mode];
2689
2690 (void)dict_add_string(v_event, "complete_word",
2691 word == NULL ? (char_u *)"" : word);
2692 (void)dict_add_string(v_event, "complete_type",
2693 mode_str != NULL ? mode_str : (char_u *)"");
2694
2695 dict_set_items_ro(v_event);
2696#endif
2697 ins_apply_autocmds(EVENT_COMPLETEDONE);
2698
2699#if defined(FEAT_EVAL)
2700 restore_v_event(v_event, &save_v_event);
2701#endif
2702}
2703
2704/*
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002705 * Stop insert completion mode
2706 */
2707 static int
2708ins_compl_stop(int c, int prev_mode, int retval)
2709{
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002710 int want_cindent;
glepnir1c5a1202024-12-04 20:27:34 +01002711 char_u *word = NULL;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002712
glepnir84a75032025-03-15 09:59:22 +01002713 // Remove pre-inserted text when present.
zeertzjq13436812025-04-23 20:46:35 +02002714 if (ins_compl_preinsert_effect() && ins_compl_win_active(curwin))
glepnir84a75032025-03-15 09:59:22 +01002715 ins_compl_delete();
2716
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002717 // Get here when we have finished typing a sequence of ^N and
2718 // ^P or other completion characters in CTRL-X mode. Free up
2719 // memory that was used, and make sure we can redo the insert.
John Marriott5e6ea922024-11-23 14:01:57 +01002720 if (compl_curr_match != NULL || compl_leader.string != NULL || c == Ctrl_E)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002721 {
glepnir40891ba2025-02-10 22:18:00 +01002722 char_u *ptr = NULL;
John Marriott5e6ea922024-11-23 14:01:57 +01002723
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002724 // If any of the original typed text has been changed, eg when
2725 // ignorecase is set, we must add back-spaces to the redo
2726 // buffer. We add as few as necessary to delete just the part
2727 // of the original text that has changed.
2728 // When using the longest match, edited the match or used
2729 // CTRL-E then don't use the current match.
2730 if (compl_curr_match != NULL && compl_used_match && c != Ctrl_E)
John Marriott5e6ea922024-11-23 14:01:57 +01002731 ptr = compl_curr_match->cp_str.string;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002732 ins_compl_fixRedoBufForLeader(ptr);
2733 }
2734
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002735 want_cindent = (get_can_cindent() && cindent_on());
Bram Moolenaar8e145b82022-05-21 20:17:31 +01002736
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002737 // When completing whole lines: fix indent for 'cindent'.
2738 // Otherwise, break line if it's too long.
2739 if (compl_cont_mode == CTRL_X_WHOLE_LINE)
2740 {
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002741 // re-indent the current line
2742 if (want_cindent)
2743 {
2744 do_c_expr_indent();
2745 want_cindent = FALSE; // don't do it again
2746 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002747 }
2748 else
2749 {
2750 int prev_col = curwin->w_cursor.col;
2751
2752 // put the cursor on the last char, for 'tw' formatting
2753 if (prev_col > 0)
2754 dec_cursor();
2755 // only format when something was inserted
2756 if (!arrow_used && !ins_need_undo_get() && c != Ctrl_E)
2757 insertchar(NUL, 0, -1);
2758 if (prev_col > 0
2759 && ml_get_curline()[curwin->w_cursor.col] != NUL)
2760 inc_cursor();
2761 }
2762
2763 // If the popup menu is displayed pressing CTRL-Y means accepting
2764 // the selection without inserting anything. When
2765 // compl_enter_selects is set the Enter key does the same.
2766 if ((c == Ctrl_Y || (compl_enter_selects
2767 && (c == CAR || c == K_KENTER || c == NL)))
2768 && pum_visible())
glepnir1c5a1202024-12-04 20:27:34 +01002769 {
2770 word = vim_strsave(compl_shown_match->cp_str.string);
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002771 retval = TRUE;
glepnir1c5a1202024-12-04 20:27:34 +01002772 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002773
2774 // CTRL-E means completion is Ended, go back to the typed text.
2775 // but only do this, if the Popup is still visible
2776 if (c == Ctrl_E)
2777 {
Bram Moolenaarf12129f2022-07-01 19:58:30 +01002778 char_u *p = NULL;
John Marriott5e6ea922024-11-23 14:01:57 +01002779 size_t plen = 0;
Bram Moolenaarf12129f2022-07-01 19:58:30 +01002780
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002781 ins_compl_delete();
John Marriott5e6ea922024-11-23 14:01:57 +01002782 if (compl_leader.string != NULL)
2783 {
2784 p = compl_leader.string;
2785 plen = compl_leader.length;
2786 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002787 else if (compl_first_match != NULL)
John Marriott5e6ea922024-11-23 14:01:57 +01002788 {
2789 p = compl_orig_text.string;
2790 plen = compl_orig_text.length;
2791 }
Bram Moolenaarf12129f2022-07-01 19:58:30 +01002792 if (p != NULL)
2793 {
2794 int compl_len = get_compl_len();
Bram Moolenaarf12129f2022-07-01 19:58:30 +01002795
John Marriott5e6ea922024-11-23 14:01:57 +01002796 if ((int)plen > compl_len)
zeertzjqf25d8f92024-12-18 21:12:25 +01002797 ins_compl_insert_bytes(p + compl_len, (int)plen - compl_len);
Bram Moolenaarf12129f2022-07-01 19:58:30 +01002798 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002799 retval = TRUE;
2800 }
2801
2802 auto_format(FALSE, TRUE);
2803
2804 // Trigger the CompleteDonePre event to give scripts a chance to
2805 // act upon the completion before clearing the info, and restore
2806 // ctrl_x_mode, so that complete_info() can be used.
2807 ctrl_x_mode = prev_mode;
2808 ins_apply_autocmds(EVENT_COMPLETEDONEPRE);
2809
2810 ins_compl_free();
2811 compl_started = FALSE;
2812 compl_matches = 0;
2813 if (!shortmess(SHM_COMPLETIONMENU))
2814 msg_clr_cmdline(); // necessary for "noshowmode"
2815 ctrl_x_mode = CTRL_X_NORMAL;
2816 compl_enter_selects = FALSE;
2817 if (edit_submode != NULL)
2818 {
2819 edit_submode = NULL;
2820 showmode();
2821 }
2822
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002823 if (c == Ctrl_C && cmdwin_type != 0)
2824 // Avoid the popup menu remains displayed when leaving the
2825 // command line window.
2826 update_screen(0);
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002827 // Indent now if a key was typed that is in 'cinkeys'.
2828 if (want_cindent && in_cinkeys(KEY_COMPLETE, ' ', inindent(0)))
2829 do_c_expr_indent();
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002830 // Trigger the CompleteDone event to give scripts a chance to act
2831 // upon the end of completion.
glepnir1c5a1202024-12-04 20:27:34 +01002832 trigger_complete_done_event(prev_mode, word);
2833 vim_free(word);
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002834
2835 return retval;
2836}
2837
2838/*
glepnircf7f0122025-04-15 19:02:00 +02002839 * Cancel completion.
2840 */
2841 int
2842ins_compl_cancel(void)
2843{
2844 return ins_compl_stop(' ', ctrl_x_mode, TRUE);
2845}
2846
2847/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002848 * Prepare for Insert mode completion, or stop it.
2849 * Called just after typing a character in Insert mode.
2850 * Returns TRUE when the character is not to be inserted;
2851 */
2852 int
2853ins_compl_prep(int c)
2854{
glepnir19e1dd62025-05-08 22:50:38 +02002855 int retval = FALSE;
2856 int prev_mode = ctrl_x_mode;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002857
2858 // Forget any previous 'special' messages if this is actually
2859 // a ^X mode key - bar ^R, in which case we wait to see what it gives us.
2860 if (c != Ctrl_R && vim_is_ctrl_x_key(c))
2861 edit_submode_extra = NULL;
2862
zeertzjq440d4cb2023-03-02 17:51:32 +00002863 // Ignore end of Select mode mapping and mouse scroll/movement.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002864 if (c == K_SELECT || c == K_MOUSEDOWN || c == K_MOUSEUP
zeertzjq440d4cb2023-03-02 17:51:32 +00002865 || c == K_MOUSELEFT || c == K_MOUSERIGHT || c == K_MOUSEMOVE
Bram Moolenaare32c3c42022-01-15 18:26:04 +00002866 || c == K_COMMAND || c == K_SCRIPT_COMMAND)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002867 return retval;
2868
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01002869#ifdef FEAT_PROP_POPUP
Bram Moolenaarf0bc15c2019-08-18 19:23:45 +02002870 // Ignore mouse events in a popup window
2871 if (is_mouse_key(c))
2872 {
2873 // Ignore drag and release events, the position does not need to be in
2874 // the popup and it may have just closed.
2875 if (c == K_LEFTRELEASE
2876 || c == K_LEFTRELEASE_NM
2877 || c == K_MIDDLERELEASE
2878 || c == K_RIGHTRELEASE
2879 || c == K_X1RELEASE
2880 || c == K_X2RELEASE
2881 || c == K_LEFTDRAG
2882 || c == K_MIDDLEDRAG
2883 || c == K_RIGHTDRAG
2884 || c == K_X1DRAG
2885 || c == K_X2DRAG)
2886 return retval;
2887 if (popup_visible)
2888 {
2889 int row = mouse_row;
2890 int col = mouse_col;
2891 win_T *wp = mouse_find_win(&row, &col, FIND_POPUP);
2892
2893 if (wp != NULL && WIN_IS_POPUP(wp))
2894 return retval;
2895 }
2896 }
2897#endif
2898
zeertzjqdca29d92021-08-31 19:12:51 +02002899 if (ctrl_x_mode == CTRL_X_CMDLINE_CTRL_X && c != Ctrl_X)
2900 {
2901 if (c == Ctrl_V || c == Ctrl_Q || c == Ctrl_Z || ins_compl_pum_key(c)
2902 || !vim_is_ctrl_x_key(c))
2903 {
2904 // Not starting another completion mode.
2905 ctrl_x_mode = CTRL_X_CMDLINE;
2906
2907 // CTRL-X CTRL-Z should stop completion without inserting anything
2908 if (c == Ctrl_Z)
2909 retval = TRUE;
2910 }
2911 else
2912 {
2913 ctrl_x_mode = CTRL_X_CMDLINE;
2914
2915 // Other CTRL-X keys first stop completion, then start another
2916 // completion mode.
2917 ins_compl_prep(' ');
2918 ctrl_x_mode = CTRL_X_NOT_DEFINED_YET;
2919 }
2920 }
2921
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002922 // Set "compl_get_longest" when finding the first matches.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002923 if (ctrl_x_mode_not_defined_yet()
2924 || (ctrl_x_mode_normal() && !compl_started))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002925 {
zeertzjq529b9ad2024-06-05 20:27:06 +02002926 compl_get_longest = (get_cot_flags() & COT_LONGEST) != 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002927 compl_used_match = TRUE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002928 }
2929
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002930 if (ctrl_x_mode_not_defined_yet())
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002931 // We have just typed CTRL-X and aren't quite sure which CTRL-X mode
2932 // it will be yet. Now we decide.
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002933 retval = set_ctrl_x_mode(c);
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002934 else if (ctrl_x_mode_not_default())
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002935 {
2936 // We're already in CTRL-X mode, do we stay in it?
2937 if (!vim_is_ctrl_x_key(c))
2938 {
glepnir40891ba2025-02-10 22:18:00 +01002939 ctrl_x_mode = ctrl_x_mode_scroll() ? CTRL_X_NORMAL : CTRL_X_FINISHED;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002940 edit_submode = NULL;
2941 }
2942 showmode();
2943 }
2944
2945 if (compl_started || ctrl_x_mode == CTRL_X_FINISHED)
2946 {
2947 // Show error message from attempted keyword completion (probably
2948 // 'Pattern not found') until another key is hit, then go back to
2949 // showing what mode we are in.
2950 showmode();
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00002951 if ((ctrl_x_mode_normal() && c != Ctrl_N && c != Ctrl_P
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002952 && c != Ctrl_R && !ins_compl_pum_key(c))
2953 || ctrl_x_mode == CTRL_X_FINISHED)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00002954 retval = ins_compl_stop(c, prev_mode, retval);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002955 }
2956 else if (ctrl_x_mode == CTRL_X_LOCAL_MSG)
2957 // Trigger the CompleteDone event to give scripts a chance to act
2958 // upon the (possibly failed) completion.
glepnir1c5a1202024-12-04 20:27:34 +01002959 trigger_complete_done_event(ctrl_x_mode, NULL);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002960
LemonBoy2bf52dd2022-04-09 18:17:34 +01002961 may_trigger_modechanged();
=?UTF-8?q?Magnus=20Gro=C3=9F?=25def2c2021-10-22 18:56:39 +01002962
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002963 // reset continue_* if we left expansion-mode, if we stay they'll be
2964 // (re)set properly in ins_complete()
2965 if (!vim_is_ctrl_x_key(c))
2966 {
2967 compl_cont_status = 0;
2968 compl_cont_mode = 0;
2969 }
2970
2971 return retval;
2972}
2973
2974/*
2975 * Fix the redo buffer for the completion leader replacing some of the typed
2976 * text. This inserts backspaces and appends the changed text.
2977 * "ptr" is the known leader text or NUL.
2978 */
2979 static void
2980ins_compl_fixRedoBufForLeader(char_u *ptr_arg)
2981{
John Marriott5e6ea922024-11-23 14:01:57 +01002982 int len = 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002983 char_u *p;
2984 char_u *ptr = ptr_arg;
2985
2986 if (ptr == NULL)
2987 {
John Marriott5e6ea922024-11-23 14:01:57 +01002988 if (compl_leader.string != NULL)
2989 ptr = compl_leader.string;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002990 else
2991 return; // nothing to do
2992 }
John Marriott5e6ea922024-11-23 14:01:57 +01002993 if (compl_orig_text.string != NULL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01002994 {
John Marriott5e6ea922024-11-23 14:01:57 +01002995 p = compl_orig_text.string;
glepnir40891ba2025-02-10 22:18:00 +01002996 // Find length of common prefix between original text and new completion
2997 while (p[len] != NUL && p[len] == ptr[len])
2998 len++;
2999 // Adjust length to not break inside a multi-byte character
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003000 if (len > 0)
3001 len -= (*mb_head_off)(p, p + len);
glepnir40891ba2025-02-10 22:18:00 +01003002 // Add backspace characters for each remaining character in
3003 // original text
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003004 for (p += len; *p != NUL; MB_PTR_ADV(p))
3005 AppendCharToRedobuff(K_BS);
3006 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003007 if (ptr != NULL)
3008 AppendToRedobuffLit(ptr + len, -1);
3009}
3010
3011/*
3012 * Loops through the list of windows, loaded-buffers or non-loaded-buffers
3013 * (depending on flag) starting from buf and looking for a non-scanned
3014 * buffer (other than curbuf). curbuf is special, if it is called with
3015 * buf=curbuf then it has to be the first call for a given flag/expansion.
3016 *
3017 * Returns the buffer to scan, if any, otherwise returns curbuf -- Acevedo
3018 */
3019 static buf_T *
3020ins_compl_next_buf(buf_T *buf, int flag)
3021{
glepnir19e1dd62025-05-08 22:50:38 +02003022 static win_T *wp = NULL;
3023 int skip_buffer;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003024
3025 if (flag == 'w') // just windows
3026 {
Bram Moolenaar0ff01832022-09-24 19:20:30 +01003027 if (buf == curbuf || !win_valid(wp))
3028 // first call for this flag/expansion or window was closed
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003029 wp = curwin;
glepnir40891ba2025-02-10 22:18:00 +01003030
3031 while (TRUE)
3032 {
Hirohito Higashi355db992025-04-28 18:07:02 +02003033 // Move to next window (wrap to first window if at the end)
3034 wp = (wp->w_next != NULL) ? wp->w_next : firstwin;
3035 // Break if we're back at start or found an unscanned buffer
3036 if (wp == curwin || !wp->w_buffer->b_scanned)
3037 break;
glepnir40891ba2025-02-10 22:18:00 +01003038 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003039 buf = wp->w_buffer;
3040 }
3041 else
glepnir40891ba2025-02-10 22:18:00 +01003042 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003043 // 'b' (just loaded buffers), 'u' (just non-loaded buffers) or 'U'
3044 // (unlisted buffers)
3045 // When completing whole lines skip unloaded buffers.
glepnir40891ba2025-02-10 22:18:00 +01003046 while (TRUE)
3047 {
3048 // Move to next buffer (wrap to first buffer if at the end)
3049 buf = (buf->b_next != NULL) ? buf->b_next : firstbuf;
3050 // Break if we're back at start buffer
3051 if (buf == curbuf)
3052 break;
3053
3054 // Check buffer conditions based on flag
3055 if (flag == 'U')
3056 skip_buffer = buf->b_p_bl;
3057 else
3058 skip_buffer = !buf->b_p_bl ||
3059 (buf->b_ml.ml_mfp == NULL) != (flag == 'u');
3060
3061 // Break if we found a buffer that matches our criteria
3062 if (!skip_buffer && !buf->b_scanned)
3063 break;
3064 }
3065 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003066 return buf;
3067}
3068
3069#ifdef FEAT_COMPL_FUNC
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003070
3071# ifdef FEAT_EVAL
3072static callback_T cfu_cb; // 'completefunc' callback function
3073static callback_T ofu_cb; // 'omnifunc' callback function
3074static callback_T tsrfu_cb; // 'thesaurusfunc' callback function
3075# endif
3076
3077/*
3078 * Copy a global callback function to a buffer local callback.
3079 */
3080 static void
3081copy_global_to_buflocal_cb(callback_T *globcb, callback_T *bufcb)
3082{
3083 free_callback(bufcb);
3084 if (globcb->cb_name != NULL && *globcb->cb_name != NUL)
3085 copy_callback(bufcb, globcb);
3086}
3087
3088/*
3089 * Parse the 'completefunc' option value and set the callback function.
3090 * Invoked when the 'completefunc' option is set. The option value can be a
3091 * name of a function (string), or function(<name>) or funcref(<name>) or a
3092 * lambda expression.
3093 */
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003094 char *
Yegappan Lakshmananaf936912023-02-20 12:16:39 +00003095did_set_completefunc(optset_T *args UNUSED)
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003096{
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003097 if (option_set_callback_func(curbuf->b_p_cfu, &cfu_cb) == FAIL)
3098 return e_invalid_argument;
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003099
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003100 set_buflocal_cfu_callback(curbuf);
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003101
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003102 return NULL;
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003103}
3104
3105/*
3106 * Copy the global 'completefunc' callback function to the buffer-local
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003107 * 'completefunc' callback for "buf".
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003108 */
3109 void
3110set_buflocal_cfu_callback(buf_T *buf UNUSED)
3111{
3112# ifdef FEAT_EVAL
3113 copy_global_to_buflocal_cb(&cfu_cb, &buf->b_cfu_cb);
3114# endif
3115}
3116
3117/*
3118 * Parse the 'omnifunc' option value and set the callback function.
3119 * Invoked when the 'omnifunc' option is set. The option value can be a
3120 * name of a function (string), or function(<name>) or funcref(<name>) or a
3121 * lambda expression.
3122 */
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003123 char *
Yegappan Lakshmananaf936912023-02-20 12:16:39 +00003124did_set_omnifunc(optset_T *args UNUSED)
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003125{
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003126 if (option_set_callback_func(curbuf->b_p_ofu, &ofu_cb) == FAIL)
3127 return e_invalid_argument;
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003128
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003129 set_buflocal_ofu_callback(curbuf);
3130 return NULL;
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003131}
3132
3133/*
3134 * Copy the global 'omnifunc' callback function to the buffer-local 'omnifunc'
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003135 * callback for "buf".
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003136 */
3137 void
3138set_buflocal_ofu_callback(buf_T *buf UNUSED)
3139{
3140# ifdef FEAT_EVAL
3141 copy_global_to_buflocal_cb(&ofu_cb, &buf->b_ofu_cb);
3142# endif
3143}
3144
3145/*
3146 * Parse the 'thesaurusfunc' option value and set the callback function.
3147 * Invoked when the 'thesaurusfunc' option is set. The option value can be a
3148 * name of a function (string), or function(<name>) or funcref(<name>) or a
3149 * lambda expression.
3150 */
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003151 char *
Yegappan Lakshmananaf936912023-02-20 12:16:39 +00003152did_set_thesaurusfunc(optset_T *args UNUSED)
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003153{
3154 int retval;
3155
zeertzjq6eda2692024-11-03 09:23:33 +01003156 if (args->os_flags & OPT_LOCAL)
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003157 // buffer-local option set
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003158 retval = option_set_callback_func(curbuf->b_p_tsrfu,
3159 &curbuf->b_tsrfu_cb);
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003160 else
3161 {
3162 // global option set
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003163 retval = option_set_callback_func(p_tsrfu, &tsrfu_cb);
zeertzjq6eda2692024-11-03 09:23:33 +01003164 // when using :set, free the local callback
3165 if (!(args->os_flags & OPT_GLOBAL))
3166 free_callback(&curbuf->b_tsrfu_cb);
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003167 }
3168
Yegappan Lakshmananf2e30d02023-01-30 13:04:42 +00003169 return retval == FAIL ? e_invalid_argument : NULL;
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003170}
3171
Yegappan Lakshmanan6ae8fae2021-12-12 16:26:44 +00003172/*
3173 * Mark the global 'completefunc' 'omnifunc' and 'thesaurusfunc' callbacks with
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003174 * "copyID" so that they are not garbage collected.
Yegappan Lakshmanan6ae8fae2021-12-12 16:26:44 +00003175 */
3176 int
3177set_ref_in_insexpand_funcs(int copyID)
3178{
glepnir19e1dd62025-05-08 22:50:38 +02003179 int abort = set_ref_in_callback(&cfu_cb, copyID);
Yegappan Lakshmanan6ae8fae2021-12-12 16:26:44 +00003180 abort = abort || set_ref_in_callback(&ofu_cb, copyID);
3181 abort = abort || set_ref_in_callback(&tsrfu_cb, copyID);
3182
3183 return abort;
3184}
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003185
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003186/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003187 * Get the user-defined completion function name for completion "type"
Yegappan Lakshmanan160e9942021-10-16 15:41:29 +01003188 */
3189 static char_u *
3190get_complete_funcname(int type)
3191{
3192 switch (type)
3193 {
3194 case CTRL_X_FUNCTION:
3195 return curbuf->b_p_cfu;
3196 case CTRL_X_OMNI:
3197 return curbuf->b_p_ofu;
3198 case CTRL_X_THESAURUS:
Bram Moolenaarf4d8b762021-10-17 14:13:09 +01003199 return *curbuf->b_p_tsrfu == NUL ? p_tsrfu : curbuf->b_p_tsrfu;
Yegappan Lakshmanan160e9942021-10-16 15:41:29 +01003200 default:
3201 return (char_u *)"";
3202 }
3203}
3204
3205/*
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003206 * Get the callback to use for insert mode completion.
3207 */
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003208 static callback_T *
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003209get_insert_callback(int type)
3210{
3211 if (type == CTRL_X_FUNCTION)
3212 return &curbuf->b_cfu_cb;
3213 if (type == CTRL_X_OMNI)
3214 return &curbuf->b_ofu_cb;
3215 // CTRL_X_THESAURUS
3216 return (*curbuf->b_p_tsrfu != NUL) ? &curbuf->b_tsrfu_cb : &tsrfu_cb;
3217}
3218
3219/*
Yegappan Lakshmanan05e59e32021-12-01 10:30:07 +00003220 * Execute user defined complete function 'completefunc', 'omnifunc' or
3221 * 'thesaurusfunc', and get matches in "matches".
Girish Palyacbe53192025-04-14 22:13:15 +02003222 * "type" can be one of CTRL_X_OMNI, CTRL_X_FUNCTION, or CTRL_X_THESAURUS.
3223 * Callback function "cb" is set if triggered by a function in the 'cpt'
3224 * option; otherwise, it is NULL.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003225 */
3226 static void
Girish Palyacbe53192025-04-14 22:13:15 +02003227expand_by_function(int type, char_u *base, callback_T *cb)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003228{
3229 list_T *matchlist = NULL;
3230 dict_T *matchdict = NULL;
3231 typval_T args[3];
3232 char_u *funcname;
3233 pos_T pos;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003234 typval_T rettv;
3235 int save_State = State;
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003236 int retval;
Girish Palyacbe53192025-04-14 22:13:15 +02003237 int is_cpt_function = (cb != NULL);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003238
Girish Palyacbe53192025-04-14 22:13:15 +02003239 if (!is_cpt_function)
3240 {
3241 funcname = get_complete_funcname(type);
3242 if (*funcname == NUL)
3243 return;
3244 cb = get_insert_callback(type);
3245 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003246
3247 // Call 'completefunc' to obtain the list of matches.
3248 args[0].v_type = VAR_NUMBER;
3249 args[0].vval.v_number = 0;
3250 args[1].v_type = VAR_STRING;
3251 args[1].vval.v_string = base != NULL ? base : (char_u *)"";
3252 args[2].v_type = VAR_UNKNOWN;
3253
3254 pos = curwin->w_cursor;
Bram Moolenaar28976e22021-01-29 21:07:07 +01003255 // Lock the text to avoid weird things from happening. Also disallow
3256 // switching to another window, it should not be needed and may end up in
3257 // Insert mode in another buffer.
zeertzjqcfe45652022-05-27 17:26:55 +01003258 ++textlock;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003259
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003260 retval = call_callback(cb, 0, &rettv, 2, args);
3261
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003262 // Call a function, which returns a list or dict.
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00003263 if (retval == OK)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003264 {
3265 switch (rettv.v_type)
3266 {
3267 case VAR_LIST:
3268 matchlist = rettv.vval.v_list;
3269 break;
3270 case VAR_DICT:
3271 matchdict = rettv.vval.v_dict;
3272 break;
3273 case VAR_SPECIAL:
3274 if (rettv.vval.v_number == VVAL_NONE)
3275 compl_opt_suppress_empty = TRUE;
3276 // FALLTHROUGH
3277 default:
3278 // TODO: Give error message?
3279 clear_tv(&rettv);
3280 break;
3281 }
3282 }
zeertzjqcfe45652022-05-27 17:26:55 +01003283 --textlock;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003284
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003285 curwin->w_cursor = pos; // restore the cursor position
zeertzjq0a419e02024-04-02 19:01:14 +02003286 check_cursor(); // make sure cursor position is valid, just in case
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003287 validate_cursor();
3288 if (!EQUAL_POS(curwin->w_cursor, pos))
3289 {
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00003290 emsg(_(e_complete_function_deleted_text));
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003291 goto theend;
3292 }
3293
3294 if (matchlist != NULL)
3295 ins_compl_add_list(matchlist);
3296 else if (matchdict != NULL)
3297 ins_compl_add_dict(matchdict);
3298
3299theend:
3300 // Restore State, it might have been changed.
3301 State = save_State;
3302
3303 if (matchdict != NULL)
3304 dict_unref(matchdict);
3305 if (matchlist != NULL)
3306 list_unref(matchlist);
3307}
3308#endif // FEAT_COMPL_FUNC
3309
3310#if defined(FEAT_COMPL_FUNC) || defined(FEAT_EVAL) || defined(PROTO)
glepnir38f99a12024-08-23 18:31:06 +02003311
3312 static inline int
3313get_user_highlight_attr(char_u *hlname)
3314{
3315 if (hlname != NULL && *hlname != NUL)
Hirohito Higashi355db992025-04-28 18:07:02 +02003316 return syn_name2attr(hlname);
glepnir38f99a12024-08-23 18:31:06 +02003317 return -1;
3318}
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003319/*
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003320 * Add a match to the list of matches from a typeval_T.
3321 * If the given string is already in the list of completions, then return
3322 * NOTDONE, otherwise add it to the list and return OK. If there is an error,
3323 * maybe because alloc() returns NULL, then FAIL is returned.
Bram Moolenaar440cf092021-04-03 20:13:30 +02003324 * When "fast" is TRUE use fast_breakcheck() instead of ui_breakcheck().
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003325 */
3326 static int
Bram Moolenaar440cf092021-04-03 20:13:30 +02003327ins_compl_add_tv(typval_T *tv, int dir, int fast)
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003328{
3329 char_u *word;
3330 int dup = FALSE;
3331 int empty = FALSE;
Bram Moolenaar440cf092021-04-03 20:13:30 +02003332 int flags = fast ? CP_FAST : 0;
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003333 char_u *(cptext[CPT_COUNT]);
Bram Moolenaar08928322020-01-04 14:32:48 +01003334 typval_T user_data;
Yegappan Lakshmanan37079142022-01-08 10:38:48 +00003335 int status;
glepnir0fe17f82024-10-08 22:26:44 +02003336 char_u *user_abbr_hlname;
glepnir38f99a12024-08-23 18:31:06 +02003337 char_u *user_kind_hlname;
glepnir80b66202024-11-27 21:53:53 +01003338 int user_hl[2] = { -1, -1 };
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003339
Bram Moolenaar08928322020-01-04 14:32:48 +01003340 user_data.v_type = VAR_UNKNOWN;
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003341 if (tv->v_type == VAR_DICT && tv->vval.v_dict != NULL)
3342 {
Bram Moolenaard61efa52022-07-23 09:52:04 +01003343 word = dict_get_string(tv->vval.v_dict, "word", FALSE);
3344 cptext[CPT_ABBR] = dict_get_string(tv->vval.v_dict, "abbr", FALSE);
3345 cptext[CPT_MENU] = dict_get_string(tv->vval.v_dict, "menu", FALSE);
3346 cptext[CPT_KIND] = dict_get_string(tv->vval.v_dict, "kind", FALSE);
3347 cptext[CPT_INFO] = dict_get_string(tv->vval.v_dict, "info", FALSE);
glepnir38f99a12024-08-23 18:31:06 +02003348
glepnir0fe17f82024-10-08 22:26:44 +02003349 user_abbr_hlname = dict_get_string(tv->vval.v_dict, "abbr_hlgroup", FALSE);
glepnir80b66202024-11-27 21:53:53 +01003350 user_hl[0] = get_user_highlight_attr(user_abbr_hlname);
glepnir38f99a12024-08-23 18:31:06 +02003351
3352 user_kind_hlname = dict_get_string(tv->vval.v_dict, "kind_hlgroup", FALSE);
glepnir80b66202024-11-27 21:53:53 +01003353 user_hl[1] = get_user_highlight_attr(user_kind_hlname);
glepnir508e7852024-07-25 21:39:08 +02003354
Bram Moolenaard61efa52022-07-23 09:52:04 +01003355 dict_get_tv(tv->vval.v_dict, "user_data", &user_data);
3356 if (dict_get_string(tv->vval.v_dict, "icase", FALSE) != NULL
3357 && dict_get_number(tv->vval.v_dict, "icase"))
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003358 flags |= CP_ICASE;
Bram Moolenaard61efa52022-07-23 09:52:04 +01003359 if (dict_get_string(tv->vval.v_dict, "dup", FALSE) != NULL)
3360 dup = dict_get_number(tv->vval.v_dict, "dup");
3361 if (dict_get_string(tv->vval.v_dict, "empty", FALSE) != NULL)
3362 empty = dict_get_number(tv->vval.v_dict, "empty");
3363 if (dict_get_string(tv->vval.v_dict, "equal", FALSE) != NULL
3364 && dict_get_number(tv->vval.v_dict, "equal"))
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003365 flags |= CP_EQUAL;
3366 }
3367 else
3368 {
3369 word = tv_get_string_chk(tv);
Bram Moolenaara80faa82020-04-12 19:37:17 +02003370 CLEAR_FIELD(cptext);
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003371 }
3372 if (word == NULL || (!empty && *word == NUL))
Yegappan Lakshmanan37079142022-01-08 10:38:48 +00003373 {
3374 clear_tv(&user_data);
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003375 return FAIL;
Yegappan Lakshmanan37079142022-01-08 10:38:48 +00003376 }
glepnir508e7852024-07-25 21:39:08 +02003377 status = ins_compl_add(word, -1, NULL, cptext,
glepnirf31cfa22025-03-06 21:59:13 +01003378 &user_data, dir, flags, dup, user_hl, 0);
Yegappan Lakshmanan37079142022-01-08 10:38:48 +00003379 if (status != OK)
3380 clear_tv(&user_data);
3381 return status;
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003382}
3383
3384/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003385 * Add completions from a list.
3386 */
3387 static void
3388ins_compl_add_list(list_T *list)
3389{
3390 listitem_T *li;
3391 int dir = compl_direction;
3392
3393 // Go through the List with matches and add each of them.
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003394 CHECK_LIST_MATERIALIZE(list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003395 FOR_ALL_LIST_ITEMS(list, li)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003396 {
Bram Moolenaar440cf092021-04-03 20:13:30 +02003397 if (ins_compl_add_tv(&li->li_tv, dir, TRUE) == OK)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003398 // if dir was BACKWARD then honor it just once
3399 dir = FORWARD;
3400 else if (did_emsg)
3401 break;
3402 }
3403}
3404
3405/*
3406 * Add completions from a dict.
3407 */
3408 static void
3409ins_compl_add_dict(dict_T *dict)
3410{
3411 dictitem_T *di_refresh;
3412 dictitem_T *di_words;
3413
3414 // Check for optional "refresh" item.
3415 compl_opt_refresh_always = FALSE;
3416 di_refresh = dict_find(dict, (char_u *)"refresh", 7);
3417 if (di_refresh != NULL && di_refresh->di_tv.v_type == VAR_STRING)
3418 {
3419 char_u *v = di_refresh->di_tv.vval.v_string;
3420
3421 if (v != NULL && STRCMP(v, (char_u *)"always") == 0)
3422 compl_opt_refresh_always = TRUE;
3423 }
3424
3425 // Add completions from a "words" list.
3426 di_words = dict_find(dict, (char_u *)"words", 5);
3427 if (di_words != NULL && di_words->di_tv.v_type == VAR_LIST)
3428 ins_compl_add_list(di_words->di_tv.vval.v_list);
3429}
3430
3431/*
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003432 * Start completion for the complete() function.
3433 * "startcol" is where the matched text starts (1 is first column).
3434 * "list" is the list of matches.
3435 */
3436 static void
3437set_completion(colnr_T startcol, list_T *list)
3438{
3439 int save_w_wrow = curwin->w_wrow;
3440 int save_w_leftcol = curwin->w_leftcol;
3441 int flags = CP_ORIGINAL_TEXT;
zeertzjqaa925ee2024-06-09 18:24:05 +02003442 unsigned int cur_cot_flags = get_cot_flags();
zeertzjq529b9ad2024-06-05 20:27:06 +02003443 int compl_longest = (cur_cot_flags & COT_LONGEST) != 0;
3444 int compl_no_insert = (cur_cot_flags & COT_NOINSERT) != 0;
3445 int compl_no_select = (cur_cot_flags & COT_NOSELECT) != 0;
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003446
3447 // If already doing completions stop it.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003448 if (ctrl_x_mode_not_default())
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003449 ins_compl_prep(' ');
3450 ins_compl_clear();
3451 ins_compl_free();
bfredl87af60c2022-09-24 11:17:51 +01003452 compl_get_longest = compl_longest;
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003453
3454 compl_direction = FORWARD;
3455 if (startcol > curwin->w_cursor.col)
3456 startcol = curwin->w_cursor.col;
3457 compl_col = startcol;
glepnir76bdb822025-02-08 19:04:51 +01003458 compl_lnum = curwin->w_cursor.lnum;
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003459 compl_length = (int)curwin->w_cursor.col - (int)startcol;
3460 // compl_pattern doesn't need to be set
glepniredd4ac32025-01-29 18:53:51 +01003461 compl_orig_text.string = vim_strnsave(ml_get_curline() + compl_col,
3462 (size_t)compl_length);
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003463 if (p_ic)
3464 flags |= CP_ICASE;
John Marriott5e6ea922024-11-23 14:01:57 +01003465 if (compl_orig_text.string == NULL)
3466 {
3467 compl_orig_text.length = 0;
3468 return;
3469 }
3470 compl_orig_text.length = (size_t)compl_length;
3471 if (ins_compl_add(compl_orig_text.string,
Hirohito Higashi355db992025-04-28 18:07:02 +02003472 (int)compl_orig_text.length, NULL, NULL, NULL, 0,
3473 flags | CP_FAST, FALSE, NULL, 0) != OK)
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003474 return;
3475
3476 ctrl_x_mode = CTRL_X_EVAL;
3477
3478 ins_compl_add_list(list);
3479 compl_matches = ins_compl_make_cyclic();
3480 compl_started = TRUE;
3481 compl_used_match = TRUE;
3482 compl_cont_status = 0;
3483
3484 compl_curr_match = compl_first_match;
bfredl87af60c2022-09-24 11:17:51 +01003485 int no_select = compl_no_select || compl_longest;
3486 if (compl_no_insert || no_select)
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003487 {
3488 ins_complete(K_DOWN, FALSE);
bfredl87af60c2022-09-24 11:17:51 +01003489 if (no_select)
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003490 // Down/Up has no real effect.
3491 ins_complete(K_UP, FALSE);
3492 }
3493 else
3494 ins_complete(Ctrl_N, FALSE);
3495 compl_enter_selects = compl_no_insert;
3496
3497 // Lazily show the popup menu, unless we got interrupted.
3498 if (!compl_interrupted)
3499 show_pum(save_w_wrow, save_w_leftcol);
LemonBoy2bf52dd2022-04-09 18:17:34 +01003500 may_trigger_modechanged();
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003501 out_flush();
3502}
3503
3504/*
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003505 * "complete()" function
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003506 */
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003507 void
3508f_complete(typval_T *argvars, typval_T *rettv UNUSED)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003509{
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02003510 if (in_vim9script()
3511 && (check_for_number_arg(argvars, 0) == FAIL
3512 || check_for_list_arg(argvars, 1) == FAIL))
3513 return;
3514
Bram Moolenaar24959102022-05-07 20:01:16 +01003515 if ((State & MODE_INSERT) == 0)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003516 {
Bram Moolenaar677658a2022-01-05 16:09:06 +00003517 emsg(_(e_complete_can_only_be_used_in_insert_mode));
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003518 return;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003519 }
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003520
3521 // Check for undo allowed here, because if something was already inserted
3522 // the line was already saved for undo and this check isn't done.
3523 if (!undo_allowed())
3524 return;
3525
Bram Moolenaard83392a2022-09-01 12:22:46 +01003526 if (check_for_nonnull_list_arg(argvars, 1) != FAIL)
Bram Moolenaarff06f282020-04-21 22:01:14 +02003527 {
glepnir19e1dd62025-05-08 22:50:38 +02003528 int startcol = (int)tv_get_number_chk(&argvars[0], NULL);
Bram Moolenaarff06f282020-04-21 22:01:14 +02003529 if (startcol > 0)
3530 set_completion(startcol - 1, argvars[1].vval.v_list);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003531 }
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003532}
3533
3534/*
3535 * "complete_add()" function
3536 */
3537 void
3538f_complete_add(typval_T *argvars, typval_T *rettv)
3539{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02003540 if (in_vim9script() && check_for_string_or_dict_arg(argvars, 0) == FAIL)
Yegappan Lakshmanan83494b42021-07-20 17:51:51 +02003541 return;
3542
Bram Moolenaar440cf092021-04-03 20:13:30 +02003543 rettv->vval.v_number = ins_compl_add_tv(&argvars[0], 0, FALSE);
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003544}
3545
3546/*
3547 * "complete_check()" function
3548 */
3549 void
3550f_complete_check(typval_T *argvars UNUSED, typval_T *rettv)
3551{
Bram Moolenaar79cdf022023-05-20 14:07:00 +01003552 int save_RedrawingDisabled = RedrawingDisabled;
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003553 RedrawingDisabled = 0;
Bram Moolenaar79cdf022023-05-20 14:07:00 +01003554
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003555 ins_compl_check_keys(0, TRUE);
3556 rettv->vval.v_number = ins_compl_interrupted();
Bram Moolenaar79cdf022023-05-20 14:07:00 +01003557
3558 RedrawingDisabled = save_RedrawingDisabled;
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003559}
3560
3561/*
glepnirbcd59952025-04-24 21:48:35 +02003562 * Add match item to the return list.
3563 * Returns FAIL if out of memory, OK otherwise.
3564 */
3565 static int
3566add_match_to_list(
3567 typval_T *rettv,
3568 char_u *str,
3569 int len,
3570 int pos)
3571{
3572 list_T *match;
3573 int ret;
3574
3575 match = list_alloc();
3576 if (match == NULL)
Hirohito Higashi355db992025-04-28 18:07:02 +02003577 return FAIL;
glepnirbcd59952025-04-24 21:48:35 +02003578
3579 if ((ret = list_append_number(match, pos + 1)) == FAIL
3580 || (ret = list_append_string(match, str, len)) == FAIL
3581 || (ret = list_append_list(rettv->vval.v_list, match)) == FAIL)
3582 {
Hirohito Higashi355db992025-04-28 18:07:02 +02003583 vim_free(match);
3584 return FAIL;
glepnirbcd59952025-04-24 21:48:35 +02003585 }
3586
3587 return OK;
3588}
3589
3590/*
3591 * "complete_match()" function
3592 */
3593 void
3594f_complete_match(typval_T *argvars, typval_T *rettv)
3595{
3596 linenr_T lnum;
3597 colnr_T col;
3598 char_u *line = NULL;
3599 char_u *ise = NULL;
3600 regmatch_T regmatch;
3601 char_u *before_cursor = NULL;
3602 char_u *cur_end = NULL;
glepnirbcd59952025-04-24 21:48:35 +02003603 int bytepos = 0;
3604 char_u part[MAXPATHL];
3605 int ret;
3606
3607 if (rettv_list_alloc(rettv) == FAIL)
3608 return;
3609
3610 ise = curbuf->b_p_ise[0] != NUL ? curbuf->b_p_ise : p_ise;
3611
3612 if (argvars[0].v_type == VAR_UNKNOWN)
3613 {
3614 lnum = curwin->w_cursor.lnum;
3615 col = curwin->w_cursor.col;
3616 }
3617 else if (argvars[1].v_type == VAR_UNKNOWN)
3618 {
3619 emsg(_(e_invalid_argument));
3620 return;
3621 }
3622 else
3623 {
3624 lnum = (linenr_T)tv_get_number(&argvars[0]);
3625 col = (colnr_T)tv_get_number(&argvars[1]);
3626 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count)
3627 {
3628 semsg(_(e_invalid_line_number_nr), lnum);
3629 return;
3630 }
3631 if (col < 1 || col > ml_get_buf_len(curbuf, lnum))
3632 {
3633 semsg(_(e_invalid_column_number_nr), col + 1);
3634 return;
3635 }
3636 }
3637
3638 line = ml_get_buf(curbuf, lnum, FALSE);
3639 if (line == NULL)
3640 return;
3641
3642 before_cursor = vim_strnsave(line, col);
3643 if (before_cursor == NULL)
3644 return;
3645
3646 if (ise == NULL || *ise == NUL)
3647 {
3648 regmatch.regprog = vim_regcomp((char_u *)"\\k\\+$", RE_MAGIC);
3649 if (regmatch.regprog != NULL)
3650 {
3651 if (vim_regexec_nl(&regmatch, before_cursor, (colnr_T)0))
3652 {
Christian Brabandt3accf042025-04-25 19:01:06 +02003653 char_u *trig = vim_strnsave(regmatch.startp[0],
glepnirbcd59952025-04-24 21:48:35 +02003654 regmatch.endp[0] - regmatch.startp[0]);
3655 if (trig == NULL)
3656 {
3657 vim_free(before_cursor);
Christian Brabandt3accf042025-04-25 19:01:06 +02003658 vim_regfree(regmatch.regprog);
glepnirbcd59952025-04-24 21:48:35 +02003659 return;
3660 }
3661
Christian Brabandt3accf042025-04-25 19:01:06 +02003662 bytepos = (int)(regmatch.startp[0] - before_cursor);
glepnirbcd59952025-04-24 21:48:35 +02003663 ret = add_match_to_list(rettv, trig, -1, bytepos);
3664 vim_free(trig);
3665 if (ret == FAIL)
3666 {
Christian Brabandt3accf042025-04-25 19:01:06 +02003667 vim_free(before_cursor);
glepnirbcd59952025-04-24 21:48:35 +02003668 vim_regfree(regmatch.regprog);
3669 return;
3670 }
3671 }
3672 vim_regfree(regmatch.regprog);
3673 }
3674 }
3675 else
3676 {
glepnir08db2f42025-05-14 20:26:19 +02003677 char_u *p = ise;
3678 char_u *p_space = NULL;
3679
glepnirbcd59952025-04-24 21:48:35 +02003680 cur_end = before_cursor + (int)STRLEN(before_cursor);
3681
3682 while (*p != NUL)
3683 {
glepnir8d0e42b2025-05-12 20:28:28 +02003684 int len = 0;
glepnir08db2f42025-05-14 20:26:19 +02003685 if (p_space)
glepnir8d0e42b2025-05-12 20:28:28 +02003686 {
glepnir08db2f42025-05-14 20:26:19 +02003687 len = p - p_space - 1;
3688 memcpy(part, p_space + 1, len);
3689 p_space = NULL;
glepnir8d0e42b2025-05-12 20:28:28 +02003690 }
3691 else
glepnir08db2f42025-05-14 20:26:19 +02003692 {
3693 char_u *next_comma = vim_strchr((*p == ',') ? p + 1 : p, ',');
3694 if (next_comma && *(next_comma + 1) == ' ')
3695 p_space = next_comma;
3696
glepnir8d0e42b2025-05-12 20:28:28 +02003697 len = copy_option_part(&p, part, MAXPATHL, ",");
glepnir08db2f42025-05-14 20:26:19 +02003698 }
glepnirbcd59952025-04-24 21:48:35 +02003699
3700 if (len > 0 && len <= col)
3701 {
3702 if (STRNCMP(cur_end - len, part, len) == 0)
3703 {
3704 bytepos = col - len;
3705 if (add_match_to_list(rettv, part, len, bytepos) == FAIL)
3706 {
3707 vim_free(before_cursor);
3708 return;
3709 }
3710 }
3711 }
3712 }
3713 }
3714
3715 vim_free(before_cursor);
3716}
3717
3718/*
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003719 * Return Insert completion mode name string
3720 */
3721 static char_u *
3722ins_compl_mode(void)
3723{
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003724 if (ctrl_x_mode_not_defined_yet() || ctrl_x_mode_scroll() || compl_started)
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003725 return (char_u *)ctrl_x_mode_names[ctrl_x_mode & ~CTRL_X_WANT_IDENT];
3726
3727 return (char_u *)"";
3728}
3729
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00003730/*
3731 * Assign the sequence number to all the completion matches which don't have
3732 * one assigned yet.
3733 */
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003734 static void
Yegappan Lakshmanana23a11b2023-02-21 14:27:41 +00003735ins_compl_update_sequence_numbers(void)
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003736{
3737 int number = 0;
3738 compl_T *match;
3739
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003740 if (compl_dir_forward())
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003741 {
Bram Moolenaarf39d9e92023-04-22 22:54:40 +01003742 // Search backwards for the first valid (!= -1) number.
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003743 // This should normally succeed already at the first loop
3744 // cycle, so it's fast!
3745 for (match = compl_curr_match->cp_prev; match != NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003746 && !is_first_match(match); match = match->cp_prev)
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003747 if (match->cp_number != -1)
3748 {
3749 number = match->cp_number;
3750 break;
3751 }
3752 if (match != NULL)
Bram Moolenaarf39d9e92023-04-22 22:54:40 +01003753 // go up and assign all numbers which are not assigned yet
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003754 for (match = match->cp_next;
3755 match != NULL && match->cp_number == -1;
3756 match = match->cp_next)
3757 match->cp_number = ++number;
3758 }
3759 else // BACKWARD
3760 {
Bram Moolenaarf39d9e92023-04-22 22:54:40 +01003761 // Search forwards (upwards) for the first valid (!= -1)
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003762 // number. This should normally succeed already at the
3763 // first loop cycle, so it's fast!
3764 for (match = compl_curr_match->cp_next; match != NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00003765 && !is_first_match(match); match = match->cp_next)
glepnir40891ba2025-02-10 22:18:00 +01003766 {
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003767 if (match->cp_number != -1)
3768 {
3769 number = match->cp_number;
3770 break;
3771 }
glepnir40891ba2025-02-10 22:18:00 +01003772 }
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003773 if (match != NULL)
glepnir40891ba2025-02-10 22:18:00 +01003774 {
Bram Moolenaarf39d9e92023-04-22 22:54:40 +01003775 // go down and assign all numbers which are not assigned yet
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003776 for (match = match->cp_prev; match
3777 && match->cp_number == -1;
3778 match = match->cp_prev)
3779 match->cp_number = ++number;
glepnir40891ba2025-02-10 22:18:00 +01003780 }
Bram Moolenaarf9d51352020-10-26 19:22:42 +01003781 }
3782}
3783
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003784/*
glepnir037b0282025-01-16 14:37:44 +01003785 * Fill the dict of complete_info
3786 */
3787 static void
3788fill_complete_info_dict(dict_T *di, compl_T *match, int add_match)
3789{
3790 dict_add_string(di, "word", match->cp_str.string);
3791 dict_add_string(di, "abbr", match->cp_text[CPT_ABBR]);
3792 dict_add_string(di, "menu", match->cp_text[CPT_MENU]);
3793 dict_add_string(di, "kind", match->cp_text[CPT_KIND]);
3794 dict_add_string(di, "info", match->cp_text[CPT_INFO]);
3795 if (add_match)
Hirohito Higashi355db992025-04-28 18:07:02 +02003796 dict_add_bool(di, "match", match->cp_in_match_array);
glepnir037b0282025-01-16 14:37:44 +01003797 if (match->cp_user_data.v_type == VAR_UNKNOWN)
Hirohito Higashi355db992025-04-28 18:07:02 +02003798 // Add an empty string for backwards compatibility
3799 dict_add_string(di, "user_data", (char_u *)"");
glepnir037b0282025-01-16 14:37:44 +01003800 else
Hirohito Higashi355db992025-04-28 18:07:02 +02003801 dict_add_tv(di, "user_data", &match->cp_user_data);
glepnir037b0282025-01-16 14:37:44 +01003802}
3803
3804/*
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003805 * Get complete information
3806 */
3807 static void
3808get_complete_info(list_T *what_list, dict_T *retdict)
3809{
3810 int ret = OK;
3811 listitem_T *item;
3812#define CI_WHAT_MODE 0x01
3813#define CI_WHAT_PUM_VISIBLE 0x02
3814#define CI_WHAT_ITEMS 0x04
3815#define CI_WHAT_SELECTED 0x08
glepnir037b0282025-01-16 14:37:44 +01003816#define CI_WHAT_COMPLETED 0x10
glepnird4088ed2024-12-31 10:55:22 +01003817#define CI_WHAT_MATCHES 0x20
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003818#define CI_WHAT_ALL 0xff
3819 int what_flag;
3820
3821 if (what_list == NULL)
glepnir037b0282025-01-16 14:37:44 +01003822 what_flag = CI_WHAT_ALL & ~(CI_WHAT_MATCHES | CI_WHAT_COMPLETED);
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003823 else
3824 {
3825 what_flag = 0;
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02003826 CHECK_LIST_MATERIALIZE(what_list);
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003827 FOR_ALL_LIST_ITEMS(what_list, item)
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003828 {
3829 char_u *what = tv_get_string(&item->li_tv);
3830
3831 if (STRCMP(what, "mode") == 0)
3832 what_flag |= CI_WHAT_MODE;
3833 else if (STRCMP(what, "pum_visible") == 0)
3834 what_flag |= CI_WHAT_PUM_VISIBLE;
3835 else if (STRCMP(what, "items") == 0)
3836 what_flag |= CI_WHAT_ITEMS;
3837 else if (STRCMP(what, "selected") == 0)
3838 what_flag |= CI_WHAT_SELECTED;
glepnir037b0282025-01-16 14:37:44 +01003839 else if (STRCMP(what, "completed") == 0)
3840 what_flag |= CI_WHAT_COMPLETED;
glepnird4088ed2024-12-31 10:55:22 +01003841 else if (STRCMP(what, "matches") == 0)
3842 what_flag |= CI_WHAT_MATCHES;
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003843 }
3844 }
3845
3846 if (ret == OK && (what_flag & CI_WHAT_MODE))
3847 ret = dict_add_string(retdict, "mode", ins_compl_mode());
3848
3849 if (ret == OK && (what_flag & CI_WHAT_PUM_VISIBLE))
3850 ret = dict_add_number(retdict, "pum_visible", pum_visible());
3851
glepnir037b0282025-01-16 14:37:44 +01003852 if (ret == OK && (what_flag & (CI_WHAT_ITEMS | CI_WHAT_SELECTED
3853 | CI_WHAT_MATCHES | CI_WHAT_COMPLETED)))
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003854 {
Christian Brabandt9eb236f2024-03-21 20:12:59 +01003855 list_T *li = NULL;
Girish Palya8950bf72024-03-20 20:07:29 +01003856 dict_T *di;
3857 compl_T *match;
3858 int selected_idx = -1;
glepnird4088ed2024-12-31 10:55:22 +01003859 int has_items = what_flag & CI_WHAT_ITEMS;
3860 int has_matches = what_flag & CI_WHAT_MATCHES;
glepnir037b0282025-01-16 14:37:44 +01003861 int has_completed = what_flag & CI_WHAT_COMPLETED;
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003862
glepnird4088ed2024-12-31 10:55:22 +01003863 if (has_items || has_matches)
Girish Palya8950bf72024-03-20 20:07:29 +01003864 {
3865 li = list_alloc();
3866 if (li == NULL)
3867 return;
glepnird4088ed2024-12-31 10:55:22 +01003868 ret = dict_add_list(retdict, (has_matches && !has_items)
3869 ? "matches" : "items", li);
Girish Palya8950bf72024-03-20 20:07:29 +01003870 }
3871 if (ret == OK && what_flag & CI_WHAT_SELECTED)
3872 if (compl_curr_match != NULL && compl_curr_match->cp_number == -1)
3873 ins_compl_update_sequence_numbers();
3874 if (ret == OK && compl_first_match != NULL)
3875 {
3876 int list_idx = 0;
3877 match = compl_first_match;
3878 do
3879 {
3880 if (!match_at_original_text(match))
3881 {
glepnird4088ed2024-12-31 10:55:22 +01003882 if (has_items
3883 || (has_matches && match->cp_in_match_array))
Girish Palya8950bf72024-03-20 20:07:29 +01003884 {
3885 di = dict_alloc();
3886 if (di == NULL)
3887 return;
3888 ret = list_append_dict(li, di);
3889 if (ret != OK)
3890 return;
glepnir037b0282025-01-16 14:37:44 +01003891 fill_complete_info_dict(di, match, has_matches && has_items);
Girish Palya8950bf72024-03-20 20:07:29 +01003892 }
glepnird4088ed2024-12-31 10:55:22 +01003893 if (compl_curr_match != NULL
3894 && compl_curr_match->cp_number == match->cp_number)
Girish Palya8950bf72024-03-20 20:07:29 +01003895 selected_idx = list_idx;
Girish Palya8cd42a52025-06-05 21:04:29 +02003896 if (match->cp_in_match_array)
Girish Palya0ac1eb32025-04-16 20:18:33 +02003897 list_idx += 1;
Girish Palya8950bf72024-03-20 20:07:29 +01003898 }
3899 match = match->cp_next;
3900 }
3901 while (match != NULL && !is_first_match(match));
3902 }
3903 if (ret == OK && (what_flag & CI_WHAT_SELECTED))
3904 ret = dict_add_number(retdict, "selected", selected_idx);
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003905
glepnir037b0282025-01-16 14:37:44 +01003906 if (ret == OK && selected_idx != -1 && has_completed)
3907 {
3908 di = dict_alloc();
3909 if (di == NULL)
3910 return;
3911 fill_complete_info_dict(di, compl_curr_match, FALSE);
3912 ret = dict_add_dict(retdict, "completed", di);
3913 }
Yegappan Lakshmanan6b085b92022-09-04 12:47:21 +01003914 }
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02003915}
3916
3917/*
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003918 * "complete_info()" function
3919 */
3920 void
3921f_complete_info(typval_T *argvars, typval_T *rettv)
3922{
3923 list_T *what_list = NULL;
3924
Bram Moolenaar93a10962022-06-16 11:42:09 +01003925 if (rettv_dict_alloc(rettv) == FAIL)
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003926 return;
3927
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02003928 if (in_vim9script() && check_for_opt_list_arg(argvars, 0) == FAIL)
3929 return;
3930
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003931 if (argvars[0].v_type != VAR_UNKNOWN)
3932 {
Bram Moolenaard83392a2022-09-01 12:22:46 +01003933 if (check_for_list_arg(argvars, 0) == FAIL)
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003934 return;
Bram Moolenaar9bca58f2019-08-15 21:31:52 +02003935 what_list = argvars[0].vval.v_list;
3936 }
3937 get_complete_info(what_list, rettv->vval.v_dict);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01003938}
3939#endif
3940
3941/*
Yegappan Lakshmanan160e9942021-10-16 15:41:29 +01003942 * Returns TRUE when using a user-defined function for thesaurus completion.
3943 */
3944 static int
3945thesaurus_func_complete(int type UNUSED)
3946{
3947#ifdef FEAT_COMPL_FUNC
Bram Moolenaarf4d8b762021-10-17 14:13:09 +01003948 return type == CTRL_X_THESAURUS
3949 && (*curbuf->b_p_tsrfu != NUL || *p_tsrfu != NUL);
Yegappan Lakshmanan160e9942021-10-16 15:41:29 +01003950#else
3951 return FALSE;
3952#endif
3953}
3954
3955/*
Girish Palya7c621052025-05-26 19:41:59 +02003956 * Check if 'cpt' list index can be advanced to the next completion source.
3957 */
3958 static int
3959may_advance_cpt_index(char_u *cpt)
3960{
3961 char_u *p = cpt;
3962
Girish Palya98c29db2025-06-01 19:40:00 +02003963 if (cpt_sources_index == -1)
3964 return FALSE;
Girish Palya7c621052025-05-26 19:41:59 +02003965 while (*p == ',' || *p == ' ') // Skip delimiters
3966 p++;
3967 return (*p != NUL);
3968}
3969
3970/*
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00003971 * Return value of process_next_cpt_value()
3972 */
3973enum
3974{
3975 INS_COMPL_CPT_OK = 1,
3976 INS_COMPL_CPT_CONT,
3977 INS_COMPL_CPT_END
3978};
3979
3980/*
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00003981 * state information used for getting the next set of insert completion
3982 * matches.
3983 */
3984typedef struct
3985{
Bram Moolenaar0ff01832022-09-24 19:20:30 +01003986 char_u *e_cpt_copy; // copy of 'complete'
3987 char_u *e_cpt; // current entry in "e_cpt_copy"
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00003988 buf_T *ins_buf; // buffer being scanned
Bram Moolenaar0ff01832022-09-24 19:20:30 +01003989 pos_T *cur_match_pos; // current match position
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00003990 pos_T prev_match_pos; // previous match position
3991 int set_match_pos; // save first_match_pos/last_match_pos
3992 pos_T first_match_pos; // first match position
3993 pos_T last_match_pos; // last match position
3994 int found_all; // found all matches of a certain type.
3995 char_u *dict; // dictionary file to search
3996 int dict_f; // "dict" is an exact file name or not
Girish Palyacbe53192025-04-14 22:13:15 +02003997 callback_T *func_cb; // callback of function in 'cpt' option
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00003998} ins_compl_next_state_T;
3999
4000/*
4001 * Process the next 'complete' option value in st->e_cpt.
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004002 *
4003 * If successful, the arguments are set as below:
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004004 * st->cpt - pointer to the next option value in "st->cpt"
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004005 * compl_type_arg - type of insert mode completion to use
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004006 * st->found_all - all matches of this type are found
4007 * st->ins_buf - search for completions in this buffer
4008 * st->first_match_pos - position of the first completion match
4009 * st->last_match_pos - position of the last completion match
4010 * st->set_match_pos - TRUE if the first match position should be saved to
Bram Moolenaar6ed545e2022-05-09 20:09:23 +01004011 * avoid loops after the search wraps around.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004012 * st->dict - name of the dictionary or thesaurus file to search
4013 * st->dict_f - flag specifying whether "dict" is an exact file name or not
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004014 *
4015 * Returns INS_COMPL_CPT_OK if the next value is processed successfully.
Yegappan Lakshmanan37079142022-01-08 10:38:48 +00004016 * Returns INS_COMPL_CPT_CONT to skip the current completion source matching
4017 * the "st->e_cpt" option value and process the next matching source.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004018 * Returns INS_COMPL_CPT_END if all the values in "st->e_cpt" are processed.
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004019 */
4020 static int
4021process_next_cpt_value(
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004022 ins_compl_next_state_T *st,
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004023 int *compl_type_arg,
glepnir8159fb12024-07-17 20:32:54 +02004024 pos_T *start_match_pos,
Girish Palya7c621052025-05-26 19:41:59 +02004025 int fuzzy_collect,
4026 int *advance_cpt_idx)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004027{
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004028 int compl_type = -1;
4029 int status = INS_COMPL_CPT_OK;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004030
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004031 st->found_all = FALSE;
Girish Palya7c621052025-05-26 19:41:59 +02004032 *advance_cpt_idx = FALSE;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004033
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004034 while (*st->e_cpt == ',' || *st->e_cpt == ' ')
4035 st->e_cpt++;
4036
4037 if (*st->e_cpt == '.' && !curbuf->b_scanned)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004038 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004039 st->ins_buf = curbuf;
4040 st->first_match_pos = *start_match_pos;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004041 // Move the cursor back one character so that ^N can match the
4042 // word immediately after the cursor.
glepnir53b14572025-03-12 21:28:39 +01004043 if (ctrl_x_mode_normal() && (!fuzzy_collect && dec(&st->first_match_pos) < 0))
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004044 {
4045 // Move the cursor to after the last character in the
4046 // buffer, so that word at start of buffer is found
4047 // correctly.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004048 st->first_match_pos.lnum = st->ins_buf->b_ml.ml_line_count;
zeertzjq94b7c322024-03-12 21:50:32 +01004049 st->first_match_pos.col = ml_get_len(st->first_match_pos.lnum);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004050 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004051 st->last_match_pos = st->first_match_pos;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004052 compl_type = 0;
4053
4054 // Remember the first match so that the loop stops when we
4055 // wrap and come back there a second time.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004056 st->set_match_pos = TRUE;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004057 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004058 else if (vim_strchr((char_u *)"buwU", *st->e_cpt) != NULL
Bram Moolenaar0ff01832022-09-24 19:20:30 +01004059 && (st->ins_buf = ins_compl_next_buf(
4060 st->ins_buf, *st->e_cpt)) != curbuf)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004061 {
4062 // Scan a buffer, but not the current one.
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004063 if (st->ins_buf->b_ml.ml_mfp != NULL) // loaded buffer
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004064 {
4065 compl_started = TRUE;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004066 st->first_match_pos.col = st->last_match_pos.col = 0;
4067 st->first_match_pos.lnum = st->ins_buf->b_ml.ml_line_count + 1;
4068 st->last_match_pos.lnum = 0;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004069 compl_type = 0;
4070 }
4071 else // unloaded buffer, scan like dictionary
4072 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004073 st->found_all = TRUE;
4074 if (st->ins_buf->b_fname == NULL)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004075 {
4076 status = INS_COMPL_CPT_CONT;
4077 goto done;
4078 }
4079 compl_type = CTRL_X_DICTIONARY;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004080 st->dict = st->ins_buf->b_fname;
4081 st->dict_f = DICT_EXACT;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004082 }
=?UTF-8?q?Bj=C3=B6rn=20Linse?=91ccbad2022-10-13 12:51:13 +01004083 if (!shortmess(SHM_COMPLETIONSCAN))
4084 {
4085 msg_hist_off = TRUE; // reset in msg_trunc_attr()
4086 vim_snprintf((char *)IObuff, IOSIZE, _("Scanning: %s"),
4087 st->ins_buf->b_fname == NULL
4088 ? buf_spname(st->ins_buf)
4089 : st->ins_buf->b_sfname == NULL
4090 ? st->ins_buf->b_fname
4091 : st->ins_buf->b_sfname);
4092 (void)msg_trunc_attr((char *)IObuff, TRUE, HL_ATTR(HLF_R));
4093 }
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004094 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004095 else if (*st->e_cpt == NUL)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004096 status = INS_COMPL_CPT_END;
4097 else
4098 {
4099 if (ctrl_x_mode_line_or_eval())
4100 compl_type = -1;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004101 else if (*st->e_cpt == 'k' || *st->e_cpt == 's')
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004102 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004103 if (*st->e_cpt == 'k')
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004104 compl_type = CTRL_X_DICTIONARY;
4105 else
4106 compl_type = CTRL_X_THESAURUS;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004107 if (*++st->e_cpt != ',' && *st->e_cpt != NUL)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004108 {
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004109 st->dict = st->e_cpt;
4110 st->dict_f = DICT_FIRST;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004111 }
4112 }
Girish Palyacbe53192025-04-14 22:13:15 +02004113#ifdef FEAT_COMPL_FUNC
Girish Palya14f6da52025-05-26 19:04:25 +02004114 else if (*st->e_cpt == 'F' || *st->e_cpt == 'o')
Girish Palyacbe53192025-04-14 22:13:15 +02004115 {
4116 compl_type = CTRL_X_FUNCTION;
Girish Palya98c29db2025-06-01 19:40:00 +02004117 st->func_cb = get_callback_if_cpt_func(st->e_cpt);
Girish Palyacbe53192025-04-14 22:13:15 +02004118 if (!st->func_cb)
4119 compl_type = -1;
4120 }
4121#endif
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004122#ifdef FEAT_FIND_ID
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004123 else if (*st->e_cpt == 'i')
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004124 compl_type = CTRL_X_PATH_PATTERNS;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004125 else if (*st->e_cpt == 'd')
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004126 compl_type = CTRL_X_PATH_DEFINES;
4127#endif
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004128 else if (*st->e_cpt == ']' || *st->e_cpt == 't')
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004129 {
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004130 compl_type = CTRL_X_TAGS;
=?UTF-8?q?Bj=C3=B6rn=20Linse?=91ccbad2022-10-13 12:51:13 +01004131 if (!shortmess(SHM_COMPLETIONSCAN))
4132 {
4133 msg_hist_off = TRUE; // reset in msg_trunc_attr()
4134 vim_snprintf((char *)IObuff, IOSIZE, _("Scanning tags."));
4135 (void)msg_trunc_attr((char *)IObuff, TRUE, HL_ATTR(HLF_R));
4136 }
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004137 }
4138 else
4139 compl_type = -1;
4140
4141 // in any case e_cpt is advanced to the next entry
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004142 (void)copy_option_part(&st->e_cpt, IObuff, IOSIZE, ",");
Girish Palya7c621052025-05-26 19:41:59 +02004143 *advance_cpt_idx = may_advance_cpt_index(st->e_cpt);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004144
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004145 st->found_all = TRUE;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004146 if (compl_type == -1)
4147 status = INS_COMPL_CPT_CONT;
4148 }
4149
4150done:
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004151 *compl_type_arg = compl_type;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004152 return status;
4153}
4154
4155#ifdef FEAT_FIND_ID
4156/*
4157 * Get the next set of identifiers or defines matching "compl_pattern" in
4158 * included files.
4159 */
4160 static void
4161get_next_include_file_completion(int compl_type)
4162{
John Marriott5e6ea922024-11-23 14:01:57 +01004163 find_pattern_in_path(compl_pattern.string, compl_direction,
4164 (int)compl_pattern.length, FALSE, FALSE,
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004165 (compl_type == CTRL_X_PATH_DEFINES
4166 && !(compl_cont_status & CONT_SOL))
4167 ? FIND_DEFINE : FIND_ANY, 1L, ACTION_EXPAND,
Colin Kennedy21570352024-03-03 16:16:47 +01004168 (linenr_T)1, (linenr_T)MAXLNUM, FALSE);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004169}
4170#endif
4171
4172/*
4173 * Get the next set of words matching "compl_pattern" in dictionary or
4174 * thesaurus files.
4175 */
4176 static void
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004177get_next_dict_tsr_completion(int compl_type, char_u *dict, int dict_f)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004178{
4179#ifdef FEAT_COMPL_FUNC
4180 if (thesaurus_func_complete(compl_type))
Girish Palyacbe53192025-04-14 22:13:15 +02004181 expand_by_function(compl_type, compl_pattern.string, NULL);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004182 else
4183#endif
4184 ins_compl_dictionaries(
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004185 dict != NULL ? dict
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004186 : (compl_type == CTRL_X_THESAURUS
4187 ? (*curbuf->b_p_tsr == NUL ? p_tsr : curbuf->b_p_tsr)
4188 : (*curbuf->b_p_dict == NUL ? p_dict : curbuf->b_p_dict)),
John Marriott5e6ea922024-11-23 14:01:57 +01004189 compl_pattern.string,
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004190 dict != NULL ? dict_f : 0,
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004191 compl_type == CTRL_X_THESAURUS);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004192}
4193
4194/*
4195 * Get the next set of tag names matching "compl_pattern".
4196 */
4197 static void
4198get_next_tag_completion(void)
4199{
4200 int save_p_ic;
4201 char_u **matches;
4202 int num_matches;
4203
4204 // set p_ic according to p_ic, p_scs and pat for find_tags().
4205 save_p_ic = p_ic;
John Marriott5e6ea922024-11-23 14:01:57 +01004206 p_ic = ignorecase(compl_pattern.string);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004207
4208 // Find up to TAG_MANY matches. Avoids that an enormous number
4209 // of matches is found when compl_pattern is empty
4210 g_tag_at_cursor = TRUE;
John Marriott5e6ea922024-11-23 14:01:57 +01004211 if (find_tags(compl_pattern.string, &num_matches, &matches,
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004212 TAG_REGEXP | TAG_NAMES | TAG_NOIC | TAG_INS_COMP
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00004213 | (ctrl_x_mode_not_default() ? TAG_VERBOSE : 0),
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004214 TAG_MANY, curbuf->b_ffname) == OK && num_matches > 0)
4215 ins_compl_add_matches(num_matches, matches, p_ic);
4216 g_tag_at_cursor = FALSE;
4217 p_ic = save_p_ic;
4218}
4219
4220/*
glepnir8159fb12024-07-17 20:32:54 +02004221 * Compare function for qsort
4222 */
glepnirf31cfa22025-03-06 21:59:13 +01004223 static int
4224compare_scores(const void *a, const void *b)
glepnir8159fb12024-07-17 20:32:54 +02004225{
4226 int idx_a = *(const int *)a;
4227 int idx_b = *(const int *)b;
4228 int score_a = compl_fuzzy_scores[idx_a];
4229 int score_b = compl_fuzzy_scores[idx_b];
zeertzjq58d70522024-08-31 17:05:39 +02004230 return score_a == score_b ? (idx_a == idx_b ? 0 : (idx_a < idx_b ? -1 : 1))
4231 : (score_a > score_b ? -1 : 1);
glepnir8159fb12024-07-17 20:32:54 +02004232}
4233
4234/*
glepnirf31cfa22025-03-06 21:59:13 +01004235 * insert prefix with redraw
4236 */
4237 static void
4238ins_compl_longest_insert(char_u *prefix)
4239{
4240 ins_compl_delete();
4241 ins_compl_insert_bytes(prefix + get_compl_len(), -1);
4242 ins_redraw(FALSE);
4243}
4244
4245/*
4246 * Calculate the longest common prefix among the best fuzzy matches
4247 * stored in compl_best_matches, and insert it as the longest.
4248 */
4249 static void
4250fuzzy_longest_match(void)
4251{
4252 char_u *prefix = NULL;
4253 int prefix_len = 0;
4254 int i = 0;
4255 int j = 0;
4256 char_u *match_str = NULL;
4257 char_u *prefix_ptr = NULL;
4258 char_u *match_ptr = NULL;
4259 char_u *leader = NULL;
4260 size_t leader_len = 0;
4261 compl_T *compl = NULL;
4262 int more_candidates = FALSE;
4263 compl_T *nn_compl = NULL;
4264
4265 if (compl_num_bests == 0)
Hirohito Higashi355db992025-04-28 18:07:02 +02004266 return;
glepnirf31cfa22025-03-06 21:59:13 +01004267
4268 nn_compl = compl_first_match->cp_next->cp_next;
4269 if (nn_compl && nn_compl != compl_first_match)
4270 more_candidates = TRUE;
4271
4272 compl = ctrl_x_mode_whole_line() ? compl_first_match
4273 : compl_first_match->cp_next;
4274 if (compl_num_bests == 1)
4275 {
4276 // no more candidates insert the match str
4277 if (!more_candidates)
4278 {
4279 ins_compl_longest_insert(compl->cp_str.string);
4280 compl_num_bests = 0;
4281 }
4282 compl_num_bests = 0;
4283 return;
4284 }
4285
4286 compl_best_matches = (compl_T **)alloc(compl_num_bests * sizeof(compl_T *));
4287 if (compl_best_matches == NULL)
Hirohito Higashi355db992025-04-28 18:07:02 +02004288 return;
glepnirf31cfa22025-03-06 21:59:13 +01004289 while (compl != NULL && i < compl_num_bests)
4290 {
Hirohito Higashi355db992025-04-28 18:07:02 +02004291 compl_best_matches[i] = compl;
4292 compl = compl->cp_next;
4293 i++;
glepnirf31cfa22025-03-06 21:59:13 +01004294 }
4295
4296 prefix = compl_best_matches[0]->cp_str.string;
zeertzjq4422de62025-03-07 19:06:02 +01004297 prefix_len = (int)compl_best_matches[0]->cp_str.length;
glepnirf31cfa22025-03-06 21:59:13 +01004298
4299 for (i = 1; i < compl_num_bests; i++)
4300 {
4301 match_str = compl_best_matches[i]->cp_str.string;
Hirohito Higashi355db992025-04-28 18:07:02 +02004302 prefix_ptr = prefix;
4303 match_ptr = match_str;
4304 j = 0;
glepnirf31cfa22025-03-06 21:59:13 +01004305
4306 while (j < prefix_len && *match_ptr != NUL && *prefix_ptr != NUL)
4307 {
4308 if (STRNCMP(prefix_ptr, match_ptr, mb_ptr2len(prefix_ptr)) != 0)
4309 break;
4310
4311 MB_PTR_ADV(prefix_ptr);
4312 MB_PTR_ADV(match_ptr);
4313 j++;
4314 }
4315
4316 if (j > 0)
4317 prefix_len = j;
4318 }
4319
4320 leader = ins_compl_leader();
zeertzjq4422de62025-03-07 19:06:02 +01004321 leader_len = ins_compl_leader_len();
glepnirf31cfa22025-03-06 21:59:13 +01004322
4323 // skip non-consecutive prefixes
zeertzjq4422de62025-03-07 19:06:02 +01004324 if (leader_len > 0 && STRNCMP(prefix, leader, leader_len) != 0)
glepnirf31cfa22025-03-06 21:59:13 +01004325 goto end;
4326
zeertzjq4422de62025-03-07 19:06:02 +01004327 prefix = vim_strnsave(prefix, prefix_len);
glepnirf31cfa22025-03-06 21:59:13 +01004328 if (prefix != NULL)
4329 {
4330 ins_compl_longest_insert(prefix);
4331 compl_cfc_longest_ins = TRUE;
4332 vim_free(prefix);
4333 }
4334
4335end:
4336 vim_free(compl_best_matches);
4337 compl_best_matches = NULL;
4338 compl_num_bests = 0;
4339}
4340
4341/*
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004342 * Get the next set of filename matching "compl_pattern".
4343 */
4344 static void
4345get_next_filename_completion(void)
4346{
4347 char_u **matches;
4348 int num_matches;
glepnir8159fb12024-07-17 20:32:54 +02004349 char_u *ptr;
4350 garray_T fuzzy_indices;
4351 int i;
4352 int score;
4353 char_u *leader = ins_compl_leader();
John Marriott5e6ea922024-11-23 14:01:57 +01004354 size_t leader_len = ins_compl_leader_len();;
glepnirf31cfa22025-03-06 21:59:13 +01004355 int in_fuzzy_collect = (cfc_has_mode() && leader_len > 0);
glepnir8159fb12024-07-17 20:32:54 +02004356 int *fuzzy_indices_data;
glepnir0be03e12024-07-19 16:45:05 +02004357 char_u *last_sep = NULL;
glepnirf31cfa22025-03-06 21:59:13 +01004358 int need_collect_bests = in_fuzzy_collect && compl_get_longest;
4359 int max_score = 0;
4360 int current_score = 0;
4361 int dir = compl_direction;
glepnir8159fb12024-07-17 20:32:54 +02004362
glepnirb9de1a02024-08-02 19:14:38 +02004363#ifdef BACKSLASH_IN_FILENAME
4364 char pathsep = (curbuf->b_p_csl[0] == 's') ?
4365 '/' : (curbuf->b_p_csl[0] == 'b') ? '\\' : PATHSEP;
4366#else
4367 char pathsep = PATHSEP;
4368#endif
4369
glepnirf31cfa22025-03-06 21:59:13 +01004370 if (in_fuzzy_collect)
glepnir8159fb12024-07-17 20:32:54 +02004371 {
glepnirb9de1a02024-08-02 19:14:38 +02004372#ifdef BACKSLASH_IN_FILENAME
4373 if (curbuf->b_p_csl[0] == 's')
4374 {
John Marriott5e6ea922024-11-23 14:01:57 +01004375 for (i = 0; i < (int)leader_len; i++)
glepnirb9de1a02024-08-02 19:14:38 +02004376 {
4377 if (leader[i] == '\\')
4378 leader[i] = '/';
4379 }
4380 }
4381 else if (curbuf->b_p_csl[0] == 'b')
4382 {
John Marriott5e6ea922024-11-23 14:01:57 +01004383 for (i = 0; i < (int)leader_len; i++)
glepnirb9de1a02024-08-02 19:14:38 +02004384 {
4385 if (leader[i] == '/')
4386 leader[i] = '\\';
4387 }
4388 }
4389#endif
4390 last_sep = vim_strrchr(leader, pathsep);
glepnir0be03e12024-07-19 16:45:05 +02004391 if (last_sep == NULL)
4392 {
4393 // No path separator or separator is the last character,
4394 // fuzzy match the whole leader
John Marriott5e6ea922024-11-23 14:01:57 +01004395 VIM_CLEAR_STRING(compl_pattern);
4396 compl_pattern.string = vim_strnsave((char_u *)"*", 1);
4397 if (compl_pattern.string == NULL)
4398 return;
4399 compl_pattern.length = 1;
glepnir0be03e12024-07-19 16:45:05 +02004400 }
4401 else if (*(last_sep + 1) == '\0')
glepnirf31cfa22025-03-06 21:59:13 +01004402 in_fuzzy_collect = FALSE;
glepnir0be03e12024-07-19 16:45:05 +02004403 else
4404 {
4405 // Split leader into path and file parts
4406 int path_len = last_sep - leader + 1;
John Marriott5e6ea922024-11-23 14:01:57 +01004407 char_u *path_with_wildcard;
4408
4409 path_with_wildcard = alloc(path_len + 2);
glepnir0be03e12024-07-19 16:45:05 +02004410 if (path_with_wildcard != NULL)
4411 {
John Marriott5e6ea922024-11-23 14:01:57 +01004412 vim_snprintf((char *)path_with_wildcard, path_len + 2, "%*.*s*", path_len, path_len, leader);
4413 VIM_CLEAR_STRING(compl_pattern);
4414 compl_pattern.string = path_with_wildcard;
4415 compl_pattern.length = path_len + 1;
glepnir0be03e12024-07-19 16:45:05 +02004416
4417 // Move leader to the file part
4418 leader = last_sep + 1;
John Marriott5e6ea922024-11-23 14:01:57 +01004419 leader_len -= path_len;
glepnir0be03e12024-07-19 16:45:05 +02004420 }
4421 }
glepnir8159fb12024-07-17 20:32:54 +02004422 }
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004423
John Marriott5e6ea922024-11-23 14:01:57 +01004424 if (expand_wildcards(1, &compl_pattern.string, &num_matches, &matches,
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004425 EW_FILE|EW_DIR|EW_ADDSLASH|EW_SILENT) != OK)
4426 return;
4427
4428 // May change home directory back to "~".
John Marriott5e6ea922024-11-23 14:01:57 +01004429 tilde_replace(compl_pattern.string, num_matches, matches);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004430#ifdef BACKSLASH_IN_FILENAME
4431 if (curbuf->b_p_csl[0] != NUL)
4432 {
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004433 for (i = 0; i < num_matches; ++i)
4434 {
glepnir8159fb12024-07-17 20:32:54 +02004435 ptr = matches[i];
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004436 while (*ptr != NUL)
4437 {
4438 if (curbuf->b_p_csl[0] == 's' && *ptr == '\\')
4439 *ptr = '/';
4440 else if (curbuf->b_p_csl[0] == 'b' && *ptr == '/')
4441 *ptr = '\\';
4442 ptr += (*mb_ptr2len)(ptr);
4443 }
4444 }
4445 }
4446#endif
glepnir8159fb12024-07-17 20:32:54 +02004447
glepnirf31cfa22025-03-06 21:59:13 +01004448 if (in_fuzzy_collect)
glepnir8159fb12024-07-17 20:32:54 +02004449 {
4450 ga_init2(&fuzzy_indices, sizeof(int), 10);
4451 compl_fuzzy_scores = (int *)alloc(sizeof(int) * num_matches);
4452
4453 for (i = 0; i < num_matches; i++)
4454 {
4455 ptr = matches[i];
4456 score = fuzzy_match_str(ptr, leader);
4457 if (score > 0)
4458 {
4459 if (ga_grow(&fuzzy_indices, 1) == OK)
4460 {
4461 ((int *)fuzzy_indices.ga_data)[fuzzy_indices.ga_len] = i;
4462 compl_fuzzy_scores[i] = score;
4463 fuzzy_indices.ga_len++;
4464 }
4465 }
4466 }
4467
Christian Brabandt220474d2024-07-20 13:26:44 +02004468 // prevent qsort from deref NULL pointer
4469 if (fuzzy_indices.ga_len > 0)
4470 {
glepnirf31cfa22025-03-06 21:59:13 +01004471 char_u *match = NULL;
Christian Brabandt220474d2024-07-20 13:26:44 +02004472 fuzzy_indices_data = (int *)fuzzy_indices.ga_data;
4473 qsort(fuzzy_indices_data, fuzzy_indices.ga_len, sizeof(int), compare_scores);
glepnir8159fb12024-07-17 20:32:54 +02004474
Christian Brabandt220474d2024-07-20 13:26:44 +02004475 for (i = 0; i < fuzzy_indices.ga_len; ++i)
glepnirf31cfa22025-03-06 21:59:13 +01004476 {
Hirohito Higashi355db992025-04-28 18:07:02 +02004477 match = matches[fuzzy_indices_data[i]];
glepnirf31cfa22025-03-06 21:59:13 +01004478 current_score = compl_fuzzy_scores[fuzzy_indices_data[i]];
4479 if (ins_compl_add(match, -1, NULL, NULL, NULL, dir,
Hirohito Higashi355db992025-04-28 18:07:02 +02004480 CP_FAST | ((p_fic || p_wic) ? CP_ICASE : 0),
4481 FALSE, NULL, current_score) == OK)
glepnirf31cfa22025-03-06 21:59:13 +01004482 dir = FORWARD;
4483
4484 if (need_collect_bests)
4485 {
4486 if (i == 0 || current_score == max_score)
4487 {
4488 compl_num_bests++;
4489 max_score = current_score;
4490 }
4491 }
4492 }
glepnir8159fb12024-07-17 20:32:54 +02004493
Christian Brabandt220474d2024-07-20 13:26:44 +02004494 FreeWild(num_matches, matches);
Christian Brabandt220474d2024-07-20 13:26:44 +02004495 }
glepnir6b6280c2024-07-27 16:25:45 +02004496 else if (leader_len > 0)
4497 {
4498 FreeWild(num_matches, matches);
4499 num_matches = 0;
4500 }
Christian Brabandt220474d2024-07-20 13:26:44 +02004501
glepnir8159fb12024-07-17 20:32:54 +02004502 vim_free(compl_fuzzy_scores);
4503 ga_clear(&fuzzy_indices);
glepnirf31cfa22025-03-06 21:59:13 +01004504
4505 if (compl_num_bests > 0 && compl_get_longest)
4506 fuzzy_longest_match();
4507 return;
glepnir8159fb12024-07-17 20:32:54 +02004508 }
4509
glepnir6b6280c2024-07-27 16:25:45 +02004510 if (num_matches > 0)
4511 ins_compl_add_matches(num_matches, matches, p_fic || p_wic);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004512}
4513
4514/*
4515 * Get the next set of command-line completions matching "compl_pattern".
4516 */
4517 static void
Yegappan Lakshmanana23a11b2023-02-21 14:27:41 +00004518get_next_cmdline_completion(void)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004519{
4520 char_u **matches;
4521 int num_matches;
4522
John Marriott5e6ea922024-11-23 14:01:57 +01004523 if (expand_cmdline(&compl_xp, compl_pattern.string,
4524 (int)compl_pattern.length, &num_matches, &matches) == EXPAND_OK)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004525 ins_compl_add_matches(num_matches, matches, FALSE);
4526}
4527
4528/*
4529 * Get the next set of spell suggestions matching "compl_pattern".
4530 */
4531 static void
4532get_next_spell_completion(linenr_T lnum UNUSED)
4533{
4534#ifdef FEAT_SPELL
4535 char_u **matches;
4536 int num_matches;
4537
John Marriott5e6ea922024-11-23 14:01:57 +01004538 num_matches = expand_spelling(lnum, compl_pattern.string, &matches);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004539 if (num_matches > 0)
4540 ins_compl_add_matches(num_matches, matches, p_ic);
Bram Moolenaar8e7cc6b2021-12-30 10:32:25 +00004541 else
4542 vim_free(matches);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004543#endif
4544}
4545
4546/*
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004547 * Return the next word or line from buffer "ins_buf" at position
4548 * "cur_match_pos" for completion. The length of the match is set in "len".
4549 */
4550 static char_u *
zeertzjq440d4cb2023-03-02 17:51:32 +00004551ins_compl_get_next_word_or_line(
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004552 buf_T *ins_buf, // buffer being scanned
4553 pos_T *cur_match_pos, // current match position
4554 int *match_len,
4555 int *cont_s_ipos) // next ^X<> will set initial_pos
4556{
4557 char_u *ptr;
4558 int len;
4559
4560 *match_len = 0;
4561 ptr = ml_get_buf(ins_buf, cur_match_pos->lnum, FALSE) +
4562 cur_match_pos->col;
John Marriott5e6ea922024-11-23 14:01:57 +01004563 len = (int)ml_get_buf_len(ins_buf, cur_match_pos->lnum) - cur_match_pos->col;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004564 if (ctrl_x_mode_line_or_eval())
4565 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00004566 if (compl_status_adding())
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004567 {
4568 if (cur_match_pos->lnum >= ins_buf->b_ml.ml_line_count)
4569 return NULL;
4570 ptr = ml_get_buf(ins_buf, cur_match_pos->lnum + 1, FALSE);
John Marriott5e6ea922024-11-23 14:01:57 +01004571 len = ml_get_buf_len(ins_buf, cur_match_pos->lnum + 1);
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004572 if (!p_paste)
John Marriott5e6ea922024-11-23 14:01:57 +01004573 {
4574 char_u *tmp_ptr = ptr;
4575
4576 ptr = skipwhite(tmp_ptr);
4577 len -= (int)(ptr - tmp_ptr);
4578 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004579 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004580 }
4581 else
4582 {
4583 char_u *tmp_ptr = ptr;
4584
John Marriott5e6ea922024-11-23 14:01:57 +01004585 if (compl_status_adding() && compl_length <= len)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004586 {
4587 tmp_ptr += compl_length;
4588 // Skip if already inside a word.
4589 if (vim_iswordp(tmp_ptr))
4590 return NULL;
4591 // Find start of next word.
4592 tmp_ptr = find_word_start(tmp_ptr);
4593 }
4594 // Find end of this word.
4595 tmp_ptr = find_word_end(tmp_ptr);
4596 len = (int)(tmp_ptr - ptr);
4597
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00004598 if (compl_status_adding() && len == compl_length)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004599 {
4600 if (cur_match_pos->lnum < ins_buf->b_ml.ml_line_count)
4601 {
4602 // Try next line, if any. the new word will be
4603 // "join" as if the normal command "J" was used.
4604 // IOSIZE is always greater than
4605 // compl_length, so the next STRNCPY always
4606 // works -- Acevedo
4607 STRNCPY(IObuff, ptr, len);
4608 ptr = ml_get_buf(ins_buf, cur_match_pos->lnum + 1, FALSE);
4609 tmp_ptr = ptr = skipwhite(ptr);
4610 // Find start of next word.
4611 tmp_ptr = find_word_start(tmp_ptr);
4612 // Find end of next word.
4613 tmp_ptr = find_word_end(tmp_ptr);
4614 if (tmp_ptr > ptr)
4615 {
4616 if (*ptr != ')' && IObuff[len - 1] != TAB)
4617 {
4618 if (IObuff[len - 1] != ' ')
4619 IObuff[len++] = ' ';
4620 // IObuf =~ "\k.* ", thus len >= 2
4621 if (p_js
4622 && (IObuff[len - 2] == '.'
4623 || (vim_strchr(p_cpo, CPO_JOINSP)
4624 == NULL
4625 && (IObuff[len - 2] == '?'
4626 || IObuff[len - 2] == '!'))))
4627 IObuff[len++] = ' ';
4628 }
4629 // copy as much as possible of the new word
4630 if (tmp_ptr - ptr >= IOSIZE - len)
4631 tmp_ptr = ptr + IOSIZE - len - 1;
4632 STRNCPY(IObuff + len, ptr, tmp_ptr - ptr);
4633 len += (int)(tmp_ptr - ptr);
4634 *cont_s_ipos = TRUE;
4635 }
4636 IObuff[len] = NUL;
4637 ptr = IObuff;
4638 }
4639 if (len == compl_length)
4640 return NULL;
4641 }
4642 }
4643
4644 *match_len = len;
4645 return ptr;
4646}
4647
4648/*
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004649 * Get the next set of words matching "compl_pattern" for default completion(s)
4650 * (normal ^P/^N and ^X^L).
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004651 * Search for "compl_pattern" in the buffer "st->ins_buf" starting from the
4652 * position "st->start_pos" in the "compl_direction" direction. If
4653 * "st->set_match_pos" is TRUE, then set the "st->first_match_pos" and
4654 * "st->last_match_pos".
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004655 * Returns OK if a new next match is found, otherwise returns FAIL.
4656 */
4657 static int
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004658get_next_default_completion(ins_compl_next_state_T *st, pos_T *start_pos)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004659{
4660 int found_new_match = FAIL;
4661 int save_p_scs;
4662 int save_p_ws;
4663 int looped_around = FALSE;
glepnir8159fb12024-07-17 20:32:54 +02004664 char_u *ptr = NULL;
4665 int len = 0;
glepnirf31cfa22025-03-06 21:59:13 +01004666 int in_collect = (cfc_has_mode() && compl_length > 0);
glepnir8159fb12024-07-17 20:32:54 +02004667 char_u *leader = ins_compl_leader();
glepnirf31cfa22025-03-06 21:59:13 +01004668 int score = 0;
Girish Palyab1565882025-04-15 20:16:00 +02004669 int in_curbuf = st->ins_buf == curbuf;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004670
4671 // If 'infercase' is set, don't use 'smartcase' here
4672 save_p_scs = p_scs;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004673 if (st->ins_buf->b_p_inf)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004674 p_scs = FALSE;
4675
4676 // Buffers other than curbuf are scanned from the beginning or the
4677 // end but never from the middle, thus setting nowrapscan in this
4678 // buffer is a good idea, on the other hand, we always set
4679 // wrapscan for curbuf to avoid missing matches -- Acevedo,Webb
4680 save_p_ws = p_ws;
Girish Palyab1565882025-04-15 20:16:00 +02004681 if (!in_curbuf)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004682 p_ws = FALSE;
glepnirf31cfa22025-03-06 21:59:13 +01004683 else if (*st->e_cpt == '.')
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004684 p_ws = TRUE;
4685 looped_around = FALSE;
4686 for (;;)
4687 {
4688 int cont_s_ipos = FALSE;
4689
4690 ++msg_silent; // Don't want messages for wrapscan.
4691
glepnirf31cfa22025-03-06 21:59:13 +01004692 if (in_collect)
4693 {
Hirohito Higashi355db992025-04-28 18:07:02 +02004694 found_new_match = search_for_fuzzy_match(st->ins_buf,
glepnir8159fb12024-07-17 20:32:54 +02004695 st->cur_match_pos, leader, compl_direction,
glepnirf31cfa22025-03-06 21:59:13 +01004696 start_pos, &len, &ptr, &score);
4697 }
4698 // ctrl_x_mode_line_or_eval() || word-wise search that
4699 // has added a word that was at the beginning of the line
4700 else if (ctrl_x_mode_whole_line() || ctrl_x_mode_eval() || (compl_cont_status & CONT_SOL))
4701 found_new_match = search_for_exact_line(st->ins_buf,
4702 st->cur_match_pos, compl_direction, compl_pattern.string);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004703 else
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004704 found_new_match = searchit(NULL, st->ins_buf, st->cur_match_pos,
John Marriott5e6ea922024-11-23 14:01:57 +01004705 NULL, compl_direction, compl_pattern.string, (int)compl_pattern.length,
John Marriott8c85a2a2024-05-20 19:18:26 +02004706 1L, SEARCH_KEEP + SEARCH_NFMSG, RE_LAST, NULL);
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004707 --msg_silent;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004708 if (!compl_started || st->set_match_pos)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004709 {
4710 // set "compl_started" even on fail
4711 compl_started = TRUE;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004712 st->first_match_pos = *st->cur_match_pos;
4713 st->last_match_pos = *st->cur_match_pos;
4714 st->set_match_pos = FALSE;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004715 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004716 else if (st->first_match_pos.lnum == st->last_match_pos.lnum
4717 && st->first_match_pos.col == st->last_match_pos.col)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004718 {
4719 found_new_match = FAIL;
4720 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00004721 else if (compl_dir_forward()
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004722 && (st->prev_match_pos.lnum > st->cur_match_pos->lnum
4723 || (st->prev_match_pos.lnum == st->cur_match_pos->lnum
4724 && st->prev_match_pos.col >= st->cur_match_pos->col)))
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004725 {
4726 if (looped_around)
4727 found_new_match = FAIL;
4728 else
4729 looped_around = TRUE;
4730 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00004731 else if (!compl_dir_forward()
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004732 && (st->prev_match_pos.lnum < st->cur_match_pos->lnum
4733 || (st->prev_match_pos.lnum == st->cur_match_pos->lnum
4734 && st->prev_match_pos.col <= st->cur_match_pos->col)))
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004735 {
4736 if (looped_around)
4737 found_new_match = FAIL;
4738 else
4739 looped_around = TRUE;
4740 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004741 st->prev_match_pos = *st->cur_match_pos;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004742 if (found_new_match == FAIL)
4743 break;
4744
4745 // when ADDING, the text before the cursor matches, skip it
Girish Palyab1565882025-04-15 20:16:00 +02004746 if (compl_status_adding() && in_curbuf
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004747 && start_pos->lnum == st->cur_match_pos->lnum
4748 && start_pos->col == st->cur_match_pos->col)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004749 continue;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004750
glepnirf31cfa22025-03-06 21:59:13 +01004751 if (!in_collect)
glepnir8159fb12024-07-17 20:32:54 +02004752 ptr = ins_compl_get_next_word_or_line(st->ins_buf, st->cur_match_pos,
4753 &len, &cont_s_ipos);
glepnirbfb4eea2025-01-31 15:28:29 +01004754 if (ptr == NULL || (ins_compl_has_preinsert() && STRCMP(ptr, compl_pattern.string) == 0))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00004755 continue;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004756
Girish Palyab1565882025-04-15 20:16:00 +02004757 if (is_nearest_active() && in_curbuf)
4758 {
4759 score = st->cur_match_pos->lnum - curwin->w_cursor.lnum;
4760 if (score < 0)
4761 score = -score;
4762 score++;
4763 }
4764
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004765 if (ins_compl_add_infercase(ptr, len, p_ic,
Girish Palyab1565882025-04-15 20:16:00 +02004766 in_curbuf ? NULL : st->ins_buf->b_sfname,
glepnirf31cfa22025-03-06 21:59:13 +01004767 0, cont_s_ipos, score) != NOTDONE)
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004768 {
glepnirf31cfa22025-03-06 21:59:13 +01004769 if (in_collect && score == compl_first_match->cp_next->cp_score)
4770 compl_num_bests++;
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004771 found_new_match = OK;
4772 break;
4773 }
4774 }
4775 p_scs = save_p_scs;
4776 p_ws = save_p_ws;
4777
4778 return found_new_match;
4779}
4780
Girish Palyacbe53192025-04-14 22:13:15 +02004781#ifdef FEAT_COMPL_FUNC
Girish Palya98c29db2025-06-01 19:40:00 +02004782/*
4783 * Return the callback function associated with "p" if it points to a
4784 * userfunc.
4785 */
Girish Palyacbe53192025-04-14 22:13:15 +02004786 static callback_T *
Girish Palya98c29db2025-06-01 19:40:00 +02004787get_callback_if_cpt_func(char_u *p)
Girish Palyacbe53192025-04-14 22:13:15 +02004788{
4789 static callback_T cb;
4790 char_u buf[LSIZE];
4791 int slen;
4792
Girish Palya98c29db2025-06-01 19:40:00 +02004793 if (*p == 'o')
4794 return &curbuf->b_ofu_cb;
4795 if (*p == 'F')
4796 {
4797 if (*++p != ',' && *p != NUL)
4798 {
4799 free_callback(&cb);
4800 slen = copy_option_part(&p, buf, LSIZE, ",");
4801 if (slen > 0 && option_set_callback_func(buf, &cb))
4802 return &cb;
4803 return NULL;
4804 }
4805 else
4806 return &curbuf->b_cfu_cb;
4807 }
Girish Palyacbe53192025-04-14 22:13:15 +02004808 return NULL;
4809}
Girish Palyacbe53192025-04-14 22:13:15 +02004810#endif
4811
4812/*
glepnir05460682025-05-26 18:23:27 +02004813 * Get completion matches from register contents.
4814 * Extracts words from all available registers and adds them to the completion list.
4815 */
4816 static void
4817get_register_completion(void)
4818{
4819 int dir = compl_direction;
4820 yankreg_T *reg = NULL;
4821 void *reg_ptr = NULL;
glepnir86d46a72025-06-03 21:04:44 +02004822 int adding_mode = compl_status_adding();
glepnir05460682025-05-26 18:23:27 +02004823
4824 for (int i = 0; i < NUM_REGISTERS; i++)
4825 {
4826 int regname = 0;
glepnir05460682025-05-26 18:23:27 +02004827 if (i == 0)
4828 regname = '"'; // unnamed register
4829 else if (i < 10)
4830 regname = '0' + i;
4831 else if (i == DELETION_REGISTER)
4832 regname = '-';
4833#ifdef FEAT_CLIPBOARD
4834 else if (i == STAR_REGISTER)
4835 regname = '*';
4836 else if (i == PLUS_REGISTER)
4837 regname = '+';
4838#endif
4839 else
4840 regname = 'a' + i - 10;
4841
4842 // Skip invalid or black hole register
4843 if (!valid_yank_reg(regname, FALSE) || regname == '_')
4844 continue;
4845
4846 reg_ptr = get_register(regname, FALSE);
4847 if (reg_ptr == NULL)
4848 continue;
4849
4850 reg = (yankreg_T *)reg_ptr;
4851
glepnir86d46a72025-06-03 21:04:44 +02004852 for (int j = 0; j < reg->y_size; j++)
glepnir05460682025-05-26 18:23:27 +02004853 {
glepnir86d46a72025-06-03 21:04:44 +02004854 char_u *str = reg->y_array[j].string;
4855 if (str == NULL)
4856 continue;
glepnir05460682025-05-26 18:23:27 +02004857
glepnir86d46a72025-06-03 21:04:44 +02004858 if (adding_mode)
4859 {
glepnird5fdfa52025-06-02 19:45:41 +02004860 int str_len = (int)STRLEN(str);
4861 if (str_len == 0)
4862 continue;
glepnir05460682025-05-26 18:23:27 +02004863
glepnird5fdfa52025-06-02 19:45:41 +02004864 if (!compl_orig_text.string
4865 || (p_ic ? STRNICMP(str, compl_orig_text.string,
4866 compl_orig_text.length) == 0
4867 : STRNCMP(str, compl_orig_text.string,
4868 compl_orig_text.length) == 0))
glepnir05460682025-05-26 18:23:27 +02004869 {
glepnir86d46a72025-06-03 21:04:44 +02004870 if (ins_compl_add_infercase(str, str_len, p_ic, NULL,
4871 dir, FALSE, 0) == OK)
glepnir05460682025-05-26 18:23:27 +02004872 dir = FORWARD;
4873 }
glepnird5fdfa52025-06-02 19:45:41 +02004874 }
glepnir86d46a72025-06-03 21:04:44 +02004875 else
glepnird5fdfa52025-06-02 19:45:41 +02004876 {
glepnird5fdfa52025-06-02 19:45:41 +02004877 // Calculate the safe end of string to avoid null byte issues
4878 char_u *str_end = str + STRLEN(str);
4879 char_u *p = str;
4880
4881 // Safely iterate through the string
4882 while (p < str_end && *p != NUL)
4883 {
4884 char_u *old_p = p;
4885 p = find_word_start(p);
4886 if (p >= str_end || *p == NUL)
4887 break;
4888
4889 char_u *word_end = find_word_end(p);
4890
4891 if (word_end <= p)
4892 {
4893 if (has_mbyte)
4894 word_end = p + (*mb_ptr2len)(p);
4895 else
4896 word_end = p + 1;
4897 }
4898
4899 if (word_end > str_end)
4900 word_end = str_end;
4901
4902 int len = (int)(word_end - p);
4903 if (len > 0 && (!compl_orig_text.string
4904 || (p_ic ? STRNICMP(p, compl_orig_text.string,
4905 compl_orig_text.length) == 0
4906 : STRNCMP(p, compl_orig_text.string,
4907 compl_orig_text.length) == 0)))
4908 {
4909 if (ins_compl_add_infercase(p, len, p_ic, NULL,
4910 dir, FALSE, 0) == OK)
4911 dir = FORWARD;
4912 }
4913
4914 p = word_end;
4915
4916 if (p <= old_p)
4917 {
4918 p = old_p + 1;
4919 if (has_mbyte && p < str_end)
4920 p = old_p + (*mb_ptr2len)(old_p);
4921 }
4922 }
glepnir05460682025-05-26 18:23:27 +02004923 }
4924 }
4925
4926 // Free the register copy
4927 put_register(regname, reg_ptr);
4928 }
4929}
4930
4931/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00004932 * get the next set of completion matches for "type".
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004933 * Returns TRUE if a new match is found. Otherwise returns FALSE.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004934 */
4935 static int
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004936get_next_completion_match(int type, ins_compl_next_state_T *st, pos_T *ini)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004937{
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004938 int found_new_match = FALSE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004939
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004940 switch (type)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004941 {
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004942 case -1:
4943 break;
4944#ifdef FEAT_FIND_ID
4945 case CTRL_X_PATH_PATTERNS:
4946 case CTRL_X_PATH_DEFINES:
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004947 get_next_include_file_completion(type);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004948 break;
4949#endif
4950
4951 case CTRL_X_DICTIONARY:
4952 case CTRL_X_THESAURUS:
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004953 get_next_dict_tsr_completion(type, st->dict, st->dict_f);
4954 st->dict = NULL;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004955 break;
4956
4957 case CTRL_X_TAGS:
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004958 get_next_tag_completion();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004959 break;
4960
4961 case CTRL_X_FILES:
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004962 get_next_filename_completion();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004963 break;
4964
4965 case CTRL_X_CMDLINE:
zeertzjqdca29d92021-08-31 19:12:51 +02004966 case CTRL_X_CMDLINE_CTRL_X:
Yegappan Lakshmananedc6f102021-12-29 17:38:46 +00004967 get_next_cmdline_completion();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004968 break;
4969
4970#ifdef FEAT_COMPL_FUNC
4971 case CTRL_X_FUNCTION:
Girish Palyacbe53192025-04-14 22:13:15 +02004972 if (ctrl_x_mode_normal()) // Invoked by a func in 'cpt' option
Girish Palyafa16c7a2025-06-30 20:14:43 +02004973 get_cpt_func_completion_matches(st->func_cb, TRUE);
Girish Palyacbe53192025-04-14 22:13:15 +02004974 else
4975 expand_by_function(type, compl_pattern.string, NULL);
4976 break;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004977 case CTRL_X_OMNI:
Girish Palyacbe53192025-04-14 22:13:15 +02004978 expand_by_function(type, compl_pattern.string, NULL);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004979 break;
4980#endif
4981
4982 case CTRL_X_SPELL:
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004983 get_next_spell_completion(st->first_match_pos.lnum);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004984 break;
4985
glepnir05460682025-05-26 18:23:27 +02004986 case CTRL_X_REGISTER:
4987 get_register_completion();
4988 break;
4989
Bram Moolenaar7591bb32019-03-30 13:53:47 +01004990 default: // normal ^P/^N and ^X^L
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00004991 found_new_match = get_next_default_completion(st, ini);
4992 if (found_new_match == FAIL && st->ins_buf == curbuf)
4993 st->found_all = TRUE;
4994 }
4995
4996 // check if compl_curr_match has changed, (e.g. other type of
4997 // expansion added something)
4998 if (type != 0 && compl_curr_match != compl_old_match)
4999 found_new_match = OK;
5000
5001 return found_new_match;
5002}
5003
5004/*
Girish Palya0ac1eb32025-04-16 20:18:33 +02005005 * Strips carets followed by numbers. This suffix typically represents the
5006 * max_matches setting.
5007 */
5008 static void
5009strip_caret_numbers_in_place(char_u *str)
5010{
5011 char_u *read = str, *write = str, *p;
5012
5013 if (str == NULL)
5014 return;
5015
5016 while (*read)
5017 {
5018 if (*read == '^')
5019 {
5020 p = read + 1;
5021 while (vim_isdigit(*p))
5022 p++;
5023 if ((*p == ',' || *p == '\0') && p != read + 1)
5024 {
5025 read = p;
5026 continue;
5027 }
5028 else
5029 *write++ = *read++;
5030 }
5031 else
5032 *write++ = *read++;
5033 }
5034 *write = '\0';
5035}
5036
5037/*
Girish Palya98c29db2025-06-01 19:40:00 +02005038 * Call functions specified in the 'cpt' option with findstart=1,
5039 * and retrieve the startcol.
5040 */
5041 static void
5042prepare_cpt_compl_funcs(void)
5043{
5044#ifdef FEAT_COMPL_FUNC
5045 char_u *cpt;
5046 char_u *p;
5047 callback_T *cb = NULL;
5048 int idx = 0;
5049 int startcol;
5050
5051 // Make a copy of 'cpt' in case the buffer gets wiped out
5052 cpt = vim_strsave(curbuf->b_p_cpt);
5053 strip_caret_numbers_in_place(cpt);
5054
5055 // Re-insert the text removed by ins_compl_delete().
5056 ins_compl_insert_bytes(compl_orig_text.string + get_compl_len(), -1);
5057
5058 for (p = cpt; *p;)
5059 {
5060 while (*p == ',' || *p == ' ') // Skip delimiters
5061 p++;
5062 cb = get_callback_if_cpt_func(p);
5063 if (cb)
5064 {
5065 if (get_userdefined_compl_info(curwin->w_cursor.col, cb, &startcol)
5066 == FAIL)
5067 {
5068 if (startcol == -3)
5069 cpt_sources_array[idx].cs_refresh_always = FALSE;
5070 else
5071 startcol = -2;
5072 }
5073 cpt_sources_array[idx].cs_startcol = startcol;
5074 }
5075 (void)copy_option_part(&p, IObuff, IOSIZE, ","); // Advance p
5076 idx++;
5077 }
5078
5079 // Undo insertion
5080 ins_compl_delete();
5081
5082 vim_free(cpt);
5083#endif
5084}
5085
5086/*
Girish Palya7c621052025-05-26 19:41:59 +02005087 * Safely advance the cpt_sources_index by one.
5088 */
5089 static int
5090advance_cpt_sources_index_safe(void)
5091{
5092 if (cpt_sources_index < cpt_sources_count - 1)
5093 {
5094 cpt_sources_index++;
5095 return OK;
5096 }
5097#ifdef FEAT_EVAL
5098 semsg(_(e_list_index_out_of_range_nr), cpt_sources_index + 1);
5099#endif
5100 return FAIL;
5101}
5102
5103/*
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005104 * Get the next expansion(s), using "compl_pattern".
5105 * The search starts at position "ini" in curbuf and in the direction
5106 * compl_direction.
5107 * When "compl_started" is FALSE start at that position, otherwise continue
5108 * where we stopped searching before.
5109 * This may return before finding all the matches.
5110 * Return the total number of matches or -1 if still unknown -- Acevedo
5111 */
5112 static int
5113ins_compl_get_exp(pos_T *ini)
5114{
Bram Moolenaar0ff01832022-09-24 19:20:30 +01005115 static ins_compl_next_state_T st;
5116 static int st_cleared = FALSE;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005117 int i;
5118 int found_new_match;
5119 int type = ctrl_x_mode;
Girish Palya7c621052025-05-26 19:41:59 +02005120 int may_advance_cpt_idx = FALSE;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005121
5122 if (!compl_started)
5123 {
Bram Moolenaar0ff01832022-09-24 19:20:30 +01005124 buf_T *buf;
5125
5126 FOR_ALL_BUFFERS(buf)
5127 buf->b_scanned = 0;
5128 if (!st_cleared)
5129 {
5130 CLEAR_FIELD(st);
5131 st_cleared = TRUE;
5132 }
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005133 st.found_all = FALSE;
5134 st.ins_buf = curbuf;
Bram Moolenaar0ff01832022-09-24 19:20:30 +01005135 vim_free(st.e_cpt_copy);
Christian Brabandtee17b6f2023-09-09 11:23:50 +02005136 // Make a copy of 'complete', in case the buffer is wiped out.
Bram Moolenaar0ff01832022-09-24 19:20:30 +01005137 st.e_cpt_copy = vim_strsave((compl_cont_status & CONT_LOCAL)
5138 ? (char_u *)"." : curbuf->b_p_cpt);
Girish Palya0ac1eb32025-04-16 20:18:33 +02005139 strip_caret_numbers_in_place(st.e_cpt_copy);
Bram Moolenaar0ff01832022-09-24 19:20:30 +01005140 st.e_cpt = st.e_cpt_copy == NULL ? (char_u *)"" : st.e_cpt_copy;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005141 st.last_match_pos = st.first_match_pos = *ini;
5142 }
5143 else if (st.ins_buf != curbuf && !buf_valid(st.ins_buf))
5144 st.ins_buf = curbuf; // In case the buffer was wiped out.
5145
5146 compl_old_match = compl_curr_match; // remember the last current match
Christian Brabandt13032a42024-07-28 21:16:48 +02005147 st.cur_match_pos = (compl_dir_forward())
glepnir8159fb12024-07-17 20:32:54 +02005148 ? &st.last_match_pos : &st.first_match_pos;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005149
Girish Palya98c29db2025-06-01 19:40:00 +02005150 if (ctrl_x_mode_normal() && !ctrl_x_mode_line_or_eval() &&
5151 !(compl_cont_status & CONT_LOCAL))
5152 {
5153 // ^N completion, not ^X^L or complete() or ^X^N
5154 if (!compl_started) // Before showing menu the first time
5155 {
5156 if (setup_cpt_sources() == FAIL)
5157 return FAIL;
5158 }
5159 prepare_cpt_compl_funcs();
5160 cpt_sources_index = 0;
5161 }
5162
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005163 // For ^N/^P loop over all the flags/windows/buffers in 'complete'.
Girish Palya7c621052025-05-26 19:41:59 +02005164 for (;;)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005165 {
5166 found_new_match = FAIL;
5167 st.set_match_pos = FALSE;
5168
5169 // For ^N/^P pick a new entry from e_cpt if compl_started is off,
5170 // or if found_all says this entry is done. For ^X^L only use the
5171 // entries from 'complete' that look in loaded buffers.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005172 if ((ctrl_x_mode_normal() || ctrl_x_mode_line_or_eval())
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005173 && (!compl_started || st.found_all))
5174 {
Girish Palya7c621052025-05-26 19:41:59 +02005175 int status = process_next_cpt_value(&st, &type, ini,
5176 cfc_has_mode(), &may_advance_cpt_idx);
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005177
5178 if (status == INS_COMPL_CPT_END)
5179 break;
5180 if (status == INS_COMPL_CPT_CONT)
Girish Palyacbe53192025-04-14 22:13:15 +02005181 {
Girish Palya7c621052025-05-26 19:41:59 +02005182 if (may_advance_cpt_idx && !advance_cpt_sources_index_safe())
5183 break;
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005184 continue;
Girish Palyacbe53192025-04-14 22:13:15 +02005185 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005186 }
5187
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005188 // If complete() was called then compl_pattern has been reset. The
5189 // following won't work then, bail out.
John Marriott5e6ea922024-11-23 14:01:57 +01005190 if (compl_pattern.string == NULL)
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005191 break;
5192
5193 // get the next set of completion matches
5194 found_new_match = get_next_completion_match(type, &st, ini);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005195
Girish Palya7c621052025-05-26 19:41:59 +02005196 if (may_advance_cpt_idx && !advance_cpt_sources_index_safe())
5197 break;
Girish Palyacbe53192025-04-14 22:13:15 +02005198
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005199 // break the loop for specialized modes (use 'complete' just for the
5200 // generic ctrl_x_mode == CTRL_X_NORMAL) or when we've found a new
5201 // match
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005202 if ((ctrl_x_mode_not_default() && !ctrl_x_mode_line_or_eval())
5203 || found_new_match != FAIL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005204 {
5205 if (got_int)
5206 break;
5207 // Fill the popup menu as soon as possible.
5208 if (type != -1)
5209 ins_compl_check_keys(0, FALSE);
5210
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005211 if ((ctrl_x_mode_not_default()
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005212 && !ctrl_x_mode_line_or_eval()) || compl_interrupted)
5213 break;
5214 compl_started = TRUE;
5215 }
5216 else
5217 {
5218 // Mark a buffer scanned when it has been scanned completely
Christian Brabandtee9166e2023-09-03 21:24:33 +02005219 if (buf_valid(st.ins_buf) && (type == 0 || type == CTRL_X_PATH_PATTERNS))
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005220 st.ins_buf->b_scanned = TRUE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005221
5222 compl_started = FALSE;
5223 }
Girish Palya5fbe72e2025-06-18 19:15:45 +02005224
5225 // For `^P` completion, reset `compl_curr_match` to the head to avoid
5226 // mixing matches from different sources.
5227 if (!compl_dir_forward())
5228 while (compl_curr_match->cp_prev)
5229 compl_curr_match = compl_curr_match->cp_prev;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005230 }
Girish Palya0ac1eb32025-04-16 20:18:33 +02005231 cpt_sources_index = -1;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005232 compl_started = TRUE;
5233
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005234 if ((ctrl_x_mode_normal() || ctrl_x_mode_line_or_eval())
Yegappan Lakshmanan6ad84ab2021-12-31 12:59:53 +00005235 && *st.e_cpt == NUL) // Got to end of 'complete'
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005236 found_new_match = FAIL;
5237
5238 i = -1; // total of matches, unknown
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005239 if (found_new_match == FAIL || (ctrl_x_mode_not_default()
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005240 && !ctrl_x_mode_line_or_eval()))
5241 i = ins_compl_make_cyclic();
5242
glepnirf31cfa22025-03-06 21:59:13 +01005243 if (cfc_has_mode() && compl_get_longest && compl_num_bests > 0)
5244 fuzzy_longest_match();
5245
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005246 if (compl_old_match != NULL)
5247 {
5248 // If several matches were added (FORWARD) or the search failed and has
5249 // just been made cyclic then we have to move compl_curr_match to the
5250 // next or previous entry (if any) -- Acevedo
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005251 compl_curr_match = compl_dir_forward() ? compl_old_match->cp_next
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005252 : compl_old_match->cp_prev;
5253 if (compl_curr_match == NULL)
5254 compl_curr_match = compl_old_match;
5255 }
LemonBoy2bf52dd2022-04-09 18:17:34 +01005256 may_trigger_modechanged();
=?UTF-8?q?Magnus=20Gro=C3=9F?=25def2c2021-10-22 18:56:39 +01005257
Girish Palyab8ee1cf2025-06-08 16:20:06 +02005258 if (is_nearest_active())
5259 sort_compl_match_list(cp_compare_nearest);
5260
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005261 return i;
5262}
5263
5264/*
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005265 * Update "compl_shown_match" to the actually shown match, it may differ when
5266 * "compl_leader" is used to omit some of the matches.
5267 */
5268 static void
5269ins_compl_update_shown_match(void)
5270{
5271 while (!ins_compl_equal(compl_shown_match,
John Marriott5e6ea922024-11-23 14:01:57 +01005272 compl_leader.string, (int)compl_leader.length)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005273 && compl_shown_match->cp_next != NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005274 && !is_first_match(compl_shown_match->cp_next))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005275 compl_shown_match = compl_shown_match->cp_next;
5276
5277 // If we didn't find it searching forward, and compl_shows_dir is
5278 // backward, find the last match.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005279 if (compl_shows_dir_backward()
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005280 && !ins_compl_equal(compl_shown_match,
John Marriott5e6ea922024-11-23 14:01:57 +01005281 compl_leader.string, (int)compl_leader.length)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005282 && (compl_shown_match->cp_next == NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005283 || is_first_match(compl_shown_match->cp_next)))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005284 {
5285 while (!ins_compl_equal(compl_shown_match,
John Marriott5e6ea922024-11-23 14:01:57 +01005286 compl_leader.string, (int)compl_leader.length)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005287 && compl_shown_match->cp_prev != NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005288 && !is_first_match(compl_shown_match->cp_prev))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005289 compl_shown_match = compl_shown_match->cp_prev;
5290 }
5291}
5292
5293/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005294 * Delete the old text being completed.
5295 */
5296 void
5297ins_compl_delete(void)
5298{
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005299 // In insert mode: Delete the typed part.
5300 // In replace mode: Put the old characters back, if any.
glepniredd4ac32025-01-29 18:53:51 +01005301 int col = compl_col + (compl_status_adding() ? compl_length : 0);
John Marriottf4b36412025-02-23 09:09:59 +01005302 string_T remaining = {NULL, 0};
glepnir76bdb822025-02-08 19:04:51 +01005303 int orig_col;
glepniredd4ac32025-01-29 18:53:51 +01005304 int has_preinsert = ins_compl_preinsert_effect();
5305 if (has_preinsert)
5306 {
Yegappan Lakshmanan7b6add02025-04-01 20:38:37 +02005307 col += (int)ins_compl_leader_len();
glepniredd4ac32025-01-29 18:53:51 +01005308 curwin->w_cursor.col = compl_ins_end_col;
5309 }
5310
glepnir76bdb822025-02-08 19:04:51 +01005311 if (curwin->w_cursor.lnum > compl_lnum)
5312 {
5313 if (curwin->w_cursor.col < ml_get_curline_len())
5314 {
John Marriottf4b36412025-02-23 09:09:59 +01005315 char_u *line = ml_get_cursor();
5316 remaining.length = ml_get_cursor_len();
5317 remaining.string = vim_strnsave(line, remaining.length);
5318 if (remaining.string == NULL)
glepnir76bdb822025-02-08 19:04:51 +01005319 return;
5320 }
5321 while (curwin->w_cursor.lnum > compl_lnum)
5322 {
5323 if (ml_delete(curwin->w_cursor.lnum) == FAIL)
5324 {
John Marriottf4b36412025-02-23 09:09:59 +01005325 if (remaining.string)
5326 vim_free(remaining.string);
glepnir76bdb822025-02-08 19:04:51 +01005327 return;
5328 }
zeertzjq060e6552025-02-21 20:06:26 +01005329 deleted_lines_mark(curwin->w_cursor.lnum, 1L);
glepnir76bdb822025-02-08 19:04:51 +01005330 curwin->w_cursor.lnum--;
5331 }
5332 // move cursor to end of line
5333 curwin->w_cursor.col = ml_get_curline_len();
5334 }
5335
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005336 if ((int)curwin->w_cursor.col > col)
5337 {
5338 if (stop_arrow() == FAIL)
glepnire3647c82025-02-10 21:16:32 +01005339 {
John Marriottf4b36412025-02-23 09:09:59 +01005340 if (remaining.string)
5341 vim_free(remaining.string);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005342 return;
glepnire3647c82025-02-10 21:16:32 +01005343 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005344 backspace_until_column(col);
zeertzjqf25d8f92024-12-18 21:12:25 +01005345 compl_ins_end_col = curwin->w_cursor.col;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005346 }
5347
John Marriottf4b36412025-02-23 09:09:59 +01005348 if (remaining.string != NULL)
glepnir76bdb822025-02-08 19:04:51 +01005349 {
5350 orig_col = curwin->w_cursor.col;
John Marriottf4b36412025-02-23 09:09:59 +01005351 ins_str(remaining.string, remaining.length);
glepnir76bdb822025-02-08 19:04:51 +01005352 curwin->w_cursor.col = orig_col;
John Marriottf4b36412025-02-23 09:09:59 +01005353 vim_free(remaining.string);
glepnir76bdb822025-02-08 19:04:51 +01005354 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005355 // TODO: is this sufficient for redrawing? Redrawing everything causes
5356 // flicker, thus we can't do that.
5357 changed_cline_bef_curs();
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02005358#ifdef FEAT_EVAL
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005359 // clear v:completed_item
5360 set_vim_var_dict(VV_COMPLETED_ITEM, dict_alloc_lock(VAR_FIXED));
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02005361#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005362}
5363
5364/*
glepnir76bdb822025-02-08 19:04:51 +01005365 * Insert a completion string that contains newlines.
5366 * The string is split and inserted line by line.
5367 */
5368 static void
5369ins_compl_expand_multiple(char_u *str)
5370{
5371 char_u *start = str;
5372 char_u *curr = str;
glepnir5090a1f2025-02-24 19:10:37 +01005373 int base_indent = get_indent();
glepnir76bdb822025-02-08 19:04:51 +01005374
5375 while (*curr != NUL)
5376 {
5377 if (*curr == '\n')
5378 {
5379 // Insert the text chunk before newline
5380 if (curr > start)
5381 ins_char_bytes(start, (int)(curr - start));
5382
5383 // Handle newline
glepnir5090a1f2025-02-24 19:10:37 +01005384 open_line(FORWARD, OPENLINE_KEEPTRAIL | OPENLINE_FORCE_INDENT, base_indent, NULL);
glepnir76bdb822025-02-08 19:04:51 +01005385 start = curr + 1;
5386 }
5387 curr++;
5388 }
5389
5390 // Handle remaining text after last newline (if any)
5391 if (curr > start)
5392 ins_char_bytes(start, (int)(curr - start));
5393
5394 compl_ins_end_col = curwin->w_cursor.col;
5395}
5396
5397/*
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005398 * Insert the new text being completed.
glepniredd4ac32025-01-29 18:53:51 +01005399 * "move_cursor" is used when 'completeopt' includes "preinsert" and when TRUE
5400 * cursor needs to move back from the inserted text to the compl_leader.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005401 */
5402 void
Girish Palya5fbe72e2025-06-18 19:15:45 +02005403ins_compl_insert(int move_cursor)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005404{
glepnir76bdb822025-02-08 19:04:51 +01005405 int compl_len = get_compl_len();
5406 int preinsert = ins_compl_has_preinsert();
5407 char_u *cp_str = compl_shown_match->cp_str.string;
5408 size_t cp_str_len = compl_shown_match->cp_str.length;
5409 size_t leader_len = ins_compl_leader_len();
5410 char_u *has_multiple = vim_strchr(cp_str, '\n');
Bram Moolenaar4b28ba32021-12-27 19:28:37 +00005411
5412 // Make sure we don't go over the end of the string, this can happen with
5413 // illegal bytes.
zeertzjq8297e2c2025-01-30 11:04:47 +01005414 if (compl_len < (int)cp_str_len)
glepniredd4ac32025-01-29 18:53:51 +01005415 {
glepnir76bdb822025-02-08 19:04:51 +01005416 if (has_multiple)
5417 ins_compl_expand_multiple(cp_str + compl_len);
5418 else
5419 {
5420 ins_compl_insert_bytes(cp_str + compl_len, -1);
5421 if (preinsert && move_cursor)
5422 curwin->w_cursor.col -= (colnr_T)(cp_str_len - leader_len);
5423 }
glepniredd4ac32025-01-29 18:53:51 +01005424 }
5425 if (match_at_original_text(compl_shown_match) || preinsert)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005426 compl_used_match = FALSE;
5427 else
5428 compl_used_match = TRUE;
Bram Moolenaar9cb698d2019-08-21 15:30:45 +02005429#ifdef FEAT_EVAL
5430 {
5431 dict_T *dict = ins_compl_dict_alloc(compl_shown_match);
5432
5433 set_vim_var_dict(VV_COMPLETED_ITEM, dict);
5434 }
5435#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005436}
5437
5438/*
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005439 * show the file name for the completion match (if any). Truncate the file
5440 * name to avoid a wait for return.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005441 */
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005442 static void
5443ins_compl_show_filename(void)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005444{
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005445 char *lead = _("match in file");
5446 int space = sc_col - vim_strsize((char_u *)lead) - 2;
5447 char_u *s;
5448 char_u *e;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005449
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005450 if (space <= 0)
5451 return;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005452
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005453 // We need the tail that fits. With double-byte encoding going
5454 // back from the end is very slow, thus go from the start and keep
5455 // the text that fits in "space" between "s" and "e".
5456 for (s = e = compl_shown_match->cp_fname; *e != NUL; MB_PTR_ADV(e))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005457 {
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005458 space -= ptr2cells(e);
5459 while (space < 0)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005460 {
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005461 space += ptr2cells(s);
5462 MB_PTR_ADV(s);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005463 }
5464 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005465 msg_hist_off = TRUE;
5466 vim_snprintf((char *)IObuff, IOSIZE, "%s %s%s", lead,
5467 s > compl_shown_match->cp_fname ? "<" : "", s);
5468 msg((char *)IObuff);
5469 msg_hist_off = FALSE;
5470 redraw_cmdline = FALSE; // don't overwrite!
5471}
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005472
glepnirdca57fb2024-06-04 22:01:21 +02005473/*
Girish Palya0ac1eb32025-04-16 20:18:33 +02005474 * Find the appropriate completion item when 'complete' ('cpt') includes
5475 * a 'max_matches' postfix. In this case, we search for a match where
5476 * 'cp_in_match_array' is set, indicating that the match is also present
5477 * in 'compl_match_array'.
5478 */
5479 static compl_T *
Girish Palyab8ee1cf2025-06-08 16:20:06 +02005480find_next_match_in_menu(void)
Girish Palya0ac1eb32025-04-16 20:18:33 +02005481{
5482 int is_forward = compl_shows_dir_forward();
5483 compl_T *match = compl_shown_match;
5484
5485 do
5486 match = is_forward ? match->cp_next : match->cp_prev;
5487 while (match->cp_next && !match->cp_in_match_array
5488 && !match_at_original_text(match));
5489 return match;
5490}
5491
5492/*
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005493 * Find the next set of matches for completion. Repeat the completion "todo"
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005494 * times. The number of matches found is returned in 'num_matches'.
5495 *
5496 * If "allow_get_expansion" is TRUE, then ins_compl_get_exp() may be called to
5497 * get more completions. If it is FALSE, then do nothing when there are no more
5498 * completions in the given direction.
5499 *
5500 * If "advance" is TRUE, then completion will move to the first match.
5501 * Otherwise, the original text will be shown.
5502 *
5503 * Returns OK on success and -1 if the number of matches are unknown.
5504 */
5505 static int
5506find_next_completion_match(
5507 int allow_get_expansion,
5508 int todo, // repeat completion this many times
5509 int advance,
5510 int *num_matches)
5511{
5512 int found_end = FALSE;
5513 compl_T *found_compl = NULL;
zeertzjqaa925ee2024-06-09 18:24:05 +02005514 unsigned int cur_cot_flags = get_cot_flags();
zeertzjq529b9ad2024-06-05 20:27:06 +02005515 int compl_no_select = (cur_cot_flags & COT_NOSELECT) != 0;
5516 int compl_fuzzy_match = (cur_cot_flags & COT_FUZZY) != 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005517
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005518 while (--todo >= 0)
5519 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005520 if (compl_shows_dir_forward() && compl_shown_match->cp_next != NULL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005521 {
Girish Palyab8ee1cf2025-06-08 16:20:06 +02005522 if (compl_match_array != NULL)
5523 compl_shown_match = find_next_match_in_menu();
Girish Palya0ac1eb32025-04-16 20:18:33 +02005524 else
5525 compl_shown_match = compl_shown_match->cp_next;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005526 found_end = (compl_first_match != NULL
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005527 && (is_first_match(compl_shown_match->cp_next)
5528 || is_first_match(compl_shown_match)));
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005529 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005530 else if (compl_shows_dir_backward()
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005531 && compl_shown_match->cp_prev != NULL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005532 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005533 found_end = is_first_match(compl_shown_match);
Girish Palyab8ee1cf2025-06-08 16:20:06 +02005534 if (compl_match_array != NULL)
5535 compl_shown_match = find_next_match_in_menu();
Girish Palya0ac1eb32025-04-16 20:18:33 +02005536 else
5537 compl_shown_match = compl_shown_match->cp_prev;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005538 found_end |= is_first_match(compl_shown_match);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005539 }
5540 else
5541 {
5542 if (!allow_get_expansion)
5543 {
5544 if (advance)
5545 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005546 if (compl_shows_dir_backward())
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005547 compl_pending -= todo + 1;
5548 else
5549 compl_pending += todo + 1;
5550 }
5551 return -1;
5552 }
5553
5554 if (!compl_no_select && advance)
5555 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005556 if (compl_shows_dir_backward())
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005557 --compl_pending;
5558 else
5559 ++compl_pending;
5560 }
5561
5562 // Find matches.
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005563 *num_matches = ins_compl_get_exp(&compl_startpos);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005564
5565 // handle any pending completions
5566 while (compl_pending != 0 && compl_direction == compl_shows_dir
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005567 && advance)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005568 {
5569 if (compl_pending > 0 && compl_shown_match->cp_next != NULL)
5570 {
5571 compl_shown_match = compl_shown_match->cp_next;
5572 --compl_pending;
5573 }
5574 if (compl_pending < 0 && compl_shown_match->cp_prev != NULL)
5575 {
5576 compl_shown_match = compl_shown_match->cp_prev;
5577 ++compl_pending;
5578 }
5579 else
5580 break;
5581 }
5582 found_end = FALSE;
5583 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005584 if (!match_at_original_text(compl_shown_match)
John Marriott5e6ea922024-11-23 14:01:57 +01005585 && compl_leader.string != NULL
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005586 && !ins_compl_equal(compl_shown_match,
John Marriott5e6ea922024-11-23 14:01:57 +01005587 compl_leader.string, (int)compl_leader.length)
glepnira218cc62024-06-03 19:32:39 +02005588 && !(compl_fuzzy_match && compl_shown_match->cp_score > 0))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005589 ++todo;
5590 else
5591 // Remember a matching item.
5592 found_compl = compl_shown_match;
5593
5594 // Stop at the end of the list when we found a usable match.
5595 if (found_end)
5596 {
5597 if (found_compl != NULL)
5598 {
5599 compl_shown_match = found_compl;
5600 break;
5601 }
5602 todo = 1; // use first usable match after wrapping around
5603 }
5604 }
5605
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005606 return OK;
5607}
5608
5609/*
5610 * Fill in the next completion in the current direction.
5611 * If "allow_get_expansion" is TRUE, then we may call ins_compl_get_exp() to
5612 * get more completions. If it is FALSE, then we just do nothing when there
5613 * are no more completions in a given direction. The latter case is used when
5614 * we are still in the middle of finding completions, to allow browsing
5615 * through the ones found so far.
5616 * Return the total number of matches, or -1 if still unknown -- webb.
5617 *
5618 * compl_curr_match is currently being used by ins_compl_get_exp(), so we use
5619 * compl_shown_match here.
5620 *
5621 * Note that this function may be called recursively once only. First with
5622 * "allow_get_expansion" TRUE, which calls ins_compl_get_exp(), which in turn
5623 * calls this function with "allow_get_expansion" FALSE.
5624 */
5625 static int
5626ins_compl_next(
5627 int allow_get_expansion,
5628 int count, // repeat completion this many times; should
5629 // be at least 1
Girish Palya5fbe72e2025-06-18 19:15:45 +02005630 int insert_match) // Insert the newly selected match
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005631{
5632 int num_matches = -1;
5633 int todo = count;
5634 int advance;
5635 int started = compl_started;
Bram Moolenaar0ff01832022-09-24 19:20:30 +01005636 buf_T *orig_curbuf = curbuf;
zeertzjqaa925ee2024-06-09 18:24:05 +02005637 unsigned int cur_cot_flags = get_cot_flags();
zeertzjq529b9ad2024-06-05 20:27:06 +02005638 int compl_no_insert = (cur_cot_flags & COT_NOINSERT) != 0;
5639 int compl_fuzzy_match = (cur_cot_flags & COT_FUZZY) != 0;
glepniredd4ac32025-01-29 18:53:51 +01005640 int compl_preinsert = ins_compl_has_preinsert();
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005641
5642 // When user complete function return -1 for findstart which is next
5643 // time of 'always', compl_shown_match become NULL.
5644 if (compl_shown_match == NULL)
5645 return -1;
5646
John Marriott5e6ea922024-11-23 14:01:57 +01005647 if (compl_leader.string != NULL
glepnira218cc62024-06-03 19:32:39 +02005648 && !match_at_original_text(compl_shown_match)
5649 && !compl_fuzzy_match)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005650 // Update "compl_shown_match" to the actually shown match
5651 ins_compl_update_shown_match();
5652
5653 if (allow_get_expansion && insert_match
nwounkn2e3cd522023-10-17 11:05:38 +02005654 && (!compl_get_longest || compl_used_match))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005655 // Delete old text to be replaced
5656 ins_compl_delete();
5657
5658 // When finding the longest common text we stick at the original text,
5659 // don't let CTRL-N or CTRL-P move to the first match.
5660 advance = count != 1 || !allow_get_expansion || !compl_get_longest;
5661
5662 // When restarting the search don't insert the first match either.
5663 if (compl_restarting)
5664 {
5665 advance = FALSE;
5666 compl_restarting = FALSE;
5667 }
5668
5669 // Repeat this for when <PageUp> or <PageDown> is typed. But don't wrap
5670 // around.
5671 if (find_next_completion_match(allow_get_expansion, todo, advance,
5672 &num_matches) == -1)
5673 return -1;
5674
Bram Moolenaar0ff01832022-09-24 19:20:30 +01005675 if (curbuf != orig_curbuf)
5676 {
5677 // In case some completion function switched buffer, don't want to
5678 // insert the completion elsewhere.
5679 return -1;
5680 }
5681
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005682 // Insert the text of the new completion, or the compl_leader.
glepniredd4ac32025-01-29 18:53:51 +01005683 if (compl_no_insert && !started && !compl_preinsert)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005684 {
glepnir6a38aff2024-12-16 21:56:16 +01005685 ins_compl_insert_bytes(compl_orig_text.string + get_compl_len(), -1);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005686 compl_used_match = FALSE;
5687 }
5688 else if (insert_match)
5689 {
5690 if (!compl_get_longest || compl_used_match)
Girish Palya5fbe72e2025-06-18 19:15:45 +02005691 ins_compl_insert(TRUE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005692 else
glepnir6a38aff2024-12-16 21:56:16 +01005693 ins_compl_insert_bytes(compl_leader.string + get_compl_len(), -1);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005694 }
5695 else
5696 compl_used_match = FALSE;
5697
5698 if (!allow_get_expansion)
5699 {
5700 // may undisplay the popup menu first
5701 ins_compl_upd_pum();
5702
5703 if (pum_enough_matches())
5704 // Will display the popup menu, don't redraw yet to avoid flicker.
5705 pum_call_update_screen();
5706 else
5707 // Not showing the popup menu yet, redraw to show the user what was
5708 // inserted.
5709 update_screen(0);
5710
5711 // display the updated popup menu
5712 ins_compl_show_pum();
5713#ifdef FEAT_GUI
5714 if (gui.in_use)
5715 {
5716 // Show the cursor after the match, not after the redrawn text.
5717 setcursor();
5718 out_flush_cursor(FALSE, FALSE);
5719 }
5720#endif
5721
5722 // Delete old text to be replaced, since we're still searching and
5723 // don't want to match ourselves!
5724 ins_compl_delete();
5725 }
5726
5727 // Enter will select a match when the match wasn't inserted and the popup
5728 // menu is visible.
5729 if (compl_no_insert && !started)
5730 compl_enter_selects = TRUE;
5731 else
5732 compl_enter_selects = !insert_match && compl_match_array != NULL;
5733
5734 // Show the file name for the match (if any)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005735 if (compl_shown_match->cp_fname != NULL)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00005736 ins_compl_show_filename();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005737
5738 return num_matches;
5739}
5740
5741/*
5742 * Call this while finding completions, to check whether the user has hit a key
5743 * that should change the currently displayed completion, or exit completion
5744 * mode. Also, when compl_pending is not zero, show a completion as soon as
5745 * possible. -- webb
5746 * "frequency" specifies out of how many calls we actually check.
5747 * "in_compl_func" is TRUE when called from complete_check(), don't set
5748 * compl_curr_match.
5749 */
5750 void
5751ins_compl_check_keys(int frequency, int in_compl_func)
5752{
5753 static int count = 0;
5754 int c;
5755
5756 // Don't check when reading keys from a script, :normal or feedkeys().
5757 // That would break the test scripts. But do check for keys when called
5758 // from complete_check().
5759 if (!in_compl_func && (using_script() || ex_normal_busy))
5760 return;
5761
5762 // Only do this at regular intervals
5763 if (++count < frequency)
5764 return;
5765 count = 0;
5766
5767 // Check for a typed key. Do use mappings, otherwise vim_is_ctrl_x_key()
5768 // can't do its work correctly.
5769 c = vpeekc_any();
5770 if (c != NUL)
5771 {
5772 if (vim_is_ctrl_x_key(c) && c != Ctrl_X && c != Ctrl_R)
5773 {
5774 c = safe_vgetc(); // Eat the character
5775 compl_shows_dir = ins_compl_key2dir(c);
5776 (void)ins_compl_next(FALSE, ins_compl_key2count(c),
Girish Palya5fbe72e2025-06-18 19:15:45 +02005777 c != K_UP && c != K_DOWN);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005778 }
5779 else
5780 {
5781 // Need to get the character to have KeyTyped set. We'll put it
5782 // back with vungetc() below. But skip K_IGNORE.
5783 c = safe_vgetc();
5784 if (c != K_IGNORE)
5785 {
5786 // Don't interrupt completion when the character wasn't typed,
5787 // e.g., when doing @q to replay keys.
5788 if (c != Ctrl_R && KeyTyped)
5789 compl_interrupted = TRUE;
5790
5791 vungetc(c);
5792 }
5793 }
5794 }
zeertzjq529b9ad2024-06-05 20:27:06 +02005795 if (compl_pending != 0 && !got_int && !(cot_flags & COT_NOINSERT))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005796 {
5797 int todo = compl_pending > 0 ? compl_pending : -compl_pending;
5798
5799 compl_pending = 0;
Girish Palya5fbe72e2025-06-18 19:15:45 +02005800 (void)ins_compl_next(FALSE, todo, TRUE);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005801 }
5802}
5803
5804/*
5805 * Decide the direction of Insert mode complete from the key typed.
5806 * Returns BACKWARD or FORWARD.
5807 */
5808 static int
5809ins_compl_key2dir(int c)
5810{
5811 if (c == Ctrl_P || c == Ctrl_L
5812 || c == K_PAGEUP || c == K_KPAGEUP || c == K_S_UP || c == K_UP)
5813 return BACKWARD;
5814 return FORWARD;
5815}
5816
5817/*
5818 * Return TRUE for keys that are used for completion only when the popup menu
5819 * is visible.
5820 */
5821 static int
5822ins_compl_pum_key(int c)
5823{
5824 return pum_visible() && (c == K_PAGEUP || c == K_KPAGEUP || c == K_S_UP
5825 || c == K_PAGEDOWN || c == K_KPAGEDOWN || c == K_S_DOWN
5826 || c == K_UP || c == K_DOWN);
5827}
5828
5829/*
5830 * Decide the number of completions to move forward.
5831 * Returns 1 for most keys, height of the popup menu for page-up/down keys.
5832 */
5833 static int
5834ins_compl_key2count(int c)
5835{
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005836 if (ins_compl_pum_key(c) && c != K_UP && c != K_DOWN)
5837 {
glepnir19e1dd62025-05-08 22:50:38 +02005838 int h = pum_get_height();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01005839 if (h > 3)
5840 h -= 2; // keep some context
5841 return h;
5842 }
5843 return 1;
5844}
5845
5846/*
5847 * Return TRUE if completion with "c" should insert the match, FALSE if only
5848 * to change the currently selected completion.
5849 */
5850 static int
5851ins_compl_use_match(int c)
5852{
5853 switch (c)
5854 {
5855 case K_UP:
5856 case K_DOWN:
5857 case K_PAGEDOWN:
5858 case K_KPAGEDOWN:
5859 case K_S_DOWN:
5860 case K_PAGEUP:
5861 case K_KPAGEUP:
5862 case K_S_UP:
5863 return FALSE;
5864 }
5865 return TRUE;
5866}
5867
5868/*
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005869 * Get the pattern, column and length for normal completion (CTRL-N CTRL-P
5870 * completion)
John Marriott5e6ea922024-11-23 14:01:57 +01005871 * Sets the global variables: compl_col, compl_length and compl_pattern.
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005872 * Uses the global variables: compl_cont_status and ctrl_x_mode
5873 */
5874 static int
5875get_normal_compl_info(char_u *line, int startcol, colnr_T curs_col)
5876{
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005877 if ((compl_cont_status & CONT_SOL) || ctrl_x_mode_path_defines())
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005878 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005879 if (!compl_status_adding())
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005880 {
5881 while (--startcol >= 0 && vim_isIDc(line[startcol]))
5882 ;
5883 compl_col += ++startcol;
5884 compl_length = curs_col - startcol;
5885 }
5886 if (p_ic)
John Marriott8c85a2a2024-05-20 19:18:26 +02005887 {
John Marriott5e6ea922024-11-23 14:01:57 +01005888 compl_pattern.string = str_foldcase(line + compl_col,
5889 compl_length, NULL, 0);
5890 if (compl_pattern.string == NULL)
5891 {
5892 compl_pattern.length = 0;
5893 return FAIL;
5894 }
5895 compl_pattern.length = STRLEN(compl_pattern.string);
5896 }
5897 else
5898 {
5899 compl_pattern.string = vim_strnsave(line + compl_col, (size_t)compl_length);
5900 if (compl_pattern.string == NULL)
5901 {
5902 compl_pattern.length = 0;
5903 return FAIL;
5904 }
5905 compl_pattern.length = (size_t)compl_length;
John Marriott8c85a2a2024-05-20 19:18:26 +02005906 }
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005907 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00005908 else if (compl_status_adding())
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005909 {
5910 char_u *prefix = (char_u *)"\\<";
John Marriott8c85a2a2024-05-20 19:18:26 +02005911 size_t prefixlen = STRLEN_LITERAL("\\<");
John Marriott5e6ea922024-11-23 14:01:57 +01005912 size_t n;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005913
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005914 if (!vim_iswordp(line + compl_col)
5915 || (compl_col > 0
5916 && (vim_iswordp(mb_prevptr(line, line + compl_col)))))
John Marriott8c85a2a2024-05-20 19:18:26 +02005917 {
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005918 prefix = (char_u *)"";
John Marriott8c85a2a2024-05-20 19:18:26 +02005919 prefixlen = 0;
5920 }
John Marriott5e6ea922024-11-23 14:01:57 +01005921
5922 // we need up to 2 extra chars for the prefix
5923 n = quote_meta(NULL, line + compl_col, compl_length) + prefixlen;
5924 compl_pattern.string = alloc(n);
5925 if (compl_pattern.string == NULL)
5926 {
5927 compl_pattern.length = 0;
5928 return FAIL;
5929 }
5930 STRCPY((char *)compl_pattern.string, prefix);
5931 (void)quote_meta(compl_pattern.string + prefixlen,
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005932 line + compl_col, compl_length);
John Marriott5e6ea922024-11-23 14:01:57 +01005933 compl_pattern.length = n - 1;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005934 }
5935 else if (--startcol < 0
5936 || !vim_iswordp(mb_prevptr(line, line + startcol + 1)))
5937 {
John Marriott5e6ea922024-11-23 14:01:57 +01005938 size_t len = STRLEN_LITERAL("\\<\\k\\k");
5939
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005940 // Match any word of at least two chars
John Marriott5e6ea922024-11-23 14:01:57 +01005941 compl_pattern.string = vim_strnsave((char_u *)"\\<\\k\\k", len);
5942 if (compl_pattern.string == NULL)
John Marriott8c85a2a2024-05-20 19:18:26 +02005943 {
John Marriott5e6ea922024-11-23 14:01:57 +01005944 compl_pattern.length = 0;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005945 return FAIL;
John Marriott8c85a2a2024-05-20 19:18:26 +02005946 }
John Marriott5e6ea922024-11-23 14:01:57 +01005947 compl_pattern.length = len;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005948 compl_col += curs_col;
5949 compl_length = 0;
5950 }
5951 else
5952 {
5953 // Search the point of change class of multibyte character
5954 // or not a word single byte character backward.
5955 if (has_mbyte)
5956 {
5957 int base_class;
5958 int head_off;
5959
5960 startcol -= (*mb_head_off)(line, line + startcol);
5961 base_class = mb_get_class(line + startcol);
5962 while (--startcol >= 0)
5963 {
5964 head_off = (*mb_head_off)(line, line + startcol);
5965 if (base_class != mb_get_class(line + startcol
5966 - head_off))
5967 break;
5968 startcol -= head_off;
5969 }
5970 }
5971 else
glepnir19e1dd62025-05-08 22:50:38 +02005972 {
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005973 while (--startcol >= 0 && vim_iswordc(line[startcol]))
5974 ;
glepnir19e1dd62025-05-08 22:50:38 +02005975 }
5976
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005977 compl_col += ++startcol;
5978 compl_length = (int)curs_col - startcol;
5979 if (compl_length == 1)
5980 {
5981 // Only match word with at least two chars -- webb
5982 // there's no need to call quote_meta,
5983 // alloc(7) is enough -- Acevedo
John Marriott5e6ea922024-11-23 14:01:57 +01005984 compl_pattern.string = alloc(7);
5985 if (compl_pattern.string == NULL)
John Marriott8c85a2a2024-05-20 19:18:26 +02005986 {
John Marriott5e6ea922024-11-23 14:01:57 +01005987 compl_pattern.length = 0;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005988 return FAIL;
John Marriott8c85a2a2024-05-20 19:18:26 +02005989 }
John Marriott5e6ea922024-11-23 14:01:57 +01005990 STRCPY((char *)compl_pattern.string, "\\<");
5991 (void)quote_meta(compl_pattern.string + 2, line + compl_col, 1);
5992 STRCAT((char *)compl_pattern.string, "\\k");
5993 compl_pattern.length = STRLEN(compl_pattern.string);
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00005994 }
5995 else
5996 {
John Marriott5e6ea922024-11-23 14:01:57 +01005997 size_t n = quote_meta(NULL, line + compl_col, compl_length) + 2;
5998
5999 compl_pattern.string = alloc(n);
6000 if (compl_pattern.string == NULL)
John Marriott8c85a2a2024-05-20 19:18:26 +02006001 {
John Marriott5e6ea922024-11-23 14:01:57 +01006002 compl_pattern.length = 0;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006003 return FAIL;
John Marriott8c85a2a2024-05-20 19:18:26 +02006004 }
John Marriott5e6ea922024-11-23 14:01:57 +01006005 STRCPY((char *)compl_pattern.string, "\\<");
6006 (void)quote_meta(compl_pattern.string + 2, line + compl_col,
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006007 compl_length);
John Marriott5e6ea922024-11-23 14:01:57 +01006008 compl_pattern.length = n - 1;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006009 }
6010 }
6011
6012 return OK;
6013}
6014
6015/*
6016 * Get the pattern, column and length for whole line completion or for the
6017 * complete() function.
6018 * Sets the global variables: compl_col, compl_length and compl_pattern.
6019 */
6020 static int
6021get_wholeline_compl_info(char_u *line, colnr_T curs_col)
6022{
6023 compl_col = (colnr_T)getwhitecols(line);
6024 compl_length = (int)curs_col - (int)compl_col;
6025 if (compl_length < 0) // cursor in indent: empty pattern
6026 compl_length = 0;
6027 if (p_ic)
John Marriott8c85a2a2024-05-20 19:18:26 +02006028 {
John Marriott5e6ea922024-11-23 14:01:57 +01006029 compl_pattern.string = str_foldcase(line + compl_col, compl_length,
6030 NULL, 0);
6031 if (compl_pattern.string == NULL)
6032 {
6033 compl_pattern.length = 0;
6034 return FAIL;
6035 }
6036 compl_pattern.length = STRLEN(compl_pattern.string);
John Marriott8c85a2a2024-05-20 19:18:26 +02006037 }
John Marriott5e6ea922024-11-23 14:01:57 +01006038 else
6039 {
6040 compl_pattern.string = vim_strnsave(line + compl_col, (size_t)compl_length);
6041 if (compl_pattern.string == NULL)
6042 {
6043 compl_pattern.length = 0;
6044 return FAIL;
6045 }
6046 compl_pattern.length = (size_t)compl_length;
6047 }
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006048
6049 return OK;
6050}
6051
6052/*
6053 * Get the pattern, column and length for filename completion.
6054 * Sets the global variables: compl_col, compl_length and compl_pattern.
6055 */
6056 static int
6057get_filename_compl_info(char_u *line, int startcol, colnr_T curs_col)
6058{
6059 // Go back to just before the first filename character.
6060 if (startcol > 0)
6061 {
6062 char_u *p = line + startcol;
6063
6064 MB_PTR_BACK(line, p);
6065 while (p > line && vim_isfilec(PTR2CHAR(p)))
6066 MB_PTR_BACK(line, p);
6067 if (p == line && vim_isfilec(PTR2CHAR(p)))
6068 startcol = 0;
6069 else
6070 startcol = (int)(p - line) + 1;
6071 }
6072
6073 compl_col += startcol;
6074 compl_length = (int)curs_col - startcol;
John Marriott5e6ea922024-11-23 14:01:57 +01006075 compl_pattern.string = addstar(line + compl_col, compl_length, EXPAND_FILES);
6076 if (compl_pattern.string == NULL)
John Marriott8c85a2a2024-05-20 19:18:26 +02006077 {
John Marriott5e6ea922024-11-23 14:01:57 +01006078 compl_pattern.length = 0;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006079 return FAIL;
John Marriott8c85a2a2024-05-20 19:18:26 +02006080 }
6081
John Marriott5e6ea922024-11-23 14:01:57 +01006082 compl_pattern.length = STRLEN(compl_pattern.string);
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006083
6084 return OK;
6085}
6086
6087/*
6088 * Get the pattern, column and length for command-line completion.
6089 * Sets the global variables: compl_col, compl_length and compl_pattern.
6090 */
6091 static int
6092get_cmdline_compl_info(char_u *line, colnr_T curs_col)
6093{
John Marriott5e6ea922024-11-23 14:01:57 +01006094 compl_pattern.string = vim_strnsave(line, curs_col);
6095 if (compl_pattern.string == NULL)
John Marriott8c85a2a2024-05-20 19:18:26 +02006096 {
John Marriott5e6ea922024-11-23 14:01:57 +01006097 compl_pattern.length = 0;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006098 return FAIL;
John Marriott8c85a2a2024-05-20 19:18:26 +02006099 }
John Marriott5e6ea922024-11-23 14:01:57 +01006100 compl_pattern.length = curs_col;
6101 set_cmd_context(&compl_xp, compl_pattern.string,
6102 (int)compl_pattern.length, curs_col, FALSE);
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006103 if (compl_xp.xp_context == EXPAND_UNSUCCESSFUL
6104 || compl_xp.xp_context == EXPAND_NOTHING)
6105 // No completion possible, use an empty pattern to get a
6106 // "pattern not found" message.
6107 compl_col = curs_col;
6108 else
John Marriott5e6ea922024-11-23 14:01:57 +01006109 compl_col = (int)(compl_xp.xp_pattern - compl_pattern.string);
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006110 compl_length = curs_col - compl_col;
6111
6112 return OK;
6113}
6114
Girish Palya98c29db2025-06-01 19:40:00 +02006115#ifdef FEAT_COMPL_FUNC
6116/*
6117 * Set global variables related to completion:
6118 * compl_col, compl_length, compl_pattern, and cpt_compl_pattern.
6119 */
6120 static int
6121set_compl_globals(
6122 int startcol UNUSED,
6123 colnr_T curs_col UNUSED,
6124 int is_cpt_compl UNUSED)
6125{
6126 char_u *line = NULL;
6127 string_T *pattern = NULL;
6128 int len;
6129
6130 if (startcol < 0 || startcol > curs_col)
6131 startcol = curs_col;
6132 len = curs_col - startcol;
6133
6134 // Re-obtain line in case it has changed
6135 line = ml_get(curwin->w_cursor.lnum);
6136
6137 pattern = is_cpt_compl ? &cpt_compl_pattern : &compl_pattern;
6138 pattern->string = vim_strnsave(line + startcol, (size_t)len);
6139 if (pattern->string == NULL)
6140 {
6141 pattern->length = 0;
6142 return FAIL;
6143 }
6144 pattern->length = (size_t)len;
6145 if (!is_cpt_compl)
6146 {
6147 compl_col = startcol;
6148 compl_length = len;
6149 }
6150 return OK;
6151}
6152#endif
6153
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006154/*
6155 * Get the pattern, column and length for user defined completion ('omnifunc',
6156 * 'completefunc' and 'thesaurusfunc')
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006157 * Uses the global variable: spell_bad_len
Girish Palyacbe53192025-04-14 22:13:15 +02006158 * Callback function "cb" is set if triggered by a function in the 'cpt'
6159 * option; otherwise, it is NULL.
6160 * "startcol", when not NULL, contains the column returned by function.
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006161 */
6162 static int
glepnir19e1dd62025-05-08 22:50:38 +02006163get_userdefined_compl_info(
6164 colnr_T curs_col UNUSED,
6165 callback_T *cb UNUSED,
6166 int *startcol UNUSED)
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006167{
6168 int ret = FAIL;
6169
6170#ifdef FEAT_COMPL_FUNC
6171 // Call user defined function 'completefunc' with "a:findstart"
6172 // set to 1 to obtain the length of text to use for completion.
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006173 typval_T args[3];
6174 int col;
glepnir19e1dd62025-05-08 22:50:38 +02006175 char_u *funcname = NULL;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006176 pos_T pos;
6177 int save_State = State;
Girish Palyacbe53192025-04-14 22:13:15 +02006178 int is_cpt_function = (cb != NULL);
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006179
Girish Palyacbe53192025-04-14 22:13:15 +02006180 if (!is_cpt_function)
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006181 {
Girish Palyacbe53192025-04-14 22:13:15 +02006182 // Call 'completefunc' or 'omnifunc' or 'thesaurusfunc' and get pattern
6183 // length as a string
6184 funcname = get_complete_funcname(ctrl_x_mode);
6185 if (*funcname == NUL)
6186 {
6187 semsg(_(e_option_str_is_not_set), ctrl_x_mode_function()
6188 ? "completefunc" : "omnifunc");
6189 return FAIL;
6190 }
6191 cb = get_insert_callback(ctrl_x_mode);
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006192 }
6193
6194 args[0].v_type = VAR_NUMBER;
6195 args[0].vval.v_number = 1;
6196 args[1].v_type = VAR_STRING;
6197 args[1].vval.v_string = (char_u *)"";
6198 args[2].v_type = VAR_UNKNOWN;
6199 pos = curwin->w_cursor;
zeertzjqcfe45652022-05-27 17:26:55 +01006200 ++textlock;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006201 col = call_callback_retnr(cb, 2, args);
zeertzjqcfe45652022-05-27 17:26:55 +01006202 --textlock;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006203
6204 State = save_State;
6205 curwin->w_cursor = pos; // restore the cursor position
zeertzjq0a419e02024-04-02 19:01:14 +02006206 check_cursor(); // make sure cursor position is valid, just in case
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006207 validate_cursor();
6208 if (!EQUAL_POS(curwin->w_cursor, pos))
6209 {
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00006210 emsg(_(e_complete_function_deleted_text));
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006211 return FAIL;
6212 }
6213
Girish Palyacbe53192025-04-14 22:13:15 +02006214 if (startcol != NULL)
6215 *startcol = col;
6216
LemonBoy9bcb9ca2022-05-26 15:23:26 +01006217 // Return value -2 means the user complete function wants to cancel the
6218 // complete without an error, do the same if the function did not execute
6219 // successfully.
6220 if (col == -2 || aborting())
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006221 return FAIL;
glepnir19e1dd62025-05-08 22:50:38 +02006222
LemonBoy9bcb9ca2022-05-26 15:23:26 +01006223 // Return value -3 does the same as -2 and leaves CTRL-X mode.
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006224 if (col == -3)
6225 {
Girish Palyacbe53192025-04-14 22:13:15 +02006226 if (is_cpt_function)
6227 return FAIL;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006228 ctrl_x_mode = CTRL_X_NORMAL;
6229 edit_submode = NULL;
6230 if (!shortmess(SHM_COMPLETIONMENU))
6231 msg_clr_cmdline();
6232 return FAIL;
6233 }
6234
Yegappan Lakshmanan37079142022-01-08 10:38:48 +00006235 // Reset extended parameters of completion, when starting new
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006236 // completion.
6237 compl_opt_refresh_always = FALSE;
6238 compl_opt_suppress_empty = FALSE;
6239
Girish Palya98c29db2025-06-01 19:40:00 +02006240 ret = !is_cpt_function ? set_compl_globals(col, curs_col, FALSE) : OK;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006241#endif
6242
6243 return ret;
6244}
6245
6246/*
6247 * Get the pattern, column and length for spell completion.
6248 * Sets the global variables: compl_col, compl_length and compl_pattern.
6249 * Uses the global variable: spell_bad_len
6250 */
6251 static int
6252get_spell_compl_info(int startcol UNUSED, colnr_T curs_col UNUSED)
6253{
6254 int ret = FAIL;
6255#ifdef FEAT_SPELL
glepnir19e1dd62025-05-08 22:50:38 +02006256 char_u *line = NULL;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006257
6258 if (spell_bad_len > 0)
6259 compl_col = curs_col - spell_bad_len;
6260 else
6261 compl_col = spell_word_start(startcol);
glepnir19e1dd62025-05-08 22:50:38 +02006262
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006263 if (compl_col >= (colnr_T)startcol)
6264 {
6265 compl_length = 0;
6266 compl_col = curs_col;
6267 }
6268 else
6269 {
6270 spell_expand_check_cap(compl_col);
6271 compl_length = (int)curs_col - compl_col;
6272 }
6273 // Need to obtain "line" again, it may have become invalid.
6274 line = ml_get(curwin->w_cursor.lnum);
John Marriott5e6ea922024-11-23 14:01:57 +01006275 compl_pattern.string = vim_strnsave(line + compl_col, (size_t)compl_length);
6276 if (compl_pattern.string == NULL)
John Marriott8c85a2a2024-05-20 19:18:26 +02006277 {
John Marriott5e6ea922024-11-23 14:01:57 +01006278 compl_pattern.length = 0;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006279 return FAIL;
John Marriott8c85a2a2024-05-20 19:18:26 +02006280 }
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006281
John Marriott5e6ea922024-11-23 14:01:57 +01006282 compl_pattern.length = (size_t)compl_length;
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006283 ret = OK;
6284#endif
6285
6286 return ret;
6287}
6288
6289/*
6290 * Get the completion pattern, column and length.
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006291 * "startcol" - start column number of the completion pattern/text
6292 * "cur_col" - current cursor column
6293 * On return, "line_invalid" is set to TRUE, if the current line may have
6294 * become invalid and needs to be fetched again.
6295 * Returns OK on success.
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006296 */
6297 static int
6298compl_get_info(char_u *line, int startcol, colnr_T curs_col, int *line_invalid)
6299{
glepnird5fdfa52025-06-02 19:45:41 +02006300 if (ctrl_x_mode_normal() || ctrl_x_mode_register()
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006301 || (ctrl_x_mode & CTRL_X_WANT_IDENT
6302 && !thesaurus_func_complete(ctrl_x_mode)))
6303 {
6304 return get_normal_compl_info(line, startcol, curs_col);
6305 }
6306 else if (ctrl_x_mode_line_or_eval())
6307 {
6308 return get_wholeline_compl_info(line, curs_col);
6309 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006310 else if (ctrl_x_mode_files())
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006311 {
6312 return get_filename_compl_info(line, startcol, curs_col);
6313 }
6314 else if (ctrl_x_mode == CTRL_X_CMDLINE)
6315 {
6316 return get_cmdline_compl_info(line, curs_col);
6317 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006318 else if (ctrl_x_mode_function() || ctrl_x_mode_omni()
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006319 || thesaurus_func_complete(ctrl_x_mode))
6320 {
Girish Palyacbe53192025-04-14 22:13:15 +02006321 if (get_userdefined_compl_info(curs_col, NULL, NULL) != OK)
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006322 return FAIL;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006323 *line_invalid = TRUE; // "line" may have become invalid
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006324 }
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006325 else if (ctrl_x_mode_spell())
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006326 {
6327 if (get_spell_compl_info(startcol, curs_col) == FAIL)
6328 return FAIL;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006329 *line_invalid = TRUE; // "line" may have become invalid
Bram Moolenaarbf7ff612021-12-27 12:52:07 +00006330 }
6331 else
6332 {
6333 internal_error("ins_complete()");
6334 return FAIL;
6335 }
6336
6337 return OK;
6338}
6339
6340/*
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006341 * Continue an interrupted completion mode search in "line".
6342 *
6343 * If this same ctrl_x_mode has been interrupted use the text from
6344 * "compl_startpos" to the cursor as a pattern to add a new word instead of
6345 * expand the one before the cursor, in word-wise if "compl_startpos" is not in
6346 * the same line as the cursor then fix it (the line has been split because it
6347 * was longer than 'tw'). if SOL is set then skip the previous pattern, a word
6348 * at the beginning of the line has been inserted, we'll look for that.
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006349 */
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006350 static void
6351ins_compl_continue_search(char_u *line)
6352{
6353 // it is a continued search
6354 compl_cont_status &= ~CONT_INTRPT; // remove INTRPT
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006355 if (ctrl_x_mode_normal() || ctrl_x_mode_path_patterns()
6356 || ctrl_x_mode_path_defines())
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006357 {
6358 if (compl_startpos.lnum != curwin->w_cursor.lnum)
6359 {
6360 // line (probably) wrapped, set compl_startpos to the
6361 // first non_blank in the line, if it is not a wordchar
6362 // include it to get a better pattern, but then we don't
6363 // want the "\\<" prefix, check it below
6364 compl_col = (colnr_T)getwhitecols(line);
6365 compl_startpos.col = compl_col;
6366 compl_startpos.lnum = curwin->w_cursor.lnum;
6367 compl_cont_status &= ~CONT_SOL; // clear SOL if present
6368 }
6369 else
6370 {
6371 // S_IPOS was set when we inserted a word that was at the
6372 // beginning of the line, which means that we'll go to SOL
6373 // mode but first we need to redefine compl_startpos
6374 if (compl_cont_status & CONT_S_IPOS)
6375 {
6376 compl_cont_status |= CONT_SOL;
6377 compl_startpos.col = (colnr_T)(skipwhite(
6378 line + compl_length
6379 + compl_startpos.col) - line);
6380 }
6381 compl_col = compl_startpos.col;
6382 }
6383 compl_length = curwin->w_cursor.col - (int)compl_col;
6384 // IObuff is used to add a "word from the next line" would we
6385 // have enough space? just being paranoid
6386#define MIN_SPACE 75
6387 if (compl_length > (IOSIZE - MIN_SPACE))
6388 {
6389 compl_cont_status &= ~CONT_SOL;
6390 compl_length = (IOSIZE - MIN_SPACE);
6391 compl_col = curwin->w_cursor.col - compl_length;
6392 }
6393 compl_cont_status |= CONT_ADDING | CONT_N_ADDS;
6394 if (compl_length < 1)
6395 compl_cont_status &= CONT_LOCAL;
6396 }
glepnird5fdfa52025-06-02 19:45:41 +02006397 else if (ctrl_x_mode_line_or_eval() || ctrl_x_mode_register())
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006398 compl_cont_status = CONT_ADDING | CONT_N_ADDS;
6399 else
6400 compl_cont_status = 0;
6401}
6402
6403/*
6404 * start insert mode completion
6405 */
6406 static int
6407ins_compl_start(void)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006408{
glepnir19e1dd62025-05-08 22:50:38 +02006409 char_u *line = NULL;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006410 int startcol = 0; // column where searched text starts
6411 colnr_T curs_col; // cursor column
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006412 int line_invalid = FALSE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006413 int save_did_ai = did_ai;
Bram Moolenaard9eefe32019-04-06 14:22:21 +02006414 int flags = CP_ORIGINAL_TEXT;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006415
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006416 // First time we hit ^N or ^P (in a row, I mean)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006417 did_ai = FALSE;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006418 did_si = FALSE;
6419 can_si = FALSE;
6420 can_si_back = FALSE;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006421 if (stop_arrow() == FAIL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006422 return FAIL;
6423
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006424 line = ml_get(curwin->w_cursor.lnum);
6425 curs_col = curwin->w_cursor.col;
6426 compl_pending = 0;
glepnir76bdb822025-02-08 19:04:51 +01006427 compl_lnum = curwin->w_cursor.lnum;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006428
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006429 if ((compl_cont_status & CONT_INTRPT) == CONT_INTRPT
6430 && compl_cont_mode == ctrl_x_mode)
6431 // this same ctrl-x_mode was interrupted previously. Continue the
6432 // completion.
6433 ins_compl_continue_search(line);
6434 else
6435 compl_cont_status &= CONT_LOCAL;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006436
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006437 if (!compl_status_adding()) // normal expansion
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006438 {
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006439 compl_cont_mode = ctrl_x_mode;
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006440 if (ctrl_x_mode_not_default())
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006441 // Remove LOCAL if ctrl_x_mode != CTRL_X_NORMAL
6442 compl_cont_status = 0;
6443 compl_cont_status |= CONT_N_ADDS;
6444 compl_startpos = curwin->w_cursor;
6445 startcol = (int)curs_col;
6446 compl_col = 0;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006447 }
6448
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006449 // Work out completion pattern and original text -- webb
6450 if (compl_get_info(line, startcol, curs_col, &line_invalid) == FAIL)
6451 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006452 if (ctrl_x_mode_function() || ctrl_x_mode_omni()
6453 || thesaurus_func_complete(ctrl_x_mode))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006454 // restore did_ai, so that adding comment leader works
6455 did_ai = save_did_ai;
6456 return FAIL;
6457 }
6458 // If "line" was changed while getting completion info get it again.
6459 if (line_invalid)
6460 line = ml_get(curwin->w_cursor.lnum);
6461
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006462 if (compl_status_adding())
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006463 {
6464 edit_submode_pre = (char_u *)_(" Adding");
6465 if (ctrl_x_mode_line_or_eval())
6466 {
6467 // Insert a new line, keep indentation but ignore 'comments'.
6468 char_u *old = curbuf->b_p_com;
6469
6470 curbuf->b_p_com = (char_u *)"";
6471 compl_startpos.lnum = curwin->w_cursor.lnum;
6472 compl_startpos.col = compl_col;
6473 ins_eol('\r');
6474 curbuf->b_p_com = old;
6475 compl_length = 0;
6476 compl_col = curwin->w_cursor.col;
glepnir76bdb822025-02-08 19:04:51 +01006477 compl_lnum = curwin->w_cursor.lnum;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006478 }
glepnircfe502c2025-04-17 20:17:53 +02006479 else if (ctrl_x_mode_normal() && cfc_has_mode())
glepnir7cfe6932024-09-15 20:06:28 +02006480 {
6481 compl_startpos = curwin->w_cursor;
6482 compl_cont_status &= CONT_S_IPOS;
6483 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006484 }
6485 else
6486 {
6487 edit_submode_pre = NULL;
6488 compl_startpos.col = compl_col;
6489 }
6490
6491 if (compl_cont_status & CONT_LOCAL)
6492 edit_submode = (char_u *)_(ctrl_x_msgs[CTRL_X_LOCAL_MSG]);
6493 else
6494 edit_submode = (char_u *)_(CTRL_X_MSG(ctrl_x_mode));
6495
6496 // If any of the original typed text has been changed we need to fix
6497 // the redo buffer.
6498 ins_compl_fixRedoBufForLeader(NULL);
6499
6500 // Always add completion for the original text.
John Marriott5e6ea922024-11-23 14:01:57 +01006501 VIM_CLEAR_STRING(compl_orig_text);
6502 compl_orig_text.length = (size_t)compl_length;
6503 compl_orig_text.string = vim_strnsave(line + compl_col, (size_t)compl_length);
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006504 if (p_ic)
6505 flags |= CP_ICASE;
zeertzjq4422de62025-03-07 19:06:02 +01006506 if (compl_orig_text.string == NULL
6507 || ins_compl_add(compl_orig_text.string,
6508 (int)compl_orig_text.length,
6509 NULL, NULL, NULL, 0, flags, FALSE, NULL, 0) != OK)
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006510 {
John Marriott5e6ea922024-11-23 14:01:57 +01006511 VIM_CLEAR_STRING(compl_pattern);
6512 VIM_CLEAR_STRING(compl_orig_text);
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006513 return FAIL;
6514 }
6515
6516 // showmode might reset the internal line pointers, so it must
6517 // be called before line = ml_get(), or when this address is no
6518 // longer needed. -- Acevedo.
6519 edit_submode_extra = (char_u *)_("-- Searching...");
6520 edit_submode_highl = HLF_COUNT;
6521 showmode();
6522 edit_submode_extra = NULL;
6523 out_flush();
6524
6525 return OK;
6526}
6527
6528/*
6529 * display the completion status message
6530 */
6531 static void
6532ins_compl_show_statusmsg(void)
6533{
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006534 // we found no match if the list has only the "compl_orig_text"-entry
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006535 if (is_first_match(compl_first_match->cp_next))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006536 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006537 edit_submode_extra = compl_status_adding() && compl_length > 1
Bram Moolenaarb53a1902022-11-15 13:57:38 +00006538 ? (char_u *)_("Hit end of paragraph")
6539 : (char_u *)_("Pattern not found");
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006540 edit_submode_highl = HLF_E;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006541 }
6542
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006543 if (edit_submode_extra == NULL)
6544 {
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006545 if (match_at_original_text(compl_curr_match))
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006546 {
6547 edit_submode_extra = (char_u *)_("Back at original");
6548 edit_submode_highl = HLF_W;
6549 }
6550 else if (compl_cont_status & CONT_S_IPOS)
6551 {
6552 edit_submode_extra = (char_u *)_("Word from other line");
6553 edit_submode_highl = HLF_COUNT;
6554 }
6555 else if (compl_curr_match->cp_next == compl_curr_match->cp_prev)
6556 {
6557 edit_submode_extra = (char_u *)_("The only match");
6558 edit_submode_highl = HLF_COUNT;
Bram Moolenaarf9d51352020-10-26 19:22:42 +01006559 compl_curr_match->cp_number = 1;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006560 }
6561 else
6562 {
Bram Moolenaar977fd0b2020-10-27 09:12:45 +01006563#if defined(FEAT_COMPL_FUNC) || defined(FEAT_EVAL)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006564 // Update completion sequence number when needed.
6565 if (compl_curr_match->cp_number == -1)
Bram Moolenaarf9d51352020-10-26 19:22:42 +01006566 ins_compl_update_sequence_numbers();
Bram Moolenaar977fd0b2020-10-27 09:12:45 +01006567#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006568 // The match should always have a sequence number now, this is
6569 // just a safety check.
6570 if (compl_curr_match->cp_number != -1)
6571 {
6572 // Space for 10 text chars. + 2x10-digit no.s = 31.
6573 // Translations may need more than twice that.
6574 static char_u match_ref[81];
6575
6576 if (compl_matches > 0)
6577 vim_snprintf((char *)match_ref, sizeof(match_ref),
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006578 _("match %d of %d"),
6579 compl_curr_match->cp_number, compl_matches);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006580 else
6581 vim_snprintf((char *)match_ref, sizeof(match_ref),
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006582 _("match %d"),
6583 compl_curr_match->cp_number);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006584 edit_submode_extra = match_ref;
6585 edit_submode_highl = HLF_R;
6586 if (dollar_vcol >= 0)
6587 curs_columns(FALSE);
6588 }
6589 }
6590 }
6591
6592 // Show a message about what (completion) mode we're in.
6593 if (!compl_opt_suppress_empty)
6594 {
6595 showmode();
6596 if (!shortmess(SHM_COMPLETIONMENU))
6597 {
6598 if (edit_submode_extra != NULL)
6599 {
6600 if (!p_smd)
Bram Moolenaarcc233582020-12-12 13:32:07 +01006601 {
6602 msg_hist_off = TRUE;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006603 msg_attr((char *)edit_submode_extra,
6604 edit_submode_highl < HLF_COUNT
6605 ? HL_ATTR(edit_submode_highl) : 0);
Bram Moolenaarcc233582020-12-12 13:32:07 +01006606 msg_hist_off = FALSE;
6607 }
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006608 }
6609 else
6610 msg_clr_cmdline(); // necessary for "noshowmode"
6611 }
6612 }
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006613}
6614
6615/*
6616 * Do Insert mode completion.
6617 * Called when character "c" was typed, which has a meaning for completion.
6618 * Returns OK if completion was done, FAIL if something failed (out of mem).
6619 */
6620 int
6621ins_complete(int c, int enable_pum)
6622{
6623 int n;
6624 int save_w_wrow;
6625 int save_w_leftcol;
6626 int insert_match;
6627
6628 compl_direction = ins_compl_key2dir(c);
6629 insert_match = ins_compl_use_match(c);
6630
6631 if (!compl_started)
6632 {
6633 if (ins_compl_start() == FAIL)
6634 return FAIL;
6635 }
6636 else if (insert_match && stop_arrow() == FAIL)
6637 return FAIL;
6638
glepnircf7f0122025-04-15 19:02:00 +02006639 compl_curr_win = curwin;
6640 compl_curr_buf = curwin->w_buffer;
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006641 compl_shown_match = compl_curr_match;
6642 compl_shows_dir = compl_direction;
6643
6644 // Find next match (and following matches).
6645 save_w_wrow = curwin->w_wrow;
6646 save_w_leftcol = curwin->w_leftcol;
Girish Palya5fbe72e2025-06-18 19:15:45 +02006647 n = ins_compl_next(TRUE, ins_compl_key2count(c), insert_match);
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006648
6649 // may undisplay the popup menu
6650 ins_compl_upd_pum();
6651
6652 if (n > 1) // all matches have been found
6653 compl_matches = n;
6654 compl_curr_match = compl_shown_match;
6655 compl_direction = compl_shows_dir;
6656
6657 // Eat the ESC that vgetc() returns after a CTRL-C to avoid leaving Insert
6658 // mode.
6659 if (got_int && !global_busy)
6660 {
6661 (void)vgetc();
6662 got_int = FALSE;
6663 }
6664
6665 // we found no match if the list has only the "compl_orig_text"-entry
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006666 if (is_first_match(compl_first_match->cp_next))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006667 {
6668 // remove N_ADDS flag, so next ^X<> won't try to go to ADDING mode,
6669 // because we couldn't expand anything at first place, but if we used
6670 // ^P, ^N, ^X^I or ^X^D we might want to add-expand a single-char-word
6671 // (such as M in M'exico) if not tried already. -- Acevedo
6672 if (compl_length > 1
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006673 || compl_status_adding()
6674 || (ctrl_x_mode_not_default()
6675 && !ctrl_x_mode_path_patterns()
6676 && !ctrl_x_mode_path_defines()))
Yegappan Lakshmanan5d2e0072021-12-30 11:40:53 +00006677 compl_cont_status &= ~CONT_N_ADDS;
6678 }
6679
6680 if (compl_curr_match->cp_flags & CP_CONT_S_IPOS)
6681 compl_cont_status |= CONT_S_IPOS;
6682 else
6683 compl_cont_status &= ~CONT_S_IPOS;
6684
6685 ins_compl_show_statusmsg();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006686
6687 // Show the popup menu, unless we got interrupted.
6688 if (enable_pum && !compl_interrupted)
6689 show_pum(save_w_wrow, save_w_leftcol);
6690
6691 compl_was_interrupted = compl_interrupted;
6692 compl_interrupted = FALSE;
6693
6694 return OK;
6695}
6696
Yegappan Lakshmanane9825862022-01-03 11:03:48 +00006697/*
6698 * Remove (if needed) and show the popup menu
6699 */
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006700 static void
6701show_pum(int prev_w_wrow, int prev_w_leftcol)
6702{
6703 // RedrawingDisabled may be set when invoked through complete().
Bram Moolenaar79cdf022023-05-20 14:07:00 +01006704 int save_RedrawingDisabled = RedrawingDisabled;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006705 RedrawingDisabled = 0;
6706
6707 // If the cursor moved or the display scrolled we need to remove the pum
6708 // first.
6709 setcursor();
6710 if (prev_w_wrow != curwin->w_wrow || prev_w_leftcol != curwin->w_leftcol)
6711 ins_compl_del_pum();
6712
6713 ins_compl_show_pum();
6714 setcursor();
Bram Moolenaar79cdf022023-05-20 14:07:00 +01006715
6716 RedrawingDisabled = save_RedrawingDisabled;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006717}
6718
6719/*
6720 * Looks in the first "len" chars. of "src" for search-metachars.
6721 * If dest is not NULL the chars. are copied there quoting (with
6722 * a backslash) the metachars, and dest would be NUL terminated.
6723 * Returns the length (needed) of dest
6724 */
6725 static unsigned
6726quote_meta(char_u *dest, char_u *src, int len)
6727{
6728 unsigned m = (unsigned)len + 1; // one extra for the NUL
6729
6730 for ( ; --len >= 0; src++)
6731 {
6732 switch (*src)
6733 {
6734 case '.':
6735 case '*':
6736 case '[':
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006737 if (ctrl_x_mode_dictionary() || ctrl_x_mode_thesaurus())
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006738 break;
6739 // FALLTHROUGH
6740 case '~':
Bram Moolenaarf4e20992020-12-21 19:59:08 +01006741 if (!magic_isset()) // quote these only if magic is set
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006742 break;
6743 // FALLTHROUGH
6744 case '\\':
Yegappan Lakshmanand94fbfc2022-01-04 17:01:44 +00006745 if (ctrl_x_mode_dictionary() || ctrl_x_mode_thesaurus())
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006746 break;
6747 // FALLTHROUGH
6748 case '^': // currently it's not needed.
6749 case '$':
6750 m++;
6751 if (dest != NULL)
6752 *dest++ = '\\';
6753 break;
6754 }
6755 if (dest != NULL)
6756 *dest++ = *src;
6757 // Copy remaining bytes of a multibyte character.
6758 if (has_mbyte)
6759 {
glepnir19e1dd62025-05-08 22:50:38 +02006760 int mb_len = (*mb_ptr2len)(src) - 1;
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006761 if (mb_len > 0 && len >= mb_len)
glepnir19e1dd62025-05-08 22:50:38 +02006762 for (int i = 0; i < mb_len; ++i)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006763 {
6764 --len;
6765 ++src;
6766 if (dest != NULL)
6767 *dest++ = *src;
6768 }
6769 }
6770 }
6771 if (dest != NULL)
6772 *dest = NUL;
6773
6774 return m;
6775}
6776
Bram Moolenaare2c453d2019-08-21 14:37:09 +02006777#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006778 void
6779free_insexpand_stuff(void)
6780{
John Marriott5e6ea922024-11-23 14:01:57 +01006781 VIM_CLEAR_STRING(compl_orig_text);
Yegappan Lakshmanan8658c752021-12-03 11:09:29 +00006782# ifdef FEAT_EVAL
6783 free_callback(&cfu_cb);
6784 free_callback(&ofu_cb);
6785 free_callback(&tsrfu_cb);
6786# endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006787}
Bram Moolenaare2c453d2019-08-21 14:37:09 +02006788#endif
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006789
Bram Moolenaare2c453d2019-08-21 14:37:09 +02006790#ifdef FEAT_SPELL
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006791/*
6792 * Called when starting CTRL_X_SPELL mode: Move backwards to a previous badly
6793 * spelled word, if there is one.
6794 */
6795 static void
6796spell_back_to_badword(void)
6797{
6798 pos_T tpos = curwin->w_cursor;
6799
Christ van Willegen - van Noort8e4c4c72024-05-17 18:49:27 +02006800 spell_bad_len = spell_move_to(curwin, BACKWARD, SMT_ALL, TRUE, NULL);
Bram Moolenaar7591bb32019-03-30 13:53:47 +01006801 if (curwin->w_cursor.col != tpos.col)
6802 start_arrow(&tpos);
6803}
Bram Moolenaare2c453d2019-08-21 14:37:09 +02006804#endif
Girish Palyacbe53192025-04-14 22:13:15 +02006805
6806/*
6807 * Reset the info associated with completion sources.
6808 */
6809 static void
Girish Palya0ac1eb32025-04-16 20:18:33 +02006810cpt_sources_clear(void)
Girish Palyacbe53192025-04-14 22:13:15 +02006811{
Girish Palya0ac1eb32025-04-16 20:18:33 +02006812 VIM_CLEAR(cpt_sources_array);
6813 cpt_sources_index = -1;
6814 cpt_sources_count = 0;
Girish Palyacbe53192025-04-14 22:13:15 +02006815}
6816
6817/*
Girish Palya98c29db2025-06-01 19:40:00 +02006818 * Setup completion sources.
Girish Palyacbe53192025-04-14 22:13:15 +02006819 */
6820 static int
Girish Palya98c29db2025-06-01 19:40:00 +02006821setup_cpt_sources(void)
Girish Palyacbe53192025-04-14 22:13:15 +02006822{
Girish Palya0ac1eb32025-04-16 20:18:33 +02006823 char_u buf[LSIZE];
6824 int slen;
Girish Palya98c29db2025-06-01 19:40:00 +02006825 int count = 0, idx = 0;
Girish Palya0ac1eb32025-04-16 20:18:33 +02006826 char_u *p;
Girish Palyacbe53192025-04-14 22:13:15 +02006827
Girish Palya0ac1eb32025-04-16 20:18:33 +02006828 for (p = curbuf->b_p_cpt; *p;)
Girish Palyacbe53192025-04-14 22:13:15 +02006829 {
6830 while (*p == ',' || *p == ' ') // Skip delimiters
6831 p++;
6832 if (*p) // If not end of string, count this segment
6833 {
Girish Palya0ac1eb32025-04-16 20:18:33 +02006834 (void)copy_option_part(&p, buf, LSIZE, ","); // Advance p
Girish Palyacbe53192025-04-14 22:13:15 +02006835 count++;
Girish Palyacbe53192025-04-14 22:13:15 +02006836 }
6837 }
Girish Palya98c29db2025-06-01 19:40:00 +02006838 if (count == 0)
6839 return OK;
6840
Girish Palya0ac1eb32025-04-16 20:18:33 +02006841 cpt_sources_clear();
6842 cpt_sources_count = count;
Girish Palya98c29db2025-06-01 19:40:00 +02006843 cpt_sources_array = ALLOC_CLEAR_MULT(cpt_source_T, count);
6844 if (cpt_sources_array == NULL)
Girish Palyacbe53192025-04-14 22:13:15 +02006845 {
Girish Palya98c29db2025-06-01 19:40:00 +02006846 cpt_sources_count = 0;
6847 return FAIL;
6848 }
Girish Palya0ac1eb32025-04-16 20:18:33 +02006849
Girish Palya98c29db2025-06-01 19:40:00 +02006850 for (p = curbuf->b_p_cpt; *p;)
6851 {
6852 while (*p == ',' || *p == ' ') // Skip delimiters
6853 p++;
6854 if (*p) // If not end of string, count this segment
6855 {
6856 char_u *t;
6857
6858 vim_memset(buf, 0, LSIZE);
6859 slen = copy_option_part(&p, buf, LSIZE, ","); // Advance p
6860 if (slen > 0 && (t = vim_strchr(buf, '^')) != NULL)
6861 cpt_sources_array[idx].cs_max_matches = atoi((char *)t + 1);
6862 idx++;
Girish Palya0ac1eb32025-04-16 20:18:33 +02006863 }
Girish Palyacbe53192025-04-14 22:13:15 +02006864 }
6865 return OK;
6866}
6867
6868/*
6869 * Return TRUE if any of the completion sources have 'refresh' set to 'always'.
6870 */
6871 static int
6872is_cpt_func_refresh_always(void)
6873{
6874#ifdef FEAT_COMPL_FUNC
glepnir19e1dd62025-05-08 22:50:38 +02006875 for (int i = 0; i < cpt_sources_count; i++)
Girish Palya98c29db2025-06-01 19:40:00 +02006876 if (cpt_sources_array[i].cs_refresh_always)
Girish Palyacbe53192025-04-14 22:13:15 +02006877 return TRUE;
6878#endif
6879 return FALSE;
6880}
6881
6882/*
6883 * Make the completion list non-cyclic.
6884 */
Girish Palyacbe53192025-04-14 22:13:15 +02006885 static void
6886ins_compl_make_linear(void)
6887{
6888 compl_T *m;
6889
6890 if (compl_first_match == NULL || compl_first_match->cp_prev == NULL)
6891 return;
6892 m = compl_first_match->cp_prev;
6893 m->cp_next = NULL;
6894 compl_first_match->cp_prev = NULL;
6895}
Girish Palyacbe53192025-04-14 22:13:15 +02006896
6897/*
6898 * Remove the matches linked to the current completion source (as indicated by
Girish Palya0ac1eb32025-04-16 20:18:33 +02006899 * cpt_sources_index) from the completion list.
Girish Palyacbe53192025-04-14 22:13:15 +02006900 */
6901#ifdef FEAT_COMPL_FUNC
6902 static compl_T *
6903remove_old_matches(void)
6904{
6905 compl_T *sublist_start = NULL, *sublist_end = NULL, *insert_at = NULL;
6906 compl_T *current, *next;
Girish Palya0ac1eb32025-04-16 20:18:33 +02006907 int compl_shown_removed = FALSE;
6908 int forward = (compl_first_match->cp_cpt_source_idx < 0);
6909
6910 compl_direction = forward ? FORWARD : BACKWARD;
6911 compl_shows_dir = compl_direction;
Girish Palyacbe53192025-04-14 22:13:15 +02006912
6913 // Identify the sublist of old matches that needs removal
6914 for (current = compl_first_match; current != NULL; current = current->cp_next)
6915 {
Girish Palya0ac1eb32025-04-16 20:18:33 +02006916 if (current->cp_cpt_source_idx < cpt_sources_index &&
6917 (forward || (!forward && !insert_at)))
Girish Palyacbe53192025-04-14 22:13:15 +02006918 insert_at = current;
6919
Girish Palya0ac1eb32025-04-16 20:18:33 +02006920 if (current->cp_cpt_source_idx == cpt_sources_index)
Girish Palyacbe53192025-04-14 22:13:15 +02006921 {
6922 if (!sublist_start)
6923 sublist_start = current;
6924 sublist_end = current;
6925 if (!compl_shown_removed && compl_shown_match == current)
6926 compl_shown_removed = TRUE;
6927 }
6928
Girish Palya0ac1eb32025-04-16 20:18:33 +02006929 if ((forward && current->cp_cpt_source_idx > cpt_sources_index)
6930 || (!forward && insert_at))
Girish Palyacbe53192025-04-14 22:13:15 +02006931 break;
6932 }
6933
6934 // Re-assign compl_shown_match if necessary
6935 if (compl_shown_removed)
6936 {
6937 if (forward)
6938 compl_shown_match = compl_first_match;
6939 else
6940 { // Last node will have the prefix that is being completed
Girish Palya0ac1eb32025-04-16 20:18:33 +02006941 for (current = compl_first_match; current->cp_next != NULL;
6942 current = current->cp_next)
Girish Palyacbe53192025-04-14 22:13:15 +02006943 ;
6944 compl_shown_match = current;
6945 }
6946 }
6947
6948 if (!sublist_start) // No nodes to remove
6949 return insert_at;
6950
6951 // Update links to remove sublist
6952 if (sublist_start->cp_prev)
6953 sublist_start->cp_prev->cp_next = sublist_end->cp_next;
6954 else
6955 compl_first_match = sublist_end->cp_next;
6956
6957 if (sublist_end->cp_next)
6958 sublist_end->cp_next->cp_prev = sublist_start->cp_prev;
6959
6960 // Free all nodes in the sublist
6961 sublist_end->cp_next = NULL;
6962 for (current = sublist_start; current != NULL; current = next)
6963 {
6964 next = current->cp_next;
6965 ins_compl_item_free(current);
6966 }
6967
6968 return insert_at;
6969}
6970#endif
6971
6972/*
6973 * Retrieve completion matches using the callback function "cb" and store the
6974 * 'refresh:always' flag.
6975 */
6976#ifdef FEAT_COMPL_FUNC
6977 static void
Girish Palyafa16c7a2025-06-30 20:14:43 +02006978get_cpt_func_completion_matches(callback_T *cb UNUSED, int restore_leader)
Girish Palyacbe53192025-04-14 22:13:15 +02006979{
Girish Palya98c29db2025-06-01 19:40:00 +02006980 int startcol = cpt_sources_array[cpt_sources_index].cs_startcol;
Girish Palyafa16c7a2025-06-30 20:14:43 +02006981 int result;
Girish Palyacbe53192025-04-14 22:13:15 +02006982
6983 VIM_CLEAR_STRING(cpt_compl_pattern);
Girish Palya98c29db2025-06-01 19:40:00 +02006984
6985 if (startcol == -2 || startcol == -3)
6986 return;
6987
Girish Palyafa16c7a2025-06-30 20:14:43 +02006988 if (restore_leader) // Re-insert the text removed by ins_compl_delete()
6989 ins_compl_insert_bytes(compl_orig_text.string + get_compl_len(), -1);
6990 result = set_compl_globals(startcol, curwin->w_cursor.col, TRUE);
6991 if (restore_leader)
6992 ins_compl_delete(); // Undo insertion
6993
6994 if (result == OK)
Girish Palyacbe53192025-04-14 22:13:15 +02006995 {
6996 expand_by_function(0, cpt_compl_pattern.string, cb);
Girish Palya98c29db2025-06-01 19:40:00 +02006997 cpt_sources_array[cpt_sources_index].cs_refresh_always =
Girish Palya0ac1eb32025-04-16 20:18:33 +02006998 compl_opt_refresh_always;
Girish Palyacbe53192025-04-14 22:13:15 +02006999 compl_opt_refresh_always = FALSE;
7000 }
7001}
7002#endif
7003
7004/*
7005 * Retrieve completion matches from functions in the 'cpt' option where the
7006 * 'refresh:always' flag is set.
7007 */
7008 static void
7009cpt_compl_refresh(void)
7010{
7011#ifdef FEAT_COMPL_FUNC
7012 char_u *cpt;
7013 char_u *p;
Christian Brabandtd2079cf2025-04-15 18:10:26 +02007014 callback_T *cb = NULL;
Girish Palya98c29db2025-06-01 19:40:00 +02007015 int startcol, ret;
Girish Palyacbe53192025-04-14 22:13:15 +02007016
7017 // Make the completion list linear (non-cyclic)
7018 ins_compl_make_linear();
7019 // Make a copy of 'cpt' in case the buffer gets wiped out
7020 cpt = vim_strsave(curbuf->b_p_cpt);
Girish Palya0ac1eb32025-04-16 20:18:33 +02007021 strip_caret_numbers_in_place(cpt);
Girish Palyacbe53192025-04-14 22:13:15 +02007022
Girish Palya0ac1eb32025-04-16 20:18:33 +02007023 cpt_sources_index = 0;
Girish Palya7c621052025-05-26 19:41:59 +02007024 for (p = cpt; *p;)
Girish Palyacbe53192025-04-14 22:13:15 +02007025 {
7026 while (*p == ',' || *p == ' ') // Skip delimiters
7027 p++;
7028
Girish Palya98c29db2025-06-01 19:40:00 +02007029 if (cpt_sources_array[cpt_sources_index].cs_refresh_always)
Girish Palyacbe53192025-04-14 22:13:15 +02007030 {
Girish Palya98c29db2025-06-01 19:40:00 +02007031 cb = get_callback_if_cpt_func(p);
Girish Palyacbe53192025-04-14 22:13:15 +02007032 if (cb)
7033 {
7034 compl_curr_match = remove_old_matches();
Girish Palya98c29db2025-06-01 19:40:00 +02007035 ret = get_userdefined_compl_info(curwin->w_cursor.col, cb,
7036 &startcol);
7037 if (ret == FAIL)
7038 {
7039 if (startcol == -3)
7040 cpt_sources_array[cpt_sources_index].cs_refresh_always
7041 = FALSE;
7042 else
7043 startcol = -2;
7044 }
7045 cpt_sources_array[cpt_sources_index].cs_startcol = startcol;
7046 if (ret == OK)
Girish Palyafa16c7a2025-06-30 20:14:43 +02007047 get_cpt_func_completion_matches(cb, FALSE);
Girish Palyacbe53192025-04-14 22:13:15 +02007048 }
7049 }
7050
Girish Palya7c621052025-05-26 19:41:59 +02007051 (void)copy_option_part(&p, IObuff, IOSIZE, ","); // Advance p
7052 if (may_advance_cpt_index(p))
7053 (void)advance_cpt_sources_index_safe();
Girish Palyacbe53192025-04-14 22:13:15 +02007054 }
Girish Palya0ac1eb32025-04-16 20:18:33 +02007055 cpt_sources_index = -1;
Girish Palyacbe53192025-04-14 22:13:15 +02007056
7057 vim_free(cpt);
7058 // Make the list cyclic
7059 compl_matches = ins_compl_make_cyclic();
7060#endif
7061}