function! s:not_supported(what) abort return lsp#utils#error(printf("%s not supported for filetype '%s'", a:what, &filetype)) endfunction function! lsp#ui#vim#implementation(in_preview, ...) abort let l:ctx = { 'in_preview': a:in_preview } if a:0 let l:ctx['mods'] = a:1 endif call s:list_location('implementation', l:ctx) endfunction function! lsp#ui#vim#type_definition(in_preview, ...) abort let l:ctx = { 'in_preview': a:in_preview } if a:0 let l:ctx['mods'] = a:1 endif call s:list_location('typeDefinition', l:ctx) endfunction function! lsp#ui#vim#declaration(in_preview, ...) abort let l:ctx = { 'in_preview': a:in_preview } if a:0 let l:ctx['mods'] = a:1 endif call s:list_location('declaration', l:ctx) endfunction function! lsp#ui#vim#definition(in_preview, ...) abort let l:ctx = { 'in_preview': a:in_preview } if a:0 let l:ctx['mods'] = a:1 endif call s:list_location('definition', l:ctx) endfunction function! lsp#ui#vim#references(ctx) abort let l:ctx = extend({ 'jump_if_one': 0 }, a:ctx) let l:request_params = { 'context': { 'includeDeclaration': v:true } } call s:list_location('references', l:ctx, l:request_params) endfunction function! lsp#ui#vim#add_tree_references() abort let l:ctx = { 'add_tree': v:true } call lsp#ui#vim#references(l:ctx) endfunction function! s:list_location(method, ctx, ...) abort " typeDefinition => type definition let l:operation = substitute(a:method, '\u', ' \l\0', 'g') let l:capabilities_func = printf('lsp#capabilities#has_%s_provider(v:val)', substitute(l:operation, ' ', '_', 'g')) let l:servers = filter(lsp#get_allowed_servers(), l:capabilities_func) let l:command_id = lsp#_new_command() let l:ctx = extend({ 'counter': len(l:servers), 'list':[], 'last_command_id': l:command_id, 'jump_if_one': 1, 'mods': '', 'in_preview': 0 }, a:ctx) if len(l:servers) == 0 call s:not_supported('Retrieving ' . l:operation) return endif let l:params = { \ 'textDocument': lsp#get_text_document_identifier(), \ 'position': lsp#get_position(), \ } if a:0 call extend(l:params, a:1) endif for l:server in l:servers call lsp#send_request(l:server, { \ 'method': 'textDocument/' . a:method, \ 'params': l:params, \ 'on_notification': function('s:handle_location', [l:ctx, l:server, l:operation]), \ }) endfor echo printf('Retrieving %s ...', l:operation) endfunction function! s:rename(server, new_name, pos) abort if empty(a:new_name) echo '... Renaming aborted ...' return endif " needs to flush existing open buffers call lsp#send_request(a:server, { \ 'method': 'textDocument/rename', \ 'params': { \ 'textDocument': lsp#get_text_document_identifier(), \ 'position': a:pos, \ 'newName': a:new_name, \ }, \ 'on_notification': function('s:handle_workspace_edit', [a:server, lsp#_last_command(), 'rename']), \ }) echo ' ... Renaming ...' endfunction function! lsp#ui#vim#rename() abort let l:servers = filter(lsp#get_allowed_servers(), 'lsp#capabilities#has_rename_prepare_provider(v:val)') let l:prepare_support = 1 if len(l:servers) == 0 let l:servers = filter(lsp#get_allowed_servers(), 'lsp#capabilities#has_rename_provider(v:val)') let l:prepare_support = 0 endif let l:command_id = lsp#_new_command() if len(l:servers) == 0 call s:not_supported('Renaming') return endif " TODO: ask the user which server it should use to rename if there are multiple let l:server = l:servers[0] if l:prepare_support call lsp#send_request(l:server, { \ 'method': 'textDocument/prepareRename', \ 'params': { \ 'textDocument': lsp#get_text_document_identifier(), \ 'position': lsp#get_position(), \ }, \ 'on_notification': function('s:handle_rename_prepare', [l:server, l:command_id, 'rename_prepare', expand(''), lsp#get_position()]), \ }) return endif call s:rename(l:server, input('new name: ', expand('')), lsp#get_position()) endfunction function! s:stop_all_servers() abort for l:server in lsp#get_server_names() if !lsp#is_server_running(l:server) continue endif echo 'Stopping' l:server 'server ...' call lsp#stop_server(l:server) endfor endfunction function! s:stop_named_server(name) abort if !lsp#is_valid_server_name(a:name) call lsp#utils#warning('No LSP servers named "' . a:name . '"') return endif if lsp#is_server_running(a:name) echo 'Stopping "' . a:name . '" server...' call lsp#stop_server(a:name) else call lsp#utils#warning( \ 'Server "' . a:name . '" is not running: ' \ . lsp#get_server_status(a:name) \ ) endif endfunction function! s:stop_buffer_servers() abort let l:servers = lsp#get_allowed_servers() let l:servers = \ filter(l:servers, {idx, name -> lsp#is_server_running(name)}) if empty(l:servers) call lsp#utils#warning('No active LSP servers for the current buffer') return endif for l:server in l:servers echo 'Stopping "' . l:server . '" server ...' call lsp#stop_server(l:server) endfor endfunction function! lsp#ui#vim#stop_server(stop_all, ...) abort if a:0 != 0 && a:0 != 1 call lsp#utils#error( \ 'lsp#ui#vim#stop_server(): expected 1 optional "name" argument.' \ . ' Got: "' . join(a:000, '", "') . '".') return endif let l:stop_all = a:stop_all ==# '!' let l:name = get(a:000, 0, '') if l:stop_all if !empty(l:name) call lsp#utils#error( \ '"!" stops all servers: name is ignored: "' . l:name . '"') endif call s:stop_all_servers() return endif if !empty(l:name) call s:stop_named_server(l:name) return endif call s:stop_buffer_servers() endfunction function! lsp#ui#vim#workspace_symbol(query) abort let l:servers = filter(lsp#get_allowed_servers(), 'lsp#capabilities#has_workspace_symbol_provider(v:val)') let l:command_id = lsp#_new_command() if len(l:servers) == 0 call s:not_supported('Retrieving workspace symbols') return endif if !empty(a:query) let l:query = a:query else let l:query = inputdialog('query>', '', "\") if l:query ==# "\" return endif endif for l:server in l:servers call lsp#send_request(l:server, { \ 'method': 'workspace/symbol', \ 'params': { \ 'query': l:query, \ }, \ 'on_notification': function('s:handle_symbol', [l:server, l:command_id, 'workspaceSymbol']), \ }) endfor redraw echo 'Retrieving workspace symbols ...' endfunction function! lsp#ui#vim#document_symbol() abort let l:servers = filter(lsp#get_allowed_servers(), 'lsp#capabilities#has_document_symbol_provider(v:val)') let l:command_id = lsp#_new_command() if len(l:servers) == 0 call s:not_supported('Retrieving symbols') return endif for l:server in l:servers call lsp#send_request(l:server, { \ 'method': 'textDocument/documentSymbol', \ 'params': { \ 'textDocument': lsp#get_text_document_identifier(), \ }, \ 'on_notification': function('s:handle_symbol', [l:server, l:command_id, 'documentSymbol']), \ }) endfor echo 'Retrieving document symbols ...' endfunction function! s:handle_symbol(server, last_command_id, type, data) abort if a:last_command_id != lsp#_last_command() return endif if lsp#client#is_error(a:data['response']) call lsp#utils#error('Failed to retrieve '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) return endif let l:list = lsp#ui#vim#utils#symbols_to_loc_list(a:server, a:data) call lsp#ui#vim#utils#setqflist(l:list, a:type) if empty(l:list) call lsp#utils#error('No ' . a:type .' found') else echo 'Retrieved ' . a:type botright copen endif endfunction function! s:handle_location(ctx, server, type, data) abort "ctx = {counter, list, last_command_id, jump_if_one, mods, in_preview} if a:ctx['last_command_id'] != lsp#_last_command() return endif let a:ctx['counter'] = a:ctx['counter'] - 1 if lsp#client#is_error(a:data['response']) || !has_key(a:data['response'], 'result') call lsp#utils#error('Failed to retrieve '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) else let a:ctx['list'] = a:ctx['list'] + lsp#utils#location#_lsp_to_vim_list(a:data['response']['result']) endif if a:ctx['counter'] == 0 if empty(a:ctx['list']) call lsp#utils#error('No ' . a:type .' found') else call lsp#utils#tagstack#_update() let l:loc = a:ctx['list'][0] if len(a:ctx['list']) == 1 && a:ctx['jump_if_one'] && !a:ctx['in_preview'] call lsp#utils#location#_open_vim_list_item(l:loc, a:ctx['mods']) echo 'Retrieved ' . a:type redraw elseif !a:ctx['in_preview'] if get(a:ctx, 'add_tree', v:false) let l:qf = getqflist({'idx' : 0, 'items': []}) let l:pos = l:qf.idx let l:parent = l:qf.items let l:level = count(l:parent[l:pos-1].text, g:lsp_tree_incoming_prefix) let a:ctx['list'] = extend(l:parent, map(a:ctx['list'], 'extend(v:val, {"text": repeat("' . g:lsp_tree_incoming_prefix . '", l:level+1) . v:val.text})'), l:pos) endif call lsp#ui#vim#utils#setqflist(a:ctx['list'], a:type) echo 'Retrieved ' . a:type botright copen if get(a:ctx, 'add_tree', v:false) " move the cursor to the newly added item execute l:pos + 1 endif else let l:lines = readfile(l:loc['filename']) if has_key(l:loc,'viewstart') " showing a locationLink let l:view = l:lines[l:loc['viewstart'] : l:loc['viewend']] call lsp#ui#vim#output#preview(a:server, l:view, { \ 'statusline': ' LSP Peek ' . a:type, \ 'filetype': &filetype \ }) else " showing a location call lsp#ui#vim#output#preview(a:server, l:lines, { \ 'statusline': ' LSP Peek ' . a:type, \ 'cursor': { 'line': l:loc['lnum'], 'col': l:loc['col'], 'align': g:lsp_peek_alignment }, \ 'filetype': &filetype \ }) endif endif endif endif endfunction function! s:handle_rename_prepare(server, last_command_id, type, cword, position, data) abort if a:last_command_id != lsp#_last_command() return endif if lsp#client#is_error(a:data['response']) call lsp#utils#error('Failed to retrieve '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) return endif let l:result = a:data['response']['result'] " Check response: null. if empty(l:result) echo 'The ' . a:server . ' returns for ' . a:type . ' (The rename request may be invalid at the given position).' return endif " Check response: { defaultBehavior: boolean }. if has_key(l:result, 'defaultBehavior') call timer_start(1, {x->s:rename(a:server, input('new name: ', a:cword), a:position)}) return endif " Check response: { placeholder: string } if has_key(l:result, 'placeholder') && !empty(l:result['placeholder']) call timer_start(1, {x->s:rename(a:server, input('new name: ', a:cword), a:position)}) return endif " Check response: { range: Range } | Range let l:range = get(l:result, 'range', l:result) let l:lines = getline(1, '$') let [l:start_line, l:start_col] = lsp#utils#position#lsp_to_vim('%', l:range['start']) let [l:end_line, l:end_col] = lsp#utils#position#lsp_to_vim('%', l:range['end']) if l:start_line ==# l:end_line let l:name = l:lines[l:start_line - 1][l:start_col - 1 : l:end_col - 2] else let l:name = l:lines[l:start_line - 1][l:start_col - 1 :] for l:i in range(l:start_line, l:end_line - 2) let l:name .= "\n" . l:lines[l:i] endfor if l:end_col - 2 < 0 let l:name .= "\n" else let l:name .= l:lines[l:end_line - 1][: l:end_col - 2] endif endif call timer_start(1, {x->s:rename(a:server, input('new name: ', l:name), l:range['start'])}) endfunction function! s:handle_workspace_edit(server, last_command_id, type, data) abort if a:last_command_id != lsp#_last_command() return endif if lsp#client#is_error(a:data['response']) call lsp#utils#error('Failed to retrieve '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) return endif call lsp#utils#workspace_edit#apply_workspace_edit(a:data['response']['result']) echo 'Renamed' endfunction function! s:handle_text_edit(server, last_command_id, type, data) abort if a:last_command_id != lsp#_last_command() return endif if lsp#client#is_error(a:data['response']) call lsp#utils#error('Failed to '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) return endif call lsp#utils#text_edit#apply_text_edits(a:data['request']['params']['textDocument']['uri'], a:data['response']['result']) redraw | echo 'Document formatted' endfunction function! lsp#ui#vim#code_action(opts) abort call lsp#ui#vim#code_action#do(extend({ \ 'sync': v:false, \ 'selection': v:false, \ 'query': '', \ }, a:opts)) endfunction function! lsp#ui#vim#code_lens() abort call lsp#ui#vim#code_lens#do({ \ 'sync': v:false, \ }) endfunction function! lsp#ui#vim#add_tree_call_hierarchy_incoming() abort let l:ctx = { 'add_tree': v:true } call lsp#ui#vim#call_hierarchy_incoming(l:ctx) endfunction function! lsp#ui#vim#call_hierarchy_incoming(ctx) abort let l:ctx = extend({ 'method': 'incomingCalls', 'key': 'from' }, a:ctx) call s:prepare_call_hierarchy(l:ctx) endfunction function! lsp#ui#vim#call_hierarchy_outgoing() abort let l:ctx = { 'method': 'outgoingCalls', 'key': 'to' } call s:prepare_call_hierarchy(l:ctx) endfunction function! s:prepare_call_hierarchy(ctx) abort let l:servers = filter(lsp#get_allowed_servers(), 'lsp#capabilities#has_call_hierarchy_provider(v:val)') let l:command_id = lsp#_new_command() let l:ctx = extend({ 'counter': len(l:servers), 'list':[], 'last_command_id': l:command_id }, a:ctx) if len(l:servers) == 0 call s:not_supported('Retrieving call hierarchy') return endif for l:server in l:servers call lsp#send_request(l:server, { \ 'method': 'textDocument/prepareCallHierarchy', \ 'params': { \ 'textDocument': lsp#get_text_document_identifier(), \ 'position': lsp#get_position(), \ }, \ 'on_notification': function('s:handle_prepare_call_hierarchy', [l:ctx, l:server, 'prepare_call_hierarchy']), \ }) endfor echo 'Preparing call hierarchy ...' endfunction function! s:handle_prepare_call_hierarchy(ctx, server, type, data) abort if a:ctx['last_command_id'] != lsp#_last_command() return endif if lsp#client#is_error(a:data['response']) || !has_key(a:data['response'], 'result') call lsp#utils#error('Failed to '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) return endif if empty(a:data['response']['result']) call lsp#utils#warning('Failed to '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) return endif for l:item in a:data['response']['result'] call s:call_hierarchy(a:ctx, a:server, l:item) endfor endfunction function! s:call_hierarchy(ctx, server, item) abort call lsp#send_request(a:server, { \ 'method': 'callHierarchy/' . a:ctx['method'], \ 'params': { \ 'item': a:item, \ }, \ 'on_notification': function('s:handle_call_hierarchy', [a:ctx, a:server, 'call_hierarchy']), \ }) endfunction function! s:handle_call_hierarchy(ctx, server, type, data) abort if a:ctx['last_command_id'] != lsp#_last_command() return endif let a:ctx['counter'] = a:ctx['counter'] - 1 if lsp#client#is_error(a:data['response']) || !has_key(a:data['response'], 'result') call lsp#utils#error('Failed to retrieve '. a:type . ' for ' . a:server . ': ' . lsp#client#error_message(a:data['response'])) elseif a:data['response']['result'] isnot v:null for l:item in a:data['response']['result'] let l:loc = s:hierarchy_item_to_vim(l:item[a:ctx['key']], a:server) if l:loc isnot v:null let a:ctx['list'] += [l:loc] endif endfor endif if a:ctx['counter'] == 0 if empty(a:ctx['list']) call lsp#utils#error('No ' . a:type .' found') else call lsp#utils#tagstack#_update() if get(a:ctx, 'add_tree', v:false) let l:qf = getqflist({'idx' : 0, 'items': []}) let l:pos = l:qf.idx let l:parent = l:qf.items let l:level = count(l:parent[l:pos-1].text, g:lsp_tree_incoming_prefix) let a:ctx['list'] = extend(l:parent, map(a:ctx['list'], 'extend(v:val, {"text": repeat("' . g:lsp_tree_incoming_prefix . '", l:level+1) . v:val.text})'), l:pos) endif call lsp#ui#vim#utils#setqflist(a:ctx['list'], a:type) echo 'Retrieved ' . a:type botright copen if get(a:ctx, 'add_tree', v:false) " move the cursor to the newly added item execute l:pos + 1 endif endif endif endfunction function! s:hierarchy_item_to_vim(item, server) abort let l:uri = a:item['uri'] if !lsp#utils#is_file_uri(l:uri) return v:null endif let l:path = lsp#utils#uri_to_path(l:uri) let [l:line, l:col] = lsp#utils#position#lsp_to_vim(l:path, a:item['range']['start']) let l:text = '[' . lsp#ui#vim#utils#_get_symbol_text_from_kind(a:server, a:item['kind']) . '] ' . a:item['name'] if has_key(a:item, 'detail') let l:text .= ": " . a:item['detail'] endif return { \ 'filename': l:path, \ 'lnum': l:line, \ 'col': l:col, \ 'text': l:text, \ } endfunction