TidyLuaHeadStatements.vim

" 清理 Lua 文件头多余的 local 赋值语句 TidyLuaHeadStatements {

fun! IsLuaLocalAssignStatement()
    return getline(".") =~ '^local\s\+\S\+\s*='
endfun

fun! IsNecessaryLuaAssignment()
    let v1 = line('.')
    silent! exe "norm! #"
    let v2 = line('.')
    silent! exe "norm! N"
    return v1 != v2
endfun

fun! TidyLuaHeadStatements()
    if getline('.') !~ '^\s*$'
        if !IsLuaLocalAssignStatement()
            echo "not lua local assignment statement"
            return
        endif 

        exec "norm! 0w"

        if !IsNecessaryLuaAssignment()
            exec "norm! dd"
        endif
    endif

    if line('.') <= 1
        echo "first line reached"
        return
    endif

    exec "norm! k"
    call TidyLuaHeadStatements()
endfun

command! TidyLuaHeadStatements call TidyLuaHeadStatements()

" }