|
| 1 | +if exists('b:did_indent') |
| 2 | + finish |
| 3 | +endif |
| 4 | +let b:did_indent = 1 |
| 5 | + |
| 6 | +setlocal autoindent |
| 7 | +setlocal indentexpr=GetMarkdownIndent() |
| 8 | +setlocal indentkeys=!^F,o,O,0&,<Space> |
| 9 | + |
| 10 | +function! s:in_mathzone(...) abort |
| 11 | + return !exists('g:loaded_vimtex') || !g:loaded_vimtex |
| 12 | + \ || call('vimtex#syntax#in_mathzone', a:000) |
| 13 | +endfunction |
| 14 | + |
| 15 | +function! s:in_normalzone(...) abort |
| 16 | + return !exists('g:loaded_vimtex') || !g:loaded_vimtex |
| 17 | + \ || !call('vimtex#syntax#in_mathzone', a:000) |
| 18 | +endfunction |
| 19 | + |
| 20 | +function! s:in_codeblock(...) abort |
| 21 | + let l:lnum = get(a:000, 0, v:lnum) |
| 22 | + let l:bufnr = get(a:000, 1, bufnr('%')) |
| 23 | + return luaeval(printf('require("utils.ft.markdown").in_codeblock(%d, %d)' |
| 24 | + \ , l:lnum, l:bufnr)) |
| 25 | +endfunction |
| 26 | + |
| 27 | +function! s:ts_active() abort |
| 28 | + return luaeval('require("utils.treesitter").is_active()') |
| 29 | +endfunction |
| 30 | + |
| 31 | +" Find the first previous non-blank line that matches the given pattern if |
| 32 | +" trimmed, stops on the first line that has a larger indent than its next |
| 33 | +" non-blank line |
| 34 | +" Returns the line number, the index of the match, and the substring that |
| 35 | +" matches the pattern |
| 36 | +function! s:trimmed_prevnonblank_matches(lnum, pattern) abort |
| 37 | + let l:lnum = prevnonblank(a:lnum - 1) |
| 38 | + let l:indent = indent(l:lnum) |
| 39 | + while l:lnum >= 1 |
| 40 | + let l:line = trim(getline(l:lnum)) |
| 41 | + let l:match_idx = match(l:line, a:pattern) |
| 42 | + let l:match = matchstr(l:line, a:pattern) |
| 43 | + if l:match_idx >= 0 |
| 44 | + return [l:lnum, l:match_idx, l:match] |
| 45 | + endif |
| 46 | + let l:lnum = prevnonblank(l:lnum - 1) |
| 47 | + let l:new_indent = indent(l:lnum) |
| 48 | + if l:new_indent <= l:indent |
| 49 | + let l:indent = l:new_indent |
| 50 | + else |
| 51 | + break |
| 52 | + endif |
| 53 | + endwhile |
| 54 | + return [0, -1, ''] |
| 55 | +endfunction |
| 56 | + |
| 57 | +function! GetMarkdownIndent() abort |
| 58 | + let l:line = getline(v:lnum) |
| 59 | + let l:prev_lnum = prevnonblank(v:lnum - 1) |
| 60 | + let l:prev_line = getline(l:prev_lnum) |
| 61 | + let l:sw = shiftwidth() |
| 62 | + let l:default = indent(v:lnum) |
| 63 | + if l:prev_lnum == 0 |
| 64 | + return l:default |
| 65 | + endif |
| 66 | + |
| 67 | + if s:ts_active() && s:in_codeblock() |
| 68 | + return nvim_treesitter#indent() |
| 69 | + endif |
| 70 | + |
| 71 | + if s:in_mathzone() |
| 72 | + " Align to the first previous line that has |
| 73 | + " '='/'>'/'<'/'\le'/'\ge'/'\sim'/'\approx'/'\gg'/'\ll' or variants with |
| 74 | + " '&' at the beginning if the current line starts with one of these |
| 75 | + " patterns |
| 76 | + let align_patterns = |
| 77 | + \ '^\s*\(\(&\s*\)\?\(=\|>\|<\|\\\(le\|ge\|sim\|approx\|gg\|ll\)\>\)\|&\)' |
| 78 | + if l:line =~# align_patterns |
| 79 | + let [l:prev_eq_lnum, l:eq_pos, l:_] = |
| 80 | + \ s:trimmed_prevnonblank_matches(v:lnum, align_patterns) |
| 81 | + if l:prev_eq_lnum > 0 && s:in_mathzone(l:prev_eq_lnum, 1) |
| 82 | + return indent(l:prev_eq_lnum) + l:eq_pos |
| 83 | + endif |
| 84 | + endif |
| 85 | + return l:default |
| 86 | + endif |
| 87 | + |
| 88 | + if s:in_normalzone() |
| 89 | + " Reindent unordered list bullet points |
| 90 | + if l:line =~# '^\s*[-*+]\s\+$' |
| 91 | + return l:default / l:sw * l:sw |
| 92 | + endif |
| 93 | + " Reindent ordered list items |
| 94 | + let l:order = str2nr(matchstr(l:line, '\(^\s*\)\@<=\d\+\(\.\ \)\@=')) |
| 95 | + if l:order > 0 |
| 96 | + let [l:prev_heading_lnum, l:_, l:_] = |
| 97 | + \ s:trimmed_prevnonblank_matches(v:lnum, '^#') |
| 98 | + let [l:prev_item_lnum, l:_, l:prev_item_order] = |
| 99 | + \ s:trimmed_prevnonblank_matches(v:lnum, '^\d\+\.\@=') |
| 100 | + if s:in_normalzone(l:prev_item_lnum, 1) && |
| 101 | + \ l:prev_heading_lnum < l:prev_item_lnum |
| 102 | + return str2nr(l:order) > l:prev_item_order |
| 103 | + \ ? indent(l:prev_item_lnum) |
| 104 | + \ : indent(l:prev_item_lnum) + l:sw |
| 105 | + endif |
| 106 | + endif |
| 107 | + " Only align multi-line list items if adjacent previous line is non-empty |
| 108 | + " --------------------------------------------- |
| 109 | + " 1234. aaa-bbb |
| 110 | + " ccc <- should align with the 'aaa' |
| 111 | + " --------------------------------------------- |
| 112 | + " 5678. ddd |
| 113 | + " |
| 114 | + " eee <- should not align with 'ddd' |
| 115 | + " --------------------------------------------- |
| 116 | + if l:line =~# '^\s*$' && l:prev_lnum == v:lnum - 1 && |
| 117 | + \ s:in_normalzone(l:prev_lnum, 1) |
| 118 | + " Align unordered list multi-line items |
| 119 | + let l:prev_line_char_pos = match(l:prev_line, '\(^\s*[-*+]\s\+\)\@<=\S') |
| 120 | + if l:prev_line_char_pos >= 0 |
| 121 | + return l:prev_line_char_pos |
| 122 | + endif |
| 123 | + " Align ordered list multi-line items |
| 124 | + let l:ordered_list_extra_indent = |
| 125 | + \ match(l:prev_line, '\(^\s*\d\+\.\s*\)\@<=\S') |
| 126 | + if l:ordered_list_extra_indent >= 0 |
| 127 | + return l:ordered_list_extra_indent |
| 128 | + endif |
| 129 | + endif |
| 130 | + endif |
| 131 | + |
| 132 | + return l:default |
| 133 | +endfunction |
0 commit comments