spacevim/bundle/deoplete.nvim/rplugin/python3/deoplete/source/member.py

# ============================================================================
# FILE: member.py
# AUTHOR: Shougo Matsushita <Shougo.Matsu at gmail.com>
# License: MIT license
# ============================================================================

from pynvim import Nvim
import re
import typing

from deoplete.base.source import Base
from deoplete.util import (
    convert2list, parse_buffer_pattern, set_pattern, getlines)
from deoplete.util import UserContext, Candidates


class Source(Base):

    def __init__(self, vim: Nvim) -> None:
        super().__init__(vim)

        self.name = 'member'
        self.mark = '[M]'
        self.min_pattern_length = 0

        self._object_pattern = r'[a-zA-Z_]\w*(?:\(\)?)?'
        self._prefix = ''

        prefix_patterns: typing.Dict[str, str] = {}
        set_pattern(prefix_patterns,
                    '_', r'\.')
        set_pattern(prefix_patterns,
                    'c,objc', [r'\.', '->'])
        set_pattern(prefix_patterns,
                    'cpp,objcpp', [r'\.', '->', '::'])
        set_pattern(prefix_patterns,
                    'perl,php', ['->'])
        set_pattern(prefix_patterns,
                    'ruby', [r'\.', '::'])
        set_pattern(prefix_patterns,
                    'lua', [r'\.', ':'])
        self.vars = {
            'prefix_patterns': prefix_patterns,
        }

    def get_complete_position(self, context: UserContext) -> int:
        # Check member prefix pattern.
        for prefix_pattern in convert2list(
                self.get_filetype_var(
                    context['filetype'], 'prefix_patterns')):
            m = re.search(self._object_pattern + prefix_pattern + r'\w*$',
                          context['input'])
            if m is None or prefix_pattern == '':
                continue
            self._prefix = re.sub(r'\w*$', '', m.group(0))
            m = re.search(r'\w*$', context['input'])
            if m:
                return m.start()
        return -1

    def gather_candidates(self, context: UserContext) -> Candidates:
        return [{'word': x} for x in
                parse_buffer_pattern(
                    getlines(self.vim),
                    r'(?<=' + re.escape(self._prefix) + r')\w+'
                )
                if x != context['complete_str']]