let b:bufnr = bufnr('%')
let b:fenced_block = 0
+ let l:front_matter = 0
let b:header_list = []
let l:header_max_len = 0
+ let g:vim_markdown_toc_autofit = get(g:, "vim_markdown_toc_autofit", 0)
+ let l:vim_markdown_frontmatter = get(g:, "vim_markdown_frontmatter", 0)
for i in range(1, line('$'))
let l:lineraw = getline(i)
let l:l1 = getline(i+1)
elseif b:fenced_block == 1
let b:fenced_block = 0
endif
+ elseif l:vim_markdown_frontmatter == 1
+ if l:front_matter == 1
+ if l:line == '---'
+ let l:front_matter = 0
+ endif
+ elseif i == 1
+ if l:line == '---'
+ let l:front_matter = 1
+ endif
+ endif
endif
- if l:line =~ '^#\+' || l:l1 =~ '^==\+\s*' || l:l1 =~ '^--\+\s*'
+ if l:line =~ '^#\+' || (l:l1 =~ '^=\+\s*$' || l:l1 =~ '^-\+\s*$') && l:line =~ '^\S'
let b:is_header = 1
else
let b:is_header = 0
endif
- if b:is_header == 1 && b:fenced_block == 0
+ if b:is_header == 1 && b:fenced_block == 0 && l:front_matter == 0
" append line to location list
let b:item = {'lnum': i, 'text': l:line, 'valid': 1, 'bufnr': b:bufnr, 'col': 1}
let b:header_list = b:header_list + [b:item]
+ " keep track of the longest header size (heading level + title)
+ let b:total_len = stridx(l:line, ' ') + len(l:line)
+ if b:total_len > l:header_max_len
+ let l:header_max_len = b:total_len
+ endif
endif
endfor
if len(b:header_list) == 0
lopen
elseif l:window_type ==# 'vertical'
vertical lopen
- let &winwidth=(&columns/2)
+ " auto-fit toc window when possible to shrink it
+ if (&columns/2) > l:header_max_len && g:vim_markdown_toc_autofit == 1
+ execute 'vertical resize ' . (l:header_max_len + 1)
+ else
+ execute 'vertical resize ' . (&columns/2)
+ endif
elseif l:window_type ==# 'tab'
tab lopen
else