update lsp_server

server/src/lsp_server.py

@@ -54,7 +54,7 @@ from tclint.violations import Violation
 from plugins.poco_plugin import commands
 from tools import checks, parser
 from tools.semantic_tokens import _Highlighter, TOKEN_TYPES, TokenModifier
-from tools.completion_items import _Completion
+from tools.completion_items import completion
 
 DIAGNOSTIC_SOURCE = "nx-post-support"
 
@@ -62,7 +62,7 @@ DIAGNOSTIC_SOURCE = "nx-post-support"
 class TclLanguageServer(server.LanguageServer):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.parser = parser.CustomParser()  # Parser()
+        self.parser = Parser()
         for command in commands:
             self.parser._commands.update(command)
         self.diagnostics = {}
@@ -215,10 +215,10 @@ def did_change(params: lsp.DidChangeTextDocumentParams) -> None:
     """LSP handler for textDocument/didChange request"""
     document = LSP_SERVER.workspace.get_text_document(params.text_document.uri)
     LSP_SERVER.compute_diagnostics(document)
-    ci = _Completion()
+    completion.reset()
     tree = LSP_SERVER.parser.parse(document.source)
-    tree.accept(ci, recurse=True)
-    LSP_SERVER._custom_functions = ci.custom_functions
+    tree.accept(completion, recurse=True)
+    log_to_output(tree.pretty(2))
 
 
 @LSP_SERVER.feature(
@@ -254,7 +254,7 @@ def on_completion(params: lsp.CompletionParams) -> list[lsp.CompletionItem]:
         standard_items.tcl_keyword_list
         + standard_items.nx_procs
         + standard_items.nx_variables
-        + LSP_SERVER._custom_functions
+        + completion.custom_functions
     )
     return lsp.CompletionList(is_incomplete=False, items=items)