runtime/indent/lua.vim

   1 " Vim indent file
   2 " Language:     Lua script
   3 " Maintainer:   Marcus Aurelius Farias <marcus.cf 'at' bol.com.br>
   4 " First Author: Max Ischenko <mfi 'at' ukr.net>
   5 " Last Change:  2005 Jun 23
   6
   7 " Only load this indent file when no other was loaded.
   8 if exists("b:did_indent")
   9   finish
  10 endif
  11 let b:did_indent = 1
  12
  13 setlocal indentexpr=GetLuaIndent()
  14
  15 " To make Vim call GetLuaIndent() when it finds '\s*end' or '\s*until'
  16 " on the current line ('else' is default and includes 'elseif').
  17 setlocal indentkeys+=0=end,0=until
  18
  19 setlocal autoindent
  20
  21 " Only define the function once.
  22 if exists("*GetLuaIndent")
  23   finish
  24 endif
  25
  26 function! GetLuaIndent()
  27   " Find a non-blank line above the current line.
  28   let lnum = prevnonblank(v:lnum - 1)
  29
  30   " Hit the start of the file, use zero indent.
  31   if lnum == 0
  32     return 0
  33   endif
  34
  35   " Add a 'shiftwidth' after lines that start a block:
  36   " 'function', 'if', 'for', 'while', 'repeat', 'else', 'elseif', '{'
  37   let ind = indent(lnum)
  38   let flag = 0
  39   let prevline = getline(lnum)
  40   if prevline =~ '^\s*\%(if\>\|for\>\|while\>\|repeat\>\|else\>\|elseif\>\|do\>\|then\>\)'
  41         \ || prevline =~ '{\s*$' || prevline =~ '\<function\>\s*\%(\k\|[.:]\)\{-}\s*('
  42     let ind = ind + &shiftwidth
  43     let flag = 1
  44   endif
  45
  46   " Subtract a 'shiftwidth' after lines ending with
  47   " 'end' when they begin with 'while', 'if', 'for', etc. too.
  48   if flag == 1 && prevline =~ '\<end\>\|\<until\>'
  49     let ind = ind - &shiftwidth
  50   endif
  51
  52   " Subtract a 'shiftwidth' on end, else (and elseif), until and '}'
  53   " This is the part that requires 'indentkeys'.
  54   if getline(v:lnum) =~ '^\s*\%(end\|else\|until\|}\)'
  55     let ind = ind - &shiftwidth
  56   endif
  57
  58   return ind
  59 endfunction