1
0
mirror of https://github.com/SpaceVim/SpaceVim.git synced 2025-01-24 09:30:04 +08:00
SpaceVim/bundle/defx.nvim/rplugin/python3/defx/rplugin.py
2020-10-31 15:58:52 +08:00

79 lines
2.7 KiB
Python

# ============================================================================
# FILE: rplugin.py
# AUTHOR: Shougo Matsushita <Shougo.Matsu at gmail.com>
# License: MIT license
# ============================================================================
import typing
from defx.clipboard import Clipboard
from defx.view import View
from defx.util import Nvim
class Rplugin:
def __init__(self, vim: Nvim) -> None:
self._vim = vim
self._views: typing.List[View] = []
self._clipboard = Clipboard()
def init_channel(self) -> None:
self._vim.vars['defx#_channel_id'] = self._vim.channel_id
def start(self, args: typing.List[typing.Any]) -> None:
[paths, context] = args
self.get_view(context).init_paths(
paths, context, self._clipboard)
def do_action(self, args: typing.List[typing.Any]) -> None:
views = [x for x in self._views
if x._bufnr == self._vim.current.buffer.number]
if not views:
return
view = views[0]
prev_paths = [x._cwd for x in view._defxs]
prev_candidates = view._candidates
view.do_action(args[0], args[1], args[2])
paths = [x._cwd for x in view._defxs]
if paths == prev_paths and view._candidates != prev_candidates:
self.redraw([x for x in self._views if x != view])
def get_candidate(self) -> typing.Dict[str, typing.Union[str, bool]]:
cursor = self._vim.call('line', '.')
for view in [x for x in self._views
if x._bufnr == self._vim.current.buffer.number]:
candidate = view.get_cursor_candidate(cursor)
return {
'word': candidate['word'],
'is_directory': candidate['is_directory'],
'is_opened_tree': candidate['is_opened_tree'],
'level': candidate['level'],
'action__path': str(candidate['action__path']),
}
return {}
def get_context(self) -> typing.Dict[str, typing.Any]:
for view in [x for x in self._views
if x._bufnr == self._vim.current.buffer.number]:
return view._context._asdict()
return {}
def get_view(self, context: typing.Dict[str, typing.Any]) -> View:
views = [x for x in self._views
if context['buffer_name'] == x._context.buffer_name]
if not views or context['new']:
view = View(self._vim, len(self._views))
views = [view]
self._views.append(view)
return views[0]
def redraw(self, views: typing.List[View]) -> None:
call = self._vim.call
for view in [x for x in views
if call('bufwinnr', x._bufnr) > 0]:
view.redraw(True)