" Vim plugin to provide multiline analogs to t & f

if !exists('g:supert_provide_bindings')
  let g:supert_provide_bindings = 1
endif

let s:last = []

function! s:do_search(type, vis) abort
  if a:type == ';'
    call s:do_last(0, a:vis)
  elseif a:type == ','
    call s:do_last(1, a:vis)
  else
    call s:do_search_ch(a:type, a:vis, nr2char(getchar()))
  endif
endfunction

function! s:do_last(inv, vis) abort
  if len(s:last) < 2
    return
  endif

  let type = s:last[0]
  let ch = s:last[1]

  if a:inv
    if type =~ '[a-z]'
      let type = toupper(type)
    else
      let type = tolower(type)
    endif
  endif

  return s:do_search_ch(type, a:vis, ch)
endfunction

function! s:do_search_ch(type, vis, ch)
  let s:last = [a:type, a:ch]

  let flags = 'W'
  let pattern = ''

  if a:type =~ '[A-Z]'
    let flags .= 'b'
  endif

  let pattern = a:ch

  if a:type == 't'
    let pattern = '\zs\_.\ze' . pattern
  elseif a:type == 'T'
    let pattern = pattern . '\zs\_.\ze'
  endif

  if a:vis == 'v'
    exec "norm gv"
  elseif a:vis == 'o'
    exec "norm v"
  endif

  let i = 0
  while i < v:count1
    call search(pattern, flags)
    let i += 1
  endwhile
endfunction

nnoremap <silent> <Plug>(supert-replace-t) :<c-u>silent call <SID>do_search('t', '')<cr>
nnoremap <silent> <Plug>(supert-replace-T) :<c-u>silent call <SID>do_search('T', '')<cr>
nnoremap <silent> <Plug>(supert-replace-f) :<c-u>silent call <SID>do_search('f', '')<cr>
nnoremap <silent> <Plug>(supert-replace-F) :<c-u>silent call <SID>do_search('F', '')<cr>

vnoremap <silent> <Plug>(supert-replace-t) :<c-u>silent call <SID>do_search('t', 'v')<cr>
vnoremap <silent> <Plug>(supert-replace-T) :<c-u>silent call <SID>do_search('T', 'v')<cr>
vnoremap <silent> <Plug>(supert-replace-f) :<c-u>silent call <SID>do_search('f', 'v')<cr>
vnoremap <silent> <Plug>(supert-replace-F) :<c-u>silent call <SID>do_search('F', 'v')<cr>

onoremap <silent> <Plug>(supert-replace-t) :<c-u>silent call <SID>do_search('t', 'o')<cr>
onoremap <silent> <Plug>(supert-replace-T) :<c-u>silent call <SID>do_search('T', 'o')<cr>
onoremap <silent> <Plug>(supert-replace-f) :<c-u>silent call <SID>do_search('f', 'o')<cr>
onoremap <silent> <Plug>(supert-replace-F) :<c-u>silent call <SID>do_search('F', 'o')<cr>

onoremap <silent> <Plug>(supert-replace-,) :<c-u>silent call <SID>do_search(',', 'o')<cr>
onoremap <silent> <Plug>(supert-replace-;) :<c-u>silent call <SID>do_search(';', 'o')<cr>

vnoremap <silent> <Plug>(supert-replace-,) :<c-u>silent call <SID>do_search(',', 'v')<cr>
vnoremap <silent> <Plug>(supert-replace-;) :<c-u>silent call <SID>do_search(';', 'v')<cr>

nnoremap <silent> <Plug>(supert-replace-,) :<c-u>silent call <SID>do_search(',', '')<cr>
nnoremap <silent> <Plug>(supert-replace-;) :<c-u>silent call <SID>do_search(';', '')<cr>


if g:supert_provide_bindings
  nnoremap <silent> <M-t> <Plug>(supert-replace-t)
  nnoremap <silent> <M-T> <Plug>(supert-replace-T)
  nnoremap <silent> <M-f> <Plug>(supert-replace-f)
  nnoremap <silent> <M-F> <Plug>(supert-replace-F)

  vnoremap <silent> <M-t> <Plug>(supert-replace-t)
  vnoremap <silent> <M-T> <Plug>(supert-replace-T)
  vnoremap <silent> <M-f> <Plug>(supert-replace-f)
  vnoremap <silent> <M-F> <Plug>(supert-replace-F)

  vnoremap <silent> <M-t> <Plug>(supert-replace-t)
  vnoremap <silent> <M-T> <Plug>(supert-replace-T)
  vnoremap <silent> <M-f> <Plug>(supert-replace-f)
  vnoremap <silent> <M-F> <Plug>(supert-replace-F)

  vnoremap <silent> <M-;> <Plug>(supert-replace-;)
  vnoremap <silent> <M-,> <Plug>(supert-replace-,)

  nnoremap <silent> <M-;> <Plug>(supert-replace-;)
  nnoremap <silent> <M-,> <Plug>(supert-replace-,)
endif