youcompleteme.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809
  1. # Copyright (C) 2011-2018 YouCompleteMe contributors
  2. #
  3. # This file is part of YouCompleteMe.
  4. #
  5. # YouCompleteMe is free software: you can redistribute it and/or modify
  6. # it under the terms of the GNU General Public License as published by
  7. # the Free Software Foundation, either version 3 of the License, or
  8. # (at your option) any later version.
  9. #
  10. # YouCompleteMe is distributed in the hope that it will be useful,
  11. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. # GNU General Public License for more details.
  14. #
  15. # You should have received a copy of the GNU General Public License
  16. # along with YouCompleteMe. If not, see <http://www.gnu.org/licenses/>.
  17. import base64
  18. import json
  19. import logging
  20. import os
  21. import signal
  22. import vim
  23. from subprocess import PIPE
  24. from tempfile import NamedTemporaryFile
  25. from ycm import base, paths, signature_help, vimsupport
  26. from ycm.buffer import BufferDict
  27. from ycmd import utils
  28. from ycmd.request_wrap import RequestWrap
  29. from ycm.omni_completer import OmniCompleter
  30. from ycm import syntax_parse
  31. from ycm.client.ycmd_keepalive import YcmdKeepalive
  32. from ycm.client.base_request import BaseRequest, BuildRequestData
  33. from ycm.client.completer_available_request import SendCompleterAvailableRequest
  34. from ycm.client.command_request import SendCommandRequest
  35. from ycm.client.completion_request import CompletionRequest
  36. from ycm.client.signature_help_request import SignatureHelpRequest
  37. from ycm.client.signature_help_request import SigHelpAvailableByFileType
  38. from ycm.client.debug_info_request import ( SendDebugInfoRequest,
  39. FormatDebugInfoResponse )
  40. from ycm.client.omni_completion_request import OmniCompletionRequest
  41. from ycm.client.event_notification import SendEventNotificationAsync
  42. from ycm.client.shutdown_request import SendShutdownRequest
  43. from ycm.client.messages_request import MessagesPoll
  44. def PatchNoProxy():
  45. current_value = os.environ.get( 'no_proxy', '' )
  46. additions = '127.0.0.1,localhost'
  47. os.environ[ 'no_proxy' ] = ( additions if not current_value
  48. else current_value + ',' + additions )
  49. # We need this so that Requests doesn't end up using the local HTTP proxy when
  50. # talking to ycmd. Users should actually be setting this themselves when
  51. # configuring a proxy server on their machine, but most don't know they need to
  52. # or how to do it, so we do it for them.
  53. # Relevant issues:
  54. # https://github.com/Valloric/YouCompleteMe/issues/641
  55. # https://github.com/kennethreitz/requests/issues/879
  56. PatchNoProxy()
  57. # Force the Python interpreter embedded in Vim (in which we are running) to
  58. # ignore the SIGINT signal. This helps reduce the fallout of a user pressing
  59. # Ctrl-C in Vim.
  60. signal.signal( signal.SIGINT, signal.SIG_IGN )
  61. HMAC_SECRET_LENGTH = 16
  62. SERVER_SHUTDOWN_MESSAGE = (
  63. "The ycmd server SHUT DOWN (restart with ':YcmRestartServer')." )
  64. EXIT_CODE_UNEXPECTED_MESSAGE = (
  65. "Unexpected exit code {code}. "
  66. "Type ':YcmToggleLogs {logfile}' to check the logs." )
  67. CORE_UNEXPECTED_MESSAGE = (
  68. "Unexpected error while loading the YCM core library. "
  69. "Type ':YcmToggleLogs {logfile}' to check the logs." )
  70. CORE_MISSING_MESSAGE = (
  71. 'YCM core library not detected; you need to compile YCM before using it. '
  72. 'Follow the instructions in the documentation.' )
  73. CORE_OUTDATED_MESSAGE = (
  74. 'YCM core library too old; PLEASE RECOMPILE by running the install.py '
  75. 'script. See the documentation for more details.' )
  76. NO_PYTHON2_SUPPORT_MESSAGE = (
  77. 'YCM has dropped support for python2. '
  78. 'You need to recompile it with python3 instead.' )
  79. SERVER_IDLE_SUICIDE_SECONDS = 1800 # 30 minutes
  80. CLIENT_LOGFILE_FORMAT = 'ycm_'
  81. SERVER_LOGFILE_FORMAT = 'ycmd_{port}_{std}_'
  82. # Flag to set a file handle inheritable by child processes on Windows. See
  83. # https://msdn.microsoft.com/en-us/library/ms724935.aspx
  84. HANDLE_FLAG_INHERIT = 0x00000001
  85. class YouCompleteMe:
  86. def __init__( self ):
  87. self._available_completers = {}
  88. self._user_options = None
  89. self._user_notified_about_crash = False
  90. self._omnicomp = None
  91. self._buffers = None
  92. self._latest_completion_request = None
  93. self._latest_signature_help_request = None
  94. self._signature_help_available_requests = SigHelpAvailableByFileType()
  95. self._signature_help_state = signature_help.SignatureHelpState()
  96. self._logger = logging.getLogger( 'ycm' )
  97. self._client_logfile = None
  98. self._server_stdout = None
  99. self._server_stderr = None
  100. self._server_popen = None
  101. self._filetypes_with_keywords_loaded = set()
  102. self._ycmd_keepalive = YcmdKeepalive()
  103. self._server_is_ready_with_cache = False
  104. self._SetUpLogging()
  105. self._SetUpServer()
  106. self._ycmd_keepalive.Start()
  107. def _SetUpServer( self ):
  108. self._available_completers = {}
  109. self._user_notified_about_crash = False
  110. self._filetypes_with_keywords_loaded = set()
  111. self._server_is_ready_with_cache = False
  112. self._message_poll_requests = {}
  113. self._user_options = base.GetUserOptions()
  114. self._omnicomp = OmniCompleter( self._user_options )
  115. self._buffers = BufferDict( self._user_options )
  116. self._SetLogLevel()
  117. hmac_secret = os.urandom( HMAC_SECRET_LENGTH )
  118. options_dict = dict( self._user_options )
  119. options_dict[ 'hmac_secret' ] = utils.ToUnicode(
  120. base64.b64encode( hmac_secret ) )
  121. options_dict[ 'server_keep_logfiles' ] = self._user_options[
  122. 'keep_logfiles' ]
  123. # The temp options file is deleted by ycmd during startup.
  124. with NamedTemporaryFile( delete = False, mode = 'w+' ) as options_file:
  125. json.dump( options_dict, options_file )
  126. server_port = utils.GetUnusedLocalhostPort()
  127. BaseRequest.server_location = 'http://127.0.0.1:' + str( server_port )
  128. BaseRequest.hmac_secret = hmac_secret
  129. try:
  130. python_interpreter = paths.PathToPythonInterpreter()
  131. except RuntimeError as error:
  132. error_message = (
  133. "Unable to start the ycmd server. {0}. "
  134. "Correct the error then restart the server "
  135. "with ':YcmRestartServer'.".format( str( error ).rstrip( '.' ) ) )
  136. self._logger.exception( error_message )
  137. vimsupport.PostVimMessage( error_message )
  138. return
  139. args = [ python_interpreter,
  140. paths.PathToServerScript(),
  141. '--port={0}'.format( server_port ),
  142. '--options_file={0}'.format( options_file.name ),
  143. '--log={0}'.format( self._user_options[ 'log_level' ] ),
  144. '--idle_suicide_seconds={0}'.format(
  145. SERVER_IDLE_SUICIDE_SECONDS ) ]
  146. self._server_stdout = utils.CreateLogfile(
  147. SERVER_LOGFILE_FORMAT.format( port = server_port, std = 'stdout' ) )
  148. self._server_stderr = utils.CreateLogfile(
  149. SERVER_LOGFILE_FORMAT.format( port = server_port, std = 'stderr' ) )
  150. args.append( '--stdout={0}'.format( self._server_stdout ) )
  151. args.append( '--stderr={0}'.format( self._server_stderr ) )
  152. if self._user_options[ 'keep_logfiles' ]:
  153. args.append( '--keep_logfiles' )
  154. self._server_popen = utils.SafePopen( args, stdin_windows = PIPE,
  155. stdout = PIPE, stderr = PIPE )
  156. def _SetUpLogging( self ):
  157. def FreeFileFromOtherProcesses( file_object ):
  158. if utils.OnWindows():
  159. from ctypes import windll
  160. import msvcrt
  161. file_handle = msvcrt.get_osfhandle( file_object.fileno() )
  162. windll.kernel32.SetHandleInformation( file_handle,
  163. HANDLE_FLAG_INHERIT,
  164. 0 )
  165. self._client_logfile = utils.CreateLogfile( CLIENT_LOGFILE_FORMAT )
  166. handler = logging.FileHandler( self._client_logfile )
  167. # On Windows and Python prior to 3.4, file handles are inherited by child
  168. # processes started with at least one replaced standard stream, which is the
  169. # case when we start the ycmd server (we are redirecting all standard
  170. # outputs into a pipe). These files cannot be removed while the child
  171. # processes are still up. This is not desirable for a logfile because we
  172. # want to remove it at Vim exit without having to wait for the ycmd server
  173. # to be completely shut down. We need to make the logfile handle
  174. # non-inheritable. See https://www.python.org/dev/peps/pep-0446 for more
  175. # details.
  176. FreeFileFromOtherProcesses( handler.stream )
  177. formatter = logging.Formatter( '%(asctime)s - %(levelname)s - %(message)s' )
  178. handler.setFormatter( formatter )
  179. self._logger.addHandler( handler )
  180. def _SetLogLevel( self ):
  181. log_level = self._user_options[ 'log_level' ]
  182. numeric_level = getattr( logging, log_level.upper(), None )
  183. if not isinstance( numeric_level, int ):
  184. raise ValueError( 'Invalid log level: {0}'.format( log_level ) )
  185. self._logger.setLevel( numeric_level )
  186. def IsServerAlive( self ):
  187. # When the process hasn't finished yet, poll() returns None.
  188. return bool( self._server_popen ) and self._server_popen.poll() is None
  189. def CheckIfServerIsReady( self ):
  190. if not self._server_is_ready_with_cache and self.IsServerAlive():
  191. self._server_is_ready_with_cache = BaseRequest().GetDataFromHandler(
  192. 'ready', display_message = False )
  193. return self._server_is_ready_with_cache
  194. def IsServerReady( self ):
  195. return self._server_is_ready_with_cache
  196. def NotifyUserIfServerCrashed( self ):
  197. if ( not self._server_popen or self._user_notified_about_crash or
  198. self.IsServerAlive() ):
  199. return
  200. self._user_notified_about_crash = True
  201. return_code = self._server_popen.poll()
  202. logfile = os.path.basename( self._server_stderr )
  203. # See https://github.com/Valloric/ycmd#exit-codes for the list of exit
  204. # codes.
  205. if return_code == 3:
  206. error_message = CORE_UNEXPECTED_MESSAGE.format( logfile = logfile )
  207. elif return_code == 4:
  208. error_message = CORE_MISSING_MESSAGE
  209. elif return_code == 7:
  210. error_message = CORE_OUTDATED_MESSAGE
  211. elif return_code == 8:
  212. error_message = NO_PYTHON2_SUPPORT_MESSAGE
  213. else:
  214. error_message = EXIT_CODE_UNEXPECTED_MESSAGE.format( code = return_code,
  215. logfile = logfile )
  216. if return_code != 8:
  217. error_message = SERVER_SHUTDOWN_MESSAGE + ' ' + error_message
  218. self._logger.error( error_message )
  219. vimsupport.PostVimMessage( error_message )
  220. def ServerPid( self ):
  221. if not self._server_popen:
  222. return -1
  223. return self._server_popen.pid
  224. def _ShutdownServer( self ):
  225. SendShutdownRequest()
  226. def RestartServer( self ):
  227. vimsupport.PostVimMessage( 'Restarting ycmd server...' )
  228. self._ShutdownServer()
  229. self._SetUpServer()
  230. def SendCompletionRequest( self, force_semantic = False ):
  231. request_data = BuildRequestData()
  232. request_data[ 'force_semantic' ] = force_semantic
  233. if not self.NativeFiletypeCompletionUsable():
  234. wrapped_request_data = RequestWrap( request_data )
  235. if self._omnicomp.ShouldUseNow( wrapped_request_data ):
  236. self._latest_completion_request = OmniCompletionRequest(
  237. self._omnicomp, wrapped_request_data )
  238. self._latest_completion_request.Start()
  239. return
  240. self._AddExtraConfDataIfNeeded( request_data )
  241. self._latest_completion_request = CompletionRequest( request_data )
  242. self._latest_completion_request.Start()
  243. def CompletionRequestReady( self ):
  244. return bool( self._latest_completion_request and
  245. self._latest_completion_request.Done() )
  246. def GetCompletionResponse( self ):
  247. response = self._latest_completion_request.Response()
  248. response[ 'completions' ] = base.AdjustCandidateInsertionText(
  249. response[ 'completions' ] )
  250. return response
  251. def SendSignatureHelpRequest( self ):
  252. """Send a signature help request, if we're ready to. Return whether or not a
  253. request was sent (and should be checked later)"""
  254. if not self.NativeFiletypeCompletionUsable():
  255. return False
  256. if not self._latest_completion_request:
  257. return False
  258. for filetype in vimsupport.CurrentFiletypes():
  259. if not self._signature_help_available_requests[ filetype ].Done():
  260. continue
  261. sig_help_available = self._signature_help_available_requests[
  262. filetype ].Response()
  263. if sig_help_available == 'NO':
  264. continue
  265. if sig_help_available == 'PENDING':
  266. # Send another /signature_help_available request
  267. self._signature_help_available_requests[ filetype ].Start( filetype )
  268. return False
  269. request_data = self._latest_completion_request.request_data.copy()
  270. request_data[ 'signature_help_state' ] = self._signature_help_state.state
  271. self._AddExtraConfDataIfNeeded( request_data )
  272. self._latest_signature_help_request = SignatureHelpRequest( request_data )
  273. self._latest_signature_help_request.Start()
  274. return True
  275. def SignatureHelpRequestReady( self ):
  276. return bool( self._latest_signature_help_request and
  277. self._latest_signature_help_request.Done() )
  278. def GetSignatureHelpResponse( self ):
  279. return self._latest_signature_help_request.Response()
  280. def ClearSignatureHelp( self ):
  281. self.UpdateSignatureHelp( {} )
  282. if self._latest_signature_help_request:
  283. self._latest_signature_help_request.Reset()
  284. def UpdateSignatureHelp( self, signature_info ):
  285. self._signature_help_state = signature_help.UpdateSignatureHelp(
  286. self._signature_help_state,
  287. signature_info )
  288. def SendCommandRequest( self,
  289. arguments,
  290. modifiers,
  291. has_range,
  292. start_line,
  293. end_line ):
  294. final_arguments = []
  295. for argument in arguments:
  296. # The ft= option which specifies the completer when running a command is
  297. # ignored because it has not been working for a long time. The option is
  298. # still parsed to not break users that rely on it.
  299. if argument.startswith( 'ft=' ):
  300. continue
  301. final_arguments.append( argument )
  302. extra_data = {
  303. 'options': {
  304. 'tab_size': vimsupport.GetIntValue( 'shiftwidth()' ),
  305. 'insert_spaces': vimsupport.GetBoolValue( '&expandtab' )
  306. }
  307. }
  308. if has_range:
  309. extra_data.update( vimsupport.BuildRange( start_line, end_line ) )
  310. self._AddExtraConfDataIfNeeded( extra_data )
  311. return SendCommandRequest( final_arguments,
  312. modifiers,
  313. self._user_options[ 'goto_buffer_command' ],
  314. extra_data )
  315. def GetDefinedSubcommands( self ):
  316. subcommands = BaseRequest().PostDataToHandler( BuildRequestData(),
  317. 'defined_subcommands' )
  318. return subcommands if subcommands else []
  319. def GetCurrentCompletionRequest( self ):
  320. return self._latest_completion_request
  321. def GetOmniCompleter( self ):
  322. return self._omnicomp
  323. def FiletypeCompleterExistsForFiletype( self, filetype ):
  324. try:
  325. return self._available_completers[ filetype ]
  326. except KeyError:
  327. pass
  328. exists_completer = SendCompleterAvailableRequest( filetype )
  329. if exists_completer is None:
  330. return False
  331. self._available_completers[ filetype ] = exists_completer
  332. return exists_completer
  333. def NativeFiletypeCompletionAvailable( self ):
  334. return any( self.FiletypeCompleterExistsForFiletype( x ) for x in
  335. vimsupport.CurrentFiletypes() )
  336. def NativeFiletypeCompletionUsable( self ):
  337. disabled_filetypes = self._user_options[
  338. 'filetype_specific_completion_to_disable' ]
  339. return ( vimsupport.CurrentFiletypesEnabled( disabled_filetypes ) and
  340. self.NativeFiletypeCompletionAvailable() )
  341. def NeedsReparse( self ):
  342. return self.CurrentBuffer().NeedsReparse()
  343. def UpdateWithNewDiagnosticsForFile( self, filepath, diagnostics ):
  344. if not self._user_options[ 'show_diagnostics_ui' ]:
  345. return
  346. bufnr = vimsupport.GetBufferNumberForFilename( filepath )
  347. if bufnr in self._buffers and vimsupport.BufferIsVisible( bufnr ):
  348. # Note: We only update location lists, etc. for visible buffers, because
  349. # otherwise we default to using the current location list and the results
  350. # are that non-visible buffer errors clobber visible ones.
  351. self._buffers[ bufnr ].UpdateWithNewDiagnostics( diagnostics )
  352. else:
  353. # The project contains errors in file "filepath", but that file is not
  354. # open in any buffer. This happens for Language Server Protocol-based
  355. # completers, as they return diagnostics for the entire "project"
  356. # asynchronously (rather than per-file in the response to the parse
  357. # request).
  358. #
  359. # There are a number of possible approaches for
  360. # this, but for now we simply ignore them. Other options include:
  361. # - Use the QuickFix list to report project errors?
  362. # - Use a special buffer for project errors
  363. # - Put them in the location list of whatever the "current" buffer is
  364. # - Store them in case the buffer is opened later
  365. # - add a :YcmProjectDiags command
  366. # - Add them to errror/warning _counts_ but not any actual location list
  367. # or other
  368. # - etc.
  369. #
  370. # However, none of those options are great, and lead to their own
  371. # complexities. So for now, we just ignore these diagnostics for files not
  372. # open in any buffer.
  373. pass
  374. def OnPeriodicTick( self ):
  375. if not self.IsServerAlive():
  376. # Server has died. We'll reset when the server is started again.
  377. return False
  378. elif not self.IsServerReady():
  379. # Try again in a jiffy
  380. return True
  381. for w in vim.windows:
  382. for filetype in vimsupport.FiletypesForBuffer( w.buffer ):
  383. if filetype not in self._message_poll_requests:
  384. self._message_poll_requests[ filetype ] = MessagesPoll( w.buffer )
  385. # None means don't poll this filetype
  386. if ( self._message_poll_requests[ filetype ] and
  387. not self._message_poll_requests[ filetype ].Poll( self ) ):
  388. self._message_poll_requests[ filetype ] = None
  389. return any( self._message_poll_requests.values() )
  390. def OnFileReadyToParse( self ):
  391. if not self.IsServerAlive():
  392. self.NotifyUserIfServerCrashed()
  393. return
  394. if not self.IsServerReady():
  395. return
  396. extra_data = {}
  397. self._AddTagsFilesIfNeeded( extra_data )
  398. self._AddSyntaxDataIfNeeded( extra_data )
  399. self._AddExtraConfDataIfNeeded( extra_data )
  400. self.CurrentBuffer().SendParseRequest( extra_data )
  401. def OnBufferUnload( self, deleted_buffer_number ):
  402. SendEventNotificationAsync( 'BufferUnload', deleted_buffer_number )
  403. def UpdateMatches( self ):
  404. self.CurrentBuffer().UpdateMatches()
  405. def OnFileTypeSet( self ):
  406. buffer_number = vimsupport.GetCurrentBufferNumber()
  407. filetypes = vimsupport.CurrentFiletypes()
  408. self._buffers[ buffer_number ].UpdateFromFileTypes( filetypes )
  409. self.OnBufferVisit()
  410. def OnBufferVisit( self ):
  411. filetype = vimsupport.CurrentFiletypes()[ 0 ]
  412. # The constructor of dictionary values starts the request,
  413. # so the line below fires a new request only if the dictionary
  414. # value is accessed for the first time.
  415. self._signature_help_available_requests[ filetype ].Done()
  416. extra_data = {}
  417. self._AddUltiSnipsDataIfNeeded( extra_data )
  418. SendEventNotificationAsync( 'BufferVisit', extra_data = extra_data )
  419. def CurrentBuffer( self ):
  420. return self._buffers[ vimsupport.GetCurrentBufferNumber() ]
  421. def OnInsertLeave( self ):
  422. SendEventNotificationAsync( 'InsertLeave' )
  423. def OnCursorMoved( self ):
  424. self.CurrentBuffer().OnCursorMoved()
  425. def _CleanLogfile( self ):
  426. logging.shutdown()
  427. if not self._user_options[ 'keep_logfiles' ]:
  428. if self._client_logfile:
  429. utils.RemoveIfExists( self._client_logfile )
  430. def OnVimLeave( self ):
  431. self._ShutdownServer()
  432. self._CleanLogfile()
  433. def OnCurrentIdentifierFinished( self ):
  434. SendEventNotificationAsync( 'CurrentIdentifierFinished' )
  435. def OnCompleteDone( self ):
  436. completion_request = self.GetCurrentCompletionRequest()
  437. if completion_request:
  438. completion_request.OnCompleteDone()
  439. def GetErrorCount( self ):
  440. return self.CurrentBuffer().GetErrorCount()
  441. def GetWarningCount( self ):
  442. return self.CurrentBuffer().GetWarningCount()
  443. def _PopulateLocationListWithLatestDiagnostics( self ):
  444. return self.CurrentBuffer().PopulateLocationList()
  445. def FileParseRequestReady( self ):
  446. # Return True if server is not ready yet, to stop repeating check timer.
  447. return ( not self.IsServerReady() or
  448. self.CurrentBuffer().FileParseRequestReady() )
  449. def HandleFileParseRequest( self, block = False ):
  450. if not self.IsServerReady():
  451. return
  452. current_buffer = self.CurrentBuffer()
  453. # Order is important here:
  454. # FileParseRequestReady has a low cost, while
  455. # NativeFiletypeCompletionUsable is a blocking server request
  456. if ( not current_buffer.IsResponseHandled() and
  457. current_buffer.FileParseRequestReady( block ) and
  458. self.NativeFiletypeCompletionUsable() ):
  459. if self._user_options[ 'show_diagnostics_ui' ]:
  460. # Forcefuly update the location list, etc. from the parse request when
  461. # doing something like :YcmDiags
  462. current_buffer.UpdateDiagnostics( block )
  463. else:
  464. # If the user disabled diagnostics, we just want to check
  465. # the _latest_file_parse_request for any exception or UnknownExtraConf
  466. # response, to allow the server to raise configuration warnings, etc.
  467. # to the user. We ignore any other supplied data.
  468. current_buffer.GetResponse()
  469. # We set the file parse request as handled because we want to prevent
  470. # repeated issuing of the same warnings/errors/prompts. Setting this
  471. # makes IsRequestHandled return True until the next request is created.
  472. #
  473. # Note: it is the server's responsibility to determine the frequency of
  474. # error/warning/prompts when receiving a FileReadyToParse event, but
  475. # it is our responsibility to ensure that we only apply the
  476. # warning/error/prompt received once (for each event).
  477. current_buffer.MarkResponseHandled()
  478. def ShouldResendFileParseRequest( self ):
  479. return self.CurrentBuffer().ShouldResendParseRequest()
  480. def DebugInfo( self ):
  481. debug_info = ''
  482. if self._client_logfile:
  483. debug_info += 'Client logfile: {0}\n'.format( self._client_logfile )
  484. extra_data = {}
  485. self._AddExtraConfDataIfNeeded( extra_data )
  486. debug_info += FormatDebugInfoResponse( SendDebugInfoRequest( extra_data ) )
  487. debug_info += 'Server running at: {0}\n'.format(
  488. BaseRequest.server_location )
  489. if self._server_popen:
  490. debug_info += 'Server process ID: {0}\n'.format( self._server_popen.pid )
  491. if self._server_stdout and self._server_stderr:
  492. debug_info += ( 'Server logfiles:\n'
  493. ' {0}\n'
  494. ' {1}'.format( self._server_stdout,
  495. self._server_stderr ) )
  496. return debug_info
  497. def GetLogfiles( self ):
  498. logfiles_list = [ self._client_logfile,
  499. self._server_stdout,
  500. self._server_stderr ]
  501. extra_data = {}
  502. self._AddExtraConfDataIfNeeded( extra_data )
  503. debug_info = SendDebugInfoRequest( extra_data )
  504. if debug_info:
  505. completer = debug_info[ 'completer' ]
  506. if completer:
  507. for server in completer[ 'servers' ]:
  508. logfiles_list.extend( server[ 'logfiles' ] )
  509. logfiles = {}
  510. for logfile in logfiles_list:
  511. logfiles[ os.path.basename( logfile ) ] = logfile
  512. return logfiles
  513. def _OpenLogfile( self, logfile ):
  514. # Open log files in a horizontal window with the same behavior as the
  515. # preview window (same height and winfixheight enabled). Automatically
  516. # watch for changes. Set the cursor position at the end of the file.
  517. options = {
  518. 'size': vimsupport.GetIntValue( '&previewheight' ),
  519. 'fix': True,
  520. 'focus': False,
  521. 'watch': True,
  522. 'position': 'end'
  523. }
  524. vimsupport.OpenFilename( logfile, options )
  525. def _CloseLogfile( self, logfile ):
  526. vimsupport.CloseBuffersForFilename( logfile )
  527. def ToggleLogs( self, *filenames ):
  528. logfiles = self.GetLogfiles()
  529. if not filenames:
  530. sorted_logfiles = sorted( logfiles )
  531. try:
  532. logfile_index = vimsupport.SelectFromList(
  533. 'Which logfile do you wish to open (or close if already open)?',
  534. sorted_logfiles )
  535. except RuntimeError as e:
  536. vimsupport.PostVimMessage( str( e ) )
  537. return
  538. logfile = logfiles[ sorted_logfiles[ logfile_index ] ]
  539. if not vimsupport.BufferIsVisibleForFilename( logfile ):
  540. self._OpenLogfile( logfile )
  541. else:
  542. self._CloseLogfile( logfile )
  543. return
  544. for filename in set( filenames ):
  545. if filename not in logfiles:
  546. continue
  547. logfile = logfiles[ filename ]
  548. if not vimsupport.BufferIsVisibleForFilename( logfile ):
  549. self._OpenLogfile( logfile )
  550. continue
  551. self._CloseLogfile( logfile )
  552. def ShowDetailedDiagnostic( self ):
  553. detailed_diagnostic = BaseRequest().PostDataToHandler(
  554. BuildRequestData(), 'detailed_diagnostic' )
  555. if detailed_diagnostic and 'message' in detailed_diagnostic:
  556. vimsupport.PostVimMessage( detailed_diagnostic[ 'message' ],
  557. warning = False )
  558. def ForceCompileAndDiagnostics( self ):
  559. if not self.NativeFiletypeCompletionUsable():
  560. vimsupport.PostVimMessage(
  561. 'Native filetype completion not supported for current file, '
  562. 'cannot force recompilation.', warning = False )
  563. return False
  564. vimsupport.PostVimMessage(
  565. 'Forcing compilation, this will block Vim until done.',
  566. warning = False )
  567. self.OnFileReadyToParse()
  568. self.HandleFileParseRequest( block = True )
  569. vimsupport.PostVimMessage( 'Diagnostics refreshed', warning = False )
  570. return True
  571. def ShowDiagnostics( self ):
  572. if not self.ForceCompileAndDiagnostics():
  573. return
  574. if not self._PopulateLocationListWithLatestDiagnostics():
  575. vimsupport.PostVimMessage( 'No warnings or errors detected.',
  576. warning = False )
  577. return
  578. if self._user_options[ 'open_loclist_on_ycm_diags' ]:
  579. vimsupport.OpenLocationList( focus = True )
  580. def _AddSyntaxDataIfNeeded( self, extra_data ):
  581. if not self._user_options[ 'seed_identifiers_with_syntax' ]:
  582. return
  583. filetype = vimsupport.CurrentFiletypes()[ 0 ]
  584. if filetype in self._filetypes_with_keywords_loaded:
  585. return
  586. if self.IsServerReady():
  587. self._filetypes_with_keywords_loaded.add( filetype )
  588. extra_data[ 'syntax_keywords' ] = list(
  589. syntax_parse.SyntaxKeywordsForCurrentBuffer() )
  590. def _AddTagsFilesIfNeeded( self, extra_data ):
  591. def GetTagFiles():
  592. tag_files = vim.eval( 'tagfiles()' )
  593. return [ os.path.join( utils.GetCurrentDirectory(), tag_file )
  594. for tag_file in tag_files ]
  595. if not self._user_options[ 'collect_identifiers_from_tags_files' ]:
  596. return
  597. extra_data[ 'tag_files' ] = GetTagFiles()
  598. def _AddExtraConfDataIfNeeded( self, extra_data ):
  599. def BuildExtraConfData( extra_conf_vim_data ):
  600. extra_conf_data = {}
  601. for expr in extra_conf_vim_data:
  602. try:
  603. extra_conf_data[ expr ] = vimsupport.VimExpressionToPythonType( expr )
  604. except vim.error:
  605. message = (
  606. "Error evaluating '{expr}' in the 'g:ycm_extra_conf_vim_data' "
  607. "option.".format( expr = expr ) )
  608. vimsupport.PostVimMessage( message, truncate = True )
  609. self._logger.exception( message )
  610. return extra_conf_data
  611. extra_conf_vim_data = self._user_options[ 'extra_conf_vim_data' ]
  612. if extra_conf_vim_data:
  613. extra_data[ 'extra_conf_data' ] = BuildExtraConfData(
  614. extra_conf_vim_data )
  615. def _AddUltiSnipsDataIfNeeded( self, extra_data ):
  616. # See :h UltiSnips#SnippetsInCurrentScope.
  617. try:
  618. vim.eval( 'UltiSnips#SnippetsInCurrentScope( 1 )' )
  619. except vim.error:
  620. return
  621. snippets = vimsupport.GetVariableValue( 'g:current_ulti_dict_info' )
  622. extra_data[ 'ultisnips_snippets' ] = [
  623. { 'trigger': trigger,
  624. 'description': snippet[ 'description' ] }
  625. for trigger, snippet in snippets.items()
  626. ]