vim-polyglot/indent/elixir.vim

203 lines
5.9 KiB
VimL
Raw Normal View History

if !exists('g:polyglot_disabled') || index(g:polyglot_disabled, 'elixir') == -1
2013-09-12 17:31:56 +02:00
if exists("b:did_indent")
finish
2016-09-11 13:24:17 +02:00
end
2013-09-12 17:31:56 +02:00
let b:did_indent = 1
setlocal nosmartindent
2014-01-24 18:06:22 +01:00
setlocal indentexpr=GetElixirIndent()
2016-09-11 13:24:17 +02:00
setlocal indentkeys+=0),0],0=end,0=else,0=match,0=elsif,0=catch,0=after,0=rescue,0=\|>
2013-09-12 17:31:56 +02:00
if exists("*GetElixirIndent")
finish
2016-09-11 13:24:17 +02:00
end
2013-09-12 17:31:56 +02:00
let s:cpo_save = &cpo
set cpo&vim
2016-09-11 13:24:17 +02:00
let s:no_colon_before = ':\@<!'
let s:no_colon_after = ':\@!'
let s:symbols_end = '\]\|}\|)'
let s:symbols_start = '\[\|{\|('
let s:arrow = '^.*->$'
let s:skip_syntax = '\%(Comment\|String\)$'
let s:block_skip = "synIDattr(synID(line('.'),col('.'),1),'name') =~? '".s:skip_syntax."'"
let s:block_start = '\<\%(do\|fn\)\>'
let s:block_middle = 'else\|match\|elsif\|catch\|after\|rescue'
let s:block_end = 'end'
let s:starts_with_pipeline = '^\s*|>.*$'
2016-05-02 10:42:37 +02:00
let s:ending_with_assignment = '=\s*$'
2016-01-22 09:08:00 +01:00
2016-09-11 13:24:17 +02:00
let s:indent_keywords = '\<'.s:no_colon_before.'\%('.s:block_start.'\|'.s:block_middle.'\)$'.'\|'.s:arrow
2016-01-22 09:08:00 +01:00
let s:deindent_keywords = '^\s*\<\%('.s:block_end.'\|'.s:block_middle.'\)\>'.'\|'.s:arrow
2013-09-12 17:31:56 +02:00
2016-09-11 13:24:17 +02:00
let s:pair_start = '\<\%('.s:no_colon_before.s:block_start.'\)\>'.s:no_colon_after
let s:pair_middle = '^\s*\%('.s:block_middle.'\)\>'.s:no_colon_after.'\zs'
let s:pair_end = '\<\%('.s:no_colon_before.s:block_end.'\)\>\zs'
2016-05-02 10:42:37 +02:00
2016-09-11 13:24:17 +02:00
function! s:is_indentable_syntax()
2014-01-24 18:06:22 +01:00
" TODO: Remove these 2 lines
" I don't know why, but for the test on spec/indent/lists_spec.rb:24.
" Vim is making some mess on parsing the syntax of 'end', it is being
" recognized as 'elixirString' when should be recognized as 'elixirBlock'.
2016-09-11 13:24:17 +02:00
call synID(s:current_line_ref, 1, 1)
2014-01-24 18:06:22 +01:00
" This forces vim to sync the syntax.
syntax sync fromstart
2016-09-11 13:24:17 +02:00
return synIDattr(synID(s:current_line_ref, 1, 1), "name")
\ !~ s:skip_syntax
endfunction
2016-05-02 10:42:37 +02:00
2016-09-11 13:24:17 +02:00
function! s:indent_opened_symbol(ind)
if s:opened_symbol > 0
if s:pending_parenthesis > 0
\ && s:last_line !~ '^\s*def'
\ && s:last_line !~ s:arrow
let b:old_ind = a:ind
return matchend(s:last_line, '(')
" if start symbol is followed by a character, indent based on the
" whitespace after the symbol, otherwise use the default shiftwidth
" Avoid negative indentation index
elseif s:last_line =~ '\('.s:symbols_start.'\).'
let regex = '\('.s:symbols_start.'\)\s*'
let opened_prefix = matchlist(s:last_line, regex)[0]
return a:ind + (s:opened_symbol * strlen(opened_prefix))
2016-05-02 10:42:37 +02:00
else
2016-09-11 13:24:17 +02:00
return a:ind + (s:opened_symbol * &sw)
end
elseif s:opened_symbol < 0
let ind = get(b:, 'old_ind', a:ind + (s:opened_symbol * &sw))
let ind = float2nr(ceil(floor(ind)/&sw)*&sw)
return ind <= 0 ? 0 : ind
else
return a:ind
end
endfunction
2015-05-11 15:05:13 +02:00
2016-09-11 13:24:17 +02:00
function! s:indent_last_line_end_symbol_or_indent_keyword(ind)
if s:last_line =~ '^\s*\('.s:symbols_end.'\)'
\ || s:last_line =~ s:indent_keywords
return a:ind + &sw
else
return a:ind
end
endfunction
2014-03-16 14:48:30 +01:00
2016-09-11 13:24:17 +02:00
function! s:indent_symbols_ending(ind)
if s:current_line =~ '^\s*\('.s:symbols_end.'\)'
return a:ind - &sw
else
return a:ind
end
endfunction
2016-05-02 10:42:37 +02:00
2016-09-11 13:24:17 +02:00
function! s:indent_assignment(ind)
if s:last_line =~ s:ending_with_assignment
let b:old_ind = indent(s:last_line_ref) " FIXME: side effect
return a:ind + &sw
else
return a:ind
end
endfunction
2014-12-09 23:09:20 +01:00
2016-09-11 13:24:17 +02:00
function! s:indent_pipeline(ind)
if s:last_line =~ s:starts_with_pipeline
\ && s:current_line =~ s:starts_with_pipeline
indent(s:last_line_ref)
elseif s:current_line =~ s:starts_with_pipeline
\ && s:last_line =~ '^[^=]\+=.\+$'
let b:old_ind = indent(s:last_line_ref)
2013-11-02 23:27:57 +01:00
" if line starts with pipeline
2014-01-09 11:59:09 +01:00
" and last line is an attribution
" indents pipeline in same level as attribution
2016-09-11 13:24:17 +02:00
return match(s:last_line, '=\s*\zs[^ ]')
else
return a:ind
end
endfunction
function! s:indent_after_pipeline(ind)
if s:last_line =~ s:starts_with_pipeline
if empty(substitute(s:current_line, ' ', '', 'g'))
\ || s:current_line =~ s:starts_with_pipeline
return indent(s:last_line_ref)
elseif s:last_line !~ s:indent_keywords
return b:old_ind
else
return a:ind
end
else
return a:ind
end
endfunction
function! s:deindent_keyword(ind)
if s:current_line =~ s:deindent_keywords
let bslnum = searchpair(
\ s:pair_start,
\ s:pair_middle,
\ s:pair_end,
\ 'nbW',
\ s:block_skip
\ )
return indent(bslnum)
else
return a:ind
end
endfunction
2014-01-09 11:59:09 +01:00
2016-09-11 13:24:17 +02:00
function! s:indent_arrow(ind)
if s:current_line =~ s:arrow
2014-01-09 11:59:09 +01:00
" indent case statements '->'
2016-09-11 13:24:17 +02:00
return a:ind + &sw
else
return a:ind
end
endfunction
2013-09-12 17:31:56 +02:00
2016-09-11 13:24:17 +02:00
function! GetElixirIndent()
let s:current_line_ref = v:lnum
let s:last_line_ref = prevnonblank(s:current_line_ref - 1)
let s:current_line = getline(s:current_line_ref)
let s:last_line = getline(s:last_line_ref)
let s:pending_parenthesis = 0
let s:opened_symbol = 0
if s:last_line !~ s:arrow
let splitted_line = split(s:last_line, '\zs')
let s:pending_parenthesis =
\ + count(splitted_line, '(') - count(splitted_line, ')')
let s:opened_symbol =
\ + s:pending_parenthesis
\ + count(splitted_line, '[') - count(splitted_line, ']')
\ + count(splitted_line, '{') - count(splitted_line, '}')
end
if s:last_line_ref == 0
" At the start of the file use zero indent.
return 0
elseif !s:is_indentable_syntax()
" Current syntax is not indentable, keep last line indentation
return indent(s:last_line_ref)
else
let ind = indent(s:last_line_ref)
let ind = s:indent_opened_symbol(ind)
let ind = s:indent_symbols_ending(ind)
let ind = s:indent_pipeline(ind)
let ind = s:indent_after_pipeline(ind)
let ind = s:indent_assignment(ind)
let ind = s:indent_last_line_end_symbol_or_indent_keyword(ind)
let ind = s:deindent_keyword(ind)
let ind = s:indent_arrow(ind)
return ind
end
2013-09-12 17:31:56 +02:00
endfunction
let &cpo = s:cpo_save
unlet s:cpo_save
endif