drop user@ syntax
[mirror/dsa-puppet.git] / tools / vim / ftplugin / puppet.vim
1 " Vim filetype plugin
2 " Language:     Puppet
3 " Maintainer:   Todd Zullinger <tmz@pobox.com>
4 " Last Change:  2009 Aug 19
5 " vim: set sw=4 sts=4:
6
7 if exists("b:did_ftplugin")
8     finish
9 endif
10 let b:did_ftplugin = 1
11
12 if !exists("no_plugin_maps") && !exists("no_puppet_maps")
13     if !hasmapto("<Plug>AlignRange")
14         map <buffer> <LocalLeader>= <Plug>AlignRange
15     endif
16 endif
17
18 noremap <buffer> <unique> <script> <Plug>AlignArrows :call <SID>AlignArrows()<CR>
19 noremap <buffer> <unique> <script> <Plug>AlignRange :call <SID>AlignRange()<CR>
20
21 iabbrev => =><C-R>=<SID>AlignArrows('=>')<CR>
22 iabbrev +> +><C-R>=<SID>AlignArrows('+>')<CR>
23
24 if exists('*s:AlignArrows')
25     finish
26 endif
27
28 let s:arrow_re = '[=+]>'
29 let s:selector_re = '[=+]>\s*\$.*\s*?\s*{\s*$'
30
31 function! s:AlignArrows(op)
32     let cursor_pos = getpos('.')
33     let lnum = line('.')
34     let line = getline(lnum)
35     if line !~ s:arrow_re
36         return
37     endif
38     let pos = stridx(line, a:op)
39     let start = lnum
40     let end = lnum
41     let pnum = lnum - 1
42     while 1
43         let pline = getline(pnum)
44         if pline !~ s:arrow_re || pline =~ s:selector_re
45             break
46         endif
47         let start = pnum
48         let pnum -= 1
49     endwhile
50     let cnum = end
51     while 1
52         let cline = getline(cnum)
53         if cline !~ s:arrow_re ||
54                 \ (indent(cnum) != indent(cnum+1) && getline(cnum+1) !~ '\s*}')
55             break
56         endif
57         let end = cnum
58         let cnum += 1
59     endwhile
60     call s:AlignSection(start, end)
61     let cursor_pos[2] = stridx(getline('.'), a:op) + strlen(a:op) + 1
62     call setpos('.', cursor_pos)
63     return ''
64 endfunction
65
66 function! s:AlignRange() range
67     call s:AlignSection(a:firstline, a:lastline)
68 endfunction
69
70 " AlignSection and AlignLine are from the vim wiki:
71 " http://vim.wikia.com/wiki/Regex-based_text_alignment
72 function! s:AlignSection(start, end)
73     let extra = 1
74     let sep = s:arrow_re
75     let maxpos = 0
76     let section = getline(a:start, a:end)
77     for line in section
78         let pos = match(line, ' *'.sep)
79         if maxpos < pos
80             let maxpos = pos
81         endif
82     endfor
83     call map(section, 's:AlignLine(v:val, sep, maxpos, extra)')
84     call setline(a:start, section)
85 endfunction
86
87 function! s:AlignLine(line, sep, maxpos, extra)
88     let m = matchlist(a:line, '\(.\{-}\) \{-}\('.a:sep.'.*\)')
89     if empty(m)
90         return a:line
91     endif
92     let spaces = repeat(' ', a:maxpos - strlen(m[1]) + a:extra)
93     return m[1] . spaces . m[2]
94 endfunction