Add :OmniSharpRepeatCodeAction command

Author: nickspoons

autoload/OmniSharp/actions/codeactions.vim

@@ -42,6 +42,11 @@ function! s:StdioGet(mode, Callback) abort
   \ 'ResponseHandler': function('s:StdioGetRH', [a:Callback]),
   \ 'SavePosition': 1
   \}
+  call s:PrepareParameters(opts, a:mode)
+  call OmniSharp#stdio#Request('/v2/getcodeactions', opts)
+endfunction
+
+function! s:PrepareParameters(opts, mode) abort
   if a:mode ==# 'visual'
     let start = getpos("'<")
     let end = getpos("'>")
@@ -63,13 +68,12 @@ function! s:StdioGet(mode, Callback) abort
     \   }
     \ }
     \}
-    let opts.Parameters = s:codeActionParameters
+    let a:opts.Parameters = s:codeActionParameters
   else
     if exists('s:codeActionParameters')
       unlet s:codeActionParameters
     endif
   endif
-  call OmniSharp#stdio#Request('/v2/getcodeactions', opts)
 endfunction
 
 function! s:StdioGetRH(Callback, response) abort
@@ -156,16 +160,35 @@ function! s:CBGetCodeActions(mode, actions) abort
 endfunction
 
 
+function! OmniSharp#actions#codeactions#Repeat(mode) abort
+  if !g:OmniSharp_server_stdio
+    echomsg 'This functionality is only available with the stdio server'
+    return
+  endif
+  if !exists('s:lastCodeActionIdentifier')
+    echomsg 'There is no last code action to repeat'
+    return
+  endif
+  call s:PrepareParameters({}, a:mode)
+  let RH = function('OmniSharp#buffer#PerformChanges', [{}])
+  call s:RunCodeAction(s:lastCodeActionIdentifier, 1, RH)
+endfunction
+
 function! OmniSharp#actions#codeactions#Run(action, ...) abort
   let RH = function('OmniSharp#buffer#PerformChanges', [a:0 ? a:1 : {}])
+  let s:lastCodeActionIdentifier = a:action.Identifier
+  call s:RunCodeAction(a:action.Identifier, 0, RH)
+endfunction
+
+function! s:RunCodeAction(identifier, repeating, ResponseHandler) abort
   let opts = {
-  \ 'ResponseHandler': RH,
+  \ 'ResponseHandler': a:ResponseHandler,
   \ 'Parameters': {
-  \   'Identifier': a:action.Identifier,
+  \   'Identifier': a:identifier,
   \   'WantsTextChanges': 1,
   \   'WantsAllCodeActionOperations': 1
   \ },
-  \ 'UsePreviousPosition': 1
+  \ 'UsePreviousPosition': !a:repeating
   \}
   if exists('s:codeActionParameters')
     call extend(opts.Parameters, s:codeActionParameters, 'force')

autoload/OmniSharp/stdio.vim

@@ -97,7 +97,7 @@ function! s:HandleServerEvent(job, res) abort
 endfunction
 
 function! OmniSharp#stdio#Request(command, opts) abort
-  if has_key(a:opts, 'UsePreviousPosition')
+  if get(a:opts, 'UsePreviousPosition', 0)
     let [bufnr, lnum, cnum] = s:lastPosition
   elseif has_key(a:opts, 'BufNum') && a:opts.BufNum != bufnr('%')
     let bufnr = a:opts.BufNum

ftplugin/cs/OmniSharp.vim

@@ -45,6 +45,7 @@ command! -buffer -bar OmniSharpPreviewDefinition call OmniSharp#actions#definiti
 command! -buffer -bar OmniSharpPreviewImplementation call OmniSharp#actions#implementations#Preview()
 command! -buffer -bar OmniSharpRename call OmniSharp#actions#rename#Prompt()
 command! -buffer -nargs=1 OmniSharpRenameTo call OmniSharp#actions#rename#To(<q-args>)
+command! -buffer -bar OmniSharpRepeatCodeAction call OmniSharp#actions#codeactions#Repeat('normal')
 command! -buffer -bar OmniSharpRunTest call OmniSharp#actions#test#Run()
 command! -buffer -bar -nargs=* -complete=file OmniSharpRunTestsInFile call OmniSharp#actions#test#RunInFile(<f-args>)
 command! -buffer -bar OmniSharpSignatureHelp call OmniSharp#actions#signature#SignatureHelp()
@@ -60,6 +61,8 @@ nnoremap <buffer> <Plug>(omnisharp_find_usages) :OmniSharpFindUsages<CR>
 nnoremap <buffer> <Plug>(omnisharp_fix_usings) :OmniSharpFixUsings<CR>
 nnoremap <buffer> <Plug>(omnisharp_code_actions) :OmniSharpGetCodeActions<CR>
 xnoremap <buffer> <Plug>(omnisharp_code_actions) :call OmniSharp#actions#codeactions#Get('visual')<CR>
+nnoremap <buffer> <Plug>(omnisharp_code_action_repeat) :OmniSharpRepeatCodeAction<CR>
+xnoremap <buffer> <Plug>(omnisharp_code_action_repeat) :call OmniSharp#actions#codeactions#Repeat('visual')<CR>
 nnoremap <buffer> <Plug>(omnisharp_global_code_check) :OmniSharpGlobalCodeCheck<CR>
 nnoremap <buffer> <Plug>(omnisharp_go_to_definition) :OmniSharpGotoDefinition<CR>
 nnoremap <buffer> <Plug>(omnisharp_highlight) :OmniSharpHighlight<CR>
@@ -117,6 +120,7 @@ let b:undo_ftplugin .= '
 \| delcommand OmniSharpRenameTo
 \| delcommand OmniSharpRestartAllServers
 \| delcommand OmniSharpRestartServer
+\| delcommand OmniSharpRepeatCodeAction
 \| delcommand OmniSharpRunTest
 \| delcommand OmniSharpRunTestsInFile
 \| delcommand OmniSharpSignatureHelp