fix: format behavior on line break between import stmt and other stmts

Signed-off-by: xiarui.xr <[email protected]>

kclvm/ast_pretty/src/node.rs

@@ -904,8 +904,18 @@ impl<'p> Printer<'p> {
     }
 
     pub fn stmts(&mut self, stmts: &[ast::NodeRef<ast::Stmt>]) {
+        let mut prev_stmt: Option<ast::Stmt> = None;
         for stmt in stmts {
+            let import_stmt_alter = match (prev_stmt.as_ref(), stmt.as_ref().node.to_owned()) {
+                (Some(ast::Stmt::Import(_)), ast::Stmt::Import(_)) => false,
+                (Some(ast::Stmt::Import(_)), _) => true,
+                _ => false,
+            };
+            if import_stmt_alter {
+                self.write_newline();
+            }
             self.stmt(stmt);
+            prev_stmt = Some(stmt.node.to_owned());
         }
     }
 }

kclvm/tools/src/format/test_data/format_data/breakline.golden

@@ -1,4 +1,5 @@
 import math
+
 schema Base:
     name: str
 

kclvm/tools/src/format/test_data/format_data/codelayout.golden

@@ -1,4 +1,5 @@
 import math as alias_math
+
 schema Person(Base):
     # inline comment
     name: str

kclvm/tools/src/format/test_data/format_data/different_stmts_line_breaks.golden

@@ -4,5 +4,4 @@ schema A:
     name: str = a.name
 
 A {}
-
 # Break one blank line between different statements e.g., import, schema and expression statements.

kclvm/tools/src/format/test_data/format_data/different_stmts_line_breaks.input

@@ -2,5 +2,4 @@ import a
 schema A:
     name: str = a.name
 A{}
-
 # Break one blank line between different statements e.g., import, schema and expression statements.

kclvm/tools/src/format/test_data/format_data/import.golden

@@ -3,4 +3,5 @@ import b
 import c
 import d
 import e as e
+
 a = 1

kclvm/tools/src/format/test_data/format_data/import_only.golden

@@ -0,0 +1,2 @@
+import math
+import abc

kclvm/tools/src/format/test_data/format_data/import_only.input

@@ -0,0 +1,2 @@
+import math
+import abc

kclvm/tools/src/format/test_data/format_data/layout_import_stmt.golden

@@ -0,0 +1,6 @@
+import a.b.c
+import d
+
+schema A:
+    name: str
+

kclvm/tools/src/format/test_data/format_data/layout_import_stmt.input

@@ -0,0 +1,4 @@
+import a.b.c
+import d
+schema A:
+    name: str

kclvm/tools/src/format/test_data/format_data/schema.golden

@@ -1,4 +1,5 @@
 import math
+
 mixin XXMixin:
     nameVar: str
 

kclvm/tools/src/format/tests.rs

@@ -5,7 +5,7 @@ use walkdir::WalkDir;
 
 const FILE_INPUT_SUFFIX: &str = ".input";
 const FILE_OUTPUT_SUFFIX: &str = ".golden";
-const TEST_CASES: &[&str; 19] = &[
+const TEST_CASES: &[&str; 22] = &[
     "assert",
     "check",
     "blankline",
@@ -16,6 +16,7 @@ const TEST_CASES: &[&str; 19] = &[
     "comp_for",
     "empty",
     "import",
+    "import_only",
     "indent",
     "inline_comment",
     "lambda",
@@ -25,7 +26,8 @@ const TEST_CASES: &[&str; 19] = &[
     "type_alias",
     "unary",
     "union_types",
-    // "different_stmts_line_breaks",
+    "layout_import_stmt",
+    "different_stmts_line_breaks",
     // "list_dict_schema_expr",
 ];