feat(langserver): support for folding WHILE and TRY/EXCEPT/FINALLY statements

Author: d-biehl

.vscode/launch.json

@@ -167,6 +167,7 @@
 			"args": [
 				"--full-trace",
 				"--tb=native",
+				"--regtest2-reset",
 				//"--regtest-reset",
 				//"tests/robotcode/language_server/robotframework/parts/test_foldingrange.py"
 				//"tests/robotcode/language_server/robotframework/parts/test_references.py"
@@ -175,7 +176,8 @@
 				//"tests/robotcode/utils/test_dataclasses.py::test_really_complex_data"
 				//"tests/robotcode/language_server/robotframework/parts/test_foldingrange.py"
 				//"tests/robotcode/language_server/robotframework/parts/test_document_symbols.py",
-				"./tests/robotcode/language_server/robotframework/parts/test_semantic_tokens.py "
+				// "./tests/robotcode/language_server/robotframework/parts/test_semantic_tokens.py "
+				"tests/robotcode/language_server/robotframework/parts/test_foldingrange.py::test[False-foldingrange.robot-123-001-try end]"
 			],
 			"console": "integratedTerminal",
 			"justMyCode": false,

hatch.toml

@@ -37,7 +37,7 @@ pre-install-commands = ["install-packages"]
 cov = "pytest --cov-report=term-missing --cov-config=pyproject.toml --cov=robotcode --cov=tests {args}"
 no-cov = "cov --no-cov {args}"
 test = "pytest {args}"
-test-reset = "test --regtest2-reset"
+test-reset = "pytest --regtest2-reset {args}"
 install-bundled-editable = "python ./scripts/install_bundled_editable.py"
 create-json-schema = "python ./scripts/create_robot_toml_json_schema.py"
 generate-rf-options = "python ./scripts/generate_rf_options.py"

package.json

@@ -607,7 +607,7 @@
         },
         "extensions": [
           ".robotrepl",
-          ".robotsript"
+          ".robotscript"
         ],
         "configuration": "./language-configuration.json"
       },

packages/language_server/src/robotcode/language_server/robotframework/parts/folding_range.py

@@ -10,13 +10,17 @@
 from robotcode.core.lsp.types import FoldingRange
 from robotcode.core.text_document import TextDocument
 from robotcode.core.utils.logging import LoggingDescriptor
+from robotcode.robot.utils import get_robot_version
 from robotcode.robot.utils.visitor import Visitor
 
 from .protocol_part import RobotLanguageServerProtocolPart
 
 if TYPE_CHECKING:
     from ..protocol import RobotLanguageServerProtocol
 
+if get_robot_version() >= (5, 0):
+    from robot.parsing.model.blocks import Try, While
+
 
 class _Visitor(Visitor):
     def __init__(self, parent: RobotFoldingRangeProtocolPart) -> None:
@@ -117,6 +121,19 @@ def visit_If(self, node: If) -> None:  # noqa: N802
         if node.type == "IF":
             self.current_if.remove(node)
 
+    if get_robot_version() >= (5, 0):
+
+        def visit_While(self, node: While) -> None:  # noqa: N802
+            self.__append(node, kind="while")
+            self.generic_visit(node)
+
+        def visit_Try(self, node: Try) -> None:  # noqa: N802
+            self.__append(node, kind="try", end_node=node.body[-1] if node.body else None)
+            self.generic_visit(node)
+
+
+#
+
 
 class RobotFoldingRangeProtocolPart(RobotLanguageServerProtocolPart):
     _logger = LoggingDescriptor()

tests/robotcode/language_server/robotframework/parts/_regtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-082-001-while_start].out

@@ -0,0 +1,5 @@
+data: !GeneratedTestData
+  character: 1
+  line: 82
+  name: while start
+result: []

tests/robotcode/language_server/robotframework/parts/_regtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-086-001-while_end].out

@@ -0,0 +1,5 @@
+data: !GeneratedTestData
+  character: 1
+  line: 86
+  name: while end
+result: []

tests/robotcode/language_server/robotframework/parts/_regtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-093-001-try_start].out

@@ -0,0 +1,5 @@
+data: !GeneratedTestData
+  character: 1
+  line: 93
+  name: try start
+result: []

tests/robotcode/language_server/robotframework/parts/_regtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-096-001-except_start].out

@@ -0,0 +1,5 @@
+data: !GeneratedTestData
+  character: 1
+  line: 96
+  name: except start
+result: []

tests/robotcode/language_server/robotframework/parts/_regtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-099-001-try_end].out

@@ -0,0 +1,5 @@
+data: !GeneratedTestData
+  character: 1
+  line: 99
+  name: try end
+result: []

tests/robotcode/language_server/robotframework/parts/_regtest_outputs/rf41/test_foldingrange.test[False-foldingrange.robot-102-001-try_start].out

@@ -0,0 +1,5 @@
+data: !GeneratedTestData
+  character: 1
+  line: 102
+  name: try start
+result: []