Improved line spacing for comments

Jaseci-Labs · Sep 9, 2024 · 54b3424 · 54b3424
1 parent c5e40e5
commit 54b3424
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/jaclang/compiler/passes/tool/jac_formatter_pass.py b/jaclang/compiler/passes/tool/jac_formatter_pass.py
@@ -315,7 +315,16 @@ def exit_sub_node_list(self, node: ast.SubNodeList) -> None:
                         and isinstance(prev_token.kid[-1], ast.SubNodeList)
                         and isinstance(prev_token.kid[-1].kid[-1], ast.CommentToken)
                     ):
-                        self.emit(node, "")
+                        if (
+                            prev_token
+                            and stmt.loc.first_line - prev_token.kid[-1].kid[-1].line_no
+                            > 1
+                        ):
+                            self.indent_level -= 1
+                            self.emit_ln(node, "")
+                            self.indent_level += 1
+                        else:
+                            self.emit(node, "")
                     else:
                         self.indent_level -= 1
                         self.emit_ln(node, "")

diff --git a/jaclang/compiler/passes/tool/tests/fixtures/general_format_checks/line_spacing.jac b/jaclang/compiler/passes/tool/tests/fixtures/general_format_checks/line_spacing.jac
@@ -3,6 +3,7 @@ import:py math;
 glob RAD = 5;
 glob DIA = 10;
 
+# this comment is for walker
 walker decorator_walk {
     can hash(func: Any) {
         can inner(a: Any) {
@@ -36,12 +37,16 @@ walker decorator_walk {
     }
 
     # Entry point for the walker
+
     can start with entry {
         # Apply decorators to greeter
         decorated_greeter = hash(exclaim(tilde(greeter)));
 
         # Call the decorated greeter function
         decorated_greeter("World");
+
+        # this is another comment
+
     }
 }