" Copyright (c) 2015 Junegunn Choi
"
" MIT License
"
" Permission is hereby granted, free of charge, to any person obtaining
" a copy of this software and associated documentation files (the
" "Software"), to deal in the Software without restriction, including
" without limitation the rights to use, copy, modify, merge, publish,
" distribute, sublicense, and/or sell copies of the Software, and to
" permit persons to whom the Software is furnished to do so, subject to
" the following conditions:
"
" The above copyright notice and this permission notice shall be
" included in all copies or substantial portions of the Software.
"
" THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
" EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
" MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
" NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
" LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
" OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
" WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

if exists('g:loaded_after_object')
  finish
endif
let g:loaded_after_object = 1

let s:cpo_save = &cpo
set cpo&vim

function! s:after(str, cnt, vis, bw)
  let col = a:vis ? col("'<") : col('.')
  let line = getline('.')
  let parts = split(line, '\V'.a:str.'\zs', 1)

  try
    if len(parts) == 1
      throw 'exit'
    endif

    for i in reverse(range(1, a:cnt))
      let len = 0
      let idx = 0
      for part in parts[0 : (a:bw ? -2 : -3)]
        let len += strchars(part)
        if len > col - 1
          break
        endif
        let idx += 1
      endfor

      if a:bw
        let idx = max([idx - 2, 0])
      endif
      let col = strchars(join(parts[0 : idx], '')) + (i > 1 && a:cnt > 1)
    endfor

    let rest = substitute(line, '^.\{'.col.'}', '', '')
    if empty(rest)
      throw 'exit'
    else
      let idx = max([match(rest, '\S'), 0])
      execute 'normal! 0'.(col + idx).'lv$h'
    endif
  catch 'exit'
    call s:textobj_cancel()
    if a:vis
      normal! gv
    endif
    return
  finally
    if histget(':', -1) =~ '<SNR>[0-9_]*after('
      call histdel(':', -1)
    endif
    echo
  endtry
endfunction

function! s:textobj_cancel()
  if v:operator == 'c'
    augroup afterobj_undo_empty_change
      autocmd InsertLeave <buffer> execute 'normal! u'
            \| execute 'autocmd! afterobj_undo_empty_change'
            \| execute 'augroup! afterobj_undo_empty_change'
    augroup END
  endif
endfunction

noremap         <Plug>(AfterAfterObject) <nop>
inoremap <expr> <Plug>(AfterAfterObject) exists('#afterobj_undo_empty_change')?"\<esc>":''

function! s:esc(c)
  " TODO: anything else?
  return substitute(substitute(a:c, ' ', '<space>', 'g'), '|', '<bar>', 'g')
endfunction

function! s:parse_args(args)
  let lists = filter(copy(a:args), 'type(v:val) == type([])')
  let chars = filter(copy(a:args), 'type(v:val) == type("")')
  return [get(lists, '-1', ['a', 'aa']), chars]
endfunction

function! after_object#enable(...)
  let [prefixes, chars] = s:parse_args(a:000)
  let prefixes = map(prefixes[0 : 1], '[v:val, v:key]')
  for c in chars
    for [p, b] in prefixes
      for [m, v] in items({ 'x': 1, 'o': 0 })
        execute printf(
        \ '%smap <silent> %s%s :<c-u>call <sid>after(%s, v:count1, %d, %d)<cr><Plug>(AfterAfterObject)',
        \  m, p, s:esc(c), string(s:esc(c)), v, b)
      endfor
    endfor
  endfor
endfunction

function! after_object#disable(...)
  let [prefixes, chars] = s:parse_args(a:000)
  for c in chars
    for p in prefixes[0 : 1]
      for m in ['x', 'o']
        execute printf('%sunmap <silent> %s%s', m, p, s:esc(c))
      endfor
    endfor
  endfor
endfunction

let &cpo = s:cpo_save
unlet s:cpo_save