mirror of
https://github.com/SpaceVim/SpaceVim.git
synced 2025-02-03 21:50:05 +08:00
63 lines
1.6 KiB
Python
Vendored
63 lines
1.6 KiB
Python
Vendored
# ============================================================================
|
|
# FILE: _main.py
|
|
# AUTHOR: Shougo Matsushita <Shougo.Matsu at gmail.com>
|
|
# License: MIT license
|
|
# ============================================================================
|
|
|
|
import sys
|
|
import io
|
|
|
|
from importlib.util import find_spec
|
|
if find_spec('pynvim'):
|
|
from pynvim import attach
|
|
else:
|
|
from neovim import attach
|
|
|
|
|
|
def attach_vim(serveraddr):
|
|
if len(serveraddr.split(':')) == 2:
|
|
serveraddr, port = serveraddr.split(':')
|
|
port = int(port)
|
|
vim = attach('tcp', address=serveraddr, port=port)
|
|
else:
|
|
vim = attach('socket', path=serveraddr)
|
|
|
|
# sync path
|
|
for path in vim.call(
|
|
'globpath', vim.options['runtimepath'],
|
|
'rplugin/python3', 1).split('\n'):
|
|
sys.path.append(path)
|
|
# Remove current path
|
|
del sys.path[0]
|
|
|
|
return vim
|
|
|
|
|
|
class RedirectStream(io.IOBase):
|
|
def __init__(self, handler):
|
|
self.handler = handler
|
|
|
|
def write(self, line):
|
|
self.handler(line)
|
|
|
|
def writelines(self, lines):
|
|
self.handler('\n'.join(lines))
|
|
|
|
|
|
def main(serveraddr):
|
|
vim = attach_vim(serveraddr)
|
|
from deoplete.child import Child
|
|
from deoplete.util import error_tb
|
|
stdout = sys.stdout
|
|
sys.stdout = RedirectStream(lambda data: vim.out_write(data))
|
|
sys.stderr = RedirectStream(lambda data: vim.err_write(data))
|
|
try:
|
|
child = Child(vim)
|
|
child.main_loop(stdout)
|
|
except Exception as exc:
|
|
error_tb(vim, 'Error in child: %r' % exc)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv[1])
|