vim-easymotion/plugin/EasyMotion.vim

404 lines
10 KiB
VimL
Raw Normal View History

2011-03-28 04:18:00 -04:00
" EasyMotion - Vim motions on speed!
2011-03-27 18:08:06 -04:00
"
2011-03-29 08:10:00 -04:00
" Author: Kim Silkebækken <kim.silkebaekken+vim@gmail.com>
" Source repository: https://github.com/Lokaltog/vim-easymotion
2011-03-27 18:08:06 -04:00
2011-03-30 08:18:39 -04:00
" Script initialization {{{
if exists('g:EasyMotion_loaded') || &compatible || version < 702
2011-03-27 18:08:06 -04:00
finish
endif
let g:EasyMotion_loaded = 1
" }}}
" Default configuration {{{
function! s:InitOption(option, default) " {{{
if ! exists('g:EasyMotion_' . a:option)
exec 'let g:EasyMotion_' . a:option . ' = ' . string(a:default)
endif
endfunction " }}}
function! s:InitHL(group, gui, cterm256, cterm) " {{{
if ! hlexists(a:group)
let guihl = printf('guibg=%s guifg=#%s gui=%s', a:gui[0], a:gui[1], a:gui[2])
let ctermhl = &t_Co == 256
\ ? printf('ctermbg=%s ctermfg=%s cterm=%s', a:cterm256[0], a:cterm256[1], a:cterm256[2])
\ : printf('ctermbg=%s ctermfg=%s cterm=%s', a:cterm[0], a:cterm[1], a:cterm[2])
execute printf('hi %s %s %s', a:group, guihl, ctermhl)
endif
endfunction " }}}
call s:InitOption('keys', 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ')
call s:InitOption('target_hl', 'EasyMotionTarget')
call s:InitOption('shade_hl', 'EasyMotionShade')
call s:InitOption('do_shade', 1)
call s:InitOption('do_mapping', 1)
2011-03-28 16:00:52 -04:00
call s:InitHL(g:EasyMotion_target_hl, ['NONE', 'ff0000', 'bold'], ['NONE', '196', 'bold'], ['NONE', 'red', 'bold'])
call s:InitHL(g:EasyMotion_shade_hl, ['NONE', '585858', 'NONE'], ['NONE', '240', 'NONE'], ['NONE', 'darkgrey', 'NONE'])
2011-03-27 18:08:06 -04:00
" }}}
" Default key mapping {{{
if g:EasyMotion_do_mapping
2011-03-30 12:03:48 -04:00
nnoremap <silent> <Leader>f :call EasyMotionF(0, 0)<CR>
vnoremap <silent> <Leader>f :<C-U>call EasyMotionF(1, 0)<CR>
2011-03-29 09:39:25 -04:00
2011-03-30 12:03:48 -04:00
nnoremap <silent> <Leader>F :call EasyMotionF(0, 1)<CR>
vnoremap <silent> <Leader>F :<C-U>call EasyMotionF(1, 1)<CR>
2011-03-29 09:39:25 -04:00
2011-03-30 12:03:48 -04:00
nnoremap <silent> <Leader>t :call EasyMotionT(0, 0)<CR>
vnoremap <silent> <Leader>t :<C-U>call EasyMotionT(1, 0)<CR>
2011-03-29 09:39:25 -04:00
2011-03-30 12:03:48 -04:00
nnoremap <silent> <Leader>T :call EasyMotionT(0, 1)<CR>
vnoremap <silent> <Leader>T :<C-U>call EasyMotionT(1, 1)<CR>
2011-03-30 12:22:42 -04:00
nnoremap <silent> <Leader>w :call EasyMotionWB(0, 0)<CR>
vnoremap <silent> <Leader>w :<C-U>call EasyMotionWB(1, 0)<CR>
nnoremap <silent> <Leader>b :call EasyMotionWB(0, 1)<CR>
vnoremap <silent> <Leader>b :<C-U>call EasyMotionWB(1, 1)<CR>
2011-03-30 12:03:48 -04:00
nnoremap <silent> <Leader>e :call EasyMotionE(0, 0)<CR>
vnoremap <silent> <Leader>e :<C-U>call EasyMotionE(1, 0)<CR>
2011-03-30 12:03:48 -04:00
nnoremap <silent> <Leader>ge :call EasyMotionE(0, 1)<CR>
vnoremap <silent> <Leader>ge :<C-U>call EasyMotionE(1, 1)<CR>
2011-03-30 12:47:02 -04:00
nnoremap <silent> <Leader>j :call EasyMotionJK(0, 0)<CR>
vnoremap <silent> <Leader>j :<C-U>call EasyMotionJK(1, 0)<CR>
nnoremap <silent> <Leader>k :call EasyMotionJK(0, 1)<CR>
vnoremap <silent> <Leader>k :<C-U>call EasyMotionJK(1, 1)<CR>
2011-03-27 18:08:06 -04:00
endif
" }}}
" Initialize variables {{{
function! s:CreateIndex(chars) " {{{
let index_to_key = {}
let key_to_index = {}
2011-03-27 18:08:06 -04:00
let idx = 0
for char in split(a:chars, '\zs')
let index_to_key[idx] = char
let key_to_index[char] = idx
2011-03-30 07:35:35 -04:00
let idx += 1
endfor
return [index_to_key, key_to_index]
endfunction "}}}
let [s:index_to_key, s:key_to_index] = s:CreateIndex(g:EasyMotion_keys)
let s:var_reset = {}
2011-03-27 18:08:06 -04:00
" }}}
" Motion functions {{{
function! EasyMotionF(visualmode, direction) " {{{
let char = s:GetSearchChar(a:visualmode)
2011-03-27 18:08:06 -04:00
if empty(char)
return
endif
2011-03-27 18:08:06 -04:00
let re = '\C' . escape(char, '.$^~')
2011-03-27 18:08:06 -04:00
call s:EasyMotion(re, a:direction, a:visualmode ? visualmode() : '')
endfunction " }}}
function! EasyMotionT(visualmode, direction) " {{{
let char = s:GetSearchChar(a:visualmode)
if empty(char)
return
endif
2011-03-27 18:08:06 -04:00
if a:direction == 1
let re = '\C' . escape(char, '.$^~') . '\zs.'
else
let re = '\C.' . escape(char, '.$^~')
endif
2011-03-27 18:08:06 -04:00
call s:EasyMotion(re, a:direction, a:visualmode ? visualmode() : '')
endfunction " }}}
2011-03-30 12:22:42 -04:00
function! EasyMotionWB(visualmode, direction) " {{{
call s:EasyMotion('\<.', a:direction, a:visualmode ? visualmode() : '')
endfunction " }}}
2011-03-30 12:03:48 -04:00
function! EasyMotionE(visualmode, direction) " {{{
call s:EasyMotion('.\>', a:direction, a:visualmode ? visualmode() : '')
endfunction " }}}
2011-03-30 12:47:02 -04:00
function! EasyMotionJK(visualmode, direction) " {{{
call s:EasyMotion('\%1v', a:direction, a:visualmode ? visualmode() : '')
endfunction " }}}
2011-03-27 18:08:06 -04:00
" }}}
" Helper functions {{{
function! s:Message(message) " {{{
echo 'EasyMotion: ' . a:message
endfunction " }}}
function! s:Prompt(message) " {{{
echohl Question
echo a:message . ': '
echohl None
endfunction " }}}
function! s:VarReset(var, ...) " {{{
2011-03-29 11:01:47 -04:00
let buf = bufname("")
if a:0 == 0 && has_key(s:var_reset, a:var)
" Reset var to original value
2011-03-29 11:01:47 -04:00
call setbufvar(buf, a:var, s:var_reset[a:var])
elseif a:0 == 1
let new_value = a:0 == 1 ? a:1 : ''
" Store original value
2011-03-29 11:01:47 -04:00
let s:var_reset[a:var] = getbufvar(buf, a:var)
" Set new var value
2011-03-29 11:01:47 -04:00
call setbufvar(buf, a:var, new_value)
endif
endfunction " }}}
2011-03-28 12:04:35 -04:00
function! s:SetLines(lines, key) " {{{
try
" Try to join changes with previous undo block
undojoin
catch
endtry
2011-03-28 12:04:35 -04:00
for [line_num, line] in a:lines
2011-03-28 12:04:35 -04:00
call setline(line_num, line[a:key])
endfor
endfunction " }}}
function! s:GetChar() " {{{
let char = getchar()
if char == 27
" Escape key pressed
redraw
2011-03-29 02:25:44 -04:00
call s:Message('Cancelled')
return ''
endif
return nr2char(char)
endfunction " }}}
2011-03-31 03:27:28 -04:00
function! s:GetSearchChar(visualmode) " {{{
call s:Prompt('Search for character')
let char = s:GetChar()
" Check that we have an input char
if empty(char)
" Restore selection
if ! empty(a:visualmode)
silent exec 'normal! gv'
endif
return ''
endif
return char
2011-03-31 03:27:28 -04:00
endfunction " }}}
2011-03-27 18:08:06 -04:00
" }}}
" Core functions {{{
function! s:PromptUser(groups) "{{{
let single_group = len(a:groups) == 1
let targets_len = single_group ? len(a:groups[0]) : len(a:groups)
" Only one possible match {{{
if single_group && targets_len == 1
redraw
return a:groups[0][0]
endif
" }}}
2011-03-28 03:03:49 -04:00
" Prepare marker lines {{{
let lines = {}
2011-03-27 18:08:06 -04:00
let hl_coords = []
2011-03-28 03:03:49 -04:00
let current_group = 0
2011-03-27 18:08:06 -04:00
for group in a:groups
let element = 0
2011-03-28 03:03:49 -04:00
for [line_num, col_num] in group
" Add original line and marker line
if ! has_key(lines, line_num)
let current_line = getline(line_num)
let lines[line_num] = { 'orig': current_line, 'marker': current_line }
2011-03-28 03:03:49 -04:00
endif
let marker_char = s:index_to_key[single_group ? element : current_group]
if strlen(lines[line_num]['marker']) > 0
" Substitute marker character if line length > 0
let lines[line_num]['marker'] = substitute(lines[line_num]['marker'], '\%' . col_num . 'c.', marker_char, '')
else
" Set the line to the marker character if the line is empty
let lines[line_num]['marker'] = marker_char
endif
2011-03-27 18:08:06 -04:00
" Add highlighting coordinates
2011-03-28 03:03:49 -04:00
call add(hl_coords, '\%' . line_num . 'l\%' . col_num . 'c')
let element += 1
2011-03-27 18:08:06 -04:00
endfor
let current_group += 1
endfor
2011-03-28 03:04:22 -04:00
2011-03-28 03:03:49 -04:00
let lines_items = items(lines)
2011-03-27 18:08:06 -04:00
" }}}
2011-03-28 16:36:39 -04:00
" Highlight source
let target_hl_id = matchadd(g:EasyMotion_target_hl, join(hl_coords, '\|'), 1)
2011-03-28 03:04:22 -04:00
2011-03-29 11:04:19 -04:00
try
" Set lines with markers
call s:SetLines(lines_items, 'marker')
2011-03-27 18:08:06 -04:00
2011-03-29 11:04:19 -04:00
redraw
2011-03-27 18:08:06 -04:00
2011-03-29 11:04:19 -04:00
" Get target/group character
if single_group
call s:Prompt('Target character')
else
call s:Prompt('Group character')
endif
2011-03-27 18:08:06 -04:00
2011-03-29 11:04:19 -04:00
let char = s:GetChar()
finally
" Restore original lines
call s:SetLines(lines_items, 'orig')
2011-03-27 18:08:06 -04:00
2011-03-29 11:04:19 -04:00
" Un-highlight code
2011-03-30 06:51:28 -04:00
if exists('target_hl_id')
call matchdelete(target_hl_id)
endif
2011-03-27 18:08:06 -04:00
2011-03-29 11:04:19 -04:00
redraw
endtry
2011-03-27 18:08:06 -04:00
" Check that we have an input char
2011-03-29 11:04:19 -04:00
if empty(char)
throw 'Cancelled'
endif
2011-03-28 16:36:39 -04:00
" Check if the input char is valid
2011-03-29 11:04:19 -04:00
if ! has_key(s:key_to_index, char) || s:key_to_index[char] >= targets_len
throw 'Invalid target'
2011-03-28 16:36:39 -04:00
endif
if single_group
" Return target coordinates
2011-03-29 11:04:19 -04:00
return a:groups[0][s:key_to_index[char]]
2011-03-28 16:36:39 -04:00
else
" Prompt for target character
2011-03-29 11:04:19 -04:00
return s:PromptUser([a:groups[s:key_to_index[char]]])
2011-03-28 16:36:39 -04:00
endif
2011-03-27 18:08:06 -04:00
endfunction "}}}
function! s:EasyMotion(regexp, direction, visualmode) " {{{
2011-03-27 18:08:06 -04:00
let orig_pos = [line('.'), col('.')]
let targets = []
try
" Reset properties
2011-03-29 02:25:44 -04:00
call s:VarReset('&scrolloff', 0)
call s:VarReset('&modified', 0)
call s:VarReset('&modifiable', 1)
call s:VarReset('&readonly', 0)
" Find motion targets
2011-03-30 04:47:11 -04:00
let search_direction = (a:direction == 1 ? 'b' : '')
let search_stopline = line(a:direction == 1 ? 'w0' : 'w$')
2011-03-30 04:47:11 -04:00
while 1
let pos = searchpos(a:regexp, search_direction, search_stopline)
" Reached end of search range
if pos == [0, 0]
break
endif
" Skip folded lines
if foldclosed(pos[0]) != -1
continue
endif
2011-03-27 18:08:06 -04:00
call add(targets, pos)
endwhile
2011-03-27 18:08:06 -04:00
let targets_len = len(targets)
if targets_len == 0
throw 'No matches'
2011-03-27 18:22:40 -04:00
endif
2011-03-30 06:48:12 -04:00
let groups_len = len(s:index_to_key)
2011-03-28 04:14:51 -04:00
" Split targets into key groups {{{
let groups = []
let i = 0
while i < targets_len
call add(groups, targets[i : i + groups_len - 1])
2011-03-27 18:08:06 -04:00
let i += groups_len
endwhile
" }}}
" Too many groups; only display the first ones {{{
if len(groups) > groups_len
2011-03-29 02:25:44 -04:00
call s:Message('Only displaying the first matches')
2011-03-27 18:08:06 -04:00
let groups = groups[0 : groups_len - 1]
endif
" }}}
2011-03-27 18:08:06 -04:00
" Shade inactive source
2011-03-29 02:48:23 -04:00
if g:EasyMotion_do_shade
let shade_hl_pos = '\%' . orig_pos[0] . 'l\%'. orig_pos[1] .'c'
2011-03-27 18:08:06 -04:00
if a:direction == 1
" Backward
let shade_hl_re = '\%'. line('w0') .'l\_.*' . shade_hl_pos
else
" Forward
let shade_hl_re = shade_hl_pos . '\_.*\%'. line('w$') .'l'
endif
2011-03-27 18:08:06 -04:00
let shade_hl_id = matchadd(g:EasyMotion_shade_hl, shade_hl_re, 0)
2011-03-27 18:08:06 -04:00
endif
" Prompt user for target group/character
2011-03-29 02:25:44 -04:00
let coords = s:PromptUser(groups)
2011-03-27 18:08:06 -04:00
2011-03-28 16:36:39 -04:00
if ! empty(a:visualmode)
" Update selection
2011-03-30 04:32:42 -04:00
call cursor(orig_pos[0], orig_pos[1])
2011-03-27 18:08:06 -04:00
2011-03-29 11:03:46 -04:00
exec 'normal! ' . a:visualmode
2011-03-28 04:14:51 -04:00
endif
2011-03-28 16:36:39 -04:00
2011-03-29 11:03:46 -04:00
" Update cursor position
2011-03-30 04:32:42 -04:00
call cursor(coords[0], coords[1])
2011-03-29 11:03:46 -04:00
2011-03-29 02:25:44 -04:00
call s:Message('Jumping to [' . coords[0] . ', ' . coords[1] . ']')
2011-03-29 11:04:19 -04:00
catch
redraw
2011-03-28 04:14:51 -04:00
" Show exception message
2011-03-29 02:25:44 -04:00
call s:Message(v:exception)
" Restore cursor position/selection
if ! empty(a:visualmode)
silent exec 'normal! gv'
2011-03-28 04:14:51 -04:00
else
2011-03-30 04:32:42 -04:00
call cursor(orig_pos[0], orig_pos[1])
2011-03-28 04:14:51 -04:00
endif
finally
" Restore properties
2011-03-29 02:25:44 -04:00
call s:VarReset('&scrolloff')
call s:VarReset('&modified')
call s:VarReset('&modifiable')
call s:VarReset('&readonly')
2011-03-29 11:04:19 -04:00
" Remove shading
if g:EasyMotion_do_shade && exists('shade_hl_id')
call matchdelete(shade_hl_id)
endif
endtry
2011-03-27 18:08:06 -04:00
endfunction " }}}
" }}}