111 lines
4.4 KiB
Python
111 lines
4.4 KiB
Python
#=============================================================================
|
|
# FILE: racer.py
|
|
# AUTHOR: Shougo Matsushita <Shougo.Matsu at gmail.com>
|
|
# License: MIT license {{{
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included
|
|
# in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
|
# CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
# }}}
|
|
#=============================================================================
|
|
|
|
import re
|
|
import os
|
|
import subprocess
|
|
import tempfile
|
|
from .base import Base
|
|
|
|
class Source(Base):
|
|
def __init__(self, vim):
|
|
Base.__init__(self, vim)
|
|
|
|
self.name = 'racer'
|
|
self.mark = '[racer]'
|
|
self.filetypes = ['rust']
|
|
self.input_pattern = r'(\.|::)\w*'
|
|
self.rank = 500
|
|
|
|
def on_init(self, context):
|
|
self.__executable_racer = self.vim.funcs.executable(
|
|
self.vim.eval('g:racer_cmd'))
|
|
self.__racer = self.vim.eval('g:racer_cmd')
|
|
|
|
def get_complete_position(self, context):
|
|
if not self.__executable_racer:
|
|
return -1
|
|
|
|
m = re.search('\w*$', context['input'])
|
|
return m.start() if m else -1
|
|
|
|
|
|
def gather_candidates(self, context):
|
|
typeMap = {
|
|
'Struct': 's', 'Module': 'M', 'Function': 'f',
|
|
'Crate': 'C', 'Let': 'v', 'StructField': 'm',
|
|
'Impl': 'i', 'Enum': 'e', 'EnumVariant': 'E',
|
|
'Type': 't', 'FnArg': 'v', 'Trait': 'T',
|
|
'Const': 'c'
|
|
}
|
|
|
|
candidates = []
|
|
insert_paren = int(self.vim.eval('g:racer_insert_paren'))
|
|
for line in [l[6:] for l
|
|
in self.get_results(context, 'complete',
|
|
context['complete_position'] + 1)
|
|
if l.startswith('MATCH')]:
|
|
completions = line.split(',')
|
|
kind = typeMap.get(completions[4], '')
|
|
completion = { 'kind': kind, 'word': completions[0], 'dup': 1 }
|
|
if kind == 'f': # function
|
|
completion['menu'] = ','.join(completions[5:]).replace(
|
|
'pub ', '').replace('fn ', '').rstrip('{')
|
|
if ' where ' in completion['menu'] or completion[
|
|
'menu'].endswith(' where') :
|
|
where = completion['menu'].rindex(' where')
|
|
completion['menu'] = completion['menu'][: where]
|
|
if insert_paren:
|
|
completion['abbr'] = completions[0]
|
|
completion['word'] += '('
|
|
elif kind == 's' : # struct
|
|
completion['menu'] = ','.join(completions[5:]).replace(
|
|
'pub ', '').replace( 'struct ', '').rstrip('{')
|
|
candidates.append(completion)
|
|
return candidates
|
|
|
|
def get_results(self, context, command, col):
|
|
with tempfile.NamedTemporaryFile(mode='w') as tf:
|
|
tf.write("\n".join(self.vim.current.buffer))
|
|
tf.flush()
|
|
|
|
args = [
|
|
self.__racer, command,
|
|
str(self.vim.funcs.line('.')),
|
|
str(col - 1),
|
|
tf.name
|
|
] if command == 'prefix' else [
|
|
self.__racer, command,
|
|
str(self.vim.funcs.line('.')),
|
|
str(col - 1),
|
|
self.vim.current.buffer.name,
|
|
tf.name
|
|
]
|
|
try:
|
|
results = subprocess.check_output(args).decode(
|
|
context['encoding']).splitlines()
|
|
except subprocess.CalledProcessError:
|
|
return []
|
|
return results
|