summaryrefslogtreecommitdiff
path: root/plugin/vim-ai.vim
blob: b276355ac98127f31b944e1e9e121655995ffc58 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
" Ensure python3 is available
if !has('python3')
  echoerr "Python 3 support is required for vim-ai plugin"
  finish
endif

" detect if a visual selection is pending
let g:vim_ai_is_selection_pending = 0
augroup vim_ai
  autocmd!
  if exists('##ModeChanged')
    autocmd ModeChanged         *:[vV\x16]*
          \ let g:vim_ai_is_selection_pending = 1
    autocmd ModeChanged [vV\x16]*:*
          \ let g:vim_ai_is_selection_pending =
          \   v:event.new_mode =~# '^c' &&
          \     (getcmdtype() =~# '[/?]' ||
          \      getcmdtype() ==# ':'    && getcmdline() =~# "^\s*'<'>")
  else
    " workaround for version < 8.2.3424 from https://stackoverflow.com/a/20133772
    autocmd CursorMoved *
          \ let g:vim_ai_is_selection_pending = mode() =~# "^[vV\<C-v>]"
    autocmd CmdLineEnter,CmdwinEnter *
          \ if g:vim_ai_is_selection_pending && getcmdtype() ==# ':' |
          \   let g:vim_ai_is_selection_pending = getcmdline() =~# "^\s*'<'>" |
          \ endif
    autocmd CmdLineLeave,CmdwinLeave *
          \ if g:vim_ai_is_selection_pending && getcmdtype() !~# '[/?]' |
          \   let g:vim_ai_is_selection_pending = 0 |
          \ endif
  endif
augroup END

command! -range   -nargs=? AI        <line1>,<line2>call vim_ai#AIRun({}, <q-args>)
command! -range   -nargs=? AIEdit    <line1>,<line2>call vim_ai#AIEditRun({}, <q-args>)
" Whereas AI and AIEdit default to passing the current line as range
" AIChat defaults to passing nothing which is achieved by -range=0 and passing
" <count> as described at https://stackoverflow.com/a/20133772
command! -range=0 -nargs=? AIChat    <line1>,<line2>call vim_ai#AIChatRun(<count>, {}, <q-args>)
command! -nargs=?          AINewChat                call vim_ai#AINewChatRun(<q-args>)
command!                   AIRedo                   call vim_ai#AIRedoRun()