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
"=============================================================================
" File    : autoload/unite/sources/outline/defaults/lua.vim
" Author  : meryngii <meryngii+git@gmail.com>
" Updated : 2013-03-31
"
" Licensed under the MIT license:
" http://www.opensource.org/licenses/mit-license.php
"
"=============================================================================

function! unite#sources#outline#defaults#lua#outline_info() abort
  return s:outline_info
endfunction

let s:Util = unite#sources#outline#import('Util')

"-----------------------------------------------------------------------------
" Outline Info

let s:outline_info = {
      \ 'heading'  : '^\s*\%(local\s\+\)\=function\s\+\h',
      \ 'skip'     : {
      \   'block' : ['--\[\[', '--\]\]'],
      \ },
      \ 'highlight_rules': [
      \     { 'name'     : 'comment',
      \       'pattern'  : '/--.*/' },
      \     { 'name'     : 'function',
      \       'pattern'  : '/\h\w*/' },
      \     { 'name'     : '_after_colon',
      \       'pattern'  : '/ : \h\w*/',
      \       'highlight': unite#sources#outline#get_highlight('normal') },
      \ ]
      \}

function! s:outline_info.create_heading(which, heading_line, matched_line, context) abort
    let h_lnum = a:context.heading_lnum
    let level = s:Util.get_indent_level(a:context, h_lnum)

    let heading = {
        \ 'word' : a:heading_line,
        \ 'level': level,
        \ 'type' : 'generic',
        \ }

    let suffix = ''
    if heading.word =~ '\<local\>'
        let suffix = ' : local'
    end

    let heading.word = substitute(heading.word, '^\s*\%(local\s\+\)\=function\s\+', '', '') . suffix

    return heading
endfunction