fix: schema index signature comment format location

Signed-off-by: peefy <[email protected]>
Peefy · Aug 20, 2024 · 3260726 · 3260726
1 parent 24dafd8
commit 3260726
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/kclvm/ast_pretty/src/node.rs b/kclvm/ast_pretty/src/node.rs
@@ -222,6 +222,7 @@ impl<'p, 'ctx> MutSelfTypedResultWalker<'ctx> for Printer<'p> {
         }
         if let Some(index_signature) = &schema_stmt.index_signature {
             self.fill("");
+            self.write_ast_comments(index_signature);
             self.write_token(TokenKind::OpenDelim(DelimToken::Bracket));
             if index_signature.node.any_other {
                 self.write_token(TokenKind::DotDotDot);

diff --git a/kclvm/ast_pretty/src/test_data/comment.input b/kclvm/ast_pretty/src/test_data/comment.input
@@ -37,3 +37,9 @@ appConfiguration = AppConfiguration {
     overQuota: True
 }
 # Comment Nine
+
+schema Foo:
+    # Comment for index signature
+    [k: str]: int
+    # Comment for `x` field
+    x: int
diff --git a/kclvm/ast_pretty/src/test_data/comment.output b/kclvm/ast_pretty/src/test_data/comment.output
@@ -38,3 +38,9 @@ appConfiguration = AppConfiguration {
     overQuota: True
 }
 # Comment Nine
+schema Foo:
+    # Comment for index signature
+    [k: str]: int
+    # Comment for `x` field
+    x: int
+