averello/vim-lsp-mirror
1
0
Fork 0
mirror of https://github.com/prabirshrestha/vim-lsp.git synced 2025-12-14 20:35:59 +01:00
Code Issues Packages Projects Releases Wiki Activity

add support for codeLens/resolve (#1007)

Browse Source
This commit is contained in:
Prabir Shrestha
2021-01-02 19:25:19 -08:00
committed by GitHub
parent 2ba5289300
commit 1a7a6e08de
2 changed files with 76 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
autoload/lsp.vim
View File

@@ -467,6 +467,9 @@ function! lsp#default_get_supported_capabilities(server_info) abort
\ }
\ }
\ },
\ 'codeLens': {
\ 'dynamicRegistration': v:false,
\ },
\ 'completion': {
\ 'dynamicRegistration': v:false,
\ 'completionItem': {

148
autoload/lsp/ui/vim/code_lens.vim
View File

@@ -8,93 +8,91 @@
function! lsp#ui#vim#code_lens#do(option) abort
let l:sync = get(a:option, 'sync', v:false)
let l:server_names = filter(lsp#get_allowed_servers(), 'lsp#capabilities#has_code_lens_provider(v:val)')
if len(l:server_names) == 0
let s:items = []
let l:servers = filter(lsp#get_allowed_servers(), 'lsp#capabilities#has_code_lens_provider(v:val)')
if len(l:servers) == 0
return lsp#utils#error('Code lens not supported for ' . &filetype)
endif
let l:ctx = {
\ 'count': len(l:server_names),
\ 'results': [],
\}
redraw | echo 'Retrieving codelens ...'
let l:bufnr = bufnr('%')
let l:command_id = lsp#_new_command()
for l:server_name in l:server_names
call lsp#send_request(l:server_name, {
\ 'method': 'textDocument/codeLens',
\ 'params': {
\ 'textDocument': lsp#get_text_document_identifier(),
\ },
\ 'sync': l:sync,
\ 'on_notification': function('s:handle_code_lens', [l:ctx, l:server_name, l:command_id, l:sync, l:bufnr]),
\ })
endfor
echo 'Retrieving code lenses ...'
call lsp#callbag#pipe(
\ lsp#callbag#fromList(l:servers),
\ lsp#callbag#flatMap({server->
\ lsp#callbag#pipe(
\ lsp#request(server, {
\ 'method': 'textDocument/codeLens',
\ 'params': {
\ 'textDocument': lsp#get_text_document_identifier(),
\ },
\ }),
\ lsp#callbag#flatMap({x->s:resolve_if_required(server, x['response'])}),
\ lsp#callbag#map({x->{ 'server': server, 'codelens': x }}),
\ )
\ }),
\ lsp#callbag#takeUntil(lsp#callbag#pipe(
\ lsp#stream(),
\ lsp#callbag#filter({x->has_key(x, 'command')}),
\ )),
\ lsp#callbag#subscribe({
\ 'next':{x->add(s:items, x)},
\ 'complete': {->s:chooseCodeLens(s:items, l:bufnr)},
\ 'error': {e->s:error(x)},
\ }),
\ )
endfunction
function! s:handle_code_lens(ctx, server_name, command_id, sync, bufnr, data) abort
" Ignore old request.
if a:command_id != lsp#_last_command()
return
function! s:resolve_if_required(server, response) abort
let l:codelens = a:response['result']
if empty(l:codelens)
return lsp#callbag#empty()
endif
call add(a:ctx['results'], {
\ 'server_name': a:server_name,
\ 'data': a:data,
\})
let a:ctx['count'] -= 1
if a:ctx['count'] ># 0
return lsp#callbag#pipe(
\ lsp#callbag#fromList(l:codelens),
\ lsp#callbag#flatMap({codelens-> has_key(codelens, 'command') ? lsp#callbag#of(codelens) : s:resolve_codelens(a:server, codelens) }),
\ )
endfunction
function! s:resolve_codelens(server, codelens) abort
" TODO: return callbag#lsp#empty() if codelens resolve not supported by server
return lsp#callbag#pipe(
\ lsp#request(a:server, {
\ 'method': 'codeLens/resolve',
\ 'params': a:codelens
\ }),
\ lsp#callbag#map({x->x['response']['result']}),
\ )
endfunction
function! s:chooseCodeLens(items, bufnr) abort
redraw | echo 'Select codelens:'
if empty(a:items)
call lsp#utils#error('No codelens found')
return
endif
let l:total_code_lenses = []
for l:result in a:ctx['results']
let l:server_name = l:result['server_name']
let l:data = l:result['data']
" Check response error.
if lsp#client#is_error(l:data['response'])
call lsp#utils#error('Failed to CodeLens for ' . l:server_name . ': ' . lsp#client#error_message(l:data['response']))
continue
endif
" Check code lenses.
let l:code_lenses = l:data['response']['result']
if empty(l:code_lenses)
continue
endif
for l:code_lens in l:code_lenses
call add(l:total_code_lenses, {
\ 'server_name': l:server_name,
\ 'code_lens': l:code_lens,
\})
endfor
endfor
if len(l:total_code_lenses) == 0
echo 'No code lenses found'
return
endif
call lsp#log('s:handle_code_lens', l:total_code_lenses)
" Prompt to choose code lenses.
let l:index = inputlist(map(copy(l:total_code_lenses), { i, lens ->
\ printf('%s - [%s] %s', i + 1, lens['server_name'], lens['code_lens']['command']['title'])
\ }))
" Execute code lens.
if 0 < l:index && l:index <= len(l:total_code_lenses)
let l:selected = l:total_code_lenses[l:index - 1]
call s:handle_one_code_lens(l:selected['server_name'], a:sync, a:bufnr, l:selected['code_lens'])
let l:index = inputlist(map(copy(a:items), {i, value ->
\ printf('%s - [%s] %s', i + 1, value['server'], value['codelens']['command']['title'])
\ }))
if l:index > 0 && l:index <= len(a:items)
let l:selected = a:items[l:index - 1]
call s:handle_code_lens_command(l:selected['server'], l:selected['codelens'], a:bufnr)
endif
endfunction
function! s:handle_one_code_lens(server_name, sync, bufnr, code_lens) abort
function! s:error(e) abort
call lsp#utils#error('Echo occured during CodeLens' . a:e)
endfunction
function! s:handle_code_lens_command(server, codelens, bufnr) abort
call lsp#ui#vim#execute_command#_execute({
\ 'server_name': a:server_name,
\ 'command_name': get(a:code_lens['command'], 'command', ''),
\ 'command_args': get(a:code_lens['command'], 'arguments', v:null),
\ 'sync': a:sync,
\ 'bufnr': a:bufnr,
\ })
\ 'server_name': a:server,
\ 'command_name': get(a:codelens['command'], 'command', ''),
\ 'command_args': get(a:codelens['command'], 'arguments', v:null),
\ 'sync': 0,
\ 'bufnr': a:bufnr,
\ })
endfunction