mirror of
https://github.com/sheerun/vim-polyglot.git
synced 2025-11-08 11:33:52 -05:00
80 lines
2.3 KiB
VimL
80 lines
2.3 KiB
VimL
let s:base = expand("<sfile>:h:h")
|
|
let Filter = { _, v -> stridx(v, s:base) == -1 && stridx(v, $VIMRUNTIME) == -1 && v !~ "after" }
|
|
let files = filter(globpath(&rtp, 'autoload/puppet/align.vim', 1, 1), Filter)
|
|
if len(files) > 0
|
|
exec 'source ' . files[0]
|
|
finish
|
|
endif
|
|
if !exists('g:polyglot_disabled') || index(g:polyglot_disabled, 'puppet') == -1
|
|
|
|
function! puppet#align#IndentLevel(lnum)
|
|
return indent(a:lnum) / &shiftwidth
|
|
endfunction
|
|
|
|
function! puppet#align#LinesInBlock(lnum)
|
|
let lines = []
|
|
let indent_level = puppet#align#IndentLevel(a:lnum)
|
|
|
|
let marker = a:lnum - 1
|
|
while marker >= 1
|
|
let line_text = getline(marker)
|
|
let line_indent = puppet#align#IndentLevel(marker)
|
|
|
|
if line_text =~? '\v\S'
|
|
if line_indent < indent_level
|
|
break
|
|
elseif line_indent == indent_level
|
|
call add(lines, marker)
|
|
endif
|
|
endif
|
|
|
|
let marker -= 1
|
|
endwhile
|
|
|
|
let marker = a:lnum
|
|
while marker <= line('$')
|
|
let line_text = getline(marker)
|
|
let line_indent = puppet#align#IndentLevel(marker)
|
|
|
|
if line_text =~? '\v\S'
|
|
if line_indent < indent_level
|
|
break
|
|
elseif line_indent == indent_level
|
|
call add(lines, marker)
|
|
endif
|
|
endif
|
|
|
|
let marker += 1
|
|
endwhile
|
|
|
|
return lines
|
|
endfunction
|
|
|
|
""
|
|
" Format lines with hashrocket (=>)
|
|
" @param a:1 a line where function should search for first hashrocket
|
|
" expression, if param is not given, line with active cursor is used
|
|
function! puppet#align#AlignHashrockets(...) abort
|
|
let l:lnum = get(a:, 1, line('.'))
|
|
let lines_in_block = puppet#align#LinesInBlock(l:lnum)
|
|
let max_left_len = 0
|
|
let indent_str = printf('%' . indent(l:lnum) . 's', '')
|
|
|
|
for line_num in lines_in_block
|
|
let data = matchlist(getline(line_num), '^\s*\(.\{-}\S\)\s*=>\s*\(.*\)$')
|
|
if !empty(data)
|
|
let max_left_len = max([max_left_len, strlen(data[1])])
|
|
endif
|
|
endfor
|
|
|
|
for line_num in lines_in_block
|
|
let data = matchlist(getline(line_num), '^\s*\(.\{-}\S\)\s*=>\s*\(.*\)$')
|
|
if !empty(data)
|
|
let new_line = printf('%s%-' . max_left_len . 's => %s', indent_str, data[1], data[2])
|
|
call setline(line_num, new_line)
|
|
endif
|
|
endfor
|
|
endfunction
|
|
|
|
endif
|