mirror of
https://github.com/SpaceVim/SpaceVim.git
synced 2025-02-23 19:21:33 +08:00
304 lines
8.8 KiB
VimL
304 lines
8.8 KiB
VimL
"=============================================================================
|
|
" iedit.vim --- iedit mode for SpaceVim
|
|
" Copyright (c) 2016-2017 Shidong Wang & Contributors
|
|
" Author: Shidong Wang < wsdjeg at 163.com >
|
|
" URL: https://spacevim.org
|
|
" License: MIT license
|
|
"=============================================================================
|
|
|
|
let s:stack = []
|
|
let s:index = -1
|
|
let s:cursor_col = -1
|
|
let s:mode = ''
|
|
let s:hi_id = ''
|
|
let s:Operator = ''
|
|
|
|
" prompt
|
|
|
|
let s:symbol_begin = ''
|
|
let s:symbol_cursor = ''
|
|
let s:symbol_end = ''
|
|
|
|
let s:VIMH = SpaceVim#api#import('vim#highlight')
|
|
let s:STRING = SpaceVim#api#import('data#string')
|
|
|
|
let s:cursor_stack = []
|
|
|
|
|
|
function! s:highlight_cursor() abort
|
|
let info = {
|
|
\ 'name' : 'SpaceVimGuideCursor',
|
|
\ 'guibg' : synIDattr(synIDtrans(synID(line('.'), col('.'), 1)), 'guifg'),
|
|
\ 'guifg' : synIDattr(synIDtrans(synID(line('.'), col('.'), 1)), 'guibg'),
|
|
\ 'ctermbg' : synIDattr(synIDtrans(synID(line('.'), col('.'), 1)), 'ctermfg'),
|
|
\ 'ctermfg' : synIDattr(synIDtrans(synID(line('.'), col('.'), 1)), 'ctermbg'),
|
|
\ }
|
|
hi def link SpaceVimGuideCursor Cursor
|
|
call s:VIMH.hi(info)
|
|
for pos in s:stack
|
|
call matchaddpos('Underlined', [pos])
|
|
call matchadd('SpaceVimGuideCursor', '\%' . pos[0] . 'l\%' . (pos[1] + len(s:symbol_begin)) . 'c', 99999)
|
|
endfor
|
|
endfunction
|
|
|
|
function! s:remove_cursor_highlight() abort
|
|
call clearmatches()
|
|
endfunction
|
|
|
|
function! SpaceVim#plugins#iedit#start(...)
|
|
let save_tve = &t_ve
|
|
let save_cl = &l:cursorline
|
|
setlocal nocursorline
|
|
setlocal t_ve=
|
|
let s:mode = 'n'
|
|
let w:spacevim_iedit_mode = s:mode
|
|
let w:spacevim_statusline_mode = 'in'
|
|
let curpos = getcurpos()
|
|
let save_reg_k = @k
|
|
if get(a:000, 0, 0) == 1
|
|
normal! gv"ky
|
|
else
|
|
normal! viw"ky
|
|
endif
|
|
call setpos('.', curpos)
|
|
let symbol = split(@k, "\n")[0]
|
|
let @k = save_reg_k
|
|
echomsg string(a:000)
|
|
echom symbol
|
|
let begin = get(a:000, 1, 1)
|
|
let end = get(a:000, 2, line('$'))
|
|
call s:parse_symbol(begin, end, symbol)
|
|
call s:highlight_cursor()
|
|
redrawstatus!
|
|
while s:mode != ''
|
|
redraw!
|
|
let char = getchar()
|
|
if s:mode ==# 'n' && char == 27
|
|
let s:mode = ''
|
|
else
|
|
call s:handle(s:mode, char)
|
|
endif
|
|
endwhile
|
|
let s:stack = []
|
|
let s:cursor_stack = []
|
|
let s:index = -1
|
|
let s:mode = ''
|
|
let w:spacevim_iedit_mode = s:mode
|
|
let w:spacevim_statusline_mode = 'in'
|
|
let &t_ve = save_tve
|
|
call s:remove_cursor_highlight()
|
|
try
|
|
call matchdelete(s:hi_id)
|
|
catch
|
|
endtry
|
|
let s:hi_id = ''
|
|
let &l:cursorline = save_cl
|
|
endfunction
|
|
|
|
|
|
function! s:handle(mode, char) abort
|
|
if a:mode ==# 'n'
|
|
call s:handle_normal(a:char)
|
|
elseif a:mode ==# 'i'
|
|
call s:handle_insert(a:char)
|
|
endif
|
|
endfunction
|
|
|
|
|
|
let s:toggle_stack = {}
|
|
|
|
function! s:handle_normal(char) abort
|
|
silent! call s:remove_cursor_highlight()
|
|
if a:char ==# 105 " i
|
|
let s:mode = 'i'
|
|
let w:spacevim_iedit_mode = s:mode
|
|
let w:spacevim_statusline_mode = 'ii'
|
|
redrawstatus!
|
|
elseif a:char == 9 " <tab>
|
|
if index(keys(s:toggle_stack), s:index . '') == -1
|
|
call extend(s:toggle_stack, {s:index : s:stack[s:index]})
|
|
call remove(s:stack, s:index)
|
|
else
|
|
call insert(s:stack, s:toggle_stack[s:index] , s:index)
|
|
call remove(s:toggle_stack, s:index)
|
|
endif
|
|
elseif a:char == 97 " a
|
|
let s:mode = 'i'
|
|
let w:spacevim_iedit_mode = s:mode
|
|
let w:spacevim_statusline_mode = 'ii'
|
|
let s:symbol_begin = s:symbol_begin . s:symbol_cursor
|
|
let s:symbol_cursor = matchstr(s:symbol_end, '^.')
|
|
let s:symbol_end = substitute(s:symbol_end, '^.', '', 'g')
|
|
redrawstatus!
|
|
elseif a:char == "\<Left>"
|
|
let s:symbol_end = s:symbol_cursor . s:symbol_end
|
|
let s:symbol_cursor = matchstr(s:symbol_begin, '.$')
|
|
let s:symbol_begin = substitute(s:symbol_begin, '.$', '', 'g')
|
|
elseif a:char == "\<Right>"
|
|
let s:symbol_begin = s:symbol_begin . s:symbol_cursor
|
|
let s:symbol_cursor = matchstr(s:symbol_end, '^.')
|
|
let s:symbol_end = substitute(s:symbol_end, '^.', '', 'g')
|
|
elseif a:char == 48 " 0
|
|
let s:symbol_end = substitute(s:symbol_begin . s:symbol_cursor . s:symbol_end, '^.', '', 'g')
|
|
let s:symbol_cursor = matchstr(s:symbol_begin, '^.')
|
|
let s:symbol_begin = ''
|
|
elseif a:char == 36 " $
|
|
let s:symbol_begin = substitute(s:symbol_begin . s:symbol_cursor . s:symbol_end, '.$', '', 'g')
|
|
let s:symbol_cursor = matchstr(s:symbol_end, '.$')
|
|
let s:symbol_end = ''
|
|
elseif a:char == 68 " D
|
|
let s:symbol_begin = ''
|
|
let s:symbol_cursor = ''
|
|
let s:symbol_end = ''
|
|
call s:replace_symbol(s:symbol_begin . s:symbol_cursor . s:symbol_end)
|
|
elseif a:char == 112 " p
|
|
let s:symbol_begin = @"
|
|
let s:symbol_cursor = ''
|
|
let s:symbol_end = ''
|
|
call s:replace_symbol(s:symbol_begin . s:symbol_cursor . s:symbol_end)
|
|
elseif a:char == 83 " S
|
|
let s:symbol_begin = ''
|
|
let s:symbol_cursor = ''
|
|
let s:symbol_end = ''
|
|
let s:mode = 'i'
|
|
let w:spacevim_iedit_mode = s:mode
|
|
let w:spacevim_statusline_mode = 'ii'
|
|
redrawstatus!
|
|
call s:replace_symbol(s:symbol_begin . s:symbol_cursor . s:symbol_end)
|
|
elseif a:char == 71 " G
|
|
exe s:stack[-1][0]
|
|
elseif a:char == 103 "g
|
|
if s:Operator ==# 'g'
|
|
exe s:stack[0][0]
|
|
let s:Operator = ''
|
|
else
|
|
let s:Operator = 'g'
|
|
call s:timeout()
|
|
endif
|
|
elseif a:char == 110 " n
|
|
if s:index == len(s:stack) - 1
|
|
let s:index = 0
|
|
else
|
|
let s:index += 1
|
|
endif
|
|
call cursor(s:stack[s:index][0], s:stack[s:index][1] + len(s:symbol_begin))
|
|
elseif a:char == 78 " N
|
|
if s:index == 0
|
|
let s:index = len(s:stack) - 1
|
|
else
|
|
let s:index -= 1
|
|
endif
|
|
call cursor(s:stack[s:index][0], s:stack[s:index][1] + len(s:symbol_begin))
|
|
endif
|
|
silent! call s:highlight_cursor()
|
|
endfunction
|
|
|
|
if exists('*timer_start')
|
|
function! s:timeout() abort
|
|
call timer_start(1000, function('s:reset_Operator'))
|
|
endfunction
|
|
else
|
|
function! s:timeout() abort
|
|
endfunction
|
|
endif
|
|
|
|
|
|
function! s:reset_Operator(...) abort
|
|
let s:Operator = ''
|
|
endfunction
|
|
|
|
function! s:handle_insert(char) abort
|
|
silent! call s:remove_cursor_highlight()
|
|
if a:char == 27
|
|
let s:mode = 'n'
|
|
let w:spacevim_iedit_mode = s:mode
|
|
let w:spacevim_statusline_mode = 'in'
|
|
silent! call s:highlight_cursor()
|
|
redraw!
|
|
redrawstatus!
|
|
return
|
|
elseif a:char == 23
|
|
let s:symbol_begin = ''
|
|
elseif a:char == 11
|
|
let s:symbol_cursor = ''
|
|
let s:symbol_end = ''
|
|
elseif a:char == "\<bs>"
|
|
let s:symbol_begin = substitute(s:symbol_begin, '.$', '', 'g')
|
|
elseif a:char == "\<Left>"
|
|
let s:symbol_end = s:symbol_cursor . s:symbol_end
|
|
let s:symbol_cursor = matchstr(s:symbol_begin, '.$')
|
|
let s:symbol_begin = substitute(s:symbol_begin, '.$', '', 'g')
|
|
elseif a:char == "\<Right>"
|
|
let s:symbol_begin = s:symbol_begin . s:symbol_cursor
|
|
let s:symbol_cursor = matchstr(s:symbol_end, '^.')
|
|
let s:symbol_end = substitute(s:symbol_end, '^.', '', 'g')
|
|
else
|
|
let s:symbol_begin .= nr2char(a:char)
|
|
endif
|
|
call s:replace_symbol(s:symbol_begin . s:symbol_cursor . s:symbol_end)
|
|
silent! call s:highlight_cursor()
|
|
endfunction
|
|
|
|
function! s:parse_symbol(begin, end, symbol) abort
|
|
let len = len(a:symbol)
|
|
let cursor = [line('.'), col('.')]
|
|
for l in range(a:begin, a:end)
|
|
let line = getline(l)
|
|
let idx = s:STRING.strAllIndex(line, a:symbol)
|
|
for pos_c in idx
|
|
call add(s:stack, [l, pos_c + 1, len])
|
|
if l == cursor[0] && pos_c + 1 <= cursor[1] && pos_c + 1 + len >= cursor[1]
|
|
let s:index = len(s:stack) - 1
|
|
if pos_c + 1 < cursor[1]
|
|
let s:symbol_begin = line[pos_c : cursor[1] - 2]
|
|
else
|
|
let s:symbol_begin = ''
|
|
endif
|
|
let s:symbol_cursor = line[ cursor[1] - 1 : cursor[1] - 1]
|
|
if pos_c + 1 + len > cursor[1]
|
|
let s:symbol_end = line[ cursor[1] : pos_c + len - 1]
|
|
else
|
|
let s:symbol_end = ''
|
|
endif
|
|
endif
|
|
endfor
|
|
endfor
|
|
endfunction
|
|
|
|
|
|
" TODO current only support one line symbol
|
|
function! s:replace_symbol(symbol) abort
|
|
let lines = split(a:symbol, "\n")
|
|
if len(lines) > 1
|
|
let len = len(s:stack)
|
|
for idx in range(len)
|
|
let pos = s:stack[len-1-idx]
|
|
let line = getline(pos[0])
|
|
if pos[1] == 1
|
|
let begin = ''
|
|
else
|
|
let begin = line[:pos[1] - 2]
|
|
endif
|
|
let end = line[pos[1] + pos[2]:]
|
|
let line = begin . lines[0] . end
|
|
call setline(pos[0], line)
|
|
let s:stack[len-1-idx][2] = len(lines[0])
|
|
endfor
|
|
else
|
|
let len = len(s:stack)
|
|
for idx in range(len)
|
|
let pos = s:stack[len-1-idx]
|
|
let line = getline(pos[0])
|
|
if pos[1] == 1
|
|
let begin = ''
|
|
else
|
|
let begin = line[:pos[1] - 2]
|
|
endif
|
|
let end = line[pos[1] + pos[2] - 1:]
|
|
let line = begin . a:symbol . end
|
|
call setline(pos[0], line)
|
|
let s:stack[len-1-idx][2] = len(a:symbol)
|
|
endfor
|
|
endif
|
|
endfunction
|