fix: support pygls v2 LanguageServer import

Author: nisedo

slither_lsp/app/logging/lsp_handler.py

@@ -1,5 +1,8 @@
 from logging import Handler, LogRecord, FATAL, ERROR, WARNING, INFO, DEBUG, NOTSET
-from pygls.server import LanguageServer
+try:
+    from pygls.server import LanguageServer
+except ImportError:
+    from pygls.lsp.server import LanguageServer
 from lsprotocol.types import MessageType
 
 _level_to_type = {

slither_lsp/app/request_handlers/analysis/get_detector_list.py

@@ -1,6 +1,9 @@
 # pylint: disable=unused-argument
 
-from pygls.server import LanguageServer
+try:
+    from pygls.server import LanguageServer
+except ImportError:
+    from pygls.lsp.server import LanguageServer
 from slither.__main__ import get_detectors_and_printers, output_detectors_json
 
 

slither_lsp/app/request_handlers/analysis/get_version.py

@@ -2,7 +2,10 @@
 
 from importlib.metadata import version as pkg_version
 
-from pygls.server import LanguageServer
+try:
+    from pygls.server import LanguageServer
+except ImportError:
+    from pygls.lsp.server import LanguageServer
 
 
 def get_version(ls: LanguageServer, params):

slither_lsp/app/request_handlers/compilation/get_command_line_args.py

@@ -3,7 +3,10 @@
 from argparse import ArgumentParser
 
 from crytic_compile.cryticparser.cryticparser import init as crytic_parser_init
-from pygls.server import LanguageServer
+try:
+    from pygls.server import LanguageServer
+except ImportError:
+    from pygls.lsp.server import LanguageServer
 
 
 def get_command_line_args(ls: LanguageServer, params):

slither_lsp/app/slither_server.py

@@ -12,7 +12,10 @@
 from crytic_compile.crytic_compile import CryticCompile
 from pygls.lsp import METHOD_TO_OPTIONS
 from pygls.protocol import LanguageServerProtocol
-from pygls.server import LanguageServer
+try:
+    from pygls.server import LanguageServer
+except ImportError:
+    from pygls.lsp.server import LanguageServer
 from slither import Slither
 from slither.__main__ import (
     _process as process_detectors_and_printers,