Add support for locally scoped constants

Author: withersail

gdtoolkit/formatter/function_statement.py

@@ -15,6 +15,7 @@ def format_func_statement(statement: Tree, context: Context) -> Outcome:
     handlers = {
         "pass_stmt": partial(format_simple_statement, "pass"),
         "func_var_stmt": format_var_statement,
+        "const_stmt": _format_const_statement,
         "expr_stmt": _format_expr_statement,
         "return_stmt": _format_return_statement,
         "break_stmt": partial(format_simple_statement, "break"),
@@ -29,6 +30,21 @@ def format_func_statement(statement: Tree, context: Context) -> Outcome:
     return handlers[statement.data](statement, context)
 
 
+def _format_const_statement(statement: Tree, context: Context) -> Outcome:
+    if len(statement.children) == 4:
+        prefix = f"const {statement.children[1].value} = "
+    elif len(statement.children) == 5:
+        prefix = f"const {statement.children[1].value} := "
+    elif len(statement.children) == 6:
+        prefix = (
+            f"const {statement.children[1].value}: {statement.children[3].value} = "
+        )
+    expression_context = ExpressionContext(
+        prefix, statement.line, "", statement.end_line
+    )
+    return format_expression(statement.children[-1], expression_context, context)
+
+
 def _format_expr_statement(statement: Tree, context: Context) -> Outcome:
     expr = statement.children[0]
     expression_context = ExpressionContext("", statement.line, "", statement.end_line)

gdtoolkit/parser/gdscript.lark

@@ -78,6 +78,7 @@ _simple_func_stmt: single_func_stmt (";" single_func_stmt)* [";"]
                  | break_stmt
                  | continue_stmt
                  | expr_stmt
+                 | const_stmt
 ?compound_func_stmt: if_stmt
                    | while_stmt
                    | for_stmt