Skip to content

Commit

Permalink
Add support for DocumentSymbol in document outline requests
Browse files Browse the repository at this point in the history
We are intentionally not advertising the capability.
We do want a flat response, so receiving a DocumentSymbol is a
pessimisation.
Not advertising the capability means that conforming servers take the
faster code path and the likes of OmniSharp, that assume capabilities,
still work.

Yes, it's messy, but so is LSP.
  • Loading branch information
bstaletic committed Aug 21, 2024
1 parent 1026c83 commit 5de7052
Show file tree
Hide file tree
Showing 2 changed files with 104 additions and 23 deletions.
61 changes: 38 additions & 23 deletions ycmd/completers/language_server/language_server_completer.py
Original file line number Diff line number Diff line change
Expand Up @@ -2638,7 +2638,7 @@ def GoToSymbol( self, request_data, args ):
REQUEST_TIMEOUT_COMMAND )

result = response.get( 'result' ) or []
return _SymbolInfoListToGoTo( request_data, result )
return _LspSymbolListToGoTo( request_data, result )


def GoToDocumentOutline( self, request_data ):
Expand All @@ -2655,12 +2655,11 @@ def GoToDocumentOutline( self, request_data ):

result = response.get( 'result' ) or []

# We should only receive SymbolInformation (not DocumentSymbol)
if any( 'range' in s for s in result ):
raise ValueError(
"Invalid server response; DocumentSymbol not supported" )
LOGGER.debug( 'Hierarchical DocumentSymbol not supported.' )
result = _FlattenDocumentSymbolHierarchy( result )

return _SymbolInfoListToGoTo( request_data, result )
return _LspSymbolListToGoTo( request_data, result )


def InitialHierarchy( self, request_data, args ):
Expand Down Expand Up @@ -3396,26 +3395,10 @@ def _LocationListToGoTo( request_data, positions ):
raise RuntimeError( 'Cannot jump to location' )


def _SymbolInfoListToGoTo( request_data, symbols ):
def _LspSymbolListToGoTo( request_data, symbols ):
"""Convert a list of LSP SymbolInformation into a YCM GoTo response"""

def BuildGoToLocationFromSymbol( symbol ):
location, line_value = _LspLocationToLocationAndDescription(
request_data,
symbol[ 'location' ] )

description = ( f'{ lsp.SYMBOL_KIND[ symbol[ "kind" ] ] }: '
f'{ symbol[ "name" ] }' )

goto = responses.BuildGoToResponseFromLocation( location,
description )
goto[ 'extra_data' ] = {
'kind': lsp.SYMBOL_KIND[ symbol[ 'kind' ] ],
'name': symbol[ 'name' ],
}
return goto

locations = [ BuildGoToLocationFromSymbol( s ) for s in
locations = [ _BuildGoToLocationFromSymbol( s, request_data ) for s in
sorted( symbols,
key = lambda s: ( s[ 'kind' ], s[ 'name' ] ) ) ]

Expand All @@ -3427,6 +3410,38 @@ def BuildGoToLocationFromSymbol( symbol ):
return locations


def _FlattenDocumentSymbolHierarchy( symbols ):
result = []
for s in symbols:
result.append( s )
if children := s.get( 'children' ):
result.extend( _FlattenDocumentSymbolHierarchy( children ) )
return result


def _BuildGoToLocationFromSymbol( symbol, request_data ):
""" Convert a LSP SymbolInfo or DocumentSymbol into a YCM GoTo response"""
lsp_location = symbol.get( 'location' )
if not lsp_location: # This is a DocumentSymbol
lsp_location = symbol
lsp_location[ 'uri' ] = lsp.FilePathToUri( request_data[ 'filepath' ] )

location, line_value = _LspLocationToLocationAndDescription(
request_data,
lsp_location )

description = ( f'{ lsp.SYMBOL_KIND[ symbol[ "kind" ] ] }: '
f'{ symbol[ "name" ] }' )

goto = responses.BuildGoToResponseFromLocation( location,
description )
goto[ 'extra_data' ] = {
'kind': lsp.SYMBOL_KIND[ symbol[ 'kind' ] ],
'name': symbol[ 'name' ],
}
return goto


def _LspLocationToLocationAndDescription( request_data,
location,
range_property = 'range' ):
Expand Down
66 changes: 66 additions & 0 deletions ycmd/tests/language_server/language_server_completer_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
empty,
ends_with,
equal_to,
instance_of,
contains_exactly,
has_entries,
has_entry,
Expand Down Expand Up @@ -103,6 +104,71 @@ def _Check_Distance( point, start, end, expected ):


class LanguageServerCompleterTest( TestCase ):
@IsolatedYcmd()
def test_LanguageServerCompleter_DocumentSymbol_Hierarchical( self, app ):
completer = MockCompleter()
completer._server_capabilities = { 'documentSymbolProvider': True }
request_data = RequestWrap( BuildRequest( filepath = '/foo' ) )
server_response = {
'result': [
{
"name": "testy",
"kind": 3,
"range": {
"start": { "line": 2, "character": 0 },
"end": { "line": 12, "character": 1 }
},
"children": [
{
"name": "MainClass",
"kind": 5,
"range": {
"start": { "line": 4, "character": 1 },
"end": { "line": 11, "character": 2 }
}
}
]
},
{
"name": "other",
"kind": 3,
"range": {
"start": { "line": 14, "character": 0 },
"end": { "line": 15, "character": 1 }
},
"children": []
}
]
}

with patch.object( completer, '_ServerIsInitialized', return_value = True ):
with patch.object( completer.GetConnection(),
'GetResponse',
return_value = server_response ):
document_outline = completer.GoToDocumentOutline( request_data )
print( f'result: { document_outline }' )
assert_that( document_outline, contains_exactly(
has_entries( {
'line_num': 15,
'column_num': 1,
'filepath': instance_of( str ),
'description': 'Namespace: other',
} ),
has_entries( {
'line_num': 3,
'column_num': 1,
'filepath': instance_of( str ),
'description': 'Namespace: testy',
} ),
has_entries( {
'line_num': 5,
'column_num': 2,
'filepath': instance_of( str ),
'description': 'Class: MainClass',
} )
) )


@IsolatedYcmd( { 'global_ycm_extra_conf':
PathToTestFile( 'extra_confs', 'settings_extra_conf.py' ) } )
def test_LanguageServerCompleter_ExtraConf_ServerReset( self, app ):
Expand Down

0 comments on commit 5de7052

Please sign in to comment.