boot.py

from __future__ import annotations
import os
import sublime
import sublime_plugin

# Please keep this list sorted (Edit -> Sort Lines)
from .plugin.code_actions import LspCodeActionsCommand
from .plugin.code_actions import LspRefactorCommand
from .plugin.code_actions import LspSourceActionCommand
from .plugin.code_lens import LspCodeLensCommand
from .plugin.code_lens import LspToggleCodeLensesCommand
from .plugin.color import LspColorPresentationCommand
from .plugin.completion import LspCommitCompletionWithOppositeInsertMode
from .plugin.completion import LspResolveDocsCommand
from .plugin.completion import LspSelectCompletionCommand
from .plugin.configuration import LspDisableLanguageServerGloballyCommand
from .plugin.configuration import LspDisableLanguageServerInProjectCommand
from .plugin.configuration import LspEnableLanguageServerGloballyCommand
from .plugin.configuration import LspEnableLanguageServerInProjectCommand
from .plugin.core.css import load as load_css
from .plugin.core.open import opening_files
from .plugin.core.panels import PanelName
from .plugin.core.registry import LspNextDiagnosticCommand
from .plugin.core.registry import LspOpenLocationCommand
from .plugin.core.registry import LspPrevDiagnosticCommand
from .plugin.core.registry import LspRestartServerCommand
from .plugin.core.registry import windows
from .plugin.core.sessions import AbstractPlugin
from .plugin.core.sessions import register_plugin
from .plugin.core.settings import client_configs
from .plugin.core.settings import load_settings
from .plugin.core.settings import unload_settings
from .plugin.core.signature_help import LspSignatureHelpNavigateCommand
from .plugin.core.signature_help import LspSignatureHelpShowCommand
from .plugin.core.transports import kill_all_subprocesses
from .plugin.core.tree_view import LspCollapseTreeItemCommand
from .plugin.core.tree_view import LspExpandTreeItemCommand
from .plugin.core.views import LspRunTextCommandHelperCommand
from .plugin.document_link import LspOpenLinkCommand
from .plugin.documents import DocumentSyncListener
from .plugin.documents import TextChangeListener
from .plugin.edit import LspApplyDocumentEditCommand
from .plugin.edit import LspApplyWorkspaceEditCommand
from .plugin.execute_command import LspExecuteCommand
from .plugin.folding_range import LspFoldAllCommand
from .plugin.folding_range import LspFoldCommand
from .plugin.formatting import LspFormatCommand
from .plugin.formatting import LspFormatDocumentCommand
from .plugin.formatting import LspFormatDocumentRangeCommand
from .plugin.goto import LspSymbolDeclarationCommand
from .plugin.goto import LspSymbolDefinitionCommand
from .plugin.goto import LspSymbolImplementationCommand
from .plugin.goto import LspSymbolTypeDefinitionCommand
from .plugin.goto_diagnostic import LspGotoDiagnosticCommand
from .plugin.hierarchy import LspCallHierarchyCommand
from .plugin.hierarchy import LspHierarchyToggleCommand
from .plugin.hierarchy import LspTypeHierarchyCommand
from .plugin.hover import LspHoverCommand
from .plugin.hover import LspToggleHoverPopupsCommand
from .plugin.inlay_hint import LspInlayHintClickCommand
from .plugin.inlay_hint import LspToggleInlayHintsCommand
from .plugin.panels import LspClearLogPanelCommand
from .plugin.panels import LspClearPanelCommand
from .plugin.panels import LspShowDiagnosticsPanelCommand
from .plugin.panels import LspToggleLogPanelLinesLimitCommand
from .plugin.panels import LspToggleServerPanelCommand
from .plugin.panels import LspUpdateLogPanelCommand
from .plugin.panels import LspUpdatePanelCommand
from .plugin.references import LspSymbolReferencesCommand
from .plugin.rename import LspHideRenameButtonsCommand
from .plugin.rename import LspSymbolRenameCommand
from .plugin.save_command import LspSaveAllCommand
from .plugin.save_command import LspSaveCommand
from .plugin.selection_range import LspExpandSelectionCommand
from .plugin.semantic_highlighting import LspShowScopeNameCommand
from .plugin.symbols import LspDocumentSymbolsCommand
from .plugin.symbols import LspSelectionAddCommand
from .plugin.symbols import LspSelectionClearCommand
from .plugin.symbols import LspSelectionSetCommand
from .plugin.symbols import LspWorkspaceSymbolsCommand
from .plugin.tooling import LspCopyToClipboardFromBase64Command
from .plugin.tooling import LspDumpBufferCapabilities
from .plugin.tooling import LspDumpWindowConfigs
from .plugin.tooling import LspOnDoubleClickCommand
from .plugin.tooling import LspParseVscodePackageJson
from .plugin.tooling import LspTroubleshootServerCommand
from typing import Any

__all__ = (
    "DocumentSyncListener",
    "Listener",
    "LspApplyDocumentEditCommand",
    "LspApplyWorkspaceEditCommand",
    "LspCallHierarchyCommand",
    "LspClearLogPanelCommand",
    "LspClearPanelCommand",
    "LspCodeActionsCommand",
    "LspCodeLensCommand",
    "LspCollapseTreeItemCommand",
    "LspColorPresentationCommand",
    "LspCommitCompletionWithOppositeInsertMode",
    "LspCopyToClipboardFromBase64Command",
    "LspDisableLanguageServerGloballyCommand",
    "LspDisableLanguageServerInProjectCommand",
    "LspDocumentSymbolsCommand",
    "LspDumpBufferCapabilities",
    "LspDumpWindowConfigs",
    "LspEnableLanguageServerGloballyCommand",
    "LspEnableLanguageServerInProjectCommand",
    "LspExecuteCommand",
    "LspExpandSelectionCommand",
    "LspExpandTreeItemCommand",
    "LspFoldAllCommand",
    "LspFoldCommand",
    "LspFormatCommand",
    "LspFormatDocumentCommand",
    "LspFormatDocumentRangeCommand",
    "LspGotoDiagnosticCommand",
    "LspHideRenameButtonsCommand",
    "LspHierarchyToggleCommand",
    "LspHoverCommand",
    "LspInlayHintClickCommand",
    "LspNextDiagnosticCommand",
    "LspOnDoubleClickCommand",
    "LspOpenLinkCommand",
    "LspOpenLocationCommand",
    "LspParseVscodePackageJson",
    "LspPrevDiagnosticCommand",
    "LspRefactorCommand",
    "LspResolveDocsCommand",
    "LspRestartServerCommand",
    "LspRunTextCommandHelperCommand",
    "LspSaveAllCommand",
    "LspSaveCommand",
    "LspSelectCompletionCommand",
    "LspSelectionAddCommand",
    "LspSelectionClearCommand",
    "LspSelectionSetCommand",
    "LspShowDiagnosticsPanelCommand",
    "LspShowScopeNameCommand",
    "LspSignatureHelpNavigateCommand",
    "LspSignatureHelpShowCommand",
    "LspSourceActionCommand",
    "LspSymbolDeclarationCommand",
    "LspSymbolDefinitionCommand",
    "LspSymbolImplementationCommand",
    "LspSymbolReferencesCommand",
    "LspSymbolRenameCommand",
    "LspSymbolTypeDefinitionCommand",
    "LspToggleCodeLensesCommand",
    "LspToggleHoverPopupsCommand",
    "LspToggleInlayHintsCommand",
    "LspToggleLogPanelLinesLimitCommand",
    "LspToggleServerPanelCommand",
    "LspTroubleshootServerCommand",
    "LspTypeHierarchyCommand",
    "LspUpdateLogPanelCommand",
    "LspUpdatePanelCommand",
    "LspWorkspaceSymbolsCommand",
    "TextChangeListener",
    "plugin_loaded",
    "plugin_unloaded",
)


def _get_final_subclasses(derived: list[type], results: list[type]) -> None:
    for d in derived:
        d_subclasses = d.__subclasses__()
        if len(d_subclasses) > 0:
            _get_final_subclasses(d_subclasses, results)
        else:
            results.append(d)


def _register_all_plugins() -> None:
    plugin_classes: list[type[AbstractPlugin]] = []
    _get_final_subclasses(AbstractPlugin.__subclasses__(), plugin_classes)
    for plugin_class in plugin_classes:
        try:
            if not plugin_class.name():
                continue
        except NotImplementedError:
            continue
        register_plugin(plugin_class, notify_listener=False)


def _unregister_all_plugins() -> None:
    from LSP.plugin.core.sessions import _plugins

    _plugins.clear()
    client_configs.external.clear()
    client_configs.all.clear()


def plugin_loaded() -> None:
    load_settings()
    load_css()
    windows.enable()
    _register_all_plugins()
    client_configs.update_configs()


def plugin_unloaded() -> None:
    _unregister_all_plugins()
    windows.disable()
    unload_settings()


class Listener(sublime_plugin.EventListener):
    def on_exit(self) -> None:
        kill_all_subprocesses()

    def on_load_project_async(self, w: sublime.Window) -> None:
        manager = windows.lookup(w)
        if manager:
            manager.on_load_project_async()

    def on_post_save_project_async(self, w: sublime.Window) -> None:
        manager = windows.lookup(w)
        if manager:
            manager.on_post_save_project_async()

    def on_new_window_async(self, w: sublime.Window) -> None:
        sublime.set_timeout(lambda: windows.lookup(w))

    def on_pre_close_window(self, w: sublime.Window) -> None:
        windows.discard(w)

    # Note: EventListener.on_post_move_async does not fire when a tab is moved out of the current window in such a way
    # that a new window is created: https://github.com/sublimehq/sublime_text/issues/4630
    # Hence, as a workaround we use on_pre_move, which still works in that case.
    def on_pre_move(self, view: sublime.View) -> None:
        listeners = sublime_plugin.view_event_listeners.get(view.id())
        if not isinstance(listeners, list):
            return
        for listener in listeners:
            if isinstance(listener, DocumentSyncListener):
                # we need a small delay here, so that the DocumentSyncListener will recognize a possible new window
                sublime.set_timeout_async(listener.on_post_move_window_async, 1)
                return

    def on_load(self, view: sublime.View) -> None:
        file_name = view.file_name()
        if not file_name:
            return
        for fn in opening_files.keys():
            if fn == file_name or os.path.samefile(fn, file_name):
                # Remove it from the pending opening files, and resolve the promise.
                opening_files.pop(fn)[1](view)
                break

    def on_pre_close(self, view: sublime.View) -> None:
        file_name = view.file_name()
        if not file_name:
            return
        for fn in opening_files.keys():
            if fn == file_name or os.path.samefile(fn, file_name):
                tup = opening_files.pop(fn, None)
                if tup:
                    # The view got closed before it finished loading. This can happen.
                    tup[1](None)
                    break

    def on_post_window_command(self, window: sublime.Window, command_name: str, args: dict[str, Any] | None) -> None:
        if command_name == "show_panel":
            wm = windows.lookup(window)
            if not wm:
                return
            panel_manager = wm.panel_manager
            if not panel_manager:
                return
            if panel_manager.is_panel_open(PanelName.Diagnostics):
                sublime.set_timeout_async(wm.update_diagnostics_panel_async)
            elif panel_manager.is_panel_open(PanelName.Log):
                sublime.set_timeout(lambda: panel_manager.update_log_panel(scroll_to_selection=True))