blob: 3c38b4ec373055c53d9ef57eb83e07dfd2f21074 [file] [log] [blame]
Bram Moolenaar4b471622019-01-31 13:48:09 +01001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * indent.c: Indentation related functions
12 */
13
14#include "vim.h"
15
Bram Moolenaare677df82019-09-02 22:31:11 +020016#if defined(FEAT_VARTABS) || defined(PROTO)
17
18/*
19 * Set the integer values corresponding to the string setting of 'vartabstop'.
20 * "array" will be set, caller must free it if needed.
Bram Moolenaarb7081e12021-09-04 18:47:28 +020021 * Return FAIL for an error.
Bram Moolenaare677df82019-09-02 22:31:11 +020022 */
23 int
24tabstop_set(char_u *var, int **array)
25{
Bram Moolenaarb7081e12021-09-04 18:47:28 +020026 int valcount = 1;
27 int t;
28 char_u *cp;
Bram Moolenaare677df82019-09-02 22:31:11 +020029
30 if (var[0] == NUL || (var[0] == '0' && var[1] == NUL))
31 {
32 *array = NULL;
Bram Moolenaarb7081e12021-09-04 18:47:28 +020033 return OK;
Bram Moolenaare677df82019-09-02 22:31:11 +020034 }
35
36 for (cp = var; *cp != NUL; ++cp)
37 {
38 if (cp == var || cp[-1] == ',')
39 {
40 char_u *end;
41
42 if (strtol((char *)cp, (char **)&end, 10) <= 0)
43 {
44 if (cp != end)
Bram Moolenaar460ae5d2022-01-01 14:19:49 +000045 emsg(_(e_argument_must_be_positive));
Bram Moolenaare677df82019-09-02 22:31:11 +020046 else
Bram Moolenaar436b5ad2021-12-31 22:49:24 +000047 semsg(_(e_invalid_argument_str), cp);
Bram Moolenaarb7081e12021-09-04 18:47:28 +020048 return FAIL;
Bram Moolenaare677df82019-09-02 22:31:11 +020049 }
50 }
51
52 if (VIM_ISDIGIT(*cp))
53 continue;
54 if (cp[0] == ',' && cp > var && cp[-1] != ',' && cp[1] != NUL)
55 {
56 ++valcount;
57 continue;
58 }
Bram Moolenaar436b5ad2021-12-31 22:49:24 +000059 semsg(_(e_invalid_argument_str), var);
Bram Moolenaarb7081e12021-09-04 18:47:28 +020060 return FAIL;
Bram Moolenaare677df82019-09-02 22:31:11 +020061 }
62
63 *array = ALLOC_MULT(int, valcount + 1);
64 if (*array == NULL)
Bram Moolenaarb7081e12021-09-04 18:47:28 +020065 return FAIL;
Bram Moolenaare677df82019-09-02 22:31:11 +020066 (*array)[0] = valcount;
67
68 t = 1;
69 for (cp = var; *cp != NUL;)
70 {
Bram Moolenaarb7081e12021-09-04 18:47:28 +020071 int n = atoi((char *)cp);
72
Bram Moolenaar2ddb89f2021-09-04 21:20:41 +020073 // Catch negative values, overflow and ridiculous big values.
Bram Moolenaarfc88df42022-02-05 11:13:05 +000074 if (n <= 0 || n > TABSTOP_MAX)
Bram Moolenaarb7081e12021-09-04 18:47:28 +020075 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +000076 semsg(_(e_invalid_argument_str), cp);
Yegappan Lakshmanan960dcbd2023-03-07 17:45:11 +000077 VIM_CLEAR(*array);
Bram Moolenaarb7081e12021-09-04 18:47:28 +020078 return FAIL;
79 }
80 (*array)[t++] = n;
81 while (*cp != NUL && *cp != ',')
Bram Moolenaare677df82019-09-02 22:31:11 +020082 ++cp;
83 if (*cp != NUL)
84 ++cp;
85 }
86
Bram Moolenaarb7081e12021-09-04 18:47:28 +020087 return OK;
Bram Moolenaare677df82019-09-02 22:31:11 +020088}
89
90/*
91 * Calculate the number of screen spaces a tab will occupy.
92 * If "vts" is set then the tab widths are taken from that array,
93 * otherwise the value of ts is used.
94 */
95 int
96tabstop_padding(colnr_T col, int ts_arg, int *vts)
97{
98 int ts = ts_arg == 0 ? 8 : ts_arg;
99 int tabcount;
100 colnr_T tabcol = 0;
101 int t;
102 int padding = 0;
103
104 if (vts == NULL || vts[0] == 0)
105 return ts - (col % ts);
106
107 tabcount = vts[0];
108
109 for (t = 1; t <= tabcount; ++t)
110 {
111 tabcol += vts[t];
112 if (tabcol > col)
113 {
114 padding = (int)(tabcol - col);
115 break;
116 }
117 }
118 if (t > tabcount)
119 padding = vts[tabcount] - (int)((col - tabcol) % vts[tabcount]);
120
121 return padding;
122}
123
124/*
125 * Find the size of the tab that covers a particular column.
126 */
127 int
128tabstop_at(colnr_T col, int ts, int *vts)
129{
130 int tabcount;
131 colnr_T tabcol = 0;
132 int t;
133 int tab_size = 0;
134
135 if (vts == 0 || vts[0] == 0)
136 return ts;
137
138 tabcount = vts[0];
139 for (t = 1; t <= tabcount; ++t)
140 {
141 tabcol += vts[t];
142 if (tabcol > col)
143 {
144 tab_size = vts[t];
145 break;
146 }
147 }
148 if (t > tabcount)
149 tab_size = vts[tabcount];
150
151 return tab_size;
152}
153
154/*
155 * Find the column on which a tab starts.
156 */
157 colnr_T
158tabstop_start(colnr_T col, int ts, int *vts)
159{
160 int tabcount;
161 colnr_T tabcol = 0;
162 int t;
Bram Moolenaar6ed545e2022-05-09 20:09:23 +0100163 int excess;
Bram Moolenaare677df82019-09-02 22:31:11 +0200164
165 if (vts == NULL || vts[0] == 0)
166 return (col / ts) * ts;
167
168 tabcount = vts[0];
169 for (t = 1; t <= tabcount; ++t)
170 {
171 tabcol += vts[t];
172 if (tabcol > col)
173 return tabcol - vts[t];
174 }
175
176 excess = tabcol % vts[tabcount];
177 return excess + ((col - excess) / vts[tabcount]) * vts[tabcount];
178}
179
180/*
181 * Find the number of tabs and spaces necessary to get from one column
182 * to another.
183 */
184 void
185tabstop_fromto(
186 colnr_T start_col,
187 colnr_T end_col,
188 int ts_arg,
189 int *vts,
190 int *ntabs,
191 int *nspcs)
192{
193 int spaces = end_col - start_col;
194 colnr_T tabcol = 0;
195 int padding = 0;
196 int tabcount;
197 int t;
198 int ts = ts_arg == 0 ? curbuf->b_p_ts : ts_arg;
199
200 if (vts == NULL || vts[0] == 0)
201 {
202 int tabs = 0;
203 int initspc = 0;
204
205 initspc = ts - (start_col % ts);
206 if (spaces >= initspc)
207 {
208 spaces -= initspc;
209 tabs++;
210 }
211 tabs += spaces / ts;
212 spaces -= (spaces / ts) * ts;
213
214 *ntabs = tabs;
215 *nspcs = spaces;
216 return;
217 }
218
219 // Find the padding needed to reach the next tabstop.
220 tabcount = vts[0];
221 for (t = 1; t <= tabcount; ++t)
222 {
223 tabcol += vts[t];
224 if (tabcol > start_col)
225 {
226 padding = (int)(tabcol - start_col);
227 break;
228 }
229 }
230 if (t > tabcount)
231 padding = vts[tabcount] - (int)((start_col - tabcol) % vts[tabcount]);
232
233 // If the space needed is less than the padding no tabs can be used.
234 if (spaces < padding)
235 {
236 *ntabs = 0;
237 *nspcs = spaces;
238 return;
239 }
240
241 *ntabs = 1;
242 spaces -= padding;
243
244 // At least one tab has been used. See if any more will fit.
245 while (spaces != 0 && ++t <= tabcount)
246 {
247 padding = vts[t];
248 if (spaces < padding)
249 {
250 *nspcs = spaces;
251 return;
252 }
253 ++*ntabs;
254 spaces -= padding;
255 }
256
257 *ntabs += spaces / vts[tabcount];
258 *nspcs = spaces % vts[tabcount];
259}
260
261/*
262 * See if two tabstop arrays contain the same values.
263 */
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200264 static int
Bram Moolenaare677df82019-09-02 22:31:11 +0200265tabstop_eq(int *ts1, int *ts2)
266{
267 int t;
268
269 if ((ts1 == 0 && ts2) || (ts1 && ts2 == 0))
270 return FALSE;
271 if (ts1 == ts2)
272 return TRUE;
273 if (ts1[0] != ts2[0])
274 return FALSE;
275
276 for (t = 1; t <= ts1[0]; ++t)
277 if (ts1[t] != ts2[t])
278 return FALSE;
279
280 return TRUE;
281}
282
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200283# if defined(FEAT_BEVAL) || defined(PROTO)
Bram Moolenaare677df82019-09-02 22:31:11 +0200284/*
285 * Copy a tabstop array, allocating space for the new array.
286 */
287 int *
288tabstop_copy(int *oldts)
289{
290 int *newts;
291 int t;
292
293 if (oldts == NULL)
294 return NULL;
295 newts = ALLOC_MULT(int, oldts[0] + 1);
296 if (newts != NULL)
297 for (t = 0; t <= oldts[0]; ++t)
298 newts[t] = oldts[t];
299 return newts;
300}
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200301# endif
Bram Moolenaare677df82019-09-02 22:31:11 +0200302
303/*
304 * Return a count of the number of tabstops.
305 */
306 int
307tabstop_count(int *ts)
308{
309 return ts != NULL ? ts[0] : 0;
310}
311
312/*
313 * Return the first tabstop, or 8 if there are no tabstops defined.
314 */
315 int
316tabstop_first(int *ts)
317{
318 return ts != NULL ? ts[1] : 8;
319}
320
321#endif
322
323/*
324 * Return the effective shiftwidth value for current buffer, using the
325 * 'tabstop' value when 'shiftwidth' is zero.
326 */
327 long
328get_sw_value(buf_T *buf)
329{
330 return get_sw_value_col(buf, 0);
331}
332
333/*
334 * Idem, using "pos".
335 */
336 static long
337get_sw_value_pos(buf_T *buf, pos_T *pos)
338{
339 pos_T save_cursor = curwin->w_cursor;
340 long sw_value;
341
342 curwin->w_cursor = *pos;
343 sw_value = get_sw_value_col(buf, get_nolist_virtcol());
344 curwin->w_cursor = save_cursor;
345 return sw_value;
346}
347
348/*
349 * Idem, using the first non-black in the current line.
350 */
351 long
352get_sw_value_indent(buf_T *buf)
353{
354 pos_T pos = curwin->w_cursor;
355
356 pos.col = getwhitecols_curline();
357 return get_sw_value_pos(buf, &pos);
358}
359
360/*
361 * Idem, using virtual column "col".
362 */
363 long
364get_sw_value_col(buf_T *buf, colnr_T col UNUSED)
365{
366 return buf->b_p_sw ? buf->b_p_sw :
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200367#ifdef FEAT_VARTABS
Bram Moolenaare677df82019-09-02 22:31:11 +0200368 tabstop_at(col, buf->b_p_ts, buf->b_p_vts_array);
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200369#else
Bram Moolenaare677df82019-09-02 22:31:11 +0200370 buf->b_p_ts;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200371#endif
Bram Moolenaare677df82019-09-02 22:31:11 +0200372}
373
374/*
375 * Return the effective softtabstop value for the current buffer, using the
376 * 'shiftwidth' value when 'softtabstop' is negative.
377 */
378 long
379get_sts_value(void)
380{
381 return curbuf->b_p_sts < 0 ? get_sw_value(curbuf) : curbuf->b_p_sts;
382}
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200383
384/*
385 * Count the size (in window cells) of the indent in the current line.
386 */
387 int
388get_indent(void)
389{
390#ifdef FEAT_VARTABS
391 return get_indent_str_vtab(ml_get_curline(), (int)curbuf->b_p_ts,
392 curbuf->b_p_vts_array, FALSE);
393#else
394 return get_indent_str(ml_get_curline(), (int)curbuf->b_p_ts, FALSE);
395#endif
396}
397
398/*
399 * Count the size (in window cells) of the indent in line "lnum".
400 */
401 int
402get_indent_lnum(linenr_T lnum)
403{
404#ifdef FEAT_VARTABS
405 return get_indent_str_vtab(ml_get(lnum), (int)curbuf->b_p_ts,
406 curbuf->b_p_vts_array, FALSE);
407#else
408 return get_indent_str(ml_get(lnum), (int)curbuf->b_p_ts, FALSE);
409#endif
410}
411
412#if defined(FEAT_FOLDING) || defined(PROTO)
413/*
414 * Count the size (in window cells) of the indent in line "lnum" of buffer
415 * "buf".
416 */
417 int
418get_indent_buf(buf_T *buf, linenr_T lnum)
419{
420# ifdef FEAT_VARTABS
421 return get_indent_str_vtab(ml_get_buf(buf, lnum, FALSE),
zeertzjq07146ad2022-12-19 15:51:44 +0000422 (int)buf->b_p_ts, buf->b_p_vts_array, FALSE);
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200423# else
424 return get_indent_str(ml_get_buf(buf, lnum, FALSE), (int)buf->b_p_ts, FALSE);
425# endif
426}
427#endif
428
429/*
430 * count the size (in window cells) of the indent in line "ptr", with
431 * 'tabstop' at "ts"
432 */
433 int
434get_indent_str(
435 char_u *ptr,
436 int ts,
437 int list) // if TRUE, count only screen size for tabs
438{
439 int count = 0;
440
441 for ( ; *ptr; ++ptr)
442 {
443 if (*ptr == TAB)
444 {
Bram Moolenaareed9d462021-02-15 20:38:25 +0100445 if (!list || curwin->w_lcs_chars.tab1)
446 // count a tab for what it is worth
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200447 count += ts - (count % ts);
448 else
449 // In list mode, when tab is not set, count screen char width
450 // for Tab, displays: ^I
451 count += ptr2cells(ptr);
452 }
453 else if (*ptr == ' ')
454 ++count; // count a space for one
455 else
456 break;
457 }
458 return count;
459}
460
461#ifdef FEAT_VARTABS
462/*
463 * Count the size (in window cells) of the indent in line "ptr", using
464 * variable tabstops.
465 * if "list" is TRUE, count only screen size for tabs.
466 */
467 int
468get_indent_str_vtab(char_u *ptr, int ts, int *vts, int list)
469{
470 int count = 0;
471
472 for ( ; *ptr; ++ptr)
473 {
474 if (*ptr == TAB) // count a tab for what it is worth
475 {
Bram Moolenaareed9d462021-02-15 20:38:25 +0100476 if (!list || curwin->w_lcs_chars.tab1)
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200477 count += tabstop_padding(count, ts, vts);
478 else
479 // In list mode, when tab is not set, count screen char width
480 // for Tab, displays: ^I
481 count += ptr2cells(ptr);
482 }
483 else if (*ptr == ' ')
484 ++count; // count a space for one
485 else
486 break;
487 }
488 return count;
489}
490#endif
491
492/*
493 * Set the indent of the current line.
494 * Leaves the cursor on the first non-blank in the line.
495 * Caller must take care of undo.
496 * "flags":
497 * SIN_CHANGED: call changed_bytes() if the line was changed.
498 * SIN_INSERT: insert the indent in front of the line.
499 * SIN_UNDO: save line for undo before changing it.
500 * Returns TRUE if the line was changed.
501 */
502 int
503set_indent(
504 int size, // measured in spaces
505 int flags)
506{
507 char_u *p;
508 char_u *newline;
509 char_u *oldline;
510 char_u *s;
511 int todo;
512 int ind_len; // measured in characters
513 int line_len;
514 int doit = FALSE;
515 int ind_done = 0; // measured in spaces
516#ifdef FEAT_VARTABS
517 int ind_col = 0;
518#endif
519 int tab_pad;
520 int retval = FALSE;
521 int orig_char_len = -1; // number of initial whitespace chars when
522 // 'et' and 'pi' are both set
523
524 // First check if there is anything to do and compute the number of
525 // characters needed for the indent.
526 todo = size;
527 ind_len = 0;
528 p = oldline = ml_get_curline();
529
530 // Calculate the buffer size for the new indent, and check to see if it
531 // isn't already set
532
533 // if 'expandtab' isn't set: use TABs; if both 'expandtab' and
534 // 'preserveindent' are set count the number of characters at the
535 // beginning of the line to be copied
536 if (!curbuf->b_p_et || (!(flags & SIN_INSERT) && curbuf->b_p_pi))
537 {
538 // If 'preserveindent' is set then reuse as much as possible of
539 // the existing indent structure for the new indent
540 if (!(flags & SIN_INSERT) && curbuf->b_p_pi)
541 {
542 ind_done = 0;
543
544 // count as many characters as we can use
545 while (todo > 0 && VIM_ISWHITE(*p))
546 {
547 if (*p == TAB)
548 {
549#ifdef FEAT_VARTABS
550 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
551 curbuf->b_p_vts_array);
552#else
553 tab_pad = (int)curbuf->b_p_ts
554 - (ind_done % (int)curbuf->b_p_ts);
555#endif
556 // stop if this tab will overshoot the target
557 if (todo < tab_pad)
558 break;
559 todo -= tab_pad;
560 ++ind_len;
561 ind_done += tab_pad;
562 }
563 else
564 {
565 --todo;
566 ++ind_len;
567 ++ind_done;
568 }
569 ++p;
570 }
571
572#ifdef FEAT_VARTABS
573 // These diverge from this point.
574 ind_col = ind_done;
575#endif
576 // Set initial number of whitespace chars to copy if we are
577 // preserving indent but expandtab is set
578 if (curbuf->b_p_et)
579 orig_char_len = ind_len;
580
581 // Fill to next tabstop with a tab, if possible
582#ifdef FEAT_VARTABS
583 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
584 curbuf->b_p_vts_array);
585#else
586 tab_pad = (int)curbuf->b_p_ts - (ind_done % (int)curbuf->b_p_ts);
587#endif
588 if (todo >= tab_pad && orig_char_len == -1)
589 {
590 doit = TRUE;
591 todo -= tab_pad;
592 ++ind_len;
593 // ind_done += tab_pad;
594#ifdef FEAT_VARTABS
595 ind_col += tab_pad;
596#endif
597 }
598 }
599
600 // count tabs required for indent
601#ifdef FEAT_VARTABS
602 for (;;)
603 {
604 tab_pad = tabstop_padding(ind_col, curbuf->b_p_ts,
605 curbuf->b_p_vts_array);
606 if (todo < tab_pad)
607 break;
608 if (*p != TAB)
609 doit = TRUE;
610 else
611 ++p;
612 todo -= tab_pad;
613 ++ind_len;
614 ind_col += tab_pad;
615 }
616#else
617 while (todo >= (int)curbuf->b_p_ts)
618 {
619 if (*p != TAB)
620 doit = TRUE;
621 else
622 ++p;
623 todo -= (int)curbuf->b_p_ts;
624 ++ind_len;
625 // ind_done += (int)curbuf->b_p_ts;
626 }
627#endif
628 }
629 // count spaces required for indent
630 while (todo > 0)
631 {
632 if (*p != ' ')
633 doit = TRUE;
634 else
635 ++p;
636 --todo;
637 ++ind_len;
638 // ++ind_done;
639 }
640
641 // Return if the indent is OK already.
642 if (!doit && !VIM_ISWHITE(*p) && !(flags & SIN_INSERT))
643 return FALSE;
644
645 // Allocate memory for the new line.
646 if (flags & SIN_INSERT)
647 p = oldline;
648 else
649 p = skipwhite(p);
650 line_len = (int)STRLEN(p) + 1;
651
652 // If 'preserveindent' and 'expandtab' are both set keep the original
653 // characters and allocate accordingly. We will fill the rest with spaces
654 // after the if (!curbuf->b_p_et) below.
655 if (orig_char_len != -1)
656 {
657 newline = alloc(orig_char_len + size - ind_done + line_len);
658 if (newline == NULL)
659 return FALSE;
660 todo = size - ind_done;
661 ind_len = orig_char_len + todo; // Set total length of indent in
662 // characters, which may have been
663 // undercounted until now
664 p = oldline;
665 s = newline;
666 while (orig_char_len > 0)
667 {
668 *s++ = *p++;
669 orig_char_len--;
670 }
671
672 // Skip over any additional white space (useful when newindent is less
673 // than old)
674 while (VIM_ISWHITE(*p))
675 ++p;
676
677 }
678 else
679 {
680 todo = size;
681 newline = alloc(ind_len + line_len);
682 if (newline == NULL)
683 return FALSE;
684 s = newline;
685 }
686
687 // Put the characters in the new line.
688 // if 'expandtab' isn't set: use TABs
689 if (!curbuf->b_p_et)
690 {
691 // If 'preserveindent' is set then reuse as much as possible of
692 // the existing indent structure for the new indent
693 if (!(flags & SIN_INSERT) && curbuf->b_p_pi)
694 {
695 p = oldline;
696 ind_done = 0;
697
698 while (todo > 0 && VIM_ISWHITE(*p))
699 {
700 if (*p == TAB)
701 {
702#ifdef FEAT_VARTABS
703 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
704 curbuf->b_p_vts_array);
705#else
706 tab_pad = (int)curbuf->b_p_ts
707 - (ind_done % (int)curbuf->b_p_ts);
708#endif
709 // stop if this tab will overshoot the target
710 if (todo < tab_pad)
711 break;
712 todo -= tab_pad;
713 ind_done += tab_pad;
714 }
715 else
716 {
717 --todo;
718 ++ind_done;
719 }
720 *s++ = *p++;
721 }
722
723 // Fill to next tabstop with a tab, if possible
724#ifdef FEAT_VARTABS
725 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
726 curbuf->b_p_vts_array);
727#else
728 tab_pad = (int)curbuf->b_p_ts - (ind_done % (int)curbuf->b_p_ts);
729#endif
730 if (todo >= tab_pad)
731 {
732 *s++ = TAB;
733 todo -= tab_pad;
734#ifdef FEAT_VARTABS
735 ind_done += tab_pad;
736#endif
737 }
738
739 p = skipwhite(p);
740 }
741
742#ifdef FEAT_VARTABS
743 for (;;)
744 {
745 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
746 curbuf->b_p_vts_array);
747 if (todo < tab_pad)
748 break;
749 *s++ = TAB;
750 todo -= tab_pad;
751 ind_done += tab_pad;
752 }
753#else
754 while (todo >= (int)curbuf->b_p_ts)
755 {
756 *s++ = TAB;
757 todo -= (int)curbuf->b_p_ts;
758 }
759#endif
760 }
761 while (todo > 0)
762 {
763 *s++ = ' ';
764 --todo;
765 }
766 mch_memmove(s, p, (size_t)line_len);
767
768 // Replace the line (unless undo fails).
769 if (!(flags & SIN_UNDO) || u_savesub(curwin->w_cursor.lnum) == OK)
770 {
Bram Moolenaarcf306432020-06-29 20:40:37 +0200771 colnr_T old_offset = (colnr_T)(p - oldline);
772 colnr_T new_offset = (colnr_T)(s - newline);
773
774 // this may free "newline"
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200775 ml_replace(curwin->w_cursor.lnum, newline, FALSE);
776 if (flags & SIN_CHANGED)
777 changed_bytes(curwin->w_cursor.lnum, 0);
778
779 // Correct saved cursor position if it is in this line.
780 if (saved_cursor.lnum == curwin->w_cursor.lnum)
781 {
Bram Moolenaarcf306432020-06-29 20:40:37 +0200782 if (saved_cursor.col >= old_offset)
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200783 // cursor was after the indent, adjust for the number of
784 // bytes added/removed
Bram Moolenaarcf306432020-06-29 20:40:37 +0200785 saved_cursor.col += ind_len - old_offset;
786 else if (saved_cursor.col >= new_offset)
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200787 // cursor was in the indent, and is now after it, put it back
788 // at the start of the indent (replacing spaces with TAB)
Bram Moolenaarcf306432020-06-29 20:40:37 +0200789 saved_cursor.col = new_offset;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200790 }
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +0100791#ifdef FEAT_PROP_POPUP
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200792 {
Bram Moolenaarcf306432020-06-29 20:40:37 +0200793 int added = ind_len - old_offset;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200794
795 // When increasing indent this behaves like spaces were inserted at
796 // the old indent, when decreasing indent it behaves like spaces
797 // were deleted at the new indent.
798 adjust_prop_columns(curwin->w_cursor.lnum,
Bram Moolenaar8a77d202022-08-15 16:29:37 +0100799 added > 0 ? old_offset : (colnr_T)ind_len,
800 added, APC_INDENT);
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200801 }
802#endif
803 retval = TRUE;
804 }
805 else
806 vim_free(newline);
807
808 curwin->w_cursor.col = ind_len;
809 return retval;
810}
811
812/*
813 * Return the indent of the current line after a number. Return -1 if no
814 * number was found. Used for 'n' in 'formatoptions': numbered list.
815 * Since a pattern is used it can actually handle more than numbers.
816 */
817 int
818get_number_indent(linenr_T lnum)
819{
820 colnr_T col;
821 pos_T pos;
822
823 regmatch_T regmatch;
824 int lead_len = 0; // length of comment leader
825
826 if (lnum > curbuf->b_ml.ml_line_count)
827 return -1;
828 pos.lnum = 0;
829
830 // In format_lines() (i.e. not insert mode), fo+=q is needed too...
Bram Moolenaar24959102022-05-07 20:01:16 +0100831 if ((State & MODE_INSERT) || has_format_option(FO_Q_COMS))
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200832 lead_len = get_leader_len(ml_get(lnum), NULL, FALSE, TRUE);
833
834 regmatch.regprog = vim_regcomp(curbuf->b_p_flp, RE_MAGIC);
835 if (regmatch.regprog != NULL)
836 {
837 regmatch.rm_ic = FALSE;
838
839 // vim_regexec() expects a pointer to a line. This lets us
840 // start matching for the flp beyond any comment leader...
841 if (vim_regexec(&regmatch, ml_get(lnum) + lead_len, (colnr_T)0))
842 {
843 pos.lnum = lnum;
844 pos.col = (colnr_T)(*regmatch.endp - ml_get(lnum));
845 pos.coladd = 0;
846 }
847 vim_regfree(regmatch.regprog);
848 }
849
850 if (pos.lnum == 0 || *ml_get_pos(&pos) == NUL)
851 return -1;
852 getvcol(curwin, &pos, &col, NULL, NULL);
853 return (int)col;
854}
855
856#if defined(FEAT_LINEBREAK) || defined(PROTO)
857/*
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100858 * This is called when 'breakindentopt' is changed and when a window is
859 * initialized.
860 */
861 int
862briopt_check(win_T *wp)
863{
864 char_u *p;
865 int bri_shift = 0;
866 long bri_min = 20;
867 int bri_sbr = FALSE;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +0200868 int bri_list = 0;
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100869 int bri_vcol = 0;
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100870
871 p = wp->w_p_briopt;
872 while (*p != NUL)
873 {
874 if (STRNCMP(p, "shift:", 6) == 0
875 && ((p[6] == '-' && VIM_ISDIGIT(p[7])) || VIM_ISDIGIT(p[6])))
876 {
877 p += 6;
878 bri_shift = getdigits(&p);
879 }
880 else if (STRNCMP(p, "min:", 4) == 0 && VIM_ISDIGIT(p[4]))
881 {
882 p += 4;
883 bri_min = getdigits(&p);
884 }
885 else if (STRNCMP(p, "sbr", 3) == 0)
886 {
887 p += 3;
888 bri_sbr = TRUE;
889 }
Christian Brabandt4a0b85a2021-07-14 20:00:27 +0200890 else if (STRNCMP(p, "list:", 5) == 0)
891 {
892 p += 5;
893 bri_list = getdigits(&p);
894 }
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100895 else if (STRNCMP(p, "column:", 7) == 0)
896 {
897 p += 7;
898 bri_vcol = getdigits(&p);
899 }
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100900 if (*p != ',' && *p != NUL)
901 return FAIL;
902 if (*p == ',')
903 ++p;
904 }
905
Bram Moolenaarb81f56f2020-02-23 15:29:46 +0100906 wp->w_briopt_shift = bri_shift;
907 wp->w_briopt_min = bri_min;
908 wp->w_briopt_sbr = bri_sbr;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +0200909 wp->w_briopt_list = bri_list;
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100910 wp->w_briopt_vcol = bri_vcol;
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100911
912 return OK;
913}
914
915/*
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200916 * Return appropriate space number for breakindent, taking influencing
917 * parameters into account. Window must be specified, since it is not
918 * necessarily always the current one.
919 */
920 int
921get_breakindent_win(
922 win_T *wp,
923 char_u *line) // start of the line
924{
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000925 static int prev_indent = 0; // cached indent value
926 static long prev_ts = 0L; // cached tabstop value
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100927 static int prev_fnum = 0; // cached buffer number
928 static char_u *prev_line = NULL; // cached copy of "line"
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200929 static varnumber_T prev_tick = 0; // changedtick of cached value
930# ifdef FEAT_VARTABS
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000931 static int *prev_vts = NULL; // cached vartabs values
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200932# endif
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000933 static int prev_list = 0; // cached list value
934 static int prev_listopt = 0; // cached w_p_briopt_list value
Christian Brabandtc53b4672022-01-15 10:01:05 +0000935 // cached formatlistpat value
936 static char_u *prev_flp = NULL;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200937 int bri = 0;
938 // window width minus window margin space, i.e. what rests for text
939 const int eff_wwidth = wp->w_width
940 - ((wp->w_p_nu || wp->w_p_rnu)
941 && (vim_strchr(p_cpo, CPO_NUMCOL) == NULL)
942 ? number_width(wp) + 1 : 0);
943
Christian Brabandtc53b4672022-01-15 10:01:05 +0000944 // used cached indent, unless
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100945 // - buffer changed
Christian Brabandtc53b4672022-01-15 10:01:05 +0000946 // - 'tabstop' changed
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100947 // - buffer was changed
Christian Brabandtc53b4672022-01-15 10:01:05 +0000948 // - 'briopt_list changed' changed or
949 // - 'formatlistpattern' changed
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100950 // - line changed
951 // - 'vartabs' changed
952 if (prev_fnum != wp->w_buffer->b_fnum
953 || prev_ts != wp->w_buffer->b_p_ts
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200954 || prev_tick != CHANGEDTICK(wp->w_buffer)
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000955 || prev_listopt != wp->w_briopt_list
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100956 || prev_flp == NULL
957 || STRCMP(prev_flp, get_flp_value(wp->w_buffer)) != 0
958 || prev_line == NULL || STRCMP(prev_line, line) != 0
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200959# ifdef FEAT_VARTABS
960 || prev_vts != wp->w_buffer->b_p_vts_array
961# endif
962 )
963 {
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100964 prev_fnum = wp->w_buffer->b_fnum;
965 vim_free(prev_line);
966 prev_line = vim_strsave(line);
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200967 prev_ts = wp->w_buffer->b_p_ts;
968 prev_tick = CHANGEDTICK(wp->w_buffer);
969# ifdef FEAT_VARTABS
970 prev_vts = wp->w_buffer->b_p_vts_array;
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100971 if (wp->w_briopt_vcol == 0)
972 prev_indent = get_indent_str_vtab(line,
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200973 (int)wp->w_buffer->b_p_ts,
974 wp->w_buffer->b_p_vts_array, wp->w_p_list);
975# else
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100976 if (wp->w_briopt_vcol == 0)
977 prev_indent = get_indent_str(line,
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200978 (int)wp->w_buffer->b_p_ts, wp->w_p_list);
979# endif
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000980 prev_listopt = wp->w_briopt_list;
Christian Brabandtc53b4672022-01-15 10:01:05 +0000981 prev_list = 0;
982 vim_free(prev_flp);
983 prev_flp = vim_strsave(get_flp_value(wp->w_buffer));
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000984 // add additional indent for numbered lists
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100985 if (wp->w_briopt_list != 0 && wp->w_briopt_vcol == 0)
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000986 {
987 regmatch_T regmatch;
988
Christian Brabandtc53b4672022-01-15 10:01:05 +0000989 regmatch.regprog = vim_regcomp(prev_flp,
990 RE_MAGIC + RE_STRING + RE_AUTO + RE_STRICT);
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000991
992 if (regmatch.regprog != NULL)
993 {
994 regmatch.rm_ic = FALSE;
995 if (vim_regexec(&regmatch, line, 0))
996 {
997 if (wp->w_briopt_list > 0)
998 prev_list = wp->w_briopt_list;
999 else
Maxim Kim11916722022-09-02 14:08:53 +01001000 prev_indent = (*regmatch.endp - *regmatch.startp);
Bram Moolenaarb2d85e32022-01-07 16:55:32 +00001001 }
1002 vim_regfree(regmatch.regprog);
1003 }
1004 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001005 }
Christian Brabandte7d6dbc2022-05-06 12:21:04 +01001006 if (wp->w_briopt_vcol != 0)
1007 {
1008 // column value has priority
1009 bri = wp->w_briopt_vcol;
1010 prev_list = 0;
1011 }
1012 else
1013 bri = prev_indent + wp->w_briopt_shift;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001014
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001015 // Add offset for number column, if 'n' is in 'cpoptions'
1016 bri += win_col_off2(wp);
1017
Christian Brabandt4a0b85a2021-07-14 20:00:27 +02001018 // add additional indent for numbered lists
Maxim Kim11916722022-09-02 14:08:53 +01001019 if (wp->w_briopt_list > 0)
1020 bri += prev_list;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +02001021
Maxim Kimf674b352021-07-22 11:46:59 +02001022 // indent minus the length of the showbreak string
1023 if (wp->w_briopt_sbr)
1024 bri -= vim_strsize(get_showbreak_value(wp));
1025
1026
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001027 // never indent past left window margin
1028 if (bri < 0)
1029 bri = 0;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +02001030
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001031 // always leave at least bri_min characters on the left,
1032 // if text width is sufficient
Bram Moolenaarb81f56f2020-02-23 15:29:46 +01001033 else if (bri > eff_wwidth - wp->w_briopt_min)
1034 bri = (eff_wwidth - wp->w_briopt_min < 0)
1035 ? 0 : eff_wwidth - wp->w_briopt_min;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001036
1037 return bri;
1038}
1039#endif
1040
1041/*
1042 * When extra == 0: Return TRUE if the cursor is before or on the first
1043 * non-blank in the line.
1044 * When extra == 1: Return TRUE if the cursor is before the first non-blank in
1045 * the line.
1046 */
1047 int
1048inindent(int extra)
1049{
1050 char_u *ptr;
1051 colnr_T col;
1052
1053 for (col = 0, ptr = ml_get_curline(); VIM_ISWHITE(*ptr); ++col)
1054 ++ptr;
1055 if (col >= curwin->w_cursor.col + extra)
1056 return TRUE;
1057 else
1058 return FALSE;
1059}
1060
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001061/*
1062 * op_reindent - handle reindenting a block of lines.
1063 */
1064 void
1065op_reindent(oparg_T *oap, int (*how)(void))
1066{
Bram Moolenaar4c84dd32022-04-20 10:22:54 +01001067 long i = 0;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001068 char_u *l;
1069 int amount;
1070 linenr_T first_changed = 0;
1071 linenr_T last_changed = 0;
1072 linenr_T start_lnum = curwin->w_cursor.lnum;
1073
1074 // Don't even try when 'modifiable' is off.
1075 if (!curbuf->b_p_ma)
1076 {
Bram Moolenaar108010a2021-06-27 22:03:33 +02001077 emsg(_(e_cannot_make_changes_modifiable_is_off));
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001078 return;
1079 }
1080
Bram Moolenaare4686982022-04-19 18:28:45 +01001081 // Save for undo. Do this once for all lines, much faster than doing this
1082 // for each line separately, especially when undoing.
1083 if (u_savecommon(start_lnum - 1, start_lnum + oap->line_count,
1084 start_lnum + oap->line_count, FALSE) == OK)
1085 for (i = oap->line_count; --i >= 0 && !got_int; )
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001086 {
Bram Moolenaare4686982022-04-19 18:28:45 +01001087 // it's a slow thing to do, so give feedback so there's no worry
1088 // that the computer's just hung.
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001089
Bram Moolenaare4686982022-04-19 18:28:45 +01001090 if (i > 1
1091 && (i % 50 == 0 || i == oap->line_count - 1)
1092 && oap->line_count > p_report)
1093 smsg(_("%ld lines to indent... "), i);
1094
1095 // Be vi-compatible: For lisp indenting the first line is not
1096 // indented, unless there is only one line.
Bram Moolenaare4686982022-04-19 18:28:45 +01001097 if (i != oap->line_count - 1 || oap->line_count == 1
1098 || how != get_lisp_indent)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001099 {
Bram Moolenaare4686982022-04-19 18:28:45 +01001100 l = skipwhite(ml_get_curline());
1101 if (*l == NUL) // empty or blank line
1102 amount = 0;
1103 else
1104 amount = how(); // get the indent for this line
1105
1106 if (amount >= 0 && set_indent(amount, 0))
1107 {
1108 // did change the indent, call changed_lines() later
1109 if (first_changed == 0)
1110 first_changed = curwin->w_cursor.lnum;
1111 last_changed = curwin->w_cursor.lnum;
1112 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001113 }
Bram Moolenaare4686982022-04-19 18:28:45 +01001114 ++curwin->w_cursor.lnum;
1115 curwin->w_cursor.col = 0; // make sure it's valid
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001116 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001117
1118 // put cursor on first non-blank of indented line
1119 curwin->w_cursor.lnum = start_lnum;
1120 beginline(BL_SOL | BL_FIX);
1121
1122 // Mark changed lines so that they will be redrawn. When Visual
1123 // highlighting was present, need to continue until the last line. When
1124 // there is no change still need to remove the Visual highlighting.
1125 if (last_changed != 0)
1126 changed_lines(first_changed, 0,
1127 oap->is_VIsual ? start_lnum + oap->line_count :
1128 last_changed + 1, 0L);
1129 else if (oap->is_VIsual)
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001130 redraw_curbuf_later(UPD_INVERTED);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001131
1132 if (oap->line_count > p_report)
1133 {
1134 i = oap->line_count - (i + 1);
1135 smsg(NGETTEXT("%ld line indented ",
1136 "%ld lines indented ", i), i);
1137 }
Bram Moolenaare1004402020-10-24 20:49:43 +02001138 if ((cmdmod.cmod_flags & CMOD_LOCKMARKS) == 0)
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +01001139 {
1140 // set '[ and '] marks
1141 curbuf->b_op_start = oap->start;
1142 curbuf->b_op_end = oap->end;
1143 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001144}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001145
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001146/*
1147 * Return TRUE if lines starting with '#' should be left aligned.
1148 */
1149 int
1150preprocs_left(void)
1151{
1152 return
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001153 (curbuf->b_p_si && !curbuf->b_p_cin) ||
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001154 (curbuf->b_p_cin && in_cinkeys('#', ' ', TRUE)
1155 && curbuf->b_ind_hash_comment == 0)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001156 ;
1157}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001158
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001159/*
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001160 * Return TRUE if the conditions are OK for smart indenting.
1161 */
1162 int
Yegappan Lakshmanana23a11b2023-02-21 14:27:41 +00001163may_do_si(void)
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001164{
1165 return curbuf->b_p_si
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001166 && !curbuf->b_p_cin
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001167# ifdef FEAT_EVAL
1168 && *curbuf->b_p_inde == NUL
1169# endif
1170 && !p_paste;
1171}
1172
1173/*
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001174 * Try to do some very smart auto-indenting.
1175 * Used when inserting a "normal" character.
1176 */
1177 void
1178ins_try_si(int c)
1179{
1180 pos_T *pos, old_pos;
1181 char_u *ptr;
1182 int i;
1183 int temp;
1184
1185 // do some very smart indenting when entering '{' or '}'
Bram Moolenaar2e444bb2022-05-14 12:54:23 +01001186 if (((did_si || can_si_back) && c == '{')
1187 || (can_si && c == '}' && inindent(0)))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001188 {
1189 // for '}' set indent equal to indent of line containing matching '{'
1190 if (c == '}' && (pos = findmatch(NULL, '{')) != NULL)
1191 {
1192 old_pos = curwin->w_cursor;
1193 // If the matching '{' has a ')' immediately before it (ignoring
1194 // white-space), then line up with the start of the line
1195 // containing the matching '(' if there is one. This handles the
1196 // case where an "if (..\n..) {" statement continues over multiple
1197 // lines -- webb
1198 ptr = ml_get(pos->lnum);
1199 i = pos->col;
1200 if (i > 0) // skip blanks before '{'
1201 while (--i > 0 && VIM_ISWHITE(ptr[i]))
1202 ;
1203 curwin->w_cursor.lnum = pos->lnum;
1204 curwin->w_cursor.col = i;
1205 if (ptr[i] == ')' && (pos = findmatch(NULL, '(')) != NULL)
1206 curwin->w_cursor = *pos;
1207 i = get_indent();
1208 curwin->w_cursor = old_pos;
1209 if (State & VREPLACE_FLAG)
1210 change_indent(INDENT_SET, i, FALSE, NUL, TRUE);
1211 else
1212 (void)set_indent(i, SIN_CHANGED);
1213 }
1214 else if (curwin->w_cursor.col > 0)
1215 {
1216 // when inserting '{' after "O" reduce indent, but not
1217 // more than indent of previous line
1218 temp = TRUE;
1219 if (c == '{' && can_si_back && curwin->w_cursor.lnum > 1)
1220 {
1221 old_pos = curwin->w_cursor;
1222 i = get_indent();
1223 while (curwin->w_cursor.lnum > 1)
1224 {
1225 ptr = skipwhite(ml_get(--(curwin->w_cursor.lnum)));
1226
1227 // ignore empty lines and lines starting with '#'.
1228 if (*ptr != '#' && *ptr != NUL)
1229 break;
1230 }
1231 if (get_indent() >= i)
1232 temp = FALSE;
1233 curwin->w_cursor = old_pos;
1234 }
1235 if (temp)
1236 shift_line(TRUE, FALSE, 1, TRUE);
1237 }
1238 }
1239
1240 // set indent of '#' always to 0
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001241 if (curwin->w_cursor.col > 0 && can_si && c == '#' && inindent(0))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001242 {
1243 // remember current indent for next line
1244 old_indent = get_indent();
1245 (void)set_indent(0, SIN_CHANGED);
1246 }
1247
1248 // Adjust ai_col, the char at this position can be deleted.
1249 if (ai_col > curwin->w_cursor.col)
1250 ai_col = curwin->w_cursor.col;
1251}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001252
1253/*
1254 * Insert an indent (for <Tab> or CTRL-T) or delete an indent (for CTRL-D).
1255 * Keep the cursor on the same character.
1256 * type == INDENT_INC increase indent (for CTRL-T or <Tab>)
1257 * type == INDENT_DEC decrease indent (for CTRL-D)
1258 * type == INDENT_SET set indent to "amount"
1259 * if round is TRUE, round the indent to 'shiftwidth' (only with _INC and _Dec).
1260 */
1261 void
1262change_indent(
1263 int type,
1264 int amount,
1265 int round,
1266 int replaced, // replaced character, put on replace stack
1267 int call_changed_bytes) // call changed_bytes()
1268{
1269 int vcol;
1270 int last_vcol;
1271 int insstart_less; // reduction for Insstart.col
1272 int new_cursor_col;
1273 int i;
1274 char_u *ptr;
1275 int save_p_list;
1276 int start_col;
1277 colnr_T vc;
1278 colnr_T orig_col = 0; // init for GCC
1279 char_u *new_line, *orig_line = NULL; // init for GCC
1280
Bram Moolenaar24959102022-05-07 20:01:16 +01001281 // MODE_VREPLACE state needs to know what the line was like before changing
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001282 if (State & VREPLACE_FLAG)
1283 {
1284 orig_line = vim_strsave(ml_get_curline()); // Deal with NULL below
1285 orig_col = curwin->w_cursor.col;
1286 }
1287
1288 // for the following tricks we don't want list mode
1289 save_p_list = curwin->w_p_list;
1290 curwin->w_p_list = FALSE;
Bram Moolenaar702bd6c2022-09-14 16:09:57 +01001291#ifdef FEAT_PROP_POPUP
1292 ignore_text_props = TRUE;
1293#endif
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001294 vc = getvcol_nolist(&curwin->w_cursor);
1295 vcol = vc;
1296
1297 // For Replace mode we need to fix the replace stack later, which is only
1298 // possible when the cursor is in the indent. Remember the number of
1299 // characters before the cursor if it's possible.
1300 start_col = curwin->w_cursor.col;
1301
1302 // determine offset from first non-blank
1303 new_cursor_col = curwin->w_cursor.col;
1304 beginline(BL_WHITE);
1305 new_cursor_col -= curwin->w_cursor.col;
1306
1307 insstart_less = curwin->w_cursor.col;
1308
1309 // If the cursor is in the indent, compute how many screen columns the
1310 // cursor is to the left of the first non-blank.
1311 if (new_cursor_col < 0)
1312 vcol = get_indent() - vcol;
1313
1314 if (new_cursor_col > 0) // can't fix replace stack
1315 start_col = -1;
1316
1317 // Set the new indent. The cursor will be put on the first non-blank.
1318 if (type == INDENT_SET)
1319 (void)set_indent(amount, call_changed_bytes ? SIN_CHANGED : 0);
1320 else
1321 {
1322 int save_State = State;
1323
1324 // Avoid being called recursively.
1325 if (State & VREPLACE_FLAG)
Bram Moolenaar24959102022-05-07 20:01:16 +01001326 State = MODE_INSERT;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001327 shift_line(type == INDENT_DEC, round, 1, call_changed_bytes);
1328 State = save_State;
1329 }
1330 insstart_less -= curwin->w_cursor.col;
1331
1332 // Try to put cursor on same character.
1333 // If the cursor is at or after the first non-blank in the line,
1334 // compute the cursor column relative to the column of the first
1335 // non-blank character.
1336 // If we are not in insert mode, leave the cursor on the first non-blank.
1337 // If the cursor is before the first non-blank, position it relative
1338 // to the first non-blank, counted in screen columns.
1339 if (new_cursor_col >= 0)
1340 {
1341 // When changing the indent while the cursor is touching it, reset
1342 // Insstart_col to 0.
1343 if (new_cursor_col == 0)
1344 insstart_less = MAXCOL;
1345 new_cursor_col += curwin->w_cursor.col;
1346 }
Bram Moolenaar24959102022-05-07 20:01:16 +01001347 else if (!(State & MODE_INSERT))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001348 new_cursor_col = curwin->w_cursor.col;
1349 else
1350 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001351 chartabsize_T cts;
1352
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001353 // Compute the screen column where the cursor should be.
1354 vcol = get_indent() - vcol;
1355 curwin->w_virtcol = (colnr_T)((vcol < 0) ? 0 : vcol);
1356
1357 // Advance the cursor until we reach the right screen column.
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001358 last_vcol = 0;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001359 ptr = ml_get_curline();
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001360 init_chartabsize_arg(&cts, curwin, 0, 0, ptr, ptr);
1361 while (cts.cts_vcol <= (int)curwin->w_virtcol)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001362 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001363 last_vcol = cts.cts_vcol;
1364 if (cts.cts_vcol > 0)
1365 MB_PTR_ADV(cts.cts_ptr);
1366 if (*cts.cts_ptr == NUL)
Bram Moolenaar4e889f92022-02-21 19:36:12 +00001367 break;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001368 cts.cts_vcol += lbr_chartabsize(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001369 }
1370 vcol = last_vcol;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001371 new_cursor_col = cts.cts_ptr - cts.cts_line;
1372 clear_chartabsize_arg(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001373
1374 // May need to insert spaces to be able to position the cursor on
1375 // the right screen column.
1376 if (vcol != (int)curwin->w_virtcol)
1377 {
1378 curwin->w_cursor.col = (colnr_T)new_cursor_col;
1379 i = (int)curwin->w_virtcol - vcol;
1380 ptr = alloc(i + 1);
1381 if (ptr != NULL)
1382 {
1383 new_cursor_col += i;
1384 ptr[i] = NUL;
1385 while (--i >= 0)
1386 ptr[i] = ' ';
1387 ins_str(ptr);
1388 vim_free(ptr);
1389 }
1390 }
1391
1392 // When changing the indent while the cursor is in it, reset
1393 // Insstart_col to 0.
1394 insstart_less = MAXCOL;
1395 }
1396
1397 curwin->w_p_list = save_p_list;
1398
1399 if (new_cursor_col <= 0)
1400 curwin->w_cursor.col = 0;
1401 else
1402 curwin->w_cursor.col = (colnr_T)new_cursor_col;
1403 curwin->w_set_curswant = TRUE;
1404 changed_cline_bef_curs();
1405
1406 // May have to adjust the start of the insert.
Bram Moolenaar24959102022-05-07 20:01:16 +01001407 if (State & MODE_INSERT)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001408 {
1409 if (curwin->w_cursor.lnum == Insstart.lnum && Insstart.col != 0)
1410 {
1411 if ((int)Insstart.col <= insstart_less)
1412 Insstart.col = 0;
1413 else
1414 Insstart.col -= insstart_less;
1415 }
1416 if ((int)ai_col <= insstart_less)
1417 ai_col = 0;
1418 else
1419 ai_col -= insstart_less;
1420 }
1421
Bram Moolenaar24959102022-05-07 20:01:16 +01001422 // For MODE_REPLACE state, may have to fix the replace stack, if it's
1423 // possible. If the number of characters before the cursor decreased, need
1424 // to pop a few characters from the replace stack.
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001425 // If the number of characters before the cursor increased, need to push a
1426 // few NULs onto the replace stack.
1427 if (REPLACE_NORMAL(State) && start_col >= 0)
1428 {
1429 while (start_col > (int)curwin->w_cursor.col)
1430 {
1431 replace_join(0); // remove a NUL from the replace stack
1432 --start_col;
1433 }
1434 while (start_col < (int)curwin->w_cursor.col || replaced)
1435 {
1436 replace_push(NUL);
1437 if (replaced)
1438 {
1439 replace_push(replaced);
1440 replaced = NUL;
1441 }
1442 ++start_col;
1443 }
1444 }
Bram Moolenaar702bd6c2022-09-14 16:09:57 +01001445#ifdef FEAT_PROP_POPUP
1446 ignore_text_props = FALSE;
1447#endif
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001448
Bram Moolenaar24959102022-05-07 20:01:16 +01001449 // For MODE_VREPLACE state, we also have to fix the replace stack. In this
1450 // case it is always possible because we backspace over the whole line and
1451 // then put it back again the way we wanted it.
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001452 if (State & VREPLACE_FLAG)
1453 {
1454 // If orig_line didn't allocate, just return. At least we did the job,
1455 // even if you can't backspace.
1456 if (orig_line == NULL)
1457 return;
1458
1459 // Save new line
1460 new_line = vim_strsave(ml_get_curline());
1461 if (new_line == NULL)
1462 return;
1463
1464 // We only put back the new line up to the cursor
1465 new_line[curwin->w_cursor.col] = NUL;
1466
1467 // Put back original line
1468 ml_replace(curwin->w_cursor.lnum, orig_line, FALSE);
1469 curwin->w_cursor.col = orig_col;
1470
1471 // Backspace from cursor to start of line
1472 backspace_until_column(0);
1473
1474 // Insert new stuff into line again
1475 ins_bytes(new_line);
1476
1477 vim_free(new_line);
1478 }
1479}
1480
1481/*
1482 * Copy the indent from ptr to the current line (and fill to size)
1483 * Leaves the cursor on the first non-blank in the line.
1484 * Returns TRUE if the line was changed.
1485 */
1486 int
1487copy_indent(int size, char_u *src)
1488{
1489 char_u *p = NULL;
1490 char_u *line = NULL;
1491 char_u *s;
1492 int todo;
1493 int ind_len;
1494 int line_len = 0;
1495 int tab_pad;
1496 int ind_done;
1497 int round;
1498#ifdef FEAT_VARTABS
1499 int ind_col;
1500#endif
1501
1502 // Round 1: compute the number of characters needed for the indent
1503 // Round 2: copy the characters.
1504 for (round = 1; round <= 2; ++round)
1505 {
1506 todo = size;
1507 ind_len = 0;
1508 ind_done = 0;
1509#ifdef FEAT_VARTABS
1510 ind_col = 0;
1511#endif
1512 s = src;
1513
1514 // Count/copy the usable portion of the source line
1515 while (todo > 0 && VIM_ISWHITE(*s))
1516 {
1517 if (*s == TAB)
1518 {
1519#ifdef FEAT_VARTABS
1520 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
1521 curbuf->b_p_vts_array);
1522#else
1523 tab_pad = (int)curbuf->b_p_ts
1524 - (ind_done % (int)curbuf->b_p_ts);
1525#endif
1526 // Stop if this tab will overshoot the target
1527 if (todo < tab_pad)
1528 break;
1529 todo -= tab_pad;
1530 ind_done += tab_pad;
1531#ifdef FEAT_VARTABS
1532 ind_col += tab_pad;
1533#endif
1534 }
1535 else
1536 {
1537 --todo;
1538 ++ind_done;
1539#ifdef FEAT_VARTABS
1540 ++ind_col;
1541#endif
1542 }
1543 ++ind_len;
1544 if (p != NULL)
1545 *p++ = *s;
1546 ++s;
1547 }
1548
1549 // Fill to next tabstop with a tab, if possible
1550#ifdef FEAT_VARTABS
1551 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
1552 curbuf->b_p_vts_array);
1553#else
1554 tab_pad = (int)curbuf->b_p_ts - (ind_done % (int)curbuf->b_p_ts);
1555#endif
1556 if (todo >= tab_pad && !curbuf->b_p_et)
1557 {
1558 todo -= tab_pad;
1559 ++ind_len;
1560#ifdef FEAT_VARTABS
1561 ind_col += tab_pad;
1562#endif
1563 if (p != NULL)
1564 *p++ = TAB;
1565 }
1566
1567 // Add tabs required for indent
1568 if (!curbuf->b_p_et)
1569 {
1570#ifdef FEAT_VARTABS
1571 for (;;)
1572 {
1573 tab_pad = tabstop_padding(ind_col, curbuf->b_p_ts,
1574 curbuf->b_p_vts_array);
1575 if (todo < tab_pad)
1576 break;
1577 todo -= tab_pad;
1578 ++ind_len;
1579 ind_col += tab_pad;
1580 if (p != NULL)
1581 *p++ = TAB;
1582 }
1583#else
1584 while (todo >= (int)curbuf->b_p_ts)
1585 {
1586 todo -= (int)curbuf->b_p_ts;
1587 ++ind_len;
1588 if (p != NULL)
1589 *p++ = TAB;
1590 }
1591#endif
1592 }
1593
1594 // Count/add spaces required for indent
1595 while (todo > 0)
1596 {
1597 --todo;
1598 ++ind_len;
1599 if (p != NULL)
1600 *p++ = ' ';
1601 }
1602
1603 if (p == NULL)
1604 {
1605 // Allocate memory for the result: the copied indent, new indent
1606 // and the rest of the line.
1607 line_len = (int)STRLEN(ml_get_curline()) + 1;
1608 line = alloc(ind_len + line_len);
1609 if (line == NULL)
1610 return FALSE;
1611 p = line;
1612 }
1613 }
1614
1615 // Append the original line
1616 mch_memmove(p, ml_get_curline(), (size_t)line_len);
1617
1618 // Replace the line
1619 ml_replace(curwin->w_cursor.lnum, line, FALSE);
1620
1621 // Put the cursor after the indent.
1622 curwin->w_cursor.col = ind_len;
1623 return TRUE;
1624}
1625
1626/*
Bram Moolenaar308660b2022-06-16 12:10:48 +01001627 * Give a "resulting text too long" error and maybe set got_int.
1628 */
1629 static void
1630emsg_text_too_long(void)
1631{
1632 emsg(_(e_resulting_text_too_long));
1633#ifdef FEAT_EVAL
1634 // when not inside a try/catch set got_int to break out of any loop
1635 if (trylevel == 0)
1636#endif
1637 got_int = TRUE;
1638}
1639
1640/*
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001641 * ":retab".
1642 */
1643 void
1644ex_retab(exarg_T *eap)
1645{
1646 linenr_T lnum;
1647 int got_tab = FALSE;
1648 long num_spaces = 0;
1649 long num_tabs;
1650 long len;
1651 long col;
1652 long vcol;
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001653 long start_col = 0; // For start of white-space string
1654 long start_vcol = 0; // For start of white-space string
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001655 long old_len;
Bram Moolenaar33f3c592022-02-12 20:46:15 +00001656 long new_len;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001657 char_u *ptr;
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001658 char_u *new_line = (char_u *)1; // init to non-NULL
1659 int did_undo; // called u_save for current line
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001660#ifdef FEAT_VARTABS
1661 int *new_vts_array = NULL;
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001662 char_u *new_ts_str; // string value of tab argument
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001663#else
1664 int temp;
1665 int new_ts;
1666#endif
1667 int save_list;
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001668 linenr_T first_line = 0; // first changed line
1669 linenr_T last_line = 0; // last changed line
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001670
1671 save_list = curwin->w_p_list;
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001672 curwin->w_p_list = 0; // don't want list mode here
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001673
1674#ifdef FEAT_VARTABS
1675 new_ts_str = eap->arg;
Bram Moolenaarb7081e12021-09-04 18:47:28 +02001676 if (tabstop_set(eap->arg, &new_vts_array) == FAIL)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001677 return;
1678 while (vim_isdigit(*(eap->arg)) || *(eap->arg) == ',')
1679 ++(eap->arg);
1680
1681 // This ensures that either new_vts_array and new_ts_str are freshly
1682 // allocated, or new_vts_array points to an existing array and new_ts_str
1683 // is null.
1684 if (new_vts_array == NULL)
1685 {
1686 new_vts_array = curbuf->b_p_vts_array;
1687 new_ts_str = NULL;
1688 }
1689 else
1690 new_ts_str = vim_strnsave(new_ts_str, eap->arg - new_ts_str);
1691#else
Bram Moolenaar2ddb89f2021-09-04 21:20:41 +02001692 ptr = eap->arg;
1693 new_ts = getdigits(&ptr);
1694 if (new_ts < 0 && *eap->arg == '-')
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001695 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001696 emsg(_(e_argument_must_be_positive));
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001697 return;
1698 }
Bram Moolenaar652dee42022-01-28 20:47:49 +00001699 if (new_ts < 0 || new_ts > TABSTOP_MAX)
Bram Moolenaar2ddb89f2021-09-04 21:20:41 +02001700 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001701 semsg(_(e_invalid_argument_str), eap->arg);
Bram Moolenaar2ddb89f2021-09-04 21:20:41 +02001702 return;
1703 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001704 if (new_ts == 0)
1705 new_ts = curbuf->b_p_ts;
1706#endif
1707 for (lnum = eap->line1; !got_int && lnum <= eap->line2; ++lnum)
1708 {
1709 ptr = ml_get(lnum);
1710 col = 0;
1711 vcol = 0;
1712 did_undo = FALSE;
1713 for (;;)
1714 {
1715 if (VIM_ISWHITE(ptr[col]))
1716 {
1717 if (!got_tab && num_spaces == 0)
1718 {
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001719 // First consecutive white-space
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001720 start_vcol = vcol;
1721 start_col = col;
1722 }
1723 if (ptr[col] == ' ')
1724 num_spaces++;
1725 else
1726 got_tab = TRUE;
1727 }
1728 else
1729 {
1730 if (got_tab || (eap->forceit && num_spaces > 1))
1731 {
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001732 // Retabulate this string of white-space
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001733
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001734 // len is virtual length of white string
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001735 len = num_spaces = vcol - start_vcol;
1736 num_tabs = 0;
1737 if (!curbuf->b_p_et)
1738 {
1739#ifdef FEAT_VARTABS
1740 int t, s;
1741
1742 tabstop_fromto(start_vcol, vcol,
1743 curbuf->b_p_ts, new_vts_array, &t, &s);
1744 num_tabs = t;
1745 num_spaces = s;
1746#else
1747 temp = new_ts - (start_vcol % new_ts);
1748 if (num_spaces >= temp)
1749 {
1750 num_spaces -= temp;
1751 num_tabs++;
1752 }
1753 num_tabs += num_spaces / new_ts;
1754 num_spaces -= (num_spaces / new_ts) * new_ts;
1755#endif
1756 }
1757 if (curbuf->b_p_et || got_tab ||
1758 (num_spaces + num_tabs < len))
1759 {
1760 if (did_undo == FALSE)
1761 {
1762 did_undo = TRUE;
1763 if (u_save((linenr_T)(lnum - 1),
1764 (linenr_T)(lnum + 1)) == FAIL)
1765 {
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001766 new_line = NULL; // flag out-of-memory
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001767 break;
1768 }
1769 }
1770
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001771 // len is actual number of white characters used
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001772 len = num_spaces + num_tabs;
1773 old_len = (long)STRLEN(ptr);
Bram Moolenaar33f3c592022-02-12 20:46:15 +00001774 new_len = old_len - col + start_col + len + 1;
Bram Moolenaar45491662022-02-12 21:59:51 +00001775 if (new_len <= 0 || new_len >= MAXCOL)
Bram Moolenaar33f3c592022-02-12 20:46:15 +00001776 {
Bram Moolenaar308660b2022-06-16 12:10:48 +01001777 emsg_text_too_long();
Bram Moolenaar33f3c592022-02-12 20:46:15 +00001778 break;
1779 }
1780 new_line = alloc(new_len);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001781 if (new_line == NULL)
1782 break;
1783 if (start_col > 0)
1784 mch_memmove(new_line, ptr, (size_t)start_col);
1785 mch_memmove(new_line + start_col + len,
1786 ptr + col, (size_t)(old_len - col + 1));
1787 ptr = new_line + start_col;
1788 for (col = 0; col < len; col++)
1789 ptr[col] = (col < num_tabs) ? '\t' : ' ';
Bram Moolenaar0dcd39b2021-02-03 19:44:25 +01001790 if (ml_replace(lnum, new_line, FALSE) == OK)
1791 // "new_line" may have been copied
1792 new_line = curbuf->b_ml.ml_line_ptr;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001793 if (first_line == 0)
1794 first_line = lnum;
1795 last_line = lnum;
1796 ptr = new_line;
1797 col = start_col + len;
1798 }
1799 }
1800 got_tab = FALSE;
1801 num_spaces = 0;
1802 }
1803 if (ptr[col] == NUL)
1804 break;
1805 vcol += chartabsize(ptr + col, (colnr_T)vcol);
Bram Moolenaar6e287032022-02-12 15:42:18 +00001806 if (vcol >= MAXCOL)
1807 {
Bram Moolenaar308660b2022-06-16 12:10:48 +01001808 emsg_text_too_long();
Bram Moolenaar6e287032022-02-12 15:42:18 +00001809 break;
1810 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001811 if (has_mbyte)
1812 col += (*mb_ptr2len)(ptr + col);
1813 else
1814 ++col;
1815 }
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001816 if (new_line == NULL) // out of memory
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001817 break;
1818 line_breakcheck();
1819 }
1820 if (got_int)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001821 emsg(_(e_interrupted));
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001822
1823#ifdef FEAT_VARTABS
1824 // If a single value was given then it can be considered equal to
1825 // either the value of 'tabstop' or the value of 'vartabstop'.
1826 if (tabstop_count(curbuf->b_p_vts_array) == 0
1827 && tabstop_count(new_vts_array) == 1
1828 && curbuf->b_p_ts == tabstop_first(new_vts_array))
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001829 ; // not changed
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001830 else if (tabstop_count(curbuf->b_p_vts_array) > 0
Bram Moolenaar6ed545e2022-05-09 20:09:23 +01001831 && tabstop_eq(curbuf->b_p_vts_array, new_vts_array))
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001832 ; // not changed
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001833 else
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001834 redraw_curbuf_later(UPD_NOT_VALID);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001835#else
1836 if (curbuf->b_p_ts != new_ts)
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001837 redraw_curbuf_later(UPD_NOT_VALID);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001838#endif
1839 if (first_line != 0)
1840 changed_lines(first_line, 0, last_line + 1, 0L);
1841
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001842 curwin->w_p_list = save_list; // restore 'list'
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001843
1844#ifdef FEAT_VARTABS
Bram Moolenaaraa2f0ee2019-12-21 18:47:26 +01001845 if (new_ts_str != NULL) // set the new tabstop
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001846 {
1847 // If 'vartabstop' is in use or if the value given to retab has more
1848 // than one tabstop then update 'vartabstop'.
1849 int *old_vts_ary = curbuf->b_p_vts_array;
1850
1851 if (tabstop_count(old_vts_ary) > 0 || tabstop_count(new_vts_array) > 1)
1852 {
1853 set_string_option_direct((char_u *)"vts", -1, new_ts_str,
1854 OPT_FREE|OPT_LOCAL, 0);
1855 curbuf->b_p_vts_array = new_vts_array;
1856 vim_free(old_vts_ary);
1857 }
1858 else
1859 {
1860 // 'vartabstop' wasn't in use and a single value was given to
1861 // retab then update 'tabstop'.
1862 curbuf->b_p_ts = tabstop_first(new_vts_array);
1863 vim_free(new_vts_array);
1864 }
1865 vim_free(new_ts_str);
1866 }
1867#else
1868 curbuf->b_p_ts = new_ts;
1869#endif
1870 coladvance(curwin->w_curswant);
1871
1872 u_clearline();
1873}
1874
Bram Moolenaar8e145b82022-05-21 20:17:31 +01001875#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001876/*
1877 * Get indent level from 'indentexpr'.
1878 */
1879 int
1880get_expr_indent(void)
1881{
1882 int indent = -1;
1883 char_u *inde_copy;
1884 pos_T save_pos;
1885 colnr_T save_curswant;
1886 int save_set_curswant;
1887 int save_State;
1888 int use_sandbox = was_set_insecurely((char_u *)"indentexpr",
1889 OPT_LOCAL);
Bram Moolenaar28e60cc2022-01-22 20:32:00 +00001890 sctx_T save_sctx = current_sctx;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001891
1892 // Save and restore cursor position and curswant, in case it was changed
1893 // via :normal commands
1894 save_pos = curwin->w_cursor;
1895 save_curswant = curwin->w_curswant;
1896 save_set_curswant = curwin->w_set_curswant;
1897 set_vim_var_nr(VV_LNUM, curwin->w_cursor.lnum);
1898 if (use_sandbox)
1899 ++sandbox;
zeertzjqcfe45652022-05-27 17:26:55 +01001900 ++textlock;
Bram Moolenaar28e60cc2022-01-22 20:32:00 +00001901 current_sctx = curbuf->b_p_script_ctx[BV_INDE];
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001902
1903 // Need to make a copy, the 'indentexpr' option could be changed while
1904 // evaluating it.
1905 inde_copy = vim_strsave(curbuf->b_p_inde);
1906 if (inde_copy != NULL)
1907 {
Bram Moolenaar3292a222022-10-01 20:17:17 +01001908 indent = (int)eval_to_number(inde_copy, TRUE);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001909 vim_free(inde_copy);
1910 }
1911
1912 if (use_sandbox)
1913 --sandbox;
zeertzjqcfe45652022-05-27 17:26:55 +01001914 --textlock;
Bram Moolenaar28e60cc2022-01-22 20:32:00 +00001915 current_sctx = save_sctx;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001916
1917 // Restore the cursor position so that 'indentexpr' doesn't need to.
1918 // Pretend to be in Insert mode, allow cursor past end of line for "o"
1919 // command.
1920 save_State = State;
Bram Moolenaar24959102022-05-07 20:01:16 +01001921 State = MODE_INSERT;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001922 curwin->w_cursor = save_pos;
1923 curwin->w_curswant = save_curswant;
1924 curwin->w_set_curswant = save_set_curswant;
1925 check_cursor();
1926 State = save_State;
1927
Bram Moolenaar620c9592021-07-31 21:32:31 +02001928 // Reset did_throw, unless 'debug' has "throw" and inside a try/catch.
1929 if (did_throw && (vim_strchr(p_debug, 't') == NULL || trylevel == 0))
1930 {
1931 handle_did_throw();
1932 did_throw = FALSE;
1933 }
1934
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001935 // If there is an error, just keep the current indent.
1936 if (indent < 0)
1937 indent = get_indent();
1938
1939 return indent;
1940}
1941#endif
1942
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001943 static int
1944lisp_match(char_u *p)
1945{
1946 char_u buf[LSIZE];
1947 int len;
1948 char_u *word = *curbuf->b_p_lw != NUL ? curbuf->b_p_lw : p_lispwords;
1949
1950 while (*word != NUL)
1951 {
1952 (void)copy_option_part(&word, buf, LSIZE, ",");
1953 len = (int)STRLEN(buf);
Bram Moolenaard26c5802022-10-13 12:30:08 +01001954 if (STRNCMP(buf, p, len) == 0 && IS_WHITE_OR_NUL(p[len]))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001955 return TRUE;
1956 }
1957 return FALSE;
1958}
1959
1960/*
1961 * When 'p' is present in 'cpoptions, a Vi compatible method is used.
1962 * The incompatible newer method is quite a bit better at indenting
1963 * code in lisp-like languages than the traditional one; it's still
1964 * mostly heuristics however -- Dirk van Deun, dirk@rave.org
1965 *
1966 * TODO:
1967 * Findmatch() should be adapted for lisp, also to make showmatch
1968 * work correctly: now (v5.3) it seems all C/C++ oriented:
1969 * - it does not recognize the #\( and #\) notations as character literals
1970 * - it doesn't know about comments starting with a semicolon
1971 * - it incorrectly interprets '(' as a character literal
1972 * All this messes up get_lisp_indent in some rare cases.
1973 * Update from Sergey Khorev:
1974 * I tried to fix the first two issues.
1975 */
1976 int
1977get_lisp_indent(void)
1978{
1979 pos_T *pos, realpos, paren;
1980 int amount;
1981 char_u *that;
1982 colnr_T col;
1983 colnr_T firsttry;
1984 int parencount, quotecount;
1985 int vi_lisp;
1986
1987 // Set vi_lisp to use the vi-compatible method
1988 vi_lisp = (vim_strchr(p_cpo, CPO_LISP) != NULL);
1989
1990 realpos = curwin->w_cursor;
1991 curwin->w_cursor.col = 0;
1992
1993 if ((pos = findmatch(NULL, '(')) == NULL)
1994 pos = findmatch(NULL, '[');
1995 else
1996 {
1997 paren = *pos;
1998 pos = findmatch(NULL, '[');
1999 if (pos == NULL || LT_POSP(pos, &paren))
2000 pos = &paren;
2001 }
2002 if (pos != NULL)
2003 {
2004 // Extra trick: Take the indent of the first previous non-white
2005 // line that is at the same () level.
2006 amount = -1;
2007 parencount = 0;
2008
2009 while (--curwin->w_cursor.lnum >= pos->lnum)
2010 {
2011 if (linewhite(curwin->w_cursor.lnum))
2012 continue;
2013 for (that = ml_get_curline(); *that != NUL; ++that)
2014 {
2015 if (*that == ';')
2016 {
2017 while (*(that + 1) != NUL)
2018 ++that;
2019 continue;
2020 }
2021 if (*that == '\\')
2022 {
2023 if (*(that + 1) != NUL)
2024 ++that;
2025 continue;
2026 }
2027 if (*that == '"' && *(that + 1) != NUL)
2028 {
2029 while (*++that && *that != '"')
2030 {
2031 // skipping escaped characters in the string
2032 if (*that == '\\')
2033 {
2034 if (*++that == NUL)
2035 break;
2036 if (that[1] == NUL)
2037 {
2038 ++that;
2039 break;
2040 }
2041 }
2042 }
Bram Moolenaar0e8e9382022-06-18 12:51:11 +01002043 if (*that == NUL)
2044 break;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002045 }
2046 if (*that == '(' || *that == '[')
2047 ++parencount;
2048 else if (*that == ')' || *that == ']')
2049 --parencount;
2050 }
2051 if (parencount == 0)
2052 {
2053 amount = get_indent();
2054 break;
2055 }
2056 }
2057
2058 if (amount == -1)
2059 {
2060 curwin->w_cursor.lnum = pos->lnum;
2061 curwin->w_cursor.col = pos->col;
2062 col = pos->col;
2063
2064 that = ml_get_curline();
2065
2066 if (vi_lisp && get_indent() == 0)
2067 amount = 2;
2068 else
2069 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002070 char_u *line = that;
2071 chartabsize_T cts;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002072
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002073 init_chartabsize_arg(&cts, curwin, pos->lnum, 0, line, line);
2074 while (*cts.cts_ptr != NUL && col > 0)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002075 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002076 cts.cts_vcol += lbr_chartabsize_adv(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002077 col--;
2078 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002079 amount = cts.cts_vcol;
2080 that = cts.cts_ptr;
2081 clear_chartabsize_arg(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002082
2083 // Some keywords require "body" indenting rules (the
2084 // non-standard-lisp ones are Scheme special forms):
2085 //
2086 // (let ((a 1)) instead (let ((a 1))
2087 // (...)) of (...))
2088
2089 if (!vi_lisp && (*that == '(' || *that == '[')
2090 && lisp_match(that + 1))
2091 amount += 2;
2092 else
2093 {
Bram Moolenaar8eba2bd2022-06-22 19:59:28 +01002094 if (*that != NUL)
2095 {
2096 that++;
2097 amount++;
2098 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002099 firsttry = amount;
2100
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002101 init_chartabsize_arg(&cts, curwin, (colnr_T)(that - line),
2102 amount, line, that);
2103 while (VIM_ISWHITE(*cts.cts_ptr))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002104 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002105 cts.cts_vcol += lbr_chartabsize(&cts);
2106 ++cts.cts_ptr;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002107 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002108 that = cts.cts_ptr;
2109 amount = cts.cts_vcol;
2110 clear_chartabsize_arg(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002111
2112 if (*that && *that != ';') // not a comment line
2113 {
2114 // test *that != '(' to accommodate first let/do
2115 // argument if it is more than one line
2116 if (!vi_lisp && *that != '(' && *that != '[')
2117 firsttry++;
2118
2119 parencount = 0;
2120 quotecount = 0;
2121
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002122 init_chartabsize_arg(&cts, curwin,
2123 (colnr_T)(that - line), amount, line, that);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002124 if (vi_lisp
2125 || (*that != '"'
2126 && *that != '\''
2127 && *that != '#'
2128 && (*that < '0' || *that > '9')))
2129 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002130 while (*cts.cts_ptr
2131 && (!VIM_ISWHITE(*cts.cts_ptr)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002132 || quotecount
2133 || parencount)
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002134 && (!((*cts.cts_ptr == '('
2135 || *cts.cts_ptr == '[')
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002136 && !quotecount
2137 && !parencount
2138 && vi_lisp)))
2139 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002140 if (*cts.cts_ptr == '"')
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002141 quotecount = !quotecount;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002142 if ((*cts.cts_ptr == '(' || *cts.cts_ptr == '[')
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002143 && !quotecount)
2144 ++parencount;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002145 if ((*cts.cts_ptr == ')' || *cts.cts_ptr == ']')
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002146 && !quotecount)
2147 --parencount;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002148 if (*cts.cts_ptr == '\\'
2149 && *(cts.cts_ptr+1) != NUL)
2150 cts.cts_vcol += lbr_chartabsize_adv(&cts);
2151 cts.cts_vcol += lbr_chartabsize_adv(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002152 }
2153 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002154 while (VIM_ISWHITE(*cts.cts_ptr))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002155 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002156 cts.cts_vcol += lbr_chartabsize(&cts);
2157 ++cts.cts_ptr;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002158 }
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01002159 that = cts.cts_ptr;
2160 amount = cts.cts_vcol;
2161 clear_chartabsize_arg(&cts);
2162
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002163 if (!*that || *that == ';')
2164 amount = firsttry;
2165 }
2166 }
2167 }
2168 }
2169 }
2170 else
2171 amount = 0; // no matching '(' or '[' found, use zero indent
2172
2173 curwin->w_cursor = realpos;
2174
2175 return amount;
2176}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002177
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002178/*
2179 * Re-indent the current line, based on the current contents of it and the
2180 * surrounding lines. Fixing the cursor position seems really easy -- I'm very
2181 * confused what all the part that handles Control-T is doing that I'm not.
2182 * "get_the_indent" should be get_c_indent, get_expr_indent or get_lisp_indent.
2183 */
2184
2185 void
2186fixthisline(int (*get_the_indent)(void))
2187{
2188 int amount = get_the_indent();
2189
Yegappan Lakshmanan0233bdf2023-01-12 12:33:30 +00002190 if (amount < 0)
2191 return;
2192
2193 change_indent(INDENT_SET, amount, FALSE, 0, TRUE);
2194 if (linewhite(curwin->w_cursor.lnum))
2195 did_ai = TRUE; // delete the indent if the line stays empty
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002196}
2197
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00002198/*
Bram Moolenaar49846fb2022-10-15 16:05:33 +01002199 * Return TRUE if 'indentexpr' should be used for Lisp indenting.
2200 * Caller may want to check 'autoindent'.
2201 */
2202 int
2203use_indentexpr_for_lisp(void)
2204{
2205#ifdef FEAT_EVAL
2206 return curbuf->b_p_lisp
2207 && *curbuf->b_p_inde != NUL
2208 && STRCMP(curbuf->b_p_lop, "expr:1") == 0;
2209#else
2210 return FALSE;
2211#endif
2212}
2213
2214/*
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00002215 * Fix indent for 'lisp' and 'cindent'.
2216 */
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002217 void
2218fix_indent(void)
2219{
2220 if (p_paste)
Bram Moolenaar49846fb2022-10-15 16:05:33 +01002221 return; // no auto-indenting when 'paste' is set
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002222 if (curbuf->b_p_lisp && curbuf->b_p_ai)
Bram Moolenaar49846fb2022-10-15 16:05:33 +01002223 {
2224 if (use_indentexpr_for_lisp())
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002225 do_c_expr_indent();
Bram Moolenaar49846fb2022-10-15 16:05:33 +01002226 else
2227 fixthisline(get_lisp_indent);
2228 }
2229 else if (cindent_on())
2230 do_c_expr_indent();
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002231}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002232
2233#if defined(FEAT_EVAL) || defined(PROTO)
2234/*
2235 * "indent()" function
2236 */
2237 void
2238f_indent(typval_T *argvars, typval_T *rettv)
2239{
2240 linenr_T lnum;
2241
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02002242 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
2243 return;
2244
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002245 lnum = tv_get_lnum(argvars);
2246 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
2247 rettv->vval.v_number = get_indent_lnum(lnum);
2248 else
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +00002249 {
2250 if (in_vim9script())
2251 semsg(_(e_invalid_line_number_nr), lnum);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002252 rettv->vval.v_number = -1;
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +00002253 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002254}
2255
2256/*
2257 * "lispindent(lnum)" function
2258 */
2259 void
2260f_lispindent(typval_T *argvars UNUSED, typval_T *rettv)
2261{
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002262 pos_T pos;
2263 linenr_T lnum;
2264
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02002265 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
2266 return;
2267
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002268 pos = curwin->w_cursor;
2269 lnum = tv_get_lnum(argvars);
2270 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
2271 {
2272 curwin->w_cursor.lnum = lnum;
2273 rettv->vval.v_number = get_lisp_indent();
2274 curwin->w_cursor = pos;
2275 }
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +00002276 else if (in_vim9script())
2277 semsg(_(e_invalid_line_number_nr), lnum);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002278 else
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002279 rettv->vval.v_number = -1;
2280}
2281#endif