1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039 |
- # Copyright (C) 2011-2024 YouCompleteMe contributors
- #
- # This file is part of YouCompleteMe.
- #
- # YouCompleteMe is free software: you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation, either version 3 of the License, or
- # (at your option) any later version.
- #
- # YouCompleteMe is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with YouCompleteMe. If not, see <http://www.gnu.org/licenses/>.
- import base64
- import json
- import logging
- import os
- import signal
- import vim
- from subprocess import PIPE
- from tempfile import NamedTemporaryFile
- from ycm import base, paths, signature_help, vimsupport
- from ycm.buffer import BufferDict
- from ycmd import utils
- from ycmd.request_wrap import RequestWrap
- from ycm.omni_completer import OmniCompleter
- from ycm import syntax_parse
- from ycm.hierarchy_tree import HierarchyTree
- from ycm.client.ycmd_keepalive import YcmdKeepalive
- from ycm.client.base_request import BaseRequest, BuildRequestData
- from ycm.client.completer_available_request import SendCompleterAvailableRequest
- from ycm.client.command_request import ( SendCommandRequest,
- SendCommandRequestAsync,
- GetCommandResponse,
- GetRawCommandResponse )
- from ycm.client.completion_request import CompletionRequest
- from ycm.client.resolve_completion_request import ResolveCompletionItem
- from ycm.client.signature_help_request import ( SignatureHelpRequest,
- SigHelpAvailableByFileType )
- from ycm.client.debug_info_request import ( SendDebugInfoRequest,
- FormatDebugInfoResponse )
- from ycm.client.omni_completion_request import OmniCompletionRequest
- from ycm.client.event_notification import SendEventNotificationAsync
- from ycm.client.shutdown_request import SendShutdownRequest
- from ycm.client.messages_request import MessagesPoll
- def PatchNoProxy():
- current_value = os.environ.get( 'no_proxy', '' )
- additions = '127.0.0.1,localhost'
- os.environ[ 'no_proxy' ] = ( additions if not current_value
- else current_value + ',' + additions )
- # We need this so that Requests doesn't end up using the local HTTP proxy when
- # talking to ycmd. Users should actually be setting this themselves when
- # configuring a proxy server on their machine, but most don't know they need to
- # or how to do it, so we do it for them.
- # Relevant issues:
- # https://github.com/Valloric/YouCompleteMe/issues/641
- # https://github.com/kennethreitz/requests/issues/879
- PatchNoProxy()
- # Force the Python interpreter embedded in Vim (in which we are running) to
- # ignore the SIGINT signal. This helps reduce the fallout of a user pressing
- # Ctrl-C in Vim.
- signal.signal( signal.SIGINT, signal.SIG_IGN )
- HMAC_SECRET_LENGTH = 16
- SERVER_SHUTDOWN_MESSAGE = (
- "The ycmd server SHUT DOWN (restart with ':YcmRestartServer')." )
- EXIT_CODE_UNEXPECTED_MESSAGE = (
- "Unexpected exit code {code}. "
- "Type ':YcmToggleLogs {logfile}' to check the logs." )
- CORE_UNEXPECTED_MESSAGE = (
- "Unexpected error while loading the YCM core library. "
- "Type ':YcmToggleLogs {logfile}' to check the logs." )
- CORE_MISSING_MESSAGE = (
- 'YCM core library not detected; you need to compile YCM before using it. '
- 'Follow the instructions in the documentation.' )
- CORE_OUTDATED_MESSAGE = (
- 'YCM core library too old; PLEASE RECOMPILE by running the install.py '
- 'script. See the documentation for more details.' )
- PYTHON_TOO_OLD_MESSAGE = (
- 'Your python is too old to run YCM server. '
- 'Please see troubleshooting guide on YCM GitHub wiki.'
- )
- SERVER_IDLE_SUICIDE_SECONDS = 1800 # 30 minutes
- CLIENT_LOGFILE_FORMAT = 'ycm_'
- SERVER_LOGFILE_FORMAT = 'ycmd_{port}_{std}_'
- # Flag to set a file handle inheritable by child processes on Windows. See
- # https://msdn.microsoft.com/en-us/library/ms724935.aspx
- HANDLE_FLAG_INHERIT = 0x00000001
- class YouCompleteMe:
- def __init__( self, default_options = {} ):
- self._logger = logging.getLogger( 'ycm' )
- self._client_logfile = None
- self._server_stdout = None
- self._server_stderr = None
- self._server_popen = None
- self._default_options = default_options
- self._ycmd_keepalive = YcmdKeepalive()
- self._SetUpLogging()
- self._SetUpServer()
- self._ycmd_keepalive.Start()
- self._current_hierarchy = HierarchyTree()
- def InitializeCurrentHierarchy( self, items, kind ):
- return self._current_hierarchy.SetRootNode( items, kind )
- def UpdateCurrentHierarchy( self, handle : int, direction : str ):
- if not self._current_hierarchy.UpdateChangesRoot( handle, direction ):
- items = self._ResolveHierarchyItem( handle, direction )
- self._current_hierarchy.UpdateHierarchy( handle, items, direction )
- if items is not None and direction == 'up':
- offset = sum( len( item[ 'locations' ] ) for item in items )
- else:
- offset = 0
- return self._current_hierarchy.HierarchyToLines(), offset
- else:
- location = self._current_hierarchy.HandleToRootLocation( handle )
- kind = self._current_hierarchy._kind
- self._current_hierarchy.Reset()
- items = GetRawCommandResponse(
- [ f'{ kind.title() }Hierarchy' ],
- silent = False,
- location = location
- )
- # [ 0 ] chooses the data for the 1st (and only) line.
- # [ 1 ] chooses only the handle
- handle = self.InitializeCurrentHierarchy( items, kind )[ 0 ][ 1 ]
- return self.UpdateCurrentHierarchy( handle, direction )
- def _ResolveHierarchyItem( self, handle : int, direction : str ):
- return GetRawCommandResponse(
- self._current_hierarchy.ResolveArguments( handle, direction ),
- silent = False
- )
- def ShouldResolveItem( self, handle : int, direction : str ):
- return self._current_hierarchy.ShouldResolveItem( handle, direction )
- def ResetCurrentHierarchy( self ):
- self._current_hierarchy.Reset()
- def JumpToHierarchyItem( self, handle ):
- self._current_hierarchy.JumpToItem(
- handle,
- self._user_options[ 'goto_buffer_command' ] )
- def _SetUpServer( self ):
- self._available_completers = {}
- self._user_notified_about_crash = False
- self._filetypes_with_keywords_loaded = set()
- self._server_is_ready_with_cache = False
- self._message_poll_requests = {}
- self._latest_completion_request = None
- self._latest_signature_help_request = None
- self._signature_help_available_requests = SigHelpAvailableByFileType()
- self._command_requests = {}
- self._next_command_request_id = 0
- self._signature_help_state = signature_help.SignatureHelpState()
- self._user_options = base.GetUserOptions( self._default_options )
- self._omnicomp = OmniCompleter( self._user_options )
- self._buffers = BufferDict( self._user_options )
- self._SetLogLevel()
- hmac_secret = os.urandom( HMAC_SECRET_LENGTH )
- options_dict = dict( self._user_options )
- options_dict[ 'hmac_secret' ] = utils.ToUnicode(
- base64.b64encode( hmac_secret ) )
- options_dict[ 'server_keep_logfiles' ] = self._user_options[
- 'keep_logfiles' ]
- # The temp options file is deleted by ycmd during startup.
- with NamedTemporaryFile( delete = False, mode = 'w+' ) as options_file:
- json.dump( options_dict, options_file )
- server_port = utils.GetUnusedLocalhostPort()
- BaseRequest.server_location = 'http://127.0.0.1:' + str( server_port )
- BaseRequest.hmac_secret = hmac_secret
- try:
- python_interpreter = paths.PathToPythonInterpreter()
- except RuntimeError as error:
- error_message = (
- f"Unable to start the ycmd server. { str( error ).rstrip( '.' ) }. "
- "Correct the error then restart the server "
- "with ':YcmRestartServer'." )
- self._logger.exception( error_message )
- vimsupport.PostVimMessage( error_message )
- return
- args = [ python_interpreter,
- paths.PathToServerScript(),
- f'--port={ server_port }',
- f'--options_file={ options_file.name }',
- f'--log={ self._user_options[ "log_level" ] }',
- f'--idle_suicide_seconds={ SERVER_IDLE_SUICIDE_SECONDS }' ]
- self._server_stdout = utils.CreateLogfile(
- SERVER_LOGFILE_FORMAT.format( port = server_port, std = 'stdout' ) )
- self._server_stderr = utils.CreateLogfile(
- SERVER_LOGFILE_FORMAT.format( port = server_port, std = 'stderr' ) )
- args.append( f'--stdout={ self._server_stdout }' )
- args.append( f'--stderr={ self._server_stderr }' )
- if self._user_options[ 'keep_logfiles' ]:
- args.append( '--keep_logfiles' )
- self._server_popen = utils.SafePopen( args, stdin_windows = PIPE,
- stdout = PIPE, stderr = PIPE )
- def _SetUpLogging( self ):
- def FreeFileFromOtherProcesses( file_object ):
- if utils.OnWindows():
- from ctypes import windll
- import msvcrt
- file_handle = msvcrt.get_osfhandle( file_object.fileno() )
- windll.kernel32.SetHandleInformation( file_handle,
- HANDLE_FLAG_INHERIT,
- 0 )
- self._client_logfile = utils.CreateLogfile( CLIENT_LOGFILE_FORMAT )
- handler = logging.FileHandler( self._client_logfile )
- # On Windows and Python prior to 3.4, file handles are inherited by child
- # processes started with at least one replaced standard stream, which is the
- # case when we start the ycmd server (we are redirecting all standard
- # outputs into a pipe). These files cannot be removed while the child
- # processes are still up. This is not desirable for a logfile because we
- # want to remove it at Vim exit without having to wait for the ycmd server
- # to be completely shut down. We need to make the logfile handle
- # non-inheritable. See https://www.python.org/dev/peps/pep-0446 for more
- # details.
- FreeFileFromOtherProcesses( handler.stream )
- formatter = logging.Formatter( '%(asctime)s - %(levelname)s - %(message)s' )
- handler.setFormatter( formatter )
- self._logger.addHandler( handler )
- def _SetLogLevel( self ):
- log_level = self._user_options[ 'log_level' ]
- numeric_level = getattr( logging, log_level.upper(), None )
- if not isinstance( numeric_level, int ):
- raise ValueError( f'Invalid log level: { log_level }' )
- self._logger.setLevel( numeric_level )
- def IsServerAlive( self ):
- # When the process hasn't finished yet, poll() returns None.
- return bool( self._server_popen ) and self._server_popen.poll() is None
- def CheckIfServerIsReady( self ):
- if not self._server_is_ready_with_cache and self.IsServerAlive():
- self._server_is_ready_with_cache = BaseRequest().GetDataFromHandler(
- 'ready', display_message = False )
- return self._server_is_ready_with_cache
- def IsServerReady( self ):
- return self._server_is_ready_with_cache
- def NotifyUserIfServerCrashed( self ):
- if ( not self._server_popen or self._user_notified_about_crash or
- self.IsServerAlive() ):
- return
- self._user_notified_about_crash = True
- return_code = self._server_popen.poll()
- logfile = os.path.basename( self._server_stderr )
- # See https://github.com/Valloric/ycmd#exit-codes for the list of exit
- # codes.
- if return_code == 3:
- error_message = CORE_UNEXPECTED_MESSAGE.format( logfile = logfile )
- elif return_code == 4:
- error_message = CORE_MISSING_MESSAGE
- elif return_code == 7:
- error_message = CORE_OUTDATED_MESSAGE
- elif return_code == 8:
- # TODO: here we could retry but discard g:ycm_server_python_interpreter
- error_message = PYTHON_TOO_OLD_MESSAGE
- else:
- error_message = EXIT_CODE_UNEXPECTED_MESSAGE.format( code = return_code,
- logfile = logfile )
- if return_code != 8:
- error_message = SERVER_SHUTDOWN_MESSAGE + ' ' + error_message
- self._logger.error( error_message )
- vimsupport.PostVimMessage( error_message )
- def ServerPid( self ):
- if not self._server_popen:
- return -1
- return self._server_popen.pid
- def _ShutdownServer( self ):
- SendShutdownRequest()
- def RestartServer( self ):
- vimsupport.PostVimMessage( 'Restarting ycmd server...' )
- self._ShutdownServer()
- self._SetUpServer()
- def SendCompletionRequest( self, force_semantic = False ):
- request_data = BuildRequestData()
- request_data[ 'force_semantic' ] = force_semantic
- if not self.NativeFiletypeCompletionUsable():
- wrapped_request_data = RequestWrap( request_data )
- if self._omnicomp.ShouldUseNow( wrapped_request_data ):
- self._latest_completion_request = OmniCompletionRequest(
- self._omnicomp, wrapped_request_data )
- self._latest_completion_request.Start()
- return
- self._AddExtraConfDataIfNeeded( request_data )
- self._latest_completion_request = CompletionRequest( request_data )
- self._latest_completion_request.Start()
- def CompletionRequestReady( self ):
- return bool( self._latest_completion_request and
- self._latest_completion_request.Done() )
- def GetCompletionResponse( self ):
- return self._latest_completion_request.Response()
- def SignatureHelpAvailableRequestComplete( self, filetype, send_new=True ):
- """Triggers or polls signature help available request. Returns whether or
- not the request is complete. When send_new is False, won't send a new
- request, only return the current status (This is used by the tests)"""
- if not send_new and filetype not in self._signature_help_available_requests:
- return False
- return self._signature_help_available_requests[ filetype ].Done()
- def SendSignatureHelpRequest( self ):
- """Send a signature help request, if we're ready to. Return whether or not a
- request was sent (and should be checked later)"""
- if not self.NativeFiletypeCompletionUsable():
- return False
- for filetype in vimsupport.CurrentFiletypes():
- if not self.SignatureHelpAvailableRequestComplete( filetype ):
- continue
- sig_help_available = self._signature_help_available_requests[
- filetype ].Response()
- if sig_help_available == 'NO':
- continue
- if sig_help_available == 'PENDING':
- # Send another /signature_help_available request
- self._signature_help_available_requests[ filetype ].Start( filetype )
- continue
- if not self._latest_completion_request:
- return False
- request_data = self._latest_completion_request.request_data.copy()
- request_data[ 'signature_help_state' ] = (
- self._signature_help_state.IsActive()
- )
- self._AddExtraConfDataIfNeeded( request_data )
- self._latest_signature_help_request = SignatureHelpRequest( request_data )
- self._latest_signature_help_request.Start()
- return True
- return False
- def SignatureHelpRequestReady( self ):
- return bool( self._latest_signature_help_request and
- self._latest_signature_help_request.Done() )
- def GetSignatureHelpResponse( self ):
- return self._latest_signature_help_request.Response()
- def ClearSignatureHelp( self ):
- self.UpdateSignatureHelp( {} )
- if self._latest_signature_help_request:
- self._latest_signature_help_request.Reset()
- def UpdateSignatureHelp( self, signature_info ):
- self._signature_help_state = signature_help.UpdateSignatureHelp(
- self._signature_help_state,
- signature_info )
- def _GetCommandRequestArguments( self,
- arguments,
- has_range,
- start_line,
- end_line ):
- extra_data = {
- 'options': {
- 'tab_size': vimsupport.GetIntValue( 'shiftwidth()' ),
- 'insert_spaces': vimsupport.GetBoolValue( '&expandtab' )
- }
- }
- final_arguments = []
- for argument in arguments:
- if argument.startswith( 'ft=' ):
- extra_data[ 'completer_target' ] = argument[ 3: ]
- continue
- elif argument.startswith( '--bufnr=' ):
- extra_data[ 'bufnr' ] = int( argument[ len( '--bufnr=' ): ] )
- continue
- final_arguments.append( argument )
- if has_range:
- extra_data.update( vimsupport.BuildRange( start_line, end_line ) )
- self._AddExtraConfDataIfNeeded( extra_data )
- return final_arguments, extra_data
- def SendCommandRequest( self,
- arguments,
- modifiers,
- has_range,
- start_line,
- end_line ):
- final_arguments, extra_data = self._GetCommandRequestArguments(
- arguments,
- has_range,
- start_line,
- end_line )
- return SendCommandRequest(
- final_arguments,
- modifiers,
- self._user_options[ 'goto_buffer_command' ],
- extra_data )
- def GetCommandResponse( self, arguments ):
- final_arguments, extra_data = self._GetCommandRequestArguments(
- arguments,
- False,
- 0,
- 0 )
- return GetCommandResponse( final_arguments, extra_data )
- def SendCommandRequestAsync( self,
- arguments,
- silent = True,
- location = None ):
- final_arguments, extra_data = self._GetCommandRequestArguments(
- arguments,
- False,
- 0,
- 0 )
- request_id = self._next_command_request_id
- self._next_command_request_id += 1
- self._command_requests[ request_id ] = SendCommandRequestAsync(
- final_arguments,
- extra_data,
- silent,
- location = location )
- return request_id
- def GetCommandRequest( self, request_id ):
- return self._command_requests.get( request_id )
- def FlushCommandRequest( self, request_id ):
- self._command_requests.pop( request_id, None )
- def GetDefinedSubcommands( self ):
- request = BaseRequest()
- subcommands = request.PostDataToHandler( BuildRequestData(),
- 'defined_subcommands' )
- return subcommands if subcommands else []
- def GetCurrentCompletionRequest( self ):
- return self._latest_completion_request
- def GetOmniCompleter( self ):
- return self._omnicomp
- def FiletypeCompleterExistsForFiletype( self, filetype ):
- try:
- return self._available_completers[ filetype ]
- except KeyError:
- pass
- exists_completer = SendCompleterAvailableRequest( filetype )
- if exists_completer is None:
- return False
- self._available_completers[ filetype ] = exists_completer
- return exists_completer
- def NativeFiletypeCompletionAvailable( self ):
- return any( self.FiletypeCompleterExistsForFiletype( x ) for x in
- vimsupport.CurrentFiletypes() )
- def NativeFiletypeCompletionUsable( self ):
- disabled_filetypes = self._user_options[
- 'filetype_specific_completion_to_disable' ]
- return ( vimsupport.CurrentFiletypesEnabled( disabled_filetypes ) and
- self.NativeFiletypeCompletionAvailable() )
- def NeedsReparse( self ):
- return self.CurrentBuffer().NeedsReparse()
- def UpdateWithNewDiagnosticsForFile( self, filepath, diagnostics ):
- if not self._user_options[ 'show_diagnostics_ui' ]:
- return
- bufnr = vimsupport.GetBufferNumberForFilename( filepath )
- if bufnr in self._buffers and vimsupport.BufferIsVisible( bufnr ):
- # Note: We only update location lists, etc. for visible buffers, because
- # otherwise we default to using the current location list and the results
- # are that non-visible buffer errors clobber visible ones.
- self._buffers[ bufnr ].UpdateWithNewDiagnostics( diagnostics, True )
- else:
- # The project contains errors in file "filepath", but that file is not
- # open in any buffer. This happens for Language Server Protocol-based
- # completers, as they return diagnostics for the entire "project"
- # asynchronously (rather than per-file in the response to the parse
- # request).
- #
- # There are a number of possible approaches for
- # this, but for now we simply ignore them. Other options include:
- # - Use the QuickFix list to report project errors?
- # - Use a special buffer for project errors
- # - Put them in the location list of whatever the "current" buffer is
- # - Store them in case the buffer is opened later
- # - add a :YcmProjectDiags command
- # - Add them to errror/warning _counts_ but not any actual location list
- # or other
- # - etc.
- #
- # However, none of those options are great, and lead to their own
- # complexities. So for now, we just ignore these diagnostics for files not
- # open in any buffer.
- pass
- def OnPeriodicTick( self ):
- if not self.IsServerAlive():
- # Server has died. We'll reset when the server is started again.
- return False
- elif not self.IsServerReady():
- # Try again in a jiffy
- return True
- for w in vim.windows:
- for filetype in vimsupport.FiletypesForBuffer( w.buffer ):
- if filetype not in self._message_poll_requests:
- self._message_poll_requests[ filetype ] = MessagesPoll( w.buffer )
- # None means don't poll this filetype
- if ( self._message_poll_requests[ filetype ] and
- not self._message_poll_requests[ filetype ].Poll( self ) ):
- self._message_poll_requests[ filetype ] = None
- return any( self._message_poll_requests.values() )
- def OnFileReadyToParse( self ):
- if not self.IsServerAlive():
- self.NotifyUserIfServerCrashed()
- return
- if not self.IsServerReady():
- return
- extra_data = {}
- self._AddTagsFilesIfNeeded( extra_data )
- self._AddSyntaxDataIfNeeded( extra_data )
- self._AddExtraConfDataIfNeeded( extra_data )
- self.CurrentBuffer().SendParseRequest( extra_data )
- def OnFileSave( self, saved_buffer_number ):
- SendEventNotificationAsync( 'FileSave', saved_buffer_number )
- def OnBufferUnload( self, deleted_buffer_number ):
- SendEventNotificationAsync( 'BufferUnload', deleted_buffer_number )
- def UpdateMatches( self ):
- self.CurrentBuffer().UpdateMatches()
- def OnFileTypeSet( self ):
- buffer_number = vimsupport.GetCurrentBufferNumber()
- filetypes = vimsupport.CurrentFiletypes()
- self._buffers[ buffer_number ].UpdateFromFileTypes( filetypes )
- self.OnBufferVisit()
- def OnBufferVisit( self ):
- for filetype in vimsupport.CurrentFiletypes():
- # Send the signature help available request for these filetypes if we need
- # to (as a side effect of checking if it is complete)
- self.SignatureHelpAvailableRequestComplete( filetype, True )
- extra_data = {}
- self._AddUltiSnipsDataIfNeeded( extra_data )
- SendEventNotificationAsync( 'BufferVisit', extra_data = extra_data )
- def CurrentBuffer( self ):
- return self.Buffer( vimsupport.GetCurrentBufferNumber() )
- def Buffer( self, bufnr ):
- return self._buffers[ bufnr ]
- def OnInsertEnter( self ):
- if not self._user_options[ 'update_diagnostics_in_insert_mode' ]:
- self.CurrentBuffer().ClearDiagnosticsUI()
- def OnInsertLeave( self ):
- async_diags = any( self._message_poll_requests.get( filetype )
- for filetype in vimsupport.CurrentFiletypes() )
- if ( not self._user_options[ 'update_diagnostics_in_insert_mode' ] and
- ( async_diags or not self.CurrentBuffer().ParseRequestPending() ) ):
- self.CurrentBuffer().RefreshDiagnosticsUI()
- SendEventNotificationAsync( 'InsertLeave' )
- def OnCursorMoved( self ):
- self.CurrentBuffer().OnCursorMoved()
- def _CleanLogfile( self ):
- logging.shutdown()
- if not self._user_options[ 'keep_logfiles' ]:
- if self._client_logfile:
- utils.RemoveIfExists( self._client_logfile )
- def OnVimLeave( self ):
- self._ShutdownServer()
- self._CleanLogfile()
- def OnCurrentIdentifierFinished( self ):
- SendEventNotificationAsync( 'CurrentIdentifierFinished' )
- def OnCompleteDone( self ):
- completion_request = self.GetCurrentCompletionRequest()
- if completion_request:
- completion_request.OnCompleteDone()
- def ResolveCompletionItem( self, item ):
- # Note: As mentioned elsewhere, we replace the current completion request
- # with a resolve request. It's not valid to have simultaneous resolve and
- # completion requests, because the resolve request uses the request data
- # from the last completion request and is therefore dependent on it not
- # having changed.
- #
- # The result of this is that self.GetCurrentCompletionRequest() might return
- # either a completion request of a resolve request and it's the
- # responsibility of the vimscript code to ensure that it only does one at a
- # time. This is handled by re-using the same poller for completions and
- # resolves.
- completion_request = self.GetCurrentCompletionRequest()
- if not completion_request:
- return False
- request = ResolveCompletionItem( completion_request, item )
- if not request:
- return False
- self._latest_completion_request = request
- return True
- def GetErrorCount( self ):
- return self.CurrentBuffer().GetErrorCount()
- def GetWarningCount( self ):
- return self.CurrentBuffer().GetWarningCount()
- def _PopulateLocationListWithLatestDiagnostics( self ):
- return self.CurrentBuffer().PopulateLocationList(
- self._user_options[ 'open_loclist_on_ycm_diags' ] )
- def FileParseRequestReady( self ):
- # Return True if server is not ready yet, to stop repeating check timer.
- return ( not self.IsServerReady() or
- self.CurrentBuffer().FileParseRequestReady() )
- def HandleFileParseRequest( self, block = False ):
- if not self.IsServerReady():
- return
- current_buffer = self.CurrentBuffer()
- # Order is important here:
- # FileParseRequestReady has a low cost, while
- # NativeFiletypeCompletionUsable is a blocking server request
- if ( not current_buffer.IsResponseHandled() and
- current_buffer.FileParseRequestReady( block ) and
- self.NativeFiletypeCompletionUsable() ):
- if self._user_options[ 'show_diagnostics_ui' ]:
- # Forcefuly update the location list, etc. from the parse request when
- # doing something like :YcmDiags
- async_diags = any( self._message_poll_requests.get( filetype )
- for filetype in vimsupport.CurrentFiletypes() )
- current_buffer.UpdateDiagnostics( block or not async_diags )
- else:
- # If the user disabled diagnostics, we just want to check
- # the _latest_file_parse_request for any exception or UnknownExtraConf
- # response, to allow the server to raise configuration warnings, etc.
- # to the user. We ignore any other supplied data.
- current_buffer.GetResponse()
- # We set the file parse request as handled because we want to prevent
- # repeated issuing of the same warnings/errors/prompts. Setting this
- # makes IsRequestHandled return True until the next request is created.
- #
- # Note: it is the server's responsibility to determine the frequency of
- # error/warning/prompts when receiving a FileReadyToParse event, but
- # it is our responsibility to ensure that we only apply the
- # warning/error/prompt received once (for each event).
- current_buffer.MarkResponseHandled()
- def ShouldResendFileParseRequest( self ):
- return self.CurrentBuffer().ShouldResendParseRequest()
- def DebugInfo( self ):
- debug_info = ''
- if self._client_logfile:
- debug_info += f'Client logfile: { self._client_logfile }\n'
- extra_data = {}
- self._AddExtraConfDataIfNeeded( extra_data )
- debug_info += FormatDebugInfoResponse( SendDebugInfoRequest( extra_data ) )
- debug_info += f'Server running at: { BaseRequest.server_location }\n'
- if self._server_popen:
- debug_info += f'Server process ID: { self._server_popen.pid }\n'
- if self._server_stdout and self._server_stderr:
- debug_info += ( 'Server logfiles:\n'
- f' { self._server_stdout }\n'
- f' { self._server_stderr }' )
- debug_info += ( '\nSemantic highlighting supported: ' +
- str( not vimsupport.VimIsNeovim() ) )
- debug_info += ( '\nVirtual text supported: ' +
- str( not vimsupport.VimIsNeovim() ) )
- debug_info += ( '\nPopup windows supported: ' +
- str( vimsupport.VimSupportsPopupWindows() ) )
- return debug_info
- def GetLogfiles( self ):
- logfiles_list = [ self._client_logfile,
- self._server_stdout,
- self._server_stderr ]
- extra_data = {}
- self._AddExtraConfDataIfNeeded( extra_data )
- debug_info = SendDebugInfoRequest( extra_data )
- if debug_info:
- completer = debug_info[ 'completer' ]
- if completer:
- for server in completer[ 'servers' ]:
- logfiles_list.extend( server[ 'logfiles' ] )
- logfiles = {}
- for logfile in logfiles_list:
- logfiles[ os.path.basename( logfile ) ] = logfile
- return logfiles
- def _OpenLogfile( self, size, mods, logfile ):
- # Open log files in a horizontal window with the same behavior as the
- # preview window (same height and winfixheight enabled). Automatically
- # watch for changes. Set the cursor position at the end of the file.
- if not size:
- size = vimsupport.GetIntValue( '&previewheight' )
- options = {
- 'size': size,
- 'fix': True,
- 'focus': False,
- 'watch': True,
- 'position': 'end',
- 'mods': mods
- }
- vimsupport.OpenFilename( logfile, options )
- def _CloseLogfile( self, logfile ):
- vimsupport.CloseBuffersForFilename( logfile )
- def ToggleLogs( self, size, mods, *filenames ):
- logfiles = self.GetLogfiles()
- if not filenames:
- sorted_logfiles = sorted( logfiles )
- try:
- logfile_index = vimsupport.SelectFromList(
- 'Which logfile do you wish to open (or close if already open)?',
- sorted_logfiles )
- except RuntimeError as e:
- vimsupport.PostVimMessage( str( e ) )
- return
- logfile = logfiles[ sorted_logfiles[ logfile_index ] ]
- if not vimsupport.BufferIsVisibleForFilename( logfile ):
- self._OpenLogfile( size, mods, logfile )
- else:
- self._CloseLogfile( logfile )
- return
- for filename in set( filenames ):
- if filename not in logfiles:
- continue
- logfile = logfiles[ filename ]
- if not vimsupport.BufferIsVisibleForFilename( logfile ):
- self._OpenLogfile( size, mods, logfile )
- continue
- self._CloseLogfile( logfile )
- def ShowDetailedDiagnostic( self, message_in_popup ):
- detailed_diagnostic = BaseRequest().PostDataToHandler(
- BuildRequestData(), 'detailed_diagnostic' )
- if detailed_diagnostic and 'message' in detailed_diagnostic:
- message = detailed_diagnostic[ 'message' ]
- if message_in_popup and vimsupport.VimSupportsPopupWindows():
- window = vim.current.window
- buffer_number = vimsupport.GetCurrentBufferNumber()
- diags_on_this_line = self._buffers[ buffer_number ].DiagnosticsForLine(
- window.cursor[ 0 ] )
- lines = message.split( '\n' )
- available_columns = vimsupport.GetIntValue( '&columns' )
- col = window.cursor[ 1 ] + 1
- if col > available_columns - 2: # -2 accounts for padding.
- col = 0
- options = {
- 'col': col,
- 'padding': [ 0, 1, 0, 1 ],
- 'maxwidth': available_columns,
- 'close': 'click',
- 'fixed': 0,
- 'highlight': 'YcmErrorPopup',
- 'border': [ 1, 1, 1, 1 ],
- # Close when moving cursor
- 'moved': 'expr',
- }
- popup_func = 'popup_atcursor'
- for diag in diags_on_this_line:
- if message == diag[ 'text' ]:
- popup_func = 'popup_create'
- prop = vimsupport.GetTextPropertyForDiag( buffer_number,
- window.cursor[ 0 ],
- diag )
- options.update( {
- 'textpropid': prop[ 'id' ],
- 'textprop': prop[ 'type' ],
- } )
- options.pop( 'col' )
- break
- vim.eval( f'{ popup_func }( { json.dumps( lines ) }, '
- f'{ json.dumps( options ) } )' )
- else:
- vimsupport.PostVimMessage( message, warning = False )
- def ForceCompileAndDiagnostics( self ):
- if not self.NativeFiletypeCompletionUsable():
- vimsupport.PostVimMessage(
- 'Native filetype completion not supported for current file, '
- 'cannot force recompilation.', warning = False )
- return False
- vimsupport.PostVimMessage(
- 'Forcing compilation, this will block Vim until done.',
- warning = False )
- self.OnFileReadyToParse()
- self.HandleFileParseRequest( block = True )
- vimsupport.PostVimMessage( 'Diagnostics refreshed', warning = False )
- return True
- def ShowDiagnostics( self ):
- if not self.ForceCompileAndDiagnostics():
- return
- if not self._PopulateLocationListWithLatestDiagnostics():
- vimsupport.PostVimMessage( 'No warnings or errors detected.',
- warning = False )
- return
- if self._user_options[ 'open_loclist_on_ycm_diags' ]:
- vimsupport.OpenLocationList( focus = True )
- def FilterAndSortItems( self,
- items,
- sort_property,
- query,
- max_items = 0 ):
- return BaseRequest().PostDataToHandler( {
- 'candidates': items,
- 'sort_property': sort_property,
- 'max_num_candidates': max_items,
- 'query': vimsupport.ToUnicode( query )
- }, 'filter_and_sort_candidates' )
- def ToggleSignatureHelp( self ):
- self._signature_help_state.ToggleVisibility()
- def _AddSyntaxDataIfNeeded( self, extra_data ):
- if not self._user_options[ 'seed_identifiers_with_syntax' ]:
- return
- filetype = vimsupport.CurrentFiletypes()[ 0 ]
- if filetype in self._filetypes_with_keywords_loaded:
- return
- if self.IsServerReady():
- self._filetypes_with_keywords_loaded.add( filetype )
- extra_data[ 'syntax_keywords' ] = list(
- syntax_parse.SyntaxKeywordsForCurrentBuffer() )
- def _AddTagsFilesIfNeeded( self, extra_data ):
- def GetTagFiles():
- tag_files = vim.eval( 'tagfiles()' )
- return [ ( os.path.join( utils.GetCurrentDirectory(), tag_file )
- if not os.path.isabs( tag_file ) else tag_file )
- for tag_file in tag_files ]
- if not self._user_options[ 'collect_identifiers_from_tags_files' ]:
- return
- extra_data[ 'tag_files' ] = GetTagFiles()
- def _AddExtraConfDataIfNeeded( self, extra_data ):
- def BuildExtraConfData( extra_conf_vim_data ):
- extra_conf_data = {}
- for expr in extra_conf_vim_data:
- try:
- extra_conf_data[ expr ] = vimsupport.VimExpressionToPythonType( expr )
- except vim.error:
- message = (
- f"Error evaluating '{ expr }' in the 'g:ycm_extra_conf_vim_data' "
- "option." )
- vimsupport.PostVimMessage( message, truncate = True )
- self._logger.exception( message )
- return extra_conf_data
- extra_conf_vim_data = self._user_options[ 'extra_conf_vim_data' ]
- if extra_conf_vim_data:
- extra_data[ 'extra_conf_data' ] = BuildExtraConfData(
- extra_conf_vim_data )
- def _AddUltiSnipsDataIfNeeded( self, extra_data ):
- # See :h UltiSnips#SnippetsInCurrentScope.
- try:
- vim.eval( 'UltiSnips#SnippetsInCurrentScope( 1 )' )
- except vim.error:
- return
- snippets = vimsupport.GetVariableValue( 'g:current_ulti_dict_info' )
- extra_data[ 'ultisnips_snippets' ] = [
- { 'trigger': trigger,
- 'description': snippet[ 'description' ] }
- for trigger, snippet in snippets.items()
- ]
|