1
0
mirror of https://github.com/gryf/snipmate.vim.git synced 2025-12-29 01:42:36 +01:00

refactored again; s:BuildTabStops() may actually be somewhat comprehensible now

This commit is contained in:
Michael Sanders
2009-03-03 21:21:54 -05:00
parent 392226a847
commit 6c978f274d
2 changed files with 107 additions and 107 deletions

View File

@@ -32,3 +32,4 @@ fun s:GetSnippets()
endif endif
endfor endfor
endf endf
" vim:noet:sw=4:ts=4:ft=vim

View File

@@ -26,7 +26,7 @@ fun! Filename(...)
return !a:0 || a:1 == '' ? filename : substitute(a:1, '$1', filename, 'g') return !a:0 || a:1 == '' ? filename : substitute(a:1, '$1', filename, 'g')
endf endf
" escapes special characters in snippet triggers " Escapes special characters in snippet triggers
fun s:Hash(text) fun s:Hash(text)
return substitute(a:text, '\W', '\="_".char2nr(submatch(0))."_"', 'g') return substitute(a:text, '\W', '\="_".char2nr(submatch(0))."_"', 'g')
endf endf
@@ -44,7 +44,7 @@ fun s:MakeSnippet(text, ft, multisnip)
let var = 's:snippets' let var = 's:snippets'
endif endif
if !has_key({var}, a:ft) | let {var}[a:ft] = {} | endif if !has_key({var}, a:ft) | let {var}[a:ft] = {} | endif
let end = strpart(a:text, space+1) let end = strpart(a:text, space + 1)
if end == '' || space == '' || (a:multisnip && name == '') if end == '' || space == '' || (a:multisnip && name == '')
echom 'Error in snipMate.vim: Snippet '.a:text.' is undefined.' echom 'Error in snipMate.vim: Snippet '.a:text.' is undefined.'
@@ -59,25 +59,24 @@ fun s:MakeSnippet(text, ft, multisnip)
endf endf
fun! ExtractSnips(dir, ft) fun! ExtractSnips(dir, ft)
let s:slash = has('win16') || has('win32') || has('win64') ? '\\' : '/' let slash = has('win16') || has('win32') || has('win64') ? '\\' : '/'
for path in split(globpath(a:dir, '*'), "\n") for path in split(globpath(a:dir, '*'), "\n")
if isdirectory(path) if isdirectory(path)
for snipFile in split(globpath(path, '*.snippet'), "\n") for snipFile in split(globpath(path, '*.snippet'), "\n")
call s:ProcessFile(snipFile, a:ft, call s:ProcessFile(snipFile, a:ft,
\ strpart(path, strridx(path, s:slash)+1)) \ strpart(path, strridx(path, slash)+1, slash))
endfor endfor
else else
call s:ProcessFile(path, a:ft) call s:ProcessFile(path, a:ft, slash)
endif endif
endfor endfor
unl s:slash
let g:did_ft_{a:ft} = 1 let g:did_ft_{a:ft} = 1
endf endf
" Processes a snippet file; optionally add the name of the parent directory " Processes a snippet file; optionally add the name of the parent directory
" for a snippet with multiple matches. " for a snippet with multiple matches.
fun s:ProcessFile(file, ft, ...) fun s:ProcessFile(file, ft, slash, ...)
let keyword = matchstr(a:file, '.*'.s:slash.'\zs.*\ze\.snippet') let keyword = matchstr(a:file, '.*'.a:slash.'\zs.*\ze\.snippet')
if keyword == '' | return | endif if keyword == '' | return | endif
try try
let text = join(readfile(a:file), "\n") let text = join(readfile(a:file), "\n")
@@ -109,7 +108,7 @@ fun s:ChooseSnippet(ft, trigger)
endf endf
fun! TriggerSnippet() fun! TriggerSnippet()
if pumvisible() " update snippet if completion is used, or deal with supertab if pumvisible() " Update snippet if completion is used, or deal with supertab
if exists('s:sid') | return "\<c-n>" | endif if exists('s:sid') | return "\<c-n>" | endif
call feedkeys("\<esc>a", 'n') | call s:UpdateChangedSnip(0) call feedkeys("\<esc>a", 'n') | call s:UpdateChangedSnip(0)
endif endif
@@ -119,18 +118,18 @@ fun! TriggerSnippet()
endif endif
if exists('s:snipPos') && s:endSnip == s:snipPos[s:curPos][1]+s:snipPos[s:curPos][2] if exists('s:snipPos') && s:endSnip == s:snipPos[s:curPos][1]+s:snipPos[s:curPos][2]
return s:JumpTabStop() " don't treat placeholder text as a trigger return s:JumpTabStop() " Don't treat placeholder text as a trigger.
endif endif
let word = matchstr(getline('.'), '\S\+\%'.col('.').'c') let word = matchstr(getline('.'), '\S\+\%'.col('.').'c')
for filetype in split(&ft, '\.') + ['_'] " deal with dotted file-types for filetype in split(&ft, '\.') + ['_'] " Deal with dotted file-types
let trigger = s:GetSnippet(word, filetype) let trigger = s:GetSnippet(word, filetype)
if exists('s:snippet') | break | endif if exists('s:snippet') | break | endif
endfor endfor
" if word is a trigger for a snippet, delete the trigger & expand the snippet " If word is a trigger for a snippet, delete the trigger & expand the snippet.
if exists('s:snippet') if exists('s:snippet')
if s:snippet == '' " if user cancelled a multi snippet, quit if s:snippet == '' " If user cancelled a multi snippet, quit.
return unl s:snippet return unl s:snippet
endif endif
let col = col('.') - len(trigger) let col = col('.') - len(trigger)
@@ -141,6 +140,13 @@ fun! TriggerSnippet()
return exists('s:sid') ? {s:sid}_SuperTab('n') : "\<tab>" return exists('s:sid') ? {s:sid}_SuperTab('n') : "\<tab>"
endf endf
fun s:GetSuperTabSID()
let old = @a
redir @a | exe 'sil fun /SuperTab$' | redir END
let s:sid = matchstr(@a, '<SNR>\d\+\ze_SuperTab(command)')
let @a = old
endf
" Check if word under cursor is snippet trigger; if it isn't, try checking if " Check if word under cursor is snippet trigger; if it isn't, try checking if
" the text after non-word characters is (e.g. check for "foo" in "bar.foo") " the text after non-word characters is (e.g. check for "foo" in "bar.foo")
fun s:GetSnippet(word, ft) fun s:GetSnippet(word, ft)
@@ -159,49 +165,45 @@ fun s:GetSnippet(word, ft)
return origWord return origWord
endf endf
fun s:GetSuperTabSID()
let old = @a
redir @a | exe 'sil fun /SuperTab$' | redir END
let s:sid = matchstr(@a, '<SNR>\d\+\ze_SuperTab(command)')
let @a = old
endf
fun s:ExpandSnippet(col) fun s:ExpandSnippet(col)
let lnum = line('.') | let col = a:col let lnum = line('.') | let col = a:col
let afterCursor = strpart(getline('.'), col-1)
if afterCursor != "\t" && afterCursor != ' '
sil exe 's/\%'.col.'c.*//'
else | let afterCursor = '' | endif
" for some reason the cursor needs to move one right after this
let line = getline(lnum)
if line != '' && col == 1 && afterCursor == '' && &ve !~ 'all\|onemore'
let col += 1
endif
call s:ProcessSnippet() call s:ProcessSnippet()
if s:snippet == '' if s:snippet == ''
return unl s:snippet " avoid an error if the snippet is now empty return unl s:snippet " Avoid an error if the snippet is now empty
endif endif
let snip = split(substitute(s:snippet, '$\d\|${\d.\{-}}', '', 'g'), "\n", 1) let snip = split(substitute(s:snippet, '$\d\|${\d.\{-}}', '', 'g'), "\n", 1)
if afterCursor != '' | let snip[-1] .= afterCursor | endif
call setline(lnum, line.snip[0])
" autoindent snippet according to previous indentation let afterCursor = strpart(getline('.'), col-1)
let indent = matchend(line, '^.\{-}\ze\(\S\|$\)')+1 if afterCursor != "\t" && afterCursor != ' '
call append(lnum, map(snip[1:], "'".strpart(line, 0, indent-1)."'.v:val")) sil exe 's/\%'.col.'c.*//'
let snip[-1] .= afterCursor
if exists('s:snipPos') && stridx(s:snippet, '${1') != -1 else
if exists('s:update') let afterCursor = ''
call s:UpdateSnip(len(snip[-1])-len(afterCursor)) " For some reason the cursor needs to move one right after this
call s:UpdatePlaceholderTabStops() if getline(lnum) != '' && col == 1 && &ve !~ 'all\|onemore'
else let col += 1
call s:UpdateTabStops(len(snip)-1, len(snip[-1])-len(afterCursor))
endif endif
endif endif
let snipLen = s:BuildTabStops(lnum, col-indent, indent) let line = getline(lnum)
call setline(lnum, line.snip[0])
" Autoindent snippet according to previous indentation
let indent = matchend(line, '^.\{-}\ze\(\S\|$\)') + 1
call append(lnum, map(snip[1:], "'".strpart(line, 0, indent - 1)."'.v:val"))
if exists('s:snipPos') && stridx(s:snippet, '${1') != -1
if exists('s:update')
call s:UpdateSnip(len(snip[-1]) - len(afterCursor))
call s:UpdatePlaceholderTabStops()
else
call s:UpdateTabStops(len(snip) - 1, len(snip[-1]) - len(afterCursor))
endif
endif
let snipLen = s:BuildTabStops(lnum, col - indent, indent)
unl s:snippet unl s:snippet
if snipLen if snipLen
@@ -218,17 +220,17 @@ fun s:ExpandSnippet(col)
if s:snipPos[s:curPos][2] != -1 | return s:SelectWord() | endif if s:snipPos[s:curPos][2] != -1 | return s:SelectWord() | endif
else else
if !exists('s:snipLen') | unl s:snipPos | endif if !exists('s:snipLen') | unl s:snipPos | endif
" place cursor at end of snippet if no tab stop is given " Place cursor at end of snippet if no tab stop is given
let newlines = len(snip)-1 let newlines = len(snip) - 1
call cursor(lnum + newlines, indent + len(snip[-1]) - len(afterCursor) call cursor(lnum + newlines, indent + len(snip[-1]) - len(afterCursor)
\ + (newlines ? 0: col-1)) \ + (newlines ? 0: col - 1))
endif endif
return '' return ''
endf endf
fun s:ProcessSnippet() fun s:ProcessSnippet()
" evaluate eval (`...`) expressions " Evaluate eval (`...`) expressions
" Using a loop here instead of a regex fixes a bug with nested "\=" " Using a loop here instead of a regex fixes a bug with nested "\=".
if stridx(s:snippet, '`') != -1 if stridx(s:snippet, '`') != -1
wh match(s:snippet, '`.\{-}`') != -1 wh match(s:snippet, '`.\{-}`') != -1
let s:snippet = substitute(s:snippet, '`.\{-}`', let s:snippet = substitute(s:snippet, '`.\{-}`',
@@ -238,12 +240,12 @@ fun s:ProcessSnippet()
let s:snippet = substitute(s:snippet, "\r", "\n", 'g') let s:snippet = substitute(s:snippet, "\r", "\n", 'g')
endif endif
" place all text after a colon in a tab stop after the tab stop " Place all text after a colon in a tab stop after the tab stop
" (e.g. "${#:foo}" becomes "${:foo}foo") " (e.g. "${#:foo}" becomes "${:foo}foo")
" this helps tell the position of the tab stops later. " This helps tell the position of the tab stops later.
let s:snippet = substitute(s:snippet, '${\d:\(.\{-}\)}', '&\1', 'g') let s:snippet = substitute(s:snippet, '${\d:\(.\{-}\)}', '&\1', 'g')
" update the s:snippet so that all the $# become " Update the s:snippet so that all the $# become
" the text after the colon in their associated ${#} " the text after the colon in their associated ${#}
" (e.g. "${1:foo}" turns all "$1"'s into "foo") " (e.g. "${1:foo}" turns all "$1"'s into "foo")
let i = 1 let i = 1
@@ -255,7 +257,7 @@ fun s:ProcessSnippet()
let i += 1 let i += 1
endw endw
if &et " expand tabs to spaces if 'expandtab' is set if &et " Expand tabs to spaces if 'expandtab' is set.
let s:snippet = substitute(s:snippet, '\t', let s:snippet = substitute(s:snippet, '\t',
\ repeat(' ', &sts ? &sts : &sw), 'g') \ repeat(' ', &sts ? &sts : &sw), 'g')
endif endif
@@ -271,55 +273,52 @@ fun s:Count(haystack, needle)
return counter return counter
endf endf
" Sorry, this next function is a bit convoluted... " This function builds a list of a list of each tab stop in the
" This function builds a list of a list of each tab stop in the snippet " snippet containing:
" containing: " 1.) The tab stop's line number.
" 1.) The number of the current line plus the number of "\n"s (line " 2.) The tab stop's column number
" breaks) before the tab stop " (by getting the length of the string between the last "\n" and the
" 2.) The current column plus the position of the next "${#}" on " tab stop).
" the line by getting the length of the string between the last "\n"
" and the "${#}" tab stop,
" 3.) The length of the text after the colon for the current tab stop " 3.) The length of the text after the colon for the current tab stop
" (e.g. "${#:foo}" would returnrn 3). If there is no text, -1 is returnrned. " (e.g. "${1:foo}" would return 3). If there is no text, -1 is returned.
" 4.) If the "${#:}" construct is given, the fourth part of the list " 4.) If the "${#:}" construct is given, another list containing all
" is another list containing all the matches of "$#", to be replaced " the matches of "$#", to be replaced with the placeholder. This list is
" with the variable. This list is composed the same way as the parent: " composed the same way as the parent; the first item is the line number,
" the first part is the number of "\n"s before the tab stop, and " and the second is the column.
" second is the position (column) of the "$#" tab stop on the line.
" If there are none of these tab stop, an empty list ([]) is returnrned
fun s:BuildTabStops(lnum, col, indent) fun s:BuildTabStops(lnum, col, indent)
let snipPos = [] | let i = 1 let snipPos = []
" temporarily delete placeholders let i = 1
let cut_snip = substitute(s:snippet, '$\d', '', 'g')
wh stridx(s:snippet, '${'.i) != -1 wh stridx(s:snippet, '${'.i) != -1
let snipPos += [[a:lnum+s:Count(matchstr(cut_snip, '^.*\ze${'.i), "\n"), let beforeTabStop = matchstr(s:snippet, '^.*\ze${'.i)
\ a:indent+len(matchstr(substitute(cut_snip, '${'.i.'\@!\d.\{-}}', '', 'g'), let withoutOthers = substitute(s:snippet, '$\d\|${'.i.'\@!\d.\{-}}', '', 'g')
\ "^.*\\(\n\\|^\\)\\zs.*\\ze${".i.'.\{-}}')), -1]] let snipPos += [[a:lnum + s:Count(beforeTabStop, "\n"),
\ a:indent + len(matchstr(withoutOthers,
\ "^.*\\(\n\\|^\\)\\zs.*\\ze${".i)), -1]]
if snipPos[i-1][0] == a:lnum if snipPos[i-1][0] == a:lnum
let snipPos[i-1][1] += a:col let snipPos[i-1][1] += a:col
endif endif
" get all $# matches in another list, if ${#:name} is given " Get all $# matches in another list, if ${#:name} is given
if stridx(cut_snip, '${'.i.':') != -1 if stridx(s:snippet, '${'.i.':') != -1
let j = i-1 let j = i-1
let snipPos[j][2] = len(matchstr(cut_snip, '${'.i.':\zs.\{-}\ze}')) let snipPos[j][2] = len(matchstr(s:snippet, '${'.i.':\zs.\{-}\ze}'))
let snipPos[j] += [[]] let snipPos[j] += [[]]
" temporarily delete all other tab stops/placeholders let withoutOthers = substitute(s:snippet, '${\d.\{-}}\|$'.i.'\@!\d', '', 'g')
let tempstr = substitute(s:snippet, '$'.i.'\@!\d\|${\d.\{-}}', '', 'g') wh stridx(withoutOthers, '$'.i) != -1
wh stridx(tempstr, '$'.i) != -1 let beforeMark = matchstr(withoutOthers, '^.\{-}\ze$'.i)
let beforeMark = matchstr(tempstr, '^.\{-}\ze$'.i) let linecount = a:lnum + s:Count(beforeMark, "\n")
let linecount = a:lnum+s:Count(beforeMark, "\n")
let snipPos[j][3] += [[linecount, let snipPos[j][3] += [[linecount,
\ a:indent+(linecount > a:lnum ? \ a:indent + (linecount > a:lnum
\ len(matchstr(beforeMark, "^.*\n\\zs.*")) \ ? len(matchstr(beforeMark, "^.*\n\\zs.*"))
\ : a:col+len(beforeMark))]] \ : a:col + len(beforeMark))]]
let tempstr = substitute(tempstr, '$'.i, '', '') let withoutOthers = substitute(withoutOthers, '$'.i, '', '')
endw endw
endif endif
let i += 1 let i += 1
endw endw
if exists('s:snipPos') " allow nested snippets
let s:snipPos = extend(s:snipPos, snipPos, s:curPos+1) if exists('s:snipPos') " Build a nested snippet
let s:snipPos = extend(s:snipPos, snipPos, s:curPos + 1)
else else
let s:snipPos = snipPos let s:snipPos = snipPos
endif endif
@@ -350,7 +349,7 @@ fun s:JumpTabStop()
endf endf
fun s:UpdatePlaceholderTabStops() fun s:UpdatePlaceholderTabStops()
" update tab stops in snippet if text has been added via "$#", " Update tab stops in snippet if text has been added via "$#",
" e.g. in "${1:foo}bar$1${2}" " e.g. in "${1:foo}bar$1${2}"
if exists('s:origPos') if exists('s:origPos')
let changeLen = s:origWordLen - s:snipPos[s:curPos][2] let changeLen = s:origWordLen - s:snipPos[s:curPos][2]
@@ -359,15 +358,15 @@ fun s:UpdatePlaceholderTabStops()
if changeLen != 0 if changeLen != 0
let lnum = line('.') let lnum = line('.')
let len = len(s:origPos) let len = len(s:origPos)
for pos in s:snipPos[(s:curPos+1):] for pos in s:snipPos[(s:curPos + 1):]
let i = 0 | let j = 0 | let k = 0 let i = 0 | let j = 0 | let k = 0
let endSnip = pos[2]+pos[1]-1 let endSnip = pos[2] + pos[1] - 1
wh i < len && s:origPos[i][0] <= pos[0] wh i < len && s:origPos[i][0] <= pos[0]
if pos[0] == s:origPos[i][0] if pos[0] == s:origPos[i][0]
if pos[1] > s:origPos[i][1] if pos[1] > s:origPos[i][1]
\ || (pos[2] == -1 && pos[1] == s:origPos[i][1]) \ || (pos[2] == -1 && pos[1] == s:origPos[i][1])
let j += 1 let j += 1
elseif s:origPos[i][1] < endSnip " parse variables within placeholders elseif s:origPos[i][1] < endSnip " Parse variables within placeholders
let k += 1 let k += 1
endif endif
endif endif
@@ -408,13 +407,13 @@ fun s:UpdateTabStops(...)
if exists('s:origWordLen') if exists('s:origWordLen')
let changeCol -= s:origWordLen | unl s:origWordLen let changeCol -= s:origWordLen | unl s:origWordLen
endif endif
" there's probably a more efficient way to do this as well... " There's probably a more efficient way to do this as well...
let lnum = s:snipPos[s:curPos][0] let lnum = s:snipPos[s:curPos][0]
let col = s:snipPos[s:curPos][1] let col = s:snipPos[s:curPos][1]
" update the line number of all proceeding tab stops if <cr> has " Update the line number of all proceeding tab stops if <cr> has
" been inserted " been inserted.
if changeLine != 0 if changeLine != 0
for pos in s:snipPos[(s:curPos+1):] for pos in s:snipPos[(s:curPos + 1):]
if pos[0] >= lnum if pos[0] >= lnum
if pos[0] == lnum if pos[0] == lnum
let pos[1] += changeCol let pos[1] += changeCol
@@ -433,9 +432,9 @@ fun s:UpdateTabStops(...)
endif endif
endfor endfor
elseif changeCol != 0 elseif changeCol != 0
" update the column of all proceeding tab stops if text has " Update the column of all proceeding tab stops if text has
" been inserted/deleted in the current line " been inserted/deleted in the current line.
for pos in s:snipPos[(s:curPos+1):] for pos in s:snipPos[(s:curPos + 1):]
if pos[1] >= col && pos[0] == lnum if pos[1] >= col && pos[0] == lnum
let pos[1] += changeCol let pos[1] += changeCol
endif endif
@@ -482,10 +481,10 @@ au CursorMovedI * call s:UpdateChangedSnip(0)
au InsertEnter * call s:UpdateChangedSnip(1) au InsertEnter * call s:UpdateChangedSnip(1)
fun s:UpdateChangedSnip(entering) fun s:UpdateChangedSnip(entering)
if exists('s:update') if exists('s:update')
if !exists('s:origPos') && s:curPos+1 < s:snipLen if !exists('s:origPos') && s:curPos + 1 < s:snipLen
" save the old snippet & word length before it's updated " Save the old snippet & word length before it's updated
" s:startSnip must be saved too, in case text is added " s:startSnip must be saved too, in case text is added
" before the snippet (e.g. in "foo$1${2}bar${1:foo}") " before the snippet (e.g. in "foo$1${2}bar${1:foo}").
let s:origSnipPos = s:startSnip let s:origSnipPos = s:startSnip
let s:origPos = deepcopy(s:snipPos[s:curPos][3]) let s:origPos = deepcopy(s:snipPos[s:curPos][3])
endif endif
@@ -494,12 +493,12 @@ fun s:UpdateChangedSnip(entering)
if s:endSnip != -1 if s:endSnip != -1
let changeLen = col('$') - s:prevLen[1] let changeLen = col('$') - s:prevLen[1]
let s:endSnip += changeLen let s:endSnip += changeLen
else " when being updated the first time, after leaving select mode else " When being updated the first time, after leaving select mode
if a:entering | return | endif if a:entering | return | endif
let s:endSnip = col-1 let s:endSnip = col-1
endif endif
" if the cursor moves outside the snippet, quit it " If the cursor moves outside the snippet, quit it
if line('.') != s:snipPos[s:curPos][0] || col < s:startSnip || if line('.') != s:snipPos[s:curPos][0] || col < s:startSnip ||
\ col-1 > s:endSnip \ col-1 > s:endSnip
unl! s:startSnip s:origWordLen s:origPos s:update unl! s:startSnip s:origWordLen s:origPos s:update
@@ -522,7 +521,7 @@ fun s:UpdateChangedSnip(entering)
let s:endSnip = col let s:endSnip = col
endif endif
" delete snippet if cursor moves out of it in insert mode " Delete snippet if cursor moves out of it in insert mode
if (lnum == s:endSnipLine && (col > s:endSnip || col < s:snipPos[s:curPos][1])) if (lnum == s:endSnipLine && (col > s:endSnip || col < s:snipPos[s:curPos][1]))
\ || lnum > s:endSnipLine || lnum < s:snipPos[s:curPos][0] \ || lnum > s:endSnipLine || lnum < s:snipPos[s:curPos][0]
call s:RemoveSnippet() call s:RemoveSnippet()
@@ -531,7 +530,7 @@ fun s:UpdateChangedSnip(entering)
endf endf
fun s:UpdateSnip(...) fun s:UpdateSnip(...)
" using strpart() here avoids a bug if s:endSnip is negative that would " Using strpart() here avoids a bug if s:endSnip was negative that would
" happen with the getline('.')[(s:startSnip):(s:endSnip)] syntax " happen with the getline('.')[(s:startSnip):(s:endSnip)] syntax
let newWordLen = a:0 ? a:1 : s:endSnip - s:startSnip + 1 let newWordLen = a:0 ? a:1 : s:endSnip - s:startSnip + 1
let newWord = strpart(getline('.'), s:startSnip, newWordLen) let newWord = strpart(getline('.'), s:startSnip, newWordLen)