SpaceVim/bundle/defx-icons/rplugin/python3/defx/column/icons.py

# ============================================================================
# FILE: icons.py
# AUTHOR: Kristijan Husak <husakkristijan at gmail.com>
# License: MIT license
# ============================================================================

import re
import typing
from pathlib import Path
from defx.base.column import Base
from defx.context import Context
from defx.clipboard import ClipboardAction
from defx.view import View
from neovim import Nvim


class Column(Base):
    def __init__(self, vim: Nvim) -> None:
        super().__init__(vim)
        self.vim = vim
        self.name = 'icons'
        self.opts = self.vim.call('defx_icons#get')

    def on_init(self, view: View, context: Context) -> None:
        self._context = context
        self._view = view

    def on_redraw(self, view: View, context: Context) -> None:
        self._context = context
        self._view = view

    def get(self, context: Context, candidate: dict) -> str:
        path: Path = candidate['action__path']
        filename = path.name
        if 'mark' not in context.columns and candidate['is_selected']:
            return self.icon(self.opts['mark_icon'])

        if self._view and self._view._clipboard.candidates:
            for clipboard_candidate in self._view._clipboard.candidates:
                if str(clipboard_candidate['action__path']) == str(path):
                    return self.clipboard_icon()

        if candidate.get('is_root', False):
            return self.icon(self.opts['parent_icon'])

        if candidate['is_directory']:
            if filename in self.opts['exact_dir_matches']:
                return self.icon(self.opts['exact_dir_matches'][filename]['icon'])

            if candidate.get('level', 0) > 0:
                if candidate.get('is_opened_tree'):
                    return self.icon(self.opts['nested_opened_tree_icon'])
                return self.icon(self.opts['nested_closed_tree_icon'])

            if candidate.get('is_opened_tree', False):
                return self.icon(self.opts['root_opened_tree_icon'])

            if path.is_symlink():
                return self.icon(self.opts['directory_symlink_icon'])

            return self.icon(self.opts['directory_icon'])

        filename = filename.lower()
        ext = path.suffix[1:].lower()

        for pattern, pattern_data in self.opts['pattern_matches'].items():
            if re.search(pattern, filename) is not None:
                return self.icon(pattern_data['icon'])

        if filename in self.opts['exact_matches']:
            return self.icon(self.opts['exact_matches'][filename]['icon'])

        if ext in self.opts['extensions']:
            return self.icon(self.opts['extensions'][ext]['icon'])

        return self.icon(self.opts['default_icon'])

    def length(self, context: Context) -> int:
        return self.opts['column_length']

    def icon(self, icon: str) -> str:
        return format(icon, f'<{self.opts["column_length"]}')

    def clipboard_icon(self) -> str:
        if  self._view._clipboard.action == ClipboardAction.COPY:
            return self.opts['copy_icon']
        if self._view._clipboard.action == ClipboardAction.MOVE:
            return self.opts['move_icon']
        return ''

    def syn_item(self, name, opt_name, hi_group_name) -> typing.List[str]:
        commands: typing.List[str] = []
        commands.append(f'silent! syntax clear {self.syntax_name}_{name}')
        commands.append((
            'syntax match {0}_{1} /[{2}]/ contained containedin={0}'
        ).format(self.syntax_name, name, self.opts[opt_name]))
        commands.append('highlight default link {0}_{1} {2}'.format(
            self.syntax_name, name, hi_group_name
        ))
        return commands

    def syn_list(self, opt) -> typing.List[str]:
        commands: typing.List[str] = []
        for name, opts in self.opts[opt].items():
            text = re.sub('[^A-Za-z]', '', name)
            commands.append(f'silent! syntax clear {self.syntax_name}_{text}')
            commands.append((
                'syntax match {0}_{1} /[{2}]/ contained containedin={0}'
            ).format(self.syntax_name, text, opts['icon']))
            commands.append('highlight default {0}_{1} guifg=#{2} ctermfg={3}'.format(
                self.syntax_name, text, opts['color'], opts.get('term_color',
                                                                'NONE')))
        return commands

    def highlight_commands(self) -> typing.List[str]:
        commands: typing.List[str] = []

        if not self.opts['enable_syntax_highlight']:
            return commands

        commands += self.syn_item('icon_mark', 'mark_icon', 'DefxIconsMarkIcon')
        commands += self.syn_item('icon_copy', 'copy_icon', 'DefxIconsCopyIcon')
        commands += self.syn_item('icon_move', 'move_icon', 'DefxIconsMoveIcon')

        commands += self.syn_item('directory', 'directory_icon', 'DefxIconsDirectory')
        commands += self.syn_item('parent_directory', 'parent_icon', 'DefxIconsParentDirectory')
        commands += self.syn_item('symlink_directory', 'directory_symlink_icon', 'DefxIconsSymlinkDirectory')
        commands += self.syn_item('root_opened_tree_icon', 'root_opened_tree_icon', 'DefxIconsOpenedTreeIcon')
        commands += self.syn_item('nested_opened_tree_icon', 'nested_opened_tree_icon', 'DefxIconsNestedTreeIcon')
        commands += self.syn_item('nested_closed_tree_icon', 'nested_closed_tree_icon', 'DefxIconsClosedTreeIcon')

        commands += self.syn_list('pattern_matches')
        commands += self.syn_list('exact_matches')
        commands += self.syn_list('exact_dir_matches')
        commands += self.syn_list('extensions')

        return commands