blob: d7a2e550ad54d57a63a86537563002441b2c8c59 [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);
Bram Moolenaar2ddb89f2021-09-04 21:20:41 +020077 vim_free(*array);
78 *array = NULL;
Bram Moolenaarb7081e12021-09-04 18:47:28 +020079 return FAIL;
80 }
81 (*array)[t++] = n;
82 while (*cp != NUL && *cp != ',')
Bram Moolenaare677df82019-09-02 22:31:11 +020083 ++cp;
84 if (*cp != NUL)
85 ++cp;
86 }
87
Bram Moolenaarb7081e12021-09-04 18:47:28 +020088 return OK;
Bram Moolenaare677df82019-09-02 22:31:11 +020089}
90
91/*
92 * Calculate the number of screen spaces a tab will occupy.
93 * If "vts" is set then the tab widths are taken from that array,
94 * otherwise the value of ts is used.
95 */
96 int
97tabstop_padding(colnr_T col, int ts_arg, int *vts)
98{
99 int ts = ts_arg == 0 ? 8 : ts_arg;
100 int tabcount;
101 colnr_T tabcol = 0;
102 int t;
103 int padding = 0;
104
105 if (vts == NULL || vts[0] == 0)
106 return ts - (col % ts);
107
108 tabcount = vts[0];
109
110 for (t = 1; t <= tabcount; ++t)
111 {
112 tabcol += vts[t];
113 if (tabcol > col)
114 {
115 padding = (int)(tabcol - col);
116 break;
117 }
118 }
119 if (t > tabcount)
120 padding = vts[tabcount] - (int)((col - tabcol) % vts[tabcount]);
121
122 return padding;
123}
124
125/*
126 * Find the size of the tab that covers a particular column.
127 */
128 int
129tabstop_at(colnr_T col, int ts, int *vts)
130{
131 int tabcount;
132 colnr_T tabcol = 0;
133 int t;
134 int tab_size = 0;
135
136 if (vts == 0 || vts[0] == 0)
137 return ts;
138
139 tabcount = vts[0];
140 for (t = 1; t <= tabcount; ++t)
141 {
142 tabcol += vts[t];
143 if (tabcol > col)
144 {
145 tab_size = vts[t];
146 break;
147 }
148 }
149 if (t > tabcount)
150 tab_size = vts[tabcount];
151
152 return tab_size;
153}
154
155/*
156 * Find the column on which a tab starts.
157 */
158 colnr_T
159tabstop_start(colnr_T col, int ts, int *vts)
160{
161 int tabcount;
162 colnr_T tabcol = 0;
163 int t;
Bram Moolenaar6ed545e2022-05-09 20:09:23 +0100164 int excess;
Bram Moolenaare677df82019-09-02 22:31:11 +0200165
166 if (vts == NULL || vts[0] == 0)
167 return (col / ts) * ts;
168
169 tabcount = vts[0];
170 for (t = 1; t <= tabcount; ++t)
171 {
172 tabcol += vts[t];
173 if (tabcol > col)
174 return tabcol - vts[t];
175 }
176
177 excess = tabcol % vts[tabcount];
178 return excess + ((col - excess) / vts[tabcount]) * vts[tabcount];
179}
180
181/*
182 * Find the number of tabs and spaces necessary to get from one column
183 * to another.
184 */
185 void
186tabstop_fromto(
187 colnr_T start_col,
188 colnr_T end_col,
189 int ts_arg,
190 int *vts,
191 int *ntabs,
192 int *nspcs)
193{
194 int spaces = end_col - start_col;
195 colnr_T tabcol = 0;
196 int padding = 0;
197 int tabcount;
198 int t;
199 int ts = ts_arg == 0 ? curbuf->b_p_ts : ts_arg;
200
201 if (vts == NULL || vts[0] == 0)
202 {
203 int tabs = 0;
204 int initspc = 0;
205
206 initspc = ts - (start_col % ts);
207 if (spaces >= initspc)
208 {
209 spaces -= initspc;
210 tabs++;
211 }
212 tabs += spaces / ts;
213 spaces -= (spaces / ts) * ts;
214
215 *ntabs = tabs;
216 *nspcs = spaces;
217 return;
218 }
219
220 // Find the padding needed to reach the next tabstop.
221 tabcount = vts[0];
222 for (t = 1; t <= tabcount; ++t)
223 {
224 tabcol += vts[t];
225 if (tabcol > start_col)
226 {
227 padding = (int)(tabcol - start_col);
228 break;
229 }
230 }
231 if (t > tabcount)
232 padding = vts[tabcount] - (int)((start_col - tabcol) % vts[tabcount]);
233
234 // If the space needed is less than the padding no tabs can be used.
235 if (spaces < padding)
236 {
237 *ntabs = 0;
238 *nspcs = spaces;
239 return;
240 }
241
242 *ntabs = 1;
243 spaces -= padding;
244
245 // At least one tab has been used. See if any more will fit.
246 while (spaces != 0 && ++t <= tabcount)
247 {
248 padding = vts[t];
249 if (spaces < padding)
250 {
251 *nspcs = spaces;
252 return;
253 }
254 ++*ntabs;
255 spaces -= padding;
256 }
257
258 *ntabs += spaces / vts[tabcount];
259 *nspcs = spaces % vts[tabcount];
260}
261
262/*
263 * See if two tabstop arrays contain the same values.
264 */
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200265 static int
Bram Moolenaare677df82019-09-02 22:31:11 +0200266tabstop_eq(int *ts1, int *ts2)
267{
268 int t;
269
270 if ((ts1 == 0 && ts2) || (ts1 && ts2 == 0))
271 return FALSE;
272 if (ts1 == ts2)
273 return TRUE;
274 if (ts1[0] != ts2[0])
275 return FALSE;
276
277 for (t = 1; t <= ts1[0]; ++t)
278 if (ts1[t] != ts2[t])
279 return FALSE;
280
281 return TRUE;
282}
283
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200284# if defined(FEAT_BEVAL) || defined(PROTO)
Bram Moolenaare677df82019-09-02 22:31:11 +0200285/*
286 * Copy a tabstop array, allocating space for the new array.
287 */
288 int *
289tabstop_copy(int *oldts)
290{
291 int *newts;
292 int t;
293
294 if (oldts == NULL)
295 return NULL;
296 newts = ALLOC_MULT(int, oldts[0] + 1);
297 if (newts != NULL)
298 for (t = 0; t <= oldts[0]; ++t)
299 newts[t] = oldts[t];
300 return newts;
301}
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200302# endif
Bram Moolenaare677df82019-09-02 22:31:11 +0200303
304/*
305 * Return a count of the number of tabstops.
306 */
307 int
308tabstop_count(int *ts)
309{
310 return ts != NULL ? ts[0] : 0;
311}
312
313/*
314 * Return the first tabstop, or 8 if there are no tabstops defined.
315 */
316 int
317tabstop_first(int *ts)
318{
319 return ts != NULL ? ts[1] : 8;
320}
321
322#endif
323
324/*
325 * Return the effective shiftwidth value for current buffer, using the
326 * 'tabstop' value when 'shiftwidth' is zero.
327 */
328 long
329get_sw_value(buf_T *buf)
330{
331 return get_sw_value_col(buf, 0);
332}
333
334/*
335 * Idem, using "pos".
336 */
337 static long
338get_sw_value_pos(buf_T *buf, pos_T *pos)
339{
340 pos_T save_cursor = curwin->w_cursor;
341 long sw_value;
342
343 curwin->w_cursor = *pos;
344 sw_value = get_sw_value_col(buf, get_nolist_virtcol());
345 curwin->w_cursor = save_cursor;
346 return sw_value;
347}
348
349/*
350 * Idem, using the first non-black in the current line.
351 */
352 long
353get_sw_value_indent(buf_T *buf)
354{
355 pos_T pos = curwin->w_cursor;
356
357 pos.col = getwhitecols_curline();
358 return get_sw_value_pos(buf, &pos);
359}
360
361/*
362 * Idem, using virtual column "col".
363 */
364 long
365get_sw_value_col(buf_T *buf, colnr_T col UNUSED)
366{
367 return buf->b_p_sw ? buf->b_p_sw :
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200368#ifdef FEAT_VARTABS
Bram Moolenaare677df82019-09-02 22:31:11 +0200369 tabstop_at(col, buf->b_p_ts, buf->b_p_vts_array);
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200370#else
Bram Moolenaare677df82019-09-02 22:31:11 +0200371 buf->b_p_ts;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200372#endif
Bram Moolenaare677df82019-09-02 22:31:11 +0200373}
374
375/*
376 * Return the effective softtabstop value for the current buffer, using the
377 * 'shiftwidth' value when 'softtabstop' is negative.
378 */
379 long
380get_sts_value(void)
381{
382 return curbuf->b_p_sts < 0 ? get_sw_value(curbuf) : curbuf->b_p_sts;
383}
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200384
385/*
386 * Count the size (in window cells) of the indent in the current line.
387 */
388 int
389get_indent(void)
390{
391#ifdef FEAT_VARTABS
392 return get_indent_str_vtab(ml_get_curline(), (int)curbuf->b_p_ts,
393 curbuf->b_p_vts_array, FALSE);
394#else
395 return get_indent_str(ml_get_curline(), (int)curbuf->b_p_ts, FALSE);
396#endif
397}
398
399/*
400 * Count the size (in window cells) of the indent in line "lnum".
401 */
402 int
403get_indent_lnum(linenr_T lnum)
404{
405#ifdef FEAT_VARTABS
406 return get_indent_str_vtab(ml_get(lnum), (int)curbuf->b_p_ts,
407 curbuf->b_p_vts_array, FALSE);
408#else
409 return get_indent_str(ml_get(lnum), (int)curbuf->b_p_ts, FALSE);
410#endif
411}
412
413#if defined(FEAT_FOLDING) || defined(PROTO)
414/*
415 * Count the size (in window cells) of the indent in line "lnum" of buffer
416 * "buf".
417 */
418 int
419get_indent_buf(buf_T *buf, linenr_T lnum)
420{
421# ifdef FEAT_VARTABS
422 return get_indent_str_vtab(ml_get_buf(buf, lnum, FALSE),
423 (int)curbuf->b_p_ts, buf->b_p_vts_array, FALSE);
424# else
425 return get_indent_str(ml_get_buf(buf, lnum, FALSE), (int)buf->b_p_ts, FALSE);
426# endif
427}
428#endif
429
430/*
431 * count the size (in window cells) of the indent in line "ptr", with
432 * 'tabstop' at "ts"
433 */
434 int
435get_indent_str(
436 char_u *ptr,
437 int ts,
438 int list) // if TRUE, count only screen size for tabs
439{
440 int count = 0;
441
442 for ( ; *ptr; ++ptr)
443 {
444 if (*ptr == TAB)
445 {
Bram Moolenaareed9d462021-02-15 20:38:25 +0100446 if (!list || curwin->w_lcs_chars.tab1)
447 // count a tab for what it is worth
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200448 count += ts - (count % ts);
449 else
450 // In list mode, when tab is not set, count screen char width
451 // for Tab, displays: ^I
452 count += ptr2cells(ptr);
453 }
454 else if (*ptr == ' ')
455 ++count; // count a space for one
456 else
457 break;
458 }
459 return count;
460}
461
462#ifdef FEAT_VARTABS
463/*
464 * Count the size (in window cells) of the indent in line "ptr", using
465 * variable tabstops.
466 * if "list" is TRUE, count only screen size for tabs.
467 */
468 int
469get_indent_str_vtab(char_u *ptr, int ts, int *vts, int list)
470{
471 int count = 0;
472
473 for ( ; *ptr; ++ptr)
474 {
475 if (*ptr == TAB) // count a tab for what it is worth
476 {
Bram Moolenaareed9d462021-02-15 20:38:25 +0100477 if (!list || curwin->w_lcs_chars.tab1)
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200478 count += tabstop_padding(count, ts, vts);
479 else
480 // In list mode, when tab is not set, count screen char width
481 // for Tab, displays: ^I
482 count += ptr2cells(ptr);
483 }
484 else if (*ptr == ' ')
485 ++count; // count a space for one
486 else
487 break;
488 }
489 return count;
490}
491#endif
492
493/*
494 * Set the indent of the current line.
495 * Leaves the cursor on the first non-blank in the line.
496 * Caller must take care of undo.
497 * "flags":
498 * SIN_CHANGED: call changed_bytes() if the line was changed.
499 * SIN_INSERT: insert the indent in front of the line.
500 * SIN_UNDO: save line for undo before changing it.
501 * Returns TRUE if the line was changed.
502 */
503 int
504set_indent(
505 int size, // measured in spaces
506 int flags)
507{
508 char_u *p;
509 char_u *newline;
510 char_u *oldline;
511 char_u *s;
512 int todo;
513 int ind_len; // measured in characters
514 int line_len;
515 int doit = FALSE;
516 int ind_done = 0; // measured in spaces
517#ifdef FEAT_VARTABS
518 int ind_col = 0;
519#endif
520 int tab_pad;
521 int retval = FALSE;
522 int orig_char_len = -1; // number of initial whitespace chars when
523 // 'et' and 'pi' are both set
524
525 // First check if there is anything to do and compute the number of
526 // characters needed for the indent.
527 todo = size;
528 ind_len = 0;
529 p = oldline = ml_get_curline();
530
531 // Calculate the buffer size for the new indent, and check to see if it
532 // isn't already set
533
534 // if 'expandtab' isn't set: use TABs; if both 'expandtab' and
535 // 'preserveindent' are set count the number of characters at the
536 // beginning of the line to be copied
537 if (!curbuf->b_p_et || (!(flags & SIN_INSERT) && curbuf->b_p_pi))
538 {
539 // If 'preserveindent' is set then reuse as much as possible of
540 // the existing indent structure for the new indent
541 if (!(flags & SIN_INSERT) && curbuf->b_p_pi)
542 {
543 ind_done = 0;
544
545 // count as many characters as we can use
546 while (todo > 0 && VIM_ISWHITE(*p))
547 {
548 if (*p == TAB)
549 {
550#ifdef FEAT_VARTABS
551 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
552 curbuf->b_p_vts_array);
553#else
554 tab_pad = (int)curbuf->b_p_ts
555 - (ind_done % (int)curbuf->b_p_ts);
556#endif
557 // stop if this tab will overshoot the target
558 if (todo < tab_pad)
559 break;
560 todo -= tab_pad;
561 ++ind_len;
562 ind_done += tab_pad;
563 }
564 else
565 {
566 --todo;
567 ++ind_len;
568 ++ind_done;
569 }
570 ++p;
571 }
572
573#ifdef FEAT_VARTABS
574 // These diverge from this point.
575 ind_col = ind_done;
576#endif
577 // Set initial number of whitespace chars to copy if we are
578 // preserving indent but expandtab is set
579 if (curbuf->b_p_et)
580 orig_char_len = ind_len;
581
582 // Fill to next tabstop with a tab, if possible
583#ifdef FEAT_VARTABS
584 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
585 curbuf->b_p_vts_array);
586#else
587 tab_pad = (int)curbuf->b_p_ts - (ind_done % (int)curbuf->b_p_ts);
588#endif
589 if (todo >= tab_pad && orig_char_len == -1)
590 {
591 doit = TRUE;
592 todo -= tab_pad;
593 ++ind_len;
594 // ind_done += tab_pad;
595#ifdef FEAT_VARTABS
596 ind_col += tab_pad;
597#endif
598 }
599 }
600
601 // count tabs required for indent
602#ifdef FEAT_VARTABS
603 for (;;)
604 {
605 tab_pad = tabstop_padding(ind_col, curbuf->b_p_ts,
606 curbuf->b_p_vts_array);
607 if (todo < tab_pad)
608 break;
609 if (*p != TAB)
610 doit = TRUE;
611 else
612 ++p;
613 todo -= tab_pad;
614 ++ind_len;
615 ind_col += tab_pad;
616 }
617#else
618 while (todo >= (int)curbuf->b_p_ts)
619 {
620 if (*p != TAB)
621 doit = TRUE;
622 else
623 ++p;
624 todo -= (int)curbuf->b_p_ts;
625 ++ind_len;
626 // ind_done += (int)curbuf->b_p_ts;
627 }
628#endif
629 }
630 // count spaces required for indent
631 while (todo > 0)
632 {
633 if (*p != ' ')
634 doit = TRUE;
635 else
636 ++p;
637 --todo;
638 ++ind_len;
639 // ++ind_done;
640 }
641
642 // Return if the indent is OK already.
643 if (!doit && !VIM_ISWHITE(*p) && !(flags & SIN_INSERT))
644 return FALSE;
645
646 // Allocate memory for the new line.
647 if (flags & SIN_INSERT)
648 p = oldline;
649 else
650 p = skipwhite(p);
651 line_len = (int)STRLEN(p) + 1;
652
653 // If 'preserveindent' and 'expandtab' are both set keep the original
654 // characters and allocate accordingly. We will fill the rest with spaces
655 // after the if (!curbuf->b_p_et) below.
656 if (orig_char_len != -1)
657 {
658 newline = alloc(orig_char_len + size - ind_done + line_len);
659 if (newline == NULL)
660 return FALSE;
661 todo = size - ind_done;
662 ind_len = orig_char_len + todo; // Set total length of indent in
663 // characters, which may have been
664 // undercounted until now
665 p = oldline;
666 s = newline;
667 while (orig_char_len > 0)
668 {
669 *s++ = *p++;
670 orig_char_len--;
671 }
672
673 // Skip over any additional white space (useful when newindent is less
674 // than old)
675 while (VIM_ISWHITE(*p))
676 ++p;
677
678 }
679 else
680 {
681 todo = size;
682 newline = alloc(ind_len + line_len);
683 if (newline == NULL)
684 return FALSE;
685 s = newline;
686 }
687
688 // Put the characters in the new line.
689 // if 'expandtab' isn't set: use TABs
690 if (!curbuf->b_p_et)
691 {
692 // If 'preserveindent' is set then reuse as much as possible of
693 // the existing indent structure for the new indent
694 if (!(flags & SIN_INSERT) && curbuf->b_p_pi)
695 {
696 p = oldline;
697 ind_done = 0;
698
699 while (todo > 0 && VIM_ISWHITE(*p))
700 {
701 if (*p == TAB)
702 {
703#ifdef FEAT_VARTABS
704 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
705 curbuf->b_p_vts_array);
706#else
707 tab_pad = (int)curbuf->b_p_ts
708 - (ind_done % (int)curbuf->b_p_ts);
709#endif
710 // stop if this tab will overshoot the target
711 if (todo < tab_pad)
712 break;
713 todo -= tab_pad;
714 ind_done += tab_pad;
715 }
716 else
717 {
718 --todo;
719 ++ind_done;
720 }
721 *s++ = *p++;
722 }
723
724 // Fill to next tabstop with a tab, if possible
725#ifdef FEAT_VARTABS
726 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
727 curbuf->b_p_vts_array);
728#else
729 tab_pad = (int)curbuf->b_p_ts - (ind_done % (int)curbuf->b_p_ts);
730#endif
731 if (todo >= tab_pad)
732 {
733 *s++ = TAB;
734 todo -= tab_pad;
735#ifdef FEAT_VARTABS
736 ind_done += tab_pad;
737#endif
738 }
739
740 p = skipwhite(p);
741 }
742
743#ifdef FEAT_VARTABS
744 for (;;)
745 {
746 tab_pad = tabstop_padding(ind_done, curbuf->b_p_ts,
747 curbuf->b_p_vts_array);
748 if (todo < tab_pad)
749 break;
750 *s++ = TAB;
751 todo -= tab_pad;
752 ind_done += tab_pad;
753 }
754#else
755 while (todo >= (int)curbuf->b_p_ts)
756 {
757 *s++ = TAB;
758 todo -= (int)curbuf->b_p_ts;
759 }
760#endif
761 }
762 while (todo > 0)
763 {
764 *s++ = ' ';
765 --todo;
766 }
767 mch_memmove(s, p, (size_t)line_len);
768
769 // Replace the line (unless undo fails).
770 if (!(flags & SIN_UNDO) || u_savesub(curwin->w_cursor.lnum) == OK)
771 {
Bram Moolenaarcf306432020-06-29 20:40:37 +0200772 colnr_T old_offset = (colnr_T)(p - oldline);
773 colnr_T new_offset = (colnr_T)(s - newline);
774
775 // this may free "newline"
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200776 ml_replace(curwin->w_cursor.lnum, newline, FALSE);
777 if (flags & SIN_CHANGED)
778 changed_bytes(curwin->w_cursor.lnum, 0);
779
780 // Correct saved cursor position if it is in this line.
781 if (saved_cursor.lnum == curwin->w_cursor.lnum)
782 {
Bram Moolenaarcf306432020-06-29 20:40:37 +0200783 if (saved_cursor.col >= old_offset)
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200784 // cursor was after the indent, adjust for the number of
785 // bytes added/removed
Bram Moolenaarcf306432020-06-29 20:40:37 +0200786 saved_cursor.col += ind_len - old_offset;
787 else if (saved_cursor.col >= new_offset)
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200788 // cursor was in the indent, and is now after it, put it back
789 // at the start of the indent (replacing spaces with TAB)
Bram Moolenaarcf306432020-06-29 20:40:37 +0200790 saved_cursor.col = new_offset;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200791 }
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +0100792#ifdef FEAT_PROP_POPUP
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200793 {
Bram Moolenaarcf306432020-06-29 20:40:37 +0200794 int added = ind_len - old_offset;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200795
796 // When increasing indent this behaves like spaces were inserted at
797 // the old indent, when decreasing indent it behaves like spaces
798 // were deleted at the new indent.
799 adjust_prop_columns(curwin->w_cursor.lnum,
Bram Moolenaar8a77d202022-08-15 16:29:37 +0100800 added > 0 ? old_offset : (colnr_T)ind_len,
801 added, APC_INDENT);
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200802 }
803#endif
804 retval = TRUE;
805 }
806 else
807 vim_free(newline);
808
809 curwin->w_cursor.col = ind_len;
810 return retval;
811}
812
813/*
814 * Return the indent of the current line after a number. Return -1 if no
815 * number was found. Used for 'n' in 'formatoptions': numbered list.
816 * Since a pattern is used it can actually handle more than numbers.
817 */
818 int
819get_number_indent(linenr_T lnum)
820{
821 colnr_T col;
822 pos_T pos;
823
824 regmatch_T regmatch;
825 int lead_len = 0; // length of comment leader
826
827 if (lnum > curbuf->b_ml.ml_line_count)
828 return -1;
829 pos.lnum = 0;
830
831 // In format_lines() (i.e. not insert mode), fo+=q is needed too...
Bram Moolenaar24959102022-05-07 20:01:16 +0100832 if ((State & MODE_INSERT) || has_format_option(FO_Q_COMS))
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200833 lead_len = get_leader_len(ml_get(lnum), NULL, FALSE, TRUE);
834
835 regmatch.regprog = vim_regcomp(curbuf->b_p_flp, RE_MAGIC);
836 if (regmatch.regprog != NULL)
837 {
838 regmatch.rm_ic = FALSE;
839
840 // vim_regexec() expects a pointer to a line. This lets us
841 // start matching for the flp beyond any comment leader...
842 if (vim_regexec(&regmatch, ml_get(lnum) + lead_len, (colnr_T)0))
843 {
844 pos.lnum = lnum;
845 pos.col = (colnr_T)(*regmatch.endp - ml_get(lnum));
846 pos.coladd = 0;
847 }
848 vim_regfree(regmatch.regprog);
849 }
850
851 if (pos.lnum == 0 || *ml_get_pos(&pos) == NUL)
852 return -1;
853 getvcol(curwin, &pos, &col, NULL, NULL);
854 return (int)col;
855}
856
857#if defined(FEAT_LINEBREAK) || defined(PROTO)
858/*
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100859 * This is called when 'breakindentopt' is changed and when a window is
860 * initialized.
861 */
862 int
863briopt_check(win_T *wp)
864{
865 char_u *p;
866 int bri_shift = 0;
867 long bri_min = 20;
868 int bri_sbr = FALSE;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +0200869 int bri_list = 0;
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100870 int bri_vcol = 0;
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100871
872 p = wp->w_p_briopt;
873 while (*p != NUL)
874 {
875 if (STRNCMP(p, "shift:", 6) == 0
876 && ((p[6] == '-' && VIM_ISDIGIT(p[7])) || VIM_ISDIGIT(p[6])))
877 {
878 p += 6;
879 bri_shift = getdigits(&p);
880 }
881 else if (STRNCMP(p, "min:", 4) == 0 && VIM_ISDIGIT(p[4]))
882 {
883 p += 4;
884 bri_min = getdigits(&p);
885 }
886 else if (STRNCMP(p, "sbr", 3) == 0)
887 {
888 p += 3;
889 bri_sbr = TRUE;
890 }
Christian Brabandt4a0b85a2021-07-14 20:00:27 +0200891 else if (STRNCMP(p, "list:", 5) == 0)
892 {
893 p += 5;
894 bri_list = getdigits(&p);
895 }
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100896 else if (STRNCMP(p, "column:", 7) == 0)
897 {
898 p += 7;
899 bri_vcol = getdigits(&p);
900 }
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100901 if (*p != ',' && *p != NUL)
902 return FAIL;
903 if (*p == ',')
904 ++p;
905 }
906
Bram Moolenaarb81f56f2020-02-23 15:29:46 +0100907 wp->w_briopt_shift = bri_shift;
908 wp->w_briopt_min = bri_min;
909 wp->w_briopt_sbr = bri_sbr;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +0200910 wp->w_briopt_list = bri_list;
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100911 wp->w_briopt_vcol = bri_vcol;
Bram Moolenaar7bae0b12019-11-21 22:14:18 +0100912
913 return OK;
914}
915
916/*
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200917 * Return appropriate space number for breakindent, taking influencing
918 * parameters into account. Window must be specified, since it is not
919 * necessarily always the current one.
920 */
921 int
922get_breakindent_win(
923 win_T *wp,
924 char_u *line) // start of the line
925{
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000926 static int prev_indent = 0; // cached indent value
927 static long prev_ts = 0L; // cached tabstop value
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100928 static int prev_fnum = 0; // cached buffer number
929 static char_u *prev_line = NULL; // cached copy of "line"
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200930 static varnumber_T prev_tick = 0; // changedtick of cached value
931# ifdef FEAT_VARTABS
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000932 static int *prev_vts = NULL; // cached vartabs values
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200933# endif
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000934 static int prev_list = 0; // cached list value
935 static int prev_listopt = 0; // cached w_p_briopt_list value
Christian Brabandtc53b4672022-01-15 10:01:05 +0000936 // cached formatlistpat value
937 static char_u *prev_flp = NULL;
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200938 int bri = 0;
939 // window width minus window margin space, i.e. what rests for text
940 const int eff_wwidth = wp->w_width
941 - ((wp->w_p_nu || wp->w_p_rnu)
942 && (vim_strchr(p_cpo, CPO_NUMCOL) == NULL)
943 ? number_width(wp) + 1 : 0);
944
Christian Brabandtc53b4672022-01-15 10:01:05 +0000945 // used cached indent, unless
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100946 // - buffer changed
Christian Brabandtc53b4672022-01-15 10:01:05 +0000947 // - 'tabstop' changed
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100948 // - buffer was changed
Christian Brabandtc53b4672022-01-15 10:01:05 +0000949 // - 'briopt_list changed' changed or
950 // - 'formatlistpattern' changed
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100951 // - line changed
952 // - 'vartabs' changed
953 if (prev_fnum != wp->w_buffer->b_fnum
954 || prev_ts != wp->w_buffer->b_p_ts
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200955 || prev_tick != CHANGEDTICK(wp->w_buffer)
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000956 || prev_listopt != wp->w_briopt_list
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100957 || prev_flp == NULL
958 || STRCMP(prev_flp, get_flp_value(wp->w_buffer)) != 0
959 || prev_line == NULL || STRCMP(prev_line, line) != 0
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200960# ifdef FEAT_VARTABS
961 || prev_vts != wp->w_buffer->b_p_vts_array
962# endif
963 )
964 {
Bram Moolenaarc2a79b82022-07-01 13:15:35 +0100965 prev_fnum = wp->w_buffer->b_fnum;
966 vim_free(prev_line);
967 prev_line = vim_strsave(line);
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200968 prev_ts = wp->w_buffer->b_p_ts;
969 prev_tick = CHANGEDTICK(wp->w_buffer);
970# ifdef FEAT_VARTABS
971 prev_vts = wp->w_buffer->b_p_vts_array;
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100972 if (wp->w_briopt_vcol == 0)
973 prev_indent = get_indent_str_vtab(line,
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200974 (int)wp->w_buffer->b_p_ts,
975 wp->w_buffer->b_p_vts_array, wp->w_p_list);
976# else
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100977 if (wp->w_briopt_vcol == 0)
978 prev_indent = get_indent_str(line,
Bram Moolenaar14c01f82019-10-09 22:53:08 +0200979 (int)wp->w_buffer->b_p_ts, wp->w_p_list);
980# endif
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000981 prev_listopt = wp->w_briopt_list;
Christian Brabandtc53b4672022-01-15 10:01:05 +0000982 prev_list = 0;
983 vim_free(prev_flp);
984 prev_flp = vim_strsave(get_flp_value(wp->w_buffer));
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000985 // add additional indent for numbered lists
Christian Brabandte7d6dbc2022-05-06 12:21:04 +0100986 if (wp->w_briopt_list != 0 && wp->w_briopt_vcol == 0)
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000987 {
988 regmatch_T regmatch;
989
Christian Brabandtc53b4672022-01-15 10:01:05 +0000990 regmatch.regprog = vim_regcomp(prev_flp,
991 RE_MAGIC + RE_STRING + RE_AUTO + RE_STRICT);
Bram Moolenaarb2d85e32022-01-07 16:55:32 +0000992
993 if (regmatch.regprog != NULL)
994 {
995 regmatch.rm_ic = FALSE;
996 if (vim_regexec(&regmatch, line, 0))
997 {
998 if (wp->w_briopt_list > 0)
999 prev_list = wp->w_briopt_list;
1000 else
1001 prev_list = (*regmatch.endp - *regmatch.startp);
1002 }
1003 vim_regfree(regmatch.regprog);
1004 }
1005 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001006 }
Christian Brabandte7d6dbc2022-05-06 12:21:04 +01001007 if (wp->w_briopt_vcol != 0)
1008 {
1009 // column value has priority
1010 bri = wp->w_briopt_vcol;
1011 prev_list = 0;
1012 }
1013 else
1014 bri = prev_indent + wp->w_briopt_shift;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001015
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001016 // Add offset for number column, if 'n' is in 'cpoptions'
1017 bri += win_col_off2(wp);
1018
Christian Brabandt4a0b85a2021-07-14 20:00:27 +02001019 // add additional indent for numbered lists
Maxim Kimf674b352021-07-22 11:46:59 +02001020 if (wp->w_briopt_list != 0)
Christian Brabandt4a0b85a2021-07-14 20:00:27 +02001021 {
Bram Moolenaarb2d85e32022-01-07 16:55:32 +00001022 if (wp->w_briopt_list > 0)
1023 bri += prev_list;
1024 else
1025 bri = prev_list;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +02001026 }
1027
Maxim Kimf674b352021-07-22 11:46:59 +02001028 // indent minus the length of the showbreak string
1029 if (wp->w_briopt_sbr)
1030 bri -= vim_strsize(get_showbreak_value(wp));
1031
1032
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001033 // never indent past left window margin
1034 if (bri < 0)
1035 bri = 0;
Christian Brabandt4a0b85a2021-07-14 20:00:27 +02001036
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001037 // always leave at least bri_min characters on the left,
1038 // if text width is sufficient
Bram Moolenaarb81f56f2020-02-23 15:29:46 +01001039 else if (bri > eff_wwidth - wp->w_briopt_min)
1040 bri = (eff_wwidth - wp->w_briopt_min < 0)
1041 ? 0 : eff_wwidth - wp->w_briopt_min;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001042
1043 return bri;
1044}
1045#endif
1046
1047/*
1048 * When extra == 0: Return TRUE if the cursor is before or on the first
1049 * non-blank in the line.
1050 * When extra == 1: Return TRUE if the cursor is before the first non-blank in
1051 * the line.
1052 */
1053 int
1054inindent(int extra)
1055{
1056 char_u *ptr;
1057 colnr_T col;
1058
1059 for (col = 0, ptr = ml_get_curline(); VIM_ISWHITE(*ptr); ++col)
1060 ++ptr;
1061 if (col >= curwin->w_cursor.col + extra)
1062 return TRUE;
1063 else
1064 return FALSE;
1065}
1066
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001067/*
1068 * op_reindent - handle reindenting a block of lines.
1069 */
1070 void
1071op_reindent(oparg_T *oap, int (*how)(void))
1072{
Bram Moolenaar4c84dd32022-04-20 10:22:54 +01001073 long i = 0;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001074 char_u *l;
1075 int amount;
1076 linenr_T first_changed = 0;
1077 linenr_T last_changed = 0;
1078 linenr_T start_lnum = curwin->w_cursor.lnum;
1079
1080 // Don't even try when 'modifiable' is off.
1081 if (!curbuf->b_p_ma)
1082 {
Bram Moolenaar108010a2021-06-27 22:03:33 +02001083 emsg(_(e_cannot_make_changes_modifiable_is_off));
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001084 return;
1085 }
1086
Bram Moolenaare4686982022-04-19 18:28:45 +01001087 // Save for undo. Do this once for all lines, much faster than doing this
1088 // for each line separately, especially when undoing.
1089 if (u_savecommon(start_lnum - 1, start_lnum + oap->line_count,
1090 start_lnum + oap->line_count, FALSE) == OK)
1091 for (i = oap->line_count; --i >= 0 && !got_int; )
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001092 {
Bram Moolenaare4686982022-04-19 18:28:45 +01001093 // it's a slow thing to do, so give feedback so there's no worry
1094 // that the computer's just hung.
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001095
Bram Moolenaare4686982022-04-19 18:28:45 +01001096 if (i > 1
1097 && (i % 50 == 0 || i == oap->line_count - 1)
1098 && oap->line_count > p_report)
1099 smsg(_("%ld lines to indent... "), i);
1100
1101 // Be vi-compatible: For lisp indenting the first line is not
1102 // indented, unless there is only one line.
Bram Moolenaare4686982022-04-19 18:28:45 +01001103 if (i != oap->line_count - 1 || oap->line_count == 1
1104 || how != get_lisp_indent)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001105 {
Bram Moolenaare4686982022-04-19 18:28:45 +01001106 l = skipwhite(ml_get_curline());
1107 if (*l == NUL) // empty or blank line
1108 amount = 0;
1109 else
1110 amount = how(); // get the indent for this line
1111
1112 if (amount >= 0 && set_indent(amount, 0))
1113 {
1114 // did change the indent, call changed_lines() later
1115 if (first_changed == 0)
1116 first_changed = curwin->w_cursor.lnum;
1117 last_changed = curwin->w_cursor.lnum;
1118 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001119 }
Bram Moolenaare4686982022-04-19 18:28:45 +01001120 ++curwin->w_cursor.lnum;
1121 curwin->w_cursor.col = 0; // make sure it's valid
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001122 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001123
1124 // put cursor on first non-blank of indented line
1125 curwin->w_cursor.lnum = start_lnum;
1126 beginline(BL_SOL | BL_FIX);
1127
1128 // Mark changed lines so that they will be redrawn. When Visual
1129 // highlighting was present, need to continue until the last line. When
1130 // there is no change still need to remove the Visual highlighting.
1131 if (last_changed != 0)
1132 changed_lines(first_changed, 0,
1133 oap->is_VIsual ? start_lnum + oap->line_count :
1134 last_changed + 1, 0L);
1135 else if (oap->is_VIsual)
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001136 redraw_curbuf_later(UPD_INVERTED);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001137
1138 if (oap->line_count > p_report)
1139 {
1140 i = oap->line_count - (i + 1);
1141 smsg(NGETTEXT("%ld line indented ",
1142 "%ld lines indented ", i), i);
1143 }
Bram Moolenaare1004402020-10-24 20:49:43 +02001144 if ((cmdmod.cmod_flags & CMOD_LOCKMARKS) == 0)
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +01001145 {
1146 // set '[ and '] marks
1147 curbuf->b_op_start = oap->start;
1148 curbuf->b_op_end = oap->end;
1149 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001150}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001151
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001152/*
1153 * Return TRUE if lines starting with '#' should be left aligned.
1154 */
1155 int
1156preprocs_left(void)
1157{
1158 return
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001159 (curbuf->b_p_si && !curbuf->b_p_cin) ||
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001160 (curbuf->b_p_cin && in_cinkeys('#', ' ', TRUE)
1161 && curbuf->b_ind_hash_comment == 0)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001162 ;
1163}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001164
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001165/*
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001166 * Return TRUE if the conditions are OK for smart indenting.
1167 */
1168 int
1169may_do_si()
1170{
1171 return curbuf->b_p_si
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001172 && !curbuf->b_p_cin
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001173# ifdef FEAT_EVAL
1174 && *curbuf->b_p_inde == NUL
1175# endif
1176 && !p_paste;
1177}
1178
1179/*
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001180 * Try to do some very smart auto-indenting.
1181 * Used when inserting a "normal" character.
1182 */
1183 void
1184ins_try_si(int c)
1185{
1186 pos_T *pos, old_pos;
1187 char_u *ptr;
1188 int i;
1189 int temp;
1190
1191 // do some very smart indenting when entering '{' or '}'
Bram Moolenaar2e444bb2022-05-14 12:54:23 +01001192 if (((did_si || can_si_back) && c == '{')
1193 || (can_si && c == '}' && inindent(0)))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001194 {
1195 // for '}' set indent equal to indent of line containing matching '{'
1196 if (c == '}' && (pos = findmatch(NULL, '{')) != NULL)
1197 {
1198 old_pos = curwin->w_cursor;
1199 // If the matching '{' has a ')' immediately before it (ignoring
1200 // white-space), then line up with the start of the line
1201 // containing the matching '(' if there is one. This handles the
1202 // case where an "if (..\n..) {" statement continues over multiple
1203 // lines -- webb
1204 ptr = ml_get(pos->lnum);
1205 i = pos->col;
1206 if (i > 0) // skip blanks before '{'
1207 while (--i > 0 && VIM_ISWHITE(ptr[i]))
1208 ;
1209 curwin->w_cursor.lnum = pos->lnum;
1210 curwin->w_cursor.col = i;
1211 if (ptr[i] == ')' && (pos = findmatch(NULL, '(')) != NULL)
1212 curwin->w_cursor = *pos;
1213 i = get_indent();
1214 curwin->w_cursor = old_pos;
1215 if (State & VREPLACE_FLAG)
1216 change_indent(INDENT_SET, i, FALSE, NUL, TRUE);
1217 else
1218 (void)set_indent(i, SIN_CHANGED);
1219 }
1220 else if (curwin->w_cursor.col > 0)
1221 {
1222 // when inserting '{' after "O" reduce indent, but not
1223 // more than indent of previous line
1224 temp = TRUE;
1225 if (c == '{' && can_si_back && curwin->w_cursor.lnum > 1)
1226 {
1227 old_pos = curwin->w_cursor;
1228 i = get_indent();
1229 while (curwin->w_cursor.lnum > 1)
1230 {
1231 ptr = skipwhite(ml_get(--(curwin->w_cursor.lnum)));
1232
1233 // ignore empty lines and lines starting with '#'.
1234 if (*ptr != '#' && *ptr != NUL)
1235 break;
1236 }
1237 if (get_indent() >= i)
1238 temp = FALSE;
1239 curwin->w_cursor = old_pos;
1240 }
1241 if (temp)
1242 shift_line(TRUE, FALSE, 1, TRUE);
1243 }
1244 }
1245
1246 // set indent of '#' always to 0
Bram Moolenaarde5cf282022-05-14 11:52:23 +01001247 if (curwin->w_cursor.col > 0 && can_si && c == '#' && inindent(0))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001248 {
1249 // remember current indent for next line
1250 old_indent = get_indent();
1251 (void)set_indent(0, SIN_CHANGED);
1252 }
1253
1254 // Adjust ai_col, the char at this position can be deleted.
1255 if (ai_col > curwin->w_cursor.col)
1256 ai_col = curwin->w_cursor.col;
1257}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001258
1259/*
1260 * Insert an indent (for <Tab> or CTRL-T) or delete an indent (for CTRL-D).
1261 * Keep the cursor on the same character.
1262 * type == INDENT_INC increase indent (for CTRL-T or <Tab>)
1263 * type == INDENT_DEC decrease indent (for CTRL-D)
1264 * type == INDENT_SET set indent to "amount"
1265 * if round is TRUE, round the indent to 'shiftwidth' (only with _INC and _Dec).
1266 */
1267 void
1268change_indent(
1269 int type,
1270 int amount,
1271 int round,
1272 int replaced, // replaced character, put on replace stack
1273 int call_changed_bytes) // call changed_bytes()
1274{
1275 int vcol;
1276 int last_vcol;
1277 int insstart_less; // reduction for Insstart.col
1278 int new_cursor_col;
1279 int i;
1280 char_u *ptr;
1281 int save_p_list;
1282 int start_col;
1283 colnr_T vc;
1284 colnr_T orig_col = 0; // init for GCC
1285 char_u *new_line, *orig_line = NULL; // init for GCC
1286
Bram Moolenaar24959102022-05-07 20:01:16 +01001287 // MODE_VREPLACE state needs to know what the line was like before changing
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001288 if (State & VREPLACE_FLAG)
1289 {
1290 orig_line = vim_strsave(ml_get_curline()); // Deal with NULL below
1291 orig_col = curwin->w_cursor.col;
1292 }
1293
1294 // for the following tricks we don't want list mode
1295 save_p_list = curwin->w_p_list;
1296 curwin->w_p_list = FALSE;
1297 vc = getvcol_nolist(&curwin->w_cursor);
1298 vcol = vc;
1299
1300 // For Replace mode we need to fix the replace stack later, which is only
1301 // possible when the cursor is in the indent. Remember the number of
1302 // characters before the cursor if it's possible.
1303 start_col = curwin->w_cursor.col;
1304
1305 // determine offset from first non-blank
1306 new_cursor_col = curwin->w_cursor.col;
1307 beginline(BL_WHITE);
1308 new_cursor_col -= curwin->w_cursor.col;
1309
1310 insstart_less = curwin->w_cursor.col;
1311
1312 // If the cursor is in the indent, compute how many screen columns the
1313 // cursor is to the left of the first non-blank.
1314 if (new_cursor_col < 0)
1315 vcol = get_indent() - vcol;
1316
1317 if (new_cursor_col > 0) // can't fix replace stack
1318 start_col = -1;
1319
1320 // Set the new indent. The cursor will be put on the first non-blank.
1321 if (type == INDENT_SET)
1322 (void)set_indent(amount, call_changed_bytes ? SIN_CHANGED : 0);
1323 else
1324 {
1325 int save_State = State;
1326
1327 // Avoid being called recursively.
1328 if (State & VREPLACE_FLAG)
Bram Moolenaar24959102022-05-07 20:01:16 +01001329 State = MODE_INSERT;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001330 shift_line(type == INDENT_DEC, round, 1, call_changed_bytes);
1331 State = save_State;
1332 }
1333 insstart_less -= curwin->w_cursor.col;
1334
1335 // Try to put cursor on same character.
1336 // If the cursor is at or after the first non-blank in the line,
1337 // compute the cursor column relative to the column of the first
1338 // non-blank character.
1339 // If we are not in insert mode, leave the cursor on the first non-blank.
1340 // If the cursor is before the first non-blank, position it relative
1341 // to the first non-blank, counted in screen columns.
1342 if (new_cursor_col >= 0)
1343 {
1344 // When changing the indent while the cursor is touching it, reset
1345 // Insstart_col to 0.
1346 if (new_cursor_col == 0)
1347 insstart_less = MAXCOL;
1348 new_cursor_col += curwin->w_cursor.col;
1349 }
Bram Moolenaar24959102022-05-07 20:01:16 +01001350 else if (!(State & MODE_INSERT))
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001351 new_cursor_col = curwin->w_cursor.col;
1352 else
1353 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001354 chartabsize_T cts;
1355
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001356 // Compute the screen column where the cursor should be.
1357 vcol = get_indent() - vcol;
1358 curwin->w_virtcol = (colnr_T)((vcol < 0) ? 0 : vcol);
1359
1360 // Advance the cursor until we reach the right screen column.
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001361 last_vcol = 0;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001362 ptr = ml_get_curline();
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001363 init_chartabsize_arg(&cts, curwin, 0, 0, ptr, ptr);
1364 while (cts.cts_vcol <= (int)curwin->w_virtcol)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001365 {
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001366 last_vcol = cts.cts_vcol;
1367 if (cts.cts_vcol > 0)
1368 MB_PTR_ADV(cts.cts_ptr);
1369 if (*cts.cts_ptr == NUL)
Bram Moolenaar4e889f92022-02-21 19:36:12 +00001370 break;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001371 cts.cts_vcol += lbr_chartabsize(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001372 }
1373 vcol = last_vcol;
Bram Moolenaar7f9969c2022-07-25 18:13:54 +01001374 new_cursor_col = cts.cts_ptr - cts.cts_line;
1375 clear_chartabsize_arg(&cts);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001376
1377 // May need to insert spaces to be able to position the cursor on
1378 // the right screen column.
1379 if (vcol != (int)curwin->w_virtcol)
1380 {
1381 curwin->w_cursor.col = (colnr_T)new_cursor_col;
1382 i = (int)curwin->w_virtcol - vcol;
1383 ptr = alloc(i + 1);
1384 if (ptr != NULL)
1385 {
1386 new_cursor_col += i;
1387 ptr[i] = NUL;
1388 while (--i >= 0)
1389 ptr[i] = ' ';
1390 ins_str(ptr);
1391 vim_free(ptr);
1392 }
1393 }
1394
1395 // When changing the indent while the cursor is in it, reset
1396 // Insstart_col to 0.
1397 insstart_less = MAXCOL;
1398 }
1399
1400 curwin->w_p_list = save_p_list;
1401
1402 if (new_cursor_col <= 0)
1403 curwin->w_cursor.col = 0;
1404 else
1405 curwin->w_cursor.col = (colnr_T)new_cursor_col;
1406 curwin->w_set_curswant = TRUE;
1407 changed_cline_bef_curs();
1408
1409 // May have to adjust the start of the insert.
Bram Moolenaar24959102022-05-07 20:01:16 +01001410 if (State & MODE_INSERT)
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001411 {
1412 if (curwin->w_cursor.lnum == Insstart.lnum && Insstart.col != 0)
1413 {
1414 if ((int)Insstart.col <= insstart_less)
1415 Insstart.col = 0;
1416 else
1417 Insstart.col -= insstart_less;
1418 }
1419 if ((int)ai_col <= insstart_less)
1420 ai_col = 0;
1421 else
1422 ai_col -= insstart_less;
1423 }
1424
Bram Moolenaar24959102022-05-07 20:01:16 +01001425 // For MODE_REPLACE state, may have to fix the replace stack, if it's
1426 // possible. If the number of characters before the cursor decreased, need
1427 // to pop a few characters from the replace stack.
Bram Moolenaar14c01f82019-10-09 22:53:08 +02001428 // If the number of characters before the cursor increased, need to push a
1429 // few NULs onto the replace stack.
1430 if (REPLACE_NORMAL(State) && start_col >= 0)
1431 {
1432 while (start_col > (int)curwin->w_cursor.col)
1433 {
1434 replace_join(0); // remove a NUL from the replace stack
1435 --start_col;
1436 }
1437 while (start_col < (int)curwin->w_cursor.col || replaced)
1438 {
1439 replace_push(NUL);
1440 if (replaced)
1441 {
1442 replace_push(replaced);
1443 replaced = NUL;
1444 }
1445 ++start_col;
1446 }
1447 }
1448
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 {
1908 indent = (int)eval_to_number(inde_copy);
1909 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);
1954 if (STRNCMP(buf, p, len) == 0 && p[len] == ' ')
1955 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
2190 if (amount >= 0)
2191 {
2192 change_indent(INDENT_SET, amount, FALSE, 0, TRUE);
2193 if (linewhite(curwin->w_cursor.lnum))
2194 did_ai = TRUE; // delete the indent if the line stays empty
2195 }
2196}
2197
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00002198/*
2199 * Fix indent for 'lisp' and 'cindent'.
2200 */
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002201 void
2202fix_indent(void)
2203{
2204 if (p_paste)
2205 return;
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002206 if (curbuf->b_p_lisp && curbuf->b_p_ai)
2207 fixthisline(get_lisp_indent);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002208 else
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002209 if (cindent_on())
2210 do_c_expr_indent();
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002211}
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002212
2213#if defined(FEAT_EVAL) || defined(PROTO)
2214/*
2215 * "indent()" function
2216 */
2217 void
2218f_indent(typval_T *argvars, typval_T *rettv)
2219{
2220 linenr_T lnum;
2221
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02002222 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
2223 return;
2224
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002225 lnum = tv_get_lnum(argvars);
2226 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
2227 rettv->vval.v_number = get_indent_lnum(lnum);
2228 else
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +00002229 {
2230 if (in_vim9script())
2231 semsg(_(e_invalid_line_number_nr), lnum);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002232 rettv->vval.v_number = -1;
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +00002233 }
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002234}
2235
2236/*
2237 * "lispindent(lnum)" function
2238 */
2239 void
2240f_lispindent(typval_T *argvars UNUSED, typval_T *rettv)
2241{
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002242 pos_T pos;
2243 linenr_T lnum;
2244
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02002245 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
2246 return;
2247
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002248 pos = curwin->w_cursor;
2249 lnum = tv_get_lnum(argvars);
2250 if (lnum >= 1 && lnum <= curbuf->b_ml.ml_line_count)
2251 {
2252 curwin->w_cursor.lnum = lnum;
2253 rettv->vval.v_number = get_lisp_indent();
2254 curwin->w_cursor = pos;
2255 }
Bram Moolenaar8dac2ac2021-12-27 20:57:06 +00002256 else if (in_vim9script())
2257 semsg(_(e_invalid_line_number_nr), lnum);
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002258 else
Bram Moolenaar14c01f82019-10-09 22:53:08 +02002259 rettv->vval.v_number = -1;
2260}
2261#endif