summaryrefslogtreecommitdiffstats
path: root/runtime/indent/thrift.vim
blob: e0860e12a2bdb6865faf6f844dd0173b2fcb2d6b (plain)
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
" Vim indent file
" Language: Apache Thrift
" Maintainer: Yinzuo Jiang <jiangyinzuo@foxmail.com>
" Last Change: 2024/07/29

" Only load this indent file when no other was loaded.
if exists("b:did_indent")
  finish
endif
let b:did_indent = 1

setlocal cindent
setlocal indentexpr=GetThriftIndent()

let b:undo_indent = "set cindent< indentexpr<"

" Only define the function once.
if exists("*GetThriftIndent")
  finish
endif

let s:keepcpo= &cpo
set cpo&vim

function! SkipThriftBlanksAndComments(startline)
  let lnum = a:startline
  while lnum > 1
    let lnum = prevnonblank(lnum)
    if getline(lnum) =~ '\*/\s*$'
      while getline(lnum) !~ '/\*' && lnum > 1
        let lnum = lnum - 1
      endwhile
      if getline(lnum) =~ '^\s*/\*'
        let lnum = lnum - 1
      else
        break
      endif
    elseif getline(lnum) =~ '^\s*\(//\|#\)'
      let lnum = lnum - 1
    else
      break
    endif
  endwhile
  return lnum
endfunction

function GetThriftIndent()
  " Thrift is just like C; use the built-in C indenting and then correct a few
  " specific cases.
  let theIndent = cindent(v:lnum)

  " If we're in the middle of a comment then just trust cindent
  if getline(v:lnum) =~ '^\s*\*'
    return theIndent
  endif

  let line = substitute(getline(v:lnum), '\(//\|#\).*$', '', '')
  let previousNum = SkipThriftBlanksAndComments(v:lnum - 1)
  let previous = substitute(getline(previousNum), '\(//\|#\).*$', '', '')

  let l:indent = indent(previousNum)
  if previous =~ "{" && previous !~ "}"
    let l:indent += shiftwidth()
  endif
  if line =~ "}" && line !~ "{"
    let l:indent -= shiftwidth()
  endif
  return l:indent
endfunction

let &cpo = s:keepcpo
unlet s:keepcpo

" vim: sw=2 sts=2 et