alex/vim-argwrap
1
Fork 0
Code Releases 1 Activity

Wrapping

Browse Source
This commit is contained in:
Alex Yatskov 2014-12-01 18:21:05 +09:00
parent 39da034ec5
commit a98f27847f
2 changed files with 24 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sample.txt
View File

@ -1,4 +1,4 @@
Foo(a, [b, z], c) a = Foo(a, [b, z], "hello, world", c)
Foo( Foo(
a, a,

38
wrap.vim
View File

@ -21,11 +21,6 @@
function! FindRange() function! FindRange()
let [l:lineStart, l:colStart] = searchpairpos("(", "", ")", "Wnb") let [l:lineStart, l:colStart] = searchpairpos("(", "", ")", "Wnb")
let [l:lineEnd, l:colEnd] = searchpairpos("(", "", ")", "Wn") let [l:lineEnd, l:colEnd] = searchpairpos("(", "", ")", "Wn")
if l:lineStart == l:lineEnd && l:colStart == l:colEnd
return {}
endif
return {"lineStart": l:lineStart, "colStart": l:colStart, "lineEnd": l:lineEnd, "colEnd": l:colEnd} return {"lineStart": l:lineStart, "colStart": l:colStart, "lineEnd": l:lineEnd, "colEnd": l:colEnd}
endfunction endfunction
@ -48,7 +43,7 @@ function! ExtractArgumentText(range)
if l:extractStart < l:extractEnd if l:extractStart < l:extractEnd
let l:extract = l:lineText[l:extractStart : l:extractEnd - 1] let l:extract = l:lineText[l:extractStart : l:extractEnd - 1]
let l:extract = substitute(l:extract, "^\\s\\+", "", "g") let l:extract = substitute(l:extract, "^\\s\\+", "", "g")
let l:extract = substitute(l:extract, ",", ", ", "g") let l:extract = substitute(l:extract, ",$", ", ", "g")
let l:text .= l:extract let l:text .= l:extract
endif endif
endfor endfor
@ -96,23 +91,34 @@ endfunction
function! ExtractContainer(range) function! ExtractContainer(range)
let l:line = getline(a:range.lineStart) let l:line = getline(a:range.lineStart)
let l:indent = match(l:line, "\\S") let l:prefix = l:line[: a:range.colStart - 1]
let l:prefix = l:line[l:indent : a:range.colStart - 1]
let l:line = getline(a:range.lineEnd) let l:line = getline(a:range.lineEnd)
let l:suffix = l:line[a:range.colEnd - 1:] let l:suffix = l:line[a:range.colEnd - 1:]
return {"indent": l:indent, "prefix": l:prefix, "suffix": l:suffix} return {"prefix": l:prefix, "suffix": l:suffix}
endfunction
function! RebuildContainer(range, container, arguments)
let l:line = a:range.lineStart
call setline(l:line, a:container.prefix)
for l:argument in a:arguments
call append(l:line, l:argument)
let l:line += 1
call cursor(l:line, 1)
normal! >>
endfor
call append(l:line, a:container.suffix)
endfunction endfunction
function! Wrap() function! Wrap()
let l:range = FindRange() let l:range = FindRange()
if len(l:range) == 0 let l:argText = ExtractArgumentText(l:range)
return let l:arguments = ExtractArguments(l:argText)
endif let l:container = ExtractContainer(l:range)
let l:text = ExtractArgumentText(l:range) echo l:arguments
let l:args = ExtractArguments(l:text)
let l:cont = ExtractContainer(l:range) call RebuildContainer(l:range, l:container, l:arguments)
echo l:cont
endfunction endfunction