156 lines
4.6 KiB
Python
156 lines
4.6 KiB
Python
from logging import DEBUG, basicConfig, getLogger
|
|
from pathlib import Path
|
|
from typing import Any
|
|
from lsprotocol.types import (
|
|
TEXT_DOCUMENT_DID_CHANGE,
|
|
TEXT_DOCUMENT_DID_CLOSE,
|
|
TEXT_DOCUMENT_DID_OPEN,
|
|
INITIALIZE,
|
|
TEXT_DOCUMENT_DID_SAVE,
|
|
TEXT_DOCUMENT_DOCUMENT_SYMBOL,
|
|
TEXT_DOCUMENT_INLAY_HINT,
|
|
Diagnostic,
|
|
DiagnosticSeverity,
|
|
DidChangeTextDocumentParams,
|
|
DidCloseTextDocumentParams,
|
|
DidOpenTextDocumentParams,
|
|
DocumentSymbol,
|
|
DocumentSymbolParams,
|
|
InitializeParams,
|
|
InlayHint,
|
|
InlayHintKind,
|
|
InlayHintParams,
|
|
NotebookDocumentSyncOptions,
|
|
PublishDiagnosticsParams,
|
|
TextDocumentSyncKind,
|
|
)
|
|
from pygls.lsp.server import LanguageServer
|
|
|
|
|
|
from skillls.parser import SkillParser
|
|
from skillls.types import URI, Node
|
|
|
|
basicConfig(
|
|
filename="skillls.log",
|
|
filemode="w",
|
|
level=DEBUG,
|
|
format="%(asctime)s [%(levelname)s]: %(message)s",
|
|
)
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
class SkillLanguageServer(LanguageServer):
|
|
ws_files: set[URI]
|
|
opened_files: set[URI]
|
|
scopes: dict[URI, list[Node]]
|
|
diagnostics: dict[URI, list[Diagnostic]]
|
|
|
|
def __init__(
|
|
self,
|
|
name: str,
|
|
version: str,
|
|
text_document_sync_kind: TextDocumentSyncKind = TextDocumentSyncKind.Incremental,
|
|
notebook_document_sync: NotebookDocumentSyncOptions | None = None,
|
|
):
|
|
super().__init__(name, version, text_document_sync_kind, notebook_document_sync)
|
|
self.ws_files = set()
|
|
self.opened_files = set()
|
|
self.scopes: dict[URI, list[DocumentSymbol]] = {}
|
|
self.diagnostics: dict[URI, list[Diagnostic]] = {}
|
|
self.parser = SkillParser()
|
|
|
|
def update_diagnostics(self) -> None:
|
|
for uri in self.opened_files:
|
|
diags = self.diagnostics.get(uri, [])
|
|
|
|
self.text_document_publish_diagnostics(
|
|
PublishDiagnosticsParams(
|
|
uri=uri,
|
|
version=self.workspace.get_text_document(uri).version,
|
|
diagnostics=diags,
|
|
)
|
|
)
|
|
|
|
|
|
server = SkillLanguageServer("SkillLS", "0.2.0")
|
|
|
|
|
|
@server.feature(INITIALIZE)
|
|
def lsp_initialize(server: SkillLanguageServer, params: InitializeParams) -> None:
|
|
init_options: dict[str, Any] = params.initialization_options or {}
|
|
|
|
logger.info("done init")
|
|
logger.debug(init_options)
|
|
ws_dir = server.workspace.root_path
|
|
|
|
logger.debug(ws_dir)
|
|
|
|
if ws_dir:
|
|
root_dir = Path(ws_dir)
|
|
for file in (*root_dir.rglob("*.il"), *root_dir.rglob("*.ocn")):
|
|
uri = file.as_uri()
|
|
logger.debug(uri)
|
|
|
|
server.ws_files.add(uri)
|
|
try:
|
|
text_doc = server.workspace.get_text_document(uri)
|
|
symbols, diagnostics = server.parser.parse_document(text_doc)
|
|
server.scopes[uri] = symbols
|
|
server.diagnostics[uri] = diagnostics
|
|
except Exception as e:
|
|
logger.error(f"Error initializing file {uri}: {e}")
|
|
|
|
|
|
@server.feature(TEXT_DOCUMENT_DID_OPEN)
|
|
def on_open(server: SkillLanguageServer, params: DidOpenTextDocumentParams) -> None:
|
|
server.opened_files.add(params.text_document.uri)
|
|
|
|
server.update_diagnostics()
|
|
|
|
|
|
@server.feature(TEXT_DOCUMENT_DID_CLOSE)
|
|
def on_close(server: SkillLanguageServer, params: DidCloseTextDocumentParams) -> None:
|
|
server.opened_files.remove(params.text_document.uri)
|
|
|
|
|
|
@server.feature(TEXT_DOCUMENT_DID_CHANGE)
|
|
@server.feature(TEXT_DOCUMENT_DID_SAVE)
|
|
def on_change(server: SkillLanguageServer, params: DidChangeTextDocumentParams) -> None:
|
|
try:
|
|
text_doc = server.workspace.get_text_document(params.text_document.uri)
|
|
symbols, diagnostics = server.parser.parse_document(text_doc)
|
|
server.scopes[params.text_document.uri] = symbols
|
|
server.diagnostics[params.text_document.uri] = diagnostics
|
|
except Exception as e:
|
|
logger.error(f"Error changing file {params.text_document.uri}: {e}")
|
|
|
|
server.update_diagnostics()
|
|
|
|
|
|
@server.feature(TEXT_DOCUMENT_INLAY_HINT)
|
|
def on_inlay(server: SkillLanguageServer, params: InlayHintParams) -> list[InlayHint]:
|
|
hints: list[InlayHint] = []
|
|
uri = params.text_document.uri
|
|
for symbol in server.scopes.get(uri, []):
|
|
hints.append(
|
|
InlayHint(
|
|
label=symbol.name,
|
|
kind=InlayHintKind.Type,
|
|
padding_left=True,
|
|
position=symbol.range.end,
|
|
)
|
|
)
|
|
|
|
return hints
|
|
|
|
|
|
@server.feature(TEXT_DOCUMENT_DOCUMENT_SYMBOL)
|
|
def on_symbols(
|
|
server: SkillLanguageServer, params: DocumentSymbolParams
|
|
) -> list[DocumentSymbol] | None:
|
|
return server.scopes[params.text_document.uri]
|
|
|
|
|
|
def main():
|
|
server.start_io()
|