summaryrefslogtreecommitdiffstats
path: root/indent/kotlin.vim
blob: 20bc1ee9ab6bb868c6bc962562e1315e96172d38 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
if !exists('g:polyglot_disabled') || index(g:polyglot_disabled, 'kotlin') == -1
  
" Vim indent file
" Language: Kotlin
" Maintainer: Alexander Udalov
" Latest Revision: 27 June 2015

if exists("b:did_indent")
    finish
endif
let b:did_indent = 1

setlocal indentexpr=GetKotlinIndent()
setlocal indentkeys=0},0),!^F,o,O,e,<CR>
setlocal autoindent " TODO ?

" TODO teach it to count bracket balance, etc.
function! GetKotlinIndent()
    if v:lnum == 0
        return 0
    endif

    let prev_num = prevnonblank(v:lnum - 1)
    let prev = getline(prev_num)
    let prev_indent = indent(prev_num)
    let cur = getline(v:lnum)

    let prev_open_paren = prev =~ '^.*(\s*$'
    let cur_close_paren = cur =~ '^\s*).*$'

    if prev_open_paren && !cur_close_paren
        return prev_indent + 2 * &shiftwidth
    endif

    if cur_close_paren && !prev_open_paren
        return prev_indent - 2 * &shiftwidth
    endif


    let prev_open_brace = prev =~ '^.*\({\|->\)\s*$'
    let cur_close_brace = cur =~ '^\s*}.*$'

    if prev_open_brace && !cur_close_brace
        return prev_indent + &shiftwidth
    endif

    if cur_close_brace && !prev_open_brace
        return prev_indent - &shiftwidth
    endif

    return prev_indent
endfunction

endif